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); | ||