1 /* 2 * This program is free software; you can redistribute it and/or modify 3 * it under the terms of the GNU General Public License as published by 4 * the Free Software Foundation; either version 2 of the License, or 5 * (at your option) any later version. 6 * 7 * This program is distributed in the hope that it will be useful, 8 * but WITHOUT ANY WARRANTY; without even the implied warranty of 9 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 10 * GNU General Public License for more details. 11 * 12 * You should have received a copy of the GNU General Public License 13 * along with this program; if not, write to the Free Software 14 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. 15 * 16 * Copyright (C) IBM Corporation, 2010 17 * 18 * Author: Anton Blanchard <anton@au.ibm.com> 19 */ 20 #include <linux/export.h> 21 #include <linux/compiler.h> 22 #include <linux/types.h> 23 #include <asm/checksum.h> 24 #include <linux/uaccess.h> 25 26 __wsum csum_and_copy_from_user(const void __user *src, void *dst, 27 int len, __wsum sum, int *err_ptr) 28 { 29 unsigned int csum; 30 31 might_sleep(); 32 33 *err_ptr = 0; 34 35 if (!len) { 36 csum = 0; 37 goto out; 38 } 39 40 if (unlikely((len < 0) || !access_ok(src, len))) { 41 *err_ptr = -EFAULT; 42 csum = (__force unsigned int)sum; 43 goto out; 44 } 45 46 csum = csum_partial_copy_generic((void __force *)src, dst, 47 len, sum, err_ptr, NULL); 48 49 if (unlikely(*err_ptr)) { 50 int missing = __copy_from_user(dst, src, len); 51 52 if (missing) { 53 memset(dst + len - missing, 0, missing); 54 *err_ptr = -EFAULT; 55 } else { 56 *err_ptr = 0; 57 } 58 59 csum = csum_partial(dst, len, sum); 60 } 61 62 out: 63 return (__force __wsum)csum; 64 } 65 EXPORT_SYMBOL(csum_and_copy_from_user); 66 67 __wsum csum_and_copy_to_user(const void *src, void __user *dst, int len, 68 __wsum sum, int *err_ptr) 69 { 70 unsigned int csum; 71 72 might_sleep(); 73 74 *err_ptr = 0; 75 76 if (!len) { 77 csum = 0; 78 goto out; 79 } 80 81 if (unlikely((len < 0) || !access_ok(dst, len))) { 82 *err_ptr = -EFAULT; 83 csum = -1; /* invalid checksum */ 84 goto out; 85 } 86 87 csum = csum_partial_copy_generic(src, (void __force *)dst, 88 len, sum, NULL, err_ptr); 89 90 if (unlikely(*err_ptr)) { 91 csum = csum_partial(src, len, sum); 92 93 if (copy_to_user(dst, src, len)) { 94 *err_ptr = -EFAULT; 95 csum = -1; /* invalid checksum */ 96 } 97 } 98 99 out: 100 return (__force __wsum)csum; 101 } 102 EXPORT_SYMBOL(csum_and_copy_to_user); 103