diff options
Diffstat (limited to 'mm/util.c')
| -rw-r--r-- | mm/util.c | 37 |
1 files changed, 37 insertions, 0 deletions
| @@ -1,6 +1,8 @@ | |||
| 1 | #include <linux/slab.h> | 1 | #include <linux/slab.h> |
| 2 | #include <linux/string.h> | 2 | #include <linux/string.h> |
| 3 | #include <linux/module.h> | 3 | #include <linux/module.h> |
| 4 | #include <linux/err.h> | ||
| 5 | #include <asm/uaccess.h> | ||
| 4 | 6 | ||
| 5 | /** | 7 | /** |
| 6 | * kzalloc - allocate memory. The memory is set to zero. | 8 | * kzalloc - allocate memory. The memory is set to zero. |
| @@ -37,3 +39,38 @@ char *kstrdup(const char *s, gfp_t gfp) | |||
| 37 | return buf; | 39 | return buf; |
| 38 | } | 40 | } |
| 39 | EXPORT_SYMBOL(kstrdup); | 41 | EXPORT_SYMBOL(kstrdup); |
| 42 | |||
| 43 | /* | ||
| 44 | * strndup_user - duplicate an existing string from user space | ||
| 45 | * | ||
| 46 | * @s: The string to duplicate | ||
| 47 | * @n: Maximum number of bytes to copy, including the trailing NUL. | ||
| 48 | */ | ||
| 49 | char *strndup_user(const char __user *s, long n) | ||
| 50 | { | ||
| 51 | char *p; | ||
| 52 | long length; | ||
| 53 | |||
| 54 | length = strnlen_user(s, n); | ||
| 55 | |||
| 56 | if (!length) | ||
| 57 | return ERR_PTR(-EFAULT); | ||
| 58 | |||
| 59 | if (length > n) | ||
| 60 | return ERR_PTR(-EINVAL); | ||
| 61 | |||
| 62 | p = kmalloc(length, GFP_KERNEL); | ||
| 63 | |||
| 64 | if (!p) | ||
| 65 | return ERR_PTR(-ENOMEM); | ||
| 66 | |||
| 67 | if (copy_from_user(p, s, length)) { | ||
| 68 | kfree(p); | ||
| 69 | return ERR_PTR(-EFAULT); | ||
| 70 | } | ||
| 71 | |||
| 72 | p[length - 1] = '\0'; | ||
| 73 | |||
| 74 | return p; | ||
| 75 | } | ||
| 76 | EXPORT_SYMBOL(strndup_user); | ||
