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)
69 SPCheckersWg sync.WaitGroup
72 type FdAndFullSize struct {
77 type HasherAndOffset struct {
85 SPTypeInfo SPType = iota
86 SPTypeFreq SPType = iota
87 SPTypeFile SPType = iota
88 SPTypeDone SPType = iota
89 SPTypeHalt SPType = iota
90 SPTypePing SPType = iota
123 type FreqWithNice struct {
128 type ConnDeadlined interface {
130 SetReadDeadline(t time.Time) error
131 SetWriteDeadline(t time.Time) error
136 spHead := SPHead{Type: SPTypeHalt}
137 if _, err := xdr.Marshal(&buf, spHead); err != nil {
140 SPHaltMarshalized = make([]byte, SPHeadOverhead)
141 copy(SPHaltMarshalized, buf.Bytes())
144 spHead = SPHead{Type: SPTypePing}
145 if _, err := xdr.Marshal(&buf, spHead); err != nil {
148 SPPingMarshalized = make([]byte, SPHeadOverhead)
149 copy(SPPingMarshalized, buf.Bytes())
152 spInfo := SPInfo{Nice: 123, Size: 123, Hash: new([32]byte)}
153 if _, err := xdr.Marshal(&buf, spInfo); err != nil {
156 SPInfoOverhead = buf.Len()
159 spFreq := SPFreq{Hash: new([32]byte), Offset: 123}
160 if _, err := xdr.Marshal(&buf, spFreq); err != nil {
163 SPFreqOverhead = buf.Len()
166 spFile := SPFile{Hash: new([32]byte), Offset: 123}
167 if _, err := xdr.Marshal(&buf, spFile); err != nil {
170 SPFileOverhead = buf.Len()
173 func MarshalSP(typ SPType, sp interface{}) []byte {
175 if _, err := xdr.Marshal(&buf, SPHead{typ}); err != nil {
178 if _, err := xdr.Marshal(&buf, sp); err != nil {
184 func payloadsSplit(payloads [][]byte) [][]byte {
185 var outbounds [][]byte
186 outbound := make([]byte, 0, MaxSPSize)
187 for i, payload := range payloads {
188 outbound = append(outbound, payload...)
189 if i+1 < len(payloads) && len(outbound)+len(payloads[i+1]) > MaxSPSize {
190 outbounds = append(outbounds, outbound)
191 outbound = make([]byte, 0, MaxSPSize)
194 if len(outbound) > 0 {
195 outbounds = append(outbounds, outbound)
200 type SPState struct {
205 onlineDeadline time.Duration
206 maxOnlineTime time.Duration
207 hs *noise.HandshakeState
208 csOur *noise.CipherState
209 csTheir *noise.CipherState
212 infosTheir map[[32]byte]*SPInfo
213 infosOurSeen map[[32]byte]uint8
214 queueTheir []*FreqWithNice
218 RxLastNonPing time.Time
221 TxLastNonPing time.Time
223 mustFinishAt time.Time
224 Duration time.Duration
234 onlyPkts map[[32]byte]bool
235 writeSPBuf bytes.Buffer
236 fds map[string]FdAndFullSize
238 fileHashers map[string]*HasherAndOffset
239 checkerQueues SPCheckerQueues
240 progressBars map[string]struct{}
244 func (state *SPState) SetDead() {
249 // Already closed channel, dead
255 for range state.payloads {
259 for range state.pings {
264 func (state *SPState) NotAlive() bool {
273 func (state *SPState) dirUnlock() {
274 state.Ctx.UnlockDir(state.rxLock)
275 state.Ctx.UnlockDir(state.txLock)
278 func SPChecker(ctx *Ctx, nodeId *NodeId, appeared, checked chan *[32]byte) {
279 for hshValue := range appeared {
280 pktName := Base32Codec.EncodeToString(hshValue[:])
287 ctx.LogD("sp-checker", les, func(les LEs) string {
288 return fmt.Sprintf("Checksumming %s/rx/%s", ctx.NodeName(nodeId), pktName)
290 size, err := ctx.CheckNoCK(nodeId, hshValue)
291 les = append(les, LE{"Size", size})
293 ctx.LogE("sp-checker", les, err, func(les LEs) string {
295 "Checksumming %s/rx/%s (%s)", ctx.NodeName(nodeId), pktName,
296 humanize.IBytes(uint64(size)),
301 ctx.LogI("sp-checker-done", les, func(les LEs) string {
303 "Packet %s is retreived (%s)",
304 pktName, humanize.IBytes(uint64(size)),
308 go func(hsh *[32]byte) { checked <- hsh }(hshValue)
312 func (state *SPState) WriteSP(dst io.Writer, payload []byte, ping bool) error {
313 state.writeSPBuf.Reset()
314 n, err := xdr.Marshal(&state.writeSPBuf, SPRaw{
321 if n, err = dst.Write(state.writeSPBuf.Bytes()); err == nil {
322 state.TxLastSeen = time.Now()
323 state.TxBytes += int64(n)
325 state.TxLastNonPing = state.TxLastSeen
331 func (state *SPState) ReadSP(src io.Reader) ([]byte, error) {
333 n, err := xdr.UnmarshalLimited(src, &sp, 1<<17)
335 ue := err.(*xdr.UnmarshalError)
336 if ue.Err == io.EOF {
341 state.RxLastSeen = time.Now()
342 state.RxBytes += int64(n)
343 if sp.Magic != MagicNNCPLv1 {
346 return sp.Payload, nil
349 func (ctx *Ctx) infosOur(nodeId *NodeId, nice uint8, seen *map[[32]byte]uint8) [][]byte {
352 for job := range ctx.Jobs(nodeId, TTx) {
353 if job.PktEnc.Nice > nice {
356 if _, known := (*seen)[*job.HshValue]; known {
359 totalSize += job.Size
360 infos = append(infos, &SPInfo{
361 Nice: job.PktEnc.Nice,
362 Size: uint64(job.Size),
365 (*seen)[*job.HshValue] = job.PktEnc.Nice
367 sort.Sort(ByNice(infos))
368 var payloads [][]byte
369 for _, info := range infos {
370 payloads = append(payloads, MarshalSP(SPTypeInfo, info))
371 pktName := Base32Codec.EncodeToString(info.Hash[:])
372 ctx.LogD("sp-info-our", LEs{
376 }, func(les LEs) string {
378 "Our info: %s/tx/%s (%s)",
379 ctx.NodeName(nodeId),
381 humanize.IBytes(info.Size),
386 ctx.LogI("sp-infos-tx", LEs{
389 {"Pkts", len(payloads)},
391 }, func(les LEs) string {
393 "We have got for %s: %d packets, %s",
394 ctx.NodeName(nodeId),
396 humanize.IBytes(uint64(totalSize)),
400 return payloadsSplit(payloads)
403 func (state *SPState) StartI(conn ConnDeadlined) error {
404 nodeId := state.Node.Id
405 err := state.Ctx.ensureRxDir(nodeId)
410 if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TRx) {
411 rxLock, err = state.Ctx.LockDir(nodeId, string(TRx))
417 if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TTx) {
418 txLock, err = state.Ctx.LockDir(nodeId, string(TTx))
423 started := time.Now()
424 conf := noise.Config{
425 CipherSuite: NoiseCipherSuite,
426 Pattern: noise.HandshakeIK,
428 StaticKeypair: noise.DHKey{
429 Private: state.Ctx.Self.NoisePrv[:],
430 Public: state.Ctx.Self.NoisePub[:],
432 PeerStatic: state.Node.NoisePub[:],
434 hs, err := noise.NewHandshakeState(conf)
439 state.payloads = make(chan []byte)
440 state.pings = make(chan struct{})
441 state.infosTheir = make(map[[32]byte]*SPInfo)
442 state.infosOurSeen = make(map[[32]byte]uint8)
443 state.progressBars = make(map[string]struct{})
444 state.started = started
445 state.rxLock = rxLock
446 state.txLock = txLock
448 var infosPayloads [][]byte
449 if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TTx) {
450 infosPayloads = state.Ctx.infosOur(nodeId, state.Nice, &state.infosOurSeen)
452 var firstPayload []byte
453 if len(infosPayloads) > 0 {
454 firstPayload = infosPayloads[0]
456 // Pad first payload, to hide actual number of existing files
457 for i := 0; i < (MaxSPSize-len(firstPayload))/SPHeadOverhead; i++ {
458 firstPayload = append(firstPayload, SPHaltMarshalized...)
463 buf, _, _, err = state.hs.WriteMessage(nil, firstPayload)
468 les := LEs{{"Node", nodeId}, {"Nice", int(state.Nice)}}
469 state.Ctx.LogD("sp-startI", les, func(les LEs) string {
471 "SP with %s (nice %s): sending first message",
473 NicenessFmt(state.Nice),
476 conn.SetWriteDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
477 if err = state.WriteSP(conn, buf, false); err != nil {
478 state.Ctx.LogE("sp-startI", les, err, func(les LEs) string {
480 "SP with %s (nice %s): writing",
482 NicenessFmt(state.Nice),
488 state.Ctx.LogD("sp-startI-wait", les, func(les LEs) string {
490 "SP with %s (nice %s): waiting for first message",
492 NicenessFmt(state.Nice),
495 conn.SetReadDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
496 if buf, err = state.ReadSP(conn); err != nil {
497 state.Ctx.LogE("sp-startI-read", les, err, func(les LEs) string {
499 "SP with %s (nice %s): reading",
501 NicenessFmt(state.Nice),
507 payload, state.csOur, state.csTheir, err = state.hs.ReadMessage(nil, buf)
509 state.Ctx.LogE("sp-startI-read", les, err, func(les LEs) string {
511 "SP with %s (nice %s): reading Noise message",
513 NicenessFmt(state.Nice),
519 state.Ctx.LogD("sp-startI-workers", les, func(les LEs) string {
521 "SP with %s (nice %s): starting workers",
523 NicenessFmt(state.Nice),
526 err = state.StartWorkers(conn, infosPayloads, payload)
528 state.Ctx.LogE("sp-startI-workers", les, err, func(les LEs) string {
530 "SP with %s (nice %s): starting workers",
532 NicenessFmt(state.Nice),
540 func (state *SPState) StartR(conn ConnDeadlined) error {
541 started := time.Now()
542 conf := noise.Config{
543 CipherSuite: NoiseCipherSuite,
544 Pattern: noise.HandshakeIK,
546 StaticKeypair: noise.DHKey{
547 Private: state.Ctx.Self.NoisePrv[:],
548 Public: state.Ctx.Self.NoisePub[:],
551 hs, err := noise.NewHandshakeState(conf)
557 state.payloads = make(chan []byte)
558 state.pings = make(chan struct{})
559 state.infosOurSeen = make(map[[32]byte]uint8)
560 state.infosTheir = make(map[[32]byte]*SPInfo)
561 state.progressBars = make(map[string]struct{})
562 state.started = started
563 state.xxOnly = xxOnly
567 logMsg := func(les LEs) string {
569 "SP nice %s: waiting for first message",
570 NicenessFmt(state.Nice),
573 les := LEs{{"Nice", int(state.Nice)}}
574 state.Ctx.LogD("sp-startR", les, logMsg)
575 conn.SetReadDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
576 if buf, err = state.ReadSP(conn); err != nil {
577 state.Ctx.LogE("sp-startR-read", les, err, logMsg)
580 if payload, _, _, err = state.hs.ReadMessage(nil, buf); err != nil {
581 state.Ctx.LogE("sp-startR-read", les, err, logMsg)
586 for _, n := range state.Ctx.Neigh {
587 if subtle.ConstantTimeCompare(state.hs.PeerStatic(), n.NoisePub[:]) == 1 {
593 peerId := Base32Codec.EncodeToString(state.hs.PeerStatic())
594 err = errors.New("unknown peer: " + peerId)
595 state.Ctx.LogE("sp-startR-unknown", append(les, LE{"Peer", peerId}), err, logMsg)
599 state.rxRate = node.RxRate
600 state.txRate = node.TxRate
601 state.onlineDeadline = node.OnlineDeadline
602 state.maxOnlineTime = node.MaxOnlineTime
603 les = LEs{{"Node", node.Id}, {"Nice", int(state.Nice)}}
605 if err = state.Ctx.ensureRxDir(node.Id); err != nil {
609 if xxOnly == "" || xxOnly == TRx {
610 rxLock, err = state.Ctx.LockDir(node.Id, string(TRx))
615 state.rxLock = rxLock
617 if xxOnly == "" || xxOnly == TTx {
618 txLock, err = state.Ctx.LockDir(node.Id, string(TTx))
623 state.txLock = txLock
625 var infosPayloads [][]byte
626 if xxOnly == "" || xxOnly == TTx {
627 infosPayloads = state.Ctx.infosOur(node.Id, state.Nice, &state.infosOurSeen)
629 var firstPayload []byte
630 if len(infosPayloads) > 0 {
631 firstPayload = infosPayloads[0]
633 // Pad first payload, to hide actual number of existing files
634 for i := 0; i < (MaxSPSize-len(firstPayload))/SPHeadOverhead; i++ {
635 firstPayload = append(firstPayload, SPHaltMarshalized...)
638 state.Ctx.LogD("sp-startR-write", les, func(les LEs) string {
640 "SP with %s (nice %s): sending first message",
641 node.Name, NicenessFmt(state.Nice),
644 buf, state.csTheir, state.csOur, err = state.hs.WriteMessage(nil, firstPayload)
649 conn.SetWriteDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
650 if err = state.WriteSP(conn, buf, false); err != nil {
651 state.Ctx.LogE("sp-startR-write", les, err, func(les LEs) string {
653 "SP with %s (nice %s): writing",
654 node.Name, NicenessFmt(state.Nice),
660 state.Ctx.LogD("sp-startR-workers", les, func(les LEs) string {
662 "SP with %s (nice %s): starting workers",
663 node.Name, NicenessFmt(state.Nice),
666 err = state.StartWorkers(conn, infosPayloads, payload)
673 func (state *SPState) closeFd(pth string) {
675 if s, exists := state.fds[pth]; exists {
676 delete(state.fds, pth)
679 state.fdsLock.Unlock()
682 func (state *SPState) StartWorkers(
684 infosPayloads [][]byte,
687 les := LEs{{"Node", state.Node.Id}, {"Nice", int(state.Nice)}}
688 state.fds = make(map[string]FdAndFullSize)
689 state.fileHashers = make(map[string]*HasherAndOffset)
690 state.isDead = make(chan struct{})
691 if state.maxOnlineTime > 0 {
692 state.mustFinishAt = state.started.Add(state.maxOnlineTime)
697 queues := spCheckers[*state.Node.Id]
699 queues = &SPCheckerQueues{
700 appeared: make(chan *[32]byte),
701 checked: make(chan *[32]byte),
703 spCheckers[*state.Node.Id] = queues
704 go SPChecker(state.Ctx, state.Node.Id, queues.appeared, queues.checked)
706 state.checkerQueues = *queues
708 for job := range state.Ctx.JobsNoCK(state.Node.Id) {
709 if job.PktEnc.Nice <= state.Nice {
710 state.checkerQueues.appeared <- job.HshValue
716 defer state.wg.Done()
721 case hsh := <-state.checkerQueues.checked:
722 state.payloads <- MarshalSP(SPTypeDone, SPDone{hsh})
728 // Remaining handshake payload sending
729 if len(infosPayloads) > 1 {
732 for _, payload := range infosPayloads[1:] {
734 "sp-queue-remaining",
735 append(les, LE{"Size", int64(len(payload))}),
736 func(les LEs) string {
738 "SP with %s (nice %s): queuing remaining payload (%s)",
739 state.Node.Name, NicenessFmt(state.Nice),
740 humanize.IBytes(uint64(len(payload))),
744 state.payloads <- payload
750 // Processing of first payload and queueing its responses
751 logMsg := func(les LEs) string {
753 "SP with %s (nice %s): processing first payload (%s)",
754 state.Node.Name, NicenessFmt(state.Nice),
755 humanize.IBytes(uint64(len(payload))),
758 state.Ctx.LogD("sp-process", append(les, LE{"Size", int64(len(payload))}), logMsg)
759 replies, err := state.ProcessSP(payload)
761 state.Ctx.LogE("sp-process", les, err, logMsg)
766 for _, reply := range replies {
769 append(les, LE{"Size", int64(len(reply))}),
770 func(les LEs) string {
772 "SP with %s (nice %s): queuing reply (%s)",
773 state.Node.Name, NicenessFmt(state.Nice),
774 humanize.IBytes(uint64(len(payload))),
778 state.payloads <- reply
786 deadlineTicker := time.NewTicker(time.Second)
787 pingTicker := time.NewTicker(PingTimeout)
792 deadlineTicker.Stop()
795 case now := <-deadlineTicker.C:
796 if now.Sub(state.RxLastNonPing) >= state.onlineDeadline &&
797 now.Sub(state.TxLastNonPing) >= state.onlineDeadline {
800 if state.maxOnlineTime > 0 && state.mustFinishAt.Before(now) {
803 if now.Sub(state.RxLastSeen) >= 2*PingTimeout {
809 conn.Close() // #nosec G104
810 case now := <-pingTicker.C:
811 if now.After(state.TxLastSeen.Add(PingTimeout)) {
814 state.pings <- struct{}{}
822 // Spool checker and INFOs sender of appearing files
823 if !state.listOnly && (state.xxOnly == "" || state.xxOnly == TTx) {
826 ticker := time.NewTicker(time.Second)
834 for _, payload := range state.Ctx.infosOur(
841 append(les, LE{"Size", int64(len(payload))}),
842 func(les LEs) string {
844 "SP with %s (nice %s): queuing new info (%s)",
845 state.Node.Name, NicenessFmt(state.Nice),
846 humanize.IBytes(uint64(len(payload))),
850 state.payloads <- payload
861 defer state.SetDead()
862 defer state.wg.Done()
864 if state.NotAlive() {
871 state.Ctx.LogD("sp-got-ping", les, func(les LEs) string {
873 "SP with %s (nice %s): got ping",
874 state.Node.Name, NicenessFmt(state.Nice),
877 payload = SPPingMarshalized
879 case payload = <-state.payloads:
882 append(les, LE{"Size", int64(len(payload))}),
883 func(les LEs) string {
885 "SP with %s (nice %s): got payload (%s)",
886 state.Node.Name, NicenessFmt(state.Nice),
887 humanize.IBytes(uint64(len(payload))),
893 if len(state.queueTheir) == 0 {
895 time.Sleep(100 * time.Millisecond)
898 freq := state.queueTheir[0].freq
900 if state.txRate > 0 {
901 time.Sleep(time.Second / time.Duration(state.txRate))
903 pktName := Base32Codec.EncodeToString(freq.Hash[:])
906 LE{"XX", string(TTx)},
908 LE{"Size", int64(freq.Offset)},
910 logMsg := func(les LEs) string {
912 "SP with %s (nice %s): tx/%s (%s)",
913 state.Node.Name, NicenessFmt(state.Nice),
915 humanize.IBytes(freq.Offset),
918 state.Ctx.LogD("sp-queue", lesp, func(les LEs) string {
919 return logMsg(les) + ": queueing"
921 pth := filepath.Join(
923 state.Node.Id.String(),
925 Base32Codec.EncodeToString(freq.Hash[:]),
927 state.fdsLock.RLock()
928 fdAndFullSize, exists := state.fds[pth]
929 state.fdsLock.RUnlock()
931 fd, err := os.Open(pth)
933 state.Ctx.LogE("sp-queue-open", lesp, err, func(les LEs) string {
934 return logMsg(les) + ": opening"
940 state.Ctx.LogE("sp-queue-stat", lesp, err, func(les LEs) string {
941 return logMsg(les) + ": stating"
945 fdAndFullSize = FdAndFullSize{fd: fd, fullSize: fi.Size()}
947 state.fds[pth] = fdAndFullSize
948 state.fdsLock.Unlock()
950 fd := fdAndFullSize.fd
951 fullSize := fdAndFullSize.fullSize
953 if freq.Offset < uint64(fullSize) {
954 state.Ctx.LogD("sp-file-seek", lesp, func(les LEs) string {
955 return logMsg(les) + ": seeking"
957 if _, err = fd.Seek(int64(freq.Offset), io.SeekStart); err != nil {
958 state.Ctx.LogE("sp-file-seek", lesp, err, func(les LEs) string {
959 return logMsg(les) + ": seeking"
963 buf = make([]byte, MaxSPSize-SPHeadOverhead-SPFileOverhead)
964 n, err := fd.Read(buf)
966 state.Ctx.LogE("sp-file-read", lesp, err, func(les LEs) string {
967 return logMsg(les) + ": reading"
974 LE{"XX", string(TTx)},
976 LE{"Size", int64(n)},
978 state.Ctx.LogD("sp-file-read", lesp, func(les LEs) string {
981 logMsg(les), humanize.IBytes(uint64(n)),
986 payload = MarshalSP(SPTypeFile, SPFile{
991 ourSize := freq.Offset + uint64(len(buf))
994 LE{"XX", string(TTx)},
996 LE{"Size", int64(ourSize)},
997 LE{"FullSize", fullSize},
999 if state.Ctx.ShowPrgrs {
1000 state.progressBars[pktName] = struct{}{}
1001 Progress("Tx", lesp)
1004 if len(state.queueTheir) > 0 && *state.queueTheir[0].freq.Hash == *freq.Hash {
1005 if ourSize == uint64(fullSize) {
1006 state.Ctx.LogD("sp-file-finished", lesp, func(les LEs) string {
1007 return logMsg(les) + ": finished"
1009 if len(state.queueTheir) > 1 {
1010 state.queueTheir = state.queueTheir[1:]
1012 state.queueTheir = state.queueTheir[:0]
1014 if state.Ctx.ShowPrgrs {
1015 delete(state.progressBars, pktName)
1018 state.queueTheir[0].freq.Offset += uint64(len(buf))
1021 state.Ctx.LogD("sp-file-disappeared", lesp, func(les LEs) string {
1022 return logMsg(les) + ": queue disappeared"
1027 logMsg := func(les LEs) string {
1029 "SP with %s (nice %s): sending %s",
1030 state.Node.Name, NicenessFmt(state.Nice),
1031 humanize.IBytes(uint64(len(payload))),
1034 state.Ctx.LogD("sp-sending", append(les, LE{"Size", int64(len(payload))}), logMsg)
1035 conn.SetWriteDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
1036 if err := state.WriteSP(conn, state.csOur.Encrypt(nil, nil, payload), ping); err != nil {
1037 state.Ctx.LogE("sp-sending", les, err, logMsg)
1047 if state.NotAlive() {
1050 logMsg := func(les LEs) string {
1052 "SP with %s (nice %s): waiting for payload",
1053 state.Node.Name, NicenessFmt(state.Nice),
1056 state.Ctx.LogD("sp-recv-wait", les, logMsg)
1057 conn.SetReadDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
1058 payload, err := state.ReadSP(conn)
1063 unmarshalErr := err.(*xdr.UnmarshalError)
1064 if os.IsTimeout(unmarshalErr.Err) {
1067 if unmarshalErr.ErrorCode == xdr.ErrIO {
1070 state.Ctx.LogE("sp-recv-wait", les, err, logMsg)
1073 logMsg = func(les LEs) string {
1075 "SP with %s (nice %s): payload (%s)",
1076 state.Node.Name, NicenessFmt(state.Nice),
1077 humanize.IBytes(uint64(len(payload))),
1082 append(les, LE{"Size", int64(len(payload))}),
1083 func(les LEs) string { return logMsg(les) + ": got" },
1085 payload, err = state.csTheir.Decrypt(nil, nil, payload)
1087 state.Ctx.LogE("sp-recv-got", les, err, func(les LEs) string {
1088 return logMsg(les) + ": got"
1094 append(les, LE{"Size", int64(len(payload))}),
1095 func(les LEs) string {
1096 return logMsg(les) + ": processing"
1099 replies, err := state.ProcessSP(payload)
1101 state.Ctx.LogE("sp-recv-process", les, err, func(les LEs) string {
1102 return logMsg(les) + ": processing"
1108 for _, reply := range replies {
1111 append(les[:len(les)-1], LE{"Size", int64(len(reply))}),
1112 func(les LEs) string {
1114 "SP with %s (nice %s): queuing reply (%s)",
1115 state.Node.Name, NicenessFmt(state.Nice),
1116 humanize.IBytes(uint64(len(reply))),
1120 state.payloads <- reply
1124 if state.rxRate > 0 {
1125 time.Sleep(time.Second / time.Duration(state.rxRate))
1131 conn.Close() // #nosec G104
1137 func (state *SPState) Wait() {
1139 close(state.payloads)
1141 state.Duration = time.Now().Sub(state.started)
1144 state.RxSpeed = state.RxBytes
1145 state.TxSpeed = state.TxBytes
1146 rxDuration := int64(state.RxLastSeen.Sub(state.started).Seconds())
1147 txDuration := int64(state.TxLastSeen.Sub(state.started).Seconds())
1149 state.RxSpeed = state.RxBytes / rxDuration
1152 state.TxSpeed = state.TxBytes / txDuration
1154 for _, s := range state.fds {
1157 for pktName := range state.progressBars {
1158 ProgressKill(pktName)
1162 func (state *SPState) ProcessSP(payload []byte) ([][]byte, error) {
1163 les := LEs{{"Node", state.Node.Id}, {"Nice", int(state.Nice)}}
1164 r := bytes.NewReader(payload)
1166 var replies [][]byte
1169 state.Ctx.LogD("sp-process-unmarshal", les, func(les LEs) string {
1171 "SP with %s (nice %s): unmarshaling header",
1172 state.Node.Name, NicenessFmt(state.Nice),
1176 if _, err = xdr.Unmarshal(r, &head); err != nil {
1177 state.Ctx.LogE("sp-process-unmarshal", les, err, func(les LEs) string {
1179 "SP with %s (nice %s): unmarshaling header",
1180 state.Node.Name, NicenessFmt(state.Nice),
1185 if head.Type != SPTypePing {
1186 state.RxLastNonPing = state.RxLastSeen
1192 append(les, LE{"Type", "halt"}), func(les LEs) string {
1194 "SP with %s (nice %s): got HALT",
1195 state.Node.Name, NicenessFmt(state.Nice),
1200 state.queueTheir = nil
1206 append(les, LE{"Type", "ping"}),
1207 func(les LEs) string {
1209 "SP with %s (nice %s): got PING",
1210 state.Node.Name, NicenessFmt(state.Nice),
1217 lesp := append(les, LE{"Type", "info"})
1219 "sp-process-info-unmarshal", lesp,
1220 func(les LEs) string {
1222 "SP with %s (nice %s): unmarshaling INFO",
1223 state.Node.Name, NicenessFmt(state.Nice),
1228 if _, err = xdr.Unmarshal(r, &info); err != nil {
1230 "sp-process-info-unmarshal", lesp, err,
1231 func(les LEs) string {
1233 "SP with %s (nice %s): unmarshaling INFO",
1234 state.Node.Name, NicenessFmt(state.Nice),
1240 pktName := Base32Codec.EncodeToString(info.Hash[:])
1244 LE{"Size", int64(info.Size)},
1245 LE{"PktNice", int(info.Nice)},
1247 logMsg := func(les LEs) string {
1249 "SP with %s (nice %s): INFO %s (%s) nice %s",
1250 state.Node.Name, NicenessFmt(state.Nice),
1252 humanize.IBytes(info.Size),
1253 NicenessFmt(info.Nice),
1256 if !state.listOnly && info.Nice > state.Nice {
1257 state.Ctx.LogD("sp-process-info-too-nice", lesp, func(les LEs) string {
1258 return logMsg(les) + ": too nice"
1262 state.Ctx.LogD("sp-process-info-got", lesp, func(les LEs) string {
1263 return logMsg(les) + ": received"
1265 if !state.listOnly && state.xxOnly == TTx {
1269 state.infosTheir[*info.Hash] = &info
1271 state.Ctx.LogD("sp-process-info-stat", lesp, func(les LEs) string {
1272 return logMsg(les) + ": stating part"
1274 pktPath := filepath.Join(
1276 state.Node.Id.String(),
1278 Base32Codec.EncodeToString(info.Hash[:]),
1280 logMsg = func(les LEs) string {
1282 "Packet %s (%s) (nice %s)",
1284 humanize.IBytes(info.Size),
1285 NicenessFmt(info.Nice),
1288 if _, err = os.Stat(pktPath); err == nil {
1289 state.Ctx.LogI("sp-info-done", lesp, func(les LEs) string {
1290 return logMsg(les) + ": already done"
1292 if !state.listOnly {
1293 replies = append(replies, MarshalSP(SPTypeDone, SPDone{info.Hash}))
1297 if _, err = os.Stat(pktPath + SeenSuffix); err == nil {
1298 state.Ctx.LogI("sp-info-seen", lesp, func(les LEs) string {
1299 return logMsg(les) + ": already seen"
1301 if !state.listOnly {
1302 replies = append(replies, MarshalSP(SPTypeDone, SPDone{info.Hash}))
1306 if _, err = os.Stat(pktPath + NoCKSuffix); err == nil {
1307 state.Ctx.LogI("sp-info-nock", lesp, func(les LEs) string {
1308 return logMsg(les) + ": still not checksummed"
1312 fi, err := os.Stat(pktPath + PartSuffix)
1317 if !state.Ctx.IsEnoughSpace(int64(info.Size) - offset) {
1318 state.Ctx.LogI("sp-info-no-space", lesp, func(les LEs) string {
1319 return logMsg(les) + ": not enough space"
1325 append(lesp, LE{"Offset", offset}),
1326 func(les LEs) string {
1328 "%s: %d%%", logMsg(les), 100*uint64(offset)/info.Size,
1332 if !state.listOnly && (state.onlyPkts == nil || state.onlyPkts[*info.Hash]) {
1333 replies = append(replies, MarshalSP(
1335 SPFreq{info.Hash, uint64(offset)},
1340 lesp := append(les, LE{"Type", "file"})
1341 state.Ctx.LogD("sp-process-file", lesp, func(les LEs) string {
1343 "SP with %s (nice %s): unmarshaling FILE",
1344 state.Node.Name, NicenessFmt(state.Nice),
1348 if _, err = xdr.Unmarshal(r, &file); err != nil {
1349 state.Ctx.LogE("sp-process-file", lesp, err, func(les LEs) string {
1351 "SP with %s (nice %s): unmarshaling FILE",
1352 state.Node.Name, NicenessFmt(state.Nice),
1357 pktName := Base32Codec.EncodeToString(file.Hash[:])
1360 LE{"XX", string(TRx)},
1362 LE{"Size", int64(len(file.Payload))},
1364 logMsg := func(les LEs) string {
1366 "Got packet %s (%s)",
1367 pktName, humanize.IBytes(uint64(len(file.Payload))),
1370 dirToSync := filepath.Join(
1372 state.Node.Id.String(),
1375 filePath := filepath.Join(dirToSync, pktName)
1376 filePathPart := filePath + PartSuffix
1377 state.Ctx.LogD("sp-file-open", lesp, func(les LEs) string {
1378 return logMsg(les) + ": opening part"
1380 state.fdsLock.RLock()
1381 fdAndFullSize, exists := state.fds[filePathPart]
1382 state.fdsLock.RUnlock()
1385 fd = fdAndFullSize.fd
1387 fd, err = os.OpenFile(
1389 os.O_RDWR|os.O_CREATE,
1393 state.Ctx.LogE("sp-file-open", lesp, err, func(les LEs) string {
1394 return logMsg(les) + ": opening part"
1398 state.fdsLock.Lock()
1399 state.fds[filePathPart] = FdAndFullSize{fd: fd}
1400 state.fdsLock.Unlock()
1401 if file.Offset == 0 {
1402 h, err := blake2b.New256(nil)
1406 state.fileHashers[filePath] = &HasherAndOffset{h: h}
1411 append(lesp, LE{"Offset", file.Offset}),
1412 func(les LEs) string {
1413 return fmt.Sprintf("%s: seeking %d", logMsg(les), file.Offset)
1415 if _, err = fd.Seek(int64(file.Offset), io.SeekStart); err != nil {
1416 state.Ctx.LogE("sp-file-seek", lesp, err, func(les LEs) string {
1417 return logMsg(les) + ": seeking"
1419 state.closeFd(filePathPart)
1422 state.Ctx.LogD("sp-file-write", lesp, func(les LEs) string {
1423 return logMsg(les) + ": writing"
1425 if _, err = fd.Write(file.Payload); err != nil {
1426 state.Ctx.LogE("sp-file-write", lesp, err, func(les LEs) string {
1427 return logMsg(les) + ": writing"
1429 state.closeFd(filePathPart)
1432 hasherAndOffset, hasherExists := state.fileHashers[filePath]
1434 if hasherAndOffset.offset == file.Offset {
1435 if _, err = hasherAndOffset.h.Write(file.Payload); err != nil {
1438 hasherAndOffset.offset += uint64(len(file.Payload))
1441 "sp-file-offset-differs", lesp,
1442 func(les LEs) string {
1443 return logMsg(les) + ": offset differs, deleting hasher"
1446 delete(state.fileHashers, filePath)
1447 hasherExists = false
1450 ourSize := int64(file.Offset + uint64(len(file.Payload)))
1451 lesp[len(lesp)-1].V = ourSize
1452 fullsize := int64(0)
1454 infoTheir, ok := state.infosTheir[*file.Hash]
1457 fullsize = int64(infoTheir.Size)
1459 lesp = append(lesp, LE{"FullSize", fullsize})
1460 if state.Ctx.ShowPrgrs {
1461 state.progressBars[pktName] = struct{}{}
1462 Progress("Rx", lesp)
1464 if fullsize != ourSize {
1467 if state.Ctx.ShowPrgrs {
1468 delete(state.progressBars, pktName)
1470 logMsg = func(les LEs) string {
1472 "Got packet %s %d%% (%s / %s)",
1473 pktName, 100*ourSize/fullsize,
1474 humanize.IBytes(uint64(ourSize)),
1475 humanize.IBytes(uint64(fullsize)),
1480 state.Ctx.LogE("sp-file-sync", lesp, err, func(les LEs) string {
1481 return logMsg(les) + ": syncing"
1483 state.closeFd(filePathPart)
1487 if bytes.Compare(hasherAndOffset.h.Sum(nil), file.Hash[:]) != 0 {
1489 "sp-file-bad-checksum", lesp,
1490 errors.New("checksum mismatch"),
1495 if err = os.Rename(filePathPart, filePath); err != nil {
1496 state.Ctx.LogE("sp-file-rename", lesp, err, func(les LEs) string {
1497 return logMsg(les) + ": renaming"
1501 if err = DirSync(dirToSync); err != nil {
1502 state.Ctx.LogE("sp-file-dirsync", lesp, err, func(les LEs) string {
1503 return logMsg(les) + ": dirsyncing"
1507 state.Ctx.LogI("sp-file-done", lesp, func(les LEs) string {
1508 return logMsg(les) + ": done"
1512 state.payloads <- MarshalSP(SPTypeDone, SPDone{file.Hash})
1516 delete(state.infosTheir, *file.Hash)
1518 if !state.Ctx.HdrUsage {
1519 state.closeFd(filePathPart)
1522 if _, err = fd.Seek(0, io.SeekStart); err != nil {
1523 state.Ctx.LogE("sp-file-seek", lesp, err, func(les LEs) string {
1524 return logMsg(les) + ": seeking"
1526 state.closeFd(filePathPart)
1529 _, pktEncRaw, err := state.Ctx.HdrRead(fd)
1530 state.closeFd(filePathPart)
1532 state.Ctx.LogE("sp-file-hdr-read", lesp, err, func(les LEs) string {
1533 return logMsg(les) + ": HdrReading"
1537 state.Ctx.HdrWrite(pktEncRaw, filePath)
1540 state.closeFd(filePathPart)
1541 if err = os.Rename(filePathPart, filePath+NoCKSuffix); err != nil {
1542 state.Ctx.LogE("sp-file-rename", lesp, err, func(les LEs) string {
1543 return logMsg(les) + ": renaming"
1547 if err = DirSync(dirToSync); err != nil {
1548 state.Ctx.LogE("sp-file-dirsync", lesp, err, func(les LEs) string {
1549 return logMsg(les) + ": dirsyncing"
1553 state.Ctx.LogI("sp-file-downloaded", lesp, func(les LEs) string {
1554 return logMsg(les) + ": downloaded"
1557 delete(state.infosTheir, *file.Hash)
1560 state.checkerQueues.appeared <- file.Hash
1564 lesp := append(les, LE{"Type", "done"})
1565 state.Ctx.LogD("sp-process-done-unmarshal", lesp, func(les LEs) string {
1567 "SP with %s (nice %s): unmarshaling DONE",
1568 state.Node.Name, NicenessFmt(state.Nice),
1572 if _, err = xdr.Unmarshal(r, &done); err != nil {
1573 state.Ctx.LogE("sp-process-done-unmarshal", lesp, err, func(les LEs) string {
1575 "SP with %s (nice %s): unmarshaling DONE",
1576 state.Node.Name, NicenessFmt(state.Nice),
1581 pktName := Base32Codec.EncodeToString(done.Hash[:])
1582 lesp = append(lesp, LE{"Pkt", pktName}, LE{"XX", string(TTx)})
1583 logMsg := func(les LEs) string {
1585 "SP with %s (nice %s): DONE: removing %s",
1586 state.Node.Name, NicenessFmt(state.Nice), pktName,
1589 state.Ctx.LogD("sp-done", lesp, logMsg)
1590 pth := filepath.Join(
1592 state.Node.Id.String(),
1596 if err = os.Remove(pth); err == nil {
1597 state.Ctx.LogI("sp-done", lesp, func(les LEs) string {
1598 return fmt.Sprintf("Packet %s is sent", pktName)
1600 if state.Ctx.HdrUsage {
1601 os.Remove(pth + HdrSuffix)
1604 state.Ctx.LogE("sp-done", lesp, err, logMsg)
1608 lesp := append(les, LE{"Type", "freq"})
1609 state.Ctx.LogD("sp-process-freq", lesp, func(les LEs) string {
1611 "SP with %s (nice %s): unmarshaling FREQ",
1612 state.Node.Name, NicenessFmt(state.Nice),
1616 if _, err = xdr.Unmarshal(r, &freq); err != nil {
1617 state.Ctx.LogE("sp-process-freq", lesp, err, func(les LEs) string {
1619 "SP with %s (nice %s): unmarshaling FREQ",
1620 state.Node.Name, NicenessFmt(state.Nice),
1625 pktName := Base32Codec.EncodeToString(freq.Hash[:])
1626 lesp = append(lesp, LE{"Pkt", pktName}, LE{"Offset", freq.Offset})
1627 state.Ctx.LogD("sp-process-freq-queueing", lesp, func(les LEs) string {
1629 "SP with %s (nice %s): FREQ %s: queuing",
1630 state.Node.Name, NicenessFmt(state.Nice), pktName,
1633 nice, exists := state.infosOurSeen[*freq.Hash]
1635 if state.onlyPkts == nil || !state.onlyPkts[*freq.Hash] {
1638 var freqWithNice *FreqWithNice
1639 for insertIdx, freqWithNice = range state.queueTheir {
1640 if freqWithNice.nice > nice {
1644 state.queueTheir = append(state.queueTheir, nil)
1645 copy(state.queueTheir[insertIdx+1:], state.queueTheir[insertIdx:])
1646 state.queueTheir[insertIdx] = &FreqWithNice{&freq, nice}
1649 state.Ctx.LogD("sp-process-freq-skip", lesp, func(les LEs) string {
1651 "SP with %s (nice %s): FREQ %s: skipping",
1652 state.Node.Name, NicenessFmt(state.Nice), pktName,
1657 state.Ctx.LogD("sp-process-freq-unknown", lesp, func(les LEs) string {
1659 "SP with %s (nice %s): FREQ %s: unknown",
1660 state.Node.Name, NicenessFmt(state.Nice), pktName,
1667 "sp-process-type-unknown",
1668 append(les, LE{"Type", head.Type}),
1669 errors.New("unknown type"),
1670 func(les LEs) string {
1672 "SP with %s (nice %s): %d",
1673 state.Node.Name, NicenessFmt(state.Nice), head.Type,
1677 return nil, BadPktType
1685 for _, info := range state.infosTheir {
1690 state.Ctx.LogI("sp-infos-rx", LEs{
1691 {"XX", string(TRx)},
1692 {"Node", state.Node.Id},
1694 {"Size", int64(size)},
1695 }, func(les LEs) string {
1697 "%s has got for us: %d packets, %s",
1698 state.Node.Name, pkts, humanize.IBytes(size),
1702 return payloadsSplit(replies), nil