Added.
This commit is contained in:
Saturneic
parent
fc386d83de
commit
dbaa0fc064
@ -16,6 +16,14 @@
|
||||
|
||||
class Proj;
|
||||
|
||||
//检查数据库表专用信息储存结构
|
||||
struct check_table_column{
|
||||
string name;
|
||||
string type;
|
||||
int notnull;
|
||||
int pk;
|
||||
};
|
||||
|
||||
//配置文件通用方法类
|
||||
class setting_file{
|
||||
protected:
|
||||
@ -172,6 +180,8 @@ class Proj:public setting_file{
|
||||
void write_proj_info(void);
|
||||
// 检查数据库是否正确
|
||||
void check_database(void);
|
||||
// 检查数据库表
|
||||
void check_table(int cnum, vector<check_table_column> tctc,sqlite3_stmt *psqlsmt);
|
||||
public:
|
||||
// 读取Proj文件
|
||||
Proj(string t_projpath, string t_projfile);
|
||||
@ -191,6 +201,7 @@ public:
|
||||
void DBProcess(void);
|
||||
// 获得工程名
|
||||
string GetName(void);
|
||||
// 更新工程
|
||||
void UpdateProcess(void);
|
||||
};
|
||||
|
||||
|
||||
@ -230,6 +230,7 @@ void Proj::write_src_info(void){
|
||||
sqlite3_reset(psqlsmt);
|
||||
sqlite3_clear_bindings(psqlsmt);
|
||||
}
|
||||
sqlite3_finalize(psqlsmt);
|
||||
}
|
||||
|
||||
//写入动态链接库信息到数据库中
|
||||
@ -262,6 +263,7 @@ void Proj::write_lib_info(void){
|
||||
sqlite3_reset(psqlsmt);
|
||||
sqlite3_clear_bindings(psqlsmt);
|
||||
}
|
||||
sqlite3_finalize(psqlsmt);
|
||||
}
|
||||
|
||||
// 写入入口函数信息到数据库中
|
||||
@ -341,6 +343,7 @@ void Proj::write_func_info(void){
|
||||
sqlite3_reset(psqlsmt);
|
||||
sqlite3_clear_bindings(psqlsmt);
|
||||
}
|
||||
sqlite3_finalize(psqlsmt);
|
||||
}
|
||||
|
||||
//写入口函数入输入输出参数信息到数据库中
|
||||
@ -437,6 +440,7 @@ void Proj::write_args_info(void){
|
||||
sqlite3_reset(psqlsmt);
|
||||
sqlite3_clear_bindings(psqlsmt);
|
||||
}
|
||||
sqlite3_finalize(psqlsmt);
|
||||
}
|
||||
}
|
||||
|
||||
@ -482,6 +486,7 @@ void Proj::write_cpt_info(void){
|
||||
sqlite3_reset(psqlsmt);
|
||||
sqlite3_clear_bindings(psqlsmt);
|
||||
}
|
||||
sqlite3_finalize(psqlsmt);
|
||||
}
|
||||
//写入工程描述文件信息到数据库中
|
||||
void Proj::write_proj_info(void){
|
||||
@ -520,9 +525,170 @@ void Proj::write_proj_info(void){
|
||||
}
|
||||
sqlite3_reset(psqlsmt);
|
||||
sqlite3_clear_bindings(psqlsmt);
|
||||
sqlite3_finalize(psqlsmt);
|
||||
|
||||
}
|
||||
|
||||
void Proj::check_database(void){
|
||||
// 编译SQL语句
|
||||
string sql_quote = "SELECT count(*) FROM sqlite_master WHERE type='table' AND name=?1";
|
||||
sqlite3_stmt *psqlsmt;
|
||||
const char *pzTail;
|
||||
sqlite3_prepare(psql, sql_quote.data(), -1, &psqlsmt, &pzTail);
|
||||
// 检查数据库表是否存在
|
||||
#ifdef DEBUG
|
||||
printf("Start To Check Tables In Database\n");
|
||||
#endif
|
||||
const char *tables[] = {"srcfiles","cptfiles","functions","projfile","libfiles","projfile"};
|
||||
for(auto strc : tables){
|
||||
sqlite3_bind_text(psqlsmt, 1, strc, -1, SQLITE_STATIC);
|
||||
// 执行SQL语句
|
||||
int rtn = sqlite3_step(psqlsmt);
|
||||
if(rtn == SQLITE_OK || rtn == SQLITE_DONE || rtn == SQLITE_ROW){
|
||||
#ifdef DEBUG
|
||||
printf("[*]Succeed In Selecting Table %s\n",strc);
|
||||
#endif
|
||||
}
|
||||
else{
|
||||
#ifdef DEBUG
|
||||
printf("[*]Failed to Select Table %s\n",strc);
|
||||
#endif
|
||||
throw "fail to select";
|
||||
}
|
||||
int ifhas = sqlite3_column_int(psqlsmt, 0);
|
||||
if(ifhas != 1){
|
||||
#ifdef DEBUG
|
||||
printf("Table Is Abnormal %s\n",strc);
|
||||
#endif
|
||||
throw "table is abnormal";
|
||||
}
|
||||
sqlite3_reset(psqlsmt);
|
||||
sqlite3_clear_bindings(psqlsmt);
|
||||
}
|
||||
sqlite3_finalize(psqlsmt);
|
||||
|
||||
// 检查数据库表srcfiles其项及其类型是否正确
|
||||
sql_quote = "PRAGMA table_info( srcfiles );";
|
||||
sqlite3_prepare(psql, sql_quote.data(), -1, &psqlsmt, &pzTail);
|
||||
#ifdef DEBUG
|
||||
printf("Start To Check Table Columns srcfiles\n");
|
||||
#endif
|
||||
vector<check_table_column> sctc={
|
||||
{"id","INT",1,1},
|
||||
{"name","TEXT",1,0},
|
||||
{"path","TEXT",1,0},
|
||||
{"content","NONE",0,0},
|
||||
{"md5","TEXT",1,0}
|
||||
};
|
||||
check_table(5, sctc, psqlsmt);
|
||||
sqlite3_finalize(psqlsmt);
|
||||
|
||||
// 检查数据库表libfiles其项及其类型是否正确
|
||||
sql_quote = "PRAGMA table_info( libfiles );";
|
||||
sqlite3_prepare(psql, sql_quote.data(), -1, &psqlsmt, &pzTail);
|
||||
#ifdef DEBUG
|
||||
printf("Start To Check Table Columns libfiles\n");
|
||||
#endif
|
||||
sctc={
|
||||
{"id","INT",1,1},
|
||||
{"name","TEXT",1,0},
|
||||
{"content","NONE",0,0}
|
||||
};
|
||||
check_table(3, sctc, psqlsmt);
|
||||
sqlite3_finalize(psqlsmt);
|
||||
|
||||
// 检查数据库表cptfiles其项及其类型是否正确
|
||||
sql_quote = "PRAGMA table_info( cptfiles );";
|
||||
sqlite3_prepare(psql, sql_quote.data(), -1, &psqlsmt, &pzTail);
|
||||
#ifdef DEBUG
|
||||
printf("Start To Check Table Columns cptfiles\n");
|
||||
#endif
|
||||
sctc={
|
||||
{"id","INT",1,1},
|
||||
{"path","TEXT",1,0},
|
||||
{"content","NONE",0,0},
|
||||
{"md5","TEXT",1,0}
|
||||
};
|
||||
check_table(4, sctc, psqlsmt);
|
||||
sqlite3_finalize(psqlsmt);
|
||||
|
||||
// 检查数据库表functions其项及其类型是否正确
|
||||
sql_quote = "PRAGMA table_info( functions );";
|
||||
sqlite3_prepare(psql, sql_quote.data(), -1, &psqlsmt, &pzTail);
|
||||
#ifdef DEBUG
|
||||
printf("Start To Check Table Columns functions\n");
|
||||
#endif
|
||||
sctc={
|
||||
{"id","INT",1,1},
|
||||
{"name","TEXT",1,0},
|
||||
{"srcfile_id","INT",1,0},
|
||||
{"srcfile_id","INT",1,0}
|
||||
};
|
||||
check_table(4, sctc, psqlsmt);
|
||||
sqlite3_finalize(psqlsmt);
|
||||
|
||||
// 检查数据库表projfile其项及其类型是否正确
|
||||
sql_quote = "PRAGMA table_info( projfile );";
|
||||
sqlite3_prepare(psql, sql_quote.data(), -1, &psqlsmt, &pzTail);
|
||||
#ifdef DEBUG
|
||||
printf("Start To Check Table Columns projfile\n");
|
||||
#endif
|
||||
sctc={
|
||||
{"project_name","TEXT",1,0},
|
||||
{"content","NONE",0,0},
|
||||
{"md5","TEXT",1,0}
|
||||
};
|
||||
check_table(3, sctc, psqlsmt);
|
||||
sqlite3_finalize(psqlsmt);
|
||||
#ifdef DEBUG
|
||||
printf("Succeed In Checking Table Columns\n");
|
||||
#endif
|
||||
|
||||
sql_quote = "SELECT project_name FROM projfile;";
|
||||
sqlite3_prepare(psql, sql_quote.data(), -1, &psqlsmt, &pzTail);
|
||||
sqlite3_step(psqlsmt);
|
||||
string tname = (char *)sqlite3_column_text(psqlsmt, 0);
|
||||
if(name != tname){
|
||||
#ifdef DEBUG
|
||||
printf("Database Conflict %s\n",tname.data());
|
||||
#endif
|
||||
throw "database conflict";
|
||||
}
|
||||
sqlite3_finalize(psqlsmt);
|
||||
#ifdef DEBUG
|
||||
printf("Succeed In Checking Project Information\n");
|
||||
#endif
|
||||
|
||||
}
|
||||
|
||||
void Proj::check_table(int cnum,vector<check_table_column> sctc,sqlite3_stmt *psqlsmt){
|
||||
sqlite3_step(psqlsmt);
|
||||
int tcnum = sqlite3_data_count(psqlsmt);
|
||||
if(tcnum != 6){
|
||||
#ifdef DEBUG
|
||||
printf("[*]Column Number Is Abnormal.\n");
|
||||
#endif
|
||||
throw "table abnormal srcfiles";
|
||||
}
|
||||
struct check_table_column tctc;
|
||||
for(int i = 1; i < cnum; i++){
|
||||
tctc.name = (char *) sqlite3_column_text(psqlsmt, 1);
|
||||
tctc.type = (char *) sqlite3_column_text(psqlsmt, 2);
|
||||
tctc.notnull = sqlite3_column_int(psqlsmt, 3);
|
||||
tctc.pk = sqlite3_column_int(psqlsmt, 5);
|
||||
#ifdef DEBUG
|
||||
printf("[*]N:(%s) T:(%s) IN:(%d) IP:(%d)\n",tctc.name.data(),tctc.type.data(),tctc.notnull,tctc.pk);
|
||||
#endif
|
||||
if(tctc.name != sctc[i-1].name || tctc.type != sctc[i-1].type || tctc.notnull != sctc[i-1].notnull || tctc.pk != sctc[i-1].pk){
|
||||
throw "table abnormal";
|
||||
}
|
||||
sqlite3_step(psqlsmt);
|
||||
}
|
||||
}
|
||||
|
||||
/*******************************************************************
|
||||
*Public
|
||||
*************/
|
||||
|
||||
Proj::Proj(string t_projpath, string t_projfile){
|
||||
// 检查工程描述文件是否可读
|
||||
@ -741,3 +907,17 @@ void Proj::DBProcess(void){
|
||||
string Proj::GetName(void){
|
||||
return name;
|
||||
}
|
||||
|
||||
//更新工程
|
||||
void Proj::UpdateProcess(void){
|
||||
db_path = proj_path +"dbs/"+ name+".db";
|
||||
if(access(db_path.data(), F_OK) == -1){
|
||||
#ifdef DEBUG
|
||||
printf("Database Not Exist.\n");
|
||||
#endif
|
||||
throw "database not exist";
|
||||
}
|
||||
sqlite3_open(db_path.data(), &psql);
|
||||
// 检查数据库的完成性
|
||||
check_database();
|
||||
}
|
||||
|
||||
@ -82,7 +82,12 @@ bool config_search(vector<string> &configs,string tfg){
|
||||
}
|
||||
|
||||
int update(string instruct, vector<string> &configs, vector<string> &lconfigs, vector<string> &targets){
|
||||
|
||||
try {
|
||||
Proj nproj(targets[0], "netc.proj");
|
||||
nproj.UpdateProcess();
|
||||
} catch (const char *err_info) {
|
||||
printf("\033[31mError: %s\n\033[0m",err_info);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user