src/sp.go

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