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