src/cmd_neonserv_register.c

   1 /* cmd_neonserv_register.c - NeonServ v5.0
   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_neonserv.h"
  19
  20 /*
  21 * argv[0] - channel
  22 * argv[0/1] - nick / *auth
  23 */
  24 static AUTHLOOKUP_CALLBACK(neonserv_cmd_register_auth_lookup);
  25 static USERAUTH_CALLBACK(neonserv_cmd_register_nick_lookup);
  26 static void neonserv_cmd_register_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event, char *channel, char *auth);
  27
  28 struct neonserv_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 };
  36
  37 CMD_BIND(neonserv_cmd_register) {
  38     MYSQL_RES *res;
  39     MYSQL_ROW row;
  40     char *channel = argv[0];
  41     if(!is_valid_chan(channel)) {
  42         reply(getTextBot(), user, "NS_INVALID_CHANNEL_NAME", argv[0]);
  43         return;
  44     }
  45     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), client->botid);
  46     res = mysql_use();
  47     if ((row = mysql_fetch_row(res)) != NULL) {
  48         reply(getTextBot(), user, "NS_REGISTER_ALREADY", argv[0], client->user->nick);
  49         return;
  50     }
  51     //check own access
  52     if(argv[1][0] == '*') {
  53         //we've got an auth
  54         argv[1]++;
  55         printf_mysql_query("SELECT `user_user` FROM `users` WHERE `user_user` = '%s'", escape_string(argv[1]));
  56         res = mysql_use();
  57         if ((row = mysql_fetch_row(res)) != NULL) {
  58             neonserv_cmd_register_async1(client, getTextBot(), user, chan, event, channel, row[0]);
  59         } else {
  60             //we need to create a new user...
  61             //but first lookup the auth to check if it really exists
  62             struct neonserv_cmd_register_cache *cache = malloc(sizeof(*cache));
  63             if (!cache) {
  64                 perror("malloc() failed");
  65                 return;
  66             }
  67             cache->client = client;
  68             cache->textclient = getTextBot();
  69             cache->user = user;
  70             cache->chan = chan;
  71             cache->event = event;
  72             cache->nick = strdup(argv[1]);
  73             cache->channel = strdup(channel);
  74             lookup_authname(argv[1], neonserv_cmd_register_auth_lookup, cache);
  75         }
  76     } else {
  77         struct UserNode *cuser = getUserByNick(argv[1]);
  78         if(!cuser) {
  79             cuser = createTempUser(argv[1]);
  80             cuser->flags |= USERFLAG_ISTMPUSER;
  81         }
  82         if(cuser->flags & USERFLAG_ISAUTHED) {
  83             neonserv_cmd_register_async1(client, getTextBot(), user, chan, event, channel, cuser->auth);
  84         } else {
  85             struct neonserv_cmd_register_cache *cache = malloc(sizeof(*cache));
  86             if (!cache) {
  87                 perror("malloc() failed");
  88                 return;
  89             }
  90             cache->client = client;
  91             cache->textclient = getTextBot();
  92             cache->user = user;
  93             cache->chan = chan;
  94             cache->event = event;
  95             cache->nick = strdup(argv[1]);
  96             cache->channel = strdup(channel);
  97             get_userauth(cuser, neonserv_cmd_register_nick_lookup, cache);
  98         }
  99     }
 100 }
 101
 102 static AUTHLOOKUP_CALLBACK(neonserv_cmd_register_auth_lookup) {
 103     struct neonserv_cmd_register_cache *cache = data;
 104     if(!exists) {
 105         //AUTH_DOES_NOT_EXIST
 106         reply(cache->textclient, cache->user, "NS_AUTH_UNKNOWN", cache->nick);
 107     } else
 108         neonserv_cmd_register_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, cache->channel, auth);
 109     free(cache->channel);
 110     free(cache->nick);
 111     free(cache);
 112 }
 113
 114 static USERAUTH_CALLBACK(neonserv_cmd_register_nick_lookup) {
 115     struct neonserv_cmd_register_cache *cache = data;
 116     if(!user) {
 117         //USER_DOES_NOT_EXIST
 118         reply(cache->textclient, cache->user, "NS_USER_UNKNOWN", cache->nick);
 119     }
 120     else if(!(user->flags & USERFLAG_ISAUTHED)) {
 121         //USER_NOT_AUTHED
 122         reply(cache->textclient, cache->user, "NS_USER_NEED_AUTH", cache->nick);
 123     }
 124     else
 125         neonserv_cmd_register_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, cache->channel, user->auth);
 126     free(cache->channel);
 127     free(cache->nick);
 128     free(cache);
 129 }
 130
 131 static void neonserv_cmd_register_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event, char *channel, char *auth) {
 132     //we've got a valid auth now...
 133     MYSQL_RES *res;
 134     MYSQL_ROW row, row2;
 135     int userid, adminid;
 136     printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
 137     res = mysql_use();
 138     if ((row = mysql_fetch_row(res)) != NULL)
 139         adminid = atoi(row[0]);
 140     else
 141         adminid = 0;
 142     printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(auth));
 143     res = mysql_use();
 144     if ((row = mysql_fetch_row(res)) != NULL) {
 145         userid = atoi(row[0]);
 146     } else {
 147         printf_mysql_query("INSERT INTO `users` (`user_user`) VALUES ('%s')", escape_string(auth));
 148         userid = (int) mysql_insert_id(mysql_conn);
 149     }
 150     printf_mysql_query("SELECT `id`, `max_channels`, `defaulttrigger` FROM `bots` WHERE `botclass` = '%d' ORDER BY `register_priority` DESC", client->botid);
 151     res = mysql_use();
 152     int botid = 0;
 153     char *bottrigger;
 154     while ((row = mysql_fetch_row(res)) != NULL) {
 155         //check channel count
 156         printf_mysql_query("SELECT COUNT(*) FROM `bot_channels` WHERE `botid` = '%s'", row[0]);
 157         row2 = mysql_fetch_row(mysql_use());
 158         if(atoi(row2[0]) < atoi(row[1])) {
 159             botid = atoi(row[0]);
 160             bottrigger = row[2];
 161             break;
 162         }
 163     }
 164     if(!botid) {
 165         reply(textclient, user, "NS_REGISTER_FULL");
 166         return;
 167     }
 168     int chanid;
 169     printf_mysql_query("SELECT `channel_id` FROM `channels` WHERE `channel_name` = '%s'", escape_string(channel));
 170     res = mysql_use();
 171     if ((row = mysql_fetch_row(res)) != NULL) {
 172         chanid = atoi(row[0]);
 173         printf_mysql_query("UPDATE `channels` SET `channel_registered` = UNIX_TIMESTAMP(), `channel_registrator` = '%d' WHERE `channel_id` = '%d'", adminid, chanid);
 174     } else {
 175         printf_mysql_query("INSERT INTO `channels` (`channel_name`, `channel_registered`, `channel_registrator`) VALUES ('%s', UNIX_TIMESTAMP(), '%d')", escape_string(channel), adminid);
 176         chanid = (int) mysql_insert_id(mysql_conn);
 177     }
 178     struct ClientSocket *bot;
 179     for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 180         if(bot->clientid == botid)
 181             break;
 182     }
 183     if(bot) {
 184         putsock(bot, "JOIN %s", channel);
 185     } else
 186         reply(textclient, user, "NS_REGISTER_DISCONNECTED");
 187     printf_mysql_query("INSERT INTO `bot_channels` (`botid`, `chanid`, `trigger`) VALUES ('%d', '%d', '%s')", botid, chanid, bottrigger);
 188     printf_mysql_query("DELETE FROM `chanusers` WHERE `chanuser_cid` = '%d'", chanid);
 189     printf_mysql_query("INSERT INTO `chanusers` (`chanuser_cid`, `chanuser_uid`, `chanuser_access`) VALUES ('%d', '%d', '%d')", chanid, userid, 500);
 190     reply(textclient, user, "NS_REGISTER_DONE", channel, auth);
 191     logEvent(event);
 192 }