win: Retrieve thread context for x64

snapshot/win/thread_snapshot_win.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 "snapshot/win/thread_snapshot_win.h"
 
 #include "base/logging.h"
 #include "snapshot/win/process_reader_win.h"
 
 namespace crashpad {
 namespace internal {
 
+namespace {
+
+void InitializeX64Context(const CONTEXT& context,
+                          CPUContextX86_64* out) {
+  out->rax = context.Rax;
+  out->rbx = context.Rbx;
+  out->rcx = context.Rcx;
+  out->rdx = context.Rdx;
+  out->rdi = context.Rdi;
+  out->rsi = context.Rsi;
+  out->rbp = context.Rbp;
+  out->rsp = context.Rsp;
+  out->r8 = context.R8;
+  out->r9 = context.R9;
+  out->r10 = context.R10;
+  out->r11 = context.R11;
+  out->r12 = context.R12;
+  out->r13 = context.R13;
+  out->r14 = context.R14;
+  out->r15 = context.R15;
+  out->rip = context.Rip;
+  out->rflags = context.EFlags;
+  out->cs = context.SegCs;
+  out->fs = context.SegFs;
+  out->gs = context.SegGs;
+
+  out->dr0 = context.Dr0;
+  out->dr1 = context.Dr1;
+  out->dr2 = context.Dr2;
+  out->dr3 = context.Dr3;

[Robert Sesek, 2015/05/12 22:33:22]

Dr4, 5, and 6 are missing?

[scottmg, 2015/05/12 23:09:24]

They don't appear in the CONTEXT structure on Windows. Updated to set them to
6/7 per comment.

+  out->dr6 = context.Dr6;
+  out->dr7 = context.Dr7;
+
+  static_assert(sizeof(out->fxsave) == sizeof(context.FltSave),
+                "types must be equivalent");
+  memcpy(&out->fxsave, &context.FltSave.ControlWord, sizeof(out->fxsave));
+}
+
+void InitializeX86Context(const CONTEXT& context,
+                          CPUContextX86* out) {
+  CHECK(false) << "TODO(scottmg) InitializeX86Context()";
+}
+
+}  // namespace
+
 ThreadSnapshotWin::ThreadSnapshotWin()
     : ThreadSnapshot(), context_(), stack_(), thread_(), initialized_() {
 }
 
 ThreadSnapshotWin::~ThreadSnapshotWin() {
 }
 
 bool ThreadSnapshotWin::Initialize(
     ProcessReaderWin* process_reader,
     const ProcessReaderWin::Thread& process_reader_thread) {
   INITIALIZATION_STATE_SET_INITIALIZING(initialized_);
 
   thread_ = process_reader_thread;
   stack_.Initialize(
       process_reader, thread_.stack_region_address, thread_.stack_region_size);
 
+#if defined(ARCH_CPU_X86_FAMILY)
+  if (process_reader->Is64Bit()) {
+    context_.architecture = kCPUArchitectureX86_64;
+    context_.x86_64 = &context_union_.x86_64;
+    InitializeX64Context(process_reader_thread.context, context_.x86_64);
+  } else {
+    context_.architecture = kCPUArchitectureX86;
+    context_.x86 = &context_union_.x86;
+    InitializeX86Context(process_reader_thread.context, context_.x86);
+  }
+#endif
+
   INITIALIZATION_STATE_SET_VALID(initialized_);
   return true;
 }
 
 const CPUContext* ThreadSnapshotWin::Context() const {
   INITIALIZATION_STATE_DCHECK_VALID(initialized_);
-  LOG(ERROR) << "TODO(scottmg): CPUContext";
   return &context_;
 }
 
 const MemorySnapshot* ThreadSnapshotWin::Stack() const {
   INITIALIZATION_STATE_DCHECK_VALID(initialized_);
   return &stack_;
 }
 
 uint64_t ThreadSnapshotWin::ThreadID() const {
   INITIALIZATION_STATE_DCHECK_VALID(initialized_);
@@ skipping 10 matching lines @@
   return thread_.priority;
 }
 
 uint64_t ThreadSnapshotWin::ThreadSpecificDataAddress() const {
   INITIALIZATION_STATE_DCHECK_VALID(initialized_);
   return thread_.teb;
 }
 
 }  // namespace internal
 }  // namespace crashpad