xref: /kvmtool/net/uip/csum.c (revision 07197967df794fbf133694e1fbf16cfbeabe2d85)
1d8db9f85SAsias He #include "kvm/uip.h"
2d8db9f85SAsias He 
3d8db9f85SAsias He static u16 uip_csum(u16 csum, u8 *addr, u16 count)
4d8db9f85SAsias He {
5d8db9f85SAsias He 	long sum = csum;
6d8db9f85SAsias He 
7d8db9f85SAsias He 	while (count > 1) {
8d8db9f85SAsias He 		sum	+= *(u16 *)addr;
9d8db9f85SAsias He 		addr	+= 2;
10d8db9f85SAsias He 		count	-= 2;
11d8db9f85SAsias He 	}
12d8db9f85SAsias He 
13d8db9f85SAsias He 	if (count > 0)
14d8db9f85SAsias He 		sum += *(unsigned char *)addr;
15d8db9f85SAsias He 
16d8db9f85SAsias He 	while (sum>>16)
17d8db9f85SAsias He 		sum = (sum & 0xffff) + (sum >> 16);
18d8db9f85SAsias He 
19d8db9f85SAsias He 	return ~sum;
20d8db9f85SAsias He }
21d8db9f85SAsias He 
22d8db9f85SAsias He u16 uip_csum_ip(struct uip_ip *ip)
23d8db9f85SAsias He {
24d8db9f85SAsias He 	return uip_csum(0, &ip->vhl, uip_ip_hdrlen(ip));
25d8db9f85SAsias He }
260134a2a7SAsias He 
270134a2a7SAsias He u16 uip_csum_icmp(struct uip_icmp *icmp)
280134a2a7SAsias He {
290134a2a7SAsias He 	struct uip_ip *ip;
300134a2a7SAsias He 
310134a2a7SAsias He 	ip = &icmp->ip;
320134a2a7SAsias He 	return icmp->csum = uip_csum(0, &icmp->type, htons(ip->len) - uip_ip_hdrlen(ip) - 8); /* icmp header len = 8 */
330134a2a7SAsias He }
34*07197967SAsias He 
35*07197967SAsias He u16 uip_csum_udp(struct uip_udp *udp)
36*07197967SAsias He {
37*07197967SAsias He 	struct uip_pseudo_hdr hdr;
38*07197967SAsias He 	struct uip_ip *ip;
39*07197967SAsias He 	int udp_len;
40*07197967SAsias He 	u8 *pad;
41*07197967SAsias He 
42*07197967SAsias He 	ip	  = &udp->ip;
43*07197967SAsias He 
44*07197967SAsias He 	hdr.sip   = ip->sip;
45*07197967SAsias He 	hdr.dip	  = ip->dip;
46*07197967SAsias He 	hdr.zero  = 0;
47*07197967SAsias He 	hdr.proto = ip->proto;
48*07197967SAsias He 	hdr.len   = udp->len;
49*07197967SAsias He 
50*07197967SAsias He 	udp_len	  = uip_udp_len(udp);
51*07197967SAsias He 
52*07197967SAsias He 	if (udp_len % 2) {
53*07197967SAsias He 		pad = (u8 *)&udp->sport + udp_len;
54*07197967SAsias He 		*pad = 0;
55*07197967SAsias He 		memcpy((u8 *)&udp->sport + udp_len + 1, &hdr, sizeof(hdr));
56*07197967SAsias He 		return uip_csum(0, (u8 *)&udp->sport, udp_len + 1 + sizeof(hdr));
57*07197967SAsias He 	} else {
58*07197967SAsias He 		memcpy((u8 *)&udp->sport + udp_len, &hdr, sizeof(hdr));
59*07197967SAsias He 		return uip_csum(0, (u8 *)&udp->sport, udp_len + sizeof(hdr));
60*07197967SAsias He 	}
61*07197967SAsias He 
62*07197967SAsias He }
63