X-Git-Url: http://www.git.cypherpunks.ru/?a=blobdiff_plain;f=src%2Fcmd%2Fnncp-toss%2Fmain.go;h=4eca181635d2281eef161407fed31e93dd347214;hb=0367cce2741e1ce6a89a49fd5c4e9df6005c9744;hp=5cf0a933496fc546ba679dec8314917033253134;hpb=1d2ce674b042d07fd9b37a46578c8b62bb0345b7;p=nncp.git

diff --git a/src/cmd/nncp-toss/main.go b/src/cmd/nncp-toss/main.go
index 5cf0a93..4eca181 100644
--- a/src/cmd/nncp-toss/main.go
+++ b/src/cmd/nncp-toss/main.go
@@ -1,6 +1,6 @@
 /*
 NNCP -- Node to Node copy, utilities for store-and-forward data exchange
-Copyright (C) 2016-2019 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2016-2022 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
@@ -23,9 +23,10 @@ import (
 	"fmt"
 	"log"
 	"os"
+	"path/filepath"
 	"time"
 
-	"go.cypherpunks.ru/nncp/v5"
+	"go.cypherpunks.ru/nncp/v8"
 )
 
 func usage() {
@@ -41,12 +42,13 @@ func main() {
 		nodeRaw   = flag.String("node", "", "Process only that node")
 		niceRaw   = flag.String("nice", nncp.NicenessFmt(255), "Minimal required niceness")
 		dryRun    = flag.Bool("dryrun", false, "Do not actually write any tossed data")
-		doSeen    = flag.Bool("seen", false, "Create .seen files")
+		doSeen    = flag.Bool("seen", false, "Create seen/ files")
 		cycle     = flag.Uint("cycle", 0, "Repeat tossing after N seconds in infinite loop")
-		noFile    = flag.Bool("nofile", false, "Do not process packets with type: file")
-		noFreq    = flag.Bool("nofreq", false, "Do not process packets with type: freq")
-		noExec    = flag.Bool("noexec", false, "Do not process packets with type: exec")
-		noTrns    = flag.Bool("notrns", false, "Do not process packets with type: transitional")
+		noFile    = flag.Bool("nofile", false, "Do not process \"file\" packets")
+		noFreq    = flag.Bool("nofreq", false, "Do not process \"freq\" packets")
+		noExec    = flag.Bool("noexec", false, "Do not process \"exec\" packets")
+		noTrns    = flag.Bool("notrns", false, "Do not process \"transitional\" packets")
+		noArea    = flag.Bool("noarea", false, "Do not process \"area\" packets")
 		spoolPath = flag.String("spool", "", "Override path to spool")
 		logPath   = flag.String("log", "", "Override path to logfile")
 		quiet     = flag.Bool("quiet", false, "Print only errors")
@@ -56,6 +58,7 @@ func main() {
 		version   = flag.Bool("version", false, "Print version information")
 		warranty  = flag.Bool("warranty", false, "Print warranty information")
 	)
+	log.SetFlags(log.Lshortfile)
 	flag.Usage = usage
 	flag.Parse()
 	if *warranty {
@@ -97,28 +100,65 @@ func main() {
 
 	ctx.Umask()
 
-Cycle:
-	isBad := false
+	if *cycle == 0 {
+		isBad := false
+		for nodeId, node := range ctx.Neigh {
+			if nodeOnly != nil && nodeId != *nodeOnly.Id {
+				continue
+			}
+			isBad = ctx.Toss(
+				node.Id,
+				nncp.TRx,
+				nice,
+				*dryRun, *doSeen, *noFile, *noFreq, *noExec, *noTrns, *noArea,
+			) || isBad
+			if nodeId == *ctx.SelfId {
+				isBad = ctx.Toss(
+					node.Id,
+					nncp.TTx,
+					nice,
+					*dryRun, false, true, true, true, true, *noArea,
+				) || isBad
+			}
+		}
+		if isBad {
+			os.Exit(1)
+		}
+		return
+	}
+
+	nodeIds := make(chan *nncp.NodeId)
 	for nodeId, node := range ctx.Neigh {
 		if nodeOnly != nil && nodeId != *nodeOnly.Id {
 			continue
 		}
-		isBad = ctx.Toss(
-			node.Id,
-			nice,
-			*dryRun,
-			*doSeen,
-			*noFile,
-			*noFreq,
-			*noExec,
-			*noTrns,
+		dw, err := ctx.NewDirWatcher(
+			filepath.Join(ctx.Spool, node.Id.String(), string(nncp.TRx)),
+			time.Second*time.Duration(*cycle),
 		)
+		if err != nil {
+			log.Fatalln(err)
+		}
+		go func(nodeId *nncp.NodeId) {
+			for range dw.C {
+				nodeIds <- nodeId
+			}
+		}(node.Id)
 	}
-	if *cycle > 0 {
-		time.Sleep(time.Duration(*cycle) * time.Second)
-		goto Cycle
-	}
-	if isBad {
-		os.Exit(1)
+	for nodeId := range nodeIds {
+		ctx.Toss(
+			nodeId,
+			nncp.TRx,
+			nice,
+			*dryRun, *doSeen, *noFile, *noFreq, *noExec, *noTrns, *noArea,
+		)
+		if *nodeId == *ctx.SelfId {
+			ctx.Toss(
+				nodeId,
+				nncp.TTx,
+				nice,
+				*dryRun, false, true, true, true, true, *noArea,
+			)
+		}
 	}
 }