cmd_neonserv_register.c

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