前言
本文主要給大家介紹了關于golang用原始套接字構造UDP包的相關內容，分享出來供大家參考學習，下面話不多說了，來一起看看詳細的介紹吧。
RAW SOCKET 介紹
TCP/IP協議中，最常見的就是原始(SOCKET_RAW)、tcp(SOCKET_STREAM)、udp(SOCKET_DGRA)三種套接字。原始套接字能夠對底層傳輸進行控制，允許自行組裝數據包，比如修改本地IP，發送Ping包，進行網絡監聽。這里不做詳細介紹，要了解更多可以網上自己查詢。
實現
這里先看IP頭結構：
其中16位總長度包括IP頭長度和數據的長度，8位協議填寫17，因為UDP協議類型為17。這里要說明一下IP頭中的首部校驗，這個值只校驗IP頭部，不包含數據。
這里給出校驗算法，IP頭和UDP頭中使用的校驗算法是一樣的。
|
1
2
3
4
5
6
7
8
9
10
|
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包
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
//目的IPdst := net.IPv4(192, 168, 1, 2)//源IPsrc := net.IPv4(192, 168, 1, 3)//填充ip首部iph := &ipv4.Header{ Version: ipv4.Version, //IP頭長一般是20 Len: ipv4.HeaderLen, TOS: 0x00, //buff為數據 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)}//計算IP頭部校驗值iph.Checksum = int(checkSum(h)) |
下面開始處理UDP頭部，先來看UDP頭結構：
UDP頭結構就很簡單了，16位UDP校驗和涉及到一個UDP偽首部的東西，我們先來看下UDP偽首部的構成。
|
1
2
3
4
5
6
7
|
-----------------------------------------| 32bit Source IP address |-----------------------------------------| 32bit Destination IP addr |-----------------------------------------| 0 | 8bit Proto| 16bit header length|----------------------------------------- |
偽首部包含了源IP，目的IP，協議號，16位的長度。這個偽首部僅僅參與校驗計算。
下面開始填充UDP頭：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
//填充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]//協議類型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)//校驗和udph[18], udph[19] = 0x00, 0x00//計算校驗值check := checkSum(append(udph, buff...))udph[18], udph[19] = byte(check>>8&255), byte(check&255） |
下面我們需要發送自己構造的UDP包，可以使用net下的ListenPacket。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
listener, err := net.ListenPacket("ip4:udp", "192.168.1.104")if err != nil { log.Fatal(err)}defer listener.Close()//listener 實現了net.PacketConn接口r, err := ipv4.NewRawConn(c)if err != nil { log.Fatal(err)}//發送自己構造的UDP包if err = r.WriteTo(iph, append(udph[12:20], buff...), nil); err != nil { log.Fatal(err)} |
這個實現只在linux和mac上測試過，windows上需要借助于第三方吧，比如winpcap。
結語
這里只給出了UDP的實現，TCP的實現比較復雜，以后也會給出TCP實現的例子。
總結
以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對服務器之家的支持。
原文鏈接：https://segmentfault.com/a/1190000009253715
