80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
package cmd
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/bdlm/log"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/FreifunkBremen/yanic/respond"
|
|
"github.com/FreifunkBremen/yanic/runtime"
|
|
)
|
|
|
|
var (
|
|
wait int
|
|
port int
|
|
ipAddress string
|
|
)
|
|
|
|
// queryCmd represents the query command
|
|
var queryCmd = &cobra.Command{
|
|
Use: "query <interfaces> <destination>",
|
|
Short: "Sends a query on the interface to the destination and waits for a response",
|
|
Example: `yanic query "eth0,wlan0" "fe80::eade:27ff:dead:beef"`,
|
|
Args: cobra.ExactArgs(2),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
ifaces := strings.Split(args[0], ",")
|
|
dstAddress := net.ParseIP(args[1])
|
|
|
|
log.WithFields(map[string]interface{}{
|
|
"address": dstAddress,
|
|
"ifaces": ifaces,
|
|
}).Info("sending request")
|
|
|
|
var ifacesConfigs []respond.InterfaceConfig
|
|
for _, iface := range ifaces {
|
|
ifaceConfig := respond.InterfaceConfig{
|
|
InterfaceName: iface,
|
|
Port: port,
|
|
IPAddress: ipAddress,
|
|
}
|
|
ifacesConfigs = append(ifacesConfigs, ifaceConfig)
|
|
}
|
|
var config respond.Config
|
|
config.Interfaces = ifacesConfigs
|
|
|
|
nodes := runtime.NewNodes(&runtime.NodesConfig{})
|
|
|
|
collector := respond.NewCollector(nil, nodes, &config)
|
|
defer collector.Close()
|
|
collector.SendPacket(dstAddress)
|
|
|
|
time.Sleep(time.Second * time.Duration(wait))
|
|
|
|
for id, data := range nodes.List {
|
|
jq, err := json.Marshal(data)
|
|
if err != nil {
|
|
fmt.Printf("%s: %+v", id, data)
|
|
} else {
|
|
jqNeighbours, err := json.Marshal(data.Neighbours)
|
|
if err != nil {
|
|
fmt.Printf("%s: %s neighbours: %+v", id, string(jq), data.Neighbours)
|
|
} else {
|
|
fmt.Printf("%s: %s neighbours: %s", id, string(jq), string(jqNeighbours))
|
|
}
|
|
}
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
RootCmd.AddCommand(queryCmd)
|
|
queryCmd.Flags().IntVar(&wait, "wait", 1, "Seconds to wait for a response")
|
|
queryCmd.Flags().IntVar(&port, "port", 0, "define a port to listen (if not set or set to 0 the kernel will use a random free port at its own)")
|
|
queryCmd.Flags().StringVar(&ipAddress, "ip", "", "ip address which is used for sending (optional - without definition used the link-local address)")
|
|
}
|