2bd75ee14111b2a6431f69a19c7cb90019f0d1b3
[yaz-moved-to-github.git] / 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 }