IRCParser.c

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