]> Cypherpunks.ru repositories - nncp.git/blobdiff - src/cypherpunks.ru/nncp/cmd/nncp-bundle/main.go
Replace Twofish/HKDF with ChaCha20/BLAKE2X for speed and simplicity
[nncp.git] / src / cypherpunks.ru / nncp / cmd / nncp-bundle / main.go
index 10682511f627af9190d78c0a1935ea3588e2e91f..bfa48169e6edd8dc872968b07287738c3019cbeb 100644 (file)
@@ -46,24 +46,27 @@ func usage() {
        fmt.Fprintf(os.Stderr, nncp.UsageHeader())
        fmt.Fprintln(os.Stderr, "nncp-bundle -- Create/digest stream of NNCP encrypted packets\n")
        fmt.Fprintf(os.Stderr, "Usage: %s [options] -tx [-delete] NODE [NODE ...] > ...\n", os.Args[0])
-       fmt.Fprintf(os.Stderr, "       %s [options] -rx -delete [NODE ...] < ...\n", os.Args[0])
-       fmt.Fprintf(os.Stderr, "       %s [options] -rx [-check] [NODE ...] < ...\n", os.Args[0])
+       fmt.Fprintf(os.Stderr, "       %s [options] -rx -delete [-dryrun] [NODE ...] < ...\n", os.Args[0])
+       fmt.Fprintf(os.Stderr, "       %s [options] -rx [-check] [-dryrun] [NODE ...] < ...\n", os.Args[0])
        fmt.Fprintln(os.Stderr, "Options:")
        flag.PrintDefaults()
 }
 
 func main() {
        var (
-               cfgPath  = flag.String("cfg", nncp.DefaultCfgPath, "Path to configuration file")
-               niceRaw  = flag.Int("nice", 255, "Minimal required niceness")
-               doRx     = flag.Bool("rx", false, "Receive packets")
-               doTx     = flag.Bool("tx", false, "Transfer packets")
-               doDelete = flag.Bool("delete", false, "Delete transferred packets")
-               doCheck  = flag.Bool("check", false, "Check integrity while receiving")
-               quiet    = flag.Bool("quiet", false, "Print only errors")
-               debug    = flag.Bool("debug", false, "Print debug messages")
-               version  = flag.Bool("version", false, "Print version information")
-               warranty = flag.Bool("warranty", false, "Print warranty information")
+               cfgPath   = flag.String("cfg", nncp.DefaultCfgPath, "Path to configuration file")
+               niceRaw   = flag.Int("nice", 255, "Minimal required niceness")
+               doRx      = flag.Bool("rx", false, "Receive packets")
+               doTx      = flag.Bool("tx", false, "Transfer packets")
+               doDelete  = flag.Bool("delete", false, "Delete transferred packets")
+               doCheck   = flag.Bool("check", false, "Check integrity while receiving")
+               dryRun    = flag.Bool("dryrun", false, "Do not writings")
+               spoolPath = flag.String("spool", "", "Override path to spool")
+               logPath   = flag.String("log", "", "Override path to logfile")
+               quiet     = flag.Bool("quiet", false, "Print only errors")
+               debug     = flag.Bool("debug", false, "Print debug messages")
+               version   = flag.Bool("version", false, "Print version information")
+               warranty  = flag.Bool("warranty", false, "Print warranty information")
        )
        flag.Usage = usage
        flag.Parse()
@@ -86,16 +89,10 @@ func main() {
                log.Fatalln("At least one of -rx and -tx must be specified")
        }
 
-       cfgRaw, err := ioutil.ReadFile(nncp.CfgPathFromEnv(cfgPath))
+       ctx, err := nncp.CtxFromCmdline(*cfgPath, *spoolPath, *logPath, *quiet, *debug)
        if err != nil {
-               log.Fatalln("Can not read config:", err)
+               log.Fatalln("Error during initialization:", err)
        }
-       ctx, err := nncp.CfgParse(cfgRaw)
-       if err != nil {
-               log.Fatalln("Can not parse config:", err)
-       }
-       ctx.Quiet = *quiet
-       ctx.Debug = *debug
 
        nodeIds := make(map[nncp.NodeId]struct{}, flag.NArg())
        for i := 0; i < flag.NArg(); i++ {
@@ -215,7 +212,7 @@ func main() {
                                ctx.LogD("nncp-bundle", sds, "Bad packet structure")
                                continue
                        }
-                       if pktEnc.Magic != nncp.MagicNNCPEv2 {
+                       if pktEnc.Magic != nncp.MagicNNCPEv3 {
                                ctx.LogD("nncp-bundle", sds, "Bad packet magic number")
                                continue
                        }
@@ -256,7 +253,9 @@ func main() {
                                }
                                if nncp.ToBase32(hsh.Sum(nil)) == pktName {
                                        ctx.LogI("nncp-bundle", sds, "removed")
-                                       os.Remove(dstPath)
+                                       if !*dryRun {
+                                               os.Remove(dstPath)
+                                       }
                                } else {
                                        ctx.LogE("nncp-bundle", sds, "bad checksum")
                                }
@@ -280,55 +279,78 @@ func main() {
                                ctx.LogD("nncp-bundle", sds, "Packet already exists")
                                continue
                        }
-                       if _, err = os.Stat(dstPath + nncp.SeenPostfix); err == nil || !os.IsNotExist(err) {
+                       if _, err = os.Stat(dstPath + nncp.SeenSuffix); err == nil || !os.IsNotExist(err) {
                                ctx.LogD("nncp-bundle", sds, "Packet already exists")
                                continue
                        }
                        if *doCheck {
-                               tmp, err := ctx.NewTmpFileWHash()
-                               if err != nil {
-                                       log.Fatalln("Error during temporary file creation:", err)
-                               }
-                               if _, err = tmp.W.Write(pktEncBuf); err != nil {
-                                       log.Fatalln("Error during writing:", err)
-                               }
-                               if _, err = io.Copy(tmp.W, tarR); err != nil {
-                                       log.Fatalln("Error during copying:", err)
-                               }
-                               if err = tmp.W.Flush(); err != nil {
-                                       log.Fatalln("Error during flusing:", err)
-                               }
-                               if nncp.ToBase32(tmp.Hsh.Sum(nil)) == pktName {
-                                       if err = tmp.Commit(selfPath); err != nil {
-                                               log.Fatalln("Error during commiting:", err)
+                               if *dryRun {
+                                       hsh, err := blake2b.New256(nil)
+                                       if err != nil {
+                                               log.Fatalln("Error during hasher creation:", err)
+                                       }
+                                       if _, err = hsh.Write(pktEncBuf); err != nil {
+                                               log.Fatalln("Error during writing:", err)
+                                       }
+                                       if _, err = io.Copy(hsh, tarR); err != nil {
+                                               log.Fatalln("Error during copying:", err)
+                                       }
+                                       if nncp.ToBase32(hsh.Sum(nil)) != pktName {
+                                               ctx.LogE("nncp-bundle", sds, "bad checksum")
+                                               continue
                                        }
                                } else {
-                                       ctx.LogE("nncp-bundle", sds, "bad checksum")
-                                       tmp.Cancel()
-                                       continue
+                                       tmp, err := ctx.NewTmpFileWHash()
+                                       if err != nil {
+                                               log.Fatalln("Error during temporary file creation:", err)
+                                       }
+                                       if _, err = tmp.W.Write(pktEncBuf); err != nil {
+                                               log.Fatalln("Error during writing:", err)
+                                       }
+                                       if _, err = io.Copy(tmp.W, tarR); err != nil {
+                                               log.Fatalln("Error during copying:", err)
+                                       }
+                                       if err = tmp.W.Flush(); err != nil {
+                                               log.Fatalln("Error during flusing:", err)
+                                       }
+                                       if nncp.ToBase32(tmp.Hsh.Sum(nil)) == pktName {
+                                               if err = tmp.Commit(selfPath); err != nil {
+                                                       log.Fatalln("Error during commiting:", err)
+                                               }
+                                       } else {
+                                               ctx.LogE("nncp-bundle", sds, "bad checksum")
+                                               tmp.Cancel()
+                                               continue
+                                       }
                                }
                        } else {
-                               tmp, err := ctx.NewTmpFile()
-                               if err != nil {
-                                       log.Fatalln("Error during temporary file creation:", err)
-                               }
-                               bufTmp := bufio.NewWriterSize(tmp, CopyBufSize)
-                               if _, err = bufTmp.Write(pktEncBuf); err != nil {
-                                       log.Fatalln("Error during writing:", err)
-                               }
-                               if _, err = io.Copy(bufTmp, tarR); err != nil {
-                                       log.Fatalln("Error during copying:", err)
-                               }
-                               if err = bufTmp.Flush(); err != nil {
-                                       log.Fatalln("Error during flushing:", err)
-                               }
-                               tmp.Sync()
-                               tmp.Close()
-                               if err = os.MkdirAll(selfPath, os.FileMode(0700)); err != nil {
-                                       log.Fatalln("Error during mkdir:", err)
-                               }
-                               if err = os.Rename(tmp.Name(), dstPath); err != nil {
-                                       log.Fatalln("Error during renaming:", err)
+                               if *dryRun {
+                                       if _, err = io.Copy(ioutil.Discard, tarR); err != nil {
+                                               log.Fatalln("Error during copying:", err)
+                                       }
+                               } else {
+                                       tmp, err := ctx.NewTmpFile()
+                                       if err != nil {
+                                               log.Fatalln("Error during temporary file creation:", err)
+                                       }
+                                       bufTmp := bufio.NewWriterSize(tmp, CopyBufSize)
+                                       if _, err = bufTmp.Write(pktEncBuf); err != nil {
+                                               log.Fatalln("Error during writing:", err)
+                                       }
+                                       if _, err = io.Copy(bufTmp, tarR); err != nil {
+                                               log.Fatalln("Error during copying:", err)
+                                       }
+                                       if err = bufTmp.Flush(); err != nil {
+                                               log.Fatalln("Error during flushing:", err)
+                                       }
+                                       tmp.Sync()
+                                       tmp.Close()
+                                       if err = os.MkdirAll(selfPath, os.FileMode(0700)); err != nil {
+                                               log.Fatalln("Error during mkdir:", err)
+                                       }
+                                       if err = os.Rename(tmp.Name(), dstPath); err != nil {
+                                               log.Fatalln("Error during renaming:", err)
+                                       }
                                }
                        }
                        ctx.LogI("nncp-bundle", nncp.SdsAdd(sds, nncp.SDS{