1 /* main.c - NeonServ v5.3
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/>.
19 #include "ClientSocket.h"
22 #include "IRCEvents.h"
23 #include "IRCParser.h"
25 #include "WHOHandler.h"
27 #include "mysqlConn.h"
28 #include "HandleInfoHandler.h"
32 #include "EventLogger.h"
37 #include "ConfigParser.h"
42 static int running, hard_restart;
43 static int statistics_requested_lusers = 0;
44 int statistics_enabled;
45 TIMEQ_CALLBACK(main_statistics);
46 TIMEQ_CALLBACK(main_checkauths);
49 pthread_mutex_t cache_sync;
50 pthread_mutex_t whohandler_sync, whohandler_mass_sync;
64 free_handleinfohandler();
68 static int load_mysql_config() {
69 char *mysql_host, *mysql_user, *mysql_pass, *mysql_base;
71 if(loadConfig("neonserv.conf")) {
72 mysql_host = get_string_field("MySQL.host");
74 perror("invalid neonserv.conf: missing MySQL.host");
77 mysql_serverport = get_int_field("MySQL.port");
79 mysql_serverport = 3306;
80 mysql_user = get_string_field("MySQL.user");
82 perror("invalid neonserv.conf: missing MySQL.user");
85 mysql_pass = get_string_field("MySQL.pass");
87 perror("invalid neonserv.conf: missing MySQL.pass");
90 mysql_base = get_string_field("MySQL.base");
92 perror("invalid neonserv.conf: missing MySQL base");
96 perror("Unable to load neonserv.conf");
99 init_mysql(mysql_host, mysql_serverport, mysql_user, mysql_pass, mysql_base);
104 pthread_t *current_threads = NULL;
106 void * thread_main(void *arg) {
109 socket_wait = time(0) + SOCKET_SELECT_TIME;
111 socket_loop(SOCKET_SELECT_TIME);
112 } while(time(0) < socket_wait);
120 int getCurrentThreadID() {
121 if(!current_threads) return 0;
123 unsigned int my_tid = (unsigned int) pthread_self_tid();
124 for(i = 0; i < running_threads; i++) {
126 if((unsigned int) current_threads[i].p == my_tid)
128 if((unsigned int) current_threads[i] == my_tid)
137 int main(int argc, char *argv[]) {
140 start_time = time(0);
145 // Start Windows Sockets.
146 res = WSAStartup(MAKEWORD(2, 0), &wsadata);
149 perror("Unable to start Windows Sockets");
154 if(!load_mysql_config()) return 0;
156 statistics_enabled = get_int_field("statistics.enable");
159 THREAD_MUTEX_INIT(cache_sync);
160 THREAD_MUTEX_INIT(whohandler_sync);
161 THREAD_MUTEX_INIT(whohandler_mass_sync);
175 init_handleinfohandler();
183 int update_minutes = get_int_field("statistics.frequency");
184 if(!update_minutes) update_minutes = 2;
185 timeq_add(update_minutes * 60 + 10, main_statistics, NULL);
187 timeq_add(90, main_checkauths, NULL);
189 int worker_threads = get_int_field("General.worker_threads");
190 if(!worker_threads) worker_threads = 1;
194 current_threads = calloc(worker_threads, sizeof(*current_threads));
195 for(tid_id = 0; tid_id < worker_threads; tid_id++) {
197 pthread_create(¤t_threads[tid_id], NULL, thread_main, NULL);
199 int usleep_delay = 1000000 / TICKS_PER_SECOND;
205 usleep(usleep_delay);
207 for(tid_id = 0; tid_id < worker_threads; tid_id++) {
208 pthread_join(current_threads[tid_id], NULL);
214 socket_wait = time(0) + SOCKET_SELECT_TIME;
216 socket_loop(SOCKET_SELECT_TIME);
217 } while(time(0) < socket_wait);
228 /* Append a NULL to the end of argv[]. */
229 char **restart_argv = (char **)alloca((argc + 1) * sizeof(char *));
230 memcpy(restart_argv, argv, argc * sizeof(char *));
231 restart_argv[argc] = NULL;
234 execv(argv[0], (const char * const*)restart_argv);
236 execv(argv[0], restart_argv);
242 int stricmp (const char *s1, const char *s2)
244 if (s1 == NULL) return s2 == NULL ? 0 : -(*s2);
245 if (s2 == NULL) return *s1;
247 while ((c1 = tolower (*s1)) == (c2 = tolower (*s2)))
249 if (*s1 == '\0') break;
255 int stricmplen (const char *s1, const char *s2, int len)
257 if (s1 == NULL) return s2 == NULL ? 0 : -(*s2);
258 if (s2 == NULL) return *s1;
261 while ((c1 = tolower (*s1)) == (c2 = tolower (*s2)))
264 if (*s1 == '\0') break;
271 void restart_bot(int do_hard_restart) {
272 hard_restart = do_hard_restart;
281 void reload_config() {
282 loadConfig("neonserv.conf");
285 static int getCurrentSecondsOfDay() {
286 time_t now = time(0);
287 struct tm *timeofday = localtime(&now);
289 seconds += timeofday->tm_hour * 3600;
290 seconds += timeofday->tm_min * 60;
291 seconds += timeofday->tm_sec;
295 static AUTHLOOKUP_CALLBACK(main_checkauths_callback) {
296 //check if registered is still valid
299 printf_mysql_query("SELECT `user_id`, `user_registered` FROM `users` WHERE `user_user` = '%s'", escape_string(auth));
301 if ((row = mysql_fetch_row(res)) != NULL) {
302 if(!exists || (strcmp(row[1], "0") && registered != atoi(row[1]))) {
303 //User is no longer valid! Delete it...
304 deleteUser(atoi(row[0]));
305 char *alertchan = get_string_field("General.CheckAuths.alertchan");
307 struct ChanNode *alertchan_chan = getChanByName(alertchan);
308 struct ClientSocket *alertclient;
309 if(alertchan_chan && (alertclient = getChannelBot(alertchan_chan, 0)) != NULL) {
310 putsock(alertclient, "PRIVMSG %s :Deleted User %s", alertchan_chan->name, auth);
313 } else if(exists && !strcmp(row[1], "0")) {
314 printf_mysql_query("UPDATE `users` SET `user_registered` = '%lu', `user_lastcheck` = UNIX_TIMESTAMP() WHERE `user_id` = '%s'", (unsigned long) registered, row[0]);
316 printf_mysql_query("UPDATE `users` SET `user_lastcheck` = UNIX_TIMESTAMP() WHERE `user_id` = '%s'", row[0]);
321 TIMEQ_CALLBACK(main_checkauths) {
323 if(get_int_field("General.CheckAuths.enabled")) {
324 int check_start_time = get_int_field("General.CheckAuths.start_time") * 3600;
325 int duration = get_int_field("General.CheckAuths.duration") * 60;
326 int now = getCurrentSecondsOfDay();
327 if(now < check_start_time && check_start_time+duration >= 86400) {
328 check_start_time -= 86400;
330 if(now >= check_start_time && now < (check_start_time + duration)) {
331 next_call = get_int_field("General.CheckAuths.interval");
332 //get the "longest-unchecked-user"
336 time_t unixtime = time(0);
337 int min_unckecked = get_int_field("General.CheckAuths.min_unckecked");
338 printf_mysql_query("SELECT `user_user`, `user_lastcheck` FROM `users` ORDER BY `user_lastcheck` ASC LIMIT 1");
340 if ((row = mysql_fetch_row(res)) != NULL) {
341 lastcheck = atoi(row[1]);
342 if(!lastcheck || unixtime - lastcheck >= min_unckecked) {
343 lookup_authname(row[0], main_checkauths_callback, NULL);
349 if(now > check_start_time)
350 pending = 86400 - now + check_start_time;
352 pending = check_start_time - now;
358 timeq_add(next_call, main_checkauths, NULL);
361 TIMEQ_CALLBACK(main_statistics) {
362 int update_minutes = get_int_field("statistics.frequency");
363 if(!update_minutes) update_minutes = 2;
364 timeq_add(update_minutes * 60, main_statistics, NULL);
365 if(get_int_field("statistics.enable")) {
366 statistics_enabled = 1;
367 statistics_requested_lusers = 1;
368 if(get_int_field("statistics.include_lusers")) {
369 struct ClientSocket *bot, *lusersbot = NULL;
370 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
371 if(bot->flags & SOCKET_FLAG_PREFERRED)
375 if(bot == NULL) bot = getBots(SOCKET_FLAG_READY, NULL);
376 putsock(bot, "LUSERS");
381 statistics_enabled = 0;
384 void statistics_update() {
385 if(get_int_field("statistics.enable") && statistics_requested_lusers && get_string_field("statistics.execute")) {
386 statistics_requested_lusers = 0;
387 char command[MAXLEN];
393 - commands per minute
397 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);
398 statistics_privmsg = 0;
399 statistics_commands = 0;