+/* Blerg is (C) 2011 The Dominion of Awesome, and is distributed under a
+ * BSD-style license. Please see the COPYING file for details.
+ */
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include "database.h"
#include "tags.h"
#include "auth.h"
+#include "canned_responses.h"
#include "app.h"
#include "config.h"
yajl_gen_config yajl_c = { 0, 0 };
-void respond_simple_data(unsigned const char *data, int len) {
- printf("Content-length: %d\r\n\r\n", len);
- write(0, data, len);
-}
-
-void respond_404() {
- printf("Status: 404 Not Found\r\n");
- printf("Content-type: text/html\r\n");
- printf("Content-length: %d\r\n\r\n", strlen(CONTENT_404));
-
- printf(CONTENT_404);
-}
-
-void respond_405() {
- printf("Status: 405 Method Not Allowed\r\n");
- printf("Content-type: text/html\r\n");
- printf("Content-length: %d\r\n\r\n", strlen(CONTENT_405));
-
- printf(CONTENT_405);
-}
-
-void respond_JSON_Failure() {
- respond_simple_data(JSON_FAILURE, strlen(JSON_FAILURE));
-}
-
-void respond_JSON_Success() {
- respond_simple_data(JSON_SUCCESS, strlen(JSON_SUCCESS));
-}
-
void respond_for_range(struct blerg *b, uint64_t from, uint64_t to) {
const unsigned char *ybuf;
- char number[21];
unsigned int len;
uint64_t i;
yajl_gen g;
for (i = to; i != from - 1; i--) {
json_generate_one_record(g, NULL, b, i);
yajl_gen_get_buf(g, &ybuf, &len);
- write(0, ybuf, len);
+ fwrite(ybuf, len, 1, stdout);
+ yajl_gen_clear(g);
+ }
+
+ yajl_gen_array_close(g);
+ yajl_gen_get_buf(g, &ybuf, &len);
+ fwrite(ybuf, len, 1, stdout);
+ yajl_gen_free(g);
+}
+
+void respond_taglist(struct tag *results, int i) {
+ const unsigned char *ybuf;
+ unsigned int len;
+ struct blerg *b;
+ yajl_gen g;
+
+ i--;
+
+ printf("Content-type: application/json\r\n\r\n");
+ g = yajl_gen_alloc(&yajl_c, NULL);
+
+ yajl_gen_array_open(g);
+
+ while (i >= 0) {
+ b = blerg_open(results[i].author);
+ if (b != NULL) {
+ json_generate_one_record(g, results[i].author, b, results[i].record);
+ blerg_close(b);
+ }
+ yajl_gen_get_buf(g, &ybuf, &len);
+ fwrite(ybuf, len, 1, stdout);
yajl_gen_clear(g);
+
+ i--;
}
yajl_gen_array_close(g);
yajl_gen_get_buf(g, &ybuf, &len);
- write(0, ybuf, len);
+ fwrite(ybuf, len, 1, stdout);
yajl_gen_free(g);
}
exit(0);
}
request_method = getenv("REQUEST_METHOD");
+ if (request_method == NULL) {
+ fprintf(stderr, "Request method is null!?\n");
+ exit(0);
+ }
if (strncmp(path, "/get", 4) == 0 && strlen(path) > 4) {
if (strncmp(request_method, "GET", 4) != 0) {
}
blerg_close(b);
- } else if (strncmp(path, "/tag", 4) == 0 && strlen(url) > 4) {
+ } else if (strncmp(path, "/tag", 4) == 0 && strlen(path) > 4) {
if (strcmp(request_method, "GET") != 0) {
respond_405();
exit(0);
exit(0);
}
+ if (info.author[0] == 'H')
+ info.author[0] = '#';
if (!tag_exists(info.author)) {
respond_404();
exit(0);
}
- uint64_t recs = 50;
+ int recs = 50;
struct tag *taglist = tag_list(info.author, 0, &recs, -1);
if (recs == 0) {
} else {
respond_taglist(taglist, recs);
}
- } else if (strncmp(url, "/put", 4) == 0) {
+ } else if (strncmp(path, "/put", 4) == 0) {
+ if (strcmp(request_method, "POST") != 0) {
+ respond_405();
+ exit(0);
+ }
+
+ if (path[4] == '/') {
+ respond_404();
+ exit(0);
+ }
+
+ const char *username = cgi_getentrystr("username");
+ const char *data = cgi_getentrystr("data");
+ if (username == NULL || username[0] == 0 ||
+ data == NULL || data[0] == 0) {
+ respond_JSON_Failure();
+ exit(0);
+ }
+
+ const char *given_token = cgi_getcookie("auth");
+ if (!auth_check_token(username, given_token)) {
+ respond_JSON_Failure();
+ exit(0);
+ }
+
+ struct blerg *b = blerg_open(username);
+ if (b == NULL) {
+ respond_JSON_Failure();
+ exit(0);
+ }
+ ret = blerg_store(b, data, strlen(data));
+ blerg_close(b);
+ if (ret == -1) {
+ respond_JSON_Failure();
+ exit(0);
+ }
+
+ respond_JSON_Success();
} else if (strncmp(path, "/info", 5) == 0) {
if (strcmp(request_method, "GET") != 0) {
respond_405();
printf("Content-type: application/json\r\n");
printf("Content-length: %d\r\n\r\n", len);
- write(0, ybuf, len);
+ fwrite(ybuf, len, 1, stdout);
yajl_gen_free(g);
- } else if (strncmp(url, "/create", 8) == 0) {
+ } else if (strncmp(path, "/create", 8) == 0) {
+ if (strcmp(request_method, "POST") != 0) {
+ respond_405();
+ exit(0);
+ }
+
+ const char *username = cgi_getentrystr("username");
+ const char *password = cgi_getentrystr("password");
+ if (username == NULL || username[0] == 0 ||
+ password == NULL || password[0] == 0) {
+ respond_JSON_Failure();
+ exit(0);
+ }
+
+ if (blerg_exists(username)) {
+ respond_JSON_Failure();
+ exit(0);
+ }
+
+ struct blerg *b = blerg_open(username);
+ blerg_close(b);
+ auth_set_password(username, password);
+
+ respond_JSON_Success();
+ } else if (strncmp(path, "/login", 7) == 0) {
+ if (strcmp(request_method, "POST") != 0) {
+ respond_405();
+ exit(0);
+ }
+
+ const char *username = cgi_getentrystr("username");
+ const char *password = cgi_getentrystr("password");
+ if (username == NULL || username[0] == 0 ||
+ password == NULL || password[0] == 0) {
+ respond_JSON_Failure();
+ exit(0);
+ }
+
+ if (!auth_login(username, password)) {
+ respond_JSON_Failure();
+ exit(0);
+ }
+
+ char *token = auth_get_token(username);
+ printf("Set-Cookie: auth=%s\r\n", token);
+ free(token);
+
+ respond_JSON_Success();
+ } else if (strncmp(path, "/logout", 8) == 0) {
+ if (strcmp(request_method, "POST") != 0) {
+ respond_405();
+ exit(0);
+ }
+
+ const char *username = cgi_getentrystr("username");
+ if (username == NULL || username[0] == 0) {
+ respond_JSON_Failure();
+ exit(0);
+ }
+
+
+ const char *given_token = cgi_getcookie("auth");
+ if (auth_check_token(username, given_token)) {
+ auth_logout(username);
+ respond_JSON_Success();
+ } else {
+ respond_JSON_Failure();
+ }
} else {
respond_404();
exit(0);