| OLD | NEW |
| (Empty) |
| 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/native_library.h" | |
| 6 | |
| 7 #include <windows.h> | |
| 8 | |
| 9 #include "base/files/file_util.h" | |
| 10 #include "base/strings/stringprintf.h" | |
| 11 #include "base/strings/utf_string_conversions.h" | |
| 12 #include "base/threading/thread_restrictions.h" | |
| 13 | |
| 14 namespace base { | |
| 15 | |
| 16 typedef HMODULE (WINAPI* LoadLibraryFunction)(const wchar_t* file_name); | |
| 17 | |
| 18 namespace { | |
| 19 | |
| 20 NativeLibrary LoadNativeLibraryHelper(const FilePath& library_path, | |
| 21 LoadLibraryFunction load_library_api, | |
| 22 NativeLibraryLoadError* error) { | |
| 23 // LoadLibrary() opens the file off disk. | |
| 24 ThreadRestrictions::AssertIOAllowed(); | |
| 25 | |
| 26 // Switch the current directory to the library directory as the library | |
| 27 // may have dependencies on DLLs in this directory. | |
| 28 bool restore_directory = false; | |
| 29 FilePath current_directory; | |
| 30 if (GetCurrentDirectory(¤t_directory)) { | |
| 31 FilePath plugin_path = library_path.DirName(); | |
| 32 if (!plugin_path.empty()) { | |
| 33 SetCurrentDirectory(plugin_path); | |
| 34 restore_directory = true; | |
| 35 } | |
| 36 } | |
| 37 | |
| 38 HMODULE module = (*load_library_api)(library_path.value().c_str()); | |
| 39 if (!module && error) { | |
| 40 // GetLastError() needs to be called immediately after |load_library_api|. | |
| 41 error->code = GetLastError(); | |
| 42 } | |
| 43 | |
| 44 if (restore_directory) | |
| 45 SetCurrentDirectory(current_directory); | |
| 46 | |
| 47 return module; | |
| 48 } | |
| 49 | |
| 50 } // namespace | |
| 51 | |
| 52 std::string NativeLibraryLoadError::ToString() const { | |
| 53 return StringPrintf("%u", code); | |
| 54 } | |
| 55 | |
| 56 // static | |
| 57 NativeLibrary LoadNativeLibrary(const FilePath& library_path, | |
| 58 NativeLibraryLoadError* error) { | |
| 59 return LoadNativeLibraryHelper(library_path, LoadLibraryW, error); | |
| 60 } | |
| 61 | |
| 62 NativeLibrary LoadNativeLibraryDynamically(const FilePath& library_path) { | |
| 63 typedef HMODULE (WINAPI* LoadLibraryFunction)(const wchar_t* file_name); | |
| 64 | |
| 65 LoadLibraryFunction load_library; | |
| 66 load_library = reinterpret_cast<LoadLibraryFunction>( | |
| 67 GetProcAddress(GetModuleHandle(L"kernel32.dll"), "LoadLibraryW")); | |
| 68 | |
| 69 return LoadNativeLibraryHelper(library_path, load_library, NULL); | |
| 70 } | |
| 71 | |
| 72 // static | |
| 73 void UnloadNativeLibrary(NativeLibrary library) { | |
| 74 FreeLibrary(library); | |
| 75 } | |
| 76 | |
| 77 // static | |
| 78 void* GetFunctionPointerFromNativeLibrary(NativeLibrary library, | |
| 79 const char* name) { | |
| 80 return GetProcAddress(library, name); | |
| 81 } | |
| 82 | |
| 83 // static | |
| 84 string16 GetNativeLibraryName(const string16& name) { | |
| 85 return name + ASCIIToUTF16(".dll"); | |
| 86 } | |
| 87 | |
| 88 } // namespace base | |
| OLD | NEW |