mirror of https://github.com/40t/go-sniffer.git
145 lines
3.2 KiB
Go
145 lines
3.2 KiB
Go
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)
|
|
}
|
|
|
|
|
|
|
|
|