src/cypherpunks.ru/nncp/sp.go

   1 /*
   2 NNCP -- Node to Node copy, utilities for store-and-forward data exchange
   3 Copyright (C) 2016-2019 Sergey Matveev <stargrave@stargrave.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation, either version 3 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 package nncp
  20
  21 import (
  22         "bytes"
  23         "crypto/subtle"
  24         "errors"
  25         "io"
  26         "net"
  27         "os"
  28         "path/filepath"
  29         "sort"
  30         "strconv"
  31         "sync"
  32         "time"
  33
  34         "github.com/davecgh/go-xdr/xdr2"
  35         "github.com/flynn/noise"
  36 )
  37
  38 const (
  39         MaxSPSize       = 1<<16 - 256
  40         PartSuffix      = ".part"
  41         DefaultDeadline = 10
  42 )
  43
  44 var (
  45         MagicNNCPLv1 [8]byte = [8]byte{'N', 'N', 'C', 'P', 'S', 0, 0, 1}
  46
  47         SPHeadOverhead    int
  48         SPInfoOverhead    int
  49         SPFreqOverhead    int
  50         SPFileOverhead    int
  51         SPHaltMarshalized []byte
  52
  53         NoiseCipherSuite noise.CipherSuite = noise.NewCipherSuite(
  54                 noise.DH25519,
  55                 noise.CipherChaChaPoly,
  56                 noise.HashBLAKE2b,
  57         )
  58
  59         spWorkersGroup sync.WaitGroup
  60 )
  61
  62 type SPType uint8
  63
  64 const (
  65         SPTypeInfo SPType = iota
  66         SPTypeFreq SPType = iota
  67         SPTypeFile SPType = iota
  68         SPTypeDone SPType = iota
  69         SPTypeHalt SPType = iota
  70 )
  71
  72 type SPHead struct {
  73         Type SPType
  74 }
  75
  76 type SPInfo struct {
  77         Nice uint8
  78         Size uint64
  79         Hash *[32]byte
  80 }
  81
  82 type SPFreq struct {
  83         Hash   *[32]byte
  84         Offset uint64
  85 }
  86
  87 type SPFile struct {
  88         Hash    *[32]byte
  89         Offset  uint64
  90         Payload []byte
  91 }
  92
  93 type SPDone struct {
  94         Hash *[32]byte
  95 }
  96
  97 type SPRaw struct {
  98         Magic   [8]byte
  99         Payload []byte
 100 }
 101
 102 type FreqWithNice struct {
 103         freq *SPFreq
 104         nice uint8
 105 }
 106
 107 type ConnDeadlined interface {
 108         io.ReadWriter
 109         SetReadDeadline(t time.Time) error
 110         SetWriteDeadline(t time.Time) error
 111 }
 112
 113 func init() {
 114         var buf bytes.Buffer
 115         spHead := SPHead{Type: SPTypeHalt}
 116         if _, err := xdr.Marshal(&buf, spHead); err != nil {
 117                 panic(err)
 118         }
 119         copy(SPHaltMarshalized, buf.Bytes())
 120         SPHeadOverhead = buf.Len()
 121         buf.Reset()
 122
 123         spInfo := SPInfo{Nice: 123, Size: 123, Hash: new([32]byte)}
 124         if _, err := xdr.Marshal(&buf, spInfo); err != nil {
 125                 panic(err)
 126         }
 127         SPInfoOverhead = buf.Len()
 128         buf.Reset()
 129
 130         spFreq := SPFreq{Hash: new([32]byte), Offset: 123}
 131         if _, err := xdr.Marshal(&buf, spFreq); err != nil {
 132                 panic(err)
 133         }
 134         SPFreqOverhead = buf.Len()
 135         buf.Reset()
 136
 137         spFile := SPFile{Hash: new([32]byte), Offset: 123}
 138         if _, err := xdr.Marshal(&buf, spFile); err != nil {
 139                 panic(err)
 140         }
 141         SPFileOverhead = buf.Len()
 142 }
 143
 144 func MarshalSP(typ SPType, sp interface{}) []byte {
 145         var buf bytes.Buffer
 146         var err error
 147         if _, err = xdr.Marshal(&buf, SPHead{typ}); err != nil {
 148                 panic(err)
 149         }
 150         if _, err = xdr.Marshal(&buf, sp); err != nil {
 151                 panic(err)
 152         }
 153         return buf.Bytes()
 154 }
 155
 156 func payloadsSplit(payloads [][]byte) [][]byte {
 157         var outbounds [][]byte
 158         outbound := make([]byte, 0, MaxSPSize)
 159         for i, payload := range payloads {
 160                 outbound = append(outbound, payload...)
 161                 if i+1 < len(payloads) && len(outbound)+len(payloads[i+1]) > MaxSPSize {
 162                         outbounds = append(outbounds, outbound)
 163                         outbound = make([]byte, 0, MaxSPSize)
 164                 }
 165         }
 166         if len(outbound) > 0 {
 167                 outbounds = append(outbounds, outbound)
 168         }
 169         return outbounds
 170 }
 171
 172 type SPState struct {
 173         Ctx            *Ctx
 174         Node           *Node
 175         Nice           uint8
 176         onlineDeadline uint
 177         maxOnlineTime  uint
 178         hs             *noise.HandshakeState
 179         csOur          *noise.CipherState
 180         csTheir        *noise.CipherState
 181         payloads       chan []byte
 182         infosTheir     map[[32]byte]*SPInfo
 183         infosOurSeen   map[[32]byte]uint8
 184         queueTheir     []*FreqWithNice
 185         wg             sync.WaitGroup
 186         RxBytes        int64
 187         RxLastSeen     time.Time
 188         TxBytes        int64
 189         TxLastSeen     time.Time
 190         started        time.Time
 191         Duration       time.Duration
 192         RxSpeed        int64
 193         TxSpeed        int64
 194         rxLock         *os.File
 195         txLock         *os.File
 196         xxOnly         TRxTx
 197         rxRate         int
 198         txRate         int
 199         isDead         bool
 200         listOnly       bool
 201         onlyPkts       map[[32]byte]bool
 202         sync.RWMutex
 203 }
 204
 205 func (state *SPState) NotAlive() bool {
 206         if state.isDead {
 207                 return true
 208         }
 209         now := time.Now()
 210         if state.maxOnlineTime > 0 && state.started.Add(time.Duration(state.maxOnlineTime)*time.Second).Before(now) {
 211                 return true
 212         }
 213         return uint(now.Sub(state.RxLastSeen).Seconds()) >= state.onlineDeadline &&
 214                 uint(now.Sub(state.TxLastSeen).Seconds()) >= state.onlineDeadline
 215 }
 216
 217 func (state *SPState) dirUnlock() {
 218         state.Ctx.UnlockDir(state.rxLock)
 219         state.Ctx.UnlockDir(state.txLock)
 220 }
 221
 222 func (state *SPState) WriteSP(dst io.Writer, payload []byte) error {
 223         n, err := xdr.Marshal(dst, SPRaw{Magic: MagicNNCPLv1, Payload: payload})
 224         if err == nil {
 225                 state.TxLastSeen = time.Now()
 226                 state.TxBytes += int64(n)
 227         }
 228         return err
 229 }
 230
 231 func (state *SPState) ReadSP(src io.Reader) ([]byte, error) {
 232         var sp SPRaw
 233         n, err := xdr.UnmarshalLimited(src, &sp, 1<<17)
 234         if err != nil {
 235                 return nil, err
 236         }
 237         state.RxLastSeen = time.Now()
 238         state.RxBytes += int64(n)
 239         if sp.Magic != MagicNNCPLv1 {
 240                 return nil, BadMagic
 241         }
 242         return sp.Payload, nil
 243 }
 244
 245 func (ctx *Ctx) infosOur(nodeId *NodeId, nice uint8, seen *map[[32]byte]uint8) [][]byte {
 246         var infos []*SPInfo
 247         var totalSize int64
 248         for job := range ctx.Jobs(nodeId, TTx) {
 249                 job.Fd.Close()
 250                 if job.PktEnc.Nice > nice {
 251                         continue
 252                 }
 253                 if _, known := (*seen)[*job.HshValue]; known {
 254                         continue
 255                 }
 256                 totalSize += job.Size
 257                 infos = append(infos, &SPInfo{
 258                         Nice: job.PktEnc.Nice,
 259                         Size: uint64(job.Size),
 260                         Hash: job.HshValue,
 261                 })
 262                 (*seen)[*job.HshValue] = job.PktEnc.Nice
 263         }
 264         sort.Sort(ByNice(infos))
 265         var payloads [][]byte
 266         for _, info := range infos {
 267                 payloads = append(payloads, MarshalSP(SPTypeInfo, info))
 268                 ctx.LogD("sp-info-our", SDS{
 269                         "node": nodeId,
 270                         "name": ToBase32(info.Hash[:]),
 271                         "size": strconv.FormatInt(int64(info.Size), 10),
 272                 }, "")
 273         }
 274         if totalSize > 0 {
 275                 ctx.LogI("sp-infos", SDS{
 276                         "xx":   string(TTx),
 277                         "node": nodeId,
 278                         "pkts": strconv.Itoa(len(payloads)),
 279                         "size": strconv.FormatInt(totalSize, 10),
 280                 }, "")
 281         }
 282         return payloadsSplit(payloads)
 283 }
 284
 285 func (state *SPState) StartI(conn ConnDeadlined) error {
 286         nodeId := state.Node.Id
 287         err := state.Ctx.ensureRxDir(nodeId)
 288         if err != nil {
 289                 return err
 290         }
 291         var rxLock *os.File
 292         if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TRx) {
 293                 rxLock, err = state.Ctx.LockDir(nodeId, TRx)
 294                 if err != nil {
 295                         return err
 296                 }
 297         }
 298         var txLock *os.File
 299         if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TTx) {
 300                 txLock, err = state.Ctx.LockDir(nodeId, TTx)
 301                 if err != nil {
 302                         return err
 303                 }
 304         }
 305         started := time.Now()
 306         conf := noise.Config{
 307                 CipherSuite: NoiseCipherSuite,
 308                 Pattern:     noise.HandshakeIK,
 309                 Initiator:   true,
 310                 StaticKeypair: noise.DHKey{
 311                         Private: state.Ctx.Self.NoisePrv[:],
 312                         Public:  state.Ctx.Self.NoisePub[:],
 313                 },
 314                 PeerStatic: state.Node.NoisePub[:],
 315         }
 316         hs, err := noise.NewHandshakeState(conf)
 317         if err != nil {
 318                 return err
 319         }
 320         state.hs = hs
 321         state.payloads = make(chan []byte)
 322         state.infosTheir = make(map[[32]byte]*SPInfo)
 323         state.infosOurSeen = make(map[[32]byte]uint8)
 324         state.started = started
 325         state.rxLock = rxLock
 326         state.txLock = txLock
 327
 328         var infosPayloads [][]byte
 329         if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TTx) {
 330                 infosPayloads = state.Ctx.infosOur(nodeId, state.Nice, &state.infosOurSeen)
 331         }
 332         var firstPayload []byte
 333         if len(infosPayloads) > 0 {
 334                 firstPayload = infosPayloads[0]
 335         }
 336         // Pad first payload, to hide actual number of existing files
 337         for i := 0; i < (MaxSPSize-len(firstPayload))/SPHeadOverhead; i++ {
 338                 firstPayload = append(firstPayload, SPHaltMarshalized...)
 339         }
 340
 341         var buf []byte
 342         var payload []byte
 343         buf, _, _, err = state.hs.WriteMessage(nil, firstPayload)
 344         if err != nil {
 345                 state.dirUnlock()
 346                 return err
 347         }
 348         sds := SDS{"node": nodeId, "nice": strconv.Itoa(int(state.Nice))}
 349         state.Ctx.LogD("sp-start", sds, "sending first message")
 350         conn.SetWriteDeadline(time.Now().Add(DefaultDeadline * time.Second))
 351         if err = state.WriteSP(conn, buf); err != nil {
 352                 state.Ctx.LogE("sp-start", SdsAdd(sds, SDS{"err": err}), "")
 353                 state.dirUnlock()
 354                 return err
 355         }
 356         state.Ctx.LogD("sp-start", sds, "waiting for first message")
 357         conn.SetReadDeadline(time.Now().Add(DefaultDeadline * time.Second))
 358         if buf, err = state.ReadSP(conn); err != nil {
 359                 state.Ctx.LogE("sp-start", SdsAdd(sds, SDS{"err": err}), "")
 360                 state.dirUnlock()
 361                 return err
 362         }
 363         payload, state.csOur, state.csTheir, err = state.hs.ReadMessage(nil, buf)
 364         if err != nil {
 365                 state.Ctx.LogE("sp-start", SdsAdd(sds, SDS{"err": err}), "")
 366                 state.dirUnlock()
 367                 return err
 368         }
 369         state.Ctx.LogD("sp-start", sds, "starting workers")
 370         err = state.StartWorkers(conn, infosPayloads, payload)
 371         if err != nil {
 372                 state.Ctx.LogE("sp-start", SdsAdd(sds, SDS{"err": err}), "")
 373                 state.dirUnlock()
 374                 return err
 375         }
 376         return err
 377 }
 378
 379 func (state *SPState) StartR(conn ConnDeadlined) error {
 380         started := time.Now()
 381         conf := noise.Config{
 382                 CipherSuite: NoiseCipherSuite,
 383                 Pattern:     noise.HandshakeIK,
 384                 Initiator:   false,
 385                 StaticKeypair: noise.DHKey{
 386                         Private: state.Ctx.Self.NoisePrv[:],
 387                         Public:  state.Ctx.Self.NoisePub[:],
 388                 },
 389         }
 390         hs, err := noise.NewHandshakeState(conf)
 391         if err != nil {
 392                 return err
 393         }
 394         xxOnly := TRxTx("")
 395         state.hs = hs
 396         state.payloads = make(chan []byte)
 397         state.infosOurSeen = make(map[[32]byte]uint8)
 398         state.infosTheir = make(map[[32]byte]*SPInfo)
 399         state.started = started
 400         state.xxOnly = xxOnly
 401         var buf []byte
 402         var payload []byte
 403         state.Ctx.LogD(
 404                 "sp-start",
 405                 SDS{"nice": strconv.Itoa(int(state.Nice))},
 406                 "waiting for first message",
 407         )
 408         conn.SetReadDeadline(time.Now().Add(DefaultDeadline * time.Second))
 409         if buf, err = state.ReadSP(conn); err != nil {
 410                 state.Ctx.LogE("sp-start", SDS{"err": err}, "")
 411                 return err
 412         }
 413         if payload, _, _, err = state.hs.ReadMessage(nil, buf); err != nil {
 414                 state.Ctx.LogE("sp-start", SDS{"err": err}, "")
 415                 return err
 416         }
 417
 418         var node *Node
 419         for _, node = range state.Ctx.Neigh {
 420                 if subtle.ConstantTimeCompare(state.hs.PeerStatic(), node.NoisePub[:]) == 1 {
 421                         break
 422                 }
 423         }
 424         if node == nil {
 425                 peerId := ToBase32(state.hs.PeerStatic())
 426                 state.Ctx.LogE("sp-start", SDS{"peer": peerId}, "unknown")
 427                 return errors.New("Unknown peer: " + peerId)
 428         }
 429         state.Node = node
 430         state.rxRate = node.RxRate
 431         state.txRate = node.TxRate
 432         state.onlineDeadline = node.OnlineDeadline
 433         state.maxOnlineTime = node.MaxOnlineTime
 434         sds := SDS{"node": node.Id, "nice": strconv.Itoa(int(state.Nice))}
 435
 436         if state.Ctx.ensureRxDir(node.Id); err != nil {
 437                 return err
 438         }
 439         var rxLock *os.File
 440         if xxOnly == "" || xxOnly == TRx {
 441                 rxLock, err = state.Ctx.LockDir(node.Id, TRx)
 442                 if err != nil {
 443                         return err
 444                 }
 445         }
 446         state.rxLock = rxLock
 447         var txLock *os.File
 448         if xxOnly == "" || xxOnly == TTx {
 449                 txLock, err = state.Ctx.LockDir(node.Id, TTx)
 450                 if err != nil {
 451                         return err
 452                 }
 453         }
 454         state.txLock = txLock
 455
 456         var infosPayloads [][]byte
 457         if xxOnly == "" || xxOnly == TTx {
 458                 infosPayloads = state.Ctx.infosOur(node.Id, state.Nice, &state.infosOurSeen)
 459         }
 460         var firstPayload []byte
 461         if len(infosPayloads) > 0 {
 462                 firstPayload = infosPayloads[0]
 463         }
 464         // Pad first payload, to hide actual number of existing files
 465         for i := 0; i < (MaxSPSize-len(firstPayload))/SPHeadOverhead; i++ {
 466                 firstPayload = append(firstPayload, SPHaltMarshalized...)
 467         }
 468
 469         state.Ctx.LogD("sp-start", sds, "sending first message")
 470         buf, state.csTheir, state.csOur, err = state.hs.WriteMessage(nil, firstPayload)
 471         if err != nil {
 472                 state.dirUnlock()
 473                 return err
 474         }
 475         conn.SetWriteDeadline(time.Now().Add(DefaultDeadline * time.Second))
 476         if err = state.WriteSP(conn, buf); err != nil {
 477                 state.Ctx.LogE("sp-start", SdsAdd(sds, SDS{"err": err}), "")
 478                 state.dirUnlock()
 479                 return err
 480         }
 481         state.Ctx.LogD("sp-start", sds, "starting workers")
 482         err = state.StartWorkers(conn, infosPayloads, payload)
 483         if err != nil {
 484                 state.dirUnlock()
 485                 return err
 486         }
 487         return err
 488 }
 489
 490 func (state *SPState) StartWorkers(
 491         conn ConnDeadlined,
 492         infosPayloads [][]byte,
 493         payload []byte) error {
 494         sds := SDS{"node": state.Node.Id, "nice": strconv.Itoa(int(state.Nice))}
 495         if len(infosPayloads) > 1 {
 496                 go func() {
 497                         for _, payload := range infosPayloads[1:] {
 498                                 state.Ctx.LogD(
 499                                         "sp-work",
 500                                         SdsAdd(sds, SDS{"size": strconv.Itoa(len(payload))}),
 501                                         "queuing remaining payload",
 502                                 )
 503                                 state.payloads <- payload
 504                         }
 505                 }()
 506         }
 507         state.Ctx.LogD(
 508                 "sp-work",
 509                 SdsAdd(sds, SDS{"size": strconv.Itoa(len(payload))}),
 510                 "processing first payload",
 511         )
 512         replies, err := state.ProcessSP(payload)
 513         if err != nil {
 514                 state.Ctx.LogE("sp-work", SdsAdd(sds, SDS{"err": err}), "")
 515                 return err
 516         }
 517
 518         go func() {
 519                 for _, reply := range replies {
 520                         state.Ctx.LogD(
 521                                 "sp-work",
 522                                 SdsAdd(sds, SDS{"size": strconv.Itoa(len(reply))}),
 523                                 "queuing reply",
 524                         )
 525                         state.payloads <- reply
 526                 }
 527         }()
 528
 529         if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TTx) {
 530                 go func() {
 531                         for range time.Tick(time.Second) {
 532                                 if state.NotAlive() {
 533                                         return
 534                                 }
 535                                 for _, payload := range state.Ctx.infosOur(
 536                                         state.Node.Id,
 537                                         state.Nice,
 538                                         &state.infosOurSeen,
 539                                 ) {
 540                                         state.Ctx.LogD(
 541                                                 "sp-work",
 542                                                 SdsAdd(sds, SDS{"size": strconv.Itoa(len(payload))}),
 543                                                 "queuing new info",
 544                                         )
 545                                         state.payloads <- payload
 546                                 }
 547                         }
 548                 }()
 549         }
 550
 551         state.wg.Add(1)
 552         go func() {
 553                 defer func() {
 554                         state.isDead = true
 555                         state.wg.Done()
 556                 }()
 557                 for {
 558                         if state.NotAlive() {
 559                                 return
 560                         }
 561                         var payload []byte
 562                         select {
 563                         case payload = <-state.payloads:
 564                                 state.Ctx.LogD(
 565                                         "sp-xmit",
 566                                         SdsAdd(sds, SDS{"size": strconv.Itoa(len(payload))}),
 567                                         "got payload",
 568                                 )
 569                         default:
 570                         }
 571                         if payload == nil {
 572                                 state.RLock()
 573                                 if len(state.queueTheir) == 0 {
 574                                         state.Ctx.LogD("sp-xmit", sds, "file queue is empty")
 575                                         state.RUnlock()
 576                                         time.Sleep(100 * time.Millisecond)
 577                                         continue
 578                                 }
 579                                 freq := state.queueTheir[0].freq
 580                                 state.RUnlock()
 581
 582                                 if state.txRate > 0 {
 583                                         time.Sleep(time.Second / time.Duration(state.txRate))
 584                                 }
 585
 586                                 sdsp := SdsAdd(sds, SDS{
 587                                         "xx":   string(TTx),
 588                                         "hash": ToBase32(freq.Hash[:]),
 589                                         "size": strconv.FormatInt(int64(freq.Offset), 10),
 590                                 })
 591                                 state.Ctx.LogD("sp-file", sdsp, "queueing")
 592                                 fd, err := os.Open(filepath.Join(
 593                                         state.Ctx.Spool,
 594                                         state.Node.Id.String(),
 595                                         string(TTx),
 596                                         ToBase32(freq.Hash[:]),
 597                                 ))
 598                                 if err != nil {
 599                                         state.Ctx.LogE("sp-file", SdsAdd(sdsp, SDS{"err": err}), "")
 600                                         break
 601                                 }
 602                                 fi, err := fd.Stat()
 603                                 if err != nil {
 604                                         state.Ctx.LogE("sp-file", SdsAdd(sdsp, SDS{"err": err}), "")
 605                                         break
 606                                 }
 607                                 fullSize := uint64(fi.Size())
 608                                 var buf []byte
 609                                 if freq.Offset < fullSize {
 610                                         state.Ctx.LogD("sp-file", sdsp, "seeking")
 611                                         if _, err = fd.Seek(int64(freq.Offset), io.SeekStart); err != nil {
 612                                                 state.Ctx.LogE("sp-file", SdsAdd(sdsp, SDS{"err": err}), "")
 613                                                 break
 614                                         }
 615                                         buf = make([]byte, MaxSPSize-SPHeadOverhead-SPFileOverhead)
 616                                         n, err := fd.Read(buf)
 617                                         if err != nil {
 618                                                 state.Ctx.LogE("sp-file", SdsAdd(sdsp, SDS{"err": err}), "")
 619                                                 break
 620                                         }
 621                                         buf = buf[:n]
 622                                         state.Ctx.LogD(
 623                                                 "sp-file",
 624                                                 SdsAdd(sdsp, SDS{"size": strconv.Itoa(n)}),
 625                                                 "read",
 626                                         )
 627                                 }
 628                                 fd.Close()
 629                                 payload = MarshalSP(SPTypeFile, SPFile{
 630                                         Hash:    freq.Hash,
 631                                         Offset:  freq.Offset,
 632                                         Payload: buf,
 633                                 })
 634                                 ourSize := freq.Offset + uint64(len(buf))
 635                                 sdsp["size"] = strconv.FormatInt(int64(ourSize), 10)
 636                                 sdsp["fullsize"] = strconv.FormatInt(int64(fullSize), 10)
 637                                 state.Ctx.LogP("sp-file", sdsp, "")
 638                                 state.Lock()
 639                                 if len(state.queueTheir) > 0 && *state.queueTheir[0].freq.Hash == *freq.Hash {
 640                                         if ourSize == fullSize {
 641                                                 state.Ctx.LogD("sp-file", sdsp, "finished")
 642                                                 if len(state.queueTheir) > 1 {
 643                                                         state.queueTheir = state.queueTheir[1:]
 644                                                 } else {
 645                                                         state.queueTheir = state.queueTheir[:0]
 646                                                 }
 647                                         } else {
 648                                                 state.queueTheir[0].freq.Offset += uint64(len(buf))
 649                                         }
 650                                 } else {
 651                                         state.Ctx.LogD("sp-file", sdsp, "queue disappeared")
 652                                 }
 653                                 state.Unlock()
 654                         }
 655                         state.Ctx.LogD(
 656                                 "sp-xmit",
 657                                 SdsAdd(sds, SDS{"size": strconv.Itoa(len(payload))}),
 658                                 "sending",
 659                         )
 660                         conn.SetWriteDeadline(time.Now().Add(DefaultDeadline * time.Second))
 661                         if err := state.WriteSP(conn, state.csOur.Encrypt(nil, nil, payload)); err != nil {
 662                                 state.Ctx.LogE("sp-xmit", SdsAdd(sds, SDS{"err": err}), "")
 663                                 break
 664                         }
 665                 }
 666         }()
 667
 668         state.wg.Add(1)
 669         go func() {
 670                 defer func() {
 671                         state.isDead = true
 672                         state.wg.Done()
 673                 }()
 674                 for {
 675                         if state.NotAlive() {
 676                                 return
 677                         }
 678                         state.Ctx.LogD("sp-recv", sds, "waiting for payload")
 679                         conn.SetReadDeadline(time.Now().Add(DefaultDeadline * time.Second))
 680                         payload, err := state.ReadSP(conn)
 681                         if err != nil {
 682                                 unmarshalErr := err.(*xdr.UnmarshalError)
 683                                 netErr, ok := unmarshalErr.Err.(net.Error)
 684                                 if ok && netErr.Timeout() {
 685                                         continue
 686                                 }
 687                                 if unmarshalErr.ErrorCode == xdr.ErrIO {
 688                                         break
 689                                 }
 690                                 state.Ctx.LogE("sp-recv", SdsAdd(sds, SDS{"err": err}), "")
 691                                 break
 692                         }
 693                         state.Ctx.LogD(
 694                                 "sp-recv",
 695                                 SdsAdd(sds, SDS{"size": strconv.Itoa(len(payload))}),
 696                                 "got payload",
 697                         )
 698                         payload, err = state.csTheir.Decrypt(nil, nil, payload)
 699                         if err != nil {
 700                                 state.Ctx.LogE("sp-recv", SdsAdd(sds, SDS{"err": err}), "")
 701                                 break
 702                         }
 703                         state.Ctx.LogD(
 704                                 "sp-recv",
 705                                 SdsAdd(sds, SDS{"size": strconv.Itoa(len(payload))}),
 706                                 "processing",
 707                         )
 708                         replies, err := state.ProcessSP(payload)
 709                         if err != nil {
 710                                 state.Ctx.LogE("sp-recv", SdsAdd(sds, SDS{"err": err}), "")
 711                                 break
 712                         }
 713                         go func() {
 714                                 for _, reply := range replies {
 715                                         state.Ctx.LogD(
 716                                                 "sp-recv",
 717                                                 SdsAdd(sds, SDS{"size": strconv.Itoa(len(reply))}),
 718                                                 "queuing reply",
 719                                         )
 720                                         state.payloads <- reply
 721                                 }
 722                         }()
 723                         if state.rxRate > 0 {
 724                                 time.Sleep(time.Second / time.Duration(state.rxRate))
 725                         }
 726                 }
 727         }()
 728
 729         return nil
 730 }
 731
 732 func (state *SPState) Wait() {
 733         state.wg.Wait()
 734         state.dirUnlock()
 735         state.Duration = time.Now().Sub(state.started)
 736         state.RxSpeed = state.RxBytes
 737         state.TxSpeed = state.TxBytes
 738         rxDuration := int64(state.RxLastSeen.Sub(state.started).Seconds())
 739         txDuration := int64(state.TxLastSeen.Sub(state.started).Seconds())
 740         if rxDuration > 0 {
 741                 state.RxSpeed = state.RxBytes / rxDuration
 742         }
 743         if txDuration > 0 {
 744                 state.TxSpeed = state.TxBytes / txDuration
 745         }
 746 }
 747
 748 func (state *SPState) ProcessSP(payload []byte) ([][]byte, error) {
 749         sds := SDS{"node": state.Node.Id, "nice": strconv.Itoa(int(state.Nice))}
 750         r := bytes.NewReader(payload)
 751         var err error
 752         var replies [][]byte
 753         var infosGot bool
 754         for r.Len() > 0 {
 755                 state.Ctx.LogD("sp-process", sds, "unmarshaling header")
 756                 var head SPHead
 757                 if _, err = xdr.Unmarshal(r, &head); err != nil {
 758                         state.Ctx.LogE("sp-process", SdsAdd(sds, SDS{"err": err}), "")
 759                         return nil, err
 760                 }
 761                 switch head.Type {
 762                 case SPTypeInfo:
 763                         infosGot = true
 764                         sdsp := SdsAdd(sds, SDS{"type": "info"})
 765                         state.Ctx.LogD("sp-process", sdsp, "unmarshaling packet")
 766                         var info SPInfo
 767                         if _, err = xdr.Unmarshal(r, &info); err != nil {
 768                                 state.Ctx.LogE("sp-process", SdsAdd(sdsp, SDS{"err": err}), "")
 769                                 return nil, err
 770                         }
 771                         sdsp = SdsAdd(sds, SDS{
 772                                 "hash": ToBase32(info.Hash[:]),
 773                                 "size": strconv.FormatInt(int64(info.Size), 10),
 774                                 "nice": strconv.Itoa(int(info.Nice)),
 775                         })
 776                         if !state.listOnly && info.Nice > state.Nice {
 777                                 state.Ctx.LogD("sp-process", sdsp, "too nice")
 778                                 continue
 779                         }
 780                         state.Ctx.LogD("sp-process", sdsp, "received")
 781                         if !state.listOnly && state.xxOnly == TTx {
 782                                 continue
 783                         }
 784                         state.Lock()
 785                         state.infosTheir[*info.Hash] = &info
 786                         state.Unlock()
 787                         state.Ctx.LogD("sp-process", sdsp, "stating part")
 788                         pktPath := filepath.Join(
 789                                 state.Ctx.Spool,
 790                                 state.Node.Id.String(),
 791                                 string(TRx),
 792                                 ToBase32(info.Hash[:]),
 793                         )
 794                         if _, err = os.Stat(pktPath); err == nil {
 795                                 state.Ctx.LogI("sp-info", sdsp, "already done")
 796                                 if !state.listOnly {
 797                                         replies = append(replies, MarshalSP(SPTypeDone, SPDone{info.Hash}))
 798                                 }
 799                                 continue
 800                         }
 801                         if _, err = os.Stat(pktPath + SeenSuffix); err == nil {
 802                                 state.Ctx.LogI("sp-info", sdsp, "already seen")
 803                                 if !state.listOnly {
 804                                         replies = append(replies, MarshalSP(SPTypeDone, SPDone{info.Hash}))
 805                                 }
 806                                 continue
 807                         }
 808                         fi, err := os.Stat(pktPath + PartSuffix)
 809                         var offset int64
 810                         if err == nil {
 811                                 offset = fi.Size()
 812                         }
 813                         if !state.Ctx.IsEnoughSpace(int64(info.Size) - offset) {
 814                                 state.Ctx.LogI("sp-info", sdsp, "not enough space")
 815                                 continue
 816                         }
 817                         state.Ctx.LogI(
 818                                 "sp-info",
 819                                 SdsAdd(sdsp, SDS{"offset": strconv.FormatInt(offset, 10)}),
 820                                 "",
 821                         )
 822                         if !state.listOnly && (state.onlyPkts == nil || state.onlyPkts[*info.Hash]) {
 823                                 replies = append(replies, MarshalSP(
 824                                         SPTypeFreq,
 825                                         SPFreq{info.Hash, uint64(offset)},
 826                                 ))
 827                         }
 828                 case SPTypeFile:
 829                         sdsp := SdsAdd(sds, SDS{"type": "file"})
 830                         state.Ctx.LogD("sp-process", sdsp, "unmarshaling packet")
 831                         var file SPFile
 832                         if _, err = xdr.Unmarshal(r, &file); err != nil {
 833                                 state.Ctx.LogE("sp-process", SdsAdd(sds, SDS{
 834                                         "err":  err,
 835                                         "type": "file",
 836                                 }), "")
 837                                 return nil, err
 838                         }
 839                         sdsp["xx"] = string(TRx)
 840                         sdsp["hash"] = ToBase32(file.Hash[:])
 841                         sdsp["size"] = strconv.Itoa(len(file.Payload))
 842                         filePath := filepath.Join(
 843                                 state.Ctx.Spool,
 844                                 state.Node.Id.String(),
 845                                 string(TRx),
 846                                 ToBase32(file.Hash[:]),
 847                         )
 848                         state.Ctx.LogD("sp-file", sdsp, "opening part")
 849                         fd, err := os.OpenFile(
 850                                 filePath+PartSuffix,
 851                                 os.O_RDWR|os.O_CREATE,
 852                                 os.FileMode(0600),
 853                         )
 854                         if err != nil {
 855                                 state.Ctx.LogE("sp-file", SdsAdd(sdsp, SDS{"err": err}), "")
 856                                 return nil, err
 857                         }
 858                         state.Ctx.LogD(
 859                                 "sp-file",
 860                                 SdsAdd(sdsp, SDS{"offset": strconv.FormatInt(int64(file.Offset), 10)}),
 861                                 "seeking",
 862                         )
 863                         if _, err = fd.Seek(int64(file.Offset), io.SeekStart); err != nil {
 864                                 state.Ctx.LogE("sp-file", SdsAdd(sdsp, SDS{"err": err}), "")
 865                                 fd.Close()
 866                                 return nil, err
 867                         }
 868                         state.Ctx.LogD("sp-file", sdsp, "writing")
 869                         _, err = fd.Write(file.Payload)
 870                         if err != nil {
 871                                 state.Ctx.LogE("sp-file", SdsAdd(sdsp, SDS{"err": err}), "")
 872                                 fd.Close()
 873                                 return nil, err
 874                         }
 875                         ourSize := uint64(file.Offset) + uint64(len(file.Payload))
 876                         state.RLock()
 877                         sdsp["fullsize"] = strconv.FormatInt(int64(state.infosTheir[*file.Hash].Size), 10)
 878                         sdsp["size"] = strconv.FormatInt(int64(ourSize), 10)
 879                         state.Ctx.LogP("sp-file", sdsp, "")
 880                         if state.infosTheir[*file.Hash].Size != ourSize {
 881                                 state.RUnlock()
 882                                 fd.Close()
 883                                 continue
 884                         }
 885                         state.RUnlock()
 886                         spWorkersGroup.Wait()
 887                         spWorkersGroup.Add(1)
 888                         go func() {
 889                                 if err := fd.Sync(); err != nil {
 890                                         state.Ctx.LogE("sp-file", SdsAdd(sdsp, SDS{"err": err}), "sync")
 891                                         fd.Close()
 892                                         return
 893                                 }
 894                                 state.wg.Add(1)
 895                                 defer state.wg.Done()
 896                                 fd.Seek(0, io.SeekStart)
 897                                 state.Ctx.LogD("sp-file", sdsp, "checking")
 898                                 gut, err := Check(fd, file.Hash[:])
 899                                 fd.Close()
 900                                 if err != nil || !gut {
 901                                         state.Ctx.LogE("sp-file", sdsp, "checksum mismatch")
 902                                         return
 903                                 }
 904                                 state.Ctx.LogI("sp-done", SdsAdd(sdsp, SDS{"xx": string(TRx)}), "")
 905                                 os.Rename(filePath+PartSuffix, filePath)
 906                                 state.Lock()
 907                                 delete(state.infosTheir, *file.Hash)
 908                                 state.Unlock()
 909                                 spWorkersGroup.Done()
 910                                 go func() {
 911                                         state.payloads <- MarshalSP(SPTypeDone, SPDone{file.Hash})
 912                                 }()
 913                         }()
 914                 case SPTypeDone:
 915                         sdsp := SdsAdd(sds, SDS{"type": "done"})
 916                         state.Ctx.LogD("sp-process", sdsp, "unmarshaling packet")
 917                         var done SPDone
 918                         if _, err = xdr.Unmarshal(r, &done); err != nil {
 919                                 state.Ctx.LogE("sp-process", SdsAdd(sds, SDS{
 920                                         "type": "done",
 921                                         "err":  err,
 922                                 }), "")
 923                                 return nil, err
 924                         }
 925                         sdsp["hash"] = ToBase32(done.Hash[:])
 926                         state.Ctx.LogD("sp-done", sdsp, "removing")
 927                         err := os.Remove(filepath.Join(
 928                                 state.Ctx.Spool,
 929                                 state.Node.Id.String(),
 930                                 string(TTx),
 931                                 ToBase32(done.Hash[:]),
 932                         ))
 933                         sdsp["xx"] = string(TTx)
 934                         if err == nil {
 935                                 state.Ctx.LogI("sp-done", sdsp, "")
 936                         } else {
 937                                 state.Ctx.LogE("sp-done", sdsp, "")
 938                         }
 939                 case SPTypeFreq:
 940                         sdsp := SdsAdd(sds, SDS{"type": "freq"})
 941                         state.Ctx.LogD("sp-process", sdsp, "unmarshaling packet")
 942                         var freq SPFreq
 943                         if _, err = xdr.Unmarshal(r, &freq); err != nil {
 944                                 state.Ctx.LogE("sp-process", SdsAdd(sdsp, SDS{"err": err}), "")
 945                                 return nil, err
 946                         }
 947                         sdsp["hash"] = ToBase32(freq.Hash[:])
 948                         sdsp["offset"] = strconv.FormatInt(int64(freq.Offset), 10)
 949                         state.Ctx.LogD("sp-process", sdsp, "queueing")
 950                         nice, exists := state.infosOurSeen[*freq.Hash]
 951                         if exists {
 952                                 if state.onlyPkts == nil || !state.onlyPkts[*freq.Hash] {
 953                                         state.Lock()
 954                                         insertIdx := 0
 955                                         var freqWithNice *FreqWithNice
 956                                         for insertIdx, freqWithNice = range state.queueTheir {
 957                                                 if freqWithNice.nice > nice {
 958                                                         break
 959                                                 }
 960                                         }
 961                                         state.queueTheir = append(state.queueTheir, nil)
 962                                         copy(state.queueTheir[insertIdx+1:], state.queueTheir[insertIdx:])
 963                                         state.queueTheir[insertIdx] = &FreqWithNice{&freq, nice}
 964                                         state.Unlock()
 965                                 } else {
 966                                         state.Ctx.LogD("sp-process", sdsp, "skipping")
 967                                 }
 968                         } else {
 969                                 state.Ctx.LogD("sp-process", sdsp, "unknown")
 970                         }
 971                 case SPTypeHalt:
 972                         state.Ctx.LogD("sp-process", SdsAdd(sds, SDS{"type": "halt"}), "")
 973                         state.Lock()
 974                         state.queueTheir = nil
 975                         state.Unlock()
 976                 default:
 977                         state.Ctx.LogE(
 978                                 "sp-process",
 979                                 SdsAdd(sds, SDS{"type": head.Type}),
 980                                 "unknown",
 981                         )
 982                         return nil, BadPktType
 983                 }
 984         }
 985         if infosGot {
 986                 var pkts int
 987                 var size uint64
 988                 state.RLock()
 989                 for _, info := range state.infosTheir {
 990                         pkts++
 991                         size += info.Size
 992                 }
 993                 state.RUnlock()
 994                 state.Ctx.LogI("sp-infos", SDS{
 995                         "xx":   string(TRx),
 996                         "node": state.Node.Id,
 997                         "pkts": strconv.Itoa(pkts),
 998                         "size": strconv.FormatInt(int64(size), 10),
 999                 }, "")
1000         }
1001         return payloadsSplit(replies), nil
1002 }