示例#1
0
文件: main.go 项目: jxwr/pcap-agent
// main uses golibpcap to build a simple tcpdump binary.
func main() {
	flag.Parse()

	var h *pcap.Pcap
	var err error

	if *device == "" {
		flag.Usage()
		log.Fatal("main: device == \"\"")
	}
	if *buffLimit != 0 {
		// Set things up by hand.
		h, err = pcap.Create(*device)
		if err != nil {
			log.Fatalf("main:pcap.Create: %v", err)
		}
		err = h.SetSnaplen(int32(*snaplen))
		if err != nil {
			log.Fatalf("main:h.SetSnaplen: %v", err)
		}
		err = h.SetBufferSize(int32(*buffLimit))
		if err != nil {
			log.Fatalf("main:h.SetBufferSize: %v", err)
		}
		err = h.SetPromisc(true)
		if err != nil {
			log.Fatalf("main:h.SetPromisc: %v", err)
		}
		err = h.SetTimeout(int32(0))
		if err != nil {
			log.Fatalf("main:h.SetTimeout: %v", err)
		}
		err = h.Activate()
		if err != nil {
			log.Fatalf("main:h.Activate: %v", err)
		}
	} else {
		h, err = pcap.OpenLive(*device, int32(*snaplen), true, 0)
		if err != nil {
			log.Fatalf("main:pcap.OpenLive: %v", err)
		}
	}

	if *expr == "" {
		*expr = fmt.Sprintf("port %d", *serverPort)
	}
	err = h.Setfilter(*expr)
	if err != nil {
		log.Fatalf("main:h.Setfilter: %v", err)
	}

	if *pCount > 0 {
		go h.Loop(*pCount)
	} else {
		go h.Loop(-1)
	}

	// Start decoding packets until we receive the signal to stop (nil pkt).
	var p *pkt.Packet
	for {
		p = <-h.Pchan
		if p == nil {
			break
		}
		ipHdr, ok := p.Headers[pkt.NetworkLayer].(*pkt.IpHdr)
		if !ok {
			log.Fatalf("read iphdr failed")
		}
		tcpHdr, ok := p.Headers[pkt.TransportLayer].(*pkt.TcpHdr)
		if !ok {
			log.Fatalf("read tcphdr failed")
		}
		handle(uint16(*serverPort), tcpHdr, tcpHdr.GetPayloadBytes(ipHdr.PayloadLen))
	}

	s, err := h.Getstats()
	if err == nil {
		fmt.Printf("%s\n", s)
	}
	h.Close()
}
示例#2
0
// main uses golibpcap to build a simple tcpdump binary.
func main() {
	flag.Parse()

	var h *pcap.Pcap
	var err error

	// First we check to see if the user is passing us a pcap save file to
	// read.  If so, then we will open that off-line.
	if *dumpFile != "" {
		h, err = pcap.OpenOffline(*dumpFile)
		if err != nil {
			log.Fatalf("main:pcap.OpenOffline: %v", err)
		}
	}

	if h == nil {
		if *device == "" {
			flag.Usage()
			log.Fatal("main: *device == \"\"")
		}
		if *buffLimit != 0 {
			// If we have a custom buffer limit then we have to set things up
			// by hand.  Most of these settings have to be set before and cannot
			// be changed once the pcap is active.
			h, err = pcap.Create(*device)
			if err != nil {
				log.Fatalf("main:pcap.Create: %v", err)
			}
			err = h.SetSnaplen(int32(*snaplen))
			if err != nil {
				log.Fatalf("main:h.SetSnaplen: %v", err)
			}
			err = h.SetBufferSize(int32(*buffLimit))
			if err != nil {
				log.Fatalf("main:h.SetBufferSize: %v", err)
			}
			err = h.SetPromisc(true)
			if err != nil {
				log.Fatalf("main:h.SetPromisc: %v", err)
			}
			err = h.SetTimeout(int32(0))
			if err != nil {
				log.Fatalf("main:h.SetTimeout: %v", err)
			}
			err = h.Activate()
			if err != nil {
				log.Fatalf("main:h.Activate: %v", err)
			}
		}
	}
	if h == nil {
		// Given a device we will open a live trace of that device.
		h, err = pcap.OpenLive(*device, int32(*snaplen), true, 0)
		if err != nil {
			log.Fatalf("main:pcap.OpenLive: %v", err)
		}
	}

	// If given a filter string to use then try to apply that filter.
	if *expr != "" {
		err = h.Setfilter(*expr)
		if err != nil {
			log.Fatalf("main:h.Setfilter: %v", err)
		}
	}

	// If given a fixed amount of time to trace the device then start a
	// goroutine that will terminate the live capture after that amount of
	// time.
	if *tLimit > 0 {
		go h.DelayBreakLoop(*tLimit)
	}

	// If given a fixed number of packets to grab them just grab that many
	// otherwise we should enter an loop that does not end.  In the latter
	// case either the time limit will trigger an exit or the program will
	// have to be killed by an external force.
	if *pCount > 0 {
		go h.Loop(*pCount)
	} else {
		go h.Loop(-1)
	}

	// If given a writeFile we should listen quietly.
	if *writeFile != "" {
		r := make(chan *[]*pkt.Packet, 1)

		// h.Listen will accumulate the results that will be passed back
		// on the channel r that was just created.
		go h.Listen(r)

		// Here we convert the container that was used to accumulate the
		// results into an easier to work with trace.PktTrace.
		t, err := h.NewPktTrace(<-r)
		if err != nil {
			log.Printf("main:h.NewPktTrace: %v", err)
		}
		w, err := os.Create(*writeFile)
		if err != nil {
			log.Printf("main:os.Create: %v", err)
		}
		err = t.Archive(w)
		if err != nil {
			log.Printf("main:t.Archive: %v", err)
		}
		w.Close()
	} else {
		// Start decoding packets until we receive the signal to stop (nil pkt).
		var p *pkt.Packet
		if *verbose {
			for {
				p = <-h.Pchan
				if p == nil {
					break
				}
				fmt.Println(p.JsonString())
			}
		} else if *quiet {
			for {
				p = <-h.Pchan
				if p == nil {
					break
				}
			}
		} else {
			for {
				p = <-h.Pchan
				if p == nil {
					break
				}
				fmt.Println(p.String())
			}
		}
	}
	s, err := h.Getstats()
	if err == nil {
		fmt.Printf("%s\n", s)
	}
	h.Close()
}