修复一些内存泄漏

This commit is contained in:
hugy 2023-02-06 11:21:13 +08:00
parent 1216733b82
commit ebde430aec
12 changed files with 98 additions and 47 deletions

View File

@ -5,6 +5,8 @@
#include "get_db_handle.h" #include "get_db_handle.h"
#include "wechat_data.h" #include "wechat_data.h"
#include "base64.h" #include "base64.h"
using namespace std;
#define WX_CHAT_ROOM_MGR_OFFSET 0x67ee70 #define WX_CHAT_ROOM_MGR_OFFSET 0x67ee70
#define WX_GET_CHAT_ROOM_DETAIL_INFO_OFFSET 0xa73a80 #define WX_GET_CHAT_ROOM_DETAIL_INFO_OFFSET 0xa73a80
#define WX_NEW_CHAT_ROOM_INFO_OFFSET 0xd07010 #define WX_NEW_CHAT_ROOM_INFO_OFFSET 0xd07010
@ -45,21 +47,54 @@ int GetChatRoomDetailInfo(wchar_t* chat_room_id, ChatRoomInfoInner& room_info) {
MOV success,EAX MOV success,EAX
POPAD POPAD
} }
room_info.chat_room_id.ptr = *(wchar_t**)(chat_room_info + 0x4); DWORD room_id_len = *(DWORD*)(chat_room_info + 0x8);
room_info.chat_room_id.length = *(DWORD*)(chat_room_info + 0x8); DWORD room_id_max_len = *(DWORD*)(chat_room_info + 0xC);
room_info.chat_room_id.max_length = *(DWORD*)(chat_room_info + 0xC); wchar_t * room_id = new wchar_t[room_id_len + 1];
wmemcpy(room_id,*(wchar_t**)(chat_room_info + 0x4),room_id_len + 1);
room_info.chat_room_id.ptr = room_id;
room_info.chat_room_id.length = room_id_len;
room_info.chat_room_id.max_length = room_id_max_len;
room_info.notice.ptr = *(wchar_t**)(chat_room_info + 0x18);
room_info.notice.length = *(DWORD*)(chat_room_info + 0x1C);
room_info.notice.max_length = *(DWORD*)(chat_room_info + 0x20);
room_info.admin.ptr = *(wchar_t**)(chat_room_info + 0x2C); DWORD notice_len = *(DWORD*)(chat_room_info + 0x1C);
room_info.admin.length = *(DWORD*)(chat_room_info + 0x30); DWORD notice_max_len = *(DWORD*)(chat_room_info + 0x20);
room_info.admin.max_length = *(DWORD*)(chat_room_info + 0x34); wchar_t* notice_ptr = *(wchar_t**)(chat_room_info + 0x18);
if(notice_len <= 0){
room_info.notice.ptr = nullptr;
}else{
wchar_t * notice = new wchar_t[notice_len + 1];
wmemcpy(notice,notice_ptr,notice_len+1);
room_info.notice.ptr = notice;
}
room_info.notice.length = notice_len;
room_info.notice.max_length = notice_max_len;
DWORD admin_len = *(DWORD*)(chat_room_info + 0x30);
DWORD admin_max_len = *(DWORD*)(chat_room_info + 0x34);
wchar_t* admin_ptr = *(wchar_t**)(chat_room_info + 0x2C);
if(admin_len <= 0){
room_info.admin.ptr = nullptr;
}else{
wchar_t * admin = new wchar_t[admin_len + 1];
wmemcpy(admin,admin_ptr,admin_len+1);
room_info.admin.ptr = admin;
}
room_info.admin.length = admin_len;
room_info.admin.max_length = admin_max_len;
DWORD xml_len = *(DWORD*)(chat_room_info + 0x54);
DWORD xml_max_len = *(DWORD*)(chat_room_info + 0x58);
wchar_t* xml_ptr = *(wchar_t**)(chat_room_info + 0x50);
if (xml_len <= 0){
room_info.xml.ptr = nullptr;
}else{
wchar_t * xml = new wchar_t[xml_len + 1];
wmemcpy(xml,xml_ptr,xml_len+1);
room_info.xml.ptr = xml;
}
room_info.xml.length = xml_len;
room_info.xml.max_length = xml_max_len;
room_info.xml.ptr = *(wchar_t**)(chat_room_info + 0x50);
room_info.xml.length = *(DWORD*)(chat_room_info + 0x54);
room_info.xml.max_length = *(DWORD*)(chat_room_info + 0x58);
__asm { __asm {
PUSHAD PUSHAD
LEA ECX,chat_room_info LEA ECX,chat_room_info

View File

@ -25,7 +25,7 @@ wstring utf8_to_unicode(const char *buffer) {
/// @param wstr unicode /// @param wstr unicode
/// @return string utf8 /// @return string utf8
string unicode_to_utf8(wchar_t *wstr) { string unicode_to_utf8(wchar_t *wstr) {
int c_size = WideCharToMultiByte(CP_UTF8, 0, wstr, -1, NULL, 0, NULL, FALSE); int c_size = WideCharToMultiByte(CP_UTF8, 0, wstr, -1, NULL, 0, NULL, FALSE);
if (c_size > 0) { if (c_size > 0) {
char *buffer = new char[c_size + 1]; char *buffer = new char[c_size + 1];
WideCharToMultiByte(CP_UTF8, 0, wstr, -1, buffer, c_size, NULL, FALSE); WideCharToMultiByte(CP_UTF8, 0, wstr, -1, buffer, c_size, NULL, FALSE);

View File

@ -1,18 +1,17 @@
#ifndef COMMON_H_ #ifndef COMMON_H_
#define COMMON_H_ #define COMMON_H_
#include <string> #include <string>
using namespace std; #define READ_WSTRING(addr, offset) ((*(DWORD *)(addr + offset + 0x4) == 0) ? std::wstring(L"") : std::wstring((wchar_t *)(*(DWORD *)(addr + offset)), *(DWORD *)(addr + offset + 0x4)))
#define READ_WSTRING(addr, offset) ((*(DWORD *)(addr + offset + 0x4) == 0) ? wstring(L"") : wstring((wchar_t *)(*(DWORD *)(addr + offset)), *(DWORD *)(addr + offset + 0x4)))
/// @brief utf8 转换成unicode /// @brief utf8 转换成unicode
/// @param buffer utf8 /// @param buffer utf8
/// @return unicode /// @return unicode
wstring utf8_to_unicode(const char *buffer); std::wstring utf8_to_unicode(const char *buffer);
/// @brief unicode转换utf8 /// @brief unicode转换utf8
/// @param wstr unicode /// @param wstr unicode
/// @return utf8 /// @return utf8
string unicode_to_utf8(wchar_t *wstr); std::string unicode_to_utf8(wchar_t *wstr);
/// @brief 获取WeChatWin.dll基址 /// @brief 获取WeChatWin.dll基址
/// @return 基址 /// @return 基址
@ -35,7 +34,7 @@ void UnHookAnyAddress(DWORD hook_addr, char *origin);
/// @brief get timeW /// @brief get timeW
/// @param timestamp timestamp /// @param timestamp timestamp
/// @return str /// @return str
wstring GetTimeW(long long timestamp); std::wstring GetTimeW(long long timestamp);
/// @brief unicode trans utf8 /// @brief unicode trans utf8
/// @param str unicode str /// @param str unicode str
/// @return utf8 str /// @return utf8 str
@ -43,11 +42,11 @@ std::string UnicodeToUtf8(const wchar_t *str);
/// @brief string convert wstring /// @brief string convert wstring
/// @param str /// @param str
/// @return /// @return
wstring String2Wstring(string str); std::wstring String2Wstring(std::string str);
/// @brief wstring convert string /// @brief wstring convert string
/// @param str /// @param str
/// @return /// @return
string Wstring2String(wstring wstr); std::string Wstring2String(std::wstring wstr);
/// @brief create dir /// @brief create dir
/// @param path /// @param path
@ -56,7 +55,7 @@ BOOL FindOrCreateDirectoryW(const wchar_t *path);
template <typename T1, typename T2> template <typename T1, typename T2>
vector<T1> split(T1 str, T2 letter) { std::vector<T1> split(T1 str, T2 letter) {
vector<T1> arr; vector<T1> arr;
size_t pos; size_t pos;
while ((pos = str.find_first_of(letter)) != T1::npos) { while ((pos = str.find_first_of(letter)) != T1::npos) {

View File

@ -3,7 +3,7 @@
#include "common.h" #include "common.h"
#include "wechat_data.h" #include "wechat_data.h"
using namespace std;
#define WX_CONTACT_MGR_INSTANCE_OFFSET 0x64dc30 #define WX_CONTACT_MGR_INSTANCE_OFFSET 0x64dc30
#define WX_CONTACT_GET_LIST_OFFSET 0xa9b000 #define WX_CONTACT_GET_LIST_OFFSET 0xa9b000
#define WX_CONTACT_DEL_OFFSET 0xa9ef40 #define WX_CONTACT_DEL_OFFSET 0xa9ef40

View File

@ -3,7 +3,7 @@
#include <vector> #include <vector>
#include "wechat_data.h" #include "wechat_data.h"
int GetAllContact(vector<Contact> &vec); int GetAllContact(std::vector<Contact> &vec);

View File

@ -4,7 +4,7 @@
#include "base64.h" #include "base64.h"
#include "common.h" #include "common.h"
#include "new_sqlite3.h" #include "new_sqlite3.h"
using namespace std;
/// @brief free data /// @brief free data
void FreeResult(vector<vector<SqlResult>> &data) { void FreeResult(vector<vector<SqlResult>> &data) {
@ -15,11 +15,11 @@ void FreeResult(vector<vector<SqlResult>> &data) {
for (unsigned j = 0; j < data[i].size(); j++) { for (unsigned j = 0; j < data[i].size(); j++) {
SqlResult *sr = (SqlResult *)&data[i][j]; SqlResult *sr = (SqlResult *)&data[i][j];
if (sr->column_name) { if (sr->column_name) {
delete sr->column_name; delete[] sr->column_name;
sr->column_name = NULL; sr->column_name = NULL;
} }
if (sr->content) { if (sr->content) {
delete sr->content; delete[] sr->content;
sr->content = NULL; sr->content = NULL;
} }
} }

View File

@ -2,7 +2,6 @@
#define DB_OPERATION_H_ #define DB_OPERATION_H_
#include <windows.h> #include <windows.h>
#include <vector> #include <vector>
using namespace std;
struct SqlResult { struct SqlResult {
char *column_name; char *column_name;
DWORD column_name_len; DWORD column_name_len;
@ -18,6 +17,6 @@ struct SqlResult {
/// @return /// @return
int ExecuteSQL(DWORD db, const char *sql, DWORD callback, void *data); int ExecuteSQL(DWORD db, const char *sql, DWORD callback, void *data);
int Select(DWORD db_hanle, const char *sql,vector<vector<string>> &query_result); int Select(DWORD db_hanle, const char *sql,std::vector<std::vector<std::string>> &query_result);
#endif #endif

View File

@ -2,7 +2,7 @@
#include "hook_img.h" #include "hook_img.h"
#include "common.h" #include "common.h"
using namespace std;
// #define WX_HOOK_IMG_OFFSET 0xd7eaa5 // #define WX_HOOK_IMG_OFFSET 0xd7eaa5
// #define WX_HOOK_IMG_NEXT_OFFSET 0xda56e0 // #define WX_HOOK_IMG_NEXT_OFFSET 0xda56e0

View File

@ -1,9 +1,8 @@
#ifndef HOOK_IMG_H_ #ifndef HOOK_IMG_H_
#define HOOK_IMG_H_ #define HOOK_IMG_H_
#include "windows.h" #include "windows.h"
using namespace std;
int HookImg(wstring save_path); int HookImg(std::wstring save_path);
int UnHookImg(); int UnHookImg();
int GetImgByName(wchar_t* file_path,wchar_t* save_dir); int GetImgByName(wchar_t* file_path,wchar_t* save_dir);

View File

@ -7,7 +7,7 @@
#define WX_INIT_OBJ_OFFSET 0x6cbab0 #define WX_INIT_OBJ_OFFSET 0x6cbab0
#define WX_OCR_MANAGER_OFFSET 0x6cff00 #define WX_OCR_MANAGER_OFFSET 0x6cff00
#define WX_DO_OCR_TASK_OFFSET 0x11e3210 #define WX_DO_OCR_TASK_OFFSET 0x11e3210
using namespace std;
int DoOCRTask(wchar_t *img_path, std::string &result) { int DoOCRTask(wchar_t *img_path, std::string &result) {
int success = -1; int success = -1;
WeChatString path(img_path); WeChatString path(img_path);

View File

@ -4,6 +4,7 @@
#include "common.h" #include "common.h"
#include "wechat_data.h" #include "wechat_data.h"
using namespace std;
#define WX_SELF_NAME_OFFSET 0x2C426E8 #define WX_SELF_NAME_OFFSET 0x2C426E8
#define WX_SELF_MOBILE_OFFSET 0x2C42658 #define WX_SELF_MOBILE_OFFSET 0x2C42658

View File

@ -4,7 +4,6 @@
// #include <windows.h> // #include <windows.h>
#include <vector> #include <vector>
using namespace std;
struct WeChatString { struct WeChatString {
wchar_t *ptr; wchar_t *ptr;
DWORD length; DWORD length;
@ -13,7 +12,7 @@ struct WeChatString {
DWORD c_len = 0; DWORD c_len = 0;
WeChatString() { WeChatString(NULL); } WeChatString() { WeChatString(NULL); }
WeChatString(wstring &s) { WeChatString(std::wstring &s) {
ptr = (wchar_t *)(s.c_str()); ptr = (wchar_t *)(s.c_str());
length = s.length(); length = s.length();
max_length = s.length() * 2; max_length = s.length() * 2;
@ -51,7 +50,7 @@ struct DatabaseInfo {
DWORD handle = 0; DWORD handle = 0;
wchar_t *db_name = NULL; wchar_t *db_name = NULL;
DWORD db_name_len = 0; DWORD db_name_len = 0;
vector<TableInfo> tables; std::vector<TableInfo> tables;
DWORD count = 0; DWORD count = 0;
DWORD extrainfo = 0; DWORD extrainfo = 0;
}; };
@ -101,6 +100,25 @@ struct ChatRoomInfoInner {
WeChatString notice; WeChatString notice;
WeChatString admin; WeChatString admin;
WeChatString xml; WeChatString xml;
~ChatRoomInfoInner(){
if(chat_room_id.ptr){
delete []chat_room_id.ptr;
chat_room_id.ptr = nullptr;
}
if(notice.ptr){
delete []notice.ptr;
notice.ptr = nullptr;
}
if(admin.ptr){
delete []admin.ptr;
admin.ptr = nullptr;
}
if(xml.ptr){
delete []xml.ptr;
xml.ptr = nullptr;
}
}
}; };
struct VectorInner { struct VectorInner {
@ -124,17 +142,17 @@ struct ChatRoomInner{
}; };
struct SelfInfoInner{ struct SelfInfoInner{
string name; std::string name;
string city; std::string city;
string province; std::string province;
string country; std::string country;
string account; std::string account;
string wxid; std::string wxid;
string mobile; std::string mobile;
string small_img; std::string small_img;
string big_img; std::string big_img;
string data_root_path; std::string data_root_path;
string data_save_path; std::string data_save_path;
string current_data_path; std::string current_data_path;
}; };
#endif #endif