src/heap.c

   1 /*
   2  * Copyright (C) 2013 Intel Corporation.
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  17  *
  18  * Authors:
  19  *   José Bollo <jose.bollo@open.eurogiciel.org>
  20  *   Stéphane Desneux <stephane.desneux@open.eurogiciel.org>
  21  *   Jean-Benoit Martin <jean-benoit.martin@open.eurogiciel.org>
  22  *
  23  */
  24 #define _GNU_SOURCE
  25
  26 #ifdef HAVE_CONFIG_H
  27 # include "config.h"
  28 #endif
  29
  30 #include <unistd.h>
  31 #include <string.h>
  32 #include <assert.h>
  33 #include <sys/mman.h>
  34
  35 #include "heap.h"
  36
  37 /* align to a size_t size */
  38 inline static size_t align(size_t size)
  39 {
  40     /* we assume that sizeof(size_t) is a power of 2 */
  41     assert( (sizeof(size_t) & (sizeof(size_t)-1)) == 0 );
  42     return (size + (sizeof(size_t)-1)) & ~(sizeof(size_t)-1);
  43 }
  44
  45 /* align to a page size */
  46 inline static size_t pagealign( size_t size)
  47 {
  48     static size_t pagemask = 0;
  49     /* we assume that pagesize is a power of 2 */
  50     if (!pagemask) {
  51         pagemask = (size_t)sysconf(_SC_PAGE_SIZE) - 1;
  52         assert( pagemask );
  53         assert( (pagemask & (pagemask+1)) == 0 );
  54     }
  55     return (size + pagemask) & ~pagemask;
  56 }
  57
  58 int heap_create( struct heap *heap, size_t capacity)
  59 {
  60     char *data;
  61
  62     /* allocation of the heap */
  63     capacity = pagealign(capacity ? capacity : 1);
  64     data = mmap(NULL, capacity, PROT_READ|PROT_WRITE,
  65                                     MAP_PRIVATE|MAP_ANONYMOUS, -1, 0);
  66
  67     /* success of the allocation? */
  68     if (data == MAP_FAILED)
  69         return -1;
  70
  71     /* yes. initialisae the heap */
  72     heap->data = data;
  73     heap->capacity = capacity;
  74     heap->size = 0;
  75     return 0;
  76 }
  77
  78 int heap_destroy( struct heap *heap)
  79 {
  80     return munmap( heap->data, heap->capacity);
  81 }
  82
  83 int heap_resize( struct heap *heap, size_t size)
  84 {
  85     /* has heap enought data? */
  86     if (size > heap->capacity) {
  87
  88         /* no. resizing of the heap. */
  89         /* compute the sizes and realloc */
  90         size_t capa = pagealign(size);
  91         char *data = mremap(heap->data, heap->capacity, capa, MREMAP_MAYMOVE);
  92
  93         /* error if failure */
  94         if (data == MAP_FAILED)
  95             return -1;
  96
  97         /* record new parameters. */
  98         heap->data = data;
  99         heap->capacity = capa;
 100     }
 101     heap->size = size;
 102
 103     return 0;
 104 }
 105
 106 size_t heap_alloc( struct heap *heap, size_t count)
 107 {
 108     size_t result = heap->size;
 109     if( heap_resize( heap, align( result + count)) != 0)
 110         result = HNULL;
 111     return result;
 112 }
 113
 114 size_t heap_strndup( struct heap *heap, const char *value, size_t length)
 115 {
 116     size_t offset;
 117     char *string;
 118
 119     offset = heap_alloc( heap, length+1);
 120     if (offset != HNULL) {
 121         string = heap_address( heap, offset);
 122         memcpy( string, value, length);
 123         string[length] = 0;
 124     }
 125     return offset;
 126 }
 127
 128 size_t heap_strdup( struct heap *heap, const char *string)
 129 {
 130     return heap_strndup( heap, string, strlen(string));
 131 }
 132
 133 int heap_read_write( struct heap *heap)
 134 {
 135     return mprotect(heap->data, heap->capacity, PROT_READ|PROT_WRITE);
 136 }
 137
 138 int heap_read_only( struct heap *heap)
 139 {
 140     return mprotect(heap->data, heap->capacity, PROT_READ);
 141 }
 142
 143
 144