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

Support MediaMsgi.db

Browse Source
This commit is contained in:
Changhua 2024-06-16 15:28:34 +08:00
parent 9030017e5d
commit 7a8ee47473
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
WeChatFerry/spy/exec_sql.cpp
View File

@ -2,6 +2,7 @@
#include "exec_sql.h" #include "exec_sql.h"
#include "load_calls.h" #include "load_calls.h"
#include "log.h"
#include "sqlite3.h" #include "sqlite3.h"
#include "util.h" #include "util.h"
@ -40,9 +41,9 @@ static void GetMsgDbHandle(QWORD msgMgrAddr)
dbMap[dbname] = GET_QWORD(dbAddr + 0x78); dbMap[dbname] = GET_QWORD(dbAddr + 0x78);
// MediaMsgi.db // MediaMsgi.db
// QWORD mmdbAddr = GET_QWORD(dbAddr + 0x14); QWORD mmdbAddr = GET_QWORD(dbAddr + 0x20);
// string mmdbname = Wstring2String(GET_WSTRING(mmdbAddr + 0x4C)); string mmdbname = Wstring2String(GET_WSTRING(mmdbAddr + 0x78));
// dbMap[mmdbname] = GET_QWORD(mmdbAddr + 0x38); dbMap[mmdbname] = GET_QWORD(mmdbAddr + 0x50);
} }
} }
} }
@ -133,6 +134,12 @@ DbRows_t ExecDbQuery(const string db, const string sql)
} }
QWORD *stmt; QWORD *stmt;
QWORD handle = dbMap[db];
if (handle == 0){
LOG_WARN("Empty handle, retrying...");
dbMap = GetDbHandles();
}
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 rows; return rows;
@ -200,7 +207,7 @@ vector<uint8_t> GetAudioData(uint64_t id)
QWORD msgMgrAddr = GET_QWORD(g_WeChatWinDllAddr + OFFSET_DB_MSG_MGR); QWORD msgMgrAddr = GET_QWORD(g_WeChatWinDllAddr + OFFSET_DB_MSG_MGR);
QWORD dbIndex = GET_QWORD(msgMgrAddr + 0x68); QWORD dbIndex = GET_QWORD(msgMgrAddr + 0x68);
string sql = "SELECT Buf from Media WHERE Reserved0=" + to_string(id) + ";"; string sql = "SELECT Buf FROM Media WHERE Reserved0=" + to_string(id) + ";";
for (int i = dbIndex - 1; i >= 0; i--) { for (int i = dbIndex - 1; i >= 0; i--) {
string dbname = "MediaMSG" + to_string(i) + ".db"; string dbname = "MediaMSG" + to_string(i) + ".db";
DbRows_t rows = ExecDbQuery(dbname, sql); DbRows_t rows = ExecDbQuery(dbname, sql);