/* snac - A simple, minimalistic ActivityPub instance */ /* copyright (c) 2022 grunfink - MIT license */ #include "xs.h" #include "xs_io.h" #include "xs_encdec.h" #include "xs_json.h" #include "xs_socket.h" #include "xs_httpd.h" #include "xs_mime.h" #include "snac.h" #include #include /* susie.png */ const char *susie = "iVBORw0KGgoAAAANSUhEUgAAAEAAAABAAQAAAAC" "CEkxzAAAAUUlEQVQoz43R0QkAMQwCUDdw/y3dwE" "vsvzlL4X1IoQkAisKmwfAFT3RgJHbQezpSRoXEq" "eqCL9BJBf7h3QbOCCxV5EVWMEMwG7K1/WODtlvx" "AYTtEsDU9F34AAAAAElFTkSuQmCC"; int server_get_handler(d_char *req, char *q_path, char **body, int *b_size, char **ctype) /* basic server services */ { int status = 0; /* is it the server root? */ if (*q_path == '\0') { /* try to open greeting.html */ xs *fn = xs_fmt("%s/greeting.html", srv_basedir); FILE *f; if ((f = fopen(fn, "r")) != NULL) { d_char *s = xs_readall(f); fclose(f); status = 200; /* replace %host% */ s = xs_replace_i(s, "%host%", xs_dict_get(srv_config, "host")); /* does it have a %userlist% mark? */ if (xs_str_in(s, "%userlist%") != -1) { char *host = xs_dict_get(srv_config, "host"); xs *list = user_list(); char *p, *uid; xs *ul = xs_str_new("\n"); s = xs_replace_i(s, "%userlist%", ul); } *body = s; } } else if (strcmp(q_path, "/susie.png") == 0 || strcmp(q_path, "/favicon.ico") == 0 ) { status = 200; *body = xs_base64_dec(susie, b_size); *ctype = "image/png"; } if (status != 0) srv_debug(1, xs_fmt("server_get_handler serving '%s' %d", q_path, status)); return status; } void httpd_connection(FILE *f) /* the connection processor */ { xs *req; char *method; int status = 0; d_char *body = NULL; int b_size = 0; char *ctype = NULL; xs *headers = NULL; xs *q_path = NULL; xs *payload = NULL; int p_size = 0; char *p; req = xs_httpd_request(f, &payload, &p_size); if (req == NULL) { /* probably because a timeout */ fclose(f); return; } method = xs_dict_get(req, "method"); q_path = xs_dup(xs_dict_get(req, "path")); /* crop the q_path from leading / and the prefix */ if (xs_endswith(q_path, "/")) q_path = xs_crop(q_path, 0, -1); p = xs_dict_get(srv_config, "prefix"); if (xs_startswith(q_path, p)) q_path = xs_crop(q_path, strlen(p), 0); if (strcmp(method, "GET") == 0 || strcmp(method, "HEAD") == 0) { /* cascade through */ if (status == 0) status = server_get_handler(req, q_path, &body, &b_size, &ctype); if (status == 0) status = webfinger_get_handler(req, q_path, &body, &b_size, &ctype); if (status == 0) status = activitypub_get_handler(req, q_path, &body, &b_size, &ctype); if (status == 0) status = html_get_handler(req, q_path, &body, &b_size, &ctype); } else if (strcmp(method, "POST") == 0) { if (status == 0) status = activitypub_post_handler(req, q_path, payload, p_size, &body, &b_size, &ctype); if (status == 0) status = html_post_handler(req, q_path, payload, p_size, &body, &b_size, &ctype); } /* let's go */ headers = xs_dict_new(); /* unattended? it's an error */ if (status == 0) { srv_debug(1, xs_fmt("httpd_connection unattended %s %s", method, q_path)); status = 404; } if (status == 404) body = xs_str_new("

404 Not Found

"); if (status == 400) body = xs_str_new("

400 Bad Request

"); if (status == 303) headers = xs_dict_append(headers, "location", body); if (status == 401) headers = xs_dict_append(headers, "WWW-Authenticate", "Basic realm=\"IDENTIFY\""); if (ctype == NULL) ctype = "text/html; charset=utf-8"; headers = xs_dict_append(headers, "content-type", ctype); headers = xs_dict_append(headers, "x-creator", USER_AGENT); if (b_size == 0 && body != NULL) b_size = strlen(body); /* if it was a HEAD, no body will be sent */ if (strcmp(method, "HEAD") == 0) body = xs_free(body); xs_httpd_response(f, status, headers, body, b_size); fclose(f); srv_archive("RECV", req, payload, p_size, status, headers, body, b_size); xs_free(body); } static jmp_buf on_break; void term_handler(int s) { longjmp(on_break, 1); } static void *purge_thread(void *arg) /* spawned purge */ { srv_log(xs_dup("purge start")); purge_all(); srv_log(xs_dup("purge end")); return NULL; } static void *queue_thread(void *arg) /* queue thread (queue management) */ { pthread_mutex_t dummy_mutex = PTHREAD_MUTEX_INITIALIZER; pthread_cond_t dummy_cond = PTHREAD_COND_INITIALIZER; time_t purge_time; /* first purge time */ purge_time = time(NULL) + 10 * 60; srv_log(xs_fmt("queue thread start")); while (srv_running) { time_t t; { xs *list = user_list(); char *p, *uid; /* process queues for all users */ p = list; while (xs_list_iter(&p, &uid)) { snac snac; if (user_open(&snac, uid)) { process_queue(&snac); user_free(&snac); } } } /* time to purge? */ if ((t = time(NULL)) > purge_time) { pthread_t pth; pthread_create(&pth, NULL, purge_thread, NULL); /* next purge time is tomorrow */ purge_time = t + 24 * 60 * 60; } /* sleep 3 seconds */ struct timespec ts; clock_gettime(CLOCK_REALTIME, &ts); ts.tv_sec += 3; pthread_mutex_lock(&dummy_mutex); while (pthread_cond_timedwait(&dummy_cond, &dummy_mutex, &ts) == 0); pthread_mutex_unlock(&dummy_mutex); } srv_log(xs_fmt("queue thread stop")); return NULL; } static void *connection_thread(void *arg) /* connection thread */ { httpd_connection((FILE *)arg); return NULL; } int threaded_connections = 1; void httpd(void) /* starts the server */ { char *address; int port; int rs; pthread_t htid; address = xs_dict_get(srv_config, "address"); port = xs_number_get(xs_dict_get(srv_config, "port")); if ((rs = xs_socket_server(address, port)) == -1) { srv_log(xs_fmt("cannot bind socket to %s:%d", address, port)); return; } srv_running = 1; signal(SIGPIPE, SIG_IGN); signal(SIGTERM, term_handler); signal(SIGINT, term_handler); srv_log(xs_fmt("httpd start %s:%d %s", address, port, USER_AGENT)); pthread_create(&htid, NULL, queue_thread, NULL); if (setjmp(on_break) == 0) { for (;;) { FILE *f = xs_socket_accept(rs); if (threaded_connections) { pthread_t cth; pthread_create(&cth, NULL, connection_thread, f); pthread_detach(cth); } else httpd_connection(f); } } srv_running = 0; /* wait for the helper thread to end */ pthread_join(htid, NULL); srv_log(xs_fmt("httpd stop %s:%d", address, port)); }