src/odr_mem.c

   1 /*
   2  * Copyright (C) 1995-2005, Index Data ApS
   3  * See the file LICENSE for details.
   4  *
   5  * $Id: odr_mem.c,v 1.4 2005-01-15 19:47:14 adam Exp $
   6  */
   7 /**
   8  * \file odr_mem.c
   9  * \brief Implements ODR memory management
  10  */
  11 #if HAVE_CONFIG_H
  12 #include <config.h>
  13 #endif
  14
  15 #include <stdlib.h>
  16 #include "odr-priv.h"
  17 #include <yaz/xmalloc.h>
  18
  19 /* ------------------------ NIBBLE MEMORY ---------------------- */
  20
  21 /*
  22  * Extract the memory control block from o.
  23  */
  24 NMEM odr_extract_mem(ODR o)
  25 {
  26     NMEM r = o->mem;
  27
  28     o->mem = 0;
  29     return r;
  30 }
  31
  32 void *odr_malloc(ODR o, int size)
  33 {
  34     if (o && !o->mem)
  35         o->mem = nmem_create();
  36     return nmem_malloc(o ? o->mem : 0, size);
  37 }
  38
  39 char *odr_strdup(ODR o, const char *str)
  40 {
  41     if (o && !o->mem)
  42         o->mem = nmem_create();
  43     return nmem_strdup(o->mem, str);
  44 }
  45
  46 char *odr_strdupn(ODR o, const char *str, size_t n)
  47 {
  48     return nmem_strdupn(o->mem, str, n);
  49 }
  50
  51 int *odr_intdup(ODR o, int v)
  52 {
  53     if (o && !o->mem)
  54         o->mem = nmem_create();
  55     return nmem_intdup(o->mem, v);
  56 }
  57
  58 int odr_total(ODR o)
  59 {
  60     return o->mem ? nmem_total(o->mem) : 0;
  61 }
  62
  63 /* ---------- memory management for data encoding ----------*/
  64
  65
  66 int odr_grow_block(ODR b, int min_bytes)
  67 {
  68     int togrow;
  69
  70     if (!b->can_grow)
  71         return -1;
  72     if (!b->size)
  73         togrow = 1024;
  74     else
  75         togrow = b->size;
  76     if (togrow < min_bytes)
  77         togrow = min_bytes;
  78     if (b->size && !(b->buf =
  79                      (unsigned char *) xrealloc(b->buf, b->size += togrow)))
  80         abort();
  81     else if (!b->size && !(b->buf = (unsigned char *)
  82                            xmalloc(b->size = togrow)))
  83         abort();
  84 #ifdef ODR_DEBUG
  85     fprintf(stderr, "New size for encode_buffer: %d\n", b->size);
  86 #endif
  87     return 0;
  88 }
  89
  90 int odr_write(ODR o, unsigned char *buf, int bytes)
  91 {
  92     if (o->pos + bytes >= o->size && odr_grow_block(o, bytes))
  93     {
  94         odr_seterror(o, OSPACE, 40);
  95         return -1;
  96     }
  97     memcpy(o->buf + o->pos, buf, bytes);
  98     o->pos += bytes;
  99     if (o->pos > o->top)
 100         o->top = o->pos;
 101     return 0;
 102 }
 103
 104 int odr_seek(ODR o, int whence, int offset)
 105 {
 106     if (whence == ODR_S_CUR)
 107         offset += o->pos;
 108     else if (whence == ODR_S_END)
 109         offset += o->top;
 110     if (offset > o->size && odr_grow_block(o, offset - o->size))
 111     {
 112         odr_seterror(o, OSPACE, 41);
 113         return -1;
 114     }
 115     o->pos = offset;
 116     return 0;
 117 }