/* $Id: config.c 349 2024-10-15 20:08:08Z nishi $ */

#define SOURCE

#include "../config.h"

#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>

#if !defined(_MSC_VER) && !defined(__BORLANDC__)
#include <unistd.h>
#endif

#if defined(__MINGW32__) || defined(_MSC_VER) || defined(__BORLANDC__) || (defined(__WATCOMC__) && !defined(__OS2__) && !defined(__NETWARE__))
#ifdef USE_WINSOCK1
#include <winsock.h>
#else
#include <winsock2.h>
#endif
#endif

#include <cm_string.h>
#include <cm_log.h>

#ifdef __OS2__
#include <types.h>
#include <netinet/in.h>
#include <tcpustd.h>
#endif

#ifdef __NETWARE__
#include <sys/socket.h>
#endif
#include "tw_config.h"
#include "tw_module.h"

struct tw_config config;

struct tw_config_entry* tw_vhost_match(const char* name, int port) {
	int i;
	for(i = 0; i < config.vhost_count; i++) {
		if(strcmp(config.vhosts[i].name, name) == 0 && (config.vhosts[i].port == -1 ? 1 : config.vhosts[i].port == port)) {
			return &config.vhosts[i];
		}
	}
	return &config.root;
}

bool tw_permission_allowed(const char* path, SOCKADDR addr, struct tw_http_request req, struct tw_config_entry* vhost) {
	int i;
	bool found = false;
	bool pathstart = false;
	bool perm = false;
again:
	for(i = 0; i < vhost->dir_count; i++) {
		char* noslash;
		struct tw_dir_entry* e = &vhost->dirs[i];
		pathstart = false;
		if(strlen(path) >= strlen(e->dir)) {
			int j;
			pathstart = true;
			for(j = 0; path[j] != 0 && e->dir[j] != 0; j++) {
				if(path[j] != e->dir[j]) {
					pathstart = false;
					break;
				}
			}
		}
		noslash = cm_strdup(e->dir);
		noslash[strlen(noslash) - 1] = 0;
		if(strcmp(e->dir, path) == 0 || strcmp(noslash, path) == 0 || pathstart) {
			found = true;
			if(strcmp(e->name, "all") == 0) {
				perm = e->type == TW_DIR_ALLOW;
			}
		}
		free(noslash);
	}
	if(!found && vhost != &config.root) {
		vhost = &config.root;
		goto again;
	}
	return perm;
}

void tw_config_init(void) {
	int i;
	for(i = 0; i < MAX_PORTS + 1; i++) {
		config.ports[i] = -1;
	}
	for(i = 0; i < MAX_VHOSTS; i++) {
#ifndef NO_SSL
		config.vhosts[i].sslkey = NULL;
		config.vhosts[i].sslcert = NULL;
#endif
		config.vhosts[i].root = NULL;
#ifdef HAS_CHROOT
		config.vhosts[i].chroot_path = NULL;
#endif
	}
#ifndef NO_SSL
	config.root.sslkey = NULL;
	config.root.sslcert = NULL;
#endif
	config.root.root = NULL;
	config.root.mime_count = 0;
	config.root.dir_count = 0;
	config.root.icon_count = 0;
	config.root.index_count = 0;
	config.root.readme_count = 0;
	config.root.hideport = 0;
#ifdef HAS_CHROOT
	config.root.chroot_path = NULL;
#endif
	config.vhost_count = 0;
	config.module_count = 0;
	config.extension = NULL;
	config.server_root = cm_strdup(PREFIX);
	config.server_admin = cm_strdup(SERVER_ADMIN);
	config.defined[0] = NULL;
#if defined(_PSP)
	strcpy(config.hostname, "psp");
#elif defined(__PPU__)
	strcpy(config.hostname, "ps3");
#elif defined(__ps2sdk__)
	strcpy(config.hostname, "ps2");
#elif defined(__NETWARE__)
	strcpy(config.hostname, "netware");
#else
	gethostname(config.hostname, 1024);
#endif
#ifdef HAS_CHROOT
	tw_add_define("HAS_CHROOT");
#endif
#ifndef NO_SSL
	tw_add_define("HAS_SSL");
#endif
}

int tw_config_read(const char* path) {
	char cbuf[2];
	int ln = 0;
	int ifbr = 0;
	int ignore = -1;
	int portcount;
	FILE* f;
	cm_log("Config", "Reading %s", path);
	f = fopen(path, "r");
	cbuf[1] = 0;
	if(f != NULL) {
		char* line = malloc(1);
		int stop = 0;
		struct tw_config_entry* current = &config.root;
		char* vhost = NULL;
		char* dir = NULL;
		line[0] = 0;
		while(stop == 0) {
			int c = fread(cbuf, 1, 1, f);
			if(cbuf[0] == '\n' || c <= 0) {
				char* l = cm_trim(line);
				ln++;
				if(strlen(l) > 0 && l[0] != '#') {
					char** r = cm_split(l, " \t");
					int i;
					if(ignore != -1 && ifbr >= ignore) {
						if(cm_strcaseequ(r[0], "EndIf")) ifbr--;
						if(ifbr == 0) {
							ignore = -1;
						}
					} else if(cm_strcaseequ(r[0], "Include") || cm_strcaseequ(r[0], "IncludeOptional")) {
						for(i = 1; r[i] != NULL; i++) {
							if(tw_config_read(r[i]) != 0 && cm_strcaseequ(r[0], "Include")) {
								stop = 1;
								break;
							}
						}
					} else if(cm_strcaseequ(r[0], "Define")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing name at line %d", ln);
							stop = 1;
						} else {
							tw_add_define(r[1]);
						}
					} else if(cm_strcaseequ(r[0], "Undefine")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing name at line %d", ln);
							stop = 1;
						} else {
							tw_delete_define(r[1]);
						}
					} else if(cm_strcaseequ(r[0], "BeginDirectory")) {
						if(dir != NULL) {
							cm_log("Config", "Already in directory section at line %d", ln);
							stop = 1;
						} else {
							if(r[1] == NULL) {
								cm_log("Config", "Missing directory at line %d", ln);
								stop = 1;
							} else {
								dir = cm_strcat(r[1], r[1][strlen(r[1]) - 1] == '/' ? "" : "/");
							}
						}
					} else if(cm_strcaseequ(r[0], "EndDirectory")) {
						if(dir == NULL) {
							cm_log("Config", "Not in directory section at line %d", ln);
							stop = 1;
						} else {
							free(dir);
							dir = NULL;
						}
					} else if(cm_strcaseequ(r[0], "Allow")) {
						if(dir == NULL) {
							cm_log("Config", "Not in directory section at line %d", ln);
							stop = 1;
						} else {
							if(r[1] == NULL) {
								cm_log("Config", "Missing argument at line %d", ln);
								stop = 1;
							} else {
								struct tw_dir_entry* e = &current->dirs[current->dir_count++];
								e->name = cm_strdup(r[1]);
								e->dir = cm_strdup(dir);
								e->type = TW_DIR_ALLOW;
							}
						}
					} else if(cm_strcaseequ(r[0], "Deny")) {
						if(dir == NULL) {
							cm_log("Config", "Not in directory section at line %d", ln);
							stop = 1;
						} else {
							if(r[1] == NULL) {
								cm_log("Config", "Missing argument at line %d", ln);
								stop = 1;
							} else {
								struct tw_dir_entry* e = &current->dirs[current->dir_count++];
								e->name = cm_strdup(r[1]);
								e->dir = cm_strdup(dir);
								e->type = TW_DIR_DENY;
							}
						}
					} else if(cm_strcaseequ(r[0], "BeginVirtualHost")) {
						if(vhost != NULL) {
							cm_log("Config", "Already in virtual host section at line %d", ln);
							stop = 1;
						} else {
							if(r[1] == NULL) {
								cm_log("Config", "Missing virtual host at line %d", ln);
								stop = 1;
							} else {
								int i;
								vhost = cm_strdup(r[1]);
								current = &config.vhosts[config.vhost_count++];
								current->dir_count = 0;
								current->mime_count = 0;
								current->icon_count = 0;
								current->index_count = 0;
								current->readme_count = 0;
								current->hideport = -1;
								current->name = cm_strdup(vhost);
								current->port = -1;
								for(i = 0; vhost[i] != 0; i++) {
									if(vhost[i] == ':') {
										current->name[i] = 0;
										current->port = atoi(current->name + i + 1);
										break;
									}
								}
							}
						}
					} else if(cm_strcaseequ(r[0], "EndVirtualHost")) {
						if(vhost == NULL) {
							cm_log("Config", "Not in virtual host section at line %d", ln);
							stop = 1;
						} else {
							free(vhost);
							vhost = NULL;
							current = &config.root;
						}
					} else if(cm_strcaseequ(r[0], "Listen")
#ifndef NO_SSL
						  || cm_strcaseequ(r[0], "ListenSSL")
#endif
					) {
						for(i = 1; r[i] != NULL; i++) {
#if defined(_MSC_VER) || defined(__BORLANDC__)
							uint32_t port = atoi(r[i]);
#else
							uint64_t port = atoi(r[i]);
#endif
							int j;
							cm_log("Config", "Going to listen at port %d%s", (int)port, cm_strcaseequ(r[0], "ListenSSL") ? " with SSL" : "");
#if defined(_MSC_VER) || defined(__BORLANDC__)
							if(cm_strcaseequ(r[0], "ListenSSL")) port |= (1UL << 31);
#else
							if(cm_strcaseequ(r[0], "ListenSSL")) port |= (1ULL << 31);
#endif
							for(j = 0; config.ports[j] != -1; j++)
								;
							config.ports[j] = port;
						}
					} else if(cm_strcaseequ(r[0], "HidePort")) {
						current->hideport = 1;
					} else if(cm_strcaseequ(r[0], "ShowPort")) {
						current->hideport = 0;
#ifndef NO_SSL
					} else if(cm_strcaseequ(r[0], "SSLKey")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing path at line %d", ln);
							stop = 1;
						} else {
							if(current->sslkey != NULL) free(current->sslkey);
							current->sslkey = cm_strdup(r[1]);
						}
					} else if(cm_strcaseequ(r[0], "SSLCertificate")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing path at line %d", ln);
							stop = 1;
						} else {
							if(current->sslcert != NULL) free(current->sslcert);
							current->sslcert = cm_strdup(r[1]);
						}
#endif
#ifdef HAS_CHROOT
					} else if(cm_strcaseequ(r[0], "ChrootDirectory")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing path at line %d", ln);
							stop = 1;
						} else {
							if(current->chroot_path != NULL) free(current->chroot_path);
							current->chroot_path = cm_strdup(r[1]);
						}
#endif
					} else if(cm_strcaseequ(r[0], "ForceLog")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing log at line %d", ln);
							stop = 1;
						} else {
							cm_force_log(r[1]);
						}
					} else if(cm_strcaseequ(r[0], "EndIf")) {
						if(ifbr == 0) {
							cm_log("Config", "Missing BeginIf at line %d", ln);
							stop = 1;
						}
						ifbr--;
					} else if(cm_strcaseequ(r[0], "BeginIf") || cm_strcaseequ(r[0], "BeginIfNot")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing condition type at line %d", ln);
						} else {
							bool ign = false;
							ifbr++;
							if(cm_strcaseequ(r[1], "False")) {
								ign = true;
							} else if(cm_strcaseequ(r[1], "True")) {
							} else if(cm_strcaseequ(r[1], "Defined")) {
								if(r[2] == NULL) {
									cm_log("Config", "Missing name at line %d", ln);
									stop = 1;
								} else {
									int i;
									bool fndit = false;
									for(i = 0; config.defined[i] != NULL; i++) {
										if(strcmp(config.defined[i], r[2]) == 0) {
											fndit = true;
											break;
										}
									}
									if(!fndit) {
										ign = true;
									}
								}
							} else {
								cm_log("Config", "Unknown condition type at line %d", ln);
								stop = 1;
							}
							if(cm_strcaseequ(r[0], "BeginIfNot")) ign = !ign;
							if(ign) {
								ignore = ifbr - 1;
							}
						}
					} else if(cm_strcaseequ(r[0], "ServerRoot")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing path at line %d", ln);
							stop = 1;
						} else {
							chdir(r[1]);
							free(config.server_root);
							config.server_root = cm_strdup(r[1]);
						}
					} else if(cm_strcaseequ(r[0], "ServerAdmin")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing email at line %d", ln);
							stop = 1;
						} else {
							free(config.server_admin);
							config.server_admin = cm_strdup(r[1]);
						}
					} else if(cm_strcaseequ(r[0], "DocumentRoot")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing path at line %d", ln);
							stop = 1;
						} else {
							if(current->root != NULL) free(current->root);
							current->root = cm_strdup(strcmp(r[1], "/") == 0 ? "" : r[1]);
						}
					} else if(cm_strcaseequ(r[0], "MIMEType")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing extension at line %d", ln);
							stop = 1;
						} else if(r[2] == NULL) {
							cm_log("Config", "Missing MIME at line %d", ln);
							stop = 1;
						} else {
							struct tw_mime_entry* e = &current->mimes[current->mime_count++];
							e->ext = cm_strdup(r[1]);
							e->mime = cm_strdup(r[2]);
						}
					} else if(cm_strcaseequ(r[0], "MIMEFile")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing path at line %d", ln);
							stop = 1;
						} else {
							FILE* mimefile = fopen(r[1], "r");
							if(mimefile == NULL) {
								cm_log("Config", "Could not load the file at line %d", ln);
								stop = 1;
							} else {
								char* line = malloc(1);
								int i;
								struct stat st;
								char* buf;
								int incr = 0;
								stat(r[1], &st);

								buf = malloc(st.st_size + 1);
								fread(buf, st.st_size, 1, mimefile);

								for(i = 0;; i++) {
									if(buf[i] == '\n' || buf[i] == 0) {
										char oldc = buf[i];
										char* line;
										buf[i] = 0;

										line = buf + incr;

										if(strlen(line) > 0 && line[0] != '#') {
											int j;
											for(j = 0; line[j] != 0; j++) {
												if(line[j] == ' ' || line[j] == '\t') {
													line[j] = 0;
													j++;
													for(; line[j] != 0; j++) {
														if(line[j] != ' ' && line[j] != '\t') {
															char* name = line;
															char* mimes = line + j;
															int k = 0;
															int incr2 = 0;
															for(k = 0;; k++) {
																if(mimes[k] == ' ' || mimes[k] == 0) {
																	char oldc2 = mimes[k];
																	struct tw_mime_entry* e;
																	mimes[k] = 0;

																	e = &current->mimes[current->mime_count++];
																	e->ext = cm_strcat(".", mimes + incr2);
																	e->mime = cm_strdup(name);
																	if(current->mime_count == MAX_MIME) {
																		cm_log("Config", "Too many MIME types, cannot handle");
																		stop = 1;
																		break;
																	}

																	incr2 = k + 1;
																	if(oldc2 == 0) break;
																}
															}
															break;
														}
													}
													break;
												}
												if(stop) break;
											}
										}

										incr = i + 1;
										if(oldc == 0) break;
									}
								}

								free(buf);

								fclose(mimefile);
							}
						}
					} else if(cm_strcaseequ(r[0], "Icon")) {
						if(r[1] == NULL) {
							cm_log("Config", "Missing MIME at line %d", ln);
							stop = 1;
						} else if(r[2] == NULL) {
							cm_log("Config", "Missing path at line %d", ln);
							stop = 1;
						} else {
							struct tw_icon_entry* e = &current->icons[current->icon_count++];
							e->mime = cm_strdup(r[1]);
							e->icon = cm_strdup(r[2]);
						}
					} else if(cm_strcaseequ(r[0], "LoadModule")) {
						for(i = 1; r[i] != NULL; i++) {
							void* mod = tw_module_load(r[i]);
							if(mod != NULL) {
								config.modules[config.module_count++] = mod;
								if(tw_module_init(mod) != 0) {
									stop = 1;
									break;
								}
							} else {
								cm_log("Config", "Could not load the module at line %d", ln);
								stop = 1;
								break;
							}
						}
					} else if(cm_strcaseequ(r[0], "DirectoryIndex")) {
						for(i = 1; r[i] != NULL; i++) {
							current->indexes[current->index_count++] = cm_strdup(r[i]);
						}
					} else if(cm_strcaseequ(r[0], "ReadmeFile") || cm_strcaseequ(r[0], "Readme")) {
						if(cm_strcaseequ(r[0], "Readme")) {
							cm_force_log("NOTE: Readme directive is deprecated.");
						}
						for(i = 1; r[i] != NULL; i++) {
							current->readmes[current->readme_count++] = cm_strdup(r[i]);
						}
					} else {
						stop = 1;
						if(r[0] != NULL) {
							int argc;
							int i;
							bool called = false;
							struct tw_tool tools;
							for(argc = 0; r[argc] != NULL; argc++)
								;
							stop = 0;
							tw_init_tools(&tools);
							for(i = 0; i < config.module_count; i++) {
								tw_mod_config_t mod_config = (tw_mod_config_t)tw_module_symbol(config.modules[i], "mod_config");
								int resp;
								if(mod_config != NULL && (resp = mod_config(&tools, r, argc)) == TW_CONFIG_PARSED) {
									called = true;
									break;
								}
								if(resp == TW_CONFIG_ERROR) {
									stop = 1;
									called = true;
									break;
								}
							}
							if(!called) {
								cm_log("Config", "Unknown directive `%s' at line %d", r[0], ln);
								stop = 1;
							}
						}
					}
					for(i = 0; r[i] != NULL; i++) free(r[i]);
					free(r);
				}
				free(l);
				free(line);
				line = malloc(1);
				line[0] = 0;
				if(c <= 0) break;
			} else if(cbuf[0] != '\r') {
				char* tmp = line;
				line = cm_strcat(tmp, cbuf);
				free(tmp);
			}
		}
		free(line);
		fclose(f);
		for(portcount = 0; config.ports[portcount] != -1; portcount++)
			;
		if(portcount == 0) {
			return 1;
		} else {
			return stop;
		}
	} else {
		cm_log("Config", "Could not open the file");
		return 1;
	}
}