acl/app/net_tools/net_store.cpp
zsxxsz cc05b877a2 first commit acl to github
first commit acl to github
2013-08-18 17:42:25 +08:00

293 lines
6.5 KiB
C++
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#include "StdAfx.h"
#include "global/global.h"
#include "global/passwd_crypt.h"
#include "net_store.h"
net_store::net_store(const char* smtp_addr, int smtp_port,
const char* pop3_addr, int pop3_port,
const char* user, const char* pass,
const char* recipients, net_store_callback* callback,
bool store /* = false */)
: smtp_addr_(smtp_addr)
, smtp_port_(smtp_port)
, pop3_addr_(pop3_addr)
, pop3_port_(pop3_port)
, user_(user)
, pass_(pass)
, recipients_(recipients)
, callback_(callback)
, store_(store)
{
}
net_store::~net_store()
{
}
//////////////////////////////////////////////////////////////////////////
bool create_option_tbl(acl::db_handle& db, const char* tbl_name,
const char* sql_create)
{
if (db.tbl_exists(tbl_name))
{
logger("table(%s) exist", tbl_name);
return (true);
}
else if (db.sql_update(sql_create) == false)
{
logger_error("sql(%s) error", sql_create);
return (false);
}
else
{
logger("create table %s ok", tbl_name);
return (true);
}
}
//////////////////////////////////////////////////////////////////////////
static const char* CREATE_OPT_TBL =
"create table option_tbl"
"(\r\n"
"name varchar(128) not null default '',\r\n"
"value varchar(256) not null default '',\r\n"
"PRIMARY KEY(name)"
")";
static acl::db_handle* open_option_tbl()
{
const char* path = global::get_instance().get_path();
acl::string dbpath;
dbpath.format("%s/net_store.db", path);
acl::db_handle* db = new acl::db_sqlite(dbpath.c_str());
if (db->open() == false)
{
logger_error("open db: %s failed", dbpath.c_str());
delete db;
return NULL;
}
else if (create_option_tbl(*db, "option_tbl", CREATE_OPT_TBL) == false)
{
delete db;
return NULL;
}
else
return db;
}
bool net_store::get_key(const char* name, acl::string& out)
{
acl::db_handle* db = open_option_tbl();
if (db == NULL)
return false;
acl::string sql;
sql.format("select value from option_tbl where name='%s'", name);
if (db->sql_select(sql.c_str()) == false)
{
delete db;
return false;
}
const acl::db_row* row = db->get_first_row();
if (row == NULL)
{
delete db;
return false;
}
const char* value = (*row)["value"];
if (value == NULL)
{
db->free_result();
delete db;
return false;
}
out = value;
return true;
}
bool net_store::set_key(const char* name, const char* value)
{
acl::db_handle* db = open_option_tbl();
if (db == NULL)
return false;
acl::string sql;
sql.format("insert into option_tbl(name, value) values('%s', '%s')",
name, value);
if (db->sql_update(sql.c_str()) == false)
{
sql.format("update option_tbl set value='%s' where name='%s'",
value, name);
db->sql_update(sql.c_str());
delete db;
return false;
}
else
{
delete db;
return true;
}
}
//////////////////////////////////////////////////////////////////////////
// <20><><EFBFBD>߳<EFBFBD><DFB3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
void net_store::rpc_onover()
{
callback_->load_db_callback(smtp_addr_.c_str(), smtp_port_,
pop3_addr_.c_str(), pop3_port_, user_.c_str(),
pass_.c_str(), recipients_.c_str(), store_);
delete this;
}
//////////////////////////////////////////////////////////////////////////
// <20><><EFBFBD>߳<EFBFBD><DFB3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
static const char* CREATE_MAIL_TBL =
"create table mail_tbl\r\n"
"(\r\n"
"smtp_addr varchar(128) not null,\r\n"
"smtp_port int not null,\r\n"
"pop3_addr varchar(128) not null default '',\r\n"
"pop3_port int not null,\r\n"
"user varchar(128) not null default '',\r\n"
"pass varchar(128) not null default '',\r\n"
"recipients varchar(256) not null default '',\r\n"
"PRIMARY KEY(user)"
")";
void net_store::rpc_run()
{
const char* path = global::get_instance().get_path();
dbpath_.format("%s/net_store.db", path);
acl::db_sqlite db(dbpath_.c_str());
if (db.open() == false)
logger_error("open db: %s failed", dbpath_.c_str());
else if (create_mail_tbl(db, "mail_tbl", CREATE_MAIL_TBL) == false)
logger_error("create table failed for %s", dbpath_.c_str());
else if (store_)
save_mail_db(db);
else
load_mail_db(db);
}
bool net_store::create_mail_tbl(acl::db_handle& db, const char* tbl_name,
const char* sql_create)
{
if (db.tbl_exists(tbl_name))
{
logger("table(%s) exist", tbl_name);
return (true);
}
else if (db.sql_update(sql_create) == false)
{
logger_error("sql(%s) error", sql_create);
return (false);
}
else
{
save_mail_db(db);
logger("create table %s ok", tbl_name);
return (true);
}
}
void net_store::save_mail_db(acl::db_handle& db)
{
acl::string sql;
// <20><>ɾ<EFBFBD><C9BE>ԭ<EFBFBD><D4AD><EFBFBD><EFBFBD><EFBFBD>еļ<D0B5>¼
sql.format("delete from mail_tbl");
db.sql_update(sql.c_str());
acl::string smtp_addr, pop3_addr, user, pass, recipients;
db.escape_string(smtp_addr_.c_str(), smtp_addr_.length(), smtp_addr);
db.escape_string(pop3_addr_.c_str(), pop3_addr_.length(), pop3_addr);
db.escape_string(user_.c_str(), user_.length(), user);
db.escape_string(pass_.c_str(), pass_.length(), pass);
ACL_ARGV* tokens = acl_argv_split(recipients_.c_str(), "\t,;\r\n");
ACL_ITER iter;
acl::string buf;
acl_foreach(iter, tokens)
{
if (iter.i > 0)
buf << ",";
buf << (char*) iter.data;
}
acl_argv_free(tokens);
db.escape_string(buf.c_str(), recipients_.length(), recipients);
// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҫ<EFBFBD><D2AA><EFBFBD>ܴ洢
char* pass_crypted = passwd_crypt(pass.c_str());
sql.format("insert into mail_tbl(smtp_addr, smtp_port, pop3_addr,"
" pop3_port, user, pass, recipients)"
" values('%s', %d, '%s', %d, '%s', '%s', '%s')",
smtp_addr.c_str(), smtp_port_, pop3_addr.c_str(), pop3_port_,
user.c_str(), pass_crypted, recipients.c_str());
acl_myfree(pass_crypted);
const char* ptr = sql.c_str();
ptr = pop3_addr.c_str();
if (db.sql_update(sql.c_str()) == false)
logger_error("sql(%s) error", sql.c_str());
}
void net_store::load_mail_db(acl::db_handle& db)
{
acl::string sql;
sql.format("select smtp_addr, smtp_port, pop3_addr, pop3_port, user, pass, recipients from mail_tbl");
if (db.sql_select(sql.c_str()) == false)
{
logger_error("sql(%s) error", sql.c_str());
return;
}
const acl::db_row* first_row = db.get_first_row();
if (first_row == NULL)
return;
const char* ptr = (*first_row)["smtp_addr"];
if (ptr)
smtp_addr_ = ptr;
ptr = (*first_row)["smtp_port"];
int n;
if (ptr && (n = atoi(ptr)) > 0)
smtp_port_ = n;
ptr = (*first_row)["pop3_addr"];
if (ptr)
pop3_addr_ = ptr;
ptr = (*first_row)["pop3_port"];
if (ptr && (n = atoi(ptr)) > 0)
pop3_port_ = n;
ptr = (*first_row)["user"];
if (ptr)
user_ = ptr;
ptr = (*first_row)["pass"];
if (ptr)
{
// <20>Լ<EFBFBD><D4BC>ܵ<EFBFBD><DCB5><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҫ<EFBFBD>Ƚ<EFBFBD><C8BD><EFBFBD>
char* pass_plain = passwd_decrypt(ptr);
if (pass_plain)
{
pass_ = pass_plain;
acl_myfree(pass_plain);
}
}
ptr = (*first_row)["recipients"];
if (ptr)
recipients_ = ptr;
}