diff options
Diffstat (limited to 'sandbox/src/handle_table.cc')
-rw-r--r-- | sandbox/src/handle_table.cc | 181 |
1 files changed, 0 insertions, 181 deletions
diff --git a/sandbox/src/handle_table.cc b/sandbox/src/handle_table.cc deleted file mode 100644 index c7fcf0a..0000000 --- a/sandbox/src/handle_table.cc +++ /dev/null @@ -1,181 +0,0 @@ -// Copyright (c) 2011 The Chromium Authors. All rights reserved. -// Use of this source code is governed by a BSD-style license that can be -// found in the LICENSE file. - -#include "sandbox/src/handle_table.h" - -#include <algorithm> -#include <cstdlib> - -#include "base/memory/scoped_ptr.h" -#include "sandbox/src/win_utils.h" - -namespace { - -bool CompareHandleEntries(const SYSTEM_HANDLE_INFORMATION& a, - const SYSTEM_HANDLE_INFORMATION& b) { - return a.ProcessId < b.ProcessId; -} - -} // namespace - -namespace sandbox { - -const char16* HandleTable::kTypeProcess = L"Process"; -const char16* HandleTable::kTypeThread = L"Thread"; -const char16* HandleTable::kTypeFile = L"File"; -const char16* HandleTable::kTypeDirectory = L"Directory"; -const char16* HandleTable::kTypeKey = L"Key"; -const char16* HandleTable::kTypeWindowStation = L"WindowStation"; -const char16* HandleTable::kTypeDesktop = L"Desktop"; -const char16* HandleTable::kTypeService = L"Service"; -const char16* HandleTable::kTypeMutex = L"Mutex"; -const char16* HandleTable::kTypeSemaphore = L"Semaphore"; -const char16* HandleTable::kTypeEvent = L"Event"; -const char16* HandleTable::kTypeTimer = L"Timer"; -const char16* HandleTable::kTypeNamedPipe = L"NamedPipe"; -const char16* HandleTable::kTypeJobObject = L"JobObject"; -const char16* HandleTable::kTypeFileMap = L"FileMap"; -const char16* HandleTable::kTypeAlpcPort = L"ALPC Port"; - -HandleTable::HandleTable() { - static NtQuerySystemInformation QuerySystemInformation = NULL; - if (!QuerySystemInformation) - ResolveNTFunctionPtr("NtQuerySystemInformation", &QuerySystemInformation); - - ULONG size = 0x15000; - NTSTATUS result; - do { - handle_info_buffer_.resize(size); - result = QuerySystemInformation(SystemHandleInformation, - handle_info_internal(), size, &size); - } while (result == STATUS_INFO_LENGTH_MISMATCH); - - // We failed, so make an empty table. - if (!NT_SUCCESS(result)) { - handle_info_buffer_.resize(0); - return; - } - - // Sort it to make process lookups faster. - std::sort(handle_info_internal()->Information, - handle_info_internal()->Information + - handle_info_internal()->NumberOfHandles, CompareHandleEntries); -} - -HandleTable::Iterator HandleTable::HandlesForProcess(ULONG process_id) const { - SYSTEM_HANDLE_INFORMATION key; - key.ProcessId = process_id; - - const SYSTEM_HANDLE_INFORMATION* start = handle_info()->Information; - const SYSTEM_HANDLE_INFORMATION* finish = - &handle_info()->Information[handle_info()->NumberOfHandles]; - - start = std::lower_bound(start, finish, key, CompareHandleEntries); - if (start->ProcessId != process_id) - return Iterator(*this, finish, finish); - finish = std::upper_bound(start, finish, key, CompareHandleEntries); - return Iterator(*this, start, finish); -} - -HandleTable::HandleEntry::HandleEntry( - const SYSTEM_HANDLE_INFORMATION* handle_info_entry) - : handle_entry_(handle_info_entry), last_entry_(0) { -} - -void HandleTable::HandleEntry::UpdateInfo(UpdateType flag) { - static NtQueryObject QueryObject = NULL; - if (!QueryObject) - ResolveNTFunctionPtr("NtQueryObject", &QueryObject); - - NTSTATUS result; - - // Always update the basic type info, but grab the names as needed. - if (needs_info_update()) { - handle_name_.clear(); - type_name_.clear(); - last_entry_ = handle_entry_; - - // Most handle names are very short, so start small and reuse this buffer. - if (type_info_buffer_.empty()) - type_info_buffer_.resize(sizeof(OBJECT_TYPE_INFORMATION) - + (32 * sizeof(wchar_t))); - ULONG size = static_cast<ULONG>(type_info_buffer_.size()); - result = QueryObject(reinterpret_cast<HANDLE>(handle_entry_->Handle), - ObjectTypeInformation, type_info_internal(), size, &size); - while (result == STATUS_INFO_LENGTH_MISMATCH) { - type_info_buffer_.resize(size); - result = QueryObject(reinterpret_cast<HANDLE>(handle_entry_->Handle), - ObjectTypeInformation, type_info_internal(), size, &size); - } - - if (!NT_SUCCESS(result)) { - type_info_buffer_.clear(); - return; - } - } - - // Don't bother copying out names until we ask for them, and then cache them. - switch (flag) { - case UPDATE_INFO_AND_NAME: - if (type_info_buffer_.size() && handle_name_.empty()) { - ULONG size = MAX_PATH; - scoped_ptr<UNICODE_STRING> name; - do { - name.reset(reinterpret_cast<UNICODE_STRING*>(new BYTE[size])); - result = QueryObject(reinterpret_cast<HANDLE>( - handle_entry_->Handle), ObjectNameInformation, name.get(), - size, &size); - } while (result == STATUS_INFO_LENGTH_MISMATCH); - - if (NT_SUCCESS(result)) { - handle_name_.assign(name->Buffer, name->Length / sizeof(wchar_t)); - } - } - break; - - case UPDATE_INFO_AND_TYPE_NAME: - if (!type_info_buffer_.empty() && type_info_internal()->Name.Buffer && - type_name_.empty()) { - type_name_.assign(type_info_internal()->Name.Buffer, - type_info_internal()->Name.Length / sizeof(wchar_t)); - } - break; - } -} - -const OBJECT_TYPE_INFORMATION* HandleTable::HandleEntry::TypeInfo() { - UpdateInfo(UPDATE_INFO_ONLY); - return type_info_buffer_.empty() ? NULL : type_info_internal(); -} - -const string16& HandleTable::HandleEntry::Name() { - UpdateInfo(UPDATE_INFO_AND_NAME); - return handle_name_; -} - -const string16& HandleTable::HandleEntry::Type() { - UpdateInfo(UPDATE_INFO_AND_TYPE_NAME); - return type_name_; -} - -bool HandleTable::HandleEntry::IsType(const string16& type_string) { - UpdateInfo(UPDATE_INFO_ONLY); - if (type_info_buffer_.empty()) - return false; - return type_string.compare(0, - type_info_internal()->Name.Length / sizeof(wchar_t), - type_info_internal()->Name.Buffer) == 0; -} - -HandleTable::Iterator::Iterator(const HandleTable& table, - const SYSTEM_HANDLE_INFORMATION* start, - const SYSTEM_HANDLE_INFORMATION* end) - : table_(table), current_(start), end_(end) { -} - -HandleTable::Iterator::Iterator(const Iterator& it) - : table_(it.table_), current_(it.current_.handle_entry_), end_(it.end_) { -} - -} // namespace sandbox |