Another year is about to end... So we have to update these damn copyright information :P
[NeonServV5.git] / src / bot_NeonSpam.c
1 /* bot_NeonSpam.c - NeonServ v5.3
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
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 #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 int loadNeonSpamSettings(struct ChanNode *chan);
70 static void createSpamNode(struct ChanUser *chanuser);
71 static void freeJoinNode(struct NeonSpamJoinNode *joinnode);
72 static struct NeonSpamJoinNode *getNeonSpamJoinNode(struct ChanUser *chanuser);
73
74 #define SPAMSERV_CHECK_IGNORE 0
75 #define SPAMSERV_CHECK_WARN   1
76 #define SPAMSERV_CHECK_PUNISH 2
77 #define SPAMSERV_CHECK_DEAD   3 /* scanner has already killed the user */
78
79 #define SPAMSERV_MSG_SPAM       "Spamming"
80 #define SPAMSERV_MSG_FLOOD      "Flooding the channel/network"
81 #define SPAMSERV_MSG_ADV        "Advertising"
82 #define SPAMSERV_MSG_JOINFLOOD  "Join flooding the channel"
83 #define SPAMSERV_MSG_WARNING    "%s is against the channel rules"
84 #define SPAMSERV_MSG_BOTNET     "BotNet detected."
85 #define SPAMSERV_MSG_CAPS       "Using too many chars in UPPER CASE"
86 #define SPAMSERV_MSG_DIGIT      "Using too many numeric chars"
87
88 //EVENTS
89 #include "event_neonspam_join.c"
90 #include "event_neonspam_chanmsg.c"
91
92 static void neonspam_event_kick(struct UserNode *user, struct ChanUser *target, char *reason) {
93     struct ChanNode *chan = target->chan;
94     if(isBot(target->user)) {
95         struct ClientSocket *client = getChannelBot(chan, 0);
96         struct ClientSocket *bot = client;
97         for(client = getBots(SOCKET_FLAG_READY, NULL); client; client = getBots(SOCKET_FLAG_READY, client)) {
98             if(client->user == target->user) {
99                 break;
100             }
101         }
102         if(!client) return;
103         if(bot && bot != client && (isModeSet(chan->modes, 'i') || isModeSet(chan->modes, 'a') || isModeSet(chan->modes, 'l'))) {
104             struct ChanUser *chanuser = getChanUser(bot->user, chan);
105             if(chanuser && chanuser->flags & CHANUSERFLAG_OPPED)
106                 putsock(bot, "INVITE %s %s", target->user->nick, chan->name);
107         }
108         char *key = "";
109         if(isModeSet(chan->modes, 'k')) {
110             key = getModeValue(chan->modes, 'k');
111         }
112         putsock(client, "JOIN %s %s", chan->name, key);
113         return;
114     }
115 }
116
117 static void neonspam_bot_ready(struct ClientSocket *client) {
118     MYSQL_RES *res;
119     MYSQL_ROW row;
120     
121     printf_mysql_query("SELECT `automodes` FROM `bots` WHERE `id` = '%d'", client->clientid);
122     res = mysql_use();
123     if ((row = mysql_fetch_row(res)) != NULL) {
124         putsock(client, "MODE %s +%s", client->user->nick, row[0]);
125     }
126     
127     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);
128     res = mysql_use();
129     
130     while ((row = mysql_fetch_row(res)) != NULL) {
131         putsock(client, "JOIN %s %s", row[0], row[1]);
132     }
133 }
134
135 static void neonspam_trigger_callback(int clientid, struct ChanNode *chan, char *trigger) {
136     MYSQL_RES *res;
137     MYSQL_ROW row;
138     loadChannelSettings(chan);
139     if(!(chan->flags & CHANFLAG_CHAN_REGISTERED)) {
140         strcpy(trigger, "~");
141         return;
142     }
143     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);
144     res = mysql_use();
145     row = mysql_fetch_row(res);
146     if(row[0] && *row[0])
147         strcpy(trigger, row[0]);
148     else
149         strcpy(trigger, ((row[1] && *row[1]) ? row[1] : "~"));
150 }
151
152 static void start_bots() {
153     struct ClientSocket *client;
154     MYSQL_RES *res, *res2;
155     MYSQL_ROW row;
156     
157     printf_mysql_query("SELECT `nick`, `ident`, `realname`, `server`, `port`, `pass`, `textbot`, `id`, `queue`, `ssl`, `bind` FROM `bots` WHERE `botclass` = '%d' AND `active` = '1'", BOTID);
158     res = mysql_use();
159     
160     while ((row = mysql_fetch_row(res)) != NULL) {
161         client = create_socket(row[3], atoi(row[4]), row[10], row[5], row[0], row[1], row[2]);
162         client->flags |= (strcmp(row[6], "0") ? SOCKET_FLAG_PREFERRED : 0);
163         client->flags |= (strcmp(row[8], "0") ? SOCKET_FLAG_USE_QUEUE : 0);
164         client->flags |= (strcmp(row[9], "0") ? SOCKET_FLAG_SSL : 0);
165         client->botid = BOTID;
166         client->clientid = atoi(row[7]);
167         connect_socket(client);
168     }
169     
170     printf_mysql_query("SELECT `command`, `function`, `parameters`, `global_access`, `chan_access`, `flags` FROM `bot_binds` WHERE `botclass` = '%d'", BOTID);
171     res2 = mysql_use();
172     while ((row = mysql_fetch_row(res2)) != NULL) {
173         if(bind_cmd_to_command(BOTID, row[0], row[1])) {
174             if(row[2] && strcmp(row[2], "")) {
175                 bind_set_parameters(BOTID, row[0], row[2]);
176             }
177             if(row[3]) {
178                 bind_set_global_access(BOTID, row[0], atoi(row[3]));
179             }
180             if(row[4]) {
181                 bind_set_channel_access(BOTID, row[0], row[4]);
182             }
183             if(strcmp(row[5], "0"))
184                 bind_set_bind_flags(BOTID, row[0], atoi(row[5]));
185         }
186     }
187     bind_unbound_required_functions(BOTID);
188 }
189
190 char* convertNeonSpamSettingsToString(unsigned int flags, char *buffer) {
191     int pos = 0;
192     unsigned int i;
193     int j = 0;
194     char *chars = SPAMSETTINGS_CHARS;
195     for(i = 1; i <= SPAMSETTINGS_FLAGS; i = i << 1) {
196         if(flags & i)
197             buffer[pos++] = chars[j];
198         j++;
199     }
200     buffer[pos] = '\0';
201     return buffer;
202 }
203
204 static unsigned int convertNeonSpamSettingsToFlags(char *str) {
205     unsigned int i = 1, flags = 0;
206     int j = 0;
207     char *chars = SPAMSETTINGS_CHARS;
208     while(*str) {
209         for(; i <= SPAMSETTINGS_FLAGS; i = i << 1) {
210             if(*str == chars[j]) {
211                 flags |= i;
212                 j++;
213                 i = i << 1;
214                 break;
215             }
216             j++;
217         }
218         str++;
219     }
220     return flags;
221 }
222
223 static int loadNeonSpamSettings(struct ChanNode *chan) {
224     if(chan->spam_settings) return 0;
225     struct NeonSpamSettings *settings = malloc(sizeof(*settings));
226     if(!settings) {
227         perror("malloc() failed");
228         return 0;
229     }
230     MYSQL_RES *res;
231     MYSQL_ROW row, defaults = NULL;
232     loadChannelSettings(chan);
233     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);
234     res = mysql_use();
235     row = mysql_fetch_row(res);
236     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]) {
237         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'");
238         res = mysql_use();
239         defaults = mysql_fetch_row(res);
240     }
241     settings->flags = convertNeonSpamSettingsToFlags(row[0] ? row[0] : defaults[0]);
242     settings->spam_amount = atoi(row[1] ? row[1] : defaults[1]);
243     settings->exceptlevel[SPAMSETTINGS_SPAMEXCINDEX] = atoi(row[2] ? row[2] : defaults[2]);
244     settings->sensibility_amount[SPAMSETTINGS_FLOODSENINDEX] = atoi(row[3] ? row[3] : defaults[3]);
245     settings->sensibility_time[SPAMSETTINGS_FLOODSENINDEX] = atoi(row[4] ? row[4] : defaults[4]);
246     settings->exceptlevel[SPAMSETTINGS_FLOODEXCINDEX] = atoi(row[5] ? row[5] : defaults[5]);
247     settings->sensibility_amount[SPAMSETTINGS_JOINSENINDEX] = atoi(row[6] ? row[6] : defaults[6]);
248     settings->sensibility_time[SPAMSETTINGS_JOINSENINDEX] = atoi(row[7] ? row[7] : defaults[7]);
249     settings->exceptlevel[SPAMSETTINGS_JOINEXCINDEX] = atoi(row[8] ? row[8] : defaults[8]);
250     settings->percent[SPAMSETTINGS_CAPSPERCENTINDEX] = atoi(row[9] ? row[9] : defaults[9]);
251     settings->exceptlevel[SPAMSETTINGS_CAPSEXCINDEX] = atoi(row[10] ? row[10] : defaults[10]);
252     settings->percent[SPAMSETTINGS_DIGITPERCENTINDEX] = atoi(row[11] ? row[11] : defaults[11]);
253     settings->exceptlevel[SPAMSETTINGS_DIGITEXCINDEX] = atoi(row[12] ? row[12] : defaults[12]);
254     settings->join_nodes = NULL;
255     settings->lastmsg_time = 0;
256     int i;
257     for(i = 0; i < BOTNETSCAN_USERS; i++) 
258         settings->botnicks[i] = NULL;
259     chan->spam_settings = settings;
260     return 1;
261 }
262
263 void freeNeonSpamSettings(struct NeonSpamSettings *settings) {
264     struct NeonSpamJoinNode *joinnode, *nextjoinnode;
265     for(joinnode = settings->join_nodes; joinnode; joinnode = nextjoinnode) {
266         nextjoinnode = joinnode->next;
267         freeJoinNode(joinnode);
268     }
269     free(settings);
270 }
271
272 static void freeJoinNode(struct NeonSpamJoinNode *joinnode) {
273     free(joinnode->ident);
274     free(joinnode->host);
275     free(joinnode);
276 }
277
278 static struct NeonSpamJoinNode *getNeonSpamJoinNode(struct ChanUser *chanuser) {
279     struct NeonSpamJoinNode *joinnode, *prevjoinnode = NULL, *nextjoinnode, *result = NULL;
280     for(joinnode = chanuser->chan->spam_settings->join_nodes; joinnode; joinnode = nextjoinnode) {
281         nextjoinnode = joinnode->next;
282         if(!stricmp(joinnode->ident, chanuser->user->ident) && !stricmp(joinnode->host, chanuser->user->host)) {
283             prevjoinnode = joinnode;
284             result = joinnode;
285         } else if(time(0) - joinnode->last_penalty_update > MAX_JOIN_TIME) {
286             freeJoinNode(joinnode);
287             if(prevjoinnode)
288                 prevjoinnode->next = nextjoinnode;
289             else
290                 chanuser->chan->spam_settings->join_nodes = nextjoinnode;
291         } else 
292             prevjoinnode = joinnode;
293     }
294     if(result)
295         return result;
296     joinnode = malloc(sizeof(*joinnode));
297     if(!joinnode) {
298         perror("malloc() failed");
299         return NULL;
300     }
301     joinnode->ident = strdup(chanuser->user->ident);
302     joinnode->host = strdup(chanuser->user->host);
303     joinnode->last_penalty_update = time(0);
304     joinnode->joinpenalty = 0;
305     joinnode->next = chanuser->chan->spam_settings->join_nodes;
306     chanuser->chan->spam_settings->join_nodes = joinnode;
307     return joinnode;
308 }
309
310 static void createSpamNode(struct ChanUser *chanuser) {
311     struct NeonSpamNode *spamnode = malloc(sizeof(*spamnode));
312     if(!spamnode) {
313         perror("malloc() failed");
314         return;
315     }
316     spamnode->lastmsg = 0;
317     spamnode->spamcount = 0;
318     spamnode->floodpenalty = 0;
319     spamnode->last_penalty_update = time(0);
320     chanuser->spamnode = spamnode;
321 }
322
323 void init_NeonSpam() {
324     
325     set_bot_alias(BOTID, BOTALIAS);
326     start_bots();
327     
328     //register events
329     bind_bot_ready(neonspam_bot_ready);
330     bind_join(neonspam_event_join);
331     bind_chanmsg(neonspam_event_chanmsg);
332     bind_privctcp(general_event_privctcp);
333     bind_kick(neonspam_event_kick);
334     
335     set_trigger_callback(BOTID, neonspam_trigger_callback);
336     
337     register_default_language_table(msgtab);
338 }
339
340 void loop_NeonSpam() {
341     
342 }
343
344 void free_NeonSpam() {
345     
346 }
347
348 #undef BOTID
349 #undef BOTALIAS