X-Git-Url: http://git.bytex64.net/?a=blobdiff_plain;f=http%2Fhttp_blerg.c;fp=http%2Fhttp_blerg.c;h=a6d3a5fea02cc57f9f532fb5b2bb9e9e60a8b519;hb=bfc01c15f56c58cefc8680f7faed4c5e3650fa38;hp=025fbb01bc3c4bcd1b9a5a5563c43d8856e49452;hpb=d119f69edc7c2a9ad2ded2be7fe71eae89926f92;p=blerg.git

diff --git a/http/http_blerg.c b/http/http_blerg.c
index 025fbb0..a6d3a5f 100644
--- a/http/http_blerg.c
+++ b/http/http_blerg.c
@@ -28,6 +28,12 @@ struct put_state {
 	int data_size;
 };
 
+struct subscribe_state {
+	struct MHD_PostProcessor *pp;
+	char username[33];
+	char to[33];
+};
+
 struct get_state {
 	struct blerg *b;
 	yajl_gen g;
@@ -209,6 +215,22 @@ int POST_put_iterator(void *cls, enum MHD_ValueKind kind, const char *key, const
 	return MHD_YES;
 }
 
+int POST_subscribe_iterator(void *cls, enum MHD_ValueKind kind, const char *key, const char *filename, const char *content_type, const char *transfer_encoding, const char *data, uint64_t off, size_t size) {
+	struct subscribe_state *ss = cls;
+
+	if (strncmp(key, "username", 9) == 0) {
+		if (size > 32) size = 32;
+		memcpy(ss->username, data, size);
+		ss->username[size] = 0;
+	} else if (strncmp(key, "to", 3) == 0) {
+		if (size > 32) size = 32;
+		memcpy(ss->to, data, size);
+		ss->to[size] = 0;
+	}
+
+	return MHD_YES;
+}
+
 struct MHD_Response *create_response_for_range(struct blerg *b, uint64_t from, uint64_t to) {
 	struct MHD_Response *response;
 	struct get_state *gs = malloc(sizeof(struct get_state));
@@ -535,6 +557,33 @@ ahc_derp (void *cls, struct MHD_Connection *connection, const char *url, const c
 		} else {
 			return respond_JSON_Failure(connection);
 		}
+	} else if (strncmp(url, "/subscribe", 11) == 0) {
+		struct subscribe_state *ss = (struct subscribe_state *) *ptr;
+
+		if (ss == NULL) {
+			if (strcmp(method, MHD_HTTP_METHOD_POST) != 0)
+				return respond_405(connection);
+
+			struct subscribe_state *ss = malloc(sizeof(struct subscribe_state));
+			ss->username[0] = ss->to[0] = 0;
+			ss->pp = MHD_create_post_processor(connection, 1024, &POST_subscribe_iterator, ss);
+			*ptr = ss;
+			return MHD_YES;
+		}
+
+		if (*upload_data_size) {
+			MHD_post_process(ss->pp, upload_data, *upload_data_size);
+			*upload_data_size = 0;
+			return MHD_YES;
+		}
+
+		const char *given_token = MHD_lookup_connection_value(connection, MHD_COOKIE_KIND, "auth");
+		if (auth_check_token(ss->username, given_token)) {
+			subscription_add(ss->username, ss->to);
+			return respond_JSON_Success(connection);
+		} else {
+			return respond_JSON_Failure(connection);
+		}
 	} else {
 		return respond_404(connection);
 	}