src/cypherpunks.ru/nncp/sp.go

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