mlCommom

pkg/packetbreeze.go

@@ -7,7 +7,7 @@ import (
 	"github.com/deepfence/FlowMeter/pkg/constants"
 	"github.com/deepfence/FlowMeter/pkg/packetAnalyzer"
 	"github.com/google/gopacket"
-	"github.com/google/gopacket/p cap"
+	"github.com/google/gopacket/pcap"
 
 	"github.com/sirupsen/logrus"
 )
@@ -17,7 +17,7 @@ func packetCollect() {
 	done := make(chan struct{}) // signal channel
 
 	// Command line arguments.
-	ifLiveCapturePtr := flag.Bool("ifLiveCapture", false, "a bool")
+	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")
@@ -36,14 +36,14 @@ func packetCollect() {
 
 	logrus.Info("Start Main method")
 
+	//File Handling with error checks is done here
+
 	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 {
@@ -57,7 +57,7 @@ func packetCollect() {
 		log.Fatal(constants.Err)
 	}
 
-	go packetAnalyzer.FlowMeter(ch, done, maxNumPackets, localIP, ifLocalIPKnown, filename)
+	go packetAnalyzer.PacketBreeze(ch, done, maxNumPackets, localIP, ifLocalIPKnown, filename)
 
 	defer constants.Handle.Close()