gotelem/cmd/xbee.go

147 lines
3.5 KiB
Go
Raw Normal View History

2023-05-05 00:15:03 +00:00
package cmd
// this file contains xbee utilities.
// we can do network discovery and netcat-like things.
import (
2023-05-07 05:00:35 +00:00
"fmt"
"io"
"os"
2023-05-07 13:39:10 +00:00
"runtime"
"strings"
2023-05-07 05:00:35 +00:00
"github.com/kschamplin/gotelem/xbee"
2023-05-05 00:15:03 +00:00
"github.com/urfave/cli/v2"
2023-05-07 05:00:35 +00:00
"go.bug.st/serial"
"golang.org/x/exp/slog"
2023-05-05 00:15:03 +00:00
)
var xbeeCmd = &cli.Command{
Name: "xbee",
Aliases: []string{"x"},
Usage: "Utilities for XBee",
Description: `
Allows for testing and debugging XBee networks and devices.
The "device" parameter is not optional, and can be any of the following formats:
tcp://192.168.4.5:8430
COM1
/dev/ttyUSB0:115200
For serial devices (COM1 and /dev/ttyUSB0), you can specify the baud rate
using a ':'. If excluded the baud rate will default to 9600. Note that
if using the native USB of the XLR Pro, the baud rate setting has no effect.
2023-05-07 05:00:35 +00:00
TCP/UDP connections require a port and will fail if one is not provided.
2023-05-05 00:15:03 +00:00
`,
Flags: []cli.Flag{
&cli.StringFlag{
2023-05-07 05:00:35 +00:00
Name: "device",
Aliases: []string{"d"},
Usage: "The XBee to connect to",
Required: true,
EnvVars: []string{"XBEE_DEVICE"},
2023-05-05 00:15:03 +00:00
},
},
Subcommands: []*cli.Command{
{
2023-05-07 05:00:35 +00:00
Name: "info",
Usage: "get information about an xbee device",
Action: xbeeInfo,
HideHelpCommand: true,
2023-05-05 00:15:03 +00:00
},
{
Name: "netcat",
Aliases: []string{"nc"},
ArgsUsage: "[addr]",
Usage: "send data from stdio over the xbee",
2023-05-07 05:00:35 +00:00
Description: `
netcat emulates the nc command. It reads data from stdin and transmits it to
[addr] on the XBee network. If [addr] is FFFF or not present, it will broadcast
the data to all listening devices. Data received from the network will be
writtend to stdout.
`,
Action: netcat,
HideHelpCommand: true,
2023-05-05 00:15:03 +00:00
},
},
}
2023-05-07 05:00:35 +00:00
func xbeeInfo(ctx *cli.Context) error {
logger := slog.New(slog.NewTextHandler(os.Stderr))
2023-05-07 13:39:10 +00:00
serialDevice, _ := serial.Open("/dev/ttyACM0", &serial.Mode{})
xb, err := xbee.NewSession(serialDevice, logger.With("name", ""))
2023-05-07 05:00:35 +00:00
if err != nil {
return cli.Exit(err, 1)
}
b, err := xb.ATCommand([2]rune{'I', 'D'}, nil, false)
if err != nil {
return cli.Exit(err, 1)
}
fmt.Println(b)
return nil
}
func netcat(ctx *cli.Context) error {
if ctx.Args().Len() < 1 {
cli.ShowSubcommandHelp(ctx)
return cli.Exit("missing [addr] argument", 1)
}
// basically create two pipes.
logger := slog.New(slog.NewTextHandler(os.Stderr))
2023-05-07 13:39:10 +00:00
transport, _ := parseDeviceString("/dev/ttyUSB0")
xb, _ := xbee.NewSession(transport, logger.With("devtype", transport.Type()))
2023-05-07 05:00:35 +00:00
sent := make(chan int64)
streamCopy := func(r io.ReadCloser, w io.WriteCloser) {
defer r.Close()
defer w.Close()
n, err := io.Copy(w, r)
if err != nil {
logger.Warn("got error copying", "err", err)
}
sent <- n
}
go streamCopy(os.Stdin, xb)
go streamCopy(xb, os.Stdout)
<-sent
return nil
}
2023-05-07 13:39:10 +00:00
type xbeeTransport struct {
io.ReadWriteCloser
devType string
}
func (xbt *xbeeTransport) Type() string {
return xbt.devType
}
// parseDeviceString parses the device parameter and sets up the associated
// device. The device is returned in an xbeeTransport which also stores
// the underlying type of the device with Type() string
func parseDeviceString(dev string) (*xbeeTransport, error) {
// FIXME: implement properly
serialDevice, _ := serial.Open(dev, &serial.Mode{})
xbt := &xbeeTransport{
ReadWriteCloser: serialDevice,
devType: "serial",
}
if strings.HasPrefix(dev, "tcp://") {
} else if strings.HasPrefix(dev, "COM") && runtime.GOOS == "windows" {
} else if strings.HasPrefix(dev, "/") && runtime.GOOS != "windows" {
}
return xbt, nil
}