Init

pkg/packetbreeze.go

@@ -0,0 +1,80 @@
+package main
+
+import (
+	"flag"
+	"log"
+
+	"github.com/deepfence/FlowMeter/pkg/constants"
+	"github.com/deepfence/FlowMeter/pkg/packetAnalyzer"
+	"github.com/google/gopacket"
+	"github.com/google/gopacket/p cap"
+
+	"github.com/sirupsen/logrus"
+)
+
+func packetCollect() {
+	ch := make(chan gopacket.Packet)
+	done := make(chan struct{}) // signal channel
+
+	// Command line arguments.
+	ifLiveCapturePtr := flag.Bool("ifLiveCapture", true, "a bool")
+	filenamePtr := flag.String("fname", "foo", "a string")
+	maxNumPacketsPtr := flag.Int("maxNumPackets", 42, "an int")
+	ifLocalIPKnownPtr := flag.Bool("ifLocalIPKnown", false, "a bool")
+	localIPPtr := flag.String("localIP", "143.198.72.237", "a string")
+
+	flag.Parse()
+
+	ifLiveCapture := *ifLiveCapturePtr
+	filename := *filenamePtr
+	maxNumPackets := *maxNumPacketsPtr
+	ifLocalIPKnown := *ifLocalIPKnownPtr
+	localIP := *localIPPtr
+
+	logrus.Info("Live Capture: ", ifLiveCapture)
+	logrus.Info("Target max number packets: ", maxNumPackets)
+
+	logrus.Info("Start Main method")
+
+	if ifLiveCapture {
+		// Open device
+		constants.Handle, constants.Err = pcap.OpenLive(constants.Device, constants.SnapshotLen, constants.Promiscuous, constants.Timeout)
+		localIP = packetAnalyzer.GetOutboundIP().String()
+		ifLocalIPKnown = true
+		logrus.Info("Live capture of packets.")
+	} else {
+		// Open file instead of device
+		constants.Handle, constants.Err = pcap.OpenOffline(constants.PacketFolder + "/" + filename + constants.PacketFileType)
+
+		if !ifLocalIPKnown {
+			localIP = ""
+		}
+
+		logrus.Info("Analyzing offline pcap files.")
+	}
+
+	if constants.Err != nil {
+		log.Fatal(constants.Err)
+	}
+
+	go packetAnalyzer.FlowMeter(ch, done, maxNumPackets, localIP, ifLocalIPKnown, filename)
+
+	defer constants.Handle.Close()
+
+	packetSource := gopacket.NewPacketSource(constants.Handle, constants.Handle.LinkType())
+
+loop:
+	for packet := range packetSource.Packets() {
+		select {
+		case ch <- packet:
+		case <-done:
+			close(ch)
+			close(done)
+			break loop
+		}
+	}
+}
+
+func main() {
+	packetCollect()
+}