surf.c

   1 /* See LICENSE file for copyright and license details.
   2  *
   3  * To understand surf, start reading main().
   4  */
   5 #include <signal.h>
   6 #include <X11/X.h>
   7 #include <X11/Xatom.h>
   8 #include <gtk/gtk.h>
   9 #include <gdk/gdkx.h>
  10 #include <gdk/gdk.h>
  11 #include <gdk/gdkkeysyms.h>
  12 #include <string.h>
  13 #include <sys/types.h>
  14 #include <sys/wait.h>
  15 #include <unistd.h>
  16 #include <stdlib.h>
  17 #include <stdio.h>
  18 #include <webkit/webkit.h>
  19 #include <glib/gstdio.h>
  20 #include <JavaScriptCore/JavaScript.h>
  21
  22 #define LENGTH(x)               (sizeof x / sizeof x[0])
  23 #define CLEANMASK(mask)         (mask & ~(GDK_MOD2_MASK))
  24
  25 typedef union Arg Arg;
  26 union Arg {
  27         gboolean b;
  28         gint i;
  29         const void *v;
  30 };
  31
  32 typedef struct Client {
  33         GtkWidget *win, *scroll, *vbox, *indicator;
  34         GtkWidget **items;
  35         WebKitWebView *view;
  36         WebKitDownload *download;
  37         char *title, *linkhover;
  38         const char *uri, *needle;
  39         gint progress;
  40         struct Client *next;
  41         gboolean zoomed;
  42 } Client;
  43
  44 typedef struct {
  45         char *label;
  46         void (*func)(Client *c, const Arg *arg);
  47         const Arg arg;
  48 } Item;
  49
  50 typedef struct {
  51         guint mod;
  52         guint keyval;
  53         void (*func)(Client *c, const Arg *arg);
  54         const Arg arg;
  55 } Key;
  56
  57 static Display *dpy;
  58 static Atom uriprop, findprop;
  59 static Client *clients = NULL;
  60 static GdkNativeWindow embed = 0;
  61 static gboolean showxid = FALSE;
  62 static int ignorexprop = 0;
  63 static char winid[64];
  64 static char *progname;
  65 static gboolean loadimage = 1, plugin = 1, script = 1;
  66
  67 static char *buildpath(const char *path);
  68 static void cleanup(void);
  69 static void clipboard(Client *c, const Arg *arg);
  70 static void context(WebKitWebView *v, GtkMenu *m, Client *c);
  71 static char *copystr(char **str, const char *src);
  72 static WebKitWebView *createwindow(WebKitWebView *v, WebKitWebFrame *f, Client *c);
  73 static gboolean decidedownload(WebKitWebView *v, WebKitWebFrame *f, WebKitNetworkRequest *r, gchar *m,  WebKitWebPolicyDecision *p, Client *c);
  74 static gboolean decidewindow(WebKitWebView *v, WebKitWebFrame *f, WebKitNetworkRequest *r, WebKitWebNavigationAction *n, WebKitWebPolicyDecision *p, Client *c);
  75 static void destroyclient(Client *c);
  76 static void destroywin(GtkWidget* w, Client *c);
  77 static void die(char *str);
  78 static void download(Client *c, const Arg *arg);
  79 static void drawindicator(Client *c);
  80 static gboolean exposeindicator(GtkWidget *w, GdkEventExpose *e, Client *c);
  81 static void find(Client *c, const Arg *arg);
  82 static const char *getatom(Client *c, Atom a);
  83 static const char *getcookies(SoupURI *uri);
  84 static char *geturi(Client *c);
  85 void gotheaders(SoupMessage *msg, gpointer user_data);
  86 static gboolean initdownload(WebKitWebView *v, WebKitDownload *o, Client *c);
  87 static void itemclick(GtkMenuItem *mi, Client *c);
  88 static gboolean keypress(GtkWidget *w, GdkEventKey *ev, Client *c);
  89 static void linkhover(WebKitWebView *v, const char* t, const char* l, Client *c);
  90 static void loadstatuschange(WebKitWebView *view, GParamSpec *pspec, Client *c);
  91 static void loaduri(Client *c, const Arg *arg);
  92 static void navigate(Client *c, const Arg *arg);
  93 static Client *newclient(void);
  94 static void newwindow(Client *c, const Arg *arg);
  95 static void newrequest(SoupSession *s, SoupMessage *msg, gpointer v);
  96 static void pasteuri(GtkClipboard *clipboard, const char *text, gpointer d);
  97 static void print(Client *c, const Arg *arg);
  98 static GdkFilterReturn processx(GdkXEvent *xevent, GdkEvent *event, gpointer d);
  99 static void progresschange(WebKitWebView *view, GParamSpec *pspec, Client *c);
 100 static void reload(Client *c, const Arg *arg);
 101 static void resize(GtkWidget *w, GtkAllocation *a, Client *c);
 102 static void scroll(Client *c, const Arg *arg);
 103 static void setatom(Client *c, Atom a, const char *v);
 104 static void setcookie(SoupCookie *c);
 105 static void setup(void);
 106 static void sigchld(int unused);
 107 static void source(Client *c, const Arg *arg);
 108 static void spawn(Client *c, const Arg *arg);
 109 static void stop(Client *c, const Arg *arg);
 110 static void titlechange(WebKitWebView *v, WebKitWebFrame* frame, const char* title, Client *c);
 111 static void update(Client *c);
 112 static void updatedownload(WebKitDownload *o, GParamSpec *pspec, Client *c);
 113 static void updatewinid(Client *c);
 114 static void usage(void);
 115 static void windowobjectcleared(GtkWidget *w, WebKitWebFrame *frame, JSContextRef js, JSObjectRef win, Client *c);
 116 static void zoom(Client *c, const Arg *arg);
 117
 118 /* configuration, allows nested code to access above variables */
 119 #include "config.h"
 120
 121 char *
 122 buildpath(const char *path) {
 123         char *apath, *p;
 124         FILE *f;
 125
 126         /* creating directory */
 127         if(path[0] == '/')
 128                 apath = g_strdup(path);
 129         else
 130                 apath = g_strconcat(g_get_home_dir(), "/", path, NULL);
 131         if((p = strrchr(apath, '/'))) {
 132                 *p = '\0';
 133                 g_mkdir_with_parents(apath, 0755);
 134                 *p = '/';
 135         }
 136         /* creating file (gives error when apath ends with "/") */
 137         if((f = g_fopen(apath, "a")))
 138                 fclose(f);
 139         return apath;
 140 }
 141
 142 void
 143 cleanup(void) {
 144         while(clients)
 145                 destroyclient(clients);
 146         g_free(cookiefile);
 147         g_free(dldir);
 148         g_free(scriptfile);
 149         g_free(stylefile);
 150 }
 151
 152 void
 153 clipboard(Client *c, const Arg *arg) {
 154         gboolean paste = *(gboolean *)arg;
 155
 156         if(paste)
 157                 gtk_clipboard_request_text(gtk_clipboard_get(GDK_SELECTION_PRIMARY), pasteuri, c);
 158         else
 159                 gtk_clipboard_set_text(gtk_clipboard_get(GDK_SELECTION_PRIMARY), c->linkhover ? c->linkhover : geturi(c), -1);
 160 }
 161
 162 void
 163 context(WebKitWebView *v, GtkMenu *m, Client *c) {
 164         int i;
 165         GtkContainer *parent;
 166
 167         gtk_widget_hide_all(GTK_WIDGET(m));
 168         gtk_widget_show(GTK_WIDGET(m));
 169         for(i = 0; i < LENGTH(items); i++) {
 170                 parent = GTK_CONTAINER(gtk_widget_get_parent(c->items[i]));
 171                 if(parent)
 172                         gtk_container_remove(parent, c->items[i]);
 173                 gtk_menu_shell_append(GTK_MENU_SHELL(m), c->items[i]);
 174                 gtk_widget_show(c->items[i]);
 175         }
 176 }
 177
 178 char *
 179 copystr(char **str, const char *src) {
 180         char *tmp;
 181         tmp = g_strdup(src);
 182
 183         if(str && *str) {
 184                 g_free(*str);
 185                 *str = tmp;
 186         }
 187         return tmp;
 188 }
 189
 190 WebKitWebView *
 191 createwindow(WebKitWebView  *v, WebKitWebFrame *f, Client *c) {
 192         Client *n = newclient();
 193         return n->view;
 194 }
 195
 196 gboolean
 197 decidedownload(WebKitWebView *v, WebKitWebFrame *f, WebKitNetworkRequest *r, gchar *m,  WebKitWebPolicyDecision *p, Client *c) {
 198         if(!webkit_web_view_can_show_mime_type(v, m)) {
 199                 webkit_web_policy_decision_download(p);
 200                 return TRUE;
 201         }
 202         return FALSE;
 203 }
 204
 205 gboolean
 206 decidewindow(WebKitWebView *view, WebKitWebFrame *f, WebKitNetworkRequest *r, WebKitWebNavigationAction *n, WebKitWebPolicyDecision *p, Client *c) {
 207         Arg arg;
 208
 209         if(webkit_web_navigation_action_get_reason(n) == WEBKIT_WEB_NAVIGATION_REASON_LINK_CLICKED) {
 210                 webkit_web_policy_decision_ignore(p);
 211                 arg.v = (void *)webkit_network_request_get_uri(r);
 212                 newwindow(NULL, &arg);
 213                 return TRUE;
 214         }
 215         return FALSE;
 216 }
 217
 218 void
 219 destroyclient(Client *c) {
 220         int i;
 221         Client *p;
 222
 223         gtk_widget_destroy(c->indicator);
 224         gtk_widget_destroy(GTK_WIDGET(c->view));
 225         gtk_widget_destroy(c->scroll);
 226         gtk_widget_destroy(c->vbox);
 227         gtk_widget_destroy(c->win);
 228         for(i = 0; i < LENGTH(items); i++)
 229                 gtk_widget_destroy(c->items[i]);
 230         free(c->items);
 231
 232         for(p = clients; p && p->next != c; p = p->next);
 233         if(p)
 234                 p->next = c->next;
 235         else
 236                 clients = c->next;
 237         free(c);
 238         if(clients == NULL)
 239                 gtk_main_quit();
 240 }
 241
 242 void
 243 destroywin(GtkWidget* w, Client *c) {
 244         destroyclient(c);
 245 }
 246
 247 void
 248 die(char *str) {
 249         fputs(str, stderr);
 250         exit(EXIT_FAILURE);
 251 }
 252
 253 void
 254 download(Client *c, const Arg *arg) {
 255         char *uri;
 256         WebKitNetworkRequest *r;
 257         WebKitDownload       *dl;
 258
 259         if(arg->v)
 260                 uri = (char *)arg->v;
 261         else
 262                 uri = c->linkhover ? c->linkhover : geturi(c);
 263         r = webkit_network_request_new(uri);
 264         dl = webkit_download_new(r);
 265         initdownload(c->view, dl, c);
 266         webkit_download_start(c->download);
 267 }
 268
 269 void
 270 drawindicator(Client *c) {
 271         gint width;
 272         const char *uri;
 273         GtkWidget *w;
 274         GdkGC *gc;
 275         GdkColor fg;
 276
 277         uri = getatom(c, uriprop);
 278         w = c->indicator;
 279         width = c->progress * w->allocation.width / 100;
 280         gc = gdk_gc_new(w->window);
 281         gdk_color_parse(strstr(uri, "https://") == uri ?
 282                         progress_trust : progress, &fg);
 283         gdk_gc_set_rgb_fg_color(gc, &fg);
 284         gdk_draw_rectangle(w->window,
 285                         w->style->bg_gc[GTK_WIDGET_STATE(w)],
 286                         TRUE, 0, 0, w->allocation.width, w->allocation.height);
 287         gdk_draw_rectangle(w->window, gc, TRUE, 0, 0, width,
 288                         w->allocation.height);
 289         g_object_unref(gc);
 290 }
 291
 292 gboolean
 293 exposeindicator(GtkWidget *w, GdkEventExpose *e, Client *c) {
 294         drawindicator(c);
 295         return TRUE;
 296 }
 297
 298 void
 299 find(Client *c, const Arg *arg) {
 300         const char *s;
 301
 302         s = getatom(c, findprop);
 303         gboolean forward = *(gboolean *)arg;
 304         webkit_web_view_search_text(c->view, s, FALSE, forward, TRUE);
 305 }
 306
 307 const char *
 308 getcookies(SoupURI *uri) {
 309         const char *c;
 310         SoupCookieJar *j = soup_cookie_jar_text_new(cookiefile, TRUE);
 311         c = soup_cookie_jar_get_cookies(j, uri, TRUE);
 312         g_object_unref(j);
 313         return c;
 314 }
 315
 316 const char *
 317 getatom(Client *c, Atom a) {
 318         static char buf[BUFSIZ];
 319         Atom adummy;
 320         int idummy;
 321         unsigned long ldummy;
 322         unsigned char *p = NULL;
 323
 324         XGetWindowProperty(dpy, GDK_WINDOW_XID(GTK_WIDGET(c->win)->window),
 325                         a, 0L, BUFSIZ, False, XA_STRING,
 326                         &adummy, &idummy, &ldummy, &ldummy, &p);
 327         if(p)
 328                 strncpy(buf, (char *)p, LENGTH(buf)-1);
 329         else
 330                 buf[0] = '\0';
 331         XFree(p);
 332         return buf;
 333 }
 334
 335 char *
 336 geturi(Client *c) {
 337         char *uri;
 338
 339         if(!(uri = (char *)webkit_web_view_get_uri(c->view)))
 340                 uri = "about:blank";
 341         return uri;
 342 }
 343
 344 void
 345 gotheaders(SoupMessage *msg, gpointer v) {
 346         SoupURI *uri;
 347         GSList *l, *p;
 348
 349         uri = soup_message_get_uri(msg);
 350         for(p = l = soup_cookies_from_response(msg); p;
 351                 p = g_slist_next(p))  {
 352                 setcookie((SoupCookie *)p->data);
 353         }
 354         soup_cookies_free(l);
 355 }
 356
 357 gboolean
 358 initdownload(WebKitWebView *view, WebKitDownload *o, Client *c) {
 359         const char *filename;
 360         char *uri;
 361         WebKitWebBackForwardList *h;
 362         WebKitWebHistoryItem *i;
 363
 364         c->download = o;
 365         filename = webkit_download_get_suggested_filename(o);
 366         if(!strcmp("", filename))
 367                 filename = "index.html";
 368         uri = g_strconcat("file://", dldir, "/", filename, NULL);
 369         webkit_download_set_destination_uri(c->download, uri);
 370         c->progress = 0;
 371         h = webkit_web_view_get_back_forward_list(c->view);
 372         i = webkit_web_history_item_new_with_data(webkit_download_get_uri(c->download), filename);
 373         webkit_web_back_forward_list_add_item(h, i);
 374         g_signal_connect(c->download, "notify::progress", G_CALLBACK(updatedownload), c);
 375         g_signal_connect(c->download, "notify::status", G_CALLBACK(updatedownload), c);
 376
 377         c->title = copystr(&c->title, filename);
 378         update(c);
 379         return TRUE;
 380 }
 381
 382 void
 383 itemclick(GtkMenuItem *mi, Client *c) {
 384         int i;
 385         const char *label;
 386
 387         label = gtk_menu_item_get_label(mi);
 388         for(i = 0; i < LENGTH(items); i++)
 389                 if(!strcmp(items[i].label, label))
 390                         items[i].func(c, &(items[i].arg));
 391 }
 392
 393 gboolean
 394 keypress(GtkWidget* w, GdkEventKey *ev, Client *c) {
 395         guint i;
 396         gboolean processed = FALSE;
 397
 398         updatewinid(c);
 399         for(i = 0; i < LENGTH(keys); i++) {
 400                 if(gdk_keyval_to_lower(ev->keyval) == keys[i].keyval
 401                                 && CLEANMASK(ev->state) == keys[i].mod
 402                                 && keys[i].func) {
 403                         keys[i].func(c, &(keys[i].arg));
 404                         processed = TRUE;
 405                 }
 406         }
 407         return processed;
 408 }
 409
 410 void
 411 linkhover(WebKitWebView *v, const char* t, const char* l, Client *c) {
 412         if(l)
 413                 c->linkhover = copystr(&c->linkhover, l);
 414         else if(c->linkhover) {
 415                 free(c->linkhover);
 416                 c->linkhover = NULL;
 417         }
 418         update(c);
 419 }
 420
 421 void
 422 loadstatuschange(WebKitWebView *view, GParamSpec *pspec, Client *c) {
 423         if(c->download)
 424                 stop(c, NULL);
 425         switch(webkit_web_view_get_load_status (c->view)) {
 426         case WEBKIT_LOAD_COMMITTED:
 427                 setatom(c, uriprop, geturi(c));
 428                 break;
 429         case WEBKIT_LOAD_FINISHED:
 430                 c->progress = 0;
 431                 update(c);
 432                 break;
 433         case WEBKIT_LOAD_PROVISIONAL:
 434         case WEBKIT_LOAD_FIRST_VISUALLY_NON_EMPTY_LAYOUT:
 435                 break;
 436         }
 437 }
 438
 439 void
 440 loaduri(Client *c, const Arg *arg) {
 441         char *u;
 442         const char *uri = (char *)arg->v;
 443         Arg a = { .b = FALSE };
 444
 445         if(strcmp(uri, "") == 0)
 446                 return;
 447         u = g_strrstr(uri, "://") ? g_strdup(uri)
 448                 : g_strdup_printf("http://%s", uri);
 449         /* prevents endless loop */
 450         if(c->uri && strcmp(u, c->uri) == 0) {
 451                 reload(c, &a);
 452         }
 453         else {
 454                 webkit_web_view_load_uri(c->view, u);
 455                 c->progress = 0;
 456                 c->title = copystr(&c->title, u);
 457                 g_free(u);
 458                 update(c);
 459         }
 460 }
 461
 462 void
 463 navigate(Client *c, const Arg *arg) {
 464         int steps = *(int *)arg;
 465         webkit_web_view_go_back_or_forward(c->view, steps);
 466 }
 467
 468 Client *
 469 newclient(void) {
 470         int i;
 471         Client *c;
 472         WebKitWebSettings *settings;
 473         GdkGeometry hints = { 1, 1 };
 474         char *uri, *ua;
 475
 476         if(!(c = calloc(1, sizeof(Client))))
 477                 die("Cannot malloc!\n");
 478         /* Window */
 479         if(embed) {
 480                 c->win = gtk_plug_new(embed);
 481         }
 482         else {
 483                 c->win = gtk_window_new(GTK_WINDOW_TOPLEVEL);
 484                 /* TA:  20091214:  Despite what the GNOME docs say, the ICCCM
 485                  * is always correct, so we should still call this function.
 486                  * But when doing so, we *must* differentiate between a
 487                  * WM_CLASS and a resource on the window.  By convention, the
 488                  * window class (WM_CLASS) is capped, while the resource is in
 489                  * lowercase.   Both these values come as a pair.
 490                  */
 491                 gtk_window_set_wmclass(GTK_WINDOW(c->win), "surf", "surf");
 492
 493                 /* TA:  20091214:  And set the role here as well -- so that
 494                  * sessions can pick this up.
 495                  */
 496                 gtk_window_set_role(GTK_WINDOW(c->win), "Surf");
 497         }
 498         gtk_window_set_default_size(GTK_WINDOW(c->win), 800, 600);
 499         g_signal_connect(G_OBJECT(c->win), "destroy", G_CALLBACK(destroywin), c);
 500         g_signal_connect(G_OBJECT(c->win), "key-press-event", G_CALLBACK(keypress), c);
 501         g_signal_connect(G_OBJECT(c->win), "size-allocate", G_CALLBACK(resize), c);
 502
 503         if(!(c->items = calloc(1, sizeof(GtkWidget *) * LENGTH(items))))
 504                 die("Cannot malloc!\n");
 505
 506         /* contextmenu */
 507         for(i = 0; i < LENGTH(items); i++) {
 508                 c->items[i] = gtk_menu_item_new_with_label(items[i].label);
 509                 g_signal_connect(G_OBJECT(c->items[i]), "activate",
 510                                 G_CALLBACK(itemclick), c);
 511         }
 512
 513         /* VBox */
 514         c->vbox = gtk_vbox_new(FALSE, 0);
 515
 516         /* Scrolled Window */
 517         c->scroll = gtk_scrolled_window_new(NULL, NULL);
 518         gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(c->scroll),
 519                         GTK_POLICY_NEVER, GTK_POLICY_NEVER);
 520
 521         /* Webview */
 522         c->view = WEBKIT_WEB_VIEW(webkit_web_view_new());
 523         g_signal_connect(G_OBJECT(c->view), "title-changed", G_CALLBACK(titlechange), c);
 524         g_signal_connect(G_OBJECT(c->view), "hovering-over-link", G_CALLBACK(linkhover), c);
 525         g_signal_connect(G_OBJECT(c->view), "create-web-view", G_CALLBACK(createwindow), c);
 526         g_signal_connect(G_OBJECT(c->view), "new-window-policy-decision-requested", G_CALLBACK(decidewindow), c);
 527         g_signal_connect(G_OBJECT(c->view), "mime-type-policy-decision-requested", G_CALLBACK(decidedownload), c);
 528         g_signal_connect(G_OBJECT(c->view), "download-requested", G_CALLBACK(initdownload), c);
 529         g_signal_connect(G_OBJECT(c->view), "window-object-cleared", G_CALLBACK(windowobjectcleared), c);
 530         g_signal_connect(G_OBJECT(c->view), "populate-popup", G_CALLBACK(context), c);
 531         g_signal_connect(G_OBJECT(c->view), "notify::load-status", G_CALLBACK(loadstatuschange), c);
 532         g_signal_connect(G_OBJECT(c->view), "notify::progress", G_CALLBACK(progresschange), c);
 533
 534         /* Indicator */
 535         c->indicator = gtk_drawing_area_new();
 536         gtk_widget_set_size_request(c->indicator, 0, 2);
 537         g_signal_connect (G_OBJECT (c->indicator), "expose_event",
 538                         G_CALLBACK (exposeindicator), c);
 539
 540         /* Arranging */
 541         gtk_container_add(GTK_CONTAINER(c->scroll), GTK_WIDGET(c->view));
 542         gtk_container_add(GTK_CONTAINER(c->win), c->vbox);
 543         gtk_container_add(GTK_CONTAINER(c->vbox), c->scroll);
 544         gtk_container_add(GTK_CONTAINER(c->vbox), c->indicator);
 545
 546         /* Setup */
 547         gtk_box_set_child_packing(GTK_BOX(c->vbox), c->indicator, FALSE, FALSE, 0, GTK_PACK_START);
 548         gtk_box_set_child_packing(GTK_BOX(c->vbox), c->scroll, TRUE, TRUE, 0, GTK_PACK_START);
 549         gtk_widget_grab_focus(GTK_WIDGET(c->view));
 550         gtk_widget_show(c->vbox);
 551         gtk_widget_show(c->indicator);
 552         gtk_widget_show(c->scroll);
 553         gtk_widget_show(GTK_WIDGET(c->view));
 554         gtk_widget_show(c->win);
 555         gtk_window_set_geometry_hints(GTK_WINDOW(c->win), NULL, &hints, GDK_HINT_MIN_SIZE);
 556         gdk_window_set_events(GTK_WIDGET(c->win)->window, GDK_ALL_EVENTS_MASK);
 557         gdk_window_add_filter(GTK_WIDGET(c->win)->window, processx, c);
 558         webkit_web_view_set_full_content_zoom(c->view, TRUE);
 559         settings = webkit_web_view_get_settings(c->view);
 560         if(!(ua = getenv("SURF_USERAGENT")))
 561                 ua = useragent;
 562         g_object_set(G_OBJECT(settings), "user-agent", ua, NULL);
 563         uri = g_strconcat("file://", stylefile, NULL);
 564         g_object_set(G_OBJECT(settings), "user-stylesheet-uri", uri, NULL);
 565         g_object_set(G_OBJECT(settings), "auto-load-images", loadimage, NULL);
 566         g_object_set(G_OBJECT(settings), "enable-plugins", plugin, NULL);
 567         g_object_set(G_OBJECT(settings), "enable-scripts", script, NULL);
 568         g_free(uri);
 569         setatom(c, findprop, "");
 570         setatom(c, uriprop, "");
 571
 572         c->download = NULL;
 573         c->title = NULL;
 574         c->next = clients;
 575         clients = c;
 576         if(showxid) {
 577                 gdk_display_sync(gtk_widget_get_display(c->win));
 578                 printf("%u\n", (guint)GDK_WINDOW_XID(GTK_WIDGET(c->win)->window));
 579                 fflush(NULL);
 580         }
 581         return c;
 582 }
 583
 584 void
 585 newrequest(SoupSession *s, SoupMessage *msg, gpointer v) {
 586         SoupMessageHeaders *h = msg->request_headers;
 587         SoupURI *uri;
 588         const char *c;
 589
 590         soup_message_headers_remove(h, "Cookie");
 591         uri = soup_message_get_uri(msg);
 592         if((c = getcookies(uri))) {
 593                 soup_message_headers_append(h, "Cookie", c);
 594         }
 595         g_signal_connect_after(G_OBJECT(msg), "got-headers", G_CALLBACK(gotheaders), NULL);
 596 }
 597
 598 void
 599 newwindow(Client *c, const Arg *arg) {
 600         guint i = 0;
 601         const char *cmd[10], *uri;
 602         const Arg a = { .v = (void *)cmd };
 603         char tmp[64];
 604
 605         cmd[i++] = progname;
 606         if(embed) {
 607                 cmd[i++] = "-e";
 608                 snprintf(tmp, LENGTH(tmp), "%u\n", (int)embed);
 609                 cmd[i++] = tmp;
 610         }
 611         if(!script)
 612                 cmd[i++] = "-s";
 613         if(!plugin)
 614                 cmd[i++] = "-p";
 615         if(!loadimage)
 616                 cmd[i++] = "-i";
 617         if(showxid)
 618                 cmd[i++] = "-x";
 619         cmd[i++] = "--";
 620         uri = arg->v ? (char *)arg->v : c->linkhover;
 621         if(uri)
 622                 cmd[i++] = uri;
 623         cmd[i++] = NULL;
 624         spawn(NULL, &a);
 625 }
 626
 627 void
 628 pasteuri(GtkClipboard *clipboard, const char *text, gpointer d) {
 629         Arg arg = {.v = text };
 630         if(text != NULL)
 631                 loaduri((Client *) d, &arg);
 632 }
 633
 634 void
 635 print(Client *c, const Arg *arg) {
 636         webkit_web_frame_print(webkit_web_view_get_main_frame(c->view));
 637 }
 638
 639 GdkFilterReturn
 640 processx(GdkXEvent *e, GdkEvent *event, gpointer d) {
 641         Client *c = (Client *)d;
 642         XPropertyEvent *ev;
 643         Arg arg;
 644
 645         if(((XEvent *)e)->type == PropertyNotify) {
 646                 ev = &((XEvent *)e)->xproperty;
 647                 if(ignorexprop)
 648                         ignorexprop--;
 649                 else if(ev->state == PropertyNewValue) {
 650                         if(ev->atom == uriprop) {
 651                                 arg.v = getatom(c, uriprop);
 652                                 loaduri(c, &arg);
 653                         }
 654                         else if(ev->atom == findprop) {
 655                                 arg.b = TRUE;
 656                                 find(c, &arg);
 657                         }
 658                         return GDK_FILTER_REMOVE;
 659                 }
 660         }
 661         return GDK_FILTER_CONTINUE;
 662 }
 663
 664 void
 665 progresschange(WebKitWebView *view, GParamSpec *pspec, Client *c) {
 666         c->progress = webkit_web_view_get_progress(c->view) * 100;
 667         update(c);
 668 }
 669
 670 void
 671 reload(Client *c, const Arg *arg) {
 672         gboolean nocache = *(gboolean *)arg;
 673         if(nocache)
 674                  webkit_web_view_reload_bypass_cache(c->view);
 675         else
 676                  webkit_web_view_reload(c->view);
 677 }
 678
 679 void
 680 resize(GtkWidget *w, GtkAllocation *a, Client *c) {
 681         double zoom;
 682
 683         if(c->zoomed)
 684                 return;
 685         zoom = webkit_web_view_get_zoom_level(c->view);
 686         if(a->width * a->height < 300 * 400 && zoom != 0.2)
 687                 webkit_web_view_set_zoom_level(c->view, 0.2);
 688         else if(zoom != 1.0)
 689                 webkit_web_view_set_zoom_level(c->view, 1.0);
 690 }
 691
 692 void
 693 scroll(Client *c, const Arg *arg) {
 694         gdouble v;
 695         GtkAdjustment *a;
 696
 697         a = gtk_scrolled_window_get_vadjustment(GTK_SCROLLED_WINDOW(c->scroll));
 698         v = gtk_adjustment_get_value(a);
 699         v += gtk_adjustment_get_step_increment(a) * arg->i;
 700         v = MAX(v, 0.0);
 701         v = MIN(v, gtk_adjustment_get_upper(a) - gtk_adjustment_get_page_size(a));
 702         gtk_adjustment_set_value(a, v);
 703 }
 704
 705 void
 706 setcookie(SoupCookie *c) {
 707         SoupDate *e;
 708         SoupCookieJar *j = soup_cookie_jar_text_new(cookiefile, FALSE);
 709         c = soup_cookie_copy(c);
 710         if(c->expires == NULL) {
 711                 e = soup_date_new_from_time_t(time(NULL) + sessiontime);
 712                 soup_cookie_set_expires(c, e);
 713         }
 714         soup_cookie_jar_add_cookie(j, c);
 715         g_object_unref(j);
 716 }
 717
 718 void
 719 setatom(Client *c, Atom a, const char *v) {
 720         XSync(dpy, False);
 721         ignorexprop++;
 722         XChangeProperty(dpy, GDK_WINDOW_XID(GTK_WIDGET(c->win)->window), a,
 723                         XA_STRING, 8, PropModeReplace, (unsigned char *)v,
 724                         strlen(v) + 1);
 725 }
 726
 727 void
 728 setup(void) {
 729         char *proxy;
 730         char *new_proxy;
 731         SoupURI *puri;
 732         SoupSession *s;
 733
 734         /* clean up any zombies immediately */
 735         sigchld(0);
 736         gtk_init(NULL, NULL);
 737         if (!g_thread_supported())
 738                 g_thread_init(NULL);
 739
 740         dpy = GDK_DISPLAY();
 741         s = webkit_get_default_session();
 742         uriprop = XInternAtom(dpy, "_SURF_URI", False);
 743         findprop = XInternAtom(dpy, "_SURF_FIND", False);
 744
 745         /* dirs and files */
 746         cookiefile = buildpath(cookiefile);
 747         dldir = buildpath(dldir);
 748         scriptfile = buildpath(scriptfile);
 749         stylefile = buildpath(stylefile);
 750
 751         /* request handler */
 752         s = webkit_get_default_session();
 753         soup_session_remove_feature_by_type(s, soup_cookie_get_type());
 754         soup_session_remove_feature_by_type(s, soup_cookie_jar_get_type());
 755         g_signal_connect_after(G_OBJECT(s), "request-started", G_CALLBACK(newrequest), NULL);
 756
 757
 758         /* proxy */
 759         if((proxy = getenv("http_proxy")) && strcmp(proxy, "")) {
 760                 new_proxy = g_strrstr(proxy, "http://") ? g_strdup(proxy) :
 761                         g_strdup_printf("http://%s", proxy);
 762
 763                 puri = soup_uri_new(new_proxy);
 764                 g_object_set(G_OBJECT(s), "proxy-uri", puri, NULL);
 765                 soup_uri_free(puri);
 766                 g_free(new_proxy);
 767         }
 768 }
 769
 770 void
 771 sigchld(int unused) {
 772         if(signal(SIGCHLD, sigchld) == SIG_ERR)
 773                 die("Can't install SIGCHLD handler");
 774         while(0 < waitpid(-1, NULL, WNOHANG));
 775 }
 776
 777 void
 778 source(Client *c, const Arg *arg) {
 779         Arg a = { .b = FALSE };
 780         gboolean s;
 781
 782         s = webkit_web_view_get_view_source_mode(c->view);
 783         webkit_web_view_set_view_source_mode(c->view, !s);
 784         reload(c, &a);
 785 }
 786
 787 void
 788 spawn(Client *c, const Arg *arg) {
 789         if(fork() == 0) {
 790                 if(dpy)
 791                         close(ConnectionNumber(dpy));
 792                 setsid();
 793                 execvp(((char **)arg->v)[0], (char **)arg->v);
 794                 fprintf(stderr, "surf: execvp %s", ((char **)arg->v)[0]);
 795                 perror(" failed");
 796                 exit(0);
 797         }
 798 }
 799
 800 void
 801 stop(Client *c, const Arg *arg) {
 802         if(c->download)
 803                 webkit_download_cancel(c->download);
 804         else
 805                 webkit_web_view_stop_loading(c->view);
 806         c->download = NULL;
 807 }
 808
 809 void
 810 titlechange(WebKitWebView *v, WebKitWebFrame *f, const char *t, Client *c) {
 811         c->title = copystr(&c->title, t);
 812         update(c);
 813 }
 814
 815 void
 816 update(Client *c) {
 817         char *t;
 818
 819         if(c->progress != 100)
 820                 t = g_strdup_printf("[%i%%] %s", c->progress, c->title);
 821         else if(c->linkhover)
 822                 t = g_strdup(c->linkhover);
 823         else
 824                 t = g_strdup(c->title);
 825         drawindicator(c);
 826         gtk_window_set_title(GTK_WINDOW(c->win), t);
 827         g_free(t);
 828 }
 829
 830 void
 831 updatedownload(WebKitDownload *o, GParamSpec *pspec, Client *c) {
 832         WebKitDownloadStatus status;
 833
 834         status = webkit_download_get_status(c->download);
 835         if(status == WEBKIT_DOWNLOAD_STATUS_STARTED || status == WEBKIT_DOWNLOAD_STATUS_CREATED) {
 836                 c->progress = (gint)(webkit_download_get_progress(c->download)*100);
 837         }
 838         else
 839                 stop(c, NULL);
 840         update(c);
 841 }
 842
 843 void
 844 updatewinid(Client *c) {
 845         snprintf(winid, LENGTH(winid), "%u",
 846                         (int)GDK_WINDOW_XID(GTK_WIDGET(c->win)->window));
 847 }
 848
 849 void
 850 usage(void) {
 851         fputs("surf - simple browser\n", stderr);
 852         die("usage: surf [-e xid] [-i] [-p] [-s] [-v] [-x] [uri]\n");
 853 }
 854
 855 void
 856 windowobjectcleared(GtkWidget *w, WebKitWebFrame *frame, JSContextRef js, JSObjectRef win, Client *c) {
 857         JSStringRef jsscript;
 858         char *script;
 859         JSValueRef exception = NULL;
 860         GError *error;
 861
 862         if(g_file_get_contents(scriptfile, &script, NULL, &error)) {
 863                 jsscript = JSStringCreateWithUTF8CString(script);
 864                 JSEvaluateScript(js, jsscript, JSContextGetGlobalObject(js), NULL, 0, &exception);
 865         }
 866 }
 867
 868 void
 869 zoom(Client *c, const Arg *arg) {
 870         c->zoomed = TRUE;
 871         if(arg->i < 0)          /* zoom out */
 872                 webkit_web_view_zoom_out(c->view);
 873         else if(arg->i > 0)     /* zoom in */
 874                 webkit_web_view_zoom_in(c->view);
 875         else {                  /* reset */
 876                 c->zoomed = FALSE;
 877                 webkit_web_view_set_zoom_level(c->view, 1.0);
 878         }
 879 }
 880
 881 int
 882 main(int argc, char *argv[]) {
 883         int i;
 884         Arg arg;
 885
 886         progname = argv[0];
 887         /* command line args */
 888         for(i = 1, arg.v = NULL; i < argc && argv[i][0] == '-' &&
 889                         argv[i][1] != '\0' && argv[i][2] == '\0'; i++) {
 890                 if(!strcmp(argv[i], "--")) {
 891                         i++;
 892                         break;
 893                 }
 894                 switch(argv[i][1]) {
 895                 case 'e':
 896                         if(++i < argc)
 897                                 embed = atoi(argv[i]);
 898                         else
 899                                 usage();
 900                         break;
 901                 case 'i':
 902                         loadimage = 0;
 903                         break;
 904                 case 'p':
 905                         plugin = 0;
 906                         break;
 907                 case 's':
 908                         script = 0;
 909                         break;
 910                 case 'x':
 911                         showxid = TRUE;
 912                         break;
 913                 case 'v':
 914                         die("surf-"VERSION", © 2009 surf engineers, see LICENSE for details\n");
 915                 default:
 916                         usage();
 917                 }
 918         }
 919         if(i < argc)
 920                 arg.v = argv[i];
 921         setup();
 922         newclient();
 923         if(arg.v)
 924                 loaduri(clients, &arg);
 925         gtk_main();
 926         cleanup();
 927         return EXIT_SUCCESS;
 928 }