cyonjan/WeChatFerry
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix GetDbTables

Browse Source
This commit is contained in:
Changhua 2022-10-16 22:08:05 +08:00
parent eb7ace08c1
commit f757c95e0d
1 changed files with 21 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
spy/exec_sql.cpp
View File

@ -72,9 +72,9 @@ typedef int(__cdecl *Sqlite3_finalize)(DWORD *);
static int cbGetTables(void *ret, int argc, char **argv, char **azColName) static int cbGetTables(void *ret, int argc, char **argv, char **azColName)
{ {
wcf::DbTables* tbls = (wcf::DbTables*)ret; wcf::DbTables *tbls = (wcf::DbTables *)ret;
wcf::DbTable *tbl = tbls->add_tables();
for (int i = 0; i < argc; i++) { for (int i = 0; i < argc; i++) {
wcf::DbTable* tbl = tbls->add_tables();
if (strcmp(azColName[i], "name") == 0) { if (strcmp(azColName[i], "name") == 0) {
tbl->set_name(argv[i] ? argv[i] : ""); tbl->set_name(argv[i] ? argv[i] : "");
} else if (strcmp(azColName[i], "sql") == 0) { } else if (strcmp(azColName[i], "sql") == 0) {
@ -95,7 +95,7 @@ dbMap_t GetDbHandles()
DWORD sqlHandleEndAddr = *(DWORD *)(sqlHandleBaseAddr + g_WxCalls.sql.end); DWORD sqlHandleEndAddr = *(DWORD *)(sqlHandleBaseAddr + g_WxCalls.sql.end);
while (sqlHandleBeginAddr < sqlHandleEndAddr) { while (sqlHandleBeginAddr < sqlHandleEndAddr) {
DWORD dwHandle = *(DWORD *)sqlHandleBeginAddr; DWORD dwHandle = *(DWORD *)sqlHandleBeginAddr;
string dbName = Wstring2String(wstring((wchar_t*)(*(DWORD*)(dwHandle + g_WxCalls.sql.name)))); string dbName = Wstring2String(wstring((wchar_t *)(*(DWORD *)(dwHandle + g_WxCalls.sql.name))));
DWORD handle = *(DWORD *)(dwHandle + g_WxCalls.sql.slot); DWORD handle = *(DWORD *)(dwHandle + g_WxCalls.sql.slot);
if (handle) { if (handle) {
dbMap[dbName] = handle; dbMap[dbName] = handle;
@ -106,20 +106,19 @@ dbMap_t GetDbHandles()
return dbMap; return dbMap;
} }
void GetDbNames(wcf::DbNames *names)
void GetDbNames(wcf::DbNames* names)
{ {
if (dbMap.empty()) { if (dbMap.empty()) {
dbMap = GetDbHandles(); dbMap = GetDbHandles();
} }
for (auto& [k, v] : dbMap) { for (auto &[k, v] : dbMap) {
auto* name = names->add_names(); auto *name = names->add_names();
name->assign(k); name->assign(k);
} }
} }
void GetDbTables(const string db, wcf::DbTables* tables) void GetDbTables(const string db, wcf::DbTables *tables)
{ {
if (dbMap.empty()) { if (dbMap.empty()) {
dbMap = GetDbHandles(); dbMap = GetDbHandles();
@ -130,13 +129,13 @@ void GetDbTables(const string db, wcf::DbTables* tables)
return; // DB not found return; // DB not found
} }
const char *sql = "select * from sqlite_master where type=\"table\";"; const char *sql = "select name, sql from sqlite_master where type=\"table\";";
Sqlite3_exec p_Sqlite3_exec = (Sqlite3_exec)(g_WeChatWinDllAddr + g_WxCalls.sql.exec); Sqlite3_exec p_Sqlite3_exec = (Sqlite3_exec)(g_WeChatWinDllAddr + g_WxCalls.sql.exec);
p_Sqlite3_exec(it->second, sql, (sqlite3_callback)cbGetTables, tables, 0); p_Sqlite3_exec(it->second, sql, (sqlite3_callback)cbGetTables, tables, 0);
} }
void ExecDbQuery(const string db, const string sql, wcf::DbRows* rows) void ExecDbQuery(const string db, const string sql, wcf::DbRows *rows)
{ {
Sqlite3_prepare func_prepare = (Sqlite3_prepare)(g_WeChatWinDllAddr + 0x14227F0); Sqlite3_prepare func_prepare = (Sqlite3_prepare)(g_WeChatWinDllAddr + 0x14227F0);
Sqlite3_step func_step = (Sqlite3_step)(g_WeChatWinDllAddr + 0x13EA780); Sqlite3_step func_step = (Sqlite3_step)(g_WeChatWinDllAddr + 0x13EA780);
@ -151,21 +150,21 @@ void ExecDbQuery(const string db, const string sql, wcf::DbRows* rows)
dbMap = GetDbHandles(); dbMap = GetDbHandles();
} }
DWORD* stmt; DWORD *stmt;
int rc = func_prepare(dbMap[db], sql.c_str(), -1, &stmt, 0); int rc = func_prepare(dbMap[db], sql.c_str(), -1, &stmt, 0);
if (rc != SQLITE_OK) { if (rc != SQLITE_OK) {
return; return;
} }
while (func_step(stmt) == SQLITE_ROW) { while (func_step(stmt) == SQLITE_ROW) {
wcf::DbRow* row = rows->add_rows(); wcf::DbRow *row = rows->add_rows();
int col_count = func_column_count(stmt); int col_count = func_column_count(stmt);
for (int i = 0; i < col_count; i++) { for (int i = 0; i < col_count; i++) {
wcf::DbField* field = row->add_fields(); wcf::DbField *field = row->add_fields();
field->set_type(func_column_type(stmt, i)); field->set_type(func_column_type(stmt, i));
field->set_column(func_column_name(stmt, i)); field->set_column(func_column_name(stmt, i));
int length = func_column_bytes(stmt, i); int length = func_column_bytes(stmt, i);
const void* blob = func_column_blob(stmt, i); const void *blob = func_column_blob(stmt, i);
if (length && (field->type() != 5)) { if (length && (field->type() != 5)) {
field->set_content(string((char *)blob, length)); field->set_content(string((char *)blob, length));
} }