1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "base/scoped_native_library.h" 6 #if defined(OS_WIN) 7 #include "base/files/file_path.h" 8 #endif 9 10 #include "testing/gtest/include/gtest/gtest.h" 11 12 namespace base { 13 14 // Tests whether or not a function pointer retrieved via ScopedNativeLibrary 15 // is available only in a scope. 16 TEST(ScopedNativeLibrary, Basic) { 17 #if defined(OS_WIN) 18 // Get the pointer to DirectDrawCreate() from "ddraw.dll" and verify it 19 // is valid only in this scope. 20 // FreeLibrary() doesn't actually unload a DLL until its reference count 21 // becomes zero, i.e. function pointer is still valid if the DLL used 22 // in this test is also used by another part of this executable. 23 // So, this test uses "ddraw.dll", which is not used by Chrome at all but 24 // installed on all versions of Windows. 25 const char kFunctionName[] = "DirectDrawCreate"; 26 NativeLibrary native_library; 27 { 28 FilePath path(GetNativeLibraryName(L"ddraw")); 29 native_library = LoadNativeLibrary(path, NULL); 30 ScopedNativeLibrary library(native_library); 31 FARPROC test_function = 32 reinterpret_cast<FARPROC>(library.GetFunctionPointer(kFunctionName)); 33 EXPECT_EQ(0, IsBadCodePtr(test_function)); 34 EXPECT_EQ( 35 GetFunctionPointerFromNativeLibrary(native_library, kFunctionName), 36 test_function); 37 } 38 EXPECT_EQ(NULL, 39 GetFunctionPointerFromNativeLibrary(native_library, kFunctionName)); 40 #endif 41 } 42 43 } // namespace base 44
