-
Notifications
You must be signed in to change notification settings - Fork 8
Expand file tree
/
Copy pathchdr.cpp
More file actions
1678 lines (1379 loc) · 63.3 KB
/
chdr.cpp
File metadata and controls
1678 lines (1379 loc) · 63.3 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
#include "chdr.h"
// Process_t definitions and functions.
namespace chdr
{
// Get target proces by name.
Process_t::Process_t(const wchar_t* m_wszProcessName, std::int32_t m_ParseType, DWORD m_dDesiredAccess)
{
HANDLE m_hSnapShot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, NULL);
PROCESSENTRY32 entry = { 0 };
entry.dwSize = sizeof(entry);
while (Process32Next(m_hSnapShot, &entry))
{
if (std::wcscmp(entry.szExeFile, m_wszProcessName) != 0)
continue;
this->m_nTargetProcessID = entry.th32ProcessID;
this->m_hTargetProcessHandle = OpenProcess(m_dDesiredAccess, false, this->m_nTargetProcessID);
break;
}
CloseHandle(m_hSnapShot);
CH_ASSERT(true, this->m_hTargetProcessHandle && this->m_hTargetProcessHandle != INVALID_HANDLE_VALUE,
"Couldn't obtain valid HANDLE for process %ws", m_wszProcessName);
this->m_bShouldFreeHandleAtDestructor = this->m_hTargetProcessHandle && this->m_hTargetProcessHandle != INVALID_HANDLE_VALUE;
this->m_eProcessArchitecture = this->GetProcessArchitecture_Internal();
this->m_szProcessPath = this->GetProcessPath_Internal();
this->m_szProcessName = this->GetProcessName_Internal();
this->m_PEHeaderData = PEHeaderData_t(*this, m_ParseType);
}
// Get target proces by PID.
Process_t::Process_t(std::uint32_t m_nProcessID, std::int32_t m_ParseType, DWORD m_dDesiredAccess)
{
this->m_nTargetProcessID = m_nProcessID;
this->m_hTargetProcessHandle = OpenProcess(m_dDesiredAccess, false, this->m_nTargetProcessID);
CH_ASSERT(true, this->m_hTargetProcessHandle && this->m_hTargetProcessHandle != INVALID_HANDLE_VALUE,
"Couldn't obtain valid HANDLE for PID %i", m_nProcessID);
this->m_bShouldFreeHandleAtDestructor = this->m_hTargetProcessHandle && this->m_hTargetProcessHandle != INVALID_HANDLE_VALUE;
this->m_eProcessArchitecture = this->GetProcessArchitecture_Internal();
this->m_szProcessPath = this->GetProcessPath_Internal();
this->m_szProcessName = this->GetProcessName_Internal();
this->m_PEHeaderData = PEHeaderData_t(*this, m_ParseType);
}
// Get target proces by HANDLE.
Process_t::Process_t(HANDLE m_hProcessHandle, std::int32_t m_ParseType)
{
this->m_hTargetProcessHandle = m_hProcessHandle;
this->m_nTargetProcessID = GetProcessId(this->m_hTargetProcessHandle);
this->m_bShouldFreeHandleAtDestructor = this->m_hTargetProcessHandle && this->m_hTargetProcessHandle != INVALID_HANDLE_VALUE;
this->m_eProcessArchitecture = this->GetProcessArchitecture_Internal();
this->m_szProcessPath = this->GetProcessPath_Internal();
this->m_szProcessName = this->GetProcessName_Internal();
this->m_PEHeaderData = PEHeaderData_t(*this, m_ParseType);
}
// Default dtor
Process_t::~Process_t()
{
// Just to be safe, free all of our allocated memory from this process.
if (!this->m_AllocatedMemoryTracker.empty())
{
for (const auto& ForgottenMemory : this->m_AllocatedMemoryTracker)
if (this->Free(ForgottenMemory.first)) // Has to be this ugly since I've included string enc in the log macros... :(
CH_LOG("Successful free of memory at 0x%X with size 0x%X", ForgottenMemory.first, ForgottenMemory.second)
else
CH_LOG("Couldn't free memory at 0x%X with size 0x%X", ForgottenMemory.first, ForgottenMemory.second);
}
// Not allowed to release this HANDLE, or was already released.
CH_ASSERT(true,
this->m_bShouldFreeHandleAtDestructor &&
this->m_hTargetProcessHandle &&
this->m_hTargetProcessHandle != INVALID_HANDLE_VALUE,
"adawdasd");
CloseHandle(m_hTargetProcessHandle);
}
// The process ID of the target process. (lol)
std::uint32_t Process_t::GetProcessID()
{
return this->m_nTargetProcessID;
}
// Ensure we found a HANDLE to the target process.
bool Process_t::IsValid()
{
return this->m_hTargetProcessHandle && this->m_hTargetProcessHandle != INVALID_HANDLE_VALUE;
}
// Is this process 32-bit running on 64-bit OS?
bool Process_t::IsWow64()
{
BOOL m_bIsWow64 = FALSE;
IsWow64Process(this->m_hTargetProcessHandle, &m_bIsWow64);
return m_bIsWow64;
}
// Get name of target process.
std::string Process_t::GetProcessName_Internal()
{
TCHAR m_szProcessNameBuffer[MAX_PATH];
if (!GetModuleBaseName(this->m_hTargetProcessHandle, NULL, m_szProcessNameBuffer, MAX_PATH))
return "";
m_szProcessNameBuffer[MAX_PATH - 1] = '\0';
// TCHAR->string
_bstr_t m_szPreProcessName(m_szProcessNameBuffer);
return std::string(m_szPreProcessName);
}
// The base address of the target process.
std::uintptr_t Process_t::GetBaseAddress()
{
for (auto& CurrentModule : this->EnumerateModules(true))
{
if (std::strcmp(CurrentModule.m_szName.c_str(), this->m_szProcessName.c_str()) != 0)
continue;
return CurrentModule.m_BaseAddress;
}
return NULL;
}
// Helper function to get name of target process.
std::string Process_t::GetName()
{
return this->m_szProcessName;
}
// Get filesystem path of target process.
std::string Process_t::GetProcessPath_Internal()
{
TCHAR m_szProcessPathBuffer[MAX_PATH];
if (!GetModuleFileNameEx(this->m_hTargetProcessHandle, NULL, m_szProcessPathBuffer, MAX_PATH))
return "";
m_szProcessPathBuffer[MAX_PATH - 1] = '\0';
// TCHAR->string
_bstr_t m_szPreProcessPath(m_szProcessPathBuffer);
return std::string(m_szPreProcessPath);
}
// Helper function to get filesystem path of target process.
std::string Process_t::GetPath()
{
return this->m_szProcessPath;
}
// Get architecture of target process.
Process_t::eProcessArchitecture Process_t::GetProcessArchitecture_Internal()
{
SYSTEM_INFO m_SystemInformation = { 0 };
GetNativeSystemInfo(&m_SystemInformation);
// Native x86, or WOW64 process.
if (m_SystemInformation.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_INTEL || this->IsWow64())
return eProcessArchitecture::ARCHITECTURE_x86;
// Everything else should be native x64.
return eProcessArchitecture::ARCHITECTURE_x64;
}
// Helper function to get architecture of target process.
Process_t::eProcessArchitecture Process_t::GetProcessArchitecture()
{
return this->m_eProcessArchitecture;
}
// Helper function to get PE header data of target process.
PEHeaderData_t Process_t::GetPEHeaderData()
{
return this->m_PEHeaderData;
}
// Did we suspend the target process ourselves?
bool Process_t::IsManuallySuspended()
{
return this->m_bIsProcessManuallySuspended;
}
// Is the target process suspended?
bool Process_t::IsSuspended()
{
// Traverse all threads, and ensure each is in a suspended state.
for (const auto& CurrentThread : this->EnumerateThreads())
{
if (CurrentThread.m_bIsThreadSuspended)
continue;
return false;
}
return true;
}
// Is the target process running under a debugger?
bool Process_t::IsBeingDebugged()
{
BOOL m_bHasRemoteDebugger = FALSE;
CheckRemoteDebuggerPresent(this->m_hTargetProcessHandle, &m_bHasRemoteDebugger);
return m_bHasRemoteDebugger;
}
// The PEB of the target process.
PEB Process_t::GetPEB()
{
NtQueryInformationProcess_fn NtQueryInformationProcess =
CH_R_CAST<NtQueryInformationProcess_fn>(GetProcAddress(GetModuleHandleA("ntdll.dll"), "NtQueryInformationProcess"));
PROCESS_BASIC_INFORMATION m_ProcessBasicInformation;
// Get address where PEB resides in this target process.
if (NtQueryInformationProcess(this->m_hTargetProcessHandle, PROCESSINFOCLASS::ProcessBasicInformation,
&m_ProcessBasicInformation, sizeof(PROCESS_BASIC_INFORMATION), nullptr) != 0x00000000/*STATUS_SUCCESS*/)
{
CH_LOG("NtQueryInformationProcess failure!");
return {};
}
// Read PEB from found base address.
const PEB m_PEB = this->Read<PEB>(CH_R_CAST<std::uintptr_t>(m_ProcessBasicInformation.PebBaseAddress));
return m_PEB;
}
// Custom struct to pass through as lpReserved to communicate extra data to module.
struct TransmittedData_t {
char szKey[256];
};
// Data to pass through our shellcode.
struct LoaderData_t {
std::uintptr_t m_ModuleBase = 0u, m_ImageBase = 0u, m_EntryPoint = 0u, m_LoadLibrary = 0u, m_GetProcAddress = 0u, m_Memset = 0u;
std::uint32_t m_RelocDirVA = 0u, m_RelocDirSize = 0u, m_ImportDirVA = 0u, m_PEHeaderSize = 0u, m_eInjectionFlags = 0u, m_Reason = 0u;
TransmittedData_t m_CustomTransmitted = {};
} LoaderData;
// Code to fix up needed data, then execute DllMain in target process.
void __stdcall Shellcode(LoaderData_t* m_LoaderData)
{
const std::uintptr_t m_TargetBase = m_LoaderData->m_ModuleBase;
const std::uintptr_t m_ImageBase = m_LoaderData->m_ImageBase;
// Calculate delta to relocate.
const std::uintptr_t m_Delta = m_TargetBase - m_ImageBase;
if (m_Delta && m_LoaderData->m_RelocDirVA)
// Relocate image.
{
PIMAGE_BASE_RELOCATION m_pRelocation = CH_R_CAST<PIMAGE_BASE_RELOCATION>(m_TargetBase + m_LoaderData->m_RelocDirVA);
PIMAGE_BASE_RELOCATION m_pRelocationEnd = CH_R_CAST<PIMAGE_BASE_RELOCATION>(CH_R_CAST<std::uintptr_t>(m_pRelocation) + m_LoaderData->m_RelocDirSize - sizeof(IMAGE_BASE_RELOCATION)/*?*/);
for (; m_pRelocation < m_pRelocationEnd;
m_pRelocation = CH_R_CAST<PIMAGE_BASE_RELOCATION>(CH_R_CAST<std::uintptr_t>(m_pRelocation) + m_pRelocation->SizeOfBlock))
{
std::uint16_t* m_pRelocationType = CH_R_CAST<std::uint16_t*>(m_pRelocation + 0x1);
const std::size_t m_nRelocationAmount = CH_S_CAST<std::size_t>((m_pRelocation->SizeOfBlock - sizeof(IMAGE_BASE_RELOCATION)) / sizeof(std::uint16_t));
for (std::size_t i = 0u; i < m_nRelocationAmount; ++i, ++m_pRelocationType)
{
switch (*m_pRelocationType >> 0xC)
{
#if defined (_WIN64)
case IMAGE_REL_BASED_DIR64:
#else
case IMAGE_REL_BASED_HIGHLOW:
#endif
*CH_R_CAST<std::uintptr_t*>(m_TargetBase + m_pRelocation->VirtualAddress + ((*m_pRelocationType) & 0xFFF)) += m_Delta;
break;
}
}
}
}
typedef int(__stdcall* DllMain_fn)(std::uintptr_t, std::uint32_t, void*);
DllMain_fn DllMain = CH_R_CAST<DllMain_fn>(m_TargetBase + m_LoaderData->m_EntryPoint);
if (!m_LoaderData->m_ImportDirVA)
{
// Call EP of our module.
DllMain(m_TargetBase, m_LoaderData->m_Reason, CH_R_CAST<void*>(&m_LoaderData->m_CustomTransmitted));
return;
}
typedef HMODULE(__stdcall* LoadLibraryA_fn)(LPCSTR);
LoadLibraryA_fn _LoadLibraryA = CH_R_CAST<LoadLibraryA_fn>(m_LoaderData->m_LoadLibrary);
typedef FARPROC(__stdcall* GetProcAddress_fn)(HMODULE, LPCSTR);
GetProcAddress_fn _GetProcAddress = CH_R_CAST<GetProcAddress_fn>(m_LoaderData->m_GetProcAddress);
// Fix up imports.
PIMAGE_IMPORT_DESCRIPTOR m_pImports = CH_R_CAST<PIMAGE_IMPORT_DESCRIPTOR>(m_TargetBase + m_LoaderData->m_ImportDirVA);
for (; m_pImports->Name; ++m_pImports)
{
const HMODULE m_hImportModule = _LoadLibraryA(CH_R_CAST<char*>(m_TargetBase + m_pImports->Name));
ULONG_PTR* m_pNameReference = CH_R_CAST<ULONG_PTR*>(m_TargetBase + m_pImports->OriginalFirstThunk);
ULONG_PTR* m_pThunk = CH_R_CAST<ULONG_PTR*>(m_TargetBase + m_pImports->FirstThunk);
for (; *m_pNameReference; ++m_pNameReference, ++m_pThunk)
{
if (IMAGE_SNAP_BY_ORDINAL(*m_pNameReference))
*CH_R_CAST<FARPROC*>(m_pThunk) = _GetProcAddress(m_hImportModule, CH_R_CAST<char*>(*m_pNameReference & 0xFFFF));
else
{
PIMAGE_IMPORT_BY_NAME m_pThunkData = CH_R_CAST<PIMAGE_IMPORT_BY_NAME>(m_TargetBase + *m_pNameReference);
*CH_R_CAST<FARPROC*>(m_pThunk) = _GetProcAddress(m_hImportModule, m_pThunkData->Name);
}
}
}
// Call EP of our module.
DllMain(m_TargetBase, m_LoaderData->m_Reason, CH_R_CAST<void*>(&m_LoaderData->m_CustomTransmitted));
typedef void*(__stdcall* memset_fn)(std::uintptr_t, std::int32_t, std::size_t);
memset_fn _memset = CH_R_CAST<memset_fn>(m_LoaderData->m_Memset);
if (m_LoaderData->m_eInjectionFlags & Process_t::eManualMapInjectionFlags::INJECTION_EXTRA_WIPEPEHEADERS)
_memset(m_TargetBase, NULL, m_LoaderData->m_PEHeaderSize);
if (m_LoaderData->m_eInjectionFlags & Process_t::eManualMapInjectionFlags::INJECTION_EXTRA_WIPEENTRYPOINT)
_memset(m_TargetBase + m_LoaderData->m_EntryPoint, NULL, 0x20u);
};
// Internal manual map function.
bool Process_t::ManualMapInject_Internal(std::uint8_t* m_ImageBuffer, std::int32_t m_eInjectionFlags)
{
// Grab DOS header.
const PIMAGE_DOS_HEADER m_pDosHeaders = CH_R_CAST<PIMAGE_DOS_HEADER>(m_ImageBuffer);
if (m_pDosHeaders->e_magic != IMAGE_DOS_SIGNATURE)
{
CH_LOG("Couldn't find IMAGE_DOS_SIGNATURE for m_ImageBuffer");
return false;
}
// Grab NT header.
const PIMAGE_NT_HEADERS m_pNTHeaders = CH_R_CAST<PIMAGE_NT_HEADERS>(m_ImageBuffer + m_pDosHeaders->e_lfanew);
if (m_pNTHeaders->Signature != IMAGE_NT_SIGNATURE)
{
CH_LOG("Couldn't find IMAGE_NT_SIGNATURE for m_ImageBuffer");
return false;
}
// Address our module will be in context of target process.
const std::uintptr_t m_TargetBaseAddress = this->Allocate(m_pNTHeaders->OptionalHeader.SizeOfImage, PAGE_EXECUTE_READWRITE, false);
// Copy over PE Header to target process.
if (!this->Write(m_TargetBaseAddress, m_ImageBuffer, m_pNTHeaders->OptionalHeader.SizeOfImage))
{
CH_LOG("Couldn't copy over PE header data to target process.");
return false;
}
// Copy over needed sections to target process.
PIMAGE_SECTION_HEADER m_pSectionHeaders = IMAGE_FIRST_SECTION(m_pNTHeaders);
for (std::size_t i = 0u; i < m_pNTHeaders->FileHeader.NumberOfSections; ++i, ++m_pSectionHeaders)
{
const std::uintptr_t m_Address = m_TargetBaseAddress + m_pSectionHeaders->VirtualAddress;
if (!this->Write(m_Address, &m_ImageBuffer[m_pSectionHeaders->PointerToRawData], m_pSectionHeaders->SizeOfRawData))
{
CH_LOG("Couldn't copy over %s section's data to target process.", CH_R_CAST<char*>(m_pSectionHeaders->Name));
return false;
}
}
// Can use this to communicate some data to the target process.
// lpReserved parameter in DllMain is unused, so popping our custom struct here is all good.
TransmittedData_t m_CustomTransmittedData = {};
strcpy_s<256>(m_CustomTransmittedData.szKey, "Hello from the other side!");
// Populate loader data structure.
LoaderData = {
m_TargetBaseAddress,
CH_S_CAST<std::uintptr_t>(m_pNTHeaders->OptionalHeader.ImageBase),
CH_S_CAST<std::uintptr_t>(m_pNTHeaders->OptionalHeader.AddressOfEntryPoint),
CH_R_CAST<std::uintptr_t>(LoadLibraryA),
CH_R_CAST<std::uintptr_t>(GetProcAddress),
CH_R_CAST<std::uintptr_t>(std::memset),
CH_S_CAST<std::uint32_t>(m_pNTHeaders->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress),
CH_S_CAST<std::uint32_t>(m_pNTHeaders->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].Size),
CH_S_CAST<std::uint32_t>(m_pNTHeaders->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress),
CH_S_CAST<std::uint32_t>(m_pNTHeaders->OptionalHeader.SizeOfHeaders),
CH_S_CAST<std::uint32_t>(m_eInjectionFlags),
DLL_PROCESS_ATTACH,
NULL
};
if (m_eInjectionFlags & eManualMapInjectionFlags::INJECTION_EXTRA_CUSTOMARGUMENTS)
LoaderData.m_CustomTransmitted = m_CustomTransmittedData;
// Fix up data if we've built with mismatched architecture.
if (m_pNTHeaders->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR32_MAGIC)
{
#if defined (_WIN64) // Didn't already have NT_HEADERS32 data.
const PIMAGE_NT_HEADERS32 m_NT32Temporary = CH_R_CAST<PIMAGE_NT_HEADERS32>(m_pNTHeaders);
LoaderData.m_EntryPoint = CH_S_CAST<std::uintptr_t>(m_NT32Temporary->OptionalHeader.AddressOfEntryPoint);
LoaderData.m_RelocDirVA = CH_S_CAST<std::uint32_t>(m_NT32Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress);
LoaderData.m_RelocDirSize = CH_S_CAST<std::uint32_t>(m_NT32Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].Size);
LoaderData.m_ImportDirVA = CH_S_CAST<std::uint32_t>(m_NT32Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress);
#endif
}
else if (m_pNTHeaders->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)
{
#if defined (_WIN32) // Didn't already have NT_HEADERS64 data.
const PIMAGE_NT_HEADERS64 m_NT64Temporary = CH_R_CAST<PIMAGE_NT_HEADERS64>(m_pNTHeaders);
LoaderData.m_EntryPoint = CH_S_CAST<std::uintptr_t>(m_NT64Temporary->OptionalHeader.AddressOfEntryPoint);
LoaderData.m_RelocDirVA = CH_S_CAST<std::uint32_t>(m_NT64Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress);
LoaderData.m_RelocDirSize = CH_S_CAST<std::uint32_t>(m_NT64Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].Size);
LoaderData.m_ImportDirVA = CH_S_CAST<std::uint32_t>(m_NT64Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress);
#endif
}
// Address our loader data will be in context of target process.
const std::uintptr_t m_LoaderDataAddress = this->Allocate(sizeof(LoaderData_t), PAGE_EXECUTE_READWRITE, false);
// Copy over loader data to target process.
if (!this->Write(m_LoaderDataAddress, &LoaderData, sizeof(LoaderData_t)))
{
CH_LOG("Couldn't copy over loader data to target process.");
return false;
}
if (m_eInjectionFlags & eManualMapInjectionFlags::INJECTION_MODE_THREADHIJACK)
// Hijack an existing thread in target process to execute our shellcode.
{
}
else
// Simply CreateRemoteThread in target process to execute our shellcode.
{
// Address our shellcode will be in context of target process.
const std::uintptr_t m_ShellcodeAddress = this->Allocate(4096, PAGE_EXECUTE_READWRITE, false);
// Copy over shellcode to target process.
if (!this->Write(m_ShellcodeAddress, Shellcode, 4096))
{
CH_LOG("Couldn't copy over loader shellcode to target process.");
return false;
}
const std::int32_t m_nThreadResult = this->_CreateRemoteThread(CH_R_CAST<LPVOID>(m_ShellcodeAddress), CH_R_CAST<LPVOID>(m_LoaderDataAddress));
if (m_nThreadResult <= 0)
{
CH_LOG("Failed to create remote thread in target process with code %i.", m_nThreadResult);
return false;
}
}
return true;
}
// Manual map injection from module on disk.
bool Process_t::ManualMapInject(const char* m_szDLLPath, std::int32_t m_eInjectionFlags)
{
ByteArray_t m_FileImageBuffer = { 0 };
// Fill local image buffer from file on disk.
std::ifstream m_fFile(m_szDLLPath, std::ios::binary);
(&m_FileImageBuffer)->assign((std::istreambuf_iterator<char>(m_fFile)), std::istreambuf_iterator<char>());
m_fFile.close();
if (!m_FileImageBuffer.size())
{
CH_LOG("Couldn't parse desired m_ImageBuffer to manual map.");
return false;
}
return ManualMapInject_Internal(m_FileImageBuffer.data(), m_eInjectionFlags);
}
// Manual map injection from module in memory.
bool Process_t::ManualMapInject(std::uint8_t* m_ImageBuffer, std::int32_t m_eInjectionFlags)
{
return ManualMapInject_Internal(m_ImageBuffer, m_eInjectionFlags);
}
// Manual map injection from ImageFile_t.
bool Process_t::ManualMapInject(ImageFile_t& m_ImageFile, std::int32_t m_eInjectionFlags)
{
if (!m_ImageFile.m_ImageBuffer.size())
{
CH_LOG("Couldn't parse desired ImageFile_t to manual map.");
return false;
}
return ManualMapInject_Internal(m_ImageFile.m_ImageBuffer.data(), m_eInjectionFlags);
}
// LoadLibrary injection from module on disk.
bool Process_t::LoadLibraryInject(const char* m_szDLLPath)
{
// Allocate memory in target process.
const std::uintptr_t m_AllocatedMemory = this->Allocate(strlen(m_szDLLPath), PAGE_READWRITE);
if (!m_AllocatedMemory)
{
CH_LOG("Couldn't allocate memory to target process. Error code was #%i", GetLastError());
return false;
}
// Write string name for our module in previously allocated space.
const std::size_t m_nWrittenBytes = this->Write(m_AllocatedMemory, CH_C_CAST<char*>(m_szDLLPath));
if (!m_nWrittenBytes)
{
CH_LOG("Couldn't write module name to target process. Error code was #%i", GetLastError());
return false;
}
// Load DLL by invoking LoadLibrary(m_szDLLPath) in a target process
const std::int32_t m_nThreadResult = this->_CreateRemoteThread(CH_R_CAST<LPVOID>(LoadLibraryA), CH_R_CAST<LPVOID>(m_AllocatedMemory));
if (m_nThreadResult <= 0)
{
CH_LOG("Failed to create remote thread in target process with code %i.", m_nThreadResult);
return false;
}
return true;
}
// Traverse and cache data about all threads in a target process.
std::vector<Process_t::ThreadInformation_t> Process_t::EnumerateThreads()
{
std::vector<Process_t::ThreadInformation_t> m_EnumeratedThreads = {};
HANDLE m_hSnapShot = CreateToolhelp32Snapshot(TH32CS_SNAPTHREAD, this->m_nTargetProcessID);
Thread_t::NtQueryInformationThread_fn NtQueryInformationThread =
CH_R_CAST<Thread_t::NtQueryInformationThread_fn>(GetProcAddress(GetModuleHandleA("ntdll.dll"), "NtQueryInformationThread"));
THREADENTRY32 mEntry = { 0 };
mEntry.dwSize = sizeof(mEntry);
if (!Thread32First(m_hSnapShot, &mEntry))
return {};
while (Thread32Next(m_hSnapShot, &mEntry))
{
// Ensure our target process owns this thread.
if (mEntry.th32OwnerProcessID != this->m_nTargetProcessID)
continue;
// Open handle to this specific thread.
HANDLE m_hThreadHandle = OpenThread(THREAD_ALL_ACCESS, FALSE, mEntry.th32ThreadID);
if (!m_hThreadHandle || m_hThreadHandle == INVALID_HANDLE_VALUE)
continue;
ADD_SCOPE_HANDLER(CloseHandle, m_hThreadHandle);
DWORD m_dThreadStartAddress = 0;
if (NtQueryInformationThread(m_hThreadHandle,
Thread_t::THREADINFOCLASS::ThreadQuerySetWin32StartAddress,
&m_dThreadStartAddress,
this->m_eProcessArchitecture == eProcessArchitecture::ARCHITECTURE_x64 ? sizeof(DWORD) * 2 : sizeof(DWORD),
nullptr) != 0x00000000/*STATUS_SUCCESS*/)
continue;
const bool m_bIsThreadSuspended = WaitForSingleObject(m_hThreadHandle, 0) == WAIT_ABANDONED;
m_EnumeratedThreads.push_back(
{ mEntry.th32ThreadID,
m_dThreadStartAddress,
m_bIsThreadSuspended }
);
}
CloseHandle(m_hSnapShot);
return m_EnumeratedThreads;
}
// Traverse and cache data about all loaded modules in a target process.
std::vector<Process_t::ModuleInformation_t> Process_t::EnumerateModules(bool m_bUseCachedData)
{
// To take up less processing power, or if you know nothing will be loaded into the target process unexpectedly.
if (m_bUseCachedData && this->m_bHasCachedProcessesModules)
{
if (!this->m_EnumeratedModulesCached.empty())
return this->m_EnumeratedModulesCached;
}
if (!this->m_EnumeratedModulesCached.empty())
// Wipe any previously cached data.
this->m_EnumeratedModulesCached.clear();
HANDLE m_hSnapShot = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, this->m_nTargetProcessID);
MODULEENTRY32 mEntry = { 0 };
mEntry.dwSize = sizeof(mEntry);
while (Module32Next(m_hSnapShot, &mEntry))
{
wchar_t m_wszModPath[MAX_PATH];
if (!GetModuleFileNameEx(this->m_hTargetProcessHandle, mEntry.hModule, m_wszModPath, sizeof(m_wszModPath) / sizeof(wchar_t)))
continue;
m_wszModPath[MAX_PATH - 1] = '\0';
// Convert wstring->string.
_bstr_t m_bszPreModulePath(m_wszModPath);
_bstr_t m_bszPreModuleName(mEntry.szModule);
std::string m_szModulePath(m_bszPreModulePath);
std::string m_szModuleName(m_bszPreModuleName);
this->m_EnumeratedModulesCached.push_back({
m_szModuleName, m_szModulePath,
CH_S_CAST<std::uint32_t>(mEntry.modBaseSize),
CH_R_CAST<std::uintptr_t>(mEntry.modBaseAddr) }
);
}
CloseHandle(m_hSnapShot);
return this->m_EnumeratedModulesCached;
}
// Sets debug privileges of a target process.
bool Process_t::SetDebugPrivilege(bool m_bShouldEnable)
{
HANDLE m_hToken;
if (!OpenProcessToken(this->m_hTargetProcessHandle, TOKEN_ALL_ACCESS, &m_hToken))
return false;
ADD_SCOPE_HANDLER(CloseHandle, m_hToken);
LUID m_LUID;
if (!LookupPrivilegeValue(NULL, L"SeDebugPrivilege", &m_LUID))
return false;
TOKEN_PRIVILEGES m_TokenPrivileges;
m_TokenPrivileges.PrivilegeCount = 1;
m_TokenPrivileges.Privileges[0].Luid = m_LUID;
m_TokenPrivileges.Privileges[0].Attributes = m_bShouldEnable ? SE_PRIVILEGE_ENABLED : 0;
const bool result =
AdjustTokenPrivileges(m_hToken, false, &m_TokenPrivileges, sizeof(m_TokenPrivileges), NULL, NULL)
&& GetLastError() != ERROR_NOT_ALL_ASSIGNED;
return result;
}
// Suspend every thread in a target process.
void Process_t::Suspend()
{
NtSuspendProcess_fn NtSuspendProcess = CH_R_CAST<NtSuspendProcess_fn>(GetProcAddress(GetModuleHandleA("ntdll.dll"), "NtSuspendProcess"));
this->m_bIsProcessManuallySuspended = NtSuspendProcess(this->m_hTargetProcessHandle) == 0x00000000/*STATUS_SUCCESS*/;
CH_ASSERT(false, this->m_bIsProcessManuallySuspended, "Failed to suspend process!");
}
// Resume every previously suspended thread in a target process.
void Process_t::Resume()
{
// TODO: Is there any use case of resuming a suspended process (that WE didn't suspend??).
CH_ASSERT(true, this->m_bIsProcessManuallySuspended, "Attempted to resume process that was never suspended!");
NtResumeProcess_fn NtResumeProcess = CH_R_CAST<NtResumeProcess_fn>(GetProcAddress(GetModuleHandleA("ntdll.dll"), "NtResumeProcess"));
this->m_bIsProcessManuallySuspended = NtResumeProcess(this->m_hTargetProcessHandle) != 0x00000000/*STATUS_SUCCESS*/;
CH_ASSERT(false, this->m_bIsProcessManuallySuspended == false, "Failed to resume suspended process!");
}
// Get desired export address by name.
std::uintptr_t Process_t::GetRemoteProcAddress(const char* m_szModuleName, const char* m_szExportName)
{
chdr::Module_t m_Module(*this,
m_szModuleName,
PEHeaderData_t::PEHEADER_PARSING_TYPE::TYPE_EXPORT_DIRECTORY // Only parse exports.
);
if (!m_Module.IsValid() || !m_Module.GetPEHeaderData().IsValid())
return 0u;
auto ExportData = m_Module.GetPEHeaderData().GetExportData();
if (ExportData.empty() || ExportData.find(m_szExportName) == ExportData.end())
// Ensure we even have this export in our map.
return 0u;
return ExportData[m_szExportName].m_nAddress;
}
// VirtualAllocEx implementation.
std::uintptr_t Process_t::Allocate(std::size_t m_AllocationSize, DWORD m_dProtectionType, bool m_bShouldTrack)
{
const std::uintptr_t m_AllocatedMemory = CH_R_CAST<std::uintptr_t>(VirtualAllocEx(this->m_hTargetProcessHandle, nullptr, m_AllocationSize, MEM_COMMIT | MEM_RESERVE, m_dProtectionType));
if (m_AllocatedMemory && m_bShouldTrack)
this->m_AllocatedMemoryTracker.insert({ m_AllocatedMemory, m_AllocationSize });
return m_AllocatedMemory;
}
// VirtualFreeEx implementation.
bool Process_t::Free(std::uintptr_t m_FreeAddress)
{
const bool m_bDidFree = VirtualFreeEx(this->m_hTargetProcessHandle, (LPVOID)m_FreeAddress, NULL, MEM_RELEASE);
if (m_bDidFree && this->m_AllocatedMemoryTracker.find(m_FreeAddress) != this->m_AllocatedMemoryTracker.end())
this->m_AllocatedMemoryTracker.erase(this->m_AllocatedMemoryTracker.find(m_FreeAddress));
return m_bDidFree;
}
// VirtualQueryEx implementation.
std::size_t Process_t::Query(LPCVOID m_QueryAddress, MEMORY_BASIC_INFORMATION* m_MemoryInformation)
{
return VirtualQueryEx(this->m_hTargetProcessHandle, m_QueryAddress, m_MemoryInformation, sizeof(MEMORY_BASIC_INFORMATION));
}
// CreateRemoteThread implementation.
std::int32_t Process_t::_CreateRemoteThread(LPVOID m_lpStartAddress, LPVOID m_lpParameter)
{
HANDLE m_hRemoteThread = CreateRemoteThread(this->m_hTargetProcessHandle,
NULL,
NULL,
CH_R_CAST<LPTHREAD_START_ROUTINE>(m_lpStartAddress),
m_lpParameter,
NULL,
NULL
);
if (!m_hRemoteThread || m_hRemoteThread == INVALID_HANDLE_VALUE)
return 0;
ADD_SCOPE_HANDLER(CloseHandle, m_hRemoteThread);
if (WaitForSingleObject(m_hRemoteThread, INFINITE) == WAIT_FAILED)
return -1;
return 1;
}
// GetModule implementation.
Module_t& Process_t::GetModule(const char* m_szModuleName, std::int32_t m_ParseType)
{
if (!IsValid())
CH_LOG("Invalid process called GetModule : %s", m_szProcessName);
if (m_AllocatedModules.count(m_szModuleName) > 0u)
return m_AllocatedModules[m_szModuleName];
Module_t m_Module(*this, m_szModuleName, m_ParseType);
m_AllocatedModules[m_szModuleName] = m_Module;
return m_AllocatedModules[m_szModuleName];
}
}
// PEHeaderData_t definitions and functions.
namespace chdr
{
// Parsing data out of this image's buffer.
PEHeaderData_t::PEHeaderData_t(std::uint8_t* m_ImageBuffer, std::size_t m_ImageSize, std::int32_t m_ParseType)
{
if (m_ParseType & PEHEADER_PARSING_TYPE::TYPE_NONE)
return;
CH_ASSERT(true, m_ImageBuffer && m_ImageSize, "Failed to read PE image.");
this->m_pDOSHeaders = CH_R_CAST<PIMAGE_DOS_HEADER>(m_ImageBuffer);
this->m_pNTHeaders = CH_R_CAST<PIMAGE_NT_HEADERS>(m_ImageBuffer + this->m_pDOSHeaders->e_lfanew);
this->m_bIsValidInternal =
this->m_pDOSHeaders->e_magic == IMAGE_DOS_SIGNATURE &&
this->m_pNTHeaders->Signature == IMAGE_NT_SIGNATURE;
// Ensure image PE headers was valid.
CH_ASSERT(true, this->IsValid(), "Couldn't find MZ&NT header.");
PIMAGE_SECTION_HEADER m_pSectionHeaders = IMAGE_FIRST_SECTION(this->m_pNTHeaders);
for (std::size_t i = 0u; i < m_pNTHeaders->FileHeader.NumberOfSections; ++i, ++m_pSectionHeaders)
{
this->m_SectionData.push_back(
{ CH_R_CAST<char*>(m_pSectionHeaders->Name),
m_pSectionHeaders->VirtualAddress,
m_pSectionHeaders->Misc.VirtualSize,
m_pSectionHeaders->Characteristics,
m_pSectionHeaders->PointerToRawData,
m_pSectionHeaders->PointerToRawData }
);
}
for (std::size_t i = 0u; i < IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR/*Maximum*/; ++i)
this->m_DirectoryData.push_back(this->m_pNTHeaders->OptionalHeader.DataDirectory[i]);
// Gather all needed directories, then ensure we're using the correct type for the image.
IMAGE_DATA_DIRECTORY m_ExportDataDirectory = this->GetDataDirectory(IMAGE_DIRECTORY_ENTRY_EXPORT);
IMAGE_DATA_DIRECTORY m_ImportDataDirectory = this->GetDataDirectory(IMAGE_DIRECTORY_ENTRY_IMPORT);
IMAGE_DATA_DIRECTORY m_DebugDataDirectory = this->GetDataDirectory(IMAGE_DIRECTORY_ENTRY_DEBUG);
if (this->m_pNTHeaders->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR32_MAGIC)
{
#if defined (_WIN64) // Didn't already have NT_HEADERS32 data.
const PIMAGE_NT_HEADERS32 m_NT32Temporary = CH_R_CAST<PIMAGE_NT_HEADERS32>(this->m_pNTHeaders);
m_ExportDataDirectory = m_NT32Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT];
m_ImportDataDirectory = m_NT32Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT];
m_DebugDataDirectory = m_NT32Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG];
if (!this->m_DirectoryData.empty())
// Reset directory data to be filled with one of correct architecture.
this->m_DirectoryData.clear();
for (std::size_t i = 0u; i < IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR/*Maximum*/; ++i)
this->m_DirectoryData.push_back(m_NT32Temporary->OptionalHeader.DataDirectory[i]);
#endif
}
else if (this->m_pNTHeaders->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)
{
#if defined (_WIN32) // Didn't already have NT_HEADERS64 data.
const PIMAGE_NT_HEADERS64 m_NT64Temporary = CH_R_CAST<PIMAGE_NT_HEADERS64>(this->m_pNTHeaders);
m_ExportDataDirectory = m_NT64Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT];
m_ImportDataDirectory = m_NT64Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT];
m_DebugDataDirectory = m_NT64Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG];
if (!this->m_DirectoryData.empty())
// Reset directory data to be filled with one of correct architecture.
this->m_DirectoryData.clear();
for (std::size_t i = 0u; i < IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR/*Maximum*/; ++i)
this->m_DirectoryData.push_back(m_NT64Temporary->OptionalHeader.DataDirectory[i]);
#endif
}
const bool bDebugEnabled = m_ParseType & PEHEADER_PARSING_TYPE::TYPE_DEBUG_DIRECTORY || m_ParseType & PEHEADER_PARSING_TYPE::TYPE_ALL;
if (!bDebugEnabled ||
!m_DebugDataDirectory.VirtualAddress ||
!m_DebugDataDirectory.Size)
{
if (bDebugEnabled)
CH_LOG("Debug table didn't exist for current region. 0x%x | 0X%x", m_DebugDataDirectory.VirtualAddress, m_DebugDataDirectory.Size);
}
else // Debug table parsing.
{
const PIMAGE_DEBUG_DIRECTORY m_DebugDirectoryData = CH_R_CAST<PIMAGE_DEBUG_DIRECTORY>(m_ImageBuffer + this->RvaToOffset(m_DebugDataDirectory.VirtualAddress));
if (m_DebugDirectoryData->Type == IMAGE_DEBUG_TYPE_CODEVIEW)
{
const CV_INFO_PDB70* m_PDBInfo = CH_R_CAST<CV_INFO_PDB70*>(m_ImageBuffer + this->RvaToOffset(m_DebugDirectoryData->AddressOfRawData));
wchar_t m_wszGUIDStr[MAX_PATH];
if (!StringFromGUID2(m_PDBInfo->Signature, m_wszGUIDStr, sizeof(m_wszGUIDStr)))
// Couldn't read GUID, whatever lol.
this->m_DebugData = { (char*)m_PDBInfo->PdbFileName, "", m_PDBInfo->Age, m_PDBInfo->CvSignature };
else
{
m_wszGUIDStr[MAX_PATH - 1] = '\0';
// wchar_t->string
_bstr_t m_szPreGUIDStr(m_wszGUIDStr);
this->m_DebugData = {
(char*)m_PDBInfo->PdbFileName, std::string(m_szPreGUIDStr),
m_PDBInfo->Age, m_PDBInfo->CvSignature
};
}
}
}
const bool bExportsEnabled = m_ParseType & PEHEADER_PARSING_TYPE::TYPE_EXPORT_DIRECTORY || m_ParseType & PEHEADER_PARSING_TYPE::TYPE_ALL;
if (!bExportsEnabled ||
!m_ExportDataDirectory.VirtualAddress ||
!m_ExportDataDirectory.Size)
{
if (bExportsEnabled)
CH_LOG("Export table didn't exist for current region. 0x%x | 0X%x", m_ExportDataDirectory.VirtualAddress, m_ExportDataDirectory.Size);
}
else // Export table parsing.
{
const PIMAGE_EXPORT_DIRECTORY m_pExportDirectory = CH_R_CAST<PIMAGE_EXPORT_DIRECTORY>(m_ImageBuffer + this->RvaToOffset(m_ExportDataDirectory.VirtualAddress));
const std::uint16_t* m_pOrdinalAddress = CH_R_CAST<std::uint16_t*>(m_ImageBuffer + this->RvaToOffset(m_pExportDirectory->AddressOfNameOrdinals));
const std::uint32_t* m_pNamesAddress = CH_R_CAST<std::uint32_t*>(m_ImageBuffer + this->RvaToOffset(m_pExportDirectory->AddressOfNames));
const std::uint32_t* m_pFunctionAddress = CH_R_CAST<std::uint32_t*>(m_ImageBuffer + this->RvaToOffset(m_pExportDirectory->AddressOfFunctions));
// Traverse export table and cache desired data.
for (std::size_t i = 0u; i < m_pExportDirectory->NumberOfNames; ++i)
{
const std::uint16_t m_CurrentOrdinal = m_pOrdinalAddress[i];
const std::uint32_t m_CurrentName = m_pNamesAddress[i];
if (m_CurrentName == 0u || m_CurrentOrdinal > m_pExportDirectory->NumberOfNames)
// Happend a few times, dunno.
continue;
char* m_szExportName = CH_R_CAST<char*>(m_ImageBuffer + this->RvaToOffset(m_CurrentName));
this->m_ExportData[m_szExportName] = { m_pFunctionAddress[m_CurrentOrdinal], m_CurrentOrdinal };
}
}
const bool bImportsEnabled = m_ParseType & PEHEADER_PARSING_TYPE::TYPE_IMPORT_DIRECTORY || m_ParseType & PEHEADER_PARSING_TYPE::TYPE_ALL;
if (!bImportsEnabled ||
!m_ImportDataDirectory.VirtualAddress ||
!m_ImportDataDirectory.Size)
{
if (bImportsEnabled)
CH_LOG("Import table didn't exist for current region. 0x%X | 0x%X",
m_ImportDataDirectory.VirtualAddress, m_ImportDataDirectory.Size);
}
else // Import table parsing.
{
PIMAGE_IMPORT_DESCRIPTOR m_pImportDescriptor = CH_R_CAST<PIMAGE_IMPORT_DESCRIPTOR>(m_ImageBuffer + this->RvaToOffset(m_ImportDataDirectory.VirtualAddress));
for (; m_pImportDescriptor->Name; ++m_pImportDescriptor)
{
// Read module name.
char* m_szModuleName = CH_R_CAST<char*>(m_ImageBuffer + this->RvaToOffset(m_pImportDescriptor->Name));
PIMAGE_THUNK_DATA m_pThunkData = CH_R_CAST<PIMAGE_THUNK_DATA>(m_ImageBuffer + this->RvaToOffset(m_pImportDescriptor->OriginalFirstThunk));
for (; m_pThunkData->u1.AddressOfData; ++m_pThunkData)
{
if (m_pThunkData->u1.Ordinal & IMAGE_ORDINAL_FLAG32)
// TODO: Imports by ordinal, dunno how I will make this nice.
continue;
// Read function name.
char* m_szFunctionName = CH_R_CAST<char*>(m_ImageBuffer + this->RvaToOffset(std::uint32_t(m_pThunkData->u1.AddressOfData + 2)));
// Cache desired data.
this->m_ImportData[m_szFunctionName] = { m_szModuleName };
}
}
}
}
// Parsing data out of this image's process.
PEHeaderData_t::PEHeaderData_t(Process_t& m_Process, std::int32_t m_ParseType, std::uintptr_t m_CustomBaseAddress)
{
if (m_ParseType & PEHEADER_PARSING_TYPE::TYPE_NONE)
return;
const std::uintptr_t m_BaseAddress = m_CustomBaseAddress != 0u ? m_CustomBaseAddress : m_Process.GetBaseAddress();
CH_ASSERT(true, m_BaseAddress, "Couldn't find base address of target process.");
IMAGE_DOS_HEADER m_pDOSHeadersTemporary = m_Process.Read<IMAGE_DOS_HEADER>(m_BaseAddress);
this->m_pDOSHeaders = &m_pDOSHeadersTemporary;
IMAGE_NT_HEADERS m_pNTHeadersTemporary = m_Process.Read<IMAGE_NT_HEADERS>(m_BaseAddress + this->m_pDOSHeaders->e_lfanew);
this->m_pNTHeaders = &m_pNTHeadersTemporary;
this->m_bIsValidInternal =
this->m_pDOSHeaders->e_magic == IMAGE_DOS_SIGNATURE &&
this->m_pNTHeaders->Signature == IMAGE_NT_SIGNATURE;
// Ensure image PE headers was valid.
CH_ASSERT(true, this->IsValid(), "Couldn't find MZ&NT header.");
for (std::size_t i = 0u; i < m_pNTHeaders->FileHeader.NumberOfSections; ++i)
{
std::size_t m_nSectionOffset = sizeof(IMAGE_NT_HEADERS) + (i * sizeof(IMAGE_SECTION_HEADER));
IMAGE_SECTION_HEADER m_pSectionHeaders = m_Process.Read<IMAGE_SECTION_HEADER>(m_BaseAddress + this->m_pDOSHeaders->e_lfanew + m_nSectionOffset);
this->m_SectionData.push_back(
{ CH_R_CAST<char*>(m_pSectionHeaders.Name),
m_pSectionHeaders.VirtualAddress,
m_pSectionHeaders.Misc.VirtualSize,
m_pSectionHeaders.Characteristics,
m_pSectionHeaders.PointerToRawData,
m_pSectionHeaders.SizeOfRawData }
);
}
for (std::size_t i = 0u; i < IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR/*Maximum*/; ++i)
this->m_DirectoryData.push_back(this->m_pNTHeaders->OptionalHeader.DataDirectory[i]);
// Gather all needed directories, then ensure we're using the correct type for the image.
IMAGE_DATA_DIRECTORY m_ExportDataDirectory = this->GetDataDirectory(IMAGE_DIRECTORY_ENTRY_EXPORT);
IMAGE_DATA_DIRECTORY m_ImportDataDirectory = this->GetDataDirectory(IMAGE_DIRECTORY_ENTRY_IMPORT);
IMAGE_DATA_DIRECTORY m_DebugDataDirectory = this->GetDataDirectory(IMAGE_DIRECTORY_ENTRY_DEBUG);
if (this->m_pNTHeaders->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR32_MAGIC)
{
#if defined (_WIN64) // Didn't already have NT_HEADERS32 data.
const PIMAGE_NT_HEADERS32 m_NT32Temporary = CH_R_CAST<PIMAGE_NT_HEADERS32>(this->m_pNTHeaders);
m_ExportDataDirectory = m_NT32Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT];
m_ImportDataDirectory = m_NT32Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT];
m_DebugDataDirectory = m_NT32Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG];
this->m_nMismatchedArchitecture = CH_S_CAST<std::uint32_t>(Process_t::eProcessArchitecture::ARCHITECTURE_x64); // For future notifications of dangerous actions.
#endif
}
else if (this->m_pNTHeaders->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)
{
#if defined (_WIN32) // Didn't already have NT_HEADERS64 data.
const PIMAGE_NT_HEADERS64 m_NT64Temporary = CH_R_CAST<PIMAGE_NT_HEADERS64>(this->m_pNTHeaders);
m_ExportDataDirectory = m_NT64Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT];
m_ImportDataDirectory = m_NT64Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT];
m_DebugDataDirectory = m_NT64Temporary->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG];
this->m_nMismatchedArchitecture = CH_S_CAST<std::uint32_t>(Process_t::eProcessArchitecture::ARCHITECTURE_x64); // For future notifications of dangerous actions.
#endif