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(!ret) {
  82         //fprintf(stderr,"PARSE ERROR: %s", cmd); commented out right now - we're still developing and this is very spammy
  83     }
  84 }
  85
  86 static USERLIST_CALLBACK(got_channel_userlist) {
  87     struct ChanUser *chanuser = data;
  88     event_join(chanuser);
  89     putsock(client, "PRIVMSG %s :[BOT JOIN] Users on this Channel:", chan->name);
  90     for(chanuser = getChannelUsers(chan, NULL); chanuser; chanuser = getChannelUsers(chan, chanuser)) {
  91         putsock(client, "PRIVMSG %s :  %s!%s@%s [%s]  rights: %d", chan->name, chanuser->user->nick, chanuser->user->ident, chanuser->user->host, ((chanuser->user->flags & USERFLAG_ISAUTHED) ? chanuser->user->auth : "*"), chanuser->flags);
  92     }
  93 }
  94
  95 static IRC_CMD(raw_001) {
  96     client->flags |= SOCKET_FLAG_READY;
  97     putsock(client, "JOIN #pktest");
  98     return 1;
  99 }
 100
 101 static IRC_CMD(raw_join) {
 102     if(from == NULL || argc < 1) return 0;
 103     struct UserNode *user = getUserByMask(from);
 104     if(user == NULL) {
 105         user = addUserMask(from);
 106     }
 107     struct ChanNode *chan = getChanByName(argv[0]);
 108     if(chan == NULL) {
 109         chan = addChannel(argv[0]);
 110         //request member list
 111         struct ChanUser *chanuser = addChanUser(chan, user); //it must be a bot
 112         get_userlist(chan, got_channel_userlist, chanuser);
 113     } else if(!isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
 114         struct ChanUser *chanuser = addChanUser(chan, user);
 115         event_join(chanuser);
 116     }
 117     return 1;
 118 }
 119
 120 static IRC_CMD(raw_part) {
 121     if(from == NULL || argc < 2) return 0;
 122     struct UserNode *user = getUserByMask(from);
 123     if(user == NULL) return 0;
 124     struct ChanNode *chan = getChanByName(argv[0]);
 125     if(chan == NULL) return 0;
 126     if(isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
 127         struct ChanUser *chanuser = getChanUser(user, chan);
 128         delChanUser(chanuser, 0); //we need to free the chanuser manually!
 129         event_part(chanuser, argv[1]);
 130         free(chanuser);
 131         if(user->flags & USERFLAG_ISBOT) {
 132             //check if theres another bot in the channel - otherwise free it
 133             checkChannelVisibility(chan);
 134         }
 135     }
 136     if(user->channel == NULL && !(user->flags & USERFLAG_ISBOT)) {
 137         //remove the user
 138         delUser(user, 1);
 139     }
 140     return 1;
 141 }
 142
 143 static IRC_CMD(raw_quit) {
 144     if(from == NULL || argc < 2) return 0;
 145     struct UserNode *user = getUserByMask(from);
 146     if(user == NULL) return 0;
 147     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
 148     event_quit(user, argv[1]);
 149     if(user->flags & USERFLAG_ISBOT) {
 150         //check if there are other bots in the users channel - otherwise free them
 151         struct ChanUser *chanuser, *next;
 152         for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
 153             next = getUserChannels(user, chanuser);
 154             checkChannelVisibility(chanuser->chan);
 155         }
 156     }
 157     delUser(user, 1); //now we fully free the user
 158     return 1;
 159 }
 160
 161 void bot_disconnect(struct ClientSocket *client) {
 162     struct UserNode *user = client->user;
 163     struct ChanUser *chanuser, *next;
 164     delUser(user, 0);
 165     event_quit(user, "disconnected");
 166     for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
 167         next = getUserChannels(user, chanuser);
 168         checkChannelVisibility(chanuser->chan);
 169         free(chanuser);
 170     }
 171     user->channel = NULL;
 172 }
 173
 174 static IRC_CMD(raw_kick) {
 175     if(from == NULL || argc < 3) return 0;
 176     struct UserNode *user = getUserByMask(from);
 177     struct UserNode *target = getUserByNick(argv[1]);
 178     struct ChanNode *chan = getChanByName(argv[0]);
 179     if(chan == NULL || target == NULL) return 0;
 180     if(isUserOnChan(target, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
 181         if(user == NULL) {
 182             user = createTempUser(from);
 183             user->flags |= USERFLAG_ISTMPUSER;
 184         }
 185         struct ChanUser *chanuser = getChanUser(target, chan);
 186         delChanUser(chanuser, 0); //we need to free the chanuser manually!
 187         event_kick(user, chanuser, argv[1]);
 188         free(chanuser);
 189         if(user->flags & USERFLAG_ISTMPUSER) {
 190             free(user);
 191         }
 192         if(target->flags & USERFLAG_ISBOT) {
 193             //check if theres another bot in the channel - otherwise free it
 194             checkChannelVisibility(chan);
 195         }
 196     }
 197     if(target->channel == NULL && !(target->flags & USERFLAG_ISBOT)) {
 198         //remove the user
 199         delUser(user, 1);
 200     }
 201     return 1;
 202 }
 203
 204 static IRC_CMD(raw_topic) {
 205     if(from == NULL || argc < 2) return 0;
 206     struct UserNode *user = getUserByMask(from);
 207     struct ChanNode *chan = getChanByName(argv[0]);
 208     if(chan == NULL) return 0;
 209     if(user == NULL) {
 210         user = createTempUser(from);
 211         user->flags |= USERFLAG_ISTMPUSER;
 212     }
 213     event_topic(user, chan, argv[1]);
 214     strcpy(chan->topic, argv[1]);
 215     if(user->flags & USERFLAG_ISTMPUSER) {
 216         free(user);
 217     }
 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     struct ChanNode *chan = getChanByName(argv[0]);
 225     if(chan == NULL) return 0;
 226     if(user == NULL) {
 227         user = createTempUser(from);
 228         user->flags |= USERFLAG_ISTMPUSER;
 229     }
 230     //TEST
 231     struct ChanUser *chanuser;
 232     putsock(client, "PRIVMSG %s :[PRIVMSG] Users on this Channel:", chan->name);
 233     for(chanuser = getChannelUsers(chan, NULL); chanuser; chanuser = getChannelUsers(chan, chanuser)) {
 234         putsock(client, "PRIVMSG %s :  %s!%s@%s [%s]  rights: %d", chan->name, chanuser->user->nick, chanuser->user->ident, chanuser->user->host, ((chanuser->user->flags & USERFLAG_ISAUTHED) ? chanuser->user->auth : "*"), chanuser->flags);
 235     }
 236     //END TEST
 237     if(user->flags & USERFLAG_ISTMPUSER) {
 238         free(user);
 239     }
 240     return 1;
 241 }
 242
 243 static IRC_CMD(raw_ping) {
 244     if(argc == 0) return 0;
 245     putsock(client, "PONG :%s", argv[0]);
 246     return 1;
 247 }
 248
 249 static IRC_CMD(raw_354) {
 250     recv_whohandler_354(client, argv, argc);
 251     return 1;
 252 }
 253
 254 static IRC_CMD(raw_315) {
 255     recv_whohandler_315(client, argv, argc);
 256     return 1;
 257 }
 258
 259 void parser_init() {
 260     //all the raws we receive...
 261     register_irc_function("001", raw_001);
 262     register_irc_function("TOPIC", raw_topic);
 263     register_irc_function("KICK", raw_kick);
 264     register_irc_function("JOIN", raw_join);
 265     register_irc_function("PART", raw_part);
 266     register_irc_function("QUIT", raw_quit);
 267     register_irc_function("354", raw_354);
 268     register_irc_function("315", raw_315);
 269     register_irc_function("PING", raw_ping);
 270     register_irc_function("PRIVMSG", raw_privmsg);
 271 }