src/cypherpunks.ru/nncp/sp.go

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