compositor: Add socket based activation for X server
authorKristian Høgsberg <krh@bitplanet.net>
Tue, 21 Jun 2011 21:32:25 +0000 (17:32 -0400)
committerKristian Høgsberg <krh@bitplanet.net>
Tue, 21 Jun 2011 21:32:25 +0000 (17:32 -0400)
compositor/Makefile.am
compositor/compositor.c
compositor/compositor.h
compositor/x11-integration.c [new file with mode: 0644]

index ac63487..43a077e 100644 (file)
@@ -15,7 +15,8 @@ wayland_compositor_SOURCES =                  \
        image-loader.c                          \
        screenshooter.c                         \
        screenshooter-protocol.c                \
-       screenshooter-server-protocol.h
+       screenshooter-server-protocol.h         \
+       x11-integration.c
 
 udevrulesddir = $(sysconfdir)/udev/rules.d
 
index 73affad..3257240 100644 (file)
@@ -28,6 +28,7 @@
 #include <stdarg.h>
 #include <assert.h>
 #include <sys/ioctl.h>
+#include <sys/wait.h>
 #include <fcntl.h>
 #include <unistd.h>
 #include <math.h>
@@ -46,6 +47,38 @@ static const char *option_socket_name = NULL;
 static const char *option_background = "background.jpg";
 static int option_idle_time = 300;
 
+static struct wl_list child_process_list;
+
+static int
+sigchld_handler(int signal_number, void *data)
+{
+       struct wlsc_process *p;
+       int status;
+       pid_t pid;
+
+       pid = wait(&status);
+       wl_list_for_each(p, &child_process_list, link) {
+               if (p->pid == pid)
+                       break;
+       }
+
+       if (&p->link == &child_process_list) {
+               fprintf(stderr, "unknown child process exited\n");
+               return 1;
+       }
+
+       wl_list_remove(&p->link);
+       p->cleanup(p, status);
+
+       return 1;
+}
+
+WL_EXPORT void
+wlsc_watch_process(struct wlsc_process *process)
+{
+       wl_list_insert(&child_process_list, &process->link);
+}
+
 WL_EXPORT void
 wlsc_matrix_init(struct wlsc_matrix *matrix)
 {
@@ -1885,7 +1918,7 @@ int main(int argc, char *argv[])
        struct wl_display *display;
        struct wlsc_compositor *ec;
        struct wl_event_loop *loop;
-       int o;
+       int o, xserver = 0;
        void *shell_module, *backend_module;
        int (*shell_init)(struct wlsc_compositor *ec);
        struct wlsc_compositor
@@ -1895,7 +1928,7 @@ int main(int argc, char *argv[])
        char *shell = NULL;
        char *p;
 
-       static const char opts[] = "B:b:o:S:i:s:";
+       static const char opts[] = "B:b:o:S:i:s:x";
        static const struct option longopts[ ] = {
                { "backend", 1, NULL, 'B' },
                { "backend-options", 1, NULL, 'o' },
@@ -1903,6 +1936,7 @@ int main(int argc, char *argv[])
                { "socket", 1, NULL, 'S' },
                { "idle-time", 1, NULL, 'i' },
                { "shell", 1, NULL, 's' },
+               { "xserver", 0, NULL, 'x' },
                { NULL, }
        };
 
@@ -1932,6 +1966,9 @@ int main(int argc, char *argv[])
                case 's':
                        shell = optarg;
                        break;
+               case 'x':
+                       xserver = 1;
+                       break;
                }
        }
 
@@ -1970,6 +2007,9 @@ int main(int argc, char *argv[])
        if (shell_init(ec) < 0)
                exit(EXIT_FAILURE);
 
+       if (xserver)
+               wlsc_xserver_init(display);
+
        if (wl_display_add_socket(display, option_socket_name)) {
                fprintf(stderr, "failed to add socket: %m\n");
                exit(EXIT_FAILURE);
@@ -1979,6 +2019,9 @@ int main(int argc, char *argv[])
        wl_event_loop_add_signal(loop, SIGTERM, on_term_signal, ec);
        wl_event_loop_add_signal(loop, SIGINT, on_term_signal, ec);
 
+       wl_list_init(&child_process_list);
+       wl_event_loop_add_signal(loop, SIGCHLD, sigchld_handler, NULL);
+
        wl_display_run(display);
 
        if (ec->has_bind_display)
index fc8442e..31ace44 100644 (file)
@@ -378,4 +378,16 @@ uint32_t *
 wlsc_load_image(const char *filename,
                int32_t *width_arg, int32_t *height_arg, uint32_t *stride_arg);
 
+struct wlsc_process {
+       pid_t pid;
+       void (*cleanup)(struct wlsc_process *process, int status);
+       struct wl_list link;
+};
+
+void
+wlsc_watch_process(struct wlsc_process *process);
+
+int
+wlsc_xserver_init(struct wl_display *display);
+
 #endif
diff --git a/compositor/x11-integration.c b/compositor/x11-integration.c
new file mode 100644 (file)
index 0000000..f55d631
--- /dev/null
@@ -0,0 +1,206 @@
+/*
+ * Copyright © 2011 Intel Corporation
+ *
+ * Permission to use, copy, modify, distribute, and sell this software and
+ * its documentation for any purpose is hereby granted without fee, provided
+ * that the above copyright notice appear in all copies and that both that
+ * copyright notice and this permission notice appear in supporting
+ * documentation, and that the name of the copyright holders not be used in
+ * advertising or publicity pertaining to distribution of the software
+ * without specific, written prior permission.  The copyright holders make
+ * no representations about the suitability of this software for any
+ * purpose.  It is provided "as is" without express or implied warranty.
+ *
+ * THE COPYRIGHT HOLDERS DISCLAIM ALL WARRANTIES WITH REGARD TO THIS
+ * SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
+ * FITNESS, IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY
+ * SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER
+ * RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF
+ * CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
+ * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+#define _GNU_SOURCE
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+#include <fcntl.h>
+#include <errno.h>
+#include <unistd.h>
+
+#include <xcb/xcb.h>
+
+#include <wayland-server.h>
+
+#include "compositor.h"
+
+/*
+ * TODO:
+ *  - Clean X socket and lock file on exit
+ *  - Nuke lock file if process doesn't exist.
+ */
+
+struct wlsc_xserver {
+       struct wl_display *wl_display;
+       struct wl_event_loop *loop;
+       struct wl_event_source *source;
+       struct wl_event_source *sigchld_source;
+       struct wl_client *client;
+       int fd;
+       struct sockaddr_un addr;
+       char lock_addr[113];
+       int display;
+       struct wlsc_process process;
+};
+
+static int
+wlsc_xserver_handle_event(int listen_fd, uint32_t mask, void *data)
+{
+       struct wlsc_xserver *mxs = data;
+       char listen_fd_str[8], display[8], s[8];
+       int sv[2], flags;
+
+       if (socketpair(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0, sv) < 0) {
+               fprintf(stderr, "socketpair failed\n");
+               return 1;
+       }
+
+       mxs->process.pid = fork();
+       switch (mxs->process.pid) {
+       case 0:
+               /* SOCK_CLOEXEC closes both ends, so we need to unset
+                * the flag on the client fd. */
+               flags = fcntl(sv[1], F_GETFD);
+               if (flags != -1)
+                       fcntl(sv[1], F_SETFD, flags & ~FD_CLOEXEC);
+
+               flags = fcntl(listen_fd, F_GETFD);
+               if (flags != -1)
+                       fcntl(listen_fd, F_SETFD, flags & ~FD_CLOEXEC);
+
+               snprintf(s, sizeof s, "%d", sv[1]);
+               setenv("WAYLAND_SOCKET", s, 1);
+
+               snprintf(listen_fd_str, sizeof listen_fd_str, "%d", listen_fd);
+               snprintf(display, sizeof display, ":%d", mxs->display);
+
+               if (execl("/usr/bin/Xorg",
+                         "/usr/bin/Xorg",
+                         display,
+                         "-wayland",
+                         // "-rootless",
+                         "-retro",
+                         "-logfile", "/tmp/foo",
+                         "-terminate",
+                         "-socket", listen_fd_str,
+                         NULL) < 0)
+                       fprintf(stderr, "exec failed: %m\n");
+               exit(-1);
+
+       default:
+               fprintf(stderr, "forked X server, pid %d\n", mxs->process.pid);
+
+               close(sv[1]);
+               mxs->client = wl_client_create(mxs->wl_display, sv[0]);
+
+               wlsc_watch_process(&mxs->process);
+
+               wl_event_source_remove(mxs->source);
+               break;
+
+       case -1:
+               fprintf(stderr, "failed to fork\n");
+               break;
+       }
+
+       return 1;
+}
+
+static void
+wlsc_xserver_cleanup(struct wlsc_process *process, int status)
+{
+       struct wlsc_xserver *mxs =
+               container_of(process, struct wlsc_xserver, process);
+
+       fprintf(stderr, "xserver exited, code %d\n", status);
+       mxs->process.pid = 0;
+       mxs->source =
+               wl_event_loop_add_fd(mxs->loop, mxs->fd, WL_EVENT_READABLE,
+                                    wlsc_xserver_handle_event, mxs);
+}
+
+int
+wlsc_xserver_init(struct wl_display *display)
+{
+       struct wlsc_xserver *mxs;
+       char lockfile[256], pid[16];
+       socklen_t size, name_size;
+       int fd;
+
+       mxs = malloc(sizeof *mxs);
+       memset(mxs, 0, sizeof mxs);
+
+       mxs->process.cleanup = wlsc_xserver_cleanup;
+       mxs->wl_display = display;
+       mxs->addr.sun_family = AF_LOCAL;
+       mxs->fd = socket(PF_LOCAL, SOCK_STREAM | SOCK_CLOEXEC, 0);
+       if (mxs->fd < 0) {
+               free(mxs);
+               return -1;
+       }
+
+       mxs->display = 0;
+       do {
+               snprintf(lockfile, sizeof lockfile,
+                        "/tmp/.X%d-lock", mxs->display);
+               fd = open(lockfile,
+                         O_WRONLY | O_CLOEXEC | O_CREAT | O_EXCL, 0444);
+               if (fd < 0 && errno == EEXIST) {
+                       fprintf(stderr, "server %d exists\n", mxs->display);
+                       mxs->display++;
+                       continue;
+               } else if (fd < 0) {
+                       close(mxs->fd);
+                       free(mxs);
+                       return -1;
+               }
+
+               size = snprintf(pid, sizeof pid, "%10d\n", getpid());
+               write(fd, pid, size);
+               close(fd);
+
+               name_size = snprintf(mxs->addr.sun_path,
+                                    sizeof mxs->addr.sun_path,
+                                    "/tmp/.X11-unix/X%d", mxs->display) + 1;
+               size = offsetof(struct sockaddr_un, sun_path) + name_size;
+               if (bind(mxs->fd, (struct sockaddr *) &mxs->addr, size) < 0) {
+                       fprintf(stderr, "failed to bind to %s (%m)\n",
+                               mxs->addr.sun_path);
+                       unlink(lockfile);
+                       close(mxs->fd);
+                       free(mxs);
+                       return -1;
+               }
+               break;
+       } while (errno != 0);
+
+       fprintf(stderr, "listening on display %d\n", mxs->display);
+
+       if (listen(mxs->fd, 1) < 0) {
+               unlink(mxs->addr.sun_path);
+               unlink(lockfile);
+               close(mxs->fd);
+               free(mxs);
+               return -1;
+       }
+
+       mxs->loop = wl_display_get_event_loop(display);
+       mxs->source =
+               wl_event_loop_add_fd(mxs->loop, mxs->fd, WL_EVENT_READABLE,
+                                    wlsc_xserver_handle_event, mxs);
+
+       return 0;
+}