1 /* IRCParser.c - NeonServ v5.0
2 * Copyright (C) 2011 Philipp Kreil (pk910)
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 #include "IRCParser.h"
22 #include "IRCEvents.h"
23 #include "ClientSocket.h"
24 #include "WHOHandler.h"
30 struct irc_cmd *irc_commands = NULL;
32 static void parse_line(struct ClientSocket *client, char *line);
33 static void register_irc_function(char *command, irc_cmd_t *func);
34 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc);
36 int parse_lines(struct ClientSocket *client, char *lines, int len) {
39 for(i = 0; i < len; i++) {
40 if(lines[i] == '\r') //just zero it out :D
42 if(lines[i] == '\n') {
44 parse_line(client, line);
52 static void parse_line(struct ClientSocket *client, char *line) {
54 char *argv[MAXNUMPARAMS];
55 printf("[recv %lu] %s\n", (unsigned long) strlen(line), line);
65 //the rest is a single parameter
66 argv[argc++] = line + 1;
70 if (argc >= MAXNUMPARAMS)
72 while (*line != ' ' && *line)
76 parse_raw(client, argv[0], argv[1], argv+2, argc-2);
80 static void register_irc_function(char *command, irc_cmd_t *func) {
81 struct irc_cmd *irc_cmd = malloc(sizeof(*irc_cmd));
84 perror("malloc() failed");
87 irc_cmd->cmd = command;
89 irc_cmd->next = irc_commands;
90 irc_commands = irc_cmd;
93 static void parse_raw(struct ClientSocket *client, char *from, char *cmd, char **argv, int argc) {
94 struct irc_cmd *irc_cmd;
96 for(irc_cmd = irc_commands; irc_cmd; irc_cmd = irc_cmd->next) {
97 if(!stricmp(irc_cmd->cmd, cmd)) {
98 ret = irc_cmd->func(client, from, argv, argc);
103 event_raw(client, from, cmd, argv, argc);
105 fprintf(stderr,"PARSE ERROR: %s\n", cmd);
109 static USERLIST_CALLBACK(got_channel_userlist) {
110 struct ChanUser *chanuser = data;
111 event_join(chanuser);
114 static IRC_CMD(raw_001) {
115 client->flags |= SOCKET_FLAG_READY;
116 event_bot_ready(client);
120 static IRC_CMD(raw_join) {
121 if(from == NULL || argc < 1) return 0;
122 struct UserNode *user = getUserByMask(from);
123 struct ChanNode *chan = getChanByName(argv[0]);
124 if(!chan && !(user->flags & USERFLAG_ISBOT)) return 0;
126 user = addUserMask(from);
129 chan = addChannel(argv[0]);
130 //request member list
131 chan->chanbot = user;
132 struct ChanUser *chanuser = addChanUser(chan, user); //it must be a bot
133 get_userlist(chan, got_channel_userlist, chanuser);
134 putsock(client, "MODE %s", chan->name);
135 putsock(client, "MODE %s +b", chan->name);
136 } else if(!isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
137 struct ChanUser *chanuser = addChanUser(chan, user);
138 event_join(chanuser);
143 static IRC_CMD(raw_part) {
144 if(from == NULL || argc < 1) return 0;
145 struct UserNode *user = getUserByMask(from);
146 if(user == NULL) return 0;
147 struct ChanNode *chan = getChanByName(argv[0]);
148 if(chan == NULL) return 0;
149 if(isUserOnChan(user, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
150 struct ChanUser *chanuser = getChanUser(user, chan);
151 delChanUser(chanuser, 0); //we need to free the chanuser manually!
152 event_part(chanuser, (argc > 1 ? argv[1] : NULL));
154 if(chan->chanbot == user) {
155 //check if theres another bot in the channel - otherwise free it
156 checkChannelVisibility(chan);
159 if(user->channel == NULL && !(user->flags & USERFLAG_ISBOT)) {
166 static IRC_CMD(raw_quit) {
167 if(from == NULL || argc < 1) return 0;
168 struct UserNode *user = getUserByMask(from);
169 if(user == NULL) return 0;
170 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
171 event_quit(user, argv[0]);
172 if(user->flags & USERFLAG_ISBOT) {
173 //check if there are other bots in the users channel - otherwise free them
174 struct ChanUser *chanuser, *next;
175 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
176 next = getUserChannels(user, chanuser);
177 checkChannelVisibility(chanuser->chan);
180 delUser(user, 1); //now we fully free the user
184 void bot_disconnect(struct ClientSocket *client) {
185 struct UserNode *user = client->user;
186 struct ChanUser *chanuser, *next;
188 event_quit(user, "disconnected");
189 for(chanuser = getUserChannels(user, NULL); chanuser; chanuser = next) {
190 next = getUserChannels(user, chanuser);
191 checkChannelVisibility(chanuser->chan);
194 user->channel = NULL;
197 static IRC_CMD(raw_kick) {
198 if(from == NULL || argc < 3) return 0;
199 struct UserNode *user = getUserByMask(from);
200 struct UserNode *target = getUserByNick(argv[1]);
201 struct ChanNode *chan = getChanByName(argv[0]);
202 if(chan == NULL || target == NULL) return 0;
203 if(isUserOnChan(target, chan) && (chan->flags & CHANFLAG_RECEIVED_USERLIST)) {
205 user = createTempUser(from);
206 user->flags |= USERFLAG_ISTMPUSER;
208 struct ChanUser *chanuser = getChanUser(target, chan);
209 delChanUser(chanuser, 0); //we need to free the chanuser manually!
210 event_kick(user, chanuser, argv[1]);
212 if(target->flags & USERFLAG_ISBOT) {
213 //check if theres another bot in the channel - otherwise free it
214 checkChannelVisibility(chan);
217 if(target->channel == NULL && !(target->flags & USERFLAG_ISBOT)) {
224 static IRC_CMD(raw_topic) {
225 if(from == NULL || argc < 2) return 0;
226 struct UserNode *user = getUserByMask(from);
227 struct ChanNode *chan = getChanByName(argv[0]);
228 if(chan == NULL) return 0;
229 if(chan->chanbot != client->user) return 1; //just ignore it to prevent event duplicates
231 user = createTempUser(from);
232 user->flags |= USERFLAG_ISTMPUSER;
234 event_topic(user, chan, argv[1]);
235 strcpy(chan->topic, argv[1]);
239 static IRC_CMD(raw_privmsg) {
240 if(from == NULL || argc < 2) return 0;
241 struct UserNode *user = getUserByMask(from);
243 user = createTempUser(from);
244 user->flags |= USERFLAG_ISTMPUSER;
246 if(argv[0][0] == '#') { //Channel message
247 struct ChanNode *chan = getChanByName(argv[0]);
248 if(chan && chan->chanbot == client->user) {
249 if(argv[1][0] == '\001') {
250 char *cmd = &argv[1][1];
251 char *text = strstr(cmd, " ");
255 if(strlen(text) && text[strlen(text)-1] == '\001')
256 text[strlen(text)-1] = '\0';
257 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
258 cmd[strlen(cmd)-1] = '\0';
259 event_chanctcp(user, chan, cmd, text);
261 event_chanmsg(user, chan, argv[1]);
264 struct UserNode *target = getUserByNick(argv[0]);
266 if(argv[1][0] == '\001') {
267 char *cmd = &argv[1][1];
268 char *text = strstr(cmd, " ");
272 if(strlen(text) && text[strlen(text)-1] == '\001')
273 text[strlen(text)-1] = '\0';
274 } else if(strlen(cmd) && cmd[strlen(cmd)-1] == '\001')
275 cmd[strlen(cmd)-1] = '\0';
276 event_privctcp(user, target, cmd, text);
278 event_privmsg(user, target, argv[1]);
284 static IRC_CMD(raw_notice) {
285 if(from == NULL || argc < 2) return 0;
286 struct UserNode *user = getUserByMask(from);
288 user = createTempUser(from);
289 user->flags |= USERFLAG_ISTMPUSER;
291 if(argv[0][0] == '#') { //Channel notice
292 struct ChanNode *chan = getChanByName(argv[0]);
293 if(chan && chan->chanbot == client->user)
294 event_channotice(user, chan, argv[1]);
296 struct UserNode *target = getUserByNick(argv[0]);
298 event_privnotice(user, target, argv[1]);
303 static IRC_CMD(raw_nick) {
304 if(from == NULL || argc == 0) return 0;
305 struct UserNode *user = getUserByMask(from);
306 if(user == NULL) return 0;
307 event_nick(user, argv[0]);
308 renameUser(user, argv[0]);
312 static IRC_CMD(raw_ping) {
313 if(argc == 0) return 0;
314 putsock(client, "PONG :%s", argv[0]);
318 static IRC_CMD(raw_354) {
319 recv_whohandler_354(client, argv, argc);
323 static IRC_CMD(raw_315) {
324 recv_whohandler_315(client, argv, argc);
328 static IRC_CMD(raw_324) { //MODE LIST
329 //Watchcat #pktest +stnzN
330 if(from == NULL || argc < 3) return 0;
331 struct ChanNode *chan = getChanByName(argv[1]);
332 if(chan == NULL) return 0;
333 parseModes(chan->modes, argv[2], argv+3, argc-3);
337 static IRC_CMD(raw_invite) {
338 if(from == NULL || argc < 2) return 0;
339 struct UserNode *user = getUserByMask(from);
341 user = createTempUser(from);
342 user->flags |= USERFLAG_ISTMPUSER;
344 event_invite(client, user, argv[1]);
348 static IRC_CMD(raw_mode) {
349 if(from == NULL || argc < 2) return 0;
350 struct UserNode *user = getUserByMask(from);
352 user = createTempUser(from);
353 user->flags |= USERFLAG_ISTMPUSER;
355 if(argv[0][0] == '#') {
357 struct ChanNode *chan = getChanByName(argv[0]);
359 if(chan->chanbot != client->user) return 1;
360 event_mode(user, chan, argv[1], argv+2, argc-2);
361 parseModes(chan->modes, argv[1], argv+2, argc-2);
368 static IRC_CMD(raw_367) {
369 //Watchcat #pktest pk911!*@* TestBot!~bot@pktest.user.WebGamesNet 1315863279
370 struct ChanNode *chan = getChanByName(argv[1]);
373 while((ban = getMatchingChannelBan(chan, argv[2]))) {
374 removeChannelBan(ban);
376 addChannelBan(chan, argv[2]);
381 //all the raws we receive...
382 register_irc_function("001", raw_001);
383 register_irc_function("324", raw_324);
384 register_irc_function("367", raw_367);
385 register_irc_function("INVITE", raw_invite);
386 register_irc_function("NOTICE", raw_notice);
387 register_irc_function("TOPIC", raw_topic);
388 register_irc_function("KICK", raw_kick);
389 register_irc_function("PART", raw_part);
390 register_irc_function("QUIT", raw_quit);
391 register_irc_function("JOIN", raw_join);
392 register_irc_function("MODE", raw_mode);
393 register_irc_function("NICK", raw_nick);
394 register_irc_function("354", raw_354);
395 register_irc_function("315", raw_315);
396 register_irc_function("PING", raw_ping);
397 register_irc_function("PRIVMSG", raw_privmsg);
401 struct irc_cmd *cmd, *next;
402 for(cmd = irc_commands; cmd; cmd = next) {
408 void reply(struct ClientSocket *client, struct UserNode *user, const char *text, ...) {
409 const char *reply_format = get_language_string(user, text);
410 if(reply_format == NULL)
412 loadUserSettings(user);
413 char formatBuf[MAXLEN];
414 sprintf(formatBuf, "%s %s :%s", ((user->flags & USERFLAG_REPLY_PRIVMSG) ? "PRIVMSG" : "NOTICE"), user->nick, reply_format);
416 char sendBuf[MAXLEN];
418 if (!(client->flags & SOCKET_FLAG_CONNECTED)) return;
420 va_start(arg_list, text);
421 pos = vsnprintf(sendBuf, MAXLEN - 2, formatBuf, arg_list);
423 if (pos < 0 || pos > (MAXLEN - 2)) pos = MAXLEN - 2;
425 sendBuf[pos+1] = '\0';
426 write_socket(client, sendBuf, pos+1);
429 char* merge_argv(char **argv, int start, int end) {
430 return merge_argv_char(argv, start, end, ' ');
433 char* merge_argv_char(char **argv, int start, int end, char seperator) {
436 for(i = start; i < end; i++) {