checksum.h 3.2 KB
Newer Older
L
Linus Torvalds 已提交
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
/*
 * INET		An implementation of the TCP/IP protocol suite for the LINUX
 *		operating system.  INET is implemented using the  BSD Socket
 *		interface as the means of communication with the user level.
 *
 *		Checksumming functions for IP, TCP, UDP and so on
 *
 * Authors:	Jorge Cwik, <jorge@laser.satlink.net>
 *		Arnt Gulbrandsen, <agulbra@nvg.unit.no>
 *		Borrows very liberally from tcp.c and ip.c, see those
 *		files for more names.
 *
 *		This program is free software; you can redistribute it and/or
 *		modify it under the terms of the GNU General Public License
 *		as published by the Free Software Foundation; either version
 *		2 of the License, or (at your option) any later version.
 */

#ifndef _CHECKSUM_H
#define _CHECKSUM_H

#include <linux/errno.h>
#include <asm/types.h>
#include <asm/byteorder.h>
#include <asm/uaccess.h>
#include <asm/checksum.h>

#ifndef _HAVE_ARCH_COPY_AND_CSUM_FROM_USER
static inline
30 31
__wsum csum_and_copy_from_user (const void __user *src, void *dst,
				      int len, __wsum sum, int *err_ptr)
L
Linus Torvalds 已提交
32 33 34 35 36 37 38 39 40 41 42 43
{
	if (access_ok(VERIFY_READ, src, len))
		return csum_partial_copy_from_user(src, dst, len, sum, err_ptr);

	if (len)
		*err_ptr = -EFAULT;

	return sum;
}
#endif

#ifndef HAVE_CSUM_COPY_USER
44 45
static __inline__ __wsum csum_and_copy_to_user
(const void *src, void __user *dst, int len, __wsum sum, int *err_ptr)
L
Linus Torvalds 已提交
46 47 48 49 50 51 52 53 54 55
{
	sum = csum_partial(src, len, sum);

	if (access_ok(VERIFY_WRITE, dst, len)) {
		if (copy_to_user(dst, src, len) == 0)
			return sum;
	}
	if (len)
		*err_ptr = -EFAULT;

56
	return (__force __wsum)-1; /* invalid checksum */
L
Linus Torvalds 已提交
57 58 59
}
#endif

60
static inline __wsum csum_add(__wsum csum, __wsum addend)
L
Linus Torvalds 已提交
61
{
62 63 64
	u32 res = (__force u32)csum;
	res += (__force u32)addend;
	return (__force __wsum)(res + (res < (__force u32)addend));
L
Linus Torvalds 已提交
65 66
}

67
static inline __wsum csum_sub(__wsum csum, __wsum addend)
L
Linus Torvalds 已提交
68 69 70 71
{
	return csum_add(csum, ~addend);
}

72 73
static inline __wsum
csum_block_add(__wsum csum, __wsum csum2, int offset)
L
Linus Torvalds 已提交
74
{
75
	u32 sum = (__force u32)csum2;
L
Linus Torvalds 已提交
76
	if (offset&1)
77 78
		sum = ((sum&0xFF00FF)<<8)+((sum>>8)&0xFF00FF);
	return csum_add(csum, (__force __wsum)sum);
L
Linus Torvalds 已提交
79 80
}

81 82
static inline __wsum
csum_block_sub(__wsum csum, __wsum csum2, int offset)
L
Linus Torvalds 已提交
83
{
84
	u32 sum = (__force u32)csum2;
L
Linus Torvalds 已提交
85
	if (offset&1)
86 87 88 89 90 91 92
		sum = ((sum&0xFF00FF)<<8)+((sum>>8)&0xFF00FF);
	return csum_sub(csum, (__force __wsum)sum);
}

static inline __wsum csum_unfold(__sum16 n)
{
	return (__force __wsum)n;
L
Linus Torvalds 已提交
93 94
}

95
#define CSUM_MANGLED_0 ((__force __sum16)0xffff)
96 97 98 99 100

static inline void csum_replace4(__sum16 *sum, __be32 from, __be32 to)
{
	__be32 diff[] = { ~from, to };

101
	*sum = csum_fold(csum_partial(diff, sizeof(diff), ~csum_unfold(*sum)));
102 103 104 105 106 107 108 109 110 111
}

static inline void csum_replace2(__sum16 *sum, __be16 from, __be16 to)
{
	csum_replace4(sum, (__force __be32)from, (__force __be32)to);
}

struct sk_buff;
extern void inet_proto_csum_replace4(__sum16 *sum, struct sk_buff *skb,
				     __be32 from, __be32 to, int pseudohdr);
112 113 114
extern void inet_proto_csum_replace16(__sum16 *sum, struct sk_buff *skb,
				      const __be32 *from, const __be32 *to,
				      int pseudohdr);
115 116 117 118 119 120 121 122 123

static inline void inet_proto_csum_replace2(__sum16 *sum, struct sk_buff *skb,
					    __be16 from, __be16 to,
					    int pseudohdr)
{
	inet_proto_csum_replace4(sum, skb, (__force __be32)from,
				 (__force __be32)to, pseudohdr);
}

L
Linus Torvalds 已提交
124
#endif