golang利用原始套接字構(gòu)造UDP包的示例

小編給大家分享一下golang利用原始套接字構(gòu)造UDP包的示例，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

RAW SOCKET 介紹

TCP/IP協(xié)議中，最常見的就是原始(SOCKET_RAW)、tcp(SOCKET_STREAM)、udp(SOCKET_DGRA)三種套接字。原始套接字能夠?qū)Φ讓觽鬏斶M(jìn)行控制，允許自行組裝數(shù)據(jù)包，比如修改本地IP，發(fā)送Ping包，進(jìn)行網(wǎng)絡(luò)監(jiān)聽。這里不做詳細(xì)介紹，要了解更多可以網(wǎng)上自己查詢。

實(shí)現(xiàn)

這里先看IP頭結(jié)構(gòu)：

其中16位總長度包括IP頭長度和數(shù)據(jù)的長度，8位協(xié)議填寫17，因?yàn)閁DP協(xié)議類型為17。這里要說明一下IP頭中的首部校驗(yàn)，這個(gè)值只校驗(yàn)IP頭部，不包含數(shù)據(jù)。

這里給出校驗(yàn)算法，IP頭和UDP頭中使用的校驗(yàn)算法是一樣的。

func checkSum(msg []byte) uint16 {
 sum := 0
 for n := 1; n < len(msg)-1; n += 2 {
  sum += int(msg[n])*256 + int(msg[n+1])
 }
 sum = (sum >> 16) + (sum & 0xffff)
 sum += (sum >> 16)
 var ans = uint16(^sum)
 return ans
}

下面開始填充IP頭，這里使用了golang.org/x/net下的ipv4包

 //目的IP
 dst := net.IPv4(192, 168, 1, 2)
 //源IP
 src := net.IPv4(192, 168, 1, 3)
 //填充ip首部
 iph := &ipv4.Header{
  Version: ipv4.Version,
  //IP頭長一般是20
  Len:  ipv4.HeaderLen,
  TOS:  0x00,
  //buff為數(shù)據(jù)
  TotalLen: ipv4.HeaderLen + len(buff),
  TTL:  64,
  Flags: ipv4.DontFragment,
  FragOff: 0,
  Protocol: 17,
  Checksum: 0,
  Src:  src,
  Dst:  dst,
 }
 
 h, err := iph.Marshal()
 if err != nil {
  log.Fatalln(err)
 }
 //計(jì)算IP頭部校驗(yàn)值
 iph.Checksum = int(checkSum(h))

下面開始處理UDP頭部，先來看UDP頭結(jié)構(gòu)：

UDP頭結(jié)構(gòu)就很簡單了，16位UDP校驗(yàn)和涉及到一個(gè)UDP偽首部的東西，我們先來看下UDP偽首部的構(gòu)成。

-----------------------------------------
|   32bit Source IP address  |
-----------------------------------------
|   32bit Destination IP addr  |
-----------------------------------------
| 0 | 8bit Proto| 16bit header length|
-----------------------------------------

偽首部包含了源IP，目的IP，協(xié)議號，16位的長度。這個(gè)偽首部僅僅參與校驗(yàn)計(jì)算。

下面開始填充UDP頭：

 //填充udp首部
 //udp偽首部
 udph := make([]byte, 20)
 //源ip地址
 udph[0], udph[1], udph[2], udph[3] = src[12], src[13], src[14], src[15]
 //目的ip地址
 udph[4], udph[5], udph[6], udph[7] = dst.IP[12], dst.IP[13], dst.IP[14], dst.IP[15]
 //協(xié)議類型
 udph[8], udph[9] = 0x00, 0x11
 //udp頭長度
 udph[10], udph[11] = 0x00, byte(len(buff)+8)
 //下面開始就真正的udp頭部
 //源端口號
 udph[12], udph[13] = 0x27, 0x10
 //目的端口號
 udph[14], udph[15] = 0x17, 0x70
 //udp頭長度
 udph[16], udph[17] = 0x00, byte(len(buff)+8)
 //校驗(yàn)和
 udph[18], udph[19] = 0x00, 0x00
 //計(jì)算校驗(yàn)值
 check := checkSum(append(udph, buff...))
 udph[18], udph[19] = byte(check>>8&255), byte(check&255）

下面我們需要發(fā)送自己構(gòu)造的UDP包，可以使用net下的ListenPacket。

 listener, err := net.ListenPacket("ip4:udp", "192.168.1.104")
 if err != nil {
  log.Fatal(err)
 }
 defer listener.Close()
 
 //listener 實(shí)現(xiàn)了net.PacketConn接口
 r, err := ipv4.NewRawConn(c)
 if err != nil {
  log.Fatal(err)
 }

 //發(fā)送自己構(gòu)造的UDP包
 if err = r.WriteTo(iph, append(udph[12:20], buff...), nil); err != nil {
  log.Fatal(err)
 }

這個(gè)實(shí)現(xiàn)只在linux和mac上測試過，windows上需要借助于第三方吧，比如winpcap。

以上是“golang利用原始套接字構(gòu)造UDP包的示例”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對大家有所幫助，如果還想學(xué)習(xí)更多知識，歡迎關(guān)注億速云行業(yè)資訊頻道！