提交 a8203725 编写于 作者: X Xi Wang 提交者: Pekka Enberg

slab: introduce kmalloc_array()

Introduce a kmalloc_array() wrapper that performs integer overflow
checking without zeroing the memory.
Suggested-by: NAndrew Morton <akpm@linux-foundation.org>
Suggested-by: NJens Axboe <axboe@kernel.dk>
Signed-off-by: NXi Wang <xi.wang@gmail.com>
Cc: Dan Carpenter <dan.carpenter@oracle.com>
Acked-by: NDavid Rientjes <rientjes@google.com>
Signed-off-by: NAndrew Morton <akpm@linux-foundation.org>
Signed-off-by: NPekka Enberg <penberg@kernel.org>
上级 8028dcea
...@@ -190,7 +190,7 @@ size_t ksize(const void *); ...@@ -190,7 +190,7 @@ size_t ksize(const void *);
#endif #endif
/** /**
* kcalloc - allocate memory for an array. The memory is set to zero. * kmalloc_array - allocate memory for an array.
* @n: number of elements. * @n: number of elements.
* @size: element size. * @size: element size.
* @flags: the type of memory to allocate. * @flags: the type of memory to allocate.
...@@ -240,11 +240,22 @@ size_t ksize(const void *); ...@@ -240,11 +240,22 @@ size_t ksize(const void *);
* for general use, and so are not documented here. For a full list of * for general use, and so are not documented here. For a full list of
* potential flags, always refer to linux/gfp.h. * potential flags, always refer to linux/gfp.h.
*/ */
static inline void *kcalloc(size_t n, size_t size, gfp_t flags) static inline void *kmalloc_array(size_t n, size_t size, gfp_t flags)
{ {
if (size != 0 && n > ULONG_MAX / size) if (size != 0 && n > ULONG_MAX / size)
return NULL; return NULL;
return __kmalloc(n * size, flags | __GFP_ZERO); return __kmalloc(n * size, flags);
}
/**
* kcalloc - allocate memory for an array. The memory is set to zero.
* @n: number of elements.
* @size: element size.
* @flags: the type of memory to allocate (see kmalloc).
*/
static inline void *kcalloc(size_t n, size_t size, gfp_t flags)
{
return kmalloc_array(n, size, flags | __GFP_ZERO);
} }
#if !defined(CONFIG_NUMA) && !defined(CONFIG_SLOB) #if !defined(CONFIG_NUMA) && !defined(CONFIG_SLOB)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册