moth/packages/mcp/src/common.c

536 lines
10 KiB
C
Raw Normal View History

2010-09-03 21:00:02 -06:00
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdarg.h>
#include <stdlib.h>
2010-09-03 21:00:02 -06:00
#include <stdio.h>
2010-09-16 12:21:16 -06:00
#include <string.h>
2010-09-03 21:00:02 -06:00
#include <ctype.h>
#include <values.h>
2010-09-03 21:00:02 -06:00
#include <time.h>
#include "common.h"
2010-09-14 18:04:33 -06:00
#ifdef NODUMP
# define DUMPf(fmt, args...)
#else
# define DUMPf(fmt, args...) fprintf(stderr, "%s:%s:%d " fmt "\n", __FILE__, __FUNCTION__, __LINE__, ##args)
#endif
#define DUMP() DUMPf("")
#define DUMP_d(v) DUMPf("%s = %d", #v, v)
#define DUMP_x(v) DUMPf("%s = 0x%x", #v, v)
#define DUMP_s(v) DUMPf("%s = %s", #v, v)
#define DUMP_c(v) DUMPf("%s = '%c' (0x%02x)", #v, v, v)
#define DUMP_p(v) DUMPf("%s = %p", #v, v)
#define POST_MAX 1024
/*
* CGI
*/
static int is_cgi = 0;
static char **argv = NULL;
static int
read_char_argv()
{
static int arg = 0;
static char *p;
if (NULL == argv) {
return EOF;
}
if (0 == arg) {
arg = 1;
p = argv[1];
}
if (! p) {
return EOF;
} else if (! *p) {
arg += 1;
p = argv[arg];
return '&';
}
return *(p++);
}
static int
read_char_stdin()
{
static int inlen = -1;
if (-1 == inlen) {
char *p = getenv("CONTENT_LENGTH");
if (p) {
inlen = atoi(p);
2010-09-14 18:04:33 -06:00
if (inlen > POST_MAX) {
inlen = POST_MAX;
}
2010-10-22 15:03:51 -06:00
if (inlen < 0) {
inlen = 0;
}
} else {
inlen = 0;
}
}
if (inlen) {
inlen -= 1;
return getchar();
}
return EOF;
}
static int
read_char_query_string()
{
static char *p = (char *)-1;
if ((char *)-1 == p) {
p = getenv("QUERY_STRING");
}
if (! p) {
return EOF;
} else if (! *p) {
return EOF;
} else {
return *(p++);
}
}
static int (* read_char)() = read_char_argv;
int
cgi_init(char *global_argv[])
{
char *rm = getenv("REQUEST_METHOD");
if (! rm) {
read_char = read_char_argv;
argv = global_argv;
} else if (0 == strcmp(rm, "POST")) {
read_char = read_char_stdin;
is_cgi = 1;
} else if (0 == strcmp(rm, "GET")) {
read_char = read_char_query_string;
is_cgi = 1;
} else {
printf(("405 Method not allowed\r\n"
"Allow: GET, POST\r\n"
"Content-type: text/plain\r\n"
"\r\n"
"%s is not allowed.\n"),
rm);
return -1;
}
return 0;
}
static char
tonum(int c)
{
if ((c >= '0') && (c <= '9')) {
return c - '0';
}
if ((c >= 'a') && (c <= 'f')) {
return 10 + c - 'a';
}
if ((c >= 'A') && (c <= 'F')) {
return 10 + c - 'A';
}
return 0;
}
static char
read_hex()
{
int a = read_char();
int b = read_char();
return tonum(a)*16 + tonum(b);
}
/* Read a key or a value. Since & and = aren't supposed to appear
outside of boundaries, we can use the same function for both.
*/
size_t
cgi_item(char *str, size_t maxlen)
{
int c;
size_t pos = 0;
while (1) {
c = read_char();
switch (c) {
case EOF:
case '=':
case '&':
str[pos] = '\0';
return pos;
case '%':
c = read_hex();
break;
case '+':
c = ' ';
break;
}
if (pos < maxlen - 1) {
str[pos] = c;
pos += 1;
}
}
}
void
2010-09-12 22:00:58 -06:00
cgi_head(char *title)
{
2010-09-12 22:15:06 -06:00
if (is_cgi) {
printf("Content-type: text/html\r\n\r\n");
}
printf(("<!DOCTYPE html>\n"
"<html>\n"
" <head>\n"
" <title>%s</title>\n"
" <link rel=\"stylesheet\" href=\"ctf.css\" type=\"text/css\">\n"
" </head>\n"
" <body>\n"
" <h1>%s</h1>\n"),
title, title);
2010-09-12 22:00:58 -06:00
}
void
cgi_foot()
{
printf("\n"
" </body>\n"
"</html>\n");
2010-09-12 22:00:58 -06:00
}
void
2011-03-14 17:13:25 -06:00
cgi_result(int code, char *desc, char *fmt, ...)
2010-09-12 22:00:58 -06:00
{
2011-03-14 17:13:25 -06:00
va_list ap;
2010-09-12 22:00:58 -06:00
2011-03-14 17:13:25 -06:00
if (is_cgi) {
printf("%d %s\r\n", code, desc);
}
cgi_head(desc);
2010-09-12 22:00:58 -06:00
va_start(ap, fmt);
vprintf(fmt, ap);
va_end(ap);
cgi_foot();
exit(0);
}
void
2011-03-14 17:13:25 -06:00
cgi_page(char *title, char *fmt, ...)
{
2011-03-14 17:13:25 -06:00
va_list ap;
2011-03-14 17:13:25 -06:00
cgi_head(title);
va_start(ap, fmt);
vprintf(fmt, ap);
va_end(ap);
2011-03-14 17:13:25 -06:00
cgi_foot();
exit(0);
}
2011-03-14 17:13:25 -06:00
void
cgi_error(char *text)
{
cgi_result(500, "Internal error", "<p>%s</p>", text);
}
/*
* Common routines
*/
2010-09-14 18:04:33 -06:00
#define EOL(c) ((EOF == (c)) || ('\n' == (c)))
int
fgrepx(char const *needle, char const *filename)
{
FILE *f;
int found = 0;
char const *p = needle;
f = fopen(filename, "r");
if (f) {
while (1) {
int c = fgetc(f);
/* This list of cases would have looked so much nicer in OCaml. I
apologize. */
2010-09-14 18:04:33 -06:00
if (EOL(c) && ('\0' == *p)) {
found = 1;
break;
2010-09-14 18:04:33 -06:00
} else if (EOF == c) { /* End of file */
break;
2011-02-23 17:29:34 -07:00
} else if (('\0' == p) || (*p != (char)c)) {
p = needle;
2010-09-14 18:04:33 -06:00
/* Discard the rest of the line */
do {
c = fgetc(f);
} while (! EOL(c));
2010-09-14 18:04:33 -06:00
} else if (EOL(c)) {
p = needle;
2010-09-14 18:04:33 -06:00
} else { /* It matched */
p += 1;
}
}
fclose(f);
}
return found;
}
void
urandom(char *buf, size_t buflen)
{
static int fd = -2;
if (-2 == fd) {
srandom(time(NULL) * getpid());
fd = open("/dev/urandom", O_RDONLY);
}
if (-1 != fd) {
int len;
len = read(fd, buf, buflen);
if (len == buflen) {
return;
}
}
/* Fall back to libc's crappy thing */
{
int i;
for (i = 0; i < buflen; i += 1) {
buf[i] = (char)random();
}
}
}
2010-09-03 21:00:02 -06:00
int
my_snprintf(char *buf, size_t buflen, char *fmt, ...)
{
int len;
va_list ap;
va_start(ap, fmt);
len = vsnprintf(buf, buflen - 1, fmt, ap);
va_end(ap);
2010-09-14 18:04:33 -06:00
buf[buflen - 1] = '\0';
if (len >= buflen) {
return buflen - 1;
} else {
return len;
}
}
2010-09-23 18:23:00 -06:00
static char *
mkpath(char const *base, char const *fmt, va_list ap)
{
char relpath[PATH_MAX];
static char path[PATH_MAX];
2010-09-23 18:23:00 -06:00
char const *var;
int len;
len = vsnprintf(relpath, sizeof(relpath) - 1, fmt, ap);
relpath[sizeof(relpath) - 1] = '\0';
2010-09-23 18:23:00 -06:00
var = getenv("CTF_BASE");
if (! var) {
var = base;
}
2010-09-23 18:23:00 -06:00
my_snprintf(path, sizeof(path), "%s/%s", var, relpath);
return path;
}
2010-09-23 18:23:00 -06:00
char *
state_path(char const *fmt, ...)
{
va_list ap;
char *ret;
va_start(ap, fmt);
ret = mkpath("/var/lib/ctf", fmt, ap);
va_end(ap);
return ret;
}
char *
package_path(char const *fmt, ...)
{
va_list ap;
char *ret;
va_start(ap, fmt);
ret = mkpath("/opt", fmt, ap);
va_end(ap);
return ret;
}
int
team_exists(char const *teamhash)
2010-09-03 21:00:02 -06:00
{
struct stat buf;
int ret;
int i;
if ((! teamhash) || (! *teamhash)) {
return 0;
}
2010-09-03 21:00:02 -06:00
/* Check for invalid characters. */
for (i = 0; teamhash[i]; i += 1) {
if (! isalnum(teamhash[i])) {
return 0;
}
}
2010-09-14 18:04:33 -06:00
/* stat seems to be the preferred way to check for existence. */
2010-09-23 18:23:00 -06:00
ret = stat(state_path("teams/names/%s", teamhash), &buf);
2010-09-03 21:00:02 -06:00
if (-1 == ret) {
return 0;
}
return 1;
}
int
award_points(char const *teamhash,
char const *category,
2010-09-12 22:00:58 -06:00
long points)
2010-09-03 21:00:02 -06:00
{
2010-09-06 21:39:59 -06:00
char line[100];
int linelen;
char *filename;
2010-09-06 21:39:59 -06:00
int fd;
time_t now = time(NULL);
2010-09-03 21:00:02 -06:00
if (! team_exists(teamhash)) {
return -2;
}
linelen = snprintf(line, sizeof(line),
2010-09-16 12:21:16 -06:00
"%lu %s %s %ld\n",
(unsigned long)now, teamhash, category, points);
if (sizeof(line) <= linelen) {
2010-09-03 21:00:02 -06:00
return -1;
}
/* At one time I had this writing to a single log file, using lockf.
This works, as long as nobody ever tries to edit the log file.
Editing the log file would require locking it, which would block
everything trying to score, effectively taking down the entire
contest. If you can't lock it first--and nothing in busybox lets
you do this--you have to bring down pretty much everything manually
anyway.
2010-09-03 21:00:02 -06:00
By putting new scores into new files and periodically appending
those files to the main log file, it is possible to stop the thing
that appends, edit the file at leisure, and then start the appender
back up, all without affecting things trying to score: they're
still able to record their score and move on. You don't even
really need an appender, but it does make things look a little
nicer on the fs.
The fact that this makes the code simpler is just gravy.
Note that doing this means there's a little time between when a
score's written and when the scoreboard (which only reads the log
file) picks it up. It's not a big deal for the points log, but
this situation makes this technique unsuitable for writing log
files that prevent people from double-scoring, like the puzzler or
token log.
*/
2010-09-23 23:27:14 -06:00
filename = state_path("points.tmp/%d.%d.%s.%s.%ld",
2010-09-23 18:23:00 -06:00
now, getpid(),
teamhash, category, points);
2010-09-03 21:00:02 -06:00
fd = open(filename, O_WRONLY | O_CREAT | O_EXCL, 0666);
if (-1 == fd) {
2010-09-03 21:00:02 -06:00
return -1;
}
2010-09-07 22:43:24 -06:00
if (-1 == write(fd, line, linelen)) {
close(fd);
return -1;
}
2010-09-03 21:00:02 -06:00
close(fd);
2010-09-23 23:27:14 -06:00
/* Rename into points.new */
{
char ofn[PATH_MAX];
strncpy(ofn, filename, sizeof(ofn));
filename = state_path("points.new/%d.%d.%s.%s.%ld",
now, getpid(),
teamhash, category, points);
rename(ofn, filename);
}
2010-09-03 21:00:02 -06:00
return 0;
}
2010-09-14 18:04:33 -06:00
/** Award points iff they haven't been logged.
If [line] is not in [dbfile], append it and give [points] to [team]
in [category].
*/
void
award_and_log_uniquely(char const *team,
char const *category,
2010-09-12 22:00:58 -06:00
long points,
2010-09-23 18:23:00 -06:00
char const *dbpath,
2010-09-14 18:04:33 -06:00
char const *line)
{
2011-03-14 17:13:25 -06:00
int fd;
/* Make sure they haven't already claimed these points */
if (fgrepx(line, dbpath)) {
2011-03-14 17:13:25 -06:00
cgi_result(409, "Already claimed",
"<p>Your team has already claimed these points.</p>");
}
/* Open and lock logfile */
fd = open(dbpath, O_WRONLY | O_CREAT, 0666);
if (-1 == fd) {
cgi_error("Unable to open log");
}
if (-1 == lockf(fd, F_LOCK, 0)) {
cgi_error("Unable to lock log");
}
/* Award points */
if (0 != award_points(team, category, points)) {
cgi_error("Unable to award points");
}
/* Log that we did so */
lseek(fd, 0, SEEK_END);
2010-09-14 18:04:33 -06:00
if (-1 == write(fd, line, strlen(line))) {
cgi_error("Unable to append log");
}
if (-1 == write(fd, "\n", 1)) {
cgi_error("Unable to append log");
}
close(fd);
}
2010-10-25 17:17:10 -06:00