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