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