From: Maksym Mamontov Date: Mon, 1 Feb 2021 16:16:58 +0000 (+0200) Subject: Merge remote-tracking branch 'github/master' X-Git-Url: https://git.stg.codes/stg.git/commitdiff_plain/8e80bb9cec2c90dd61f810fb1525932a434288eb?ds=sidebyside;hp=-c Merge remote-tracking branch 'github/master' --- 8e80bb9cec2c90dd61f810fb1525932a434288eb diff --combined projects/stargazer/plugins/store/mysql/mysql_store.cpp index 3ad79b30,00000000..c441a708 mode 100644,000000..100644 --- a/projects/stargazer/plugins/store/mysql/mysql_store.cpp +++ b/projects/stargazer/plugins/store/mysql/mysql_store.cpp @@@ -1,2153 -1,0 +1,2166 @@@ +#include "mysql_store.h" + +#include "stg/common.h" +#include "stg/user_ips.h" +#include "stg/user_conf.h" +#include "stg/user_stat.h" +#include "stg/admin_conf.h" +#include "stg/tariff_conf.h" +#include "stg/blowfish.h" +#include "stg/logger.h" + +#include +#include +#include +#include +#include + +#include + +#define adm_enc_passwd "cjeifY8m3" + +namespace +{ +char qbuf[4096]; + +const int pt_mega = 1024 * 1024; +const std::string badSyms = "'`"; +const char repSym = '\"'; +const int RepitTimes = 3; + +template +int GetInt(const std::string & str, T * val, T defaultVal = T()) +{ + char *res; + + *val = static_cast(strtoll(str.c_str(), &res, 10)); + + if (*res != 0) + { + *val = defaultVal; //Error! + return EINVAL; + } + + return 0; +} + +int GetDouble(const std::string & str, double * val, double defaultVal) +{ + char *res; + + *val = strtod(str.c_str(), &res); + + if (*res != 0) + { + *val = defaultVal; //Error! + return EINVAL; + } + + return 0; +} + +int GetTime(const std::string & str, time_t * val, time_t defaultVal) +{ + char *res; + + *val = strtol(str.c_str(), &res, 10); + + if (*res != 0) + { + *val = defaultVal; //Error! + return EINVAL; + } + + return 0; +} + +//----------------------------------------------------------------------------- +std::string ReplaceStr(std::string source, const std::string & symlist, const char chgsym) +{ + std::string::size_type pos=0; + + while( (pos = source.find_first_of(symlist,pos)) != std::string::npos) + source.replace(pos, 1,1, chgsym); + + return source; +} + +int GetULongLongInt(const std::string & str, uint64_t * val, uint64_t defaultVal) +{ + char *res; + + *val = strtoull(str.c_str(), &res, 10); + + if (*res != 0) + { + *val = defaultVal; //Error! + return EINVAL; + } + + return 0; +} + +} + +extern "C" STG::Store* GetStore() +{ + static MYSQL_STORE plugin; + return &plugin; +} +//----------------------------------------------------------------------------- +MYSQL_STORE_SETTINGS::MYSQL_STORE_SETTINGS() + : settings(NULL) ++ , dbPort(0) +{ +} +//----------------------------------------------------------------------------- +int MYSQL_STORE_SETTINGS::ParseParam(const std::vector & moduleParams, + const std::string & name, std::string & result) +{ +STG::ParamValue pv; +pv.param = name; +std::vector::const_iterator pvi; +pvi = find(moduleParams.begin(), moduleParams.end(), pv); +if (pvi == moduleParams.end() || pvi->value.empty()) + { + errorStr = "Parameter \'" + name + "\' not found."; + return -1; + } + +result = pvi->value[0]; + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE_SETTINGS::ParseSettings(const STG::ModuleSettings & s) +{ +if (ParseParam(s.moduleParams, "user", dbUser) < 0 && + ParseParam(s.moduleParams, "dbuser", dbUser) < 0) + return -1; +if (ParseParam(s.moduleParams, "password", dbPass) < 0 && + ParseParam(s.moduleParams, "rootdbpass", dbPass) < 0) + return -1; +if (ParseParam(s.moduleParams, "database", dbName) < 0 && + ParseParam(s.moduleParams, "dbname", dbName) < 0) + return -1; +if (ParseParam(s.moduleParams, "server", dbHost) < 0 && + ParseParam(s.moduleParams, "dbhost", dbHost) < 0) + return -1; + ++// not required ++std::string dbPortAsString; ++if (ParseParam(s.moduleParams, "port", dbPortAsString) == 0 || ++ ParseParam(s.moduleParams, "dbport", dbPortAsString) == 0) ++{ ++ if (GetInt(dbPortAsString, &dbPort, 0) != 0) ++ { ++ errorStr = "Can't parse db port from string: \"" + dbPortAsString + "\"\n"; ++ return -1; ++ } ++} ++ +return 0; +} +//----------------------------------------------------------------------------- +//----------------------------------------------------------------------------- +//----------------------------------------------------------------------------- +MYSQL_STORE::MYSQL_STORE() - : version("mysql_store v.0.67"), ++ : version("mysql_store v.0.68"), + schemaVersion(0), + logger(STG::PluginLogger::get("store_mysql")) +{ +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::MysqlQuery(const char* sQuery,MYSQL * sock) const +{ + int ret; + + if( (ret = mysql_query(sock,sQuery)) ) + { + for(int i=0; i info; + if (GetAllParams(&info, "info", "version")) + schemaVersion = 0; + else + { + if (info.empty()) + schemaVersion = 0; + else + GetInt(info.front(), &schemaVersion, 0); + } +} +//admins----------------------------------------------------------------------- +if(!IsTablePresent("admins",sock)) +{ + sprintf(qbuf,"CREATE TABLE admins (login VARCHAR(40) DEFAULT '' PRIMARY KEY,"\ + "password VARCHAR(150) DEFAULT '*',ChgConf TINYINT DEFAULT 0,"\ + "ChgPassword TINYINT DEFAULT 0,ChgStat TINYINT DEFAULT 0,"\ + "ChgCash TINYINT DEFAULT 0,UsrAddDel TINYINT DEFAULT 0,"\ + "ChgTariff TINYINT DEFAULT 0,ChgAdmin TINYINT DEFAULT 0)"); + + if(MysqlQuery(qbuf,sock)) + { + errorStr = "Couldn't create admin table list With error:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } + + sprintf(qbuf,"INSERT INTO admins SET login='admin',"\ + "password='geahonjehjfofnhammefahbbbfbmpkmkmmefahbbbfbmpkmkmmefahbbbfbmpkmkaa',"\ + "ChgConf=1,ChgPassword=1,ChgStat=1,ChgCash=1,UsrAddDel=1,ChgTariff=1,ChgAdmin=1"); + + if(MysqlQuery(qbuf,sock)) + { + errorStr = "Couldn't create default admin. With error:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } +} + +//tariffs----------------------------------------------------------------------- +std::string param, res; +if(!IsTablePresent("tariffs",sock)) +{ + res = "CREATE TABLE tariffs (name VARCHAR(40) DEFAULT '' PRIMARY KEY,"; + + for (int i = 0; i < DIR_NUM; i++) + { + strprintf(¶m, " PriceDayA%d DOUBLE DEFAULT 0.0,", i); + res += param; + + strprintf(¶m, " PriceDayB%d DOUBLE DEFAULT 0.0,", i); + res += param; + + strprintf(¶m, " PriceNightA%d DOUBLE DEFAULT 0.0,", i); + res += param; + + strprintf(¶m, " PriceNightB%d DOUBLE DEFAULT 0.0,", i); + res += param; + + strprintf(¶m, " Threshold%d INT DEFAULT 0,", i); + res += param; + + strprintf(¶m, " Time%d VARCHAR(15) DEFAULT '0:0-0:0',", i); + res += param; + + strprintf(¶m, " NoDiscount%d INT DEFAULT 0,", i); + res += param; + + strprintf(¶m, " SinglePrice%d INT DEFAULT 0,", i); + res += param; + } + + res += "PassiveCost DOUBLE DEFAULT 0.0, Fee DOUBLE DEFAULT 0.0," + "Free DOUBLE DEFAULT 0.0, TraffType VARCHAR(10) DEFAULT ''," + "period VARCHAR(32) NOT NULL DEFAULT 'month'," + "change_policy VARCHAR(32) NOT NULL DEFAULT 'allow'," + "change_policy_timeout TIMESTAMP NOT NULL DEFAULT 0)"; + + if(MysqlQuery(res.c_str(),sock)) + { + errorStr = "Couldn't create tariffs table list With error:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } + + res = "INSERT INTO tariffs SET name='tariff',"; + + for (int i = 0; i < DIR_NUM; i++) + { + strprintf(¶m, " NoDiscount%d=1,", i); + res += param; + + strprintf(¶m, " Threshold%d=0,", i); + res += param; + + strprintf(¶m, " Time%d='0:0-0:0',", i); + res += param; + + if(i != 0 && i != 1) + { + strprintf(¶m, " SinglePrice%d=0,", i); + res += param; + } + + if(i != 1) + { + strprintf(¶m, " PriceDayA%d=0.0,", i); + res += param; + } + if(i != 1) + { + strprintf(¶m, " PriceDayB%d=0.0,", i); + res += param; + } + + if(i != 0) + { + strprintf(¶m, " PriceNightA%d=0.0,", i); + res += param; + } + if(i != 0) + { + strprintf(¶m, " PriceNightB%d=0.0,", i); + res += param; + } + } + + res += "PassiveCost=0.0, Fee=10.0, Free=0,"\ + "SinglePrice0=1, SinglePrice1=1,PriceDayA1=0.75,PriceDayB1=0.75,"\ + "PriceNightA0=1.0,PriceNightB0=1.0,TraffType='up+down',period='month',"\ + "change_policy='allow', change_policy_timeout=0"; + + if(MysqlQuery(res.c_str(),sock)) + { + errorStr = "Couldn't create default tariff. With error:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } + + sprintf(qbuf,"UPDATE info SET version=1"); + + if(MysqlQuery(qbuf,sock)) + { + errorStr = "Couldn't write default version. With error:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } + schemaVersion = 2; +} + +//users----------------------------------------------------------------------- +if(!IsTablePresent("users",sock)) +{ + res = "CREATE TABLE users (login VARCHAR(50) NOT NULL DEFAULT '' PRIMARY KEY, Password VARCHAR(150) NOT NULL DEFAULT '*',"\ + "Passive INT(3) DEFAULT 0,Down INT(3) DEFAULT 0,DisabledDetailStat INT(3) DEFAULT 0,AlwaysOnline INT(3) DEFAULT 0,Tariff VARCHAR(40) NOT NULL DEFAULT '',"\ + "Address VARCHAR(254) NOT NULL DEFAULT '',Phone VARCHAR(128) NOT NULL DEFAULT '',Email VARCHAR(50) NOT NULL DEFAULT '',"\ + "Note TEXT NOT NULL,RealName VARCHAR(254) NOT NULL DEFAULT '',StgGroup VARCHAR(40) NOT NULL DEFAULT '',"\ + "Credit DOUBLE DEFAULT 0, TariffChange VARCHAR(40) NOT NULL DEFAULT '',"; + + for (int i = 0; i < USERDATA_NUM; i++) + { + strprintf(¶m, " Userdata%d VARCHAR(254) NOT NULL,", i); + res += param; + } + + param = " CreditExpire INT(11) DEFAULT 0,"; + res += param; + + strprintf(¶m, " IP VARCHAR(254) DEFAULT '*',"); + res += param; + + for (int i = 0; i < DIR_NUM; i++) + { + strprintf(¶m, " D%d BIGINT(30) DEFAULT 0,", i); + res += param; + + strprintf(¶m, " U%d BIGINT(30) DEFAULT 0,", i); + res += param; + } + + strprintf(¶m, "Cash DOUBLE DEFAULT 0,FreeMb DOUBLE DEFAULT 0,LastCashAdd DOUBLE DEFAULT 0,"\ + "LastCashAddTime INT(11) DEFAULT 0,PassiveTime INT(11) DEFAULT 0,LastActivityTime INT(11) DEFAULT 0,"\ + "NAS VARCHAR(17) NOT NULL, INDEX (AlwaysOnline), INDEX (IP), INDEX (Address),"\ + " INDEX (Tariff),INDEX (Phone),INDEX (Email),INDEX (RealName))"); + res += param; + + if(MysqlQuery(res.c_str(),sock)) + { + errorStr = "Couldn't create users table list With error:\n"; + errorStr += mysql_error(sock); + errorStr += "\n\n" + res; + mysql_close(sock); + return -1; + } + + res = "INSERT INTO users SET login='test',Address='',AlwaysOnline=0,"\ + "Credit=0.0,CreditExpire=0,Down=0,Email='',DisabledDetailStat=0,"\ + "StgGroup='',IP='192.168.1.1',Note='',Passive=0,Password='123456',"\ + "Phone='', RealName='',Tariff='tariff',TariffChange='',NAS='',"; + + for (int i = 0; i < USERDATA_NUM; i++) + { + strprintf(¶m, " Userdata%d='',", i); + res += param; + } + + for (int i = 0; i < DIR_NUM; i++) + { + strprintf(¶m, " D%d=0,", i); + res += param; + + strprintf(¶m, " U%d=0,", i); + res += param; + } + + res += "Cash=10.0,FreeMb=0.0,LastActivityTime=0,LastCashAdd=0,"\ + "LastCashAddTime=0, PassiveTime=0"; + + if(MysqlQuery(res.c_str(),sock)) + { + errorStr = "Couldn't create default user. With error:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } +} +/* +//logs----------------------------------------------------------------------- +if(!IsTablePresent("logs")) +{ + sprintf(qbuf,"CREATE TABLE logs (unid INT UNSIGNED NOT NULL AUTO_INCREMENT PRIMARY KEY, login VARCHAR(40),text TEXT)"); + + if(MysqlQuery(qbuf)) + { + errorStr = "Couldn't create admin table list With error:\n"; + errorStr += mysql_error(sock); + return -1; + } +} +*/ +//messages--------------------------------------------------------------------- +if(!IsTablePresent("messages",sock)) +{ + sprintf(qbuf,"CREATE TABLE messages (login VARCHAR(40) DEFAULT '', id BIGINT, "\ + "type INT, lastSendTime INT, creationTime INT, showTime INT,"\ + "stgRepeat INT, repeatPeriod INT, text TEXT)"); + + if(MysqlQuery(qbuf,sock)) + { + errorStr = "Couldn't create messages table. With error:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } +} + +//month_stat------------------------------------------------------------------- +if(!IsTablePresent("stat",sock)) +{ + res = "CREATE TABLE stat (login VARCHAR(50), month TINYINT, year SMALLINT,"; + + for (int i = 0; i < DIR_NUM; i++) + { + strprintf(¶m, " U%d BIGINT,", i); + res += param; + + strprintf(¶m, " D%d BIGINT,", i); + res += param; + } + + res += " cash DOUBLE, INDEX (login))"; + + if(MysqlQuery(res.c_str(),sock)) + { + errorStr = "Couldn't create stat table. With error:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } +} + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::MakeUpdates(MYSQL * sock) +{ +if (schemaVersion < 1) + { + if (MysqlQuery("ALTER TABLE tariffs ADD period VARCHAR(32) NOT NULL DEFAULT 'month'", sock)) + { + errorStr = "Couldn't update tariffs table to version 1. With error:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } + if (MysqlQuery("UPDATE info SET version = 1", sock)) + { + errorStr = "Couldn't update DB schema version to 1. With error:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } + schemaVersion = 1; + logger("MYSQL_STORE: Updated DB schema to version %d", schemaVersion); + } + +if (schemaVersion < 2) + { + if (MysqlQuery("ALTER TABLE tariffs ADD change_policy VARCHAR(32) NOT NULL DEFAULT 'allow'", sock) || + MysqlQuery("ALTER TABLE tariffs ADD change_policy_timeout TIMESTAMP NOT NULL DEFAULT 0", sock)) + { + errorStr = "Couldn't update tariffs table to version 2. With error:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } + if (MysqlQuery("UPDATE info SET version = 2", sock)) + { + errorStr = "Couldn't update DB schema version to 2. With error:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } + schemaVersion = 2; + logger("MYSQL_STORE: Updated DB schema to version %d", schemaVersion); + } +return 0; +} +//----------------------------------------------------------------------------- + +int MYSQL_STORE::GetAllParams(std::vector * ParamList, + const std::string & table, const std::string & name) const +{ +MYSQL_RES *res; +MYSQL_ROW row; +MYSQL * sock=NULL; +my_ulonglong num, i; + +ParamList->clear(); + +sprintf(qbuf,"SELECT %s FROM %s", name.c_str(), table.c_str()); + +if(MysqlGetQuery(qbuf,sock)) +{ + errorStr = "Couldn't GetAllParams Query for: "; + errorStr += name + " - " + table + "\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +if (!(res=mysql_store_result(sock))) +{ + errorStr = "Couldn't GetAllParams Results for: "; + errorStr += name + " - " + table + "\n"; + errorStr += mysql_error(sock); + return -1; +} + +num = mysql_num_rows(res); + +for(i = 0; i < num; i++) +{ + row = mysql_fetch_row(res); + ParamList->push_back(row[0]); +} + +mysql_free_result(res); +mysql_close(sock); + +return 0; +} + +//----------------------------------------------------------------------------- +int MYSQL_STORE::GetUsersList(std::vector * usersList) const +{ +if(GetAllParams(usersList, "users", "login")) + return -1; + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::GetAdminsList(std::vector * adminsList) const +{ +if(GetAllParams(adminsList, "admins", "login")) + return -1; + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::GetTariffsList(std::vector * tariffsList) const +{ +if(GetAllParams(tariffsList, "tariffs", "name")) + return -1; + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::AddUser(const std::string & login) const +{ +std::string query = "INSERT INTO users SET login='" + login + "',Note='',NAS=''"; + +for (int i = 0; i < USERDATA_NUM; i++) + query += ",Userdata" + std::to_string(i) + "=''"; + +if(MysqlSetQuery(query.c_str())) +{ + errorStr = "Couldn't add user:\n"; + //errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::DelUser(const std::string & login) const +{ +sprintf(qbuf,"DELETE FROM users WHERE login='%s' LIMIT 1", login.c_str()); + +if(MysqlSetQuery(qbuf)) +{ + errorStr = "Couldn't delete user:\n"; + //errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::RestoreUserConf(STG::UserConf * conf, const std::string & login) const +{ +MYSQL_RES *res; +MYSQL_ROW row; +MYSQL * sock; +std::string query; + +query = "SELECT login, Password, Passive, Down, DisabledDetailStat, \ + AlwaysOnline, Tariff, Address, Phone, Email, Note, \ + RealName, StgGroup, Credit, TariffChange, "; + +for (int i = 0; i < USERDATA_NUM; i++) +{ + sprintf(qbuf, "Userdata%d, ", i); + query += qbuf; +} + +query += "CreditExpire, IP FROM users WHERE login='"; +query += login + "' LIMIT 1"; + +//sprintf(qbuf,"SELECT * FROM users WHERE login='%s' LIMIT 1", login.c_str()); + +if(MysqlGetQuery(query.c_str(),sock)) +{ + errorStr = "Couldn't restore Tariff(on query):\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +if (!(res=mysql_store_result(sock))) +{ + errorStr = "Couldn't restore Tariff(on getting result):\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +if (mysql_num_rows(res) != 1) +{ + errorStr = "User not found"; + mysql_close(sock); + return -1; +} + +row = mysql_fetch_row(res); + +conf->password = row[1]; + +if (conf->password.empty()) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' password is blank."; + mysql_close(sock); + return -1; + } + +if (GetInt(row[2],&conf->passive) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' data not read. Parameter Passive."; + mysql_close(sock); + return -1; + } + +if (GetInt(row[3], &conf->disabled) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' data not read. Parameter Down."; + mysql_close(sock); + return -1; + } + +if (GetInt(row[4], &conf->disabledDetailStat) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' data not read. Parameter DisabledDetailStat."; + mysql_close(sock); + return -1; + } + +if (GetInt(row[5], &conf->alwaysOnline) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' data not read. Parameter AlwaysOnline."; + mysql_close(sock); + return -1; + } + +conf->tariffName = row[6]; + +if (conf->tariffName.empty()) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' tariff is blank."; + mysql_close(sock); + return -1; + } + +conf->address = row[7]; +conf->phone = row[8]; +conf->email = row[9]; +conf->note = row[10]; +conf->realName = row[11]; +conf->group = row[12]; + +if (GetDouble(row[13], &conf->credit, 0) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' data not read. Parameter Credit."; + mysql_close(sock); + return -1; + } + +conf->nextTariff = row[14]; + +for (int i = 0; i < USERDATA_NUM; i++) + { + conf->userdata[i] = row[15+i]; + } + +GetTime(row[15+USERDATA_NUM], &conf->creditExpire, 0); + +std::string ipStr = row[16+USERDATA_NUM]; +STG::UserIPs i; +try + { + i = STG::UserIPs::parse(ipStr); + } +catch (const std::string & s) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' data not read. Parameter IP address. " + s; + mysql_close(sock); + return -1; + } +conf->ips = i; + +mysql_free_result(res); +mysql_close(sock); + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::RestoreUserStat(STG::UserStat * stat, const std::string & login) const +{ +MYSQL_RES *res; +MYSQL_ROW row; +MYSQL * sock; + +std::string query; + +query = "SELECT "; + +for (int i = 0; i < DIR_NUM; i++) +{ + sprintf(qbuf, "D%d, U%d, ", i, i); + query += qbuf; +} + +query += "Cash, FreeMb, LastCashAdd, LastCashAddTime, PassiveTime, LastActivityTime \ + FROM users WHERE login = '"; +query += login + "'"; + +//sprintf(qbuf,"SELECT * FROM users WHERE login='%s' LIMIT 1", login.c_str()); + +if(MysqlGetQuery(query.c_str() ,sock)) +{ + errorStr = "Couldn't restore UserStat(on query):\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +if (!(res=mysql_store_result(sock))) +{ + errorStr = "Couldn't restore UserStat(on getting result):\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +row = mysql_fetch_row(res); + +unsigned int startPos=0; + +char s[22]; + +for (int i = 0; i < DIR_NUM; i++) + { + uint64_t traff; + sprintf(s, "D%d", i); + if (GetULongLongInt(row[startPos+i*2], &traff, 0) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' stat not read. Parameter " + std::string(s); + mysql_close(sock); + return -1; + } + stat->monthDown[i] = traff; + + sprintf(s, "U%d", i); + if (GetULongLongInt(row[startPos+i*2+1], &traff, 0) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' stat not read. Parameter " + std::string(s); + mysql_close(sock); + return -1; + } + stat->monthUp[i] = traff; + }//for + +startPos += (2*DIR_NUM); + +if (GetDouble(row[startPos], &stat->cash, 0) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' stat not read. Parameter Cash"; + mysql_close(sock); + return -1; + } + +if (GetDouble(row[startPos+1],&stat->freeMb, 0) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' stat not read. Parameter FreeMb"; + mysql_close(sock); + return -1; + } + +if (GetDouble(row[startPos+2], &stat->lastCashAdd, 0) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' stat not read. Parameter LastCashAdd"; + mysql_close(sock); + return -1; + } + +if (GetTime(row[startPos+3], &stat->lastCashAddTime, 0) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' stat not read. Parameter LastCashAddTime"; + mysql_close(sock); + return -1; + } + +if (GetTime(row[startPos+4], &stat->passiveTime, 0) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' stat not read. Parameter PassiveTime"; + mysql_close(sock); + return -1; + } + +if (GetTime(row[startPos+5], &stat->lastActivityTime, 0) != 0) + { + mysql_free_result(res); + errorStr = "User \'" + login + "\' stat not read. Parameter LastActivityTime"; + mysql_close(sock); + return -1; + } + +mysql_free_result(res); +mysql_close(sock); +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::SaveUserConf(const STG::UserConf & conf, const std::string & login) const +{ +std::string param; +std::string res; + +strprintf(&res,"UPDATE users SET Password='%s', Passive=%d, Down=%d, DisabledDetailStat = %d, "\ + "AlwaysOnline=%d, Tariff='%s', Address='%s', Phone='%s', Email='%s', "\ + "Note='%s', RealName='%s', StgGroup='%s', Credit=%f, TariffChange='%s', ", + conf.password.c_str(), + conf.passive, + conf.disabled, + conf.disabledDetailStat, + conf.alwaysOnline, + conf.tariffName.c_str(), + (ReplaceStr(conf.address,badSyms,repSym)).c_str(), + (ReplaceStr(conf.phone,badSyms,repSym)).c_str(), + (ReplaceStr(conf.email,badSyms,repSym)).c_str(), + (ReplaceStr(conf.note,badSyms,repSym)).c_str(), + (ReplaceStr(conf.realName,badSyms,repSym)).c_str(), + (ReplaceStr(conf.group,badSyms,repSym)).c_str(), + conf.credit, + conf.nextTariff.c_str() + ); + +for (int i = 0; i < USERDATA_NUM; i++) + { + strprintf(¶m, " Userdata%d='%s',", i, + (ReplaceStr(conf.userdata[i],badSyms,repSym)).c_str()); + res += param; + } + +strprintf(¶m, " CreditExpire=%d,", conf.creditExpire); +res += param; + +std::ostringstream ipStr; +ipStr << conf.ips; + +strprintf(¶m, " IP='%s'", ipStr.str().c_str()); +res += param; + +strprintf(¶m, " WHERE login='%s' LIMIT 1", login.c_str()); +res += param; + +if(MysqlSetQuery(res.c_str())) +{ + errorStr = "Couldn't save user conf:\n"; + //errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::SaveUserStat(const STG::UserStat & stat, const std::string & login) const +{ +std::string param; +std::string res; + +res = "UPDATE users SET"; + +for (int i = 0; i < DIR_NUM; i++) + { + strprintf(¶m, " D%d=%lld,", i, stat.monthDown[i]); + res += param; + + strprintf(¶m, " U%d=%lld,", i, stat.monthUp[i]); + res += param; + } + +strprintf(¶m, " Cash=%f, FreeMb=%f, LastCashAdd=%f, LastCashAddTime=%d,"\ + " PassiveTime=%d, LastActivityTime=%d", + stat.cash, + stat.freeMb, + stat.lastCashAdd, + stat.lastCashAddTime, + stat.passiveTime, + stat.lastActivityTime + ); +res += param; + +strprintf(¶m, " WHERE login='%s' LIMIT 1", login.c_str()); +res += param; + +if(MysqlSetQuery(res.c_str())) +{ + errorStr = "Couldn't save user stat:\n"; +// errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::WriteLogString(const std::string & str, const std::string & login) const +{ +std::string res, tempStr; +time_t t; +tm * lt; + +t = time(NULL); +lt = localtime(&t); + +MYSQL_RES* result; +MYSQL * sock; +strprintf(&tempStr, "logs_%02d_%4d", lt->tm_mon+1, lt->tm_year+1900); +if (!(sock=MysqlConnect())){ + errorStr = "Couldn't connect to Server"; + return -1; +} +if (!(result=mysql_list_tables(sock,tempStr.c_str() ))) +{ + errorStr = "Couldn't get table " + tempStr + ":\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +my_ulonglong num_rows = mysql_num_rows(result); + +mysql_free_result(result); + +if (num_rows < 1) +{ + sprintf(qbuf,"CREATE TABLE logs_%02d_%4d (unid INT UNSIGNED NOT NULL AUTO_INCREMENT PRIMARY KEY, login VARCHAR(40),text TEXT)", + lt->tm_mon+1, lt->tm_year+1900); + + if(MysqlQuery(qbuf,sock)) + { + errorStr = "Couldn't create WriteDetailedStat table:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } +} + +strprintf(&res, "%s -- %s",LogDate(t), str.c_str()); + +std::string send; + +strprintf(&send,"INSERT INTO logs_%02d_%4d SET login='%s', text='%s'", + lt->tm_mon+1, lt->tm_year+1900, + login.c_str(), (ReplaceStr(res,badSyms,repSym)).c_str()); + +if(MysqlQuery(send.c_str(),sock)) +{ + errorStr = "Couldn't write log string:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} +mysql_close(sock); +return 0; + +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::WriteUserChgLog(const std::string & login, + const std::string & admLogin, + uint32_t admIP, + const std::string & paramName, + const std::string & oldValue, + const std::string & newValue, + const std::string & message) const +{ +std::string userLogMsg = "Admin \'" + admLogin + "\', " + inet_ntostring(admIP) + ": \'" + + paramName + "\' parameter changed from \'" + oldValue + + "\' to \'" + newValue + "\'. " + message; + +return WriteLogString(userLogMsg, login); +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::WriteUserConnect(const std::string & login, uint32_t ip) const +{ +std::string logStr = "Connect, " + inet_ntostring(ip); +return WriteLogString(logStr, login); +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::WriteUserDisconnect(const std::string & login, + const STG::DirTraff & up, + const STG::DirTraff & down, + const STG::DirTraff & sessionUp, + const STG::DirTraff & sessionDown, + double cash, + double /*freeMb*/, + const std::string & /*reason*/) const +{ +std::string logStr = "Disconnect, "; +std::ostringstream sssu; +std::ostringstream sssd; +std::ostringstream ssmu; +std::ostringstream ssmd; +std::ostringstream sscash; + +ssmu << up; +ssmd << down; + +sssu << sessionUp; +sssd << sessionDown; + +sscash << cash; + +logStr += " session upload: \'"; +logStr += sssu.str(); +logStr += "\' session download: \'"; +logStr += sssd.str(); +logStr += "\' month upload: \'"; +logStr += ssmu.str(); +logStr += "\' month download: \'"; +logStr += ssmd.str(); +logStr += "\' cash: \'"; +logStr += sscash.str(); +logStr += "\'"; + +return WriteLogString(logStr, login); +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::SaveMonthStat(const STG::UserStat & stat, int month, int year, + const std::string & login) const +{ +std::string param, res; + +strprintf(&res, "INSERT INTO stat SET login='%s', month=%d, year=%d,", + login.c_str(), month+1, year+1900); + +for (int i = 0; i < DIR_NUM; i++) + { + strprintf(¶m, " U%d=%lld,", i, stat.monthUp[i]); + res += param; + + strprintf(¶m, " D%d=%lld,", i, stat.monthDown[i]); + res += param; + } + +strprintf(¶m, " cash=%f", stat.cash); +res += param; + +if(MysqlSetQuery(res.c_str())) +{ + errorStr = "Couldn't SaveMonthStat:\n"; + //errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//-----------------------------------------------------------------------------*/ +int MYSQL_STORE::AddAdmin(const std::string & login) const +{ +sprintf(qbuf,"INSERT INTO admins SET login='%s'", login.c_str()); + +if(MysqlSetQuery(qbuf)) +{ + errorStr = "Couldn't add admin:\n"; + //errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//-----------------------------------------------------------------------------*/ +int MYSQL_STORE::DelAdmin(const std::string & login) const +{ +sprintf(qbuf,"DELETE FROM admins where login='%s' LIMIT 1", login.c_str()); + +if(MysqlSetQuery(qbuf)) +{ + errorStr = "Couldn't delete admin:\n"; + //errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//-----------------------------------------------------------------------------*/ +int MYSQL_STORE::SaveAdmin(const STG::AdminConf & ac) const +{ +char passwordE[2 * ADM_PASSWD_LEN + 2]; +char pass[ADM_PASSWD_LEN + 1]; +char adminPass[ADM_PASSWD_LEN + 1]; + +memset(pass, 0, sizeof(pass)); +memset(adminPass, 0, sizeof(adminPass)); + +BLOWFISH_CTX ctx; +InitContext(adm_enc_passwd, strlen(adm_enc_passwd), &ctx); + +strncpy(adminPass, ac.password.c_str(), ADM_PASSWD_LEN); +adminPass[ADM_PASSWD_LEN - 1] = 0; + +for (int i = 0; i < ADM_PASSWD_LEN/8; i++) + { + EncryptBlock(pass + 8*i, adminPass + 8*i, &ctx); + } + +pass[ADM_PASSWD_LEN - 1] = 0; +Encode12(passwordE, pass, ADM_PASSWD_LEN); + +sprintf(qbuf,"UPDATE admins SET password='%s', ChgConf=%d, ChgPassword=%d, "\ + "ChgStat=%d, ChgCash=%d, UsrAddDel=%d, ChgTariff=%d, ChgAdmin=%d "\ + "WHERE login='%s' LIMIT 1", + passwordE, + ac.priv.userConf, + ac.priv.userPasswd, + ac.priv.userStat, + ac.priv.userCash, + ac.priv.userAddDel, + ac.priv.tariffChg, + ac.priv.adminChg, + ac.login.c_str() + ); + +if(MysqlSetQuery(qbuf)) +{ + errorStr = "Couldn't save admin:\n"; + //errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::RestoreAdmin(STG::AdminConf * ac, const std::string & login) const +{ +char pass[ADM_PASSWD_LEN + 1]; +char password[ADM_PASSWD_LEN + 1]; +char passwordE[2*ADM_PASSWD_LEN + 2]; +BLOWFISH_CTX ctx; + +memset(password, 0, sizeof(password)); + +std::string p; +MYSQL_RES *res; +MYSQL_ROW row; +MYSQL * sock; +sprintf(qbuf,"SELECT * FROM admins WHERE login='%s' LIMIT 1", login.c_str()); + +if(MysqlGetQuery(qbuf,sock)) +{ + errorStr = "Couldn't restore admin:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +if (!(res=mysql_store_result(sock))) +{ + errorStr = "Couldn't restore admin:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +if ( mysql_num_rows(res) == 0) +{ + mysql_free_result(res); + errorStr = "Couldn't restore admin as couldn't found him in table.\n"; + mysql_close(sock); + return -1; +} + +row = mysql_fetch_row(res); + +p = row[1]; + +if(p.length() == 0) +{ + mysql_free_result(res); + errorStr = "Error in parameter password"; + mysql_close(sock); + return -1; +} + +memset(passwordE, 0, sizeof(passwordE)); +strncpy(passwordE, p.c_str(), 2*ADM_PASSWD_LEN); + +memset(pass, 0, sizeof(pass)); + +if (passwordE[0] != 0) + { + Decode21(pass, passwordE); + InitContext(adm_enc_passwd, strlen(adm_enc_passwd), &ctx); + + for (int i = 0; i < ADM_PASSWD_LEN/8; i++) + { + DecryptBlock(password + 8*i, pass + 8*i, &ctx); + } + } +else + { + password[0] = 0; + } + +ac->password = password; + +uint16_t a; + +if (GetInt(row[2], &a) == 0) + ac->priv.userConf = a; +else + { + mysql_free_result(res); + errorStr = "Error in parameter ChgConf"; + mysql_close(sock); + return -1; + } + +if (GetInt(row[3], &a) == 0) + ac->priv.userPasswd = a; +else + { + mysql_free_result(res); + errorStr = "Error in parameter ChgPassword"; + mysql_close(sock); + return -1; + } + +if (GetInt(row[4], &a) == 0) + ac->priv.userStat = a; +else + { + mysql_free_result(res); + errorStr = "Error in parameter ChgStat"; + mysql_close(sock); + return -1; + } + +if (GetInt(row[5], &a) == 0) + ac->priv.userCash = a; +else + { + mysql_free_result(res); + errorStr = "Error in parameter ChgCash"; + mysql_close(sock); + return -1; + } + +if (GetInt(row[6], &a) == 0) + ac->priv.userAddDel = a; +else + { + mysql_free_result(res); + errorStr = "Error in parameter UsrAddDel"; + mysql_close(sock); + return -1; + } + +if (GetInt(row[7], &a) == 0) + ac->priv.tariffChg = a; +else + { + mysql_free_result(res); + errorStr = "Error in parameter ChgTariff"; + mysql_close(sock); + return -1; + } + +if (GetInt(row[8], &a) == 0) + ac->priv.adminChg = a; +else + { + mysql_free_result(res); + errorStr = "Error in parameter ChgAdmin"; + mysql_close(sock); + return -1; + } + +mysql_free_result(res); +mysql_close(sock); +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::AddTariff(const std::string & name) const +{ +sprintf(qbuf,"INSERT INTO tariffs SET name='%s'", name.c_str()); + +if(MysqlSetQuery(qbuf)) +{ + errorStr = "Couldn't add tariff:\n"; +// errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::DelTariff(const std::string & name) const +{ +sprintf(qbuf,"DELETE FROM tariffs WHERE name='%s' LIMIT 1", name.c_str()); + +if(MysqlSetQuery(qbuf)) +{ + errorStr = "Couldn't delete tariff: "; +// errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::RestoreTariff(STG::TariffData * td, const std::string & tariffName) const +{ +MYSQL_RES *res; +MYSQL_ROW row; +MYSQL * sock; +sprintf(qbuf,"SELECT * FROM tariffs WHERE name='%s' LIMIT 1", tariffName.c_str()); + +if(MysqlGetQuery(qbuf,sock)) +{ + errorStr = "Couldn't restore Tariff:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +if (!(res=mysql_store_result(sock))) +{ + errorStr = "Couldn't restore Tariff:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +std::string str; +td->tariffConf.name = tariffName; + +row = mysql_fetch_row(res); + +std::string param; +for (int i = 0; idirPrice[i].hDay, + td->dirPrice[i].mDay, + td->dirPrice[i].hNight, + td->dirPrice[i].mNight); + + strprintf(¶m, "PriceDayA%d", i); + if (GetDouble(row[1+i*8], &td->dirPrice[i].priceDayA, 0.0) < 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param; + mysql_close(sock); + return -1; + } + td->dirPrice[i].priceDayA /= (1024*1024); + + strprintf(¶m, "PriceDayB%d", i); + if (GetDouble(row[2+i*8], &td->dirPrice[i].priceDayB, 0.0) < 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param; + mysql_close(sock); + return -1; + } + td->dirPrice[i].priceDayB /= (1024*1024); + + strprintf(¶m, "PriceNightA%d", i); + if (GetDouble(row[3+i*8], &td->dirPrice[i].priceNightA, 0.0) < 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param; + mysql_close(sock); + return -1; + } + td->dirPrice[i].priceNightA /= (1024*1024); + + strprintf(¶m, "PriceNightB%d", i); + if (GetDouble(row[4+i*8], &td->dirPrice[i].priceNightB, 0.0) < 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param; + mysql_close(sock); + return -1; + } + td->dirPrice[i].priceNightB /= (1024*1024); + + strprintf(¶m, "Threshold%d", i); + if (GetInt(row[5+i*8], &td->dirPrice[i].threshold) < 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param; + mysql_close(sock); + return -1; + } + + strprintf(¶m, "SinglePrice%d", i); + if (GetInt(row[8+i*8], &td->dirPrice[i].singlePrice) < 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param; + mysql_close(sock); + return -1; + } + + strprintf(¶m, "NoDiscount%d", i); + if (GetInt(row[7+i*8], &td->dirPrice[i].noDiscount) < 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param; + mysql_close(sock); + return -1; + } + }//main for + +if (GetDouble(row[2+8*DIR_NUM], &td->tariffConf.fee, 0.0) < 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter Fee"; + mysql_close(sock); + return -1; + } + +if (GetDouble(row[3+8*DIR_NUM], &td->tariffConf.free, 0.0) < 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter Free"; + mysql_close(sock); + return -1; + } + +if (GetDouble(row[1+8*DIR_NUM], &td->tariffConf.passiveCost, 0.0) < 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter PassiveCost"; + mysql_close(sock); + return -1; + } + + str = row[4+8*DIR_NUM]; + param = "TraffType"; + + if (str.length() == 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param; + mysql_close(sock); + return -1; + } + +td->tariffConf.traffType = STG::Tariff::parseTraffType(str); + +if (schemaVersion > 0) +{ + str = row[5+8*DIR_NUM]; + param = "Period"; + + if (str.length() == 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param; + mysql_close(sock); + return -1; + } + + td->tariffConf.period = STG::Tariff::parsePeriod(str); + } +else + { + td->tariffConf.period = STG::Tariff::MONTH; + } + +if (schemaVersion > 1) + { + str = row[6+8*DIR_NUM]; + param = "ChangePolicy"; + + if (str.length() == 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param; + mysql_close(sock); + return -1; + } + + td->tariffConf.changePolicy = STG::Tariff::parseChangePolicy(str); + + str = row[7+8*DIR_NUM]; + param = "ChangePolicyTimeout"; + + if (str.length() == 0) + { + mysql_free_result(res); + errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param; + mysql_close(sock); + return -1; + } + + td->tariffConf.changePolicyTimeout = readTime(str); + } +else + { + td->tariffConf.changePolicy = STG::Tariff::ALLOW; + td->tariffConf.changePolicyTimeout = 0; + } + +mysql_free_result(res); +mysql_close(sock); +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::SaveTariff(const STG::TariffData & td, const std::string & tariffName) const +{ +std::string param; + +std::string res="UPDATE tariffs SET"; + +for (int i = 0; i < DIR_NUM; i++) + { + strprintf(¶m, " PriceDayA%d=%f,", i, + td.dirPrice[i].priceDayA * pt_mega); + res += param; + + strprintf(¶m, " PriceDayB%d=%f,", i, + td.dirPrice[i].priceDayB * pt_mega); + res += param; + + strprintf(¶m, " PriceNightA%d=%f,", i, + td.dirPrice[i].priceNightA * pt_mega); + res += param; + + strprintf(¶m, " PriceNightB%d=%f,", i, + td.dirPrice[i].priceNightB * pt_mega); + res += param; + + strprintf(¶m, " Threshold%d=%d,", i, + td.dirPrice[i].threshold); + res += param; + + std::string s; + strprintf(¶m, " Time%d", i); + + strprintf(&s, "%0d:%0d-%0d:%0d", + td.dirPrice[i].hDay, + td.dirPrice[i].mDay, + td.dirPrice[i].hNight, + td.dirPrice[i].mNight); + + res += (param + "='" + s + "',"); + + strprintf(¶m, " NoDiscount%d=%d,", i, + td.dirPrice[i].noDiscount); + res += param; + + strprintf(¶m, " SinglePrice%d=%d,", i, + td.dirPrice[i].singlePrice); + res += param; + } + +strprintf(¶m, " PassiveCost=%f,", td.tariffConf.passiveCost); +res += param; + +strprintf(¶m, " Fee=%f,", td.tariffConf.fee); +res += param; + +strprintf(¶m, " Free=%f,", td.tariffConf.free); +res += param; + +res += " TraffType='" + STG::Tariff::toString(td.tariffConf.traffType) + "'"; + +if (schemaVersion > 0) + res += ", Period='" + STG::Tariff::toString(td.tariffConf.period) + "'"; + +if (schemaVersion > 1) + res += ", change_policy='" + STG::Tariff::toString(td.tariffConf.changePolicy) + "'"\ + ", change_policy_timeout='" + formatTime(td.tariffConf.changePolicy) + "'"; + +strprintf(¶m, " WHERE name='%s' LIMIT 1", tariffName.c_str()); +res += param; + +if(MysqlSetQuery(res.c_str())) +{ + errorStr = "Couldn't save tariff:\n"; + //errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::WriteDetailedStat(const STG::TraffStat & statTree, + time_t lastStat, + const std::string & login) const +{ +std::string res, stTime, endTime, tempStr; +time_t t; +tm * lt; + +t = time(NULL); +lt = localtime(&t); + +if (lt->tm_hour == 0 && lt->tm_min <= 5) + { + t -= 3600 * 24; + lt = localtime(&t); + } + +MYSQL_RES* result; +MYSQL * sock; +strprintf(&tempStr, "detailstat_%02d_%4d", lt->tm_mon+1, lt->tm_year+1900); + +if (!(sock=MysqlConnect())){ + mysql_close(sock); + return -1; +} + +if (!(result=mysql_list_tables(sock,tempStr.c_str() ))) +{ + errorStr = "Couldn't get table " + tempStr + ":\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +my_ulonglong num_rows = mysql_num_rows(result); + +mysql_free_result(result); + +if (num_rows < 1) +{ + sprintf(qbuf,"CREATE TABLE detailstat_%02d_%4d (login VARCHAR(40) DEFAULT '',"\ + "day TINYINT DEFAULT 0,startTime TIME,endTime TIME,"\ + "IP VARCHAR(17) DEFAULT '',dir INT DEFAULT 0,"\ + "down BIGINT DEFAULT 0,up BIGINT DEFAULT 0, cash DOUBLE DEFAULT 0.0, INDEX (login), INDEX(dir), INDEX(day), INDEX(IP))", + lt->tm_mon+1, lt->tm_year+1900); + + if(MysqlQuery(qbuf,sock)) + { + errorStr = "Couldn't create WriteDetailedStat table:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } +} + +struct tm * lt1; +struct tm * lt2; + +lt1 = localtime(&lastStat); + +int h1, m1, s1; +int h2, m2, s2; + +h1 = lt1->tm_hour; +m1 = lt1->tm_min; +s1 = lt1->tm_sec; + +lt2 = localtime(&t); + +h2 = lt2->tm_hour; +m2 = lt2->tm_min; +s2 = lt2->tm_sec; + +strprintf(&stTime, "%02d:%02d:%02d", h1, m1, s1); +strprintf(&endTime, "%02d:%02d:%02d", h2, m2, s2); + +strprintf(&res,"INSERT INTO detailstat_%02d_%4d SET login='%s',"\ + "day=%d,startTime='%s',endTime='%s',", + lt->tm_mon+1, lt->tm_year+1900, + login.c_str(), + lt->tm_mday, + stTime.c_str(), + endTime.c_str() + ); + +STG::TraffStat::const_iterator stIter; +stIter = statTree.begin(); + +while (stIter != statTree.end()) + { + strprintf(&tempStr,"IP='%s', dir=%d, down=%lld, up=%lld, cash=%f", + inet_ntostring(stIter->first.ip).c_str(), + stIter->first.dir, + stIter->second.down, + stIter->second.up, + stIter->second.cash + ); + + if( MysqlQuery((res+tempStr).c_str(),sock) ) + { + errorStr = "Couldn't insert data in WriteDetailedStat:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; + } + + result=mysql_store_result(sock); + if(result) + mysql_free_result(result); + + ++stIter; + } +mysql_close(sock); +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::AddMessage(STG::Message * msg, const std::string & login) const +{ +struct timeval tv; + +gettimeofday(&tv, NULL); + +msg->header.id = static_cast(tv.tv_sec) * 1000000 + static_cast(tv.tv_usec); + +sprintf(qbuf,"INSERT INTO messages SET login='%s', id=%lld", + login.c_str(), + static_cast(msg->header.id) + ); + +if(MysqlSetQuery(qbuf)) +{ + errorStr = "Couldn't add message:\n"; + //errorStr += mysql_error(sock); + return -1; +} + +return EditMessage(*msg, login); +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::EditMessage(const STG::Message & msg, const std::string & login) const +{ +std::string res; + +strprintf(&res,"UPDATE messages SET type=%d, lastSendTime=%u, creationTime=%u, "\ + "showTime=%u, stgRepeat=%d, repeatPeriod=%u, text='%s' "\ + "WHERE login='%s' AND id=%lld LIMIT 1", + msg.header.type, + msg.header.lastSendTime, + msg.header.creationTime, + msg.header.showTime, + msg.header.repeat, + msg.header.repeatPeriod, + (ReplaceStr(msg.text,badSyms,repSym)).c_str(), + login.c_str(), + msg.header.id + ); + +if(MysqlSetQuery(res.c_str())) +{ + errorStr = "Couldn't edit message:\n"; + //errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::GetMessage(uint64_t id, STG::Message * msg, const std::string & login) const +{ +MYSQL_RES *res; +MYSQL_ROW row; +MYSQL * sock; + +sprintf(qbuf,"SELECT * FROM messages WHERE login='%s' AND id=%llu LIMIT 1", + login.c_str(), static_cast(id)); + +if(MysqlGetQuery(qbuf,sock)) +{ + errorStr = "Couldn't GetMessage:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +if (!(res=mysql_store_result(sock))) +{ + errorStr = "Couldn't GetMessage:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +row = mysql_fetch_row(res); + +if(row[2]&&str2x(row[2], msg->header.type)) +{ + mysql_free_result(res); + errorStr = "Invalid value in message header for user: " + login; + mysql_close(sock); + return -1; +} + +if(row[3] && str2x(row[3], msg->header.lastSendTime)) +{ + mysql_free_result(res); + errorStr = "Invalid value in message header for user: " + login; + mysql_close(sock); + return -1; +} + +if(row[4] && str2x(row[4], msg->header.creationTime)) +{ + mysql_free_result(res); + errorStr = "Invalid value in message header for user: " + login; + mysql_close(sock); + return -1; +} + +if(row[5] && str2x(row[5], msg->header.showTime)) +{ + mysql_free_result(res); + errorStr = "Invalid value in message header for user: " + login; + mysql_close(sock); + return -1; +} + +if(row[6] && str2x(row[6], msg->header.repeat)) +{ + mysql_free_result(res); + errorStr = "Invalid value in message header for user: " + login; + mysql_close(sock); + return -1; +} + +if(row[7] && str2x(row[7], msg->header.repeatPeriod)) +{ + mysql_free_result(res); + errorStr = "Invalid value in message header for user: " + login; + mysql_close(sock); + return -1; +} + +msg->header.id = id; +msg->text = row[8]; + +mysql_free_result(res); +mysql_close(sock); +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::DelMessage(uint64_t id, const std::string & login) const +{ +sprintf(qbuf,"DELETE FROM messages WHERE login='%s' AND id=%lld LIMIT 1", + login.c_str(), static_cast(id)); + +if(MysqlSetQuery(qbuf)) +{ + errorStr = "Couldn't delete Message:\n"; + //errorStr += mysql_error(sock); + return -1; +} + +return 0; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::GetMessageHdrs(std::vector * hdrsList, const std::string & login) const +{ +MYSQL_RES *res; +MYSQL_ROW row; +MYSQL * sock; +sprintf(qbuf,"SELECT * FROM messages WHERE login='%s'", login.c_str()); + +if(MysqlGetQuery(qbuf,sock)) +{ + errorStr = "Couldn't GetMessageHdrs:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +if (!(res=mysql_store_result(sock))) +{ + errorStr = "Couldn't GetMessageHdrs:\n"; + errorStr += mysql_error(sock); + mysql_close(sock); + return -1; +} + +unsigned int i; +my_ulonglong num_rows = mysql_num_rows(res); +uint64_t id = 0; + +for (i = 0; i < num_rows; i++) +{ + row = mysql_fetch_row(res); + if (str2x(row[1], id)) + continue; + + STG::Message::Header hdr; + if (row[2]) + if(str2x(row[2], hdr.type)) + continue; + + if (row[3]) + if(str2x(row[3], hdr.lastSendTime)) + continue; + + if (row[4]) + if(str2x(row[4], hdr.creationTime)) + continue; + + if (row[5]) + if(str2x(row[5], hdr.showTime)) + continue; + + if (row[6]) + if(str2x(row[6], hdr.repeat)) + continue; + + if (row[7]) + if(str2x(row[7], hdr.repeatPeriod)) + continue; + + hdr.id = id; + hdrsList->push_back(hdr); +} + +mysql_free_result(res); +mysql_close(sock); +return 0; +} +//----------------------------------------------------------------------------- + +int MYSQL_STORE::MysqlSetQuery(const char * Query) const { + + MYSQL * sock; + int ret=MysqlGetQuery(Query,sock); + mysql_close(sock); + return ret; +} +//----------------------------------------------------------------------------- +int MYSQL_STORE::MysqlGetQuery(const char * Query,MYSQL * & sock) const { + if (!(sock=MysqlConnect())) { + return -1; + } + return MysqlQuery(Query,sock); +} +//----------------------------------------------------------------------------- +MYSQL * MYSQL_STORE::MysqlConnect() const { + MYSQL * sock; + if ( !(sock=mysql_init(NULL)) ){ + errorStr= "mysql init susck\n"; + return NULL; + } + if (!(sock = mysql_real_connect(sock,storeSettings.GetDBHost().c_str(), + storeSettings.GetDBUser().c_str(),storeSettings.GetDBPassword().c_str(), - 0,0,NULL,0))) ++ 0,storeSettings.GetDBPort(),NULL,0))) + { + errorStr = "Couldn't connect to mysql engine! With error:\n"; + errorStr += mysql_error(sock); + return NULL; + } + else{ + if(mysql_select_db(sock, storeSettings.GetDBName().c_str())){ + errorStr = "Database lost !\n"; + return NULL; + } + } + return sock; +} +//----------------------------------------------------------------------------- diff --combined projects/stargazer/plugins/store/mysql/mysql_store.h index 6d09c476,00000000..61a13203 mode 100644,000000..100644 --- a/projects/stargazer/plugins/store/mysql/mysql_store.h +++ b/projects/stargazer/plugins/store/mysql/mysql_store.h @@@ -1,140 -1,0 +1,142 @@@ +#pragma once + +#include "stg/store.h" +#include "stg/module_settings.h" +#include "stg/user_traff.h" +#include "stg/logger.h" + +#include +#include +#include + +#include + +//----------------------------------------------------------------------------- +class MYSQL_STORE_SETTINGS +{ +public: + MYSQL_STORE_SETTINGS(); + virtual ~MYSQL_STORE_SETTINGS() {} + virtual int ParseSettings(const STG::ModuleSettings & s); + virtual const std::string & GetStrError() const { return errorStr; } + + const std::string & GetDBUser() const { return dbUser; } + const std::string & GetDBPassword() const { return dbPass; } + const std::string & GetDBHost() const { return dbHost; } + const std::string & GetDBName() const { return dbName; } ++ unsigned int GetDBPort() const { return dbPort; } + +private: + MYSQL_STORE_SETTINGS(const MYSQL_STORE_SETTINGS & rvalue); + MYSQL_STORE_SETTINGS & operator=(const MYSQL_STORE_SETTINGS & rvalue); + + const STG::ModuleSettings * settings; + + int ParseParam(const std::vector & moduleParams, + const std::string & name, std::string & result); + + std::string errorStr; + + std::string dbUser; + std::string dbPass; + std::string dbName; + std::string dbHost; ++ unsigned int dbPort; +}; +//----------------------------------------------------------------------------- +class MYSQL_STORE: public STG::Store +{ +public: + MYSQL_STORE(); + const std::string & GetStrError() const override { return errorStr; } + + //User + int GetUsersList(std::vector * usersList) const override; + int AddUser(const std::string & login) const override; + int DelUser(const std::string & login) const override; + int SaveUserStat(const STG::UserStat & stat, const std::string & login) const override; + int SaveUserConf(const STG::UserConf & conf, const std::string & login) const override; + int RestoreUserStat(STG::UserStat * stat, const std::string & login) const override; + int RestoreUserConf(STG::UserConf * conf, const std::string & login) const override; + int WriteUserChgLog(const std::string & login, + const std::string & admLogin, + uint32_t admIP, + const std::string & paramName, + const std::string & oldValue, + const std::string & newValue, + const std::string & message = "") const override; + int WriteUserConnect(const std::string & login, uint32_t ip) const override; + int WriteUserDisconnect(const std::string & login, + const STG::DirTraff & up, + const STG::DirTraff & down, + const STG::DirTraff & sessionUp, + const STG::DirTraff & sessionDown, + double cash, + double freeMb, + const std::string & reason) const override; + + int WriteDetailedStat(const STG::TraffStat & statTree, + time_t lastStat, + const std::string & login) const override; + + int AddMessage(STG::Message * msg, const std::string & login) const override; + int EditMessage(const STG::Message & msg, const std::string & login) const override; + int GetMessage(uint64_t id, STG::Message * msg, const std::string & login) const override; + int DelMessage(uint64_t id, const std::string & login) const override; + int GetMessageHdrs(std::vector * hdrsList, const std::string & login) const override; + + int SaveMonthStat(const STG::UserStat & stat, int month, int year, const std::string & login) const override; + + //Admin + int GetAdminsList(std::vector * adminsList) const override; + int AddAdmin(const std::string & login) const override; + int DelAdmin(const std::string & login) const override; + int RestoreAdmin(STG::AdminConf * ac, const std::string & login) const override; + int SaveAdmin(const STG::AdminConf & ac) const override; + + //Tariff + int GetTariffsList(std::vector * tariffsList) const override; + int AddTariff(const std::string & name) const override; + int DelTariff(const std::string & name) const override; + int SaveTariff(const STG::TariffData & td, const std::string & tariffName) const override; + int RestoreTariff(STG::TariffData * td, const std::string & tariffName) const override; + + //Corparation + int GetCorpsList(std::vector *) const override {return 0;} + int SaveCorp(const STG::CorpConf &) const override {return 0;} + int RestoreCorp(STG::CorpConf *, const std::string &) const override {return 0;} + int AddCorp(const std::string &) const override {return 0;} + int DelCorp(const std::string &) const override {return 0;} + + // Services + int GetServicesList(std::vector *) const override {return 0;} + int SaveService(const STG::ServiceConf &) const override {return 0;} + int RestoreService(STG::ServiceConf *, const std::string &) const override {return 0;} + int AddService(const std::string &) const override {return 0;} + int DelService(const std::string &) const override {return 0;} + + void SetSettings(const STG::ModuleSettings & s) override { settings = s; } + int ParseSettings() override; + const std::string & GetVersion() const override { return version; } + +private: + MYSQL_STORE(const MYSQL_STORE & rvalue); + MYSQL_STORE & operator=(const MYSQL_STORE & rvalue); + + virtual int WriteLogString(const std::string & str, const std::string & login) const; + int GetAllParams(std::vector * ParamList, const std::string & table, const std::string & name) const; + int CheckAllTables(MYSQL * sock); + int MakeUpdates(MYSQL * sock); + bool IsTablePresent(const std::string & str,MYSQL * sock); + mutable std::string errorStr; + int MysqlQuery(const char* sQuery,MYSQL * sock) const; + int MysqlGetQuery(const char * Query,MYSQL * & sock) const; + int MysqlSetQuery(const char * Query) const; + MYSQL * MysqlConnect() const ; + std::string version; + MYSQL_STORE_SETTINGS storeSettings; + STG::ModuleSettings settings; + int schemaVersion; + + STG::PluginLogger logger; +};