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