1 /* IRCParser.c - NeonServ v5.2
2 * Copyright (C) 2011 Philipp Kreil (pk910)
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.
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.
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/>.
18 #include "IRCParser.h"
22 #include "IRCEvents.h"
23 #include "ClientSocket.h"
24 #include "WHOHandler.h"
30 struct irc_cmd *irc_commands = NULL;
31 static struct UserNode *registering_users = NULL;
32 int statistics_privmsg = 0;
33 int statistics_network_users = 0;
34 int statistics_network_channels = 0;
36 static void parse_line(struct ClientSocket *client, char *line);
37 static void register_irc_function(char *command, irc_cmd_t *func);
38 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc);
40 int parse_lines(struct ClientSocket *client, char *lines, int len) {
43 for(i = 0; i < len; i++) {
44 if(lines[i] == '\r') //just zero it out :D
46 if(lines[i] == '\n') {
48 parse_line(client, line);
56 static void parse_line(struct ClientSocket *client, char *line) {
58 char *argv[MAXNUMPARAMS];
59 printf("[recv %lu] %s\n", (unsigned long) strlen(line), line);
69 //the rest is a single parameter
70 argv[argc++] = line + 1;
74 if (argc >= MAXNUMPARAMS)
76 while (*line != ' ' && *line)
80 parse_raw(client, argv[0], argv[1], argv+2, argc-2);
84 static void register_irc_function(char *command, irc_cmd_t *func) {
85 struct irc_cmd *irc_cmd = malloc(sizeof(*irc_cmd));
88 perror("malloc() failed");
91 irc_cmd->cmd = command;
93 irc_cmd->next = irc_commands;
94 irc_commands = irc_cmd;
97 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc) {
98 struct irc_cmd *irc_cmd;
100 for(irc_cmd = irc_commands; irc_cmd; irc_cmd = irc_cmd->next) {
101 if(!stricmp(irc_cmd->cmd, cmd)) {
102 ret = irc_cmd->func(client, from, argv, argc);
107 event_raw(client, from, cmd, argv, argc);
109 fprintf(stderr,"PARSE ERROR: %s\n", cmd);
113 static USERLIST_CALLBACK(got_channel_userlist) {
114 struct ChanUser *chanuser = data;
115 event_join(chanuser);
118 static IRC_CMD(raw_001) {
119 struct UserNode *user = getUserByNick(argv[0]);
121 user = addUser(argv[0]);
123 client->user->flags |= USERFLAG_ISBOT;
124 client->flags |= SOCKET_FLAG_READY;
125 event_bot_ready(client);
129 static int is_firstBotSeeUser(struct ClientSocket *client, struct UserNode *user) {
130 struct ClientSocket *bot, *pref_bot = NULL, *unpref_bot = NULL;
131 struct ChanUser *chanuser;
133 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
135 for(chanuser = getUserChannels(bot->user, NULL); chanuser; chanuser = getUserChannels(bot->user, chanuser)) {
136 if(isUserOnChan(user, chanuser->chan)) {
142 if(bot->flags & SOCKET_FLAG_PREFERRED) {
148 bot = (pref_bot ? pref_bot : unpref_bot);
155 static int is_onlyBotSeeUser(struct ClientSocket *client, struct UserNode *user) {
156 struct ClientSocket *bot;
157 struct ChanUser *chanuser;
158 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
159 if(bot == client) continue;
160 for(chanuser = getUserChannels(bot->user, NULL); chanuser; chanuser = getUserChannels(bot->user, chanuser)) {
161 if(isUserOnChan(user, chanuser->chan)) {
169 static IRC_CMD(raw_join) {
170 if(from == NULL || argc < 1) return 0;
171 struct UserNode *user = getUserByMask(from);
172 struct ChanNode *chan = getChanByName(argv[0]);
173 if(!chan && !(user->flags & USERFLAG_ISBOT)) return 0;
174 if(chan && chan->chanbot != client->user) return 1; //we ignore it - but it's not a parse error
176 user = addUserMask(from);
178 struct UserNode *registering, *last_registering = NULL, *next_registering;
179 int noEvent = 0, wasRegistering = 0;
180 for(registering = registering_users; registering; registering = next_registering) {
181 next_registering = registering->next;
182 if(!strcmp(registering->nick, user->nick)) {
183 noEvent = event_registered(registering, user);
186 last_registering->next = registering->next;
188 registering_users = registering->next;
189 delUser(registering, 1);
190 } else if(time(0) - registering->created > 2) {
192 last_registering->next = registering->next;
194 registering_users = registering->next;
195 delUser(registering, 1);
197 last_registering = registering;
200 chan = addChannel(argv[0]);
201 //request member list
202 chan->chanbot = user;
203 struct ChanUser *chanuser = addChanUser(chan, user); //it must be a bot
204 get_userlist(chan, got_channel_userlist, chanuser);
205 putsock(client, "MODE %s", chan->name);
206 putsock(client, "MODE %s +b", chan->name);
207 } else if(!isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
208 struct ChanUser *chanuser = addChanUser(chan, user);
211 user->flags |= USERFLAG_WAS_REGISTRING;
212 event_join(chanuser);
214 user->flags &= ~USERFLAG_WAS_REGISTRING;
216 if(!(user->flags & USERFLAG_ISBOT) && (chan->flags & CHANFLAG_REJOINING)) {
217 //ABORT REJOIN (security break)
218 struct ClientSocket **clients = chan->rejoin_array;
220 putsock(*clients, "JOIN %s", chan->name);
223 free(chan->rejoin_array);
224 chan->flags &= ~CHANFLAG_REJOINING;
226 } else if(chan->usercount == 1 && isUserOnChan(user, chan)) {
228 struct ChanUser *chanuser = getChanUser(user, chan);
229 chanuser->flags &= ~CHANUSERFLAG_VOICED;
230 chanuser->flags |= CHANUSERFLAG_OPPED;
235 static void check_full_rejoin(struct ChanNode *chan) {
236 struct ChanUser *chanuser;
239 for(chanuser = getChannelUsers(chan, NULL); chanuser; chanuser = getChannelUsers(chan, chanuser)) {
240 if((chanuser->flags & CHANUSERFLAG_OPPED) || !(chanuser->user->flags & USERFLAG_ISBOT)) {
244 if((chanuser->user->flags & USERFLAG_ISBOT))
248 struct ClientSocket **clients = calloc(botcount, sizeof(*clients));
249 struct ClientSocket *bot, *chanbot;
251 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
252 if(bot->user != chan->chanbot && isUserOnChan(bot->user, chan)) {
254 putsock(bot, "PART %s :rejoining", chan->name);
255 } else if(bot->user == chan->chanbot)
258 chan->flags |= CHANFLAG_REJOINING;
259 chan->rejoin_array = clients;
262 putsock(chanbot, "PART %s :magic hop", chan->name);
263 putsock(chanbot, "JOIN %s", chan->name);
268 static IRC_CMD(raw_part) {
269 if(from == NULL || argc < 1) return 0;
270 struct UserNode *user = getUserByMask(from);
271 if(user == NULL) return 0;
272 struct ChanNode *chan = getChanByName(argv[0]);
273 if(chan == NULL) return 0;
274 if(chan->chanbot != client->user) return 1; //we ignore it - but it's not a parse error
275 int keep_channel = 1;
276 if(chan->chanbot == user && (chan->flags & CHANFLAG_REJOINING)) {
277 struct ClientSocket **clients = chan->rejoin_array;
279 putsock(*clients, "JOIN %s", chan->name);
282 free(chan->rejoin_array);
283 chan->flags &= ~CHANFLAG_REJOINING;
285 } else if(isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
286 struct ChanUser *chanuser = getChanUser(user, chan);
287 delChanUser(chanuser, 0); //we need to free the chanuser manually!
288 event_part(chanuser, (argc > 1 ? argv[1] : NULL));
289 freeChanUser(chanuser);
290 if(chan->chanbot == user) {
291 //check if theres another bot in the channel - otherwise free it
292 keep_channel = checkChannelVisibility(chan);
295 if(user->channel == NULL && !(user->flags & USERFLAG_ISBOT)) {
299 if(keep_channel && (chan->flags & CHANFLAG_RECEIVED_USERLIST) && !(chan->flags & CHANFLAG_REJOINING)) {
300 check_full_rejoin(chan);
302 else if(keep_channel && (chan->flags & CHANFLAG_REJOINING) && chan->usercount == 1) {
303 //bot is alone... rejoin!
304 putsock(client, "PART %s :magic hop", chan->name);
305 putsock(client, "JOIN %s", chan->name);
310 static IRC_CMD(raw_quit) {
311 if(from == NULL || argc < 1) return 0;
312 struct UserNode *user = getUserByMask(from);
313 if(user == NULL) return 0;
314 if(!is_firstBotSeeUser(client, user)) return 1; //we ignore it - but it's not a parse error
315 int keep_channel = 1;
316 int registering = !stricmp(argv[0], "Registered");
317 if((registering && (user->flags & USERFLAG_ISBOT))) return 1; //bot is registering - just ignore it
318 delUser(user, 0); //a little bit crazy, but we want to delete the user on the channel's userlists - but not the users channel list
319 event_quit(user, argv[0]);
320 if(user->flags & USERFLAG_ISBOT) {
321 //check if there are other bots in the users channel - otherwise free them
322 struct ChanUser *chanuser, *next;
323 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
324 next = getUserChannels(user, chanuser);
325 if(chanuser->chan->chanbot == user)
326 keep_channel = checkChannelVisibility(chanuser->chan);
328 //search the user representing the bot in the world of IRC
329 struct ClientSocket *bot;
330 for(bot = getBots(0, NULL); bot; bot = getBots(0, bot)) {
331 if(bot->user == user) {
336 } else if(!registering) {
337 struct ChanUser *chanuser;
338 struct ChanNode *chan;
339 for(chanuser = user->channel; chanuser; chanuser = chanuser->next_chan) {
340 chan = chanuser->chan;
341 if((chan->flags & CHANFLAG_RECEIVED_USERLIST) && !(chan->flags & CHANFLAG_REJOINING))
342 check_full_rejoin(chan);
345 if(registering && !(user->flags & USERFLAG_ISBOT)) {
346 user->next = registering_users;
347 user->created = time(0);
348 registering_users = user;
350 delUser(user, 1); //now we fully free the user
354 void bot_disconnect(struct ClientSocket *client) {
355 struct UserNode *user = client->user;
356 struct ChanUser *chanuser, *next;
358 if(is_onlyBotSeeUser(client, user)) {
359 //ok the bot-user is not seen by any other bots so we can simply free it.
361 event_quit(user, "disconnected");
362 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
363 next = getUserChannels(user, chanuser);
364 if(chanuser->chan->chanbot == user)
365 checkChannelVisibility(chanuser->chan);
366 freeChanUser(chanuser);
368 user->channel = NULL;
369 delUser(user, 1); //now we fully free the user
371 //we need to transform the bot-user back to a normal user (BNC FIX)
372 user->flags &= ~USERFLAG_ISBOT;
373 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
374 next = getUserChannels(user, chanuser);
375 if(chanuser->chan->chanbot == user)
376 checkChannelVisibility(chanuser->chan);
383 static IRC_CMD(raw_kick) {
384 if(from == NULL || argc < 3) return 0;
385 struct UserNode *user = getUserByMask(from);
386 struct UserNode *target = getUserByNick(argv[1]);
387 struct ChanNode *chan = getChanByName(argv[0]);
388 if(chan == NULL || target == NULL) return 0;
389 if(chan->chanbot != client->user) return 1; //we ignore it - but it's not a parse error
390 int keep_channel = 1;
391 if(isUserOnChan(target, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
393 user = createTempUser(from);
394 user->flags |= USERFLAG_ISTMPUSER;
396 struct ChanUser *chanuser = getChanUser(target, chan);
397 delChanUser(chanuser, 0); //we need to free the chanuser manually!
398 event_kick(user, chanuser, argv[1]);
399 if(chanuser->chan->chanbot == user) {
400 //check if theres another bot in the channel - otherwise free it
401 keep_channel = checkChannelVisibility(chan);
403 freeChanUser(chanuser);
405 if(target->channel == NULL && !(target->flags & USERFLAG_ISBOT)) {
409 if(keep_channel && (chan->flags & CHANFLAG_RECEIVED_USERLIST) && !(chan->flags & CHANFLAG_REJOINING)) {
410 check_full_rejoin(chan);
415 static IRC_CMD(raw_topic) {
416 if(from == NULL || argc < 2) return 0;
417 struct UserNode *user = getUserByMask(from);
418 struct ChanNode *chan = getChanByName(argv[0]);
419 if(chan == NULL) return 0;
420 if(chan->chanbot != client->user) return 1; //just ignore it to prevent event duplicates
422 user = createTempUser(from);
423 user->flags |= USERFLAG_ISTMPUSER;
425 event_topic(user, chan, argv[1]);
426 strcpy(chan->topic, argv[1]);
430 static IRC_CMD(raw_privmsg) {
431 if(from == NULL || argc < 2) return 0;
432 struct UserNode *user = getUserByMask(from);
434 user = createTempUser(from);
435 user->flags |= USERFLAG_ISTMPUSER;
437 if(argv[0][0] == '#') { //Channel message
438 struct ChanNode *chan = getChanByName(argv[0]);
439 if(chan && chan->chanbot == client->user) {
440 if(statistics_enabled)
441 statistics_privmsg++;
442 if(argv[1][0] == '\001') {
443 char *cmd = &argv[1][1];
444 char *text = strstr(cmd, " ");
448 if(strlen(text) && text[strlen(text)-1] == '\001')
449 text[strlen(text)-1] = '\0';
450 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
451 cmd[strlen(cmd)-1] = '\0';
452 event_chanctcp(user, chan, cmd, text);
454 event_chanmsg(user, chan, argv[1]);
457 struct UserNode *target = getUserByNick(argv[0]);
459 if(argv[1][0] == '\001') {
460 char *cmd = &argv[1][1];
461 char *text = strstr(cmd, " ");
465 if(strlen(text) && text[strlen(text)-1] == '\001')
466 text[strlen(text)-1] = '\0';
467 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
468 cmd[strlen(cmd)-1] = '\0';
469 event_privctcp(user, target, cmd, text);
471 event_privmsg(user, target, argv[1]);
477 static IRC_CMD(raw_notice) {
478 if(from == NULL || argc < 2) return 0;
479 struct UserNode *user = getUserByMask(from);
481 user = createTempUser(from);
482 user->flags |= USERFLAG_ISTMPUSER;
484 if(argv[0][0] == '#') { //Channel notice
485 struct ChanNode *chan = getChanByName(argv[0]);
486 if(chan && chan->chanbot == client->user)
487 event_channotice(user, chan, argv[1]);
489 struct UserNode *target = getUserByNick(argv[0]);
491 event_privnotice(user, target, argv[1]);
496 static IRC_CMD(raw_nick) {
497 if(from == NULL || argc == 0) return 0;
498 struct UserNode *user = getUserByMask(from);
499 if(user == NULL) return 0;
500 if(!is_firstBotSeeUser(client, user)) return 1; //we ignore it - but it's not a parse error
501 event_nick(user, argv[0]);
502 renameUser(user, argv[0]);
506 static IRC_CMD(raw_ping) {
507 if(argc == 0) return 0;
508 putsock(client, "PONG :%s", argv[0]);
512 static IRC_CMD(raw_354) {
513 recv_whohandler_354(client, argv, argc);
517 static IRC_CMD(raw_315) {
518 recv_whohandler_315(client, argv, argc);
522 static IRC_CMD(raw_324) { //MODE LIST
523 //Watchcat #pktest +stnzN
524 if(from == NULL || argc < 3) return 0;
525 struct ChanNode *chan = getChanByName(argv[1]);
526 if(chan == NULL) return 0;
527 parseModes(chan->modes, argv[2], argv+3, argc-3);
531 static IRC_CMD(raw_invite) {
532 if(from == NULL || argc < 2) return 0;
533 struct UserNode *user = getUserByMask(from);
535 user = createTempUser(from);
536 user->flags |= USERFLAG_ISTMPUSER;
538 event_invite(client, user, argv[1]);
542 static IRC_CMD(raw_mode) {
543 if(from == NULL || argc < 2) return 0;
544 struct UserNode *user = getUserByMask(from);
546 user = createTempUser(from);
547 user->flags |= USERFLAG_ISTMPUSER;
549 if(argv[0][0] == '#') {
551 struct ChanNode *chan = getChanByName(argv[0]);
553 if(chan->chanbot != client->user) return 1;
554 event_mode(user, chan, argv[1], argv+2, argc-2);
555 parseModes(chan->modes, argv[1], argv+2, argc-2);
562 static IRC_CMD(raw_367) {
563 //Watchcat #pktest pk911!*@* TestBot!~bot@pktest.user.WebGamesNet 1315863279
564 struct ChanNode *chan = getChanByName(argv[1]);
567 while((ban = getMatchingChannelBan(chan, argv[2]))) {
568 removeChannelBan(ban);
570 addChannelBan(chan, argv[2]);
574 static IRC_CMD(raw_251) {
575 if(argc < 2) return 0;
576 char *total_user_str = argv[1];
577 char total_visible[20], total_invisible[20];
578 int i, total_visible_pos = 0, total_invisible_pos = 0;
579 while(*total_user_str) {
580 if(*total_user_str == ' ') {
583 if(total_visible_pos < 20)
584 total_visible[total_visible_pos++] = *total_user_str;
586 if(total_invisible_pos < 20)
587 total_invisible[total_invisible_pos++] = *total_user_str;
591 total_visible[total_visible_pos] = '\0';
592 total_invisible[total_invisible_pos] = '\0';
593 statistics_network_users = atoi(total_visible) + atoi(total_invisible);
597 static IRC_CMD(raw_254) {
598 if(argc < 3) return 0;
599 statistics_network_channels = atoi(argv[1]);
604 static IRC_CMD(raw_332) {
605 //Skynet #neonserv :topic
606 struct ChanNode *chan = getChanByName(argv[1]);
608 strcpy(chan->topic, argv[2]);
613 //all the raws we receive...
614 register_irc_function("001", raw_001);
615 register_irc_function("251", raw_251);
616 register_irc_function("254", raw_254);
617 register_irc_function("324", raw_324);
618 register_irc_function("332", raw_332);
619 register_irc_function("367", raw_367);
620 register_irc_function("INVITE", raw_invite);
621 register_irc_function("NOTICE", raw_notice);
622 register_irc_function("TOPIC", raw_topic);
623 register_irc_function("KICK", raw_kick);
624 register_irc_function("PART", raw_part);
625 register_irc_function("QUIT", raw_quit);
626 register_irc_function("JOIN", raw_join);
627 register_irc_function("MODE", raw_mode);
628 register_irc_function("NICK", raw_nick);
629 register_irc_function("354", raw_354);
630 register_irc_function("315", raw_315);
631 register_irc_function("PING", raw_ping);
632 register_irc_function("PRIVMSG", raw_privmsg);
636 struct irc_cmd *cmd, *next;
637 for(cmd = irc_commands; cmd; cmd = next) {
643 void reply(struct ClientSocket *client, struct UserNode *user, const char *text, ...) {
644 const char *reply_format = get_language_string(user, text);
645 if(reply_format == NULL)
647 loadUserSettings(user);
648 char formatBuf[MAXLEN];
649 sprintf(formatBuf, "%s %s :%s", ((user->flags & USERFLAG_REPLY_PRIVMSG) ? "PRIVMSG" : "NOTICE"), user->nick, reply_format);
651 char sendBuf[MAXLEN];
653 if (!(client->flags & SOCKET_FLAG_CONNECTED)) return;
655 va_start(arg_list, text);
656 pos = vsnprintf(sendBuf, MAXLEN - 2, formatBuf, arg_list);
658 if (pos < 0 || pos > (MAXLEN - 2)) pos = MAXLEN - 2;
660 sendBuf[pos+1] = '\0';
661 write_socket(client, sendBuf, pos+1);
664 char* merge_argv(char **argv, int start, int end) {
665 return merge_argv_char(argv, start, end, ' ');
668 char* merge_argv_char(char **argv, int start, int end, char seperator) {
671 for(i = start; i < end; i++) {