/* The maximum size of an environment value (in bytes) */
#define MAX_ENVVAL_SIZE 1024
+extern LIST_ENTRY HandleTableListHead;
+extern EX_PUSH_LOCK HandleTableListLock;
+
FAST_MUTEX ExpEnvironmentLock;
ERESOURCE ExpFirmwareTableResource;
LIST_ENTRY ExpFirmwareTableProviderListHead;
QSI_DEF(SystemHandleInformation)
{
PSYSTEM_HANDLE_INFORMATION HandleInformation;
- KPROCESSOR_MODE PreviousMode;
- PEPROCESS Process;
- PEPROCESS SystemProcess;
- ULONG CurrentSize;
- ULONG NumberOfHandles = 0;
- ULONG Index;
+ ULONG Index = 0;
NTSTATUS Status;
PMDL Mdl;
DPRINT("NtQuerySystemInformation - SystemHandleInformation\n");
+ /* Set initial required buffer size */
+ *ReqSize = sizeof(SYSTEM_HANDLE_INFORMATION);
+
/* Check user's buffer size */
- if (Size < sizeof(SYSTEM_HANDLE_INFORMATION))
+ if (Size < *ReqSize)
{
- *ReqSize = sizeof(SYSTEM_HANDLE_INFORMATION);
return STATUS_INFO_LENGTH_MISMATCH;
}
- /* Retrieve needed buffer size to hold the list of handles */
- SystemProcess = PsGetNextProcess(NULL);
- Process = SystemProcess;
-
- do
- {
- NumberOfHandles += ObGetProcessHandleCount(Process);
- Process = PsGetNextProcess(Process);
- }
- while ((Process != SystemProcess) && (Process != NULL));
-
- /* Dereference the process which was referenced by PsGetNextProcess */
- if (Process != NULL) ObDereferenceObject(Process);
-
- /* Calculate the current size of all handles */
- CurrentSize = sizeof(SYSTEM_HANDLE_INFORMATION) +
- ((sizeof(SYSTEM_HANDLE_TABLE_ENTRY_INFO) * NumberOfHandles) -
- (sizeof(SYSTEM_HANDLE_TABLE_ENTRY_INFO)));
-
- *ReqSize = CurrentSize;
-
- /* Check user's buffer size */
- if (CurrentSize > Size) return STATUS_INFO_LENGTH_MISMATCH;
-
/* We need to lock down the memory */
- PreviousMode = ExGetPreviousMode();
Status = ExLockUserBuffer(Buffer,
Size,
- PreviousMode,
+ ExGetPreviousMode(),
IoWriteAccess,
(PVOID*)&HandleInformation,
&Mdl);
return Status;
}
- /* Initialization of count of handles */
- HandleInformation->NumberOfHandles = NumberOfHandles;
+ _SEH2_TRY
+ {
+ PLIST_ENTRY NextTableEntry;
- /* Now get handles from all processes. */
- SystemProcess = PsGetNextProcess(NULL);
- Process = SystemProcess;
+ /* Enumerate all system handles */
+ for (NextTableEntry = HandleTableListHead.Flink;
+ NextTableEntry != &HandleTableListHead;
+ NextTableEntry = NextTableEntry->Flink)
+ {
+ PHANDLE_TABLE HandleTable;
- Index = 0;
+ /* Enter a critical region */
+ KeEnterCriticalRegion();
- do
- {
- PHANDLE_TABLE_ENTRY HandleTableEntry;
- EXHANDLE Handle;
+ /* Acquire the handle table lock */
+ ExAcquirePushLockExclusive(&HandleTableListLock);
- /* Enter a critical region */
- KeEnterCriticalRegion();
+ /* Get current handle table */
+ HandleTable = CONTAINING_RECORD(NextTableEntry, HANDLE_TABLE, HandleTableList);
- _SEH2_TRY
- {
- /* Set the initial value and loop the entries */
- Handle.Value = 0;
- while ((HandleTableEntry = ExpLookupHandleTableEntry(Process->ObjectTable, Handle)))
+ _SEH2_TRY
{
- /* Validate the entry */
- if ((HandleTableEntry->Object) &&
- (HandleTableEntry->NextFreeTableEntry != -2))
+ PHANDLE_TABLE_ENTRY HandleTableEntry;
+ EXHANDLE Handle;
+
+ /* Set the initial value and loop the entries */
+ Handle.Value = 0;
+ while ((HandleTableEntry = ExpLookupHandleTableEntry(HandleTable, Handle)))
{
- /* Lock the entry */
- if (ExpLockHandleTableEntry(Process->ObjectTable, HandleTableEntry))
+ /* Validate the entry */
+ if ((HandleTableEntry->Object) &&
+ (HandleTableEntry->NextFreeTableEntry != -2))
{
- _SEH2_TRY
+ /* Increase of count of handles */
+ ++HandleInformation->NumberOfHandles;
+
+ /* Increase required buffer size */
+ *ReqSize += sizeof(SYSTEM_HANDLE_TABLE_ENTRY_INFO);
+
+ /* Check user's buffer size */
+ if (*ReqSize > Size)
{
- POBJECT_HEADER ObjectHeader = ObpGetHandleObject(HandleTableEntry);
+ /* It is correct? How it is correct to leave the enclosed SEH blocks? */
+ _SEH2_YIELD(return STATUS_INFO_LENGTH_MISMATCH);
+ }
- /* Filling handle information */
- HandleInformation->Handles[Index].UniqueProcessId =
- (USHORT)(ULONG_PTR) Process->UniqueProcessId;
+ /* Lock the entry */
+ if (ExpLockHandleTableEntry(HandleTable, HandleTableEntry))
+ {
+ _SEH2_TRY
+ {
+ POBJECT_HEADER ObjectHeader = ObpGetHandleObject(HandleTableEntry);
- HandleInformation->Handles[Index].CreatorBackTraceIndex = 0;
+ /* Filling handle information */
+ HandleInformation->Handles[Index].UniqueProcessId =
+ (USHORT)(ULONG_PTR) HandleTable->UniqueProcessId;
- HandleInformation->Handles[Index].ObjectTypeIndex =
- (UCHAR) ObjectHeader->Type->Index;
+ HandleInformation->Handles[Index].CreatorBackTraceIndex = 0;
- HandleInformation->Handles[Index].HandleAttributes =
- HandleTableEntry->ObAttributes & OBJ_HANDLE_ATTRIBUTES;
+ HandleInformation->Handles[Index].ObjectTypeIndex =
+ (UCHAR) ObjectHeader->Type->Index;
- HandleInformation->Handles[Index].HandleValue =
- (USHORT)(ULONG_PTR) Handle.GenericHandleOverlay;
+ HandleInformation->Handles[Index].HandleAttributes =
+ HandleTableEntry->ObAttributes & OBJ_HANDLE_ATTRIBUTES;
- HandleInformation->Handles[Index].Object = &ObjectHeader->Body;
+ HandleInformation->Handles[Index].HandleValue =
+ (USHORT)(ULONG_PTR) Handle.GenericHandleOverlay;
- HandleInformation->Handles[Index].GrantedAccess =
- HandleTableEntry->GrantedAccess;
+ HandleInformation->Handles[Index].Object = &ObjectHeader->Body;
- ++Index;
- }
- _SEH2_FINALLY
- {
- /* Unlock it */
- ExUnlockHandleTableEntry(Process->ObjectTable, HandleTableEntry);
+ HandleInformation->Handles[Index].GrantedAccess =
+ HandleTableEntry->GrantedAccess;
+
+ ++Index;
+ }
+ _SEH2_FINALLY
+ {
+ /* Unlock it */
+ ExUnlockHandleTableEntry(HandleTable, HandleTableEntry);
+ }
+ _SEH2_END;
}
- _SEH2_END;
}
+
+ /* Go to the next entry */
+ Handle.Value += sizeof(HANDLE);
}
+ }
+ _SEH2_FINALLY
+ {
+ /* Release the lock */
+ ExReleasePushLockExclusive(&HandleTableListLock);
- /* Go to the next entry */
- Handle.Value += sizeof(HANDLE);
+ /* Leave the critical region */
+ KeLeaveCriticalRegion();
}
+ _SEH2_END;
}
- _SEH2_FINALLY
- {
- /* Leave the critical region */
- KeLeaveCriticalRegion();
- }
- _SEH2_END;
-
- Process = PsGetNextProcess(Process);
}
- while ((Process != SystemProcess) && (Process != NULL));
-
- /* Dereference the process which was referenced by PsGetNextProcess */
- if (Process != NULL) ObDereferenceObject(Process);
-
- /* Release the locked user buffer */
- ExUnlockUserBuffer(Mdl);
+ _SEH2_FINALLY
+ {
+ /* Release the locked user buffer */
+ ExUnlockUserBuffer(Mdl);
+ }
+ _SEH2_END;
return STATUS_SUCCESS;
}