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