src/IRCParser.c

   1 /* IRCParser.c - NeonServ v5.2
   2  * Copyright (C) 2011  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 #include "IRCParser.h"
  19 #include "UserNode.h"
  20 #include "ChanNode.h"
  21 #include "ChanUser.h"
  22 #include "IRCEvents.h"
  23 #include "ClientSocket.h"
  24 #include "WHOHandler.h"
  25 #include "lang.h"
  26 #include "DBHelper.h"
  27 #include "BanNode.h"
  28 #include "ModeNode.h"
  29
  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;
  35
  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);
  39
  40 int parse_lines(struct ClientSocket *client, char *lines, int len) {
  41     int i, used = 0;
  42     char *line = lines;
  43     for(i = 0; i < len; i++) {
  44         if(lines[i] == '\r') //just zero it out :D
  45             lines[i] = 0;
  46         if(lines[i] == '\n') {
  47             lines[i] = 0;
  48             parse_line(client, line);
  49             line = lines+(i+1);
  50             used = i+1;
  51         }
  52     }
  53     return used;
  54 }
  55
  56 static void parse_line(struct ClientSocket *client, char *line) {
  57     int argc = 0;
  58     char *argv[MAXNUMPARAMS];
  59     printf("[recv %lu] %s\n", (unsigned long) strlen(line), line);
  60     if(line[0] == ':')
  61         line++;
  62     else
  63         argv[argc++] = NULL;
  64     while(*line) {
  65         //skip leading spaces
  66         while (*line == ' ')
  67             *line++ = 0;
  68         if (*line == ':') {
  69            //the rest is a single parameter
  70            argv[argc++] = line + 1;
  71            break;
  72         }
  73         argv[argc++] = line;
  74         if (argc >= MAXNUMPARAMS)
  75             break;
  76         while (*line != ' ' && *line)
  77             line++;
  78     }
  79     if(argc >= 2) {
  80         parse_raw(client, argv[0], argv[1], argv+2, argc-2);
  81     }
  82 }
  83
  84 static void register_irc_function(char *command, irc_cmd_t *func) {
  85     struct irc_cmd *irc_cmd = malloc(sizeof(*irc_cmd));
  86     if (!irc_cmd)
  87     {
  88         perror("malloc() failed");
  89         return;
  90     }
  91     irc_cmd->cmd = command;
  92     irc_cmd->func = func;
  93     irc_cmd->next = irc_commands;
  94     irc_commands = irc_cmd;
  95 }
  96
  97 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc) {
  98     struct irc_cmd *irc_cmd;
  99     int ret = 0;
 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);
 103             break;
 104         }
 105     }
 106     if(!irc_cmd) {
 107         event_raw(client, from, cmd, argv, argc);
 108     } else if(!ret) {
 109         fprintf(stderr,"PARSE ERROR: %s\n", cmd);
 110     }
 111 }
 112
 113 static USERLIST_CALLBACK(got_channel_userlist) {
 114     struct ChanUser *chanuser = data;
 115     event_join(chanuser);
 116 }
 117
 118 static IRC_CMD(raw_001) {
 119     struct UserNode *user = getUserByNick(argv[0]);
 120     if(!user)
 121         user = addUser(argv[0]);
 122     client->user = user;
 123     client->user->flags |= USERFLAG_ISBOT;
 124     client->flags |= SOCKET_FLAG_READY;
 125     event_bot_ready(client);
 126     return 1;
 127 }
 128
 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;
 132     int found;
 133     for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 134         found = 0;
 135         for(chanuser = getUserChannels(bot->user, NULL); chanuser; chanuser = getUserChannels(bot->user, chanuser)) {
 136             if(isUserOnChan(user, chanuser->chan)) {
 137                 found = 1;
 138                 break;
 139             }
 140         }
 141         if(!found) continue;
 142         if(bot->flags & SOCKET_FLAG_PREFERRED) {
 143             pref_bot = bot;
 144             break;
 145         } else
 146             unpref_bot = bot;
 147     }
 148     bot = (pref_bot ? pref_bot : unpref_bot);
 149     if(client == bot)
 150         return 1;
 151     else
 152         return 0;
 153 }
 154
 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)) {
 162                 return 0;
 163             }
 164         }
 165     }
 166     return 1;
 167 }
 168
 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
 175     if(user == NULL) {
 176         user = addUserMask(from);
 177     }
 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);
 184             wasRegistering = 1;
 185             if(last_registering)
 186                 last_registering->next = registering->next;
 187             else
 188                 registering_users = registering->next;
 189             delUser(registering, 1);
 190         } else if(time(0) - registering->created > 2) {
 191             if(last_registering)
 192                 last_registering->next = registering->next;
 193             else
 194                 registering_users = registering->next;
 195             delUser(registering, 1);
 196         } else
 197             last_registering = registering;
 198     }
 199     if(chan == NULL) {
 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);
 209         if(!noEvent) {
 210             if(wasRegistering)
 211                 user->flags |= USERFLAG_WAS_REGISTRING;
 212             event_join(chanuser);
 213             if(wasRegistering)
 214                 user->flags &= ~USERFLAG_WAS_REGISTRING;
 215         }
 216     }
 217     return 1;
 218 }
 219
 220 static IRC_CMD(raw_part) {
 221     if(from == NULL || argc < 1) return 0;
 222     struct UserNode *user = getUserByMask(from);
 223     if(user == NULL) return 0;
 224     struct ChanNode *chan = getChanByName(argv[0]);
 225     if(chan == NULL) return 0;
 226     if(chan->chanbot != client->user) return 1; //we ignore it - but it's not a parse error
 227     if(isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
 228         struct ChanUser *chanuser = getChanUser(user, chan);
 229         delChanUser(chanuser, 0); //we need to free the chanuser manually!
 230         event_part(chanuser, (argc > 1 ? argv[1] : NULL));
 231         freeChanUser(chanuser);
 232         if(chan->chanbot == user) {
 233             //check if theres another bot in the channel - otherwise free it
 234             checkChannelVisibility(chan);
 235         }
 236     }
 237     if(user->channel == NULL && !(user->flags & USERFLAG_ISBOT)) {
 238         //remove the user
 239         delUser(user, 1);
 240     }
 241     return 1;
 242 }
 243
 244 static IRC_CMD(raw_quit) {
 245     if(from == NULL || argc < 1) return 0;
 246     struct UserNode *user = getUserByMask(from);
 247     if(user == NULL) return 0;
 248     if(!is_firstBotSeeUser(client, user)) return 1; //we ignore it - but it's not a parse error
 249     int registering = !stricmp(argv[0], "Registered");
 250     if((registering && (user->flags & USERFLAG_ISBOT))) return 1; //bot is registering - just ignore it
 251     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
 252     event_quit(user, argv[0]);
 253     if(user->flags & USERFLAG_ISBOT) {
 254         //check if there are other bots in the users channel - otherwise free them
 255         struct ChanUser *chanuser, *next;
 256         for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
 257             next = getUserChannels(user, chanuser);
 258             if(chanuser->chan->chanbot == user)
 259                 checkChannelVisibility(chanuser->chan);
 260         }
 261         //search the user representing the bot in the world of IRC
 262         struct ClientSocket *bot;
 263         for(bot = getBots(0, NULL); bot; bot = getBots(0, bot)) {
 264             if(bot->user == user) {
 265                 bot->user = NULL;
 266                 break;
 267             }
 268         }
 269     }
 270     if(registering && !(user->flags & USERFLAG_ISBOT)) {
 271         user->next = registering_users;
 272         user->created = time(0);
 273         registering_users = user;
 274     } else
 275         delUser(user, 1); //now we fully free the user
 276     return 1;
 277 }
 278
 279 void bot_disconnect(struct ClientSocket *client) {
 280     struct UserNode *user = client->user;
 281     struct ChanUser *chanuser, *next;
 282     if(user) {
 283         if(is_onlyBotSeeUser(client, user)) {
 284             //ok  the bot-user is not seen by any other bots so we can simply free it.
 285             delUser(user, 0);
 286             event_quit(user, "disconnected");
 287             for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
 288                 next = getUserChannels(user, chanuser);
 289                 if(chanuser->chan->chanbot == user)
 290                     checkChannelVisibility(chanuser->chan);
 291                 freeChanUser(chanuser);
 292             }
 293             user->channel = NULL;
 294             delUser(user, 1); //now we fully free the user
 295         } else {
 296             //we need to transform the bot-user back to a normal user (BNC FIX)
 297             user->flags &= ~USERFLAG_ISBOT;
 298             for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
 299                 next = getUserChannels(user, chanuser);
 300                 if(chanuser->chan->chanbot == user)
 301                     checkChannelVisibility(chanuser->chan);
 302             }
 303         }
 304         client->user = NULL;
 305     }
 306 }
 307
 308 static IRC_CMD(raw_kick) {
 309     if(from == NULL || argc < 3) return 0;
 310     struct UserNode *user = getUserByMask(from);
 311     struct UserNode *target = getUserByNick(argv[1]);
 312     struct ChanNode *chan = getChanByName(argv[0]);
 313     if(chan == NULL || target == NULL) return 0;
 314     if(chan->chanbot != client->user) return 1; //we ignore it - but it's not a parse error
 315     if(isUserOnChan(target, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
 316         if(user == NULL) {
 317             user = createTempUser(from);
 318             user->flags |= USERFLAG_ISTMPUSER;
 319         }
 320         struct ChanUser *chanuser = getChanUser(target, chan);
 321         delChanUser(chanuser, 0); //we need to free the chanuser manually!
 322         event_kick(user, chanuser, argv[1]);
 323         if(chanuser->chan->chanbot == user) {
 324             //check if theres another bot in the channel - otherwise free it
 325             checkChannelVisibility(chan);
 326         }
 327         freeChanUser(chanuser);
 328     }
 329     if(target->channel == NULL && !(target->flags & USERFLAG_ISBOT)) {
 330         //remove the user
 331         delUser(target, 1);
 332     }
 333     return 1;
 334 }
 335
 336 static IRC_CMD(raw_topic) {
 337     if(from == NULL || argc < 2) return 0;
 338     struct UserNode *user = getUserByMask(from);
 339     struct ChanNode *chan = getChanByName(argv[0]);
 340     if(chan == NULL) return 0;
 341     if(chan->chanbot != client->user) return 1; //just ignore it to prevent event duplicates
 342     if(user == NULL) {
 343         user = createTempUser(from);
 344         user->flags |= USERFLAG_ISTMPUSER;
 345     }
 346     event_topic(user, chan, argv[1]);
 347     strcpy(chan->topic, argv[1]);
 348     return 1;
 349 }
 350
 351 static IRC_CMD(raw_privmsg) {
 352     if(from == NULL || argc < 2) return 0;
 353     struct UserNode *user = getUserByMask(from);
 354     if(user == NULL) {
 355         user = createTempUser(from);
 356         user->flags |= USERFLAG_ISTMPUSER;
 357     }
 358     if(argv[0][0] == '#') { //Channel message
 359         struct ChanNode *chan = getChanByName(argv[0]);
 360         if(chan && chan->chanbot == client->user) {
 361             if(statistics_enabled)
 362                 statistics_privmsg++;
 363             if(argv[1][0] == '\001') {
 364                 char *cmd = &argv[1][1];
 365                 char *text = strstr(cmd, " ");
 366                 if(text) {
 367                     *text = '\0';
 368                     text++;
 369                     if(strlen(text) && text[strlen(text)-1] == '\001')
 370                         text[strlen(text)-1] = '\0';
 371                 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
 372                     cmd[strlen(cmd)-1] = '\0';
 373                 event_chanctcp(user, chan, cmd, text);
 374             } else
 375                 event_chanmsg(user, chan, argv[1]);
 376         }
 377     } else {
 378         struct UserNode *target = getUserByNick(argv[0]);
 379         if(target) {
 380             if(argv[1][0] == '\001') {
 381                 char *cmd = &argv[1][1];
 382                 char *text = strstr(cmd, " ");
 383                 if(text) {
 384                     *text = '\0';
 385                     text++;
 386                     if(strlen(text) && text[strlen(text)-1] == '\001')
 387                         text[strlen(text)-1] = '\0';
 388                 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
 389                     cmd[strlen(cmd)-1] = '\0';
 390                 event_privctcp(user, target, cmd, text);
 391             } else
 392                 event_privmsg(user, target, argv[1]);
 393         }
 394     }
 395     return 1;
 396 }
 397
 398 static IRC_CMD(raw_notice) {
 399     if(from == NULL || argc < 2) return 0;
 400     struct UserNode *user = getUserByMask(from);
 401     if(user == NULL) {
 402         user = createTempUser(from);
 403         user->flags |= USERFLAG_ISTMPUSER;
 404     }
 405     if(argv[0][0] == '#') { //Channel notice
 406         struct ChanNode *chan = getChanByName(argv[0]);
 407         if(chan && chan->chanbot == client->user)
 408             event_channotice(user, chan, argv[1]);
 409     } else {
 410         struct UserNode *target = getUserByNick(argv[0]);
 411         if(target)
 412             event_privnotice(user, target, argv[1]);
 413     }
 414     return 1;
 415 }
 416
 417 static IRC_CMD(raw_nick) {
 418     if(from == NULL || argc == 0) return 0;
 419     struct UserNode *user = getUserByMask(from);
 420     if(user == NULL) return 0;
 421     if(!is_firstBotSeeUser(client, user)) return 1; //we ignore it - but it's not a parse error
 422     event_nick(user, argv[0]);
 423     renameUser(user, argv[0]);
 424     return 1;
 425 }
 426
 427 static IRC_CMD(raw_ping) {
 428     if(argc == 0) return 0;
 429     putsock(client, "PONG :%s", argv[0]);
 430     return 1;
 431 }
 432
 433 static IRC_CMD(raw_354) {
 434     recv_whohandler_354(client, argv, argc);
 435     return 1;
 436 }
 437
 438 static IRC_CMD(raw_315) {
 439     recv_whohandler_315(client, argv, argc);
 440     return 1;
 441 }
 442
 443 static IRC_CMD(raw_324) { //MODE LIST
 444     //Watchcat #pktest +stnzN
 445     if(from == NULL || argc < 3) return 0;
 446     struct ChanNode *chan = getChanByName(argv[1]);
 447     if(chan == NULL) return 0;
 448     parseModes(chan->modes, argv[2], argv+3, argc-3);
 449     return 1;
 450 }
 451
 452 static IRC_CMD(raw_invite) {
 453     if(from == NULL || argc < 2) return 0;
 454     struct UserNode *user = getUserByMask(from);
 455     if(user == NULL) {
 456         user = createTempUser(from);
 457         user->flags |= USERFLAG_ISTMPUSER;
 458     }
 459     event_invite(client, user, argv[1]);
 460     return 1;
 461 }
 462
 463 static IRC_CMD(raw_mode) {
 464     if(from == NULL || argc < 2) return 0;
 465     struct UserNode *user = getUserByMask(from);
 466     if(user == NULL) {
 467         user = createTempUser(from);
 468         user->flags |= USERFLAG_ISTMPUSER;
 469     }
 470     if(argv[0][0] == '#') {
 471         //ChannelMode
 472         struct ChanNode *chan = getChanByName(argv[0]);
 473         if(!chan) return 0;
 474         if(chan->chanbot != client->user) return 1;
 475         event_mode(user, chan, argv[1], argv+2, argc-2);
 476         parseModes(chan->modes, argv[1], argv+2, argc-2);
 477     } else {
 478         //UserMode
 479     }
 480     return 1;
 481 }
 482
 483 static IRC_CMD(raw_367) {
 484     //Watchcat #pktest pk911!*@* TestBot!~bot@pktest.user.WebGamesNet 1315863279
 485     struct ChanNode *chan = getChanByName(argv[1]);
 486     if(!chan) return 0;
 487     struct BanNode *ban;
 488     while((ban = getMatchingChannelBan(chan, argv[2]))) {
 489         removeChannelBan(ban);
 490     }
 491     addChannelBan(chan, argv[2]);
 492     return 1;
 493 }
 494
 495 static IRC_CMD(raw_251) {
 496     if(argc < 2) return 0;
 497     char *total_user_str = argv[1];
 498     char total_visible[20], total_invisible[20];
 499     int i, total_visible_pos = 0, total_invisible_pos = 0;
 500     while(*total_user_str) {
 501         if(*total_user_str == ' ') {
 502             i++;
 503         } else if(i == 2) {
 504             if(total_visible_pos < 20)
 505                 total_visible[total_visible_pos++] = *total_user_str;
 506         } else if(i == 5) {
 507             if(total_invisible_pos < 20)
 508                 total_invisible[total_invisible_pos++] = *total_user_str;
 509         }
 510         total_user_str++;
 511     }
 512     total_visible[total_visible_pos] = '\0';
 513     total_invisible[total_invisible_pos] = '\0';
 514     statistics_network_users = atoi(total_visible) + atoi(total_invisible);
 515     return 1;
 516 }
 517
 518 static IRC_CMD(raw_254) {
 519     if(argc < 3) return 0;
 520     statistics_network_channels = atoi(argv[1]);
 521     statistics_update();
 522     return 1;
 523 }
 524
 525 static IRC_CMD(raw_332) {
 526     //Skynet #neonserv :topic
 527     struct ChanNode *chan = getChanByName(argv[1]);
 528     if(!chan) return 0;
 529     strcpy(chan->topic, argv[2]);
 530     return 1;
 531 }
 532
 533 void init_parser() {
 534     //all the raws we receive...
 535     register_irc_function("001", raw_001);
 536     register_irc_function("251", raw_251);
 537     register_irc_function("254", raw_254);
 538     register_irc_function("324", raw_324);
 539     register_irc_function("332", raw_332);
 540     register_irc_function("367", raw_367);
 541     register_irc_function("INVITE", raw_invite);
 542     register_irc_function("NOTICE", raw_notice);
 543     register_irc_function("TOPIC", raw_topic);
 544     register_irc_function("KICK", raw_kick);
 545     register_irc_function("PART", raw_part);
 546     register_irc_function("QUIT", raw_quit);
 547     register_irc_function("JOIN", raw_join);
 548     register_irc_function("MODE", raw_mode);
 549     register_irc_function("NICK", raw_nick);
 550     register_irc_function("354", raw_354);
 551     register_irc_function("315", raw_315);
 552     register_irc_function("PING", raw_ping);
 553     register_irc_function("PRIVMSG", raw_privmsg);
 554 }
 555
 556 void free_parser() {
 557     struct irc_cmd *cmd, *next;
 558     for(cmd = irc_commands; cmd; cmd = next) {
 559         next = cmd->next;
 560         free(cmd);
 561     }
 562 }
 563
 564 void reply(struct ClientSocket *client, struct UserNode *user, const char *text, ...) {
 565     const char *reply_format = get_language_string(user, text);
 566     if(reply_format == NULL)
 567         reply_format = text;
 568     loadUserSettings(user);
 569     char formatBuf[MAXLEN];
 570     sprintf(formatBuf, "%s %s :%s", ((user->flags & USERFLAG_REPLY_PRIVMSG) ? "PRIVMSG" : "NOTICE"), user->nick, reply_format);
 571     va_list arg_list;
 572     char sendBuf[MAXLEN];
 573     int pos;
 574     if (!(client->flags & SOCKET_FLAG_CONNECTED)) return;
 575     sendBuf[0] = '\0';
 576     va_start(arg_list, text);
 577     pos = vsnprintf(sendBuf, MAXLEN - 2, formatBuf, arg_list);
 578     va_end(arg_list);
 579     if (pos < 0 || pos > (MAXLEN - 2)) pos = MAXLEN - 2;
 580     sendBuf[pos] = '\n';
 581     sendBuf[pos+1] = '\0';
 582     write_socket(client, sendBuf, pos+1);
 583 }
 584
 585 char* merge_argv(char **argv, int start, int end) {
 586     return merge_argv_char(argv, start, end, ' ');
 587 }
 588
 589 char* merge_argv_char(char **argv, int start, int end, char seperator) {
 590     int i;
 591     char *p = NULL;
 592     for(i = start; i < end; i++) {
 593         p = argv[i];
 594         while(*p) p++;
 595         *p = seperator;
 596     }
 597     if(p) *p = '\0';
 598     return argv[start];
 599 }