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