mirror of
https://github.com/chromium/crashpad.git
synced 2024-12-31 01:43:03 +08:00
d075a9eb2e
These wrap the GetProcAddress(LoadLibrary(), …) idiom into macros that are much less wordy. TEST=crashpad_util_test GetFunction.GetFunction and all others R=scottmg@chromium.org Review URL: https://codereview.chromium.org/1405323003 .
79 lines
3.1 KiB
C++
79 lines
3.1 KiB
C++
// 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/get_function.h"
|
||
|
||
#include <windows.h>
|
||
#include <winternl.h>
|
||
|
||
#include "gtest/gtest.h"
|
||
|
||
namespace crashpad {
|
||
namespace test {
|
||
namespace {
|
||
|
||
TEST(GetFunction, GetFunction) {
|
||
// Check equivalence of GET_FUNCTION_REQUIRED() with functions that are
|
||
// available in the SDK normally.
|
||
EXPECT_EQ(&GetProcAddress,
|
||
GET_FUNCTION_REQUIRED(L"kernel32.dll", GetProcAddress));
|
||
EXPECT_EQ(&LoadLibraryW,
|
||
GET_FUNCTION_REQUIRED(L"kernel32.dll", LoadLibraryW));
|
||
|
||
// Make sure that a function pointer retrieved by GET_FUNCTION_REQUIRED() can
|
||
// be called and that it works correctly.
|
||
const auto get_current_process_id =
|
||
GET_FUNCTION_REQUIRED(L"kernel32.dll", GetCurrentProcessId);
|
||
EXPECT_EQ(&GetCurrentProcessId, get_current_process_id);
|
||
ASSERT_TRUE(get_current_process_id);
|
||
EXPECT_EQ(GetCurrentProcessId(), get_current_process_id());
|
||
|
||
// GET_FUNCTION_REQUIRED() and GET_FUNCTION() should behave identically when
|
||
// the function is present.
|
||
EXPECT_EQ(get_current_process_id,
|
||
GET_FUNCTION(L"kernel32.dll", GetCurrentProcessId));
|
||
|
||
// Using a leading :: should also work.
|
||
EXPECT_EQ(get_current_process_id,
|
||
GET_FUNCTION(L"kernel32.dll", ::GetCurrentProcessId));
|
||
EXPECT_EQ(get_current_process_id,
|
||
GET_FUNCTION_REQUIRED(L"kernel32.dll", ::GetCurrentProcessId));
|
||
|
||
// Try a function that’s declared in the SDK’s headers but that has no import
|
||
// library.
|
||
EXPECT_TRUE(GET_FUNCTION_REQUIRED(L"ntdll.dll", RtlNtStatusToDosError));
|
||
|
||
// GetNamedPipeClientProcessId() is only available on Vista and later.
|
||
const auto get_named_pipe_client_process_id =
|
||
GET_FUNCTION(L"kernel32.dll", GetNamedPipeClientProcessId);
|
||
const DWORD version = GetVersion();
|
||
const DWORD major_version = LOBYTE(LOWORD(version));
|
||
EXPECT_EQ(major_version >= 6, get_named_pipe_client_process_id != nullptr);
|
||
|
||
// Test that GET_FUNCTION() can fail by trying a nonexistent library and a
|
||
// symbol that doesn’t exist in the specified library.
|
||
EXPECT_FALSE(GET_FUNCTION(L"not_a_real_library.dll", TerminateProcess));
|
||
EXPECT_FALSE(GET_FUNCTION(L"ntdll.dll", TerminateProcess));
|
||
EXPECT_FALSE(GET_FUNCTION(L"not_a_real_library.dll", ::TerminateProcess));
|
||
EXPECT_FALSE(GET_FUNCTION(L"ntdll.dll", ::TerminateProcess));
|
||
|
||
// Here it is!
|
||
EXPECT_TRUE(GET_FUNCTION(L"kernel32.dll", TerminateProcess));
|
||
EXPECT_TRUE(GET_FUNCTION(L"kernel32.dll", ::TerminateProcess));
|
||
}
|
||
|
||
} // namespace
|
||
} // namespace test
|
||
} // namespace crashpad
|