src/modules/NeonSpam.mod/bot_NeonSpam.c

   1 /* bot_NeonSpam.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_NeonSpam.h"
  21 #include "../../modcmd.h"
  22 #include "../../IRCParser.h"
  23 #include "../../IRCEvents.h"
  24 #include "../../UserNode.h"
  25 #include "../../ChanNode.h"
  26 #include "../../ChanUser.h"
  27 #include "../../ModeNode.h"
  28 #include "../../BanNode.h"
  29 #include "../../ClientSocket.h"
  30 #include "../../mysqlConn.h"
  31 #include "../../lang.h"
  32 #include "../../HandleInfoHandler.h"
  33 #include "../../WHOHandler.h"
  34 #include "../../DBHelper.h"
  35 #include "../../tools.h"
  36 #include "../../timeq.h"
  37 #include "../../version.h"
  38 #include "../../EventLogger.h"
  39 #include "../../bots.h"
  40 #include "cmd_neonspam.h"
  41
  42 #define BOTID NEONSPAM_BOTID
  43 #define BOTALIAS "NeonSpam"
  44
  45 static const struct default_language_entry msgtab[] = {
  46     {"SS_SET_PERCENT", "%u is an invalid percent value (valid: 0-100)"}, /* {ARGS: 120} */
  47     {"SS_SET_SENSIBILITY", "%s is an invalid sensibility format. (valid: amount:time   e.g. 5:10)"}, /* {ARGS: "möp"} */
  48     {"SS_SET_SENSIBILITY_AMOUNT", "%d is an invalid amount value. (valid: %d-%d)"}, /* {ARGS: 120, 1, 5} */
  49     {"SS_SET_SENSIBILITY_TIME", "%d is an invalid time value. (valid: %d-%d)"}, /* {ARGS: 300, 15, 180} */
  50     {"SS_SET_SPAMLIMIT", "%d is an invalid spamlimit value. (valid: %d-%d)"}, /* {ARGS: 120, 2, 8} */
  51     {"SS_SET_OPTION_SpamReaction_0", "Kick"},
  52     {"SS_SET_OPTION_SpamReaction_1", "KickBan"},
  53     {"SS_SET_OPTION_SpamReaction_2", "Timed Ban"},
  54     {"SS_SET_OPTION_FloodReaction_0", "Kick"},
  55     {"SS_SET_OPTION_FloodReaction_1", "KickBan"},
  56     {"SS_SET_OPTION_FloodReaction_2", "Timed Ban"},
  57     {"SS_SET_OPTION_JoinReaction_0", "Kick"},
  58     {"SS_SET_OPTION_JoinReaction_1", "KickBan"},
  59     {"SS_SET_OPTION_JoinReaction_2", "Timed Ban"},
  60     {"SS_SET_OPTION_CapsReaction_0", "Kick"},
  61     {"SS_SET_OPTION_CapsReaction_1", "KickBan"},
  62     {"SS_SET_OPTION_CapsReaction_2", "Timed Ban"},
  63     {"SS_SET_OPTION_DigitReaction_0", "Kick"},
  64     {"SS_SET_OPTION_DigitReaction_1", "KickBan"},
  65     {"SS_SET_OPTION_DigitReaction_2", "Timed Ban"},
  66     {"SS_SET_OPTION_BadwordReaction_0", "Kick"},
  67     {"SS_SET_OPTION_BadwordReaction_1", "KickBan"},
  68     {"SS_SET_OPTION_BadwordReaction_2", "Timed Ban"},
  69     {"SS_BADWORD_ADDED", "Badword '%s' added. (ID: %d)"}, /* {ARGS: "NeonServ is bad", 1} */
  70     {"SS_BADWORD_ID_UNKNOWN", "Badword ID %d was not found in %s."}, /* {ARGS: 1, "#channel"} */
  71     {"SS_BADWORD_DELETED", "Badword '%s' deleted. (ID: %d)"}, /* {ARGS: "NeonServ is bad", 1} */
  72     {"SS_BADWORDS_REACTION_0", "Kick"},
  73     {"SS_BADWORDS_REACTION_1", "KickBan"},
  74     {"SS_BADWORDS_REACTION_2", "Timed Ban"},
  75     {"SS_BADWORDS_ID", "Id"},
  76     {"SS_BADWORDS_PATTERN", "Pattern"},
  77     {"SS_BADWORDS_SCAN", "Scan"},
  78     {"SS_BADWORDS_REACTION", "Reaction"},
  79     {NULL, NULL}
  80 };
  81
  82 static unsigned int convertNeonSpamSettingsToFlags(char *str);
  83 static void createSpamNode(struct ChanUser *chanuser);
  84 static void freeJoinNode(struct NeonSpamJoinNode *joinnode);
  85 static struct NeonSpamJoinNode *getNeonSpamJoinNode(struct ChanUser *chanuser);
  86
  87 #define SPAMSERV_CHECK_IGNORE 0
  88 #define SPAMSERV_CHECK_WARN   1
  89 #define SPAMSERV_CHECK_PUNISH 2
  90 #define SPAMSERV_CHECK_DEAD   3 /* scanner has already killed the user */
  91
  92 #define SPAMSERV_MSG_SPAM       "Spamming"
  93 #define SPAMSERV_MSG_FLOOD      "Flooding the channel/network"
  94 #define SPAMSERV_MSG_ADV        "Advertising"
  95 #define SPAMSERV_MSG_JOINFLOOD  "Join flooding the channel"
  96 #define SPAMSERV_MSG_WARNING    "%s is against the channel rules"
  97 #define SPAMSERV_MSG_BOTNET     "BotNet detected."
  98 #define SPAMSERV_MSG_CAPS       "Using too many chars in UPPER CASE"
  99 #define SPAMSERV_MSG_DIGIT      "Using too many numeric chars"
 100 #define SPAMSERV_MSG_BADWORD    "Your message contained a forbidden word."
 101
 102 //EVENTS
 103 #include "event_neonspam_join.c"
 104 #include "event_neonspam_chanmsg.c"
 105
 106 static void neonspam_bot_ready(struct ClientSocket *client) {
 107     if(client->botid != BOTID)
 108         return;
 109     MYSQL_RES *res;
 110     MYSQL_ROW row;
 111
 112     printf_mysql_query("SELECT `automodes`, `oper_user`, `oper_pass` FROM `bots` WHERE `id` = '%d'", client->clientid);
 113     res = mysql_use();
 114     if ((row = mysql_fetch_row(res)) != NULL) {
 115         if(row[1] && row[2]) {
 116             putsock(client, "OPER %s %s", row[1], row[2]);
 117         }
 118         putsock(client, "MODE %s +%s", client->user->nick, row[0]);
 119     }
 120
 121     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);
 122     res = mysql_use();
 123
 124     while ((row = mysql_fetch_row(res)) != NULL) {
 125         putsock(client, "JOIN %s %s", row[0], row[1]);
 126     }
 127 }
 128
 129 static void neonspam_trigger_callback(int clientid, struct ChanNode *chan, char *trigger) {
 130     MYSQL_RES *res;
 131     MYSQL_ROW row;
 132     loadChannelSettings(chan);
 133     if(!(chan->flags & CHANFLAG_CHAN_REGISTERED)) {
 134         strcpy(trigger, "~");
 135         return;
 136     }
 137     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);
 138     res = mysql_use();
 139     if(!(row = mysql_fetch_row(res))) {
 140         strcpy(trigger, "~");
 141         return;
 142     }
 143     if(row[0] && *row[0])
 144         strcpy(trigger, row[0]);
 145     else
 146         strcpy(trigger, ((row[1] && *row[1]) ? row[1] : "~"));
 147 }
 148
 149 static void start_bots(int type) {
 150     struct ClientSocket *client;
 151     MYSQL_RES *res, *res2;
 152     MYSQL_ROW row;
 153
 154     if(type == MODSTATE_STARTSTOP) {
 155         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);
 156         res = mysql_use();
 157
 158         while ((row = mysql_fetch_row(res)) != NULL) {
 159             client = create_socket(row[3], atoi(row[4]), row[10], row[5], row[0], row[1], row[2]);
 160             client->flags |= (strcmp(row[6], "0") ? SOCKET_FLAG_PREFERRED : 0);
 161             client->flags |= (strcmp(row[8], "0") ? SOCKET_FLAG_USE_QUEUE : 0);
 162             client->flags |= (strcmp(row[9], "0") ? SOCKET_FLAG_SSL : 0);
 163             client->flags |= (strcmp(row[11], "0") ? SOCKET_FLAG_SECRET_BOT : 0);
 164             client->flags |= SOCKET_FLAG_REQUEST_INVITE | SOCKET_FLAG_REQUEST_OP;
 165             client->botid = BOTID;
 166             client->clientid = atoi(row[7]);
 167             connect_socket(client);
 168         }
 169     }
 170
 171     printf_mysql_query("SELECT `command`, `function`, `parameters`, `global_access`, `chan_access`, `flags` FROM `bot_binds` WHERE `botclass` = '%d'", BOTID);
 172     res2 = mysql_use();
 173     while ((row = mysql_fetch_row(res2)) != NULL) {
 174         if(bind_cmd_to_command(BOTID, row[0], row[1])) {
 175             if(row[2] && strcmp(row[2], "")) {
 176                 bind_set_parameters(BOTID, row[0], row[2]);
 177             }
 178             if(row[3]) {
 179                 bind_set_global_access(BOTID, row[0], atoi(row[3]));
 180             }
 181             if(row[4]) {
 182                 bind_set_channel_access(BOTID, row[0], row[4]);
 183             }
 184             if(strcmp(row[5], "0"))
 185                 bind_set_bind_flags(BOTID, row[0], atoi(row[5]));
 186         }
 187     }
 188     bind_unbound_required_functions(BOTID);
 189 }
 190
 191 char* convertNeonSpamSettingsToString(unsigned int flags, char *buffer) {
 192     int pos = 0;
 193     unsigned int i;
 194     int j = 0;
 195     char *chars = SPAMSETTINGS_CHARS;
 196     for(i = 1; i <= SPAMSETTINGS_FLAGS; i = i << 1) {
 197         if(flags & i)
 198             buffer[pos++] = chars[j];
 199         j++;
 200     }
 201     buffer[pos] = '\0';
 202     return buffer;
 203 }
 204
 205 static unsigned int convertNeonSpamSettingsToFlags(char *str) {
 206     unsigned int i = 1, flags = 0;
 207     int j = 0;
 208     char *chars = SPAMSETTINGS_CHARS;
 209     while(*str) {
 210         for(; i <= SPAMSETTINGS_FLAGS; i = i << 1) {
 211             if(*str == chars[j]) {
 212                 flags |= i;
 213                 j++;
 214                 i = i << 1;
 215                 break;
 216             }
 217             j++;
 218         }
 219         str++;
 220     }
 221     return flags;
 222 }
 223
 224 int loadNeonSpamSettings(struct ChanNode *chan) {
 225     if(chan->spam_settings) return 0;
 226     struct NeonSpamSettings *settings = malloc(sizeof(*settings));
 227     if(!settings) {
 228         printf_log("neonserv", LOG_ERROR, "%s:%d malloc() failed", __FILE__, __LINE__);
 229         return 0;
 230     }
 231     MYSQL_RES *res;
 232     MYSQL_ROW row, defaults = NULL;
 233     loadChannelSettings(chan);
 234     printf_mysql_query("SELECT `channel_scanner`, `channel_spam_limit`, `channel_spam_except`, `channel_flood_limit`, `channel_flood_time`, `channel_flood_except`, `channel_join_limit`, `channel_join_time`, `channel_join_except`, `channel_caps_percent`, `channel_caps_except`, `channel_digit_percent`, `channel_digit_except`, `channel_badword_except` FROM `channels` WHERE `channel_id` = '%d'", chan->channel_id);
 235     res = mysql_use();
 236     if(!(row = mysql_fetch_row(res)))
 237         return 0;
 238     if(!row[0] || !row[1] || !row[2] || !row[3] || !row[4] || !row[5] || !row[6] || !row[7] || !row[8] || !row[9] || !row[10] || !row[11] || !row[12]) {
 239         printf_mysql_query("SELECT `channel_scanner`, `channel_spam_limit`, `channel_spam_except`, `channel_flood_limit`, `channel_flood_time`, `channel_flood_except`, `channel_join_limit`, `channel_join_time`, `channel_join_except`, `channel_caps_percent`, `channel_caps_except`, `channel_digit_percent`, `channel_digit_except`, `channel_badword_except` FROM `channels` WHERE `channel_name` = 'defaults'");
 240         res = mysql_use();
 241         defaults = mysql_fetch_row(res);
 242     }
 243     settings->flags = convertNeonSpamSettingsToFlags(row[0] ? row[0] : defaults[0]);
 244     settings->spam_amount = atoi(row[1] ? row[1] : defaults[1]);
 245     settings->exceptlevel[SPAMSETTINGS_SPAMEXCINDEX] = atoi(row[2] ? row[2] : defaults[2]);
 246     settings->sensibility_amount[SPAMSETTINGS_FLOODSENINDEX] = atoi(row[3] ? row[3] : defaults[3]);
 247     settings->sensibility_time[SPAMSETTINGS_FLOODSENINDEX] = atoi(row[4] ? row[4] : defaults[4]);
 248     settings->exceptlevel[SPAMSETTINGS_FLOODEXCINDEX] = atoi(row[5] ? row[5] : defaults[5]);
 249     settings->sensibility_amount[SPAMSETTINGS_JOINSENINDEX] = atoi(row[6] ? row[6] : defaults[6]);
 250     settings->sensibility_time[SPAMSETTINGS_JOINSENINDEX] = atoi(row[7] ? row[7] : defaults[7]);
 251     settings->exceptlevel[SPAMSETTINGS_JOINEXCINDEX] = atoi(row[8] ? row[8] : defaults[8]);
 252     settings->percent[SPAMSETTINGS_CAPSPERCENTINDEX] = atoi(row[9] ? row[9] : defaults[9]);
 253     settings->exceptlevel[SPAMSETTINGS_CAPSEXCINDEX] = atoi(row[10] ? row[10] : defaults[10]);
 254     settings->percent[SPAMSETTINGS_DIGITPERCENTINDEX] = atoi(row[11] ? row[11] : defaults[11]);
 255     settings->exceptlevel[SPAMSETTINGS_DIGITEXCINDEX] = atoi(row[12] ? row[12] : defaults[12]);
 256     settings->exceptlevel[SPAMSETTINGS_BADWORDEXCINDEX] = atoi(row[13] ? row[13] : defaults[13]);
 257     settings->join_nodes = NULL;
 258     settings->lastmsg_time = 0;
 259     int i;
 260     for(i = 0; i < BOTNETSCAN_USERS; i++)
 261         settings->botnicks[i] = NULL;
 262     settings->badwords = NULL;
 263     printf_mysql_query("SELECT `badword_match`, `badword_reaction`, `badword_reaction_time`, `badword_id`, `badword_scan_ops`, `badword_scan_voice`, `badword_exceptlevel`, `badword_use_default`, `badword_use_default_reaction` FROM `spamserv_badwords` WHERE `badword_cid` = '%d'", chan->channel_id);
 264     res = mysql_use();
 265     while((row = mysql_fetch_row(res))) {
 266         struct NeonSpamBadword *badword = malloc(sizeof(*badword));
 267         badword->id = atoi(row[3]);
 268         badword->badword = strdup(row[0]);
 269         badword->use_defaults = atoi(row[7]);
 270         badword->exceptlevel = atoi(row[6]);
 271         badword->scan_ops = atoi(row[4]);
 272         badword->scan_voice = atoi(row[5]);
 273         badword->use_default_reaction = atoi(row[8]);
 274         badword->reaction = atoi(row[1]);
 275         badword->reaction_time = atoi(row[2]);
 276         badword->next = settings->badwords;
 277         settings->badwords = badword;
 278     }
 279     chan->spam_settings = settings;
 280     return 1;
 281 }
 282
 283 void freeNeonSpamSettings(struct NeonSpamSettings *settings) {
 284     struct NeonSpamJoinNode *joinnode, *nextjoinnode;
 285     for(joinnode = settings->join_nodes; joinnode; joinnode = nextjoinnode) {
 286         nextjoinnode = joinnode->next;
 287         freeJoinNode(joinnode);
 288     }
 289     struct NeonSpamBadword *badword, *nextbadword;
 290     for(badword = settings->badwords; badword; badword = nextbadword) {
 291         nextbadword = badword->next;
 292         free(badword->badword);
 293         free(badword);
 294     }
 295     free(settings);
 296 }
 297
 298 static void freeJoinNode(struct NeonSpamJoinNode *joinnode) {
 299     free(joinnode->ident);
 300     free(joinnode->host);
 301     free(joinnode);
 302 }
 303
 304 static struct NeonSpamJoinNode *getNeonSpamJoinNode(struct ChanUser *chanuser) {
 305     struct NeonSpamJoinNode *joinnode, *prevjoinnode = NULL, *nextjoinnode, *result = NULL;
 306     for(joinnode = chanuser->chan->spam_settings->join_nodes; joinnode; joinnode = nextjoinnode) {
 307         nextjoinnode = joinnode->next;
 308         if(!stricmp(joinnode->ident, chanuser->user->ident) && !stricmp(joinnode->host, chanuser->user->host)) {
 309             prevjoinnode = joinnode;
 310             result = joinnode;
 311         } else if(time(0) - joinnode->last_penalty_update > MAX_JOIN_TIME) {
 312             freeJoinNode(joinnode);
 313             if(prevjoinnode)
 314                 prevjoinnode->next = nextjoinnode;
 315             else
 316                 chanuser->chan->spam_settings->join_nodes = nextjoinnode;
 317         } else
 318             prevjoinnode = joinnode;
 319     }
 320     if(result)
 321         return result;
 322     joinnode = malloc(sizeof(*joinnode));
 323     if(!joinnode) {
 324         printf_log("neonserv", LOG_ERROR, "%s:%d malloc() failed", __FILE__, __LINE__);
 325         return NULL;
 326     }
 327     joinnode->ident = strdup(chanuser->user->ident);
 328     joinnode->host = strdup(chanuser->user->host);
 329     joinnode->last_penalty_update = time(0);
 330     joinnode->joinpenalty = 0;
 331     joinnode->next = chanuser->chan->spam_settings->join_nodes;
 332     chanuser->chan->spam_settings->join_nodes = joinnode;
 333     return joinnode;
 334 }
 335
 336 static void createSpamNode(struct ChanUser *chanuser) {
 337     struct NeonSpamNode *spamnode = malloc(sizeof(*spamnode));
 338     if(!spamnode) {
 339         printf_log("neonserv", LOG_ERROR, "%s:%d malloc() failed", __FILE__, __LINE__);
 340         return;
 341     }
 342     spamnode->lastmsg = 0;
 343     spamnode->spamcount = 0;
 344     spamnode->floodpenalty = 0;
 345     spamnode->last_penalty_update = time(0);
 346     chanuser->spamnode = spamnode;
 347 }
 348
 349 static int neonspam_event_freechan(struct ChanNode *chan) {
 350     if(chan->spam_settings)
 351         freeNeonSpamSettings(chan->spam_settings);
 352     return 1;
 353 }
 354
 355 static void neonspam_event_invite(struct ClientSocket *client, struct UserNode *user, char *channel) {
 356         if(client->botid != BOTID)
 357                 return;
 358     MYSQL_RES *res;
 359     MYSQL_ROW row;
 360     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);
 361     res = mysql_use();
 362     if ((row = mysql_fetch_row(res)) == NULL) {
 363         reply(client, user, "NS_INVITE_FAIL", channel, client->user->nick);
 364         return;
 365     }
 366     if(!strcmp(row[2], "1")) {
 367         reply(client, user, "MODCMD_CHAN_SUSPENDED");
 368         return;
 369     }
 370     int botid = atoi(row[0]);
 371     struct ClientSocket *bot;
 372     for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 373         if(bot->clientid == botid)
 374             break;
 375     }
 376     if(bot) {
 377         struct ChanNode *chan = getChanByName(channel);
 378         if(chan && isUserOnChan(bot->user, chan)) {
 379             reply(client, user, "NS_INVITE_ON_CHAN", bot->user->nick, chan->name);
 380         } else
 381             putsock(bot, "JOIN %s", channel);
 382     }
 383 }
 384
 385 void init_NeonSpam(int type) {
 386
 387     set_bot_alias(BOTID, BOTALIAS);
 388     start_bots(type);
 389
 390     if(type == MODSTATE_REBIND) return;
 391
 392     //register events
 393     bind_bot_ready(neonspam_bot_ready, module_id);
 394     bind_join(neonspam_event_join, module_id);
 395     bind_chanmsg(neonspam_event_chanmsg, module_id);
 396     bind_privctcp(general_event_privctcp, module_id);
 397     bind_freechan(neonspam_event_freechan, module_id);
 398     bind_invite(neonspam_event_invite, module_id);
 399
 400     set_trigger_callback(BOTID, module_id, neonspam_trigger_callback);
 401
 402     register_default_language_table(msgtab);
 403 }
 404
 405 void free_NeonSpam(int type) {
 406     unbind_allcmd(BOTID);
 407     if(type == MODSTATE_STARTSTOP) {
 408         //disconnect all our bots
 409         struct ClientSocket *client;
 410         for(client = getBots(0, NULL); client; client = getBots(0, client)) {
 411             if(client->botid == BOTID) {
 412                 unbind_botwise_allcmd(0, client->clientid);
 413                 close_socket(client);
 414                 break;
 415             }
 416         }
 417     }
 418 }
 419
 420 #undef BOTID
 421 #undef BOTALIAS