kern/src/kreallocarray.c

   1 /*
   2  * Copyright (c) 2008 Otto Moerbeek <otto@drijf.net>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16
  17 #include <ros/common.h>
  18 #include <kmalloc.h>
  19
  20 /*
  21  * This is sqrt(SIZE_MAX+1), as s1*s2 <= SIZE_MAX
  22  * if both s1 < MUL_NO_OVERFLOW and s2 < MUL_NO_OVERFLOW
  23  */
  24 enum {
  25         MUL_NO_OVERFLOW = ((size_t)1 << (sizeof(size_t) * 4))
  26 };
  27
  28 void *kreallocarray(void *optr, size_t nmemb, size_t size, int flags)
  29 {
  30         if (((nmemb >= MUL_NO_OVERFLOW) || (size >= MUL_NO_OVERFLOW)) &&
  31             (nmemb > 0) && ((SIZE_MAX / nmemb) < size)) {
  32                 return NULL;
  33         }
  34         return krealloc(optr, size * nmemb, flags);
  35 }
  36
  37 void *kmalloc_array(size_t nmemb, size_t size, int flags)
  38 {
  39         if (((nmemb >= MUL_NO_OVERFLOW) || (size >= MUL_NO_OVERFLOW)) &&
  40             (nmemb > 0) && ((SIZE_MAX / nmemb) < size)) {
  41                 return NULL;
  42         }
  43         return kmalloc(size * nmemb, flags);
  44 }