src/cmd_neonserv_addtimeban.c

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