func TestUnpackAllEthIPv4TCPRaw(t *testing.T) {
pkt, err := layers.UnpackAll(test_eth_ipv4_tcp_raw, packet.Eth)
if err != nil {
t.Fatalf("Error unpacking: %s", err)
}
if pkt.GetType() != packet.Eth {
t.Fatalf("Packet type mismatch, %s", pkt.GetType())
}
pkt = pkt.Payload()
if pkt.GetType() != packet.IPv4 {
t.Fatalf("Packet type mismatch, %s", pkt.GetType())
}
pkt = pkt.Payload()
if pkt.GetType() != packet.TCP {
t.Fatalf("Packet type mismatch, %s", pkt.GetType())
}
pkt = pkt.Payload()
if pkt.GetType() != packet.Raw {
t.Fatalf("Packet type mismatch, %s", pkt.GetType())
}
}
func ExampleUnpack() {
// Create the buf data
buf := []byte("random data")
// Assume Ethernet as datalink layer
pkt, err := layers.UnpackAll(buf, packet.Eth)
if err != nil {
log.Fatal(err)
}
log.Println(pkt)
}
// Capture a single packet from the given capture handle, unpack it and return
// it. This will block until a packet is received.
func Recv(c capture.Handle) (packet.Packet, error) {
buf, err := c.Capture()
if err != nil {
return nil, fmt.Errorf("Could not capture: %s", err)
}
pkt, err := layers.UnpackAll(buf, c.LinkType())
if err != nil {
return nil, fmt.Errorf("Could not unpack: %s", err)
}
return pkt, nil
}
func TestUnpackAllEthArp(t *testing.T) {
pkt, err := layers.UnpackAll(test_eth_arp, packet.Eth)
if err != nil {
t.Fatalf("Error unpacking: %s", err)
}
if pkt.GetType() != packet.Eth {
t.Fatalf("Packet type mismatch, %s", pkt.GetType())
}
if pkt.Payload().GetType() != packet.ARP {
t.Fatalf("Packet type mismatch, %s", pkt.Payload().GetType())
}
}
func TestFindLayer(t *testing.T) {
pkt, err := layers.UnpackAll(test_eth_ipv4_tcp, packet.Eth)
if err != nil {
t.Fatalf("Error unpacking: %s", err)
}
ipv4_pkt := layers.FindLayer(pkt, packet.IPv4)
if ipv4_pkt == nil || ipv4_pkt.GetType() != packet.IPv4 {
t.Fatalf("Not IPv4: %s", ipv4_pkt)
}
tcp_pkt := layers.FindLayer(pkt, packet.TCP)
if tcp_pkt == nil || tcp_pkt.GetType() != packet.TCP {
t.Fatalf("Not TCP: %s", tcp_pkt)
}
udp_pkt := layers.FindLayer(pkt, packet.UDP)
if udp_pkt != nil {
t.Fatalf("Not nil: %s", udp_pkt)
}
}
func main() {
log.SetFlags(0)
usage := `Usage: dump [options] [<expression>]
Dump the traffic on the network (like tcpdump).
Options:
-c <count> Exit after receiving count packets.
-i <iface> Listen on interface.
-r <file> Read packets from file.
-w <file> Write the raw packets to file.`
args, err := docopt.Parse(usage, nil, true, "", false)
if err != nil {
log.Fatalf("Invalid arguments: %s", err)
}
var count uint64
if args["-c"] != nil {
count, err = strconv.ParseUint(args["-c"].(string), 10, 64)
if err != nil {
log.Fatalf("Error parsing count: %s", err)
}
}
var src capture.Handle
if args["-i"] != nil {
src, err = pcap.Open(args["-i"].(string))
if err != nil {
log.Fatalf("Error opening iface: %s", err)
}
} else if args["-r"] != nil {
src, err = file.Open(args["-r"].(string))
if err != nil {
log.Fatalf("Error opening file: %s", err)
}
} else {
log.Fatalf("Must select a source (either -i or -r)")
}
defer src.Close()
var dst capture.Handle
if args["-w"] != nil {
dst, err = file.Open(args["-w"].(string))
if err != nil {
log.Fatalf("Error opening file: %s", err)
}
defer dst.Close()
}
err = src.Activate()
if err != nil {
log.Fatalf("Error activating source: %s", err)
}
if args["<expression>"] != nil {
expr := args["<expression>"].(string)
flt, err := filter.Compile(expr, src.LinkType(), false)
if err != nil {
log.Fatalf("Error parsing filter: %s", err)
}
defer flt.Cleanup()
err = src.ApplyFilter(flt)
if err != nil {
log.Fatalf("Error appying filter: %s", err)
}
}
var i uint64
for {
buf, err := src.Capture()
if err != nil {
log.Fatalf("Error: %s", err)
break
}
if buf == nil {
break
}
i++
if dst == nil {
rcv_pkt, err := layers.UnpackAll(buf, src.LinkType())
if err != nil {
log.Printf("Error: %s\n", err)
}
log.Println(rcv_pkt)
} else {
dst.Inject(buf)
}
if count > 0 && i >= count {
break
}
}
}
func BenchmarkUnpackAllEthIPv4TCPRaw(bn *testing.B) {
for n := 0; n < bn.N; n++ {
layers.UnpackAll(test_eth_ipv4_tcp_raw, packet.Eth)
}
}
func BenchmarkUnpackAllEthIPv4UDP(bn *testing.B) {
for n := 0; n < bn.N; n++ {
layers.UnpackAll(test_eth_ipv4_udp, packet.Eth)
}
}
func BenchmarkUnpackAllEthVLANArp(bn *testing.B) {
for n := 0; n < bn.N; n++ {
layers.UnpackAll(test_eth_vlan_arp, packet.Eth)
}
}