src/main.c

   1 /* main.c - NeonServ v5.3
   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 #include "main.h"
  19 #include "signal.h"
  20 #include "ClientSocket.h"
  21 #include "UserNode.h"
  22 #include "ChanNode.h"
  23 #include "IRCEvents.h"
  24 #include "IRCParser.h"
  25 #include "modcmd.h"
  26 #include "WHOHandler.h"
  27 #include "bots.h"
  28 #include "mysqlConn.h"
  29 #include "HandleInfoHandler.h"
  30 #include "lang.h"
  31 #include "tools.h"
  32 #include "timeq.h"
  33 #include "EventLogger.h"
  34 #include "ModeNode.h"
  35 #include "IRCQueue.h"
  36 #include "DBHelper.h"
  37 #include "commands.h"
  38 #include "ConfigParser.h"
  39 #include "ssl.h"
  40 #include "QServer.h"
  41
  42 time_t start_time;
  43 static int running, hard_restart;
  44 static int statistics_requested_lusers = 0;
  45 int statistics_enabled;
  46 TIMEQ_CALLBACK(main_statistics);
  47 TIMEQ_CALLBACK(main_checkauths);
  48 static int process_argc;
  49 static char **process_argv;
  50 #ifdef HAVE_THREADS
  51 int running_threads;
  52 pthread_mutex_t cache_sync;
  53 pthread_mutex_t whohandler_sync, whohandler_mass_sync;
  54 #endif
  55
  56 void cleanup() {
  57     free_sockets();
  58     qserver_free();
  59     free_parser();
  60     free_UserNode();
  61     free_ChanNode();
  62     free_bind();
  63     free_modcmd();
  64     free_whoqueue();
  65     free_bots();
  66     free_mysql();
  67     free_handleinfohandler();
  68     free_lang();
  69 }
  70
  71 static int load_mysql_config() {
  72     char *mysql_host, *mysql_user, *mysql_pass, *mysql_base;
  73     int mysql_serverport;
  74     if(loadConfig("neonserv.conf")) {
  75         mysql_host = get_string_field("MySQL.host");
  76         if(!mysql_host) {
  77             perror("invalid neonserv.conf: missing MySQL.host");
  78             return 0;
  79         }
  80         mysql_serverport = get_int_field("MySQL.port");
  81         if(!mysql_serverport)
  82             mysql_serverport = 3306;
  83         mysql_user = get_string_field("MySQL.user");
  84         if(!mysql_user) {
  85             perror("invalid neonserv.conf: missing MySQL.user");
  86             return 0;
  87         }
  88         mysql_pass = get_string_field("MySQL.pass");
  89         if(!mysql_pass) {
  90             perror("invalid neonserv.conf: missing MySQL.pass");
  91             return 0;
  92         }
  93         mysql_base = get_string_field("MySQL.base");
  94         if(!mysql_base) {
  95             perror("invalid neonserv.conf: missing MySQL base");
  96             return 0;
  97         }
  98     } else {
  99         perror("Unable to load neonserv.conf");
 100         return 0;
 101     }
 102     init_mysql(mysql_host, mysql_serverport, mysql_user, mysql_pass, mysql_base);
 103     return 1;
 104 }
 105
 106 #ifdef HAVE_THREADS
 107 pthread_t *current_threads = NULL;
 108
 109 void * thread_main(void *arg) {
 110     time_t socket_wait;
 111     while(running) {
 112         socket_wait = time(0) + SOCKET_SELECT_TIME;
 113         do {
 114             socket_loop(SOCKET_SELECT_TIME);
 115         } while(time(0) < socket_wait);
 116         clearTempUsers();
 117         destroyEvents();
 118     }
 119     running_threads--;
 120     return NULL;
 121 }
 122
 123 int getCurrentThreadID() {
 124     if(!current_threads) return 0;
 125     int i;
 126     unsigned int my_tid = (unsigned int) pthread_self_tid();
 127     for(i = 0; i < running_threads; i++) {
 128         #ifdef WIN32
 129         if((unsigned int) current_threads[i].p == my_tid)
 130         #else
 131         if((unsigned int) current_threads[i] == my_tid)
 132         #endif
 133             return i+1;
 134     }
 135     return 0;
 136 }
 137
 138 #endif
 139
 140 int main(int argc, char *argv[]) {
 141 main:
 142     process_argv = argv;
 143     process_argc = argc;
 144
 145     signal(SIGABRT, sighandler);
 146     signal(SIGFPE, sighandler);
 147     signal(SIGILL, sighandler);
 148     signal(SIGINT, sighandler);
 149     signal(SIGSEGV, sighandler);
 150     signal(SIGTERM, sighandler);
 151
 152     start_time = time(0);
 153
 154     #ifdef WIN32
 155     int res;
 156     WSADATA wsadata;
 157     // Start Windows Sockets.
 158     res = WSAStartup(MAKEWORD(2, 0), &wsadata);
 159     if (res)
 160     {
 161         perror("Unable to start Windows Sockets");
 162         return 0;
 163     }
 164     #endif
 165
 166     if(!load_mysql_config()) return 0;
 167
 168     statistics_enabled = get_int_field("statistics.enable");
 169
 170     #ifdef HAVE_THREADS
 171     THREAD_MUTEX_INIT(cache_sync);
 172     THREAD_MUTEX_INIT(whohandler_sync);
 173     THREAD_MUTEX_INIT(whohandler_mass_sync);
 174     #endif
 175
 176     queue_init();
 177     init_sockets();
 178     init_timeq();
 179     init_lang();
 180     ssl_init();
 181     init_parser();
 182     init_UserNode();
 183     init_ChanNode();
 184     init_ModeNode();
 185     init_bind();
 186         init_modcmd();
 187     init_handleinfohandler();
 188     init_tools();
 189     register_commands();
 190     init_bots();
 191     init_DBHelper();
 192     qserver_init();
 193
 194     load_languages();
 195     int update_minutes = get_int_field("statistics.frequency");
 196     if(!update_minutes) update_minutes = 2;
 197     timeq_add(update_minutes * 60 + 10, main_statistics, NULL);
 198
 199     timeq_add(90, main_checkauths, NULL);
 200
 201     int worker_threads = get_int_field("General.worker_threads");
 202     if(!worker_threads) worker_threads = 1;
 203     running = 1;
 204     #ifdef HAVE_THREADS
 205     int tid_id = 0;
 206     current_threads = calloc(worker_threads, sizeof(*current_threads));
 207     for(tid_id = 0; tid_id < worker_threads; tid_id++) {
 208         running_threads++;
 209         pthread_create(&current_threads[tid_id], NULL, thread_main, NULL);
 210     }
 211     int usleep_delay = 1000000 / TICKS_PER_SECOND;
 212     while(running) {
 213         timeq_tick();
 214         loop_bots();
 215         qserver_loop();
 216         queue_loop();
 217         usleep(usleep_delay);
 218     }
 219     for(tid_id = 0; tid_id < worker_threads; tid_id++) {
 220         pthread_join(current_threads[tid_id], NULL);
 221     }
 222     running_threads = 0;
 223     #else
 224     time_t socket_wait;
 225     while(running) {
 226         socket_wait = time(0) + SOCKET_SELECT_TIME;
 227         do {
 228             socket_loop(SOCKET_SELECT_TIME);
 229         } while(time(0) < socket_wait);
 230         timeq_tick();
 231         loop_bots();
 232         clearTempUsers();
 233         destroyEvents();
 234         qserver_loop();
 235         queue_loop();
 236     }
 237     #endif
 238     cleanup();
 239     if(hard_restart) {
 240         restart_process();
 241     }
 242     goto main;
 243 }
 244
 245 void restart_process() {
 246     /* Append a NULL to the end of argv[]. */
 247     char **restart_argv = (char **)alloca((process_argc + 1) * sizeof(char *));
 248     memcpy(restart_argv, process_argv, process_argc * sizeof(char *));
 249     restart_argv[process_argc] = NULL;
 250     #ifdef WIN32
 251     execv(process_argv[0], (const char * const*)restart_argv);
 252     #else
 253     execv(process_argv[0], restart_argv);
 254     #endif
 255 }
 256
 257 int stricmp (const char *s1, const char *s2)
 258 {
 259    if (s1 == NULL) return s2 == NULL ? 0 : -(*s2);
 260    if (s2 == NULL) return *s1;
 261    char c1, c2;
 262    while ((c1 = tolower (*s1)) == (c2 = tolower (*s2)))
 263    {
 264      if (*s1 == '\0') break;
 265      ++s1; ++s2;
 266    }
 267    return c1 - c2;
 268 }
 269
 270 int stricmplen (const char *s1, const char *s2, int len)
 271 {
 272    if (s1 == NULL) return s2 == NULL ? 0 : -(*s2);
 273    if (s2 == NULL) return *s1;
 274    char c1, c2;
 275    int i = 0;
 276    while ((c1 = tolower (*s1)) == (c2 = tolower (*s2)))
 277    {
 278      i++;
 279      if (*s1 == '\0') break;
 280      ++s1; ++s2;
 281      if(i == len) break;
 282    }
 283    return c1 - c2;
 284 }
 285
 286 void restart_bot(int do_hard_restart) {
 287     hard_restart = do_hard_restart;
 288     running = 0;
 289 }
 290
 291 void stop_bot() {
 292     cleanup();
 293     exit(0);
 294 }
 295
 296 void reload_config() {
 297     loadConfig("neonserv.conf");
 298 }
 299
 300 static int getCurrentSecondsOfDay() {
 301     time_t now = time(0);
 302     struct tm *timeofday = localtime(&now);
 303     int seconds = 0;
 304     seconds += timeofday->tm_hour * 3600;
 305     seconds += timeofday->tm_min * 60;
 306     seconds += timeofday->tm_sec;
 307     return seconds;
 308 }
 309
 310 static AUTHLOOKUP_CALLBACK(main_checkauths_callback) {
 311     //check if registered is still valid
 312     MYSQL_RES *res;
 313     MYSQL_ROW row;
 314     printf_mysql_query("SELECT `user_id`, `user_registered` FROM `users` WHERE `user_user` = '%s'", escape_string(auth));
 315     res = mysql_use();
 316     if ((row = mysql_fetch_row(res)) != NULL) {
 317         if(!exists || (strcmp(row[1], "0") && registered != atoi(row[1]))) {
 318             //User is no longer valid! Delete it...
 319             deleteUser(atoi(row[0]));
 320             char *alertchan = get_string_field("General.CheckAuths.alertchan");
 321             if(alertchan) {
 322                 struct ChanNode *alertchan_chan = getChanByName(alertchan);
 323                 struct ClientSocket *alertclient;
 324                 if(alertchan_chan && (alertclient = getChannelBot(alertchan_chan, 0)) != NULL) {
 325                     putsock(alertclient, "PRIVMSG %s :Deleted User %s", alertchan_chan->name, auth);
 326                 }
 327             }
 328         } else if(exists && !strcmp(row[1], "0")) {
 329             printf_mysql_query("UPDATE `users` SET `user_registered` = '%lu', `user_lastcheck` = UNIX_TIMESTAMP() WHERE `user_id` = '%s'", (unsigned long) registered, row[0]);
 330         } else {
 331             printf_mysql_query("UPDATE `users` SET `user_lastcheck` = UNIX_TIMESTAMP() WHERE `user_id` = '%s'", row[0]);
 332         }
 333     }
 334 }
 335
 336 TIMEQ_CALLBACK(main_checkauths) {
 337     int next_call = 600;
 338     if(get_int_field("General.CheckAuths.enabled")) {
 339         int check_start_time = get_int_field("General.CheckAuths.start_time") * 3600;
 340         int duration = get_int_field("General.CheckAuths.duration") * 60;
 341         int now = getCurrentSecondsOfDay();
 342         if(now < check_start_time && check_start_time+duration >= 86400) {
 343             check_start_time -= 86400;
 344         }
 345         if(now >= check_start_time && now < (check_start_time + duration)) {
 346             next_call = get_int_field("General.CheckAuths.interval");
 347             //get the "longest-unchecked-user"
 348             MYSQL_RES *res;
 349             MYSQL_ROW row;
 350             int lastcheck;
 351             time_t unixtime = time(0);
 352             int min_unckecked = get_int_field("General.CheckAuths.min_unckecked");
 353             printf_mysql_query("SELECT `user_user`, `user_lastcheck` FROM `users` ORDER BY `user_lastcheck` ASC LIMIT 1");
 354             res = mysql_use();
 355             if ((row = mysql_fetch_row(res)) != NULL) {
 356                 lastcheck = atoi(row[1]);
 357                 if(!lastcheck || unixtime - lastcheck >= min_unckecked) {
 358                     lookup_authname(row[0], main_checkauths_callback, NULL);
 359                 } else
 360                     next_call = 300;
 361             }
 362         } else {
 363             int pending;
 364             if(now > check_start_time)
 365                 pending = 86400 - now + check_start_time;
 366             else
 367                 pending = check_start_time - now;
 368             if(pending < 600)
 369                 next_call = pending;
 370         }
 371
 372     }
 373     timeq_add(next_call, main_checkauths, NULL);
 374 }
 375
 376 TIMEQ_CALLBACK(main_statistics) {
 377     int update_minutes = get_int_field("statistics.frequency");
 378     if(!update_minutes) update_minutes = 2;
 379     timeq_add(update_minutes * 60, main_statistics, NULL);
 380     if(get_int_field("statistics.enable")) {
 381         statistics_enabled = 1;
 382         statistics_requested_lusers = 1;
 383         if(get_int_field("statistics.include_lusers")) {
 384             struct ClientSocket *bot, *lusersbot = NULL;
 385             for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 386                 if(bot->flags & SOCKET_FLAG_PREFERRED)
 387                     lusersbot = bot;
 388             }
 389             bot = lusersbot;
 390             if(bot == NULL) bot = getBots(SOCKET_FLAG_READY, NULL);
 391             putsock(bot, "LUSERS");
 392         } else {
 393             statistics_update();
 394         }
 395     } else
 396         statistics_enabled = 0;
 397 }
 398
 399 void statistics_update() {
 400     if(get_int_field("statistics.enable") && statistics_requested_lusers && get_string_field("statistics.execute")) {
 401         statistics_requested_lusers = 0;
 402         char command[MAXLEN];
 403         /* parameters:
 404          - visible users
 405          - visible chanusers
 406          - visible channels
 407          - privmsg per minute
 408          - commands per minute
 409          - network users
 410          - network channels
 411         */
 412         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);
 413         statistics_privmsg = 0;
 414         statistics_commands = 0;
 415         system(command);
 416     }
 417 }