linux_compat: handle __GFP_ZERO in kmalloc()

[karo-tx-uboot.git] / lib / linux_compat.c

diff --git a/lib/linux_compat.c b/lib/linux_compat.c
index 8c7a7b57525aed4ef7ab53507950d470db5c3d51..a936a7eac21437fd4ac0d24d32b71e418345ac03 100644 (file)
--- a/lib/linux_compat.c
+++ b/lib/linux_compat.c
@@ -16,14 +16,13 @@ unsigned long copy_from_user(void *dest, const void *src,
 
 void *kmalloc(size_t size, int flags)
 {
-       return memalign(ARCH_DMA_MINALIGN, size);
-}
+       void *p;
 
-void *kzalloc(size_t size, int flags)
-{
-       void *ptr = kmalloc(size, flags);
-       memset(ptr, 0, size);
-       return ptr;
+       p = memalign(ARCH_DMA_MINALIGN, size);
+       if (flags & __GFP_ZERO)
+               memset(p, 0, size);
+
+       return p;
 }
 
 struct kmem_cache *get_mem(int element_sz)