/*
NNCP -- Node to Node copy, utilities for store-and-forward data exchange
-Copyright (C) 2016-2020 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2016-2021 Sergey Matveev <stargrave@stargrave.org>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
DefaultDeadline = 10 * time.Second
PingTimeout = time.Minute
- spWorkersGroup sync.WaitGroup
+ spCheckerToken chan struct{}
)
+type FdAndFullSize struct {
+ fd *os.File
+ fullSize int64
+}
+
type SPType uint8
const (
panic(err)
}
SPFileOverhead = buf.Len()
+ spCheckerToken = make(chan struct{}, 1)
+ spCheckerToken <- struct{}{}
}
func MarshalSP(typ SPType, sp interface{}) []byte {
listOnly bool
onlyPkts map[[32]byte]bool
writeSPBuf bytes.Buffer
+ fds map[string]FdAndFullSize
sync.RWMutex
}
}
close(state.isDead)
go func() {
- for _ = range state.payloads {
+ for range state.payloads {
}
}()
go func() {
- for _ = range state.pings {
+ for range state.pings {
}
}()
}
var infos []*SPInfo
var totalSize int64
for job := range ctx.Jobs(nodeId, TTx) {
- job.Fd.Close()
if job.PktEnc.Nice > nice {
continue
}
var payloads [][]byte
for _, info := range infos {
payloads = append(payloads, MarshalSP(SPTypeInfo, info))
- ctx.LogD("sp-info-our", SDS{
- "node": nodeId,
- "name": ToBase32(info.Hash[:]),
- "size": info.Size,
+ ctx.LogD("sp-info-our", LEs{
+ {"Node", nodeId},
+ {"Name", Base32Codec.EncodeToString(info.Hash[:])},
+ {"Size", info.Size},
}, "")
}
if totalSize > 0 {
- ctx.LogI("sp-infos", SDS{
- "xx": string(TTx),
- "node": nodeId,
- "pkts": len(payloads),
- "size": totalSize,
+ ctx.LogI("sp-infos", LEs{
+ {"XX", string(TTx)},
+ {"Node", nodeId},
+ {"Pkts", len(payloads)},
+ {"Size", totalSize},
}, "")
}
return payloadsSplit(payloads)
state.dirUnlock()
return err
}
- sds := SDS{"node": nodeId, "nice": int(state.Nice)}
- state.Ctx.LogD("sp-start", sds, "sending first message")
- conn.SetWriteDeadline(time.Now().Add(DefaultDeadline))
+ les := LEs{{"Node", nodeId}, {"Nice", int(state.Nice)}}
+ state.Ctx.LogD("sp-start", les, "sending first message")
+ conn.SetWriteDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
if err = state.WriteSP(conn, buf, false); err != nil {
- state.Ctx.LogE("sp-start", sds, err, "")
+ state.Ctx.LogE("sp-start", les, err, "")
state.dirUnlock()
return err
}
- state.Ctx.LogD("sp-start", sds, "waiting for first message")
- conn.SetReadDeadline(time.Now().Add(DefaultDeadline))
+ state.Ctx.LogD("sp-start", les, "waiting for first message")
+ conn.SetReadDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
if buf, err = state.ReadSP(conn); err != nil {
- state.Ctx.LogE("sp-start", sds, err, "")
+ state.Ctx.LogE("sp-start", les, err, "")
state.dirUnlock()
return err
}
payload, state.csOur, state.csTheir, err = state.hs.ReadMessage(nil, buf)
if err != nil {
- state.Ctx.LogE("sp-start", sds, err, "")
+ state.Ctx.LogE("sp-start", les, err, "")
state.dirUnlock()
return err
}
- state.Ctx.LogD("sp-start", sds, "starting workers")
+ state.Ctx.LogD("sp-start", les, "starting workers")
err = state.StartWorkers(conn, infosPayloads, payload)
if err != nil {
- state.Ctx.LogE("sp-start", sds, err, "")
+ state.Ctx.LogE("sp-start", les, err, "")
state.dirUnlock()
}
return err
state.xxOnly = xxOnly
var buf []byte
var payload []byte
- state.Ctx.LogD("sp-start", SDS{"nice": int(state.Nice)}, "waiting for first message")
- conn.SetReadDeadline(time.Now().Add(DefaultDeadline))
+ state.Ctx.LogD("sp-start", LEs{{"Nice", int(state.Nice)}}, "waiting for first message")
+ conn.SetReadDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
if buf, err = state.ReadSP(conn); err != nil {
- state.Ctx.LogE("sp-start", SDS{}, err, "")
+ state.Ctx.LogE("sp-start", LEs{}, err, "")
return err
}
if payload, _, _, err = state.hs.ReadMessage(nil, buf); err != nil {
- state.Ctx.LogE("sp-start", SDS{}, err, "")
+ state.Ctx.LogE("sp-start", LEs{}, err, "")
return err
}
}
}
if node == nil {
- peerId := ToBase32(state.hs.PeerStatic())
- state.Ctx.LogE("sp-start", SDS{"peer": peerId}, errors.New("unknown"), "")
+ peerId := Base32Codec.EncodeToString(state.hs.PeerStatic())
+ state.Ctx.LogE("sp-start", LEs{{"Peer", peerId}}, errors.New("unknown peer"), "")
return errors.New("Unknown peer: " + peerId)
}
state.Node = node
state.txRate = node.TxRate
state.onlineDeadline = node.OnlineDeadline
state.maxOnlineTime = node.MaxOnlineTime
- sds := SDS{"node": node.Id, "nice": int(state.Nice)}
+ les := LEs{{"Node", node.Id}, {"Nice", int(state.Nice)}}
- if state.Ctx.ensureRxDir(node.Id); err != nil {
+ if err = state.Ctx.ensureRxDir(node.Id); err != nil {
return err
}
var rxLock *os.File
firstPayload = append(firstPayload, SPHaltMarshalized...)
}
- state.Ctx.LogD("sp-start", sds, "sending first message")
+ state.Ctx.LogD("sp-start", les, "sending first message")
buf, state.csTheir, state.csOur, err = state.hs.WriteMessage(nil, firstPayload)
if err != nil {
state.dirUnlock()
return err
}
- conn.SetWriteDeadline(time.Now().Add(DefaultDeadline))
+ conn.SetWriteDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
if err = state.WriteSP(conn, buf, false); err != nil {
- state.Ctx.LogE("sp-start", sds, err, "")
+ state.Ctx.LogE("sp-start", les, err, "")
state.dirUnlock()
return err
}
- state.Ctx.LogD("sp-start", sds, "starting workers")
+ state.Ctx.LogD("sp-start", les, "starting workers")
err = state.StartWorkers(conn, infosPayloads, payload)
if err != nil {
state.dirUnlock()
return err
}
+func (state *SPState) closeFd(pth string) {
+ s, exists := state.fds[pth]
+ delete(state.fds, pth)
+ if exists {
+ s.fd.Close()
+ }
+}
+
func (state *SPState) StartWorkers(
conn ConnDeadlined,
infosPayloads [][]byte,
payload []byte,
) error {
- sds := SDS{"node": state.Node.Id, "nice": int(state.Nice)}
+ les := LEs{{"Node", state.Node.Id}, {"Nice", int(state.Nice)}}
+ state.fds = make(map[string]FdAndFullSize)
state.isDead = make(chan struct{})
if state.maxOnlineTime > 0 {
state.mustFinishAt = state.started.Add(state.maxOnlineTime)
for _, payload := range infosPayloads[1:] {
state.Ctx.LogD(
"sp-work",
- SdsAdd(sds, SDS{"size": len(payload)}),
+ append(les, LE{"Size", len(payload)}),
"queuing remaining payload",
)
state.payloads <- payload
// Processing of first payload and queueing its responses
state.Ctx.LogD(
"sp-work",
- SdsAdd(sds, SDS{"size": len(payload)}),
+ append(les, LE{"Size", len(payload)}),
"processing first payload",
)
replies, err := state.ProcessSP(payload)
if err != nil {
- state.Ctx.LogE("sp-work", sds, err, "")
+ state.Ctx.LogE("sp-work", les, err, "")
return err
}
state.wg.Add(1)
for _, reply := range replies {
state.Ctx.LogD(
"sp-work",
- SdsAdd(sds, SDS{"size": len(reply)}),
+ append(les, LE{"Size", len(reply)}),
"queuing reply",
)
state.payloads <- reply
(state.maxOnlineTime > 0 && state.mustFinishAt.Before(now)) ||
(now.Sub(state.RxLastSeen) >= 2*PingTimeout) {
state.SetDead()
- conn.Close()
+ conn.Close() // #nosec G104
}
case now := <-pingTicker.C:
if now.After(state.TxLastSeen.Add(PingTimeout)) {
) {
state.Ctx.LogD(
"sp-work",
- SdsAdd(sds, SDS{"size": len(payload)}),
+ append(les, LE{"Size", len(payload)}),
"queuing new info",
)
state.payloads <- payload
var ping bool
select {
case <-state.pings:
- state.Ctx.LogD("sp-xmit", sds, "got ping")
+ state.Ctx.LogD("sp-xmit", les, "got ping")
payload = SPPingMarshalized
ping = true
case payload = <-state.payloads:
state.Ctx.LogD(
"sp-xmit",
- SdsAdd(sds, SDS{"size": len(payload)}),
+ append(les, LE{"Size", len(payload)}),
"got payload",
)
default:
if state.txRate > 0 {
time.Sleep(time.Second / time.Duration(state.txRate))
}
- sdsp := SdsAdd(sds, SDS{
- "xx": string(TTx),
- "pkt": ToBase32(freq.Hash[:]),
- "size": int64(freq.Offset),
- })
- state.Ctx.LogD("sp-file", sdsp, "queueing")
- fd, err := os.Open(filepath.Join(
+ lesp := append(les, LEs{
+ {"XX", string(TTx)},
+ {"Pkt", Base32Codec.EncodeToString(freq.Hash[:])},
+ {"Size", int64(freq.Offset)},
+ }...)
+ state.Ctx.LogD("sp-file", lesp, "queueing")
+ pth := filepath.Join(
state.Ctx.Spool,
state.Node.Id.String(),
string(TTx),
- ToBase32(freq.Hash[:]),
- ))
- if err != nil {
- state.Ctx.LogE("sp-file", sdsp, err, "")
- return
- }
- fi, err := fd.Stat()
- if err != nil {
- state.Ctx.LogE("sp-file", sdsp, err, "")
- return
+ Base32Codec.EncodeToString(freq.Hash[:]),
+ )
+ fdAndFullSize, exists := state.fds[pth]
+ if !exists {
+ fd, err := os.Open(pth)
+ if err != nil {
+ state.Ctx.LogE("sp-file", lesp, err, "")
+ return
+ }
+ fi, err := fd.Stat()
+ if err != nil {
+ state.Ctx.LogE("sp-file", lesp, err, "")
+ return
+ }
+ fdAndFullSize = FdAndFullSize{fd: fd, fullSize: fi.Size()}
+ state.fds[pth] = fdAndFullSize
}
- fullSize := fi.Size()
+ fd := fdAndFullSize.fd
+ fullSize := fdAndFullSize.fullSize
var buf []byte
if freq.Offset < uint64(fullSize) {
- state.Ctx.LogD("sp-file", sdsp, "seeking")
+ state.Ctx.LogD("sp-file", lesp, "seeking")
if _, err = fd.Seek(int64(freq.Offset), io.SeekStart); err != nil {
- state.Ctx.LogE("sp-file", sdsp, err, "")
+ state.Ctx.LogE("sp-file", lesp, err, "")
return
}
buf = make([]byte, MaxSPSize-SPHeadOverhead-SPFileOverhead)
n, err := fd.Read(buf)
if err != nil {
- state.Ctx.LogE("sp-file", sdsp, err, "")
+ state.Ctx.LogE("sp-file", lesp, err, "")
return
}
buf = buf[:n]
- state.Ctx.LogD("sp-file", SdsAdd(sdsp, SDS{"size": n}), "read")
+ state.Ctx.LogD("sp-file", append(lesp, LE{"Size", n}), "read")
}
- fd.Close()
+ state.closeFd(pth)
payload = MarshalSP(SPTypeFile, SPFile{
Hash: freq.Hash,
Offset: freq.Offset,
Payload: buf,
})
ourSize := freq.Offset + uint64(len(buf))
- sdsp["size"] = int64(ourSize)
- sdsp["fullsize"] = fullSize
+ lesp = append(lesp, LE{"Size", int64(ourSize)})
+ lesp = append(lesp, LE{"FullSize", fullSize})
if state.Ctx.ShowPrgrs {
- Progress("Tx", sdsp)
+ Progress("Tx", lesp)
}
state.Lock()
if len(state.queueTheir) > 0 && *state.queueTheir[0].freq.Hash == *freq.Hash {
if ourSize == uint64(fullSize) {
- state.Ctx.LogD("sp-file", sdsp, "finished")
+ state.Ctx.LogD("sp-file", lesp, "finished")
if len(state.queueTheir) > 1 {
state.queueTheir = state.queueTheir[1:]
} else {
state.queueTheir[0].freq.Offset += uint64(len(buf))
}
} else {
- state.Ctx.LogD("sp-file", sdsp, "queue disappeared")
+ state.Ctx.LogD("sp-file", lesp, "queue disappeared")
}
state.Unlock()
}
- state.Ctx.LogD("sp-xmit", SdsAdd(sds, SDS{"size": len(payload)}), "sending")
- conn.SetWriteDeadline(time.Now().Add(DefaultDeadline))
+ state.Ctx.LogD("sp-xmit", append(les, LE{"Size", len(payload)}), "sending")
+ conn.SetWriteDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
if err := state.WriteSP(conn, state.csOur.Encrypt(nil, nil, payload), ping); err != nil {
- state.Ctx.LogE("sp-xmit", sds, err, "")
+ state.Ctx.LogE("sp-xmit", les, err, "")
return
}
}
if state.NotAlive() {
break
}
- state.Ctx.LogD("sp-recv", sds, "waiting for payload")
- conn.SetReadDeadline(time.Now().Add(DefaultDeadline))
+ state.Ctx.LogD("sp-recv", les, "waiting for payload")
+ conn.SetReadDeadline(time.Now().Add(DefaultDeadline)) // #nosec G104
payload, err := state.ReadSP(conn)
if err != nil {
if err == io.EOF {
if unmarshalErr.ErrorCode == xdr.ErrIO {
break
}
- state.Ctx.LogE("sp-recv", sds, err, "")
+ state.Ctx.LogE("sp-recv", les, err, "")
break
}
state.Ctx.LogD(
"sp-recv",
- SdsAdd(sds, SDS{"size": len(payload)}),
+ append(les, LE{"Size", len(payload)}),
"got payload",
)
payload, err = state.csTheir.Decrypt(nil, nil, payload)
if err != nil {
- state.Ctx.LogE("sp-recv", sds, err, "")
+ state.Ctx.LogE("sp-recv", les, err, "")
break
}
state.Ctx.LogD(
"sp-recv",
- SdsAdd(sds, SDS{"size": len(payload)}),
+ append(les, LE{"Size", len(payload)}),
"processing",
)
replies, err := state.ProcessSP(payload)
if err != nil {
- state.Ctx.LogE("sp-recv", sds, err, "")
+ state.Ctx.LogE("sp-recv", les, err, "")
break
}
state.wg.Add(1)
for _, reply := range replies {
state.Ctx.LogD(
"sp-recv",
- SdsAdd(sds, SDS{"size": len(reply)}),
+ append(les, LE{"Size", len(reply)}),
"queuing reply",
)
state.payloads <- reply
state.SetDead()
state.wg.Done()
state.SetDead()
- conn.Close()
+ conn.Close() // #nosec G104
+ for _, s := range state.fds {
+ s.fd.Close()
+ }
}()
return nil
}
func (state *SPState) ProcessSP(payload []byte) ([][]byte, error) {
- sds := SDS{"node": state.Node.Id, "nice": int(state.Nice)}
+ les := LEs{{"Node", state.Node.Id}, {"Nice", int(state.Nice)}}
r := bytes.NewReader(payload)
var err error
var replies [][]byte
var infosGot bool
for r.Len() > 0 {
- state.Ctx.LogD("sp-process", sds, "unmarshaling header")
+ state.Ctx.LogD("sp-process", les, "unmarshaling header")
var head SPHead
if _, err = xdr.Unmarshal(r, &head); err != nil {
- state.Ctx.LogE("sp-process", sds, err, "")
+ state.Ctx.LogE("sp-process", les, err, "")
return nil, err
}
+ if head.Type != SPTypePing {
+ state.RxLastNonPing = state.RxLastSeen
+ }
switch head.Type {
case SPTypeHalt:
- state.Ctx.LogD("sp-process", SdsAdd(sds, SDS{"type": "halt"}), "")
+ state.Ctx.LogD("sp-process", append(les, LE{"Type", "halt"}), "")
state.Lock()
state.queueTheir = nil
state.Unlock()
case SPTypePing:
- state.Ctx.LogD("sp-process", SdsAdd(sds, SDS{"type": "ping"}), "")
+ state.Ctx.LogD("sp-process", append(les, LE{"Type", "ping"}), "")
case SPTypeInfo:
infosGot = true
- sdsp := SdsAdd(sds, SDS{"type": "info"})
- state.Ctx.LogD("sp-process", sdsp, "unmarshaling packet")
+ lesp := append(les, LE{"Type", "info"})
+ state.Ctx.LogD("sp-process", lesp, "unmarshaling packet")
var info SPInfo
if _, err = xdr.Unmarshal(r, &info); err != nil {
- state.Ctx.LogE("sp-process", sdsp, err, "")
+ state.Ctx.LogE("sp-process", lesp, err, "")
return nil, err
}
- sdsp = SdsAdd(sds, SDS{
- "pkt": ToBase32(info.Hash[:]),
- "size": int64(info.Size),
- "nice": int(info.Nice),
- })
+ lesp = append(lesp, LEs{
+ {"Pkt", Base32Codec.EncodeToString(info.Hash[:])},
+ {"Size", int64(info.Size)},
+ {"Nice", int(info.Nice)},
+ }...)
if !state.listOnly && info.Nice > state.Nice {
- state.Ctx.LogD("sp-process", sdsp, "too nice")
+ state.Ctx.LogD("sp-process", lesp, "too nice")
continue
}
- state.Ctx.LogD("sp-process", sdsp, "received")
+ state.Ctx.LogD("sp-process", lesp, "received")
if !state.listOnly && state.xxOnly == TTx {
continue
}
state.Lock()
state.infosTheir[*info.Hash] = &info
state.Unlock()
- state.Ctx.LogD("sp-process", sdsp, "stating part")
+ state.Ctx.LogD("sp-process", lesp, "stating part")
pktPath := filepath.Join(
state.Ctx.Spool,
state.Node.Id.String(),
string(TRx),
- ToBase32(info.Hash[:]),
+ Base32Codec.EncodeToString(info.Hash[:]),
)
if _, err = os.Stat(pktPath); err == nil {
- state.Ctx.LogI("sp-info", sdsp, "already done")
+ state.Ctx.LogI("sp-info", lesp, "already done")
if !state.listOnly {
replies = append(replies, MarshalSP(SPTypeDone, SPDone{info.Hash}))
}
continue
}
if _, err = os.Stat(pktPath + SeenSuffix); err == nil {
- state.Ctx.LogI("sp-info", sdsp, "already seen")
+ state.Ctx.LogI("sp-info", lesp, "already seen")
if !state.listOnly {
replies = append(replies, MarshalSP(SPTypeDone, SPDone{info.Hash}))
}
offset = fi.Size()
}
if !state.Ctx.IsEnoughSpace(int64(info.Size) - offset) {
- state.Ctx.LogI("sp-info", sdsp, "not enough space")
+ state.Ctx.LogI("sp-info", lesp, "not enough space")
continue
}
- state.Ctx.LogI(
- "sp-info",
- SdsAdd(sdsp, SDS{"offset": offset}),
- "",
- )
+ state.Ctx.LogI("sp-info", append(lesp, LE{"Offset", offset}), "")
if !state.listOnly && (state.onlyPkts == nil || state.onlyPkts[*info.Hash]) {
replies = append(replies, MarshalSP(
SPTypeFreq,
))
}
case SPTypeFile:
- sdsp := SdsAdd(sds, SDS{"type": "file"})
- state.Ctx.LogD("sp-process", sdsp, "unmarshaling packet")
+ lesp := append(les, LE{"Type", "file"})
+ state.Ctx.LogD("sp-process", lesp, "unmarshaling packet")
var file SPFile
if _, err = xdr.Unmarshal(r, &file); err != nil {
- state.Ctx.LogE("sp-process", SdsAdd(sds, SDS{"type": "file"}), err, "")
+ state.Ctx.LogE("sp-process", lesp, err, "")
return nil, err
}
- sdsp["xx"] = string(TRx)
- sdsp["pkt"] = ToBase32(file.Hash[:])
- sdsp["size"] = len(file.Payload)
+ lesp = append(lesp, LEs{
+ {"XX", string(TRx)},
+ {"Pkt", Base32Codec.EncodeToString(file.Hash[:])},
+ {"Size", len(file.Payload)},
+ }...)
dirToSync := filepath.Join(
state.Ctx.Spool,
state.Node.Id.String(),
string(TRx),
)
- filePath := filepath.Join(dirToSync, ToBase32(file.Hash[:]))
- state.Ctx.LogD("sp-file", sdsp, "opening part")
- fd, err := os.OpenFile(
- filePath+PartSuffix,
- os.O_RDWR|os.O_CREATE,
- os.FileMode(0666),
- )
- if err != nil {
- state.Ctx.LogE("sp-file", sdsp, err, "")
- return nil, err
+ filePath := filepath.Join(dirToSync, Base32Codec.EncodeToString(file.Hash[:]))
+ filePathPart := filePath + PartSuffix
+ state.Ctx.LogD("sp-file", lesp, "opening part")
+ fdAndFullSize, exists := state.fds[filePathPart]
+ var fd *os.File
+ if exists {
+ fd = fdAndFullSize.fd
+ } else {
+ fd, err = os.OpenFile(
+ filePathPart,
+ os.O_RDWR|os.O_CREATE,
+ os.FileMode(0666),
+ )
+ if err != nil {
+ state.Ctx.LogE("sp-file", lesp, err, "")
+ return nil, err
+ }
+ state.fds[filePathPart] = FdAndFullSize{fd: fd}
}
- state.Ctx.LogD(
- "sp-file",
- SdsAdd(sdsp, SDS{"offset": file.Offset}),
- "seeking",
- )
+ state.Ctx.LogD("sp-file", append(lesp, LE{"Offset", file.Offset}), "seeking")
if _, err = fd.Seek(int64(file.Offset), io.SeekStart); err != nil {
- state.Ctx.LogE("sp-file", sdsp, err, "")
- fd.Close()
+ state.Ctx.LogE("sp-file", lesp, err, "")
+ state.closeFd(filePathPart)
return nil, err
}
- state.Ctx.LogD("sp-file", sdsp, "writing")
- _, err = fd.Write(file.Payload)
- if err != nil {
- state.Ctx.LogE("sp-file", sdsp, err, "")
- fd.Close()
+ state.Ctx.LogD("sp-file", lesp, "writing")
+ if _, err = fd.Write(file.Payload); err != nil {
+ state.Ctx.LogE("sp-file", lesp, err, "")
+ state.closeFd(filePathPart)
return nil, err
}
ourSize := int64(file.Offset + uint64(len(file.Payload)))
- sdsp["size"] = ourSize
+ lesp[len(lesp)-1].V = ourSize
fullsize := int64(0)
state.RLock()
infoTheir, ok := state.infosTheir[*file.Hash]
if ok {
fullsize = int64(infoTheir.Size)
}
- sdsp["fullsize"] = fullsize
+ lesp = append(lesp, LE{"FullSize", fullsize})
if state.Ctx.ShowPrgrs {
- Progress("Rx", sdsp)
+ Progress("Rx", lesp)
}
if fullsize != ourSize {
- fd.Close()
continue
}
- spWorkersGroup.Wait()
- spWorkersGroup.Add(1)
+ <-spCheckerToken
go func() {
+ defer func() {
+ spCheckerToken <- struct{}{}
+ }()
if err := fd.Sync(); err != nil {
- state.Ctx.LogE("sp-file", sdsp, err, "sync")
- fd.Close()
+ state.Ctx.LogE("sp-file", lesp, err, "sync")
+ state.closeFd(filePathPart)
return
}
state.wg.Add(1)
defer state.wg.Done()
- fd.Seek(0, io.SeekStart)
- state.Ctx.LogD("sp-file", sdsp, "checking")
- gut, err := Check(fd, file.Hash[:], sdsp, state.Ctx.ShowPrgrs)
- fd.Close()
+ if _, err = fd.Seek(0, io.SeekStart); err != nil {
+ state.closeFd(filePathPart)
+ state.Ctx.LogE("sp-file", lesp, err, "")
+ return
+ }
+ state.Ctx.LogD("sp-file", lesp, "checking")
+ gut, err := Check(fd, file.Hash[:], lesp, state.Ctx.ShowPrgrs)
+ state.closeFd(filePathPart)
if err != nil || !gut {
- state.Ctx.LogE("sp-file", sdsp, errors.New("checksum mismatch"), "")
+ state.Ctx.LogE("sp-file", lesp, errors.New("checksum mismatch"), "")
return
}
- state.Ctx.LogI("sp-done", SdsAdd(sdsp, SDS{"xx": string(TRx)}), "")
+ state.Ctx.LogI("sp-done", lesp, "")
if err = os.Rename(filePath+PartSuffix, filePath); err != nil {
- state.Ctx.LogE("sp-file", sdsp, err, "rename")
+ state.Ctx.LogE("sp-file", lesp, err, "rename")
return
}
if err = DirSync(dirToSync); err != nil {
- state.Ctx.LogE("sp-file", sdsp, err, "sync")
+ state.Ctx.LogE("sp-file", lesp, err, "sync")
return
}
state.Lock()
delete(state.infosTheir, *file.Hash)
state.Unlock()
- spWorkersGroup.Done()
state.wg.Add(1)
go func() {
state.payloads <- MarshalSP(SPTypeDone, SPDone{file.Hash})
}()
}()
case SPTypeDone:
- sdsp := SdsAdd(sds, SDS{"type": "done"})
- state.Ctx.LogD("sp-process", sdsp, "unmarshaling packet")
+ lesp := append(les, LE{"Type", "done"})
+ state.Ctx.LogD("sp-process", lesp, "unmarshaling packet")
var done SPDone
if _, err = xdr.Unmarshal(r, &done); err != nil {
- state.Ctx.LogE("sp-process", SdsAdd(sds, SDS{"type": "done"}), err, "")
+ state.Ctx.LogE("sp-process", lesp, err, "")
return nil, err
}
- sdsp["pkt"] = ToBase32(done.Hash[:])
- state.Ctx.LogD("sp-done", sdsp, "removing")
+ lesp = append(lesp, LE{"Pkt", Base32Codec.EncodeToString(done.Hash[:])})
+ state.Ctx.LogD("sp-done", lesp, "removing")
err := os.Remove(filepath.Join(
state.Ctx.Spool,
state.Node.Id.String(),
string(TTx),
- ToBase32(done.Hash[:]),
+ Base32Codec.EncodeToString(done.Hash[:]),
))
- sdsp["xx"] = string(TTx)
+ lesp = append(lesp, LE{"XX", string(TTx)})
if err == nil {
- state.Ctx.LogI("sp-done", sdsp, "")
+ state.Ctx.LogI("sp-done", lesp, "")
} else {
- state.Ctx.LogE("sp-done", sdsp, err, "")
+ state.Ctx.LogE("sp-done", lesp, err, "")
}
case SPTypeFreq:
- sdsp := SdsAdd(sds, SDS{"type": "freq"})
- state.Ctx.LogD("sp-process", sdsp, "unmarshaling packet")
+ lesp := append(les, LE{"Type", "freq"})
+ state.Ctx.LogD("sp-process", lesp, "unmarshaling packet")
var freq SPFreq
if _, err = xdr.Unmarshal(r, &freq); err != nil {
- state.Ctx.LogE("sp-process", sdsp, err, "")
+ state.Ctx.LogE("sp-process", lesp, err, "")
return nil, err
}
- sdsp["pkt"] = ToBase32(freq.Hash[:])
- sdsp["offset"] = freq.Offset
- state.Ctx.LogD("sp-process", sdsp, "queueing")
+ lesp = append(lesp, LE{"Pkt", Base32Codec.EncodeToString(freq.Hash[:])})
+ lesp = append(lesp, LE{"Offset", freq.Offset})
+ state.Ctx.LogD("sp-process", lesp, "queueing")
nice, exists := state.infosOurSeen[*freq.Hash]
if exists {
if state.onlyPkts == nil || !state.onlyPkts[*freq.Hash] {
state.queueTheir[insertIdx] = &FreqWithNice{&freq, nice}
state.Unlock()
} else {
- state.Ctx.LogD("sp-process", sdsp, "skipping")
+ state.Ctx.LogD("sp-process", lesp, "skipping")
}
} else {
- state.Ctx.LogD("sp-process", sdsp, "unknown")
+ state.Ctx.LogD("sp-process", lesp, "unknown")
}
default:
state.Ctx.LogE(
"sp-process",
- SdsAdd(sds, SDS{"type": head.Type}),
+ append(les, LE{"Type", head.Type}),
errors.New("unknown type"),
"",
)
return nil, BadPktType
}
- if head.Type != SPTypePing {
- state.RxLastNonPing = state.RxLastSeen
- }
}
if infosGot {
var pkts int
size += info.Size
}
state.RUnlock()
- state.Ctx.LogI("sp-infos", SDS{
- "xx": string(TRx),
- "node": state.Node.Id,
- "pkts": pkts,
- "size": int64(size),
+ state.Ctx.LogI("sp-infos", LEs{
+ {"XX", string(TRx)},
+ {"Node", state.Node.Id},
+ {"Pkts", pkts},
+ {"Size", int64(size)},
}, "")
}
return payloadsSplit(replies), nil