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