From dd22e96445620de12d12f06a2264f7cb43abc646 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Petr=20P=C3=ADsa=C5=99?= <petr.pisar@atlas.cz>
Date: Thu, 4 Aug 2011 22:15:37 +0200
Subject: [PATCH] test: server: read a line

---
 test/simline/http.c | 86 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)

diff --git a/test/simline/http.c b/test/simline/http.c
index eb95872..61d0df7 100644
--- a/test/simline/http.c
+++ b/test/simline/http.c
@@ -1,10 +1,96 @@
 #include "http.h"
 #include <stdlib.h>
+#include <errno.h>
+#include <unistd.h>
+#include <string.h>
+#include <stdio.h> /* fprintf() */
+
+/* Read a line from HTTP socket.
+ * @socket is descriptor to read from.
+ * @line is auto-allocated just read line. Will be NULL if EOF has been
+ * reached.
+ * @buffer is automatically reallocated buffer for the socket. It can preserve
+ * prematurately read socket data.
+ * @buffer_size is allocated size of @buffer
+ * @buffer_length is size of head of the buffer that holds read data.
+ * @return 0 in success. */
+static int http_read_line(int socket, char **line,
+        char **buffer, size_t *buffer_size, size_t *buffer_used) {
+    ssize_t got;
+    char *p, *tmp;
+
+    if (line == NULL) return -1;
+    *line = NULL;
+
+    if (buffer == NULL || buffer_size == NULL || buffer_used == NULL)
+        return -1;
+    if (*buffer == NULL && *buffer_size > 0) return -1;
+    if (*buffer_size < *buffer_used) return -1;
+
+#define BURST 1024
+    while (1) {
+        if (*buffer_size == *buffer_used) {
+            /* Grow buffer */
+            tmp = realloc(*buffer, *buffer_size + BURST);
+            if (tmp == NULL) return -1;
+            *buffer = tmp;
+            *buffer_size += BURST;
+        }
+
+        /* Read data */
+        got = read(socket, *buffer + *buffer_used, *buffer_size - *buffer_used);
+        if (got == -1 && errno != EINTR) return -1;
+
+        /* Check for EOL */
+        for (p = *buffer + *buffer_used; p < *buffer + *buffer_used + got; p++)
+        {
+            if (*p != '\r')
+                continue;
+            if (!(p + 1 < *buffer + *buffer_used + got && p[1] == '\n'))
+                continue;
+
+            /* EOL found */
+            /* Crop by zero at EOL */
+            *p = '\0';
+            p += 2;
+            /* Copy read ahead data to new buffer and point line to original
+             * buffer. */
+            tmp = malloc(BURST);
+            if (tmp == NULL) return -1;
+            memcpy(tmp, p, *buffer + *buffer_used + got - p);
+            *line = *buffer;
+            *buffer_size = BURST;
+            *buffer_used = *buffer + *buffer_used + got - p;
+            *buffer = tmp;
+            /* And exit */
+            return 0;
+        }
+        
+        /* Chek for EOF */
+        if (got == 0) return -1;
+
+        /* Move end of buffer */
+        *buffer_used += got;
+    }
+#undef BURST
+
+    return -1;
+}
 
 /* Read a HTTP request from connected socket.
  * @request is automatically allocated received HTTP request
  * @return 0 in case of success */
 int http_read_request(int socket, struct http_request *request) {
+    char *line = NULL;
+    char *buffer = NULL;
+    size_t buffer_size = 0, buffer_used = 0;
+
+    if (-1 == http_read_line(socket, &line, &buffer, &buffer_size,
+                &buffer_used))
+        return -1;
+    fprintf(stderr, "Request: <%s>\n", line);
+    free(line);
+    free(buffer);
     return -1;
 }
 
-- 
2.11.4.GIT