Minor changes.
[yaz-moved-to-github.git] / util / wrbuf.c
1 /*
2  * Copyright (c) 1995, Index Data.
3  * See the file LICENSE for details.
4  * Sebastian Hammer, Adam Dickmeiss
5  *
6  * $Log: wrbuf.c,v $
7  * Revision 1.4  1998-02-11 11:53:36  adam
8  * Changed code so that it compiles as C++.
9  *
10  * Revision 1.3  1997/05/01 15:08:15  adam
11  * Added log_mask_str_x routine.
12  *
13  * Revision 1.2  1995/11/01 13:55:06  quinn
14  * Minor adjustments
15  *
16  * Revision 1.1  1995/10/06  08:51:25  quinn
17  * Added Write-buffer.
18  *
19  *
20  */
21
22 /*
23  * Growing buffer for writing various stuff.
24  */
25
26 #include <stdlib.h>
27 #include <string.h>
28
29 #include <wrbuf.h>
30
31 WRBUF wrbuf_alloc(void)
32 {
33     WRBUF n;
34
35     if (!(n = (WRBUF)xmalloc(sizeof(*n))))
36         abort();
37     n->buf = 0;
38     n->size = 0;
39     n->pos = 0;
40     return n;
41 }
42
43 void wrbuf_free(WRBUF b, int free_buf)
44 {
45     if (free_buf && b->buf)
46         xfree(b->buf);
47     xfree(b);
48 }
49
50 void wrbuf_rewind(WRBUF b)
51 {
52     b->pos = 0;
53 }
54
55 int wrbuf_grow(WRBUF b, int minsize)
56 {
57     int togrow;
58
59     if (!b->size)
60         togrow = 1024;
61     else
62         togrow = b->size;
63     if (togrow < minsize)
64         togrow = minsize;
65     if (b->size && !(b->buf =(char *)xrealloc(b->buf, b->size += togrow)))
66         abort();
67     else if (!b->size && !(b->buf = (char *)xmalloc(b->size = togrow)))
68         abort();
69     return 0;
70 }
71
72 int wrbuf_write(WRBUF b, char *buf, int size)
73 {
74     if (b->pos + size >= b->size)
75         wrbuf_grow(b, size);
76     memcpy(b->buf + b->pos, buf, size);
77     b->pos += size;
78     return 0;
79 }