1 /* cmd_global_register.c - NeonServ v5.6
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/>.
18 #include "cmd_global.h"
22 * argv[1] - nick / *auth
23 * argv[2] - (optional) bot nick
25 static AUTHLOOKUP_CALLBACK(global_cmd_register_auth_lookup);
26 static USERAUTH_CALLBACK(global_cmd_register_nick_lookup);
27 static void global_cmd_register_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event, char *channel, char *auth, int multibot, char *botname);
29 struct global_cmd_register_cache {
30 struct ClientSocket *client, *textclient;
31 struct UserNode *user;
32 struct ChanNode *chan;
40 CMD_BIND(global_cmd_register) {
43 char *channel = argv[0];
44 char *botname = (argc > 2 ? argv[2] : NULL);
46 if(!is_valid_chan(channel)) {
47 reply(textclient, user, "NS_INVALID_CHANNEL_NAME", argv[0]);
50 printf_mysql_query("SELECT `botid`, `botclass` FROM `bot_channels` LEFT JOIN `bots` ON `bot_channels`.`botid` = `bots`.`id` LEFT JOIN `channels` ON `bot_channels`.`chanid` = `channels`.`channel_id` WHERE `channel_name` = '%s'", escape_string(channel));
52 while ((row = mysql_fetch_row(res)) != NULL) {
53 if(atoi(row[1]) == client->botid && (client->botid || client->clientid == atoi(row[0]))) {
54 reply(textclient, user, "NS_REGISTER_ALREADY", argv[0], client->user->nick);
59 printf_mysql_query("SELECT `user_user`, `dnr_timeout`, `dnr_reason`, `dnr_id` FROM `donotregister` LEFT JOIN `users` ON `dnr_user` = `user_id` WHERE `dnr_target` = '%s'", escape_string(channel));
61 if((row = mysql_fetch_row(res)) != NULL) {
62 int expire_time = atoi(row[1]);
64 if(expire_time - time(0) <= 0) {
65 printf_mysql_query("DELETE FROM `donotregister` WHERE `dnr_id` = '%s'", row[3]);
67 char expireBuf[MAXLEN];
68 reply(textclient, user, "NS_DNR_SET_EXPIRES", channel, row[0], timeToStr(user, (expire_time - time(0)), 2, expireBuf), row[2]);
72 reply(textclient, user, "NS_DNR_SET", channel, row[0], row[2]);
76 //if theres already another bot in the channel we don't need a owner parameter...
77 if(multibot && argc < 2) {
78 //skip all these owner check lines
80 global_cmd_register_async1(client, textclient, user, chan, event, channel, NULL, multibot, botname);
83 global_cmd_register_async1(client, textclient, user, chan, event, channel, user->auth, multibot, botname);
87 if(argv[1][0] == '*') {
90 printf_mysql_query("SELECT `user_user` FROM `users` WHERE `user_user` = '%s'", escape_string(argv[1]));
92 if ((row = mysql_fetch_row(res)) != NULL) {
93 global_cmd_register_async1(client, textclient, user, chan, event, channel, row[0], multibot, botname);
95 //we need to create a new user...
96 //but first lookup the auth to check if it really exists
97 struct global_cmd_register_cache *cache = malloc(sizeof(*cache));
99 perror("malloc() failed");
102 cache->client = client;
103 cache->textclient = textclient;
106 cache->event = event;
107 cache->nick = strdup(argv[1]);
108 cache->channel = strdup(channel);
109 cache->multibot = multibot;
110 cache->botname = (botname ? strdup(botname) : NULL);
111 lookup_authname(argv[1], module_id, global_cmd_register_auth_lookup, cache);
114 struct UserNode *cuser = getUserByNick(argv[1]);
116 cuser = createTempUser(argv[1]);
118 reply(textclient, user, "NS_USER_UNKNOWN", argv[1]);
121 cuser->flags |= USERFLAG_ISTMPUSER;
123 if(cuser->flags & USERFLAG_ISAUTHED) {
124 global_cmd_register_async1(client, textclient, user, chan, event, channel, cuser->auth, multibot, botname);
126 struct global_cmd_register_cache *cache = malloc(sizeof(*cache));
128 perror("malloc() failed");
131 cache->client = client;
132 cache->textclient = textclient;
135 cache->event = event;
136 cache->nick = strdup(argv[1]);
137 cache->channel = strdup(channel);
138 cache->multibot = multibot;
139 cache->botname = (botname ? strdup(botname) : NULL);
140 get_userauth(cuser, module_id, global_cmd_register_nick_lookup, cache);
145 static AUTHLOOKUP_CALLBACK(global_cmd_register_auth_lookup) {
146 struct global_cmd_register_cache *cache = data;
148 //AUTH_DOES_NOT_EXIST
149 reply(cache->textclient, cache->user, "NS_AUTH_UNKNOWN", cache->nick);
151 global_cmd_register_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, cache->channel, auth, cache->multibot, cache->botname);
153 free(cache->botname);
154 free(cache->channel);
159 static USERAUTH_CALLBACK(global_cmd_register_nick_lookup) {
160 struct global_cmd_register_cache *cache = data;
162 //USER_DOES_NOT_EXIST
163 reply(cache->textclient, cache->user, "NS_USER_UNKNOWN", cache->nick);
165 else if(!(user->flags & USERFLAG_ISAUTHED)) {
167 reply(cache->textclient, cache->user, "NS_USER_NEED_AUTH", cache->nick);
170 global_cmd_register_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, cache->channel, user->auth, cache->multibot, cache->botname);
172 free(cache->botname);
173 free(cache->channel);
178 static void global_cmd_register_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event, char *channel, char *auth, int multibot, char *botname) {
179 //we've got a valid auth now...
182 int userid = 0, adminid;
183 printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
185 if ((row = mysql_fetch_row(res)) != NULL)
186 adminid = atoi(row[0]);
190 printf_mysql_query("SELECT `user_user`, `dnr_timeout`, `dnr_reason`, `dnr_id` FROM `donotregister` LEFT JOIN `users` ON `dnr_user` = `user_id` WHERE `dnr_target` = '%s'", escape_string(auth));
192 if((row = mysql_fetch_row(res)) != NULL) {
193 int expire_time = atoi(row[1]);
195 if(expire_time - time(0) <= 0) {
196 printf_mysql_query("DELETE FROM `donotregister` WHERE `dnr_id` = '%s'", row[3]);
198 char expireBuf[MAXLEN];
199 reply(textclient, user, "NS_DNR_SET_EXPIRES", auth, row[0], timeToStr(user, (expire_time - time(0)), 2, expireBuf), row[2]);
203 reply(textclient, user, "NS_DNR_SET", auth, row[0], row[2]);
207 printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(auth));
209 if ((row = mysql_fetch_row(res)) != NULL) {
210 userid = atoi(row[0]);
212 printf_mysql_query("INSERT INTO `users` (`user_user`) VALUES ('%s')", escape_string(auth));
213 userid = (int) mysql_insert_id(get_mysql_conn());
217 printf_mysql_query("SELECT `id`, `max_channels`, `defaulttrigger`, `nick` FROM `bots` WHERE `botclass` = '%d' AND `active` = '1' ORDER BY `register_priority` DESC", client->botid);
219 printf_mysql_query("SELECT `id`, `max_channels`, `defaulttrigger`, `nick` FROM `bots` WHERE `id` = '%d' AND `active` = '1'", client->clientid);
222 while ((row = mysql_fetch_row(res)) != NULL) {
223 //check channel count
224 printf_mysql_query("SELECT COUNT(*) FROM `bot_channels` WHERE `botid` = '%s'", row[0]);
225 row2 = mysql_fetch_row(mysql_use());
226 if(atoi(row2[0]) < atoi(row[1]) && (!botname || !stricmp(botname, row[3]))) {
227 botid = atoi(row[0]);
232 reply(textclient, user, "NS_REGISTER_FULL");
236 printf_mysql_query("SELECT `channel_id` FROM `channels` WHERE `channel_name` = '%s'", escape_string(channel));
238 if ((row = mysql_fetch_row(res)) != NULL) {
239 chanid = atoi(row[0]);
240 printf_mysql_query("UPDATE `channels` SET `channel_registered` = UNIX_TIMESTAMP(), `channel_registrator` = '%d' WHERE `channel_id` = '%d'", adminid, chanid);
242 printf_mysql_query("INSERT INTO `channels` (`channel_name`, `channel_registered`, `channel_registrator`) VALUES ('%s', UNIX_TIMESTAMP(), '%d')", escape_string(channel), adminid);
243 chanid = (int) mysql_insert_id(get_mysql_conn());
245 struct ClientSocket *bot;
246 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
247 if(bot->clientid == botid)
251 putsock(bot, "JOIN %s", channel);
253 reply(textclient, user, "NS_REGISTER_DISCONNECTED", channel);
254 printf_mysql_query("INSERT INTO `bot_channels` (`botid`, `chanid`, `trigger`) VALUES ('%d', '%d', NULL)", botid, chanid);
257 printf_mysql_query("UPDATE `chanusers` SET `chanuser_access` = 499 WHERE `chanuser_cid` = '%d' AND `chanuser_access` = '500'", chanid);
258 printf_mysql_query("DELETE FROM `chanusers` WHERE `chanuser_cid` = '%d' AND `chanuser_uid` = '%d'", chanid, userid);
260 printf_mysql_query("DELETE FROM `chanusers` WHERE `chanuser_cid` = '%d'", chanid);
261 printf_mysql_query("INSERT INTO `chanusers` (`chanuser_cid`, `chanuser_uid`, `chanuser_access`) VALUES ('%d', '%d', '%d')", chanid, userid, 500);
262 reply(textclient, user, "NS_REGISTER_DONE", channel, auth);
264 reply(textclient, user, "NS_REGISTER_DONE_NOAUTH", channel);
265 // NeonBackup SubBlock
266 if(client->botid == NEONSERV_BOTID) {
268 sprintf(setting, "modules.%s.auto_backup_register", get_module_name(module_id));
269 if(get_int_field(setting))
270 module_global_cmd_register_neonbackup(channel);
275 void global_cmd_register_neonbackup(char *channel) {
278 printf_mysql_query("SELECT `botid` FROM `bot_channels` LEFT JOIN `bots` ON `bot_channels`.`botid` = `bots`.`id` LEFT JOIN `channels` ON `bot_channels`.`chanid` = `channels`.`channel_id` WHERE `channel_name` = '%s' AND botclass = '%d'", escape_string(channel), NEONBACKUP_BOTID);
280 if ((row = mysql_fetch_row(res)) != NULL)
283 printf_mysql_query("SELECT `channel_id` FROM `channels` WHERE `channel_name` = '%s'", escape_string(channel));
285 if ((row = mysql_fetch_row(res)) != NULL) {
286 chanid = atoi(row[0]);
289 printf_mysql_query("SELECT `id`, `max_channels`, `defaulttrigger`, `nick` FROM `bots` WHERE `botclass` = '%d' AND `active` = '1' ORDER BY `register_priority` DESC", NEONBACKUP_BOTID);
292 while ((row = mysql_fetch_row(res)) != NULL) {
293 //check channel count
294 printf_mysql_query("SELECT COUNT(*) FROM `bot_channels` WHERE `botid` = '%s'", row[0]);
295 row2 = mysql_fetch_row(mysql_use());
296 if(atoi(row2[0]) < atoi(row[1])) {
297 botid = atoi(row[0]);
303 struct ClientSocket *bot;
304 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
305 if(bot->clientid == botid)
309 putsock(bot, "JOIN %s", channel);
311 printf_mysql_query("INSERT INTO `bot_channels` (`botid`, `chanid`, `trigger`) VALUES ('%d', '%d', NULL)", botid, chanid);