wireshark插件開發 - 自定義協議

雖然wireshark自帶了很多知名協議的解析插件，譬如HTTP、DHCP等等，然而在實際應用環境中，有不少軟件之間的通信協議都是私有的，如游戲客戶端和服務器之間的交互協議通常都是私有的，wireshark無法具體解析出各種字段之間的含義，只能顯示接收到的二進制數據，給協議的分析和問題的排查帶來了一定的困難，尤其是協議內容比較復雜時。

本文一個自定義的簡單協議入手，分析如何基於wireshark開發自定義協議分析插件。

 

 

 

1.1. 概述

本書使用Go語言來描述協議的交互過程。Go由Google出品，號稱是互聯網上的C語言，有點類似於C語言，以其簡潔和高並發著稱。

即使不了解Go語言也沒有關系，本書給出的代碼，基本可以自釋其義，了解過程即可，重點不在於此。

1.2. 協議描述

package packet

import (
    "bytes"
    "encoding/binary"
    "fmt"
    "os"
)

type NPacket struct {
    Version int16    //兩個字節
    ID uint32       //四個字節
    Buffer [1024]byte  //簡單起見，固定為1024個字節
}

func (packet *NPacket) String() string {
    return fmt.Sprintf("%v, %v, %v\n", packet.Version,
        packet.ID, string(packet.Buffer[:]))
}

func CheckError(err error) {
    if err != nil {
        fmt.Println("Error: %s", err.Error())
        os.Exit(1)
    }
}

func ConvertToPacket(b []byte) (packet *NPacket, err error) {
    buffer := bytes.NewBuffer(b[:])

    packet = &NPacket{}
    #將緩沖區buffer的內容轉換到NPacket結構體中
    err = binary.Read(buffer, binary.BigEndian, packet)

    return packet, err
}

func ConvertToBuffer(packet *NPacket) (b []byte, err error) {
    buf := new(bytes.Buffer)

#將結構體內容轉換到緩沖區中
    err = binary.Write(buf, binary.BigEndian, packet)

    return buf.Bytes(), err
}

 

 

1.3. 客戶端

package main

import (
    "fmt"
    "net"
)

import "packet"

func main() {
    #連接服務端，UDP協議
    conn, err := net.Dial("udp", "192.168.5.4:11110")
    defer conn.Close()
    packet.CheckError(err)

    #定義報文內容
    pkt_send := packet.NPacket{Version: 1, ID: 1}
    copy(pkt_send.Buffer[:], "Ping")

    send_buff, err := packet.ConvertToBuffer(&pkt_send)
    packet.CheckError(err)
    #發送報文
    conn.Write(send_buff)

    var recv_msg [4096]byte
    #讀取報文
    n, err := conn.Read(recv_msg[0:])
    packet.CheckError(err)

    #報文轉換到結構體中
    pkt_recv, err := packet.ConvertToPacket(recv_msg[0:n])
    packet.CheckError(err)

    #會調用NPacket結構體的String方法
    fmt.Println(pkt_recv)
}

 

1.4. 服務端

package main

import (
    "fmt"
    "net"
)

import "packet"

func recvUDPMsg(conn *net.UDPConn) {
    var buf [4096]byte

    #讀取UDP報文
    n, raddr, err := conn.ReadFromUDP(buf[0:])
    packet.CheckError(err)

    #網絡數據轉換到結構體中
    pkt_recv, err := packet.ConvertToPacket(buf[0:n])
    packet.CheckError(err)

    fmt.Println(pkt_recv)

    #構造響應報文
    pkt_send := packet.NPacket{Version: pkt_recv.Version, ID: pkt_recv.ID}
    copy(pkt_send.Buffer[:], "Pong")

    send_buff, err := packet.ConvertToBuffer(&pkt_send)
    packet.CheckError(err)
    #發送報文
    _, err = conn.WriteToUDP(send_buff, raddr)
    packet.CheckError(err)
}

func main() {
    udp_addr, err := net.ResolveUDPAddr("udp", ":11110")
    packet.CheckError(err)

    conn, err := net.ListenUDP("udp", udp_addr)
    defer conn.Close()
    packet.CheckError(err)

    recvUDPMsg(conn)
}

 

 

1.5. 啟動方式

1、啟動服務端

打開CMD窗口，進入源碼所在目錄

set GOPATH=源碼所在目錄

 

go run src\main\server.go

 

2、啟動客戶端

打開CMD窗口，進入源碼所在目錄

set GOPATH=源碼所在目錄

 

go run src\main\client.go

 

1.6. 抓包

 

 

更多更完整的內容，請移步百度閱讀：

https://yuedu.baidu.com/ebook/ca33e60d3a3567ec102de2bd960590c69ec3d89b