fixed DarthTon#379: don't use process pseudo-handle

edisionLee · Jun 5, 2020 · ed951df · ed951df
1 parent 5eb0732
commit ed951df
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/BlackBone/Process/ProcessCore.cpp b/src/BlackBone/Process/ProcessCore.cpp
@@ -56,6 +56,10 @@ NTSTATUS ProcessCore::Open( HANDLE handle )
     _hProcess = handle;
     _pid = GetProcessId( _hProcess );
 
+    // Some routines in win10 do not support pseudo handle
+    if (IsWindows10OrGreater() && _pid == GetCurrentProcessId())
+        _hProcess = OpenProcess( PROCESS_ALL_ACCESS, FALSE, _pid );
+
     return Init();
 }
 

diff --git a/src/Samples/Main.cpp b/src/Samples/Main.cpp
@@ -11,6 +11,10 @@ void MapCmdFromMem();
 
 int main( int /*argc*/, char* /*argv[]*/ )
 {
+    blackbone::Process _process;
+    _ASSERT( NT_SUCCESS( _process.Attach( GetCurrentProcess() ) ) );
+    auto m = _process.modules().GetMainModule();
+
     // List all process PIDs matching name
     auto pids = Process::EnumByName( L"explorer.exe" );