X-Git-Url: https://tinc-vpn.org/git/browse?p=tinc;a=blobdiff_plain;f=src%2Fbuffer.c;h=ac57d1c459e1e97790285425fb38b436cea27a28;hp=fc4fac0f3629570f3097d65c54337db857b56f8e;hb=390d25f0b80dd7418e147de3561c70461628574d;hpb=f431fcb35f400be388a905ae0f7f50c1f5c4cd5d diff --git a/src/buffer.c b/src/buffer.c index fc4fac0f..ac57d1c4 100644 --- a/src/buffer.c +++ b/src/buffer.c @@ -22,6 +22,14 @@ #include "buffer.h" #include "xalloc.h" +void buffer_compact(buffer_t *buffer, int maxsize) { + if(buffer->len >= maxsize || buffer->offset / 7 > buffer->len / 8) { + memmove(buffer->data, buffer->data + buffer->offset, buffer->len - buffer->offset); + buffer->len -= buffer->offset; + buffer->offset = 0; + } +} + // Make sure we can add size bytes to the buffer, and return a pointer to the start of those bytes. char *buffer_prepare(buffer_t *buffer, int size) { @@ -41,13 +49,16 @@ char *buffer_prepare(buffer_t *buffer, int size) { } } + char *start = buffer->data + buffer->len; + buffer->len += size; - return buffer->data + buffer->offset; + + return start; } // Copy data into the buffer. - -char *buffer_add(buffer_t *buffer, const char *data, int size) { + +void buffer_add(buffer_t *buffer, const char *data, int size) { memcpy(buffer_prepare(buffer, size), data, size); } @@ -61,8 +72,6 @@ static char *buffer_consume(buffer_t *buffer, int size) { if(buffer->offset >= buffer->len) { buffer->offset = 0; buffer->len = 0; - } else { - buffer->offset += size; } return start; @@ -76,7 +85,7 @@ char *buffer_readline(buffer_t *buffer) { if(!newline) return NULL; - int len = newline + 1 - buffer->data + buffer->offset; + int len = newline + 1 - (buffer->data + buffer->offset); *newline = 0; return buffer_consume(buffer, len); } @@ -92,7 +101,7 @@ char *buffer_read(buffer_t *buffer, int size) { void buffer_clear(buffer_t *buffer) { free(buffer->data); - buffer->data = 0; + buffer->data = NULL; buffer->maxlen = 0; buffer->len = 0; buffer->offset = 0;