spl: spl_legacy: Use IS_ENABLED() to remove #ifdef

[oweals/u-boot.git] / lib / linux_compat.c

diff --git a/lib/linux_compat.c b/lib/linux_compat.c
index 6373b4451eb3fdafaed99cfeb7eb40f7821712b0..89a6fd6ec96d5e10ae4b890d2cb1200ca4332ab9 100644 (file)
--- a/lib/linux_compat.c
+++ b/lib/linux_compat.c
@@ -1,5 +1,6 @@
 
 #include <common.h>
+#include <malloc.h>
 #include <memalign.h>
 #include <linux/compat.h>
 
@@ -20,7 +21,7 @@ void *kmalloc(size_t size, int flags)
        void *p;
 
        p = malloc_cache_aligned(size);
-       if (flags & __GFP_ZERO)
+       if (p && flags & __GFP_ZERO)
                memset(p, 0, size);
 
        return p;
@@ -40,3 +41,22 @@ void *kmem_cache_alloc(struct kmem_cache *obj, int flag)
 {
        return malloc_cache_aligned(obj->sz);
 }
+
+/**
+ * kmemdup - duplicate region of memory
+ *
+ * @src: memory region to duplicate
+ * @len: memory region length
+ * @gfp: GFP mask to use
+ *
+ * Return: newly allocated copy of @src or %NULL in case of error
+ */
+void *kmemdup(const void *src, size_t len, gfp_t gfp)
+{
+       void *p;
+
+       p = kmalloc(len, gfp);
+       if (p)
+               memcpy(p, src, len);
+       return p;
+}