src/cmd_global_register.c

   1 /* cmd_global_register.c - NeonServ v5.2
   2  * Copyright (C) 2011  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[0/1] - nick / *auth
  23 */
  24 static AUTHLOOKUP_CALLBACK(global_cmd_register_auth_lookup);
  25 static USERAUTH_CALLBACK(global_cmd_register_nick_lookup);
  26 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);
  27
  28 struct global_cmd_register_cache {
  29     struct ClientSocket *client, *textclient;
  30     struct UserNode *user;
  31     struct ChanNode *chan;
  32     struct Event *event;
  33     char *nick;
  34     char *channel;
  35     int multibot;
  36 };
  37
  38 CMD_BIND(global_cmd_register) {
  39     MYSQL_RES *res;
  40     MYSQL_ROW row;
  41     char *channel = argv[0];
  42     int multibot = 0;
  43     if(!is_valid_chan(channel)) {
  44         reply(getTextBot(), user, "NS_INVALID_CHANNEL_NAME", argv[0]);
  45         return;
  46     }
  47     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));
  48     res = mysql_use();
  49     while ((row = mysql_fetch_row(res)) != NULL) {
  50         if(atoi(row[1]) == client->botid) {
  51             reply(getTextBot(), user, "NS_REGISTER_ALREADY", argv[0], client->user->nick);
  52             return;
  53         } else
  54             multibot = 1;
  55     }
  56     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));
  57     res = mysql_use();
  58     if((row = mysql_fetch_row(res)) != NULL) {
  59         int expire_time = atoi(row[1]);
  60         if(expire_time) {
  61             if(expire_time - time(0) <= 0) {
  62                 printf_mysql_query("DELETE FROM `donotregister` WHERE `dnr_id` = '%s'", row[3]);
  63             } else {
  64                 char expireBuf[MAXLEN];
  65                 reply(getTextBot(), user, "NS_DNR_SET_EXPIRES", channel, row[0], timeToStr(user, (expire_time - time(0)), 2, expireBuf), row[2]);
  66                 return;
  67             }
  68         } else {
  69             reply(getTextBot(), user, "NS_DNR_SET", channel, row[0], row[2]);
  70             return;
  71         }
  72     }
  73     //if theres already another bot in the channel we don't need a owner parameter...
  74     if(multibot && argc < 2) {
  75         //skip all these owner check lines
  76         multibot = 2;
  77         global_cmd_register_async1(client, getTextBot(), user, chan, event, channel, NULL, multibot);
  78         return;
  79     } else if(argc < 2) {
  80         reply(getTextBot(), user, "MODCMD_LESS_PARAM_COUNT");
  81         return;
  82     }
  83     //check own access
  84     if(argv[1][0] == '*') {
  85         //we've got an auth
  86         argv[1]++;
  87         printf_mysql_query("SELECT `user_user` FROM `users` WHERE `user_user` = '%s'", escape_string(argv[1]));
  88         res = mysql_use();
  89         if ((row = mysql_fetch_row(res)) != NULL) {
  90             global_cmd_register_async1(client, getTextBot(), user, chan, event, channel, row[0], multibot);
  91         } else {
  92             //we need to create a new user...
  93             //but first lookup the auth to check if it really exists
  94             struct global_cmd_register_cache *cache = malloc(sizeof(*cache));
  95             if (!cache) {
  96                 perror("malloc() failed");
  97                 return;
  98             }
  99             cache->client = client;
 100             cache->textclient = getTextBot();
 101             cache->user = user;
 102             cache->chan = chan;
 103             cache->event = event;
 104             cache->nick = strdup(argv[1]);
 105             cache->channel = strdup(channel);
 106             cache->multibot = multibot;
 107             lookup_authname(argv[1], global_cmd_register_auth_lookup, cache);
 108         }
 109     } else {
 110         struct UserNode *cuser = getUserByNick(argv[1]);
 111         if(!cuser) {
 112             cuser = createTempUser(argv[1]);
 113             cuser->flags |= USERFLAG_ISTMPUSER;
 114         }
 115         if(cuser->flags & USERFLAG_ISAUTHED) {
 116             global_cmd_register_async1(client, getTextBot(), user, chan, event, channel, cuser->auth, multibot);
 117         } else {
 118             struct global_cmd_register_cache *cache = malloc(sizeof(*cache));
 119             if (!cache) {
 120                 perror("malloc() failed");
 121                 return;
 122             }
 123             cache->client = client;
 124             cache->textclient = getTextBot();
 125             cache->user = user;
 126             cache->chan = chan;
 127             cache->event = event;
 128             cache->nick = strdup(argv[1]);
 129             cache->channel = strdup(channel);
 130             cache->multibot = multibot;
 131             get_userauth(cuser, global_cmd_register_nick_lookup, cache);
 132         }
 133     }
 134 }
 135
 136 static AUTHLOOKUP_CALLBACK(global_cmd_register_auth_lookup) {
 137     struct global_cmd_register_cache *cache = data;
 138     if(!exists) {
 139         //AUTH_DOES_NOT_EXIST
 140         reply(cache->textclient, cache->user, "NS_AUTH_UNKNOWN", cache->nick);
 141     } else
 142         global_cmd_register_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, cache->channel, auth, cache->multibot);
 143     free(cache->channel);
 144     free(cache->nick);
 145     free(cache);
 146 }
 147
 148 static USERAUTH_CALLBACK(global_cmd_register_nick_lookup) {
 149     struct global_cmd_register_cache *cache = data;
 150     if(!user) {
 151         //USER_DOES_NOT_EXIST
 152         reply(cache->textclient, cache->user, "NS_USER_UNKNOWN", cache->nick);
 153     }
 154     else if(!(user->flags & USERFLAG_ISAUTHED)) {
 155         //USER_NOT_AUTHED
 156         reply(cache->textclient, cache->user, "NS_USER_NEED_AUTH", cache->nick);
 157     }
 158     else
 159         global_cmd_register_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, cache->channel, user->auth, cache->multibot);
 160     free(cache->channel);
 161     free(cache->nick);
 162     free(cache);
 163 }
 164
 165 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) {
 166     //we've got a valid auth now...
 167     MYSQL_RES *res;
 168     MYSQL_ROW row, row2;
 169     int userid = 0, adminid;
 170     printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
 171     res = mysql_use();
 172     if ((row = mysql_fetch_row(res)) != NULL)
 173         adminid = atoi(row[0]);
 174     else
 175         adminid = 0;
 176     if(multibot != 2) {
 177         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));
 178         res = mysql_use();
 179         if((row = mysql_fetch_row(res)) != NULL) {
 180             int expire_time = atoi(row[1]);
 181             if(expire_time) {
 182                 if(expire_time - time(0) <= 0) {
 183                     printf_mysql_query("DELETE FROM `donotregister` WHERE `dnr_id` = '%s'", row[3]);
 184                 } else {
 185                     char expireBuf[MAXLEN];
 186                     reply(getTextBot(), user, "NS_DNR_SET_EXPIRES", auth, row[0], timeToStr(user, (expire_time - time(0)), 2, expireBuf), row[2]);
 187                     return;
 188                 }
 189             } else {
 190                 reply(getTextBot(), user, "NS_DNR_SET", auth, row[0], row[2]);
 191                 return;
 192             }
 193         }
 194         printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(auth));
 195         res = mysql_use();
 196         if ((row = mysql_fetch_row(res)) != NULL) {
 197             userid = atoi(row[0]);
 198         } else {
 199             printf_mysql_query("INSERT INTO `users` (`user_user`) VALUES ('%s')", escape_string(auth));
 200             userid = (int) mysql_insert_id(mysql_conn);
 201         }
 202     }
 203     printf_mysql_query("SELECT `id`, `max_channels`, `defaulttrigger` FROM `bots` WHERE `botclass` = '%d' ORDER BY `register_priority` DESC", client->botid);
 204     res = mysql_use();
 205     int botid = 0;
 206     char *bottrigger;
 207     while ((row = mysql_fetch_row(res)) != NULL) {
 208         //check channel count
 209         printf_mysql_query("SELECT COUNT(*) FROM `bot_channels` WHERE `botid` = '%s'", row[0]);
 210         row2 = mysql_fetch_row(mysql_use());
 211         if(atoi(row2[0]) < atoi(row[1])) {
 212             botid = atoi(row[0]);
 213             bottrigger = row[2];
 214             break;
 215         }
 216     }
 217     if(!botid) {
 218         reply(textclient, user, "NS_REGISTER_FULL");
 219         return;
 220     }
 221     int chanid;
 222     printf_mysql_query("SELECT `channel_id` FROM `channels` WHERE `channel_name` = '%s'", escape_string(channel));
 223     res = mysql_use();
 224     if ((row = mysql_fetch_row(res)) != NULL) {
 225         chanid = atoi(row[0]);
 226         printf_mysql_query("UPDATE `channels` SET `channel_registered` = UNIX_TIMESTAMP(), `channel_registrator` = '%d' WHERE `channel_id` = '%d'", adminid, chanid);
 227     } else {
 228         printf_mysql_query("INSERT INTO `channels` (`channel_name`, `channel_registered`, `channel_registrator`) VALUES ('%s', UNIX_TIMESTAMP(), '%d')", escape_string(channel), adminid);
 229         chanid = (int) mysql_insert_id(mysql_conn);
 230     }
 231     struct ClientSocket *bot;
 232     for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 233         if(bot->clientid == botid)
 234             break;
 235     }
 236     if(bot) {
 237         putsock(bot, "JOIN %s", channel);
 238     } else
 239         reply(textclient, user, "NS_REGISTER_DISCONNECTED");
 240     printf_mysql_query("INSERT INTO `bot_channels` (`botid`, `chanid`, `trigger`) VALUES ('%d', '%d', '%s')", botid, chanid, bottrigger);
 241     if(multibot != 2) {
 242         if(multibot) {
 243             printf_mysql_query("UPDATE `chanusers` SET `chanuser_access` = 499 WHERE `chanuser_cid` = '%d' AND `chanuser_access` = '500'", chanid);
 244             printf_mysql_query("DELETE FROM `chanusers` WHERE `chanuser_cid` = '%d' AND `chanuser_uid` = '%d'", chanid, userid);
 245         } else
 246             printf_mysql_query("DELETE FROM `chanusers` WHERE `chanuser_cid` = '%d'", chanid);
 247         printf_mysql_query("INSERT INTO `chanusers` (`chanuser_cid`, `chanuser_uid`, `chanuser_access`) VALUES ('%d', '%d', '%d')", chanid, userid, 500);
 248         reply(textclient, user, "NS_REGISTER_DONE", channel, auth);
 249     } else
 250         reply(textclient, user, "NS_REGISTER_DONE_NOAUTH", channel);
 251     logEvent(event);
 252 }