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 "ClientSocket.h"
  20 #include "UserNode.h"
  21 #include "ChanNode.h"
  22 #include "IRCEvents.h"
  23 #include "IRCParser.h"
  24 #include "modcmd.h"
  25 #include "WHOHandler.h"
  26 #include "bots.h"
  27 #include "mysqlConn.h"
  28 #include "HandleInfoHandler.h"
  29 #include "lang.h"
  30 #include "tools.h"
  31 #include "timeq.h"
  32 #include "EventLogger.h"
  33 #include "ModeNode.h"
  34 #include "IRCQueue.h"
  35 #include "DBHelper.h"
  36 #include "commands.h"
  37 #include "ConfigParser.h"
  38 #include "ssl.h"
  39
  40 time_t start_time;
  41 static int running, hard_restart;
  42 static int statistics_requested_lusers = 0;
  43 int statistics_enabled;
  44 TIMEQ_CALLBACK(main_statistics);
  45 #ifdef HAVE_THREADS
  46 int running_threads;
  47 #endif
  48
  49 void cleanup() {
  50     free_sockets();
  51     free_parser();
  52     free_UserNode();
  53     free_ChanNode();
  54     free_bind();
  55     free_modcmd();
  56     free_whoqueue();
  57     free_bots();
  58     free_mysql();
  59     free_handleinfohandler();
  60     free_lang();
  61 }
  62
  63 static int load_mysql_config() {
  64     char *mysql_host, *mysql_user, *mysql_pass, *mysql_base;
  65     int mysql_serverport;
  66     if(loadConfig("neonserv.conf")) {
  67         mysql_host = get_string_field("MySQL.host");
  68         if(!mysql_host) {
  69             perror("invalid neonserv.conf: missing MySQL.host");
  70             return 0;
  71         }
  72         mysql_serverport = get_int_field("MySQL.port");
  73         if(!mysql_serverport)
  74             mysql_serverport = 3306;
  75         mysql_user = get_string_field("MySQL.user");
  76         if(!mysql_user) {
  77             perror("invalid neonserv.conf: missing MySQL.user");
  78             return 0;
  79         }
  80         mysql_pass = get_string_field("MySQL.pass");
  81         if(!mysql_pass) {
  82             perror("invalid neonserv.conf: missing MySQL.pass");
  83             return 0;
  84         }
  85         mysql_base = get_string_field("MySQL.base");
  86         if(!mysql_base) {
  87             perror("invalid neonserv.conf: missing MySQL base");
  88             return 0;
  89         }
  90     } else {
  91         perror("Unable to load neonserv.conf");
  92         return 0;
  93     }
  94     init_mysql(mysql_host, mysql_serverport, mysql_user, mysql_pass, mysql_base);
  95     return 1;
  96 }
  97
  98 #ifdef HAVE_THREADS
  99 void * thread_main(void *arg) {
 100     time_t socket_wait;
 101     while(running) {
 102         socket_wait = time(0) + SOCKET_SELECT_TIME;
 103         do {
 104             socket_loop(SOCKET_SELECT_TIME);
 105         } while(time(0) < socket_wait);
 106         clearTempUsers();
 107         destroyEvents();
 108     }
 109     running_threads--;
 110     return NULL;
 111 }
 112 #endif
 113
 114 int main(int argc, char *argv[]) {
 115 main:
 116
 117     start_time = time(0);
 118
 119     #ifdef WIN32
 120     int res;
 121     WSADATA wsadata;
 122     // Start Windows Sockets.
 123     res = WSAStartup(MAKEWORD(2, 0), &wsadata);
 124     if (res)
 125     {
 126         perror("Unable to start Windows Sockets");
 127         return 0;
 128     }
 129     #endif
 130
 131     if(!load_mysql_config()) return 0;
 132
 133     statistics_enabled = get_int_field("statistics.enable");
 134
 135     queue_init();
 136     init_sockets();
 137     init_timeq();
 138     init_lang();
 139     ssl_init();
 140     init_parser();
 141     init_UserNode();
 142     init_ChanNode();
 143     init_ModeNode();
 144     init_bind();
 145         init_modcmd();
 146     init_handleinfohandler();
 147     init_tools();
 148     register_commands();
 149     init_bots();
 150     init_DBHelper();
 151
 152     load_languages();
 153     int update_minutes = get_int_field("statistics.frequency");
 154     if(!update_minutes) update_minutes = 2;
 155     timeq_add(update_minutes * 60 + 10, main_statistics, NULL);
 156
 157     int worker_threads = get_int_field("General.worker_threads");
 158     if(!worker_threads) worker_threads = 1;
 159
 160     running = 1;
 161     #ifdef HAVE_THREADS
 162     pthread_t tid[worker_threads];
 163     int tid_id = 0;
 164     for(tid_id = 0; tid_id < worker_threads; tid_id++) {
 165         running_threads++;
 166         pthread_create(&tid[tid_id], NULL, thread_main, NULL);
 167     }
 168     int usleep_delay = 1000000 / TICKS_PER_SECOND;
 169     while(running) {
 170         timeq_tick();
 171         loop_bots();
 172         queue_loop();
 173         usleep(usleep_delay);
 174     }
 175     for(tid_id = 0; tid_id < worker_threads; tid_id++) {
 176         pthread_join(tid[tid_id], NULL);
 177     }
 178     running_threads = 0;
 179     #else
 180     time_t socket_wait;
 181     while(running) {
 182         socket_wait = time(0) + SOCKET_SELECT_TIME;
 183         do {
 184             socket_loop(SOCKET_SELECT_TIME);
 185         } while(time(0) < socket_wait);
 186         timeq_tick();
 187         loop_bots();
 188         clearTempUsers();
 189         destroyEvents();
 190         queue_loop();
 191     }
 192     #endif
 193     cleanup();
 194     if(hard_restart) {
 195         /* Append a NULL to the end of argv[]. */
 196         char **restart_argv = (char **)alloca((argc + 1) * sizeof(char *));
 197         memcpy(restart_argv, argv, argc * sizeof(char *));
 198         restart_argv[argc] = NULL;
 199
 200         #ifdef WIN32
 201         execv(argv[0], (const char * const*)restart_argv);
 202         #else
 203         execv(argv[0], restart_argv);
 204         #endif
 205     }
 206     goto main;
 207 }
 208
 209 int stricmp (const char *s1, const char *s2)
 210 {
 211    if (s1 == NULL) return s2 == NULL ? 0 : -(*s2);
 212    if (s2 == NULL) return *s1;
 213    char c1, c2;
 214    while ((c1 = tolower (*s1)) == (c2 = tolower (*s2)))
 215    {
 216      if (*s1 == '\0') break;
 217      ++s1; ++s2;
 218    }
 219    return c1 - c2;
 220 }
 221
 222 int stricmplen (const char *s1, const char *s2, int len)
 223 {
 224    if (s1 == NULL) return s2 == NULL ? 0 : -(*s2);
 225    if (s2 == NULL) return *s1;
 226    char c1, c2;
 227    int i = 0;
 228    while ((c1 = tolower (*s1)) == (c2 = tolower (*s2)))
 229    {
 230      i++;
 231      if (*s1 == '\0') break;
 232      ++s1; ++s2;
 233      if(i == len) break;
 234    }
 235    return c1 - c2;
 236 }
 237
 238 void restart_bot(int do_hard_restart) {
 239     hard_restart = do_hard_restart;
 240     running = 0;
 241 }
 242
 243 void stop_bot() {
 244     cleanup();
 245     exit(0);
 246 }
 247
 248 void reload_config() {
 249     loadConfig("neonserv.conf");
 250 }
 251
 252 TIMEQ_CALLBACK(main_statistics) {
 253     int update_minutes = get_int_field("statistics.frequency");
 254     if(!update_minutes) update_minutes = 2;
 255     timeq_add(update_minutes * 60, main_statistics, NULL);
 256     if(get_int_field("statistics.enable")) {
 257         statistics_enabled = 1;
 258         statistics_requested_lusers = 1;
 259         if(get_int_field("statistics.include_lusers")) {
 260             struct ClientSocket *bot, *lusersbot = NULL;
 261             for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 262                 if(bot->flags & SOCKET_FLAG_PREFERRED)
 263                     lusersbot = bot;
 264             }
 265             bot = lusersbot;
 266             if(bot == NULL) bot = getBots(SOCKET_FLAG_READY, NULL);
 267             putsock(bot, "LUSERS");
 268         } else {
 269             statistics_update();
 270         }
 271     } else
 272         statistics_enabled = 0;
 273 }
 274
 275 void statistics_update() {
 276     if(get_int_field("statistics.enable") && statistics_requested_lusers && get_string_field("statistics.execute")) {
 277         statistics_requested_lusers = 0;
 278         char command[MAXLEN];
 279         /* parameters:
 280          - visible users
 281          - visible chanusers
 282          - visible channels
 283          - privmsg per minute
 284          - commands per minute
 285          - network users
 286          - network channels
 287         */
 288         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);
 289         statistics_privmsg = 0;
 290         statistics_commands = 0;
 291         system(command);
 292     }
 293 }