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(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(len, gfp);
  37         if (buf)
  38                 memcpy(buf, s, len);
  39         return buf;
  40 }
  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);