1 extern FILE *replay_file;
3 unsigned long boot_time;
4 unsigned long burst_begin;
6 unsigned long burst_length;
7 struct log_type *MAIN_LOG;
12 char *services_config = "srvx.conf";
17 struct cManagerNode cManager;
19 struct policer_params *oper_policer_params, *luser_policer_params, *god_policer_params;
21 static const struct message_entry msgtab[] = {
22 { "MSG_NONE", "None" },
25 { "MSG_NEVER", "Never" },
26 { "MSG_SERVICE_IMMUNE", "$b%s$b may not be kicked, killed, banned, or deopped." },
27 { "MSG_SERVICE_PRIVILEGED", "$b%s$b is a privileged service." },
28 { "MSG_NOT_A_SERVICE", "$b%s$b is not a service bot." },
29 { "MSG_COMMAND_UNKNOWN", "$b%s$b is an unknown command." },
30 { "MSG_COMMAND_PRIVILEGED", "$b%s$b is a privileged command." },
31 { "MSG_COMMAND_DISABLED", "$b%s$b is a disabled command." },
32 { "MSG_SETTING_PRIVILEGED", "$b%s$b is a privileged setting." },
33 { "MSG_AUTHENTICATE", "You must first authenticate with $b$N$b." },
34 { "MSG_USER_AUTHENTICATE", "%s must first authenticate with $b$N$b." },
35 { "MSG_SET_EMAIL_ADDR", "You must first set your account's email address. (Contact network staff if you cannot auth to your account.)" },
36 { "MSG_HANDLE_UNKNOWN", "Account $b%s$b has not been registered." },
37 { "MSG_NICK_UNKNOWN", "User with nick $b%s$b does not exist." },
38 { "MSG_CHANNEL_UNKNOWN", "Channel with name $b%s$b does not exist." },
39 { "MSG_SERVER_UNKNOWN", "Server with name $b%s$b does not exist or is not linked." },
40 { "MSG_MODULE_UNKNOWN", "No module has been registered with name $b%s$b." },
41 { "MSG_INVALID_MODES", "$b%s$b is an invalid set of channel modes." },
42 { "MSG_INVALID_GLINE", "Invalid G-line '%s'." },
43 { "MSG_INVALID_DURATION", "Invalid time span '%s'." },
44 { "MSG_NOT_TARGET_NAME", "You must provide the name of a channel or user." },
45 { "MSG_NOT_CHANNEL_NAME", "You must provide a valid channel name." },
46 { "MSG_INVALID_CHANNEL", "You must provide the name of a channel that exists." },
47 { "MSG_CHANNEL_ABSENT", "You aren't currently in $b%s$b." },
48 { "MSG_CHANNEL_USER_ABSENT", "$b%s$b isn't currently in $b%s$b." },
49 { "MSG_MISSING_PARAMS", "$b%s$b requires more parameters." },
50 { "MSG_DEPRECATED_COMMAND", "The $b%s$b command has been deprecated, and will be removed in the future; please use $b%s$b instead." },
51 { "MSG_OPER_SUSPENDED", "Your $b$O$b access has been suspended." },
52 { "MSG_USER_OUTRANKED", "$b%s$b outranks you (command has no effect)." },
53 { "MSG_STUPID_ACCESS_CHANGE", "Please ask someone $belse$b to demote you." },
54 { "MSG_NO_SEARCH_ACCESS", "You do not have enough access to search based on $b%s$b." },
55 { "MSG_INVALID_CRITERIA", "$b%s$b is an invalid search criteria." },
56 { "MSG_INVALID_FIELD", "$b%s$b is an invalid search field." },
57 { "MSG_MATCH_COUNT", "Found $b%u$b matches." },
58 { "MSG_NO_MATCHES", "Nothing matched the criteria of your search." },
59 { "MSG_TOPIC_UNKNOWN", "No help on that topic." },
60 { "MSG_INVALID_BINARY", "$b%s$b is an invalid binary value." },
61 { "MSG_INTERNAL_FAILURE", "Your command could not be processed due to an internal failure." },
62 { "MSG_DB_UNKNOWN", "I do not know of a database named %s." },
63 { "MSG_DB_IS_MONDO", "Database %s is in the \"mondo\" database and cannot be written separately." },
64 { "MSG_DB_WRITE_ERROR", "Error while writing database %s." },
65 { "MSG_DB_WROTE_DB", "Wrote database %s (in %lu.%06lu seconds)." },
66 { "MSG_DB_WROTE_ALL", "Wrote all databases (in %lu.%06lu seconds)." },
68 { "MSG_0_SECONDS", "0 seconds" },
69 { "MSG_YEAR", "year" },
70 { "MSG_YEARS", "years" },
71 { "MSG_WEEK", "week" },
72 { "MSG_WEEKS", "weeks" },
74 { "MSG_DAYS", "days" },
75 { "MSG_HOUR", "hour" },
76 { "MSG_HOURS", "hours" },
77 { "MSG_MINUTE", "minute" },
78 { "MSG_MINUTES", "minutes" },
79 { "MSG_SECOND", "second" },
80 { "MSG_SECONDS", "seconds" },
84 void uplink_select(char *name);
87 uplink_insert(const char *key, void *data, UNUSED_ARG(void *extra))
89 struct uplinkNode *uplink = malloc(sizeof(struct uplinkNode));
90 struct record_data *rd = data;
91 struct addrinfo hints, *ai;
100 uplink->name = (char *)key;
101 uplink->host = database_get_data(rd->d.object, "address", RECDB_QSTRING);
103 str = database_get_data(rd->d.object, "port", RECDB_QSTRING);
104 uplink->port = str ? atoi(str) : 6667;
105 uplink->password = database_get_data(rd->d.object, "password", RECDB_QSTRING);
106 uplink->their_password = database_get_data(rd->d.object, "uplink_password", RECDB_QSTRING);
108 str = database_get_data(rd->d.object, "enabled", RECDB_QSTRING);
111 enabled = atoi(str) ? 1 : 0;
114 cManager.enabled += enabled;
116 str = database_get_data(rd->d.object, "max_tries", RECDB_QSTRING);
117 uplink->max_tries = str ? atoi(str) : 3;
118 uplink->flags = enabled ? 0 : UPLINK_UNAVAILABLE;
119 uplink->state = DISCONNECTED;
122 str = database_get_data(rd->d.object, "bind_address", RECDB_QSTRING);
123 memset(&hints, 0, sizeof(hints));
124 hints.ai_flags = AI_PASSIVE;
125 hints.ai_socktype = SOCK_STREAM;
126 if (!getaddrinfo(str, NULL, &hints, &ai))
128 uplink->bind_addr_len = ai->ai_addrlen;
129 uplink->bind_addr = calloc(1, ai->ai_addrlen);
130 memcpy(uplink->bind_addr, ai->ai_addr, ai->ai_addrlen);
135 uplink->bind_addr = NULL;
136 uplink->bind_addr_len = 0;
139 uplink->next = cManager.uplinks;
144 cManager.uplinks->prev = uplink;
147 cManager.uplinks = uplink;
149 /* If the configuration is being reloaded, set the current uplink
150 to the reloaded equivalent, if possible. */
153 && !irccasecmp(uplink->host, cManager.uplink->host)
154 && uplink->port == cManager.uplink->port)
156 uplink->state = cManager.uplink->state;
157 uplink->tries = cManager.uplink->tries;
158 cManager.uplink = uplink;
169 struct uplinkNode *oldUplinks = NULL, *oldUplink = NULL;
171 /* Save the old uplinks, we'll remove them later. */
172 oldUplink = cManager.uplink;
173 oldUplinks = cManager.uplinks;
175 cycles = conf_get_data("server/max_cycles", RECDB_QSTRING);
176 max_cycles = cycles ? atoi(cycles) : 30;
177 if(!(conf_node = conf_get_data("uplinks", RECDB_OBJECT)))
179 log_module(MAIN_LOG, LOG_FATAL, "No uplinks configured; giving up.");
183 cManager.enabled = 0;
184 dict_foreach(conf_node, uplink_insert, NULL);
186 /* Remove the old uplinks, if any. It doesn't matter if oldUplink (below)
187 is a reference to one of these, because it won't get dereferenced. */
190 struct uplinkNode *uplink, *next;
192 oldUplinks->prev->next = NULL;
194 for(uplink = oldUplinks; uplink; uplink = next)
197 free(uplink->bind_addr);
202 /* If the uplink hasn't changed, it's either NULL or pointing at
203 an uplink that was just deleted, select a new one. */
204 if(cManager.uplink == oldUplink)
208 irc_squit(self, "Uplinks updated; selecting new uplink.", NULL);
211 cManager.uplink = NULL;
217 uplink_find(char *name)
219 struct uplinkNode *uplink;
221 if(!cManager.enabled || !cManager.uplinks)
226 for(uplink = cManager.uplinks; uplink; uplink = uplink->next)
228 if(!strcasecmp(uplink->name, name))
238 uplink_select(char *name)
240 struct uplinkNode *start, *uplink, *next;
243 if(!cManager.enabled || !cManager.uplinks)
245 log_module(MAIN_LOG, LOG_FATAL, "No uplinks enabled; giving up.");
251 start = cManager.uplinks;
255 start = cManager.uplink->next;
258 start = cManager.uplinks;
263 for(uplink = start; uplink; uplink = next)
265 next = uplink->next ? uplink->next : cManager.uplinks;
273 /* We've wrapped around the list. */
276 sleep((cManager.cycles >> 1) * 5);
279 if(max_cycles && (cManager.cycles >= max_cycles))
281 log_module(MAIN_LOG, LOG_ERROR, "Maximum uplink list cycles exceeded; giving up.");
285 /* Give the uplink currently in 'uplink' consideration,
286 and if not selected, break on the next iteration. */
290 /* Skip bad uplinks. */
291 if(uplink->flags & UPLINK_UNAVAILABLE)
296 if(name && irccasecmp(uplink->name, name))
298 /* If we were told to connect to a specific uplink, don't stop
304 /* It would be possible to track uplink health through a variety
305 of statistics and only break on the best uplink. For now, break
306 on the first available one.
314 /* We are shit outta luck if every single uplink has been passed
315 over. Use the current uplink if possible. */
316 if(!cManager.uplink || cManager.uplink->flags & UPLINK_UNAVAILABLE)
318 log_module(MAIN_LOG, LOG_ERROR, "All available uplinks exhausted; giving up.");
325 cManager.uplink = uplink;
331 struct uplinkNode *uplink = cManager.uplink;
333 if(uplink->state != DISCONNECTED)
338 if(uplink->flags & UPLINK_UNAVAILABLE)
341 uplink = cManager.uplink;
346 /* This delay could scale with the number of tries. */
350 if(!create_socket_client(uplink))
352 if(uplink->max_tries && (uplink->tries >= uplink->max_tries))
354 /* This is a bad uplink, move on. */
355 uplink->flags |= UPLINK_UNAVAILABLE;
363 uplink->state = AUTHENTICATING;
364 irc_introduce(uplink->password);
373 /* This function is called when a ping is received. Take it as
374 a sign of link health and reset the connection manager
380 static exit_func_t *ef_list;
381 static unsigned int ef_size = 0, ef_used = 0;
383 void reg_exit_func(exit_func_t handler)
385 if (ef_used == ef_size) {
388 ef_list = realloc(ef_list, ef_size*sizeof(exit_func_t));
391 ef_list = malloc(ef_size*sizeof(exit_func_t));
394 ef_list[ef_used++] = handler;
397 void call_exit_funcs(void)
399 unsigned int n = ef_used;
401 /* Call them in reverse order because we initialize logs, then
402 * nickserv, then chanserv, etc., and they register their exit
403 * funcs in that order, and there are some dependencies (for
404 * example, ChanServ requires NickServ to not have cleaned up).
411 ef_used = ef_size = 0;
415 set_policer_param(const char *param, void *data, void *extra)
417 struct record_data *rd = data;
418 const char *str = GET_RECORD_QSTRING(rd);
420 policer_params_set(extra, param, str);
431 info = conf_get_data("services/global/nick", RECDB_QSTRING);
432 if (info && (info[0] == '.'))
436 info = conf_get_data("services/nickserv/nick", RECDB_QSTRING);
437 if (info && (info[0] == '.'))
441 info = conf_get_data("services/chanserv/nick", RECDB_QSTRING);
442 if (info && (info[0] == '.'))
446 info = conf_get_data("services/spamserv/nick", RECDB_QSTRING);
447 if (info && (info[0] == '.'))
451 god_policer_params = policer_params_new();
452 if ((dict = conf_get_data("policers/commands-god", RECDB_OBJECT))) {
453 dict_foreach(dict, set_policer_param, god_policer_params);
455 policer_params_set(god_policer_params, "size", "30");
456 policer_params_set(god_policer_params, "drain-rate", "1");
458 oper_policer_params = policer_params_new();
459 if ((dict = conf_get_data("policers/commands-oper", RECDB_OBJECT))) {
460 dict_foreach(dict, set_policer_param, oper_policer_params);
462 policer_params_set(oper_policer_params, "size", "10");
463 policer_params_set(oper_policer_params, "drain-rate", "1");
465 luser_policer_params = policer_params_new();
466 if ((dict = conf_get_data("policers/commands-luser", RECDB_OBJECT))) {
467 dict_foreach(dict, set_policer_param, luser_policer_params);
469 policer_params_set(luser_policer_params, "size", "5");
470 policer_params_set(luser_policer_params, "drain-rate", "0.50");
473 info = conf_get_data("services/opserv/nick", RECDB_QSTRING);
474 if (info && (info[0] == '.'))
479 #ifdef HAVE_SYS_RESOURCE_H
482 set_item_rlimit(const char *name, void *data, void *extra)
486 struct record_data *rd = data;
490 rsrc = (long)dict_find(extra, name, &found);
492 log_module(MAIN_LOG, LOG_ERROR, "Invalid rlimit \"%s\" in rlimits section.", name);
495 if (!(str = GET_RECORD_QSTRING(rd))) {
496 log_module(MAIN_LOG, LOG_ERROR, "Missing or invalid parameter type for rlimit \"%s\".", name);
499 if (getrlimit(rsrc, &rlim) < 0) {
500 log_module(MAIN_LOG, LOG_ERROR, "Couldn't get rlimit \"%s\": errno %d: %s", name, errno, strerror(errno));
503 rlim.rlim_cur = ParseVolume(str);
504 if (setrlimit(rsrc, &rlim) < 0) {
505 log_module(MAIN_LOG, LOG_ERROR, "Couldn't set rlimit \"%s\": errno %d: %s", name, errno, strerror(errno));
516 dict_insert(values, "data", (void*)RLIMIT_DATA);
517 dict_insert(values, "stack", (void*)RLIMIT_STACK);
519 dict_insert(values, "vmem", (void*)RLIMIT_VMEM);
522 dict_insert(values, "vmem", (void*)RLIMIT_AS);
525 if ((dict = conf_get_data("rlimits", RECDB_OBJECT))) {
526 dict_foreach(dict, set_item_rlimit, values);
541 usage(char *exe_name)
543 /* We can assume we have getopt_long(). */
544 printf("Usage: %s [-c config] [-r log] [-d] [-f] [-v|-h]\n"
545 " -c, --config selects a different configuration file.\n"
546 " -d, --debug enables debug mode.\n"
547 " -f, --foreground run srvx in the foreground.\n"
548 " -h, --help prints this usage message.\n"
549 " -k, --check checks the configuration file's syntax.\n"
550 " -r, --replay replay a log file (for debugging).\n"
551 " -v, --version prints this program's version.\n"
558 printf(" --------------------------------------------------\n"
559 " - "PACKAGE_STRING" ("CODENAME"), Built: " __DATE__ ", " __TIME__".\n"
560 " - Copyright (C) 2000 - 2007, srvx Development Team\n"
561 " - Version tag %s\n"
562 " --------------------------------------------------\n",
570 "This program is free software; you can redistribute it and/or modify\n"
571 "it under the terms of the GNU General Public License as published by\n"
572 "the Free Software Foundation; either version 2 of the License, or\n"
573 "(at your option) any later version.\n"
575 "This program is distributed in the hope that it will be useful,\n"
576 "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
577 "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\n"
578 "GNU General Public License for more details.\n"
580 "You should have received a copy of the GNU General Public License\n"
581 "along with this program; if not, write to the Free Software\n"
582 "Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.\n\n");
585 void main_shutdown(void)
587 struct uplinkNode *ul, *ul_next;
589 for (ul = cManager.uplinks; ul; ul = ul_next) {
596 #if defined(PID_FILE)
599 policer_params_delete(god_policer_params);
600 policer_params_delete(oper_policer_params);
601 policer_params_delete(luser_policer_params);