csum-wrappers_64.c 3.9 KB
Newer Older
1 2
/*
 * Copyright 2002, 2003 Andi Kleen, SuSE Labs.
L
Linus Torvalds 已提交
3
 * Subject to the GNU Public License v.2
4
 *
L
Linus Torvalds 已提交
5 6 7 8
 * Wrappers of assembly checksum functions for x86-64.
 */
#include <asm/checksum.h>
#include <linux/module.h>
9
#include <asm/smap.h>
L
Linus Torvalds 已提交
10

11 12 13
/**
 * csum_partial_copy_from_user - Copy and checksum from user space.
 * @src: source address (user space)
L
Linus Torvalds 已提交
14 15 16 17
 * @dst: destination address
 * @len: number of bytes to be copied.
 * @isum: initial sum that is added into the result (32bit unfolded)
 * @errp: set to -EFAULT for an bad source address.
18
 *
L
Linus Torvalds 已提交
19
 * Returns an 32bit unfolded checksum of the buffer.
20 21
 * src and dst are best aligned to 64bits.
 */
22 23 24
__wsum
csum_partial_copy_from_user(const void __user *src, void *dst,
			    int len, __wsum isum, int *errp)
25
{
L
Linus Torvalds 已提交
26 27
	might_sleep();
	*errp = 0;
28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43

	if (!likely(access_ok(VERIFY_READ, src, len)))
		goto out_err;

	/*
	 * Why 6, not 7? To handle odd addresses aligned we
	 * would need to do considerable complications to fix the
	 * checksum which is defined as an 16bit accumulator. The
	 * fix alignment code is primarily for performance
	 * compatibility with 32bit and that will handle odd
	 * addresses slowly too.
	 */
	if (unlikely((unsigned long)src & 6)) {
		while (((unsigned long)src & 6) && len >= 2) {
			__u16 val16;

44 45
			if (__get_user(val16, (const __u16 __user *)src))
				goto out_err;
46 47 48 49 50 51 52

			*(__u16 *)dst = val16;
			isum = (__force __wsum)add32_with_carry(
					(__force unsigned)isum, val16);
			src += 2;
			dst += 2;
			len -= 2;
L
Linus Torvalds 已提交
53
		}
54
	}
55
	stac();
56 57
	isum = csum_partial_copy_generic((__force const void *)src,
				dst, len, isum, errp, NULL);
58
	clac();
59 60 61 62 63 64
	if (unlikely(*errp))
		goto out_err;

	return isum;

out_err:
L
Linus Torvalds 已提交
65
	*errp = -EFAULT;
66
	memset(dst, 0, len);
67

68 69
	return isum;
}
L
Linus Torvalds 已提交
70 71
EXPORT_SYMBOL(csum_partial_copy_from_user);

72 73
/**
 * csum_partial_copy_to_user - Copy and checksum to user space.
L
Linus Torvalds 已提交
74 75 76 77 78
 * @src: source address
 * @dst: destination address (user space)
 * @len: number of bytes to be copied.
 * @isum: initial sum that is added into the result (32bit unfolded)
 * @errp: set to -EFAULT for an bad destination address.
79
 *
L
Linus Torvalds 已提交
80 81
 * Returns an 32bit unfolded checksum of the buffer.
 * src and dst are best aligned to 64bits.
82
 */
83 84 85
__wsum
csum_partial_copy_to_user(const void *src, void __user *dst,
			  int len, __wsum isum, int *errp)
86
{
87 88
	__wsum ret;

L
Linus Torvalds 已提交
89
	might_sleep();
90

L
Linus Torvalds 已提交
91 92
	if (unlikely(!access_ok(VERIFY_WRITE, dst, len))) {
		*errp = -EFAULT;
93
		return 0;
L
Linus Torvalds 已提交
94 95 96
	}

	if (unlikely((unsigned long)dst & 6)) {
97
		while (((unsigned long)dst & 6) && len >= 2) {
L
Linus Torvalds 已提交
98
			__u16 val16 = *(__u16 *)src;
99

100 101
			isum = (__force __wsum)add32_with_carry(
					(__force unsigned)isum, val16);
L
Linus Torvalds 已提交
102 103 104
			*errp = __put_user(val16, (__u16 __user *)dst);
			if (*errp)
				return isum;
105 106
			src += 2;
			dst += 2;
L
Linus Torvalds 已提交
107 108 109 110 111
			len -= 2;
		}
	}

	*errp = 0;
112 113 114 115 116
	stac();
	ret = csum_partial_copy_generic(src, (void __force *)dst,
					len, isum, NULL, errp);
	clac();
	return ret;
117
}
L
Linus Torvalds 已提交
118 119
EXPORT_SYMBOL(csum_partial_copy_to_user);

120
/**
L
Linus Torvalds 已提交
121 122 123 124
 * csum_partial_copy_nocheck - Copy and checksum.
 * @src: source address
 * @dst: destination address
 * @len: number of bytes to be copied.
W
Wanpeng Li 已提交
125
 * @sum: initial sum that is added into the result (32bit unfolded)
126
 *
L
Linus Torvalds 已提交
127
 * Returns an 32bit unfolded checksum of the buffer.
128
 */
129 130
__wsum
csum_partial_copy_nocheck(const void *src, void *dst, int len, __wsum sum)
131 132 133
{
	return csum_partial_copy_generic(src, dst, len, sum, NULL, NULL);
}
134
EXPORT_SYMBOL(csum_partial_copy_nocheck);
L
Linus Torvalds 已提交
135

136 137 138
__sum16 csum_ipv6_magic(const struct in6_addr *saddr,
			const struct in6_addr *daddr,
			__u32 len, unsigned short proto, __wsum sum)
L
Linus Torvalds 已提交
139 140
{
	__u64 rest, sum64;
141

142 143
	rest = (__force __u64)htonl(len) + (__force __u64)htons(proto) +
		(__force __u64)sum;
144 145 146 147 148 149 150

	asm("	addq (%[saddr]),%[sum]\n"
	    "	adcq 8(%[saddr]),%[sum]\n"
	    "	adcq (%[daddr]),%[sum]\n"
	    "	adcq 8(%[daddr]),%[sum]\n"
	    "	adcq $0,%[sum]\n"

151 152
	    : [sum] "=r" (sum64)
	    : "[sum]" (rest), [saddr] "r" (saddr), [daddr] "r" (daddr));
L
Linus Torvalds 已提交
153

154 155 156
	return csum_fold(
	       (__force __wsum)add32_with_carry(sum64 & 0xffffffff, sum64>>32));
}
L
Linus Torvalds 已提交
157
EXPORT_SYMBOL(csum_ipv6_magic);