Index: sandbox/win/src/interception_unittest.cc |
diff --git a/sandbox/win/src/interception_unittest.cc b/sandbox/win/src/interception_unittest.cc |
deleted file mode 100644 |
index 7b7932af67ccdc7e33d7c580012f60bc82e21d19..0000000000000000000000000000000000000000 |
--- a/sandbox/win/src/interception_unittest.cc |
+++ /dev/null |
@@ -1,260 +0,0 @@ |
-// Copyright (c) 2011 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-// This file contains unit tests for InterceptionManager. |
-// The tests require private information so the whole interception.cc file is |
-// included from this file. |
- |
-#include <windows.h> |
-#include <stddef.h> |
- |
-#include <algorithm> |
-#include <set> |
- |
-#include "base/bits.h" |
-#include "base/memory/scoped_ptr.h" |
-#include "sandbox/win/src/interception.h" |
-#include "sandbox/win/src/interception_internal.h" |
-#include "sandbox/win/src/interceptors.h" |
-#include "sandbox/win/src/target_process.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-namespace sandbox { |
- |
-namespace internal { |
-size_t GetGranularAlignedRandomOffset(size_t size); |
-} |
- |
-// Walks the settings buffer, verifying that the values make sense and counting |
-// objects. |
-// Arguments: |
-// buffer (in): the buffer to walk. |
-// size (in): buffer size |
-// num_dlls (out): count of the dlls on the buffer. |
-// num_function (out): count of intercepted functions. |
-// num_names (out): count of named interceptor functions. |
-void WalkBuffer(void* buffer, size_t size, int* num_dlls, int* num_functions, |
- int* num_names) { |
- ASSERT_TRUE(NULL != buffer); |
- ASSERT_TRUE(NULL != num_functions); |
- ASSERT_TRUE(NULL != num_names); |
- *num_dlls = *num_functions = *num_names = 0; |
- SharedMemory *memory = reinterpret_cast<SharedMemory*>(buffer); |
- |
- ASSERT_GT(size, sizeof(SharedMemory)); |
- DllPatchInfo *dll = &memory->dll_list[0]; |
- |
- for (int i = 0; i < memory->num_intercepted_dlls; i++) { |
- ASSERT_NE(0u, wcslen(dll->dll_name)); |
- ASSERT_EQ(0u, dll->record_bytes % sizeof(size_t)); |
- ASSERT_EQ(0u, dll->offset_to_functions % sizeof(size_t)); |
- ASSERT_NE(0, dll->num_functions); |
- |
- FunctionInfo *function = reinterpret_cast<FunctionInfo*>( |
- reinterpret_cast<char*>(dll) + dll->offset_to_functions); |
- |
- for (int j = 0; j < dll->num_functions; j++) { |
- ASSERT_EQ(0u, function->record_bytes % sizeof(size_t)); |
- |
- char* name = function->function; |
- size_t length = strlen(name); |
- ASSERT_NE(0u, length); |
- name += length + 1; |
- |
- // look for overflows |
- ASSERT_GT(reinterpret_cast<char*>(buffer) + size, name + strlen(name)); |
- |
- // look for a named interceptor |
- if (strlen(name)) { |
- (*num_names)++; |
- EXPECT_TRUE(NULL == function->interceptor_address); |
- } else { |
- EXPECT_TRUE(NULL != function->interceptor_address); |
- } |
- |
- (*num_functions)++; |
- function = reinterpret_cast<FunctionInfo*>( |
- reinterpret_cast<char*>(function) + function->record_bytes); |
- } |
- |
- (*num_dlls)++; |
- dll = reinterpret_cast<DllPatchInfo*>(reinterpret_cast<char*>(dll) + |
- dll->record_bytes); |
- } |
-} |
- |
-TEST(InterceptionManagerTest, GetGranularAlignedRandomOffset) { |
- std::set<size_t> sizes; |
- |
- // 544 is current value of interceptions_.size() * sizeof(ThunkData) + |
- // sizeof(DllInterceptionData). |
- const size_t kThunkBytes = 544; |
- |
- // ciel(log2(544)) = 10. |
- // Alignment must be 2^10 = 1024. |
- const size_t kAlignmentBits = base::bits::Log2Ceiling(kThunkBytes); |
- const size_t kAlignment = static_cast<size_t>(1) << kAlignmentBits; |
- |
- const size_t kAllocGranularity = 65536; |
- |
- // Generate enough sample data to ensure there is at least one value in each |
- // potential bucket. |
- for (size_t i = 0; i < 1000000; i++) |
- sizes.insert(internal::GetGranularAlignedRandomOffset(kThunkBytes)); |
- |
- size_t prev_val = 0; |
- size_t min_val = kAllocGranularity; |
- size_t min_nonzero_val = kAllocGranularity; |
- size_t max_val = 0; |
- |
- for (size_t val : sizes) { |
- ASSERT_LT(val, kAllocGranularity); |
- if (prev_val) |
- ASSERT_EQ(val - prev_val, kAlignment); |
- if (val) |
- min_nonzero_val = std::min(val, min_nonzero_val); |
- min_val = std::min(val, min_val); |
- prev_val = val; |
- max_val = std::max(val, max_val); |
- } |
- ASSERT_EQ(max_val, kAllocGranularity - kAlignment); |
- ASSERT_EQ(0u, min_val); |
- ASSERT_EQ(min_nonzero_val, kAlignment); |
-} |
- |
-TEST(InterceptionManagerTest, BufferLayout1) { |
- wchar_t exe_name[MAX_PATH]; |
- ASSERT_NE(0u, GetModuleFileName(NULL, exe_name, MAX_PATH - 1)); |
- |
- TargetProcess *target = MakeTestTargetProcess(::GetCurrentProcess(), |
- ::GetModuleHandle(exe_name)); |
- |
- InterceptionManager interceptions(target, true); |
- |
- // Any pointer will do for a function pointer. |
- void* function = &interceptions; |
- |
- // We don't care about the interceptor id. |
- interceptions.AddToPatchedFunctions(L"ntdll.dll", "NtCreateFile", |
- INTERCEPTION_SERVICE_CALL, function, |
- OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"kernel32.dll", "CreateFileEx", |
- INTERCEPTION_EAT, function, OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"kernel32.dll", "SomeFileEx", |
- INTERCEPTION_SMART_SIDESTEP, function, |
- OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"user32.dll", "FindWindow", |
- INTERCEPTION_EAT, function, OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"kernel32.dll", "CreateMutex", |
- INTERCEPTION_EAT, function, OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"user32.dll", "PostMsg", |
- INTERCEPTION_EAT, function, OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"user32.dll", "PostMsg", |
- INTERCEPTION_EAT, "replacement", |
- OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"comctl.dll", "SaveAsDlg", |
- INTERCEPTION_EAT, function, OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"ntdll.dll", "NtClose", |
- INTERCEPTION_SERVICE_CALL, function, |
- OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"ntdll.dll", "NtOpenFile", |
- INTERCEPTION_SIDESTEP, function, |
- OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"some.dll", "Superfn", |
- INTERCEPTION_EAT, function, OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"comctl.dll", "SaveAsDlg", |
- INTERCEPTION_EAT, "a", OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"comctl.dll", "SaveAsDlg", |
- INTERCEPTION_SIDESTEP, "ab", OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"comctl.dll", "SaveAsDlg", |
- INTERCEPTION_EAT, "abc", OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"a.dll", "p", |
- INTERCEPTION_EAT, function, OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"b.dll", |
- "TheIncredibleCallToSaveTheWorld", |
- INTERCEPTION_EAT, function, OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"a.dll", "BIsLame", |
- INTERCEPTION_EAT, function, OPEN_KEY_ID); |
- interceptions.AddToPatchedFunctions(L"a.dll", "ARules", |
- INTERCEPTION_EAT, function, OPEN_KEY_ID); |
- |
- // Verify that all interceptions were added |
- ASSERT_EQ(18u, interceptions.interceptions_.size()); |
- |
- size_t buffer_size = interceptions.GetBufferSize(); |
- scoped_ptr<BYTE[]> local_buffer(new BYTE[buffer_size]); |
- |
- ASSERT_TRUE(interceptions.SetupConfigBuffer(local_buffer.get(), |
- buffer_size)); |
- |
- // At this point, the interceptions should have been separated into two |
- // groups: one group with the local ("cold") interceptions, consisting of |
- // everything from ntdll and stuff set as INTRECEPTION_SERVICE_CALL, and |
- // another group with the interceptions belonging to dlls that will be "hot" |
- // patched on the client. The second group lives on local_buffer, and the |
- // first group remains on the list of interceptions (inside the object |
- // "interceptions"). There are 3 local interceptions (of ntdll); the |
- // other 15 have to be sent to the child to be performed "hot". |
- EXPECT_EQ(3u, interceptions.interceptions_.size()); |
- |
- int num_dlls, num_functions, num_names; |
- WalkBuffer(local_buffer.get(), buffer_size, &num_dlls, &num_functions, |
- &num_names); |
- |
- // The 15 interceptions on the buffer (to the child) should be grouped on 6 |
- // dlls. Only four interceptions are using an explicit name for the |
- // interceptor function. |
- EXPECT_EQ(6, num_dlls); |
- EXPECT_EQ(15, num_functions); |
- EXPECT_EQ(4, num_names); |
-} |
- |
-TEST(InterceptionManagerTest, BufferLayout2) { |
- wchar_t exe_name[MAX_PATH]; |
- ASSERT_NE(0u, GetModuleFileName(NULL, exe_name, MAX_PATH - 1)); |
- |
- TargetProcess *target = MakeTestTargetProcess(::GetCurrentProcess(), |
- ::GetModuleHandle(exe_name)); |
- |
- InterceptionManager interceptions(target, true); |
- |
- // Any pointer will do for a function pointer. |
- void* function = &interceptions; |
- interceptions.AddToUnloadModules(L"some01.dll"); |
- // We don't care about the interceptor id. |
- interceptions.AddToPatchedFunctions(L"ntdll.dll", "NtCreateFile", |
- INTERCEPTION_SERVICE_CALL, function, |
- OPEN_FILE_ID); |
- interceptions.AddToPatchedFunctions(L"kernel32.dll", "CreateFileEx", |
- INTERCEPTION_EAT, function, OPEN_FILE_ID); |
- interceptions.AddToUnloadModules(L"some02.dll"); |
- interceptions.AddToPatchedFunctions(L"kernel32.dll", "SomeFileEx", |
- INTERCEPTION_SMART_SIDESTEP, function, |
- OPEN_FILE_ID); |
- // Verify that all interceptions were added |
- ASSERT_EQ(5u, interceptions.interceptions_.size()); |
- |
- size_t buffer_size = interceptions.GetBufferSize(); |
- scoped_ptr<BYTE[]> local_buffer(new BYTE[buffer_size]); |
- |
- ASSERT_TRUE(interceptions.SetupConfigBuffer(local_buffer.get(), |
- buffer_size)); |
- |
- // At this point, the interceptions should have been separated into two |
- // groups: one group with the local ("cold") interceptions, and another |
- // group with the interceptions belonging to dlls that will be "hot" |
- // patched on the client. The second group lives on local_buffer, and the |
- // first group remains on the list of interceptions, in this case just one. |
- EXPECT_EQ(1u, interceptions.interceptions_.size()); |
- |
- int num_dlls, num_functions, num_names; |
- WalkBuffer(local_buffer.get(), buffer_size, &num_dlls, &num_functions, |
- &num_names); |
- |
- EXPECT_EQ(3, num_dlls); |
- EXPECT_EQ(4, num_functions); |
- EXPECT_EQ(0, num_names); |
-} |
- |
-} // namespace sandbox |