7 #include "ClientSocket.h"
8 #include "WHOHandler.h"
14 struct irc_cmd *irc_commands = NULL;
16 static void parse_line(struct ClientSocket *client, char *line);
17 static void register_irc_function(char *command, irc_cmd_t *func);
18 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc);
20 int parse_lines(struct ClientSocket *client, char *lines, int len) {
23 for(i = 0; i < len; i++) {
24 if(lines[i] == '\r') //just zero it out :D
26 if(lines[i] == '\n') {
28 parse_line(client, line);
36 static void parse_line(struct ClientSocket *client, char *line) {
38 char *argv[MAXNUMPARAMS];
39 printf("[recv %lu] %s\n", (unsigned long) strlen(line), line);
49 //the rest is a single parameter
50 argv[argc++] = line + 1;
54 if (argc >= MAXNUMPARAMS)
56 while (*line != ' ' && *line)
60 parse_raw(client, argv[0], argv[1], argv+2, argc-2);
64 static void register_irc_function(char *command, irc_cmd_t *func) {
65 struct irc_cmd *irc_cmd = malloc(sizeof(*irc_cmd));
68 perror("malloc() failed");
71 irc_cmd->cmd = command;
73 irc_cmd->next = irc_commands;
74 irc_commands = irc_cmd;
77 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc) {
78 struct irc_cmd *irc_cmd;
80 for(irc_cmd = irc_commands; irc_cmd; irc_cmd = irc_cmd->next) {
81 if(!stricmp(irc_cmd->cmd, cmd)) {
82 ret = irc_cmd->func(client, from, argv, argc);
87 event_raw(client, from, cmd, argv, argc);
89 fprintf(stderr,"PARSE ERROR: %s\n", cmd);
93 static USERLIST_CALLBACK(got_channel_userlist) {
94 struct ChanUser *chanuser = data;
98 static IRC_CMD(raw_001) {
99 client->flags |= SOCKET_FLAG_READY;
100 event_bot_ready(client);
104 static IRC_CMD(raw_join) {
105 if(from == NULL || argc < 1) return 0;
106 struct UserNode *user = getUserByMask(from);
107 struct ChanNode *chan = getChanByName(argv[0]);
108 if(!chan && !(user->flags & USERFLAG_ISBOT)) return 0;
110 user = addUserMask(from);
113 chan = addChannel(argv[0]);
114 //request member list
115 chan->chanbot = user;
116 struct ChanUser *chanuser = addChanUser(chan, user); //it must be a bot
117 get_userlist(chan, got_channel_userlist, chanuser);
118 putsock(client, "MODE %s", chan->name);
119 putsock(client, "MODE %s +b", chan->name);
120 } else if(!isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
121 struct ChanUser *chanuser = addChanUser(chan, user);
122 event_join(chanuser);
127 static IRC_CMD(raw_part) {
128 if(from == NULL || argc < 2) return 0;
129 struct UserNode *user = getUserByMask(from);
130 if(user == NULL) return 0;
131 struct ChanNode *chan = getChanByName(argv[0]);
132 if(chan == NULL) return 0;
133 if(isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
134 struct ChanUser *chanuser = getChanUser(user, chan);
135 delChanUser(chanuser, 0); //we need to free the chanuser manually!
136 event_part(chanuser, argv[1]);
138 if(chan->chanbot == user) {
139 //check if theres another bot in the channel - otherwise free it
140 checkChannelVisibility(chan);
143 if(user->channel == NULL && !(user->flags & USERFLAG_ISBOT)) {
150 static IRC_CMD(raw_quit) {
151 if(from == NULL || argc < 1) return 0;
152 struct UserNode *user = getUserByMask(from);
153 if(user == NULL) return 0;
154 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
155 event_quit(user, argv[0]);
156 if(user->flags & USERFLAG_ISBOT) {
157 //check if there are other bots in the users channel - otherwise free them
158 struct ChanUser *chanuser, *next;
159 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
160 next = getUserChannels(user, chanuser);
161 checkChannelVisibility(chanuser->chan);
164 delUser(user, 1); //now we fully free the user
168 void bot_disconnect(struct ClientSocket *client) {
169 struct UserNode *user = client->user;
170 struct ChanUser *chanuser, *next;
172 event_quit(user, "disconnected");
173 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
174 next = getUserChannels(user, chanuser);
175 checkChannelVisibility(chanuser->chan);
178 user->channel = NULL;
181 static IRC_CMD(raw_kick) {
182 if(from == NULL || argc < 3) return 0;
183 struct UserNode *user = getUserByMask(from);
184 struct UserNode *target = getUserByNick(argv[1]);
185 struct ChanNode *chan = getChanByName(argv[0]);
186 if(chan == NULL || target == NULL) return 0;
187 if(isUserOnChan(target, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
189 user = createTempUser(from);
190 user->flags |= USERFLAG_ISTMPUSER;
192 struct ChanUser *chanuser = getChanUser(target, chan);
193 delChanUser(chanuser, 0); //we need to free the chanuser manually!
194 event_kick(user, chanuser, argv[1]);
196 if(target->flags & USERFLAG_ISBOT) {
197 //check if theres another bot in the channel - otherwise free it
198 checkChannelVisibility(chan);
201 if(target->channel == NULL && !(target->flags & USERFLAG_ISBOT)) {
208 static IRC_CMD(raw_topic) {
209 if(from == NULL || argc < 2) return 0;
210 struct UserNode *user = getUserByMask(from);
211 struct ChanNode *chan = getChanByName(argv[0]);
212 if(chan == NULL) return 0;
213 if(chan->chanbot != client->user) return 1; //just ignore it to prevent event duplicates
215 user = createTempUser(from);
216 user->flags |= USERFLAG_ISTMPUSER;
218 event_topic(user, chan, argv[1]);
219 strcpy(chan->topic, argv[1]);
223 static IRC_CMD(raw_privmsg) {
224 if(from == NULL || argc < 2) return 0;
225 struct UserNode *user = getUserByMask(from);
227 user = createTempUser(from);
228 user->flags |= USERFLAG_ISTMPUSER;
230 if(argv[0][0] == '#') { //Channel message
231 struct ChanNode *chan = getChanByName(argv[0]);
232 if(chan && chan->chanbot == client->user) {
233 if(argv[1][0] == '\001') {
234 char *cmd = &argv[1][1];
235 char *text = strstr(cmd, " ");
239 if(strlen(text) && text[strlen(text)-1] == '\001')
240 text[strlen(text)-1] = '\0';
241 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
242 cmd[strlen(cmd)-1] = '\0';
243 event_chanctcp(user, chan, cmd, text);
245 event_chanmsg(user, chan, argv[1]);
248 struct UserNode *target = getUserByNick(argv[0]);
250 if(argv[1][0] == '\001') {
251 char *cmd = &argv[1][1];
252 char *text = strstr(cmd, " ");
256 if(strlen(text) && text[strlen(text)-1] == '\001')
257 text[strlen(text)-1] = '\0';
258 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
259 cmd[strlen(cmd)-1] = '\0';
260 event_privctcp(user, target, cmd, text);
262 event_privmsg(user, target, argv[1]);
268 static IRC_CMD(raw_notice) {
269 if(from == NULL || argc < 2) return 0;
270 struct UserNode *user = getUserByMask(from);
272 user = createTempUser(from);
273 user->flags |= USERFLAG_ISTMPUSER;
275 if(argv[0][0] == '#') { //Channel notice
276 struct ChanNode *chan = getChanByName(argv[0]);
277 if(chan && chan->chanbot == client->user)
278 event_channotice(user, chan, argv[1]);
280 struct UserNode *target = getUserByNick(argv[0]);
282 event_privnotice(user, target, argv[1]);
287 static IRC_CMD(raw_nick) {
288 if(from == NULL || argc == 0) return 0;
289 struct UserNode *user = getUserByMask(from);
290 if(user == NULL) return 0;
291 event_nick(user, argv[0]);
292 renameUser(user, argv[0]);
296 static IRC_CMD(raw_ping) {
297 if(argc == 0) return 0;
298 putsock(client, "PONG :%s", argv[0]);
302 static IRC_CMD(raw_354) {
303 recv_whohandler_354(client, argv, argc);
307 static IRC_CMD(raw_315) {
308 recv_whohandler_315(client, argv, argc);
312 static IRC_CMD(raw_324) { //MODE LIST
313 //Watchcat #pktest +stnzN
314 if(from == NULL || argc < 3) return 0;
315 struct ChanNode *chan = getChanByName(argv[1]);
316 if(chan == NULL) return 0;
317 parseModes(chan->modes, argv[2], argv+3, argc-3);
321 static IRC_CMD(raw_invite) {
322 if(from == NULL || argc < 2) return 0;
323 struct UserNode *user = getUserByMask(from);
324 event_invite(user, argv[1]);
328 static IRC_CMD(raw_mode) {
329 if(from == NULL || argc < 2) return 0;
330 struct UserNode *user = getUserByMask(from);
332 user = createTempUser(from);
333 user->flags |= USERFLAG_ISTMPUSER;
335 if(argv[0][0] == '#') {
337 struct ChanNode *chan = getChanByName(argv[0]);
339 if(chan->chanbot != client->user) return 1;
340 event_mode(user, chan, argv[1], argv+2, argc-2);
341 parseModes(chan->modes, argv[1], argv+2, argc-2);
348 static IRC_CMD(raw_367) {
349 //Watchcat #pktest pk911!*@* TestBot!~bot@pktest.user.WebGamesNet 1315863279
350 struct ChanNode *chan = getChanByName(argv[1]);
353 while((ban = getMatchingChannelBan(chan, argv[2]))) {
354 removeChannelBan(ban);
356 addChannelBan(chan, argv[2]);
361 //all the raws we receive...
362 register_irc_function("001", raw_001);
363 register_irc_function("324", raw_324);
364 register_irc_function("367", raw_367);
365 register_irc_function("INVITE", raw_invite);
366 register_irc_function("NOTICE", raw_notice);
367 register_irc_function("TOPIC", raw_topic);
368 register_irc_function("KICK", raw_kick);
369 register_irc_function("PART", raw_part);
370 register_irc_function("QUIT", raw_quit);
371 register_irc_function("JOIN", raw_join);
372 register_irc_function("MODE", raw_mode);
373 register_irc_function("NICK", raw_nick);
374 register_irc_function("354", raw_354);
375 register_irc_function("315", raw_315);
376 register_irc_function("PING", raw_ping);
377 register_irc_function("PRIVMSG", raw_privmsg);
381 struct irc_cmd *cmd, *next;
382 for(cmd = irc_commands; cmd; cmd = next) {
388 void reply(struct ClientSocket *client, struct UserNode *user, const char *text, ...) {
389 const char *reply_format = get_language_string(user, text);
390 if(reply_format == NULL)
392 loadUserSettings(user);
393 char formatBuf[MAXLEN];
394 sprintf(formatBuf, "%s %s :%s", ((user->flags & USERFLAG_REPLY_PRIVMSG) ? "PRIVMSG" : "NOTICE"), user->nick, reply_format);
396 char sendBuf[MAXLEN];
398 if (!(client->flags & SOCKET_FLAG_CONNECTED)) return;
400 va_start(arg_list, text);
401 pos = vsnprintf(sendBuf, MAXLEN - 2, formatBuf, arg_list);
403 if (pos < 0 || pos > (MAXLEN - 2)) pos = MAXLEN - 2;
405 sendBuf[pos+1] = '\0';
406 write_socket(client, sendBuf, pos+1);
409 char* merge_argv(char **argv, int start, int end) {
410 return merge_argv_char(argv, start, end, ' ');
413 char* merge_argv_char(char **argv, int start, int end, char seperator) {
416 for(i = start; i < end; i++) {