admin/bb107-MemoryModulePP
1
0
Fork 0
mirror of https://github.com/bb107/MemoryModulePP synced 2026-06-08 13:15:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed some code formatting

Browse Source
This commit is contained in:
Boring
2023-03-31 08:03:22 +08:00
parent 9e9fd2ab94
commit 59e06ca5af
2 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
MemoryModule/MemoryModule.cpp
+1 -1
View File
@@ -248,7 +248,7 @@ NTSTATUS MemoryLoadLibrary(
nullptr,
old_header->OptionalHeader.SizeOfImage,
MEM_RESERVE,
PAGE_READWRITE
PAGE_EXECUTE_READWRITE
);
if (!base) status = STATUS_NO_MEMORY;
}
MemoryModule/MmpTls.cpp
+5 -7
View File
@@ -540,13 +540,14 @@ NTSTATUS NTAPI HookNtSetInformationProcess(
TlsLength
);
if (!NT_SUCCESS(status))break;
//
// Modify our mapping
//
EnterCriticalSection(&MmpGlobalDataPtr->MmpTls->MmpTlspLock);
for (ULONG i = 0; i < Tls->ThreadDataCount; ++i) {
bool found = false;
BOOL found = FALSE;
PLIST_ENTRY entry = MmpGlobalDataPtr->MmpTls->MmpThreadLocalStoragePointer.Flink;
// Find thread-spec tlsp
@@ -556,10 +557,7 @@ NTSTATUS NTAPI HookNtSetInformationProcess(
if (ProcessTlsInformation->OperationType == ProcessTlsReplaceVector) {
if (j->TlspMmpBlock[ProcessTlsInformation->TlsVectorLength] == ProcessTlsInformation->ThreadData[i].TlsVector[ProcessTlsInformation->TlsVectorLength]) {
found = true;
//auto tlsp = CONTAINING_RECORD(ProcessTlsInformation->ThreadData[i].TlsVector, TLS_VECTOR, TLS_VECTOR::ModuleTlsData);
//assert(tlsp->Length >= ProcessTlsInformation->TlsVectorLength);
found = TRUE;
// Copy old data to new pointer
RtlCopyMemory(
@@ -577,7 +575,7 @@ NTSTATUS NTAPI HookNtSetInformationProcess(
}
else {
if (j->TlspMmpBlock[ProcessTlsInformation->TlsIndex] == ProcessTlsInformation->ThreadData[i].TlsModulePointer) {
found = true;
found = TRUE;
if (ProcessHandle) {
j->TlspLdrBlock[ProcessTlsInformation->TlsIndex] = ProcessTlsInformation->ThreadData[i].TlsModulePointer;