2 * Copyright © 2012 Benjamin Franzke
3 * Copyright © 2013 Intel Corporation
5 * Permission to use, copy, modify, distribute, and sell this software and
6 * its documentation for any purpose is hereby granted without fee, provided
7 * that the above copyright notice appear in all copies and that both that
8 * copyright notice and this permission notice appear in supporting
9 * documentation, and that the name of the copyright holders not be used in
10 * advertising or publicity pertaining to distribution of the software
11 * without specific, written prior permission. The copyright holders make
12 * no representations about the suitability of this software for any
13 * purpose. It is provided "as is" without express or implied warranty.
15 * THE COPYRIGHT HOLDERS DISCLAIM ALL WARRANTIES WITH REGARD TO THIS
16 * SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
17 * FITNESS, IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY
18 * SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER
19 * RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF
20 * CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
21 * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
32 #include <sys/socket.h>
33 #include <sys/types.h>
36 #include <sys/ioctl.h>
41 #include <linux/major.h>
43 #include "compositor.h"
44 #include "launcher-util.h"
45 #include "logind-util.h"
46 #include "weston-launch.h"
51 #define KDSKBMUTE 0x4B51
59 is_drm_master(int drm_fd)
63 return drmGetMagic(drm_fd, &magic) == 0 &&
64 drmAuthMagic(drm_fd, magic) == 0;
70 drmDropMaster(int drm_fd)
76 drmSetMaster(int drm_fd)
82 is_drm_master(int drm_fd)
90 union cmsg_data { unsigned char b[4]; int fd; };
92 struct weston_launcher {
93 struct weston_compositor *compositor;
94 struct weston_logind *logind;
95 struct wl_event_loop *loop;
97 struct wl_event_source *source;
99 int kb_mode, tty, drm_fd;
100 struct wl_event_source *vt_source;
104 weston_launcher_open(struct weston_launcher *launcher,
105 const char *path, int flags)
109 struct cmsghdr *cmsg;
111 union cmsg_data *data;
112 char control[CMSG_SPACE(sizeof data->fd)];
114 struct weston_launcher_open *message;
117 if (launcher->logind)
118 return weston_logind_open(launcher->logind, path, flags);
120 if (launcher->fd == -1) {
121 fd = open(path, flags | O_CLOEXEC);
125 if (fstat(fd, &s) == -1) {
130 if (major(s.st_rdev) == DRM_MAJOR) {
131 launcher->drm_fd = fd;
132 if (!is_drm_master(fd)) {
133 weston_log("drm fd not master\n");
142 n = sizeof(*message) + strlen(path) + 1;
147 message->header.opcode = WESTON_LAUNCHER_OPEN;
148 message->flags = flags;
149 strcpy(message->path, path);
152 len = send(launcher->fd, message, n, 0);
153 } while (len < 0 && errno == EINTR);
156 memset(&msg, 0, sizeof msg);
158 iov.iov_len = sizeof ret;
161 msg.msg_control = control;
162 msg.msg_controllen = sizeof control;
165 len = recvmsg(launcher->fd, &msg, MSG_CMSG_CLOEXEC);
166 } while (len < 0 && errno == EINTR);
168 if (len != sizeof ret ||
172 cmsg = CMSG_FIRSTHDR(&msg);
174 cmsg->cmsg_level != SOL_SOCKET ||
175 cmsg->cmsg_type != SCM_RIGHTS) {
176 fprintf(stderr, "invalid control message\n");
180 data = (union cmsg_data *) CMSG_DATA(cmsg);
181 if (data->fd == -1) {
182 fprintf(stderr, "missing drm fd in socket request\n");
190 weston_launcher_close(struct weston_launcher *launcher, int fd)
192 if (launcher->logind)
193 return weston_logind_close(launcher->logind, fd);
199 weston_launcher_restore(struct weston_launcher *launcher)
201 struct vt_mode mode = { 0 };
203 if (launcher->logind)
204 return weston_logind_restore(launcher->logind);
206 if (ioctl(launcher->tty, KDSKBMUTE, 0) &&
207 ioctl(launcher->tty, KDSKBMODE, launcher->kb_mode))
208 weston_log("failed to restore kb mode: %m\n");
210 if (ioctl(launcher->tty, KDSETMODE, KD_TEXT))
211 weston_log("failed to set KD_TEXT mode on tty: %m\n");
213 /* We have to drop master before we switch the VT back in
214 * VT_AUTO, so we don't risk switching to a VT with another
215 * display server, that will then fail to set drm master. */
216 drmDropMaster(launcher->drm_fd);
219 if (ioctl(launcher->tty, VT_SETMODE, &mode) < 0)
220 weston_log("could not reset vt handling\n");
224 weston_launcher_data(int fd, uint32_t mask, void *data)
226 struct weston_launcher *launcher = data;
229 if (mask & (WL_EVENT_HANGUP | WL_EVENT_ERROR)) {
230 weston_log("launcher socket closed, exiting\n");
231 /* Normally the weston-launch will reset the tty, but
232 * in this case it died or something, so do it here so
233 * we don't end up with a stuck vt. */
234 weston_launcher_restore(launcher);
239 len = recv(launcher->fd, &ret, sizeof ret, 0);
240 } while (len < 0 && errno == EINTR);
243 case WESTON_LAUNCHER_ACTIVATE:
244 launcher->compositor->session_active = 1;
245 wl_signal_emit(&launcher->compositor->session_signal,
246 launcher->compositor);
248 case WESTON_LAUNCHER_DEACTIVATE:
249 launcher->compositor->session_active = 0;
250 wl_signal_emit(&launcher->compositor->session_signal,
251 launcher->compositor);
254 weston_log("unexpected event from weston-launch\n");
262 vt_handler(int signal_number, void *data)
264 struct weston_launcher *launcher = data;
265 struct weston_compositor *compositor = launcher->compositor;
267 if (compositor->session_active) {
268 compositor->session_active = 0;
269 wl_signal_emit(&compositor->session_signal, compositor);
270 drmDropMaster(launcher->drm_fd);
271 ioctl(launcher->tty, VT_RELDISP, 1);
273 ioctl(launcher->tty, VT_RELDISP, VT_ACKACQ);
274 drmSetMaster(launcher->drm_fd);
275 compositor->session_active = 1;
276 wl_signal_emit(&compositor->session_signal, compositor);
283 setup_tty(struct weston_launcher *launcher, int tty)
285 struct wl_event_loop *loop;
286 struct vt_mode mode = { 0 };
288 char tty_device[32] ="<stdin>";
292 launcher->tty = dup(tty);
293 if (launcher->tty == -1) {
294 weston_log("couldn't dup stdin: %m\n");
298 snprintf(tty_device, sizeof tty_device, "/dev/tty%d", tty);
299 launcher->tty = open(tty_device, O_RDWR | O_CLOEXEC);
300 if (launcher->tty == -1) {
301 weston_log("couldn't open tty %s: %m\n", tty_device);
306 if (fstat(launcher->tty, &buf) == -1 ||
307 major(buf.st_rdev) != TTY_MAJOR || minor(buf.st_rdev) == 0) {
308 weston_log("%s not a vt\n", tty_device);
309 weston_log("if running weston from ssh, "
310 "use --tty to specify a tty\n");
314 ret = ioctl(launcher->tty, KDGETMODE, &kd_mode);
316 weston_log("failed to get VT mode: %m\n");
319 if (kd_mode != KD_TEXT) {
320 weston_log("%s is already in graphics mode, "
321 "is another display server running?\n", tty_device);
325 ioctl(launcher->tty, VT_ACTIVATE, minor(buf.st_rdev));
326 ioctl(launcher->tty, VT_WAITACTIVE, minor(buf.st_rdev));
328 if (ioctl(launcher->tty, KDGKBMODE, &launcher->kb_mode)) {
329 weston_log("failed to read keyboard mode: %m\n");
333 if (ioctl(launcher->tty, KDSKBMUTE, 1) &&
334 ioctl(launcher->tty, KDSKBMODE, K_OFF)) {
335 weston_log("failed to set K_OFF keyboard mode: %m\n");
339 ret = ioctl(launcher->tty, KDSETMODE, KD_GRAPHICS);
341 weston_log("failed to set KD_GRAPHICS mode on tty: %m\n");
345 mode.mode = VT_PROCESS;
346 mode.relsig = SIGUSR1;
347 mode.acqsig = SIGUSR1;
348 if (ioctl(launcher->tty, VT_SETMODE, &mode) < 0) {
349 weston_log("failed to take control of vt handling\n");
353 loop = wl_display_get_event_loop(launcher->compositor->wl_display);
354 launcher->vt_source =
355 wl_event_loop_add_signal(loop, SIGUSR1, vt_handler, launcher);
356 if (!launcher->vt_source)
362 close(launcher->tty);
367 weston_launcher_activate_vt(struct weston_launcher *launcher, int vt)
369 if (launcher->logind)
370 return weston_logind_activate_vt(launcher->logind, vt);
372 return ioctl(launcher->tty, VT_ACTIVATE, vt);
375 struct weston_launcher *
376 weston_launcher_connect(struct weston_compositor *compositor, int tty,
379 struct weston_launcher *launcher;
380 struct wl_event_loop *loop;
383 launcher = malloc(sizeof *launcher);
384 if (launcher == NULL)
387 launcher->logind = NULL;
388 launcher->compositor = compositor;
389 launcher->drm_fd = -1;
390 launcher->fd = weston_environment_get_fd("WESTON_LAUNCHER_SOCK");
391 if (launcher->fd != -1) {
392 launcher->tty = weston_environment_get_fd("WESTON_TTY_FD");
393 /* We don't get a chance to read out the original kb
394 * mode for the tty, so just hard code K_UNICODE here
395 * in case we have to clean if weston-launch dies. */
396 launcher->kb_mode = K_UNICODE;
398 loop = wl_display_get_event_loop(compositor->wl_display);
399 launcher->source = wl_event_loop_add_fd(loop, launcher->fd,
401 weston_launcher_data,
403 if (launcher->source == NULL) {
408 r = weston_logind_connect(&launcher->logind, compositor,
411 launcher->logind = NULL;
412 #ifdef ENABLE_SYS_UID
413 if (geteuid() <= 499) { /* 499 = SYS_UID_MAX in login.defs, but it should be parsed */
415 if (geteuid() == 0) {
417 if (setup_tty(launcher, tty) == -1) {
432 weston_launcher_destroy(struct weston_launcher *launcher)
434 if (launcher->logind) {
435 weston_logind_destroy(launcher->logind);
436 } else if (launcher->fd != -1) {
438 wl_event_source_remove(launcher->source);
440 weston_launcher_restore(launcher);
441 wl_event_source_remove(launcher->vt_source);
444 if (launcher->tty >= 0)
445 close(launcher->tty);