Split main.c into two files (main-common.c and the old main.c).
Add main-win32.c to handle the Win32-specific parts.
# CC=i586-mingw32msvc-cc
CPPFLAGS=-I$(VPATH) -Isrc
CFLAGS=-mno-cygwin -W -Wall -O2
+EXE_EXT=.exe
+LIBS=-lws2_32
#MODULES=
MODULES=helpserv memoserv qserver snoop sockcheck
src/conf.o \
src/dict-splay.o \
src/gline.o \
+ src/global.o \
src/hash.o \
src/heap.o \
src/helpfile.o \
src/ioset-win32.o \
src/ioset.o \
src/log.o \
- src/main.o \
+ src/main-win32.o \
src/md5.o \
src/modcmd.o \
src/modules.o \
src/tools.o \
$(addsuffix .o,$(addprefix src/mod-,$(MODULES)))
-srvx: src/config.h src/modules-list.h $(RX_OBJS) $(SRVX_OBJS)
+srvx$(EXE_EXT): src/config.h src/modules-list.h $(RX_OBJS) $(SRVX_OBJS)
+ $(CC) -o $< $(RX_OBJS) $(SRVX_OBJS) $(LIBS)
src/config.h: $(VPATH)/src/config.h.win32
cp $< $@
src/modules-list.h:
ioset-epoll.c \
ioset-select.c \
ioset-win32.c \
+ main-common.c \
+ main-win32.c \
proto-bahamut.c \
proto-common.c \
proto-p10.c \
--- /dev/null
+extern FILE *replay_file;
+
+time_t boot_time, burst_begin, now;
+unsigned long burst_length;
+struct log_type *MAIN_LOG;
+
+int quit_services;
+int max_cycles;
+
+char *services_config = "srvx.conf";
+
+char **services_argv;
+int services_argc;
+
+struct cManagerNode cManager;
+
+struct policer_params *oper_policer_params, *luser_policer_params, *god_policer_params;
+
+static const struct message_entry msgtab[] = {
+ { "MSG_NONE", "None" },
+ { "MSG_ON", "On" },
+ { "MSG_OFF", "Off" },
+ { "MSG_NEVER", "Never" },
+ { "MSG_SERVICE_IMMUNE", "$b%s$b may not be kicked, killed, banned, or deopped." },
+ { "MSG_SERVICE_PRIVILEGED", "$b%s$b is a privileged service." },
+ { "MSG_NOT_A_SERVICE", "$b%s$b is not a service bot." },
+ { "MSG_COMMAND_UNKNOWN", "$b%s$b is an unknown command." },
+ { "MSG_COMMAND_PRIVILEGED", "$b%s$b is a privileged command." },
+ { "MSG_COMMAND_DISABLED", "$b%s$b is a disabled command." },
+ { "MSG_SETTING_PRIVILEGED", "$b%s$b is a privileged setting." },
+ { "MSG_AUTHENTICATE", "You must first authenticate with $b$N$b." },
+ { "MSG_USER_AUTHENTICATE", "%s must first authenticate with $b$N$b." },
+ { "MSG_SET_EMAIL_ADDR", "You must first set your account's email address. (Contact network staff if you cannot auth to your account.)" },
+ { "MSG_HANDLE_UNKNOWN", "Account $b%s$b has not been registered." },
+ { "MSG_NICK_UNKNOWN", "User with nick $b%s$b does not exist." },
+ { "MSG_CHANNEL_UNKNOWN", "Channel with name $b%s$b does not exist." },
+ { "MSG_SERVER_UNKNOWN", "Server with name $b%s$b does not exist or is not linked." },
+ { "MSG_MODULE_UNKNOWN", "No module has been registered with name $b%s$b." },
+ { "MSG_INVALID_MODES", "$b%s$b is an invalid set of channel modes." },
+ { "MSG_INVALID_GLINE", "Invalid G-line '%s'." },
+ { "MSG_INVALID_DURATION", "Invalid time span '%s'." },
+ { "MSG_NOT_TARGET_NAME", "You must provide the name of a channel or user." },
+ { "MSG_NOT_CHANNEL_NAME", "You must provide a valid channel name." },
+ { "MSG_INVALID_CHANNEL", "You must provide the name of a channel that exists." },
+ { "MSG_CHANNEL_ABSENT", "You aren't currently in $b%s$b." },
+ { "MSG_CHANNEL_USER_ABSENT", "$b%s$b isn't currently in $b%s$b." },
+ { "MSG_MISSING_PARAMS", "$b%s$b requires more parameters." },
+ { "MSG_DEPRECATED_COMMAND", "The $b%s$b command has been deprecated, and will be removed in the future; please use $b%s$b instead." },
+ { "MSG_OPER_SUSPENDED", "Your $b$O$b access has been suspended." },
+ { "MSG_USER_OUTRANKED", "$b%s$b outranks you (command has no effect)." },
+ { "MSG_STUPID_ACCESS_CHANGE", "Please ask someone $belse$b to demote you." },
+ { "MSG_NO_SEARCH_ACCESS", "You do not have enough access to search based on $b%s$b." },
+ { "MSG_INVALID_CRITERIA", "$b%s$b is an invalid search criteria." },
+ { "MSG_MATCH_COUNT", "Found $b%u$b matches." },
+ { "MSG_NO_MATCHES", "Nothing matched the criteria of your search." },
+ { "MSG_TOPIC_UNKNOWN", "No help on that topic." },
+ { "MSG_INVALID_BINARY", "$b%s$b is an invalid binary value." },
+ { "MSG_INTERNAL_FAILURE", "Your command could not be processed due to an internal failure." },
+ { "MSG_DB_UNKNOWN", "I do not know of a database named %s." },
+ { "MSG_DB_IS_MONDO", "Database %s is in the \"mondo\" database and cannot be written separately." },
+ { "MSG_DB_WRITE_ERROR", "Error while writing database %s." },
+ { "MSG_DB_WROTE_DB", "Wrote database %s (in "FMT_TIME_T".%06lu seconds)." },
+ { "MSG_DB_WROTE_ALL", "Wrote all databases (in "FMT_TIME_T".%06lu seconds)." },
+ { "MSG_AND", "and" },
+ { "MSG_0_SECONDS", "0 seconds" },
+ { "MSG_YEAR", "year" },
+ { "MSG_YEARS", "years" },
+ { "MSG_WEEK", "week" },
+ { "MSG_WEEKS", "weeks" },
+ { "MSG_DAY", "day" },
+ { "MSG_DAYS", "days" },
+ { "MSG_HOUR", "hour" },
+ { "MSG_HOURS", "hours" },
+ { "MSG_MINUTE", "minute" },
+ { "MSG_MINUTES", "minutes" },
+ { "MSG_SECOND", "second" },
+ { "MSG_SECONDS", "seconds" },
+ { NULL, NULL }
+};
+
+void uplink_select(char *name);
+
+static int
+uplink_insert(const char *key, void *data, UNUSED_ARG(void *extra))
+{
+ struct uplinkNode *uplink = malloc(sizeof(struct uplinkNode));
+ struct record_data *rd = data;
+ struct addrinfo hints, *ai;
+ int enabled = 1;
+ char *str;
+
+ if(!uplink)
+ {
+ return 0;
+ }
+
+ uplink->name = (char *)key;
+ uplink->host = database_get_data(rd->d.object, "address", RECDB_QSTRING);
+
+ str = database_get_data(rd->d.object, "port", RECDB_QSTRING);
+ uplink->port = str ? atoi(str) : 6667;
+ uplink->password = database_get_data(rd->d.object, "password", RECDB_QSTRING);
+ uplink->their_password = database_get_data(rd->d.object, "uplink_password", RECDB_QSTRING);
+
+ str = database_get_data(rd->d.object, "enabled", RECDB_QSTRING);
+ if(str)
+ {
+ enabled = atoi(str) ? 1 : 0;
+ }
+
+ cManager.enabled += enabled;
+
+ str = database_get_data(rd->d.object, "max_tries", RECDB_QSTRING);
+ uplink->max_tries = str ? atoi(str) : 3;
+ uplink->flags = enabled ? 0 : UPLINK_UNAVAILABLE;
+ uplink->state = DISCONNECTED;
+ uplink->tries = 0;
+
+ str = database_get_data(rd->d.object, "bind_address", RECDB_QSTRING);
+ memset(&hints, 0, sizeof(hints));
+ hints.ai_flags = AI_PASSIVE;
+ hints.ai_socktype = SOCK_STREAM;
+ if (!getaddrinfo(str, NULL, &hints, &ai))
+ {
+ uplink->bind_addr_len = ai->ai_addrlen;
+ uplink->bind_addr = calloc(1, ai->ai_addrlen);
+ memcpy(uplink->bind_addr, ai->ai_addr, ai->ai_addrlen);
+ freeaddrinfo(ai);
+ }
+ else
+ {
+ uplink->bind_addr = NULL;
+ uplink->bind_addr_len = 0;
+ }
+
+ uplink->next = cManager.uplinks;
+ uplink->prev = NULL;
+
+ if(cManager.uplinks)
+ {
+ cManager.uplinks->prev = uplink;
+ }
+
+ cManager.uplinks = uplink;
+
+ /* If the configuration is being reloaded, set the current uplink
+ to the reloaded equivalent, if possible. */
+ if(cManager.uplink
+ && enabled
+ && !irccasecmp(uplink->host, cManager.uplink->host)
+ && uplink->port == cManager.uplink->port)
+ {
+ uplink->state = cManager.uplink->state;
+ uplink->tries = cManager.uplink->tries;
+ cManager.uplink = uplink;
+ }
+
+ return 0;
+}
+
+void
+uplink_compile(void)
+{
+ const char *cycles;
+ dict_t conf_node;
+ struct uplinkNode *oldUplinks = NULL, *oldUplink = NULL;
+
+ /* Save the old uplinks, we'll remove them later. */
+ oldUplink = cManager.uplink;
+ oldUplinks = cManager.uplinks;
+
+ cycles = conf_get_data("server/max_cycles", RECDB_QSTRING);
+ max_cycles = cycles ? atoi(cycles) : 30;
+ if(!(conf_node = conf_get_data("uplinks", RECDB_OBJECT)))
+ {
+ log_module(MAIN_LOG, LOG_FATAL, "No uplinks configured; giving up.");
+ exit(1);
+ }
+
+ cManager.enabled = 0;
+ dict_foreach(conf_node, uplink_insert, NULL);
+
+ /* Remove the old uplinks, if any. It doesn't matter if oldUplink (below)
+ is a reference to one of these, because it won't get dereferenced. */
+ if(oldUplinks)
+ {
+ struct uplinkNode *uplink, *next;
+
+ oldUplinks->prev->next = NULL;
+
+ for(uplink = oldUplinks; uplink; uplink = next)
+ {
+ next = uplink->next;
+ free(uplink->bind_addr);
+ free(uplink);
+ }
+ }
+
+ /* If the uplink hasn't changed, it's either NULL or pointing at
+ an uplink that was just deleted, select a new one. */
+ if(cManager.uplink == oldUplink)
+ {
+ if(oldUplink)
+ {
+ irc_squit(self, "Uplinks updated; selecting new uplink.", NULL);
+ }
+
+ cManager.uplink = NULL;
+ uplink_select(NULL);
+ }
+}
+
+struct uplinkNode *
+uplink_find(char *name)
+{
+ struct uplinkNode *uplink;
+
+ if(!cManager.enabled || !cManager.uplinks)
+ {
+ return NULL;
+ }
+
+ for(uplink = cManager.uplinks; uplink; uplink = uplink->next)
+ {
+ if(!strcasecmp(uplink->name, name))
+ {
+ return uplink;
+ }
+ }
+
+ return NULL;
+}
+
+void
+uplink_select(char *name)
+{
+ struct uplinkNode *start, *uplink, *next;
+ int stop;
+
+ if(!cManager.enabled || !cManager.uplinks)
+ {
+ log_module(MAIN_LOG, LOG_FATAL, "No uplinks enabled; giving up.");
+ exit(1);
+ }
+
+ if(!cManager.uplink)
+ {
+ start = cManager.uplinks;
+ }
+ else
+ {
+ start = cManager.uplink->next;
+ if(!start)
+ {
+ start = cManager.uplinks;
+ }
+ }
+
+ stop = 0;
+ for(uplink = start; uplink; uplink = next)
+ {
+ next = uplink->next ? uplink->next : cManager.uplinks;
+
+ if(stop)
+ {
+ uplink = NULL;
+ break;
+ }
+
+ /* We've wrapped around the list. */
+ if(next == start)
+ {
+ sleep((cManager.cycles >> 1) * 5);
+ cManager.cycles++;
+
+ if(max_cycles && (cManager.cycles >= max_cycles))
+ {
+ log_module(MAIN_LOG, LOG_FATAL, "Maximum uplink list cycles exceeded; giving up.");
+ exit(1);
+ }
+
+ /* Give the uplink currently in 'uplink' consideration,
+ and if not selected, break on the next iteration. */
+ stop = 1;
+ }
+
+ /* Skip bad uplinks. */
+ if(uplink->flags & UPLINK_UNAVAILABLE)
+ {
+ continue;
+ }
+
+ if(name && irccasecmp(uplink->name, name))
+ {
+ /* If we were told to connect to a specific uplink, don't stop
+ until we find it.
+ */
+ continue;
+ }
+
+ /* It would be possible to track uplink health through a variety
+ of statistics and only break on the best uplink. For now, break
+ on the first available one.
+ */
+
+ break;
+ }
+
+ if(!uplink)
+ {
+ /* We are shit outta luck if every single uplink has been passed
+ over. Use the current uplink if possible. */
+ if(!cManager.uplink || cManager.uplink->flags & UPLINK_UNAVAILABLE)
+ {
+ log_module(MAIN_LOG, LOG_FATAL, "All available uplinks exhausted; giving up.");
+ exit(1);
+ }
+
+ return;
+ }
+
+ cManager.uplink = uplink;
+}
+
+int
+uplink_connect(void)
+{
+ struct uplinkNode *uplink = cManager.uplink;
+
+ if(uplink->state != DISCONNECTED)
+ {
+ return 0;
+ }
+
+ if(uplink->flags & UPLINK_UNAVAILABLE)
+ {
+ uplink_select(NULL);
+ uplink = cManager.uplink;
+ }
+
+ if(uplink->tries)
+ {
+ /* This delay could scale with the number of tries. */
+ sleep(2);
+ }
+
+ if(!create_socket_client(uplink))
+ {
+ if(uplink->max_tries && (uplink->tries >= uplink->max_tries))
+ {
+ /* This is a bad uplink, move on. */
+ uplink->flags |= UPLINK_UNAVAILABLE;
+ uplink_select(NULL);
+ }
+
+ return 0;
+ }
+ else
+ {
+ uplink->state = AUTHENTICATING;
+ irc_introduce(uplink->password);
+ }
+
+ return 1;
+}
+
+void
+received_ping(void)
+{
+ /* This function is called when a ping is received. Take it as
+ a sign of link health and reset the connection manager
+ information. */
+
+ cManager.cycles = 0;
+}
+
+static exit_func_t *ef_list;
+static unsigned int ef_size = 0, ef_used = 0;
+
+void reg_exit_func(exit_func_t handler)
+{
+ if (ef_used == ef_size) {
+ if (ef_size) {
+ ef_size <<= 1;
+ ef_list = realloc(ef_list, ef_size*sizeof(exit_func_t));
+ } else {
+ ef_size = 8;
+ ef_list = malloc(ef_size*sizeof(exit_func_t));
+ }
+ }
+ ef_list[ef_used++] = handler;
+}
+
+void call_exit_funcs(void)
+{
+ unsigned int n = ef_used;
+
+ /* Call them in reverse order because we initialize logs, then
+ * nickserv, then chanserv, etc., and they register their exit
+ * funcs in that order, and there are some dependencies (for
+ * example, ChanServ requires NickServ to not have cleaned up).
+ */
+
+ while (n > 0) {
+ ef_list[--n]();
+ }
+ free(ef_list);
+ ef_used = ef_size = 0;
+}
+
+int
+set_policer_param(const char *param, void *data, void *extra)
+{
+ struct record_data *rd = data;
+ const char *str = GET_RECORD_QSTRING(rd);
+ if (str) {
+ policer_params_set(extra, param, str);
+ }
+ return 0;
+}
+
+static void
+conf_globals(void)
+{
+ const char *info;
+ dict_t dict;
+
+ info = conf_get_data("services/global/nick", RECDB_QSTRING);
+ if (info && (info[0] == '.'))
+ info = NULL;
+ init_global(info);
+
+ info = conf_get_data("services/nickserv/nick", RECDB_QSTRING);
+ if (info && (info[0] == '.'))
+ info = NULL;
+ init_nickserv(info);
+
+ info = conf_get_data("services/chanserv/nick", RECDB_QSTRING);
+ if (info && (info[0] == '.'))
+ info = NULL;
+ init_chanserv(info);
+
+ god_policer_params = policer_params_new();
+ if ((dict = conf_get_data("policers/commands-god", RECDB_OBJECT))) {
+ dict_foreach(dict, set_policer_param, god_policer_params);
+ } else {
+ policer_params_set(god_policer_params, "size", "30");
+ policer_params_set(god_policer_params, "drain-rate", "1");
+ }
+ oper_policer_params = policer_params_new();
+ if ((dict = conf_get_data("policers/commands-oper", RECDB_OBJECT))) {
+ dict_foreach(dict, set_policer_param, oper_policer_params);
+ } else {
+ policer_params_set(oper_policer_params, "size", "10");
+ policer_params_set(oper_policer_params, "drain-rate", "1");
+ }
+ luser_policer_params = policer_params_new();
+ if ((dict = conf_get_data("policers/commands-luser", RECDB_OBJECT))) {
+ dict_foreach(dict, set_policer_param, luser_policer_params);
+ } else {
+ policer_params_set(luser_policer_params, "size", "5");
+ policer_params_set(luser_policer_params, "drain-rate", "0.50");
+ }
+
+ info = conf_get_data("services/opserv/nick", RECDB_QSTRING);
+ if (info && (info[0] == '.'))
+ info = NULL;
+ init_opserv(info);
+}
+
+#ifdef HAVE_SYS_RESOURCE_H
+
+static int
+set_item_rlimit(const char *name, void *data, void *extra)
+{
+ long rsrc;
+ int found;
+ struct record_data *rd = data;
+ struct rlimit rlim;
+ const char *str;
+
+ rsrc = (long)dict_find(extra, name, &found);
+ if (!found) {
+ log_module(MAIN_LOG, LOG_ERROR, "Invalid rlimit \"%s\" in rlimits section.", name);
+ return 0;
+ }
+ if (!(str = GET_RECORD_QSTRING(rd))) {
+ log_module(MAIN_LOG, LOG_ERROR, "Missing or invalid parameter type for rlimit \"%s\".", name);
+ return 0;
+ }
+ if (getrlimit(rsrc, &rlim) < 0) {
+ log_module(MAIN_LOG, LOG_ERROR, "Couldn't get rlimit \"%s\": errno %d: %s", name, errno, strerror(errno));
+ return 0;
+ }
+ rlim.rlim_cur = ParseVolume(str);
+ if (setrlimit(rsrc, &rlim) < 0) {
+ log_module(MAIN_LOG, LOG_ERROR, "Couldn't set rlimit \"%s\": errno %d: %s", name, errno, strerror(errno));
+ }
+ return 0;
+}
+
+static void
+conf_rlimits(void)
+{
+ dict_t dict, values;
+
+ values = dict_new();
+ dict_insert(values, "data", (void*)RLIMIT_DATA);
+ dict_insert(values, "stack", (void*)RLIMIT_STACK);
+#ifdef RLIMIT_VMEM
+ dict_insert(values, "vmem", (void*)RLIMIT_VMEM);
+#else
+#ifdef RLIMIT_AS
+ dict_insert(values, "vmem", (void*)RLIMIT_AS);
+#endif
+#endif
+ if ((dict = conf_get_data("rlimits", RECDB_OBJECT))) {
+ dict_foreach(dict, set_item_rlimit, values);
+ }
+ dict_delete(values);
+}
+
+#else
+
+static void
+conf_rlimits(void)
+{
+}
+
+#endif
+
+void main_shutdown(void)
+{
+ struct uplinkNode *ul, *ul_next;
+ ioset_cleanup();
+ for (ul = cManager.uplinks; ul; ul = ul_next) {
+ ul_next = ul->next;
+ free(ul->bind_addr);
+ free(ul);
+ }
+ tools_cleanup();
+ conf_close();
+#if defined(PID_FILE)
+ remove(PID_FILE);
+#endif
+ policer_params_delete(god_policer_params);
+ policer_params_delete(oper_policer_params);
+ policer_params_delete(luser_policer_params);
+ if (replay_file)
+ fclose(replay_file);
+}
--- /dev/null
+#include "common.h"
+#include "conf.h"
+#include "gline.h"
+#include "ioset.h"
+#include "modcmd.h"
+#include "saxdb.h"
+#include "sendmail.h"
+#include "timeq.h"
+
+#include "chanserv.h"
+#include "global.h"
+#include "modules.h"
+#include "opserv.h"
+
+#define sleep(x) Sleep((x) * 1000)
+#include "main-common.c"
+
+#include <windows.h>
+
+int APIENTRY WinMain(HINSTANCE hInst, HINSTANCE hPrevInst, LPSTR lpCmdLine, int nCmdShow)
+{
+ tools_init();
+ /* TODO: parse lpCmdLine */
+ log_module(MAIN_LOG, LOG_INFO, "Initializing daemon...");
+
+ if (!conf_read(services_config)) {
+ log_module(MAIN_LOG, LOG_FATAL, "Unable to read %s.", services_config);
+ exit(0);
+ }
+
+ boot_time = time(&now);
+ conf_register_reload(uplink_compile);
+ atexit(call_exit_funcs);
+ reg_exit_func(main_shutdown);
+
+ log_init();
+ MAIN_LOG = log_register_type("srvx", "file:main.log");
+ ioset_init();
+ init_structs();
+ init_parse();
+ modcmd_init();
+ saxdb_init();
+ gline_init();
+ sendmail_init();
+ helpfile_init();
+ conf_globals();
+ conf_rlimits();
+ modules_init();
+ message_register_table(msgtab);
+ modcmd_finalize();
+ saxdb_finalize();
+ helpfile_finalize();
+ modules_finalize();
+ reg_exit_func(saxdb_write_all);
+ srand(time(&now));
+ ioset_run();
+
+ return quit_services;
+ (void)hInst; (void)hPrevInst; (void)nCmdShow;
+}
+
#define SIGCHLD SIGCLD
#endif
-extern FILE *replay_file;
-
-time_t boot_time, burst_begin, now;
-unsigned long burst_length;
-struct log_type *MAIN_LOG;
-
-int quit_services, max_cycles;
-
-char *services_config = "srvx.conf";
-
-char **services_argv;
-int services_argc;
-
-struct cManagerNode cManager;
-
-struct policer_params *oper_policer_params, *luser_policer_params, *god_policer_params;
-
-static const struct message_entry msgtab[] = {
- { "MSG_NONE", "None" },
- { "MSG_ON", "On" },
- { "MSG_OFF", "Off" },
- { "MSG_NEVER", "Never" },
- { "MSG_SERVICE_IMMUNE", "$b%s$b may not be kicked, killed, banned, or deopped." },
- { "MSG_SERVICE_PRIVILEGED", "$b%s$b is a privileged service." },
- { "MSG_NOT_A_SERVICE", "$b%s$b is not a service bot." },
- { "MSG_COMMAND_UNKNOWN", "$b%s$b is an unknown command." },
- { "MSG_COMMAND_PRIVILEGED", "$b%s$b is a privileged command." },
- { "MSG_COMMAND_DISABLED", "$b%s$b is a disabled command." },
- { "MSG_SETTING_PRIVILEGED", "$b%s$b is a privileged setting." },
- { "MSG_AUTHENTICATE", "You must first authenticate with $b$N$b." },
- { "MSG_USER_AUTHENTICATE", "%s must first authenticate with $b$N$b." },
- { "MSG_SET_EMAIL_ADDR", "You must first set your account's email address. (Contact network staff if you cannot auth to your account.)" },
- { "MSG_HANDLE_UNKNOWN", "Account $b%s$b has not been registered." },
- { "MSG_NICK_UNKNOWN", "User with nick $b%s$b does not exist." },
- { "MSG_CHANNEL_UNKNOWN", "Channel with name $b%s$b does not exist." },
- { "MSG_SERVER_UNKNOWN", "Server with name $b%s$b does not exist or is not linked." },
- { "MSG_MODULE_UNKNOWN", "No module has been registered with name $b%s$b." },
- { "MSG_INVALID_MODES", "$b%s$b is an invalid set of channel modes." },
- { "MSG_INVALID_GLINE", "Invalid G-line '%s'." },
- { "MSG_INVALID_DURATION", "Invalid time span '%s'." },
- { "MSG_NOT_TARGET_NAME", "You must provide the name of a channel or user." },
- { "MSG_NOT_CHANNEL_NAME", "You must provide a valid channel name." },
- { "MSG_INVALID_CHANNEL", "You must provide the name of a channel that exists." },
- { "MSG_CHANNEL_ABSENT", "You aren't currently in $b%s$b." },
- { "MSG_CHANNEL_USER_ABSENT", "$b%s$b isn't currently in $b%s$b." },
- { "MSG_MISSING_PARAMS", "$b%s$b requires more parameters." },
- { "MSG_DEPRECATED_COMMAND", "The $b%s$b command has been deprecated, and will be removed in the future; please use $b%s$b instead." },
- { "MSG_OPER_SUSPENDED", "Your $b$O$b access has been suspended." },
- { "MSG_USER_OUTRANKED", "$b%s$b outranks you (command has no effect)." },
- { "MSG_STUPID_ACCESS_CHANGE", "Please ask someone $belse$b to demote you." },
- { "MSG_NO_SEARCH_ACCESS", "You do not have enough access to search based on $b%s$b." },
- { "MSG_INVALID_CRITERIA", "$b%s$b is an invalid search criteria." },
- { "MSG_MATCH_COUNT", "Found $b%u$b matches." },
- { "MSG_NO_MATCHES", "Nothing matched the criteria of your search." },
- { "MSG_TOPIC_UNKNOWN", "No help on that topic." },
- { "MSG_INVALID_BINARY", "$b%s$b is an invalid binary value." },
- { "MSG_INTERNAL_FAILURE", "Your command could not be processed due to an internal failure." },
- { "MSG_DB_UNKNOWN", "I do not know of a database named %s." },
- { "MSG_DB_IS_MONDO", "Database %s is in the \"mondo\" database and cannot be written separately." },
- { "MSG_DB_WRITE_ERROR", "Error while writing database %s." },
- { "MSG_DB_WROTE_DB", "Wrote database %s (in "FMT_TIME_T".%06lu seconds)." },
- { "MSG_DB_WROTE_ALL", "Wrote all databases (in "FMT_TIME_T".%06lu seconds)." },
- { "MSG_AND", "and" },
- { "MSG_0_SECONDS", "0 seconds" },
- { "MSG_YEAR", "year" },
- { "MSG_YEARS", "years" },
- { "MSG_WEEK", "week" },
- { "MSG_WEEKS", "weeks" },
- { "MSG_DAY", "day" },
- { "MSG_DAYS", "days" },
- { "MSG_HOUR", "hour" },
- { "MSG_HOURS", "hours" },
- { "MSG_MINUTE", "minute" },
- { "MSG_MINUTES", "minutes" },
- { "MSG_SECOND", "second" },
- { "MSG_SECONDS", "seconds" },
- { NULL, NULL }
-};
-
-void uplink_select(char *name);
-
-static int
-uplink_insert(const char *key, void *data, UNUSED_ARG(void *extra))
-{
- struct uplinkNode *uplink = malloc(sizeof(struct uplinkNode));
- struct record_data *rd = data;
- struct addrinfo hints, *ai;
- int enabled = 1;
- char *str;
-
- if(!uplink)
- {
- return 0;
- }
-
- uplink->name = (char *)key;
- uplink->host = database_get_data(rd->d.object, "address", RECDB_QSTRING);
-
- str = database_get_data(rd->d.object, "port", RECDB_QSTRING);
- uplink->port = str ? atoi(str) : 6667;
- uplink->password = database_get_data(rd->d.object, "password", RECDB_QSTRING);
- uplink->their_password = database_get_data(rd->d.object, "uplink_password", RECDB_QSTRING);
-
- str = database_get_data(rd->d.object, "enabled", RECDB_QSTRING);
- if(str)
- {
- enabled = atoi(str) ? 1 : 0;
- }
-
- cManager.enabled += enabled;
-
- str = database_get_data(rd->d.object, "max_tries", RECDB_QSTRING);
- uplink->max_tries = str ? atoi(str) : 3;
- uplink->flags = enabled ? 0 : UPLINK_UNAVAILABLE;
- uplink->state = DISCONNECTED;
- uplink->tries = 0;
-
- str = database_get_data(rd->d.object, "bind_address", RECDB_QSTRING);
- memset(&hints, 0, sizeof(hints));
- hints.ai_flags = AI_PASSIVE;
- hints.ai_socktype = SOCK_STREAM;
- if (!getaddrinfo(str, NULL, &hints, &ai))
- {
- uplink->bind_addr_len = ai->ai_addrlen;
- uplink->bind_addr = calloc(1, ai->ai_addrlen);
- memcpy(uplink->bind_addr, ai->ai_addr, ai->ai_addrlen);
- freeaddrinfo(ai);
- }
- else
- {
- uplink->bind_addr = NULL;
- uplink->bind_addr_len = 0;
- }
-
- uplink->next = cManager.uplinks;
- uplink->prev = NULL;
-
- if(cManager.uplinks)
- {
- cManager.uplinks->prev = uplink;
- }
-
- cManager.uplinks = uplink;
-
- /* If the configuration is being reloaded, set the current uplink
- to the reloaded equivalent, if possible. */
- if(cManager.uplink
- && enabled
- && !irccasecmp(uplink->host, cManager.uplink->host)
- && uplink->port == cManager.uplink->port)
- {
- uplink->state = cManager.uplink->state;
- uplink->tries = cManager.uplink->tries;
- cManager.uplink = uplink;
- }
-
- return 0;
-}
-
-void
-uplink_compile(void)
-{
- const char *cycles;
- dict_t conf_node;
- struct uplinkNode *oldUplinks = NULL, *oldUplink = NULL;
-
- /* Save the old uplinks, we'll remove them later. */
- oldUplink = cManager.uplink;
- oldUplinks = cManager.uplinks;
-
- cycles = conf_get_data("server/max_cycles", RECDB_QSTRING);
- max_cycles = cycles ? atoi(cycles) : 30;
- if(!(conf_node = conf_get_data("uplinks", RECDB_OBJECT)))
- {
- log_module(MAIN_LOG, LOG_FATAL, "No uplinks configured; giving up.");
- exit(1);
- }
-
- cManager.enabled = 0;
- dict_foreach(conf_node, uplink_insert, NULL);
-
- /* Remove the old uplinks, if any. It doesn't matter if oldUplink (below)
- is a reference to one of these, because it won't get dereferenced. */
- if(oldUplinks)
- {
- struct uplinkNode *uplink, *next;
-
- oldUplinks->prev->next = NULL;
-
- for(uplink = oldUplinks; uplink; uplink = next)
- {
- next = uplink->next;
- free(uplink->bind_addr);
- free(uplink);
- }
- }
-
- /* If the uplink hasn't changed, it's either NULL or pointing at
- an uplink that was just deleted, select a new one. */
- if(cManager.uplink == oldUplink)
- {
- if(oldUplink)
- {
- irc_squit(self, "Uplinks updated; selecting new uplink.", NULL);
- }
-
- cManager.uplink = NULL;
- uplink_select(NULL);
- }
-}
-
-struct uplinkNode *
-uplink_find(char *name)
-{
- struct uplinkNode *uplink;
-
- if(!cManager.enabled || !cManager.uplinks)
- {
- return NULL;
- }
-
- for(uplink = cManager.uplinks; uplink; uplink = uplink->next)
- {
- if(!strcasecmp(uplink->name, name))
- {
- return uplink;
- }
- }
-
- return NULL;
-}
-
-void
-uplink_select(char *name)
-{
- struct uplinkNode *start, *uplink, *next;
- int stop;
-
- if(!cManager.enabled || !cManager.uplinks)
- {
- log_module(MAIN_LOG, LOG_FATAL, "No uplinks enabled; giving up.");
- exit(1);
- }
-
- if(!cManager.uplink)
- {
- start = cManager.uplinks;
- }
- else
- {
- start = cManager.uplink->next;
- if(!start)
- {
- start = cManager.uplinks;
- }
- }
-
- stop = 0;
- for(uplink = start; uplink; uplink = next)
- {
- next = uplink->next ? uplink->next : cManager.uplinks;
-
- if(stop)
- {
- uplink = NULL;
- break;
- }
-
- /* We've wrapped around the list. */
- if(next == start)
- {
- sleep((cManager.cycles >> 1) * 5);
- cManager.cycles++;
-
- if(max_cycles && (cManager.cycles >= max_cycles))
- {
- log_module(MAIN_LOG, LOG_FATAL, "Maximum uplink list cycles exceeded; giving up.");
- exit(1);
- }
-
- /* Give the uplink currently in 'uplink' consideration,
- and if not selected, break on the next iteration. */
- stop = 1;
- }
-
- /* Skip bad uplinks. */
- if(uplink->flags & UPLINK_UNAVAILABLE)
- {
- continue;
- }
-
- if(name && irccasecmp(uplink->name, name))
- {
- /* If we were told to connect to a specific uplink, don't stop
- until we find it.
- */
- continue;
- }
-
- /* It would be possible to track uplink health through a variety
- of statistics and only break on the best uplink. For now, break
- on the first available one.
- */
-
- break;
- }
-
- if(!uplink)
- {
- /* We are shit outta luck if every single uplink has been passed
- over. Use the current uplink if possible. */
- if(!cManager.uplink || cManager.uplink->flags & UPLINK_UNAVAILABLE)
- {
- log_module(MAIN_LOG, LOG_FATAL, "All available uplinks exhausted; giving up.");
- exit(1);
- }
-
- return;
- }
-
- cManager.uplink = uplink;
-}
-
-int
-uplink_connect(void)
-{
- struct uplinkNode *uplink = cManager.uplink;
-
- if(uplink->state != DISCONNECTED)
- {
- return 0;
- }
-
- if(uplink->flags & UPLINK_UNAVAILABLE)
- {
- uplink_select(NULL);
- uplink = cManager.uplink;
- }
-
- if(uplink->tries)
- {
- /* This delay could scale with the number of tries. */
- sleep(2);
- }
-
- if(!create_socket_client(uplink))
- {
- if(uplink->max_tries && (uplink->tries >= uplink->max_tries))
- {
- /* This is a bad uplink, move on. */
- uplink->flags |= UPLINK_UNAVAILABLE;
- uplink_select(NULL);
- }
-
- return 0;
- }
- else
- {
- uplink->state = AUTHENTICATING;
- irc_introduce(uplink->password);
- }
-
- return 1;
-}
-
-void
-received_ping(void)
-{
- /* This function is called when a ping is received. Take it as
- a sign of link health and reset the connection manager
- information. */
-
- cManager.cycles = 0;
-}
+#include "main-common.c"
void sigaction_writedb(int x)
{
do_reopen = 1;
}
-static exit_func_t *ef_list;
-static unsigned int ef_size = 0, ef_used = 0;
-
-void reg_exit_func(exit_func_t handler)
-{
- if (ef_used == ef_size) {
- if (ef_size) {
- ef_size <<= 1;
- ef_list = realloc(ef_list, ef_size*sizeof(exit_func_t));
- } else {
- ef_size = 8;
- ef_list = malloc(ef_size*sizeof(exit_func_t));
- }
- }
- ef_list[ef_used++] = handler;
-}
-
-void call_exit_funcs(void)
-{
- unsigned int n = ef_used;
-
- /* Call them in reverse order because we initialize logs, then
- * nickserv, then chanserv, etc., and they register their exit
- * funcs in that order, and there are some dependencies (for
- * example, ChanServ requires NickServ to not have cleaned up).
- */
-
- while (n > 0) {
- ef_list[--n]();
- }
- free(ef_list);
- ef_used = ef_size = 0;
-}
-
-int
-set_policer_param(const char *param, void *data, void *extra)
-{
- struct record_data *rd = data;
- const char *str = GET_RECORD_QSTRING(rd);
- if (str) {
- policer_params_set(extra, param, str);
- }
- return 0;
-}
-
-static void
-conf_globals(void)
-{
- const char *info;
- dict_t dict;
-
- info = conf_get_data("services/global/nick", RECDB_QSTRING);
- if (info && (info[0] == '.'))
- info = NULL;
- init_global(info);
-
- info = conf_get_data("services/nickserv/nick", RECDB_QSTRING);
- if (info && (info[0] == '.'))
- info = NULL;
- init_nickserv(info);
-
- info = conf_get_data("services/chanserv/nick", RECDB_QSTRING);
- if (info && (info[0] == '.'))
- info = NULL;
- init_chanserv(info);
-
- god_policer_params = policer_params_new();
- if ((dict = conf_get_data("policers/commands-god", RECDB_OBJECT))) {
- dict_foreach(dict, set_policer_param, god_policer_params);
- } else {
- policer_params_set(god_policer_params, "size", "30");
- policer_params_set(god_policer_params, "drain-rate", "1");
- }
- oper_policer_params = policer_params_new();
- if ((dict = conf_get_data("policers/commands-oper", RECDB_OBJECT))) {
- dict_foreach(dict, set_policer_param, oper_policer_params);
- } else {
- policer_params_set(oper_policer_params, "size", "10");
- policer_params_set(oper_policer_params, "drain-rate", "1");
- }
- luser_policer_params = policer_params_new();
- if ((dict = conf_get_data("policers/commands-luser", RECDB_OBJECT))) {
- dict_foreach(dict, set_policer_param, luser_policer_params);
- } else {
- policer_params_set(luser_policer_params, "size", "5");
- policer_params_set(luser_policer_params, "drain-rate", "0.50");
- }
-
- info = conf_get_data("services/opserv/nick", RECDB_QSTRING);
- if (info && (info[0] == '.'))
- info = NULL;
- init_opserv(info);
-}
-
-#ifdef HAVE_SYS_RESOURCE_H
-
-static int
-set_item_rlimit(const char *name, void *data, void *extra)
-{
- long rsrc;
- int found;
- struct record_data *rd = data;
- struct rlimit rlim;
- const char *str;
-
- rsrc = (long)dict_find(extra, name, &found);
- if (!found) {
- log_module(MAIN_LOG, LOG_ERROR, "Invalid rlimit \"%s\" in rlimits section.", name);
- return 0;
- }
- if (!(str = GET_RECORD_QSTRING(rd))) {
- log_module(MAIN_LOG, LOG_ERROR, "Missing or invalid parameter type for rlimit \"%s\".", name);
- return 0;
- }
- if (getrlimit(rsrc, &rlim) < 0) {
- log_module(MAIN_LOG, LOG_ERROR, "Couldn't get rlimit \"%s\": errno %d: %s", name, errno, strerror(errno));
- return 0;
- }
- rlim.rlim_cur = ParseVolume(str);
- if (setrlimit(rsrc, &rlim) < 0) {
- log_module(MAIN_LOG, LOG_ERROR, "Couldn't set rlimit \"%s\": errno %d: %s", name, errno, strerror(errno));
- }
- return 0;
-}
-
-static void
-conf_rlimits(void)
-{
- dict_t dict, values;
-
- values = dict_new();
- dict_insert(values, "data", (void*)RLIMIT_DATA);
- dict_insert(values, "stack", (void*)RLIMIT_STACK);
-#ifdef RLIMIT_VMEM
- dict_insert(values, "vmem", (void*)RLIMIT_VMEM);
-#else
-#ifdef RLIMIT_AS
- dict_insert(values, "vmem", (void*)RLIMIT_AS);
-#endif
-#endif
- if ((dict = conf_get_data("rlimits", RECDB_OBJECT))) {
- dict_foreach(dict, set_item_rlimit, values);
- }
- dict_delete(values);
-}
-
-#else
-
-static void
-conf_rlimits(void)
-{
-}
-
-#endif
-
-void main_shutdown(void)
-{
- struct uplinkNode *ul, *ul_next;
- ioset_cleanup();
- for (ul = cManager.uplinks; ul; ul = ul_next) {
- ul_next = ul->next;
- free(ul->bind_addr);
- free(ul);
- }
- tools_cleanup();
- conf_close();
- remove(PID_FILE);
- policer_params_delete(god_policer_params);
- policer_params_delete(oper_policer_params);
- policer_params_delete(luser_policer_params);
- if (replay_file)
- fclose(replay_file);
-}
-
void usage(char *self) {
/* We can assume we have getopt_long(). */
printf("Usage: %s [-c config] [-r log] [-d] [-f] [-v|-h]\n"