we_adblock.c

   1 #include <stdio.h>
   2
   3 #include <glib.h>
   4 #include <webkit2/webkit-web-extension.h>
   5
   6
   7 static GSList *adblock_patterns = NULL;
   8
   9
  10 static void
  11 adblock_load(void)
  12 {
  13         GRegex *re = NULL;
  14         GError *err = NULL;
  15         GIOChannel *channel = NULL;
  16         gchar *path = NULL, *buf = NULL;
  17
  18         path = g_build_filename(g_get_user_config_dir(), __NAME__, "adblock.black",
  19                                 NULL);
  20         channel = g_io_channel_new_file(path, "r", &err);
  21         if (channel != NULL)
  22         {
  23                 while (g_io_channel_read_line(channel, &buf, NULL, NULL, NULL)
  24                        == G_IO_STATUS_NORMAL)
  25                 {
  26                         g_strstrip(buf);
  27                         if (buf[0] != '#')
  28                         {
  29                                 re = g_regex_new(buf,
  30                                                  G_REGEX_CASELESS | G_REGEX_OPTIMIZE,
  31                                                  G_REGEX_MATCH_PARTIAL, &err);
  32                                 if (err != NULL)
  33                                 {
  34                                         fprintf(stderr, __NAME__": Could not compile regex: %s\n", buf);
  35                                         g_error_free(err);
  36                                         err = NULL;
  37                                 }
  38                                 else
  39                                         adblock_patterns = g_slist_append(adblock_patterns, re);
  40                         }
  41                         g_free(buf);
  42                 }
  43                 g_io_channel_shutdown(channel, FALSE, NULL);
  44         }
  45         g_free(path);
  46 }
  47
  48 static gboolean
  49 web_page_send_request(WebKitWebPage *web_page, WebKitURIRequest *request,
  50                       WebKitURIResponse *redirected_response, gpointer user_data)
  51 {
  52         GSList *it = adblock_patterns;
  53         const gchar *uri;
  54
  55         uri = webkit_uri_request_get_uri(request);
  56
  57         while (it)
  58         {
  59                 if (g_regex_match((GRegex *)(it->data), uri, 0, NULL))
  60                         return TRUE;
  61                 it = g_slist_next(it);
  62         }
  63
  64         return FALSE;
  65 }
  66
  67 static void
  68 web_page_created_callback(WebKitWebExtension *extension, WebKitWebPage *web_page,
  69                           gpointer user_data)
  70 {
  71         g_signal_connect_object(web_page, "send-request",
  72                                 G_CALLBACK(web_page_send_request), NULL, 0);
  73 }
  74
  75 G_MODULE_EXPORT void
  76 webkit_web_extension_initialize(WebKitWebExtension *extension)
  77 {
  78         adblock_load();
  79         g_signal_connect(extension, "page-created",
  80                          G_CALLBACK(web_page_created_callback), NULL);
  81 }