static Element *get_active_element(Document *doc)
{
char *tagname;
Document *d = NULL;
Element *active, *result = NULL;
active = webkit_dom_html_document_get_active_element((void*)doc);
if (!active) {
return result;
}
tagname = webkit_dom_element_get_tag_name(active);
if (!g_strcmp0(tagname, "FRAME")) {
d = webkit_dom_html_frame_element_get_content_document(WEBKIT_DOM_HTML_FRAME_ELEMENT(active));
result = get_active_element(d);
} else if (!g_strcmp0(tagname, "IFRAME")) {
d = webkit_dom_html_iframe_element_get_content_document(WEBKIT_DOM_HTML_IFRAME_ELEMENT(active));
result = get_active_element(d);
}
g_free(tagname);
if (result) {
return result;
}
return active;
}
// active_element_change_cb is called when the active element is changed, and
// updates bookkeeping and the main process.
static void
active_element_change_cb(WebKitDOMEventTarget *target,
WebKitDOMEvent *event,
gpointer user_data)
{
Exten *exten = user_data;
WebKitDOMDocument *document;
if(WEBKIT_DOM_IS_DOCUMENT(target)) {
document = WEBKIT_DOM_DOCUMENT(target);
} else {
// target is a window.
g_object_get(target, "document", &document, NULL);
}
WebKitDOMElement *active = webkit_dom_document_get_active_element(document);
if(active == NULL || active == exten->active) {
return;
}
if(WEBKIT_DOM_IS_HTML_IFRAME_ELEMENT(active)) {
// The iframe document handles this, unless it isn't registered yet.
// In this case, register it and call recursively.
WebKitDOMDocument *doc =
webkit_dom_html_iframe_element_get_content_document(
WEBKIT_DOM_HTML_IFRAME_ELEMENT(active));
if(!g_hash_table_contains(exten->registered_documents, doc)) {
frame_document_loaded(doc, exten);
active_element_change_cb(
WEBKIT_DOM_EVENT_TARGET(doc),
NULL,
exten);
}
return;
}
exten->active = active;
exten->scroll_target = get_scroll_target(active);
// Check whether the currently active element is an input element.
// If this has changed, signal DBus.
//
// Input elements:
//
// WebKitDOMHTMLAppletElement
// WebKitDOMHTMLEmbedElement
// WebKitDOMHTMLInputElement
// WebKitDOMHTMLTextAreaElement
gboolean input_focus = (
WEBKIT_DOM_IS_HTML_APPLET_ELEMENT(active) ||
WEBKIT_DOM_IS_HTML_EMBED_ELEMENT(active) ||
WEBKIT_DOM_IS_HTML_INPUT_ELEMENT(active) ||
WEBKIT_DOM_IS_HTML_TEXT_AREA_ELEMENT(active));
if(input_focus != exten->last_input_focus) {
exten->last_input_focus = input_focus;
input_focus_changed(
exten->page_id,
input_focus,
exten);
}
}
// frame_document_loaded watches signals emitted from the given document.
static void
frame_document_loaded(WebKitDOMDocument *doc,
Exten *exten)
{
// Track document, and don't register multiple times.
if(!g_hash_table_add(exten->registered_documents, doc)) {
return;
}
WebKitDOMEventTarget *target = WEBKIT_DOM_EVENT_TARGET(
webkit_dom_document_get_default_view(doc));
// listen for focus changes
webkit_dom_event_target_add_event_listener(
target,
"blur",
G_CALLBACK(active_element_change_cb),
true,
exten);
webkit_dom_event_target_add_event_listener(
target,
"focus",
G_CALLBACK(active_element_change_cb),
true,
exten);
// listen for resource loads.
webkit_dom_event_target_add_event_listener(
target,
"beforeload",
G_CALLBACK(adblock_before_load_cb),
true,
exten);
// Scan for existing iframes, and add them as new frames.
WebKitDOMNodeList *nodes = webkit_dom_document_get_elements_by_tag_name(
WEBKIT_DOM_DOCUMENT(doc),
"IFRAME");
gulong i;
gulong len;
if(nodes == NULL) {
len = 0;
} else {
len = webkit_dom_node_list_get_length(nodes);
}
for(i = 0; i < len; i++) {
WebKitDOMDocument *subdoc =
webkit_dom_html_iframe_element_get_content_document(
WEBKIT_DOM_HTML_IFRAME_ELEMENT(
webkit_dom_node_list_item(nodes, i)));
frame_document_loaded(subdoc, exten);
}
// Element hider
inject_adblock_css(doc, exten);
}
/**
* Callback called if a editable element changes it focus state.
* Event target may be a WebKitDOMDocument (in case of iframe) or a
* WebKitDOMDOMWindow.
*/
static void on_editable_change_focus(WebKitDOMEventTarget *target,
WebKitDOMEvent *event, WebKitWebPage *page)
{
WebKitDOMDocument *doc;
WebKitDOMDOMWindow *dom_window;
WebKitDOMElement *active;
GVariant *variant;
char *message;
if (WEBKIT_DOM_IS_DOM_WINDOW(target)) {
g_object_get(target, "document", &doc, NULL);
} else {
/* target is a doc document */
doc = WEBKIT_DOM_DOCUMENT(target);
}
dom_window = webkit_dom_document_get_default_view(doc);
if (!dom_window) {
return;
}
active = webkit_dom_document_get_active_element(doc);
/* Don't do anything if there is no active element */
if (!active) {
return;
}
if (WEBKIT_DOM_IS_HTML_IFRAME_ELEMENT(active)) {
WebKitDOMHTMLIFrameElement *iframe;
WebKitDOMDocument *subdoc;
iframe = WEBKIT_DOM_HTML_IFRAME_ELEMENT(active);
subdoc = webkit_dom_html_iframe_element_get_content_document(iframe);
add_onload_event_observers(subdoc, page);
return;
}
/* Check if the active element is an editable element. */
variant = g_variant_new("(tb)", webkit_web_page_get_id(page),
ext_dom_is_editable(active));
message = g_variant_print(variant, FALSE);
g_variant_unref(variant);
if (!webkit_dom_dom_window_webkit_message_handlers_post_message(dom_window, "focus", message)) {
g_warning("Error sending focus message");
}
g_free(message);
g_object_unref(dom_window);
}
static void test_dom_document_garbage_collection(DomDocumentFixture* fixture, gconstpointer data)
{
guint count = 0;
g_assert(fixture);
WebKitWebView* view = (WebKitWebView*)fixture->webView;
g_assert(view);
WebKitDOMDocument* document = webkit_web_view_get_dom_document(view);
g_assert(document);
g_object_weak_ref(G_OBJECT(document), (GWeakNotify)weak_notify, &count);
WebKitDOMHTMLHeadElement* head = webkit_dom_document_get_head(document);
g_assert(head);
g_object_weak_ref(G_OBJECT(head), (GWeakNotify)weak_notify, &count);
WebKitDOMHTMLElement* body = webkit_dom_document_get_body(document);
g_assert(body);
g_object_weak_ref(G_OBJECT(body), (GWeakNotify)weak_notify, &count);
WebKitDOMHTMLCollection *collection = webkit_dom_document_get_links(document);
g_assert(collection);
g_object_weak_ref(G_OBJECT(collection), (GWeakNotify)weak_notify, &count);
webkit_web_view_load_string(WEBKIT_WEB_VIEW(view), HTML_DOCUMENT_LINKS, NULL, NULL, NULL);
while (g_main_context_pending(NULL))
g_main_context_iteration(NULL, FALSE);
g_assert_cmpuint(count, ==, 3);
g_object_unref(collection);
g_assert_cmpuint(count, ==, 4);
count = 0;
document = webkit_web_view_get_dom_document(view);
g_assert(document);
g_object_weak_ref(G_OBJECT(document), (GWeakNotify)weak_notify, &count);
head = webkit_dom_document_get_head(document);
g_assert(head);
g_object_weak_ref(G_OBJECT(head), (GWeakNotify)weak_notify, &count);
body = webkit_dom_document_get_body(document);
g_assert(body);
g_object_weak_ref(G_OBJECT(body), (GWeakNotify)weak_notify, &count);
collection = webkit_dom_document_get_links(document);
g_assert(collection);
g_object_weak_ref(G_OBJECT(collection), (GWeakNotify)weak_notify, &count);
/* Ask twice for the same object */
WebKitDOMHTMLCollection* collection2 = webkit_dom_document_get_links(document);
g_assert(collection2);
g_object_weak_ref(G_OBJECT(collection2), (GWeakNotify)weak_notify, &count);
g_object_unref(document);
g_object_unref(head);
g_object_unref(body);
g_object_unref(collection);
g_object_unref(collection2);
g_assert_cmpuint(count, ==, 5);
webkit_web_view_load_string(WEBKIT_WEB_VIEW(view), HTML_DOCUMENT_IFRAME, NULL, NULL, NULL);
while (g_main_context_pending(NULL))
g_main_context_iteration(NULL, FALSE);
count = 0;
document = webkit_web_view_get_dom_document(view);
WebKitDOMElement* div = webkit_dom_document_get_element_by_id(document, "test");
g_assert(div);
g_object_weak_ref(G_OBJECT(div), (GWeakNotify)weak_notify, &count);
WebKitDOMElement* iframe = webkit_dom_document_get_element_by_id(document, "iframe");
g_assert(iframe);
webkit_dom_element_set_attribute(iframe, "src", "data:<html><head></head></html>", NULL);
while (g_main_context_pending(NULL))
g_main_context_iteration(NULL, FALSE);
WebKitDOMDocument* iframeDocument = webkit_dom_html_iframe_element_get_content_document(WEBKIT_DOM_HTML_IFRAME_ELEMENT(iframe));
g_assert(iframeDocument);
head = webkit_dom_document_get_head(iframeDocument);
g_assert(head);
g_object_weak_ref(G_OBJECT(head), (GWeakNotify)weak_notify, &count);
webkit_dom_element_set_attribute(iframe, "src", "about:blank", NULL);
while (g_main_context_pending(NULL))
g_main_context_iteration(NULL, FALSE);
g_assert_cmpuint(count, ==, 1);
webkit_web_view_load_string(WEBKIT_WEB_VIEW(view), HTML_DOCUMENT_LINKS, NULL, NULL, NULL);
while (g_main_context_pending(NULL))
g_main_context_iteration(NULL, FALSE);
g_assert_cmpuint(count, ==, 2);
count = 0;
document = webkit_web_view_get_dom_document(view);
g_assert(document);
g_object_weak_ref(G_OBJECT(document), (GWeakNotify)weak_notify, &count);
/* Ask twice for the Document */
WebKitDOMDocument* document2 = webkit_web_view_get_dom_document(view);
g_assert(document2);
g_object_weak_ref(G_OBJECT(document2), (GWeakNotify)weak_notify, &count);
head = webkit_dom_document_get_head(document);
g_assert(head);
g_object_weak_ref(G_OBJECT(head), (GWeakNotify)weak_notify, &count);
body = webkit_dom_document_get_body(document);
g_assert(body);
g_object_weak_ref(G_OBJECT(body), (GWeakNotify)weak_notify, &count);
collection = webkit_dom_document_get_links(document);
g_assert(collection);
g_object_weak_ref(G_OBJECT(collection), (GWeakNotify)weak_notify, &count);
gtk_widget_destroy(fixture->webView);
fixture->webView = NULL;
g_assert_cmpuint(count, ==, 4);
g_object_unref(collection);
g_assert_cmpuint(count, ==, 5);
}
// adblock_before_load_cb is triggered when
static void
adblock_before_load_cb(WebKitDOMEventTarget *doc,
WebKitDOMEvent *event,
gpointer user_data)
{
WebKitDOMEventTarget *target = webkit_dom_event_get_target(event);
guint64 flags = 0;
gchar *uri = NULL;
if(WEBKIT_DOM_IS_HTML_LINK_ELEMENT(target)) {
WebKitDOMHTMLLinkElement *e = WEBKIT_DOM_HTML_LINK_ELEMENT(target);
gboolean isCSS = 0;
isCSS |= g_strcmp0(
webkit_dom_html_link_element_get_rel(e),
"stylesheet") == 0;
isCSS |= g_strcmp0(
webkit_dom_html_link_element_get_type_attr(e),
"text/css") == 0;
if(!isCSS) {
return;
}
uri = webkit_dom_html_link_element_get_href(e);
flags |= ADBLOCK_STYLE_SHEET;
} else if(WEBKIT_DOM_IS_HTML_OBJECT_ELEMENT(target)) {
WebKitDOMHTMLObjectElement *e = WEBKIT_DOM_HTML_OBJECT_ELEMENT(target);
uri = webkit_dom_html_object_element_get_data(e);
flags |= ADBLOCK_OBJECT;
} else if(WEBKIT_DOM_IS_HTML_EMBED_ELEMENT(target)) {
WebKitDOMHTMLEmbedElement *e = WEBKIT_DOM_HTML_EMBED_ELEMENT(target);
uri = webkit_dom_html_embed_element_get_src(e);
flags |= ADBLOCK_OBJECT;
} else if(WEBKIT_DOM_IS_HTML_IMAGE_ELEMENT(target)) {
WebKitDOMHTMLImageElement *e = WEBKIT_DOM_HTML_IMAGE_ELEMENT(target);
uri = webkit_dom_html_image_element_get_src(e);
flags |= ADBLOCK_IMAGE;
} else if(WEBKIT_DOM_IS_HTML_SCRIPT_ELEMENT(target)) {
WebKitDOMHTMLScriptElement *e = WEBKIT_DOM_HTML_SCRIPT_ELEMENT(target);
uri = webkit_dom_html_script_element_get_src(e);
flags |= ADBLOCK_SCRIPT;
} else if(WEBKIT_DOM_IS_HTML_IFRAME_ELEMENT(target)) {
WebKitDOMHTMLIFrameElement *e = WEBKIT_DOM_HTML_IFRAME_ELEMENT(target);
uri = webkit_dom_html_iframe_element_get_src(e);
flags |= ADBLOCK_SUBDOCUMENT;
if(uri_is_blocked(uri, flags, user_data)) {
webkit_dom_event_prevent_default(event);
} else {
frame_document_loaded(
webkit_dom_html_iframe_element_get_content_document(e),
user_data);
}
g_free(uri);
return;
}
if(uri == NULL) {
return;
}
if(uri_is_blocked(uri, flags, user_data)) {
webkit_dom_event_prevent_default(event);
}
g_free(uri);
}
/**
* Find the first editable element and set the focus on it and enter input
* mode.
* Returns true if there was an editable element focused.
*/
gboolean dom_focus_input(Document *doc)
{
WebKitDOMNode *html, *node;
WebKitDOMDOMWindow *win;
WebKitDOMNodeList *list;
WebKitDOMXPathNSResolver *resolver;
WebKitDOMXPathResult* result;
Document *frame_doc;
guint i, len;
win = webkit_dom_document_get_default_view(doc);
list = webkit_dom_document_get_elements_by_tag_name(doc, "html");
if (!list) {
return false;
}
html = webkit_dom_node_list_item(list, 0);
g_object_unref(list);
resolver = webkit_dom_document_create_ns_resolver(doc, html);
if (!resolver) {
return false;
}
/* Use translate to match xpath expression case insensitive so that also
* intput filed of type="TEXT" are matched. */
result = webkit_dom_document_evaluate(
doc, "//input[not(@type) "
"or translate(@type,'ETX','etx')='text' "
"or translate(@type,'ADOPRSW','adoprsw')='password' "
"or translate(@type,'CLOR','clor')='color' "
"or translate(@type,'ADET','adet')='date' "
"or translate(@type,'ADEIMT','adeimt')='datetime' "
"or translate(@type,'ACDEILMOT','acdeilmot')='datetime-local' "
"or translate(@type,'AEILM','aeilm')='email' "
"or translate(@type,'HMNOT','hmnot')='month' "
"or translate(@type,'BEMNRU','bemnru')='number' "
"or translate(@type,'ACEHRS','acehrs')='search' "
"or translate(@type,'ELT','elt')='tel' "
"or translate(@type,'EIMT','eimt')='time' "
"or translate(@type,'LRU','lru')='url' "
"or translate(@type,'EKW','ekw')='week' "
"]|//textarea",
html, resolver, 5, NULL, NULL
);
if (!result) {
return false;
}
while ((node = webkit_dom_xpath_result_iterate_next(result, NULL))) {
if (element_is_visible(win, WEBKIT_DOM_ELEMENT(node))) {
vb_enter('i');
webkit_dom_element_focus(WEBKIT_DOM_ELEMENT(node));
return true;
}
}
/* Look for editable elements in frames too. */
list = webkit_dom_document_get_elements_by_tag_name(doc, "iframe");
len = webkit_dom_node_list_get_length(list);
for (i = 0; i < len; i++) {
node = webkit_dom_node_list_item(list, i);
frame_doc = webkit_dom_html_iframe_element_get_content_document(WEBKIT_DOM_HTML_IFRAME_ELEMENT(node));
/* Stop on first frame with focused element. */
if (dom_focus_input(frame_doc)) {
g_object_unref(list);
return true;
}
}
g_object_unref(list);
return false;
}
