cmd_neonserv_adduser.c

   1
   2 /*
   3 * argv[0] - nick / *auth
   4 * argv[1] - chan access
   5 */
   6 static AUTHLOOKUP_CALLBACK(neonserv_cmd_adduser_auth_lookup);
   7 static USERAUTH_CALLBACK(neonserv_cmd_adduser_nick_lookup);
   8 static void neonserv_cmd_adduser_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, char *nick, char *auth, int access);
   9
  10 struct neonserv_cmd_adduser_cache {
  11     struct ClientSocket *client, *textclient;
  12     struct UserNode *user;
  13     struct ChanNode *chan;
  14     int access;
  15     char *nick;
  16 };
  17
  18 static CMD_BIND(neonserv_cmd_adduser) {
  19     int caccess;
  20     MYSQL_RES *res;
  21     MYSQL_ROW row;
  22     caccess = atoi(argv[1]);
  23     if(caccess <= 0 || caccess > 500) {
  24         reply(getTextBot(), user, "NS_INVALID_ACCESS", caccess);
  25         return;
  26     }
  27     if(caccess >= getChannelAccess(user, chan, 1)) {
  28         reply(getTextBot(), user, "NS_ACCESS_OUTRANKED");
  29         return;
  30     }
  31     //check own access
  32     if(argv[0][0] == '*') {
  33         //we've got an auth
  34         argv[0]++;
  35         printf_mysql_query("SELECT `user_user` FROM `users` WHERE `user_user` = '%s'", escape_string(argv[0]));
  36         res = mysql_use();
  37         if ((row = mysql_fetch_row(res)) != NULL) {
  38             neonserv_cmd_adduser_async1(client, getTextBot(), user, chan, argv[0], row[0], caccess);
  39         } else {
  40             //we need to create a new user...
  41             //but first lookup the auth to check if it really exists
  42             struct neonserv_cmd_adduser_cache *cache = malloc(sizeof(*cache));
  43             if (!cache) {
  44                 perror("malloc() failed");
  45                 return;
  46             }
  47             cache->client = client;
  48             cache->textclient = getTextBot();
  49             cache->user = user;
  50             cache->chan = chan;
  51             cache->access = caccess;
  52             cache->nick = strdup(argv[0]);
  53             lookup_authname(argv[0], neonserv_cmd_adduser_auth_lookup, cache);
  54         }
  55     } else {
  56         struct UserNode *cuser = getUserByNick(argv[0]);
  57         if(!cuser) {
  58             cuser = createTempUser(argv[0]);
  59             cuser->flags |= USERFLAG_ISTMPUSER;
  60         }
  61         if(cuser->flags & USERFLAG_ISAUTHED) {
  62             neonserv_cmd_adduser_async1(client, getTextBot(), user, chan, argv[0], cuser->auth, caccess);
  63         } else {
  64             struct neonserv_cmd_adduser_cache *cache = malloc(sizeof(*cache));
  65             if (!cache) {
  66                 perror("malloc() failed");
  67                 return;
  68             }
  69             cache->client = client;
  70             cache->textclient = getTextBot();
  71             cache->user = user;
  72             cache->chan = chan;
  73             cache->access = caccess;
  74             cache->nick = strdup(argv[0]);
  75             get_userauth(cuser, neonserv_cmd_adduser_nick_lookup, cache);
  76         }
  77     }
  78 }
  79
  80 static AUTHLOOKUP_CALLBACK(neonserv_cmd_adduser_auth_lookup) {
  81     struct neonserv_cmd_adduser_cache *cache = data;
  82     if(!exists) {
  83         //AUTH_DOES_NOT_EXIST
  84         reply(cache->textclient, cache->user, "NS_AUTH_UNKNOWN", cache->nick);
  85     } else
  86         neonserv_cmd_adduser_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->nick, auth, cache->access);
  87     free(cache->nick);
  88     free(cache);
  89 }
  90
  91 static USERAUTH_CALLBACK(neonserv_cmd_adduser_nick_lookup) {
  92     struct neonserv_cmd_adduser_cache *cache = data;
  93     if(!user) {
  94         //USER_DOES_NOT_EXIST
  95         reply(cache->textclient, cache->user, "NS_USER_UNKNOWN", cache->nick);
  96     }
  97     else if(!(user->flags & USERFLAG_ISAUTHED)) {
  98         //USER_NOT_AUTHED
  99         reply(cache->textclient, cache->user, "NS_USER_NEED_AUTH", cache->nick);
 100     }
 101     else
 102         neonserv_cmd_adduser_async1(cache->client, cache->textclient, cache->user, cache->chan, user->nick, user->auth, cache->access);
 103     free(cache->nick);
 104     free(cache);
 105 }
 106
 107 static void neonserv_cmd_adduser_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, char *nick, char *auth, int caccess) {
 108     //we've got a valid auth now...
 109     MYSQL_RES *res;
 110     MYSQL_ROW row;
 111     int userid;
 112     printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(auth));
 113     res = mysql_use();
 114     if ((row = mysql_fetch_row(res)) != NULL) {
 115         userid = atoi(row[0]);
 116         //check if the user is already added
 117         printf_mysql_query("SELECT `chanuser_access` FROM `chanusers` WHERE `chanuser_cid` = '%d' AND `chanuser_uid` = '%d'", chan->channel_id, userid);
 118         res = mysql_use();
 119         if ((row = mysql_fetch_row(res)) != NULL) {
 120             reply(textclient, user, "NS_ADDUSER_ALREADY_ADDED", nick, chan->name, atoi(row[0]));
 121             return;
 122         }
 123     } else {
 124         printf_mysql_query("INSERT INTO `users` (`user_user`) VALUES ('%s')", escape_string(auth));
 125         userid = (int) mysql_insert_id(mysql_conn);
 126     }
 127     printf_mysql_query("INSERT INTO `chanusers` (`chanuser_cid`, `chanuser_uid`, `chanuser_access`) VALUES ('%d', '%d', '%d')", chan->channel_id, userid, caccess);
 128     reply(textclient, user, "NS_ADDUSER_DONE", nick, chan->name, caccess);
 129 }