C语言计算TCP/UDP校验和

---

前言

在发送/接收报文的过程中，需要计算TCP/UDP的checksum来验证报文的准确性。

---

一、TCP/UDP校验和计算原理

TCP/UDP校验和：伪首部+tcp/udp头(校验和位 置为0)+tcp/udp数据，对该结果进行补0，相加，取反，即为最终的校验和

伪首部（12字节）：源IP+目的IP+保留字节+传输层协议号+报文总长度（tcp/udp头+tcp/udp数据的总长度）

注：IP层校验和原理和tcp/udp校验和原理类似，将校验和位 置为0，IP层数据每两个字节（16位）相加，相加的结果不满16位，前面补充0，再相加，然后对该结果取反即为IP检验和

二、TCP校验和实例分析

伪首部：

0xc0a8+0x0164+0xc0a8+0x01c8+0x0006+0x0020=0x184a2

校验和位 置为0的tcp报文：

0x0017+0xf012+0x97f9+0xa504+0x666d+0x51eb+0x8012+0x7210+0x0000+0x000+0x0204+0x05b4+0x0101+0x0402+0x0103+0x0304=0x3e862

将伪首部和报文相加

0x184a2+0x3e862=0x56d04

相加的结果不满16位，补零，相加

0x0005+0x6d04=0x6d09

按位取反

~(0x6d09)=0x92f6

最终的tcp校验和为0x92f6，和抓包报文中的结果一致

二、代码实现

传入的参数依次为tcp/udp报文的内容，协议，tcp/udp头+tcp/udp数据的总长度，源IP，目的IP。

代码如下：

static u16 checksum_compute(void *data, int protocol,int len, __be32 sip,__be32 dip)
{
	int sum, oddbyte, phlen;
	u16 *ptr;

	struct {
		__be32 sip;
		__be32 dip;
		u8 zero;
		u8 prot;
		__be16 len;
	} pseudo_hdr = { sip, dip, 0, protocol , htons(len & 0xffff) };

	sum = 0;

	phlen = sizeof(pseudo_hdr);
	ptr = (u16 *) &pseudo_hdr;

	while (phlen > 1) {
		sum += *ptr++;
		phlen -= 2;
	}

	ptr = (u16 *) data;

	while (len > 1) {
		sum += *ptr++;
		len -= 2;
	}

	if (len == 1) {
		oddbyte = 0;
		((u8 *) &oddbyte)[0] = *(u8 *) ptr;
		((u8 *) &oddbyte)[1] = 0;
		sum += oddbyte;
	}

	sum = (sum >> 16) + (sum & 0xffff);
	sum += (sum >> 16);
	sum = ~sum & 0xffff;

	return sum;
}

总结

以上就是TCP/UDP校验和计算原理和实例