Merge branch 'development'
[NeonServV5.git] / src / main.c
1 /* main.c - NeonServ v5.6
2  * Copyright (C) 2011-2012  Philipp Kreil (pk910)
3  * 
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.
8  * 
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.
13  * 
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/>. 
16  */
17
18 #define DEFAULT_PID_FILE "neonserv.pid"
19 #define DEFAULT_CONF_FILE "neonserv.conf"
20
21 #include "main.h"
22 #include "signal.h"
23 #include "ClientSocket.h"
24 #include "UserNode.h"
25 #include "ChanNode.h"
26 #include "IRCEvents.h"
27 #include "IRCParser.h"
28 #include "modcmd.h"
29 #include "WHOHandler.h"
30 #include "bots.h"
31 #include "mysqlConn.h"
32 #include "HandleInfoHandler.h"
33 #include "lang.h"
34 #include "tools.h"
35 #include "timeq.h"
36 #include "EventLogger.h"
37 #include "ModeNode.h"
38 #include "IRCQueue.h"
39 #include "DBHelper.h"
40 #include "ConfigParser.h"
41 #include "QServer.h"
42 #include "version.h"
43 #include "modules.h"
44 #include "module_commands.h"
45 #include "ModuleFunctions.h"
46 #include "IOHandler.h"
47 #include "statistics.h"
48 #include "log.h"
49
50 struct ProcessState process_state;
51
52 #ifdef HAVE_THREADS
53 pthread_mutex_t cache_sync;
54 pthread_mutex_t whohandler_sync, whohandler_mass_sync;
55 static pthread_t *current_threads = NULL;
56 #endif
57
58 static void *main_tread(void *empty);
59 static TIMEQ_CALLBACK(clear_cache);
60 static TIMEQ_CALLBACK(main_checkauths);
61 static void check_firstrun();
62
63
64 static void main_parse_arguments() {
65     int c;
66     struct option options[] = {
67         {"show", 1, 0, 's'},
68         {"foreground", 0, 0, 'f'},
69         {"config", 1, 0, 'c'},
70         {"pid", 1, 0, 'p'},
71         {"help", 0, 0, 'h'},
72         {"version", 0, 0, 'v'},
73         {0, 0, 0, 0}
74     };
75     while ((c = getopt_long(process_state.argc, process_state.argv, "s:fvh", options, NULL)) != -1) {
76         switch (c) {
77         case 'c':
78             strncpy(process_state.config, optarg, MAXLEN-1);
79             break;
80         case 'p':
81             strncpy(process_state.pidfile, optarg, MAXLEN-1);
82             break;
83         case 's':
84             process_state.loglevel = atoi(optarg);
85             break;
86         case 'f':
87             process_state.run_as_daemon = 0;
88             break;
89         case 'v':
90             printf("Version: %s.%d (%s)\n", NEONSERV_VERSION, patchlevel, (strcmp(revision, "") ? revision : "-"));
91             printf("Build: #%s %s (%s lines, " COMPILER ")\n", compilation, creation, codelines);
92             exit(0);
93             break;
94         case 'h':
95             printf("Usage: ./neonserv [-c neonserv.conf] [-p neonserv.pid] [-s loglevel] [-f] [-h|-v]\n");
96             printf(" -c, --config         use this configuration file.\n");
97             printf(" -f, --foreground     run NeonServ in the foreground.\n");
98             printf(" -h, --help           prints this usage message.\n");
99             printf(" -p, --pid            use this pid file.\n");
100             printf(" -s, --show           show log lines matching loglevel in stdout.\n");
101             printf(" -v, --version        prints this program's version.\n");
102             exit(0);
103             break;
104         }
105     }
106 }
107
108 static void main_daemon_exit() {
109     remove(process_state.pidfile);
110 }
111
112 static void main_daemonize() {
113     #ifndef WIN32
114     /* Attempt to fork into the background if daemon mode is on. */
115     pid_t pid = fork();
116     if (pid < 0) {
117         fprintf(stderr, "Unable to fork: %s\n", strerror(errno));
118     } else if (pid > 0) {
119         printf("Forking into the background (pid: %d)...\n", pid);
120         printf_log("main", LOG_INFO, "Forking into the background (pid: %d)...\n", pid);
121         exit(0);
122     }
123     setsid();
124     process_state.daemonized = 1;
125     atexit(main_daemon_exit);
126     FILE *pidfile = fopen(process_state.pidfile, "w");
127     if (pidfile == NULL) {
128         fprintf(stderr, "Unable to create PID file: %s\n", strerror(errno));
129         printf_log("main", LOG_ERROR, "Unable to create PID file: %s\n", strerror(errno));
130     } else {
131         fprintf(pidfile, "%i\n", (int)getpid());
132         fclose(pidfile);
133     }
134     FILE *retn;
135     fclose(stdin); retn = fopen("/dev/null", "r");
136     fclose(stdout); retn = fopen("/dev/null", "w");
137     fclose(stderr); retn = fopen("/dev/null", "w");
138     #endif
139 }
140
141 static int reload_configuration() {
142     printf_log("main", LOG_DEBUG, "reloading configuration file: %s", process_state.config);
143     if(!loadConfig(process_state.config)) {
144         printf_log("main", LOG_ERROR, "could not reload configuration file: %s", process_state.config);
145         return 1;
146     }
147     if(process_state.loaded_config) {
148         if(!reload_mysql())
149             return 2;
150         char **modulelist = get_all_fieldnames("modules");
151         if(!modulelist || !modulelist[0]) {
152             free(modulelist);
153             return 3;
154         }
155         free(modulelist);
156         event_reload(0);
157     }
158     process_state.loaded_config = 1;
159     return 0;
160 }
161
162
163 /* INITIALISATION OF SUBSYSTEMS */
164 void initialize_subsystems() {
165     init_bind();
166     init_log();
167     printf_log("main", LOG_INFO, "starting up NeonServ %s subsystems...", NEONSERV_VERSION);
168     init_lang();
169     init_parser();
170     init_UserNode();
171     init_ChanNode();
172     init_ModeNode();
173         init_modcmd();
174     register_module_commands();
175     init_handleinfohandler();
176     init_tools();
177     init_modulefunctions();
178     loadModules();
179     init_bots();
180     init_DBHelper();
181     qserver_init();
182     load_languages();
183     init_statistics();
184 }
185
186 void shutdown_subsystems() {
187     printf_log("main", LOG_INFO, "stopping NeonServ subsystems...");
188     free_sockets(1);
189     //wait 50ms (run iohandler)
190     {
191         struct timeval timeout, ctime1, ctime2;
192         gettimeofday(&ctime1, NULL);
193         ctime1.tv_usec += 50000;
194         if(ctime1.tv_usec > 1000000) {
195             ctime1.tv_usec -= 1000000;
196             ctime1.tv_sec++;
197         }
198         do {
199             timeout.tv_sec = 0;
200             timeout.tv_usec = 10000;
201             iohandler_poll_timeout(timeout);
202             gettimeofday(&ctime2, NULL);
203         } while(timeval_is_bigger(ctime1, ctime2));
204     }
205     stop_modules();
206     free_sockets(0);
207     qserver_free();
208     free_parser();
209     free_UserNode();
210     free_ChanNode();
211     free_bind();
212     free_modcmd();
213     free_whoqueue();
214     free_mysql();
215     free_handleinfohandler();
216     free_lang();
217 }
218
219 /* THREAD CONTROL */
220 #ifdef HAVE_THREADS
221 int getCurrentThreadID() {
222     if(!current_threads) return 0;
223     int i;
224     unsigned int my_tid = (unsigned int) pthread_self_tid();
225     for(i = 0; i < process_state.running_threads; i++) {
226         #ifdef WIN32
227         if((unsigned int) current_threads[i].p == my_tid)
228         #else
229         if((unsigned int) current_threads[i] == my_tid)
230         #endif
231             return i+1;
232     }
233     return 0;
234 }
235 #endif
236
237 int getRunningThreads() {
238         #ifdef HAVE_THREADS
239     return process_state.running_threads;
240         #else
241         return 1;
242         #endif
243 }
244
245 static void main_start_threads() {
246     int worker_threads = get_int_field("General.worker_threads");
247     if(!worker_threads) worker_threads = 1;
248     #ifdef HAVE_THREADS
249     int tid_id = 0;
250     {
251         current_threads = calloc(worker_threads, sizeof(*current_threads));
252         for(tid_id = 0; tid_id < worker_threads; tid_id++) {
253             process_state.running_threads++;
254             pthread_create(&current_threads[tid_id], NULL, main_tread, NULL);
255         }
256     }
257     #endif
258     main_tread(NULL);
259     #ifdef HAVE_THREADS
260     {
261         for(tid_id = 0; tid_id < worker_threads; tid_id++) {
262             pthread_join(current_threads[tid_id], NULL);
263         }
264         process_state.running_threads = 0;
265     }
266     #endif
267 }
268
269 /* MAIN FUNCTION(S) */
270
271 static void *main_tread(void *empty) {
272     while(process_state.running) {
273         iohandler_poll();
274     }
275     return NULL;
276 }
277
278 static void main_restart_process() {
279     /* Append a NULL to the end of argv[]. */
280     char **restart_argv = (char **)alloca((process_state.argc + 1) * sizeof(char *));
281     memcpy(restart_argv, process_state.argv, process_state.argc * sizeof(char *));
282     restart_argv[process_state.argc] = NULL;
283     #ifdef WIN32
284     execv(process_state.argv[0], (const char * const*)restart_argv);
285     #else
286     execv(process_state.argv[0], restart_argv);
287     #endif
288 }
289
290 int main(int argc, char *argv[]) {
291     memset(&process_state, 0, sizeof(process_state));
292     printf("NeonServ v%s\n\n", NEONSERV_VERSION);
293     
294     process_state.argv = argv;
295     process_state.argc = argc;
296     process_state.run_as_daemon = 1;
297     strcpy(process_state.config, DEFAULT_CONF_FILE);
298     strcpy(process_state.pidfile, DEFAULT_PID_FILE);
299     
300     //parse argv
301     main_parse_arguments();
302     
303     //initialize memory debugger BEFORE allocating memory
304     #ifdef ENABLE_MEMORY_DEBUG
305     initMemoryDebug();
306     #endif
307     
308     //initialize mutex debugger BEFORE using any mutexes
309     #ifdef ENABLE_MUTEX_DEBUG
310     initMutexDebug();
311     #endif
312     
313     //deny root startup
314     #ifndef WIN32
315     if(geteuid() == 0 || getuid() == 0) {
316         fprintf(stderr, "NeonServ may not be run with super user privileges.\n");
317         exit(0);
318     }
319     #endif
320     
321     //load configuration
322     int errid;
323     if((errid = reload_configuration())) {
324         fprintf(stderr, "Unable to load configuration file `%s`. (errid: %d)\n", process_state.config, errid);
325         exit(0);
326     }
327     
328     //check mysql configuration
329     if(!reload_mysql()) {
330         fprintf(stderr, "Unable to load MySQL configuration.\n");
331         exit(0);
332     }
333     
334     //check module configuration
335     char **modulelist = get_all_fieldnames("modules");
336     if(!modulelist || !modulelist[0]) {
337         fprintf(stderr, "Unable to load Module configuration.\n");
338         exit(0);
339     }
340     free(modulelist);
341     
342     #if HAVE_THREADS
343     THREAD_MUTEX_INIT(cache_sync);
344     THREAD_MUTEX_INIT(whohandler_sync);
345     THREAD_MUTEX_INIT(whohandler_mass_sync);
346     #endif
347     
348     //connect to mysql and check if it's the frst bot startup
349     init_mysql();
350     check_firstrun();
351     
352     //deamonize if wanted
353     if(process_state.run_as_daemon)
354         main_daemonize();
355     
356     //set signal handlers
357     signal(SIGABRT, sighandler);
358     signal(SIGFPE, sighandler);
359     signal(SIGILL, sighandler);
360     signal(SIGINT, sighandler);
361     signal(SIGSEGV, sighandler);
362     signal(SIGTERM, sighandler);
363     
364     //set start time and initialize other code parts
365     process_state.running = 1;
366     process_state.start_time = time(0);
367     initialize_subsystems();
368     
369     //start timers
370     timeq_add(CLEAR_CACHE_INTERVAL, 0, clear_cache, NULL);
371     timeq_add(90, 0, main_checkauths, NULL);
372     
373     //start worker threads
374     main_start_threads();  //BLOCKING
375     
376     //shutdown sequence...
377     shutdown_subsystems();
378     if(process_state.restart)
379         main_restart_process(); //terminates the current process on success
380     
381     //eop (end of program :P)
382     //trust me, thats the end!
383     exit(0);
384 }
385
386 /* BOT INFORMATION */
387 time_t getStartTime() {
388     return process_state.start_time;
389 }
390
391 /* BOT CONTROL */
392 void restart_bot(int crash) {
393     if(crash) {
394         main_daemon_exit();
395         main_restart_process();
396     } else {
397         process_state.restart = 1;
398         process_state.running = 0;
399     }
400 }
401
402 void stop_bot() {
403     process_state.running = 0;
404 }
405
406 void reload_config() {
407     reload_configuration();
408 }
409
410 /* TIMER FUNCTIONS */
411
412 static TIMEQ_CALLBACK(clear_cache) {
413     timeq_add(CLEAR_CACHE_INTERVAL, 0, clear_cache, NULL);
414     clearTempUsers();
415     destroyEvents();
416     mysql_free();
417 }
418
419 static AUTHLOOKUP_CALLBACK(main_checkauths_callback) {
420     //check if registered is still valid
421     MYSQL_RES *res;
422     MYSQL_ROW row;
423     printf_mysql_query("SELECT `user_id`, `user_registered` FROM `users` WHERE `user_user` = '%s'", escape_string(auth));
424     res = mysql_use();
425     if ((row = mysql_fetch_row(res)) != NULL) {
426         int diff = registered - atoi(row[1]);
427         if(diff < 0)
428             diff *= -1;
429         if(!exists || (strcmp(row[1], "0") && diff > 86400)) {
430             //User is no longer valid! Delete it...
431             deleteUser(atoi(row[0]));
432             char *alertchan = get_string_field("General.CheckAuths.alertchan");
433             if(alertchan) {
434                 char reason[MAXLEN];
435                 if(!exists) {
436                     strcpy(reason, "USER_NOT_EXISTS");
437                 } else {
438                     sprintf(reason, "USER_REGISTERED_MISSMATCH: %lu, expected %d (diff: %d)", (unsigned long) registered, atoi(row[1]), diff);
439                 }
440                 struct ChanNode *alertchan_chan = getChanByName(alertchan);
441                 struct ClientSocket *alertclient;
442                 if(alertchan_chan && (alertclient = getChannelBot(alertchan_chan, 0)) != NULL) {
443                     putsock(alertclient, "PRIVMSG %s :Deleted User %s (%s)", alertchan_chan->name, auth, reason);
444                 }
445             }
446         } else if(exists && !strcmp(row[1], "0")) {
447             printf_mysql_query("UPDATE `users` SET `user_registered` = '%lu', `user_lastcheck` = UNIX_TIMESTAMP() WHERE `user_id` = '%s'", (unsigned long) registered, row[0]);
448         } else {
449             printf_mysql_query("UPDATE `users` SET `user_lastcheck` = UNIX_TIMESTAMP() WHERE `user_id` = '%s'", row[0]);
450         }
451     }
452 }
453
454 static TIMEQ_CALLBACK(main_checkauths) {
455     int next_call = 600;
456     if(get_int_field("General.CheckAuths.enabled")) {
457         int check_start_time = get_int_field("General.CheckAuths.start_time") * 3600;
458         int duration = get_int_field("General.CheckAuths.duration") * 60;
459         int now = getCurrentSecondsOfDay();
460         if(now < check_start_time && check_start_time+duration >= 86400) {
461             check_start_time -= 86400;
462         }
463         if(now >= check_start_time && now < (check_start_time + duration)) {
464             next_call = get_int_field("General.CheckAuths.interval");
465             //get the "longest-unchecked-user"
466             MYSQL_RES *res;
467             MYSQL_ROW row;
468             int lastcheck;
469             time_t unixtime = time(0);
470             int min_unckecked = get_int_field("General.CheckAuths.min_unckecked");
471             printf_mysql_query("SELECT `user_user`, `user_lastcheck` FROM `users` ORDER BY `user_lastcheck` ASC LIMIT 1");
472             res = mysql_use();
473             if ((row = mysql_fetch_row(res)) != NULL) {
474                 lastcheck = atoi(row[1]);
475                 if(!lastcheck || unixtime - lastcheck >= min_unckecked) {
476                     lookup_authname(row[0], 0, main_checkauths_callback, NULL);
477                 } else 
478                     next_call = 300;
479             }
480         } else {
481             int pending;
482             if(now > check_start_time)
483                 pending = 86400 - now + check_start_time;
484             else
485                 pending = check_start_time - now;
486             if(pending < 600)
487                 next_call = pending;
488         }
489         
490     }
491     timeq_add(next_call, 0, main_checkauths, NULL);
492 }
493
494 /* INSTALLATION SCRIPT */
495
496 static void check_firstrun() {
497     MYSQL_RES *res;
498     MYSQL_ROW row;
499     printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_access` = 1000 LIMIT 1");
500     res = mysql_use();
501     if (mysql_fetch_row(res) == NULL) {
502         //first run!
503         printf("No superuser found...\n");
504         check_firstrun_admin:
505         printf("AuthServ account name of admin user: ");
506         char *ptr;
507         char adminuser[31];
508         ptr = fgets(adminuser, 30, stdin);
509         for(ptr = adminuser; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
510         if(strlen(adminuser) < 2) goto check_firstrun_admin;
511         printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(adminuser));
512         res = mysql_use();
513         if ((row = mysql_fetch_row(res)) != NULL)
514             printf_mysql_query("UPDATE `users` SET `user_access` = 1000 WHERE `user_id` = '%s'", row[0]);
515         else
516             printf_mysql_query("INSERT INTO `users` (`user_user`, `user_access`) VALUES ('%s', 1000)", escape_string(adminuser));
517     }
518     printf_mysql_query("SELECT `id` FROM `bots` WHERE `active` = 1 LIMIT 1");
519     res = mysql_use();
520     if (mysql_fetch_row(res) == NULL) {
521         //no bot active
522         printf("No active bot found...\n\n");
523         printf("ADD NEW BOT\n");
524         char *ptr;
525         char bot_nick[31];
526         check_firstrun_bot_nick:
527         printf("Nick: ");
528         ptr = fgets(bot_nick, 30, stdin);
529         for(ptr = bot_nick; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
530         if(strlen(bot_nick) < 2) goto check_firstrun_bot_nick;
531         char bot_ident[16];
532         check_firstrun_bot_ident:
533         printf("Ident: ");
534         ptr = fgets(bot_ident, 15, stdin);
535         for(ptr = bot_ident; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
536         if(strlen(bot_ident) < 2) goto check_firstrun_bot_ident;
537         char bot_realname[101];
538         check_firstrun_bot_realname:
539         printf("Realname: ");
540         ptr = fgets(bot_realname, 100, stdin);
541         for(ptr = bot_realname; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
542         if(strlen(bot_realname) < 2) goto check_firstrun_bot_realname;
543         char bot_server[101];
544         check_firstrun_bot_server:
545         printf("Server: [irc.onlinegamesnet.net] ");
546         ptr = fgets(bot_server, 100, stdin);
547         for(ptr = bot_server; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
548         if(*bot_server && strlen(bot_nick) < 5) goto check_firstrun_bot_server;
549         if(!*bot_server)
550             strcpy(bot_server, "irc.onlinegamesnet.net");
551         int bot_port;
552         char bot_port_buf[7];
553         printf("Port: [6667] ");
554         ptr = fgets(bot_port_buf, 6, stdin);
555         for(ptr = bot_port_buf; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
556         if(!*bot_port_buf)
557             bot_port = 6667;
558         else
559             bot_port = atoi(bot_port_buf);
560         int bot_ssl;
561         char bot_ssl_buf[5];
562         check_firstrun_bot_ssl:
563         printf("SSL: [y/N] ");
564         ptr = fgets(bot_ssl_buf, 4, stdin);
565         for(ptr = bot_ssl_buf; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
566         if(!*bot_ssl_buf || tolower(*bot_ssl_buf) == 'n')
567             bot_ssl = 0;
568         else if(tolower(*bot_ssl_buf) == 'y')
569             bot_ssl = 1;
570         else
571             goto check_firstrun_bot_ssl;
572         char bot_pass[101];
573         printf("Server Password: [] ");
574         ptr = fgets(bot_pass, 100, stdin);
575         for(ptr = bot_pass; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
576         int bot_maxchan;
577         char bot_maxchan_buf[5];
578         printf("MaxChannel: [20] ");
579         ptr = fgets(bot_maxchan_buf, 5, stdin);
580         for(ptr = bot_maxchan_buf; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
581         if(*bot_maxchan_buf)
582             bot_maxchan = atoi(bot_maxchan_buf);
583         else
584             bot_maxchan = 20;
585         int bot_queue;
586         char bot_queue_buf[5];
587         check_firstrun_bot_queue:
588         printf("Queue (prevents excess floods): [Y/n] ");
589         ptr = fgets(bot_queue_buf, 4, stdin);
590         for(ptr = bot_queue_buf; *ptr; ptr++) { if(*ptr == '\n' || *ptr == '\r') *ptr = '\0'; }
591         if(!*bot_queue_buf || tolower(*bot_queue_buf) == 'y')
592             bot_queue = 1;
593         else if(tolower(*bot_queue_buf) == 'n')
594             bot_queue = 0;
595         else
596             goto check_firstrun_bot_queue;
597         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);
598     }
599 }
600