2 * Copyright (C) 2012-2013 Robert Ancell.
3 * Author: Robert Ancell <robert.ancell@canonical.com>
5 * This program is free software: you can redistribute it and/or modify it under
6 * the terms of the GNU General Public License as published by the Free Software
7 * Foundation, either version 3 of the License, or (at your option) any later
8 * version. See http://www.gnu.org/copyleft/gpl.html the full text of the
15 #include <glib/gstdio.h>
17 #include "seat-unity.h"
18 #include "configuration.h"
19 #include "x-server-local.h"
20 #include "mir-server.h"
28 USC_MESSAGE_READY = 2,
29 USC_MESSAGE_SESSION_CONNECTED = 3,
30 USC_MESSAGE_SET_ACTIVE_SESSION = 4
33 struct SeatUnityPrivate
35 /* VT we are running on */
38 /* TRUE if waiting for X server to start before stopping Plymouth */
39 gboolean stopping_plymouth;
44 /* Filename of Mir socket */
45 gchar *mir_socket_filename;
47 /* Pipes to communicate with compositor */
48 int to_compositor_pipe[2];
49 int from_compositor_pipe[2];
51 /* IO channel listening on for messages from the compositor */
52 GIOChannel *from_compositor_channel;
54 /* TRUE when the compositor indicates it is ready */
55 gboolean compositor_ready;
57 /* Buffer reading from channel */
59 gsize read_buffer_length;
60 gsize read_buffer_n_used;
62 /* Compositor process */
63 Process *compositor_process;
65 /* Timeout when waiting for compositor to start */
66 guint compositor_timeout;
68 /* Next Mir ID to use for a compositor client */
71 /* TRUE if using VT switching fallback */
72 gboolean use_vt_switching;
74 /* The currently visible session */
75 Session *active_session;
76 DisplayServer *active_display_server;
79 G_DEFINE_TYPE (SeatUnity, seat_unity, SEAT_TYPE);
82 seat_unity_get_start_local_sessions (Seat *seat)
84 return !seat_get_string_property (seat, "xdmcp-manager");
88 seat_unity_setup (Seat *seat)
90 seat_set_can_switch (seat, TRUE);
91 SEAT_CLASS (seat_unity_parent_class)->setup (seat);
95 compositor_stopped_cb (Process *process, SeatUnity *seat)
97 if (seat->priv->compositor_timeout != 0)
98 g_source_remove (seat->priv->compositor_timeout);
99 seat->priv->compositor_timeout = 0;
101 /* Finished with the VT */
102 vt_unref (seat->priv->vt);
105 if (seat_get_is_stopping (SEAT (seat)))
107 SEAT_CLASS (seat_unity_parent_class)->stop (SEAT (seat));
111 /* If stopped before it was ready, then revert to VT mode */
112 if (!seat->priv->compositor_ready)
114 g_debug ("Compositor failed to start, switching to VT mode");
115 seat->priv->use_vt_switching = TRUE;
116 SEAT_CLASS (seat_unity_parent_class)->start (SEAT (seat));
120 g_debug ("Stopping Unity seat, compositor terminated");
122 if (seat->priv->stopping_plymouth)
124 g_debug ("Stopping Plymouth, compositor failed to start");
125 plymouth_quit (FALSE);
126 seat->priv->stopping_plymouth = FALSE;
129 seat_stop (SEAT (seat));
133 compositor_run_cb (Process *process, SeatUnity *seat)
137 /* Make input non-blocking */
138 fd = open ("/dev/null", O_RDONLY);
139 dup2 (fd, STDIN_FILENO);
142 /* Redirect output to logfile */
143 if (seat->priv->log_file)
147 fd = g_open (seat->priv->log_file, O_WRONLY | O_CREAT | O_TRUNC, 0600);
149 g_warning ("Failed to open log file %s: %s", seat->priv->log_file, g_strerror (errno));
152 dup2 (fd, STDOUT_FILENO);
153 dup2 (fd, STDERR_FILENO);
158 if (seat->priv->stopping_plymouth)
160 seat->priv->stopping_plymouth = FALSE;
161 plymouth_quit (TRUE);
166 write_message (SeatUnity *seat, guint16 id, const guint8 *payload, guint16 payload_length)
169 gsize data_length = 4 + payload_length;
171 data = g_malloc (data_length);
174 data[2] = payload_length >> 8;
175 data[3] = payload_length & 0xFF;
176 memcpy (data + 4, payload, payload_length);
179 if (write (seat->priv->to_compositor_pipe[1], data, data_length) != data_length)
180 g_warning ("Failed to write to compositor: %s", strerror (errno));
184 read_cb (GIOChannel *source, GIOCondition condition, gpointer data)
186 SeatUnity *seat = data;
188 guint16 id, payload_length;
191 if (condition == G_IO_HUP)
193 g_debug ("Compositor closed communication channel");
197 /* Work out how much required for a message */
198 if (seat->priv->read_buffer_n_used < 4)
199 n_to_read = 4 - seat->priv->read_buffer_n_used;
202 payload_length = seat->priv->read_buffer[2] << 8 | seat->priv->read_buffer[3];
203 n_to_read = 4 + payload_length - seat->priv->read_buffer_n_used;
206 /* Read from compositor */
209 gsize n_total, n_read = 0;
211 GError *error = NULL;
213 n_total = seat->priv->read_buffer_n_used + n_to_read;
214 if (seat->priv->read_buffer_length < n_total)
215 seat->priv->read_buffer = g_realloc (seat->priv->read_buffer, n_total);
217 status = g_io_channel_read_chars (source,
218 (gchar *)seat->priv->read_buffer + seat->priv->read_buffer_n_used,
223 g_warning ("Failed to read from compositor: %s", error->message);
224 if (status != G_IO_STATUS_NORMAL)
226 g_clear_error (&error);
227 seat->priv->read_buffer_n_used += n_read;
231 if (seat->priv->read_buffer_n_used < 4)
233 id = seat->priv->read_buffer[0] << 8 | seat->priv->read_buffer[1];
234 payload_length = seat->priv->read_buffer[2] << 8 | seat->priv->read_buffer[3];
237 if (seat->priv->read_buffer_n_used < 4 + payload_length)
239 /*payload = seat->priv->read_buffer + 4;*/
243 case USC_MESSAGE_PING:
245 write_message (seat, USC_MESSAGE_PONG, NULL, 0);
247 case USC_MESSAGE_PONG:
250 case USC_MESSAGE_READY:
252 if (!seat->priv->compositor_ready)
254 seat->priv->compositor_ready = TRUE;
255 g_debug ("Compositor ready");
256 g_source_remove (seat->priv->compositor_timeout);
257 seat->priv->compositor_timeout = 0;
258 SEAT_CLASS (seat_unity_parent_class)->start (SEAT (seat));
261 case USC_MESSAGE_SESSION_CONNECTED:
262 g_debug ("SESSION CONNECTED");
265 g_warning ("Ingoring unknown message %d with %d octets from system compositor", id, payload_length);
270 seat->priv->read_buffer_n_used = 0;
276 get_absolute_command (const gchar *command)
279 gchar *absolute_binary, *absolute_command = NULL;
281 tokens = g_strsplit (command, " ", 2);
283 absolute_binary = g_find_program_in_path (tokens[0]);
287 absolute_command = g_strjoin (" ", absolute_binary, tokens[1], NULL);
289 absolute_command = g_strdup (absolute_binary);
290 g_free (absolute_binary);
293 absolute_command = g_strdup (command);
297 return absolute_command;
301 compositor_timeout_cb (gpointer data)
303 SeatUnity *seat = data;
305 /* Stop the compositor - it is not working */
306 process_stop (seat->priv->compositor_process);
312 seat_unity_start (Seat *seat)
314 const gchar *compositor_command;
315 gchar *command, *absolute_command, *dir;
319 /* Replace Plymouth if it is running */
320 if (plymouth_get_is_active () && plymouth_has_active_vt ())
322 gint active_vt = vt_get_active ();
323 if (active_vt >= vt_get_min ())
325 g_debug ("Compositor will replace Plymouth");
326 SEAT_UNITY (seat)->priv->stopping_plymouth = TRUE;
327 SEAT_UNITY (seat)->priv->vt = active_vt;
328 plymouth_deactivate ();
331 g_debug ("Plymouth is running on VT %d, but this is less than the configured minimum of %d so not replacing it", active_vt, vt_get_min ());
333 if (SEAT_UNITY (seat)->priv->vt < 0)
334 SEAT_UNITY (seat)->priv->vt = vt_get_unused ();
335 if (SEAT_UNITY (seat)->priv->vt < 0)
337 g_debug ("Failed to get a VT to run on");
340 vt_ref (SEAT_UNITY (seat)->priv->vt);
342 /* Create pipes to talk to compositor */
343 if (pipe (SEAT_UNITY (seat)->priv->to_compositor_pipe) < 0 || pipe (SEAT_UNITY (seat)->priv->from_compositor_pipe) < 0)
345 g_debug ("Failed to create compositor pipes: %s", g_strerror (errno));
349 /* Don't allow the daemon end of the pipes to be accessed in the compositor */
350 fcntl (SEAT_UNITY (seat)->priv->to_compositor_pipe[1], F_SETFD, FD_CLOEXEC);
351 fcntl (SEAT_UNITY (seat)->priv->from_compositor_pipe[0], F_SETFD, FD_CLOEXEC);
353 /* Listen for messages from the compositor */
354 SEAT_UNITY (seat)->priv->from_compositor_channel = g_io_channel_unix_new (SEAT_UNITY (seat)->priv->from_compositor_pipe[0]);
355 g_io_add_watch (SEAT_UNITY (seat)->priv->from_compositor_channel, G_IO_IN | G_IO_HUP, read_cb, seat);
358 dir = config_get_string (config_get_instance (), "LightDM", "log-directory");
359 SEAT_UNITY (seat)->priv->log_file = g_build_filename (dir, "unity-system-compositor.log", NULL);
360 g_debug ("Logging to %s", SEAT_UNITY (seat)->priv->log_file);
363 SEAT_UNITY (seat)->priv->mir_socket_filename = g_strdup ("/tmp/mir_socket"); // FIXME: Use this socket by default as XMir is hardcoded to this
364 timeout = seat_get_integer_property (seat, "unity-compositor-timeout");
365 compositor_command = seat_get_string_property (seat, "unity-compositor-command");
366 command = g_strdup_printf ("%s --from-dm-fd %d --to-dm-fd %d --vt %d", compositor_command, SEAT_UNITY (seat)->priv->to_compositor_pipe[0], SEAT_UNITY (seat)->priv->from_compositor_pipe[1], SEAT_UNITY (seat)->priv->vt);
368 absolute_command = get_absolute_command (command);
371 /* Start the compositor */
372 process_set_command (SEAT_UNITY (seat)->priv->compositor_process, absolute_command);
373 g_free (absolute_command);
374 g_signal_connect (SEAT_UNITY (seat)->priv->compositor_process, "stopped", G_CALLBACK (compositor_stopped_cb), seat);
375 g_signal_connect (SEAT_UNITY (seat)->priv->compositor_process, "run", G_CALLBACK (compositor_run_cb), seat);
376 result = process_start (SEAT_UNITY (seat)->priv->compositor_process, FALSE);
378 /* Close compostor ends of the pipes */
379 close (SEAT_UNITY (seat)->priv->to_compositor_pipe[0]);
380 SEAT_UNITY (seat)->priv->to_compositor_pipe[0] = 0;
381 close (SEAT_UNITY (seat)->priv->from_compositor_pipe[1]);
382 SEAT_UNITY (seat)->priv->from_compositor_pipe[1] = 0;
387 /* Connect to the compositor */
388 timeout = seat_get_integer_property (seat, "unity-compositor-timeout");
391 g_debug ("Waiting for system compositor for %ds", timeout);
392 SEAT_UNITY (seat)->priv->compositor_timeout = g_timeout_add (timeout * 1000, compositor_timeout_cb, seat);
397 static DisplayServer *
398 create_x_server (Seat *seat)
400 XServerLocal *x_server;
401 const gchar *command = NULL, *layout = NULL, *config_file = NULL, *xdmcp_manager = NULL, *key_name = NULL;
406 g_debug ("Starting X server on Unity compositor");
408 x_server = x_server_local_new ();
410 if (!SEAT_UNITY (seat)->priv->use_vt_switching)
412 id = g_strdup_printf ("%d", SEAT_UNITY (seat)->priv->next_id);
413 SEAT_UNITY (seat)->priv->next_id++;
414 x_server_local_set_mir_id (x_server, id);
415 x_server_local_set_mir_socket (x_server, SEAT_UNITY (seat)->priv->mir_socket_filename);
419 command = seat_get_string_property (seat, "xserver-command");
421 x_server_local_set_command (x_server, command);
423 layout = seat_get_string_property (seat, "xserver-layout");
425 x_server_local_set_layout (x_server, layout);
427 config_file = seat_get_string_property (seat, "xserver-config");
429 x_server_local_set_config (x_server, config_file);
431 allow_tcp = seat_get_boolean_property (seat, "xserver-allow-tcp");
432 x_server_local_set_allow_tcp (x_server, allow_tcp);
434 xdmcp_manager = seat_get_string_property (seat, "xdmcp-manager");
436 x_server_local_set_xdmcp_server (x_server, xdmcp_manager);
438 port = seat_get_integer_property (seat, "xdmcp-port");
440 x_server_local_set_xdmcp_port (x_server, port);
442 key_name = seat_get_string_property (seat, "xdmcp-key");
448 GError *error = NULL;
450 dir = config_get_string (config_get_instance (), "LightDM", "config-directory");
451 path = g_build_filename (dir, "keys.conf", NULL);
454 keys = g_key_file_new ();
455 result = g_key_file_load_from_file (keys, path, G_KEY_FILE_NONE, &error);
457 g_debug ("Error getting key %s", error->message);
458 g_clear_error (&error);
464 if (g_key_file_has_key (keys, "keyring", key_name, NULL))
465 key = g_key_file_get_string (keys, "keyring", key_name, NULL);
467 g_debug ("Key %s not defined", key_name);
470 x_server_local_set_xdmcp_key (x_server, key);
475 g_key_file_free (keys);
478 return DISPLAY_SERVER (x_server);
481 static DisplayServer *
482 seat_unity_create_display_server (Seat *seat, const gchar *session_type)
484 if (strcmp (session_type, "x") == 0)
485 return create_x_server (seat);
486 else if (strcmp (session_type, "mir") == 0)
487 return DISPLAY_SERVER (mir_server_new ());
490 g_warning ("Can't create unsupported display server '%s'", session_type);
496 seat_unity_create_greeter_session (Seat *seat)
498 Greeter *greeter_session;
500 greeter_session = greeter_new ();
501 session_set_env (SESSION (greeter_session), "XDG_SEAT", "seat0");
503 return greeter_session;
507 seat_unity_create_session (Seat *seat)
511 session = session_new ();
512 session_set_env (session, "XDG_SEAT", "seat0");
518 seat_unity_set_active_session (Seat *seat, Session *session)
520 DisplayServer *display_server;
522 /* If no compositor, have to use VT switching */
523 if (SEAT_UNITY (seat)->priv->use_vt_switching)
525 gint vt = display_server_get_vt (session_get_display_server (session));
529 SEAT_CLASS (seat_unity_parent_class)->set_active_session (seat, session);
533 if (session == SEAT_UNITY (seat)->priv->active_session)
535 SEAT_UNITY (seat)->priv->active_session = g_object_ref (session);
537 display_server = session_get_display_server (session);
538 if (SEAT_UNITY (seat)->priv->active_display_server != display_server)
542 SEAT_UNITY (seat)->priv->active_display_server = g_object_ref (display_server);
543 id = x_server_local_get_mir_id (X_SERVER_LOCAL (display_server));
545 g_debug ("Switching to Mir session %s", id);
546 write_message (SEAT_UNITY (seat), USC_MESSAGE_SET_ACTIVE_SESSION, (const guint8 *) id, strlen (id));
549 SEAT_CLASS (seat_unity_parent_class)->set_active_session (seat, session);
553 seat_unity_get_active_session (Seat *seat)
555 if (SEAT_UNITY (seat)->priv->use_vt_switching)
559 vt = vt_get_active ();
563 for (link = seat_get_sessions (seat); link; link = link->next)
565 Session *session = link->data;
566 if (display_server_get_vt (session_get_display_server (session)) == vt)
573 return SEAT_UNITY (seat)->priv->active_session;
577 seat_unity_run_script (Seat *seat, DisplayServer *display_server, Process *script)
580 XServerLocal *x_server;
582 x_server = X_SERVER_LOCAL (display_server);
583 path = x_server_local_get_authority_file_path (x_server);
584 process_set_env (script, "DISPLAY", x_server_get_address (X_SERVER (x_server)));
585 process_set_env (script, "XAUTHORITY", path);
587 SEAT_CLASS (seat_unity_parent_class)->run_script (seat, display_server, script);
591 seat_unity_stop (Seat *seat)
593 /* Stop the compositor first */
594 if (process_get_is_running (SEAT_UNITY (seat)->priv->compositor_process))
596 process_stop (SEAT_UNITY (seat)->priv->compositor_process);
600 SEAT_CLASS (seat_unity_parent_class)->stop (seat);
604 seat_unity_init (SeatUnity *seat)
606 seat->priv = G_TYPE_INSTANCE_GET_PRIVATE (seat, SEAT_UNITY_TYPE, SeatUnityPrivate);
608 seat->priv->compositor_process = process_new ();
612 seat_unity_finalize (GObject *object)
614 SeatUnity *seat = SEAT_UNITY (object);
616 if (seat->priv->vt >= 0)
617 vt_unref (seat->priv->vt);
618 g_free (seat->priv->log_file);
619 g_free (seat->priv->mir_socket_filename);
620 close (seat->priv->to_compositor_pipe[0]);
621 close (seat->priv->to_compositor_pipe[1]);
622 close (seat->priv->from_compositor_pipe[0]);
623 close (seat->priv->from_compositor_pipe[1]);
624 g_io_channel_unref (seat->priv->from_compositor_channel);
625 g_free (seat->priv->read_buffer);
626 g_object_unref (seat->priv->compositor_process);
627 if (seat->priv->active_session)
628 g_object_unref (seat->priv->active_session);
629 if (seat->priv->active_display_server)
630 g_object_unref (seat->priv->active_display_server);
632 G_OBJECT_CLASS (seat_unity_parent_class)->finalize (object);
636 seat_unity_class_init (SeatUnityClass *klass)
638 GObjectClass *object_class = G_OBJECT_CLASS (klass);
639 SeatClass *seat_class = SEAT_CLASS (klass);
641 object_class->finalize = seat_unity_finalize;
642 seat_class->get_start_local_sessions = seat_unity_get_start_local_sessions;
643 seat_class->setup = seat_unity_setup;
644 seat_class->start = seat_unity_start;
645 seat_class->create_display_server = seat_unity_create_display_server;
646 seat_class->create_greeter_session = seat_unity_create_greeter_session;
647 seat_class->create_session = seat_unity_create_session;
648 seat_class->set_active_session = seat_unity_set_active_session;
649 seat_class->get_active_session = seat_unity_get_active_session;
650 seat_class->run_script = seat_unity_run_script;
651 seat_class->stop = seat_unity_stop;
653 g_type_class_add_private (klass, sizeof (SeatUnityPrivate));