Forbid any later GNU GPL versions autousage

[nncp.git] / src / cypherpunks.ru / nncp / cmd / nncp-call / main.go

diff --git a/src/cypherpunks.ru/nncp/cmd/nncp-call/main.go b/src/cypherpunks.ru/nncp/cmd/nncp-call/main.go
index d853443e381a469cc455ad676e895707e32cb9e4..ede5a4864208c7ae72fd9873a366d0bbdf1a42c6 100644 (file)
--- a/src/cypherpunks.ru/nncp/cmd/nncp-call/main.go
+++ b/src/cypherpunks.ru/nncp/cmd/nncp-call/main.go
@@ -4,8 +4,7 @@ Copyright (C) 2016-2019 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
-the Free Software Foundation, either version 3 of the License, or
-(at your option) any later version.
+the Free Software Foundation, version 3 of the License.
 
 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ -39,19 +38,20 @@ func usage() {
 
 func main() {
        var (
-               cfgPath   = flag.String("cfg", nncp.DefaultCfgPath, "Path to configuration file")
-               niceRaw   = flag.String("nice", nncp.NicenessFmt(255), "Minimal required niceness")
-               rxOnly    = flag.Bool("rx", false, "Only receive packets")
-               txOnly    = flag.Bool("tx", false, "Only transmit packets")
-               listOnly  = flag.Bool("list", false, "Only list remote packets")
-               rxRate    = flag.Int("rxrate", 0, "Maximal receive rate, pkts/sec")
-               txRate    = flag.Int("txrate", 0, "Maximal transmit rate, pkts/sec")
-               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")
+               cfgPath     = flag.String("cfg", nncp.DefaultCfgPath, "Path to configuration file")
+               niceRaw     = flag.String("nice", nncp.NicenessFmt(255), "Minimal required niceness")
+               rxOnly      = flag.Bool("rx", false, "Only receive packets")
+               txOnly      = flag.Bool("tx", false, "Only transmit packets")
+               listOnly    = flag.Bool("list", false, "Only list remote packets")
+               onlyPktsRaw = flag.String("pkts", "", "Recieve only that packets, comma separated")
+               rxRate      = flag.Int("rxrate", 0, "Maximal receive rate, pkts/sec")
+               txRate      = flag.Int("txrate", 0, "Maximal transmit rate, pkts/sec")
+               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")
 
                onlineDeadline = flag.Uint("onlinedeadline", 0, "Override onlinedeadline option")
                maxOnlineTime  = flag.Uint("maxonlinetime", 0, "Override maxonlinetime option")
@@ -124,6 +124,21 @@ func main() {
                }
        }
 
+       var onlyPkts map[[32]byte]bool
+       if len(*onlyPktsRaw) > 0 {
+               splitted = strings.Split(*onlyPktsRaw, ",")
+               onlyPkts = make(map[[32]byte]bool, len(splitted))
+               for _, pktIdRaw := range splitted {
+                       pktId, err := nncp.FromBase32(pktIdRaw)
+                       if err != nil {
+                               log.Fatalln("Invalid packet specified: ", err)
+                       }
+                       pktIdArr := new([32]byte)
+                       copy(pktIdArr[:], pktId)
+                       onlyPkts[*pktIdArr] = true
+               }
+       }
+
        if !ctx.CallNode(
                node,
                addrs,
@@ -134,6 +149,7 @@ func main() {
                *onlineDeadline,
                *maxOnlineTime,
                *listOnly,
+               onlyPkts,
        ) {
                os.Exit(1)
        }