X-Git-Url: http://git.pk910.de/?p=NeonServV5.git;a=blobdiff_plain;f=src%2Fbots.c;h=7da678c97ffb3b4a185f9b0dea6b74f3c69431ae;hp=2eb81d4f081e78d51b295dec2a6b3e0ebea69f9b;hb=3218b969d5dd33bbd8defb30059d666f33284779;hpb=b6a3428c1a0d04174d1f8ff93ac31a861c9a2e51 diff --git a/src/bots.c b/src/bots.c index 2eb81d4..7da678c 100644 --- a/src/bots.c +++ b/src/bots.c @@ -1,4 +1,4 @@ -/* bots.c - NeonServ v5.4 +/* bots.c - NeonServ v5.6 * Copyright (C) 2011-2012 Philipp Kreil (pk910) * * This program is free software: you can redistribute it and/or modify @@ -26,6 +26,8 @@ #include "modcmd.h" #include "DBHelper.h" #include "IRCEvents.h" +#include "tools.h" +#include "log.h" struct cmd_bot_alias { int botid; @@ -40,7 +42,7 @@ static void start_zero_bots() { MYSQL_RES *res, *res2; MYSQL_ROW row; - printf_mysql_query("SELECT `nick`, `ident`, `realname`, `server`, `port`, `pass`, `textbot`, `id`, `queue`, `ssl`, `bind` FROM `bots` WHERE `botclass` = '0' AND `active` = '1'"); + printf_mysql_query("SELECT `nick`, `ident`, `realname`, `server`, `port`, `pass`, `textbot`, `id`, `queue`, `ssl`, `bind`, `secret` FROM `bots` WHERE `botclass` = '0' AND `active` = '1'"); res = mysql_use(); while ((row = mysql_fetch_row(res)) != NULL) { @@ -48,6 +50,7 @@ static void start_zero_bots() { client->flags |= (strcmp(row[6], "0") ? SOCKET_FLAG_PREFERRED : 0); client->flags |= (strcmp(row[8], "0") ? SOCKET_FLAG_USE_QUEUE : 0); client->flags |= (strcmp(row[9], "0") ? SOCKET_FLAG_SSL : 0); + client->flags |= (strcmp(row[11], "0") ? SOCKET_FLAG_SECRET_BOT : 0); client->botid = 0; client->clientid = atoi(row[7]); connect_socket(client); @@ -116,16 +119,11 @@ static void zero_bots_bot_ready(struct ClientSocket *client) { } } -void init_bots() { - set_bot_alias(0, "0"); - start_zero_bots(); - set_trigger_callback(0, 0, zero_bots_trigger_callback); - bind_bot_ready(zero_bots_bot_ready, 0); - +static TIMEQ_CALLBACK(load_timed_bans) { MYSQL_RES *res; MYSQL_ROW row; - //load all timed bans - printf_mysql_query("SELECT `ban_id`, `ban_timeout` FROM `bans` WHERE `ban_timeout` > 0"); + //load all timed bans for the next 7 days + printf_mysql_query("SELECT `ban_id`, `ban_timeout` FROM `bans` WHERE `ban_timeout` > 0 AND `ban_timeout` < (UNIX_TIMESTAMP() + (86400 * 7))"); res = mysql_use(); char nameBuf[20]; while ((row = mysql_fetch_row(res)) != NULL) { @@ -137,6 +135,17 @@ void init_bots() { printf_mysql_query("DELETE FROM `bans` WHERE `ban_id` = '%s'", row[0]); } } + timeq_add(86400*7, 0, load_timed_bans, NULL); +} + +void init_bots() { + set_bot_alias(0, "0"); + start_zero_bots(); + set_trigger_callback(0, 0, zero_bots_trigger_callback); + bind_bot_ready(zero_bots_bot_ready, 0); + + timeq_add(10, 0, load_timed_bans, NULL); + } struct ClientSocket *getChannelBot(struct ChanNode *chan, int botid) { @@ -144,8 +153,8 @@ struct ClientSocket *getChannelBot(struct ChanNode *chan, int botid) { struct ChanUser *chanuser; for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) { if(botid && bot->botid != botid) continue; - if((chanuser = getChanUser(bot->user, chan)) != NULL) { - if((chanuser->flags & CHANUSERFLAG_OPPED)) { + if(!chan || (chanuser = getChanUser(bot->user, chan)) != NULL) { + if(chan && (chanuser->flags & CHANUSERFLAG_OPPED)) { use_bot = bot; if(bot->flags & SOCKET_FLAG_PREFERRED) break; } else if(bot->flags & SOCKET_FLAG_PREFERRED) @@ -183,7 +192,7 @@ void requestOp(struct UserNode *user, struct ChanNode *chan) { } void requestInvite(struct UserNode *user, struct ChanNode *chan) { - struct ClientSocket *bot, *userbot = NULL; + struct ClientSocket *bot; struct ChanUser *chanuser = getChanUser(user, chan); char invited = 0; if(chanuser) return; @@ -192,9 +201,6 @@ void requestInvite(struct UserNode *user, struct ChanNode *chan) { invited = 1; putsock(bot, "INVITE %s %s", user->nick, chan->name); } - if(bot->user == user) { - userbot = bot; - } } } @@ -270,7 +276,7 @@ void set_bot_alias(int botid, char *alias) { } botalias = malloc(sizeof(*botalias)); if (!botalias) { - perror("malloc() failed"); + printf_log("main", LOG_ERROR, "%s:%d malloc() failed", __FILE__, __LINE__); return; } botalias->botid = botid;