#define ObjectNameInformation  1

#define SystemHandleInformation 0x10

typedef struct _SYSTEM_HANDLE_INFORMATION {
  ULONG ProcessId;
  UCHAR ObjectTypeNumber;
  UCHAR Flags;
  USHORT Handle;
  PVOID Object;
  ACCESS_MASK GrantedAccess;
} SYSTEM_HANDLE_INFORMATION, *PSYSTEM_HANDLE_INFORMATION;

typedef struct _SYSTEM_HANDLE_INFORMATION_EX {
  ULONG NumberOfHandles;
  SYSTEM_HANDLE_INFORMATION Information[1];
} SYSTEM_HANDLE_INFORMATION_EX, *PSYSTEM_HANDLE_INFORMATION_EX;

NTSTATUS ZwQuerySystemInformation( 
  IN ULONG SystemInformationClass, 
  IN PVOID SystemInformation, 
  IN ULONG SystemInformationLength, 
  OUT PULONG ReturnLength);
/*
NTSTATUS ZwDuplicateObject(
  IN HANDLE        SourceProcessHandle,
  IN PHANDLE        SourceHandle,
  IN HANDLE        TargetProcessHandle,
  OUT PHANDLE        TargetHandle,
  IN ACCESS_MASK      DesiredAccess OPTIONAL,
  IN BOOLEAN        InheritHandle,
  IN ULONG        Options );

NTSTATUS ZwQueryObject(
  IN HANDLE                ObjectHandle,
  IN ULONG                 ObjectInformationClass,
  OUT PVOID                ObjectInformation,
  IN ULONG                 ObjectInformationLength,
  OUT PULONG               ReturnLength OPTIONAL);

NTSTATUS PsLookupProcessByProcessId(
  IN ULONG               ulProcId, 
  OUT PEPROCESS *        pEProcess);

NTSTATUS KeAttachProcess(PEPROCESS pPeb);
NTSTATUS KeDetachProcess();
*/

PVOID GetInfoTable(ULONG ATableType)
{
  ULONG mSize = 0x4000;
  PVOID mPtr = NULL;
  NTSTATUS St;

  do
  {
    mPtr = ExAllocatePoolWithTag(PagedPool, mSize, 'ACEK');
    memset(mPtr, 0, mSize);
    if (mPtr)
    {
      St = ZwQuerySystemInformation(ATableType, mPtr, mSize, NULL);
    } else return NULL;
    if (St == STATUS_INFO_LENGTH_MISMATCH)
    {
      ExFreePoolWithTag(mPtr, 'ACEK');
      mSize = mSize * 2;
    }
  } while (St == STATUS_INFO_LENGTH_MISMATCH);
  if (St == STATUS_SUCCESS) return mPtr;
  ExFreePool(mPtr);
  return NULL;
}

typedef struct _OBJECT_TYPE_INDEX {
  WCHAR  Format[0x40];
}OBJECT_TYPE_INDEX, *POBJECT_TYPE_INDEX; 

OBJECT_TYPE_INDEX TypeIndex[0x20] = { 0 };

void ObjectTypeFormat( ULONG Type, WCHAR * FormatName )
{
  if (Type >= 0x20) {
    
    return;
  }
  if (TypeIndex[Type].Format[0] == 0) {
    
    RtlStringCbPrintfW(
      TypeIndex[Type].Format,
      0x40,
      L"OB_TYPE_%s",
      FormatName
      );
  }
}

void ObjectTypeDbgPrint()
{
  ULONG i = 0;
  for (i = 0; i < 30; i++) {
    
    if (TypeIndex[i].Format[0] == 0) {
      
      DbgPrint("#define OB_TYPE_UNKNOWN  %d\n", i);
    } else {
      _wcsupr(TypeIndex[i].Format);
      DbgPrint("#define %S  %d\n", TypeIndex[i].Format, i);
    }
  }
}

HANDLE TestTable()
{
  HANDLE Process, hObject;
  HANDLE CsrId = (HANDLE)0;
  OBJECT_ATTRIBUTES obj;
  CLIENT_ID cid;
  UCHAR Buff[0x100];
  POBJECT_NAME_INFORMATION ObjName = (PVOID)&Buff;
  PSYSTEM_HANDLE_INFORMATION_EX Handles;
  ULONG r;
  
  ULONG  i;
  
  Handles = GetInfoTable(SystemHandleInformation);

  if (!Handles) return CsrId;

  for (r = 0; r < Handles->NumberOfHandles; r++)
  {
    {
      InitializeObjectAttributes(&obj, NULL, OBJ_KERNEL_HANDLE, NULL, NULL);

      cid.UniqueProcess = (HANDLE)Handles->Information[r].ProcessId;
      cid.UniqueThread = 0;

      if (NT_SUCCESS(NtOpenProcess(&Process, PROCESS_DUP_HANDLE, &obj, &cid)))
      {
        if (NT_SUCCESS(ZwDuplicateObject(
          Process, 
          (HANDLE)Handles->Information[r].Handle, 
          NtCurrentProcess(), &hObject, 0, 0, DUPLICATE_SAME_ACCESS)))
        {
          PPUBLIC_OBJECT_TYPE_INFORMATION uoti = (PPUBLIC_OBJECT_TYPE_INFORMATION)NULL;
          uoti = (PPUBLIC_OBJECT_TYPE_INFORMATION)ExAllocatePoolWithTag( 
            NonPagedPool, sizeof(PUBLIC_OBJECT_TYPE_INFORMATION) + 0x100, 'ACEK' );
          if (!uoti) {
            
            return 0;
          }
          RtlZeroMemory( uoti, sizeof(PUBLIC_OBJECT_TYPE_INFORMATION) + 0x100 );
          if (NT_SUCCESS(ZwQueryObject(hObject, ObjectTypeInformation, uoti, sizeof(PUBLIC_OBJECT_TYPE_INFORMATION) + 0x100, NULL))) {
            
            //ANSI_STRING AnsiFormat = { 0 };
            //KdPrint(("%wZ = %d\n", &uoti->TypeName, Handles->Information[r].ObjectTypeNumber));
            //if (NT_SUCCESS(ZwQueryObject(hObject, ObjectNameInformation, ObjName, 0x100, NULL)))
            //{
              //KdPrint(("TestTable: Number = %d, %wZ, ObjectName = %wZ\n", 
              //  Handles->Information[r].ObjectTypeNumber, &uoti->TypeName, &ObjName->Name));
            //}  
            //RtlUnicodeStringToAnsiString( &AnsiFormat, &uoti->TypeName, TRUE);
            ObjectTypeFormat( Handles->Information[r].ObjectTypeNumber, uoti->TypeName.Buffer );
            //RtlFreeAnsiString( &AnsiFormat );
          }
          ExFreePoolWithTag( uoti, 'ACEK' );
          ZwClose(hObject);
        }
        ZwClose(Process);
      }
    }
  }

  ObjectTypeDbgPrint();

  ExFreePool(Handles);
  return CsrId;
}