src/modules/global.mod/cmd_global_register.c

   1 /* cmd_global_register.c - NeonServ v5.6
   2  * Copyright (C) 2011-2012  Philipp Kreil (pk910)
   3  *
   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.
   8  *
   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.
  13  *
  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/>.
  16  */
  17
  18 #include "cmd_global.h"
  19
  20 /*
  21 * argv[0] - channel
  22 * argv[1] - nick / *auth
  23 * argv[2] - (optional) bot nick
  24 */
  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);
  28
  29 struct global_cmd_register_cache {
  30     struct ClientSocket *client, *textclient;
  31     struct UserNode *user;
  32     struct ChanNode *chan;
  33     struct Event *event;
  34     char *nick;
  35     char *channel;
  36     char *botname;
  37     int multibot;
  38 };
  39
  40 CMD_BIND(global_cmd_register) {
  41     MYSQL_RES *res;
  42     MYSQL_ROW row;
  43     char *channel = argv[0];
  44     char *botname = (argc > 2 ? argv[2] : NULL);
  45     int multibot = 0;
  46     if(!is_valid_chan(channel)) {
  47         reply(textclient, user, "NS_INVALID_CHANNEL_NAME", argv[0]);
  48         return;
  49     }
  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));
  51     res = mysql_use();
  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);
  55             return;
  56         } else
  57             multibot = 1;
  58     }
  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));
  60     res = mysql_use();
  61     if((row = mysql_fetch_row(res)) != NULL) {
  62         int expire_time = atoi(row[1]);
  63         if(expire_time) {
  64             if(expire_time - time(0) <= 0) {
  65                 printf_mysql_query("DELETE FROM `donotregister` WHERE `dnr_id` = '%s'", row[3]);
  66             } else {
  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]);
  69                 return;
  70             }
  71         } else {
  72             reply(textclient, user, "NS_DNR_SET", channel, row[0], row[2]);
  73             return;
  74         }
  75     }
  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
  79         multibot = 2;
  80         global_cmd_register_async1(client, textclient, user, chan, event, channel, NULL, multibot, botname);
  81         return;
  82     } else if(argc < 2) {
  83         global_cmd_register_async1(client, textclient, user, chan, event, channel, user->auth, multibot, botname);
  84         return;
  85     }
  86     //check own access
  87     if(argv[1][0] == '*') {
  88         //we've got an auth
  89         argv[1]++;
  90         printf_mysql_query("SELECT `user_user` FROM `users` WHERE `user_user` = '%s'", escape_string(argv[1]));
  91         res = mysql_use();
  92         if ((row = mysql_fetch_row(res)) != NULL) {
  93             global_cmd_register_async1(client, textclient, user, chan, event, channel, row[0], multibot, botname);
  94         } else {
  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));
  98             if (!cache) {
  99                 printf_log("global", LOG_ERROR, "%s:%d malloc() failed", __FILE__, __LINE__);
 100                 return;
 101             }
 102             cache->client = client;
 103             cache->textclient = textclient;
 104             cache->user = user;
 105             cache->chan = chan;
 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);
 112         }
 113     } else {
 114         struct UserNode *cuser = getUserByNick(argv[1]);
 115         if(!cuser) {
 116             cuser = createTempUser(argv[1]);
 117             if(!cuser) {
 118                 reply(textclient, user, "NS_USER_UNKNOWN", argv[1]);
 119                 return;
 120             }
 121             cuser->flags |= USERFLAG_ISTMPUSER;
 122         }
 123         if(cuser->flags & USERFLAG_ISAUTHED) {
 124             global_cmd_register_async1(client, textclient, user, chan, event, channel, cuser->auth, multibot, botname);
 125         } else {
 126             struct global_cmd_register_cache *cache = malloc(sizeof(*cache));
 127             if (!cache) {
 128                 printf_log("global", LOG_ERROR, "%s:%d malloc() failed", __FILE__, __LINE__);
 129                 return;
 130             }
 131             cache->client = client;
 132             cache->textclient = textclient;
 133             cache->user = user;
 134             cache->chan = chan;
 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);
 141         }
 142     }
 143 }
 144
 145 static AUTHLOOKUP_CALLBACK(global_cmd_register_auth_lookup) {
 146     struct global_cmd_register_cache *cache = data;
 147     if(!exists) {
 148         //AUTH_DOES_NOT_EXIST
 149         reply(cache->textclient, cache->user, "NS_AUTH_UNKNOWN", cache->nick);
 150     } else
 151         global_cmd_register_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, cache->channel, auth, cache->multibot, cache->botname);
 152     if(cache->botname)
 153         free(cache->botname);
 154     free(cache->channel);
 155     free(cache->nick);
 156     free(cache);
 157 }
 158
 159 static USERAUTH_CALLBACK(global_cmd_register_nick_lookup) {
 160     struct global_cmd_register_cache *cache = data;
 161     if(!user) {
 162         //USER_DOES_NOT_EXIST
 163         reply(cache->textclient, cache->user, "NS_USER_UNKNOWN", cache->nick);
 164     }
 165     else if(!(user->flags & USERFLAG_ISAUTHED)) {
 166         //USER_NOT_AUTHED
 167         reply(cache->textclient, cache->user, "NS_USER_NEED_AUTH", cache->nick);
 168     }
 169     else
 170         global_cmd_register_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, cache->channel, user->auth, cache->multibot, cache->botname);
 171     if(cache->botname)
 172         free(cache->botname);
 173     free(cache->channel);
 174     free(cache->nick);
 175     free(cache);
 176 }
 177
 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...
 180     MYSQL_RES *res;
 181     MYSQL_ROW row, row2;
 182     int userid = 0, adminid;
 183     printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
 184     res = mysql_use();
 185     if ((row = mysql_fetch_row(res)) != NULL)
 186         adminid = atoi(row[0]);
 187     else
 188         adminid = 0;
 189     if(multibot != 2) {
 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));
 191         res = mysql_use();
 192         if((row = mysql_fetch_row(res)) != NULL) {
 193             int expire_time = atoi(row[1]);
 194             if(expire_time) {
 195                 if(expire_time - time(0) <= 0) {
 196                     printf_mysql_query("DELETE FROM `donotregister` WHERE `dnr_id` = '%s'", row[3]);
 197                 } else {
 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]);
 200                     return;
 201                 }
 202             } else {
 203                 reply(textclient, user, "NS_DNR_SET", auth, row[0], row[2]);
 204                 return;
 205             }
 206         }
 207         printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(auth));
 208         res = mysql_use();
 209         if ((row = mysql_fetch_row(res)) != NULL) {
 210             userid = atoi(row[0]);
 211         } else {
 212             printf_mysql_query("INSERT INTO `users` (`user_user`) VALUES ('%s')", escape_string(auth));
 213             userid = (int) mysql_insert_id(get_mysql_conn());
 214         }
 215     }
 216     if(client->botid)
 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);
 218     else
 219         printf_mysql_query("SELECT `id`, `max_channels`, `defaulttrigger`, `nick` FROM `bots` WHERE `id` = '%d' AND `active` = '1'", client->clientid);
 220     res = mysql_use();
 221     int botid = 0;
 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]);
 228             break;
 229         }
 230     }
 231     if(!botid) {
 232         reply(textclient, user, "NS_REGISTER_FULL");
 233         return;
 234     }
 235     int chanid;
 236     printf_mysql_query("SELECT `channel_id` FROM `channels` WHERE `channel_name` = '%s'", escape_string(channel));
 237     res = mysql_use();
 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);
 241     } else {
 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());
 244     }
 245     struct ClientSocket *bot;
 246     for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 247         if(bot->clientid == botid)
 248             break;
 249     }
 250     if(bot) {
 251         putsock(bot, "JOIN %s", channel);
 252     } else
 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);
 255     if(multibot != 2) {
 256         if(multibot) {
 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);
 259         } else
 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);
 263     } else
 264         reply(textclient, user, "NS_REGISTER_DONE_NOAUTH", channel);
 265     // NeonBackup SubBlock
 266     if(client->botid == NEONSERV_BOTID) {
 267         char setting[128];
 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);
 271     }
 272     logEvent(event);
 273 }
 274
 275 void global_cmd_register_neonbackup(char *channel) {
 276     MYSQL_RES *res;
 277     MYSQL_ROW row, row2;
 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);
 279     res = mysql_use();
 280     if ((row = mysql_fetch_row(res)) != NULL)
 281         return;
 282     int chanid;
 283     printf_mysql_query("SELECT `channel_id` FROM `channels` WHERE `channel_name` = '%s'", escape_string(channel));
 284     res = mysql_use();
 285     if ((row = mysql_fetch_row(res)) != NULL) {
 286         chanid = atoi(row[0]);
 287     } else
 288         return;
 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);
 290     res = mysql_use();
 291     int botid = 0;
 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]);
 298             break;
 299         }
 300     }
 301     if(!botid)
 302         return;
 303     struct ClientSocket *bot;
 304     for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 305         if(bot->clientid == botid)
 306             break;
 307     }
 308     if(bot) {
 309         putsock(bot, "JOIN %s", channel);
 310     }
 311     printf_mysql_query("INSERT INTO `bot_channels` (`botid`, `chanid`, `trigger`) VALUES ('%d', '%d', NULL)", botid, chanid);
 312 }
 313