1 files changed, 596 insertions, 0 deletions
diff --git a/src/engine/e_engine.cpp b/src/engine/e_engine.cpp
new file mode 100644
index 00000000..4475478a
--- /dev/null
+++ b/src/engine/e_engine.cpp
@@ -0,0 +1,596 @@
+/* copyright (c) 2007 magnus auvinen, see licence.txt for more info */
+#include <string.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+#include <base/system.h>
+
+#include <engine/e_server_interface.h>
+#include <engine/e_config.h>
+#include <engine/e_console.h>
+#include <engine/e_engine.h>
+#include <engine/e_network.h>
+#include "e_linereader.h"
+
+/* compiled-in data-dir path */
+#define DATA_DIR "data"
+
+static JOBPOOL hostlookuppool;
+static int engine_find_datadir(char *argv0);
+
+static void con_dbg_dumpmem(void *result, void *user_data)
+{
+	mem_debug_dump();
+}
+
+static void con_dbg_lognetwork(void *result, void *user_data)
+{
+	CNetBase::OpenLog("network_sent.dat", "network_recv.dat");
+}
+
+
+static char application_save_path[512] = {0};
+static char datadir[512] = {0};
+
+const char *engine_savepath(const char *filename, char *buffer, int max)
+{
+	str_format(buffer, max, "%s/%s", application_save_path, filename);
+	return buffer;
+}
+
+void engine_init(const char *appname)
+{
+	dbg_logger_stdout();
+	dbg_logger_debugger();
+	
+	/* */
+	dbg_msg("engine", "running on %s-%s-%s", CONF_FAMILY_STRING, CONF_PLATFORM_STRING, CONF_ARCH_STRING);
+#ifdef CONF_ARCH_ENDIAN_LITTLE
+	dbg_msg("engine", "arch is little endian");
+#elif defined(CONF_ARCH_ENDIAN_BIG)
+	dbg_msg("engine", "arch is big endian");
+#else
+	dbg_msg("engine", "unknown endian");
+#endif
+
+	/* init the network */
+	net_init();
+	CNetBase::Init();
+	
+	/* create storage location */
+	{
+		char path[1024] = {0};
+		fs_storage_path(appname, application_save_path, sizeof(application_save_path));
+		if(fs_makedir(application_save_path) == 0)
+		{		
+			str_format(path, sizeof(path), "%s/screenshots", application_save_path);
+			fs_makedir(path);
+
+			str_format(path, sizeof(path), "%s/maps", application_save_path);
+			fs_makedir(path);
+
+			str_format(path, sizeof(path), "%s/downloadedmaps", application_save_path);
+			fs_makedir(path);
+
+			str_format(path, sizeof(path), "%s/demos", application_save_path);
+			fs_makedir(path);
+		}
+	}
+
+	/* init console and add the console logger */
+	console_init();
+	dbg_logger(console_print);
+	
+	jobs_initpool(&hostlookuppool, 1);
+
+	MACRO_REGISTER_COMMAND("dbg_dumpmem", "", CFGFLAG_SERVER|CFGFLAG_CLIENT, con_dbg_dumpmem, 0x0, "Dump the memory");
+	MACRO_REGISTER_COMMAND("dbg_lognetwork", "", CFGFLAG_SERVER|CFGFLAG_CLIENT, con_dbg_lognetwork, 0x0, "Log the network");
+	
+	/* reset the config */
+	config_reset();
+}
+
+
+void engine_listdir(int types, const char *path, FS_LISTDIR_CALLBACK cb, void *user)
+{
+	char buffer[1024];
+	
+	/* list current directory */
+	if(types&LISTDIRTYPE_CURRENT)
+	{
+		fs_listdir(path, cb, user);
+	}
+	
+	/* list users directory */
+	if(types&LISTDIRTYPE_SAVE)
+	{
+		engine_savepath(path, buffer, sizeof(buffer));
+		fs_listdir(buffer, cb, user);
+	}
+	
+	/* list datadir directory */
+	if(types&LISTDIRTYPE_DATA)
+	{
+		str_format(buffer, sizeof(buffer), "%s/%s", datadir, path);
+		fs_listdir(buffer, cb, user);
+	}
+}
+
+void engine_getpath(char *buffer, int buffer_size, const char *filename, int flags)
+{
+	if(flags&IOFLAG_WRITE)
+		engine_savepath(filename, buffer, buffer_size);
+	else
+	{
+		IOHANDLE handle = 0;
+		
+		/* check current directory */
+		handle = io_open(filename, flags);
+		if(handle)
+		{
+			str_copy(buffer, filename, buffer_size);
+			io_close(handle);
+			return;
+		}
+			
+		/* check user directory */
+		engine_savepath(filename, buffer, buffer_size);
+		handle = io_open(buffer, flags);
+		if(handle)
+		{
+			io_close(handle);
+			return;
+		}
+			
+		/* check normal data directory */
+		str_format(buffer, buffer_size, "%s/%s", datadir, filename);
+		handle = io_open(buffer, flags);
+		if(handle)
+		{
+			io_close(handle);
+			return;
+		}
+	}
+	
+	buffer[0] = 0;
+}
+
+IOHANDLE engine_openfile(const char *filename, int flags)
+{
+	char buffer[1024];
+	
+	if(flags&IOFLAG_WRITE)
+	{
+		engine_savepath(filename, buffer, sizeof(buffer));
+		return io_open(buffer, flags);
+	}
+	else
+	{
+		IOHANDLE handle = 0;
+		
+		/* check current directory */
+		handle = io_open(filename, flags);
+		if(handle)
+			return handle;
+			
+		/* check user directory */
+		engine_savepath(filename, buffer, sizeof(buffer));
+		handle = io_open(buffer, flags);
+		if(handle)
+			return handle;
+			
+		/* check normal data directory */
+		str_format(buffer, sizeof(buffer), "%s/%s", datadir, filename);
+		handle = io_open(buffer, flags);
+		if(handle)
+			return handle;
+	}
+	return 0;		
+}
+
+void engine_parse_arguments(int argc, char **argv)
+{
+	/* load the configuration */
+	int i;
+	
+	/* check for datadir override */
+	for(i = 1; i < argc; i++)
+	{
+		if(argv[i][0] == '-' && argv[i][1] == 'd' && argv[i][2] == 0 && argc - i > 1)
+		{
+			str_copy(datadir, argv[i+1], sizeof(datadir));
+			i++;
+		}
+	}
+	
+	/* search for data directory */
+	engine_find_datadir(argv[0]);
+	
+	dbg_msg("engine/datadir", "paths used:");
+	dbg_msg("engine/datadir", "\t.");
+	dbg_msg("engine/datadir", "\t%s", application_save_path);
+	dbg_msg("engine/datadir", "\t%s", datadir);
+	dbg_msg("engine/datadir", "saving files to: %s", application_save_path);
+
+
+	/* check for scripts to execute */
+	for(i = 1; i < argc; i++)
+	{
+		if(argv[i][0] == '-' && argv[i][1] == 'f' && argv[i][2] == 0 && argc - i > 1)
+		{
+			console_execute_file(argv[i+1]);
+			i++;
+		}
+	}
+
+	/* search arguments for overrides */
+	{
+		int i;
+		for(i = 1; i < argc; i++)
+			console_execute_line(argv[i]);
+	}
+
+	console_execute_file("autoexec.cfg");
+
+	/* open logfile if needed */
+	if(config.logfile[0])
+		dbg_logger_file(config.logfile);
+
+	/* set default servers and load from disk*/
+	mastersrv_default();
+	mastersrv_load();
+}
+
+
+static IOHANDLE config_file = 0;
+
+int engine_config_write_start()
+{
+	config_save();
+	config_file = engine_openfile("settings.cfg", IOFLAG_WRITE);
+	if(config_file == 0)
+		return -1;
+	return 0;
+}
+
+void engine_config_write_line(const char *line)
+{
+	if(config_file)
+	{
+#if defined(CONF_FAMILY_WINDOWS)
+		static const char newline[] = "\r\n";
+#else
+		static const char newline[] = "\n";
+#endif
+		io_write(config_file, line, strlen(line));
+		io_write(config_file, newline, sizeof(newline)-1);
+	}
+}
+
+void engine_config_write_stop()
+{
+	io_close(config_file);
+	config_file = 0;
+}
+/*
+void engine_writeconfig()
+{
+}*/
+
+static int perf_tick = 1;
+static PERFORMACE_INFO *current = 0;
+
+void perf_init()
+{
+}
+
+void perf_next()
+{
+	perf_tick++;
+	current = 0;
+}
+
+void perf_start(PERFORMACE_INFO *info)
+{
+	if(info->tick != perf_tick)
+	{
+		info->parent = current;
+		info->first_child = 0;
+		info->next_child = 0;
+		
+		if(info->parent)
+		{
+			info->next_child = info->parent->first_child;
+			info->parent->first_child = info;
+		}
+		
+		info->tick = perf_tick;
+		info->biggest = 0;
+		info->total = 0;
+	}
+	
+	current = info;
+	current->start = time_get();
+}
+
+void perf_end()
+{
+	if(!current)
+		return;
+		
+	current->last_delta = time_get()-current->start;
+	current->total += current->last_delta;
+	
+	if(current->last_delta > current->biggest)
+		current->biggest = current->last_delta;
+	
+	current = current->parent;
+}
+
+static void perf_dump_imp(PERFORMACE_INFO *info, int indent)
+{
+	char buf[512] = {0};
+	int64 freq = time_freq();
+	int i;
+	
+	for(i = 0; i < indent; i++)
+		buf[i] = ' ';
+	
+	str_format(&buf[indent], sizeof(buf)-indent, "%-20s %8.2f %8.2f", info->name, info->total*1000/(float)freq, info->biggest*1000/(float)freq);
+	dbg_msg("perf", "%s", buf);
+	
+	info = info->first_child;
+	while(info)
+	{
+		perf_dump_imp(info, indent+2);
+		info = info->next_child;
+	}
+}
+
+void perf_dump(PERFORMACE_INFO *top)
+{
+	perf_dump_imp(top, 0);
+}
+
+/* master server functions */
+typedef struct
+{
+	char hostname[128];
+	NETADDR addr;
+	
+	HOSTLOOKUP lookup;
+} MASTER_INFO;
+
+static MASTER_INFO master_servers[MAX_MASTERSERVERS] = {{{0}}};
+static int needs_update = -1;
+
+int mastersrv_refresh_addresses()
+{
+	int i;
+	
+	if(needs_update != -1)
+		return 0;
+	
+	dbg_msg("engine/mastersrv", "refreshing master server addresses");
+
+	/* add lookup jobs */
+	for(i = 0; i < MAX_MASTERSERVERS; i++)	
+		engine_hostlookup(&master_servers[i].lookup, master_servers[i].hostname);
+	
+	needs_update = 1;
+	return 0;
+}
+
+void mastersrv_update()
+{
+	int i;
+	
+	/* check if we need to update */
+	if(needs_update != 1)
+		return;
+	needs_update = 0;
+	
+	for(i = 0; i < MAX_MASTERSERVERS; i++)
+	{
+		if(jobs_status(&master_servers[i].lookup.job) != JOBSTATUS_DONE)
+			needs_update = 1;
+		else
+		{
+			master_servers[i].addr = master_servers[i].lookup.addr;
+			master_servers[i].addr.port = 8300;
+		}
+	}
+	
+	if(!needs_update)
+	{
+		dbg_msg("engine/mastersrv", "saving addresses");
+		mastersrv_save();
+	}
+}
+
+int mastersrv_refreshing()
+{
+	return needs_update;
+}
+
+NETADDR mastersrv_get(int index) 
+{
+	return master_servers[index].addr;
+}
+
+const char *mastersrv_name(int index) 
+{
+	return master_servers[index].hostname;
+}
+
+void mastersrv_dump_servers()
+{
+	int i;
+	for(i = 0; i < MAX_MASTERSERVERS; i++)
+	{
+		dbg_msg("mastersrv", "#%d = %d.%d.%d.%d", i,
+			master_servers[i].addr.ip[0], master_servers[i].addr.ip[1],
+			master_servers[i].addr.ip[2], master_servers[i].addr.ip[3]);
+	}
+}
+
+void mastersrv_default()
+{
+	int i;
+	mem_zero(master_servers, sizeof(master_servers));
+	for(i = 0; i < MAX_MASTERSERVERS; i++)
+		sprintf(master_servers[i].hostname, "master%d.teeworlds.com", i+1);
+}
+
+int mastersrv_load()
+{
+	LINEREADER lr;
+	IOHANDLE file;
+	int count = 0;
+	
+	/* try to open file */
+	file = engine_openfile("masters.cfg", IOFLAG_READ);
+	if(!file)
+		return -1;
+	
+	linereader_init(&lr, file);
+	while(1)
+	{
+		MASTER_INFO info = {{0}};
+		int ip[4];
+		const char *line = linereader_get(&lr);
+		if(!line)
+			break;
+
+		/* parse line */		
+		if(sscanf(line, "%s %d.%d.%d.%d", info.hostname, &ip[0], &ip[1], &ip[2], &ip[3]) == 5)
+		{
+			info.addr.ip[0] = (unsigned char)ip[0];
+			info.addr.ip[1] = (unsigned char)ip[1];
+			info.addr.ip[2] = (unsigned char)ip[2];
+			info.addr.ip[3] = (unsigned char)ip[3];
+			info.addr.port = 8300;
+			if(count != MAX_MASTERSERVERS)
+			{
+				master_servers[count] = info;
+				count++;
+			}
+			else
+				dbg_msg("engine/mastersrv", "warning: skipped master server '%s' due to limit of %d", line, MAX_MASTERSERVERS);
+		}
+		else
+			dbg_msg("engine/mastersrv", "warning: couldn't parse master server '%s'", line);
+	}
+	
+	io_close(file);
+	return 0;
+}
+
+int mastersrv_save()
+{
+	IOHANDLE file;
+	int i;
+
+	/* try to open file */
+	file = engine_openfile("masters.cfg", IOFLAG_WRITE);
+	if(!file)
+		return -1;
+
+	for(i = 0; i < MAX_MASTERSERVERS; i++)
+	{
+		char buf[1024];
+		str_format(buf, sizeof(buf), "%s %d.%d.%d.%d\n", master_servers[i].hostname,
+			master_servers[i].addr.ip[0], master_servers[i].addr.ip[1],
+			master_servers[i].addr.ip[2], master_servers[i].addr.ip[3]);
+			
+		io_write(file, buf, strlen(buf));
+	}
+	
+	io_close(file);
+	return 0;
+}
+
+
+int hostlookup_thread(void *user)
+{
+	HOSTLOOKUP *lookup = (HOSTLOOKUP *)user;
+	net_host_lookup(lookup->hostname, &lookup->addr, NETTYPE_IPV4);
+	return 0;
+}
+
+void engine_hostlookup(HOSTLOOKUP *lookup, const char *hostname)
+{
+	str_copy(lookup->hostname, hostname, sizeof(lookup->hostname));
+	jobs_add(&hostlookuppool, &lookup->job, hostlookup_thread, lookup);
+}
+
+static int engine_find_datadir(char *argv0)
+{
+	/* 1) use provided data-dir override */
+	if(datadir[0])
+	{
+		if(fs_is_dir(datadir))
+			return 0;
+		else
+		{
+			dbg_msg("engine/datadir", "specified data-dir '%s' does not exist", datadir);
+			return -1;
+		}
+	}
+	
+	/* 2) use data-dir in PWD if present */
+	if(fs_is_dir("data"))
+	{
+		strcpy(datadir, "data");
+		return 0;
+	}
+	
+	/* 3) use compiled-in data-dir if present */
+	if (fs_is_dir(DATA_DIR))
+	{
+		strcpy(datadir, DATA_DIR);
+		return 0;
+	}
+	
+	/* 4) check for usable path in argv[0] */
+	{
+		unsigned int pos = strrchr(argv0, '/') - argv0;
+		
+		if (pos < sizeof(datadir))
+		{
+			char basedir[sizeof(datadir)];
+			strncpy(basedir, argv0, pos);
+			basedir[pos] = '\0';
+			str_format(datadir, sizeof(datadir), "%s/data", basedir);
+			
+			if (fs_is_dir(datadir))
+				return 0;
+		}
+	}
+	
+#if defined(CONF_FAMILY_UNIX)
+	/* 5) check for all default locations */
+	{
+		const char *sdirs[] = {
+			"/usr/share/teeworlds",
+			"/usr/local/share/teeworlds"
+			"/opt/teeworlds"
+		};
+		const int sdirs_count = sizeof(sdirs) / sizeof(sdirs[0]);
+		
+		int i;
+		for (i = 0; i < sdirs_count; i++)
+		{
+			if (fs_is_dir(sdirs[i]))
+			{
+				strcpy(datadir, sdirs[i]);
+				return 0;
+			}
+		}
+	}
+#endif
+	
+	/* no data-dir found */
+	dbg_msg("engine/datadir", "warning no data directory found");
+	return -1;
+}