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         chan struct{}
 199         listOnly       bool
 200         onlyPkts       map[[32]byte]bool
 201         sync.RWMutex
 202 }
 203
 204 func (state *SPState) SetDead() {
 205         state.Lock()
 206         defer state.Unlock()
 207         select {
 208         case _, ok := <-state.isDead:
 209                 if !ok {
 210                         // Already closed channel, dead
 211                         return
 212                 }
 213         default:
 214         }
 215         close(state.isDead)
 216         go func() {
 217                 for _ = range state.payloads {
 218                 }
 219         }()
 220 }
 221
 222 func (state *SPState) NotAlive() bool {
 223         select {
 224         case _, ok := <-state.isDead:
 225                 if !ok {
 226                         return true
 227                 }
 228         default:
 229         }
 230         now := time.Now()
 231         if state.maxOnlineTime > 0 && state.started.Add(time.Duration(state.maxOnlineTime)*time.Second).Before(now) {
 232                 return true
 233         }
 234         return uint(now.Sub(state.RxLastSeen).Seconds()) >= state.onlineDeadline &&
 235                 uint(now.Sub(state.TxLastSeen).Seconds()) >= state.onlineDeadline
 236 }
 237
 238 func (state *SPState) dirUnlock() {
 239         state.Ctx.UnlockDir(state.rxLock)
 240         state.Ctx.UnlockDir(state.txLock)
 241 }
 242
 243 func (state *SPState) WriteSP(dst io.Writer, payload []byte) error {
 244         n, err := xdr.Marshal(dst, SPRaw{Magic: MagicNNCPLv1, Payload: payload})
 245         if err == nil {
 246                 state.TxLastSeen = time.Now()
 247                 state.TxBytes += int64(n)
 248         }
 249         return err
 250 }
 251
 252 func (state *SPState) ReadSP(src io.Reader) ([]byte, error) {
 253         var sp SPRaw
 254         n, err := xdr.UnmarshalLimited(src, &sp, 1<<17)
 255         if err != nil {
 256                 ue := err.(*xdr.UnmarshalError)
 257                 if ue.Err == io.EOF {
 258                         return nil, ue.Err
 259                 }
 260                 return nil, err
 261         }
 262         state.RxLastSeen = time.Now()
 263         state.RxBytes += int64(n)
 264         if sp.Magic != MagicNNCPLv1 {
 265                 return nil, BadMagic
 266         }
 267         return sp.Payload, nil
 268 }
 269
 270 func (ctx *Ctx) infosOur(nodeId *NodeId, nice uint8, seen *map[[32]byte]uint8) [][]byte {
 271         var infos []*SPInfo
 272         var totalSize int64
 273         for job := range ctx.Jobs(nodeId, TTx) {
 274                 job.Fd.Close()
 275                 if job.PktEnc.Nice > nice {
 276                         continue
 277                 }
 278                 if _, known := (*seen)[*job.HshValue]; known {
 279                         continue
 280                 }
 281                 totalSize += job.Size
 282                 infos = append(infos, &SPInfo{
 283                         Nice: job.PktEnc.Nice,
 284                         Size: uint64(job.Size),
 285                         Hash: job.HshValue,
 286                 })
 287                 (*seen)[*job.HshValue] = job.PktEnc.Nice
 288         }
 289         sort.Sort(ByNice(infos))
 290         var payloads [][]byte
 291         for _, info := range infos {
 292                 payloads = append(payloads, MarshalSP(SPTypeInfo, info))
 293                 ctx.LogD("sp-info-our", SDS{
 294                         "node": nodeId,
 295                         "name": ToBase32(info.Hash[:]),
 296                         "size": info.Size,
 297                 }, "")
 298         }
 299         if totalSize > 0 {
 300                 ctx.LogI("sp-infos", SDS{
 301                         "xx":   string(TTx),
 302                         "node": nodeId,
 303                         "pkts": len(payloads),
 304                         "size": totalSize,
 305                 }, "")
 306         }
 307         return payloadsSplit(payloads)
 308 }
 309
 310 func (state *SPState) StartI(conn ConnDeadlined) error {
 311         nodeId := state.Node.Id
 312         err := state.Ctx.ensureRxDir(nodeId)
 313         if err != nil {
 314                 return err
 315         }
 316         var rxLock *os.File
 317         if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TRx) {
 318                 rxLock, err = state.Ctx.LockDir(nodeId, TRx)
 319                 if err != nil {
 320                         return err
 321                 }
 322         }
 323         var txLock *os.File
 324         if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TTx) {
 325                 txLock, err = state.Ctx.LockDir(nodeId, TTx)
 326                 if err != nil {
 327                         return err
 328                 }
 329         }
 330         started := time.Now()
 331         conf := noise.Config{
 332                 CipherSuite: NoiseCipherSuite,
 333                 Pattern:     noise.HandshakeIK,
 334                 Initiator:   true,
 335                 StaticKeypair: noise.DHKey{
 336                         Private: state.Ctx.Self.NoisePrv[:],
 337                         Public:  state.Ctx.Self.NoisePub[:],
 338                 },
 339                 PeerStatic: state.Node.NoisePub[:],
 340         }
 341         hs, err := noise.NewHandshakeState(conf)
 342         if err != nil {
 343                 return err
 344         }
 345         state.hs = hs
 346         state.payloads = make(chan []byte)
 347         state.infosTheir = make(map[[32]byte]*SPInfo)
 348         state.infosOurSeen = make(map[[32]byte]uint8)
 349         state.started = started
 350         state.rxLock = rxLock
 351         state.txLock = txLock
 352
 353         var infosPayloads [][]byte
 354         if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TTx) {
 355                 infosPayloads = state.Ctx.infosOur(nodeId, state.Nice, &state.infosOurSeen)
 356         }
 357         var firstPayload []byte
 358         if len(infosPayloads) > 0 {
 359                 firstPayload = infosPayloads[0]
 360         }
 361         // Pad first payload, to hide actual number of existing files
 362         for i := 0; i < (MaxSPSize-len(firstPayload))/SPHeadOverhead; i++ {
 363                 firstPayload = append(firstPayload, SPHaltMarshalized...)
 364         }
 365
 366         var buf []byte
 367         var payload []byte
 368         buf, _, _, err = state.hs.WriteMessage(nil, firstPayload)
 369         if err != nil {
 370                 state.dirUnlock()
 371                 return err
 372         }
 373         sds := SDS{"node": nodeId, "nice": int(state.Nice)}
 374         state.Ctx.LogD("sp-start", sds, "sending first message")
 375         conn.SetWriteDeadline(time.Now().Add(DefaultDeadline * time.Second))
 376         if err = state.WriteSP(conn, buf); err != nil {
 377                 state.Ctx.LogE("sp-start", sds, err, "")
 378                 state.dirUnlock()
 379                 return err
 380         }
 381         state.Ctx.LogD("sp-start", sds, "waiting for first message")
 382         conn.SetReadDeadline(time.Now().Add(DefaultDeadline * time.Second))
 383         if buf, err = state.ReadSP(conn); err != nil {
 384                 state.Ctx.LogE("sp-start", sds, err, "")
 385                 state.dirUnlock()
 386                 return err
 387         }
 388         payload, state.csOur, state.csTheir, err = state.hs.ReadMessage(nil, buf)
 389         if err != nil {
 390                 state.Ctx.LogE("sp-start", sds, err, "")
 391                 state.dirUnlock()
 392                 return err
 393         }
 394         state.Ctx.LogD("sp-start", sds, "starting workers")
 395         err = state.StartWorkers(conn, infosPayloads, payload)
 396         if err != nil {
 397                 state.Ctx.LogE("sp-start", sds, err, "")
 398                 state.dirUnlock()
 399         }
 400         return err
 401 }
 402
 403 func (state *SPState) StartR(conn ConnDeadlined) error {
 404         started := time.Now()
 405         conf := noise.Config{
 406                 CipherSuite: NoiseCipherSuite,
 407                 Pattern:     noise.HandshakeIK,
 408                 Initiator:   false,
 409                 StaticKeypair: noise.DHKey{
 410                         Private: state.Ctx.Self.NoisePrv[:],
 411                         Public:  state.Ctx.Self.NoisePub[:],
 412                 },
 413         }
 414         hs, err := noise.NewHandshakeState(conf)
 415         if err != nil {
 416                 return err
 417         }
 418         xxOnly := TRxTx("")
 419         state.hs = hs
 420         state.payloads = make(chan []byte)
 421         state.infosOurSeen = make(map[[32]byte]uint8)
 422         state.infosTheir = make(map[[32]byte]*SPInfo)
 423         state.started = started
 424         state.xxOnly = xxOnly
 425         var buf []byte
 426         var payload []byte
 427         state.Ctx.LogD("sp-start", SDS{"nice": int(state.Nice)}, "waiting for first message")
 428         conn.SetReadDeadline(time.Now().Add(DefaultDeadline * time.Second))
 429         if buf, err = state.ReadSP(conn); err != nil {
 430                 state.Ctx.LogE("sp-start", SDS{}, err, "")
 431                 return err
 432         }
 433         if payload, _, _, err = state.hs.ReadMessage(nil, buf); err != nil {
 434                 state.Ctx.LogE("sp-start", SDS{}, err, "")
 435                 return err
 436         }
 437
 438         var node *Node
 439         for _, n := range state.Ctx.Neigh {
 440                 if subtle.ConstantTimeCompare(state.hs.PeerStatic(), n.NoisePub[:]) == 1 {
 441                         node = n
 442                         break
 443                 }
 444         }
 445         if node == nil {
 446                 peerId := ToBase32(state.hs.PeerStatic())
 447                 state.Ctx.LogE("sp-start", SDS{"peer": peerId}, errors.New("unknown"), "")
 448                 return errors.New("Unknown peer: " + peerId)
 449         }
 450         state.Node = node
 451         state.rxRate = node.RxRate
 452         state.txRate = node.TxRate
 453         state.onlineDeadline = node.OnlineDeadline
 454         state.maxOnlineTime = node.MaxOnlineTime
 455         sds := SDS{"node": node.Id, "nice": int(state.Nice)}
 456
 457         if state.Ctx.ensureRxDir(node.Id); err != nil {
 458                 return err
 459         }
 460         var rxLock *os.File
 461         if xxOnly == "" || xxOnly == TRx {
 462                 rxLock, err = state.Ctx.LockDir(node.Id, TRx)
 463                 if err != nil {
 464                         return err
 465                 }
 466         }
 467         state.rxLock = rxLock
 468         var txLock *os.File
 469         if xxOnly == "" || xxOnly == TTx {
 470                 txLock, err = state.Ctx.LockDir(node.Id, TTx)
 471                 if err != nil {
 472                         return err
 473                 }
 474         }
 475         state.txLock = txLock
 476
 477         var infosPayloads [][]byte
 478         if xxOnly == "" || xxOnly == TTx {
 479                 infosPayloads = state.Ctx.infosOur(node.Id, state.Nice, &state.infosOurSeen)
 480         }
 481         var firstPayload []byte
 482         if len(infosPayloads) > 0 {
 483                 firstPayload = infosPayloads[0]
 484         }
 485         // Pad first payload, to hide actual number of existing files
 486         for i := 0; i < (MaxSPSize-len(firstPayload))/SPHeadOverhead; i++ {
 487                 firstPayload = append(firstPayload, SPHaltMarshalized...)
 488         }
 489
 490         state.Ctx.LogD("sp-start", sds, "sending first message")
 491         buf, state.csTheir, state.csOur, err = state.hs.WriteMessage(nil, firstPayload)
 492         if err != nil {
 493                 state.dirUnlock()
 494                 return err
 495         }
 496         conn.SetWriteDeadline(time.Now().Add(DefaultDeadline * time.Second))
 497         if err = state.WriteSP(conn, buf); err != nil {
 498                 state.Ctx.LogE("sp-start", sds, err, "")
 499                 state.dirUnlock()
 500                 return err
 501         }
 502         state.Ctx.LogD("sp-start", sds, "starting workers")
 503         err = state.StartWorkers(conn, infosPayloads, payload)
 504         if err != nil {
 505                 state.dirUnlock()
 506         }
 507         return err
 508 }
 509
 510 func (state *SPState) StartWorkers(
 511         conn ConnDeadlined,
 512         infosPayloads [][]byte,
 513         payload []byte,
 514 ) error {
 515         state.isDead = make(chan struct{})
 516         sds := SDS{"node": state.Node.Id, "nice": int(state.Nice)}
 517
 518         if len(infosPayloads) > 1 {
 519                 state.wg.Add(1)
 520                 go func() {
 521                         for _, payload := range infosPayloads[1:] {
 522                                 state.Ctx.LogD(
 523                                         "sp-work",
 524                                         SdsAdd(sds, SDS{"size": len(payload)}),
 525                                         "queuing remaining payload",
 526                                 )
 527                                 state.payloads <- payload
 528                         }
 529                         state.wg.Done()
 530                 }()
 531         }
 532         state.Ctx.LogD(
 533                 "sp-work",
 534                 SdsAdd(sds, SDS{"size": len(payload)}),
 535                 "processing first payload",
 536         )
 537         replies, err := state.ProcessSP(payload)
 538         if err != nil {
 539                 state.Ctx.LogE("sp-work", sds, err, "")
 540                 return err
 541         }
 542
 543         state.wg.Add(1)
 544         go func() {
 545                 for _, reply := range replies {
 546                         state.Ctx.LogD(
 547                                 "sp-work",
 548                                 SdsAdd(sds, SDS{"size": len(reply)}),
 549                                 "queuing reply",
 550                         )
 551                         state.payloads <- reply
 552                 }
 553                 state.wg.Done()
 554         }()
 555
 556         if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TTx) {
 557                 state.wg.Add(1)
 558                 go func() {
 559                         ticker := time.NewTicker(time.Second)
 560                         for {
 561                                 select {
 562                                 case _, ok := <-state.isDead:
 563                                         if !ok {
 564                                                 state.wg.Done()
 565                                                 ticker.Stop()
 566                                                 return
 567                                         }
 568                                 case <-ticker.C:
 569                                         for _, payload := range state.Ctx.infosOur(
 570                                                 state.Node.Id,
 571                                                 state.Nice,
 572                                                 &state.infosOurSeen,
 573                                         ) {
 574                                                 state.Ctx.LogD(
 575                                                         "sp-work",
 576                                                         SdsAdd(sds, SDS{"size": len(payload)}),
 577                                                         "queuing new info",
 578                                                 )
 579                                                 state.payloads <- payload
 580                                         }
 581                                 }
 582                         }
 583                 }()
 584         }
 585
 586         state.wg.Add(1)
 587         go func() {
 588                 for {
 589                         if state.NotAlive() {
 590                                 break
 591                         }
 592                         var payload []byte
 593                         select {
 594                         case payload = <-state.payloads:
 595                                 state.Ctx.LogD(
 596                                         "sp-xmit",
 597                                         SdsAdd(sds, SDS{"size": len(payload)}),
 598                                         "got payload",
 599                                 )
 600                         default:
 601                         }
 602                         if payload == nil {
 603                                 state.RLock()
 604                                 if len(state.queueTheir) == 0 {
 605                                         state.RUnlock()
 606                                         time.Sleep(100 * time.Millisecond)
 607                                         continue
 608                                 }
 609                                 freq := state.queueTheir[0].freq
 610                                 state.RUnlock()
 611
 612                                 if state.txRate > 0 {
 613                                         time.Sleep(time.Second / time.Duration(state.txRate))
 614                                 }
 615
 616                                 sdsp := SdsAdd(sds, SDS{
 617                                         "xx":   string(TTx),
 618                                         "pkt":  ToBase32(freq.Hash[:]),
 619                                         "size": int64(freq.Offset),
 620                                 })
 621                                 state.Ctx.LogD("sp-file", sdsp, "queueing")
 622                                 fd, err := os.Open(filepath.Join(
 623                                         state.Ctx.Spool,
 624                                         state.Node.Id.String(),
 625                                         string(TTx),
 626                                         ToBase32(freq.Hash[:]),
 627                                 ))
 628                                 if err != nil {
 629                                         state.Ctx.LogE("sp-file", sdsp, err, "")
 630                                         break
 631                                 }
 632                                 fi, err := fd.Stat()
 633                                 if err != nil {
 634                                         state.Ctx.LogE("sp-file", sdsp, err, "")
 635                                         break
 636                                 }
 637                                 fullSize := fi.Size()
 638                                 var buf []byte
 639                                 if freq.Offset < uint64(fullSize) {
 640                                         state.Ctx.LogD("sp-file", sdsp, "seeking")
 641                                         if _, err = fd.Seek(int64(freq.Offset), io.SeekStart); err != nil {
 642                                                 state.Ctx.LogE("sp-file", sdsp, err, "")
 643                                                 break
 644                                         }
 645                                         buf = make([]byte, MaxSPSize-SPHeadOverhead-SPFileOverhead)
 646                                         n, err := fd.Read(buf)
 647                                         if err != nil {
 648                                                 state.Ctx.LogE("sp-file", sdsp, err, "")
 649                                                 break
 650                                         }
 651                                         buf = buf[:n]
 652                                         state.Ctx.LogD("sp-file", SdsAdd(sdsp, SDS{"size": n}), "read")
 653                                 }
 654                                 fd.Close()
 655                                 payload = MarshalSP(SPTypeFile, SPFile{
 656                                         Hash:    freq.Hash,
 657                                         Offset:  freq.Offset,
 658                                         Payload: buf,
 659                                 })
 660                                 ourSize := freq.Offset + uint64(len(buf))
 661                                 sdsp["size"] = int64(ourSize)
 662                                 sdsp["fullsize"] = fullSize
 663                                 if state.Ctx.ShowPrgrs {
 664                                         Progress("Tx", sdsp)
 665                                 }
 666                                 state.Lock()
 667                                 if len(state.queueTheir) > 0 && *state.queueTheir[0].freq.Hash == *freq.Hash {
 668                                         if ourSize == uint64(fullSize) {
 669                                                 state.Ctx.LogD("sp-file", sdsp, "finished")
 670                                                 if len(state.queueTheir) > 1 {
 671                                                         state.queueTheir = state.queueTheir[1:]
 672                                                 } else {
 673                                                         state.queueTheir = state.queueTheir[:0]
 674                                                 }
 675                                         } else {
 676                                                 state.queueTheir[0].freq.Offset += uint64(len(buf))
 677                                         }
 678                                 } else {
 679                                         state.Ctx.LogD("sp-file", sdsp, "queue disappeared")
 680                                 }
 681                                 state.Unlock()
 682                         }
 683                         state.Ctx.LogD(
 684                                 "sp-xmit",
 685                                 SdsAdd(sds, SDS{"size": len(payload)}),
 686                                 "sending",
 687                         )
 688                         conn.SetWriteDeadline(time.Now().Add(DefaultDeadline * time.Second))
 689                         if err := state.WriteSP(conn, state.csOur.Encrypt(nil, nil, payload)); err != nil {
 690                                 state.Ctx.LogE("sp-xmit", sds, err, "")
 691                                 break
 692                         }
 693                 }
 694                 state.SetDead()
 695                 state.wg.Done()
 696         }()
 697
 698         state.wg.Add(1)
 699         go func() {
 700                 for {
 701                         if state.NotAlive() {
 702                                 break
 703                         }
 704                         state.Ctx.LogD("sp-recv", sds, "waiting for payload")
 705                         conn.SetReadDeadline(time.Now().Add(DefaultDeadline * time.Second))
 706                         payload, err := state.ReadSP(conn)
 707                         if err != nil {
 708                                 if err == io.EOF {
 709                                         break
 710                                 }
 711                                 unmarshalErr := err.(*xdr.UnmarshalError)
 712                                 netErr, ok := unmarshalErr.Err.(net.Error)
 713                                 if ok && netErr.Timeout() {
 714                                         continue
 715                                 }
 716                                 if unmarshalErr.ErrorCode == xdr.ErrIO {
 717                                         break
 718                                 }
 719                                 state.Ctx.LogE("sp-recv", sds, err, "")
 720                                 break
 721                         }
 722                         state.Ctx.LogD(
 723                                 "sp-recv",
 724                                 SdsAdd(sds, SDS{"size": len(payload)}),
 725                                 "got payload",
 726                         )
 727                         payload, err = state.csTheir.Decrypt(nil, nil, payload)
 728                         if err != nil {
 729                                 state.Ctx.LogE("sp-recv", sds, err, "")
 730                                 break
 731                         }
 732                         state.Ctx.LogD(
 733                                 "sp-recv",
 734                                 SdsAdd(sds, SDS{"size": len(payload)}),
 735                                 "processing",
 736                         )
 737                         replies, err := state.ProcessSP(payload)
 738                         if err != nil {
 739                                 state.Ctx.LogE("sp-recv", sds, err, "")
 740                                 break
 741                         }
 742                         state.wg.Add(1)
 743                         go func() {
 744                                 for _, reply := range replies {
 745                                         state.Ctx.LogD(
 746                                                 "sp-recv",
 747                                                 SdsAdd(sds, SDS{"size": len(reply)}),
 748                                                 "queuing reply",
 749                                         )
 750                                         state.payloads <- reply
 751                                 }
 752                                 state.wg.Done()
 753                         }()
 754                         if state.rxRate > 0 {
 755                                 time.Sleep(time.Second / time.Duration(state.rxRate))
 756                         }
 757                 }
 758                 state.SetDead()
 759                 state.wg.Done()
 760         }()
 761
 762         return nil
 763 }
 764
 765 func (state *SPState) Wait() {
 766         state.wg.Wait()
 767         close(state.payloads)
 768         state.dirUnlock()
 769         state.Duration = time.Now().Sub(state.started)
 770         state.RxSpeed = state.RxBytes
 771         state.TxSpeed = state.TxBytes
 772         rxDuration := int64(state.RxLastSeen.Sub(state.started).Seconds())
 773         txDuration := int64(state.TxLastSeen.Sub(state.started).Seconds())
 774         if rxDuration > 0 {
 775                 state.RxSpeed = state.RxBytes / rxDuration
 776         }
 777         if txDuration > 0 {
 778                 state.TxSpeed = state.TxBytes / txDuration
 779         }
 780 }
 781
 782 func (state *SPState) ProcessSP(payload []byte) ([][]byte, error) {
 783         sds := SDS{"node": state.Node.Id, "nice": int(state.Nice)}
 784         r := bytes.NewReader(payload)
 785         var err error
 786         var replies [][]byte
 787         var infosGot bool
 788         for r.Len() > 0 {
 789                 state.Ctx.LogD("sp-process", sds, "unmarshaling header")
 790                 var head SPHead
 791                 if _, err = xdr.Unmarshal(r, &head); err != nil {
 792                         state.Ctx.LogE("sp-process", sds, err, "")
 793                         return nil, err
 794                 }
 795                 switch head.Type {
 796                 case SPTypeInfo:
 797                         infosGot = true
 798                         sdsp := SdsAdd(sds, SDS{"type": "info"})
 799                         state.Ctx.LogD("sp-process", sdsp, "unmarshaling packet")
 800                         var info SPInfo
 801                         if _, err = xdr.Unmarshal(r, &info); err != nil {
 802                                 state.Ctx.LogE("sp-process", sdsp, err, "")
 803                                 return nil, err
 804                         }
 805                         sdsp = SdsAdd(sds, SDS{
 806                                 "pkt":  ToBase32(info.Hash[:]),
 807                                 "size": int64(info.Size),
 808                                 "nice": int(info.Nice),
 809                         })
 810                         if !state.listOnly && info.Nice > state.Nice {
 811                                 state.Ctx.LogD("sp-process", sdsp, "too nice")
 812                                 continue
 813                         }
 814                         state.Ctx.LogD("sp-process", sdsp, "received")
 815                         if !state.listOnly && state.xxOnly == TTx {
 816                                 continue
 817                         }
 818                         state.Lock()
 819                         state.infosTheir[*info.Hash] = &info
 820                         state.Unlock()
 821                         state.Ctx.LogD("sp-process", sdsp, "stating part")
 822                         pktPath := filepath.Join(
 823                                 state.Ctx.Spool,
 824                                 state.Node.Id.String(),
 825                                 string(TRx),
 826                                 ToBase32(info.Hash[:]),
 827                         )
 828                         if _, err = os.Stat(pktPath); err == nil {
 829                                 state.Ctx.LogI("sp-info", sdsp, "already done")
 830                                 if !state.listOnly {
 831                                         replies = append(replies, MarshalSP(SPTypeDone, SPDone{info.Hash}))
 832                                 }
 833                                 continue
 834                         }
 835                         if _, err = os.Stat(pktPath + SeenSuffix); err == nil {
 836                                 state.Ctx.LogI("sp-info", sdsp, "already seen")
 837                                 if !state.listOnly {
 838                                         replies = append(replies, MarshalSP(SPTypeDone, SPDone{info.Hash}))
 839                                 }
 840                                 continue
 841                         }
 842                         fi, err := os.Stat(pktPath + PartSuffix)
 843                         var offset int64
 844                         if err == nil {
 845                                 offset = fi.Size()
 846                         }
 847                         if !state.Ctx.IsEnoughSpace(int64(info.Size) - offset) {
 848                                 state.Ctx.LogI("sp-info", sdsp, "not enough space")
 849                                 continue
 850                         }
 851                         state.Ctx.LogI(
 852                                 "sp-info",
 853                                 SdsAdd(sdsp, SDS{"offset": offset}),
 854                                 "",
 855                         )
 856                         if !state.listOnly && (state.onlyPkts == nil || state.onlyPkts[*info.Hash]) {
 857                                 replies = append(replies, MarshalSP(
 858                                         SPTypeFreq,
 859                                         SPFreq{info.Hash, uint64(offset)},
 860                                 ))
 861                         }
 862                 case SPTypeFile:
 863                         sdsp := SdsAdd(sds, SDS{"type": "file"})
 864                         state.Ctx.LogD("sp-process", sdsp, "unmarshaling packet")
 865                         var file SPFile
 866                         if _, err = xdr.Unmarshal(r, &file); err != nil {
 867                                 state.Ctx.LogE("sp-process", SdsAdd(sds, SDS{"type": "file"}), err, "")
 868                                 return nil, err
 869                         }
 870                         sdsp["xx"] = string(TRx)
 871                         sdsp["pkt"] = ToBase32(file.Hash[:])
 872                         sdsp["size"] = len(file.Payload)
 873                         dirToSync := filepath.Join(
 874                                 state.Ctx.Spool,
 875                                 state.Node.Id.String(),
 876                                 string(TRx),
 877                         )
 878                         filePath := filepath.Join(dirToSync, ToBase32(file.Hash[:]))
 879                         state.Ctx.LogD("sp-file", sdsp, "opening part")
 880                         fd, err := os.OpenFile(
 881                                 filePath+PartSuffix,
 882                                 os.O_RDWR|os.O_CREATE,
 883                                 os.FileMode(0666),
 884                         )
 885                         if err != nil {
 886                                 state.Ctx.LogE("sp-file", sdsp, err, "")
 887                                 return nil, err
 888                         }
 889                         state.Ctx.LogD(
 890                                 "sp-file",
 891                                 SdsAdd(sdsp, SDS{"offset": file.Offset}),
 892                                 "seeking",
 893                         )
 894                         if _, err = fd.Seek(int64(file.Offset), io.SeekStart); err != nil {
 895                                 state.Ctx.LogE("sp-file", sdsp, err, "")
 896                                 fd.Close()
 897                                 return nil, err
 898                         }
 899                         state.Ctx.LogD("sp-file", sdsp, "writing")
 900                         _, err = fd.Write(file.Payload)
 901                         if err != nil {
 902                                 state.Ctx.LogE("sp-file", sdsp, err, "")
 903                                 fd.Close()
 904                                 return nil, err
 905                         }
 906                         ourSize := int64(file.Offset + uint64(len(file.Payload)))
 907                         sdsp["size"] = ourSize
 908                         fullsize := int64(0)
 909                         state.RLock()
 910                         infoTheir, ok := state.infosTheir[*file.Hash]
 911                         state.RUnlock()
 912                         if ok {
 913                                 fullsize = int64(infoTheir.Size)
 914                         }
 915                         sdsp["fullsize"] = fullsize
 916                         if state.Ctx.ShowPrgrs {
 917                                 Progress("Rx", sdsp)
 918                         }
 919                         if fullsize != ourSize {
 920                                 fd.Close()
 921                                 continue
 922                         }
 923                         spWorkersGroup.Wait()
 924                         spWorkersGroup.Add(1)
 925                         go func() {
 926                                 if err := fd.Sync(); err != nil {
 927                                         state.Ctx.LogE("sp-file", sdsp, err, "sync")
 928                                         fd.Close()
 929                                         return
 930                                 }
 931                                 state.wg.Add(1)
 932                                 defer state.wg.Done()
 933                                 fd.Seek(0, io.SeekStart)
 934                                 state.Ctx.LogD("sp-file", sdsp, "checking")
 935                                 gut, err := Check(fd, file.Hash[:], sdsp, state.Ctx.ShowPrgrs)
 936                                 fd.Close()
 937                                 if err != nil || !gut {
 938                                         state.Ctx.LogE("sp-file", sdsp, errors.New("checksum mismatch"), "")
 939                                         return
 940                                 }
 941                                 state.Ctx.LogI("sp-done", SdsAdd(sdsp, SDS{"xx": string(TRx)}), "")
 942                                 if err = os.Rename(filePath+PartSuffix, filePath); err != nil {
 943                                         state.Ctx.LogE("sp-file", sdsp, err, "rename")
 944                                         return
 945                                 }
 946                                 if err = DirSync(dirToSync); err != nil {
 947                                         state.Ctx.LogE("sp-file", sdsp, err, "sync")
 948                                         return
 949                                 }
 950                                 state.Lock()
 951                                 delete(state.infosTheir, *file.Hash)
 952                                 state.Unlock()
 953                                 spWorkersGroup.Done()
 954                                 state.wg.Add(1)
 955                                 go func() {
 956                                         state.payloads <- MarshalSP(SPTypeDone, SPDone{file.Hash})
 957                                         state.wg.Done()
 958                                 }()
 959                         }()
 960                 case SPTypeDone:
 961                         sdsp := SdsAdd(sds, SDS{"type": "done"})
 962                         state.Ctx.LogD("sp-process", sdsp, "unmarshaling packet")
 963                         var done SPDone
 964                         if _, err = xdr.Unmarshal(r, &done); err != nil {
 965                                 state.Ctx.LogE("sp-process", SdsAdd(sds, SDS{"type": "done"}), err, "")
 966                                 return nil, err
 967                         }
 968                         sdsp["pkt"] = ToBase32(done.Hash[:])
 969                         state.Ctx.LogD("sp-done", sdsp, "removing")
 970                         err := os.Remove(filepath.Join(
 971                                 state.Ctx.Spool,
 972                                 state.Node.Id.String(),
 973                                 string(TTx),
 974                                 ToBase32(done.Hash[:]),
 975                         ))
 976                         sdsp["xx"] = string(TTx)
 977                         if err == nil {
 978                                 state.Ctx.LogI("sp-done", sdsp, "")
 979                         } else {
 980                                 state.Ctx.LogE("sp-done", sdsp, err, "")
 981                         }
 982                 case SPTypeFreq:
 983                         sdsp := SdsAdd(sds, SDS{"type": "freq"})
 984                         state.Ctx.LogD("sp-process", sdsp, "unmarshaling packet")
 985                         var freq SPFreq
 986                         if _, err = xdr.Unmarshal(r, &freq); err != nil {
 987                                 state.Ctx.LogE("sp-process", sdsp, err, "")
 988                                 return nil, err
 989                         }
 990                         sdsp["pkt"] = ToBase32(freq.Hash[:])
 991                         sdsp["offset"] = freq.Offset
 992                         state.Ctx.LogD("sp-process", sdsp, "queueing")
 993                         nice, exists := state.infosOurSeen[*freq.Hash]
 994                         if exists {
 995                                 if state.onlyPkts == nil || !state.onlyPkts[*freq.Hash] {
 996                                         state.Lock()
 997                                         insertIdx := 0
 998                                         var freqWithNice *FreqWithNice
 999                                         for insertIdx, freqWithNice = range state.queueTheir {
1000                                                 if freqWithNice.nice > nice {
1001                                                         break
1002                                                 }
1003                                         }
1004                                         state.queueTheir = append(state.queueTheir, nil)
1005                                         copy(state.queueTheir[insertIdx+1:], state.queueTheir[insertIdx:])
1006                                         state.queueTheir[insertIdx] = &FreqWithNice{&freq, nice}
1007                                         state.Unlock()
1008                                 } else {
1009                                         state.Ctx.LogD("sp-process", sdsp, "skipping")
1010                                 }
1011                         } else {
1012                                 state.Ctx.LogD("sp-process", sdsp, "unknown")
1013                         }
1014                 case SPTypeHalt:
1015                         state.Ctx.LogD("sp-process", SdsAdd(sds, SDS{"type": "halt"}), "")
1016                         state.Lock()
1017                         state.queueTheir = nil
1018                         state.Unlock()
1019                 default:
1020                         state.Ctx.LogE(
1021                                 "sp-process",
1022                                 SdsAdd(sds, SDS{"type": head.Type}),
1023                                 errors.New("unknown type"),
1024                                 "",
1025                         )
1026                         return nil, BadPktType
1027                 }
1028         }
1029         if infosGot {
1030                 var pkts int
1031                 var size uint64
1032                 state.RLock()
1033                 for _, info := range state.infosTheir {
1034                         pkts++
1035                         size += info.Size
1036                 }
1037                 state.RUnlock()
1038                 state.Ctx.LogI("sp-infos", SDS{
1039                         "xx":   string(TRx),
1040                         "node": state.Node.Id,
1041                         "pkts": pkts,
1042                         "size": int64(size),
1043                 }, "")
1044         }
1045         return payloadsSplit(replies), nil
1046 }