1 /* IRCParser.c - NeonServ v5.3
2 * Copyright (C) 2011-2012 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"
31 struct irc_cmd *irc_commands = NULL;
32 static struct UserNode *registering_users = NULL;
33 int statistics_privmsg = 0;
34 int statistics_network_users = 0;
35 int statistics_network_channels = 0;
37 static void parse_line(struct ClientSocket *client, char *line);
38 static void register_irc_function(char *command, irc_cmd_t *func);
39 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc);
41 int parse_lines(struct ClientSocket *client, char *lines, int len) {
44 for(i = 0; i < len; i++) {
45 if(lines[i] == '\r') //just zero it out :D
47 if(lines[i] == '\n') {
49 parse_line(client, line);
57 static void parse_line(struct ClientSocket *client, char *line) {
59 char *argv[MAXNUMPARAMS];
61 putlog(LOGLEVEL_RAW, "[%d recv %lu] %s\n", getCurrentThreadID(), (unsigned long) strlen(line), line);
63 putlog(LOGLEVEL_RAW, "[recv %lu] %s\n", (unsigned long) strlen(line), line);
74 //the rest is a single parameter
75 argv[argc++] = line + 1;
79 if (argc >= MAXNUMPARAMS)
81 while (*line != ' ' && *line)
85 parse_raw(client, argv[0], argv[1], argv+2, argc-2);
89 static void register_irc_function(char *command, irc_cmd_t *func) {
90 struct irc_cmd *irc_cmd = malloc(sizeof(*irc_cmd));
93 perror("malloc() failed");
96 irc_cmd->cmd = command;
98 irc_cmd->next = irc_commands;
99 irc_commands = irc_cmd;
102 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc) {
103 struct irc_cmd *irc_cmd;
105 for(irc_cmd = irc_commands; irc_cmd; irc_cmd = irc_cmd->next) {
106 if(!stricmp(irc_cmd->cmd, cmd)) {
107 ret = irc_cmd->func(client, from, argv, argc);
112 event_raw(client, from, cmd, argv, argc);
114 fprintf(stderr,"PARSE ERROR: %s\n", cmd);
118 static USERLIST_CALLBACK(got_channel_userlist) {
119 struct ChanUser *chanuser = data;
120 event_join(chanuser);
123 static IRC_CMD(raw_002) { //fixed: ZNC fakes a 001 raw even if we're not connected!
124 struct UserNode *user = getUserByNick(argv[0]);
126 user = addUser(argv[0]);
128 client->user->flags |= USERFLAG_ISBOT;
129 client->flags |= SOCKET_FLAG_READY;
130 event_bot_ready(client);
134 static int is_firstBotSeeUser(struct ClientSocket *client, struct UserNode *user) {
135 struct ClientSocket *bot, *pref_bot = NULL, *unpref_bot = NULL;
136 struct ChanUser *chanuser;
138 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
140 for(chanuser = getUserChannels(bot->user, NULL); chanuser; chanuser = getUserChannels(bot->user, chanuser)) {
141 if(isUserOnChan(user, chanuser->chan)) {
147 if(bot->flags & SOCKET_FLAG_PREFERRED) {
153 bot = (pref_bot ? pref_bot : unpref_bot);
160 static int is_onlyBotSeeUser(struct ClientSocket *client, struct UserNode *user) {
161 struct ClientSocket *bot;
162 struct ChanUser *chanuser;
163 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
164 if(bot == client) continue;
165 for(chanuser = getUserChannels(bot->user, NULL); chanuser; chanuser = getUserChannels(bot->user, chanuser)) {
166 if(isUserOnChan(user, chanuser->chan)) {
174 static IRC_CMD(raw_join) {
175 if(from == NULL || argc < 1) return 0;
176 SYNCHRONIZE(cache_sync);
177 struct UserNode *user = getUserByMask(from);
178 struct ChanNode *chan = getChanByName(argv[0]);
179 if(!chan && (!user || !(user->flags & USERFLAG_ISBOT))) {
180 DESYNCHRONIZE(cache_sync);
183 if(chan && (((!user || !isBot(user)) && chan->chanbot != client->user) || ((user && isBot(user)) && client->user != user))) {
184 DESYNCHRONIZE(cache_sync);
185 return 1; //we ignore it - but it's not a parse error
187 //let Bots always add themselves! (maybe they join invisible)
189 user = addUserMask(from);
191 struct UserNode *registering, *last_registering = NULL, *next_registering;
192 int noEvent = 0, wasRegistering = 0;
193 for(registering = registering_users; registering; registering = next_registering) {
194 next_registering = registering->next;
195 if(!strcmp(registering->nick, user->nick)) {
196 noEvent = event_registered(registering, user);
199 last_registering->next = registering->next;
201 registering_users = registering->next;
202 delUser(registering, 1);
203 } else if(time(0) - registering->created > 2) {
205 last_registering->next = registering->next;
207 registering_users = registering->next;
208 delUser(registering, 1);
210 last_registering = registering;
213 chan = addChannel(argv[0]);
214 //request member list
215 chan->chanbot = user;
216 struct ChanUser *chanuser = addChanUser(chan, user); //it must be a bot
217 get_userlist_with_invisible(chan, got_channel_userlist, chanuser);
218 putsock(client, "MODE %s", chan->name);
219 putsock(client, "MODE %s +b", chan->name);
220 } else if(!isUserOnChan(user, chan) && ((chan->flags & CHANFLAG_RECEIVED_USERLIST) || isBot(user))) {
221 struct ChanUser *chanuser = addChanUser(chan, user);
222 if(isBot(user) && isModeSet(chan->modes, 'D')) //if the bot joins a channel it could also be invisible
223 chanuser->flags |= CHANUSERFLAG_INVISIBLE;
226 user->flags |= USERFLAG_WAS_REGISTRING;
227 event_join(chanuser);
229 user->flags &= ~USERFLAG_WAS_REGISTRING;
231 if(!(user->flags & USERFLAG_ISBOT) && (chan->flags & CHANFLAG_REJOINING)) {
232 //ABORT REJOIN (security break)
233 struct ClientSocket **clients = chan->rejoin_array;
235 putsock(*clients, "JOIN %s", chan->name);
238 free(chan->rejoin_array);
239 chan->flags &= ~CHANFLAG_REJOINING;
241 } else if(chan->usercount == 1 && isUserOnChan(user, chan)) {
243 struct ChanUser *chanuser = getChanUser(user, chan);
244 chanuser->flags &= ~CHANUSERFLAG_VOICED;
245 chanuser->flags |= CHANUSERFLAG_OPPED;
247 DESYNCHRONIZE(cache_sync);
251 static void check_full_rejoin(struct ChanNode *chan) {
252 struct ChanUser *chanuser;
255 for(chanuser = getChannelUsers(chan, NULL); chanuser; chanuser = getChannelUsers(chan, chanuser)) {
256 if((chanuser->flags & CHANUSERFLAG_OPPED) || !(chanuser->user->flags & USERFLAG_ISBOT)) {
260 if((chanuser->user->flags & USERFLAG_ISBOT))
264 struct ClientSocket **clients = calloc(botcount, sizeof(*clients));
265 struct ClientSocket *bot, *chanbot;
267 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
268 if(bot->user != chan->chanbot && isUserOnChan(bot->user, chan)) {
270 putsock(bot, "PART %s :rejoining", chan->name);
271 } else if(bot->user == chan->chanbot)
274 chan->flags |= CHANFLAG_REJOINING;
275 chan->rejoin_array = clients;
278 putsock(chanbot, "PART %s :magic hop", chan->name);
279 putsock(chanbot, "JOIN %s", chan->name);
284 static IRC_CMD(raw_part) {
285 if(from == NULL || argc < 1) return 0;
286 SYNCHRONIZE(cache_sync);
287 struct UserNode *user = getUserByMask(from);
289 DESYNCHRONIZE(cache_sync);
292 struct ChanNode *chan = getChanByName(argv[0]);
294 DESYNCHRONIZE(cache_sync);
297 if((!isBot(user) && chan->chanbot != client->user) || (isBot(user) && client->user != user)) {
298 DESYNCHRONIZE(cache_sync);
299 return 1; //we ignore it - but it's not a parse error
301 int keep_channel = 1;
302 if(chan->chanbot == user && (chan->flags & CHANFLAG_REJOINING)) {
303 struct ClientSocket **clients = chan->rejoin_array;
305 putsock(*clients, "JOIN %s", chan->name);
308 free(chan->rejoin_array);
309 chan->flags &= ~CHANFLAG_REJOINING;
310 DESYNCHRONIZE(cache_sync);
312 } else if(isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
313 struct ChanUser *chanuser = getChanUser(user, chan);
314 delChanUser(chanuser, 0); //we need to free the chanuser manually!
315 event_part(chanuser, (argc > 1 ? argv[1] : NULL));
316 freeChanUser(chanuser);
317 if(chan->chanbot == user) {
318 //check if theres another bot in the channel - otherwise free it
319 keep_channel = checkChannelVisibility(chan);
322 if(user->channel == NULL && !(user->flags & USERFLAG_ISBOT)) {
326 if(keep_channel && (chan->flags & CHANFLAG_RECEIVED_USERLIST) && !(chan->flags & CHANFLAG_REJOINING)) {
327 check_full_rejoin(chan);
329 else if(keep_channel && (chan->flags & CHANFLAG_REJOINING) && chan->usercount == 1) {
330 //bot is alone... rejoin!
331 putsock(client, "PART %s :magic hop", chan->name);
332 putsock(client, "JOIN %s", chan->name);
334 DESYNCHRONIZE(cache_sync);
338 static IRC_CMD(raw_quit) {
339 if(from == NULL || argc < 1) return 0;
340 SYNCHRONIZE(cache_sync);
341 struct UserNode *user = getUserByMask(from);
343 DESYNCHRONIZE(cache_sync);
346 if(!is_firstBotSeeUser(client, user)) {
347 DESYNCHRONIZE(cache_sync);
348 return 1; //we ignore it - but it's not a parse error
350 int registering = !stricmp(argv[0], "Registered");
351 if((registering && (user->flags & USERFLAG_ISBOT))) {
352 DESYNCHRONIZE(cache_sync);
353 return 1; //bot is registering - just ignore it
355 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
356 event_quit(user, argv[0]);
357 if(user->flags & USERFLAG_ISBOT) {
358 //check if there are other bots in the users channel - otherwise free them
359 struct ChanUser *chanuser, *next;
360 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
361 next = getUserChannels(user, chanuser);
362 if(chanuser->chan->chanbot == user)
363 checkChannelVisibility(chanuser->chan);
365 //search the user representing the bot in the world of IRC
366 struct ClientSocket *bot;
367 for(bot = getBots(0, NULL); bot; bot = getBots(0, bot)) {
368 if(bot->user == user) {
369 bot->flags &= ~SOCKET_FLAG_READY;
374 } else if(!registering) {
375 struct ChanUser *chanuser;
376 struct ChanNode *chan;
377 for(chanuser = user->channel; chanuser; chanuser = chanuser->next_chan) {
378 chan = chanuser->chan;
379 if((chan->flags & CHANFLAG_RECEIVED_USERLIST) && !(chan->flags & CHANFLAG_REJOINING))
380 check_full_rejoin(chan);
383 if(registering && !(user->flags & USERFLAG_ISBOT)) {
384 user->next = registering_users;
385 user->created = time(0);
386 registering_users = user;
388 delUser(user, 1); //now we fully free the user
389 DESYNCHRONIZE(cache_sync);
393 void bot_disconnect(struct ClientSocket *client) {
394 struct UserNode *user = client->user;
395 struct ChanUser *chanuser, *next;
397 if(is_onlyBotSeeUser(client, user)) {
398 //ok the bot-user is not seen by any other bots so we can simply free it.
400 event_quit(user, "disconnected");
401 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
402 next = getUserChannels(user, chanuser);
403 if(chanuser->chan->chanbot == user)
404 checkChannelVisibility(chanuser->chan);
405 freeChanUser(chanuser);
407 user->channel = NULL;
408 delUser(user, 1); //now we fully free the user
410 //we need to transform the bot-user back to a normal user (BNC FIX)
411 user->flags &= ~USERFLAG_ISBOT;
412 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
413 next = getUserChannels(user, chanuser);
414 if(chanuser->chan->chanbot == user)
415 checkChannelVisibility(chanuser->chan);
419 client->flags &= ~SOCKET_FLAG_READY;
423 static IRC_CMD(raw_kick) {
424 if(from == NULL || argc < 3) return 0;
425 SYNCHRONIZE(cache_sync);
426 struct UserNode *user = getUserByMask(from);
427 struct UserNode *target = getUserByNick(argv[1]);
428 struct ChanNode *chan = getChanByName(argv[0]);
429 if(chan == NULL || target == NULL) {
430 DESYNCHRONIZE(cache_sync);
433 if(((!isBot(target) && chan->chanbot != client->user) || (isBot(target) && client->user != target))) {
434 DESYNCHRONIZE(cache_sync);
435 return 1; //we ignore it - but it's not a parse error
437 int keep_channel = 1;
438 if(isUserOnChan(target, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
440 user = createTempUserMask(from);
442 DESYNCHRONIZE(cache_sync);
445 user->flags |= USERFLAG_ISTMPUSER;
447 struct ChanUser *chanuser = getChanUser(target, chan);
448 delChanUser(chanuser, 0); //we need to free the chanuser manually!
449 event_kick(user, chanuser, argv[1]);
450 if(chanuser->chan->chanbot == user) {
451 //check if theres another bot in the channel - otherwise free it
452 keep_channel = checkChannelVisibility(chan);
454 freeChanUser(chanuser);
456 if(target->channel == NULL && !(target->flags & USERFLAG_ISBOT)) {
460 if(keep_channel && (chan->flags & CHANFLAG_RECEIVED_USERLIST) && !(chan->flags & CHANFLAG_REJOINING)) {
461 check_full_rejoin(chan);
463 DESYNCHRONIZE(cache_sync);
467 static IRC_CMD(raw_topic) {
468 if(from == NULL || argc < 2) return 0;
469 struct UserNode *user = getUserByMask(from);
470 struct ChanNode *chan = getChanByName(argv[0]);
471 if(chan == NULL) return 0;
472 if(chan->chanbot != client->user) return 1; //just ignore it to prevent event duplicates
474 user = createTempUserMask(from);
476 user->flags |= USERFLAG_ISTMPUSER;
478 event_topic(user, chan, argv[1]);
479 strcpy(chan->topic, argv[1]);
483 static IRC_CMD(raw_privmsg) {
484 if(from == NULL || argc < 2) return 0;
485 if(!stricmplen(from, "*status", 7) || !stricmplen(from, "-sBNC", 5)) {
487 unsigned int tid = (unsigned int) pthread_self_tid();
488 while(!clientsocket_parseorder_top(tid)) {
492 if(!match("Disconnected from IRC.*", argv[1])) {
494 bot_disconnect(client);
497 if(!match("* disconnected from the server.", argv[1])) {
499 bot_disconnect(client);
503 struct UserNode *user = getUserByMask(from);
505 user = createTempUserMask(from);
507 user->flags |= USERFLAG_ISTMPUSER;
509 if(argv[0][0] == '#') { //Channel message
510 struct ChanNode *chan = getChanByName(argv[0]);
511 if(chan && chan->chanbot == client->user) {
512 if(statistics_enabled)
513 statistics_privmsg++;
514 if(argv[1][0] == '\001') {
515 char *cmd = &argv[1][1];
516 char *text = strstr(cmd, " ");
520 if(strlen(text) && text[strlen(text)-1] == '\001')
521 text[strlen(text)-1] = '\0';
522 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
523 cmd[strlen(cmd)-1] = '\0';
524 event_chanctcp(user, chan, cmd, text);
526 event_chanmsg(user, chan, argv[1]);
529 struct UserNode *target = getUserByNick(argv[0]);
531 if(argv[1][0] == '\001') {
532 char *cmd = &argv[1][1];
533 char *text = strstr(cmd, " ");
537 if(strlen(text) && text[strlen(text)-1] == '\001')
538 text[strlen(text)-1] = '\0';
539 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
540 cmd[strlen(cmd)-1] = '\0';
541 event_privctcp(user, target, cmd, text);
543 event_privmsg(user, target, argv[1]);
549 static IRC_CMD(raw_notice) {
550 if(from == NULL || argc < 2) return 0;
551 struct UserNode *user = getUserByMask(from);
553 user = createTempUserMask(from);
555 user->flags |= USERFLAG_ISTMPUSER;
557 if(argv[0][0] == '#') { //Channel notice
558 struct ChanNode *chan = getChanByName(argv[0]);
559 if(chan && chan->chanbot == client->user)
560 event_channotice(user, chan, argv[1]);
562 struct UserNode *target = getUserByNick(argv[0]);
564 event_privnotice(user, target, argv[1]);
569 static IRC_CMD(raw_nick) {
570 if(from == NULL || argc == 0) return 0;
571 struct UserNode *user = getUserByMask(from);
572 if(user == NULL) return 0;
573 if(!is_firstBotSeeUser(client, user)) return 1; //we ignore it - but it's not a parse error
574 event_nick(user, argv[0]);
575 renameUser(user, argv[0]);
579 static IRC_CMD(raw_ping) {
580 if(argc == 0) return 0;
581 putsock(client, "PONG :%s", argv[0]);
585 static IRC_CMD(raw_354) {
586 recv_whohandler_354(client, argv, argc);
590 static IRC_CMD(raw_315) {
591 recv_whohandler_315(client, argv, argc);
595 static IRC_CMD(raw_324) { //MODE LIST
596 //Watchcat #pktest +stnzN
597 if(from == NULL || argc < 3) return 0;
598 struct ChanNode *chan = getChanByName(argv[1]);
599 if(chan == NULL) return 0;
600 parseModes(chan->modes, argv[2], argv+3, argc-3);
604 static IRC_CMD(raw_invite) {
605 if(from == NULL || argc < 2) return 0;
606 struct UserNode *user = getUserByMask(from);
608 user = createTempUserMask(from);
610 user->flags |= USERFLAG_ISTMPUSER;
612 event_invite(client, user, argv[1]);
616 static IRC_CMD(raw_mode) {
617 if(from == NULL || argc < 2) return 0;
618 struct UserNode *user = getUserByMask(from);
620 user = createTempUserMask(from);
622 user->flags |= USERFLAG_ISTMPUSER;
624 if(argv[0][0] == '#') {
626 struct ChanNode *chan = getChanByName(argv[0]);
628 if(chan->chanbot != client->user) return 1;
629 parseModes(chan->modes, argv[1], argv+2, argc-2);
630 event_mode(user, chan, argv[1], argv+2, argc-2);
637 static IRC_CMD(raw_367) {
638 //Watchcat #pktest pk911!*@* TestBot!~bot@pktest.user.WebGamesNet 1315863279
639 struct ChanNode *chan = getChanByName(argv[1]);
642 while((ban = getMatchingChannelBan(chan, argv[2]))) {
643 removeChannelBan(ban);
645 addChannelBan(chan, argv[2]);
649 static IRC_CMD(raw_251) {
650 if(argc < 2) return 0;
651 char *total_user_str = argv[1];
652 char total_visible[20], total_invisible[20];
653 int i = 0, total_visible_pos = 0, total_invisible_pos = 0;
654 while(*total_user_str) {
655 if(*total_user_str == ' ') {
658 if(total_visible_pos < 20)
659 total_visible[total_visible_pos++] = *total_user_str;
661 if(total_invisible_pos < 20)
662 total_invisible[total_invisible_pos++] = *total_user_str;
666 total_visible[total_visible_pos] = '\0';
667 total_invisible[total_invisible_pos] = '\0';
668 statistics_network_users = atoi(total_visible) + atoi(total_invisible);
672 static IRC_CMD(raw_254) {
673 if(argc < 3) return 0;
674 statistics_network_channels = atoi(argv[1]);
679 static IRC_CMD(raw_332) {
680 //Skynet #neonserv :topic
681 struct ChanNode *chan = getChanByName(argv[1]);
683 strcpy(chan->topic, argv[2]);
688 //all the raws we receive...
689 register_irc_function("002", raw_002);
690 register_irc_function("251", raw_251);
691 register_irc_function("254", raw_254);
692 register_irc_function("324", raw_324);
693 register_irc_function("332", raw_332);
694 register_irc_function("367", raw_367);
695 register_irc_function("INVITE", raw_invite);
696 register_irc_function("NOTICE", raw_notice);
697 register_irc_function("TOPIC", raw_topic);
698 register_irc_function("KICK", raw_kick);
699 register_irc_function("PART", raw_part);
700 register_irc_function("QUIT", raw_quit);
701 register_irc_function("JOIN", raw_join);
702 register_irc_function("MODE", raw_mode);
703 register_irc_function("NICK", raw_nick);
704 register_irc_function("354", raw_354);
705 register_irc_function("315", raw_315);
706 register_irc_function("PING", raw_ping);
707 register_irc_function("PRIVMSG", raw_privmsg);
711 struct irc_cmd *cmd, *next;
712 for(cmd = irc_commands; cmd; cmd = next) {
718 void reply(struct ClientSocket *client, struct UserNode *user, const char *text, ...) {
719 const char *reply_format = get_language_string(user, text);
720 if(reply_format == NULL)
722 loadUserSettings(user);
723 char formatBuf[MAXLEN];
724 sprintf(formatBuf, "%s %s :%s", ((user->flags & USERFLAG_REPLY_PRIVMSG) ? "PRIVMSG" : "NOTICE"), user->nick, reply_format);
726 char sendBuf[MAXLEN];
728 if (!(client->flags & SOCKET_FLAG_CONNECTED)) return;
730 va_start(arg_list, text);
731 pos = vsnprintf(sendBuf, MAXLEN - 2, formatBuf, arg_list);
733 if (pos < 0 || pos > (MAXLEN - 2)) pos = MAXLEN - 2;
735 sendBuf[pos+1] = '\0';
736 write_socket(client, sendBuf, pos+1);
739 char* merge_argv(char **argv, int start, int end) {
740 return merge_argv_char(argv, start, end, ' ');
743 char* merge_argv_char(char **argv, int start, int end, char seperator) {
746 for(i = start; i < end; i++) {
750 while(p != argv[i+1]) {