odr/odr_mem.c

   1 /*
   2  * Copyright (c) 1995, Index Data
   3  * See the file LICENSE for details.
   4  * Sebastian Hammer, Adam Dickmeiss
   5  *
   6  * $Log: odr_mem.c,v $
   7  * Revision 1.8  1995-09-29 17:12:24  quinn
   8  * Smallish
   9  *
  10  * Revision 1.7  1995/09/27  15:02:59  quinn
  11  * Modified function heads & prototypes.
  12  *
  13  * Revision 1.6  1995/08/21  09:10:41  quinn
  14  * Smallish fixes to suppport new formats.
  15  *
  16  * Revision 1.5  1995/05/16  08:50:55  quinn
  17  * License, documentation, and memory fixes
  18  *
  19  * Revision 1.4  1995/05/15  11:56:09  quinn
  20  * More work on memory management.
  21  *
  22  * Revision 1.3  1995/04/18  08:15:21  quinn
  23  * Added dynamic memory allocation on encoding (whew). Code is now somewhat
  24  * neater. We'll make the same change for decoding one day.
  25  *
  26  * Revision 1.2  1995/03/17  10:17:52  quinn
  27  * Added memory management.
  28  *
  29  * Revision 1.1  1995/03/14  10:27:40  quinn
  30  * Modified makefile to use common lib
  31  * Beginning to add memory management to odr
  32  *
  33  */
  34
  35 #include <stdlib.h>
  36 #include <odr.h>
  37 #include <dmalloc.h>
  38
  39 /* ------------------------ NIBBLE MEMORY ---------------------- */
  40
  41 #define ODR_MEM_CHUNK (10*1024)
  42
  43 typedef struct odr_memblock
  44 {
  45     char *buf;
  46     int size;
  47     int top;
  48     int total;
  49     struct odr_memblock *next;
  50 } odr_memblock;
  51
  52 static odr_memblock *freelist = 0; /* global freelist */
  53
  54 static void free_block(odr_memblock *p)
  55 {
  56     p->next = freelist;
  57     freelist = p;
  58 }
  59
  60 /*
  61  * acquire a block with a minimum of size free bytes.
  62  */
  63 static odr_memblock *get_block(int size)
  64 {
  65     odr_memblock *r, *l;
  66
  67     for (r = freelist, l = 0; r; l = r, r = r->next)
  68         if (r->size >= size)
  69             break;
  70     if (r)
  71         if (l)
  72             l->next = r->next;
  73         else
  74             freelist = r->next;
  75     else
  76     {
  77         int get = ODR_MEM_CHUNK;
  78
  79         if (get < size)
  80             get = size;
  81         if (!(r = malloc(sizeof(*r))))
  82             abort();
  83         if (!(r->buf = malloc(r->size = get)))
  84             abort();
  85     }
  86     r->top = 0;
  87     r->total = 0;
  88     return r;
  89 }
  90
  91 /*
  92  * Return p to the global freelist.
  93  */
  94 void odr_release_mem(ODR_MEM p)
  95 {
  96     odr_memblock *t;
  97
  98     while (p)
  99     {
 100         t = p;
 101         p = p->next;
 102         free_block(t);
 103     }
 104 }
 105
 106 /*
 107  * Extract the memory control block from o.
 108  */
 109 ODR_MEM odr_extract_mem(ODR o)
 110 {
 111     ODR_MEM r = o->mem;
 112
 113     o->mem = 0;
 114     return r;
 115 }
 116
 117 void *odr_malloc(ODR o, int size)
 118 {
 119     struct odr_memblock *p = o->mem;
 120     char *r;
 121
 122     if (!p || p->size - p->top < size)
 123         if (!(p = get_block(size)))
 124             abort();
 125         else
 126         {
 127             if (o->mem)
 128                 p->total = o->mem->total;
 129             p->next = o->mem;
 130             o->mem = p;
 131         }
 132     r = p->buf + p->top;
 133     /* align size */
 134     p->top += (size + (sizeof(long) - 1)) & ~(sizeof(long) - 1);
 135     p->total += size;
 136     return r;
 137 }
 138
 139 int odr_total(ODR o)
 140 {
 141     return o->mem ? o->mem->total : 0;
 142 }
 143
 144 /* ---------- memory management for data encoding ----------*/
 145
 146
 147 int odr_grow_block(odr_ecblock *b, int min_bytes)
 148 {
 149     int togrow;
 150
 151     if (!b->can_grow)
 152         return -1;
 153     if (!b->size)
 154         togrow = 1024;
 155     else
 156         togrow = b->size;
 157     if (togrow < min_bytes)
 158         togrow = min_bytes;
 159     if (b->size && !(b->buf = realloc(b->buf, b->size += togrow)))
 160         abort();
 161     else if (!b->size && !(b->buf = malloc(b->size = togrow)))
 162         abort();
 163 #ifdef ODR_DEBUG
 164     fprintf(stderr, "New size for encode_buffer: %d\n", b->size);
 165 #endif
 166     return 0;
 167 }
 168
 169 int odr_write(ODR o, unsigned char *buf, int bytes)
 170 {
 171     if (o->ecb.pos + bytes >= o->ecb.size && odr_grow_block(&o->ecb, bytes))
 172     {
 173         o->error = OSPACE;
 174         return -1;
 175     }
 176     memcpy(o->ecb.buf + o->ecb.pos, buf, bytes);
 177     o->ecb.pos += bytes;
 178     if (o->ecb.pos > o->ecb.top)
 179         o->ecb.top = o->ecb.pos;
 180     return 0;
 181 }
 182
 183 int odr_seek(ODR o, int whence, int offset)
 184 {
 185     if (whence == ODR_S_CUR)
 186         offset += o->ecb.pos;
 187     else if (whence == ODR_S_END)
 188         offset += o->ecb.top;
 189     if (offset > o->ecb.size && odr_grow_block(&o->ecb, offset - o->ecb.size))
 190     {
 191         o->error = OSPACE;
 192         return -1;
 193     }
 194     o->ecb.pos = offset;
 195     return 0;
 196 }