#include "Inject_Main.h"
#include "resource.h"
#include <Windows.h>
#include <TlHelp32.h>
#include <string>
#include <TCHAR.H>

using namespace std;


/// <summary>
/// 通過進(jìn)程名稱獲取該進(jìn)程句柄
/// </summary>
/// <param name="processName"></param>
/// <returns>成功返回 DWORD，失敗返回 0</returns>
DWORD GetProcessByName(CONST TCHAR* processName) {
? ? // 獲取到整個(gè)系統(tǒng)的進(jìn)程
? ? HANDLE processALL = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, NULL);

? ? // 定義一個(gè)容器，該容器用來接收，進(jìn)程信息
? ? PROCESSENTRY32W processInfo = { 0 };
? ? processInfo.dwSize = sizeof(PROCESSENTRY32W);

? ? // 根據(jù)進(jìn)程名稱，循環(huán)判斷是否是指定的進(jìn)程
? ? do
? ? {
? ? ? ? if (_tcscmp(processInfo.szExeFile, processName) == 0)
? ? ? ? {
? ? ? ? ? ? // 釋放進(jìn)程快照，防止內(nèi)存泄露
? ? ? ? ? ? CloseHandle(processALL);
? ? ? ? ? ? // 如果是返回指定進(jìn)程句柄
? ? ? ? ? ? return processInfo.th32ProcessID;
? ? ? ? }
? ? ? ? // 一個(gè)迭代函數(shù)
? ? } while (Process32Next(processALL, &processInfo));
? ? // 釋放進(jìn)程快照，防止內(nèi)存泄露
? ? CloseHandle(processALL);
? ? return 0;
}

/// <summary>
/// 獲取指定 DLL 的內(nèi)存地址
/// </summary>
/// <param name="pid"></param>
/// <param name="moduleName"></param>
/// <returns></returns>
HMODULE GetProcessModuleHandle(DWORD pid, CONST TCHAR* moduleName) {
? ? MODULEENTRY32 moduleEntry;
? ? HANDLE handle = NULL;
? ? handle = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, pid);
? ? if (!handle) {
? ? ? ? CloseHandle(handle);
? ? ? ? return NULL;
? ? }
? ? ZeroMemory(&moduleEntry, sizeof(MODULEENTRY32));
? ? moduleEntry.dwSize = sizeof(MODULEENTRY32);
? ? if (!Module32First(handle, &moduleEntry)) {
? ? ? ? CloseHandle(handle);
? ? ? ? return NULL;
? ? }
? ? do {
? ? ? ? if (_tcscmp(moduleEntry.szModule, moduleName) == 0) {
? ? ? ? ? ? // 釋放進(jìn)程快照，防止內(nèi)存泄露
? ? ? ? ? ? CloseHandle(handle);
? ? ? ? ? ? return moduleEntry.hModule;
? ? ? ? }
? ? } while (Module32Next(handle, &moduleEntry));
? ? CloseHandle(handle);
? ? return 0;
}

/// <summary>
/// 把指定DLL注入到指定進(jìn)程中
/// </summary>
/// <param name="processName">processName 進(jìn)程名稱</param>
/// <param name="dllPath">dllPath dll路徑</param>
void InjectDll(const wchar_t* processName, const char* dllPath) {
? ? // 獲取指定進(jìn)程的句柄
? ? DWORD dword = GetProcessByName(processName);
? ? if (dword == 0)
? ? {
? ? ? ? MessageBox(NULL, TEXT("沒有找到指定進(jìn)程"), TEXT("錯(cuò)誤"), 0);
? ? ? ? return;
? ? }
? ? // 打開指定進(jìn)程
? ? HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dword);
? ? if (hProcess == NULL)
? ? {
? ? ? ? MessageBox(NULL, TEXT("指定進(jìn)程打開失敗"), TEXT("錯(cuò)誤"), 0);
? ? ? ? return;
? ? }
? ? /*
? ? ? ? 在指定進(jìn)程的地址，開辟一塊內(nèi)存空間，用來保存 DLL的路徑信息
? ? ? ? LPVOID VirtualAllocEx(
? ? ? ? [in] ? ? ? ? ? HANDLE hProcess, 在那個(gè)進(jìn)程中開辟內(nèi)存
? ? ? ? [in, optional] LPVOID lpAddress, 開辟內(nèi)存的起始地址 (NULL，不需要控制起始位置)
? ? ? ? [in] ? ? ? ? ? SIZE_T dwSize, ?開辟內(nèi)存的大?。ó?dāng)前保存的內(nèi)容是 DLL的路徑）
? ? ? ? [in] ? ? ? ? ? DWORD ?flAllocationType, 內(nèi)存分配的類型。（開辟內(nèi)存）
? ? ? ? [in] ? ? ? ? ? DWORD ?flProtect，設(shè)置內(nèi)存的權(quán)限 （可讀可寫）
? ? ? ? );
? ? ?*/
? ? LPVOID DLLAddress = VirtualAllocEx(hProcess, NULL, strlen(dllPath), MEM_COMMIT, PAGE_READWRITE);
? ? /*
? ? ? ? 把DLL的路徑，寫入到剛開辟出來的內(nèi)存中
? ? ? ? BOOL WriteProcessMemory(
? ? ? ? [in] ?HANDLE ?hProcess, // 指定的進(jìn)程
? ? ? ? [in] ?LPVOID ?lpBaseAddress, // DLL路徑字符串，寫入的基址
? ? ? ? [in] ?LPCVOID lpBuffer, // DLL路徑字符串，的指針
? ? ? ? [in] ?SIZE_T ?nSize, // 需要寫入內(nèi)存的字節(jié)長(zhǎng)度
? ? ? ? [out] SIZE_T ?*lpNumberOfBytesWritten // [out] 返回一個(gè)指針，不需要，NULL
? ? ? ? );
? ? */

? ? if (WriteProcessMemory(hProcess, DLLAddress, dllPath, strlen(dllPath), NULL) == 0)
? ? {
? ? ? ? MessageBox(NULL, TEXT("路徑寫入失敗"), TEXT("錯(cuò)誤"), 0);
? ? ? ? return;
? ? }

? ? // 獲取 Kernel32.dll 這個(gè)模塊
? ? HMODULE k32 = GetModuleHandle(TEXT("Kernel32.dll"));
? ? // 在 Kernel32.dll 模塊中找到 LoadLibrary 這個(gè)函數(shù)的內(nèi)存地址
? ? LPVOID loadADD = GetProcAddress(k32, "LoadLibraryA");
? ? /*
? ? ? ? 在指定進(jìn)程中，創(chuàng)建一個(gè)線程
? ? ? ? 并通過這個(gè)線程，調(diào)用 LoadLibrary 函數(shù)
? ? ? ? 通過 LoadLibrary 函數(shù)，把 DLL 載入到目標(biāo)進(jìn)程中
? ? ? ? HANDLE CreateRemoteThread(
? ? ? ? [in] ?HANDLE ? ? ? ? ? ? ? ? hProcess, // 指定進(jìn)程
? ? ? ? [in] ?LPSECURITY_ATTRIBUTES ?lpThreadAttributes, // 設(shè)置線程安全屬性，表示線程是否可以繼承，NULL就夠了
? ? ? ? [in] ?SIZE_T ? ? ? ? ? ? ? ? dwStackSize, // 堆棧的初始大小，0 表示使用可執(zhí)行文件的默認(rèn)大小
? ? ? ? [in] ?LPTHREAD_START_ROUTINE lpStartAddress, // 遠(yuǎn)程進(jìn)程中，需要執(zhí)行的那個(gè)函數(shù)的指針
? ? ? ? [in] ?LPVOID ? ? ? ? ? ? ? ? lpParameter, // 目前進(jìn)程中 DLL路徑的指針
? ? ? ? [in] ?DWORD ? ? ? ? ? ? ? ? ?dwCreationFlags, // 0 線程在創(chuàng)建后立即運(yùn)行。
? ? ? ? [out] LPDWORD ? ? ? ? ? ? ? ?lpThreadId // [out] 當(dāng)前不需要這個(gè)返回值
? ? ? ? );
? ? */
? ? HANDLE hThread = CreateRemoteThread(hProcess, NULL, 0, (LPTHREAD_START_ROUTINE)loadADD, DLLAddress, 0, NULL);


? ? // 釋放指定的模塊
? ? CloseHandle(hThread);
? ? CloseHandle(hProcess);

}

/// <summary>
/// 把指定進(jìn)程中的DLL卸載掉
/// </summary>
/// <param name="processName"></param>
/// <param name="dllPath"></param>
void UnInjectDll(const wchar_t* processName) {
? ? // 通過進(jìn)程名稱獲取該進(jìn)程句柄
? ? DWORD dword = GetProcessByName(processName);
? ? if (dword == 0)
? ? {
? ? ? ? MessageBox(NULL, TEXT("沒有找到指定進(jìn)程"), TEXT("錯(cuò)誤"), 0);
? ? ? ? return;
? ? }
? ? // 獲取指定進(jìn)程中指定模塊的內(nèi)存地址
? ? HMODULE hmodule = GetProcessModuleHandle(dword, L"WX_Read_Write.dll");

? ? // 打開指定進(jìn)程
? ? HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dword);
? ? if (hProcess == NULL)
? ? {
? ? ? ? MessageBox(NULL, TEXT("指定進(jìn)程打開失敗"), TEXT("錯(cuò)誤"), 0);
? ? ? ? return;
? ? }

? ? // 獲取 Kernel32.dll 這個(gè)模塊
? ? HMODULE k32 = GetModuleHandle(TEXT("Kernel32.dll"));
? ? // 在 Kernel32.dll 模塊中找到 LoadLibrary 這個(gè)函數(shù)的內(nèi)存地址
? ? LPVOID loadADD = GetProcAddress(k32, "FreeLibrary");

? ? HANDLE hThread = CreateRemoteThread(hProcess, NULL, 0, (LPTHREAD_START_ROUTINE)loadADD, (LPVOID)hmodule, 0, NULL);

? ? // 釋放指定的模塊
? ? CloseHandle(hThread);
? ? CloseHandle(hProcess);
}


/// <summary>
///?
/// </summary>
/// <param name="hwndDlg"></param>
/// <param name="uMsg"></param>
/// <param name="wParam"></param>
/// <param name="lParam"></param>
/// <returns></returns>
INT_PTR CALLBACK dialogProc(_In_ HWND hwndDlg, _In_ UINT uMsg, _In_ WPARAM wParam, _In_ LPARAM lParam)
{
? ? wchar_t processName[100] = L"WeChat.exe";
? ? char dllPath[400] = { "C://Users//qiaoas//documents//visual studio 2015//Projects//ConsoleApplication1//Debug//WX_Read_Write.dll" };
? ? switch (uMsg)
? ? {
? ? case WM_INITDIALOG:
? ? ? ? break;
? ? case WM_CLOSE:
? ? ? ? EndDialog(hwndDlg, 0); // 關(guān)閉窗體
? ? ? ? break;
? ? case WM_COMMAND:
? ? ? ? /*GetDlgItemText(hwndDlg, Text_ProcessName, processName, sizeof(processName));
? ? ? ? GetDlgItemText(hwndDlg, Text_DLLPath, (LPWSTR)dllPath, sizeof(dllPath));*/

? ? ? ? if (wParam == Btn_Inject_DLL)
? ? ? ? {
? ? ? ? ? ? if (sizeof(processName) == 0)
? ? ? ? ? ? {
? ? ? ? ? ? ? ? MessageBox(NULL, TEXT("進(jìn)程名稱不能為空"), TEXT("錯(cuò)誤"), 0);
? ? ? ? ? ? }
? ? ? ? ? ? if (sizeof(dllPath) == 0)
? ? ? ? ? ? {
? ? ? ? ? ? ? ? MessageBox(NULL, TEXT("DLL路徑不能為空"), TEXT("錯(cuò)誤"), 0);
? ? ? ? ? ? }
? ? ? ? ? ? InjectDll(processName, dllPath); // 注入DLL
? ? ? ? }
? ? ? ? if (wParam == Btn_unInject_DLL)
? ? ? ? {
? ? ? ? ? ? UnInjectDll(processName); // 卸載DLL
? ? ? ? }
? ? ? ? break;
? ? default:
? ? ? ? break;
? ? }
? ? return FALSE;
}


/// <summary>
/// 初始化
/// </summary>
/// <param name="hInstance"></param>
/// <param name="hPrevInstance"></param>
/// <param name="lpCmdLine"></param>
/// <param name="nCmdShow"></param>
/// <returns></returns>
int APIENTRY wWinMain(_In_ HINSTANCE hInstance,
? ? _In_opt_ HINSTANCE hPrevInstance,
? ? _In_ LPWSTR ? ?lpCmdLine,
? ? _In_ int ? ? ? nCmdShow)
{
? ? DialogBox(hInstance, MAKEINTRESOURCE(IDD_DIALOG1), NULL, &dialogProc);
? ? return 0;
}

WINBASEAPI
_Ret_maybenull_
HMODULE
WINAPI
LoadLibraryA(
? ? _In_ LPCSTR lpLibFileName
? ? );

WINBASEAPI
_Ret_maybenull_
HMODULE
WINAPI
LoadLibraryW(
? ? _In_ LPCWSTR lpLibFileName
? ? );

#ifdef UNICODE
#define LoadLibrary ?LoadLibraryW
#else
#define LoadLibrary ?LoadLibraryA
#endif // !UNICODE