X-Git-Url: https://www.tinc-vpn.org/git/browse?a=blobdiff_plain;f=src%2Fbuffer.c;h=a9e79aa95a1ac39a54b0dc6fcc8cb9f4399692a4;hb=6c0584c55b99dd9814fed5c13536d831b3e5317e;hp=fc4fac0f3629570f3097d65c54337db857b56f8e;hpb=f431fcb35f400be388a905ae0f7f50c1f5c4cd5d;p=tinc

diff --git a/src/buffer.c b/src/buffer.c
index fc4fac0f..a9e79aa9 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;
@@ -73,10 +82,11 @@ static char *buffer_consume(buffer_t *buffer, int size) {
 char *buffer_readline(buffer_t *buffer) {
 	char *newline = memchr(buffer->data + buffer->offset, '\n', buffer->len - buffer->offset);
 
-	if(!newline)
+	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);
 }
@@ -84,15 +94,16 @@ char *buffer_readline(buffer_t *buffer) {
 // Check if we have enough bytes in the buffer, and if so, return a pointer to the start of them.
 
 char *buffer_read(buffer_t *buffer, int size) {
-	if(buffer->len - buffer->offset < size)
+	if(buffer->len - buffer->offset < size) {
 		return NULL;
+	}
 
 	return buffer_consume(buffer, 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;