lib: Add memdup()

Add a function to duplicate a memory region, a little like strdup().

Signed-off-by: Simon Glass <sjg@chromium.org>

diff --git a/include/linux/string.h b/include/linux/string.h
index dd255f21633a73982acd540d9348f9bdb610b611..3169c93796e929382940cf9d19819b826a0c49a3 100644 (file)
--- a/include/linux/string.h
+++ b/include/linux/string.h
@@ -129,6 +129,19 @@ extern void * memchr(const void *,int,__kernel_size_t);
 void *memchr_inv(const void *, int, size_t);
 #endif
 
+/**
+ * memdup() - allocate a buffer and copy in the contents
+ *
+ * Note that this returns a valid pointer even if @len is 0
+ *
+ * @src: data to copy in
+ * @len: number of bytes to copy
+ * @return allocated buffer with the copied contents, or NULL if not enough
+ *     memory is available
+ *
+ */
+char *memdup(const void *src, size_t len);
+
 unsigned long ustrtoul(const char *cp, char **endp, unsigned int base);
 unsigned long long ustrtoull(const char *cp, char **endp, unsigned int base);
 

diff --git a/lib/string.c b/lib/string.c
index ba176fb08f73710fc0d22bbec4312108517fa8d1..78bd65c41369cd5b0050b75013bc97bbc7b5bda5 100644 (file)
--- a/lib/string.c
+++ b/lib/string.c
@@ -659,6 +659,19 @@ void * memscan(void * addr, int c, size_t size)
 }
 #endif
 
+char *memdup(const void *src, size_t len)
+{
+       char *p;
+
+       p = malloc(len);
+       if (!p)
+               return NULL;
+
+       memcpy(p, src, len);
+
+       return p;
+}
+
 #ifndef __HAVE_ARCH_STRSTR
 /**
  * strstr - Find the first substring in a %NUL terminated string

diff --git a/test/lib/string.c b/test/lib/string.c
index 64234bef36cd741ee957ab46fe13644f1ddaec28..5dcf4d6db00389f6ed28bbb8add28409eeec7c7d 100644 (file)
--- a/test/lib/string.c
+++ b/test/lib/string.c
@@ -23,6 +23,8 @@
 /* Allow for copying up to 32 bytes */
 #define BUFLEN (SWEEP + 33)
 
+#define TEST_STR       "hello"
+
 /**
  * init_buffer() - initialize buffer
  *
@@ -193,3 +195,33 @@ static int lib_memmove(struct unit_test_state *uts)
 }
 
 LIB_TEST(lib_memmove, 0);
+
+/** lib_memdup() - unit test for memdup() */
+static int lib_memdup(struct unit_test_state *uts)
+{
+       char buf[BUFLEN];
+       size_t len;
+       char *p, *q;
+
+       /* Zero size should do nothing */
+       p = memdup(NULL, 0);
+       ut_assertnonnull(p);
+       free(p);
+
+       p = memdup(buf, 0);
+       ut_assertnonnull(p);
+       free(p);
+
+       strcpy(buf, TEST_STR);
+       len = sizeof(TEST_STR);
+       p = memdup(buf, len);
+       ut_asserteq_mem(p, buf, len);
+
+       q = memdup(p, len);
+       ut_asserteq_mem(q, buf, len);
+       free(q);
+       free(p);
+
+       return 0;
+}
+LIB_TEST(lib_memdup, 0);