Files
bb107-MemoryModulePP/test/test.cpp
T
Boring ed550ae6c7 1. Add Win10 forward export support
2. Fixed some bugs
2020-02-22 14:53:26 +08:00

117 lines
3.3 KiB
C++

#include "../MemoryModule/NativeFunctionsInternal.h"
//#include "../MemoryModule/LoadDllMemoryApi.h"
#ifndef NT_SUCCESS
#define NT_SUCCESS(Status) (((NTSTATUS)(Status)) >= 0)
#endif
#include <cstdio>
#pragma warning(disable:4996)
PLDR_DATA_TABLE_ENTRY_WIN10_2 RtlFindDllLdrEntry(LPCWSTR DllName) {
PLIST_ENTRY head = &NtCurrentPeb()->Ldr->InMemoryOrderModuleList, entry = head->Flink;
PLDR_DATA_TABLE_ENTRY_WIN10_2 cur = nullptr;
while (entry != head) {
cur = CONTAINING_RECORD(entry, LDR_DATA_TABLE_ENTRY_WIN10_2, InMemoryOrderLinks);
entry = entry->Flink;
if (!wcsicmp(DllName, cur->BaseDllName.Buffer))return cur;
}
return nullptr;
}
int main() {
LPVOID buffer;
size_t size;
FILE* f = fopen("a.dll", "rb");
if (!f)return 0;
_fseeki64(f, 0, SEEK_END);
if (!(size = _ftelli64(f))) {
fclose(f);
return 0;
}
_fseeki64(f, 0, SEEK_SET);
fread(buffer = new char[size], 1, size, f);
fclose(f);
HMEMORYMODULE m1 = nullptr, m2 = m1;
HMODULE hModule = nullptr;
FARPROC pfn = nullptr;
if (!NT_SUCCESS(NtLoadDllMemoryExW(&m1, nullptr, 0, buffer, size, L"kernel64", nullptr))) goto end;
if (!NT_SUCCESS(NtLoadDllMemoryExW(&m2, nullptr, 0, buffer, size, L"kernel128", nullptr))) goto end;
hModule = (HMODULE)m1;
pfn = (decltype(pfn))(GetProcAddress(hModule, "Socket")); //ws2_32.WSASocketW
pfn = (decltype(pfn))(GetProcAddress(hModule, "VerifyTruse")); //wintrust.WinVerifyTrust
hModule = (HMODULE)m2;
pfn = (decltype(pfn))(GetProcAddress(hModule, "Socket"));
pfn = (decltype(pfn))(GetProcAddress(hModule, "VerifyTruse"));
printf("pfn = %p\n", pfn);
end:
delete[]buffer;
if (m1)NtUnloadDllMemory(m1);
if (m2)NtUnloadDllMemory(m2);
return 0;
}
//#include <cstdio>
//#include "../MemoryModule/NativeFunctionsInternal.h"
//
//bool c;
//static thread_local int x = -1;
//
//DWORD WINAPI Thread(PVOID) {
// printf("[1] x = %d\n", x);
// x = 0;
// c = true;
// while (c)Sleep(100);
// return x;
//}
//
//int main() {
// x = 1;
// c = false;
// HANDLE hThread = CreateThread(nullptr, 0, Thread, nullptr, 0, nullptr);
// DWORD ex = 0;
// if (hThread) {
// while (!c)Sleep(100);
// printf("[0] x = %d\n", x);
// c = false;
// WaitForSingleObject(hThread, 0xffffffff);
// GetExitCodeThread(hThread, &ex);
// CloseHandle(hThread);
// printf("[0] Exit = %d\n", ex);
// }
//
// PLIST_ENTRY entry = &NtCurrentPeb()->Ldr->InLoadOrderModuleList;
// PLDR_DATA_TABLE_ENTRY_WIN7 data = nullptr;
//
// while (entry != entry->Flink) {
// entry = entry->Flink;
// data = CONTAINING_RECORD(entry, LDR_DATA_TABLE_ENTRY_WIN7, InLoadOrderLinks);
// }
//
// return 0;
//}
//#include "../MemoryModule/Native.h"
//#include <cstdio>
//
//static thread_local int x = 0xffccffdd;
//
//DWORD WINAPI Thread(PVOID) {
// printf("[1] ThreadLocalStoragePointer = %p\n", NtCurrentTeb()->ThreadLocalStoragePointer);
// return x == 0xffccffdd ? 0 : 1;
//}
//
//int main() {
// x = 2;
// printf("[0] ThreadLocalStoragePointer = %p\n", NtCurrentTeb()->ThreadLocalStoragePointer);
// HANDLE hThread = CreateThread(nullptr, 0, Thread, nullptr, 0, nullptr);
// DWORD ret = -1;
// if (hThread) {
// WaitForSingleObject(hThread, 0xffffffff);
// GetExitCodeThread(hThread, &ret);
// CloseHandle(hThread);
// return ret;
// }
// return -1;
//}