3 files changed, 116 insertions, 2 deletions

diff --git a/scrape.c b/scrape.c
index 33e22fc..57e5c6c 100644
--- a/scrape.c
+++ b/scrape.c
@@ -1,9 +1,11 @@
 #define _POSIX_C_SOURCE 200809L
 
+#include <ctype.h>
 #include <netdb.h>
 #include <netinet/in.h>
 #include <poll.h>
 #include <stdio.h>
+#include <string.h>
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <unistd.h>
@@ -22,6 +24,8 @@ struct scrape_req {
   cbuf *buf;
 };
 
+static bool handle_http(struct scrape_req *req);
+
 bool scrape_serve(const char *port, scrape_handler *handler, void *handler_ctx) {
   struct scrape_req req;
 
@@ -112,8 +116,8 @@ bool scrape_serve(const char *port, scrape_handler *handler, void *handler_ctx)
         continue;
       }
 
-      // TODO: HTTP stuff
-      handler(&req, handler_ctx);
+      if (handle_http(&req))
+        handler(&req, handler_ctx);
       close(req.socket);
     }
   }
@@ -150,3 +154,103 @@ void scrape_write(scrape_req *req, const char *metric, const char *(*labels)[2],
 void scrape_write_raw(scrape_req *req, const void *buf, size_t len) {
   write_all(req->socket, buf, len);
 }
+
+static const char http_success[] =
+    "HTTP/1.1 200 OK\r\n"
+    "Server: prometheus-nano-exporter\r\n"
+    "Content-Type: text/plain; charset=UTF-8\r\n"
+    "Connection: close\r\n"
+    "\r\n"
+    ;
+static const char http_error[] =
+    "HTTP/1.1 400 Bad Request\r\n"
+    "Server: prometheus-nano-exporter\r\n"
+    "Content-Type: text/plain; charset=UTF-8\r\n"
+    "Connection: close\r\n"
+    "\r\n"
+    "This is not a general-purpose HTTP server.\r\n"
+    ;
+
+static bool handle_http(struct scrape_req *req) {
+  unsigned char http_buf[1024];
+  cbuf *tmp_buf = req->buf;
+
+  enum {
+    state_read_method,
+    state_read_path,
+    state_read_version,
+    state_skip_headers_1,
+    state_skip_headers_2
+  } state = state_read_method;
+  cbuf_reset(tmp_buf);
+
+  while (true) {
+    ssize_t got = read(req->socket, &http_buf, sizeof http_buf);
+    if (got <= 0)
+      return false;
+    for (ssize_t i = 0; i < got; i++) {
+      int c = http_buf[i];
+      if (c == '\r')
+        continue;
+
+      switch (state) {
+        case state_read_method:
+          if (c == ' ') {
+            if (cbuf_cmp(tmp_buf, "GET") != 0)
+              goto fail;
+            state = state_read_path;
+            cbuf_reset(tmp_buf);
+            break;
+          }
+          if (!isalnum(c) || cbuf_len(tmp_buf) >= 16)
+            goto fail;
+          cbuf_putc(tmp_buf, c);
+          break;
+
+        case state_read_path:
+          if (c == ' ') {
+            if (cbuf_cmp(tmp_buf, "/metrics") != 0)
+              goto fail;
+            state = state_read_version;
+            cbuf_reset(tmp_buf);
+            break;
+          }
+          if (!isprint(c) || c == '\n' || cbuf_len(tmp_buf) >= 128)
+            goto fail;
+          cbuf_putc(tmp_buf, c);
+          break;
+
+        case state_read_version:
+          if (c == '\n') {
+            if (cbuf_cmp(tmp_buf, "HTTP/1.1") != 0)
+              goto fail;
+            state = state_skip_headers_1;
+            cbuf_reset(tmp_buf);
+            break;
+          }
+          if (!isgraph(c) || cbuf_len(tmp_buf) >= 16)
+            goto fail;
+          cbuf_putc(tmp_buf, c);
+          break;
+
+        case state_skip_headers_1:
+          if (c == '\n')
+            goto succeed;
+          state = state_skip_headers_2;
+          break;
+        case state_skip_headers_2:
+          if (c == '\n')
+            state = state_skip_headers_1;
+          break;
+      }
+    }
+  }
+
+succeed:
+  write_all(req->socket, http_success, sizeof http_success - 1);
+  return true;
+
+fail:
+  write_all(req->socket, http_error, sizeof http_error - 1);
+  return false;
+}
diff --git a/util.c b/util.c
index 6476f5c..e88ef95 100644
--- a/util.c
+++ b/util.c
@@ -48,6 +48,10 @@ void cbuf_reset(cbuf *buf) {
   buf->len = 0;
 }
 
+size_t cbuf_len(cbuf *buf) {
+  return buf->len;
+}
+
 void cbuf_put(cbuf *buf, const void *src, size_t len) {
   if (!cbuf_reserve(buf, len))
     return;
@@ -88,6 +92,10 @@ const char *cbuf_get(struct cbuf *buf, size_t *len) {
   return buf->data;
 }
 
+int cbuf_cmp(cbuf *buf, const char *other) {
+  return strncmp(buf->data, other, buf->len);
+}
+
 // string lists
 
 struct slist *slist_split(const char *str, const char *delim) {
diff --git a/util.h b/util.h
index 3fa5fd0..70cd474 100644
--- a/util.h
+++ b/util.h
@@ -11,11 +11,13 @@ typedef struct cbuf cbuf;
 
 cbuf *cbuf_alloc(size_t initial_size, size_t max_size);
 void cbuf_reset(cbuf *buf);
+size_t cbuf_len(cbuf *buf);
 void cbuf_put(cbuf *buf, const void *src, size_t len);
 void cbuf_puts(cbuf *buf, const char *src);
 void cbuf_putc(cbuf *buf, int c);
 void cbuf_putf(cbuf *buf, const char *fmt, ...);
 const char *cbuf_get(struct cbuf *buf, size_t *len);
+int cbuf_cmp(cbuf *buf, const char *other);
 
 // string lists