/*
NNCP -- Node to Node copy, utilities for store-and-forward data exchange
-Copyright (C) 2016-2017 Sergey Matveev <stargrave@stargrave.org>
+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
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
-// Call NNCP TCP daemon
+// Call NNCP TCP daemon.
package main
import (
"flag"
"fmt"
- "io/ioutil"
"log"
- "net"
"os"
- "strconv"
"strings"
"cypherpunks.ru/nncp"
func usage() {
fmt.Fprintf(os.Stderr, nncp.UsageHeader())
- fmt.Fprintln(os.Stderr, "nncp-call -- call TCP daemon\n")
+ fmt.Fprintf(os.Stderr, "nncp-call -- call TCP daemon\n\n")
fmt.Fprintf(os.Stderr, "Usage: %s [options] NODE[:ADDR] [FORCEADDR]\n", os.Args[0])
- fmt.Fprintln(os.Stderr, "You must specify either [NODE:ADDR] or [NODE FORCEADDR]\n")
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")
- rxOnly = flag.Bool("rx", false, "Only receive packets")
- txOnly = flag.Bool("tx", false, "Only transfer packets")
- debug = flag.Bool("debug", false, "Enable debugging information")
- 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")
)
flag.Usage = usage
flag.Parse()
usage()
os.Exit(1)
}
- if *niceRaw < 1 || *niceRaw > 255 {
- log.Fatalln("-nice must be between 1 and 255")
+ nice, err := nncp.NicenessParse(*niceRaw)
+ if err != nil {
+ log.Fatalln(err)
}
- nice := uint8(*niceRaw)
if *rxOnly && *txOnly {
log.Fatalln("-rx and -tx can not be set simultaneously")
}
- cfgRaw, err := ioutil.ReadFile(*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)
+ if ctx.Self == nil {
+ log.Fatalln("Config lacks private keys")
}
- ctx.Debug = *debug
splitted := strings.SplitN(flag.Arg(0), ":", 2)
node, err := ctx.FindNode(splitted[0])
if err != nil {
log.Fatalln("Invalid NODE specified:", err)
}
- if len(splitted) == 1 && flag.NArg() != 2 {
- usage()
- os.Exit(1)
- }
- var dst string
- if len(splitted) == 2 {
- var known bool
- dst, known = ctx.Neigh[*node.Id].Addrs[splitted[1]]
- if !known {
- log.Fatalln("Unknown ADDR specified")
- }
- } else {
- dst = flag.Arg(1)
+ if node.NoisePub == nil {
+ log.Fatalln("Node does not have online communication capability")
}
- conn, err := net.Dial("tcp", dst)
- if err != nil {
- log.Fatalln("Can not connect:", err)
+ if *onlineDeadline == 0 {
+ onlineDeadline = &node.OnlineDeadline
+ }
+ if *maxOnlineTime == 0 {
+ maxOnlineTime = &node.MaxOnlineTime
}
- ctx.LogD("call", nncp.SDS{"addr": dst}, "connected")
+
var xxOnly nncp.TRxTx
if *rxOnly {
xxOnly = nncp.TRx
} else if *txOnly {
xxOnly = nncp.TTx
}
- state, err := ctx.StartI(conn, node.Id, nice, &xxOnly)
- if err == nil {
- ctx.LogI("call-start", nncp.SDS{"node": state.NodeId}, "connected")
- state.Wait()
- ctx.LogI("call-finish", nncp.SDS{
- "node": state.NodeId,
- "duration": strconv.FormatInt(int64(state.Duration.Seconds()), 10),
- "rxbytes": strconv.FormatInt(state.RxBytes, 10),
- "txbytes": strconv.FormatInt(state.TxBytes, 10),
- "rxspeed": strconv.FormatInt(state.RxSpeed, 10),
- "txspeed": strconv.FormatInt(state.TxSpeed, 10),
- }, "")
+
+ var addrs []string
+ if flag.NArg() == 2 {
+ addrs = append(addrs, flag.Arg(1))
+ } else if len(splitted) == 2 {
+ addr, known := ctx.Neigh[*node.Id].Addrs[splitted[1]]
+ if !known {
+ log.Fatalln("Unknown ADDR specified")
+ }
+ addrs = append(addrs, addr)
} else {
- ctx.LogE("call-start", nncp.SDS{"node": state.NodeId, "err": err}, "")
- conn.Close()
+ for _, addr := range ctx.Neigh[*node.Id].Addrs {
+ addrs = append(addrs, addr)
+ }
+ }
+
+ 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,
+ nice,
+ xxOnly,
+ *rxRate,
+ *txRate,
+ *onlineDeadline,
+ *maxOnlineTime,
+ *listOnly,
+ onlyPkts,
+ ) {
os.Exit(1)
}
- conn.Close()
}