abbel/main.go

144 lines
3.1 KiB
Go

package main
import (
"fmt"
"log"
"git.freifunk-franken.de/jkimmel/abbel/capture"
"git.freifunk-franken.de/jkimmel/abbel/packet"
"git.freifunk-franken.de/jkimmel/abbel/tlv"
"inet.af/netaddr"
)
type options struct {
Group string
Port uint16
Ifs []string
UseCapture bool
}
func parseOpts() (options, error) {
var opt options
var err error
opt = options{
Group: "ff02:0:0:0:0:0:1:6",
Port: 6696,
Ifs: []string{"babel"},
UseCapture: false,
}
return opt, err
}
type updatesKey struct {
prefix netaddr.IPPrefix
routerID tlv.RouterID
nexthop netaddr.IP
}
var (
updates = map[updatesKey]tlv.Update{}
)
type BabelPacketConn interface {
ReadFrom(b []byte) (body []byte, src netaddr.IP, ifindex int, err error)
Close() error
}
func run(opt options) error {
var err error
var conn BabelPacketConn
if opt.UseCapture {
conn, err = capture.FromInterface(opt.Ifs[0])
} else {
conn, err = packet.Listen(opt.Group, opt.Port, opt.Ifs...)
}
if err != nil {
return err
}
defer conn.Close()
buf := [4096]byte{}
var s tlv.PacketDecoder
for {
b, src, ifindex, err := conn.ReadFrom(buf[:])
if err != nil {
fmt.Println("Skipping packet:", err)
continue
}
fmt.Print("\x1B[1m")
fmt.Printf("Packet size %4d from %s", len(b), src)
fmt.Println("\x1B[0m")
s.Reset(b, src, ifindex)
for s.Scan() {
switch t := s.TLV().(type) {
case tlv.NextHop:
//fmt.Printf("%12s %s\n", t.T(), t.Address)
case tlv.Hello:
fmt.Printf("% 12s %+v\n", t.T(), t)
case tlv.Update:
key := updatesKey{t.Prefix, t.RouterID, t.NextHop}
tmetric := t.Metric
diff := int(tmetric) - int(updates[key].Metric)
updates[key] = t
if diff == 0 && tmetric != 0xFFFF {
break
}
if t.Metric == 0xFFFF {
fmt.Print("\x1B[31m")
} else {
fmt.Print("\x1B[36m")
}
fmt.Printf("% 12s %-43s%+d %s\x1B[0m\n", t.T(), t.Prefix, diff, t.FormatHeader())
case tlv.RouterID:
//fmt.Printf("%12s %s\n", t.T(), t)
case tlv.RouteRequest:
fmt.Printf("%12s %s\n", t.T(), t.Prefix)
case tlv.Raw:
fmt.Printf(" T: %12s (%2d), L: %3d\n", t.T(), t.T(), len(t.V()))
case tlv.IHU:
fmt.Printf("%12s %-43sRxcost %5d Interval %4d \n", t.T(), t.Address, t.Rxcost, t.Interval)
default:
if t != nil {
fmt.Println("Unknown TLV", t.T(), t)
} else {
fmt.Println("got nil TLV")
}
}
subtlv := s.SubTLV()
for len(subtlv) > 0 {
switch tlv.SubType(subtlv[0]) {
case tlv.SubTypeSourcePrefix:
var pfx tlv.SourcePrefix
pfx, subtlv, err = tlv.SourcePrefixFromBytes(subtlv[2:])
fmt.Print("\x1B[7;36m")
fmt.Printf("%12s %s\x1B[0m\n", pfx.T(), pfx)
case tlv.SubTypeDiversity, tlv.SubTypePad1, tlv.SubTypePadN, tlv.SubTypeTimestamp:
fmt.Print("\x1B[7m")
fmt.Printf("%12s: %v\x1B[0m\n", "Subtlv", subtlv)
subtlv = nil
default:
panic("unknown subtlv type")
}
}
}
if s.Err() != nil {
fmt.Println(err)
}
}
}
func main() {
opt, err := parseOpts()
if err != nil {
log.Fatal(err)
}
if err := run(opt); err != nil {
log.Fatal(err)
}
}