src/cmd_neonserv_noregister.c

   1 /* cmd_neonserv_noregister.c - NeonServ v5.3
   2  * Copyright (C) 2011-2012  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]   *auth/#channel
  22 * argv[1]   duration
  23 * argv[2-*] reason
  24 */
  25
  26 static AUTHLOOKUP_CALLBACK(neonserv_cmd_noregister_auth_lookup);
  27 static USERAUTH_CALLBACK(neonserv_cmd_noregister_nick_lookup);
  28 static void neonserv_cmd_noregister_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event,char *auth, int duration, char *reason);
  29 static void neonserv_cmd_noregister_list(struct ClientSocket *client, struct UserNode *user, struct ChanNode *chan, struct Event *event, char **argv, int argc);
  30
  31 struct neonserv_cmd_noregister_cache {
  32     struct ClientSocket *client, *textclient;
  33     struct UserNode *user;
  34     struct ChanNode *chan;
  35     struct Event *event;
  36     int duration;
  37     char *reason, *nick;
  38 };
  39
  40 CMD_BIND(neonserv_cmd_noregister) {
  41     if(argc < 3) {
  42         neonserv_cmd_noregister_list(client, user, chan, event, argv, argc);
  43         return;
  44     }
  45     int duration = strToTime(user, argv[1]);
  46     if(duration == 0 && strcmp(argv[1], "0")) {
  47         reply(getTextBot(), user, "NS_NOREGISTER_INVALID_DURATION", argv[1]);
  48         return;
  49     }
  50     char *reason = merge_argv(argv, 2, argc);
  51     MYSQL_RES *res;
  52     MYSQL_ROW row;
  53     if(argv[0][0] == '#') {
  54         printf_mysql_query("SELECT `botid` FROM `bot_channels` LEFT JOIN `channels` ON `bot_channels`.`chanid` = `channels`.`channel_id` WHERE `channel_name` = '%s'", escape_string(argv[0]));
  55         res = mysql_use();
  56         if ((row = mysql_fetch_row(res)) != NULL) {
  57             reply(getTextBot(), user, "NS_NOREGISTER_REGISTERED", argv[0]);
  58             return;
  59         }
  60         neonserv_cmd_noregister_async1(client, getTextBot(), user, chan, event, argv[0], duration, reason);
  61     } else if(argv[0][0] == '*') {
  62         //we've got an auth
  63         argv[0]++;
  64         printf_mysql_query("SELECT `user_user` FROM `users` WHERE `user_user` = '%s'", escape_string(argv[0]));
  65         res = mysql_use();
  66         if ((row = mysql_fetch_row(res)) != NULL) {
  67             neonserv_cmd_noregister_async1(client, getTextBot(), user, chan, event, row[0], duration, reason);
  68         } else {
  69             //we need to create a new user...
  70             //but first lookup the auth to check if it really exists
  71             struct neonserv_cmd_noregister_cache *cache = malloc(sizeof(*cache));
  72             if (!cache) {
  73                 perror("malloc() failed");
  74                 return;
  75             }
  76             cache->client = client;
  77             cache->textclient = getTextBot();
  78             cache->user = user;
  79             cache->chan = chan;
  80             cache->event = event;
  81             cache->nick = strdup(argv[0]);
  82             cache->duration = duration;
  83             cache->reason = strdup(reason);
  84             lookup_authname(argv[0], neonserv_cmd_noregister_auth_lookup, cache);
  85         }
  86     } else {
  87         struct UserNode *cuser = getUserByNick(argv[0]);
  88         if(!cuser) {
  89             cuser = createTempUser(argv[0]);
  90             cuser->flags |= USERFLAG_ISTMPUSER;
  91         }
  92         if(cuser->flags & USERFLAG_ISAUTHED) {
  93             neonserv_cmd_noregister_async1(client, getTextBot(), user, chan, event, cuser->auth, duration, reason);
  94         } else {
  95             struct neonserv_cmd_noregister_cache *cache = malloc(sizeof(*cache));
  96             if (!cache) {
  97                 perror("malloc() failed");
  98                 return;
  99             }
 100             cache->client = client;
 101             cache->textclient = getTextBot();
 102             cache->user = user;
 103             cache->chan = chan;
 104             cache->event = event;
 105             cache->nick = strdup(user->nick);
 106             cache->duration = duration;
 107             cache->reason = strdup(reason);
 108             get_userauth(cuser, neonserv_cmd_noregister_nick_lookup, cache);
 109         }
 110     }
 111 }
 112
 113 static AUTHLOOKUP_CALLBACK(neonserv_cmd_noregister_auth_lookup) {
 114     struct neonserv_cmd_noregister_cache *cache = data;
 115     if(!exists) {
 116         //AUTH_DOES_NOT_EXIST
 117         reply(cache->textclient, cache->user, "NS_AUTH_UNKNOWN", cache->nick);
 118     } else
 119         neonserv_cmd_noregister_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, auth, cache->duration, cache->reason);
 120     free(cache->reason);
 121     free(cache->nick);
 122     free(cache);
 123 }
 124
 125 static USERAUTH_CALLBACK(neonserv_cmd_noregister_nick_lookup) {
 126     struct neonserv_cmd_noregister_cache *cache = data;
 127     if(!user) {
 128         //USER_DOES_NOT_EXIST
 129         reply(cache->textclient, cache->user, "NS_USER_UNKNOWN", cache->nick);
 130     }
 131     else if(!(user->flags & USERFLAG_ISAUTHED)) {
 132         //USER_NOT_AUTHED
 133         reply(cache->textclient, cache->user, "NS_USER_NEED_AUTH", cache->nick);
 134     }
 135     else
 136         neonserv_cmd_noregister_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, user->auth, cache->duration, cache->reason);
 137     free(cache->reason);
 138     free(cache->nick);
 139     free(cache);
 140 }
 141
 142 static void neonserv_cmd_noregister_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event,char *auth, int duration, char *reason) {
 143     MYSQL_RES *res;
 144     MYSQL_ROW row;
 145     printf_mysql_query("SELECT `dnr_id` FROM `donotregister` WHERE `dnr_target` = '%s'", escape_string(auth));
 146     res = mysql_use();
 147     if((row = mysql_fetch_row(res)) != NULL) {
 148         printf_mysql_query("DELETE FROM `donotregister` WHERE `dnr_id` = '%s'", row[0]);
 149     }
 150     int userid;
 151     printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
 152     res = mysql_use();
 153     if ((row = mysql_fetch_row(res)) != NULL)
 154         userid = atoi(row[0]);
 155     else
 156         userid = 0;
 157     printf_mysql_query("INSERT INTO `donotregister` (`dnr_target`, `dnr_timeout`, `dnr_user`, `dnr_reason`) VALUES ('%s', '%lu', '%d', '%s')", escape_string(auth), (duration ? (time(0)+duration) : 0), userid, escape_string(reason));
 158     reply(getTextBot(), user, "NS_NOREGISTER_DONE", auth);
 159 }
 160
 161 static void neonserv_cmd_noregister_list(struct ClientSocket *client, struct UserNode *user, struct ChanNode *chan, struct Event *event, char **argv, int argc) {
 162     struct Table *table;
 163     int entrys = 0;
 164     MYSQL_RES *res;
 165     MYSQL_ROW row;
 166     printf_mysql_query("SELECT `dnr_target`, `dnr_timeout`, `user_user`, `dnr_reason` FROM `donotregister` LEFT JOIN `users` ON `dnr_user` = `user_id` ORDER BY `dnr_target` ASC");
 167     res = mysql_use();
 168     table = table_init(4, mysql_num_rows(res) + 1, 0);
 169     char *content[4];
 170     content[0] = get_language_string(user, "NS_DNR_TARGET");
 171     content[1] = get_language_string(user, "NS_DNR_USER");
 172     content[2] = get_language_string(user, "NS_DNR_EXPIRES");
 173     content[3] = get_language_string(user, "NS_DNR_REASON");
 174     table_add(table, content);
 175     int duration;
 176     char expires_str[MAXLEN];
 177     while ((row = mysql_fetch_row(res)) != NULL) {
 178         entrys++;
 179         content[0] = row[0];
 180         content[1] = row[2];
 181         duration = atoi(row[1]);
 182         content[2] = (duration ? timeToStr(user, (duration - time(0)), 2, expires_str) : get_language_string(user, "NS_USERS_SEEN_NEVER"));
 183         content[3] = row[3];
 184         table_add(table, content);
 185     }
 186     //send the table
 187     char **table_lines = table_end(table);
 188     int i;
 189     for(i = 0; i < table->entrys; i++) {
 190         reply(getTextBot(), user, table_lines[i]);
 191     }
 192     if(!entrys)
 193         reply(getTextBot(), user, "NS_TABLE_NONE");
 194     table_free(table);
 195 }