欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

詳解Go程序添加遠程調用tcpdump功能

 更新時間:2022年05月23日 14:20:00   作者:superpigwin  
這篇文章主要介紹了go程序添加遠程調用tcpdump功能,本文給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下

最近開發(fā)的telemetry采集系統(tǒng)上線了。聽起來高大上,簡單來說就是一個grpc/udp服務端,用戶的機器(路由器、交換機)將它們的各種統(tǒng)計數(shù)據(jù)上報采集、整理后交后端的各類AI分析系統(tǒng)分析。目前華為/思科的大部分設備支持。

上線之后,各類用戶開始找來要求幫忙定位問題,一般是上報的數(shù)據(jù)在后端系統(tǒng)中不存在等等。

在一通抓包分析后,百分之99都是用戶自己的問題。但頻繁的抓包定位問題,嚴重的壓縮了我摸魚的時間。而且,這套系統(tǒng)采用多實例方式部署在騰X云多個容器中,一個個的登錄抓包,真的很煩。

這讓我萌生了一個需求:

  • 主動給采集器下發(fā)抓包任務。
  • 將抓包的信息返回。
  • 將抓包的文件暫存,以備進一步定位問題。

方法1

使用fabric等ssh運維工具,編寫腳本自動化登錄機器后執(zhí)行tcpdump,然后進一步處理。
很可惜的是,并沒有容器母機ssh的權限。只能通過一個web命令行觀察容器。這條路玩不轉。

方法2

  • 在采集器中添加一個接口,用以下發(fā)tcpdump命令
  • 采集器執(zhí)行tcpdump命令,并獲取返回的信息(比如captured xxx pacs),保存相關文件。
  • 將獲取的抓包信息以某種方式反發(fā)給命令下發(fā)人。

使用tcpdump定時抓取并保存信息

首先需要解決tcpdump定時的問題,以免tcpdump無限期的執(zhí)行抓包,經(jīng)過一通谷歌,命令如下:

timeout 30 tcpdump -i eth0 host 9.123.123.111 and port 6651 -w /tmp/log.cap

timeout 30 指抓取30秒,超時后tcpdump會直接退出
-i 指定抓取的端口
host xxx 源IP
port xxx 源端口

編寫tcpdump函數(shù)

下面到了我最喜歡的寫代碼階段,為了簡單,直接使用os/exec庫。不要笑,很多大廠的很多系統(tǒng)其實都是包命令行工具,解決問題最重要。

// TcpDump 執(zhí)行tcpdump命令,并返回抓到的包數(shù)
func TcpDump(sudo bool, timeout int, eth string, host string, port int) (caps int, err error) {
	portStr := ""
	if port != 0 {
		portStr = fmt.Sprintf("and port %v", port)
	}
	tcpdumpCmd := fmt.Sprintf("timeout %v tcpdump -i %v host %v %v -w /tmp/log.cap",
		timeout, eth, host, portStr)
	if sudo {
		tcpdumpCmd = "sudo " + tcpdumpCmd
	}
	logrus.Infof("call %v", tcpdumpCmd)
	cmd := exec.Command("sh", "-c", tcpdumpCmd)
	var outb, errb bytes.Buffer
	cmd.Stderr = &errb
	err = cmd.Run()
	if err != nil {
		if !errors.Is(err, &exec.ExitError{}) {
			logrus.Infof("out:%s ; %s", outb.Bytes(), errb.Bytes())
			return getPacs(errb.String()), nil
		}
		return
	}
	return 0,fmt.Errorf("unknown error")
}
func getPacs(input string) int {
	end := strings.Index(input, "packets captured")
	pos := end
	for {
		pos -= 1
		if pos <= 0 {
			return 0
		}
		if input[pos] == '\n' {
			break
		}
	}
	// logrus.Infof("captured:%s", input[pos+1:end-1])
	v, err := strconv.Atoi(input[pos+1 : end-1])
	if err != nil {
		return 0
	}
	return v
}

這里要注意幾點:

執(zhí)行cmd := exec.Command("sh", "-c", tcpdumpCmd)后,tcpdump的返回信息類似:

listening on eth1, link-type EN10MB (Ethernet), capture size 65535 bytes\n56 packets captured\n56 packets received by filter\n0 packets dropped by kernel\n

是在stderr中的。而不是stdout。

getPacs函數(shù)簡單的從xx packets received中提取出了抓包數(shù)。但是如果是中文的服務器系統(tǒng)(不會吧,不會吧),就不太好使了。

編寫api

現(xiàn)在函數(shù)已經(jīng)有了,只要再寫一個http api,就能很方便的把它暴露出去。

import "github.com/gogf/gf/v2/encoding/gjson"
// ErrJson,寫入一個error json,形如:
//{
//    "err": code,
//    "err_msg": msg
//}
func ErrJson(w http.ResponseWriter, errCode int, errStr string) error {
	w.Header().Set("Content-Type", "application/json")
	js := make(map[string]interface{})
	js["err"] = errCode
	js["err_msg"] = errStr
	jsBts, _ := json.Marshal(js)
	_, err := w.Write(jsBts)
	return err
}
/* TcpDumpHandler
req:{
	"sudo":  true,
	"eth": "eth0",
	"host": "10.99.17.135",
	"port": 0
}
rsp:{
	"err": 0,
	"caps": 14
}
*/
func TcpDumpHandler(w http.ResponseWriter, r *http.Request) {
	r.ParseForm()
	ret, err := ioutil.ReadAll(r.Body)
	if err != nil {
		ErrJson(w, 1, "數(shù)據(jù)錯誤")
		return
	}
	js := gjson.New(ret)
	sudo := js.Get("sudo").Bool()
	eth := js.Get("eth").String()
	if eth == "" {
		ErrJson(w, 1, "數(shù)據(jù)錯誤, eth不存在")
		return
	}
	host := js.Get("host").String()
	if host == "" {
		ErrJson(w, 1, "數(shù)據(jù)錯誤, host不存在")
		return
	}
	port := js.Get("port").Int()
	timeout := js.Get("timeout").Int()
	if timeout == 0 {
		ErrJson(w, 1, "數(shù)據(jù)錯誤, timeout為0或不存在")
		return
	}
	go func() {
		chatKey := config.GlobalConfigObj.Global.ChatKey
		botKey := config.GlobalConfigObj.Global.BotKey
		
		// 這里直接利用了公司的一個消息系統(tǒng),如果貴公司沒有這樣的系統(tǒng),就變通一下
		msgSender := msg.NewNiuBiMsg(chatKey, botKey)
		caps, err := TcpDump(sudo, timeout, eth, host, port)
		if err != nil {
			return
		}
		if caps > 0 {
			// 這里直接利用了公司的一個消息系統(tǒng),向企業(yè)IM發(fā)一條消息
			msgSender.Send(fmt.Sprintf("tcpdump agent_ip:%v host:%v eth:%v port:%v, captured:%v",
				config.GlobalLocalConfig.LocalIP, host, eth, port, caps))
			bts, err := ioutil.ReadFile("/tmp/log.cap")
			if err != nil {
				return
			}
			b64Caps := base64.StdEncoding.EncodeToString(bts)
			// 把抓包的文件通過這個消息系統(tǒng)也發(fā)到企業(yè)IM中
			msgSender.File(fmt.Sprintf("pacs_%v.cap", config.GlobalLocalConfig.LocalIP), b64Caps)
		}
	}()
}

然后起一個http svr

func runHttp() {
	mux := http.NewServeMux()
	server :=
		http.Server{
			Addr:         fmt.Sprintf(":%d", 3527),
			Handler:      mux,
			ReadTimeout:  3 * time.Second,
			WriteTimeout: 5 * time.Second,
		}
	// 開始添加路由
	mux.HandleFunc("/tcpdump", tcpdumpsvc.TcpDumpHandler)
	logrus.Infof("run http:%v", 3527)
	logrus.Info(server.ListenAndServe())
}

到這一步,這個系統(tǒng)就基本完成了。使用這個命令就能調用接口。

curl --header "Content-Type: application/json" --request GET --data '{"sudo":false,"eth":"eth0","host":"100.xxx.xxx.10","port":0,"timeout":5}' http://0.0.0.0:3527/tcpdump

這個系統(tǒng)有幾個硬傷。

  • 依賴了公司的消息系統(tǒng)完成抓包數(shù)據(jù)回發(fā)的功能。假如各位大佬的公司沒有這樣的系統(tǒng)msgSender.Send,可行的方法有:
  • scp到一個特定的文件夾。
  • 使用電子郵件。
  • 和領導申請自己開發(fā)一套,你看,需求就來了。
  • tcpdump可能會生成極大的抓包文件,此時使用bts, err := ioutil.ReadFile("/tmp/log.cap"),可能會直接讓系統(tǒng)OOM。所以設置timeout和抓包的大小(比如在tcpdump命令中使用-c)是很重要的。換句話說,這個api不是公有的,別讓不了解的人去調用。

不過這都是小問題?,F(xiàn)在用戶找上門來,我只需要啟動腳本,從服務發(fā)現(xiàn)api拉到所有的實例IP,然后依次調用tcpdump api,等待IM的反饋即可。又能快樂的摸魚啦。

到此這篇關于go程序添加遠程調用tcpdump功能的文章就介紹到這了,更多相關go遠程調用tcpdump內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!

相關文章

  • Go語言中的Struct結構體

    Go語言中的Struct結構體

    這篇文章介紹了Go語言中的Struct結構體,文中通過示例代碼介紹的非常詳細。對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下
    2022-07-07
  • golang:json 反序列化的[]和nil操作

    golang:json 反序列化的[]和nil操作

    這篇文章主要介紹了golang:json 反序列化的[]和nil操作,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
    2020-12-12
  • Golang異常處理之defer,panic,recover的使用詳解

    Golang異常處理之defer,panic,recover的使用詳解

    這篇文章主要為大家介紹了Go語言異常處理機制中defer、panic和recover三者的使用方法,文中示例代碼講解詳細,需要的朋友可以參考下
    2022-05-05
  • Go語言框架快速集成限流中間件詳解

    Go語言框架快速集成限流中間件詳解

    這篇文章主要為大家介紹了Go語言框架快速集成限流中間件詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪
    2022-09-09
  • 解決go echo后端處理跨域的兩種操作方式

    解決go echo后端處理跨域的兩種操作方式

    這篇文章主要介紹了解決go echo后端處理跨域的兩種操作方式,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
    2020-12-12
  • go連接mysql的項目實踐

    go連接mysql的項目實踐

    本文主要介紹了go連接mysql的項目實踐,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧
    2023-03-03
  • golang之資源釋放/異常錯誤處理解析

    golang之資源釋放/異常錯誤處理解析

    這篇文章主要為大家介紹了golang之資源釋放/異常錯誤處理解析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪
    2024-01-01
  • Go語言學習教程之指針的示例詳解

    Go語言學習教程之指針的示例詳解

    這篇文章主要通過簡單的練習來讓大家對Go語言中的指針有所了解,文中的示例代碼講解詳細,對我們學習Go語言有一定幫助,需要的可以參考一下
    2022-09-09
  • Web框架Gin中間件實現(xiàn)原理步驟解析

    Web框架Gin中間件實現(xiàn)原理步驟解析

    這篇文章主要為大家介紹了Web框架Gin中間件實現(xiàn)原理步驟解析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪
    2023-10-10
  • Go+Vue開發(fā)一個線上外賣應用的流程(用戶名密碼和圖形驗證碼)

    Go+Vue開發(fā)一個線上外賣應用的流程(用戶名密碼和圖形驗證碼)

    這篇文章主要介紹了Go+Vue開發(fā)一個線上外賣應用(用戶名密碼和圖形驗證碼),本文給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下
    2020-11-11

最新評論