src/modules/NeonServ.mod/event_neonserv_join.c

   1 /* event_neonserv_join.c - NeonServ v5.4
   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 struct neonserv_event_join_cache {
  19     struct ClientSocket *client;
  20     struct ChanUser *chanuser;
  21     int was_registering;
  22 };
  23
  24 static USERAUTH_CALLBACK(neonserv_event_join_nick_lookup);
  25 static void neonserv_event_join_async1(struct ClientSocket *client, struct ChanUser *chanuser, int was_registering);
  26 static TIMEQ_CALLBACK(neonserv_event_join_dynlimit);
  27
  28 static void neonserv_event_join(struct ChanUser *chanuser) {
  29     struct UserNode *user = chanuser->user;
  30     struct ClientSocket *client = getBotForChannel(chanuser->chan);
  31     if(!client) return; //we can't "see" this event
  32     if(chanuser->user == client->user) {
  33         requestOp(client->user, chanuser->chan);
  34         module_neonbackup_recover_chan(chanuser->chan);
  35         return;
  36     }
  37     if(chanuser->user->flags & USERFLAG_ISBOT) return;
  38     loadChannelSettings(chanuser->chan);
  39     if(!(chanuser->chan->flags & CHANFLAG_CHAN_REGISTERED)) return;
  40     char *ban;
  41     char usermask[NICKLEN+USERLEN+HOSTLEN+3];
  42     sprintf(usermask, "%s!%s@%s", user->nick, user->ident, user->host);
  43     if((ban = getBanAffectingMask(chanuser->chan, usermask)) != NULL && !(user->flags & (USERFLAG_ISBOT | USERFLAG_ISIRCOP))) {
  44         MYSQL_RES *res;
  45         MYSQL_ROW row;
  46         printf_mysql_query("SELECT `ban_reason`, `user_user` FROM `bans` LEFT JOIN `users` ON `ban_owner` = `user_id` WHERE `ban_channel` = '%d' AND `ban_mask` = '%s'", chanuser->chan->channel_id, escape_string(ban));
  47         res = mysql_use();
  48         if ((row = mysql_fetch_row(res)) != NULL) {
  49             putsock(client, "MODE %s +b %s", chanuser->chan->name, ban);
  50             putsock(client, "KICK %s %s :(%s) %s", chanuser->chan->name, chanuser->user->nick, (row[1] ? row[1] : client->user->nick), row[0]);
  51             return;
  52         }
  53     }
  54     if(!(user->flags & USERFLAG_ISAUTHED)) {
  55         struct neonserv_event_join_cache *cache = malloc(sizeof(*cache));
  56         if (!cache) {
  57             perror("malloc() failed");
  58             return;
  59         }
  60         cache->client = client;
  61         cache->chanuser = chanuser;
  62         cache->was_registering = (user->flags & USERFLAG_WAS_REGISTRING);
  63         get_userauth(user, module_id, neonserv_event_join_nick_lookup, cache);
  64     } else
  65         neonserv_event_join_async1(client, chanuser, (user->flags & USERFLAG_WAS_REGISTRING));
  66 }
  67
  68 static USERAUTH_CALLBACK(neonserv_event_join_nick_lookup) {
  69     struct neonserv_event_join_cache *cache = data;
  70     if(user) {
  71         neonserv_event_join_async1(cache->client, cache->chanuser, cache->was_registering);
  72     }
  73     free(cache);
  74 }
  75
  76 static void neonserv_event_join_async1(struct ClientSocket *client, struct ChanUser *chanuser, int was_registering) {
  77     struct ClientSocket *textclient = ((client->flags & SOCKET_FLAG_PREFERRED) ? client : get_prefered_bot(client->botid));
  78     struct ChanNode *chan = chanuser->chan;
  79     struct UserNode *user = chanuser->user;
  80     struct ModeBuffer *modeBuf;
  81     int with_halfops = get_int_field("General.have_halfop");
  82     MYSQL_RES *res;
  83     MYSQL_ROW row, chanuserrow, defaultrow = NULL;
  84     printf_mysql_query("SELECT `channel_maxusers`, `channel_greeting`, `channel_usergreeting`, `channel_getop`, `channel_getvoice`, `channel_userinfo`, `channel_dynlimit`, `channel_gethalfop` FROM `channels` WHERE `channel_id` = '%d'", chan->channel_id);
  85     res = mysql_use();
  86     if ((row = mysql_fetch_row(res)) == NULL) return;
  87     if(!row[3] || !row[4] || !row[5] || (!row[7] && with_halfops)) {
  88         printf_mysql_query("SELECT `channel_getop`, `channel_getvoice`, `channel_userinfo`, `channel_gethalfop` FROM `channels` WHERE `channel_name` = 'defaults'");
  89         res = mysql_use();
  90         defaultrow = mysql_fetch_row(res);
  91     }
  92     if(chan->usercount > atoi(row[0])) {
  93         //update maxusers
  94         printf_mysql_query("UPDATE `channels` SET `channel_maxusers` = '%d' WHERE `channel_id` = '%d'", chan->usercount, chan->channel_id);
  95     }
  96     if((user->flags & USERFLAG_ISAUTHED)) {
  97         printf_mysql_query("SELECT `chanuser_access`, `chanuser_flags`, `chanuser_infoline`, `chanuser_seen`, `chanuser_id` FROM `chanusers` LEFT JOIN `users` ON `chanuser_uid` = `user_id` WHERE `chanuser_cid` = '%d' AND `user_user` = '%s'", chan->channel_id, escape_string(user->auth));
  98         res = mysql_use();
  99         chanuserrow = mysql_fetch_row(res);
 100     } else
 101         chanuserrow = NULL;
 102     if(!chanuserrow)
 103         printf_mysql_query("UPDATE `channels` SET `channel_lastvisit` = UNIX_TIMESTAMP() WHERE `channel_id` = '%d'", chan->channel_id);
 104     int userflags = (chanuserrow ? atoi(chanuserrow[1]) : 0);
 105     int uaccess = ((chanuserrow && !(userflags & DB_CHANUSER_SUSPENDED)) ? atoi(chanuserrow[0]) : 0);
 106     //GREETING
 107     char greeting[MAXLEN];
 108     int greetingPos = 0;
 109     char *a, *b = (chanuserrow && *row[2] ? row[2] : row[1]);
 110     do {
 111         if(!b) break;
 112         a = strstr(b, "$");
 113         if(a) *a = '\0';
 114         greetingPos += sprintf(greeting + greetingPos, "%s", b);
 115         if(!a) break;
 116         switch(a[1]) {
 117             case '\0':
 118                 a = NULL;
 119                 break;
 120             case 'A':
 121                 greetingPos += sprintf(greeting + greetingPos, "%d", uaccess);
 122                 break;
 123             case 'B':
 124                 greetingPos += sprintf(greeting + greetingPos, "%s", client->user->nick);
 125                 break;
 126             case 'N':
 127                 greetingPos += sprintf(greeting + greetingPos, "%s", user->nick);
 128                 break;
 129             case 'H':
 130                 greetingPos += sprintf(greeting + greetingPos, "%s@%s", user->ident, user->host);
 131                 break;
 132             case 'U':
 133                 greetingPos += sprintf(greeting + greetingPos, "%s", ((user->flags & USERFLAG_ISAUTHED) ? user->auth : "*"));
 134                 break;
 135             default:
 136                 greeting[greetingPos++] = '$';
 137                 greeting[greetingPos++] = a[1];
 138                 break;
 139         }
 140         if(a)
 141             b = a+2;
 142     } while(a);
 143     if(greetingPos && (!was_registering || *row[2]))
 144         reply(textclient, user, "[%s] %s", chan->name, greeting);
 145     //USER RIGHTS
 146     if(!(userflags & DB_CHANUSER_NOAUTOOP)) {
 147         int getop = atoi((row[3] ? row[3] : defaultrow[0]));
 148         int gethalfop = (with_halfops ? atoi((row[7] ? row[7] : defaultrow[3])) : 0);
 149         int getvoice = atoi((row[4] ? row[4] : defaultrow[1]));
 150         modeBuf = initModeBuffer(client, chan);
 151         if(uaccess >= getop && uaccess != 0) { //we disallow auto op for all users
 152             modeBufferOp(modeBuf, user->nick);
 153         } else if(with_halfops && uaccess >= gethalfop) {
 154             modeBufferHalfop(modeBuf, user->nick);
 155         } else if(uaccess >= getvoice) {
 156             modeBufferVoice(modeBuf, user->nick);
 157         }
 158         freeModeBuffer(modeBuf);
 159     }
 160     //INFOLINE
 161     int userinfoaccess = atoi((row[5] ? row[5] : defaultrow[2]));
 162     if(chanuserrow && strcmp(chanuserrow[2], "") && uaccess > userinfoaccess) {
 163         if(!strcmp(chanuserrow[3], "0") || time(0) - atol(chanuserrow[3]) >= 30) {
 164             putsock(client, "PRIVMSG %s :[%s] %s", chan->name, user->nick, chanuserrow[2]);
 165         }
 166     }
 167     //SEEN
 168     if(chanuserrow) {
 169         printf_mysql_query("UPDATE `chanusers` SET `chanuser_seen` = UNIX_TIMESTAMP() WHERE `chanuser_id` = '%s'", chanuserrow[4]);
 170     }
 171     //DYNLIMIT
 172     if(row[6] && strcmp(row[6], "0")) {
 173         char nameBuf[CHANNELLEN + 10];
 174         sprintf(nameBuf, "dynlimit_%s", chan->name);
 175         if(!timeq_name_exists(nameBuf)) {
 176             //neonserv_event_join_dynlimit
 177             timeq_add_name(nameBuf, 30, module_id, neonserv_event_join_dynlimit, strdup(chan->name));
 178         }
 179     }
 180     //AUTOINVITE
 181     if((user->flags & USERFLAG_ISAUTHED) && (!chanuserrow || !strcmp(chanuserrow[3], "0") || time(0) - atol(chanuserrow[3]) >= 30)) {
 182         //check if it's the first channel, the user is seen by the bot (IMPORTANT: ignore other bot's channel!)
 183         char first_chan = 1;
 184         char bot_in_chan;
 185         struct ChanUser *cchanuser, *bchanuser;
 186         struct ClientSocket *bot;
 187         for(cchanuser = getUserChannels(user, NULL); cchanuser; cchanuser = getUserChannels(user, cchanuser)) {
 188             if(chanuser == cchanuser) continue; //ignore this one ;)
 189             //check if this bot is in the specific channel
 190             bot_in_chan = 0;
 191             for(bchanuser = getChannelUsers(cchanuser->chan, NULL); bchanuser; bchanuser = getChannelUsers(cchanuser->chan, bchanuser)) {
 192                 if(bchanuser->user->flags & USERFLAG_ISBOT) {
 193                     //get the botid
 194                     for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 195                         if(bot->user == bchanuser->user) {
 196                             if(bot->botid == client->botid)
 197                                 bot_in_chan = 1;
 198                             break;
 199                         }
 200                     }
 201                     if(bot_in_chan) break;
 202                 }
 203             }
 204             if(!bot_in_chan) continue;
 205             first_chan = 0;
 206             break;
 207         }
 208         if(first_chan) {
 209             //autoinvite :)
 210             defaultrow = NULL;
 211             printf_mysql_query("SELECT `chanuser_access`, `chanuser_flags`, `channel_name`, `channel_getinvite` FROM `chanusers` LEFT JOIN `channels` ON `chanuser_cid` = `channel_id` LEFT JOIN `users` ON `chanuser_uid` = `user_id` WHERE `user_user` = '%s' AND `chanuser_flags` >= '%d'", escape_string(user->auth), DB_CHANUSER_AUTOINVITE);
 212             res = mysql_use();
 213             int getinvite;
 214             while((chanuserrow = mysql_fetch_row(res)) != NULL) {
 215                 userflags = atoi(chanuserrow[1]);
 216                 if(!(userflags & DB_CHANUSER_AUTOINVITE)) continue;
 217                 if(!(chan = getChanByName(chanuserrow[2]))) continue; //no bot is in the channel
 218                 if((bchanuser = getChanUser(client->user, chan)) && (bchanuser->flags & CHANUSERFLAG_OPPED))
 219                     bot = client;
 220                 else {
 221                     bot = getBotForChannel(chan);
 222                 }
 223                 if(chanuserrow[3] == NULL && defaultrow == NULL) {
 224                     printf_mysql_query("SELECT `channel_getinvite` FROM `channels` WHERE `channel_name` = 'defaults'");
 225                     res = mysql_use();
 226                     defaultrow = mysql_fetch_row(res);
 227                 }
 228                 getinvite = atoi((chanuserrow[3] ? chanuserrow[3] : defaultrow[0]));
 229                 if(atoi(chanuserrow[0]) >= getinvite) {
 230                     putsock(bot, "INVITE %s %s", user->nick, chan->name);
 231                 }
 232             }
 233         }
 234     }
 235 }
 236
 237 static TIMEQ_CALLBACK(neonserv_event_join_dynlimit) {
 238     char *chanName = data;
 239     struct ChanNode *chan = getChanByName(chanName);
 240     free(chanName);
 241     if(!chan) return;
 242     struct ClientSocket *client = getBotForChannel(chan);
 243     if(!client) return;
 244     loadChannelSettings(chan);
 245     if(!(chan->flags & CHANFLAG_CHAN_REGISTERED)) return;
 246     MYSQL_RES *res;
 247     MYSQL_ROW row;
 248     printf_mysql_query("SELECT `channel_dynlimit` FROM `channels` WHERE `channel_id` = '%d'", chan->channel_id);
 249     res = mysql_use();
 250     if ((row = mysql_fetch_row(res)) == NULL) return;
 251     if(row[0] && strcmp(row[0], "0")) {
 252         putsock(client, "MODE %s +l %d", chan->name, chan->usercount + atoi(row[0]));
 253     }
 254 }