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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
/*
* Copyright (C) 2012 Igalia S.L.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public License
* along with this library; see the file COPYING.LIB. If not, write to
* the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#include "config.h"
#include "WebKitExtensionManager.h"
#include "APIString.h"
#include "InjectedBundle.h"
#include "WebKitWebExtensionPrivate.h"
#include <memory>
#include <wtf/FileSystem.h>
#include <wtf/text/CString.h>
namespace WebKit {
WebKitExtensionManager& WebKitExtensionManager::singleton()
{
static NeverDestroyed<WebKitExtensionManager> extensionManager;
return extensionManager;
}
WebKitExtensionManager::WebKitExtensionManager()
{
}
void WebKitExtensionManager::scanModules(const String& webExtensionsDirectory, Vector<String>& modules)
{
auto moduleNames = FileSystem::listDirectory(webExtensionsDirectory);
for (auto& moduleName : moduleNames) {
if (!moduleName.endsWith(".so"))
continue;
auto modulePath = FileSystem::pathByAppendingComponent(webExtensionsDirectory, moduleName);
if (FileSystem::fileExists(modulePath))
modules.append(modulePath);
}
}
static void parseUserData(API::Object* userData, String& webExtensionsDirectory, GRefPtr<GVariant>& initializationUserData)
{
ASSERT(userData->type() == API::Object::Type::String);
CString userDataString = static_cast<API::String*>(userData)->string().utf8();
GRefPtr<GVariant> variant = g_variant_parse(nullptr, userDataString.data(),
userDataString.data() + userDataString.length(), nullptr, nullptr);
ASSERT(variant);
ASSERT(g_variant_check_format_string(variant.get(), "(m&smv)", FALSE));
const char* directory = nullptr;
GVariant* data = nullptr;
g_variant_get(variant.get(), "(m&smv)", &directory, &data);
webExtensionsDirectory = FileSystem::stringFromFileSystemRepresentation(directory);
initializationUserData = adoptGRef(data);
}
bool WebKitExtensionManager::initializeWebExtension(Module* extensionModule, GVariant* userData)
{
WebKitWebExtensionInitializeWithUserDataFunction initializeWithUserDataFunction =
extensionModule->functionPointer<WebKitWebExtensionInitializeWithUserDataFunction>("webkit_web_extension_initialize_with_user_data");
if (initializeWithUserDataFunction) {
initializeWithUserDataFunction(m_extension.get(), userData);
return true;
}
WebKitWebExtensionInitializeFunction initializeFunction =
extensionModule->functionPointer<WebKitWebExtensionInitializeFunction>("webkit_web_extension_initialize");
if (initializeFunction) {
initializeFunction(m_extension.get());
return true;
}
return false;
}
void WebKitExtensionManager::initialize(InjectedBundle* bundle, API::Object* userDataObject)
{
ASSERT(bundle);
ASSERT(userDataObject);
m_extension = adoptGRef(webkitWebExtensionCreate(bundle));
String webExtensionsDirectory;
GRefPtr<GVariant> userData;
parseUserData(userDataObject, webExtensionsDirectory, userData);
if (webExtensionsDirectory.isNull())
return;
Vector<String> modulePaths;
scanModules(webExtensionsDirectory, modulePaths);
for (size_t i = 0; i < modulePaths.size(); ++i) {
auto module = makeUnique<Module>(modulePaths[i]);
if (!module->load())
continue;
if (initializeWebExtension(module.get(), userData.get()))
m_extensionModules.append(module.release());
}
}
} // namespace WebKit
|
