ref: 04db83211cd1607f51155c6a8e51381f45a441a5
dir: /test/register_state_check.h/
/* * Copyright (c) 2012 The WebM project authors. All Rights Reserved. * * Use of this source code is governed by a BSD-style license * that can be found in the LICENSE file in the root of the source * tree. An additional intellectual property rights grant can be found * in the file PATENTS. All contributing project authors may * be found in the AUTHORS file in the root of the source tree. */ #ifndef VPX_TEST_REGISTER_STATE_CHECK_H_ #define VPX_TEST_REGISTER_STATE_CHECK_H_ #include "third_party/googletest/src/include/gtest/gtest.h" #include "./vpx_config.h" #include "vpx/vpx_integer.h" // ASM_REGISTER_STATE_CHECK(asm_function) // Minimally validates the environment pre & post function execution. This // variant should be used with assembly functions which are not expected to // fully restore the system state. See platform implementations of // RegisterStateCheck for details. // // API_REGISTER_STATE_CHECK(api_function) // Performs all the checks done by ASM_REGISTER_STATE_CHECK() and any // additional checks to ensure the environment is in a consistent state pre & // post function execution. This variant should be used with API functions. // See platform implementations of RegisterStateCheckXXX for details. // #if defined(_WIN64) && VPX_ARCH_X86_64 #undef NOMINMAX #define NOMINMAX #ifndef WIN32_LEAN_AND_MEAN #define WIN32_LEAN_AND_MEAN #endif #include <windows.h> #include <winnt.h> inline bool operator==(const M128A &lhs, const M128A &rhs) { return (lhs.Low == rhs.Low && lhs.High == rhs.High); } namespace libvpx_test { // Compares the state of xmm[6-15] at construction with their state at // destruction. These registers should be preserved by the callee on // Windows x64. class RegisterStateCheck { public: RegisterStateCheck() { initialized_ = StoreRegisters(&pre_context_); } ~RegisterStateCheck() { Check(); } private: static bool StoreRegisters(CONTEXT *const context) { const HANDLE this_thread = GetCurrentThread(); EXPECT_TRUE(this_thread != NULL); context->ContextFlags = CONTEXT_FLOATING_POINT; const bool context_saved = GetThreadContext(this_thread, context) == TRUE; EXPECT_TRUE(context_saved) << "GetLastError: " << GetLastError(); return context_saved; } // Compares the register state. Returns true if the states match. void Check() const { ASSERT_TRUE(initialized_); CONTEXT post_context; ASSERT_TRUE(StoreRegisters(&post_context)); const M128A *xmm_pre = &pre_context_.Xmm6; const M128A *xmm_post = &post_context.Xmm6; for (int i = 6; i <= 15; ++i) { EXPECT_EQ(*xmm_pre, *xmm_post) << "xmm" << i << " has been modified!"; ++xmm_pre; ++xmm_post; } } bool initialized_; CONTEXT pre_context_; }; #define ASM_REGISTER_STATE_CHECK(statement) \ do { \ libvpx_test::RegisterStateCheck reg_check; \ statement; \ } while (false) } // namespace libvpx_test #elif defined(CONFIG_SHARED) && defined(HAVE_NEON_ASM) && \ defined(CONFIG_VP9) && !CONFIG_SHARED && HAVE_NEON_ASM && CONFIG_VP9 extern "C" { // Save the d8-d15 registers into store. void vpx_push_neon(int64_t *store); } namespace libvpx_test { // Compares the state of d8-d15 at construction with their state at // destruction. These registers should be preserved by the callee on // arm platform. class RegisterStateCheck { public: RegisterStateCheck() { vpx_push_neon(pre_store_); } ~RegisterStateCheck() { Check(); } private: // Compares the register state. Returns true if the states match. void Check() const { int64_t post_store[8]; vpx_push_neon(post_store); for (int i = 0; i < 8; ++i) { EXPECT_EQ(pre_store_[i], post_store[i]) << "d" << i + 8 << " has been modified"; } } int64_t pre_store_[8]; }; #define ASM_REGISTER_STATE_CHECK(statement) \ do { \ libvpx_test::RegisterStateCheck reg_check; \ statement; \ } while (false) } // namespace libvpx_test #else namespace libvpx_test { class RegisterStateCheck {}; #define ASM_REGISTER_STATE_CHECK(statement) statement } // namespace libvpx_test #endif // _WIN64 && VPX_ARCH_X86_64 #if VPX_ARCH_X86 || VPX_ARCH_X86_64 #if defined(__GNUC__) namespace libvpx_test { // Checks the FPU tag word pre/post execution to ensure emms has been called. class RegisterStateCheckMMX { public: RegisterStateCheckMMX() { __asm__ volatile("fstenv %0" : "=rm"(pre_fpu_env_)); } ~RegisterStateCheckMMX() { Check(); } private: // Checks the FPU tag word pre/post execution, returning false if not cleared // to 0xffff. void Check() const { EXPECT_EQ(0xffff, pre_fpu_env_[4]) << "FPU was in an inconsistent state prior to call"; uint16_t post_fpu_env[14]; __asm__ volatile("fstenv %0" : "=rm"(post_fpu_env)); EXPECT_EQ(0xffff, post_fpu_env[4]) << "FPU was left in an inconsistent state after call"; } uint16_t pre_fpu_env_[14]; }; #define API_REGISTER_STATE_CHECK(statement) \ do { \ libvpx_test::RegisterStateCheckMMX reg_check; \ ASM_REGISTER_STATE_CHECK(statement); \ } while (false) } // namespace libvpx_test #endif // __GNUC__ #endif // VPX_ARCH_X86 || VPX_ARCH_X86_64 #ifndef API_REGISTER_STATE_CHECK #define API_REGISTER_STATE_CHECK ASM_REGISTER_STATE_CHECK #endif #endif // VPX_TEST_REGISTER_STATE_CHECK_H_