7 #include "ClientSocket.h"
8 #include "WHOHandler.h"
10 struct irc_cmd *irc_commands = NULL;
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);
16 int parse_lines(struct ClientSocket *client, char *lines, int len) {
19 for(i = 0; i < len; i++) {
20 if(lines[i] == '\r') //just zero it out :D
22 if(lines[i] == '\n') {
24 parse_line(client, line);
32 static void parse_line(struct ClientSocket *client, char *line) {
34 char *argv[MAXNUMPARAMS];
35 printf("[recv %lu] %s\n", (unsigned long) strlen(line), line);
45 //the rest is a single parameter
46 argv[argc++] = line + 1;
49 if (argc >= MAXNUMPARAMS)
51 while (*line != ' ' && *line)
55 parse_raw(client, argv[0], argv[1], argv+2, argc-2);
59 static void register_irc_function(char *command, irc_cmd_t *func) {
60 struct irc_cmd *irc_cmd = malloc(sizeof(*irc_cmd));
63 perror("malloc() failed");
66 irc_cmd->cmd = command;
68 irc_cmd->next = irc_commands;
69 irc_commands = irc_cmd;
72 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc) {
73 struct irc_cmd *irc_cmd;
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);
82 event_raw(client, from, cmd, argv, argc);
84 fprintf(stderr,"PARSE ERROR: %s", cmd);
88 static USERLIST_CALLBACK(got_channel_userlist) {
89 struct ChanUser *chanuser = data;
93 static IRC_CMD(raw_001) {
94 client->flags |= SOCKET_FLAG_READY;
95 putsock(client, "JOIN #pktest");
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;
105 user = addUserMask(from);
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);
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]);
132 if(chan->chanbot == user) {
133 //check if theres another bot in the channel - otherwise free it
134 checkChannelVisibility(chan);
137 if(user->channel == NULL && !(user->flags & USERFLAG_ISBOT)) {
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);
158 delUser(user, 1); //now we fully free the user
162 void bot_disconnect(struct ClientSocket *client) {
163 struct UserNode *user = client->user;
164 struct ChanUser *chanuser, *next;
166 event_quit(user, "disconnected");
167 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
168 next = getUserChannels(user, chanuser);
169 checkChannelVisibility(chanuser->chan);
172 user->channel = NULL;
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)) {
183 user = createTempUser(from);
184 user->flags |= USERFLAG_ISTMPUSER;
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]);
190 if(target->flags & USERFLAG_ISBOT) {
191 //check if theres another bot in the channel - otherwise free it
192 checkChannelVisibility(chan);
195 if(target->channel == NULL && !(target->flags & USERFLAG_ISBOT)) {
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
209 user = createTempUser(from);
210 user->flags |= USERFLAG_ISTMPUSER;
212 event_topic(user, chan, argv[1]);
213 strcpy(chan->topic, argv[1]);
217 static IRC_CMD(raw_privmsg) {
218 if(from == NULL || argc < 2) return 0;
219 struct UserNode *user = getUserByMask(from);
221 user = createTempUser(from);
222 user->flags |= USERFLAG_ISTMPUSER;
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, " ");
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);
239 event_chanmsg(user, chan, argv[1]);
242 struct UserNode *target = getUserByNick(argv[0]);
244 if(argv[1][0] == '\001') {
245 char *cmd = &argv[1][1];
246 char *text = strstr(cmd, " ");
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);
256 event_privmsg(user, target, argv[1]);
262 static IRC_CMD(raw_notice) {
263 if(from == NULL || argc < 2) return 0;
264 struct UserNode *user = getUserByMask(from);
266 user = createTempUser(from);
267 user->flags |= USERFLAG_ISTMPUSER;
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]);
274 struct UserNode *target = getUserByNick(argv[0]);
276 event_privnotice(user, target, argv[1]);
281 static IRC_CMD(raw_nick) {
282 if(from == NULL || argc == 0) return 0;
283 struct UserNode *user = getUserByMask(from);
284 event_nick(user, argv[0]);
285 renameUser(user, argv[0]);
289 static IRC_CMD(raw_ping) {
290 if(argc == 0) return 0;
291 putsock(client, "PONG :%s", argv[0]);
295 static IRC_CMD(raw_354) {
296 recv_whohandler_354(client, argv, argc);
300 static IRC_CMD(raw_315) {
301 recv_whohandler_315(client, argv, argc);
305 static IRC_CMD(raw_324) { //MODE LIST
306 //Watchcat #pktest +stnzN
307 if(from == NULL || argc < 3) return 0;
308 struct ChanNode *chan = getChanByName(argv[1]);
309 if(chan == NULL) return 0;
310 parseModes(chan, argv[2], argv+3, argc-3);
314 static IRC_CMD(raw_invite) {
315 if(from == NULL || argc < 2) return 0;
316 struct UserNode *user = getUserByMask(from);
317 event_invite(user, argv[1]);
321 static IRC_CMD(raw_mode) {
322 if(from == NULL || argc < 2) return 0;
323 struct UserNode *user = getUserByMask(from);
325 user = createTempUser(from);
326 user->flags |= USERFLAG_ISTMPUSER;
328 if(argv[0][0] == '#') {
330 struct ChanNode *chan = getChanByName(argv[0]);
332 if(chan->chanbot != client->user) return 1;
333 event_mode(user, chan, argv[1], argv+2, argc-2);
334 parseModes(chan, argv[1], argv+2, argc-2);
342 //all the raws we receive...
343 register_irc_function("001", raw_001);
344 register_irc_function("324", raw_324);
345 register_irc_function("INVITE", raw_invite);
346 register_irc_function("NOTICE", raw_notice);
347 register_irc_function("TOPIC", raw_topic);
348 register_irc_function("KICK", raw_kick);
349 register_irc_function("PART", raw_part);
350 register_irc_function("QUIT", raw_quit);
351 register_irc_function("JOIN", raw_join);
352 register_irc_function("MODE", raw_mode);
353 register_irc_function("NICK", raw_nick);
354 register_irc_function("354", raw_354);
355 register_irc_function("315", raw_315);
356 register_irc_function("PING", raw_ping);
357 register_irc_function("PRIVMSG", raw_privmsg);