mirror of https://github.com/dirtbags/tanks.git
321 lines
6.6 KiB
C
321 lines
6.6 KiB
C
/*
|
|
* This software has been authored by an employee or employees of Los
|
|
* Alamos National Security, LLC, operator of the Los Alamos National
|
|
* Laboratory (LANL) under Contract No. DE-AC52-06NA25396 with the U.S.
|
|
* Department of Energy. The U.S. Government has rights to use,
|
|
* reproduce, and distribute this software. The public may copy,
|
|
* distribute, prepare derivative works and publicly display this
|
|
* software without charge, provided that this Notice and any statement
|
|
* of authorship are reproduced on all copies. Neither the Government
|
|
* nor LANS makes any warranty, express or implied, or assumes any
|
|
* liability or responsibility for the use of this software. If
|
|
* software is modified to produce derivative works, such modified
|
|
* software should be clearly marked, so as not to confuse it with the
|
|
* version available from LANL.
|
|
*/
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <ctype.h>
|
|
|
|
char *BASE_PATH = "";
|
|
|
|
struct {
|
|
char *name;
|
|
size_t size;
|
|
} entries[] = {
|
|
{"name", 20},
|
|
{"author", 40},
|
|
{"color", 10},
|
|
{"program", 8192},
|
|
{NULL, 0}
|
|
};
|
|
|
|
|
|
size_t inlen;
|
|
|
|
int
|
|
read_char()
|
|
{
|
|
if (inlen) {
|
|
inlen -= 1;
|
|
return getchar();
|
|
}
|
|
return EOF;
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
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
|
|
read_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;
|
|
}
|
|
}
|
|
}
|
|
|
|
size_t
|
|
copy_item(char *filename, size_t maxlen)
|
|
{
|
|
FILE *f;
|
|
char path[132];
|
|
int c;
|
|
size_t pos = 0;
|
|
|
|
snprintf(path, sizeof(path),
|
|
"%s%05d.%s", BASE_PATH,
|
|
getpid(), filename);
|
|
f = fopen(path, "w");
|
|
if (! f) {
|
|
/* Just send it to the bit bucket */
|
|
maxlen = 0;
|
|
}
|
|
|
|
while (1) {
|
|
c = read_char();
|
|
switch (c) {
|
|
case EOF:
|
|
case '=':
|
|
case '&':
|
|
if (f) fclose(f);
|
|
return pos;
|
|
case '%':
|
|
c = read_hex();
|
|
break;
|
|
case '+':
|
|
c = ' ';
|
|
break;
|
|
}
|
|
if (pos < maxlen) {
|
|
fputc(c, f);
|
|
pos += 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
int
|
|
croak(char *msg)
|
|
{
|
|
int i;
|
|
char path[132];
|
|
|
|
for (i = 0; entries[i].name; i += 1) {
|
|
snprintf(path, sizeof(path),
|
|
"%s%05d.%s", BASE_PATH,
|
|
getpid(), entries[i].name);
|
|
unlink(path);
|
|
}
|
|
|
|
printf("Content-type: text/html\n\n");
|
|
printf("<html><head>\n");
|
|
printf("<link rel=\"stylesheet\" href=\"style.css\" type=\"text/css\">\n");
|
|
printf("<title>Tank submission error</title>\n");
|
|
printf("</head><body><h1>Tank submission error</h1>\n");
|
|
if (msg) {
|
|
printf("<p>%s.</p>\n", msg);
|
|
} else {
|
|
printf("<p>Something went wrong.</p>.\n");
|
|
}
|
|
printf("<p>Sorry it didn't work out.</p>\n");
|
|
printf("<p>You could go back and try again, though.</p>\n");
|
|
printf("</body></html>\n");
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
int sensor[10][4];
|
|
char key[20];
|
|
char token[40];
|
|
size_t len;
|
|
|
|
memset(sensor, 0, sizeof(sensor));
|
|
token[0] = '\0';
|
|
|
|
BASE_PATH = getenv("BASE_PATH");
|
|
if (! BASE_PATH) {
|
|
BASE_PATH = "";
|
|
}
|
|
|
|
{
|
|
char *rm = getenv("REQUEST_METHOD");
|
|
|
|
if (! (rm && (0 == strcmp(rm, "POST")))) {
|
|
printf("405 Method not allowed\n");
|
|
printf("Allow: POST\n");
|
|
printf("Content-type: text/html\n");
|
|
printf("\n");
|
|
printf("<h1>Method not allowed</h1>\n");
|
|
printf("<p>I only speak POST. Sorry.</p>\n");
|
|
return 0;
|
|
}
|
|
|
|
inlen = atoi(getenv("CONTENT_LENGTH"));
|
|
}
|
|
|
|
while (inlen) {
|
|
len = read_item(key, sizeof(key));
|
|
if (0 == strcmp(key, "token")) {
|
|
read_item(token, sizeof(token));
|
|
} else if ((3 == len) && ('s' == key[0])) {
|
|
/* sensor dealie, key = "s[0-9][rawt]" */
|
|
char val[5];
|
|
int n = key[1] - '0';
|
|
int i;
|
|
int p;
|
|
|
|
read_item(val, sizeof(val));
|
|
|
|
if (! (n >= 0) && (n <= 9)) {
|
|
break;
|
|
}
|
|
i = atoi(val);
|
|
|
|
switch (key[2]) {
|
|
case 'r':
|
|
p = 0;
|
|
break;
|
|
case 'a':
|
|
p = 1;
|
|
break;
|
|
case 'w':
|
|
p = 2;
|
|
break;
|
|
default:
|
|
p = 3;
|
|
i = (val[0] != '\0');
|
|
break;
|
|
}
|
|
|
|
sensor[n][p] = i;
|
|
} else {
|
|
int i;
|
|
|
|
for (i = 0; entries[i].name; i += 1) {
|
|
if (0 == strcmp(key, entries[i].name)) {
|
|
len = copy_item(key, entries[i].size);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Sanitize token */
|
|
{
|
|
char *p = token;
|
|
|
|
while (*p) {
|
|
if (! isalnum(*p)) {
|
|
*p = '_';
|
|
}
|
|
p += 1;
|
|
}
|
|
|
|
if ('\0' == token[0]) {
|
|
token[0] = '_';
|
|
token[1] = '\0';
|
|
}
|
|
}
|
|
|
|
/* Move files into their directory */
|
|
{
|
|
char path[132];
|
|
char dest[132];
|
|
struct stat st;
|
|
int i;
|
|
|
|
snprintf(path, sizeof(path), "%s%s/", BASE_PATH, token);
|
|
if (-1 == stat(path, &st)) return croak("Invalid token");
|
|
if (! S_ISDIR(st.st_mode)) return croak("Invalid token");
|
|
for (i = 0; entries[i].name; i += 1) {
|
|
snprintf(path, sizeof(path),
|
|
"%s%05d.%s", BASE_PATH,
|
|
getpid(), entries[i].name);
|
|
snprintf(dest, sizeof(dest),
|
|
"%s%s/%s", BASE_PATH,
|
|
token, entries[i].name);
|
|
rename(path, dest);
|
|
}
|
|
|
|
for (i = 0; i < 10; i += 1) {
|
|
FILE *f;
|
|
|
|
snprintf(dest, sizeof(dest),
|
|
"%s%s/sensor%d", BASE_PATH,
|
|
token, i);
|
|
f = fopen(dest, "w");
|
|
if (! f) break;
|
|
|
|
fprintf(f, "%d %d %d %d\n",
|
|
sensor[i][0],
|
|
sensor[i][1],
|
|
sensor[i][2],
|
|
sensor[i][3]);
|
|
fclose(f);
|
|
}
|
|
}
|
|
|
|
printf("Content-type: text/html\n\n");
|
|
printf("<!DOCTYPE html>\n");
|
|
printf("<html><head>\n");
|
|
printf("<link rel=\"stylesheet\" href=\"style.css\" type=\"text/css\">\n");
|
|
printf("<title>Tank submitted</title>\n");
|
|
printf("</head><body><h1>Tank submitted</h1>\n");
|
|
printf("<p>You just uploaded a tank!</p>\n");
|
|
printf("<p>Let's hope it doesn't suck.</p>\n");
|
|
printf("</body></html>\n");
|
|
|
|
return 0;
|
|
}
|