Theta/modules/webio.c
2019-11-12 19:26:18 +01:00

356 lines
12 KiB
C

//
// Készítette: Toldi Balázs Ádám
// Dátum: 2019. 10. 14..
//
#include "webio.h"
int webio_create(Config config, struct Node_data myData, WebIO *webIo) {
char *port = map_getValue(config, "interface-port");
if (port == NULL)
port = DEFAULT_INTERFACE_PORT;
char *folder = map_getValue(config, "interface-folder");
if (folder == NULL)
folder = DEFAULT_WWW_FOLDER;
char *local_mode_str = map_getValue(config, "interface-local");
bool local_mode = false;
if (local_mode_str != NULL && strcmp(local_mode_str, "true") == 0)
local_mode = true;
struct addrinfo *result = NULL;
SOCKET listening;
result = tcp_createIPv4Socket(&listening, atoi(port), !local_mode);
if (result == NULL) {
return 1;
}
int res = tcp_bindnlisten(listening, result, SOMAXCONN);
if (res != 0) {
return 2;
}
WebIO wio;
wio.sockaddr = tcp_getAddr_in(listening);
wio.socket = listening;
wio.nodeData = myData;
strcpy(wio.folder, folder);
*webIo = wio;
return 0;
}
int webio_handleRequest(WebIO wio, const PeerList *list) {
SOCKET client = accept(wio.socket, NULL, NULL);
char buf[8192];
memset(buf, 0, 8192);
int res = recv(client, buf, 8192, 0);
if (res <= 0) {
logger_log("Error with web interface!");
closesocket(client);
return -1;
}
if (strncmp(buf, "GET",3) == 0) {
char file[50];
sscanf(buf, "%*s %s", file);
res = webio_handleGETrequest(client, wio, file, list);
} else if (strncmp(buf, "POST",4) == 0) {
int i = strlen(buf) - 1;
while (buf[i] != '\n') {
i--;
}
i++;
char tmp[8196];
strcpy(tmp, buf + i);
Map post = getPacketData(tmp);
res = webio_handlePOSTrequest(client, wio, list, post);
free(post.pairs);
} else
res = -1;
return res;
}
char *webio_getMIMEtype(char *filename) {
char type[10];
char *ext = webio_getFiletype(filename);
strcpy(type, ext);
free(ext);
if (strcmp(type, "html") == 0)
return "text/html";
else if (strcmp(type, "json") == 0)
return "application/json";
else if (strcmp(type, "svg") == 0)
return "image/svg+xml";
else if (strcmp(type, "js") == 0)
return "application/javascript";
else if (strcmp(type, "css") == 0)
return "text/css";
else if (strcmp(type, "jpg") == 0)
return "image/jpeg";
else if (strcmp(type, "png") == 0)
return "image/png";
else if (strcmp(type, "bmp") == 0)
return "image/bmp";
else if (strcmp(type, "ico") == 0)
return "image/x-icon";
else if (strcmp(type, "xml") == 0)
return "text/xml";
else if (strcmp(type, "zip") == 0)
return "application/zip";
else return "text/plain";
}
char *webio_getFiletype(char *filename) {
char *ext = (char *) malloc(sizeof(char) * 10);
char *tmp = strtok(filename, ".");
while (tmp != NULL) {
strncpy(ext, tmp, 10);
tmp = strtok(NULL, ".");
}
return ext;
}
static int webio_handleGETrequest(SOCKET client, WebIO wio, char *file, const PeerList *list) {
char path[129];
memset(path, 0, sizeof(path));
strcat(path, wio.folder);
int len = 0;
if (file[0] == '/')
memmove(file, file + 1, strlen(file));
if (strcmp(file, "") == 0) {
char index[8192] = "";
webio_getIndex(wio.folder, list, index);
webio_sendPage(client, index);
} else if (peer_ID_isFound(*list, file) || webio_isPeerFound(wio.folder, file)) {
char cnt[8192] = "";
webio_getPeerPage(wio.folder, file, (peer_ID_getPeer(*list, file) != -1), cnt);
webio_sendPage(client, cnt);
} else if (strcmp(file, "kill") == 0) {
webio_sendPage(client, "<h1>Server Killed!</h1>");
logger_log("Exit sign received.Exiting...");
return -2;
} else {
strcat(path, file);
//File küldés windows-on
#ifdef _WIN32
HANDLE file = CreateFileA(path,GENERIC_READ,FILE_SHARE_READ,
NULL,
OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL,
NULL);
DWORD size = GetFileSize(file,NULL);
if(file == INVALID_HANDLE_VALUE || size< 0)
webio_send404Page(client);
else {
webio_sendOKHeader_wSize(client,path,size);
TransmitFile(client, file, 0, 0, NULL, NULL, 0);
CloseHandle(file);
}
#else
logger_log(path);
int fd = open(path, O_RDONLY);
if (fd == -1)
webio_send404Page(client);
else {
struct stat stat_struct;
fstat(fd, &stat_struct);
int length = stat_struct.st_size;
webio_sendOKHeader(client, path);
size_t total_bytes_sent = 0;
ssize_t bytes_sent;
char puf[len];
while (total_bytes_sent < length) {
bytes_sent = sendfile(client,fd,0,length);
if(bytes_sent == -1) logger_log("Error: %s",strerror(errno));
total_bytes_sent += bytes_sent;
}
}
#endif
shutdown(client,SD_BOTH);
}
closesocket(client);
return 0;
}
static int webio_handlePOSTrequest(SOCKET client, WebIO wio, const PeerList *list, Map post) {
shutdown(client, SD_RECEIVE);
char *response = "HTTP/1.1 304 Not Modified ";
int res = send(client, response, strlen(response), 0);
if (res == SOCKET_ERROR) {
logger_log("Error with io");
return -1;
}
shutdown(client, SD_BOTH);
if (map_isFound(post, "id") && map_isFound(post, "message") &&
strcmp(map_getValue(post, "message"), "%0D%0A") != 0) {
char file[64];
char folder[72];
sprintf(folder, "%s/peers/", wio.folder);
#if defined(_WIN32)
mkdir(folder);
#else
mkdir(folder, 0777); // notice that 777 is different than 0777
#endif
sprintf(file, "%s%s.txt", folder, map_getValue(post, "id"));
FILE *f;
f = fopen(file, "a");
fprintf(f, "Me: %s\n", map_getValue(post, "message"));
fclose(f);
char buf[DEFAULT_BUFLEN];
int i = peer_ID_getPeer(*list, map_getValue(post, "id"));
if (i < 0) {
logger_log("Error sending message! Error: Peer offline!");
return 1;
}
sprintf(buf, "@message=%s", map_getValue(post, "message"));
res = send(list->array[i].socket, buf, DEFAULT_BUFLEN, 0);
if (res == SOCKET_ERROR) {
logger_log("Error sending message.Error: %d", errno);
return 2;
}
logger_log("Message sent to %s", map_getValue(post, "id"));
} else map_dump(post);
return 0;
}
static void webio_getHeader(char *folder, char result[]) {
char path[65];
strcpy(path, folder);
strcat(path, "/header.html");
FILE *fp;
fp = fopen(path, "r");
if (fp != NULL) {
char buf[1024];
while (fgets(buf, 1024, fp) != NULL) {
strcat(result, buf);
}
fclose(fp);
} else
result = "<html>";
}
static void webio_getIndex(char *folder, const PeerList *list, char *outputBuffer) {
char content[8192] = "";
char header[4096] = "";
webio_getHeader(folder, header);
strcpy(content, header);
strcat(content, "<h1>Peers:</h1>\n");
if (list->length > 0) {
strcat(content, "<ul>\n");
for (int i = 0; i < list->length; ++i) {
sprintf(content, "%s<li>"
"<a href=\"%s\">",
content, list->array[i].peerData.id);
if (strcmp(list->array[i].peerData.nick, "") != 0) {
sprintf(content, "%s%s %d- ", content, list->array[i].peerData.nick,
(int) strlen(list->array[i].peerData.nick));
}
sprintf(content, "%s%s</a></li>\n", content, list->array[i].peerData.id);
}
strcat(content, "</ul>\n");
} else
sprintf(content, "%s<div class=\"alert alert-warning\" role=\"alert\">\n"
" No peers connected!\n"
"</div>\n", content);
sprintf(content, "%s<script>setTimeout(function(){\n"
" window.location.reload(1);\n"
"}, 5000);</script>\n"
"</div>\n"
"</body>\n"
"</html>\n", content);
strcpy(outputBuffer, content);
}
static void webio_getPeerPage(char *folder, char *id, bool online, char *outputBuffer) {
char content[8192] = "";
char header[4096] = "";
webio_getHeader(folder, header);
strcpy(content, header);
char *img = (online) ? "<img width=\"30\" height=\"30\" src=\"assets\\img\\on.svg\">"
: "<img width=\"30\" height=\"30\" src=\"assets\\img\\off.svg\">";
char *disabled = (online) ? "" : "disabled";
sprintf(content, "%s\n"
"<h1>%s%s</h1>\n"
"<div id=\"msgs\" style=\"margin-bottom:5em\"></div>\n"
"<div id=\"end\"></div>\n"
" <form name=\"sendmsg\" class=\"form-inline\" style=\"margin: 7px;padding: 7px;position: fixed;bottom: 0;width: 100%%;\">"
"<textarea name=\"message\" id=\"message\" class=\"form-control\" style=\"width: 90%%;display: block;\" %s></textarea>"
"<input name=\"id\"class=\"form-control\" type=\"text\" style=\"display:none;\" value=\"%s\">"
"<input onclick=\"submitForm();\"name=\"s\" id=\"s\" type=\"button\" class=\"btn btn-primary\" style=\"margin: 7px;padding: 7px;display:inline;\" value=\"Send\" /></form>\n"
"<script src=\"assets/js/chat.js\"></script>"
"</body>\n"
"\n"
"</html>", header, img, id, disabled, id);
strcpy(outputBuffer, content);
}
static bool webio_isPeerFound(char *folder, char *id) {
char file[129];
sprintf(file, "%s/peers/%s.txt", folder, id);
FILE *F = fopen(file, "r");
if (F == NULL) {
return false;
} else
return true;
}
void webio_sendOKHeader(SOCKET socket, char *file) {
char response[8192];
sprintf(response, "HTTP/1.1 200 OK "
"Content-Encoding: gzip\r\n"
"Content-Language: en\r\n"
"Content-Type: %s\r\n\r\n", webio_getMIMEtype(file));
int res = send(socket, response, strlen(response), 0);
if (res == SOCKET_ERROR) {
logger_log("Error sending http ok header!");
}
}
void webio_sendOKHeader_wSize(SOCKET socket, char *file,int size) {
char response[8192];
sprintf(response, "HTTP/1.1 200 OK "
"Content-Encoding: gzip\r\n"
"Content-Language: en\r\n"
"Content-Length: %d\r\n"
"Content-Type: %s\r\n\r\n",size, webio_getMIMEtype(file));
int res = send(socket, response, strlen(response), 0);
if (res == SOCKET_ERROR) {
logger_log("Error sending http ok header!");
}
}
void webio_send404Page(SOCKET socket) {
char *response = "HTTP/1.1 404 Not Found "
"Content-Encoding: gzip\r\n"
"Content-Language: en\r\n"
"Content-Type: text/html\r\n\r\n"
"<h1>Error 404 File not found!</h1>";
int res = send(socket, response, (int)strlen(response), 0);
if (res == SOCKET_ERROR) {
logger_log("Error sending 404 page!");
}
}
void webio_sendPage(SOCKET socket, char *content) {
char f[] = "index.html";
webio_sendOKHeader(socket, f);
int res = send(socket, content,(int) strlen(content), 0);
if (res == SOCKET_ERROR) {
logger_log("Error sending page!");
}
shutdown(socket,SD_BOTH);
}