]> git.armaanb.net Git - chorizo.git/blobdiff - browser.c
Implement a simple certificate trust store
[chorizo.git] / browser.c
index b34a0c9543d86447737c2d987635e1402a03f86b..08c2b3e2d8fa569f0c2e7028bdbbd5699b2f2559 100644 (file)
--- a/browser.c
+++ b/browser.c
@@ -22,11 +22,12 @@ static void changed_download_progress(GObject *, GParamSpec *, gpointer);
 static void changed_load_progress(GObject *, GParamSpec *, gpointer);
 static void changed_title(GObject *, GParamSpec *, gpointer);
 static void changed_uri(GObject *, GParamSpec *, gpointer);
+static gboolean crashed_web_view(WebKitWebView *, gpointer);
+static gboolean crashed_web_view_reload(gpointer);
 static gboolean decide_policy(WebKitWebView *, WebKitPolicyDecision *,
                               WebKitPolicyDecisionType, gpointer);
 static gboolean download_handle(WebKitDownload *, gchar *, gpointer);
 static void download_handle_start(WebKitWebView *, WebKitDownload *, gpointer);
-static gboolean download_reset_indicator(gpointer);
 static void downloadmanager_cancel(GtkToolButton *, gpointer data);
 static void downloadmanager_setup(void);
 static gchar *ensure_uri_scheme(const gchar *);
@@ -40,6 +41,7 @@ static gboolean keywords_try_search(WebKitWebView *, const gchar *);
 static gboolean remote_msg(GIOChannel *, GIOCondition, gpointer);
 static void search(gpointer, gint);
 static Window tabbed_launch(void);
+static void trust_user_certs(WebKitWebContext *);
 static void usage(void);
 
 
@@ -49,7 +51,6 @@ struct Client
        GtkWidget *location;
        GtkWidget *progress;
        GtkWidget *scroll;
-       GtkWidget *status;
        GtkWidget *top_box;
        GtkWidget *vbox;
        GtkWidget *web_view;
@@ -69,17 +70,18 @@ static gint clients = 0;
 static gboolean cooperative_alone = TRUE;
 static gboolean cooperative_instances = TRUE;
 static int cooperative_pipe_fp = 0;
+static int crash_autoreload_delay = 2;
 static gchar *download_dir = "/tmp";
-static gint downloads_indicated = 0;
 static Window embed = 0;
 static gchar *fifo_suffix = "main";
 static gdouble global_zoom = 1.0;
 static gchar *home_uri = "about:blank";
+static gboolean initial_wc_setup_done = FALSE;
 static GHashTable *keywords = NULL;
-static gboolean language_is_set = FALSE;
 static gchar *search_text = NULL;
 static gboolean tabbed_automagic = TRUE;
 static gchar *user_agent = NULL;
+static gchar *web_extensions_dir = NULL;
 
 
 void
@@ -167,8 +169,6 @@ client_new(const gchar *uri)
                         G_CALLBACK(client_destroy_request), c);
        g_signal_connect(G_OBJECT(c->web_view), "decide-policy",
                         G_CALLBACK(decide_policy), NULL);
-       g_signal_connect(G_OBJECT(wc), "download-started",
-                        G_CALLBACK(download_handle_start), c);
        g_signal_connect(G_OBJECT(c->web_view), "key-press-event",
                         G_CALLBACK(key_web_view), c);
        g_signal_connect(G_OBJECT(c->web_view), "button-press-event",
@@ -177,11 +177,20 @@ client_new(const gchar *uri)
                         G_CALLBACK(key_web_view), c);
        g_signal_connect(G_OBJECT(c->web_view), "mouse-target-changed",
                         G_CALLBACK(hover_web_view), c);
+       g_signal_connect(G_OBJECT(c->web_view), "web-process-crashed",
+                        G_CALLBACK(crashed_web_view), c);
 
-       if (!language_is_set && accepted_language[0] != NULL)
+       if (!initial_wc_setup_done)
        {
-               webkit_web_context_set_preferred_languages(wc, accepted_language);
-               language_is_set = TRUE;
+               if (accepted_language[0] != NULL)
+                       webkit_web_context_set_preferred_languages(wc, accepted_language);
+
+               g_signal_connect(G_OBJECT(wc), "download-started",
+                                G_CALLBACK(download_handle_start), NULL);
+
+               trust_user_certs(wc);
+
+               initial_wc_setup_done = TRUE;
        }
 
        if (user_agent != NULL)
@@ -202,13 +211,8 @@ client_new(const gchar *uri)
        gtk_level_bar_set_value(GTK_LEVEL_BAR(c->progress), 0);
        gtk_widget_set_size_request(c->progress, 100, -1);
 
-       c->status = gtk_level_bar_new();
-       gtk_level_bar_set_value(GTK_LEVEL_BAR(c->status), 0);
-       gtk_widget_set_size_request(c->status, 20, -1);
-
        c->top_box = gtk_box_new(GTK_ORIENTATION_HORIZONTAL, 0);
-       gtk_box_pack_start(GTK_BOX(c->top_box), c->status, FALSE, FALSE, 2);
-       gtk_box_pack_start(GTK_BOX(c->top_box), c->location, TRUE, TRUE, 0);
+       gtk_box_pack_start(GTK_BOX(c->top_box), c->location, TRUE, TRUE, 2);
        gtk_box_pack_start(GTK_BOX(c->top_box), c->progress, FALSE, FALSE, 2);
 
        c->vbox = gtk_box_new(GTK_ORIENTATION_VERTICAL, 0);
@@ -339,6 +343,29 @@ changed_uri(GObject *obj, GParamSpec *pspec, gpointer data)
        gtk_entry_set_text(GTK_ENTRY(c->location), (t == NULL ? __NAME__ : t));
 }
 
+gboolean
+crashed_web_view(WebKitWebView *web_view, gpointer data)
+{
+       fprintf(stderr, __NAME__": WebView crashed!\n");
+       if (crash_autoreload_delay >= 1)
+       {
+               fprintf(stderr, __NAME__": Reloading WebView in %d seconds.\n",
+                       crash_autoreload_delay);
+               g_timeout_add_seconds(crash_autoreload_delay, crashed_web_view_reload,
+                                     web_view);
+       }
+
+       return TRUE;
+}
+
+gboolean
+crashed_web_view_reload(gpointer data)
+{
+       webkit_web_view_reload_bypass_cache(WEBKIT_WEB_VIEW(data));
+
+       return G_SOURCE_REMOVE;
+}
+
 gboolean
 decide_policy(WebKitWebView *web_view, WebKitPolicyDecision *decision,
               WebKitPolicyDecisionType type, gpointer data)
@@ -372,7 +399,6 @@ download_handle_start(WebKitWebView *web_view, WebKitDownload *download,
 gboolean
 download_handle(WebKitDownload *download, gchar *suggested_filename, gpointer data)
 {
-       struct Client *c = (struct Client *)data;
        gchar *path, *path2 = NULL, *uri;
        GtkToolItem *tb;
        int suffix = 1;
@@ -398,15 +424,11 @@ download_handle(WebKitDownload *download, gchar *suggested_filename, gpointer da
                webkit_download_set_destination(download, uri);
                g_free(uri);
 
-               gtk_level_bar_set_value(GTK_LEVEL_BAR(c->status), 1);
-               downloads_indicated++;
-               g_timeout_add(500, download_reset_indicator, c);
-
                tb = gtk_tool_button_new(NULL, NULL);
                gtk_tool_button_set_icon_name(GTK_TOOL_BUTTON(tb), "gtk-delete");
                gtk_tool_button_set_label(GTK_TOOL_BUTTON(tb), suggested_filename);
                gtk_toolbar_insert(GTK_TOOLBAR(dm.toolbar), tb, 0);
-               gtk_widget_show_all(dm.toolbar);
+               gtk_widget_show_all(dm.win);
 
                g_signal_connect(G_OBJECT(download), "notify::estimated-progress",
                                 G_CALLBACK(changed_download_progress), tb);
@@ -423,18 +445,6 @@ download_handle(WebKitDownload *download, gchar *suggested_filename, gpointer da
        return FALSE;
 }
 
-gboolean
-download_reset_indicator(gpointer data)
-{
-       struct Client *c = (struct Client *)data;
-
-       downloads_indicated--;
-       if (downloads_indicated == 0)
-               gtk_level_bar_set_value(GTK_LEVEL_BAR(c->status), 0);
-
-       return FALSE;
-}
-
 void
 downloadmanager_cancel(GtkToolButton *tb, gpointer data)
 {
@@ -500,6 +510,10 @@ grab_environment_configuration(void)
        if (e != NULL)
                accepted_language[0] = g_strdup(e);
 
+       e = g_getenv(__NAME_UPPERCASE__"_CRASH_AUTORELOAD_DELAY");
+       if (e != NULL)
+               crash_autoreload_delay = atoi(e);
+
        e = g_getenv(__NAME_UPPERCASE__"_DOWNLOAD_DIR");
        if (e != NULL)
                download_dir = g_strdup(e);
@@ -516,6 +530,13 @@ grab_environment_configuration(void)
        if (e != NULL)
                user_agent = g_strdup(e);
 
+       e = g_getenv(__NAME_UPPERCASE__"_WEB_EXTENSIONS_DIR");
+       if (e != NULL)
+               web_extensions_dir = g_strdup(e);
+       else
+               web_extensions_dir = g_build_filename(g_get_user_data_dir(), __NAME__,
+                                                     "web_extensions", NULL);
+
        e = g_getenv(__NAME_UPPERCASE__"_ZOOM");
        if (e != NULL)
                global_zoom = atof(e);
@@ -575,6 +596,7 @@ key_location(GtkWidget *widget, GdkEvent *event, gpointer data)
        struct Client *c = (struct Client *)data;
        const gchar *t;
        gchar *f;
+       WebKitWebContext *wc = webkit_web_view_get_context(WEBKIT_WEB_VIEW(c->web_view));
 
        if (event->type == GDK_KEY_PRESS)
        {
@@ -596,6 +618,9 @@ key_location(GtkWidget *widget, GdkEvent *event, gpointer data)
                                        gtk_entry_set_text(GTK_ENTRY(c->location), "/");
                                        gtk_editable_set_position(GTK_EDITABLE(c->location), -1);
                                        return TRUE;
+                               case GDK_KEY_c:  /* reload trusted certs (left hand) */
+                                       trust_user_certs(wc);
+                                       return TRUE;
                        }
                }
                else
@@ -635,9 +660,10 @@ gboolean
 key_web_view(GtkWidget *widget, GdkEvent *event, gpointer data)
 {
        struct Client *c = (struct Client *)data;
+       gdouble dx, dy;
        gchar *f;
        gfloat z;
-       gboolean b;
+       WebKitWebContext *wc = webkit_web_view_get_context(WEBKIT_WEB_VIEW(c->web_view));
 
        if (event->type == GDK_KEY_PRESS)
        {
@@ -662,16 +688,6 @@ key_web_view(GtkWidget *widget, GdkEvent *event, gpointer data)
                                        webkit_web_view_reload_bypass_cache(WEBKIT_WEB_VIEW(
                                                                            c->web_view));
                                        return TRUE;
-#if 0
-                               case GDK_KEY_s:  /* toggle source view (left hand) */
-                                       b = webkit_web_view_get_view_source_mode(WEBKIT_WEB_VIEW(
-                                                                                c->web_view));
-                                       b = !b;
-                                       webkit_web_view_set_view_source_mode(WEBKIT_WEB_VIEW(
-                                                                            c->web_view), b);
-                                       webkit_web_view_reload(WEBKIT_WEB_VIEW(c->web_view));
-                                       return TRUE;
-#endif
                                case GDK_KEY_d:  /* download manager (left hand) */
                                        gtk_widget_show_all(dm.win);
                                        return TRUE;
@@ -690,6 +706,9 @@ key_web_view(GtkWidget *widget, GdkEvent *event, gpointer data)
                                        gtk_entry_set_text(GTK_ENTRY(c->location), "/");
                                        gtk_editable_set_position(GTK_EDITABLE(c->location), -1);
                                        return TRUE;
+                               case GDK_KEY_c:  /* reload trusted certs (left hand) */
+                                       trust_user_certs(wc);
+                                       return TRUE;
                        }
                }
                else if (((GdkEventKey *)event)->keyval == GDK_KEY_Escape)
@@ -722,21 +741,12 @@ key_web_view(GtkWidget *widget, GdkEvent *event, gpointer data)
                if (((GdkEventScroll *)event)->state & GDK_MOD1_MASK ||
                    ((GdkEventScroll *)event)->state & GDK_CONTROL_MASK)
                {
-                       switch (((GdkEventScroll *)event)->direction)
-                       {
-                               case GDK_SCROLL_UP:
-                                       z = webkit_web_view_get_zoom_level(WEBKIT_WEB_VIEW(c->web_view));
-                                       z += 0.1;
-                                       webkit_web_view_set_zoom_level(WEBKIT_WEB_VIEW(c->web_view), z);
-                                       return TRUE;
-                               case GDK_SCROLL_DOWN:
-                                       z = webkit_web_view_get_zoom_level(WEBKIT_WEB_VIEW(c->web_view));
-                                       z -= 0.1;
-                                       webkit_web_view_set_zoom_level(WEBKIT_WEB_VIEW(c->web_view), z);
-                                       return TRUE;
-                               default:
-                                       break;
-                       }
+                       gdk_event_get_scroll_deltas(event, &dx, &dy);
+                       z = webkit_web_view_get_zoom_level(WEBKIT_WEB_VIEW(c->web_view));
+                       z += -dy * 0.1;
+                       z = dx != 0 ? global_zoom : z;
+                       webkit_web_view_set_zoom_level(WEBKIT_WEB_VIEW(c->web_view), z);
+                       return TRUE;
                }
        }
 
@@ -883,6 +893,33 @@ tabbed_launch(void)
        return plug_into;
 }
 
+void
+trust_user_certs(WebKitWebContext *wc)
+{
+       GTlsCertificate *cert;
+       const gchar *basedir, *file, *absfile;
+       GDir *dir;
+
+       basedir = g_build_filename(g_get_user_config_dir(), __NAME__, "certs", NULL);
+       dir = g_dir_open(basedir, 0, NULL);
+       if (dir != NULL)
+       {
+               file = g_dir_read_name(dir);
+               while (file != NULL)
+               {
+                       absfile = g_build_filename(g_get_user_config_dir(), __NAME__, "certs",
+                                                  file, NULL);
+                       cert = g_tls_certificate_new_from_file(absfile, NULL);
+                       if (cert == NULL)
+                               fprintf(stderr, __NAME__": Could not load trusted cert '%s'\n", file);
+                       else
+                               webkit_web_context_allow_tls_certificate_for_host(wc, cert, file);
+                       file = g_dir_read_name(dir);
+               }
+               g_dir_close(dir);
+       }
+}
+
 void
 usage(void)
 {
@@ -926,6 +963,10 @@ main(int argc, char **argv)
        if (tabbed_automagic && !(cooperative_instances && !cooperative_alone))
                embed = tabbed_launch();
 
+       if (!cooperative_instances || cooperative_alone)
+               webkit_web_context_set_web_extensions_directory(webkit_web_context_get_default(),
+                                                               web_extensions_dir);
+
        if (optind >= argc)
                client_new(home_uri);
        else