前言
本文主要给大家介绍了关于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
