X-Git-Url: https://git.kernel.dk/?p=fio.git;a=blobdiff_plain;f=fifo.c;h=5406aa572199a9c1f59756f3ecfd397c74a07e50;hp=1e55b6344ba1d4cbcb52dac65a6ddbb87d33a81d;hb=104bc4bdf55bd79c2b3f9087601c3df4aa884b2a;hpb=0db2679750e4cc8f38af7af135f1ba8481d6def6

diff --git a/fifo.c b/fifo.c
index 1e55b634..5406aa57 100644
--- a/fifo.c
+++ b/fifo.c
@@ -50,7 +50,7 @@ unsigned int fifo_put(struct fifo *fifo, void *buffer, unsigned int len)
 {
 	unsigned int l;
 
-	len = min(len, fifo->size - fifo->in + fifo->out);
+	len = min(len, fifo_room(fifo));
 
 	/* first put the data starting from fifo->in to buffer end */
 	l = min(len, fifo->size - (fifo->in & (fifo->size - 1)));
@@ -71,18 +71,27 @@ unsigned int fifo_put(struct fifo *fifo, void *buffer, unsigned int len)
 
 unsigned int fifo_get(struct fifo *fifo, void *buffer, unsigned int len)
 {
-	unsigned int l;
-
 	len = min(len, fifo->in - fifo->out);
 
-	/* first get the data from fifo->out until the end of the buffer */
-	l = min(len, fifo->size - (fifo->out & (fifo->size - 1)));
-	memcpy(buffer, fifo->buffer + (fifo->out & (fifo->size - 1)), l);
+	if (buffer) {
+		unsigned int l;
 
-	/* then get the rest (if any) from the beginning of the buffer */
-	memcpy(buffer + l, fifo->buffer, len - l);
+		/*
+		 * first get the data from fifo->out until the end of the buffer
+		 */
+		l = min(len, fifo->size - (fifo->out & (fifo->size - 1)));
+		memcpy(buffer, fifo->buffer + (fifo->out & (fifo->size - 1)),l);
+
+		/*
+		 * then get the rest (if any) from the beginning of the buffer
+		 */
+		memcpy(buffer + l, fifo->buffer, len - l);
+	}
 
 	fifo->out += len;
 
+	if (fifo->in == fifo->out)
+		fifo->in = fifo->out = 0;
+
 	return len;
 }