1 /* bots.c - NeonServ v5.5
2 * Copyright (C) 2011-2012 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/>.
20 #include "mysqlConn.h"
21 #include "ClientSocket.h"
28 #include "IRCEvents.h"
30 struct cmd_bot_alias {
33 struct cmd_bot_alias *next;
36 static struct cmd_bot_alias *bot_aliases = NULL;
38 static void start_zero_bots() {
39 struct ClientSocket *client;
40 MYSQL_RES *res, *res2;
43 printf_mysql_query("SELECT `nick`, `ident`, `realname`, `server`, `port`, `pass`, `textbot`, `id`, `queue`, `ssl`, `bind`, `secret` FROM `bots` WHERE `botclass` = '0' AND `active` = '1'");
46 while ((row = mysql_fetch_row(res)) != NULL) {
47 client = create_socket(row[3], atoi(row[4]), row[10], row[5], row[0], row[1], row[2]);
48 client->flags |= (strcmp(row[6], "0") ? SOCKET_FLAG_PREFERRED : 0);
49 client->flags |= (strcmp(row[8], "0") ? SOCKET_FLAG_USE_QUEUE : 0);
50 client->flags |= (strcmp(row[9], "0") ? SOCKET_FLAG_SSL : 0);
51 client->flags |= (strcmp(row[11], "0") ? SOCKET_FLAG_SECRET_BOT : 0);
53 client->clientid = atoi(row[7]);
54 connect_socket(client);
56 printf_mysql_query("SELECT `command`, `function`, `parameters`, `global_access`, `chan_access`, `flags` FROM `bot_binds` WHERE `botclass` = '0' AND `botid` = '%d'", client->clientid);
58 while ((row = mysql_fetch_row(res2)) != NULL) {
59 if(bind_botwise_cmd_to_command(0, client->clientid, row[0], row[1])) {
60 if(row[2] && strcmp(row[2], "")) {
61 bind_botwise_set_parameters(0, client->clientid, row[0], row[2]);
64 bind_botwise_set_global_access(0, client->clientid, row[0], atoi(row[3]));
67 bind_botwise_set_channel_access(0, client->clientid, row[0], row[4]);
69 if(strcmp(row[5], "0"))
70 bind_botwise_set_bind_flags(0, client->clientid, row[0], atoi(row[5]));
73 bind_botwise_unbound_required_functions(0, client->clientid);
77 static void zero_bots_trigger_callback(int clientid, struct ChanNode *chan, char *trigger) {
80 loadChannelSettings(chan);
81 if(!(chan->flags & CHANFLAG_CHAN_REGISTERED)) {
85 printf_mysql_query("SELECT `trigger`, `defaulttrigger` FROM `bot_channels` LEFT JOIN `bots` ON `botid` = `bots`.`id` WHERE `chanid` = '%d' AND `botclass` = '0' AND `botid` = '%d'", chan->channel_id, clientid);
87 if(!(row = mysql_fetch_row(res))) {
92 strcpy(trigger, row[0]);
94 strcpy(trigger, ((row[1] && *row[1]) ? row[1] : "+"));
97 static void zero_bots_bot_ready(struct ClientSocket *client) {
98 if(client->botid != 0)
103 printf_mysql_query("SELECT `automodes`, `oper_user`, `oper_pass` FROM `bots` WHERE `id` = '%d'", client->clientid);
105 if ((row = mysql_fetch_row(res)) != NULL) {
106 if(row[1] && row[2]) {
107 putsock(client, "OPER %s %s", row[1], row[2]);
109 putsock(client, "MODE %s +%s", client->user->nick, row[0]);
112 printf_mysql_query("SELECT `channel_name`, `channel_key` FROM `bot_channels` LEFT JOIN `channels` ON `chanid` = `channel_id` WHERE `botid` = '%d' AND `suspended` = '0'", client->clientid);
115 while ((row = mysql_fetch_row(res)) != NULL) {
116 putsock(client, "JOIN %s %s", row[0], row[1]);
120 static TIMEQ_CALLBACK(load_timed_bans) {
123 //load all timed bans for the next 7 days
124 printf_mysql_query("SELECT `ban_id`, `ban_timeout` FROM `bans` WHERE `ban_timeout` > 0 AND `ban_timeout` < (UNIX_TIMESTAMP() + (86400 * 7))");
127 while ((row = mysql_fetch_row(res)) != NULL) {
128 if(atol(row[1]) - time(0) > 0) {
129 sprintf(nameBuf, "ban_%s", row[0]);
130 timeq_add_name(nameBuf, atol(row[1]) - time(0), 0, channel_ban_timeout, strdup(row[0]));
133 printf_mysql_query("DELETE FROM `bans` WHERE `ban_id` = '%s'", row[0]);
136 timeq_add(86400*7, 0, load_timed_bans, NULL);
140 set_bot_alias(0, "0");
142 set_trigger_callback(0, 0, zero_bots_trigger_callback);
143 bind_bot_ready(zero_bots_bot_ready, 0);
145 timeq_add(10, 0, load_timed_bans, NULL);
149 struct ClientSocket *getChannelBot(struct ChanNode *chan, int botid) {
150 struct ClientSocket *bot, *use_bot = NULL, *second_bot = NULL, *third_bot = NULL;
151 struct ChanUser *chanuser;
152 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
153 if(botid && bot->botid != botid) continue;
154 if((chanuser = getChanUser(bot->user, chan)) != NULL) {
155 if((chanuser->flags & CHANUSERFLAG_OPPED)) {
157 if(bot->flags & SOCKET_FLAG_PREFERRED) break;
158 } else if(bot->flags & SOCKET_FLAG_PREFERRED)
164 if(!use_bot) use_bot = second_bot;
165 if(!use_bot) use_bot = third_bot;
169 void requestOp(struct UserNode *user, struct ChanNode *chan) {
170 struct ClientSocket *bot, *userbot = NULL;
171 struct ChanUser *chanuser = getChanUser(user, chan);
173 if(!chanuser) return;
174 if((chanuser->flags & CHANUSERFLAG_OPPED)) return;
175 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
176 if(!opped && (chanuser = getChanUser(bot->user, chan)) != NULL && (chanuser->flags & CHANUSERFLAG_OPPED)) {
178 putsock(bot, "MODE %s +o %s", chan->name, user->nick);
180 if(bot->user == user) {
185 if(userbot && (isUserModeSet(user, 'o') || isUserModeSet(user, 'O') || isUserModeSet(user, 'k') || isUserModeSet(user, 'X'))) {
186 putsock(userbot, "MODE %s +o %s", chan->name, user->nick);
187 putsock(userbot, "OPMODE %s +o %s", chan->name, user->nick);
192 void requestInvite(struct UserNode *user, struct ChanNode *chan) {
193 struct ClientSocket *bot, *userbot = NULL;
194 struct ChanUser *chanuser = getChanUser(user, chan);
197 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
198 if(!invited && (chanuser = getChanUser(bot->user, chan)) != NULL && (chanuser->flags & CHANUSERFLAG_OPPED)) {
200 putsock(bot, "INVITE %s %s", user->nick, chan->name);
202 if(bot->user == user) {
208 TIMEQ_CALLBACK(channel_ban_timeout) {
209 char *str_banid = data;
212 printf_mysql_query("SELECT `ban_mask`, `channel_name` FROM `bans` LEFT JOIN `channels` ON `ban_channel` = `channel_id` WHERE `ban_id` = '%s'", str_banid);
214 struct ChanNode *chan;
215 if((row = mysql_fetch_row(res)) != NULL && (chan = getChanByName(row[1])) != NULL) {
216 struct ClientSocket *use_bot = getChannelBot(chan, 0);
218 putsock(use_bot, "MODE %s -b %s", chan->name, row[0]);
220 printf_mysql_query("DELETE FROM `bans` WHERE `ban_id` = '%s'", str_banid);
225 static int general_ctcp(char *buffer, char *command, char *text);
227 void general_event_privctcp(struct UserNode *user, struct UserNode *target, char *command, char *text) {
228 char ctcpBuf[MAXLEN];
229 if(general_ctcp(ctcpBuf, command, text)) {
230 struct ClientSocket *bot;
231 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
232 if(bot->user == target) break;
235 putsock(bot, "NOTICE %s :\001%s\001", user->nick, ctcpBuf);
239 static int general_ctcp(char *buffer, char *command, char *text) {
240 if(!stricmp(command, "VERSION")) {
241 sprintf(buffer, "VERSION NeonServ v%s.%d by pk910 (%s)", NEONSERV_VERSION, patchlevel, (strcmp(revision, "") ? revision : "-"));
244 if(!stricmp(command, "FINGER")) {
245 sprintf(buffer, "FINGER NeonServ v%s.%d (%s) build %s lines C code using " COMPILER " (see +netinfo)", NEONSERV_VERSION, patchlevel, (strcmp(revision, "") ? revision : "-"), codelines);
248 if(!stricmp(command, "PING")) {
249 sprintf(buffer, "PING %s", (text ? text : "0"));
252 if(!stricmp(command, "TIME")) {
257 timeinfo = localtime(&rawtime);
258 strftime(timeBuf, 80, "%c", timeinfo);
259 sprintf(buffer, "TIME %s", timeBuf);
265 void set_bot_alias(int botid, char *alias) {
266 struct cmd_bot_alias *botalias, *existing_alias = NULL;
267 for(botalias = bot_aliases; botalias; botalias = botalias->next) {
268 if(!stricmp(botalias->alias, alias))
270 if(botalias->botid == botid)
271 existing_alias = botalias;
274 free(existing_alias->alias);
275 existing_alias->alias = strdup(alias);
278 botalias = malloc(sizeof(*botalias));
280 perror("malloc() failed");
283 botalias->botid = botid;
284 botalias->alias = strdup(alias);
285 botalias->next = bot_aliases;
286 bot_aliases = botalias;
289 const char *resolve_botid(int botid) {
290 struct cmd_bot_alias *botalias;
291 for(botalias = bot_aliases; botalias; botalias = botalias->next) {
292 if(botalias->botid == botid)
293 return botalias->alias;
298 int resolve_botalias(const char *alias) {
299 struct cmd_bot_alias *botalias;
300 for(botalias = bot_aliases; botalias; botalias = botalias->next) {
301 if(!stricmp(botalias->alias, alias))
302 return botalias->botid;