Add kmalloc_array() interface
authorBarret Rhoden <brho@cs.berkeley.edu>
Mon, 30 Nov 2015 21:04:08 +0000 (16:04 -0500)
committerBarret Rhoden <brho@cs.berkeley.edu>
Mon, 30 Nov 2015 21:05:31 +0000 (16:05 -0500)
One thing to be careful of is that these functions can fail and return 0,
regardless of your kmalloc flags.

Signed-off-by: Barret Rhoden <brho@cs.berkeley.edu>
kern/include/kmalloc.h
kern/src/kreallocarray.c

index 268cf29..26f23c1 100644 (file)
@@ -16,8 +16,9 @@
 #define KMALLOC_LARGEST KMALLOC_SMALLEST << NUM_KMALLOC_CACHES
 
 void kmalloc_init(void);
-void* kmalloc(size_t size, int flags);
-void* kzmalloc(size_t size, int flags);
+void *kmalloc(size_t size, int flags);
+void *kmalloc_array(size_t nmemb, size_t size, int flags);
+void *kzmalloc(size_t size, int flags);
 void *kmalloc_align(size_t size, int flags, size_t align);
 void *kzmalloc_align(size_t size, int flags, size_t align);
 void *krealloc(void *buf, size_t size, int flags);
index d099721..5051599 100644 (file)
@@ -34,3 +34,12 @@ void *kreallocarray(void *optr, size_t nmemb, size_t size, int flags)
        }
        return krealloc(optr, size * nmemb, flags);
 }
+
+void *kmalloc_array(size_t nmemb, size_t size, int flags)
+{
+       if (((nmemb >= MUL_NO_OVERFLOW) || (size >= MUL_NO_OVERFLOW)) &&
+           (nmemb > 0) && ((SIZE_MAX / nmemb) < size)) {
+               return NULL;
+       }
+       return kmalloc(size * nmemb, flags);
+}