1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
#include "../include/wayland_host_bridge.h"
#include "../include/log.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <sys/uio.h>
#include <sys/wait.h>
#include <wayland-client.h>
#define GSR_UI_UNIX_SOCKET_DOMAIN_FD 3
static const char *bridge_host_path =
"/var/lib/flatpak/app/com.dec05eba.gpu_screen_recorder/current/active/files/bin/gsr-wayland-bridge";
static int recv_fd(int sock) {
char dummy = 0;
struct iovec iov = { &dummy, 1 };
char cbuf[CMSG_SPACE(sizeof(int))];
memset(cbuf, 0, sizeof(cbuf));
struct msghdr msg;
memset(&msg, 0, sizeof(msg));
msg.msg_iov = &iov;
msg.msg_iovlen = 1;
msg.msg_control = cbuf;
msg.msg_controllen = sizeof(cbuf);
const ssize_t n = recvmsg(sock, &msg, 0);
if(n <= 0)
return -1;
for(struct cmsghdr *c = CMSG_FIRSTHDR(&msg); c; c = CMSG_NXTHDR(&msg, c)) {
if(c->cmsg_level == SOL_SOCKET && c->cmsg_type == SCM_RIGHTS) {
int fd = -1;
memcpy(&fd, CMSG_DATA(c), sizeof(int));
return fd;
}
}
return -1;
}
static struct wl_display* connect_via_bridge() {
const char *wayland_display = getenv("WAYLAND_DISPLAY");
int sv[2];
if(socketpair(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0, sv) != 0) {
perror("WaylandHostBridge: socketpair");
return NULL;
}
const pid_t pid = fork();
if(pid < 0) {
perror("WaylandHostBridge: fork");
close(sv[0]);
close(sv[1]);
return NULL;
}
if(pid == 0) {
close(sv[0]);
if(sv[1] != GSR_UI_UNIX_SOCKET_DOMAIN_FD) {
dup2(sv[1], GSR_UI_UNIX_SOCKET_DOMAIN_FD);
close(sv[1]);
}
const int flags = fcntl(GSR_UI_UNIX_SOCKET_DOMAIN_FD, F_GETFD);
if(flags >= 0)
fcntl(GSR_UI_UNIX_SOCKET_DOMAIN_FD, F_SETFD, flags & ~FD_CLOEXEC);
char forward_fd_arg[32];
snprintf(forward_fd_arg, sizeof(forward_fd_arg), "--forward-fd=%d", GSR_UI_UNIX_SOCKET_DOMAIN_FD);
execlp("flatpak-spawn", "flatpak-spawn", "--host", forward_fd_arg, bridge_host_path, wayland_display, (char*)NULL);
_exit(127);
}
close(sv[1]);
const int wayland_fd = recv_fd(sv[0]);
close(sv[0]);
int status = 0;
waitpid(pid, &status, 0);
if(wayland_fd < 0) {
gsr_log(GSR_LOG_LEVEL_ERROR, "WaylandHostBridge: gsr-wayland-bridge did not return a wayland fd");
return NULL;
}
struct wl_display *dpy = wl_display_connect_to_fd(wayland_fd);
if(!dpy) {
close(wayland_fd);
gsr_log(GSR_LOG_LEVEL_ERROR, "WaylandHostBridge: wl_display_connect_to_fd failed");
return NULL;
}
return dpy;
}
struct wl_display* wayland_connect_to_host() {
const int inside_flatpak = getenv("FLATPAK_ID") != NULL;
if(inside_flatpak) {
struct wl_display *dpy = connect_via_bridge();
if(dpy)
return dpy;
gsr_log(GSR_LOG_LEVEL_WARNING, "WaylandHostBridge: falling back to sandboxed wl_display_connect");
}
return wl_display_connect(NULL);
}
|
