git.sur5r.net Git - freertos/blob - FreeRTOS-Plus/CyaSSL/ctaocrypt/src/memory.c

   1 /* memory.c
   2  *
   3  * Copyright (C) 2006-2012 Sawtooth Consulting Ltd.
   4  *
   5  * This file is part of CyaSSL.
   6  *
   7  * CyaSSL is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * CyaSSL is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
  20  */
  21
  22 #ifdef HAVE_CONFIG_H
  23     #include <config.h>
  24 #endif
  25
  26 /* submitted by eof */
  27
  28
  29 #include <cyassl/ctaocrypt/settings.h>
  30
  31 #ifdef USE_CYASSL_MEMORY
  32
  33 #include <cyassl/ctaocrypt/memory.h>
  34 #include <cyassl/ctaocrypt/error.h>
  35
  36
  37 /* Set these to default values initially. */
  38 static CyaSSL_Malloc_cb  malloc_function = 0;
  39 static CyaSSL_Free_cb    free_function = 0;
  40 static CyaSSL_Realloc_cb realloc_function = 0;
  41
  42 int CyaSSL_SetAllocators(CyaSSL_Malloc_cb  mf,
  43                          CyaSSL_Free_cb    ff,
  44                          CyaSSL_Realloc_cb rf)
  45 {
  46     int res = 0;
  47
  48     if (mf)
  49         malloc_function = mf;
  50         else
  51         res = BAD_FUNC_ARG;
  52
  53     if (ff)
  54         free_function = ff;
  55     else
  56         res = BAD_FUNC_ARG;
  57
  58     if (rf)
  59         realloc_function = rf;
  60     else
  61         res = BAD_FUNC_ARG;
  62
  63     return res;
  64 }
  65
  66
  67 void* CyaSSL_Malloc(size_t size)
  68 {
  69     void* res = 0;
  70
  71     if (malloc_function)
  72         res = malloc_function(size);
  73     else
  74         res = malloc(size);
  75
  76     return res;
  77 }
  78
  79 void CyaSSL_Free(void *ptr)
  80 {
  81     if (free_function)
  82         free_function(ptr);
  83     else
  84         free(ptr);
  85 }
  86
  87 void* CyaSSL_Realloc(void *ptr, size_t size)
  88 {
  89     void* res = 0;
  90
  91     if (realloc_function)
  92         res = realloc_function(ptr, size);
  93     else
  94         res = realloc(ptr, size);
  95
  96     return res;
  97 }
  98
  99 #endif /* USE_CYASSL_MEMORY */