1 /* main.c - NeonServ v5.5
2 * Copyright (C) 2011-2012 Philipp Kreil (pk910)
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
20 #include "ClientSocket.h"
23 #include "IRCEvents.h"
24 #include "IRCParser.h"
26 #include "WHOHandler.h"
28 #include "mysqlConn.h"
29 #include "HandleInfoHandler.h"
33 #include "EventLogger.h"
37 #include "ConfigParser.h"
42 #include "module_commands.h"
43 #include "ModuleFunctions.h"
46 static int running, hard_restart;
47 static int statistics_requested_lusers = 0;
48 int statistics_enabled;
49 TIMEQ_CALLBACK(main_statistics);
50 TIMEQ_CALLBACK(main_checkauths);
51 static int daemonized = 0;
52 static int print_loglevel = 0;
53 static FILE *log_fptr = NULL;
54 static int process_argc;
55 static char **process_argv;
58 pthread_mutex_t cache_sync;
59 pthread_mutex_t whohandler_sync, whohandler_mass_sync;
60 static pthread_mutex_t log_sync;
63 static void check_firstrun();
76 free_handleinfohandler();
80 static int load_mysql_config() {
81 char *mysql_host, *mysql_user, *mysql_pass, *mysql_base;
84 mysql_host = get_string_field("MySQL.host");
86 perror("invalid neonserv.conf: missing MySQL.host");
89 mysql_serverport = get_int_field("MySQL.port");
91 mysql_serverport = 3306;
92 mysql_user = get_string_field("MySQL.user");
94 perror("invalid neonserv.conf: missing MySQL.user");
97 mysql_pass = get_string_field("MySQL.pass");
99 perror("invalid neonserv.conf: missing MySQL.pass");
102 mysql_base = get_string_field("MySQL.base");
104 perror("invalid neonserv.conf: missing MySQL base");
107 init_mysql(mysql_host, mysql_serverport, mysql_user, mysql_pass, mysql_base);
112 pthread_t *current_threads = NULL;
114 void * thread_main(void *arg) {
117 socket_wait = time(0) + SOCKET_SELECT_TIME;
119 if(!socket_loop(SOCKET_SELECT_TIME)) {
121 putlog(LOGLEVEL_ERROR, "No more active Bots... shutting down.\n");
125 } while(time(0) < socket_wait);
135 int getCurrentThreadID() {
136 if(!current_threads) return 0;
138 unsigned int my_tid = (unsigned int) pthread_self_tid();
139 for(i = 0; i < running_threads; i++) {
141 if((unsigned int) current_threads[i].p == my_tid)
143 if((unsigned int) current_threads[i] == my_tid)
163 int main(int argc, char *argv[]) {
164 int run_as_daemon = 1;
168 printf("NeonServ v%s\n\n", NEONSERV_VERSION);
169 struct option options[] = {
171 {"foreground", 0, 0, 'f'},
173 {"version", 0, 0, 'v'},
176 while ((c = getopt_long(argc, argv, "s:fvh", options, NULL)) != -1) {
179 print_loglevel = atoi(optarg);
185 printf("Version: %s.%d (%s)\n", NEONSERV_VERSION, patchlevel, (strcmp(revision, "") ? revision : "-"));
186 printf("Build: #%s %s (%s lines, " COMPILER ")\n", compilation, creation, codelines);
190 printf("Usage: ./neonserv [-s loglevel] [-f] [-h|-v]\n");
191 printf(" -s, --show show log lines matching loglevel in stdout.\n");
192 printf(" -f, --foreground run NeonServ in the foreground.\n");
193 printf(" -h, --help prints this usage message.\n");
194 printf(" -v, --version prints this program's version.\n");
200 if(geteuid() == 0 || getuid() == 0) {
201 fprintf(stderr, "NeonServ may not be run with super user privileges.\n");
205 #ifdef ENABLE_MEMORY_DEBUG
208 if(!loadConfig(CONF_FILE)) {
209 fprintf(stderr, "Unable to load " CONF_FILE "\n");
214 THREAD_MUTEX_INIT(log_sync);
216 if(!load_mysql_config()) {
217 fprintf(stderr, "Unable to connect to MySQL\n");
221 char **modulelist = get_all_fieldnames("modules");
222 if(!modulelist || !modulelist[0]) {
223 fprintf(stderr, "no modules loaded... Please update your configuration!\n");
229 /* Attempt to fork into the background if daemon mode is on. */
232 fprintf(stderr, "Unable to fork: %s\n", strerror(errno));
233 } else if (pid > 0) {
234 printf("Forking into the background (pid: %d)...\n", pid);
235 putlog(LOGLEVEL_INFO, "Forking into the background (pid: %d)...\n", pid);
241 FILE *pidfile = fopen(PID_FILE, "w");
242 if (pidfile == NULL) {
243 fprintf(stderr, "Unable to create PID file: %s\n", strerror(errno));
244 putlog(LOGLEVEL_ERROR, "Unable to create PID file: %s\n", strerror(errno));
246 fprintf(pidfile, "%i\n", (int)getpid());
250 fclose(stdin); retn = fopen("/dev/null", "r");
251 fclose(stdout); retn = fopen("/dev/null", "w");
252 fclose(stderr); retn = fopen("/dev/null", "w");
257 signal(SIGABRT, sighandler);
258 signal(SIGFPE, sighandler);
259 signal(SIGILL, sighandler);
260 signal(SIGINT, sighandler);
261 signal(SIGSEGV, sighandler);
262 signal(SIGTERM, sighandler);
264 start_time = time(0);
269 // Start Windows Sockets.
270 res = WSAStartup(MAKEWORD(2, 0), &wsadata);
273 perror("Unable to start Windows Sockets");
278 statistics_enabled = get_int_field("statistics.enable");
281 THREAD_MUTEX_INIT(cache_sync);
282 THREAD_MUTEX_INIT(whohandler_sync);
283 THREAD_MUTEX_INIT(whohandler_mass_sync);
297 register_module_commands();
298 init_handleinfohandler();
300 init_modulefunctions();
307 int update_minutes = get_int_field("statistics.frequency");
308 if(!update_minutes) update_minutes = 2;
309 timeq_add(update_minutes * 60 + 10, 0, main_statistics, NULL);
311 timeq_add(90, 0, main_checkauths, NULL);
313 int worker_threads = get_int_field("General.worker_threads");
314 if(!worker_threads) worker_threads = 1;
318 current_threads = calloc(worker_threads, sizeof(*current_threads));
319 for(tid_id = 0; tid_id < worker_threads; tid_id++) {
321 pthread_create(¤t_threads[tid_id], NULL, thread_main, NULL);
323 int usleep_delay = 1000000 / TICKS_PER_SECOND;
330 usleep(usleep_delay);
332 for(tid_id = 0; tid_id < worker_threads; tid_id++) {
333 pthread_join(current_threads[tid_id], NULL);
339 socket_wait = time(0) + SOCKET_SELECT_TIME;
341 if(!socket_loop(SOCKET_SELECT_TIME)) {
342 putlog(LOGLEVEL_ERROR, "No more active Bots... shutting down.\n");
346 } while(time(0) < socket_wait);
363 void restart_process() {
364 /* Append a NULL to the end of argv[]. */
365 char **restart_argv = (char **)alloca((process_argc + 1) * sizeof(char *));
366 memcpy(restart_argv, process_argv, process_argc * sizeof(char *));
367 restart_argv[process_argc] = NULL;
369 execv(process_argv[0], (const char * const*)restart_argv);
371 execv(process_argv[0], restart_argv);
375 int stricmp (const char *s1, const char *s2)
377 if (s1 == NULL) return s2 == NULL ? 0 : -(*s2);
378 if (s2 == NULL) return *s1;
380 while ((c1 = tolower (*s1)) == (c2 = tolower (*s2)))
382 if (*s1 == '\0') break;
388 int stricmplen (const char *s1, const char *s2, int len)
390 if (s1 == NULL) return s2 == NULL ? 0 : -(*s2);
391 if (s2 == NULL) return *s1;
394 while ((c1 = tolower (*s1)) == (c2 = tolower (*s2)))
397 if (*s1 == '\0') break;
404 void restart_bot(int do_hard_restart) {
405 hard_restart = do_hard_restart;
414 void reload_config() {
415 loadConfig(CONF_FILE);
419 static int getCurrentSecondsOfDay() {
420 time_t now = time(0);
421 struct tm *timeofday = localtime(&now);
423 seconds += timeofday->tm_hour * 3600;
424 seconds += timeofday->tm_min * 60;
425 seconds += timeofday->tm_sec;
429 static AUTHLOOKUP_CALLBACK(main_checkauths_callback) {
430 //check if registered is still valid
433 printf_mysql_query("SELECT `user_id`, `user_registered` FROM `users` WHERE `user_user` = '%s'", escape_string(auth));
435 if ((row = mysql_fetch_row(res)) != NULL) {
436 int diff = registered - atoi(row[1]);
439 if(!exists || (strcmp(row[1], "0") && diff > 86400)) {
440 //User is no longer valid! Delete it...
441 deleteUser(atoi(row[0]));
442 char *alertchan = get_string_field("General.CheckAuths.alertchan");
446 strcpy(reason, "USER_NOT_EXISTS");
448 sprintf(reason, "USER_REGISTERED_MISSMATCH: %lu, expected %d (diff: %d)", (unsigned long) registered, atoi(row[1]), diff);
450 struct ChanNode *alertchan_chan = getChanByName(alertchan);
451 struct ClientSocket *alertclient;
452 if(alertchan_chan && (alertclient = getChannelBot(alertchan_chan, 0)) != NULL) {
453 putsock(alertclient, "PRIVMSG %s :Deleted User %s (%s)", alertchan_chan->name, auth, reason);
456 } else if(exists && !strcmp(row[1], "0")) {
457 printf_mysql_query("UPDATE `users` SET `user_registered` = '%lu', `user_lastcheck` = UNIX_TIMESTAMP() WHERE `user_id` = '%s'", (unsigned long) registered, row[0]);
459 printf_mysql_query("UPDATE `users` SET `user_lastcheck` = UNIX_TIMESTAMP() WHERE `user_id` = '%s'", row[0]);
464 TIMEQ_CALLBACK(main_checkauths) {
466 if(get_int_field("General.CheckAuths.enabled")) {
467 int check_start_time = get_int_field("General.CheckAuths.start_time") * 3600;
468 int duration = get_int_field("General.CheckAuths.duration") * 60;
469 int now = getCurrentSecondsOfDay();
470 if(now < check_start_time && check_start_time+duration >= 86400) {
471 check_start_time -= 86400;
473 if(now >= check_start_time && now < (check_start_time + duration)) {
474 next_call = get_int_field("General.CheckAuths.interval");
475 //get the "longest-unchecked-user"
479 time_t unixtime = time(0);
480 int min_unckecked = get_int_field("General.CheckAuths.min_unckecked");
481 printf_mysql_query("SELECT `user_user`, `user_lastcheck` FROM `users` ORDER BY `user_lastcheck` ASC LIMIT 1");
483 if ((row = mysql_fetch_row(res)) != NULL) {
484 lastcheck = atoi(row[1]);
485 if(!lastcheck || unixtime - lastcheck >= min_unckecked) {
486 lookup_authname(row[0], 0, main_checkauths_callback, NULL);
492 if(now > check_start_time)
493 pending = 86400 - now + check_start_time;
495 pending = check_start_time - now;
501 timeq_add(next_call, 0, main_checkauths, NULL);
504 TIMEQ_CALLBACK(main_statistics) {
505 int update_minutes = get_int_field("statistics.frequency");
506 if(!update_minutes) update_minutes = 2;
507 timeq_add(update_minutes * 60, 0, main_statistics, NULL);
508 if(get_int_field("statistics.enable")) {
509 statistics_enabled = 1;
510 statistics_requested_lusers = 1;
511 if(get_int_field("statistics.include_lusers")) {
512 struct ClientSocket *bot, *lusersbot = NULL;
513 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
514 if(bot->flags & SOCKET_FLAG_PREFERRED)
518 if(bot == NULL) bot = getBots(SOCKET_FLAG_READY, NULL);
519 putsock(bot, "LUSERS");
524 statistics_enabled = 0;
527 int statistics_update() {
528 if(get_int_field("statistics.enable") && statistics_requested_lusers && get_string_field("statistics.execute")) {
529 statistics_requested_lusers = 0;
530 char command[MAXLEN];
536 - commands per minute
540 sprintf(command, "%s %d %d %d %d %d %d %d", get_string_field("statistics.execute"), getUserCount(), getChanUserCount(), getChannelCount(), statistics_privmsg, statistics_commands, statistics_network_users, statistics_network_channels);
541 statistics_privmsg = 0;
542 statistics_commands = 0;
543 return system(command);
548 time_t getStartTime() {
552 int getRunningThreads() {
554 return running_threads;
560 void write_log(int loglevel, const char *line, int len) {
561 SYNCHRONIZE(log_sync);
562 if(!daemonized && (print_loglevel & loglevel)) {
564 } else if(!daemonized && loglevel == LOGLEVEL_ERROR) {
565 fprintf(stderr, "%s", line);
567 if(get_int_field("log.loglevel") & loglevel) {
569 log_fptr = fopen(LOG_FILE, "a");
570 if(!log_fptr) goto write_log_end;
575 timeinfo = localtime(&rawtime);
577 int timepos = strftime(timestr, 20, "%x %X ", timeinfo);
578 fwrite(timestr, 1, timepos, log_fptr);
579 fwrite(line, 1, len, log_fptr);
582 DESYNCHRONIZE(log_sync);
585 void putlog(int loglevel, const char *text, ...) {
587 char logBuf[MAXLOGLEN];
590 va_start(arg_list, text);
591 pos = vsnprintf(logBuf, MAXLOGLEN - 1, text, arg_list);
593 if (pos < 0 || pos > (MAXLOGLEN - 1)) pos = MAXLOGLEN - 1;
595 write_log(loglevel, logBuf, pos);
598 static void check_firstrun() {
601 printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_access` = 1000 LIMIT 1");
603 if (mysql_fetch_row(res) == NULL) {
605 printf("No superuser found...\n");
606 check_firstrun_admin:
607 printf("AuthServ account name of admin user: ");
610 ptr = fgets(adminuser, 30, stdin);
611 for(ptr = adminuser; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
612 if(strlen(adminuser) < 2) goto check_firstrun_admin;
613 printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(adminuser));
615 if ((row = mysql_fetch_row(res)) != NULL)
616 printf_mysql_query("UPDATE `users` SET `user_access` = 1000 WHERE `user_id` = '%s'", row[0]);
618 printf_mysql_query("INSERT INTO `users` (`user_user`, `user_access`) VALUES ('%s', 1000)", escape_string(adminuser));
620 printf_mysql_query("SELECT `id` FROM `bots` WHERE `active` = 1 LIMIT 1");
622 if (mysql_fetch_row(res) == NULL) {
624 printf("No active bot found...\n\n");
625 printf("ADD NEW BOT\n");
628 check_firstrun_bot_nick:
630 ptr = fgets(bot_nick, 30, stdin);
631 for(ptr = bot_nick; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
632 if(strlen(bot_nick) < 2) goto check_firstrun_bot_nick;
634 check_firstrun_bot_ident:
636 ptr = fgets(bot_ident, 15, stdin);
637 for(ptr = bot_ident; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
638 if(strlen(bot_ident) < 2) goto check_firstrun_bot_ident;
639 char bot_realname[101];
640 check_firstrun_bot_realname:
641 printf("Realname: ");
642 ptr = fgets(bot_realname, 100, stdin);
643 for(ptr = bot_realname; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
644 if(strlen(bot_realname) < 2) goto check_firstrun_bot_realname;
645 char bot_server[101];
646 check_firstrun_bot_server:
647 printf("Server: [irc.onlinegamesnet.net] ");
648 ptr = fgets(bot_server, 100, stdin);
649 for(ptr = bot_server; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
650 if(*bot_server && strlen(bot_nick) < 5) goto check_firstrun_bot_server;
652 strcpy(bot_server, "irc.onlinegamesnet.net");
654 char bot_port_buf[7];
655 printf("Port: [6667] ");
656 ptr = fgets(bot_port_buf, 6, stdin);
657 for(ptr = bot_port_buf; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
661 bot_port = atoi(bot_port_buf);
664 check_firstrun_bot_ssl:
665 printf("SSL: [y/N] ");
666 ptr = fgets(bot_ssl_buf, 4, stdin);
667 for(ptr = bot_ssl_buf; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
668 if(!*bot_ssl_buf || tolower(*bot_ssl_buf) == 'n')
670 else if(tolower(*bot_ssl_buf) == 'y')
673 goto check_firstrun_bot_ssl;
675 printf("Server Password: [] ");
676 ptr = fgets(bot_pass, 100, stdin);
677 for(ptr = bot_pass; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
679 char bot_maxchan_buf[5];
680 printf("MaxChannel: [20] ");
681 ptr = fgets(bot_maxchan_buf, 5, stdin);
682 for(ptr = bot_maxchan_buf; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
684 bot_maxchan = atoi(bot_maxchan_buf);
688 char bot_queue_buf[5];
689 check_firstrun_bot_queue:
690 printf("Queue (prevents excess floods): [Y/n] ");
691 ptr = fgets(bot_queue_buf, 4, stdin);
692 for(ptr = bot_queue_buf; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
693 if(!*bot_queue_buf || tolower(*bot_queue_buf) == 'y')
695 else if(tolower(*bot_queue_buf) == 'n')
698 goto check_firstrun_bot_queue;
699 printf_mysql_query("INSERT INTO `bots` (`active`, `nick`, `server`, `port`, `pass`, `ssl`, `ident`, `realname`, `botclass`, `textbot`, `queue`, `defaulttrigger`, `max_channels`) VALUES ('1', '%s', '%s', '%d', '%s', '%d', '%s', '%s', '1', '1', '%d', '+', '%d')", escape_string(bot_nick), escape_string(bot_server), bot_port, escape_string(bot_pass), bot_ssl, escape_string(bot_ident), escape_string(bot_realname), bot_queue, bot_maxchan);