ip tcp check sum

unsigned short ip_fast_csum(unsigned char * iph,
  unsigned int ihl)
{
unsigned int sum;

__asm__ __volatile__(
    “movl (%1), %0 ;\n”
    “subl $4, %2 ;\n”
    “jbe 2f ;\n”
    “addl 4(%1), %0 ;\n”
    “adcl 8(%1), %0 ;\n”
    “adcl 12(%1), %0 ;\n”
“1:     adcl 16(%1), %0 ;\n”
    “lea 4(%1), %1 ;\n”
    “decl %2 ;\n”
    “jne 1b ;\n”
    “adcl $0, %0 ;\n”
    “movl %0, %2 ;\n”
    “shrl $16, %0 ;\n”
    “addw %w2, %w0 ;\n”
    “adcl $0, %0 ;\n”
    “notl %0 ;\n”
“2: ;\n”
/* Since the input registers which are loaded with iph and ihl
  are modified, we must also specify them as outputs, or gcc
  will assume they contain their original values. */
: “=r” (sum), “=r” (iph), “=r” (ihl)
: “1″ (iph), “2″ (ihl)
: “memory”);
return(sum);
}

unsigned short checksum(unsigned short *buf,int nword)
{
    unsigned long sum;
   
    for(sum=0;nword>0;nword–)
        sum += *buf++;
    sum = (sum>>16) + (sum&0xffff);
    sum += (sum>>16);
   
    return ~sum;
}

TCP CHECKSUM CALCULATION
(See also short Description of Internet Checksum).

To calculate TCP checksum a "pseudo header" is added to the TCP header. This includes:

IP Source Address		4 bytes
IP Destination Address		4 bytes
TCP Protocol			2 bytes
TCP Length  			2 bytes

The checksum is calculated over all the octets of the pseudo header, TCP header and data. 
If the data contains an odd number of octets a pad, zero octet is added to the end of data. The pseudo 
header and the pad are not transmitted with the packet. 

In the example code, 
u16 buff[] is an array containing all the octets in the TCP header and data.
u16 len_tcp is the length (number of octets) of the TCP header and data.
BOOL padding is 1 if data has an even number of octets and 0 for an odd number. 
u16 src_addr[4] and u16 dest_addr[4] are the IP source and destination address octets.

/*
**************************************************************************
Function: tcp_sum_calc()
**************************************************************************
Description: 
	Calculate TCP checksum
***************************************************************************
*/

typedef unsigned short u16;
typedef unsigned long u32;

u16 tcp_sum_calc(u16 len_tcp, u16 src_addr[],u16 dest_addr[], BOOL padding, u16 buff[])
{
u16 prot_tcp=6;
u16 padd=0;
u16 word16;
u32 sum;	
	
	// Find out if the length of data is even or odd number. If odd,
	// add a padding byte = 0 at the end of packet
	if (padding&1==1){
		padd=1;
		buff[len_tcp]=0;
	}
	
	//initialize sum to zero
	sum=0;
	
	// make 16 bit words out of every two adjacent 8 bit words and 
	// calculate the sum of all 16 vit words
	for (i=0;i<len_tcp+padd;i=i+2){
		word16 =((buff[i]<<8)&0xFF00)+(buff[i+1]&0xFF);
		sum = sum + (unsigned long)word16;
	}	
	// add the TCP pseudo header which contains:
	// the IP source and destinationn addresses,
	for (i=0;i<4;i=i+2){
		word16 =((src_addr[i]<<8)&0xFF00)+(src_addr[i+1]&0xFF);
		sum=sum+word16;	
	}
	for (i=0;i<4;i=i+2){
		word16 =((dest_addr[i]<<8)&0xFF00)+(dest_addr[i+1]&0xFF);
		sum=sum+word16; 	
	}
	// the protocol number and the length of the TCP packet
	sum = sum + prot_tcp + len_tcp;

	// keep only the last 16 bits of the 32 bit calculated sum and add the carries
    	while (sum>>16)
		sum = (sum & 0xFFFF)+(sum >> 16);
		
	// Take the one's complement of sum
	sum = ~sum;

return ((unsigned short) sum);
}