DBHelper.c

   1
   2 #include "DBHelper.h"
   3 #include "UserNode.h"
   4 #include "ChanNode.h"
   5 #include "ChanUser.h"
   6 #include "mysqlConn.h"
   7 #include "lang.h"
   8
   9
  10 void _loadUserSettings(struct UserNode *user) {
  11     check_mysql();
  12     MYSQL_RES *res;
  13     MYSQL_ROW row;
  14     printf_mysql_query("SELECT `user_lang`, `user_reply_privmsg`, `user_god` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
  15     res = mysql_use();
  16     if ((row = mysql_fetch_row(res)) != NULL) {
  17         user->language = get_language_by_tag(row[0]);
  18         if(user->language == NULL) user->language = get_default_language();
  19         if(strcmp(row[1], "0"))
  20             user->flags |= USERFLAG_REPLY_PRIVMSG;
  21         if(strcmp(row[2], "0"))
  22             user->flags |= USERFLAG_GOD_MODE;
  23     } else
  24         user->language = get_default_language();
  25     user->flags |= USERFLAG_LOADED_SETTINGS;
  26 }
  27
  28 int isGodMode(struct UserNode *user) {
  29     loadUserSettings(user);
  30     return (user->flags & USERFLAG_GOD_MODE);
  31 }
  32
  33 int getChannelAccess(struct UserNode *user, struct ChanNode *chan, int override) {
  34     if(!(user->flags & USERFLAG_ISAUTHED)) return 0;
  35     loadChannelSettings(chan);
  36     if(!(chan->flags & CHANFLAG_CHAN_REGISTERED)) return 0;
  37     MYSQL_RES *res;
  38     MYSQL_ROW row;
  39     check_mysql();
  40     int caccess = 0;
  41     printf_mysql_query("SELECT `user_id`, `user_access`, `user_god` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
  42     res = mysql_use();
  43     if ((row = mysql_fetch_row(res)) != NULL) {
  44         if(strcmp(row[2], "0") && override)
  45             caccess = atoi(row[1]);
  46         printf_mysql_query("SELECT `chanuser_access`, `chanuser_flags` FROM `chanusers` WHERE `chanuser_uid` = '%s' AND `chanuser_cid` = '%d'", row[0], chan->channel_id);
  47         //
  48         res = mysql_use();
  49         if ((row = mysql_fetch_row(res)) != NULL) {
  50             if(!(atoi(row[1]) & DB_CHANUSER_SUSPENDED) && atoi(row[0]) > caccess)
  51                 caccess = atoi(row[0]);
  52         }
  53         return caccess;
  54     }
  55     return 0;
  56 }
  57
  58 int checkChannelAccess(struct UserNode *user, struct ChanNode *chan, char *channel_setting, int allow_override, int allow_501) {
  59     loadChannelSettings(chan);
  60     if(!(chan->flags & CHANFLAG_CHAN_REGISTERED)) return 0;
  61     MYSQL_RES *res;
  62     MYSQL_ROW row;
  63     printf_mysql_query("SELECT `%s` FROM `channels` WHERE `channel_id` = '%d'", channel_setting, chan->channel_id);
  64     res = mysql_use();
  65     if ((row = mysql_fetch_row(res)) == NULL) return 0;
  66     int require_access = atoi(row[0]);
  67     if(require_access == 0) return 1;
  68     if(!(user->flags & USERFLAG_ISAUTHED)) return 0;
  69     int caccess = 0;
  70     printf_mysql_query("SELECT `user_id`, `user_access`, `user_god` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
  71     res = mysql_use();
  72     if ((row = mysql_fetch_row(res)) != NULL) {
  73         printf_mysql_query("SELECT `chanuser_access`, `chanuser_flags` FROM `chanusers` WHERE `chanuser_uid` = '%s' AND `chanuser_cid` = '%d'", row[0], chan->channel_id);
  74         res = mysql_use();
  75         if ((row = mysql_fetch_row(res)) != NULL) {
  76             if(!(atoi(row[1]) & DB_CHANUSER_SUSPENDED))
  77                 caccess = atoi(row[0]);
  78         }
  79     }
  80     if(caccess >= require_access) return 1;
  81     if(caccess == 500 && require_access == 501 && allow_501) return 1;
  82     return 0;
  83 }
  84
  85 void _loadChannelSettings(struct ChanNode *chan) {
  86     check_mysql();
  87     MYSQL_RES *res;
  88     MYSQL_ROW row;
  89     printf_mysql_query("SELECT `channel_id` FROM `channels` WHERE `channel_name` = '%s'", escape_string(chan->name));
  90     res = mysql_use();
  91     if ((row = mysql_fetch_row(res)) != NULL) {
  92         chan->flags |= CHANFLAG_CHAN_REGISTERED;
  93         chan->channel_id = atoi(row[0]);
  94     }
  95     chan->flags |= CHANFLAG_REQUESTED_CHANINFO;
  96 }
  97
  98