C# - 프로세스의 모든 핸들을 열람 - 두 번째 이야기
지난 글에서,
C# - 프로세스의 모든 핸들을 열람
; https://www.sysnet.pe.kr/2/0/12080
시스템의 모든 프로세스에 대한 윈도우 핸들을 열람하는 것을 알아봤는데요.
// Install-Package KernelStructOffset
using (WindowsHandleInfo whi = new WindowsHandleInfo())
{
for (int i = 0; i < whi.HandleCount; i++)
{
SYSTEM_HANDLE_ENTRY she = whi[i];
if (she.UniqueProcessId != processId)
{
continue;
}
string objName = she.GetName(out string handleTypeName);
Console.WriteLine($"{handleTypeName}: {objName}");
}
}
재미있는 것은, 특정 프로세스가 권한도 충분한데 열람이 안 되는 문제가 있었습니다. 왜냐하면 SYSTEM_HANDLE_ENTRY의,
[StructLayout(LayoutKind.Sequential)]
private struct SYSTEM_HANDLE_ENTRY
{
public int UniqueProcessId;
public byte ObjectType;
public byte HandleFlags;
public short HandleValue;
public IntPtr ObjectPointer;
public int AccessMask;
}
UniqueProcessId가 실은 int 형이 아닌,
[StructLayout(LayoutKind.Sequential)]
private struct SYSTEM_HANDLE_ENTRY
{
public short UniqueProcessId;
public short CreatorBackTraceIndex;
public byte ObjectType;
public byte HandleFlags;
public short HandleValue;
public IntPtr ObjectPointer;
public int AccessMask;
}
2개의 필드를 하나로 합친 것으로써 원래는 short형이었기 때문입니다. 그래서 PID가 65535보다 크면 overflow가 발생해 정상적인 조회가 안 된 것입니다. 다행히 이에 대해서 검색해 보면 해법이 나오는데요,
Why can't get process id that more than 65535 by 'ntQuerySystemInformation' in Win7 64bit?
; https://stackoverflow.com/questions/23951043/why-cant-get-process-id-that-more-than-65535-by-ntquerysysteminformation-in-w
따라서 NtQuerySystemInformation에 SYSTEM_EXTENDED_HANDLE_INFORMATION == 64 값을 전달하고,
NativeMethods.NtQuerySystemInformation(SYSTEM_INFORMATION_CLASS.SystemExtendedHandleInformation, ptr, guessSize, out requiredSize);
반환받은 버퍼의 구조는 다음과 같은 구조로,
[StructLayout(LayoutKind.Sequential)]
public struct _SYSTEM_HANDLE_INFORMATION_EX
{
public IntPtr HandleCount;
public IntPtr Reserved;
public _SYSTEM_HANDLE_TABLE_ENTRY_INFO_EX Handles; /* Handles[0] */
public int NumberOfHandles
{
get { return HandleCount.ToInt32(); }
}
}
IntPtr.Size * 2만큼의 위치부터 새로운 _SYSTEM_HANDLE_TABLE_ENTRY_INFO_EX 구조체로 받아오면 됩니다.
[StructLayout(LayoutKind.Sequential)]
public struct _SYSTEM_HANDLE_TABLE_ENTRY_INFO_EX
{
public IntPtr ObjectPointer;
public IntPtr UniqueProcessId;
public IntPtr HandleValue;
public uint GrantedAccess;
public ushort CreatorBackTraceIndex;
public ushort ObjectTypeIndex;
public uint HandleAttributes;
public uint Reserved;
public int OwnerPid
{
get { return UniqueProcessId.ToInt32(); }
}
}
(
DotNetSamples/WinConsole/Debugger/KernelStructOffset/WindowsHandleInfo.cs의 코드는 위의 변경 사항이 적용된 것입니다.)
열거된 핸들이 다른 프로세스에 있는 경우, 해당 핸들의 정보를 조회하기 위해서는 현재 프로세스 핸들 목록에 복사해 놓아야 합니다. 핸들 복사는 다음과 같은 식으로 하면 되는데요.
{
IntPtr targetHandle = ...[다른 프로세스의 Handle]...;
int targetPid = ...[targetHandle 핸들을 소유하고 있는 ProcessId]...;
IntPtr targetProcessHandle = OpenProcess(ProcessAccessRights.PROCESS_DUP_HANDLE, false, targetPid);
IntPtr currentProcess = NativeMethods.GetCurrentProcess();
IntPtr duplicatedHandle;
bool dupResult = DuplicateHandle(
targetProcessHandle, targetHandle, currentProcess, out duplicatedHandle
, 0, false, DuplicateHandleOptions.DUPLICATE_SAME_ACCESS);
}
대개의 경우, 원본 핸들과 동일한 권한(DUPLICATE_SAME_ACCESS)으로 복사하면 문제가 없는데 간혹 Process 타입의 핸들인 경우 GetModuleFileNameEx로 호출했을 때,
StringBuilder sb = new StringBuilder(4096);
int length = GetModuleFileNameEx(duplicatedHandle, IntPtr.Zero, sb, sb.Capacity);
length == 0이 나오면서 StringBuilder 버퍼에는 3개의 문자 정도가 다음과 같은 쓰레기 값으로 채워져 있습니다.
[0] 0x0050 'P' char
[1] 0x0001 '\u0001' char
[2] 0xfffd '' char
이런 문제가 발생하는 Process 타입 핸들은 권한을 조회해 보면 SYNCHRONIZE(0x00100000)만 가지고 있는 것을 볼 수 있습니다. 따라서 위의 결과를 반환하는 핸들은 DuplicateHandle 시 다음과 같은 옵션으로 복제해야 합니다.
// PROCESS_VM_READ = 0x10,
// PROCESS_QUERY_INFORMATION = 0x0400,
int addAccessRights = (int)(ProcessAccessRights.PROCESS_VM_READ | ProcessAccessRights.PROCESS_QUERY_INFORMATION);
bool dupResult = DuplicateHandle(targetProcessHandle, targetHandle, currentProcess, out duplicatedHandle
, addAccessRights, false, 0);
다른 타입의 핸들도 이런 식으로 그 타입에 맞는 권한을 조정해야 하는데 이게 좀 애매합니다. 왜냐하면, 해당 핸들의 타입을 알기 위해서는 DuplicateHandle을 해야 하는데 그때 미리 access rights를 부여하는 것이 올바른 동작인지에 대해서는 보장할 수 없기 때문입니다. 예를 들어 프로세스인 경우 PROCESS_QUERY_INFORMATION = 0x0400이지만 스레드인 경우에는 THREAD_QUERY_INFORMATION = 0x0040이 되므로 어떤 핸들을 복사해야 하느냐에 따라 권한을 위한 상수도 달라집니다.
결국 한 번은 핸들의 타입을 알기 위해 복제를 해야 하고, 이후 다시 한번 타입에 따른 권한 설정을 위해 복제를 해야만 하는 식입니다.
[이 글에 대해서 여러분들과 의견을 공유하고 싶습니다. 틀리거나 미흡한 부분 또는 의문 사항이 있으시면 언제든 댓글 남겨주십시오.]