Init

core/cmd.go

@@ -0,0 +1,144 @@
+package core
+
+import (
+	"os"
+	"strings"
+	"fmt"
+	"net"
+	"strconv"
+)
+
+const InternalCmdPrefix = "--"
+const (
+	InternalCmdHelp = "help"  //帮助文档
+	InternalCmdEnv  = "env"   //环境变量
+	InternalCmdList = "list"  //插件列表
+	InternalCmdVer  = "ver"   //版本信息
+	InternalDevice  = "dev"   //设备链表
+)
+
+type Cmd struct {
+	Device string
+	plugHandle *Plug
+}
+
+func NewCmd(p *Plug) *Cmd {
+
+	return &Cmd{
+		plugHandle:p,
+	}
+}
+
+//start
+func (cm *Cmd) Run() {
+
+	//使用帮助
+	if len(os.Args) <= 1 {
+		cm.printHelpMessage();
+		os.Exit(1)
+	}
+
+	//解析命令
+	firstArg := string(os.Args[1])
+	if strings.HasPrefix(firstArg, InternalCmdPrefix) {
+		cm.parseInternalCmd()
+	} else {
+		cm.parsePlugCmd()
+	}
+}
+
+//解析内部参数
+func (cm *Cmd) parseInternalCmd() {
+
+	arg := string(os.Args[1])
+	cmd := strings.Trim(arg, InternalCmdPrefix)
+
+	switch cmd {
+		case InternalCmdHelp:
+			cm.printHelpMessage()
+			break;
+		case InternalCmdEnv:
+			fmt.Println("插件路径:"+cm.plugHandle.dir)
+			break
+		case InternalCmdList:
+			cm.plugHandle.PrintList()
+			break
+		case InternalCmdVer:
+			fmt.Println(cxt.Version)
+			break
+		case InternalDevice:
+			cm.printDevice()
+			break;
+	}
+	os.Exit(1)
+}
+
+//使用说明
+func (cm *Cmd) printHelpMessage()  {
+
+	fmt.Println("==================================================================================")
+	fmt.Println("[使用说明]")
+	fmt.Println("")
+	fmt.Println("    go-sniffer [设备名] [插件名] [插件参数(可选)]")
+	fmt.Println()
+	fmt.Println("    [例子]")
+	fmt.Println("          go-sniffer en0 redis          抓取redis数据包")
+	fmt.Println("          go-sniffer en0 mysql -p 3306  抓取mysql数据包,端口3306")
+	fmt.Println()
+	fmt.Println("    go-sniffer --[命令]")
+	fmt.Println("               --help 帮助信息")
+	fmt.Println("               --env  环境变量")
+	fmt.Println("               --list 插件列表")
+	fmt.Println("               --ver  版本信息")
+	fmt.Println("               --dev  设备列表")
+	fmt.Println("    [例子]")
+	fmt.Println("          go-sniffer --list 查看可抓取的协议")
+	fmt.Println()
+	fmt.Println("==================================================================================")
+	cm.printDevice()
+	fmt.Println("==================================================================================")
+}
+
+//打印插件
+func (cm *Cmd) printPlugList() {
+	l := len(cm.plugHandle.InternalPlugList)
+	l += len(cm.plugHandle.ExternalPlugList)
+	fmt.Println("#    插件数量："+strconv.Itoa(l))
+}
+
+//打印设备
+func (cm *Cmd) printDevice() {
+	ifaces, err:= net.Interfaces()
+	if err != nil {
+		panic(err)
+	}
+	for _, iface := range ifaces {
+		addrs, _ := iface.Addrs()
+		for _,a:=range addrs {
+			if ipnet, ok := a.(*net.IPNet); ok {
+				if ip4 := ipnet.IP.To4(); ip4 != nil {
+					fmt.Println("[设备名] : "+iface.Name+" : "+iface.HardwareAddr.String()+"  "+ip4.String())
+				}
+			}
+		}
+	}
+}
+
+//解析插件需要的参数
+func (cm *Cmd) parsePlugCmd()  {
+
+	if len(os.Args) < 3 {
+		fmt.Println("缺少[插件名]")
+		fmt.Println("go-sniffer [设备名] [插件名] [插件参数(可选)]")
+		os.Exit(1)
+	}
+
+	cm.Device  = os.Args[1]
+	plugName  := os.Args[2]
+	plugParams:= os.Args[3:]
+	cm.plugHandle.SetOption(plugName, plugParams)
+}
+
+
+
+