Fix reading broken BER in gcrypt/rsa.c
[tinc] / src / buffer.c
1 /*
2     buffer.c -- buffer management
3     Copyright (C) 2011 Guus Sliepen <guus@tinc-vpn.org>,
4
5     This program is free software; you can redistribute it and/or modify
6     it under the terms of the GNU General Public License as published by
7     the Free Software Foundation; either version 2 of the License, or
8     (at your option) any later version.
9
10     This program is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13     GNU General Public License for more details.
14
15     You should have received a copy of the GNU General Public License along
16     with this program; if not, write to the Free Software Foundation, Inc.,
17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18 */
19
20 #include "system.h"
21
22 #include "buffer.h"
23 #include "xalloc.h"
24
25 void buffer_compact(buffer_t *buffer, uint32_t maxsize) {
26         if(buffer->len >= maxsize || buffer->offset / 7 > buffer->len / 8) {
27                 memmove(buffer->data, buffer->data + buffer->offset, buffer->len - buffer->offset);
28                 buffer->len -= buffer->offset;
29                 buffer->offset = 0;
30         }
31 }
32
33 // Make sure we can add size bytes to the buffer, and return a pointer to the start of those bytes.
34
35 char *buffer_prepare(buffer_t *buffer, uint32_t size) {
36         if(!buffer->data) {
37                 buffer->maxlen = size;
38                 buffer->data = xmalloc(size);
39         } else {
40                 if(buffer->offset && buffer->len + size > buffer->maxlen) {
41                         memmove(buffer->data, buffer->data + buffer->offset, buffer->len - buffer->offset);
42                         buffer->len -= buffer->offset;
43                         buffer->offset = 0;
44                 }
45
46                 if(buffer->len + size > buffer->maxlen) {
47                         buffer->maxlen = buffer->len + size;
48                         buffer->data = xrealloc(buffer->data, buffer->maxlen);
49                 }
50         }
51
52         char *start = buffer->data + buffer->len;
53
54         buffer->len += size;
55
56         return start;
57 }
58
59 // Copy data into the buffer.
60
61 void buffer_add(buffer_t *buffer, const char *data, uint32_t size) {
62         memcpy(buffer_prepare(buffer, size), data, size);
63 }
64
65 // Remove given number of bytes from the buffer, return a pointer to the start of them.
66
67 static char *buffer_consume(buffer_t *buffer, uint32_t size) {
68         char *start = buffer->data + buffer->offset;
69
70         buffer->offset += size;
71
72         if(buffer->offset >= buffer->len) {
73                 buffer->offset = 0;
74                 buffer->len = 0;
75         }
76
77         return start;
78 }
79
80 // Check if there is a complete line in the buffer, and if so, return it NULL-terminated.
81
82 char *buffer_readline(buffer_t *buffer) {
83         char *newline = memchr(buffer->data + buffer->offset, '\n', buffer->len - buffer->offset);
84
85         if(!newline) {
86                 return NULL;
87         }
88
89         uint32_t len = newline + 1 - (buffer->data + buffer->offset);
90         *newline = 0;
91         return buffer_consume(buffer, len);
92 }
93
94 // Check if we have enough bytes in the buffer, and if so, return a pointer to the start of them.
95
96 char *buffer_read(buffer_t *buffer, uint32_t size) {
97         if(buffer->len - buffer->offset < size) {
98                 return NULL;
99         }
100
101         return buffer_consume(buffer, size);
102 }
103
104 void buffer_clear(buffer_t *buffer) {
105         free(buffer->data);
106         buffer->data = NULL;
107         buffer->maxlen = 0;
108         buffer->len = 0;
109         buffer->offset = 0;
110 }