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
  33 static void parse_line(struct ClientSocket *client, char *line);
  34 static void register_irc_function(char *command, irc_cmd_t *func);
  35 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc);
  36
  37 int parse_lines(struct ClientSocket *client, char *lines, int len) {
  38     int i, used = 0;
  39     char *line = lines;
  40     for(i = 0; i < len; i++) {
  41         if(lines[i] == '\r') //just zero it out :D
  42             lines[i] = 0;
  43         if(lines[i] == '\n') {
  44             lines[i] = 0;
  45             parse_line(client, line);
  46             line = lines+(i+1);
  47             used = i+1;
  48         }
  49     }
  50     return used;
  51 }
  52
  53 static void parse_line(struct ClientSocket *client, char *line) {
  54     int argc = 0;
  55     char *argv[MAXNUMPARAMS];
  56     printf("[recv %lu] %s\n", (unsigned long) strlen(line), line);
  57     if(line[0] == ':')
  58         line++;
  59     else
  60         argv[argc++] = NULL;
  61     while(*line) {
  62         //skip leading spaces
  63         while (*line == ' ')
  64             *line++ = 0;
  65         if (*line == ':') {
  66            //the rest is a single parameter
  67            argv[argc++] = line + 1;
  68            break;
  69         }
  70         argv[argc++] = line;
  71         if (argc >= MAXNUMPARAMS)
  72             break;
  73         while (*line != ' ' && *line)
  74             line++;
  75     }
  76     if(argc >= 2) {
  77         parse_raw(client, argv[0], argv[1], argv+2, argc-2);
  78     }
  79 }
  80
  81 static void register_irc_function(char *command, irc_cmd_t *func) {
  82     struct irc_cmd *irc_cmd = malloc(sizeof(*irc_cmd));
  83     if (!irc_cmd)
  84     {
  85         perror("malloc() failed");
  86         return;
  87     }
  88     irc_cmd->cmd = command;
  89     irc_cmd->func = func;
  90     irc_cmd->next = irc_commands;
  91     irc_commands = irc_cmd;
  92 }
  93
  94 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc) {
  95     struct irc_cmd *irc_cmd;
  96     int ret = 0;
  97     for(irc_cmd = irc_commands; irc_cmd; irc_cmd = irc_cmd->next) {
  98         if(!stricmp(irc_cmd->cmd, cmd)) {
  99             ret = irc_cmd->func(client, from, argv, argc);
 100             break;
 101         }
 102     }
 103     if(!irc_cmd) {
 104         event_raw(client, from, cmd, argv, argc);
 105     } else if(!ret) {
 106         fprintf(stderr,"PARSE ERROR: %s\n", cmd);
 107     }
 108 }
 109
 110 static USERLIST_CALLBACK(got_channel_userlist) {
 111     struct ChanUser *chanuser = data;
 112     event_join(chanuser);
 113 }
 114
 115 static IRC_CMD(raw_001) {
 116     client->flags |= SOCKET_FLAG_READY;
 117     event_bot_ready(client);
 118     return 1;
 119 }
 120
 121 static int is_firstBotSeeUser(struct ClientSocket *client, struct UserNode *user) {
 122     struct ClientSocket *bot, *pref_bot = NULL, *unpref_bot = NULL;
 123     struct ChanUser *chanuser;
 124     int found;
 125     for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 126         found = 0;
 127         for(chanuser = getUserChannels(bot->user, NULL); chanuser; chanuser = getUserChannels(bot->user, chanuser)) {
 128             if(isUserOnChan(user, chanuser->chan)) {
 129                 found = 1;
 130                 break;
 131             }
 132         }
 133         if(!found) continue;
 134         if(bot->flags & SOCKET_FLAG_PREFERRED) {
 135             pref_bot = bot;
 136             break;
 137         } else
 138             unpref_bot = bot;
 139     }
 140     bot = (pref_bot ? pref_bot : unpref_bot);
 141     if(client == bot)
 142         return 1;
 143     else
 144         return 0;
 145 }
 146
 147 static IRC_CMD(raw_join) {
 148     if(from == NULL || argc < 1) return 0;
 149     struct UserNode *user = getUserByMask(from);
 150     struct ChanNode *chan = getChanByName(argv[0]);
 151     if(!chan && !(user->flags & USERFLAG_ISBOT)) return 0;
 152     if(chan && chan->chanbot != client->user) return 1; //we ignore it - but it's not a parse error
 153     if(user == NULL) {
 154         user = addUserMask(from);
 155     }
 156     struct UserNode *registering, *last_registering = NULL, *next_registering;
 157     int noEvent = 0, wasRegistering = 0;
 158     for(registering = registering_users; registering; registering = next_registering) {
 159         next_registering = registering->next;
 160         if(!strcmp(registering->nick, user->nick)) {
 161             noEvent = event_registered(registering, user);
 162             wasRegistering = 1;
 163             if(last_registering)
 164                 last_registering->next = registering->next;
 165             else
 166                 registering_users = registering->next;
 167             delUser(registering, 1);
 168         } else if(time(0) - registering->created > 2) {
 169             if(last_registering)
 170                 last_registering->next = registering->next;
 171             else
 172                 registering_users = registering->next;
 173             delUser(registering, 1);
 174         } else
 175             last_registering = registering;
 176     }
 177     if(chan == NULL) {
 178         chan = addChannel(argv[0]);
 179         //request member list
 180         chan->chanbot = user;
 181         struct ChanUser *chanuser = addChanUser(chan, user); //it must be a bot
 182         get_userlist(chan, got_channel_userlist, chanuser);
 183         putsock(client, "MODE %s", chan->name);
 184         putsock(client, "MODE %s +b", chan->name);
 185     } else if(!isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
 186         struct ChanUser *chanuser = addChanUser(chan, user);
 187         if(!noEvent) {
 188             if(wasRegistering)
 189                 user->flags |= USERFLAG_WAS_REGISTRING;
 190             event_join(chanuser);
 191             if(wasRegistering)
 192                 user->flags &= ~USERFLAG_WAS_REGISTRING;
 193         }
 194     }
 195     return 1;
 196 }
 197
 198 static IRC_CMD(raw_part) {
 199     if(from == NULL || argc < 1) return 0;
 200     struct UserNode *user = getUserByMask(from);
 201     if(user == NULL) return 0;
 202     struct ChanNode *chan = getChanByName(argv[0]);
 203     if(chan == NULL) return 0;
 204     if(chan->chanbot != client->user) return 1; //we ignore it - but it's not a parse error
 205     if(isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
 206         struct ChanUser *chanuser = getChanUser(user, chan);
 207         delChanUser(chanuser, 0); //we need to free the chanuser manually!
 208         event_part(chanuser, (argc > 1 ? argv[1] : NULL));
 209         freeChanUser(chanuser);
 210         if(chan->chanbot == user) {
 211             //check if theres another bot in the channel - otherwise free it
 212             checkChannelVisibility(chan);
 213         }
 214     }
 215     if(user->channel == NULL && !(user->flags & USERFLAG_ISBOT)) {
 216         //remove the user
 217         delUser(user, 1);
 218     }
 219     return 1;
 220 }
 221
 222 static IRC_CMD(raw_quit) {
 223     if(from == NULL || argc < 1) return 0;
 224     struct UserNode *user = getUserByMask(from);
 225     if(user == NULL) return 0;
 226     if(!is_firstBotSeeUser(client, user)) return 1; //we ignore it - but it's not a parse error
 227     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
 228     event_quit(user, argv[0]);
 229     if(user->flags & USERFLAG_ISBOT) {
 230         //check if there are other bots in the users channel - otherwise free them
 231         struct ChanUser *chanuser, *next;
 232         for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
 233             next = getUserChannels(user, chanuser);
 234             if(chanuser->chan->chanbot == user)
 235                 checkChannelVisibility(chanuser->chan);
 236         }
 237         //search the user representing the bot in the world of IRC
 238         struct ClientSocket *bot;
 239         for(bot = getBots(0, NULL); bot; bot = getBots(0, bot)) {
 240             if(bot->user == user) {
 241                 bot->user = NULL;
 242                 break;
 243             }
 244         }
 245     }
 246     if(!stricmp(argv[0], "Registered") && !(user->flags & USERFLAG_ISBOT)) {
 247         user->next = registering_users;
 248         user->created = time(0);
 249         registering_users = user;
 250     } else
 251         delUser(user, 1); //now we fully free the user
 252     return 1;
 253 }
 254
 255 void bot_disconnect(struct ClientSocket *client) {
 256     struct UserNode *user = client->user;
 257     struct ChanUser *chanuser, *next;
 258     if(user) {
 259         delUser(user, 0);
 260         event_quit(user, "disconnected");
 261         for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
 262             next = getUserChannels(user, chanuser);
 263             if(chanuser->chan->chanbot == user)
 264                 checkChannelVisibility(chanuser->chan);
 265             freeChanUser(chanuser);
 266         }
 267         user->channel = NULL;
 268         delUser(user, 1); //now we fully free the user
 269         client->user = NULL;
 270     }
 271 }
 272
 273 static IRC_CMD(raw_kick) {
 274     if(from == NULL || argc < 3) return 0;
 275     struct UserNode *user = getUserByMask(from);
 276     struct UserNode *target = getUserByNick(argv[1]);
 277     struct ChanNode *chan = getChanByName(argv[0]);
 278     if(chan == NULL || target == NULL) return 0;
 279     if(chan->chanbot != client->user) return 1; //we ignore it - but it's not a parse error
 280     if(isUserOnChan(target, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
 281         if(user == NULL) {
 282             user = createTempUser(from);
 283             user->flags |= USERFLAG_ISTMPUSER;
 284         }
 285         struct ChanUser *chanuser = getChanUser(target, chan);
 286         delChanUser(chanuser, 0); //we need to free the chanuser manually!
 287         event_kick(user, chanuser, argv[1]);
 288         freeChanUser(chanuser);
 289         if(chanuser->chan->chanbot == user) {
 290             //check if theres another bot in the channel - otherwise free it
 291             checkChannelVisibility(chan);
 292         }
 293     }
 294     if(target->channel == NULL && !(target->flags & USERFLAG_ISBOT)) {
 295         //remove the user
 296         delUser(target, 1);
 297     }
 298     return 1;
 299 }
 300
 301 static IRC_CMD(raw_topic) {
 302     if(from == NULL || argc < 2) return 0;
 303     struct UserNode *user = getUserByMask(from);
 304     struct ChanNode *chan = getChanByName(argv[0]);
 305     if(chan == NULL) return 0;
 306     if(chan->chanbot != client->user) return 1; //just ignore it to prevent event duplicates
 307     if(user == NULL) {
 308         user = createTempUser(from);
 309         user->flags |= USERFLAG_ISTMPUSER;
 310     }
 311     event_topic(user, chan, argv[1]);
 312     strcpy(chan->topic, argv[1]);
 313     return 1;
 314 }
 315
 316 static IRC_CMD(raw_privmsg) {
 317     if(from == NULL || argc < 2) return 0;
 318     struct UserNode *user = getUserByMask(from);
 319     if(user == NULL) {
 320         user = createTempUser(from);
 321         user->flags |= USERFLAG_ISTMPUSER;
 322     }
 323     if(argv[0][0] == '#') { //Channel message
 324         struct ChanNode *chan = getChanByName(argv[0]);
 325         if(chan && chan->chanbot == client->user) {
 326             if(argv[1][0] == '\001') {
 327                 char *cmd = &argv[1][1];
 328                 char *text = strstr(cmd, " ");
 329                 if(text) {
 330                     *text = '\0';
 331                     text++;
 332                     if(strlen(text) && text[strlen(text)-1] == '\001')
 333                         text[strlen(text)-1] = '\0';
 334                 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
 335                     cmd[strlen(cmd)-1] = '\0';
 336                 event_chanctcp(user, chan, cmd, text);
 337             } else
 338                 event_chanmsg(user, chan, argv[1]);
 339         }
 340     } else {
 341         struct UserNode *target = getUserByNick(argv[0]);
 342         if(target) {
 343             if(argv[1][0] == '\001') {
 344                 char *cmd = &argv[1][1];
 345                 char *text = strstr(cmd, " ");
 346                 if(text) {
 347                     *text = '\0';
 348                     text++;
 349                     if(strlen(text) && text[strlen(text)-1] == '\001')
 350                         text[strlen(text)-1] = '\0';
 351                 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
 352                     cmd[strlen(cmd)-1] = '\0';
 353                 event_privctcp(user, target, cmd, text);
 354             } else
 355                 event_privmsg(user, target, argv[1]);
 356         }
 357     }
 358     return 1;
 359 }
 360
 361 static IRC_CMD(raw_notice) {
 362     if(from == NULL || argc < 2) return 0;
 363     struct UserNode *user = getUserByMask(from);
 364     if(user == NULL) {
 365         user = createTempUser(from);
 366         user->flags |= USERFLAG_ISTMPUSER;
 367     }
 368     if(argv[0][0] == '#') { //Channel notice
 369         struct ChanNode *chan = getChanByName(argv[0]);
 370         if(chan && chan->chanbot == client->user)
 371             event_channotice(user, chan, argv[1]);
 372     } else {
 373         struct UserNode *target = getUserByNick(argv[0]);
 374         if(target)
 375             event_privnotice(user, target, argv[1]);
 376     }
 377     return 1;
 378 }
 379
 380 static IRC_CMD(raw_nick) {
 381     if(from == NULL || argc == 0) return 0;
 382     struct UserNode *user = getUserByMask(from);
 383     if(user == NULL) return 0;
 384     if(!is_firstBotSeeUser(client, user)) return 1; //we ignore it - but it's not a parse error
 385     event_nick(user, argv[0]);
 386     renameUser(user, argv[0]);
 387     return 1;
 388 }
 389
 390 static IRC_CMD(raw_ping) {
 391     if(argc == 0) return 0;
 392     putsock(client, "PONG :%s", argv[0]);
 393     return 1;
 394 }
 395
 396 static IRC_CMD(raw_354) {
 397     recv_whohandler_354(client, argv, argc);
 398     return 1;
 399 }
 400
 401 static IRC_CMD(raw_315) {
 402     recv_whohandler_315(client, argv, argc);
 403     return 1;
 404 }
 405
 406 static IRC_CMD(raw_324) { //MODE LIST
 407     //Watchcat #pktest +stnzN
 408     if(from == NULL || argc < 3) return 0;
 409     struct ChanNode *chan = getChanByName(argv[1]);
 410     if(chan == NULL) return 0;
 411     parseModes(chan->modes, argv[2], argv+3, argc-3);
 412     return 1;
 413 }
 414
 415 static IRC_CMD(raw_invite) {
 416     if(from == NULL || argc < 2) return 0;
 417     struct UserNode *user = getUserByMask(from);
 418     if(user == NULL) {
 419         user = createTempUser(from);
 420         user->flags |= USERFLAG_ISTMPUSER;
 421     }
 422     event_invite(client, user, argv[1]);
 423     return 1;
 424 }
 425
 426 static IRC_CMD(raw_mode) {
 427     if(from == NULL || argc < 2) return 0;
 428     struct UserNode *user = getUserByMask(from);
 429     if(user == NULL) {
 430         user = createTempUser(from);
 431         user->flags |= USERFLAG_ISTMPUSER;
 432     }
 433     if(argv[0][0] == '#') {
 434         //ChannelMode
 435         struct ChanNode *chan = getChanByName(argv[0]);
 436         if(!chan) return 0;
 437         if(chan->chanbot != client->user) return 1;
 438         event_mode(user, chan, argv[1], argv+2, argc-2);
 439         parseModes(chan->modes, argv[1], argv+2, argc-2);
 440     } else {
 441         //UserMode
 442     }
 443     return 1;
 444 }
 445
 446 static IRC_CMD(raw_367) {
 447     //Watchcat #pktest pk911!*@* TestBot!~bot@pktest.user.WebGamesNet 1315863279
 448     struct ChanNode *chan = getChanByName(argv[1]);
 449     if(!chan) return 0;
 450     struct BanNode *ban;
 451     while((ban = getMatchingChannelBan(chan, argv[2]))) {
 452         removeChannelBan(ban);
 453     }
 454     addChannelBan(chan, argv[2]);
 455     return 1;
 456 }
 457
 458 void init_parser() {
 459     //all the raws we receive...
 460     register_irc_function("001", raw_001);
 461     register_irc_function("324", raw_324);
 462     register_irc_function("367", raw_367);
 463     register_irc_function("INVITE", raw_invite);
 464     register_irc_function("NOTICE", raw_notice);
 465     register_irc_function("TOPIC", raw_topic);
 466     register_irc_function("KICK", raw_kick);
 467     register_irc_function("PART", raw_part);
 468     register_irc_function("QUIT", raw_quit);
 469     register_irc_function("JOIN", raw_join);
 470     register_irc_function("MODE", raw_mode);
 471     register_irc_function("NICK", raw_nick);
 472     register_irc_function("354", raw_354);
 473     register_irc_function("315", raw_315);
 474     register_irc_function("PING", raw_ping);
 475     register_irc_function("PRIVMSG", raw_privmsg);
 476 }
 477
 478 void free_parser() {
 479     struct irc_cmd *cmd, *next;
 480     for(cmd = irc_commands; cmd; cmd = next) {
 481         next = cmd->next;
 482         free(cmd);
 483     }
 484 }
 485
 486 void reply(struct ClientSocket *client, struct UserNode *user, const char *text, ...) {
 487     const char *reply_format = get_language_string(user, text);
 488     if(reply_format == NULL)
 489         reply_format = text;
 490     loadUserSettings(user);
 491     char formatBuf[MAXLEN];
 492     sprintf(formatBuf, "%s %s :%s", ((user->flags & USERFLAG_REPLY_PRIVMSG) ? "PRIVMSG" : "NOTICE"), user->nick, reply_format);
 493     va_list arg_list;
 494     char sendBuf[MAXLEN];
 495     int pos;
 496     if (!(client->flags & SOCKET_FLAG_CONNECTED)) return;
 497     sendBuf[0] = '\0';
 498     va_start(arg_list, text);
 499     pos = vsnprintf(sendBuf, MAXLEN - 2, formatBuf, arg_list);
 500     va_end(arg_list);
 501     if (pos < 0 || pos > (MAXLEN - 2)) pos = MAXLEN - 2;
 502     sendBuf[pos] = '\n';
 503     sendBuf[pos+1] = '\0';
 504     write_socket(client, sendBuf, pos+1);
 505 }
 506
 507 char* merge_argv(char **argv, int start, int end) {
 508     return merge_argv_char(argv, start, end, ' ');
 509 }
 510
 511 char* merge_argv_char(char **argv, int start, int end, char seperator) {
 512     int i;
 513     char *p = NULL;
 514     for(i = start; i < end; i++) {
 515         p = argv[i];
 516         while(*p) p++;
 517         *p = seperator;
 518     }
 519     if(p) *p = '\0';
 520     return argv[start];
 521 }