| 1 |
root |
1.1 |
/* enable the POSIX prototypes of mmap/munmap on Solaris */ |
| 2 |
|
|
#ifdef __sun |
| 3 |
|
|
# if __STDC_VERSION__ >= 199901L |
| 4 |
|
|
# define _XOPEN_SOURCE 600 |
| 5 |
|
|
# else |
| 6 |
|
|
# define _XOPEN_SOURCE 500 |
| 7 |
|
|
# endif |
| 8 |
|
|
#endif |
| 9 |
|
|
|
| 10 |
|
|
#ifndef _WIN32 |
| 11 |
|
|
#include <unistd.h> |
| 12 |
|
|
#endif |
| 13 |
|
|
|
| 14 |
|
|
#if !defined USE_MMAP \ |
| 15 |
|
|
&& _POSIX_MAPPED_FILES > 0 \ |
| 16 |
|
|
&& (_POSIX_VERSION >= 200809L || _POSIX_MEMORY_PROTECTION > 0) |
| 17 |
|
|
#include <sys/mman.h> |
| 18 |
|
|
#if !defined MAP_ANONYMOUS && defined MAP_ANON |
| 19 |
|
|
#define MAP_ANONYMOUS MAP_ANON |
| 20 |
|
|
#endif |
| 21 |
|
|
#ifdef MAP_ANONYMOUS |
| 22 |
|
|
#include <limits.h> |
| 23 |
|
|
#if PAGESIZE <= 0 |
| 24 |
|
|
static long pagesize; |
| 25 |
|
|
#define PAGESIZE pagesize ? pagesize : (pagesize = sysconf (_SC_PAGESIZE)) |
| 26 |
|
|
#endif |
| 27 |
|
|
#define USE_MMAP 1 |
| 28 |
|
|
#endif |
| 29 |
|
|
#endif |
| 30 |
|
|
|
| 31 |
|
|
/* we assume natural alignment, uulib only uses ints and chars */ |
| 32 |
|
|
#define ALIGN 1 |
| 33 |
|
|
#define GUARDS 4 |
| 34 |
|
|
|
| 35 |
|
|
static void * |
| 36 |
|
|
safe_alloc (size_t size) |
| 37 |
|
|
{ |
| 38 |
|
|
#if USE_MMAP |
| 39 |
|
|
|
| 40 |
|
|
size_t rounded = (size + ALIGN - 1) & ~(ALIGN - 1); |
| 41 |
|
|
size_t page = PAGESIZE; |
| 42 |
|
|
size_t page_rounded = (rounded + page - 1) & ~(page - 1); |
| 43 |
|
|
void *base = mmap (0, page_rounded + page * GUARDS * 2, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, -1, 0); |
| 44 |
|
|
|
| 45 |
|
|
if (base == (void *)-1) |
| 46 |
|
|
return 0; |
| 47 |
|
|
|
| 48 |
|
|
mprotect (base, page * GUARDS, PROT_NONE); /* beginning */ |
| 49 |
|
|
mprotect (page_rounded + page * GUARDS + (char *)base, page * GUARDS, PROT_NONE); /* end */ |
| 50 |
|
|
|
| 51 |
|
|
return page * GUARDS + (page_rounded - rounded) + (char *)base; |
| 52 |
|
|
|
| 53 |
|
|
#else |
| 54 |
|
|
return malloc (size); |
| 55 |
|
|
#endif |
| 56 |
|
|
} |
| 57 |
|
|
|
| 58 |
|
|
static void |
| 59 |
|
|
safe_free (void *mem, size_t size) |
| 60 |
|
|
{ |
| 61 |
|
|
#if USE_MMAP |
| 62 |
|
|
|
| 63 |
|
|
size_t rounded = (size + ALIGN - 1) & ~(ALIGN - 1); |
| 64 |
|
|
size_t page = PAGESIZE; |
| 65 |
|
|
size_t page_rounded = (rounded + page - 1) & ~(page - 1); |
| 66 |
|
|
|
| 67 |
|
|
if (!mem) |
| 68 |
|
|
return; |
| 69 |
|
|
|
| 70 |
|
|
mem = (char *)mem - page * GUARDS - (page_rounded - rounded); |
| 71 |
|
|
|
| 72 |
|
|
munmap (mem, page_rounded + page * GUARDS * 2); |
| 73 |
|
|
|
| 74 |
|
|
#else |
| 75 |
|
|
free (size); |
| 76 |
|
|
#endif |
| 77 |
|
|
} |
| 78 |
|
|
|