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