#include "web.h" #include "mytime.h" #include #include #include #include #include #include #include #include #include #include #include #include #define PAGE "libmicrohttpd demo"\ "libmicrohttpd demo" static void sqliteError(void *pArg, int iErrCode, const char *zMsg) { // @CLEANUP: We should stuff these somewhere, but for now just print it fprintf(stderr, "(%d) %s\n", iErrCode, zMsg); } void prepare_database(struct App *app) { // @CLEANUP: Move this to main? doesn't belong here at least sqlite3_config(SQLITE_CONFIG_LOG, sqliteError, NULL); sqlite3_config(SQLITE_CONFIG_URI, 1); int err; printf("Running migrations against %s\n", app->dbname); sqlite3 *conn; if((err = sqlite3_open(app->dbname, &conn)) != SQLITE_OK) { fprintf(stderr, "(%d) %s\n", err, sqlite3_errmsg(conn)); abort(); } sqlite3_db_config(conn, SQLITE_DBCONFIG_ENABLE_FKEY, 1, NULL); if(sqlite3_exec(conn, "CREATE TABLE IF NOT EXISTS migration (id INTEGER PRIMARY KEY NOT NULL)", NULL, NULL, NULL) != SQLITE_OK) { abort(); } sqlite3_stmt *stmt; if(sqlite3_prepare_v2(conn, "SELECT MAX(id) FROM migration", -1, &stmt, NULL) != SQLITE_OK) { abort(); } uint64_t maxVersion = 0; while((err = sqlite3_step(stmt)) == SQLITE_ROW) { maxVersion = sqlite3_column_int64(stmt, 0); } if(err != SQLITE_DONE) { abort(); } if(sqlite3_finalize(stmt) != SQLITE_OK) { abort(); } if(maxVersion < 1) { char *prog = "BEGIN TRANSACTION;" "CREATE TABLE backups (" "id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, " "client TEXT NOT NULL, " "status TEXT NOT NULL, " "started TEXT NOT NULL, " "completed TEXT " ") STRICT"; if(sqlite3_exec(conn, prog, NULL, NULL, NULL) != SQLITE_OK) { abort(); } sqlite3_stmt *stmt; if(sqlite3_prepare_v2(conn, "INSERT INTO migration (id) VALUES (?1)", -1, &stmt, NULL) != SQLITE_OK) { abort(); } if(sqlite3_bind_int64(stmt, 1, 1) != SQLITE_OK) { abort(); } if(sqlite3_step(stmt) != SQLITE_DONE) { abort(); } if(sqlite3_finalize(stmt) != SQLITE_OK) { abort(); } if(sqlite3_exec(conn, "COMMIT TRANSACTION", NULL, NULL, NULL) != SQLITE_OK) { abort(); } } if(maxVersion < 2) { char *prog = "BEGIN TRANSACTION;" "CREATE TABLE clients (" "id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, " "name TEXT NOT NULL " ") STRICT;" "INSERT INTO clients(name) " "SELECT b.client " "FROM backups b " "GROUP BY b.client;" "CREATE TABLE backups_next (" "id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, " "client INTEGER NOT NULL, " "status TEXT NOT NULL, " "started TEXT NOT NULL, " "completed TEXT ," "FOREIGN KEY(client) REFERENCES clients(id)" ") STRICT;" "INSERT INTO backups_next(id, client, status, started, completed) " "SELECT b.id, c.id, b.status, b.started, b.completed " "FROM backups b " "JOIN clients c ON b.client = c.name;" "DROP TABLE backups;" "ALTER TABLE backups_next RENAME TO backups;"; if(sqlite3_exec(conn, prog, NULL, NULL, NULL) != SQLITE_OK) { abort(); } sqlite3_stmt *stmt; if(sqlite3_prepare_v2(conn, "INSERT INTO migration (id) VALUES (?1)", -1, &stmt, NULL) != SQLITE_OK) { abort(); } if(sqlite3_bind_int64(stmt, 1, 2) != SQLITE_OK) { abort(); } if(sqlite3_step(stmt) != SQLITE_DONE) { abort(); } if(sqlite3_finalize(stmt) != SQLITE_OK) { abort(); } if(sqlite3_exec(conn, "COMMIT TRANSACTION", NULL, NULL, NULL) != SQLITE_OK) { abort(); } } if(maxVersion < 3) { char *prog = "BEGIN TRANSACTION;" "CREATE UNIQUE INDEX clients_name ON clients(name)"; if(sqlite3_exec(conn, prog, NULL, NULL, NULL) != SQLITE_OK) { abort(); } sqlite3_stmt *stmt; if(sqlite3_prepare_v2(conn, "INSERT INTO migration (id) VALUES (?1)", -1, &stmt, NULL) != SQLITE_OK) { abort(); } if(sqlite3_bind_int64(stmt, 1, 3) != SQLITE_OK) { abort(); } if(sqlite3_step(stmt) != SQLITE_DONE) { abort(); } if(sqlite3_finalize(stmt) != SQLITE_OK) { abort(); } if(sqlite3_exec(conn, "COMMIT TRANSACTION", NULL, NULL, NULL) != SQLITE_OK) { abort(); } } } enum ReportEvent { REV_BEGIN, REV_FINISH, }; // @CLEANUP: This should live somewhere else, but I don't have a spot for it yet. int submit_report(struct App *app, char *client, enum ReportEvent event) { sqlite3 *conn; int err; if((err = sqlite3_open(app->dbname, &conn)) != SQLITE_OK) { abort(); } struct timespec now = get_current_time(); int64_t clientId; { sqlite3_stmt *stmt; if(sqlite3_prepare_v2(conn, "SELECT id FROM clients WHERE name = ?1", -1, &stmt, NULL) != SQLITE_OK) { abort(); } if(sqlite3_bind_text(stmt, 1, client, -1, SQLITE_STATIC) != SQLITE_OK) { abort(); } if(sqlite3_step(stmt) != SQLITE_ROW) { abort(); } if(sqlite3_column_type(stmt, 0) == SQLITE_NULL) { abort(); } clientId = sqlite3_column_int64(stmt, 0); if(sqlite3_step(stmt) != SQLITE_DONE) { abort(); } if(sqlite3_finalize(stmt) != SQLITE_OK) { abort(); } } switch(event) { case REV_BEGIN: { sqlite3_stmt *stmt; if(sqlite3_prepare_v2(conn, "INSERT INTO backups (client, status, started) VALUES (?1, ?2, datetime(?3, 'unixepoch', 'subsec'))", -1, &stmt, NULL) != SQLITE_OK) { abort(); } if(sqlite3_bind_int64(stmt, 1, clientId) != SQLITE_OK) { abort(); } if(sqlite3_bind_text(stmt, 2, "RUNNING", -1, SQLITE_STATIC) != SQLITE_OK) { abort(); } if(sqlite3_bind_double(stmt, 3, TIME_AS_FLOAT(now)) != SQLITE_OK) { abort(); } if(sqlite3_step(stmt) != SQLITE_DONE) { abort(); } if(sqlite3_finalize(stmt) != SQLITE_OK) { abort(); } break; } case REV_FINISH: { uint64_t backupId; sqlite3_stmt *stmt; if(sqlite3_prepare_v2(conn, "SELECT id FROM backups WHERE client = ?1 AND status = 'RUNNING'", -1, &stmt, NULL) != SQLITE_OK) { abort(); } if(sqlite3_bind_int64(stmt, 1, clientId) != SQLITE_OK) { abort(); } if(sqlite3_step(stmt) != SQLITE_ROW) { abort(); } if(sqlite3_column_type(stmt, 0) == SQLITE_NULL) { abort(); } backupId = sqlite3_column_int64(stmt, 0); if(sqlite3_step(stmt) != SQLITE_DONE) { abort(); } if(sqlite3_finalize(stmt) != SQLITE_OK) { abort(); } stmt = NULL; if(sqlite3_prepare_v2(conn, "UPDATE backups SET status = ?1, completed = datetime(?2, 'unixepoch', 'subsec') WHERE id = ?3", -1, &stmt, NULL) != SQLITE_OK) { abort(); } if(sqlite3_bind_text(stmt, 1, "FINISHED", -1, SQLITE_STATIC) != SQLITE_OK) { abort(); } if(sqlite3_bind_double(stmt, 2, TIME_AS_FLOAT(now)) != SQLITE_OK) { abort(); } if(sqlite3_bind_int64(stmt, 3, backupId) != SQLITE_OK) { abort(); } if(sqlite3_step(stmt) != SQLITE_DONE) { abort(); } if(sqlite3_finalize(stmt) != SQLITE_OK) { abort(); } break; } } if(sqlite3_close(conn) != SQLITE_OK) { abort(); } return 0; } int create_client(struct Request *req, char *name, struct CreateClientResult *result) { sqlite3 *conn; int err; if((err = sqlite3_open(req->server->app->dbname, &conn)) != SQLITE_OK) { abort(); } { sqlite3_stmt *stmt; if(sqlite3_prepare_v2(conn, "INSERT INTO clients (name) VALUES (?1)", -1, &stmt, NULL) != SQLITE_OK) { abort(); } if(sqlite3_bind_text(stmt, 1, name, -1, SQLITE_STATIC) != SQLITE_OK) { abort(); } if(sqlite3_step(stmt) != SQLITE_DONE) { abort(); } result->id = sqlite3_last_insert_rowid(conn); if(sqlite3_finalize(stmt) != SQLITE_OK) { abort(); } } if(sqlite3_close(conn) != SQLITE_OK) { abort(); } return 0; } int delete_client(struct Request *req, int64_t client_id) { sqlite3 *conn; if(sqlite3_open(req->server->app->dbname, &conn) != SQLITE_OK) { abort(); } sqlite3_stmt *stmt; if(sqlite3_prepare_v2(conn, "DELETE FROM clients WHERE id = ?1", -1, &stmt, NULL) != SQLITE_OK) { abort(); } if(sqlite3_bind_int64(stmt, 1, client_id) != SQLITE_OK) { abort(); } if(sqlite3_step(stmt) != SQLITE_DONE) { abort(); } if(sqlite3_finalize(stmt) != SQLITE_OK) { abort(); } if(sqlite3_close(conn) != SQLITE_OK) { abort(); } return 0; } int list_clients(struct Request *req, struct ListClientResult *result) { sqlite3 *conn; int err; if((err = sqlite3_open(req->server->app->dbname, &conn)) != SQLITE_OK) { abort(); } { sqlite3_stmt *stmt; if(sqlite3_prepare_v2(conn, "SELECT c.id, c.name FROM clients c WHERE c.id > ?1 ORDER BY c.id ASC LIMIT ?2", -1, &stmt, NULL) != SQLITE_OK) { abort(); } if(sqlite3_bind_int64(stmt, 1, -1) != SQLITE_OK) { abort(); } if(sqlite3_bind_int64(stmt, 2, 16) != SQLITE_OK) { abort(); } int ret; int index = 0; while((ret = sqlite3_step(stmt)) == SQLITE_ROW) { assert(index < 16); if(sqlite3_column_type(stmt, 0) == SQLITE_NULL) { abort(); } int64_t clientId = sqlite3_column_int64(stmt, 0); const uint8_t *name = sqlite3_column_text(stmt, 1); result->clients[index].id = clientId; strncpy(result->clients[index].name, (char*)name, CLIENT_NAME_MAX); index++; } if(ret != SQLITE_DONE) { abort(); } result->clients_len = index; if(sqlite3_finalize(stmt) != SQLITE_OK) { abort(); } } if(sqlite3_close(conn) != SQLITE_OK) { abort(); } return 0; } static char *render_clients_page(struct Request *request, size_t *out_len) { int ret; struct ListClientResult *result = malloc(sizeof(struct ListClientResult) + sizeof(struct ListClientResultClient) * 16); ret = list_clients(request, result); if(ret != 0) { free(result); return NULL; } char *buf = NULL; size_t buf_len = 0; FILE *f = open_memstream(&buf, &buf_len); if (!f) { free(result); return NULL; } fprintf(f, "\n" "\n" "\n" " Clients\n" "\n" "\n" "

Clients

\n" " \n" " \n"); for (int i = 0; i < result->clients_len; i++) { fprintf(f, " \n" " \n" " \n" " \n", result->clients[i].name, result->clients[i].id ); } fprintf(f, "

Name	Actions
%s	\n" " \n" " \n" " \n" " \n" "

\n" "

Add New Client

\n" " \n" "\n" "\n"); fclose(f); *out_len = buf_len; free(result); return buf; } static enum MHD_Result create_post_collector(struct PostCollector *c) { c->data = malloc(512); c->size = 0; c->capacity = 512; return MHD_YES; } static enum MHD_Result collect_post(struct PostCollector *c, const char *data, size_t *data_len) { size_t newSize = c->size + *data_len; if(newSize > c->capacity) { while(newSize > c->capacity) c->capacity *= 2; c->data = realloc(c->data, c->capacity); } memcpy(c->data + c->size, data, *data_len); c->size = newSize; return MHD_YES; } static enum MHD_Result destroy_post_collector(struct PostCollector *c) { if(c->capacity != 0) { free(c->data); c->capacity = 0; } return MHD_YES; } static enum MHD_Result form_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 Request *request = cls; if (strcmp(key, "name") == 0 && off + size < CLIENT_NAME_MAX) { memcpy(request->form.name + off, data, size); request->form.name[off + size] = '\0'; request->form.name_set = 1; } else if (strcmp(key, "client_id") == 0 && off == 0) { char buf[32] = {0}; size_t copy_size = size < 31 ? size : 31; memcpy(buf, data, copy_size); request->form.client_id = strtoll(buf, NULL, 10); request->form.client_id_set = 1; } return MHD_YES; } static enum MHD_Result handler( void *cls, struct MHD_Connection *connection, const char *url, const char *method, const char *version, const char *upload_data, size_t *upload_data_size, void **ptr ) { struct Server *server = cls; int ret; // For the first call where con_cls points at NULL we have to create the // connection state struct Request *request = *ptr; if (request == NULL) { request = calloc(1, sizeof(struct Request)); if(request == NULL) { return MHD_NO; } *ptr = request; request->server = server; if(strcmp(method, "POST") == 0) { if(strcmp(url, "/clients") == 0 || strcmp(url, "/clients/delete") == 0) { request->post_proc = MHD_create_post_processor( connection, 1024, &form_iterator, request); if (request->post_proc == NULL) { free(request); return MHD_NO; } } else { create_post_collector(&request->pp); } } return MHD_YES; } if(strcmp(method, "POST") == 0) { if (request->post_proc != NULL) { if (*upload_data_size != 0) { MHD_post_process(request->post_proc, upload_data, *upload_data_size); *upload_data_size = 0; return MHD_YES; } } else { collect_post(&request->pp, upload_data, upload_data_size); if (*upload_data_size != 0) { *upload_data_size = 0; return MHD_YES; } } if(strcmp(url, "/clients") == 0) { MHD_destroy_post_processor(request->post_proc); request->post_proc = NULL; if (!request->form.name_set || request->form.name[0] == '\0') { return MHD_NO; } struct CreateClientResult result; ret = create_client(request, request->form.name, &result); if(ret != 0) return MHD_NO; // Post-Redirect-Get: redirect back to GET /clients struct MHD_Response *response = MHD_create_response_from_buffer( 0, "", MHD_RESPMEM_PERSISTENT); ret = MHD_add_response_header(response, "Location", "/clients"); if(ret != MHD_YES) return ret; ret = MHD_queue_response(connection, MHD_HTTP_SEE_OTHER, response); MHD_destroy_response(response); return ret; } else if(strcmp(url, "/clients/delete") == 0) { MHD_destroy_post_processor(request->post_proc); request->post_proc = NULL; if (!request->form.client_id_set || request->form.client_id <= 0) { return MHD_NO; } ret = delete_client(request, request->form.client_id); if(ret != 0) return MHD_NO; struct MHD_Response *response = MHD_create_response_from_buffer( 0, "", MHD_RESPMEM_PERSISTENT); ret = MHD_add_response_header(response, "Location", "/clients"); if(ret != MHD_YES) return ret; ret = MHD_queue_response(connection, MHD_HTTP_SEE_OTHER, response); MHD_destroy_response(response); return ret; } else if(strcmp(url, "/api/report") == 0) { cJSON *rootJson = cJSON_ParseWithLength(request->pp.data, request->pp.size); cJSON *clientJson = cJSON_GetObjectItemCaseSensitive(rootJson, "client"); if(!cJSON_IsString(clientJson) || clientJson->valuestring == NULL) { abort(); } cJSON *eventJson = cJSON_GetObjectItemCaseSensitive(rootJson, "event"); if(!cJSON_IsString(eventJson) || eventJson->valuestring == NULL) { abort(); } enum ReportEvent event; if(strcmp(eventJson->valuestring, "begin_backup") == 0) { event = REV_BEGIN; } else if(strcmp(eventJson->valuestring, "end_backup") == 0) { event = REV_FINISH; } else { abort(); } ret = submit_report(server->app, clientJson->valuestring, event); if(ret != 0) { // Should actually return an error page, but this will do for now return MHD_NO; } cJSON_Delete(rootJson); // We've taken in all the data, now we can use it destroy_post_collector(&request->pp); struct MHD_Response *response = MHD_create_response_from_buffer( 0, "", MHD_RESPMEM_PERSISTENT ); ret = MHD_add_response_header(response, "Content-Type", "application/json"); if(ret != MHD_YES) return ret; ret = MHD_queue_response( connection, MHD_HTTP_NO_CONTENT, response ); MHD_destroy_response(response); return ret; } else if(strcmp(url, "/api/client") == 0) { cJSON *rootJson = cJSON_ParseWithLength(request->pp.data, request->pp.size); cJSON *nameJson = cJSON_GetObjectItemCaseSensitive(rootJson, "name"); if(!cJSON_IsString(nameJson) || nameJson->valuestring == NULL) { abort(); } struct CreateClientResult result; ret = create_client(request, nameJson->valuestring, &result); if(ret != 0) { // Should actually return an error page, but this will do for now return MHD_NO; } cJSON_Delete(rootJson); // We've taken in all the data, now we can use it destroy_post_collector(&request->pp); char *buf = malloc(1024); size_t buf_len = snprintf(buf, 1024, "{ \"id\": \"%ld\" }", result.id); if(buf_len >= 1024) { abort(); } struct MHD_Response *response = MHD_create_response_from_buffer( buf_len, buf, MHD_RESPMEM_MUST_FREE ); ret = MHD_add_response_header(response, "Content-Type", "application/json"); if(ret != MHD_YES) return ret; ret = MHD_queue_response( connection, MHD_HTTP_OK, response ); MHD_destroy_response(response); return ret; } else { // Not found destroy_post_collector(&request->pp); struct MHD_Response *response = MHD_create_response_from_buffer( 0, "", MHD_RESPMEM_PERSISTENT ); ret = MHD_add_response_header(response, "Content-Type", "application/json"); if(ret != MHD_YES) return ret; ret = MHD_queue_response( connection, MHD_HTTP_NOT_FOUND, response ); MHD_destroy_response(response); return ret; } } else if (strcmp(method, "GET") == 0) { if (*upload_data_size != 0) return MHD_NO; if(strcmp(url, "/clients") == 0) { size_t html_len; char *html = render_clients_page(request, &html_len); if (!html) { return MHD_NO; } struct MHD_Response *response = MHD_create_response_from_buffer( html_len, html, MHD_RESPMEM_MUST_FREE ); ret = MHD_add_response_header(response, "Content-Type", "text/html"); if(ret != MHD_YES) return ret; ret = MHD_queue_response( connection, MHD_HTTP_OK, response ); MHD_destroy_response(response); return ret; } else { struct MHD_Response *response = MHD_create_response_from_buffer( strlen(server->page), (void *)server->page, MHD_RESPMEM_PERSISTENT ); ret = MHD_queue_response( connection, MHD_HTTP_OK, response ); MHD_destroy_response(response); return ret; } } return MHD_NO; } static void request_completed_callback ( void *cls, struct MHD_Connection *connection, void **con_cls, enum MHD_RequestTerminationCode toe ) { struct Request *request = *con_cls; if (request->post_proc != NULL) { MHD_destroy_post_processor(request->post_proc); } destroy_post_collector(&request->pp); free(request); } static struct MHD_Daemon *d = NULL; int web_begin(struct Server *server, int port) { assert(d == NULL); server->page = PAGE; d = MHD_start_daemon( MHD_USE_THREAD_PER_CONNECTION, port, NULL, NULL, &handler, server, MHD_OPTION_NOTIFY_COMPLETED, &request_completed_callback, NULL, MHD_OPTION_END ); if (d == NULL) return 1; return 0; } int web_join(struct Server *server) { assert(d != NULL); MHD_stop_daemon(d); d = NULL; return 0; }