コード例 #1
0
ファイル: layers_test.go プロジェクト: kelixin/go.pkt
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())
	}
}
コード例 #2
0
ファイル: layers_test.go プロジェクト: kelixin/go.pkt
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)
}
コード例 #3
0
ファイル: network.go プロジェクト: kelixin/go.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
}
コード例 #4
0
ファイル: layers_test.go プロジェクト: kelixin/go.pkt
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())
	}
}
コード例 #5
0
ファイル: layers_test.go プロジェクト: kelixin/go.pkt
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)
	}
}
コード例 #6
0
ファイル: main.go プロジェクト: kelixin/go.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
		}
	}
}
コード例 #7
0
ファイル: layers_test.go プロジェクト: kelixin/go.pkt
func BenchmarkUnpackAllEthIPv4TCPRaw(bn *testing.B) {
	for n := 0; n < bn.N; n++ {
		layers.UnpackAll(test_eth_ipv4_tcp_raw, packet.Eth)
	}
}
コード例 #8
0
ファイル: layers_test.go プロジェクト: kelixin/go.pkt
func BenchmarkUnpackAllEthIPv4UDP(bn *testing.B) {
	for n := 0; n < bn.N; n++ {
		layers.UnpackAll(test_eth_ipv4_udp, packet.Eth)
	}
}
コード例 #9
0
ファイル: layers_test.go プロジェクト: kelixin/go.pkt
func BenchmarkUnpackAllEthVLANArp(bn *testing.B) {
	for n := 0; n < bn.N; n++ {
		layers.UnpackAll(test_eth_vlan_arp, packet.Eth)
	}
}