aboutsummaryrefslogtreecommitdiffstats
path: root/modules/eng_gmp/gmp_mem.cpp
blob: ea8ef4d3bf7b2b1f210b985e7097e7aeb6c8bdb9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
/*************************************************
* GNU MP Memory Handlers Source File             *
* (C) 1999-2007 Jack Lloyd                       *
*************************************************/

#include <botan/eng_gmp.h>
#include <cstring>
#include <gmp.h>

namespace Botan {

namespace {

/*************************************************
* Allocator used by GNU MP                       *
*************************************************/
Allocator* gmp_alloc = 0;

/*************************************************
* Allocation Function for GNU MP                 *
*************************************************/
void* gmp_malloc(size_t n)
   {
   return gmp_alloc->allocate(n);
   }

/*************************************************
* Reallocation Function for GNU MP               *
*************************************************/
void* gmp_realloc(void* ptr, size_t old_n, size_t new_n)
   {
   void* new_buf = gmp_alloc->allocate(new_n);
   std::memcpy(new_buf, ptr, std::min(old_n, new_n));
   gmp_alloc->deallocate(ptr, old_n);
   return new_buf;
   }

/*************************************************
* Deallocation Function for GNU MP               *
*************************************************/
void gmp_free(void* ptr, size_t n)
   {
   gmp_alloc->deallocate(ptr, n);
   }

}

/*************************************************
* Set the GNU MP memory functions                *
*************************************************/
void GMP_Engine::set_memory_hooks()
   {
   if(gmp_alloc == 0)
      {
      gmp_alloc = Allocator::get(true);
      mp_set_memory_functions(gmp_malloc, gmp_realloc, gmp_free);
      }
   }

}