From ac159d0b47f048beb72edabbd05c893894e95721 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Anton=20Luka=20=C5=A0ijanec?= Date: Thu, 12 May 2022 22:18:41 +0200 Subject: initial commit --- .gitignore | 4 ++ Makefile | 27 +++++++++++++ prijave.c | 122 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ prijave.conf | 4 ++ 4 files changed, 157 insertions(+) create mode 100644 .gitignore create mode 100644 Makefile create mode 100644 prijave.c create mode 100644 prijave.conf diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..2623084 --- /dev/null +++ b/.gitignore @@ -0,0 +1,4 @@ +*.out +prijave +core +sqlite3 diff --git a/Makefile b/Makefile new file mode 100644 index 0000000..a49d1dc --- /dev/null +++ b/Makefile @@ -0,0 +1,27 @@ +DESTDIR=/ +CC = cc +cflags = -O0 -Wall -I. -fpic -Wformat-security -Wextra -pedantic -g3 +ldflags = $(shell pkg-config --libs sqlite3) -lmicrohttpd # -lintl +PROJ = prijave +# cflags and ldflags are used so that users specifying CFLAGS and LDFLAGS do not override my flags +# += is not used, because gcc usually accepts last option, for example -O0 -O2 will use -O2 +.NOTPARALLEL: +default: + $(CC) -L. $(cflags) $(CFLAGS) $(PROJ).c $(ldflags) $(LDFLAGS) -o$(PROJ) +install: + mkdir -p $(DESTDIR)/usr/bin $(DESTDIR)/etc + cp $(PROJ) $(DESTDIR)/usr/bin/ + -[ ! -f $(DESTDIR)/etc/$(PROJ) ] && cp $(PROJ).conf $(DESTDIR)/etc/$(PROJ) + +distclean: clean + +clean: + rm -f $(PROJ) + +uninstall: + rm -f $(DESTDIR)/usr/bin/$(PROJ) + +valgrind: + valgrind --leak-check=full --show-leak-kinds=all --track-origins=yes --verbose --log-file=valgrind-out.txt $(COMMAND) + +.PHONY: valgrind clean distclean install uninstall default diff --git a/prijave.c b/prijave.c new file mode 100644 index 0000000..d16c8c6 --- /dev/null +++ b/prijave.c @@ -0,0 +1,122 @@ +#include +#include +#include +#undef NDEBUG +#include +#include +#include +#include +#include +#include +#ifndef PR_PORT +#define PR_PORT "7745" +#endif +#define STRA(x) #x +#define STR(x) STRA(x) +struct prijave { + sqlite3 * db; + const char * pass; + const char * salt; +}; +enum question { + // types + RADIO = (0 << 0), + CHECKBOX = (1 << 0), + TEXT = (1 << 1), + // options: + NOT_NULL = (1 << 5) + +}; +static enum MHD_Result httpd (void * userdata, struct MHD_Connection * connection, const char * path, const char * meth, const char * ver, const char * upload, size_t * upload_size, void ** request_state) { + struct prijave * prijave = (struct prijave *) userdata; + char * response = "httpd"; + char * content_type = "text/plain"; + char * location = NULL; + int status_code = MHD_HTTP_OK; + enum MHD_ResponseMemoryMode rmm = MHD_RESPMEM_PERSISTENT; + static char headers_already_received; + if (strcmp(meth, "GET") && strcmp(meth, "POST")) { // pravzaprav bi bilo + status_code = MHD_HTTP_NOT_ACCEPTABLE; // bolj prav uporabiti + response = "HTTP 406\n"; // PUT request, ampak HTML form tega nima /: + goto r; + } + if (*request_state != &headers_already_received) { + *request_state = &headers_already_received; + return MHD_YES; + } + sqlite3_stmt * stmt; + int ret; + char statem[2048]; + char spaces[2048]; + memset(spaces, ' ', 2048); + spaces[2047] = '\0'; +#define RETURN_ERROR(section) \ + { \ + status_code = MHD_HTTP_BAD_GATEWAY; \ + response = malloc(strlen(sqlite3_errstr(ret))+strlen(sqlite3_errmsg(prijave->db))+128+2*strlen(statem)); \ + rmm = MHD_RESPMEM_MUST_FREE; \ + sprintf(response, "HTTP 502: " section " %s\n%.*s^\n%s\n%s\n\n", statem, sqlite3_error_offset(prijave->db) == -1 ? 0 : sqlite3_error_offset(prijave->db), spaces, sqlite3_errstr(ret), sqlite3_errmsg(prijave->db)); \ + sqlite3_finalize(stmt); /* mogoče tudi ni treba */ \ + fprintf(stderr, "%s\n", response); \ + goto r; \ + } +#define CREATE_TABLE(table, cols) \ + strcpy(statem, "CREATE TABLE IF NOT EXISTS " table " (" cols ") STRICT;"); \ + ret = sqlite3_prepare_v3(prijave->db, statem, -1, 0, &stmt, NULL); \ + if (ret != SQLITE_OK) \ + RETURN_ERROR("prepare"); \ + ret = sqlite3_step(stmt); \ + sqlite3_finalize(stmt); \ + if (ret != SQLITE_DONE) \ + RETURN_ERROR("step"); + CREATE_TABLE("responses", "person INTEGER NOT NULL, question INTEGER NOT NULL, answer ANY"); + CREATE_TABLE("options", "question INTEGER NOT NULL, text TEXT"); + CREATE_TABLE("questions", "id INTEGER PRIMARY KEY AUTOINCREMENT, poll INTEGER NOT NULL, text TEXT, type INTEGER NOT NULL"); + CREATE_TABLE("persons", "id INTEGER PRIMARY KEY AUTOINCREMENT, poll INTEGER NOT NULL, email TEXT, name TEXT"); + CREATE_TABLE("person_metadata", "person INTEGER NOT NULL, key TEXT NOT NULL, value ANY") + CREATE_TABLE("polls", "id INTEGER PRIMARY KEY AUTOINCREMENT, name TEXT, description TEXT, password TEXT"); + // TODO: CHECK() relation IDs that they exist - poll, question, responses + // TODO: CHECK() if answer is valid for RADIO/CHECKBOX questions + // TODO: CHECK() if question type is valid + struct MHD_Response * httpd_response; +r: + httpd_response = MHD_create_response_from_buffer(strlen(response), (void *) response, rmm); + MHD_add_response_header(httpd_response, "Content-Type", content_type); + if (status_code >= 300 && status_code <= 399) + MHD_add_response_header(httpd_response, "Location", location); + int r = MHD_queue_response(connection, status_code, httpd_response); + MHD_destroy_response(httpd_response); + return r; +} +static void handler (int s __attribute__((unused))) { + return; +} +int main (void) { + int r = 0; + struct prijave prijave; + memset(&prijave, 0, sizeof prijave); + umask(00077); + prijave.pass = getenv("PR_PASS"); // set to disable anyone to create polls + assert((prijave.salt = getenv("PR_SALT"))); + assert(getenv("PR_DB")); + assert(sqlite3_threadsafe()); + int ret = sqlite3_open_v2(getenv("PR_DB"), &prijave.db, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE | SQLITE_OPEN_NOMUTEX | SQLITE_OPEN_EXRESCODE, NULL); + if (ret != SQLITE_OK) { + fprintf(stderr, "sqlite3_open_v2: %s\n", sqlite3_errstr(ret)); + goto r; + } + struct MHD_Daemon * d = MHD_start_daemon(MHD_USE_INTERNAL_POLLING_THREAD, atoi(getenv("PR_PORT") ? getenv("PR_PORT") : PR_PORT), NULL, NULL, &httpd, &prijave, MHD_OPTION_END); + if (!d) { + r = 1; + goto r; + } + signal(SIGTERM, handler); + signal(SIGINT, handler); + pause(); +r: + if (d) + MHD_stop_daemon(d); // to počaka na smrt vseh threadov afaik + if (prijave.db) + sqlite3_close_v2(prijave.db); // zato se tudi vsi stmtji izlkopijo + return r; +} diff --git a/prijave.conf b/prijave.conf new file mode 100644 index 0000000..67de89b --- /dev/null +++ b/prijave.conf @@ -0,0 +1,4 @@ +PR_PORT=7745 +PR_DB=/var/lib/prijave/sqlite3 +PR_PASS=admin +PR_SALT=salt -- cgit v1.2.3