#include <mysql.h>
#include <errmsg.h>
+#include "stg/common.h"
#include "stg/user_ips.h"
#include "stg/user_conf.h"
#include "stg/user_stat.h"
#include "stg/blowfish.h"
#include "stg/plugin_creator.h"
+#include "stg/logger.h"
#include "mysql_store.h"
#define adm_enc_passwd "cjeifY8m3"
int GetInt(const std::string & str, T * val, T defaultVal = T())
{
char *res;
-
+
*val = static_cast<T>(strtoll(str.c_str(), &res, 10));
-
- if (*res != 0)
+
+ if (*res != 0)
{
*val = defaultVal; //Error!
return EINVAL;
int GetDouble(const std::string & str, double * val, double defaultVal)
{
char *res;
-
+
*val = strtod(str.c_str(), &res);
-
- if (*res != 0)
+
+ if (*res != 0)
{
*val = defaultVal; //Error!
return EINVAL;
int GetTime(const std::string & str, time_t * val, time_t defaultVal)
{
char *res;
-
+
*val = strtol(str.c_str(), &res, 10);
-
- if (*res != 0)
+
+ if (*res != 0)
{
*val = defaultVal; //Error!
return EINVAL;
}
//-----------------------------------------------------------------------------
-std::string ReplaceStr(std::string source, const std::string symlist, const char chgsym)
+std::string ReplaceStr(std::string source, const std::string & symlist, const char chgsym)
{
std::string::size_type pos=0;
int GetULongLongInt(const std::string & str, uint64_t * val, uint64_t defaultVal)
{
char *res;
-
+
*val = strtoull(str.c_str(), &res, 10);
-
- if (*res != 0)
+
+ if (*res != 0)
{
*val = defaultVal; //Error!
return EINVAL;
}
return 0;
-}
+}
PLUGIN_CREATOR<MYSQL_STORE> msc;
}
}
//-----------------------------------------------------------------------------
MYSQL_STORE_SETTINGS::MYSQL_STORE_SETTINGS()
- : settings(NULL),
- errorStr(),
- dbUser(),
- dbPass(),
- dbName(),
- dbHost()
+ : settings(NULL)
+ , dbPort(0)
{
}
//-----------------------------------------------------------------------------
-int MYSQL_STORE_SETTINGS::ParseParam(const std::vector<PARAM_VALUE> & moduleParams,
- const std::string & name, std::string & result)
+int MYSQL_STORE_SETTINGS::ParseParam(const std::vector<PARAM_VALUE> & moduleParams,
+ const std::string & name, std::string & result)
{
PARAM_VALUE pv;
pv.param = name;
std::vector<PARAM_VALUE>::const_iterator pvi;
pvi = find(moduleParams.begin(), moduleParams.end(), pv);
-if (pvi == moduleParams.end())
+if (pvi == moduleParams.end() || pvi->value.empty())
{
errorStr = "Parameter \'" + name + "\' not found.";
return -1;
}
-
+
result = pvi->value[0];
return 0;
ParseParam(s.moduleParams, "dbhost", dbHost) < 0)
return -1;
+std::string dbPortAsString;
+if (ParseParam(s.moduleParams, "port", dbPortAsString) == 0 ||
+ ParseParam(s.moduleParams, "dbport", dbPortAsString) == 0)
+{
+ if (GetInt<unsigned int>(dbPortAsString, &dbPort, 0) != 0)
+ {
+ errorStr = "Can't parse db port from string: \"" + dbPortAsString + "\"\n";
+ return -1;
+ }
+}
+
return 0;
}
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
MYSQL_STORE::MYSQL_STORE()
- : errorStr(),
- version("mysql_store v.0.67"),
- storeSettings(),
- settings(),
+ : version("mysql_store v.0.67"),
+ schemaVersion(0),
logger(GetPluginLogger(GetStgLogger(), "store_mysql"))
{
}
return 0;
}
}
-
+
return ret;
}
//-----------------------------------------------------------------------------
{
int ret = storeSettings.ParseSettings(settings);
MYSQL mysql;
-MYSQL * sock;
mysql_init(&mysql);
if (ret)
errorStr = storeSettings.GetStrError();
errorStr = "Database password must be not empty. Please read Manual.";
return -1;
}
-
+ MYSQL * sock;
if (!(sock = mysql_real_connect(&mysql,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(&mysql);
if(mysql_select_db(sock, storeSettings.GetDBName().c_str()))
{
std::string res = "CREATE DATABASE " + storeSettings.GetDBName();
-
+
if(MysqlQuery(res.c_str(),sock))
{
errorStr = "Couldn't create database! With error:\n";
{
if(mysql_select_db(sock, storeSettings.GetDBName().c_str()))
{
- errorStr = "Couldn't select database! With error:\n";
- errorStr += mysql_error(sock);
- mysql_close(sock);
- ret = -1;
+ errorStr = "Couldn't select database! With error:\n";
+ errorStr += mysql_error(sock);
+ mysql_close(sock);
+ ret = -1;
}
- ret = CheckAllTables(sock);
+ else
+ ret = CheckAllTables(sock);
}
- }
- else
- ret = CheckAllTables(sock);
- mysql_close(sock);
+ }
+ else
+ {
+ ret = CheckAllTables(sock);
+ }
+ if (!ret)
+ {
+ logger("MYSQL_STORE: Current DB schema version: %d", schemaVersion);
+ MakeUpdates(sock);
+ }
+ mysql_close(sock);
}
-
}
return ret;
}
//-----------------------------------------------------------------------------
int MYSQL_STORE::CheckAllTables(MYSQL * sock)
{
+//info-------------------------------------------------------------------------
+if(!IsTablePresent("info",sock))
+{
+ sprintf(qbuf,"CREATE TABLE info (version INTEGER NOT NULL)");
+
+ if(MysqlQuery(qbuf,sock))
+ {
+ errorStr = "Couldn't create info table With error:\n";
+ errorStr += mysql_error(sock);
+ mysql_close(sock);
+ return -1;
+ }
+
+ sprintf(qbuf,"INSERT INTO info SET version=0");
+
+ if(MysqlQuery(qbuf,sock))
+ {
+ errorStr = "Couldn't write default version. With error:\n";
+ errorStr += mysql_error(sock);
+ mysql_close(sock);
+ return -1;
+ }
+ schemaVersion = 0;
+}
+else
+{
+ std::vector<std::string> 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))
{
"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";
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";
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);
+ 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 '')";
-
+
+ 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";
}
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;
+ res += param;
}
-
+
if(i != 1)
{
- strprintf(¶m, " PriceDayA%d=0.0,", i);
- res += param;
+ strprintf(¶m, " PriceDayA%d=0.0,", i);
+ res += param;
}
if(i != 1)
{
- strprintf(¶m, " PriceDayB%d=0.0,", i);
- res += param;
+ strprintf(¶m, " PriceDayB%d=0.0,", i);
+ res += param;
}
-
+
if(i != 0)
{
- strprintf(¶m, " PriceNightA%d=0.0,", i);
- res += param;
+ strprintf(¶m, " PriceNightA%d=0.0,", i);
+ res += param;
}
if(i != 0)
{
- strprintf(¶m, " PriceNightB%d=0.0,", i);
- res += param;
+ 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'";
-
+ "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";
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-----------------------------------------------------------------------
"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";
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='',Userdata0='',"\
- "Userdata1='',";
-
+ "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";
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";
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";
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);
+ strprintf(¶m, " U%d BIGINT,", i);
res += param;
-
- strprintf(¶m, " D%d BIGINT,", i);
+
+ 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";
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<std::string> * ParamList,
+int MYSQL_STORE::GetAllParams(std::vector<std::string> * 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: ";
for(i = 0; i < num; i++)
{
- row = mysql_fetch_row(res);
+ row = mysql_fetch_row(res);
ParamList->push_back(row[0]);
}
//-----------------------------------------------------------------------------
int MYSQL_STORE::AddUser(const std::string & login) const
{
-sprintf(qbuf,"INSERT INTO users SET login='%s'", login.c_str());
-
-if(MysqlSetQuery(qbuf))
+std::string query = "INSERT INTO users SET login='" + login + "',Note='',NAS=''";
+
+for (int i = 0; i < USERDATA_NUM; i++)
+ query += ",Userdata" + x2str(i) + "=''";
+
+if(MysqlSetQuery(query.c_str()))
{
errorStr = "Couldn't add user:\n";
//errorStr += mysql_error(sock);
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";
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";
row = mysql_fetch_row(res);
-std::string param;
-
conf->password = row[1];
if (conf->password.empty())
conf->tariffName = row[6];
-if (conf->tariffName.empty())
+if (conf->tariffName.empty())
{
mysql_free_result(res);
errorStr = "User \'" + login + "\' tariff is blank.";
}
GetTime(row[15+USERDATA_NUM], &conf->creditExpire, 0);
-
+
std::string ipStr = row[16+USERDATA_NUM];
USER_IPS i;
try
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";
mysql_close(sock);
return -1;
}
- stat->down[i] = traff;
+ stat->monthDown[i] = traff;
sprintf(s, "U%d", i);
if (GetULongLongInt(row[startPos+i*2+1], &traff, 0) != 0)
mysql_close(sock);
return -1;
}
- stat->up[i] = traff;
+ stat->monthUp[i] = traff;
}//for
startPos += (2*DIR_NUM);
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', ",
+ "Note='%s', RealName='%s', StgGroup='%s', Credit=%f, TariffChange='%s', ",
conf.password.c_str(),
conf.passive,
conf.disabled,
for (int i = 0; i < USERDATA_NUM; i++)
{
- strprintf(¶m, " Userdata%d='%s',", 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;
for (int i = 0; i < DIR_NUM; i++)
{
- strprintf(¶m, " D%d=%lld,", i, stat.down[i]);
+ strprintf(¶m, " D%d=%lld,", i, stat.monthDown[i]);
res += param;
- strprintf(¶m, " U%d=%lld,", i, stat.up[i]);
+ 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",
+ " PassiveTime=%d, LastActivityTime=%d",
stat.cash,
stat.freeMb,
stat.lastCashAdd,
{
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";
return WriteLogString(logStr, login);
}
//-----------------------------------------------------------------------------
-int MYSQL_STORE::SaveMonthStat(const USER_STAT & stat, int month, int year,
+int MYSQL_STORE::SaveMonthStat(const USER_STAT & 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,",
+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.up[i]);
+ strprintf(¶m, " U%d=%lld,", i, stat.monthUp[i]);
res += param;
- strprintf(¶m, " D%d=%lld,", i, stat.down[i]);
+ strprintf(¶m, " D%d=%lld,", i, stat.monthDown[i]);
res += param;
}
-
-strprintf(¶m, " cash=%f", stat.cash);
+
+strprintf(¶m, " cash=%f", stat.cash);
res += param;
if(MysqlSetQuery(res.c_str()))
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";
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";
memset(adminPass, 0, sizeof(adminPass));
BLOWFISH_CTX ctx;
-EnDecodeInit(adm_enc_passwd, strlen(adm_enc_passwd), &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++)
{
- EncodeString(pass + 8*i, adminPass + 8*i, &ctx);
+ EncryptBlock(pass + 8*i, adminPass + 8*i, &ctx);
}
pass[ADM_PASSWD_LEN - 1] = 0;
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",
+ "WHERE login='%s' LIMIT 1",
passwordE,
ac.priv.userConf,
ac.priv.userPasswd,
char passwordE[2*ADM_PASSWD_LEN + 2];
BLOWFISH_CTX ctx;
-memset(pass, 0, sizeof(pass));
memset(password, 0, sizeof(password));
-memset(passwordE, 0, sizeof(passwordE));
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";
mysql_close(sock);
return -1;
}
-
+
row = mysql_fetch_row(res);
p = row[1];
if (passwordE[0] != 0)
{
Decode21(pass, passwordE);
- EnDecodeInit(adm_enc_passwd, strlen(adm_enc_passwd), &ctx);
+ InitContext(adm_enc_passwd, strlen(adm_enc_passwd), &ctx);
for (int i = 0; i < ADM_PASSWD_LEN/8; i++)
{
- DecodeString(password + 8*i, pass + 8*i, &ctx);
+ DecryptBlock(password + 8*i, pass + 8*i, &ctx);
}
}
else
uint16_t a;
-if (GetInt(row[2], &a) == 0)
+if (GetInt(row[2], &a) == 0)
ac->priv.userConf = a;
else
{
return -1;
}
-if (GetInt(row[3], &a) == 0)
+if (GetInt(row[3], &a) == 0)
ac->priv.userPasswd = a;
else
{
return -1;
}
-if (GetInt(row[4], &a) == 0)
+if (GetInt(row[4], &a) == 0)
ac->priv.userStat = a;
else
{
return -1;
}
-if (GetInt(row[5], &a) == 0)
+if (GetInt(row[5], &a) == 0)
ac->priv.userCash = a;
else
{
return -1;
}
-if (GetInt(row[6], &a) == 0)
+if (GetInt(row[6], &a) == 0)
ac->priv.userAddDel = a;
else
{
return -1;
}
-if (GetInt(row[7], &a) == 0)
+if (GetInt(row[7], &a) == 0)
ac->priv.tariffChg = a;
else
{
return -1;
}
-if (GetInt(row[8], &a) == 0)
+if (GetInt(row[8], &a) == 0)
ac->priv.adminChg = a;
else
{
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";
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: ";
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";
return -1;
}
- ParseTariffTimeStr(str.c_str(),
- td->dirPrice[i].hDay,
- td->dirPrice[i].mDay,
- td->dirPrice[i].hNight,
+ ParseTariffTimeStr(str.c_str(),
+ td->dirPrice[i].hDay,
+ td->dirPrice[i].mDay,
+ td->dirPrice[i].hNight,
td->dirPrice[i].mNight);
strprintf(¶m, "PriceDayA%d", i);
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 = TARIFF::StringToTraffType(str);
+
+if (schemaVersion > 0)
+{
+ str = row[5+8*DIR_NUM];
+ param = "Period";
+
if (str.length() == 0)
{
mysql_free_result(res);
return -1;
}
-if (!strcasecmp(str.c_str(), "up"))
- td->tariffConf.traffType = TRAFF_UP;
+ td->tariffConf.period = TARIFF::StringToPeriod(str);
+ }
else
- if (!strcasecmp(str.c_str(), "down"))
- td->tariffConf.traffType = TRAFF_DOWN;
- else
- if (!strcasecmp(str.c_str(), "up+down"))
- td->tariffConf.traffType = TRAFF_UP_DOWN;
- else
- if (!strcasecmp(str.c_str(), "max"))
- td->tariffConf.traffType = TRAFF_MAX;
- else
- {
- mysql_free_result(res);
- errorStr = "Cannot read tariff " + tariffName + ". Parameter TraffType incorrect";
- mysql_close(sock);
- return -1;
- }
+ {
+ td->tariffConf.period = 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 = TARIFF::StringToChangePolicy(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 = TARIFF::ALLOW;
+ td->tariffConf.changePolicyTimeout = 0;
+ }
mysql_free_result(res);
mysql_close(sock);
for (int i = 0; i < DIR_NUM; i++)
{
- strprintf(¶m, " PriceDayA%d=%f,", 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);
+ 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,
+ strprintf(¶m, " PriceNightB%d=%f,", i,
td.dirPrice[i].priceNightB * pt_mega);
res += param;
-
- strprintf(¶m, " Threshold%d=%d,", i,
+
+ 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",
+ strprintf(&s, "%0d:%0d-%0d:%0d",
td.dirPrice[i].hDay,
td.dirPrice[i].mDay,
td.dirPrice[i].hNight,
res += (param + "='" + s + "',");
- strprintf(¶m, " NoDiscount%d=%d,", i,
+ strprintf(¶m, " NoDiscount%d=%d,", i,
td.dirPrice[i].noDiscount);
res += param;
- strprintf(¶m, " SinglePrice%d=%d,", i,
+ strprintf(¶m, " SinglePrice%d=%d,", i,
td.dirPrice[i].singlePrice);
res += param;
}
strprintf(¶m, " Free=%f,", td.tariffConf.free);
res += param;
-switch (td.tariffConf.traffType)
- {
- case TRAFF_UP:
- res += " TraffType='up'";
- break;
- case TRAFF_DOWN:
- res += " TraffType='down'";
- break;
- case TRAFF_UP_DOWN:
- res += " TraffType='up+down'";
- break;
- case TRAFF_MAX:
- res += " TraffType='max'";
- break;
- }
+res += " TraffType='" + TARIFF::TraffTypeToString(td.tariffConf.traffType) + "'";
+
+if (schemaVersion > 0)
+ res += ", Period='" + TARIFF::PeriodToString(td.tariffConf.period) + "'";
+
+if (schemaVersion > 1)
+ res += ", change_policy='" + TARIFF::ChangePolicyToString(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 admin:\n";
+ errorStr = "Couldn't save tariff:\n";
//errorStr += mysql_error(sock);
return -1;
}
return 0;
}
//-----------------------------------------------------------------------------
-int MYSQL_STORE::WriteDetailedStat(const std::map<IP_DIR_PAIR, STAT_NODE> & statTree,
- time_t lastStat,
+int MYSQL_STORE::WriteDetailedStat(const std::map<IP_DIR_PAIR, STAT_NODE> & statTree,
+ time_t lastStat,
const std::string & login) const
{
std::string res, stTime, endTime, tempStr;
"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";
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',",
+ "day=%d,startTime='%s',endTime='%s',",
lt->tm_mon+1, lt->tm_year+1900,
login.c_str(),
lt->tm_mday,
while (stIter != statTree.end())
{
- strprintf(&tempStr,"IP='%s', dir=%d, down=%lld, up=%lld, cash=%f",
+ 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->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";
msg->header.id = static_cast<uint64_t>(tv.tv_sec) * 1000000 + static_cast<uint64_t>(tv.tv_usec);
-sprintf(qbuf,"INSERT INTO messages SET login='%s', id=%lld",
+sprintf(qbuf,"INSERT INTO messages SET login='%s', id=%lld",
login.c_str(),
static_cast<long long>(msg->header.id)
);
-
+
if(MysqlSetQuery(qbuf))
{
errorStr = "Couldn't add message:\n";
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",
+ "WHERE login='%s' AND id=%lld LIMIT 1",
msg.header.type,
msg.header.lastSendTime,
msg.header.creationTime,
sprintf(qbuf,"SELECT * FROM messages WHERE login='%s' AND id=%llu LIMIT 1",
login.c_str(), static_cast<unsigned long long>(id));
-
+
if(MysqlGetQuery(qbuf,sock))
{
errorStr = "Couldn't GetMessage:\n";
//-----------------------------------------------------------------------------
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",
+sprintf(qbuf,"DELETE FROM messages WHERE login='%s' AND id=%lld LIMIT 1",
login.c_str(), static_cast<long long>(id));
-
+
if(MysqlSetQuery(qbuf))
{
errorStr = "Couldn't delete Message:\n";
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";
row = mysql_fetch_row(res);
if (str2x(row[1], id))
continue;
-
+
STG_MSG_HDR hdr;
- if (row[2])
+ if (row[2])
if(str2x(row[2], hdr.type))
continue;
}
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);
git.stg.codes / stg.git / blobdiff