X-Git-Url: http://git.bytex64.net/?a=blobdiff_plain;f=cgi%2Fcgi_blerg.c;h=abf1237d22d6847ec2887c4d098cd30b4ff9d95a;hb=ef8396bf39d3be8e171bade8e50b7587e1a8b587;hp=c004a7d05a87ac9fca2ed9fe5e2ce9f89c7eabdb;hpb=ec8746b44dc85fd3e3b42835f779890684a9e90a;p=blerg.git diff --git a/cgi/cgi_blerg.c b/cgi/cgi_blerg.c index c004a7d..abf1237 100644 --- a/cgi/cgi_blerg.c +++ b/cgi/cgi_blerg.c @@ -9,12 +9,29 @@ #include "database.h" #include "tags.h" #include "auth.h" +#include "subscription.h" +#include "json.h" #include "canned_responses.h" #include "app.h" #include "config.h" yajl_gen_config yajl_c = { 0, 0 }; +int check_auth(const char *username) { + if (username == NULL || username[0] == 0) { + respond_403(); + return 0; + } + + const char *given_token = cgi_getcookie("auth"); + if (!auth_check_token(username, given_token)) { + respond_403(); + return 0; + } + return 1; +} + + void respond_for_range(struct blerg *b, uint64_t from, uint64_t to) { const unsigned char *ybuf; unsigned int len; @@ -84,6 +101,9 @@ int main(int argc, char *argv[]) { struct url_info info; char *data; + if (!blerg_init()) + exit(1); + if (cgi_init() != CGIERR_NONE) exit(0); @@ -183,21 +203,17 @@ int main(int argc, char *argv[]) { exit(0); } + const char *username = cgi_getentrystr("username"); + if (!check_auth(username)) + 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)) { + if (data == NULL || data[0] == 0) { respond_JSON_Failure(); exit(0); } @@ -244,17 +260,18 @@ int main(int argc, char *argv[]) { char number[21]; yajl_gen g = yajl_gen_alloc(&yajl_c, NULL); yajl_gen_map_open(g); - yajl_gen_string(g, "record_count", 12); + yajl_gen_string(g, (unsigned char *)"record_count", 12); snprintf(number, 21, "%llu", record_count); - yajl_gen_string(g, number, strlen(number)); + yajl_gen_string(g, (unsigned char *)number, strlen(number)); yajl_gen_map_close(g); const unsigned char *ybuf; - yajl_gen_get_buf(g, &ybuf, &len); + unsigned int content_len; + yajl_gen_get_buf(g, &ybuf, &content_len); printf("Content-type: application/json\r\n"); - printf("Content-length: %d\r\n\r\n", len); - fwrite(ybuf, len, 1, stdout); + printf("Content-length: %d\r\n\r\n", content_len); + fwrite(ybuf, content_len, 1, stdout); yajl_gen_free(g); } else if (strncmp(path, "/create", 8) == 0) { @@ -295,12 +312,12 @@ int main(int argc, char *argv[]) { exit(0); } - if (!auth_login(username, password)) { + char *token = auth_login(username, password); + if (token == NULL) { respond_JSON_Failure(); exit(0); } - char *token = auth_get_token(username); printf("Set-Cookie: auth=%s\r\n", token); free(token); @@ -312,18 +329,112 @@ int main(int argc, char *argv[]) { } const char *username = cgi_getentrystr("username"); - if (username == NULL || username[0] == 0) { - respond_JSON_Failure(); + if (!check_auth(username)) exit(0); + + const char *given_token = cgi_getcookie("auth"); + auth_logout(username, given_token); + respond_JSON_Success(); + } else if (strncmp(path, "/subscribe", 10) == 0 || strncmp(path, "/unsubscribe", 12) == 0) { + const char *username = cgi_getentrystr("username"); + if (!check_auth(username)) + exit(0); + + if (path[1] == 'u') { + if (path[12] != '/') { + respond_404(); + exit(0); + } + + ret = parse_url_info(path + 13, &info); + if ((ret & URL_INFO_NAME) == 0) { + respond_404(); + exit(0); + } + + subscription_remove(username, info.name); + } else { + if (path[10] != '/') { + respond_404(); + exit(0); + } + + ret = parse_url_info(path + 11, &info); + if ((ret & URL_INFO_NAME) == 0) { + respond_404(); + exit(0); + } + + subscription_add(username, info.name); } + respond_JSON_Success(); + } else if (strncmp(path, "/feed", 6) == 0) { + const char *username = cgi_getentrystr("username"); + if (!check_auth(username)) + exit(0); + int recs = 50; + struct blergref *feedlist = subscription_list(username, 0, &recs, -1); - const char *given_token = cgi_getcookie("auth"); - if (auth_check_token(username, given_token)) { - auth_logout(username); - respond_JSON_Success(); + if (recs == 0) { + respond_simple_data("[]", 2); } else { + respond_blergref_list(feedlist, recs); + } + + struct blerg *b = blerg_open(username); + blerg_set_subscription_mark(b); + blerg_close(b); + } else if (strncmp(path, "/feedinfo", 9) == 0) { + const char *username = cgi_getentrystr("username"); + if (!check_auth(username)) + exit(0); + + yajl_gen g = yajl_gen_alloc(&yajl_c, NULL); + yajl_gen_map_open(g); + if (path[9] == 0) { + struct blerg *b = blerg_open(username); + uint64_t subscription_mark = blerg_get_subscription_mark(b); + blerg_close(b); + + yajl_gen_string(g, (unsigned char *)"new", 3); + yajl_gen_integer(g, subscription_count_items(username) - subscription_mark); + } else { + yajl_gen_string(g, (unsigned char *)"subscribed", 10); + ret = parse_url_info(path + 10, &info); + if ((ret & URL_INFO_NAME) == 1) { + yajl_gen_bool(g, is_subscribed(username, info.name)); + } else { + yajl_gen_bool(g, 0); + } + } + yajl_gen_map_close(g); + + const unsigned char *ybuf; + unsigned int content_len; + yajl_gen_get_buf(g, &ybuf, &content_len); + + printf("Content-type: application/json\r\n"); + printf("Content-length: %d\r\n\r\n", content_len); + fwrite(ybuf, content_len, 1, stdout); + + yajl_gen_free(g); + } else if (strncmp(path, "/passwd", 7) == 0) { + const char *username = cgi_getentrystr("username"); + if (!check_auth(username)) + exit(0); + + const char *password = cgi_getentrystr("password"); + const char *new_password = cgi_getentrystr("new_password"); + if (password == NULL || new_password == NULL) { respond_JSON_Failure(); + } else { + if (auth_check_password(username, password)) { + auth_set_password(username, new_password); + respond_JSON_Success(); + } else { + respond_JSON_Failure(); + } } } else { respond_404();