mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-12-02 11:57:43 +08:00
122 lines
3.3 KiB
C++
122 lines
3.3 KiB
C++
#include "StdAfx.h"
|
|
#include "global/global.h"
|
|
#include "smtp_client.h"
|
|
#include "smtp_store.h"
|
|
|
|
smtp_store::smtp_store(const char* user, const char* smtp_ip,
|
|
const SMTP_METER& meter, smtp_callback& callback)
|
|
: callback_(callback)
|
|
, ok_(false)
|
|
{
|
|
user_ = acl_mystrdup(user);
|
|
smtp_ip_ = acl_mystrdup(smtp_ip);
|
|
|
|
meter_ = (SMTP_METER*) acl_mycalloc(1, sizeof(SMTP_METER));
|
|
memcpy(meter_, &meter, sizeof(SMTP_METER));
|
|
}
|
|
|
|
smtp_store::~smtp_store()
|
|
{
|
|
acl_myfree(user_);
|
|
acl_myfree(smtp_ip_);
|
|
acl_myfree(meter_);
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
// Ö÷Ï̹߳ý³Ì
|
|
|
|
void smtp_store::rpc_onover()
|
|
{
|
|
logger("store smtp test results %s!", ok_ ? "OK" : "Failed");
|
|
callback_.smtp_finish(dbpath_.c_str());
|
|
delete this;
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
// ×ÓÏ̹߳ý³Ì
|
|
|
|
static const char* CREATE_TBL =
|
|
"create table smtp_tbl\r\n"
|
|
"(\r\n"
|
|
"user varchar(128) not null,\r\n"
|
|
"smtp_ip varchar(32) not null,\r\n"
|
|
"nslookup_elapsed float(10,2) not null default 0.00,\r\n"
|
|
"connect_elapsed float(10,2) not null default 0.00,\r\n"
|
|
"banner_elapsed float(10,2) not null default 0.00,\r\n"
|
|
"auth_elapsed float(10,2) not null default 0.00,\r\n"
|
|
"mail_elapsed float(10,2) not null default 0.00,\r\n"
|
|
"rcpt_elapsed float(10,2) not null default 0.00,\r\n"
|
|
"data_elapsed float(10,2) not null default 0.00,\r\n"
|
|
"body_elapsed float(10,2) not null default 0.00,\r\n"
|
|
"envelope_elapsed float(10,2) not null default 0.00,\r\n"
|
|
"total_elapsed float(10,2) not null default 0.00,\r\n"
|
|
"smtp_speed int not null default 0\r\n"
|
|
");\r\n"
|
|
"create index user_idx on smtp_tbl(user);\r\n";
|
|
|
|
void smtp_store::rpc_run()
|
|
{
|
|
const char* path = global::get_instance().get_path();
|
|
dbpath_.format("%s/smtp_store2.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_tbl(db) == false)
|
|
logger_error("create table failed for %s", dbpath_.c_str());
|
|
else
|
|
{
|
|
logger("open db(%s) ok", dbpath_.c_str());
|
|
insert_tbl(db);
|
|
ok_ = true;
|
|
}
|
|
}
|
|
|
|
bool smtp_store::create_tbl(acl::db_handle& db)
|
|
{
|
|
if (db.tbl_exists("smtp_tbl"))
|
|
{
|
|
logger("smtp_tbl table exist");
|
|
return (true);
|
|
}
|
|
else if (db.sql_update(CREATE_TBL) == false)
|
|
{
|
|
logger_error("sql(%s) error", CREATE_TBL);
|
|
return (false);
|
|
}
|
|
else
|
|
{
|
|
logger("create table smtp_tbl ok");
|
|
return (true);
|
|
}
|
|
}
|
|
|
|
void smtp_store::insert_tbl(acl::db_handle& db)
|
|
{
|
|
acl::string sql;
|
|
|
|
sql.format("insert into smtp_tbl(user, smtp_ip, nslookup_elapsed, "
|
|
"connect_elapsed, banner_elapsed, auth_elapsed, "
|
|
"mail_elapsed, rcpt_elapsed, data_elapsed, "
|
|
"body_elapsed, envelope_elapsed, total_elapsed, smtp_speed) "
|
|
"values('%s', '%s', %0.2f, %0.2f, %0.2f, %0.2f, %0.2f, "
|
|
"%0.2f, %0.2f, %0.2f, %0.2f, %0.2f, %d)",
|
|
user_, smtp_ip_,
|
|
meter_->smtp_nslookup_elapsed,
|
|
meter_->smtp_connect_elapsed,
|
|
meter_->smtp_banner_elapsed,
|
|
meter_->smtp_auth_elapsed,
|
|
meter_->smtp_mail_elapsed,
|
|
meter_->smtp_rcpt_elapsed,
|
|
meter_->smtp_data_elapsed,
|
|
meter_->smtp_envelope_eplased,
|
|
meter_->smtp_body_elapsed,
|
|
meter_->smtp_total_elapsed,
|
|
meter_->smtp_speed);
|
|
|
|
if (db.sql_update(sql.c_str()) == false)
|
|
logger_error("sql(%s) error", sql.c_str());
|
|
else
|
|
logger("insert into smtp_tbl OK!");
|
|
}
|