7 #include "ClientSocket.h"
8 #include "WHOHandler.h"
12 struct irc_cmd *irc_commands = NULL;
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);
18 int parse_lines(struct ClientSocket *client, char *lines, int len) {
21 for(i = 0; i < len; i++) {
22 if(lines[i] == '\r') //just zero it out :D
24 if(lines[i] == '\n') {
26 parse_line(client, line);
34 static void parse_line(struct ClientSocket *client, char *line) {
36 char *argv[MAXNUMPARAMS];
37 printf("[recv %lu] %s\n", (unsigned long) strlen(line), line);
47 //the rest is a single parameter
48 argv[argc++] = line + 1;
52 if (argc >= MAXNUMPARAMS)
54 while (*line != ' ' && *line)
58 parse_raw(client, argv[0], argv[1], argv+2, argc-2);
62 static void register_irc_function(char *command, irc_cmd_t *func) {
63 struct irc_cmd *irc_cmd = malloc(sizeof(*irc_cmd));
66 perror("malloc() failed");
69 irc_cmd->cmd = command;
71 irc_cmd->next = irc_commands;
72 irc_commands = irc_cmd;
75 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc) {
76 struct irc_cmd *irc_cmd;
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);
85 event_raw(client, from, cmd, argv, argc);
87 fprintf(stderr,"PARSE ERROR: %s", cmd);
91 static USERLIST_CALLBACK(got_channel_userlist) {
92 struct ChanUser *chanuser = data;
96 static IRC_CMD(raw_001) {
97 client->flags |= SOCKET_FLAG_READY;
98 event_bot_ready(client);
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;
108 user = addUserMask(from);
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 } else if(!isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
118 struct ChanUser *chanuser = addChanUser(chan, user);
119 event_join(chanuser);
124 static IRC_CMD(raw_part) {
125 if(from == NULL || argc < 2) return 0;
126 struct UserNode *user = getUserByMask(from);
127 if(user == NULL) return 0;
128 struct ChanNode *chan = getChanByName(argv[0]);
129 if(chan == NULL) return 0;
130 if(isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
131 struct ChanUser *chanuser = getChanUser(user, chan);
132 delChanUser(chanuser, 0); //we need to free the chanuser manually!
133 event_part(chanuser, argv[1]);
135 if(chan->chanbot == user) {
136 //check if theres another bot in the channel - otherwise free it
137 checkChannelVisibility(chan);
140 if(user->channel == NULL && !(user->flags & USERFLAG_ISBOT)) {
147 static IRC_CMD(raw_quit) {
148 if(from == NULL || argc < 2) return 0;
149 struct UserNode *user = getUserByMask(from);
150 if(user == NULL) return 0;
151 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
152 event_quit(user, argv[1]);
153 if(user->flags & USERFLAG_ISBOT) {
154 //check if there are other bots in the users channel - otherwise free them
155 struct ChanUser *chanuser, *next;
156 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
157 next = getUserChannels(user, chanuser);
158 checkChannelVisibility(chanuser->chan);
161 delUser(user, 1); //now we fully free the user
165 void bot_disconnect(struct ClientSocket *client) {
166 struct UserNode *user = client->user;
167 struct ChanUser *chanuser, *next;
169 event_quit(user, "disconnected");
170 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
171 next = getUserChannels(user, chanuser);
172 checkChannelVisibility(chanuser->chan);
175 user->channel = NULL;
178 static IRC_CMD(raw_kick) {
179 if(from == NULL || argc < 3) return 0;
180 struct UserNode *user = getUserByMask(from);
181 struct UserNode *target = getUserByNick(argv[1]);
182 struct ChanNode *chan = getChanByName(argv[0]);
183 if(chan == NULL || target == NULL) return 0;
184 if(isUserOnChan(target, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
186 user = createTempUser(from);
187 user->flags |= USERFLAG_ISTMPUSER;
189 struct ChanUser *chanuser = getChanUser(target, chan);
190 delChanUser(chanuser, 0); //we need to free the chanuser manually!
191 event_kick(user, chanuser, argv[1]);
193 if(target->flags & USERFLAG_ISBOT) {
194 //check if theres another bot in the channel - otherwise free it
195 checkChannelVisibility(chan);
198 if(target->channel == NULL && !(target->flags & USERFLAG_ISBOT)) {
205 static IRC_CMD(raw_topic) {
206 if(from == NULL || argc < 2) return 0;
207 struct UserNode *user = getUserByMask(from);
208 struct ChanNode *chan = getChanByName(argv[0]);
209 if(chan == NULL) return 0;
210 if(chan->chanbot != client->user) return 1; //just ignore it to prevent event duplicates
212 user = createTempUser(from);
213 user->flags |= USERFLAG_ISTMPUSER;
215 event_topic(user, chan, argv[1]);
216 strcpy(chan->topic, argv[1]);
220 static IRC_CMD(raw_privmsg) {
221 if(from == NULL || argc < 2) return 0;
222 struct UserNode *user = getUserByMask(from);
224 user = createTempUser(from);
225 user->flags |= USERFLAG_ISTMPUSER;
227 if(argv[0][0] == '#') { //Channel message
228 struct ChanNode *chan = getChanByName(argv[0]);
229 if(chan && chan->chanbot == client->user) {
230 if(argv[1][0] == '\001') {
231 char *cmd = &argv[1][1];
232 char *text = strstr(cmd, " ");
236 if(strlen(text) && text[strlen(text)-1] == '\001')
237 text[strlen(text)-1] = '\0';
238 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
239 cmd[strlen(cmd)-1] = '\0';
240 event_chanctcp(user, chan, cmd, text);
242 event_chanmsg(user, chan, argv[1]);
245 struct UserNode *target = getUserByNick(argv[0]);
247 if(argv[1][0] == '\001') {
248 char *cmd = &argv[1][1];
249 char *text = strstr(cmd, " ");
253 if(strlen(text) && text[strlen(text)-1] == '\001')
254 text[strlen(text)-1] = '\0';
255 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
256 cmd[strlen(cmd)-1] = '\0';
257 event_privctcp(user, target, cmd, text);
259 event_privmsg(user, target, argv[1]);
265 static IRC_CMD(raw_notice) {
266 if(from == NULL || argc < 2) return 0;
267 struct UserNode *user = getUserByMask(from);
269 user = createTempUser(from);
270 user->flags |= USERFLAG_ISTMPUSER;
272 if(argv[0][0] == '#') { //Channel notice
273 struct ChanNode *chan = getChanByName(argv[0]);
274 if(chan && chan->chanbot == client->user)
275 event_channotice(user, chan, argv[1]);
277 struct UserNode *target = getUserByNick(argv[0]);
279 event_privnotice(user, target, argv[1]);
284 static IRC_CMD(raw_nick) {
285 if(from == NULL || argc == 0) return 0;
286 struct UserNode *user = getUserByMask(from);
287 if(user == NULL) return 0;
288 event_nick(user, argv[0]);
289 renameUser(user, argv[0]);
293 static IRC_CMD(raw_ping) {
294 if(argc == 0) return 0;
295 putsock(client, "PONG :%s", argv[0]);
299 static IRC_CMD(raw_354) {
300 recv_whohandler_354(client, argv, argc);
304 static IRC_CMD(raw_315) {
305 recv_whohandler_315(client, argv, argc);
309 static IRC_CMD(raw_324) { //MODE LIST
310 //Watchcat #pktest +stnzN
311 if(from == NULL || argc < 3) return 0;
312 struct ChanNode *chan = getChanByName(argv[1]);
313 if(chan == NULL) return 0;
314 parseModes(chan, argv[2], argv+3, argc-3);
318 static IRC_CMD(raw_invite) {
319 if(from == NULL || argc < 2) return 0;
320 struct UserNode *user = getUserByMask(from);
321 event_invite(user, argv[1]);
325 static IRC_CMD(raw_mode) {
326 if(from == NULL || argc < 2) return 0;
327 struct UserNode *user = getUserByMask(from);
329 user = createTempUser(from);
330 user->flags |= USERFLAG_ISTMPUSER;
332 if(argv[0][0] == '#') {
334 struct ChanNode *chan = getChanByName(argv[0]);
336 if(chan->chanbot != client->user) return 1;
337 event_mode(user, chan, argv[1], argv+2, argc-2);
338 parseModes(chan, argv[1], argv+2, argc-2);
346 //all the raws we receive...
347 register_irc_function("001", raw_001);
348 register_irc_function("324", raw_324);
349 register_irc_function("INVITE", raw_invite);
350 register_irc_function("NOTICE", raw_notice);
351 register_irc_function("TOPIC", raw_topic);
352 register_irc_function("KICK", raw_kick);
353 register_irc_function("PART", raw_part);
354 register_irc_function("QUIT", raw_quit);
355 register_irc_function("JOIN", raw_join);
356 register_irc_function("MODE", raw_mode);
357 register_irc_function("NICK", raw_nick);
358 register_irc_function("354", raw_354);
359 register_irc_function("315", raw_315);
360 register_irc_function("PING", raw_ping);
361 register_irc_function("PRIVMSG", raw_privmsg);
365 struct irc_cmd *cmd, *next;
366 for(cmd = irc_commands; cmd; cmd = next) {
372 void reply(struct ClientSocket *client, struct UserNode *user, const char *text, ...) {
373 const char *reply_format = get_language_string(user, text);
374 if(reply_format == NULL)
376 loadUserSettings(user);
377 char formatBuf[MAXLEN];
378 sprintf(formatBuf, "%s %s :%s", ((user->flags & USERFLAG_REPLY_PRIVMSG) ? "PRIVMSG" : "NOTICE"), user->nick, reply_format);
380 char sendBuf[MAXLEN];
382 if (!(client->flags & SOCKET_FLAG_CONNECTED)) return;
384 va_start(arg_list, text);
385 pos = vsnprintf(sendBuf, MAXLEN - 2, formatBuf, arg_list);
387 if (pos < 0 || pos > (MAXLEN - 2)) pos = MAXLEN - 2;
389 sendBuf[pos+1] = '\0';
390 write_socket(client, sendBuf, pos+1);
393 char* merge_argv(char **argv, int start, int end) {
394 return merge_argv_char(argv, start, end, ' ');
397 char* merge_argv_char(char **argv, int start, int end, char seperator) {
400 for(i = start; i < end; i++) {