src/modules/NeonSpam.mod/bot_NeonSpam.c

   1 /* bot_NeonSpam.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 #include "../module.h"
  18
  19 #include "bot_NeonSpam.h"
  20 #include "../../modcmd.h"
  21 #include "../../IRCParser.h"
  22 #include "../../IRCEvents.h"
  23 #include "../../UserNode.h"
  24 #include "../../ChanNode.h"
  25 #include "../../ChanUser.h"
  26 #include "../../ModeNode.h"
  27 #include "../../BanNode.h"
  28 #include "../../ClientSocket.h"
  29 #include "../../mysqlConn.h"
  30 #include "../../lang.h"
  31 #include "../../HandleInfoHandler.h"
  32 #include "../../WHOHandler.h"
  33 #include "../../DBHelper.h"
  34 #include "../../tools.h"
  35 #include "../../timeq.h"
  36 #include "../../version.h"
  37 #include "../../EventLogger.h"
  38 #include "../../bots.h"
  39 #include "cmd_neonspam.h"
  40
  41 #define BOTID 2
  42 #define BOTALIAS "NeonSpam"
  43
  44 static const struct default_language_entry msgtab[] = {
  45     {"SS_SET_PERCENT", "%u is an invalid percent value (valid: 0-100)"}, /* {ARGS: 120} */
  46     {"SS_SET_SENSIBILITY", "%s is an invalid sensibility format. (valid: amount:time   e.g. 5:10)"}, /* {ARGS: "möp"} */
  47     {"SS_SET_SENSIBILITY_AMOUNT", "%d is an invalid amount value. (valid: %d-%d)"}, /* {ARGS: 120, 1, 5} */
  48     {"SS_SET_SENSIBILITY_TIME", "%d is an invalid time value. (valid: %d-%d)"}, /* {ARGS: 300, 15, 180} */
  49     {"SS_SET_SPAMLIMIT", "%d is an invalid spamlimit value. (valid: %d-%d)"}, /* {ARGS: 120, 2, 8} */
  50     {"SS_SET_OPTION_SpamReaction_0", "Kick"},
  51     {"SS_SET_OPTION_SpamReaction_1", "KickBan"},
  52     {"SS_SET_OPTION_SpamReaction_2", "Timed Ban"},
  53     {"SS_SET_OPTION_FloodReaction_0", "Kick"},
  54     {"SS_SET_OPTION_FloodReaction_1", "KickBan"},
  55     {"SS_SET_OPTION_FloodReaction_2", "Timed Ban"},
  56     {"SS_SET_OPTION_JoinReaction_0", "Kick"},
  57     {"SS_SET_OPTION_JoinReaction_1", "KickBan"},
  58     {"SS_SET_OPTION_JoinReaction_2", "Timed Ban"},
  59     {"SS_SET_OPTION_CapsReaction_0", "Kick"},
  60     {"SS_SET_OPTION_CapsReaction_1", "KickBan"},
  61     {"SS_SET_OPTION_CapsReaction_2", "Timed Ban"},
  62     {"SS_SET_OPTION_DigitReaction_0", "Kick"},
  63     {"SS_SET_OPTION_DigitReaction_1", "KickBan"},
  64     {"SS_SET_OPTION_DigitReaction_2", "Timed Ban"},
  65     {NULL, NULL}
  66 };
  67
  68 static unsigned int convertNeonSpamSettingsToFlags(char *str);
  69 static void createSpamNode(struct ChanUser *chanuser);
  70 static void freeJoinNode(struct NeonSpamJoinNode *joinnode);
  71 static struct NeonSpamJoinNode *getNeonSpamJoinNode(struct ChanUser *chanuser);
  72
  73 #define SPAMSERV_CHECK_IGNORE 0
  74 #define SPAMSERV_CHECK_WARN   1
  75 #define SPAMSERV_CHECK_PUNISH 2
  76 #define SPAMSERV_CHECK_DEAD   3 /* scanner has already killed the user */
  77
  78 #define SPAMSERV_MSG_SPAM       "Spamming"
  79 #define SPAMSERV_MSG_FLOOD      "Flooding the channel/network"
  80 #define SPAMSERV_MSG_ADV        "Advertising"
  81 #define SPAMSERV_MSG_JOINFLOOD  "Join flooding the channel"
  82 #define SPAMSERV_MSG_WARNING    "%s is against the channel rules"
  83 #define SPAMSERV_MSG_BOTNET     "BotNet detected."
  84 #define SPAMSERV_MSG_CAPS       "Using too many chars in UPPER CASE"
  85 #define SPAMSERV_MSG_DIGIT      "Using too many numeric chars"
  86
  87 //EVENTS
  88 #include "event_neonspam_join.c"
  89 #include "event_neonspam_chanmsg.c"
  90
  91 static void neonspam_event_kick(struct UserNode *user, struct ChanUser *target, char *reason) {
  92     struct ChanNode *chan = target->chan;
  93     if(isBot(target->user)) {
  94         struct ClientSocket *client = getChannelBot(chan, 0);
  95         struct ClientSocket *bot = client;
  96         for(client = getBots(SOCKET_FLAG_READY, NULL); client; client = getBots(SOCKET_FLAG_READY, client)) {
  97             if(client->user == target->user) {
  98                 break;
  99             }
 100         }
 101         if(!client) return;
 102         if(bot && bot != client && (isModeSet(chan->modes, 'i') || isModeSet(chan->modes, 'a') || isModeSet(chan->modes, 'l'))) {
 103             struct ChanUser *chanuser = getChanUser(bot->user, chan);
 104             if(chanuser && chanuser->flags & CHANUSERFLAG_OPPED)
 105                 putsock(bot, "INVITE %s %s", target->user->nick, chan->name);
 106         }
 107         char *key = "";
 108         if(isModeSet(chan->modes, 'k')) {
 109             key = getModeValue(chan->modes, 'k');
 110         }
 111         putsock(client, "JOIN %s %s", chan->name, key);
 112         return;
 113     }
 114 }
 115
 116 static void neonspam_bot_ready(struct ClientSocket *client) {
 117     MYSQL_RES *res;
 118     MYSQL_ROW row;
 119
 120     printf_mysql_query("SELECT `automodes` FROM `bots` WHERE `id` = '%d'", client->clientid);
 121     res = mysql_use();
 122     if ((row = mysql_fetch_row(res)) != NULL) {
 123         putsock(client, "MODE %s +%s", client->user->nick, row[0]);
 124     }
 125
 126     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);
 127     res = mysql_use();
 128
 129     while ((row = mysql_fetch_row(res)) != NULL) {
 130         putsock(client, "JOIN %s %s", row[0], row[1]);
 131     }
 132 }
 133
 134 static void neonspam_trigger_callback(int clientid, struct ChanNode *chan, char *trigger) {
 135     MYSQL_RES *res;
 136     MYSQL_ROW row;
 137     loadChannelSettings(chan);
 138     if(!(chan->flags & CHANFLAG_CHAN_REGISTERED)) {
 139         strcpy(trigger, "~");
 140         return;
 141     }
 142     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);
 143     res = mysql_use();
 144     if(!(row = mysql_fetch_row(res))) {
 145         strcpy(trigger, "~");
 146         return;
 147     }
 148     if(row[0] && *row[0])
 149         strcpy(trigger, row[0]);
 150     else
 151         strcpy(trigger, ((row[1] && *row[1]) ? row[1] : "~"));
 152 }
 153
 154 static void start_bots(int type) {
 155     struct ClientSocket *client;
 156     MYSQL_RES *res, *res2;
 157     MYSQL_ROW row;
 158
 159     if(type == MODSTATE_STARTSTOP) {
 160         printf_mysql_query("SELECT `nick`, `ident`, `realname`, `server`, `port`, `pass`, `textbot`, `id`, `queue`, `ssl`, `bind` FROM `bots` WHERE `botclass` = '%d' AND `active` = '1'", BOTID);
 161         res = mysql_use();
 162
 163         while ((row = mysql_fetch_row(res)) != NULL) {
 164             client = create_socket(row[3], atoi(row[4]), row[10], row[5], row[0], row[1], row[2]);
 165             client->flags |= (strcmp(row[6], "0") ? SOCKET_FLAG_PREFERRED : 0);
 166             client->flags |= (strcmp(row[8], "0") ? SOCKET_FLAG_USE_QUEUE : 0);
 167             client->flags |= (strcmp(row[9], "0") ? SOCKET_FLAG_SSL : 0);
 168             client->botid = BOTID;
 169             client->clientid = atoi(row[7]);
 170             connect_socket(client);
 171         }
 172     }
 173
 174     printf_mysql_query("SELECT `command`, `function`, `parameters`, `global_access`, `chan_access`, `flags` FROM `bot_binds` WHERE `botclass` = '%d'", BOTID);
 175     res2 = mysql_use();
 176     while ((row = mysql_fetch_row(res2)) != NULL) {
 177         if(bind_cmd_to_command(BOTID, row[0], row[1])) {
 178             if(row[2] && strcmp(row[2], "")) {
 179                 bind_set_parameters(BOTID, row[0], row[2]);
 180             }
 181             if(row[3]) {
 182                 bind_set_global_access(BOTID, row[0], atoi(row[3]));
 183             }
 184             if(row[4]) {
 185                 bind_set_channel_access(BOTID, row[0], row[4]);
 186             }
 187             if(strcmp(row[5], "0"))
 188                 bind_set_bind_flags(BOTID, row[0], atoi(row[5]));
 189         }
 190     }
 191     bind_unbound_required_functions(BOTID);
 192 }
 193
 194 char* convertNeonSpamSettingsToString(unsigned int flags, char *buffer) {
 195     int pos = 0;
 196     unsigned int i;
 197     int j = 0;
 198     char *chars = SPAMSETTINGS_CHARS;
 199     for(i = 1; i <= SPAMSETTINGS_FLAGS; i = i << 1) {
 200         if(flags & i)
 201             buffer[pos++] = chars[j];
 202         j++;
 203     }
 204     buffer[pos] = '\0';
 205     return buffer;
 206 }
 207
 208 static unsigned int convertNeonSpamSettingsToFlags(char *str) {
 209     unsigned int i = 1, flags = 0;
 210     int j = 0;
 211     char *chars = SPAMSETTINGS_CHARS;
 212     while(*str) {
 213         for(; i <= SPAMSETTINGS_FLAGS; i = i << 1) {
 214             if(*str == chars[j]) {
 215                 flags |= i;
 216                 j++;
 217                 i = i << 1;
 218                 break;
 219             }
 220             j++;
 221         }
 222         str++;
 223     }
 224     return flags;
 225 }
 226
 227 int loadNeonSpamSettings(struct ChanNode *chan) {
 228     if(chan->spam_settings) return 0;
 229     struct NeonSpamSettings *settings = malloc(sizeof(*settings));
 230     if(!settings) {
 231         perror("malloc() failed");
 232         return 0;
 233     }
 234     MYSQL_RES *res;
 235     MYSQL_ROW row, defaults = NULL;
 236     loadChannelSettings(chan);
 237     if(!(chan->flags & CHANFLAG_CHAN_REGISTERED))
 238         return 0;
 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` FROM `channels` WHERE `channel_id` = '%d'", chan->channel_id);
 240     res = mysql_use();
 241     row = mysql_fetch_row(res);
 242     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]) {
 243         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` FROM `channels` WHERE `channel_name` = 'defaults'");
 244         res = mysql_use();
 245         defaults = mysql_fetch_row(res);
 246     }
 247     settings->flags = convertNeonSpamSettingsToFlags(row[0] ? row[0] : defaults[0]);
 248     settings->spam_amount = atoi(row[1] ? row[1] : defaults[1]);
 249     settings->exceptlevel[SPAMSETTINGS_SPAMEXCINDEX] = atoi(row[2] ? row[2] : defaults[2]);
 250     settings->sensibility_amount[SPAMSETTINGS_FLOODSENINDEX] = atoi(row[3] ? row[3] : defaults[3]);
 251     settings->sensibility_time[SPAMSETTINGS_FLOODSENINDEX] = atoi(row[4] ? row[4] : defaults[4]);
 252     settings->exceptlevel[SPAMSETTINGS_FLOODEXCINDEX] = atoi(row[5] ? row[5] : defaults[5]);
 253     settings->sensibility_amount[SPAMSETTINGS_JOINSENINDEX] = atoi(row[6] ? row[6] : defaults[6]);
 254     settings->sensibility_time[SPAMSETTINGS_JOINSENINDEX] = atoi(row[7] ? row[7] : defaults[7]);
 255     settings->exceptlevel[SPAMSETTINGS_JOINEXCINDEX] = atoi(row[8] ? row[8] : defaults[8]);
 256     settings->percent[SPAMSETTINGS_CAPSPERCENTINDEX] = atoi(row[9] ? row[9] : defaults[9]);
 257     settings->exceptlevel[SPAMSETTINGS_CAPSEXCINDEX] = atoi(row[10] ? row[10] : defaults[10]);
 258     settings->percent[SPAMSETTINGS_DIGITPERCENTINDEX] = atoi(row[11] ? row[11] : defaults[11]);
 259     settings->exceptlevel[SPAMSETTINGS_DIGITEXCINDEX] = atoi(row[12] ? row[12] : defaults[12]);
 260     settings->join_nodes = NULL;
 261     settings->lastmsg_time = 0;
 262     int i;
 263     for(i = 0; i < BOTNETSCAN_USERS; i++)
 264         settings->botnicks[i] = NULL;
 265     chan->spam_settings = settings;
 266     return 1;
 267 }
 268
 269 void freeNeonSpamSettings(struct NeonSpamSettings *settings) {
 270     struct NeonSpamJoinNode *joinnode, *nextjoinnode;
 271     for(joinnode = settings->join_nodes; joinnode; joinnode = nextjoinnode) {
 272         nextjoinnode = joinnode->next;
 273         freeJoinNode(joinnode);
 274     }
 275     free(settings);
 276 }
 277
 278 static void freeJoinNode(struct NeonSpamJoinNode *joinnode) {
 279     free(joinnode->ident);
 280     free(joinnode->host);
 281     free(joinnode);
 282 }
 283
 284 static struct NeonSpamJoinNode *getNeonSpamJoinNode(struct ChanUser *chanuser) {
 285     struct NeonSpamJoinNode *joinnode, *prevjoinnode = NULL, *nextjoinnode, *result = NULL;
 286     for(joinnode = chanuser->chan->spam_settings->join_nodes; joinnode; joinnode = nextjoinnode) {
 287         nextjoinnode = joinnode->next;
 288         if(!stricmp(joinnode->ident, chanuser->user->ident) && !stricmp(joinnode->host, chanuser->user->host)) {
 289             prevjoinnode = joinnode;
 290             result = joinnode;
 291         } else if(time(0) - joinnode->last_penalty_update > MAX_JOIN_TIME) {
 292             freeJoinNode(joinnode);
 293             if(prevjoinnode)
 294                 prevjoinnode->next = nextjoinnode;
 295             else
 296                 chanuser->chan->spam_settings->join_nodes = nextjoinnode;
 297         } else
 298             prevjoinnode = joinnode;
 299     }
 300     if(result)
 301         return result;
 302     joinnode = malloc(sizeof(*joinnode));
 303     if(!joinnode) {
 304         perror("malloc() failed");
 305         return NULL;
 306     }
 307     joinnode->ident = strdup(chanuser->user->ident);
 308     joinnode->host = strdup(chanuser->user->host);
 309     joinnode->last_penalty_update = time(0);
 310     joinnode->joinpenalty = 0;
 311     joinnode->next = chanuser->chan->spam_settings->join_nodes;
 312     chanuser->chan->spam_settings->join_nodes = joinnode;
 313     return joinnode;
 314 }
 315
 316 static void createSpamNode(struct ChanUser *chanuser) {
 317     struct NeonSpamNode *spamnode = malloc(sizeof(*spamnode));
 318     if(!spamnode) {
 319         perror("malloc() failed");
 320         return;
 321     }
 322     spamnode->lastmsg = 0;
 323     spamnode->spamcount = 0;
 324     spamnode->floodpenalty = 0;
 325     spamnode->last_penalty_update = time(0);
 326     chanuser->spamnode = spamnode;
 327 }
 328
 329 static int neonspam_event_freechan(struct ChanNode *chan) {
 330     if(chan->spam_settings)
 331         freeNeonSpamSettings(chan->spam_settings);
 332     return 1;
 333 }
 334
 335 void init_NeonSpam(int type) {
 336
 337     set_bot_alias(BOTID, BOTALIAS);
 338     start_bots(type);
 339
 340     if(type == MODSTATE_REBIND) return;
 341
 342     //register events
 343     bind_bot_ready(neonspam_bot_ready, module_id);
 344     bind_join(neonspam_event_join, module_id);
 345     bind_chanmsg(neonspam_event_chanmsg, module_id);
 346     bind_privctcp(general_event_privctcp, module_id);
 347     bind_kick(neonspam_event_kick, module_id);
 348     bind_freechan(neonspam_event_freechan, module_id);
 349
 350     set_trigger_callback(BOTID, module_id, neonspam_trigger_callback);
 351
 352     register_default_language_table(msgtab);
 353 }
 354
 355 void loop_NeonSpam() {
 356
 357 }
 358
 359 void free_NeonSpam(int type) {
 360     unbind_allcmd(BOTID);
 361     if(type == MODSTATE_STARTSTOP) {
 362         //disconnect all our bots
 363         struct ClientSocket *client;
 364         for(client = getBots(0, NULL); client; client = getBots(0, client)) {
 365             if(client->botid == BOTID) {
 366                 unbind_botwise_allcmd(0, client->clientid);
 367                 close_socket(client);
 368                 break;
 369             }
 370         }
 371     }
 372 }
 373
 374 #undef BOTID
 375 #undef BOTALIAS