fork download

copy

bool LoadPE_DirectoryExists(LoadPE_CONTEXT* ctx, unsigned long id)
{
	return (ctx->pPeHdr->OptionalHeader.NumberOfRvaAndSizes - 1) >= id
		&& ctx->pPeHdr->OptionalHeader.DataDirectory[id].VirtualAddress;
}
 
bool LoadPE_HasTLS(LoadPE_CONTEXT* ctx)
{
	return LoadPE_DirectoryExists(ctx, IMAGE_DIRECTORY_ENTRY_TLS);
}
 
DWORD LoadPE_GetTLS(LoadPE_CONTEXT* ctx)
{
	return ctx->pPeHdr->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_TLS].VirtualAddress;
}
 
void LoadPE_ProcessTLS(LoadPE_CONTEXT* ctx)
{
	if (LoadPE_HasTLS(ctx))
	{
		PIMAGE_TLS_DIRECTORY32 tls = PIMAGE_TLS_DIRECTORY32(ctx->pbRealImageBase + LoadPE_GetTLS(ctx));
		std::cout << "Callbacks: " << std::endl;
		PIMAGE_TLS_CALLBACK* cb_addr = (PIMAGE_TLS_CALLBACK *)(tls->AddressOfCallBacks);
		while (*cb_addr++)
		{
			std::cout << cb_addr << std::endl;
		}
 
		PTEB teb = (PTEB)__readfsdword(0x18);
		teb->ThreadLocalStoragePointer = HeapAlloc(
			GetProcessHeap(),
			HEAP_ZERO_MEMORY,
			tls->EndAddressOfRawData - tls->StartAddressOfRawData
		);
 
		auto data_begin = tls->StartAddressOfRawData;
		auto data_end = tls->EndAddressOfRawData;
		while (data_begin != data_end)
		{
			std::cout << *(DWORD*)data_begin << " at " << data_begin << std::endl;
			((DWORD *)teb->ThreadLocalStoragePointer)[0] = data_begin;
			data_begin += 4;
		}
	}
}

prog.cpp:1:29: error: ‘LoadPE_CONTEXT’ was not declared in this scope
 bool LoadPE_DirectoryExists(LoadPE_CONTEXT* ctx, unsigned long id)
                             ^~~~~~~~~~~~~~
prog.cpp:1:45: error: ‘ctx’ was not declared in this scope
 bool LoadPE_DirectoryExists(LoadPE_CONTEXT* ctx, unsigned long id)
                                             ^~~
prog.cpp:1:50: error: expected primary-expression before ‘unsigned’
 bool LoadPE_DirectoryExists(LoadPE_CONTEXT* ctx, unsigned long id)
                                                  ^~~~~~~~
prog.cpp:1:66: error: expression list treated as compound expression in initializer [-fpermissive]
 bool LoadPE_DirectoryExists(LoadPE_CONTEXT* ctx, unsigned long id)
                                                                  ^
prog.cpp:7:20: error: ‘LoadPE_CONTEXT’ was not declared in this scope
 bool LoadPE_HasTLS(LoadPE_CONTEXT* ctx)
                    ^~~~~~~~~~~~~~
prog.cpp:7:36: error: ‘ctx’ was not declared in this scope
 bool LoadPE_HasTLS(LoadPE_CONTEXT* ctx)
                                    ^~~
prog.cpp:12:1: error: ‘DWORD’ does not name a type
 DWORD LoadPE_GetTLS(LoadPE_CONTEXT* ctx)
 ^~~~~
prog.cpp:17:24: error: variable or field ‘LoadPE_ProcessTLS’ declared void
 void LoadPE_ProcessTLS(LoadPE_CONTEXT* ctx)
                        ^~~~~~~~~~~~~~
prog.cpp:17:24: error: ‘LoadPE_CONTEXT’ was not declared in this scope
prog.cpp:17:40: error: ‘ctx’ was not declared in this scope
 void LoadPE_ProcessTLS(LoadPE_CONTEXT* ctx)
                                        ^~~

Standard output is empty