// Copyright 2014 The Crashpad Authors
//
// 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/mac/cpu_context_mac.h"
#include <mach/mach.h>
#include "gtest/gtest.h"
namespace crashpad {
namespace test {
namespace {
#if defined(ARCH_CPU_X86_FAMILY)
TEST(CPUContextMac, InitializeContextX86) {
x86_thread_state32_t x86_thread_state32 = {};
x86_float_state32_t x86_float_state32 = {};
x86_debug_state32_t x86_debug_state32 = {};
x86_thread_state32.__eax = 1;
x86_float_state32.__fpu_ftw = 2;
x86_debug_state32.__dr0 = 3;
// Test the simple case, where everything in the CPUContextX86 argument is set
// directly from the supplied thread, float, and debug state parameters.
{
CPUContextX86 cpu_context_x86 = {};
internal::InitializeCPUContextX86(&cpu_context_x86,
THREAD_STATE_NONE,
nullptr,
0,
&x86_thread_state32,
&x86_float_state32,
&x86_debug_state32);
EXPECT_EQ(cpu_context_x86.eax, 1u);
EXPECT_EQ(cpu_context_x86.fxsave.ftw, 2u);
EXPECT_EQ(cpu_context_x86.dr0, 3u);
}
// Supply context in a CPU-specific “flavor” parameter expected to be used
// instead of the supplied thread, float, or debug state parameters. Do this
// once for each of the three valid flavors. This simulates how
// InitializeCPUContextX86() might be used to initialize the context in an
// exception handler, where the exception handler may have received the
// “flavor” parameter and this context should be used to initialize the
// CPUContextX86.
{
x86_thread_state32_t alt_x86_thread_state32 = {};
alt_x86_thread_state32.__eax = 4;
CPUContextX86 cpu_context_x86 = {};
internal::InitializeCPUContextX86(
&cpu_context_x86,
x86_THREAD_STATE32,
reinterpret_cast<natural_t*>(&alt_x86_thread_state32),
x86_THREAD_STATE32_COUNT,
&x86_thread_state32,
&x86_float_state32,
&x86_debug_state32);
EXPECT_EQ(cpu_context_x86.eax, 4u);
EXPECT_EQ(cpu_context_x86.fxsave.ftw, 2u);
EXPECT_EQ(cpu_context_x86.dr0, 3u);
}
{
x86_float_state32_t alt_x86_float_state32 = {};
alt_x86_float_state32.__fpu_ftw = 5;
CPUContextX86 cpu_context_x86 = {};
internal::InitializeCPUContextX86(
&cpu_context_x86,
x86_FLOAT_STATE32,
reinterpret_cast<natural_t*>(&alt_x86_float_state32),
x86_FLOAT_STATE32_COUNT,
&x86_thread_state32,
&x86_float_state32,
&x86_debug_state32);
EXPECT_EQ(cpu_context_x86.eax, 1u);
EXPECT_EQ(cpu_context_x86.fxsave.ftw, 5u);
EXPECT_EQ(cpu_context_x86.dr0, 3u);
}
{
x86_debug_state32_t alt_x86_debug_state32 = {};
alt_x86_debug_state32.__dr0 = 6;
CPUContextX86 cpu_context_x86 = {};
internal::InitializeCPUContextX86(
&cpu_context_x86,
x86_DEBUG_STATE32,
reinterpret_cast<natural_t*>(&alt_x86_debug_state32),
x86_DEBUG_STATE32_COUNT,
&x86_thread_state32,
&x86_float_state32,
&x86_debug_state32);
EXPECT_EQ(cpu_context_x86.eax, 1u);
EXPECT_EQ(cpu_context_x86.fxsave.ftw, 2u);
EXPECT_EQ(cpu_context_x86.dr0, 6u);
}
// Supply context in a universal “flavor” parameter expected to be used
// instead of the supplied thread, float, or debug state parameters. The
// universal format allows an exception handler to be registered to receive
// thread, float, or debug state without having to know in advance whether it
// will be receiving the state from a 32-bit or 64-bit process. For
// CPUContextX86, only the 32-bit form is supported.
{
x86_thread_state x86_thread_state_3264 = {};
x86_thread_state_3264.tsh.flavor = x86_THREAD_STATE32;
x86_thread_state_3264.tsh.count = x86_THREAD_STATE32_COUNT;
x86_thread_state_3264.uts.ts32.__eax = 7;
CPUContextX86 cpu_context_x86 = {};
internal::InitializeCPUContextX86(
&cpu_context_x86,
x86_THREAD_STATE,
reinterpret_cast<natural_t*>(&x86_thread_state_3264),
x86_THREAD_STATE_COUNT,
&x86_thread_state32,
&x86_float_state32,
&x86_debug_state32);
EXPECT_EQ(cpu_context_x86.eax, 7u);
EXPECT_EQ(cpu_context_x86.fxsave.ftw, 2u);
EXPECT_EQ(cpu_context_x86.dr0, 3u);
}
{
x86_float_state x86_float_state_3264 = {};
x86_float_state_3264.fsh.flavor = x86_FLOAT_STATE32;
x86_float_state_3264.fsh.count = x86_FLOAT_STATE32_COUNT;
x86_float_state_3264.ufs.fs32.__fpu_ftw = 8;
CPUContextX86 cpu_context_x86 = {};
internal::InitializeCPUContextX86(
&cpu_context_x86,
x86_FLOAT_STATE,
reinterpret_cast<natural_t*>(&x86_float_state_3264),
x86_FLOAT_STATE_COUNT,
&x86_thread_state32,
&x86_float_state32,
&x86_debug_state32);
EXPECT_EQ(cpu_context_x86.eax, 1u);
EXPECT_EQ(cpu_context_x86.fxsave.ftw, 8u);
EXPECT_EQ(cpu_context_x86.dr0, 3u);
}
{
x86_debug_state x86_debug_state_3264 = {};
x86_debug_state_3264.dsh.flavor = x86_DEBUG_STATE32;
x86_debug_state_3264.dsh.count = x86_DEBUG_STATE32_COUNT;
x86_debug_state_3264.uds.ds32.__dr0 = 9;
CPUContextX86 cpu_context_x86 = {};
internal::InitializeCPUContextX86(
&cpu_context_x86,
x86_DEBUG_STATE,
reinterpret_cast<natural_t*>(&x86_debug_state_3264),
x86_DEBUG_STATE_COUNT,
&x86_thread_state32,
&x86_float_state32,
&x86_debug_state32);
EXPECT_EQ(cpu_context_x86.eax, 1u);
EXPECT_EQ(cpu_context_x86.fxsave.ftw, 2u);
EXPECT_EQ(cpu_context_x86.dr0, 9u);
}
// Supply inappropriate “flavor” contexts to test that
// InitializeCPUContextX86() detects the problem and refuses to use the
// supplied “flavor” context, falling back to the thread, float, and debug
// states.
{
x86_thread_state64_t x86_thread_state64 = {};
CPUContextX86 cpu_context_x86 = {};
internal::InitializeCPUContextX86(
&cpu_context_x86,
x86_THREAD_STATE64,
reinterpret_cast<natural_t*>(&x86_thread_state64),
x86_THREAD_STATE64_COUNT,
&x86_thread_state32,
&x86_float_state32,
&x86_debug_state32);
EXPECT_EQ(cpu_context_x86.eax, 1u);
EXPECT_EQ(cpu_context_x86.fxsave.ftw, 2u);
EXPECT_EQ(cpu_context_x86.dr0, 3u);
}
{
x86_thread_state x86_thread_state_3264 = {};
x86_thread_state_3264.tsh.flavor = x86_THREAD_STATE64;
x86_thread_state_3264.tsh.count = x86_THREAD_STATE64_COUNT;
CPUContextX86 cpu_context_x86 = {};
internal::InitializeCPUContextX86(
&cpu_context_x86,
x86_THREAD_STATE,
reinterpret_cast<natural_t*>(&x86_thread_state_3264),
x86_THREAD_STATE_COUNT,
&x86_thread_state32,
&x86_float_state32,
&x86_debug_state32);
EXPECT_EQ(cpu_context_x86.eax, 1u);
EXPECT_EQ(cpu_context_x86.fxsave.ftw, 2u);
EXPECT_EQ(cpu_context_x86.dr0, 3u);
}
}
TEST(CPUContextMac, InitializeContextX86_64) {
x86_thread_state64_t x86_thread_state64 = {};
x86_float_state64_t x86_float_state64 = {};
x86_debug_state64_t x86_debug_state64 = {};
x86_thread_state64.__rax = 10;
x86_float_state64.__fpu_ftw = 11;
x86_debug_state64.__dr0 = 12;
// Test the simple case, where everything in the CPUContextX86_64 argument is
// set directly from the supplied thread, float, and debug state parameters.
{
CPUContextX86_64 cpu_context_x86_64 = {};
internal::InitializeCPUContextX86_64(&cpu_context_x86_64,
THREAD_STATE_NONE,
nullptr,
0,
&x86_thread_state64,
&x86_float_state64,
&x86_debug_state64);
EXPECT_EQ(cpu_context_x86_64.rax, 10u);
EXPECT_EQ(cpu_context_x86_64.fxsave.ftw, 11u);
EXPECT_EQ(cpu_context_x86_64.dr0, 12u);
}
// Supply context in a CPU-specific “flavor” parameter expected to be used
// instead of the supplied thread, float, or debug state parameters. Do this
// once for each of the three valid flavors. This simulates how
// InitializeCPUContextX86_64() might be used to initialize the context in an
// exception handler, where the exception handler may have received the
// “flavor” parameter and this context should be used to initialize the
// CPUContextX86_64.
{
x86_thread_state64_t alt_x86_thread_state64 = {};
alt_x86_thread_state64.__rax = 13;
CPUContextX86_64 cpu_context_x86_64 = {};
internal::InitializeCPUContextX86_64(
&cpu_context_x86_64,
x86_THREAD_STATE64,
reinterpret_cast<natural_t*>(&alt_x86_thread_state64),
x86_THREAD_STATE64_COUNT,
&x86_thread_state64,
&x86_float_state64,
&x86_debug_state64);
EXPECT_EQ(cpu_context_x86_64.rax, 13u);
EXPECT_EQ(cpu_context_x86_64.fxsave.ftw, 11u);
EXPECT_EQ(cpu_context_x86_64.dr0, 12u);
}
{
x86_float_state64_t alt_x86_float_state64 = {};
alt_x86_float_state64.__fpu_ftw = 14;
CPUContextX86_64 cpu_context_x86_64 = {};
internal::InitializeCPUContextX86_64(
&cpu_context_x86_64,
x86_FLOAT_STATE64,
reinterpret_cast<natural_t*>(&alt_x86_float_state64),
x86_FLOAT_STATE64_COUNT,
&x86_thread_state64,
&x86_float_state64,
&x86_debug_state64);
EXPECT_EQ(cpu_context_x86_64.rax, 10u);
EXPECT_EQ(cpu_context_x86_64.fxsave.ftw, 14u);
EXPECT_EQ(cpu_context_x86_64.dr0, 12u);
}
{
x86_debug_state64_t alt_x86_debug_state64 = {};
alt_x86_debug_state64.__dr0 = 15;
CPUContextX86_64 cpu_context_x86_64 = {};
internal::InitializeCPUContextX86_64(
&cpu_context_x86_64,
x86_DEBUG_STATE64,
reinterpret_cast<natural_t*>(&alt_x86_debug_state64),
x86_DEBUG_STATE64_COUNT,
&x86_thread_state64,
&x86_float_state64,
&x86_debug_state64);
EXPECT_EQ(cpu_context_x86_64.rax, 10u);
EXPECT_EQ(cpu_context_x86_64.fxsave.ftw, 11u);
EXPECT_EQ(cpu_context_x86_64.dr0, 15u);
}
// Supply context in a universal “flavor” parameter expected to be used
// instead of the supplied thread, float, or debug state parameters. The
// universal format allows an exception handler to be registered to receive
// thread, float, or debug state without having to know in advance whether it
// will be receiving the state from a 32-bit or 64-bit process. For
// CPUContextX86_64, only the 64-bit form is supported.
{
x86_thread_state x86_thread_state_3264 = {};
x86_thread_state_3264.tsh.flavor = x86_THREAD_STATE64;
x86_thread_state_3264.tsh.count = x86_THREAD_STATE64_COUNT;
x86_thread_state_3264.uts.ts64.__rax = 16;
CPUContextX86_64 cpu_context_x86_64 = {};
internal::InitializeCPUContextX86_64(
&cpu_context_x86_64,
x86_THREAD_STATE,
reinterpret_cast<natural_t*>(&x86_thread_state_3264),
x86_THREAD_STATE_COUNT,
&x86_thread_state64,
&x86_float_state64,
&x86_debug_state64);
EXPECT_EQ(cpu_context_x86_64.rax, 16u);
EXPECT_EQ(cpu_context_x86_64.fxsave.ftw, 11u);
EXPECT_EQ(cpu_context_x86_64.dr0, 12u);
}
{
x86_float_state x86_float_state_3264 = {};
x86_float_state_3264.fsh.flavor = x86_FLOAT_STATE64;
x86_float_state_3264.fsh.count = x86_FLOAT_STATE64_COUNT;
x86_float_state_3264.ufs.fs64.__fpu_ftw = 17;
CPUContextX86_64 cpu_context_x86_64 = {};
internal::InitializeCPUContextX86_64(
&cpu_context_x86_64,
x86_FLOAT_STATE,
reinterpret_cast<natural_t*>(&x86_float_state_3264),
x86_FLOAT_STATE_COUNT,
&x86_thread_state64,
&x86_float_state64,
&x86_debug_state64);
EXPECT_EQ(cpu_context_x86_64.rax, 10u);
EXPECT_EQ(cpu_context_x86_64.fxsave.ftw, 17u);
EXPECT_EQ(cpu_context_x86_64.dr0, 12u);
}
{
x86_debug_state x86_debug_state_3264 = {};
x86_debug_state_3264.dsh.flavor = x86_DEBUG_STATE64;
x86_debug_state_3264.dsh.count = x86_DEBUG_STATE64_COUNT;
x86_debug_state_3264.uds.ds64.__dr0 = 18;
CPUContextX86_64 cpu_context_x86_64 = {};
internal::InitializeCPUContextX86_64(
&cpu_context_x86_64,
x86_DEBUG_STATE,
reinterpret_cast<natural_t*>(&x86_debug_state_3264),
x86_DEBUG_STATE_COUNT,
&x86_thread_state64,
&x86_float_state64,
&x86_debug_state64);
EXPECT_EQ(cpu_context_x86_64.rax, 10u);
EXPECT_EQ(cpu_context_x86_64.fxsave.ftw, 11u);
EXPECT_EQ(cpu_context_x86_64.dr0, 18u);
}
// Supply inappropriate “flavor” contexts to test that
// InitializeCPUContextX86() detects the problem and refuses to use the
// supplied “flavor” context, falling back to the thread, float, and debug
// states.
{
x86_thread_state32_t x86_thread_state32 = {};
CPUContextX86_64 cpu_context_x86_64 = {};
internal::InitializeCPUContextX86_64(
&cpu_context_x86_64,
x86_THREAD_STATE32,
reinterpret_cast<natural_t*>(&x86_thread_state32),
x86_THREAD_STATE32_COUNT,
&x86_thread_state64,
&x86_float_state64,
&x86_debug_state64);
EXPECT_EQ(cpu_context_x86_64.rax, 10u);
EXPECT_EQ(cpu_context_x86_64.fxsave.ftw, 11u);
EXPECT_EQ(cpu_context_x86_64.dr0, 12u);
}
{
x86_thread_state x86_thread_state_3264 = {};
x86_thread_state_3264.tsh.flavor = x86_THREAD_STATE32;
x86_thread_state_3264.tsh.count = x86_THREAD_STATE32_COUNT;
CPUContextX86_64 cpu_context_x86_64 = {};
internal::InitializeCPUContextX86_64(
&cpu_context_x86_64,
x86_THREAD_STATE,
reinterpret_cast<natural_t*>(&x86_thread_state_3264),
x86_THREAD_STATE_COUNT,
&x86_thread_state64,
&x86_float_state64,
&x86_debug_state64);
EXPECT_EQ(cpu_context_x86_64.rax, 10u);
EXPECT_EQ(cpu_context_x86_64.fxsave.ftw, 11u);
EXPECT_EQ(cpu_context_x86_64.dr0, 12u);
}
}
#endif
} // namespace
} // namespace test
} // namespace crashpad
Codebase Browser
chromium