-
Notifications
You must be signed in to change notification settings - Fork 0
/
tcp.go
163 lines (154 loc) · 7.88 KB
/
tcp.go
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
// @@
// @ Author : Eacher
// @ Date : 2023-07-14 08:11:29
// @ LastEditTime : 2023-09-04 09:40:14
// @ LastEditors : Eacher
// @ --------------------------------------------------------------------------------<
// @ Description :
// @ --------------------------------------------------------------------------------<
// @ FilePath : /20yyq/packet/tcp.go
// @@
package packet
import (
"encoding/binary"
)
const (
SizeofTCPPacket = 0x14
)
/*
// 来源： https://support.huawei.com/enterprise/zh/doc/EDOC1100174722?section=j006
0 1 2 3
0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
+-------------------------------+-------------------------------+
| Source Port | Destination Port |
+-------------------------------+-------------------------------+
| Sequence Number |
+---------------------------------------------------------------+
| Acknowledgment Number |
+-------+-----------+-+-+-+-+-+-+-------------------------------+
| Data | |U|A|P|R|S|F| |
| Offset| Reserved |R|C|S|S|Y|I| Window |
| | |G|K|H|T|N|N| |
+-------+-----------+-+-+-+-+-+-+-------------------------------+
| Checksum | Urgent Pointer |
+-------------------------------+---------------+---------------+
| Options | Padding |
+-----------------------------------------------+---------------+
| data |
+---------------------------------------------------------------+
字段 长度 含义
Source Port 16比特 源端口，标识哪个应用程序发送。
Destination Port 16比特 目的端口，标识哪个应用程序接收。
Sequence Number 32比特 序号字段。TCP链接中传输的数据流中每个字节都编上一个序号。序号字段的值指的是本报文段所发送的数据的第一个字节的序号。
Acknowledgment Number 32比特 确认号，是期望收到对方的下一个报文段的数据的第1个字节的序号，即上次已成功接收到的数据字节序号加1。只有ACK标识为1，此字段有效。
Data Offset 4比特 数据偏移，即首部长度，指出TCP报文段的数据起始处距离TCP报文段的起始处有多远，以32比特（4字节）为计算单位。最多有60字节的首部，若无选项字段，正常为20字节。
Reserved 6比特 保留，必须填0。
URG 1比特 紧急指针有效标识。它告诉系统此报文段中有紧急数据，应尽快传送（相当于高优先级的数据）。
ACK 1比特 确认序号有效标识。只有当ACK=1时确认号字段才有效。当ACK=0时，确认号无效。
PSH 1比特 标识接收方应该尽快将这个报文段交给应用层。接收到PSH = 1的TCP报文段，应尽快的交付接收应用进程，而不再等待整个缓存都填满了后再向上交付。
RST 1比特 重建连接标识。当RST=1时，表明TCP连接中出现严重错误（如由于主机崩溃或其他原因），必须释放连接，然后再重新建立连接。
SYN 1比特 同步序号标识，用来发起一个连接。SYN=1表示这是一个连接请求或连接接受请求。
FIN 1比特 发端完成发送任务标识。用来释放一个连接。FIN=1表明此报文段的发送端的数据已经发送完毕，并要求释放连接。
Window 16比特 窗口：TCP的流量控制，窗口起始于确认序号字段指明的值，这个值是接收端期望接收的字节数。窗口最大为65535字节。
Checksum 16比特 校验字段，包括TCP首部和TCP数据，是一个强制性的字段，一定是由发端计算和存储，并由收端进行验证。在计算检验和时，要在TCP报文段的前面加上12字节的伪首部。
Urgent Pointer 16比特 紧急指针，只有当URG标志置1时紧急指针才有效。TCP的紧急方式是发送端向另一端发送紧急数据的一种方式。紧急指针指出在本报文段中紧急数据共有多少个字节（紧急数据放在本报文段数据的最前面）。
Options 可变 选项字段。TCP协议最初只规定了一种选项，即最长报文段长度（只包含数据字段，不包括TCP首部），又称为MSS。MSS告诉对方TCP“我的缓存所能接收的报文段的数据字段的最大长度是MSS个字节”。
新的RFC规定有以下几种选型：选项表结束，空操作，最大报文段长度，窗口扩大因子，时间戳。
选项表结束。
空操作：没有特殊含义，一般用于将TCP选项的总长度填充为4字节的整数倍。
最大报文段长度：又称为MSS，只包含数据字段，不包括TCP首部。
窗口扩大因子：3字节，其中一个字节表示偏移值S。新的窗口值等于TCP首部中的窗口位数增大到（16+S），相当于把窗口值向左移动S位后获得实际的窗口大小。
时间戳：10字节，其中最主要的字段是时间戳值（4字节）和时间戳回送应答字段（4字节）。
Padding 可变 填充字段，用来补位，使整个首部长度是4字节的整数倍。
*/
type TCPPacket struct {
SrcPort uint16
DstPort uint16
Sequence uint32
AckNum uint32
orgBites uint16
Window uint16
CheckSum uint16
UrgentPtr uint16
DataOffset uint8
Reserved uint8
URG bool
ACK bool
PSH bool
RST bool
SYN bool
FIN bool
Options []byte
}
// 14.byte EthernetPacket
// 20.byte IPv4Packet 或者 IPv6Packet
// 返回负载下标起始位
func NewTCPPacket(b []byte) (tcp TCPPacket, next uint8) {
if len(b) < SizeofTCPPacket {
return
}
tcp.SrcPort = binary.BigEndian.Uint16(b[:2])
tcp.DstPort = binary.BigEndian.Uint16(b[2:4])
tcp.Sequence = binary.BigEndian.Uint32(b[4:8])
tcp.AckNum = binary.BigEndian.Uint32(b[8:12])
tcp.orgBites = binary.BigEndian.Uint16(b[12:14])
tcp.Window = binary.BigEndian.Uint16(b[14:16])
tcp.CheckSum = binary.BigEndian.Uint16(b[16:18])
tcp.UrgentPtr = binary.BigEndian.Uint16(b[18:20])
tcp.DataOffset, next = uint8(tcp.orgBites & 0b1111000000000000 >> 10), SizeofTCPPacket
if tcp.DataOffset < SizeofTCPPacket || len(b) < int(tcp.DataOffset) {
tcp, next = TCPPacket{}, 0
return
}
if tcp.DataOffset > SizeofTCPPacket {
tcp.Options, next = make([]byte, tcp.DataOffset - SizeofTCPPacket), tcp.DataOffset
copy(tcp.Options, b[SizeofTCPPacket:tcp.DataOffset])
}
tcp.URG = (tcp.orgBites & 0b0000000000100000) != 0
tcp.ACK = (tcp.orgBites & 0b0000000000010000) != 0
tcp.PSH = (tcp.orgBites & 0b0000000000001000) != 0
tcp.RST = (tcp.orgBites & 0b0000000000000100) != 0
tcp.SYN = (tcp.orgBites & 0b0000000000000010) != 0
tcp.FIN = (tcp.orgBites & 0b0000000000000001) != 0
return
}
func (tcp TCPPacket) WireFormat() []byte {
opLen := len(tcp.Options)
if opLen > 40 {
return nil
}
b := make([]byte, SizeofTCPPacket + opLen)
if opLen > 0 {
copy(b[SizeofTCPPacket:], tcp.Options)
b = append(b, make([]byte, opLen % 4)...)
}
binary.BigEndian.PutUint16(b[:2], tcp.SrcPort)
binary.BigEndian.PutUint16(b[2:4], tcp.DstPort)
binary.BigEndian.PutUint32(b[4:8], tcp.Sequence)
binary.BigEndian.PutUint32(b[8:12], tcp.AckNum)
var tmp uint16
tmp = uint16(tcp.DataOffset >> 2)
if tcp.URG {
tmp |= 0b0000000000100000
}
if tcp.ACK {
tmp |= 0b0000000000010000
}
if tcp.PSH {
tmp |= 0b0000000000001000
}
if tcp.RST {
tmp |= 0b0000000000000100
}
if tcp.SYN {
tmp |= 0b0000000000000010
}
if tcp.FIN {
tmp |= 0b0000000000000001
}
binary.BigEndian.PutUint16(b[12:14], tmp)
binary.BigEndian.PutUint16(b[14:16], tcp.Window)
binary.BigEndian.PutUint16(b[16:18], tcp.CheckSum)
binary.BigEndian.PutUint16(b[18:20], tcp.UrgentPtr)
return b
}