src/cypherpunks.ru/nncp/sp.go

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