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/>.
29 "golang.org/x/crypto/blake2b"
32 const NoCKSuffix = ".nock"
34 func Check(src io.Reader, checksum []byte, les LEs, showPrgrs bool) (bool, error) {
35 hsh, err := blake2b.New256(nil)
39 if _, err = CopyProgressed(hsh, bufio.NewReader(src), "check", les, showPrgrs); err != nil {
42 return bytes.Compare(hsh.Sum(nil), checksum) == 0, nil
45 func (ctx *Ctx) checkXxIsBad(nodeId *NodeId, xx TRxTx) bool {
47 for job := range ctx.Jobs(nodeId, xx) {
51 {"Pkt", Base32Codec.EncodeToString(job.HshValue[:])},
52 {"FullSize", job.Size},
54 fd, err := os.Open(job.Path)
56 ctx.LogE("check", les, err, "")
59 gut, err := Check(fd, job.HshValue[:], les, ctx.ShowPrgrs)
60 fd.Close() // #nosec G104
62 ctx.LogE("check", les, err, "")
67 ctx.LogE("check", les, errors.New("bad"), "")
73 func (ctx *Ctx) Check(nodeId *NodeId) bool {
74 return !(ctx.checkXxIsBad(nodeId, TRx) || ctx.checkXxIsBad(nodeId, TTx))
77 func (ctx *Ctx) CheckNoCK(nodeId *NodeId, hshValue *[32]byte) (int64, error) {
78 dirToSync := filepath.Join(ctx.Spool, nodeId.String(), string(TRx))
79 pktName := Base32Codec.EncodeToString(hshValue[:])
80 pktPath := filepath.Join(dirToSync, pktName)
81 fd, err := os.Open(pktPath + NoCKSuffix)
98 gut, err := Check(fd, hshValue[:], les, ctx.ShowPrgrs)
99 if err != nil || !gut {
100 return 0, errors.New("checksum mismatch")
102 if err = os.Rename(pktPath+NoCKSuffix, pktPath); err != nil {
105 if err = DirSync(dirToSync); err != nil {
109 if _, err = fd.Seek(0, io.SeekStart); err != nil {
112 _, pktEncRaw, err := ctx.HdrRead(fd)
116 ctx.HdrWrite(pktEncRaw, pktPath)