src/cmd_neonserv_addtimeban.c

   1 /* cmd_neonserv_addtimeban.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 "cmd_neonserv.h"
  19
  20 /*
  21 * argv[0]    nick|*auth|*!*@mask
  22 * argv[1]    time
  23 * argv[2-*]  reason
  24 */
  25 static USERLIST_CALLBACK(neonserv_cmd_addtimeban_userlist_lookup);
  26 static void neonserv_cmd_addtimeban_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event, char *mask, int duration, char *reason);
  27
  28 struct neonserv_cmd_addtimeban_cache {
  29     struct ClientSocket *client, *textclient;
  30     struct UserNode *user;
  31     struct Event *event;
  32     char *mask;
  33     int duration;
  34     char *reason;
  35 };
  36
  37 CMD_BIND(neonserv_cmd_addtimeban) {
  38     int duration = strToTime(user, argv[1]);
  39     if(duration < 30) {
  40         reply(getTextBot(), user, "NS_TIMEBAN_DURATION_TOO_SHORT", 30);
  41         return;
  42     }
  43     struct neonserv_cmd_addtimeban_cache *cache = malloc(sizeof(*cache));
  44     if (!cache) {
  45         perror("malloc() failed");
  46         return;
  47     }
  48     cache->client = client;
  49     cache->textclient = getTextBot();
  50     cache->user = user;
  51     cache->event = event;
  52     cache->mask = strdup(argv[0]);
  53     cache->duration = duration;
  54     if(argc > 2) {
  55         cache->reason = strdup(merge_argv(argv, 2, argc));
  56     } else
  57         cache->reason = NULL;
  58     get_userlist(chan, neonserv_cmd_addtimeban_userlist_lookup, cache);
  59 }
  60
  61 static USERLIST_CALLBACK(neonserv_cmd_addtimeban_userlist_lookup) {
  62     struct neonserv_cmd_addtimeban_cache *cache = data;
  63     neonserv_cmd_addtimeban_async1(cache->client, cache->textclient, cache->user, chan, cache->event, cache->mask, cache->duration, (cache->reason ? cache->reason : "Bye."));
  64     free(cache->mask);
  65     if(cache->reason)
  66         free(cache->reason);
  67     free(cache);
  68 }
  69
  70 static void neonserv_cmd_addtimeban_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event, char *mask, int duration, char *reason) {
  71     int match_count = 0;
  72     char hostmask_buffer[NICKLEN+USERLEN+HOSTLEN+3];
  73     char usermask[NICKLEN+USERLEN+HOSTLEN+3];
  74     struct UserNode *cuser;
  75     struct ChanUser *chanuser;
  76     mask = make_banmask(mask, hostmask_buffer);
  77     for(chanuser = getChannelUsers(chan, NULL); chanuser; chanuser = getChannelUsers(chan, chanuser)) {
  78         cuser = chanuser->user;
  79         sprintf(usermask, "%s!%s@%s", cuser->nick, cuser->ident, cuser->host);
  80         if(!match(mask, usermask)) {
  81             if(isNetworkService(chanuser->user)) {
  82                 reply(textclient, user, "NS_SERVICE_IMMUNE", chanuser->user->nick);
  83                 return;
  84             }
  85             if(isUserProtected(chan, cuser, user)) {
  86                 reply(textclient, user, "NS_USER_PROTECTED", cuser->nick);
  87                 return;
  88             }
  89             match_count++;
  90             if(match_count > 4 && (match_count * 3) > chan->usercount && !isGodMode(user)) {
  91                 reply(textclient, user, "NS_LAME_MASK", mask);
  92                 return;
  93             }
  94         }
  95     }
  96     MYSQL_RES *res;
  97     MYSQL_ROW row;
  98     //check if the provided mask is already banned by another ban
  99     char *ban = getBanAffectingMask(chan, mask);
 100     if(ban != NULL) {
 101         reply(textclient, user, "NS_BAN_ALREADY_ADDED", mask, chan->name);
 102         return;
 103     }
 104     //check if the provided mask affects any existing bans
 105     printf_mysql_query("SELECT `ban_mask`, `ban_id` FROM `bans` WHERE `ban_channel` = '%d'", chan->channel_id);
 106     res = mysql_use();
 107     while ((row = mysql_fetch_row(res)) != NULL) {
 108         if(!match(mask, row[0])) {
 109             //remove the ban
 110             printf_mysql_query("DELETE FROM `bans` WHERE `ban_id` = '%s'", row[1]);
 111         }
 112     }
 113     printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
 114     int userid;
 115     res = mysql_use();
 116     if ((row = mysql_fetch_row(res)) != NULL)
 117         userid = atoi(row[0]);
 118     else
 119         return;
 120     //add the ban
 121     printf_mysql_query("INSERT INTO `bans` (`ban_channel`, `ban_mask`, `ban_triggered`, `ban_timeout`, `ban_owner`, `ban_reason`) VALUES ('%d', '%s', UNIX_TIMESTAMP(), '%lu', '%d', '%s')", chan->channel_id, escape_string(mask), (unsigned long) (time(0) + duration), userid, escape_string(reason));
 122     int banid = (int) mysql_insert_id(mysql_conn);
 123     putsock(client, "MODE %s +b %s", chan->name, mask);
 124     for(chanuser = getChannelUsers(chan, NULL); chanuser; chanuser = getChannelUsers(chan, chanuser)) {
 125         cuser = chanuser->user;
 126         sprintf(usermask, "%s!%s@%s", cuser->nick, cuser->ident, cuser->host);
 127         if(!match(mask, usermask)) {
 128             putsock(client, "KICK %s %s :(%s) %s", chan->name, cuser->nick, user->nick, reason);
 129         }
 130     }
 131     char nameBuf[MAXLEN];
 132     char banidBuf[20];
 133     sprintf(nameBuf, "ban_%d", banid);
 134     sprintf(banidBuf, "%d", banid);
 135     timeq_add_name(nameBuf, duration, channel_ban_timeout, strdup(banidBuf));
 136     reply(textclient, user, "NS_TIMEBAN_DONE", mask, chan->name, timeToStr(user, duration, 2, nameBuf), match_count);
 137     logEvent(event);
 138 }