2 * This file is in the public domain.
20 char dbpath[PATH_MAX];
22 void server_sig_handler(int, short, void *);
23 void server_open_db(struct env *);
24 void server_close_db(struct env *);
25 __dead void server_shutdown(struct env *);
26 int server_reply(struct client *, int, const char *);
27 char *server_getquery(struct client *);
30 server_sig_handler(int sig, short ev, void *arg)
32 struct env *env = arg;
35 * Normal signal handler rules don't apply here because libevent
41 log_info("re-opening the db");
50 fatalx("unexpected signal %d", sig);
55 loadstmt(sqlite3 *db, sqlite3_stmt **stmt, const char *sql)
59 err = sqlite3_prepare_v2(db, sql, -1, stmt, NULL);
61 fatalx("failed to prepare statement \"%s\": %s",
62 sql, sqlite3_errstr(err));
66 server_open_db(struct env *env)
70 err = sqlite3_open_v2(dbpath, &env->env_db,
71 SQLITE_OPEN_READONLY, NULL);
73 fatalx("can't open database %s: %s", dbpath,
74 sqlite3_errmsg(env->env_db));
76 loadstmt(env->env_db, &env->env_query,
77 "select mid, \"from\", date, subj"
79 " where email match ?"
80 " order by rank, date"
85 server_close_db(struct env *env)
89 sqlite3_finalize(env->env_query);
91 if ((err = sqlite3_close(env->env_db)) != SQLITE_OK)
92 log_warnx("sqlite3_close %s", sqlite3_errstr(err));
96 server_main(const char *db)
98 char path[PATH_MAX], *parent;
102 struct event sigterm;
104 signal(SIGPIPE, SIG_IGN);
106 memset(&env, 0, sizeof(env));
108 if (realpath(db, dbpath) == NULL)
109 fatal("realpath %s", db);
111 strlcpy(path, dbpath, sizeof(path));
112 parent = dirname(path);
113 if (unveil(parent, "r") == -1)
114 fatal("unveil(%s, r)", parent);
117 * rpath flock: sqlite3
120 if (pledge("stdio rpath flock unix", NULL) == -1)
123 server_open_db(&env);
129 event_set(&env.env_sockev, env.env_sockfd, EV_READ|EV_PERSIST,
131 event_add(&env.env_sockev, NULL);
133 evtimer_set(&env.env_pausev, fcgi_accept, &env);
135 signal_set(&sighup, SIGHUP, server_sig_handler, &env);
136 signal_set(&sigint, SIGINT, server_sig_handler, &env);
137 signal_set(&sigterm, SIGTERM, server_sig_handler, &env);
139 signal_add(&sighup, NULL);
140 signal_add(&sigint, NULL);
141 signal_add(&sigterm, NULL);
146 server_shutdown(&env);
150 server_shutdown(struct env *env)
152 log_info("shutting down");
153 server_close_db(env);
158 server_reply(struct client *clt, int status, const char *arg)
161 clt_printf(clt, "Status: %d\r\n", status) == -1)
165 if (clt_printf(clt, "Location: %s\r\n", arg) == -1)
171 clt_printf(clt, "Content-Type: %s\r\n", arg) == -1)
174 return (clt_puts(clt, "\r\n"));
178 server_urldecode(char *s)
182 char *q, code[3] = {0};
200 if (!isxdigit((unsigned char)s[1]) ||
201 !isxdigit((unsigned char)s[2]))
205 x = strtoul(code, NULL, 16);
214 server_getquery(struct client *clt)
218 tmp = clt->clt_query;
219 while ((field = strsep(&tmp, "&")) != NULL) {
220 if (server_urldecode(field) == -1)
223 if (!strncmp(field, "q=", 2))
225 log_info("unknown query param %s", field);
232 fts_escape(const char *p, char *buf, size_t bufsize)
237 * split p into words and quote them into buf.
238 * quoting means wrapping each word into "..." and
239 * replace every " with "".
240 * i.e. 'C++ "framework"' -> '"C++" """framework"""'
241 * flatting all the whitespaces seems fine too.
245 while (bufsize != 0) {
246 p += strspn(p, " \f\n\r\t\v");
252 while (*p && !isspace((unsigned char)*p) && bufsize != 0) {
253 if (*p == '"') { /* double the quote character */
268 if ((*p == '\0') && bufsize != 0) {
276 server_handle(struct env *env, struct client *clt)
279 char esc[QUERY_MAXLEN];
281 const char *mid, *from, *subj;
287 if ((query = server_getquery(clt)) != NULL &&
288 fts_escape(query, esc, sizeof(esc)) != -1) {
289 log_debug("searching for %s", esc);
291 err = sqlite3_bind_text(env->env_query, 1, esc, -1, NULL);
292 if (err != SQLITE_OK) {
293 sqlite3_reset(env->env_query);
294 if (server_reply(clt, 500, "text/plain") == -1)
296 if (clt_puts(clt, "Internal server error\n") == -1)
298 return (fcgi_end_request(clt, 1));
302 if (server_reply(clt, 200, "text/html") == -1)
305 if (clt_puts(clt, "<!doctype html>"
308 "<meta charset='utf-8'>"
309 "<meta name='viewport' content='width=device-width'>"
310 "<link rel='stylesheet' href='/style.css'>"
311 "<title>Game of Trees Mail Archive | Search</title>"
314 "<header class='index-header'>"
315 "<a href='https://gameoftrees.org' target='_blank'>"
316 "<img src='/got.png' srcset='/got.png, /got@2x.png 2x'"
317 " alt='\"GOT\" where the \"O\" is a cute, smiling pufferfish'"
320 "<h1>Game of Trees Mail Archive</h1>"
324 if (clt_puts(clt, "<form method='get'>"
326 "<input type='search' name='q' value='") == -1 ||
327 clt_putsan(clt, query) == -1 ||
328 clt_puts(clt, "'/></label>"
329 " <button type='submit'>search</button>"
336 if (clt_puts(clt, "<div class='thread'><ul>") == -1)
340 err = sqlite3_step(env->env_query);
341 if (err == SQLITE_DONE)
343 if (err != SQLITE_ROW) {
344 log_warnx("%s: sqlite3_step %s", __func__,
345 sqlite3_errstr(err));
349 mid = sqlite3_column_text(env->env_query, 0);
350 from = sqlite3_column_text(env->env_query, 1);
351 date = sqlite3_column_int64(env->env_query, 2);
352 subj = sqlite3_column_text(env->env_query, 3);
354 if ((sizeof(d) == 4) && date > UINT32_MAX) {
355 log_warnx("overflow of 32bit time value");
360 if ((tm = gmtime(&d)) == NULL) {
361 log_warnx("gmtime failure");
365 if (strftime(dbuf, sizeof(dbuf), "%F %R", tm) == 0) {
366 log_warnx("strftime failure");
370 if (clt_puts(clt, "<li class='mail'>"
371 "<p class='mail-meta'><time>") == -1 ||
372 clt_putsan(clt, dbuf) == -1 ||
373 clt_puts(clt, "</time> <span class='from'>") == -1 ||
374 clt_putsan(clt, from) == -1 ||
375 clt_puts(clt, "</span><span class=colon>:</span>") == -1 ||
377 "<p class='subject'>"
378 "<a href='/mail/") == -1 ||
379 clt_putsan(clt, mid) == -1 ||
380 clt_puts(clt, ".html'>") == -1 ||
381 clt_putsan(clt, subj) == -1 ||
382 clt_puts(clt, "</a></p></li>") == -1)
386 if (clt_puts(clt, "</ul></div>") == -1)
390 if (clt_puts(clt, "</body></html>\n") == -1)
393 sqlite3_reset(env->env_query);
394 return (fcgi_end_request(clt, 0));
396 sqlite3_reset(env->env_query);
401 server_client_free(struct client *clt)
403 free(clt->clt_server_name);
404 free(clt->clt_script_name);
405 free(clt->clt_path_info);
406 free(clt->clt_query);