91da9f8e0e59a86cafe32b51032bba222d2031d7
[NeonServV5.git] / cmd_neonserv_addban.c
1
2 /*
3 * argv[0]    nick|*auth|*!*@mask
4 * argv[1-*]  reason
5 */
6 static USERLIST_CALLBACK(neonserv_cmd_addban_userlist_lookup);
7 static void neonserv_cmd_addban_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event, char *mas, char *reason);
8
9 struct neonserv_cmd_addban_cache {
10     struct ClientSocket *client, *textclient;
11     struct UserNode *user;
12     struct Event *event;
13     char *mask;
14     char *reason;
15 };
16
17 static CMD_BIND(neonserv_cmd_addban) {
18     struct neonserv_cmd_addban_cache *cache = malloc(sizeof(*cache));
19     if (!cache) {
20         perror("malloc() failed");
21         return;
22     }
23     cache->client = client;
24     cache->textclient = getTextBot();
25     cache->user = user;
26     cache->event = event;
27     cache->mask = strdup(argv[0]);
28     if(argc > 1) {
29         cache->reason = strdup(merge_argv(argv, 1, argc));
30     } else
31         cache->reason = NULL;
32     get_userlist(chan, neonserv_cmd_addban_userlist_lookup, cache);
33 }
34
35 static USERLIST_CALLBACK(neonserv_cmd_addban_userlist_lookup) {
36     struct neonserv_cmd_addban_cache *cache = data;
37     neonserv_cmd_addban_async1(cache->client, cache->textclient, cache->user, chan, cache->event, cache->mask, (cache->reason ? cache->reason : "Bye."));
38     free(cache->mask);
39     if(cache->reason)
40         free(cache->reason);
41     free(cache);
42 }
43
44 static void neonserv_cmd_addban_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event, char *mask, char *reason) {
45     int match_count = 0;
46     char hostmask_buffer[NICKLEN+USERLEN+HOSTLEN+3];
47     char usermask[NICKLEN+USERLEN+HOSTLEN+3];
48     struct UserNode *cuser;
49     struct ChanUser *chanuser;
50     mask = make_banmask(mask, hostmask_buffer);
51     for(chanuser = getChannelUsers(chan, NULL); chanuser; chanuser = getChannelUsers(chan, chanuser)) {
52         cuser = chanuser->user;
53         sprintf(usermask, "%s!%s@%s", cuser->nick, cuser->ident, cuser->host);
54         if(!match(mask, usermask)) {
55             if(isNetworkService(chanuser->user)) {
56                 reply(textclient, user, "NS_SERVICE_IMMUNE", chanuser->user->nick);
57                 return;
58             }
59             if(isUserProtected(chan, cuser, user)) {
60                 reply(textclient, user, "NS_USER_PROTECTED", cuser->nick);
61                 return;
62             }
63             match_count++;
64             if(match_count > 4 && (match_count * 3) > chan->usercount && !isGodMode(user)) {
65                 reply(textclient, user, "NS_LAME_MASK", mask);
66                 return;
67             }
68         }
69     }
70     MYSQL_RES *res;
71     MYSQL_ROW row;
72     //check if the provided mask is already banned by another ban
73     char *ban = getBanAffectingMask(chan, mask);
74     if(ban != NULL) {
75         reply(textclient, user, "NS_BAN_ALREADY_ADDED", mask, chan->name);
76         return;
77     }
78     //check if the provided mask affects any existing bans
79     printf_mysql_query("SELECT `ban_mask`, `ban_id` FROM `bans` WHERE `ban_channel` = '%d'", chan->channel_id);
80     res = mysql_use();
81     while ((row = mysql_fetch_row(res)) != NULL) {
82         if(!match(mask, row[0])) {
83             //remove the ban
84             printf_mysql_query("DELETE FROM `bans` WHERE `ban_id` = '%s'", row[1]);
85         }
86     }
87     printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
88     int userid;
89     res = mysql_use();
90     if ((row = mysql_fetch_row(res)) != NULL)
91         userid = atoi(row[0]);
92     else
93         return;
94     //add the ban
95     printf_mysql_query("INSERT INTO `bans` (`ban_channel`, `ban_mask`, `ban_triggered`, `ban_owner`, `ban_reason`) VALUES ('%d', '%s', UNIX_TIMESTAMP(), '%d', '%s')", chan->channel_id, escape_string(mask), userid, escape_string(reason));
96     putsock(client, "MODE %s +b %s", chan->name, mask);
97     for(chanuser = getChannelUsers(chan, NULL); chanuser; chanuser = getChannelUsers(chan, chanuser)) {
98         cuser = chanuser->user;
99         sprintf(usermask, "%s!%s@%s", cuser->nick, cuser->ident, cuser->host);
100         if(!match(mask, usermask)) {
101             putsock(client, "KICK %s %s :%s", chan->name, cuser->nick, reason);
102         }
103     }
104     reply(textclient, user, "NS_ADDBAN_DONE", mask, chan->name, match_count);
105     logEvent(event);
106 }