src/bots.c

   1 /* bots.c - NeonServ v5.4
   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 "bots.h"
  19 #include "timeq.h"
  20 #include "mysqlConn.h"
  21 #include "ClientSocket.h"
  22 #include "UserNode.h"
  23 #include "ChanNode.h"
  24 #include "ChanUser.h"
  25 #include "version.h"
  26 #include "modcmd.h"
  27 #include "DBHelper.h"
  28
  29 struct cmd_bot_alias {
  30     int botid;
  31     char *alias;
  32     struct cmd_bot_alias *next;
  33 };
  34
  35 static struct cmd_bot_alias *bot_aliases = NULL;
  36
  37 static void start_zero_bots() {
  38     struct ClientSocket *client;
  39     MYSQL_RES *res, *res2;
  40     MYSQL_ROW row;
  41
  42     printf_mysql_query("SELECT `nick`, `ident`, `realname`, `server`, `port`, `pass`, `textbot`, `id`, `queue`, `ssl`, `bind` FROM `bots` WHERE `botclass` = '0' AND `active` = '1'");
  43     res = mysql_use();
  44
  45     while ((row = mysql_fetch_row(res)) != NULL) {
  46         client = create_socket(row[3], atoi(row[4]), row[10], row[5], row[0], row[1], row[2]);
  47         client->flags |= (strcmp(row[6], "0") ? SOCKET_FLAG_PREFERRED : 0);
  48         client->flags |= (strcmp(row[8], "0") ? SOCKET_FLAG_USE_QUEUE : 0);
  49         client->flags |= (strcmp(row[9], "0") ? SOCKET_FLAG_SSL : 0);
  50         client->botid = 0;
  51         client->clientid = atoi(row[7]);
  52         connect_socket(client);
  53
  54         printf_mysql_query("SELECT `command`, `function`, `parameters`, `global_access`, `chan_access`, `flags` FROM `bot_binds` WHERE `botclass` = '0' AND `botid` = '%d'", client->clientid);
  55         res2 = mysql_use();
  56         while ((row = mysql_fetch_row(res2)) != NULL) {
  57             if(bind_botwise_cmd_to_command(0, client->clientid, row[0], row[1])) {
  58                 if(row[2] && strcmp(row[2], "")) {
  59                     bind_botwise_set_parameters(0, client->clientid, row[0], row[2]);
  60                 }
  61                 if(row[3]) {
  62                     bind_botwise_set_global_access(0, client->clientid, row[0], atoi(row[3]));
  63                 }
  64                 if(row[4]) {
  65                     bind_botwise_set_channel_access(0, client->clientid, row[0], row[4]);
  66                 }
  67                 if(strcmp(row[5], "0"))
  68                     bind_botwise_set_bind_flags(0, client->clientid, row[0], atoi(row[5]));
  69             }
  70         }
  71         bind_botwise_unbound_required_functions(0, client->clientid);
  72     }
  73 }
  74
  75 static void zero_bots_trigger_callback(int clientid, struct ChanNode *chan, char *trigger) {
  76     MYSQL_RES *res;
  77     MYSQL_ROW row;
  78     loadChannelSettings(chan);
  79     if(!(chan->flags & CHANFLAG_CHAN_REGISTERED)) {
  80         strcpy(trigger, "+");
  81         return;
  82     }
  83     printf_mysql_query("SELECT `trigger`, `defaulttrigger` FROM `bot_channels` LEFT JOIN `bots` ON `botid` = `bots`.`id` WHERE `chanid` = '%d' AND `botclass` = '0' AND `botid` = '%d'", chan->channel_id, clientid);
  84     res = mysql_use();
  85     if(!(row = mysql_fetch_row(res))) {
  86         strcpy(trigger, "");
  87         return;
  88     }
  89     if(row[0] && *row[0])
  90         strcpy(trigger, row[0]);
  91     else
  92         strcpy(trigger, ((row[1] && *row[1]) ? row[1] : "+"));
  93 }
  94
  95 void init_bots() {
  96     set_bot_alias(0, "0");
  97     start_zero_bots();
  98     set_trigger_callback(0, 0, zero_bots_trigger_callback);
  99
 100     MYSQL_RES *res;
 101     MYSQL_ROW row;
 102     //load all timed bans
 103     printf_mysql_query("SELECT `ban_id`, `ban_timeout` FROM `bans` WHERE `ban_timeout` > 0");
 104     res = mysql_use();
 105     char nameBuf[20];
 106     while ((row = mysql_fetch_row(res)) != NULL) {
 107         if(atol(row[1]) - time(0) > 0) {
 108             sprintf(nameBuf, "ban_%s", row[0]);
 109             timeq_add_name(nameBuf, atol(row[1]) - time(0), 0, channel_ban_timeout, strdup(row[0]));
 110         } else {
 111             //timed out
 112             printf_mysql_query("DELETE FROM `bans` WHERE `ban_id` = '%s'", row[0]);
 113         }
 114     }
 115 }
 116
 117 struct ClientSocket *getChannelBot(struct ChanNode *chan, int botid) {
 118     struct ClientSocket *bot, *use_bot = NULL, *second_bot = NULL, *third_bot = NULL;
 119     struct ChanUser *chanuser;
 120     for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 121         if(botid && bot->botid != botid) continue;
 122         if((chanuser = getChanUser(bot->user, chan)) != NULL) {
 123             if((chanuser->flags & CHANUSERFLAG_OPPED)) {
 124                 use_bot = bot;
 125                 if(bot->flags & SOCKET_FLAG_PREFERRED) break;
 126             } else if(bot->flags & SOCKET_FLAG_PREFERRED)
 127                 second_bot = bot;
 128             else
 129                 third_bot = bot;
 130         }
 131     }
 132     if(!use_bot) use_bot = second_bot;
 133     if(!use_bot) use_bot = third_bot;
 134     return use_bot;
 135 }
 136
 137 void requestOp(struct UserNode *user, struct ChanNode *chan) {
 138     struct ClientSocket *bot, *userbot = NULL;
 139     struct ChanUser *chanuser = getChanUser(user, chan);
 140     char opped = 0;
 141     if(!chanuser) return;
 142     if((chanuser->flags & CHANUSERFLAG_OPPED)) return;
 143     for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 144         if(!opped && (chanuser = getChanUser(bot->user, chan)) != NULL && (chanuser->flags & CHANUSERFLAG_OPPED)) {
 145             opped = 1;
 146             putsock(bot, "MODE %s +o %s", chan->name, user->nick);
 147         }
 148         if(bot->user == user) {
 149             userbot = bot;
 150         }
 151     }
 152     if(!opped) {
 153         if(userbot && (isUserModeSet(user, 'o') || isUserModeSet(user, 'O') || isUserModeSet(user, 'k') || isUserModeSet(user, 'X'))) {
 154             putsock(userbot, "MODE %s +o %s", chan->name, user->nick);
 155             putsock(userbot, "OPMODE %s +o %s", chan->name, user->nick);
 156         }
 157     }
 158 }
 159
 160 TIMEQ_CALLBACK(channel_ban_timeout) {
 161     char *str_banid = data;
 162     MYSQL_RES *res;
 163     MYSQL_ROW row;
 164     printf_mysql_query("SELECT `ban_mask`, `channel_name` FROM `bans` LEFT JOIN `channels` ON `ban_channel` = `channel_id` WHERE `ban_id` = '%s'", str_banid);
 165     res = mysql_use();
 166     struct ChanNode *chan;
 167     if((row = mysql_fetch_row(res)) != NULL && (chan = getChanByName(row[1])) != NULL) {
 168         struct ClientSocket *use_bot = getChannelBot(chan, 0);
 169         if(use_bot) {
 170             putsock(use_bot, "MODE %s -b %s", chan->name, row[0]);
 171         }
 172         printf_mysql_query("DELETE FROM `bans` WHERE `ban_id` = '%s'", str_banid);
 173     }
 174     free(str_banid);
 175 }
 176
 177 static int general_ctcp(char *buffer, char *command, char *text);
 178
 179 void general_event_privctcp(struct UserNode *user, struct UserNode *target, char *command, char *text) {
 180     char ctcpBuf[MAXLEN];
 181     if(general_ctcp(ctcpBuf, command, text)) {
 182         struct ClientSocket *bot;
 183         for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 184             if(bot->user == target) break;
 185         }
 186         if(bot)
 187             putsock(bot, "NOTICE %s :\001%s\001", user->nick, ctcpBuf);
 188     }
 189 }
 190
 191 static int general_ctcp(char *buffer, char *command, char *text) {
 192     if(!stricmp(command, "VERSION")) {
 193         sprintf(buffer, "VERSION NeonServ v%s.%d by pk910 (%s)", NEONSERV_VERSION, patchlevel, (strcmp(revision, "") ? revision : "-"));
 194         return 1;
 195     }
 196     if(!stricmp(command, "FINGER")) {
 197         sprintf(buffer, "FINGER NeonServ v%s.%d (%s) build %s lines C code using " COMPILER " (see +netinfo)", NEONSERV_VERSION, patchlevel, (strcmp(revision, "") ? revision : "-"), codelines);
 198         return 1;
 199     }
 200     if(!stricmp(command, "PING")) {
 201         sprintf(buffer, "PING %s", (text ? text : "0"));
 202         return 1;
 203     }
 204     if(!stricmp(command, "TIME")) {
 205         time_t rawtime;
 206         struct tm *timeinfo;
 207         char timeBuf[80];
 208         time(&rawtime);
 209         timeinfo = localtime(&rawtime);
 210         strftime(timeBuf, 80, "%c", timeinfo);
 211         sprintf(buffer, "TIME %s", timeBuf);
 212         return 1;
 213     }
 214     return 0;
 215 }
 216
 217 void set_bot_alias(int botid, char *alias) {
 218     struct cmd_bot_alias *botalias, *existing_alias = NULL;
 219     for(botalias = bot_aliases; botalias; botalias = botalias->next) {
 220         if(!stricmp(botalias->alias, alias))
 221             return;
 222         if(botalias->botid == botid)
 223             existing_alias = botalias;
 224     }
 225     if(existing_alias) {
 226         free(existing_alias->alias);
 227         existing_alias->alias = strdup(alias);
 228         return;
 229     }
 230     botalias = malloc(sizeof(*botalias));
 231     if (!botalias) {
 232         perror("malloc() failed");
 233         return;
 234     }
 235     botalias->botid = botid;
 236     botalias->alias = strdup(alias);
 237     botalias->next = bot_aliases;
 238     bot_aliases = botalias;
 239 }
 240
 241 const char *resolve_botid(int botid) {
 242     struct cmd_bot_alias *botalias;
 243     for(botalias = bot_aliases; botalias; botalias = botalias->next) {
 244         if(botalias->botid == botid)
 245             return botalias->alias;
 246     }
 247     return NULL;
 248 }
 249
 250 int resolve_botalias(const char *alias) {
 251     struct cmd_bot_alias *botalias;
 252     for(botalias = bot_aliases; botalias; botalias = botalias->next) {
 253         if(!stricmp(botalias->alias, alias))
 254             return botalias->botid;
 255     }
 256     return -1;
 257 }