Home | History | Annotate | Download | only in memory
      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/memory/scoped_native_library.h"
      6 #if defined(OS_WIN)
      7 #include "base/file_path.h"
      8 #endif
      9 
     10 #include "testing/gtest/include/gtest/gtest.h"
     11 
     12 // Tests whether or not a function pointer retrieved via ScopedNativeLibrary
     13 // is available only in a scope.
     14 TEST(ScopedNativeLibrary, Basic) {
     15 #if defined(OS_WIN)
     16   // Get the pointer to DirectDrawCreate() from "ddraw.dll" and verify it
     17   // is valid only in this scope.
     18   // FreeLibrary() doesn't actually unload a DLL until its reference count
     19   // becomes zero, i.e. this function pointer is still valid if the DLL used
     20   // in this test is also used by another part of this executable.
     21   // So, this test uses "ddraw.dll", which is not used by Chrome at all but
     22   // installed on all versions of Windows.
     23   FARPROC test_function;
     24   {
     25     FilePath path(base::GetNativeLibraryName(L"ddraw"));
     26     base::ScopedNativeLibrary library(path);
     27     test_function = reinterpret_cast<FARPROC>(
     28         library.GetFunctionPointer("DirectDrawCreate"));
     29     EXPECT_EQ(0, IsBadCodePtr(test_function));
     30   }
     31   EXPECT_NE(0, IsBadCodePtr(test_function));
     32 #endif
     33 }
     34