1/*
2 * Copyright (C) 2012 Igalia S.L.
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Library General Public
6 * License as published by the Free Software Foundation; either
7 * version 2 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Library General Public License for more details.
13 *
14 * You should have received a copy of the GNU Library General Public License
15 * along with this library; see the file COPYING.LIB. If not, write to
16 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
17 * Boston, MA 02110-1301, USA.
18 */
19
20#include "config.h"
21#include "WebProcessTest.h"
22#include <JavaScriptCore/JSContextRef.h>
23#include <JavaScriptCore/JSRetainPtr.h>
24#include <gio/gio.h>
25#if USE(GSTREAMER)
26#include <gst/gst.h>
27#endif
28#include <jsc/jsc.h>
29#include <stdlib.h>
30#include <string.h>
31#include <wtf/Deque.h>
32#include <wtf/ProcessID.h>
33#include <wtf/glib/GRefPtr.h>
34#include <wtf/glib/GUniquePtr.h>
35#include <wtf/text/CString.h>
36
37#if PLATFORM(GTK)
38#include <webkit2/webkit-web-extension.h>
39#elif PLATFORM(WPE)
40#include <wpe/webkit-web-extension.h>
41#endif
42
43static const char introspectionXML[] =
44 "<node>"
45 " <interface name='org.webkit.gtk.WebExtensionTest'>"
46 " <method name='GetTitle'>"
47 " <arg type='t' name='pageID' direction='in'/>"
48 " <arg type='s' name='title' direction='out'/>"
49 " </method>"
50 " <method name='InputElementIsUserEdited'>"
51 " <arg type='t' name='pageID' direction='in'/>"
52 " <arg type='s' name='elementID' direction='in'/>"
53 " <arg type='b' name='isUserEdited' direction='out'/>"
54 " </method>"
55 " <method name='AbortProcess'>"
56 " </method>"
57 " <method name='RunJavaScriptInIsolatedWorld'>"
58 " <arg type='t' name='pageID' direction='in'/>"
59 " <arg type='s' name='script' direction='in'/>"
60 " </method>"
61 " <method name='GetProcessIdentifier'>"
62 " <arg type='u' name='identifier' direction='out'/>"
63 " </method>"
64 " <method name='RemoveAVPluginsFromGSTRegistry'>"
65 " </method>"
66 " <signal name='DocumentLoaded'/>"
67 " <signal name='FormControlsAssociated'>"
68 " <arg type='s' name='formIds' direction='out'/>"
69 " </signal>"
70 " <signal name='FormSubmissionWillSendDOMEvent'>"
71 " <arg type='s' name='formID' direction='out'/>"
72 " <arg type='s' name='textFieldNames' direction='out'/>"
73 " <arg type='s' name='textFieldValues' direction='out'/>"
74 " <arg type='b' name='targetFrameIsMainFrame' direction='out'/>"
75 " <arg type='b' name='sourceFrameIsMainFrame' direction='out'/>"
76 " </signal>"
77 " <signal name='FormSubmissionWillComplete'>"
78 " <arg type='s' name='formID' direction='out'/>"
79 " <arg type='s' name='textFieldNames' direction='out'/>"
80 " <arg type='s' name='textFieldValues' direction='out'/>"
81 " <arg type='b' name='targetFrameIsMainFrame' direction='out'/>"
82 " <arg type='b' name='sourceFrameIsMainFrame' direction='out'/>"
83 " </signal>"
84 " <signal name='URIChanged'>"
85 " <arg type='s' name='uri' direction='out'/>"
86 " </signal>"
87 " </interface>"
88 "</node>";
89
90
91typedef enum {
92 DocumentLoadedSignal,
93 URIChangedSignal,
94 FormControlsAssociatedSignal,
95 FormSubmissionWillSendDOMEventSignal,
96 FormSubmissionWillCompleteSignal,
97} DelayedSignalType;
98
99struct DelayedSignal {
100 explicit DelayedSignal(DelayedSignalType type)
101 : type(type)
102 {
103 }
104
105 DelayedSignal(DelayedSignalType type, const char* str)
106 : type(type)
107 , str(str)
108 {
109 }
110
111 DelayedSignal(DelayedSignalType type, const char* str, const char* str2, const char* str3, gboolean b, gboolean b2)
112 : type(type)
113 , str(str)
114 , str2(str2)
115 , str3(str3)
116 , b(b)
117 , b2(b2)
118 {
119 }
120
121 DelayedSignalType type;
122 CString str;
123 CString str2;
124 CString str3;
125 gboolean b;
126 gboolean b2;
127};
128
129Deque<DelayedSignal> delayedSignalsQueue;
130
131static void emitDocumentLoaded(GDBusConnection* connection)
132{
133 bool ok = g_dbus_connection_emit_signal(
134 connection,
135 0,
136 "/org/webkit/gtk/WebExtensionTest",
137 "org.webkit.gtk.WebExtensionTest",
138 "DocumentLoaded",
139 0,
140 0);
141 g_assert_true(ok);
142}
143
144static void documentLoadedCallback(WebKitWebPage* webPage, WebKitWebExtension* extension)
145{
146#if PLATFORM(GTK)
147 WebKitDOMDocument* document = webkit_web_page_get_dom_document(webPage);
148 G_GNUC_BEGIN_IGNORE_DEPRECATIONS;
149 GRefPtr<WebKitDOMDOMWindow> window = adoptGRef(webkit_dom_document_get_default_view(document));
150 webkit_dom_dom_window_webkit_message_handlers_post_message(window.get(), "dom", "DocumentLoaded");
151 G_GNUC_END_IGNORE_DEPRECATIONS;
152#endif
153
154 gpointer data = g_object_get_data(G_OBJECT(extension), "dbus-connection");
155 if (data)
156 emitDocumentLoaded(G_DBUS_CONNECTION(data));
157 else
158 delayedSignalsQueue.append(DelayedSignal(DocumentLoadedSignal));
159}
160
161static void emitURIChanged(GDBusConnection* connection, const char* uri)
162{
163 bool ok = g_dbus_connection_emit_signal(
164 connection,
165 0,
166 "/org/webkit/gtk/WebExtensionTest",
167 "org.webkit.gtk.WebExtensionTest",
168 "URIChanged",
169 g_variant_new("(s)", uri),
170 0);
171 g_assert_true(ok);
172}
173
174static void uriChangedCallback(WebKitWebPage* webPage, GParamSpec* pspec, WebKitWebExtension* extension)
175{
176 gpointer data = g_object_get_data(G_OBJECT(extension), "dbus-connection");
177 if (data)
178 emitURIChanged(G_DBUS_CONNECTION(data), webkit_web_page_get_uri(webPage));
179 else
180 delayedSignalsQueue.append(DelayedSignal(URIChangedSignal, webkit_web_page_get_uri(webPage)));
181}
182
183static gboolean sendRequestCallback(WebKitWebPage*, WebKitURIRequest* request, WebKitURIResponse* redirectResponse, gpointer)
184{
185 gboolean returnValue = FALSE;
186 const char* requestURI = webkit_uri_request_get_uri(request);
187 g_assert_nonnull(requestURI);
188
189 if (const char* suffix = g_strrstr(requestURI, "/remove-this/javascript.js")) {
190 GUniquePtr<char> prefix(g_strndup(requestURI, strlen(requestURI) - strlen(suffix)));
191 GUniquePtr<char> newURI(g_strdup_printf("%s/javascript.js", prefix.get()));
192 webkit_uri_request_set_uri(request, newURI.get());
193 } else if (const char* suffix = g_strrstr(requestURI, "/remove-this/javascript-after-redirection.js")) {
194 // Redirected from /redirected.js, redirectResponse should be nullptr.
195 g_assert_true(WEBKIT_IS_URI_RESPONSE(redirectResponse));
196 g_assert_true(g_str_has_suffix(webkit_uri_response_get_uri(redirectResponse), "/redirected.js"));
197
198 GUniquePtr<char> prefix(g_strndup(requestURI, strlen(requestURI) - strlen(suffix)));
199 GUniquePtr<char> newURI(g_strdup_printf("%s/javascript-after-redirection.js", prefix.get()));
200 webkit_uri_request_set_uri(request, newURI.get());
201 } else if (g_str_has_suffix(requestURI, "/redirected.js")) {
202 // Original request, redirectResponse should be nullptr.
203 g_assert_null(redirectResponse);
204 } else if (g_str_has_suffix(requestURI, "/add-do-not-track-header")) {
205 SoupMessageHeaders* headers = webkit_uri_request_get_http_headers(request);
206 g_assert_nonnull(headers);
207 soup_message_headers_append(headers, "DNT", "1");
208 } else if (g_str_has_suffix(requestURI, "/normal-change-request") && !g_strrstr(requestURI, "/redirect-js/")) {
209 GUniquePtr<char> prefix(g_strndup(requestURI, strlen(requestURI) - strlen("/normal-change-request")));
210 GUniquePtr<char> newURI(g_strdup_printf("%s/request-changed%s", prefix.get(), redirectResponse ? "-on-redirect" : ""));
211 webkit_uri_request_set_uri(request, newURI.get());
212 } else if (g_str_has_suffix(requestURI, "/http-get-method")) {
213 g_assert_cmpstr(webkit_uri_request_get_http_method(request), ==, "GET");
214 g_assert_cmpstr(webkit_uri_request_get_http_method(request), ==, SOUP_METHOD_GET);
215 } else if (g_str_has_suffix(requestURI, "/http-post-method")) {
216 g_assert_cmpstr(webkit_uri_request_get_http_method(request), ==, "POST");
217 g_assert_cmpstr(webkit_uri_request_get_http_method(request), ==, SOUP_METHOD_POST);
218 returnValue = TRUE;
219 } else if (g_str_has_suffix(requestURI, "/cancel-this.js"))
220 returnValue = TRUE;
221
222 return returnValue;
223}
224
225static GVariant* serializeContextMenu(WebKitContextMenu* menu)
226{
227 GVariantBuilder builder;
228 g_variant_builder_init(&builder, G_VARIANT_TYPE_ARRAY);
229 GList* items = webkit_context_menu_get_items(menu);
230 for (GList* it = items; it; it = g_list_next(it))
231 g_variant_builder_add(&builder, "u", webkit_context_menu_item_get_stock_action(WEBKIT_CONTEXT_MENU_ITEM(it->data)));
232 return g_variant_builder_end(&builder);
233}
234
235static GVariant* serializeNode(JSCValue* node)
236{
237 GVariantBuilder builder;
238 g_variant_builder_init(&builder, G_VARIANT_TYPE_ARRAY);
239 GRefPtr<JSCValue> value = adoptGRef(jsc_value_object_get_property(node, "nodeName"));
240 g_variant_builder_add(&builder, "{sv}", "Name", g_variant_new_take_string(jsc_value_to_string(value.get())));
241 value = adoptGRef(jsc_value_object_get_property(node, "nodeType"));
242 g_variant_builder_add(&builder, "{sv}", "Type", g_variant_new_uint32(jsc_value_to_int32(value.get())));
243 value = adoptGRef(jsc_value_object_get_property(node, "textContent"));
244 g_variant_builder_add(&builder, "{sv}", "Contents", g_variant_new_take_string(jsc_value_to_string(value.get())));
245 value = adoptGRef(jsc_value_object_get_property(node, "parentNode"));
246 if (jsc_value_is_null(value.get()))
247 g_variant_builder_add(&builder, "{sv}", "Parent", g_variant_new_string("ROOT"));
248 else {
249 value = jsc_value_object_get_property(value.get(), "nodeName");
250 g_variant_builder_add(&builder, "{sv}", "Parent", g_variant_new_take_string(jsc_value_to_string(value.get())));
251 }
252 return g_variant_builder_end(&builder);
253}
254
255static gboolean contextMenuCallback(WebKitWebPage* page, WebKitContextMenu* menu, WebKitWebHitTestResult* hitTestResult, gpointer)
256{
257 const char* pageURI = webkit_web_page_get_uri(page);
258 if (!g_strcmp0(pageURI, "ContextMenuTestDefault")) {
259 webkit_context_menu_set_user_data(menu, serializeContextMenu(menu));
260 return FALSE;
261 }
262
263 if (!g_strcmp0(pageURI, "ContextMenuTestCustom")) {
264 // Remove Back and Forward, and add Inspector action.
265 webkit_context_menu_remove(menu, webkit_context_menu_first(menu));
266 webkit_context_menu_remove(menu, webkit_context_menu_first(menu));
267 webkit_context_menu_append(menu, webkit_context_menu_item_new_separator());
268 webkit_context_menu_append(menu, webkit_context_menu_item_new_from_stock_action(WEBKIT_CONTEXT_MENU_ACTION_INSPECT_ELEMENT));
269 webkit_context_menu_set_user_data(menu, serializeContextMenu(menu));
270 return TRUE;
271 }
272
273 if (!g_strcmp0(pageURI, "ContextMenuTestClear")) {
274 webkit_context_menu_remove_all(menu);
275 return TRUE;
276 }
277
278 if (!g_strcmp0(pageURI, "ContextMenuTestNode")) {
279 WebKitDOMNode* node = webkit_web_hit_test_result_get_node(hitTestResult);
280 g_assert_true(WEBKIT_DOM_IS_NODE(node));
281 auto* frame = webkit_web_page_get_main_frame(page);
282 GRefPtr<JSCValue> jsNode = adoptGRef(webkit_frame_get_js_value_for_dom_object(frame, WEBKIT_DOM_OBJECT(node)));
283 webkit_context_menu_set_user_data(menu, serializeNode(jsNode.get()));
284 return TRUE;
285 }
286
287 return FALSE;
288}
289
290static void consoleMessageSentCallback(WebKitWebPage* webPage, WebKitConsoleMessage* consoleMessage)
291{
292 g_assert_nonnull(consoleMessage);
293 GRefPtr<GVariant> variant = g_variant_new("(uusus)", webkit_console_message_get_source(consoleMessage),
294 webkit_console_message_get_level(consoleMessage), webkit_console_message_get_text(consoleMessage),
295 webkit_console_message_get_line(consoleMessage), webkit_console_message_get_source_id(consoleMessage));
296 GUniquePtr<char> messageString(g_variant_print(variant.get(), FALSE));
297 GRefPtr<JSCContext> jsContext = adoptGRef(webkit_frame_get_js_context(webkit_web_page_get_main_frame(webPage)));
298 GRefPtr<JSCValue> console = adoptGRef(jsc_context_evaluate(jsContext.get(), "window.webkit.messageHandlers.console", -1));
299 g_assert_true(JSC_IS_VALUE(console.get()));
300 if (jsc_value_is_object(console.get())) {
301 GRefPtr<JSCValue> result = adoptGRef(jsc_value_object_invoke_method(console.get(), "postMessage", G_TYPE_STRING, messageString.get(), G_TYPE_NONE));
302 g_assert_true(JSC_IS_VALUE(result.get()));
303 }
304}
305
306static void emitFormControlsAssociated(GDBusConnection* connection, const char* formIds)
307{
308 bool ok = g_dbus_connection_emit_signal(
309 connection,
310 nullptr,
311 "/org/webkit/gtk/WebExtensionTest",
312 "org.webkit.gtk.WebExtensionTest",
313 "FormControlsAssociated",
314 g_variant_new("(s)", formIds),
315 nullptr);
316 g_assert_true(ok);
317}
318
319static void formControlsAssociatedForFrameCallback(WebKitWebPage* webPage, GPtrArray* formElements, WebKitFrame* frame, WebKitWebExtension* extension)
320{
321 GString* formIdsBuilder = g_string_new(nullptr);
322 for (guint i = 0; i < formElements->len; ++i) {
323 g_assert_true(WEBKIT_DOM_IS_ELEMENT(g_ptr_array_index(formElements, i)));
324 GRefPtr<JSCValue> value = adoptGRef(webkit_frame_get_js_value_for_dom_object(frame, WEBKIT_DOM_OBJECT(g_ptr_array_index(formElements, i))));
325 g_assert_true(JSC_IS_VALUE(value.get()));
326 g_assert_true(jsc_value_is_object(value.get()));
327 g_assert_true(jsc_value_object_is_instance_of(value.get(), "Element"));
328 GRefPtr<JSCValue> idValue = adoptGRef(jsc_value_object_get_property(value.get(), "id"));
329 GUniquePtr<char> elementID(jsc_value_to_string(idValue.get()));
330 g_string_append(formIdsBuilder, elementID.get());
331 }
332 if (!formIdsBuilder->len) {
333 g_string_free(formIdsBuilder, TRUE);
334 return;
335 }
336 GUniquePtr<char> formIds(g_string_free(formIdsBuilder, FALSE));
337 gpointer data = g_object_get_data(G_OBJECT(extension), "dbus-connection");
338 if (data)
339 emitFormControlsAssociated(G_DBUS_CONNECTION(data), formIds.get());
340 else
341 delayedSignalsQueue.append(DelayedSignal(FormControlsAssociatedSignal, formIds.get()));
342}
343
344static void emitFormSubmissionEvent(GDBusConnection* connection, const char* methodName, const char* formID, const char* names, const char* values, gboolean targetFrameIsMainFrame, gboolean sourceFrameIsMainFrame)
345{
346 bool ok = g_dbus_connection_emit_signal(
347 connection,
348 nullptr,
349 "/org/webkit/gtk/WebExtensionTest",
350 "org.webkit.gtk.WebExtensionTest",
351 methodName,
352 g_variant_new("(sssbb)", formID ? formID : "", names, values, targetFrameIsMainFrame, sourceFrameIsMainFrame),
353 nullptr);
354 g_assert_true(ok);
355}
356
357static void handleFormSubmissionCallback(WebKitWebPage* webPage, DelayedSignalType delayedSignalType, const char* methodName, const char* formID, WebKitFrame* sourceFrame, WebKitFrame* targetFrame, GPtrArray* textFieldNames, GPtrArray* textFieldValues, WebKitWebExtension* extension)
358{
359 GString* namesBuilder = g_string_new(nullptr);
360 for (guint i = 0; i < textFieldNames->len; ++i) {
361 auto* name = static_cast<char*>(g_ptr_array_index(textFieldNames, i));
362 g_string_append(namesBuilder, name);
363 g_string_append_c(namesBuilder, ',');
364 }
365 GUniquePtr<char> names(g_string_free(namesBuilder, FALSE));
366
367 GString* valuesBuilder = g_string_new(nullptr);
368 for (guint i = 0; i < textFieldValues->len; ++i) {
369 auto* value = static_cast<char*>(g_ptr_array_index(textFieldValues, i));
370 g_string_append(valuesBuilder, value);
371 g_string_append_c(valuesBuilder, ',');
372 }
373 GUniquePtr<char> values(g_string_free(valuesBuilder, FALSE));
374
375 gpointer data = g_object_get_data(G_OBJECT(extension), "dbus-connection");
376 if (data)
377 emitFormSubmissionEvent(G_DBUS_CONNECTION(data), methodName, formID, names.get(), values.get(), webkit_frame_is_main_frame(targetFrame), webkit_frame_is_main_frame(sourceFrame));
378 else
379 delayedSignalsQueue.append(DelayedSignal(delayedSignalType, formID, names.get(), values.get(), webkit_frame_is_main_frame(targetFrame), webkit_frame_is_main_frame(sourceFrame)));
380}
381
382static void willSubmitFormCallback(WebKitWebPage* webPage, WebKitDOMElement* formElement, WebKitFormSubmissionStep step, WebKitFrame* sourceFrame, WebKitFrame* targetFrame, GPtrArray* textFieldNames, GPtrArray* textFieldValues, WebKitWebExtension* extension)
383{
384#if PLATFORM(GTK)
385 G_GNUC_BEGIN_IGNORE_DEPRECATIONS;
386 g_assert_true(WEBKIT_DOM_IS_HTML_FORM_ELEMENT(formElement));
387 G_GNUC_END_IGNORE_DEPRECATIONS;
388#endif
389 GRefPtr<JSCValue> jsFormElement = adoptGRef(webkit_frame_get_js_value_for_dom_object(sourceFrame, WEBKIT_DOM_OBJECT(formElement)));
390 g_assert_true(JSC_IS_VALUE(jsFormElement.get()));
391 g_assert_true(jsc_value_is_object(jsFormElement.get()));
392 g_assert_true(jsc_value_object_is_instance_of(jsFormElement.get(), "HTMLFormElement"));
393 GRefPtr<JSCValue> idValue = adoptGRef(jsc_value_object_get_property(jsFormElement.get(), "id"));
394 GUniquePtr<char> formID(jsc_value_to_string(idValue.get()));
395
396 switch (step) {
397 case WEBKIT_FORM_SUBMISSION_WILL_SEND_DOM_EVENT:
398 handleFormSubmissionCallback(webPage, FormSubmissionWillSendDOMEventSignal, "FormSubmissionWillSendDOMEvent", formID.get(), sourceFrame, targetFrame, textFieldNames, textFieldValues, extension);
399 break;
400 case WEBKIT_FORM_SUBMISSION_WILL_COMPLETE:
401 handleFormSubmissionCallback(webPage, FormSubmissionWillCompleteSignal, "FormSubmissionWillComplete", formID.get(), sourceFrame, targetFrame, textFieldNames, textFieldValues, extension);
402 break;
403 default:
404 g_assert_not_reached();
405 }
406}
407
408static void pageCreatedCallback(WebKitWebExtension* extension, WebKitWebPage* webPage, gpointer)
409{
410 g_signal_connect(webPage, "document-loaded", G_CALLBACK(documentLoadedCallback), extension);
411 g_signal_connect(webPage, "notify::uri", G_CALLBACK(uriChangedCallback), extension);
412 g_signal_connect(webPage, "send-request", G_CALLBACK(sendRequestCallback), nullptr);
413 g_signal_connect(webPage, "console-message-sent", G_CALLBACK(consoleMessageSentCallback), nullptr);
414 g_signal_connect(webPage, "context-menu", G_CALLBACK(contextMenuCallback), nullptr);
415 g_signal_connect(webPage, "form-controls-associated-for-frame", G_CALLBACK(formControlsAssociatedForFrameCallback), extension);
416 g_signal_connect(webPage, "will-submit-form", G_CALLBACK(willSubmitFormCallback), extension);
417}
418
419static char* echoCallback(const char* message)
420{
421 return g_strdup(message);
422}
423
424static void windowObjectCleared(WebKitScriptWorld* world, WebKitWebPage* page, WebKitFrame* frame, gpointer)
425{
426 GRefPtr<JSCContext> jsContext = adoptGRef(webkit_frame_get_js_context_for_script_world(frame, world));
427 g_assert_true(JSC_IS_CONTEXT(jsContext.get()));
428 GRefPtr<JSCValue> function = adoptGRef(jsc_value_new_function(jsContext.get(), "echo", G_CALLBACK(echoCallback), NULL, NULL, G_TYPE_STRING, 1, G_TYPE_STRING));
429 jsc_context_set_value(jsContext.get(), "echo", function.get());
430}
431
432static WebKitWebPage* getWebPage(WebKitWebExtension* extension, uint64_t pageID, GDBusMethodInvocation* invocation)
433{
434 WebKitWebPage* page = webkit_web_extension_get_page(extension, pageID);
435 if (!page) {
436 g_dbus_method_invocation_return_error(
437 invocation, G_DBUS_ERROR, G_DBUS_ERROR_INVALID_ARGS,
438 "Invalid page ID: %" G_GUINT64_FORMAT, pageID);
439 return 0;
440 }
441
442 g_assert_cmpuint(webkit_web_page_get_id(page), ==, pageID);
443 return page;
444}
445
446static void methodCallCallback(GDBusConnection* connection, const char* sender, const char* objectPath, const char* interfaceName, const char* methodName, GVariant* parameters, GDBusMethodInvocation* invocation, gpointer userData)
447{
448 if (g_strcmp0(interfaceName, "org.webkit.gtk.WebExtensionTest"))
449 return;
450
451 if (!g_strcmp0(methodName, "GetTitle")) {
452 uint64_t pageID;
453 g_variant_get(parameters, "(t)", &pageID);
454 WebKitWebPage* page = getWebPage(WEBKIT_WEB_EXTENSION(userData), pageID, invocation);
455 if (!page)
456 return;
457
458 GRefPtr<JSCContext> jsContext = adoptGRef(webkit_frame_get_js_context(webkit_web_page_get_main_frame(page)));
459 GRefPtr<JSCValue> titleValue = adoptGRef(jsc_context_evaluate(jsContext.get(), "document.title", -1));
460 GUniquePtr<char> title(jsc_value_to_string(titleValue.get()));
461 g_dbus_method_invocation_return_value(invocation, g_variant_new("(s)", title.get()));
462 } else if (!g_strcmp0(methodName, "InputElementIsUserEdited")) {
463 uint64_t pageID;
464 const char* elementID;
465 g_variant_get(parameters, "(t&s)", &pageID, &elementID);
466 WebKitWebPage* page = getWebPage(WEBKIT_WEB_EXTENSION(userData), pageID, invocation);
467 if (!page)
468 return;
469
470 WebKitDOMDocument* document = webkit_web_page_get_dom_document(page);
471 WebKitFrame* frame = webkit_web_page_get_main_frame(page);
472 GRefPtr<JSCContext> jsContext = adoptGRef(webkit_frame_get_js_context(frame));
473 GRefPtr<JSCValue> jsDocument = adoptGRef(webkit_frame_get_js_value_for_dom_object(frame, WEBKIT_DOM_OBJECT(document)));
474 GRefPtr<JSCValue> jsInputElement = adoptGRef(jsc_value_object_invoke_method(jsDocument.get(), "getElementById", G_TYPE_STRING, elementID, G_TYPE_NONE));
475 WebKitDOMNode* node = webkit_dom_node_for_js_value(jsInputElement.get());
476 gboolean isUserEdited = webkit_dom_element_html_input_element_is_user_edited(WEBKIT_DOM_ELEMENT(node));
477 g_dbus_method_invocation_return_value(invocation, g_variant_new("(b)", isUserEdited));
478 } else if (!g_strcmp0(methodName, "RunJavaScriptInIsolatedWorld")) {
479 uint64_t pageID;
480 const char* script;
481 g_variant_get(parameters, "(t&s)", &pageID, &script);
482 WebKitWebPage* page = getWebPage(WEBKIT_WEB_EXTENSION(userData), pageID, invocation);
483 if (!page)
484 return;
485
486 GRefPtr<WebKitScriptWorld> world = adoptGRef(webkit_script_world_new());
487 g_assert_true(webkit_script_world_get_default() != world.get());
488 g_assert_true(g_str_has_prefix(webkit_script_world_get_name(world.get()), "UniqueWorld_"));
489 WebKitFrame* frame = webkit_web_page_get_main_frame(page);
490 GRefPtr<JSCContext> jsContext = adoptGRef(webkit_frame_get_js_context_for_script_world(frame, world.get()));
491 GRefPtr<JSCValue> result = adoptGRef(jsc_context_evaluate(jsContext.get(), script, -1));
492 g_dbus_method_invocation_return_value(invocation, 0);
493 } else if (!g_strcmp0(methodName, "AbortProcess")) {
494 abort();
495 } else if (!g_strcmp0(methodName, "GetProcessIdentifier")) {
496 g_dbus_method_invocation_return_value(invocation,
497 g_variant_new("(u)", static_cast<guint32>(getCurrentProcessID())));
498 } else if (!g_strcmp0(methodName, "RemoveAVPluginsFromGSTRegistry")) {
499#if USE(GSTREAMER)
500 gst_init(nullptr, nullptr);
501 static const char* avPlugins[] = { "libav", "omx", "vaapi", nullptr };
502 GstRegistry* registry = gst_registry_get();
503 for (unsigned i = 0; avPlugins[i]; ++i) {
504 if (GstPlugin* plugin = gst_registry_find_plugin(registry, avPlugins[i])) {
505 gst_registry_remove_plugin(registry, plugin);
506 gst_object_unref(plugin);
507 }
508 }
509#endif
510 g_dbus_method_invocation_return_value(invocation, nullptr);
511 }
512}
513
514static const GDBusInterfaceVTable interfaceVirtualTable = {
515 methodCallCallback, 0, 0, { 0, }
516};
517
518static void busAcquiredCallback(GDBusConnection* connection, const char* name, gpointer userData)
519{
520 static GDBusNodeInfo* introspectionData = 0;
521 if (!introspectionData)
522 introspectionData = g_dbus_node_info_new_for_xml(introspectionXML, 0);
523
524 GUniqueOutPtr<GError> error;
525 unsigned registrationID = g_dbus_connection_register_object(
526 connection,
527 "/org/webkit/gtk/WebExtensionTest",
528 introspectionData->interfaces[0],
529 &interfaceVirtualTable,
530 g_object_ref(userData),
531 static_cast<GDestroyNotify>(g_object_unref),
532 &error.outPtr());
533 if (!registrationID)
534 g_warning("Failed to register object: %s\n", error->message);
535
536 g_object_set_data(G_OBJECT(userData), "dbus-connection", connection);
537 while (delayedSignalsQueue.size()) {
538 DelayedSignal delayedSignal = delayedSignalsQueue.takeFirst();
539 switch (delayedSignal.type) {
540 case DocumentLoadedSignal:
541 emitDocumentLoaded(connection);
542 break;
543 case URIChangedSignal:
544 emitURIChanged(connection, delayedSignal.str.data());
545 break;
546 case FormControlsAssociatedSignal:
547 emitFormControlsAssociated(connection, delayedSignal.str.data());
548 break;
549 case FormSubmissionWillCompleteSignal:
550 emitFormSubmissionEvent(connection, "FormSubmissionWillComplete", delayedSignal.str.data(), delayedSignal.str2.data(), delayedSignal.str3.data(), delayedSignal.b, delayedSignal.b2);
551 break;
552 case FormSubmissionWillSendDOMEventSignal:
553 emitFormSubmissionEvent(connection, "FormSubmissionWillSendDOMEvent", delayedSignal.str.data(), delayedSignal.str2.data(), delayedSignal.str3.data(), delayedSignal.b, delayedSignal.b2);
554 break;
555 }
556 }
557}
558
559static void registerGResource(void)
560{
561 GUniquePtr<char> resourcesPath(g_build_filename(WEBKIT_TEST_RESOURCES_DIR, "webkitglib-tests-resources.gresource", nullptr));
562 GResource* resource = g_resource_load(resourcesPath.get(), nullptr);
563 g_assert_nonnull(resource);
564
565 g_resources_register(resource);
566 g_resource_unref(resource);
567}
568
569extern "C" void webkit_web_extension_initialize_with_user_data(WebKitWebExtension* extension, GVariant* userData)
570{
571 WebKitScriptWorld* isolatedWorld = webkit_script_world_new_with_name("WebExtensionTestScriptWorld");
572 g_assert_true(WEBKIT_IS_SCRIPT_WORLD(isolatedWorld));
573 g_assert_cmpstr(webkit_script_world_get_name(isolatedWorld), ==, "WebExtensionTestScriptWorld");
574 g_object_set_data_full(G_OBJECT(extension), "wk-script-world", isolatedWorld, g_object_unref);
575
576 g_signal_connect(extension, "page-created", G_CALLBACK(pageCreatedCallback), extension);
577 g_signal_connect(webkit_script_world_get_default(), "window-object-cleared", G_CALLBACK(windowObjectCleared), nullptr);
578
579 registerGResource();
580
581 g_assert_nonnull(userData);
582 g_assert_true(g_variant_is_of_type(userData, G_VARIANT_TYPE_UINT32));
583 GUniquePtr<char> busName(g_strdup_printf("org.webkit.gtk.WebExtensionTest%u", g_variant_get_uint32(userData)));
584 g_bus_own_name(
585 G_BUS_TYPE_SESSION,
586 busName.get(),
587 G_BUS_NAME_OWNER_FLAGS_NONE,
588 busAcquiredCallback,
589 0, 0,
590 g_object_ref(extension),
591 static_cast<GDestroyNotify>(g_object_unref));
592}
593