362 lines
7.7 KiB
C
362 lines
7.7 KiB
C
#if defined(ENABLE_WWW)
|
|
|
|
#include <microhttpd.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include "bbs.h"
|
|
|
|
extern struct bbs_config conf;
|
|
|
|
struct mime_type {
|
|
char *ext;
|
|
char *mime;
|
|
};
|
|
|
|
static struct mime_type **mime_types;
|
|
static int mime_types_count;
|
|
|
|
void www_init() {
|
|
FILE *fptr;
|
|
char buffer[4096];
|
|
int i;
|
|
|
|
mime_types_count = 0;
|
|
|
|
sprintf(buffer, "%s/mime.types", conf.www_path);
|
|
|
|
fptr = fopen(buffer, "r");
|
|
if (!fptr) {
|
|
return;
|
|
}
|
|
fgets(buffer, 4096, fptr);
|
|
while (!feof(fptr)) {
|
|
chomp(buffer);
|
|
|
|
for (i=0;i<strlen(buffer);i++) {
|
|
if (buffer[i] == ' ') {
|
|
buffer[i] = '\0';
|
|
if (mime_types_count == 0) {
|
|
mime_types = (struct mime_type **)malloc(sizeof(struct mime_type *));
|
|
} else {
|
|
mime_types = (struct mime_type **)realloc(mime_types, sizeof(struct mime_type *) * (mime_types_count + 1));
|
|
}
|
|
|
|
mime_types[mime_types_count] = (struct mime_type *)malloc(sizeof(struct mime_type));
|
|
|
|
mime_types[mime_types_count]->mime = strdup(buffer);
|
|
mime_types[mime_types_count]->ext = strdup(&buffer[i+1]);
|
|
|
|
mime_types_count++;
|
|
break;
|
|
}
|
|
}
|
|
|
|
fgets(buffer, 4096, fptr);
|
|
}
|
|
|
|
fclose(fptr);
|
|
}
|
|
|
|
char *www_get_mime_type(const char *extension) {
|
|
int i;
|
|
static char default_mime_type[] = "application/octet-stream";
|
|
|
|
|
|
for (i=0;i<mime_types_count;i++) {
|
|
if (strcasecmp(extension, mime_types[i]->ext) == 0) {
|
|
return mime_types[i]->mime;
|
|
}
|
|
}
|
|
return default_mime_type;
|
|
}
|
|
|
|
int www_404(char *header, char *footer, struct MHD_Connection * connection) {
|
|
char buffer[4096];
|
|
char *page;
|
|
struct stat s;
|
|
char *whole_page;
|
|
struct MHD_Response *response;
|
|
int ret;
|
|
FILE *fptr;
|
|
|
|
snprintf(buffer, 4096, "%s/404.tpl", conf.www_path);
|
|
|
|
page = NULL;
|
|
|
|
if (stat(buffer, &s) == 0) {
|
|
page = (char *)malloc(s.st_size + 1);
|
|
if (page == NULL) {
|
|
return -1;
|
|
}
|
|
memset(page, 0, s.st_size + 1);
|
|
fptr = fopen(buffer, "r");
|
|
if (fptr) {
|
|
fread(page, s.st_size, 1, fptr);
|
|
fclose(fptr);
|
|
} else {
|
|
free(page);
|
|
page = NULL;
|
|
}
|
|
}
|
|
|
|
if (page == NULL) {
|
|
page = (char *)malloc(16);
|
|
if (page == NULL) {
|
|
return -1;
|
|
}
|
|
sprintf(page, "Missing Content");
|
|
}
|
|
|
|
whole_page = (char *)malloc(strlen(header) + strlen(page) + strlen(footer) + 1);
|
|
|
|
sprintf(whole_page, "%s%s%s", header, page, footer);
|
|
|
|
response = MHD_create_response_from_buffer (strlen(whole_page), (void*)whole_page, MHD_RESPMEM_PERSISTENT);
|
|
|
|
ret = MHD_queue_response (connection, MHD_HTTP_NOT_FOUND, response);
|
|
MHD_destroy_response (response);
|
|
free(whole_page);
|
|
free(page);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int www_403(char *header, char *footer, struct MHD_Connection * connection) {
|
|
char buffer[4096];
|
|
char *page;
|
|
struct stat s;
|
|
char *whole_page;
|
|
struct MHD_Response *response;
|
|
int ret;
|
|
FILE *fptr;
|
|
|
|
snprintf(buffer, 4096, "%s/403.tpl", conf.www_path);
|
|
|
|
page = NULL;
|
|
|
|
if (stat(buffer, &s) == 0) {
|
|
page = (char *)malloc(s.st_size + 1);
|
|
if (page == NULL) {
|
|
return -1;
|
|
}
|
|
memset(page, 0, s.st_size + 1);
|
|
fptr = fopen(buffer, "r");
|
|
if (fptr) {
|
|
fread(page, s.st_size, 1, fptr);
|
|
fclose(fptr);
|
|
} else {
|
|
free(page);
|
|
page = NULL;
|
|
}
|
|
}
|
|
|
|
if (page == NULL) {
|
|
page = (char *)malloc(16);
|
|
if (page == NULL) {
|
|
return -1;
|
|
}
|
|
sprintf(page, "Missing Content");
|
|
}
|
|
|
|
whole_page = (char *)malloc(strlen(header) + strlen(page) + strlen(footer) + 1);
|
|
|
|
sprintf(whole_page, "%s%s%s", header, page, footer);
|
|
|
|
response = MHD_create_response_from_buffer (strlen(whole_page), (void*)whole_page, MHD_RESPMEM_PERSISTENT);
|
|
|
|
ret = MHD_queue_response (connection, MHD_HTTP_NOT_FOUND, response);
|
|
MHD_destroy_response (response);
|
|
free(whole_page);
|
|
free(page);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int www_handler(void * cls, struct MHD_Connection * connection, const char * url, const char * method, const char * version, const char * upload_data, size_t * upload_data_size, void ** ptr) {
|
|
struct MHD_Response *response;
|
|
|
|
int ret;
|
|
char *page;
|
|
char buffer[4096];
|
|
struct stat s;
|
|
char *header;
|
|
char *footer;
|
|
char *whole_page;
|
|
FILE *fptr;
|
|
char *mime;
|
|
int i;
|
|
int fno;
|
|
|
|
snprintf(buffer, 4096, "%s/header.tpl", conf.www_path);
|
|
|
|
header = NULL;
|
|
|
|
if (stat(buffer, &s) == 0) {
|
|
header = (char *)malloc(s.st_size + 1);
|
|
if (header == NULL) {
|
|
return MHD_NO;
|
|
}
|
|
memset(header, 0, s.st_size + 1);
|
|
fptr = fopen(buffer, "r");
|
|
if (fptr) {
|
|
fread(header, s.st_size, 1, fptr);
|
|
fclose(fptr);
|
|
} else {
|
|
free(header);
|
|
header = NULL;
|
|
}
|
|
}
|
|
|
|
if (header == NULL) {
|
|
header = (char *)malloc(strlen(conf.bbs_name) * 2 + 61);
|
|
if (header == NULL) {
|
|
return MHD_NO;
|
|
}
|
|
sprintf(header, "<HTML>\n<HEAD>\n<TITLE>%s</TITLE>\n</HEAD>\n<BODY>\n<H1>%s</H1><HR />", conf.bbs_name, conf.bbs_name);
|
|
}
|
|
|
|
snprintf(buffer, 4096, "%s/footer.tpl", conf.www_path);
|
|
|
|
footer = NULL;
|
|
|
|
if (stat(buffer, &s) == 0) {
|
|
footer = (char *)malloc(s.st_size + 1);
|
|
if (footer == NULL) {
|
|
free(header);
|
|
return MHD_NO;
|
|
}
|
|
memset(footer, 0, s.st_size + 1);
|
|
fptr = fopen(buffer, "r");
|
|
if (fptr) {
|
|
fread(footer, s.st_size, 1, fptr);
|
|
fclose(fptr);
|
|
} else {
|
|
free(footer);
|
|
footer = NULL;
|
|
}
|
|
}
|
|
|
|
if (footer == NULL) {
|
|
footer = (char *)malloc(43);
|
|
if (footer == NULL) {
|
|
free(header);
|
|
return MHD_NO;
|
|
}
|
|
sprintf(footer, "<HR />Powered by Magicka BBS</BODY></HTML>");
|
|
}
|
|
|
|
if (strcmp(method, "GET") == 0) {
|
|
if (strcasecmp(url, "/") == 0) {
|
|
|
|
snprintf(buffer, 4096, "%s/index.tpl", conf.www_path);
|
|
|
|
page = NULL;
|
|
|
|
if (stat(buffer, &s) == 0) {
|
|
page = (char *)malloc(s.st_size + 1);
|
|
if (page == NULL) {
|
|
free(header);
|
|
free(footer);
|
|
return MHD_NO;
|
|
}
|
|
memset(page, 0, s.st_size + 1);
|
|
fptr = fopen(buffer, "r");
|
|
if (fptr) {
|
|
fread(page, s.st_size, 1, fptr);
|
|
fclose(fptr);
|
|
} else {
|
|
free(page);
|
|
page = NULL;
|
|
}
|
|
}
|
|
|
|
if (page == NULL) {
|
|
page = (char *)malloc(16);
|
|
if (page == NULL) {
|
|
free(header);
|
|
free(footer);
|
|
return MHD_NO;
|
|
}
|
|
sprintf(page, "Missing Content");
|
|
}
|
|
|
|
whole_page = (char *)malloc(strlen(header) + strlen(page) + strlen(footer) + 1);
|
|
|
|
sprintf(whole_page, "%s%s%s", header, page, footer);
|
|
} else if (strncasecmp(url, "/static/", 8) == 0) {
|
|
// sanatize path
|
|
if (strstr(url, "/..") != NULL) {
|
|
return MHD_NO;
|
|
}
|
|
// get mimetype
|
|
for (i=strlen(url);i>0;--i) {
|
|
if (url[i] == '.') {
|
|
mime = www_get_mime_type(&url[i+1]);
|
|
break;
|
|
}
|
|
}
|
|
// load file
|
|
|
|
sprintf(buffer, "%s%s", conf.www_path, url);
|
|
if (stat(buffer, &s) == 0 && S_ISREG(s.st_mode)) {
|
|
fno = open(buffer, O_RDONLY);
|
|
if (fno != -1) {
|
|
response = MHD_create_response_from_fd(s.st_size, fno);
|
|
MHD_add_response_header(response, MHD_HTTP_HEADER_CONTENT_TYPE, mime);
|
|
ret = MHD_queue_response (connection, MHD_HTTP_OK, response);
|
|
MHD_destroy_response (response);
|
|
free(header);
|
|
free(footer);
|
|
return MHD_YES;
|
|
} else {
|
|
if (www_403(header, footer, connection) != 0) {
|
|
free(header);
|
|
free(footer);
|
|
return MHD_NO;
|
|
}
|
|
free(header);
|
|
free(footer);
|
|
return MHD_YES;
|
|
}
|
|
} else {
|
|
if (www_404(header, footer, connection) != 0) {
|
|
free(header);
|
|
free(footer);
|
|
return MHD_NO;
|
|
}
|
|
free(header);
|
|
free(footer);
|
|
return MHD_YES;
|
|
}
|
|
|
|
} else {
|
|
if (www_404(header, footer, connection) != 0) {
|
|
free(header);
|
|
free(footer);
|
|
return MHD_NO;
|
|
}
|
|
free(header);
|
|
free(footer);
|
|
return MHD_YES;
|
|
}
|
|
} else if (strcmp(method, "POST") == 0) {
|
|
free(header);
|
|
free(footer);
|
|
return MHD_NO;
|
|
}
|
|
|
|
response = MHD_create_response_from_buffer (strlen (whole_page), (void*) whole_page, MHD_RESPMEM_PERSISTENT);
|
|
|
|
ret = MHD_queue_response (connection, MHD_HTTP_OK, response);
|
|
MHD_destroy_response (response);
|
|
free(whole_page);
|
|
free(page);
|
|
free(header);
|
|
free(footer);
|
|
|
|
return ret;
|
|
}
|
|
#endif
|