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
|
/*
* 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 "WebGtkExtensionManager.h"
#include "InjectedBundle.h"
#include "WKBundleAPICast.h"
#include "WKDictionary.h"
#include "WKString.h"
#include "WKType.h"
#include "WebKitWebExtensionPrivate.h"
#include <WebCore/FileSystem.h>
#include <memory>
#include <wtf/text/CString.h>
namespace WebKit {
WebGtkExtensionManager& WebGtkExtensionManager::shared()
{
static NeverDestroyed<WebGtkExtensionManager> extensionManager;
return extensionManager;
}
WebGtkExtensionManager::WebGtkExtensionManager()
{
}
void WebGtkExtensionManager::scanModules(const String& webExtensionsDirectory, Vector<String>& modules)
{
Vector<String> modulePaths = WebCore::listDirectory(webExtensionsDirectory, String("*.so"));
for (size_t i = 0; i < modulePaths.size(); ++i) {
if (WebCore::fileExists(modulePaths[i]))
modules.append(modulePaths[i]);
}
}
static void parseUserData(WKTypeRef userData, String& webExtensionsDirectory, GRefPtr<GVariant>& initializationUserData)
{
ASSERT(userData);
ASSERT(WKGetTypeID(userData) == WKStringGetTypeID());
CString userDataString = toImpl(static_cast<WKStringRef>(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 = WebCore::filenameToString(directory);
initializationUserData = adoptGRef(data);
}
bool WebGtkExtensionManager::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 WebGtkExtensionManager::initialize(WKBundleRef bundle, WKTypeRef userDataString)
{
m_extension = adoptGRef(webkitWebExtensionCreate(toImpl(bundle)));
String webExtensionsDirectory;
GRefPtr<GVariant> userData;
parseUserData(userDataString, webExtensionsDirectory, userData);
if (webExtensionsDirectory.isNull())
return;
Vector<String> modulePaths;
scanModules(webExtensionsDirectory, modulePaths);
for (size_t i = 0; i < modulePaths.size(); ++i) {
auto module = std::make_unique<Module>(modulePaths[i]);
if (!module->load())
continue;
if (initializeWebExtension(module.get(), userData.get()))
m_extensionModules.append(module.release());
}
}
} // namespace WebKit
|
