2 NNCP -- Node to Node copy, utilities for store-and-forward data exchange
3 Copyright (C) 2016-2021 Sergey Matveev <stargrave@stargrave.org>
5 This program is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, version 3 of the License.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program. If not, see <http://www.gnu.org/licenses/>.
33 xdr "github.com/davecgh/go-xdr/xdr2"
34 "github.com/dustin/go-humanize"
35 "github.com/flynn/noise"
36 "golang.org/x/crypto/blake2b"
40 MaxSPSize = 1<<16 - 256
45 type SPCheckerQueues struct {
46 appeared chan *[32]byte
47 checked chan *[32]byte
51 MagicNNCPLv1 [8]byte = [8]byte{'N', 'N', 'C', 'P', 'S', 0, 0, 1}
56 SPHaltMarshalized []byte
57 SPPingMarshalized []byte
59 NoiseCipherSuite noise.CipherSuite = noise.NewCipherSuite(
61 noise.CipherChaChaPoly,
65 DefaultDeadline = 10 * time.Second
66 PingTimeout = time.Minute
68 spCheckers = make(map[NodeId]*SPCheckerQueues)
71 type FdAndFullSize struct {
76 type HasherAndOffset struct {
84 SPTypeInfo SPType = iota
85 SPTypeFreq SPType = iota
86 SPTypeFile SPType = iota
87 SPTypeDone SPType = iota
88 SPTypeHalt SPType = iota
89 SPTypePing SPType = iota
122 type FreqWithNice struct {
127 type ConnDeadlined interface {
129 SetReadDeadline(t time.Time) error
130 SetWriteDeadline(t time.Time) error
135 spHead := SPHead{Type: SPTypeHalt}
136 if _, err := xdr.Marshal(&buf, spHead); err != nil {
139 SPHaltMarshalized = make([]byte, SPHeadOverhead)
140 copy(SPHaltMarshalized, buf.Bytes())
143 spHead = SPHead{Type: SPTypePing}
144 if _, err := xdr.Marshal(&buf, spHead); err != nil {
147 SPPingMarshalized = make([]byte, SPHeadOverhead)
148 copy(SPPingMarshalized, buf.Bytes())
151 spInfo := SPInfo{Nice: 123, Size: 123, Hash: new([32]byte)}
152 if _, err := xdr.Marshal(&buf, spInfo); err != nil {
155 SPInfoOverhead = buf.Len()
158 spFreq := SPFreq{Hash: new([32]byte), Offset: 123}
159 if _, err := xdr.Marshal(&buf, spFreq); err != nil {
162 SPFreqOverhead = buf.Len()
165 spFile := SPFile{Hash: new([32]byte), Offset: 123}
166 if _, err := xdr.Marshal(&buf, spFile); err != nil {
169 SPFileOverhead = buf.Len()
172 func MarshalSP(typ SPType, sp interface{}) []byte {
174 if _, err := xdr.Marshal(&buf, SPHead{typ}); err != nil {
177 if _, err := xdr.Marshal(&buf, sp); err != nil {
183 func payloadsSplit(payloads [][]byte) [][]byte {
184 var outbounds [][]byte
185 outbound := make([]byte, 0, MaxSPSize)
186 for i, payload := range payloads {
187 outbound = append(outbound, payload...)
188 if i+1 < len(payloads) && len(outbound)+len(payloads[i+1]) > MaxSPSize {
189 outbounds = append(outbounds, outbound)
190 outbound = make([]byte, 0, MaxSPSize)
193 if len(outbound) > 0 {
194 outbounds = append(outbounds, outbound)
199 type SPState struct {
204 onlineDeadline time.Duration
205 maxOnlineTime time.Duration
206 hs *noise.HandshakeState
207 csOur *noise.CipherState
208 csTheir *noise.CipherState
211 infosTheir map[[32]byte]*SPInfo
212 infosOurSeen map[[32]byte]uint8
213 queueTheir []*FreqWithNice
217 RxLastNonPing time.Time
220 TxLastNonPing time.Time
222 mustFinishAt time.Time
223 Duration time.Duration
233 onlyPkts map[[32]byte]bool
234 writeSPBuf bytes.Buffer
235 fds map[string]FdAndFullSize
237 fileHashers map[string]*HasherAndOffset
238 checkerQueues SPCheckerQueues
239 progressBars map[string]struct{}
243 func (state *SPState) SetDead() {
248 // Already closed channel, dead
254 for range state.payloads {
258 for range state.pings {
262 for _, s := range state.fds {
268 func (state *SPState) NotAlive() bool {
277 func (state *SPState) dirUnlock() {
278 state.Ctx.UnlockDir(state.rxLock)
279 state.Ctx.UnlockDir(state.txLock)
282 func SPChecker(ctx *Ctx, nodeId *NodeId, appeared, checked chan *[32]byte) {
283 for hshValue := range appeared {
284 pktName := Base32Codec.EncodeToString(hshValue[:])
290 ctx.LogD("sp-checker", les, func(les LEs) string {
291 return fmt.Sprintf("Checksumming %s/rx/%s", ctx.NodeName(nodeId), pktName)
293 size, err := ctx.CheckNoCK(nodeId, hshValue)
294 les = append(les, LE{"Size", size})
296 ctx.LogE("sp-checker", les, err, func(les LEs) string {
298 "Checksumming %s/rx/%s (%s)", ctx.NodeName(nodeId), pktName,
299 humanize.IBytes(uint64(size)),
304 ctx.LogI("sp-checker-done", les, func(les LEs) string {
306 "Packet %s is retreived (%s)",
307 pktName, humanize.IBytes(uint64(size)),
310 go func(hsh *[32]byte) { checked <- hsh }(hshValue)
314 func (state *SPState) WriteSP(dst io.Writer, payload []byte, ping bool) error {
315 state.writeSPBuf.Reset()
316 n, err := xdr.Marshal(&state.writeSPBuf, SPRaw{
323 if n, err = dst.Write(state.writeSPBuf.Bytes()); err == nil {
324 state.TxLastSeen = time.Now()
325 state.TxBytes += int64(n)
327 state.TxLastNonPing = state.TxLastSeen
333 func (state *SPState) ReadSP(src io.Reader) ([]byte, error) {
335 n, err := xdr.UnmarshalLimited(src, &sp, 1<<17)
337 ue := err.(*xdr.UnmarshalError)
338 if ue.Err == io.EOF {
343 state.RxLastSeen = time.Now()
344 state.RxBytes += int64(n)
345 if sp.Magic != MagicNNCPLv1 {
348 return sp.Payload, nil
351 func (ctx *Ctx) infosOur(nodeId *NodeId, nice uint8, seen *map[[32]byte]uint8) [][]byte {
354 for job := range ctx.Jobs(nodeId, TTx) {
355 if job.PktEnc.Nice > nice {
358 if _, known := (*seen)[*job.HshValue]; known {
361 totalSize += job.Size
362 infos = append(infos, &SPInfo{
363 Nice: job.PktEnc.Nice,
364 Size: uint64(job.Size),
367 (*seen)[*job.HshValue] = job.PktEnc.Nice
369 sort.Sort(ByNice(infos))
370 var payloads [][]byte
371 for _, info := range infos {
372 payloads = append(payloads, MarshalSP(SPTypeInfo, info))
373 pktName := Base32Codec.EncodeToString(info.Hash[:])
374 ctx.LogD("sp-info-our", LEs{
378 }, func(les LEs) string {
380 "Our info: %s/tx/%s (%s)",
381 ctx.NodeName(nodeId),
383 humanize.IBytes(info.Size),
388 ctx.LogI("sp-infos-tx", LEs{
391 {"Pkts", len(payloads)},
393 }, func(les LEs) string {
395 "We have got for %s: %d packets, %s",
396 ctx.NodeName(nodeId),
398 humanize.IBytes(uint64(totalSize)),
402 return payloadsSplit(payloads)
405 func (state *SPState) StartI(conn ConnDeadlined) error {
406 nodeId := state.Node.Id
407 err := state.Ctx.ensureRxDir(nodeId)
412 if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TRx) {
413 rxLock, err = state.Ctx.LockDir(nodeId, string(TRx))
419 if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TTx) {
420 txLock, err = state.Ctx.LockDir(nodeId, string(TTx))
425 started := time.Now()
426 conf := noise.Config{
427 CipherSuite: NoiseCipherSuite,
428 Pattern: noise.HandshakeIK,
430 StaticKeypair: noise.DHKey{
431 Private: state.Ctx.Self.NoisePrv[:],
432 Public: state.Ctx.Self.NoisePub[:],
434 PeerStatic: state.Node.NoisePub[:],
436 hs, err := noise.NewHandshakeState(conf)
441 state.payloads = make(chan []byte)
442 state.pings = make(chan struct{})
443 state.infosTheir = make(map[[32]byte]*SPInfo)
444 state.infosOurSeen = make(map[[32]byte]uint8)
445 state.progressBars = make(map[string]struct{})
446 state.started = started
447 state.rxLock = rxLock
448 state.txLock = txLock
450 var infosPayloads [][]byte
451 if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TTx) {
452 infosPayloads = state.Ctx.infosOur(nodeId, state.Nice, &state.infosOurSeen)
454 var firstPayload []byte
455 if len(infosPayloads) > 0 {
456 firstPayload = infosPayloads[0]
458 // Pad first payload, to hide actual number of existing files
459 for i := 0; i < (MaxSPSize-len(firstPayload))/SPHeadOverhead; i++ {
460 firstPayload = append(firstPayload, SPHaltMarshalized...)
465 buf, _, _, err = state.hs.WriteMessage(nil, firstPayload)
470 les := LEs{{"Node", nodeId}, {"Nice", int(state.Nice)}}
471 state.Ctx.LogD("sp-startI", les, func(les LEs) string {
473 "SP with %s (nice %s): sending first message",
475 NicenessFmt(state.Nice),
478 conn.SetWriteDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
479 if err = state.WriteSP(conn, buf, false); err != nil {
480 state.Ctx.LogE("sp-startI", les, err, func(les LEs) string {
482 "SP with %s (nice %s): writing",
484 NicenessFmt(state.Nice),
490 state.Ctx.LogD("sp-startI-wait", les, func(les LEs) string {
492 "SP with %s (nice %s): waiting for first message",
494 NicenessFmt(state.Nice),
497 conn.SetReadDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
498 if buf, err = state.ReadSP(conn); err != nil {
499 state.Ctx.LogE("sp-startI-read", les, err, func(les LEs) string {
501 "SP with %s (nice %s): reading",
503 NicenessFmt(state.Nice),
509 payload, state.csOur, state.csTheir, err = state.hs.ReadMessage(nil, buf)
511 state.Ctx.LogE("sp-startI-read", les, err, func(les LEs) string {
513 "SP with %s (nice %s): reading Noise message",
515 NicenessFmt(state.Nice),
521 state.Ctx.LogD("sp-startI-workers", les, func(les LEs) string {
523 "SP with %s (nice %s): starting workers",
525 NicenessFmt(state.Nice),
528 err = state.StartWorkers(conn, infosPayloads, payload)
530 state.Ctx.LogE("sp-startI-workers", les, err, func(les LEs) string {
532 "SP with %s (nice %s): starting workers",
534 NicenessFmt(state.Nice),
542 func (state *SPState) StartR(conn ConnDeadlined) error {
543 started := time.Now()
544 conf := noise.Config{
545 CipherSuite: NoiseCipherSuite,
546 Pattern: noise.HandshakeIK,
548 StaticKeypair: noise.DHKey{
549 Private: state.Ctx.Self.NoisePrv[:],
550 Public: state.Ctx.Self.NoisePub[:],
553 hs, err := noise.NewHandshakeState(conf)
559 state.payloads = make(chan []byte)
560 state.pings = make(chan struct{})
561 state.infosOurSeen = make(map[[32]byte]uint8)
562 state.infosTheir = make(map[[32]byte]*SPInfo)
563 state.progressBars = make(map[string]struct{})
564 state.started = started
565 state.xxOnly = xxOnly
569 logMsg := func(les LEs) string {
571 "SP nice %s: waiting for first message",
572 NicenessFmt(state.Nice),
575 les := LEs{{"Nice", int(state.Nice)}}
576 state.Ctx.LogD("sp-startR", les, logMsg)
577 conn.SetReadDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
578 if buf, err = state.ReadSP(conn); err != nil {
579 state.Ctx.LogE("sp-startR-read", les, err, logMsg)
582 if payload, _, _, err = state.hs.ReadMessage(nil, buf); err != nil {
583 state.Ctx.LogE("sp-startR-read", les, err, logMsg)
588 for _, n := range state.Ctx.Neigh {
589 if subtle.ConstantTimeCompare(state.hs.PeerStatic(), n.NoisePub[:]) == 1 {
595 peerId := Base32Codec.EncodeToString(state.hs.PeerStatic())
596 err = errors.New("unknown peer: " + peerId)
597 state.Ctx.LogE("sp-startR-unknown", append(les, LE{"Peer", peerId}), err, logMsg)
601 state.rxRate = node.RxRate
602 state.txRate = node.TxRate
603 state.onlineDeadline = node.OnlineDeadline
604 state.maxOnlineTime = node.MaxOnlineTime
605 les = LEs{{"Node", node.Id}, {"Nice", int(state.Nice)}}
607 if err = state.Ctx.ensureRxDir(node.Id); err != nil {
611 if xxOnly == "" || xxOnly == TRx {
612 rxLock, err = state.Ctx.LockDir(node.Id, string(TRx))
617 state.rxLock = rxLock
619 if xxOnly == "" || xxOnly == TTx {
620 txLock, err = state.Ctx.LockDir(node.Id, string(TTx))
625 state.txLock = txLock
627 var infosPayloads [][]byte
628 if xxOnly == "" || xxOnly == TTx {
629 infosPayloads = state.Ctx.infosOur(node.Id, state.Nice, &state.infosOurSeen)
631 var firstPayload []byte
632 if len(infosPayloads) > 0 {
633 firstPayload = infosPayloads[0]
635 // Pad first payload, to hide actual number of existing files
636 for i := 0; i < (MaxSPSize-len(firstPayload))/SPHeadOverhead; i++ {
637 firstPayload = append(firstPayload, SPHaltMarshalized...)
640 state.Ctx.LogD("sp-startR-write", les, func(les LEs) string {
642 "SP with %s (nice %s): sending first message",
643 node.Name, NicenessFmt(state.Nice),
646 buf, state.csTheir, state.csOur, err = state.hs.WriteMessage(nil, firstPayload)
651 conn.SetWriteDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
652 if err = state.WriteSP(conn, buf, false); err != nil {
653 state.Ctx.LogE("sp-startR-write", les, err, func(les LEs) string {
655 "SP with %s (nice %s): writing",
656 node.Name, NicenessFmt(state.Nice),
662 state.Ctx.LogD("sp-startR-workers", les, func(les LEs) string {
664 "SP with %s (nice %s): starting workers",
665 node.Name, NicenessFmt(state.Nice),
668 err = state.StartWorkers(conn, infosPayloads, payload)
675 func (state *SPState) closeFd(pth string) {
677 if s, exists := state.fds[pth]; exists {
678 delete(state.fds, pth)
681 state.fdsLock.Unlock()
684 func (state *SPState) StartWorkers(
686 infosPayloads [][]byte,
689 les := LEs{{"Node", state.Node.Id}, {"Nice", int(state.Nice)}}
690 state.fds = make(map[string]FdAndFullSize)
691 state.fileHashers = make(map[string]*HasherAndOffset)
692 state.isDead = make(chan struct{})
693 if state.maxOnlineTime > 0 {
694 state.mustFinishAt = state.started.Add(state.maxOnlineTime)
699 queues := spCheckers[*state.Node.Id]
701 queues = &SPCheckerQueues{
702 appeared: make(chan *[32]byte),
703 checked: make(chan *[32]byte),
705 spCheckers[*state.Node.Id] = queues
706 go SPChecker(state.Ctx, state.Node.Id, queues.appeared, queues.checked)
708 state.checkerQueues = *queues
710 for job := range state.Ctx.JobsNoCK(state.Node.Id) {
711 if job.PktEnc.Nice <= state.Nice {
712 state.checkerQueues.appeared <- job.HshValue
718 defer state.wg.Done()
723 case hsh := <-state.checkerQueues.checked:
724 state.payloads <- MarshalSP(SPTypeDone, SPDone{hsh})
730 // Remaining handshake payload sending
731 if len(infosPayloads) > 1 {
734 for _, payload := range infosPayloads[1:] {
736 "sp-queue-remaining",
737 append(les, LE{"Size", int64(len(payload))}),
738 func(les LEs) string {
740 "SP with %s (nice %s): queuing remaining payload (%s)",
741 state.Node.Name, NicenessFmt(state.Nice),
742 humanize.IBytes(uint64(len(payload))),
746 state.payloads <- payload
752 // Processing of first payload and queueing its responses
753 logMsg := func(les LEs) string {
755 "SP with %s (nice %s): processing first payload (%s)",
756 state.Node.Name, NicenessFmt(state.Nice),
757 humanize.IBytes(uint64(len(payload))),
760 state.Ctx.LogD("sp-process", append(les, LE{"Size", int64(len(payload))}), logMsg)
761 replies, err := state.ProcessSP(payload)
763 state.Ctx.LogE("sp-process", les, err, logMsg)
768 for _, reply := range replies {
771 append(les, LE{"Size", int64(len(reply))}),
772 func(les LEs) string {
774 "SP with %s (nice %s): queuing reply (%s)",
775 state.Node.Name, NicenessFmt(state.Nice),
776 humanize.IBytes(uint64(len(payload))),
780 state.payloads <- reply
788 deadlineTicker := time.NewTicker(time.Second)
789 pingTicker := time.NewTicker(PingTimeout)
794 deadlineTicker.Stop()
797 case now := <-deadlineTicker.C:
798 if now.Sub(state.RxLastNonPing) >= state.onlineDeadline &&
799 now.Sub(state.TxLastNonPing) >= state.onlineDeadline {
802 if state.maxOnlineTime > 0 && state.mustFinishAt.Before(now) {
805 if now.Sub(state.RxLastSeen) >= 2*PingTimeout {
811 conn.Close() // #nosec G104
812 case now := <-pingTicker.C:
813 if now.After(state.TxLastSeen.Add(PingTimeout)) {
816 state.pings <- struct{}{}
824 // Spool checker and INFOs sender of appearing files
825 if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TTx) {
828 ticker := time.NewTicker(time.Second)
836 for _, payload := range state.Ctx.infosOur(
843 append(les, LE{"Size", int64(len(payload))}),
844 func(les LEs) string {
846 "SP with %s (nice %s): queuing new info (%s)",
847 state.Node.Name, NicenessFmt(state.Nice),
848 humanize.IBytes(uint64(len(payload))),
852 state.payloads <- payload
863 defer state.SetDead()
864 defer state.wg.Done()
866 if state.NotAlive() {
873 state.Ctx.LogD("sp-got-ping", les, func(les LEs) string {
875 "SP with %s (nice %s): got ping",
876 state.Node.Name, NicenessFmt(state.Nice),
879 payload = SPPingMarshalized
881 case payload = <-state.payloads:
884 append(les, LE{"Size", int64(len(payload))}),
885 func(les LEs) string {
887 "SP with %s (nice %s): got payload (%s)",
888 state.Node.Name, NicenessFmt(state.Nice),
889 humanize.IBytes(uint64(len(payload))),
895 if len(state.queueTheir) == 0 {
897 time.Sleep(100 * time.Millisecond)
900 freq := state.queueTheir[0].freq
902 if state.txRate > 0 {
903 time.Sleep(time.Second / time.Duration(state.txRate))
905 pktName := Base32Codec.EncodeToString(freq.Hash[:])
908 LE{"XX", string(TTx)},
910 LE{"Size", int64(freq.Offset)},
912 logMsg := func(les LEs) string {
914 "SP with %s (nice %s): tx/%s (%s)",
915 state.Node.Name, NicenessFmt(state.Nice),
917 humanize.IBytes(freq.Offset),
920 state.Ctx.LogD("sp-queue", lesp, func(les LEs) string {
921 return logMsg(les) + ": queueing"
923 pth := filepath.Join(
925 state.Node.Id.String(),
927 Base32Codec.EncodeToString(freq.Hash[:]),
929 state.fdsLock.RLock()
930 fdAndFullSize, exists := state.fds[pth]
931 state.fdsLock.RUnlock()
933 fd, err := os.Open(pth)
935 state.Ctx.LogE("sp-queue-open", lesp, err, func(les LEs) string {
936 return logMsg(les) + ": opening"
942 state.Ctx.LogE("sp-queue-stat", lesp, err, func(les LEs) string {
943 return logMsg(les) + ": stating"
947 fdAndFullSize = FdAndFullSize{fd: fd, fullSize: fi.Size()}
949 state.fds[pth] = fdAndFullSize
950 state.fdsLock.Unlock()
952 fd := fdAndFullSize.fd
953 fullSize := fdAndFullSize.fullSize
955 if freq.Offset < uint64(fullSize) {
956 state.Ctx.LogD("sp-file-seek", lesp, func(les LEs) string {
957 return logMsg(les) + ": seeking"
959 if _, err = fd.Seek(int64(freq.Offset), io.SeekStart); err != nil {
960 state.Ctx.LogE("sp-file-seek", lesp, err, func(les LEs) string {
961 return logMsg(les) + ": seeking"
965 buf = make([]byte, MaxSPSize-SPHeadOverhead-SPFileOverhead)
966 n, err := fd.Read(buf)
968 state.Ctx.LogE("sp-file-read", lesp, err, func(les LEs) string {
969 return logMsg(les) + ": reading"
976 LE{"XX", string(TTx)},
978 LE{"Size", int64(n)},
980 state.Ctx.LogD("sp-file-read", lesp, func(les LEs) string {
983 logMsg(les), humanize.IBytes(uint64(n)),
988 payload = MarshalSP(SPTypeFile, SPFile{
993 ourSize := freq.Offset + uint64(len(buf))
996 LE{"XX", string(TTx)},
998 LE{"Size", int64(ourSize)},
999 LE{"FullSize", fullSize},
1001 if state.Ctx.ShowPrgrs {
1002 state.progressBars[pktName] = struct{}{}
1003 Progress("Tx", lesp)
1006 if len(state.queueTheir) > 0 && *state.queueTheir[0].freq.Hash == *freq.Hash {
1007 if ourSize == uint64(fullSize) {
1008 state.Ctx.LogD("sp-file-finished", lesp, func(les LEs) string {
1009 return logMsg(les) + ": finished"
1011 if len(state.queueTheir) > 1 {
1012 state.queueTheir = state.queueTheir[1:]
1014 state.queueTheir = state.queueTheir[:0]
1016 if state.Ctx.ShowPrgrs {
1017 delete(state.progressBars, pktName)
1020 state.queueTheir[0].freq.Offset += uint64(len(buf))
1023 state.Ctx.LogD("sp-file-disappeared", lesp, func(les LEs) string {
1024 return logMsg(les) + ": queue disappeared"
1029 logMsg := func(les LEs) string {
1031 "SP with %s (nice %s): sending %s",
1032 state.Node.Name, NicenessFmt(state.Nice),
1033 humanize.IBytes(uint64(len(payload))),
1036 state.Ctx.LogD("sp-sending", append(les, LE{"Size", int64(len(payload))}), logMsg)
1037 conn.SetWriteDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
1038 if err := state.WriteSP(conn, state.csOur.Encrypt(nil, nil, payload), ping); err != nil {
1039 state.Ctx.LogE("sp-sending", les, err, logMsg)
1049 if state.NotAlive() {
1052 logMsg := func(les LEs) string {
1054 "SP with %s (nice %s): waiting for payload",
1055 state.Node.Name, NicenessFmt(state.Nice),
1058 state.Ctx.LogD("sp-recv-wait", les, logMsg)
1059 conn.SetReadDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
1060 payload, err := state.ReadSP(conn)
1065 unmarshalErr := err.(*xdr.UnmarshalError)
1066 if os.IsTimeout(unmarshalErr.Err) {
1069 if unmarshalErr.ErrorCode == xdr.ErrIO {
1072 state.Ctx.LogE("sp-recv-wait", les, err, logMsg)
1075 logMsg = func(les LEs) string {
1077 "SP with %s (nice %s): payload (%s)",
1078 state.Node.Name, NicenessFmt(state.Nice),
1079 humanize.IBytes(uint64(len(payload))),
1084 append(les, LE{"Size", int64(len(payload))}),
1085 func(les LEs) string { return logMsg(les) + ": got" },
1087 payload, err = state.csTheir.Decrypt(nil, nil, payload)
1089 state.Ctx.LogE("sp-recv-got", les, err, func(les LEs) string {
1090 return logMsg(les) + ": got"
1096 append(les, LE{"Size", int64(len(payload))}),
1097 func(les LEs) string {
1098 return logMsg(les) + ": processing"
1101 replies, err := state.ProcessSP(payload)
1103 state.Ctx.LogE("sp-recv-process", les, err, func(les LEs) string {
1104 return logMsg(les) + ": processing"
1110 for _, reply := range replies {
1113 append(les[:len(les)-1], LE{"Size", int64(len(reply))}),
1114 func(les LEs) string {
1116 "SP with %s (nice %s): queuing reply (%s)",
1117 state.Node.Name, NicenessFmt(state.Nice),
1118 humanize.IBytes(uint64(len(reply))),
1122 state.payloads <- reply
1126 if state.rxRate > 0 {
1127 time.Sleep(time.Second / time.Duration(state.rxRate))
1133 conn.Close() // #nosec G104
1139 func (state *SPState) Wait() {
1141 close(state.payloads)
1144 state.Duration = time.Now().Sub(state.started)
1145 state.RxSpeed = state.RxBytes
1146 state.TxSpeed = state.TxBytes
1147 rxDuration := int64(state.RxLastSeen.Sub(state.started).Seconds())
1148 txDuration := int64(state.TxLastSeen.Sub(state.started).Seconds())
1150 state.RxSpeed = state.RxBytes / rxDuration
1153 state.TxSpeed = state.TxBytes / txDuration
1155 for pktName := range state.progressBars {
1156 ProgressKill(pktName)
1160 func (state *SPState) ProcessSP(payload []byte) ([][]byte, error) {
1161 les := LEs{{"Node", state.Node.Id}, {"Nice", int(state.Nice)}}
1162 r := bytes.NewReader(payload)
1164 var replies [][]byte
1167 state.Ctx.LogD("sp-process-unmarshal", les, func(les LEs) string {
1169 "SP with %s (nice %s): unmarshaling header",
1170 state.Node.Name, NicenessFmt(state.Nice),
1174 if _, err = xdr.Unmarshal(r, &head); err != nil {
1175 state.Ctx.LogE("sp-process-unmarshal", les, err, func(les LEs) string {
1177 "SP with %s (nice %s): unmarshaling header",
1178 state.Node.Name, NicenessFmt(state.Nice),
1183 if head.Type != SPTypePing {
1184 state.RxLastNonPing = state.RxLastSeen
1190 append(les, LE{"Type", "halt"}), func(les LEs) string {
1192 "SP with %s (nice %s): got HALT",
1193 state.Node.Name, NicenessFmt(state.Nice),
1198 state.queueTheir = nil
1204 append(les, LE{"Type", "ping"}),
1205 func(les LEs) string {
1207 "SP with %s (nice %s): got PING",
1208 state.Node.Name, NicenessFmt(state.Nice),
1215 lesp := append(les, LE{"Type", "info"})
1217 "sp-process-info-unmarshal", lesp,
1218 func(les LEs) string {
1220 "SP with %s (nice %s): unmarshaling INFO",
1221 state.Node.Name, NicenessFmt(state.Nice),
1226 if _, err = xdr.Unmarshal(r, &info); err != nil {
1228 "sp-process-info-unmarshal", lesp, err,
1229 func(les LEs) string {
1231 "SP with %s (nice %s): unmarshaling INFO",
1232 state.Node.Name, NicenessFmt(state.Nice),
1238 pktName := Base32Codec.EncodeToString(info.Hash[:])
1242 LE{"Size", int64(info.Size)},
1243 LE{"PktNice", int(info.Nice)},
1245 logMsg := func(les LEs) string {
1247 "SP with %s (nice %s): INFO %s (%s) nice %s",
1248 state.Node.Name, NicenessFmt(state.Nice),
1250 humanize.IBytes(info.Size),
1251 NicenessFmt(info.Nice),
1254 if !state.listOnly && info.Nice > state.Nice {
1255 state.Ctx.LogD("sp-process-info-too-nice", lesp, func(les LEs) string {
1256 return logMsg(les) + ": too nice"
1260 state.Ctx.LogD("sp-process-info-got", lesp, func(les LEs) string {
1261 return logMsg(les) + ": received"
1263 if !state.listOnly && state.xxOnly == TTx {
1267 state.infosTheir[*info.Hash] = &info
1269 state.Ctx.LogD("sp-process-info-stat", lesp, func(les LEs) string {
1270 return logMsg(les) + ": stating part"
1272 pktPath := filepath.Join(
1274 state.Node.Id.String(),
1276 Base32Codec.EncodeToString(info.Hash[:]),
1278 logMsg = func(les LEs) string {
1280 "Packet %s (%s) (nice %s)",
1282 humanize.IBytes(info.Size),
1283 NicenessFmt(info.Nice),
1286 if _, err = os.Stat(pktPath); err == nil {
1287 state.Ctx.LogI("sp-info-done", lesp, func(les LEs) string {
1288 return logMsg(les) + ": already done"
1290 if !state.listOnly {
1291 replies = append(replies, MarshalSP(SPTypeDone, SPDone{info.Hash}))
1295 if _, err = os.Stat(pktPath + SeenSuffix); err == nil {
1296 state.Ctx.LogI("sp-info-seen", lesp, func(les LEs) string {
1297 return logMsg(les) + ": already seen"
1299 if !state.listOnly {
1300 replies = append(replies, MarshalSP(SPTypeDone, SPDone{info.Hash}))
1304 if _, err = os.Stat(pktPath + NoCKSuffix); err == nil {
1305 state.Ctx.LogI("sp-info-nock", lesp, func(les LEs) string {
1306 return logMsg(les) + ": still not checksummed"
1310 fi, err := os.Stat(pktPath + PartSuffix)
1315 if !state.Ctx.IsEnoughSpace(int64(info.Size) - offset) {
1316 state.Ctx.LogI("sp-info-no-space", lesp, func(les LEs) string {
1317 return logMsg(les) + ": not enough space"
1323 append(lesp, LE{"Offset", offset}),
1324 func(les LEs) string {
1326 "%s: %d%%", logMsg(les), 100*uint64(offset)/info.Size,
1330 if !state.listOnly && (state.onlyPkts == nil || state.onlyPkts[*info.Hash]) {
1331 replies = append(replies, MarshalSP(
1333 SPFreq{info.Hash, uint64(offset)},
1338 lesp := append(les, LE{"Type", "file"})
1339 state.Ctx.LogD("sp-process-file", lesp, func(les LEs) string {
1341 "SP with %s (nice %s): unmarshaling FILE",
1342 state.Node.Name, NicenessFmt(state.Nice),
1346 if _, err = xdr.Unmarshal(r, &file); err != nil {
1347 state.Ctx.LogE("sp-process-file", lesp, err, func(les LEs) string {
1349 "SP with %s (nice %s): unmarshaling FILE",
1350 state.Node.Name, NicenessFmt(state.Nice),
1355 pktName := Base32Codec.EncodeToString(file.Hash[:])
1358 LE{"XX", string(TRx)},
1360 LE{"Size", int64(len(file.Payload))},
1362 logMsg := func(les LEs) string {
1364 "Got packet %s (%s)",
1365 pktName, humanize.IBytes(uint64(len(file.Payload))),
1368 dirToSync := filepath.Join(
1370 state.Node.Id.String(),
1373 filePath := filepath.Join(dirToSync, pktName)
1374 filePathPart := filePath + PartSuffix
1375 state.Ctx.LogD("sp-file-open", lesp, func(les LEs) string {
1376 return logMsg(les) + ": opening part"
1378 state.fdsLock.RLock()
1379 fdAndFullSize, exists := state.fds[filePathPart]
1380 state.fdsLock.RUnlock()
1383 fd = fdAndFullSize.fd
1385 fd, err = os.OpenFile(
1387 os.O_RDWR|os.O_CREATE,
1391 state.Ctx.LogE("sp-file-open", lesp, err, func(les LEs) string {
1392 return logMsg(les) + ": opening part"
1396 state.fdsLock.Lock()
1397 state.fds[filePathPart] = FdAndFullSize{fd: fd}
1398 state.fdsLock.Unlock()
1399 if file.Offset == 0 {
1400 h, err := blake2b.New256(nil)
1404 state.fileHashers[filePath] = &HasherAndOffset{h: h}
1409 append(lesp, LE{"Offset", file.Offset}),
1410 func(les LEs) string {
1411 return fmt.Sprintf("%s: seeking %d", logMsg(les), file.Offset)
1413 if _, err = fd.Seek(int64(file.Offset), io.SeekStart); err != nil {
1414 state.Ctx.LogE("sp-file-seek", lesp, err, func(les LEs) string {
1415 return logMsg(les) + ": seeking"
1417 state.closeFd(filePathPart)
1420 state.Ctx.LogD("sp-file-write", lesp, func(les LEs) string {
1421 return logMsg(les) + ": writing"
1423 if _, err = fd.Write(file.Payload); err != nil {
1424 state.Ctx.LogE("sp-file-write", lesp, err, func(les LEs) string {
1425 return logMsg(les) + ": writing"
1427 state.closeFd(filePathPart)
1430 hasherAndOffset, hasherExists := state.fileHashers[filePath]
1432 if hasherAndOffset.offset == file.Offset {
1433 if _, err = hasherAndOffset.h.Write(file.Payload); err != nil {
1436 hasherAndOffset.offset += uint64(len(file.Payload))
1439 "sp-file-offset-differs", lesp,
1440 func(les LEs) string {
1441 return logMsg(les) + ": offset differs, deleting hasher"
1444 delete(state.fileHashers, filePath)
1445 hasherExists = false
1448 ourSize := int64(file.Offset + uint64(len(file.Payload)))
1449 lesp[len(lesp)-1].V = ourSize
1450 fullsize := int64(0)
1452 infoTheir, ok := state.infosTheir[*file.Hash]
1455 fullsize = int64(infoTheir.Size)
1457 lesp = append(lesp, LE{"FullSize", fullsize})
1458 if state.Ctx.ShowPrgrs {
1459 state.progressBars[pktName] = struct{}{}
1460 Progress("Rx", lesp)
1462 if fullsize != ourSize {
1465 if state.Ctx.ShowPrgrs {
1466 delete(state.progressBars, pktName)
1468 logMsg = func(les LEs) string {
1470 "Got packet %s %d%% (%s / %s)",
1471 pktName, 100*ourSize/fullsize,
1472 humanize.IBytes(uint64(ourSize)),
1473 humanize.IBytes(uint64(fullsize)),
1478 state.Ctx.LogE("sp-file-sync", lesp, err, func(les LEs) string {
1479 return logMsg(les) + ": syncing"
1481 state.closeFd(filePathPart)
1485 if bytes.Compare(hasherAndOffset.h.Sum(nil), file.Hash[:]) != 0 {
1487 "sp-file-bad-checksum", lesp,
1488 errors.New("checksum mismatch"),
1493 if err = os.Rename(filePathPart, filePath); err != nil {
1494 state.Ctx.LogE("sp-file-rename", lesp, err, func(les LEs) string {
1495 return logMsg(les) + ": renaming"
1499 if err = DirSync(dirToSync); err != nil {
1500 state.Ctx.LogE("sp-file-dirsync", lesp, err, func(les LEs) string {
1501 return logMsg(les) + ": dirsyncing"
1505 state.Ctx.LogI("sp-file-done", lesp, func(les LEs) string {
1506 return logMsg(les) + ": done"
1510 state.payloads <- MarshalSP(SPTypeDone, SPDone{file.Hash})
1514 delete(state.infosTheir, *file.Hash)
1516 if !state.Ctx.HdrUsage {
1517 state.closeFd(filePathPart)
1520 if _, err = fd.Seek(0, io.SeekStart); err != nil {
1521 state.Ctx.LogE("sp-file-seek", lesp, err, func(les LEs) string {
1522 return logMsg(les) + ": seeking"
1524 state.closeFd(filePathPart)
1527 _, pktEncRaw, err := state.Ctx.HdrRead(fd)
1528 state.closeFd(filePathPart)
1530 state.Ctx.LogE("sp-file-hdr-read", lesp, err, func(les LEs) string {
1531 return logMsg(les) + ": HdrReading"
1535 state.Ctx.HdrWrite(pktEncRaw, filePath)
1538 state.closeFd(filePathPart)
1539 if err = os.Rename(filePathPart, filePath+NoCKSuffix); err != nil {
1540 state.Ctx.LogE("sp-file-rename", lesp, err, func(les LEs) string {
1541 return logMsg(les) + ": renaming"
1545 if err = DirSync(dirToSync); err != nil {
1546 state.Ctx.LogE("sp-file-dirsync", lesp, err, func(les LEs) string {
1547 return logMsg(les) + ": dirsyncing"
1551 state.Ctx.LogI("sp-file-downloaded", lesp, func(les LEs) string {
1552 return logMsg(les) + ": downloaded"
1555 delete(state.infosTheir, *file.Hash)
1558 state.checkerQueues.appeared <- file.Hash
1562 lesp := append(les, LE{"Type", "done"})
1563 state.Ctx.LogD("sp-process-done-unmarshal", lesp, func(les LEs) string {
1565 "SP with %s (nice %s): unmarshaling DONE",
1566 state.Node.Name, NicenessFmt(state.Nice),
1570 if _, err = xdr.Unmarshal(r, &done); err != nil {
1571 state.Ctx.LogE("sp-process-done-unmarshal", lesp, err, func(les LEs) string {
1573 "SP with %s (nice %s): unmarshaling DONE",
1574 state.Node.Name, NicenessFmt(state.Nice),
1579 pktName := Base32Codec.EncodeToString(done.Hash[:])
1580 lesp = append(lesp, LE{"Pkt", pktName}, LE{"XX", string(TTx)})
1581 logMsg := func(les LEs) string {
1583 "SP with %s (nice %s): DONE: removing %s",
1584 state.Node.Name, NicenessFmt(state.Nice), pktName,
1587 state.Ctx.LogD("sp-done", lesp, logMsg)
1588 pth := filepath.Join(
1590 state.Node.Id.String(),
1594 if err = os.Remove(pth); err == nil {
1595 state.Ctx.LogI("sp-done", lesp, func(les LEs) string {
1596 return fmt.Sprintf("Packet %s is sent", pktName)
1598 if state.Ctx.HdrUsage {
1599 os.Remove(pth + HdrSuffix)
1602 state.Ctx.LogE("sp-done", lesp, err, logMsg)
1606 lesp := append(les, LE{"Type", "freq"})
1607 state.Ctx.LogD("sp-process-freq", lesp, func(les LEs) string {
1609 "SP with %s (nice %s): unmarshaling FREQ",
1610 state.Node.Name, NicenessFmt(state.Nice),
1614 if _, err = xdr.Unmarshal(r, &freq); err != nil {
1615 state.Ctx.LogE("sp-process-freq", lesp, err, func(les LEs) string {
1617 "SP with %s (nice %s): unmarshaling FREQ",
1618 state.Node.Name, NicenessFmt(state.Nice),
1623 pktName := Base32Codec.EncodeToString(freq.Hash[:])
1624 lesp = append(lesp, LE{"Pkt", pktName}, LE{"Offset", freq.Offset})
1625 state.Ctx.LogD("sp-process-freq-queueing", lesp, func(les LEs) string {
1627 "SP with %s (nice %s): FREQ %s: queuing",
1628 state.Node.Name, NicenessFmt(state.Nice), pktName,
1631 nice, exists := state.infosOurSeen[*freq.Hash]
1633 if state.onlyPkts == nil || !state.onlyPkts[*freq.Hash] {
1636 var freqWithNice *FreqWithNice
1637 for insertIdx, freqWithNice = range state.queueTheir {
1638 if freqWithNice.nice > nice {
1642 state.queueTheir = append(state.queueTheir, nil)
1643 copy(state.queueTheir[insertIdx+1:], state.queueTheir[insertIdx:])
1644 state.queueTheir[insertIdx] = &FreqWithNice{&freq, nice}
1647 state.Ctx.LogD("sp-process-freq-skip", lesp, func(les LEs) string {
1649 "SP with %s (nice %s): FREQ %s: skipping",
1650 state.Node.Name, NicenessFmt(state.Nice), pktName,
1655 state.Ctx.LogD("sp-process-freq-unknown", lesp, func(les LEs) string {
1657 "SP with %s (nice %s): FREQ %s: unknown",
1658 state.Node.Name, NicenessFmt(state.Nice), pktName,
1665 "sp-process-type-unknown",
1666 append(les, LE{"Type", head.Type}),
1667 errors.New("unknown type"),
1668 func(les LEs) string {
1670 "SP with %s (nice %s): %d",
1671 state.Node.Name, NicenessFmt(state.Nice), head.Type,
1675 return nil, BadPktType
1683 for _, info := range state.infosTheir {
1688 state.Ctx.LogI("sp-infos-rx", LEs{
1689 {"XX", string(TRx)},
1690 {"Node", state.Node.Id},
1692 {"Size", int64(size)},
1693 }, func(les LEs) string {
1695 "%s has got for us: %d packets, %s",
1696 state.Node.Name, pkts, humanize.IBytes(size),
1700 return payloadsSplit(replies), nil