Communicate through a Unix socket instead of a pipe pair

1 files changed, 33 insertions, 27 deletions

diff --git a/webext-surf.c b/webext-surf.c
index ec471d1..775c1e8 100644
--- a/webext-surf.c
+++ b/webext-surf.c
@@ -1,6 +1,8 @@
+#include <sys/socket.h>
 #include <sys/stat.h>
 #include <fcntl.h>
 #include <limits.h>
+#include <stdio.h>
 #include <stdlib.h>
 
 #include <gio/gio.h>
@@ -18,7 +20,7 @@ typedef struct Page {
 	struct Page *next;
 } Page;
 
-static int pipein, pipeout;
+static int sock;
 static Page *pages;
 
 Page *
@@ -47,60 +49,64 @@ msgsurf(Page *p, const char *s)
 	size_t sln = strlen(s);
 	int ret;
 
-	if ((ret = snprintf(msg, sizeof(msg), "%c%c%s",
-	                    2 + sln, p ? p->id : 0, s))
+	if ((ret = snprintf(msg, sizeof(msg), "%c%s", p ? p->id : 0, s))
 	    >= sizeof(msg)) {
-		fprintf(stderr, "webext: message too long: %d\n", ret);
+		fprintf(stderr, "webext: msg: message too long: %d\n", ret);
 		return;
 	}
 
-	if (pipeout && write(pipeout, msg, sizeof(msg)) < 0)
-		fprintf(stderr, "webext: error sending: %.*s\n", ret-2, msg+2);
+	if (send(sock, msg, ret, 0) < 0)
+		fprintf(stderr, "webext: error sending: %s\n", msg+1);
 }
 
 static gboolean
-readpipe(GIOChannel *s, GIOCondition c, gpointer unused)
+readsock(GIOChannel *s, GIOCondition c, gpointer unused)
 {
-	static char msg[MSGBUFSZ], msgsz;
+	static char msg[MSGBUFSZ];
 	WebKitDOMDOMWindow *view;
 	GError *gerr = NULL;
+	gsize msgsz;
 	glong wh, ww;
 	Page *p;
 
-	if (g_io_channel_read_chars(s, msg, LENGTH(msg), NULL, &gerr) !=
+	if (g_io_channel_read_chars(s, msg, LENGTH(msg), &msgsz, &gerr) !=
 	    G_IO_STATUS_NORMAL) {
-		fprintf(stderr, "webext: error reading pipe: %s\n",
-		        gerr->message);
-		g_error_free(gerr);
+		if (gerr) {
+			fprintf(stderr, "webext: error reading socket: %s\n",
+			        gerr->message);
+			g_error_free(gerr);
+		}
 		return TRUE;
 	}
-	if ((msgsz = msg[0]) < 3) {
-		fprintf(stderr, "webext: message too short: %d\n", msgsz);
+
+	if (msgsz < 2) {
+		fprintf(stderr, "webext: readsock: message too short: %d\n",
+		        msgsz);
 		return TRUE;
 	}
 
 	for (p = pages; p; p = p->next) {
-		if (p->id == msg[1])
+		if (p->id == msg[0])
 			break;
 	}
 	if (!p || !(view = webkit_dom_document_get_default_view(
 	            webkit_web_page_get_dom_document(p->webpage))))
 		return TRUE;
 
-	switch (msg[2]) {
+	switch (msg[1]) {
 	case 'h':
-		if (msgsz != 4)
+		if (msgsz != 3)
 			return TRUE;
 		ww = webkit_dom_dom_window_get_inner_width(view);
 		webkit_dom_dom_window_scroll_by(view,
-		                                (ww / 100) * msg[3], 0);
+		                                (ww / 100) * msg[2], 0);
 		break;
 	case 'v':
-		if (msgsz != 4)
+		if (msgsz != 3)
 			return TRUE;
 		wh = webkit_dom_dom_window_get_inner_height(view);
 		webkit_dom_dom_window_scroll_by(view,
-		                                0, (wh / 100) * msg[3]);
+		                                0, (wh / 100) * msg[2]);
 		break;
 	}
 
@@ -116,15 +122,15 @@ webpagecreated(WebKitWebExtension *e, WebKitWebPage *wp, gpointer unused)
 G_MODULE_EXPORT void
 webkit_web_extension_initialize_with_user_data(WebKitWebExtension *e, GVariant *gv)
 {
-	GIOChannel *gchanpipe;
+	GIOChannel *gchansock;
 
 	g_signal_connect(e, "page-created", G_CALLBACK(webpagecreated), NULL);
 
-	g_variant_get(gv, "(ii)", &pipein, &pipeout);
-	msgsurf(NULL, "i");
+	g_variant_get(gv, "i", &sock);
 
-	gchanpipe = g_io_channel_unix_new(pipein);
-	g_io_channel_set_encoding(gchanpipe, NULL, NULL);
-	g_io_channel_set_close_on_unref(gchanpipe, TRUE);
-	g_io_add_watch(gchanpipe, G_IO_IN, readpipe, NULL);
+	gchansock = g_io_channel_unix_new(sock);
+	g_io_channel_set_encoding(gchansock, NULL, NULL);
+	g_io_channel_set_flags(gchansock, G_IO_FLAG_NONBLOCK, NULL);
+	g_io_channel_set_close_on_unref(gchansock, TRUE);
+	g_io_add_watch(gchansock, G_IO_IN, readsock, NULL);
 }