1 #include <linux/slab.h> 2 #include <linux/string.h> 3 #include <linux/module.h> 4 #include <linux/err.h> 5 #include <asm/uaccess.h> 6 7 /** 8 * kstrdup - allocate space for and copy an existing string 9 * @s: the string to duplicate 10 * @gfp: the GFP mask used in the kmalloc() call when allocating memory 11 */ 12 char *kstrdup(const char *s, gfp_t gfp) 13 { 14 size_t len; 15 char *buf; 16 17 if (!s) 18 return NULL; 19 20 len = strlen(s) + 1; 21 buf = kmalloc_track_caller(len, gfp); 22 if (buf) 23 memcpy(buf, s, len); 24 return buf; 25 } 26 EXPORT_SYMBOL(kstrdup); 27 28 /** 29 * kstrndup - allocate space for and copy an existing string 30 * @s: the string to duplicate 31 * @max: read at most @max chars from @s 32 * @gfp: the GFP mask used in the kmalloc() call when allocating memory 33 */ 34 char *kstrndup(const char *s, size_t max, gfp_t gfp) 35 { 36 size_t len; 37 char *buf; 38 39 if (!s) 40 return NULL; 41 42 len = strnlen(s, max); 43 buf = kmalloc_track_caller(len+1, gfp); 44 if (buf) { 45 memcpy(buf, s, len); 46 buf[len] = '\0'; 47 } 48 return buf; 49 } 50 EXPORT_SYMBOL(kstrndup); 51 52 /** 53 * kmemdup - duplicate region of memory 54 * 55 * @src: memory region to duplicate 56 * @len: memory region length 57 * @gfp: GFP mask to use 58 */ 59 void *kmemdup(const void *src, size_t len, gfp_t gfp) 60 { 61 void *p; 62 63 p = kmalloc_track_caller(len, gfp); 64 if (p) 65 memcpy(p, src, len); 66 return p; 67 } 68 EXPORT_SYMBOL(kmemdup); 69 70 /** 71 * krealloc - reallocate memory. The contents will remain unchanged. 72 * @p: object to reallocate memory for. 73 * @new_size: how many bytes of memory are required. 74 * @flags: the type of memory to allocate. 75 * 76 * The contents of the object pointed to are preserved up to the 77 * lesser of the new and old sizes. If @p is %NULL, krealloc() 78 * behaves exactly like kmalloc(). If @size is 0 and @p is not a 79 * %NULL pointer, the object pointed to is freed. 80 */ 81 void *krealloc(const void *p, size_t new_size, gfp_t flags) 82 { 83 void *ret; 84 size_t ks; 85 86 if (unlikely(!new_size)) { 87 kfree(p); 88 return ZERO_SIZE_PTR; 89 } 90 91 ks = ksize(p); 92 if (ks >= new_size) 93 return (void *)p; 94 95 ret = kmalloc_track_caller(new_size, flags); 96 if (ret) { 97 memcpy(ret, p, min(new_size, ks)); 98 kfree(p); 99 } 100 return ret; 101 } 102 EXPORT_SYMBOL(krealloc); 103 104 /* 105 * strndup_user - duplicate an existing string from user space 106 * @s: The string to duplicate 107 * @n: Maximum number of bytes to copy, including the trailing NUL. 108 */ 109 char *strndup_user(const char __user *s, long n) 110 { 111 char *p; 112 long length; 113 114 length = strnlen_user(s, n); 115 116 if (!length) 117 return ERR_PTR(-EFAULT); 118 119 if (length > n) 120 return ERR_PTR(-EINVAL); 121 122 p = kmalloc(length, GFP_KERNEL); 123 124 if (!p) 125 return ERR_PTR(-ENOMEM); 126 127 if (copy_from_user(p, s, length)) { 128 kfree(p); 129 return ERR_PTR(-EFAULT); 130 } 131 132 p[length - 1] = '\0'; 133 134 return p; 135 } 136 EXPORT_SYMBOL(strndup_user); 137