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