src/bot_NeonSpam.c

   1 /* bot_NeonSpam.c - NeonServ v5.1
   2  * Copyright (C) 2011  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 "bot_NeonSpam.h"
  19 #include "modcmd.h"
  20 #include "IRCParser.h"
  21 #include "IRCEvents.h"
  22 #include "UserNode.h"
  23 #include "ChanNode.h"
  24 #include "ChanUser.h"
  25 #include "ModeNode.h"
  26 #include "BanNode.h"
  27 #include "ClientSocket.h"
  28 #include "mysqlConn.h"
  29 #include "lang.h"
  30 #include "HandleInfoHandler.h"
  31 #include "WHOHandler.h"
  32 #include "DBHelper.h"
  33 #include "tools.h"
  34 #include "timeq.h"
  35 #include "version.h"
  36 #include "EventLogger.h"
  37 #include "bots.h"
  38 #include "cmd_neonserv.h"
  39 #include "cmd_neonspam.h"
  40
  41 #define BOTID 2
  42
  43 static const struct default_language_entry msgtab[] = {
  44     {NULL, NULL}
  45 };
  46
  47 static unsigned int convertNeonSpamSettingsToFlags(char *str);
  48 static int loadNeonSpamSettings(struct ChanNode *chan);
  49 static void createSpamNode(struct ChanUser *chanuser);
  50 static void freeJoinNode(struct NeonSpamJoinNode *joinnode);
  51 static struct NeonSpamJoinNode *getNeonSpamJoinNode(struct ChanUser *chanuser);
  52
  53 #define SPAMSERV_CHECK_IGNORE 0
  54 #define SPAMSERV_CHECK_WARN   1
  55 #define SPAMSERV_CHECK_PUNISH 2
  56 #define SPAMSERV_CHECK_DEAD   3 /* scanner has already killed the user */
  57
  58 #define SPAMSERV_MSG_SPAM       "Spamming"
  59 #define SPAMSERV_MSG_FLOOD      "Flooding the channel/network"
  60 #define SPAMSERV_MSG_ADV        "Advertising"
  61 #define SPAMSERV_MSG_JOINFLOOD  "Join flooding the channel"
  62 #define SPAMSERV_MSG_WARNING    "%s is against the channel rules"
  63 #define SPAMSERV_MSG_BOTNET     "BotNet detected."
  64 #define SPAMSERV_MSG_CAPS       "Using too many chars in UPPER CASE"
  65 #define SPAMSERV_MSG_DIGIT      "Using too many numeric chars"
  66
  67 //EVENTS
  68 #include "event_neonspam_join.c"
  69 #include "event_neonspam_chanmsg.c"
  70
  71 static void neonspam_bot_ready(struct ClientSocket *client) {
  72     MYSQL_RES *res;
  73     MYSQL_ROW row;
  74
  75     printf_mysql_query("SELECT `automodes` FROM `bots` WHERE `id` = '%d'", client->clientid);
  76     res = mysql_use();
  77     if ((row = mysql_fetch_row(res)) != NULL) {
  78         putsock(client, "MODE %s +%s", client->user->nick, row[0]);
  79     }
  80
  81     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);
  82     res = mysql_use();
  83
  84     while ((row = mysql_fetch_row(res)) != NULL) {
  85         putsock(client, "JOIN %s %s", row[0], row[1]);
  86     }
  87 }
  88
  89 static void neonspam_trigger_callback(struct ChanNode *chan, char *trigger) {
  90     MYSQL_RES *res;
  91     MYSQL_ROW row;
  92     loadChannelSettings(chan);
  93     printf_mysql_query("SELECT `trigger` FROM `bot_channels` LEFT JOIN `bots` ON `botid` = `bots`.`id` WHERE `chanid` = '%d' AND `botclass` = '%d'", chan->channel_id, BOTID);
  94     res = mysql_use();
  95     row = mysql_fetch_row(res);
  96     strcpy(trigger, (strlen(row[0]) ? row[0] : "~"));
  97 }
  98
  99 static void start_bots() {
 100     struct UserNode *user;
 101     struct ClientSocket *client;
 102     MYSQL_RES *res, *res2;
 103     MYSQL_ROW row;
 104
 105     printf_mysql_query("SELECT `nick`, `ident`, `realname`, `server`, `port`, `pass`, `textbot`, `id`, `queue` FROM `bots` WHERE `botclass` = '%d' AND `active` = '1'", BOTID);
 106     res = mysql_use();
 107
 108     while ((row = mysql_fetch_row(res)) != NULL) {
 109
 110         user = addUser(row[0]);
 111         strcpy(user->ident, row[1]);
 112         strcpy(user->realname, row[2]);
 113         user->flags |= USERFLAG_ISBOT;
 114         client = create_socket(row[3], atoi(row[4]), row[5], user);
 115         client->flags |= (strcmp(row[6], "0") ? SOCKET_FLAG_PREFERRED : 0);
 116         client->flags |= (strcmp(row[8], "0") ? SOCKET_FLAG_USE_QUEUE : 0);
 117         client->botid = BOTID;
 118         client->clientid = atoi(row[7]);
 119         connect_socket(client);
 120         printf_mysql_query("SELECT `command`, `function`, `parameters`, `global_access`, `chan_access` FROM `bot_binds` WHERE `botclass` = '%d'", client->botid);
 121         res2 = mysql_use();
 122         while ((row = mysql_fetch_row(res2)) != NULL) {
 123             if(bind_cmd_to_command(BOTID, row[0], row[1])) {
 124                 if(row[2] && strcmp(row[2], "")) {
 125                     bind_set_parameters(BOTID, row[0], row[2]);
 126                 }
 127                 if(row[3]) {
 128                     bind_set_global_access(BOTID, row[0], atoi(row[3]));
 129                 }
 130                 if(row[4]) {
 131                     bind_set_channel_access(BOTID, row[0], row[4]);
 132                 }
 133             }
 134         }
 135     }
 136 }
 137
 138 char* convertNeonSpamSettingsToString(unsigned int flags, char *buffer) {
 139     int pos = 0;
 140     unsigned int i;
 141     int j = 0;
 142     char *chars = SPAMSETTINGS_CHARS;
 143     for(i = 1; i <= SPAMSETTINGS_FLAGS; i = i << 1) {
 144         if(flags & i)
 145             buffer[pos++] = chars[j];
 146         j++;
 147     }
 148     buffer[pos] = '\0';
 149     return buffer;
 150 }
 151
 152 static unsigned int convertNeonSpamSettingsToFlags(char *str) {
 153     unsigned int i = 1, flags = 0;
 154     int j = 0;
 155     char *chars = SPAMSETTINGS_CHARS;
 156     while(*str) {
 157         for(; i <= SPAMSETTINGS_FLAGS; i = i << 1) {
 158             if(*str == chars[j]) {
 159                 flags |= i;
 160                 j++;
 161                 break;
 162             }
 163             j++;
 164         }
 165         str++;
 166     }
 167     return flags;
 168 }
 169
 170 static int loadNeonSpamSettings(struct ChanNode *chan) {
 171     if(chan->spam_settings) return 0;
 172     struct NeonSpamSettings *settings = malloc(sizeof(*settings));
 173     if(!settings) {
 174         perror("malloc() failed");
 175         return 0;
 176     }
 177     MYSQL_RES *res;
 178     MYSQL_ROW row, defaults = NULL;
 179     loadChannelSettings(chan);
 180     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);
 181     res = mysql_use();
 182     row = mysql_fetch_row(res);
 183     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]) {
 184         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'");
 185         res = mysql_use();
 186         defaults = mysql_fetch_row(res);
 187     }
 188     settings->flags = convertNeonSpamSettingsToFlags(row[0] ? row[0] : defaults[0]);
 189     settings->spam_amount = atoi(row[1] ? row[1] : defaults[1]);
 190     settings->exceptlevel[SPAMSETTINGS_SPAMEXCINDEX] = atoi(row[2] ? row[2] : defaults[2]);
 191     settings->sensibility_amount[SPAMSETTINGS_FLOODSENINDEX] = atoi(row[3] ? row[3] : defaults[3]);
 192     settings->sensibility_time[SPAMSETTINGS_FLOODSENINDEX] = atoi(row[4] ? row[4] : defaults[4]);
 193     settings->exceptlevel[SPAMSETTINGS_FLOODEXCINDEX] = atoi(row[5] ? row[5] : defaults[5]);
 194     settings->sensibility_amount[SPAMSETTINGS_JOINSENINDEX] = atoi(row[6] ? row[6] : defaults[6]);
 195     settings->sensibility_time[SPAMSETTINGS_JOINSENINDEX] = atoi(row[7] ? row[7] : defaults[7]);
 196     settings->exceptlevel[SPAMSETTINGS_JOINEXCINDEX] = atoi(row[8] ? row[8] : defaults[8]);
 197     settings->percent[SPAMSETTINGS_CAPSPERCENTINDEX] = atoi(row[9] ? row[9] : defaults[9]);
 198     settings->exceptlevel[SPAMSETTINGS_CAPSEXCINDEX] = atoi(row[10] ? row[10] : defaults[10]);
 199     settings->percent[SPAMSETTINGS_DIGITPERCENTINDEX] = atoi(row[11] ? row[11] : defaults[11]);
 200     settings->exceptlevel[SPAMSETTINGS_DIGITEXCINDEX] = atoi(row[12] ? row[12] : defaults[12]);
 201     settings->join_nodes = NULL;
 202     settings->lastmsg_time = 0;
 203     int i;
 204     for(i = 0; i < BOTNETSCAN_USERS; i++)
 205         settings->botnicks[i] = NULL;
 206     chan->spam_settings = settings;
 207     return 1;
 208 }
 209
 210 void freeNeonSpamSettings(struct NeonSpamSettings *settings) {
 211     struct NeonSpamJoinNode *joinnode, *nextjoinnode;
 212     for(joinnode = settings->join_nodes; joinnode; joinnode = nextjoinnode) {
 213         nextjoinnode = joinnode->next;
 214         freeJoinNode(joinnode);
 215     }
 216     free(settings);
 217 }
 218
 219 static void freeJoinNode(struct NeonSpamJoinNode *joinnode) {
 220     free(joinnode->ident);
 221     free(joinnode->host);
 222     free(joinnode);
 223 }
 224
 225 static struct NeonSpamJoinNode *getNeonSpamJoinNode(struct ChanUser *chanuser) {
 226     struct NeonSpamJoinNode *joinnode, *prevjoinnode = NULL, *nextjoinnode, *result = NULL;
 227     for(joinnode = chanuser->chan->spam_settings->join_nodes; joinnode; joinnode = nextjoinnode) {
 228         nextjoinnode = joinnode->next;
 229         if(!stricmp(joinnode->ident, chanuser->user->ident) && !stricmp(joinnode->host, chanuser->user->host)) {
 230             prevjoinnode = joinnode;
 231             result = joinnode;
 232         } else if(time(0) - joinnode->last_penalty_update > MAX_JOIN_TIME) {
 233             freeJoinNode(joinnode);
 234             if(prevjoinnode)
 235                 prevjoinnode->next = nextjoinnode;
 236             else
 237                 chanuser->chan->spam_settings->join_nodes = nextjoinnode;
 238         } else
 239             prevjoinnode = joinnode;
 240     }
 241     if(result)
 242         return result;
 243     joinnode = malloc(sizeof(*joinnode));
 244     if(!joinnode) {
 245         perror("malloc() failed");
 246         return NULL;
 247     }
 248     joinnode->ident = strdup(chanuser->user->ident);
 249     joinnode->host = strdup(chanuser->user->host);
 250     joinnode->last_penalty_update = time(0);
 251     joinnode->joinpenalty = 0;
 252     joinnode->next = chanuser->chan->spam_settings->join_nodes;
 253     chanuser->chan->spam_settings->join_nodes = joinnode;
 254     return joinnode;
 255 }
 256
 257 static void createSpamNode(struct ChanUser *chanuser) {
 258     struct NeonSpamNode *spamnode = malloc(sizeof(*spamnode));
 259     if(!spamnode) {
 260         perror("malloc() failed");
 261         return;
 262     }
 263     spamnode->lastmsg = 0;
 264     spamnode->spamcount = 0;
 265     spamnode->floodpenalty = 0;
 266     spamnode->last_penalty_update = time(0);
 267     chanuser->spamnode = spamnode;
 268 }
 269
 270 void init_NeonSpam() {
 271
 272     #define USER_COMMAND(NAME,FUNCTION,PARAMCOUNT,PRIVS,FLAGS) register_command(BOTID, NAME, FUNCTION, PARAMCOUNT, PRIVS, 0, FLAGS)
 273     //               NAME              FUNCTION        PARAMS     PRIVS                FLAGS
 274     USER_COMMAND("netinfo",      neonserv_cmd_netinfo,   0, NULL,                   0);
 275     USER_COMMAND("version",      neonserv_cmd_version,   0, NULL,                   0);
 276     USER_COMMAND("set",          neonspam_cmd_set,       0, "#channel_setters",     CMDFLAG_REQUIRE_CHAN | CMDFLAG_REGISTERED_CHAN | CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_LOG);
 277     #undef USER_COMMAND
 278
 279     #define OPER_COMMAND(NAME,FUNCTION,PARAMCOUNT,GACCESS,FLAGS) register_command(BOTID, NAME, FUNCTION, PARAMCOUNT, NULL, GACCESS, FLAGS)
 280     //            NAME            FUNCTION              PARAMS  ACCS  FLAGS
 281     OPER_COMMAND("register",     neonserv_cmd_register,  1,     200,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
 282     OPER_COMMAND("unregister",   neonserv_cmd_unregister,1,     200,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
 283     OPER_COMMAND("bind",         neonserv_cmd_bind,      2,     900,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_OPLOG);
 284     OPER_COMMAND("unbind",       neonserv_cmd_unbind,    1,     900,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_OPLOG);
 285     OPER_COMMAND("csuspend",     neonserv_cmd_csuspend,  1,     100,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
 286     OPER_COMMAND("cunsuspend",   neonserv_cmd_cunsuspend,1,     100,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
 287     OPER_COMMAND("say",          neonserv_cmd_say,       2,     600,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
 288     OPER_COMMAND("emote",        neonserv_cmd_emote,     2,     600,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
 289     OPER_COMMAND("notice",       neonserv_cmd_notice,    2,     600,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
 290     OPER_COMMAND("raw",          neonserv_cmd_raw,       1,     800,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_OPLOG);
 291     OPER_COMMAND("god",          neonserv_cmd_god,       0,     1,    CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_OPLOG);
 292     #undef OPER_COMMAND
 293
 294     start_bots();
 295
 296     //register events
 297     bind_bot_ready(neonspam_bot_ready);
 298     bind_join(neonspam_event_join);
 299     bind_chanmsg(neonspam_event_chanmsg);
 300     bind_privctcp(general_event_privctcp);
 301
 302     set_trigger_callback(BOTID, neonspam_trigger_callback);
 303
 304     register_default_language_table(msgtab);
 305 }
 306
 307 void loop_NeonSpam() {
 308
 309 }
 310
 311 void free_NeonSpam() {
 312
 313 }
 314
 315 #undef BOTID