src/allocators/stdalloc.c

   1 /*
   2  * Copyright (C) the libgit2 contributors. All rights reserved.
   3  *
   4  * This file is part of libgit2, distributed under the GNU GPL v2 with
   5  * a Linking Exception. For full terms see the included COPYING file.
   6  */
   7
   8 #include "stdalloc.h"
   9
  10 static void *stdalloc__malloc(size_t len, const char *file, int line)
  11 {
  12         void *ptr = malloc(len);
  13
  14         GIT_UNUSED(file);
  15         GIT_UNUSED(line);
  16
  17         if (!ptr) git_error_set_oom();
  18         return ptr;
  19 }
  20
  21 static void *stdalloc__calloc(size_t nelem, size_t elsize, const char *file, int line)
  22 {
  23         void *ptr = calloc(nelem, elsize);
  24
  25         GIT_UNUSED(file);
  26         GIT_UNUSED(line);
  27
  28         if (!ptr) git_error_set_oom();
  29         return ptr;
  30 }
  31
  32 static char *stdalloc__strdup(const char *str, const char *file, int line)
  33 {
  34         char *ptr = strdup(str);
  35
  36         GIT_UNUSED(file);
  37         GIT_UNUSED(line);
  38
  39         if (!ptr) git_error_set_oom();
  40         return ptr;
  41 }
  42
  43 static char *stdalloc__strndup(const char *str, size_t n, const char *file, int line)
  44 {
  45         size_t length = 0, alloclength;
  46         char *ptr;
  47
  48         length = p_strnlen(str, n);
  49
  50         if (GIT_ADD_SIZET_OVERFLOW(&alloclength, length, 1) ||
  51                 !(ptr = stdalloc__malloc(alloclength, file, line)))
  52                 return NULL;
  53
  54         if (length)
  55                 memcpy(ptr, str, length);
  56
  57         ptr[length] = '\0';
  58
  59         return ptr;
  60 }
  61
  62 static char *stdalloc__substrdup(const char *start, size_t n, const char *file, int line)
  63 {
  64         char *ptr;
  65         size_t alloclen;
  66
  67         if (GIT_ADD_SIZET_OVERFLOW(&alloclen, n, 1) ||
  68                 !(ptr = stdalloc__malloc(alloclen, file, line)))
  69                 return NULL;
  70
  71         memcpy(ptr, start, n);
  72         ptr[n] = '\0';
  73         return ptr;
  74 }
  75
  76 static void *stdalloc__realloc(void *ptr, size_t size, const char *file, int line)
  77 {
  78         void *new_ptr = realloc(ptr, size);
  79
  80         GIT_UNUSED(file);
  81         GIT_UNUSED(line);
  82
  83         if (!new_ptr) git_error_set_oom();
  84         return new_ptr;
  85 }
  86
  87 static void *stdalloc__reallocarray(void *ptr, size_t nelem, size_t elsize, const char *file, int line)
  88 {
  89         size_t newsize;
  90
  91         if (GIT_MULTIPLY_SIZET_OVERFLOW(&newsize, nelem, elsize))
  92                 return NULL;
  93
  94         return stdalloc__realloc(ptr, newsize, file, line);
  95 }
  96
  97 static void *stdalloc__mallocarray(size_t nelem, size_t elsize, const char *file, int line)
  98 {
  99         return stdalloc__reallocarray(NULL, nelem, elsize, file, line);
 100 }
 101
 102 static void stdalloc__free(void *ptr)
 103 {
 104         free(ptr);
 105 }
 106
 107 int git_stdalloc_init_allocator(git_allocator *allocator)
 108 {
 109         allocator->gmalloc = stdalloc__malloc;
 110         allocator->gcalloc = stdalloc__calloc;
 111         allocator->gstrdup = stdalloc__strdup;
 112         allocator->gstrndup = stdalloc__strndup;
 113         allocator->gsubstrdup = stdalloc__substrdup;
 114         allocator->grealloc = stdalloc__realloc;
 115         allocator->greallocarray = stdalloc__reallocarray;
 116         allocator->gmallocarray = stdalloc__mallocarray;
 117         allocator->gfree = stdalloc__free;
 118         return 0;
 119 }