5cdea3b0608bfcddaeffd0a020d467429318e0e5
[NeonServV5.git] / 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     {"SS_SET_PERCENT", "%u is an invalid percent value (valid: 0-100)"},
45     {"SS_SET_SENSIBILITY", "%s is an invalid sensibility format. (valid: amount:time   e.g. 5:10)"},
46     {"SS_SET_SENSIBILITY_AMOUNT", "%d is an invalid amount value. (valid: %d-%d)"},
47     {"SS_SET_SENSIBILITY_TIME", "%d is an invalid time value. (valid: %d-%d)"},
48     {"SS_SET_SPAMLIMIT", "%d is an invalid spamlimit value. (valid: %d-%d)"},
49     {"SS_SET_OPTION_SpamReaction_0", "Kick"},
50     {"SS_SET_OPTION_SpamReaction_1", "KickBan"},
51     {"SS_SET_OPTION_SpamReaction_2", "Timed Ban"},
52     {"SS_SET_OPTION_FloodReaction_0", "Kick"},
53     {"SS_SET_OPTION_FloodReaction_1", "KickBan"},
54     {"SS_SET_OPTION_FloodReaction_2", "Timed Ban"},
55     {"SS_SET_OPTION_JoinReaction_0", "Kick"},
56     {"SS_SET_OPTION_JoinReaction_1", "KickBan"},
57     {"SS_SET_OPTION_JoinReaction_2", "Timed Ban"},
58     {"SS_SET_OPTION_CapsReaction_0", "Kick"},
59     {"SS_SET_OPTION_CapsReaction_1", "KickBan"},
60     {"SS_SET_OPTION_CapsReaction_2", "Timed Ban"},
61     {"SS_SET_OPTION_DigitReaction_0", "Kick"},
62     {"SS_SET_OPTION_DigitReaction_1", "KickBan"},
63     {"SS_SET_OPTION_DigitReaction_2", "Timed Ban"},
64     {NULL, NULL}
65 };
66
67 static unsigned int convertNeonSpamSettingsToFlags(char *str);
68 static int loadNeonSpamSettings(struct ChanNode *chan);
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_bot_ready(struct ClientSocket *client) {
92     MYSQL_RES *res;
93     MYSQL_ROW row;
94     
95     printf_mysql_query("SELECT `automodes` FROM `bots` WHERE `id` = '%d'", client->clientid);
96     res = mysql_use();
97     if ((row = mysql_fetch_row(res)) != NULL) {
98         putsock(client, "MODE %s +%s", client->user->nick, row[0]);
99     }
100     
101     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);
102     res = mysql_use();
103     
104     while ((row = mysql_fetch_row(res)) != NULL) {
105         putsock(client, "JOIN %s %s", row[0], row[1]);
106     }
107 }
108
109 static void neonspam_trigger_callback(struct ChanNode *chan, char *trigger) {
110     MYSQL_RES *res;
111     MYSQL_ROW row;
112     loadChannelSettings(chan);
113     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);
114     res = mysql_use();
115     row = mysql_fetch_row(res);
116     strcpy(trigger, (strlen(row[0]) ? row[0] : "~"));
117 }
118
119 static void start_bots() {
120     struct UserNode *user;
121     struct ClientSocket *client;
122     MYSQL_RES *res, *res2;
123     MYSQL_ROW row;
124     
125     printf_mysql_query("SELECT `nick`, `ident`, `realname`, `server`, `port`, `pass`, `textbot`, `id`, `queue` FROM `bots` WHERE `botclass` = '%d' AND `active` = '1'", BOTID);
126     res = mysql_use();
127     
128     while ((row = mysql_fetch_row(res)) != NULL) {
129         
130         user = addUser(row[0]);
131         strcpy(user->ident, row[1]);
132         strcpy(user->realname, row[2]);
133         user->flags |= USERFLAG_ISBOT;
134         client = create_socket(row[3], atoi(row[4]), row[5], user);
135         client->flags |= (strcmp(row[6], "0") ? SOCKET_FLAG_PREFERRED : 0);
136         client->flags |= (strcmp(row[8], "0") ? SOCKET_FLAG_USE_QUEUE : 0);
137         client->botid = BOTID;
138         client->clientid = atoi(row[7]);
139         connect_socket(client);
140         printf_mysql_query("SELECT `command`, `function`, `parameters`, `global_access`, `chan_access` FROM `bot_binds` WHERE `botclass` = '%d'", client->botid);
141         res2 = mysql_use();
142         while ((row = mysql_fetch_row(res2)) != NULL) {
143             if(bind_cmd_to_command(BOTID, row[0], row[1])) {
144                 if(row[2] && strcmp(row[2], "")) {
145                     bind_set_parameters(BOTID, row[0], row[2]);
146                 }
147                 if(row[3]) {
148                     bind_set_global_access(BOTID, row[0], atoi(row[3]));
149                 }
150                 if(row[4]) {
151                     bind_set_channel_access(BOTID, row[0], row[4]);
152                 }
153             }
154         }
155     }
156 }
157
158 char* convertNeonSpamSettingsToString(unsigned int flags, char *buffer) {
159     int pos = 0;
160     unsigned int i;
161     int j = 0;
162     char *chars = SPAMSETTINGS_CHARS;
163     for(i = 1; i <= SPAMSETTINGS_FLAGS; i = i << 1) {
164         if(flags & i)
165             buffer[pos++] = chars[j];
166         j++;
167     }
168     buffer[pos] = '\0';
169     return buffer;
170 }
171
172 static unsigned int convertNeonSpamSettingsToFlags(char *str) {
173     unsigned int i = 1, flags = 0;
174     int j = 0;
175     char *chars = SPAMSETTINGS_CHARS;
176     while(*str) {
177         for(; i <= SPAMSETTINGS_FLAGS; i = i << 1) {
178             if(*str == chars[j]) {
179                 flags |= i;
180                 j++;
181                 i = i << 1;
182                 break;
183             }
184             j++;
185         }
186         str++;
187     }
188     return flags;
189 }
190
191 static int loadNeonSpamSettings(struct ChanNode *chan) {
192     if(chan->spam_settings) return 0;
193     struct NeonSpamSettings *settings = malloc(sizeof(*settings));
194     if(!settings) {
195         perror("malloc() failed");
196         return 0;
197     }
198     MYSQL_RES *res;
199     MYSQL_ROW row, defaults = NULL;
200     loadChannelSettings(chan);
201     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);
202     res = mysql_use();
203     row = mysql_fetch_row(res);
204     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]) {
205         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'");
206         res = mysql_use();
207         defaults = mysql_fetch_row(res);
208     }
209     settings->flags = convertNeonSpamSettingsToFlags(row[0] ? row[0] : defaults[0]);
210     settings->spam_amount = atoi(row[1] ? row[1] : defaults[1]);
211     settings->exceptlevel[SPAMSETTINGS_SPAMEXCINDEX] = atoi(row[2] ? row[2] : defaults[2]);
212     settings->sensibility_amount[SPAMSETTINGS_FLOODSENINDEX] = atoi(row[3] ? row[3] : defaults[3]);
213     settings->sensibility_time[SPAMSETTINGS_FLOODSENINDEX] = atoi(row[4] ? row[4] : defaults[4]);
214     settings->exceptlevel[SPAMSETTINGS_FLOODEXCINDEX] = atoi(row[5] ? row[5] : defaults[5]);
215     settings->sensibility_amount[SPAMSETTINGS_JOINSENINDEX] = atoi(row[6] ? row[6] : defaults[6]);
216     settings->sensibility_time[SPAMSETTINGS_JOINSENINDEX] = atoi(row[7] ? row[7] : defaults[7]);
217     settings->exceptlevel[SPAMSETTINGS_JOINEXCINDEX] = atoi(row[8] ? row[8] : defaults[8]);
218     settings->percent[SPAMSETTINGS_CAPSPERCENTINDEX] = atoi(row[9] ? row[9] : defaults[9]);
219     settings->exceptlevel[SPAMSETTINGS_CAPSEXCINDEX] = atoi(row[10] ? row[10] : defaults[10]);
220     settings->percent[SPAMSETTINGS_DIGITPERCENTINDEX] = atoi(row[11] ? row[11] : defaults[11]);
221     settings->exceptlevel[SPAMSETTINGS_DIGITEXCINDEX] = atoi(row[12] ? row[12] : defaults[12]);
222     settings->join_nodes = NULL;
223     settings->lastmsg_time = 0;
224     int i;
225     for(i = 0; i < BOTNETSCAN_USERS; i++) 
226         settings->botnicks[i] = NULL;
227     chan->spam_settings = settings;
228     return 1;
229 }
230
231 void freeNeonSpamSettings(struct NeonSpamSettings *settings) {
232     struct NeonSpamJoinNode *joinnode, *nextjoinnode;
233     for(joinnode = settings->join_nodes; joinnode; joinnode = nextjoinnode) {
234         nextjoinnode = joinnode->next;
235         freeJoinNode(joinnode);
236     }
237     free(settings);
238 }
239
240 static void freeJoinNode(struct NeonSpamJoinNode *joinnode) {
241     free(joinnode->ident);
242     free(joinnode->host);
243     free(joinnode);
244 }
245
246 static struct NeonSpamJoinNode *getNeonSpamJoinNode(struct ChanUser *chanuser) {
247     struct NeonSpamJoinNode *joinnode, *prevjoinnode = NULL, *nextjoinnode, *result = NULL;
248     for(joinnode = chanuser->chan->spam_settings->join_nodes; joinnode; joinnode = nextjoinnode) {
249         nextjoinnode = joinnode->next;
250         if(!stricmp(joinnode->ident, chanuser->user->ident) && !stricmp(joinnode->host, chanuser->user->host)) {
251             prevjoinnode = joinnode;
252             result = joinnode;
253         } else if(time(0) - joinnode->last_penalty_update > MAX_JOIN_TIME) {
254             freeJoinNode(joinnode);
255             if(prevjoinnode)
256                 prevjoinnode->next = nextjoinnode;
257             else
258                 chanuser->chan->spam_settings->join_nodes = nextjoinnode;
259         } else 
260             prevjoinnode = joinnode;
261     }
262     if(result)
263         return result;
264     joinnode = malloc(sizeof(*joinnode));
265     if(!joinnode) {
266         perror("malloc() failed");
267         return NULL;
268     }
269     joinnode->ident = strdup(chanuser->user->ident);
270     joinnode->host = strdup(chanuser->user->host);
271     joinnode->last_penalty_update = time(0);
272     joinnode->joinpenalty = 0;
273     joinnode->next = chanuser->chan->spam_settings->join_nodes;
274     chanuser->chan->spam_settings->join_nodes = joinnode;
275     return joinnode;
276 }
277
278 static void createSpamNode(struct ChanUser *chanuser) {
279     struct NeonSpamNode *spamnode = malloc(sizeof(*spamnode));
280     if(!spamnode) {
281         perror("malloc() failed");
282         return;
283     }
284     spamnode->lastmsg = 0;
285     spamnode->spamcount = 0;
286     spamnode->floodpenalty = 0;
287     spamnode->last_penalty_update = time(0);
288     chanuser->spamnode = spamnode;
289 }
290
291 void init_NeonSpam() {
292     
293     #define USER_COMMAND(NAME,FUNCTION,PARAMCOUNT,PRIVS,FLAGS) register_command(BOTID, NAME, FUNCTION, PARAMCOUNT, PRIVS, 0, FLAGS)
294     //               NAME              FUNCTION        PARAMS     PRIVS                FLAGS
295     USER_COMMAND("netinfo",      neonserv_cmd_netinfo,   0, NULL,                   0);
296     USER_COMMAND("version",      neonserv_cmd_version,   0, NULL,                   0);
297     USER_COMMAND("set",          neonspam_cmd_set,       0, "#channel_setters",     CMDFLAG_REQUIRE_CHAN | CMDFLAG_REGISTERED_CHAN | CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_LOG);
298     #undef USER_COMMAND
299     
300     #define OPER_COMMAND(NAME,FUNCTION,PARAMCOUNT,GACCESS,FLAGS) register_command(BOTID, NAME, FUNCTION, PARAMCOUNT, NULL, GACCESS, FLAGS)
301     //            NAME            FUNCTION              PARAMS  ACCS  FLAGS
302     OPER_COMMAND("register",     neonserv_cmd_register,  1,     200,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
303     OPER_COMMAND("unregister",   neonserv_cmd_unregister,1,     200,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
304     OPER_COMMAND("bind",         neonserv_cmd_bind,      2,     900,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_OPLOG);
305     OPER_COMMAND("unbind",       neonserv_cmd_unbind,    1,     900,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_OPLOG);
306     OPER_COMMAND("csuspend",     neonserv_cmd_csuspend,  1,     100,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
307     OPER_COMMAND("cunsuspend",   neonserv_cmd_cunsuspend,1,     100,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
308     OPER_COMMAND("say",          neonserv_cmd_say,       2,     600,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
309     OPER_COMMAND("emote",        neonserv_cmd_emote,     2,     600,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
310     OPER_COMMAND("notice",       neonserv_cmd_notice,    2,     600,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_CHAN_PARAM | CMDFLAG_OPLOG);
311     OPER_COMMAND("raw",          neonserv_cmd_raw,       1,     800,  CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_OPLOG);
312     OPER_COMMAND("god",          neonserv_cmd_god,       0,     1,    CMDFLAG_REQUIRE_AUTH | CMDFLAG_CHECK_AUTH | CMDFLAG_OPLOG);
313     #undef OPER_COMMAND
314     
315     start_bots();
316     
317     //register events
318     bind_bot_ready(neonspam_bot_ready);
319     bind_join(neonspam_event_join);
320     bind_chanmsg(neonspam_event_chanmsg);
321     bind_privctcp(general_event_privctcp);
322     
323     set_trigger_callback(BOTID, neonspam_trigger_callback);
324     
325     register_default_language_table(msgtab);
326 }
327
328 void loop_NeonSpam() {
329     
330 }
331
332 void free_NeonSpam() {
333     
334 }
335
336 #undef BOTID