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