Fix some broken fwrite calls

[blerg.git] / cgi / cgi_blerg.c

diff --git a/cgi/cgi_blerg.c b/cgi/cgi_blerg.c
index 5319442..6404fa3 100644 (file)
--- a/cgi/cgi_blerg.c
+++ b/cgi/cgi_blerg.c
@@ -6,40 +6,12 @@
 #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;
        unsigned int len;
@@ -60,13 +32,13 @@ void respond_for_range(struct blerg *b, uint64_t from, uint64_t to) {
        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);
-       write(0, ybuf, len);
+       fwrite(ybuf, len, 1, stdout);
        yajl_gen_free(g);
 }
 
@@ -90,7 +62,7 @@ void respond_taglist(struct tag *results, int i) {
                        blerg_close(b);
                }
                yajl_gen_get_buf(g, &ybuf, &len);
-               write(0, ybuf, len);
+               fwrite(ybuf, len, 1, stdout);
                yajl_gen_clear(g);
 
                i--;
@@ -98,7 +70,7 @@ void respond_taglist(struct tag *results, int 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);
 }
 
@@ -118,6 +90,10 @@ int main(int argc, char *argv[]) {
                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) {
@@ -197,6 +173,43 @@ int main(int argc, char *argv[]) {
                        respond_taglist(taglist, recs);
                }
        } 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();
@@ -236,7 +249,7 @@ int main(int argc, char *argv[]) {
 
                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(path, "/create", 8) == 0) {
@@ -263,6 +276,50 @@ int main(int argc, char *argv[]) {
                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);