Initial commit

main.go

@@ -0,0 +1,62 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"os"
+
+	log "github.com/sirupsen/logrus"
+	"github.com/zr-hebo/sniffer-agent/capture"
+	"github.com/zr-hebo/sniffer-agent/exporter"
+	"github.com/zr-hebo/sniffer-agent/communicator"
+	sd "github.com/zr-hebo/sniffer-agent/session-dealer"
+)
+
+var (
+	logLevel string
+)
+
+func init()  {
+	flag.StringVar(&logLevel, "log_level", "info", "log level. Default is info")
+}
+
+func initLog()  {
+	log.SetFormatter(&log.TextFormatter{})
+	log.SetOutput(os.Stdout)
+	switch logLevel {
+	case "debug":
+		log.SetLevel(log.DebugLevel)
+	case "info":
+		log.SetLevel(log.InfoLevel)
+	case "warn":
+		log.SetLevel(log.WarnLevel)
+	case "error":
+		log.SetLevel(log.ErrorLevel)
+	default:
+		panic(fmt.Sprintf("cannot set log level:%s, there have four types can set: debug, info, warn, error", logLevel))
+	}
+}
+
+func main()  {
+	flag.Parse()
+	initLog()
+	sd.CheckParams()
+
+	go communicator.Server()
+	mainServer()
+}
+
+func mainServer()  {
+	ept := exporter.NewExporter()
+	networkCard := capture.NewNetworkCard()
+	log.Info("begin listen")
+	for queryPiece := range networkCard.Listen() {
+		err := ept.Export(queryPiece)
+		if err != nil {
+			log.Error(err.Error())
+		}
+	}
+
+	log.Errorf("cannot get network package from %s", capture.DeviceName)
+	os.Exit(1)
+}