win: Support reading process info cross-bitness

util/win/process_info.cc

 // Copyright 2015 The Crashpad Authors. All rights reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
 //
 //     http://www.apache.org/licenses/LICENSE-2.0
 //
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
 #include "util/win/process_info.h"
 
+#include <winternl.h>
+
 #include "base/logging.h"
+#include "util/numeric/safe_assignment.h"
+#include "util/win/process_structs.h"
 
 namespace crashpad {
 
 namespace {
 
 NTSTATUS NtQueryInformationProcess(HANDLE process_handle,
                                    PROCESSINFOCLASS process_information_class,
                                    PVOID process_information,
                                    ULONG process_information_length,
                                    PULONG return_length) {
@@ skipping 15 matching lines @@
   if (!is_wow64_process)
     return false;
   BOOL is_wow64;
   if (!is_wow64_process(process_handle, &is_wow64)) {
     PLOG(ERROR) << "IsWow64Process";
     return false;
   }
   return is_wow64;
 }
 
+template <class T>
 bool ReadUnicodeString(HANDLE process,
-                       const UNICODE_STRING& us,
+                       const process_types::UNICODE_STRING<T>& us,
                        std::wstring* result) {
   if (us.Length == 0) {
     result->clear();
     return true;
   }
   DCHECK_EQ(us.Length % sizeof(wchar_t), 0u);
   result->resize(us.Length / sizeof(wchar_t));
   SIZE_T bytes_read;
-  if (!ReadProcessMemory(
-          process, us.Buffer, &result->operator[](0), us.Length, &bytes_read)) {
+  if (!ReadProcessMemory(process,
+                         reinterpret_cast<const void*>(us.Buffer),
+                         &result->operator[](0),
+                         us.Length,
+                         &bytes_read)) {
     PLOG(ERROR) << "ReadProcessMemory UNICODE_STRING";
     return false;
   }
   if (bytes_read != us.Length) {
     LOG(ERROR) << "ReadProcessMemory UNICODE_STRING incorrect size";
     return false;
   }
   return true;
 }
 
 template <class T> bool ReadStruct(HANDLE process, uintptr_t at, T* into) {
   SIZE_T bytes_read;
   if (!ReadProcessMemory(process,
                          reinterpret_cast<const void*>(at),
                          into,
                          sizeof(T),
                          &bytes_read)) {
     // We don't have a name for the type we're reading, so include the signature
     // to get the type of T.
     PLOG(ERROR) << "ReadProcessMemory " << __FUNCSIG__;
     return false;
   }
   if (bytes_read != sizeof(T)) {
     LOG(ERROR) << "ReadProcessMemory " << __FUNCSIG__ << " incorrect size";
     return false;
   }
   return true;
 }
 
-// PEB_LDR_DATA in winternl.h doesn't document the trailing
-// InInitializationOrderModuleList field. See `dt ntdll!PEB_LDR_DATA`.
-struct FULL_PEB_LDR_DATA : public PEB_LDR_DATA {
-  LIST_ENTRY InInitializationOrderModuleList;
-};
-
-// LDR_DATA_TABLE_ENTRY doesn't include InInitializationOrderLinks, define a
-// complete version here. See `dt ntdll!_LDR_DATA_TABLE_ENTRY`.
-struct FULL_LDR_DATA_TABLE_ENTRY {
-  LIST_ENTRY InLoadOrderLinks;
-  LIST_ENTRY InMemoryOrderLinks;
-  LIST_ENTRY InInitializationOrderLinks;
-  PVOID DllBase;
-  PVOID EntryPoint;
-  ULONG SizeOfImage;
-  UNICODE_STRING FullDllName;
-  UNICODE_STRING BaseDllName;
-  ULONG Flags;
-  WORD LoadCount;
-  WORD TlsIndex;
-  LIST_ENTRY HashLinks;
-  ULONG TimeDateStamp;
-  _ACTIVATION_CONTEXT* EntryPointActivationContext;
-};
-
 }  // namespace
 
+template <class Traits>
+bool ReadProcessData(HANDLE process,
+                     uintptr_t peb_address_uintptr,
+                     ProcessInfo* process_info) {
+  Traits::Pointer peb_address;
+  if (!AssignIfInRange(&peb_address, peb_address_uintptr)) {
+    LOG(ERROR) << "peb_address_uintptr " << peb_address_uintptr
+               << " out of range";
+    return false;
+  }
+
+  // Try to read the process environment block.
+  process_types::PEB<Traits> peb;
+  if (!ReadStruct(process, peb_address, &peb))
+    return false;
+
+  process_types::RTL_USER_PROCESS_PARAMETERS<Traits> process_parameters;
+  if (!ReadStruct(process, peb.ProcessParameters, &process_parameters))
+    return false;
+
+  if (!ReadUnicodeString(process,
+                         process_parameters.CommandLine,
+                         &process_info->command_line_)) {
+    return false;
+  }
+
+  process_types::PEB_LDR_DATA<Traits> peb_ldr_data;
+  if (!ReadStruct(process, peb.Ldr, &peb_ldr_data))
+    return false;
+
+  std::wstring module;
+  process_types::LDR_DATA_TABLE_ENTRY<Traits> ldr_data_table_entry;
+
+  // Include the first module in the memory order list to get our the main
+  // executable's name, as it's not included in initialization order below.
+  if (!ReadStruct(process,
+                  reinterpret_cast<uintptr_t>(
+                      reinterpret_cast<const char*>(
+                          peb_ldr_data.InMemoryOrderModuleList.Flink) -
+                      offsetof(process_types::LDR_DATA_TABLE_ENTRY<Traits>,
+                               InMemoryOrderLinks)),
+                  &ldr_data_table_entry)) {
+    return false;
+  }
+  if (!ReadUnicodeString(process, ldr_data_table_entry.FullDllName, &module))
+    return false;
+  process_info->modules_.push_back(module);
+
+  // Walk the PEB LDR structure (doubly-linked list) to get the list of loaded
+  // modules. We use this method rather than EnumProcessModules to get the
+  // modules in initialization order rather than memory order.
+  Traits::Pointer last = peb_ldr_data.InInitializationOrderModuleList.Blink;
+  for (Traits::Pointer cur = peb_ldr_data.InInitializationOrderModuleList.Flink;
+       ;
+       cur = ldr_data_table_entry.InInitializationOrderLinks.Flink) {
+    // |cur| is the pointer to the LIST_ENTRY embedded in the
+    // LDR_DATA_TABLE_ENTRY, in the target process's address space. So we need
+    // to read from the target, and also offset back to the beginning of the
+    // structure.
+    if (!ReadStruct(process,
+                    reinterpret_cast<uintptr_t>(
+                        reinterpret_cast<const char*>(cur) -
+                        offsetof(process_types::LDR_DATA_TABLE_ENTRY<Traits>,
+                                 InInitializationOrderLinks)),
+                    &ldr_data_table_entry)) {
+      break;
+    }
+    // TODO(scottmg): Capture TimeDateStamp, Checksum, etc. too?
+    if (!ReadUnicodeString(process, ldr_data_table_entry.FullDllName, &module))
+      break;
+    process_info->modules_.push_back(module);
+    if (cur == last)
+      break;
+  }
+
+  return true;
+}
+
 ProcessInfo::ProcessInfo()
-    : process_basic_information_(),
+    : process_id_(),
+      inherited_from_process_id_(),
       command_line_(),
       modules_(),
       is_64_bit_(false),
       is_wow64_(false),
       initialized_() {
 }
 
 ProcessInfo::~ProcessInfo() {
 }
 
 bool ProcessInfo::Initialize(HANDLE process) {
   INITIALIZATION_STATE_SET_INITIALIZING(initialized_);
 
   is_wow64_ = IsProcessWow64(process);
 
   if (is_wow64_) {
     // If it's WoW64, then it's 32-on-64.
     is_64_bit_ = false;
   } else {
     // Otherwise, it's either 32 on 32, or 64 on 64. Use GetSystemInfo() to
     // distinguish between these two cases.
     SYSTEM_INFO system_info;
     GetSystemInfo(&system_info);
     is_64_bit_ =
         system_info.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_AMD64;
   }
 
-#if ARCH_CPU_64_BITS
-  if (!is_64_bit_) {
-    LOG(ERROR) << "Reading different bitness not yet supported";
-    return false;
-  }
-#else
+#if ARCH_CPU_32_BITS
   if (is_64_bit_) {
     LOG(ERROR) << "Reading x64 process from x86 process not supported";
     return false;
   }
 #endif
 
   ULONG bytes_returned;
+  // We assume this process is not running on Wow64. The
+  // PROCESS_BASIC_INFORMATION uses the OS size (that is, even Wow64 has a 64
+  // bit one.)
+  // TODO(scottmg): Either support running as Wow64, or check/resolve this at a
+  // higher level.
+#if ARCH_CPU_32_BITS
+  process_types::PROCESS_BASIC_INFORMATION<process_types::internal::Traits32>
+      process_basic_information;
+#else
+  process_types::PROCESS_BASIC_INFORMATION<process_types::internal::Traits64>
+      process_basic_information;
+#endif
   NTSTATUS status =
       crashpad::NtQueryInformationProcess(process,
                                           ProcessBasicInformation,
-                                          &process_basic_information_,
-                                          sizeof(process_basic_information_),
+                                          &process_basic_information,
+                                          sizeof(process_basic_information),
                                           &bytes_returned);
   if (status < 0) {
     LOG(ERROR) << "NtQueryInformationProcess: status=" << status;
     return false;
   }
-  if (bytes_returned != sizeof(process_basic_information_)) {
+  if (bytes_returned != sizeof(process_basic_information)) {
     LOG(ERROR) << "NtQueryInformationProcess incorrect size";
     return false;
   }
+  process_id_ = process_basic_information.UniqueProcessId;
+  inherited_from_process_id_ =
+      process_basic_information.InheritedFromUniqueProcessId;
 
-  // Try to read the process environment block.
-  PEB peb;
-  if (!ReadStruct(process,
-                  reinterpret_cast<uintptr_t>(
-                      process_basic_information_.PebBaseAddress),
-                  &peb)) {
-    return false;
+  // We now want to read the PEB to gather the rest of our information. The
+  // PebBaseAddress as returned above is what we want for 64-on-64 and 32-on-32,
+  // but for Wow64, we want to read the 32 bit PEB (a Wow64 process has both).
+  // The address of this is found by a second call to NtQueryInformationProcess.
+  uintptr_t peb_address = process_basic_information.PebBaseAddress;
+  if (is_wow64_) {
+    ULONG_PTR wow64_peb_address;
+    status =
+        crashpad::NtQueryInformationProcess(process,
+                                            ProcessWow64Information,
+                                            &wow64_peb_address,
+                                            sizeof(wow64_peb_address),
+                                            &bytes_returned);
+    if (status < 0) {
+      LOG(ERROR) << "NtQueryInformationProcess: status=" << status;
+      return false;
+    }
+    if (bytes_returned != sizeof(wow64_peb_address)) {
+      LOG(ERROR) << "NtQueryInformationProcess incorrect size";
+      return false;
+    }
+    peb_address = wow64_peb_address;
   }
 
-  RTL_USER_PROCESS_PARAMETERS process_parameters;
-  if (!ReadStruct(process,
-                  reinterpret_cast<uintptr_t>(peb.ProcessParameters),
-                  &process_parameters)) {
+  // Read the PEB data using the correct word size.
+  bool result = is_64_bit_ ? ReadProcessData<process_types::internal::Traits64>(
+                                 process, peb_address, this)
+                           : ReadProcessData<process_types::internal::Traits32>(
+                                 process, peb_address, this);
+  if (!result)
     return false;
-  }
-
-  if (!ReadUnicodeString(
-          process, process_parameters.CommandLine, &command_line_)) {
-    return false;
-  }
-
-  FULL_PEB_LDR_DATA peb_ldr_data;
-  if (!ReadStruct(process, reinterpret_cast<uintptr_t>(peb.Ldr), &peb_ldr_data))
-    return false;
-
-  // Include the first module in the memory order list to get our own name as
-  // it's not included in initialization order below.
-  std::wstring self_module;
-  FULL_LDR_DATA_TABLE_ENTRY self_ldr_data_table_entry;
-  if (!ReadStruct(process,
-                  reinterpret_cast<uintptr_t>(
-                      reinterpret_cast<const char*>(
-                          peb_ldr_data.InMemoryOrderModuleList.Flink) -
-                      offsetof(FULL_LDR_DATA_TABLE_ENTRY, InMemoryOrderLinks)),
-                  &self_ldr_data_table_entry)) {
-    return false;
-  }
-  if (!ReadUnicodeString(
-          process, self_ldr_data_table_entry.FullDllName, &self_module)) {
-    return false;
-  }
-  modules_.push_back(self_module);
-
-  // Walk the PEB LDR structure (doubly-linked list) to get the list of loaded
-  // modules. We use this method rather than EnumProcessModules to get the
-  // modules in initialization order rather than memory order.
-  const LIST_ENTRY* last = peb_ldr_data.InInitializationOrderModuleList.Blink;
-  FULL_LDR_DATA_TABLE_ENTRY ldr_data_table_entry;
-  for (const LIST_ENTRY* cur =
-           peb_ldr_data.InInitializationOrderModuleList.Flink;
-       ;
-       cur = ldr_data_table_entry.InInitializationOrderLinks.Flink) {
-    // |cur| is the pointer to the LIST_ENTRY embedded in the
-    // FULL_LDR_DATA_TABLE_ENTRY, in the target process's address space. So we
-    // need to read from the target, and also offset back to the beginning of
-    // the structure.
-    if (!ReadStruct(
-            process,
-            reinterpret_cast<uintptr_t>(reinterpret_cast<const char*>(cur) -
-                                        offsetof(FULL_LDR_DATA_TABLE_ENTRY,
-                                                 InInitializationOrderLinks)),
-            &ldr_data_table_entry)) {
-      break;
-    }
-    // TODO(scottmg): Capture TimeDateStamp, Checksum, etc. too?
-    std::wstring module;
-    if (!ReadUnicodeString(process, ldr_data_table_entry.FullDllName, &module))
-      break;
-    modules_.push_back(module);
-    if (cur == last)
-      break;
-  }
 
   INITIALIZATION_STATE_SET_VALID(initialized_);
   return true;
 }
 
 bool ProcessInfo::Is64Bit() const {
   INITIALIZATION_STATE_DCHECK_VALID(initialized_);
   return is_64_bit_;
 }
 
 bool ProcessInfo::IsWow64() const {
   INITIALIZATION_STATE_DCHECK_VALID(initialized_);
   return is_wow64_;
 }
 
 pid_t ProcessInfo::ProcessID() const {
   INITIALIZATION_STATE_DCHECK_VALID(initialized_);
-  return process_basic_information_.UniqueProcessId;
+  return process_id_;
 }
 
 pid_t ProcessInfo::ParentProcessID() const {
   INITIALIZATION_STATE_DCHECK_VALID(initialized_);
-  return process_basic_information_.InheritedFromUniqueProcessId;
+  return inherited_from_process_id_;
 }
 
 bool ProcessInfo::CommandLine(std::wstring* command_line) const {
   INITIALIZATION_STATE_DCHECK_VALID(initialized_);
   *command_line = command_line_;
   return true;
 }
 
 bool ProcessInfo::Modules(std::vector<std::wstring>* modules) const {
   INITIALIZATION_STATE_DCHECK_VALID(initialized_);
   *modules = modules_;
   return true;
 }
 
 }  // namespace crashpad