odr/odr_mem.c

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