webext-surf.c (3294B)
1 #include <sys/socket.h> 2 #include <sys/stat.h> 3 #include <fcntl.h> 4 #include <inttypes.h> 5 #include <limits.h> 6 #include <stdio.h> 7 #include <stdlib.h> 8 9 #include <gio/gio.h> 10 #include <webkit2/webkit-web-extension.h> 11 #include <webkitdom/webkitdom.h> 12 #include <webkitdom/WebKitDOMDOMWindowUnstable.h> 13 14 #include "common.h" 15 16 #define LENGTH(x) (sizeof(x) / sizeof(x[0])) 17 18 static WebKitWebExtension *webext; 19 static int sock; 20 21 static void 22 msgsurf(guint64 pageid, const char *s) 23 { 24 static char msg[MSGBUFSZ]; 25 size_t sln = strlen(s); 26 int ret; 27 28 if ((ret = snprintf(msg, sizeof(msg), "%c%s", pageid, s)) 29 >= sizeof(msg)) { 30 fprintf(stderr, "webext: msg: message too long: %d\n", ret); 31 return; 32 } 33 34 if (send(sock, msg, ret, 0) < 0) 35 fprintf(stderr, "webext: error sending: %s\n", msg+1); 36 } 37 38 static gboolean 39 readsock(GIOChannel *s, GIOCondition c, gpointer unused) 40 { 41 static char js[48], msg[MSGBUFSZ]; 42 WebKitWebPage *page; 43 JSCContext *jsc; 44 GError *gerr = NULL; 45 gsize msgsz; 46 47 if (g_io_channel_read_chars(s, msg, sizeof(msg), &msgsz, &gerr) != 48 G_IO_STATUS_NORMAL) { 49 if (gerr) { 50 fprintf(stderr, "webext: error reading socket: %s\n", 51 gerr->message); 52 g_error_free(gerr); 53 } 54 return TRUE; 55 } 56 57 if (msgsz < 2) { 58 fprintf(stderr, "webext: readsock: message too short: %d\n", 59 msgsz); 60 return TRUE; 61 } 62 63 if (!(page = webkit_web_extension_get_page(webext, msg[0]))) 64 return TRUE; 65 66 jsc = webkit_frame_get_js_context(webkit_web_page_get_main_frame(page)); 67 68 switch (msg[1]) { 69 case 'h': 70 if (msgsz != 3) 71 return TRUE; 72 snprintf(js, sizeof(js), 73 "window.scrollBy(window.innerWidth/100*%d,0);", 74 msg[2]); 75 jsc_context_evaluate(jsc, js, -1); 76 break; 77 case 'v': 78 if (msgsz != 3) 79 return TRUE; 80 snprintf(js, sizeof(js), 81 "window.scrollBy(0,window.innerHeight/100*%d);", 82 msg[2]); 83 jsc_context_evaluate(jsc, js, -1); 84 break; 85 } 86 87 return TRUE; 88 } 89 90 static void 91 pageusermessagereply(GObject *o, GAsyncResult *r, gpointer page) 92 { 93 WebKitUserMessage *m; 94 GUnixFDList *gfd; 95 GIOChannel *gchansock; 96 const char *name; 97 int nfd; 98 99 m = webkit_web_page_send_message_to_view_finish(page, r, NULL); 100 name = webkit_user_message_get_name(m); 101 if (strcmp(name, "surf-pipe") != 0) { 102 fprintf(stderr, "webext-surf: Unknown User Reply: %s\n", name); 103 return; 104 } 105 106 gfd = webkit_user_message_get_fd_list(m); 107 if ((nfd = g_unix_fd_list_get_length(gfd)) != 1) { 108 fprintf(stderr, "webext-surf: Too many file-descriptors: %d\n", nfd); 109 return; 110 } 111 112 sock = g_unix_fd_list_get(gfd, 0, NULL); 113 114 gchansock = g_io_channel_unix_new(sock); 115 g_io_channel_set_encoding(gchansock, NULL, NULL); 116 g_io_channel_set_flags(gchansock, g_io_channel_get_flags(gchansock) 117 | G_IO_FLAG_NONBLOCK, NULL); 118 g_io_channel_set_close_on_unref(gchansock, TRUE); 119 g_io_add_watch(gchansock, G_IO_IN, readsock, NULL); 120 } 121 122 void 123 pagecreated(WebKitWebExtension *e, WebKitWebPage *p, gpointer unused) 124 { 125 WebKitUserMessage *msg; 126 127 msg = webkit_user_message_new("page-created", NULL); 128 webkit_web_page_send_message_to_view(p, msg, NULL, pageusermessagereply, p); 129 } 130 131 G_MODULE_EXPORT void 132 webkit_web_extension_initialize(WebKitWebExtension *e) 133 { 134 webext = e; 135 136 g_signal_connect(G_OBJECT(e), "page-created", 137 G_CALLBACK(pagecreated), NULL); 138 }