src/modules/NeonBackup.mod/bot_NeonBackup.c

   1 /* bot_NeonBackup.c - NeonServ v5.6
   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 #include "../module.h"
  18 #include "../botid.h"
  19
  20 #include "bot_NeonBackup.h"
  21 #include "../../modcmd.h"
  22 #include "cmd_neonbackup.h"
  23 #include "../../lang.h"
  24 #include "../../mysqlConn.h"
  25 #include "../../ClientSocket.h"
  26 #include "../../UserNode.h"
  27 #include "../../ChanNode.h"
  28 #include "../../ChanUser.h"
  29 #include "../../IRCEvents.h"
  30 #include "../../IRCParser.h"
  31 #include "../../bots.h"
  32 #include "../../DBHelper.h"
  33 #include "../../WHOHandler.h"
  34
  35 #define BOTID NEONBACKUP_BOTID
  36 #define BOTALIAS "NeonBackup"
  37
  38 static const struct default_language_entry msgtab[] = {
  39     {NULL, NULL}
  40 };
  41
  42 static void neonbackup_bot_ready(struct ClientSocket *client) {
  43     if(client->botid != BOTID)
  44         return;
  45     MYSQL_RES *res;
  46     MYSQL_ROW row;
  47
  48     printf_mysql_query("SELECT `automodes`, `oper_user`, `oper_pass` FROM `bots` WHERE `id` = '%d'", client->clientid);
  49     res = mysql_use();
  50     if ((row = mysql_fetch_row(res)) != NULL) {
  51         if(row[1] && row[2]) {
  52             putsock(client, "OPER %s %s", row[1], row[2]);
  53         }
  54         putsock(client, "MODE %s +%s", client->user->nick, row[0]);
  55     }
  56
  57     printf_mysql_query("SELECT `channel_name`, `channel_key` FROM `bot_channels` LEFT JOIN `channels` ON `chanid` = `channel_id` WHERE `botid` = '%d' AND `suspended` = '0'", client->clientid);
  58     res = mysql_use();
  59
  60     while ((row = mysql_fetch_row(res)) != NULL) {
  61         putsock(client, "JOIN %s %s", row[0], row[1]);
  62     }
  63 }
  64
  65 static void neonbackup_trigger_callback(int clientid, struct ChanNode *chan, char *trigger) {
  66     MYSQL_RES *res;
  67     MYSQL_ROW row;
  68     loadChannelSettings(chan);
  69     if(!(chan->flags & CHANFLAG_CHAN_REGISTERED)) {
  70         strcpy(trigger, "!");
  71         return;
  72     }
  73     printf_mysql_query("SELECT `trigger`, `defaulttrigger` FROM `bot_channels` LEFT JOIN `bots` ON `botid` = `bots`.`id` WHERE `chanid` = '%d' AND `botclass` = '%d'", chan->channel_id, BOTID);
  74     res = mysql_use();
  75     if(!(row = mysql_fetch_row(res))) {
  76         strcpy(trigger, "+");
  77         return;
  78     }
  79     if(row[0] && *row[0])
  80         strcpy(trigger, row[0]);
  81     else
  82         strcpy(trigger, ((row[1] && *row[1]) ? row[1] : "+"));
  83 }
  84
  85 static void start_bots(int type) {
  86     struct ClientSocket *client;
  87     MYSQL_RES *res, *res2;
  88     MYSQL_ROW row;
  89
  90     if(type == MODSTATE_STARTSTOP) {
  91         printf_mysql_query("SELECT `nick`, `ident`, `realname`, `server`, `port`, `pass`, `textbot`, `id`, `queue`, `ssl`, `bind`, `secret` FROM `bots` WHERE `botclass` = '%d' AND `active` = '1'", BOTID);
  92         res = mysql_use();
  93
  94         while ((row = mysql_fetch_row(res)) != NULL) {
  95             client = create_socket(row[3], atoi(row[4]), row[10], row[5], row[0], row[1], row[2]);
  96             client->flags |= (strcmp(row[6], "0") ? SOCKET_FLAG_PREFERRED : 0);
  97             client->flags |= (strcmp(row[8], "0") ? SOCKET_FLAG_USE_QUEUE : 0);
  98             client->flags |= (strcmp(row[9], "0") ? SOCKET_FLAG_SSL : 0);
  99             client->flags |= (strcmp(row[11], "0") ? SOCKET_FLAG_SECRET_BOT : 0);
 100             client->flags |= SOCKET_FLAG_REQUEST_INVITE | SOCKET_FLAG_REQUEST_OP;
 101             client->botid = BOTID;
 102             client->clientid = atoi(row[7]);
 103             connect_socket(client);
 104         }
 105     }
 106
 107     printf_mysql_query("SELECT `command`, `function`, `parameters`, `global_access`, `chan_access`, `flags` FROM `bot_binds` WHERE `botclass` = '%d'", BOTID);
 108     res2 = mysql_use();
 109     while ((row = mysql_fetch_row(res2)) != NULL) {
 110         if(bind_cmd_to_command(BOTID, row[0], row[1])) {
 111             if(row[2] && strcmp(row[2], "")) {
 112                 bind_set_parameters(BOTID, row[0], row[2]);
 113             }
 114             if(row[3]) {
 115                 bind_set_global_access(BOTID, row[0], atoi(row[3]));
 116             }
 117             if(row[4]) {
 118                 bind_set_channel_access(BOTID, row[0], row[4]);
 119             }
 120             if(strcmp(row[5], "0"))
 121                 bind_set_bind_flags(BOTID, row[0], atoi(row[5]));
 122         }
 123     }
 124     bind_unbound_required_functions(BOTID);
 125 }
 126
 127 void neonbackup_recover_chan(struct ChanNode *chan) {
 128     struct ClientSocket *bot = getChannelBot(chan, BOTID); // prefer backup bot ;)
 129     struct ChanUser *chanuser = (bot ? getChanUser(bot->user, chan) : NULL);
 130     if(!chanuser || !(chanuser->flags & CHANUSERFLAG_OPPED)) {
 131         //search an opped bot
 132         for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 133             if((chanuser = getChanUser(bot->user, chan)) && (chanuser->flags & CHANUSERFLAG_OPPED))
 134                 break;
 135         }
 136     }
 137     if(!bot) //no opped bots present... channel can't be recovered
 138         return;
 139     struct ClientSocket *target;
 140     struct ModeBuffer *modeBuf;
 141     modeBuf = initModeBuffer(bot, chan);
 142     for(target = getBots(SOCKET_FLAG_READY, NULL); target; target = getBots(SOCKET_FLAG_READY, target)) {
 143         if((target->flags & SOCKET_FLAG_REQUEST_OP) && (chanuser = getChanUser(target->user, chan)) && !(chanuser->flags & CHANUSERFLAG_OPPED)) {
 144             modeBufferOp(modeBuf, target->user->nick);
 145         }
 146     }
 147     freeModeBuffer(modeBuf);
 148 }
 149
 150 static void neonbackup_event_join(struct ChanUser *chanuser) {
 151     struct ClientSocket *client = getChannelBot(chanuser->chan, BOTID);
 152     if(!client) return; //we can't "see" this event
 153     if(chanuser->user == client->user) {
 154         requestOp(client->user, chanuser->chan);
 155         neonbackup_recover_chan(chanuser->chan);
 156     }
 157 }
 158
 159 static void neonbackup_event_invite(struct ClientSocket *client, struct UserNode *user, char *channel) {
 160         if(client->botid != BOTID)
 161                 return;
 162     MYSQL_RES *res;
 163     MYSQL_ROW row;
 164     printf_mysql_query("SELECT `botid`, `bot_channels`.`id`, `suspended` FROM `bot_channels` LEFT JOIN `bots` ON `bot_channels`.`botid` = `bots`.`id` LEFT JOIN `channels` ON `chanid` = `channel_id` WHERE `channel_name` = '%s' AND `botclass` = '%d'", escape_string(channel), client->botid);
 165     res = mysql_use();
 166     if ((row = mysql_fetch_row(res)) == NULL) {
 167         reply(client, user, "NS_INVITE_FAIL", channel, client->user->nick);
 168         return;
 169     }
 170     if(!strcmp(row[2], "1")) {
 171         reply(client, user, "MODCMD_CHAN_SUSPENDED");
 172         return;
 173     }
 174     int botid = atoi(row[0]);
 175     struct ClientSocket *bot;
 176     for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 177         if(bot->clientid == botid)
 178             break;
 179     }
 180     if(bot) {
 181         struct ChanNode *chan = getChanByName(channel);
 182         if(chan && isUserOnChan(bot->user, chan)) {
 183             reply(client, user, "NS_INVITE_ON_CHAN", bot->user->nick, chan->name);
 184         } else
 185             putsock(bot, "JOIN %s", channel);
 186     }
 187 }
 188
 189 void init_NeonBackup(int type) {
 190     set_bot_alias(BOTID, BOTALIAS);
 191     start_bots(type);
 192
 193     if(type == MODSTATE_REBIND) return;
 194
 195     //register events
 196     bind_bot_ready(neonbackup_bot_ready, module_id);
 197     bind_join(neonbackup_event_join, module_id);
 198         bind_invite(neonbackup_event_invite, module_id);
 199
 200     set_trigger_callback(BOTID, module_id, neonbackup_trigger_callback);
 201
 202     register_default_language_table(msgtab);
 203 }
 204
 205 void free_NeonBackup(int type) {
 206     unbind_allcmd(BOTID);
 207     if(type == MODSTATE_STARTSTOP) {
 208         //disconnect all our bots
 209         struct ClientSocket *client;
 210         for(client = getBots(0, NULL); client; client = getBots(0, client)) {
 211             if(client->botid == BOTID) {
 212                 unbind_botwise_allcmd(0, client->clientid);
 213                 close_socket(client);
 214                 break;
 215             }
 216         }
 217     }
 218 }
 219
 220 #undef BOTID
 221 #undef BOTALIAS