src/cmd_neonserv_addban.c

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