src/x-server.c

   1 /*
   2  * Copyright (C) 2010-2011 Robert Ancell.
   3  * Author: Robert Ancell <robert.ancell@canonical.com>
   4  *
   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
   9  * license.
  10  */
  11
  12 #include <config.h>
  13 #include <string.h>
  14 #include <xcb/xcb.h>
  15
  16 #include "x-server.h"
  17 #include "configuration.h"
  18
  19 struct XServerPrivate
  20 {
  21     /* Host running the server */
  22     gchar *hostname;
  23
  24     /* Display number */
  25     guint number;
  26
  27     /* Cached server address */
  28     gchar *address;
  29
  30     /* Authority */
  31     XAuthority *authority;
  32
  33     /* Connection to this X server */
  34     xcb_connection_t *connection;
  35 };
  36
  37 G_DEFINE_TYPE (XServer, x_server, DISPLAY_SERVER_TYPE);
  38
  39 void
  40 x_server_set_hostname (XServer *server, const gchar *hostname)
  41 {
  42     g_return_if_fail (server != NULL);
  43     g_free (server->priv->hostname);
  44     server->priv->hostname = g_strdup (hostname);
  45     g_free (server->priv->address);
  46     server->priv->address = NULL;
  47 }
  48
  49 gchar *
  50 x_server_get_hostname (XServer *server)
  51 {
  52     g_return_val_if_fail (server != NULL, NULL);
  53     return server->priv->hostname;
  54 }
  55
  56 void
  57 x_server_set_display_number (XServer *server, guint number)
  58 {
  59     g_return_if_fail (server != NULL);
  60     server->priv->number = number;
  61     g_free (server->priv->address);
  62     server->priv->address = NULL;
  63 }
  64
  65 guint
  66 x_server_get_display_number (XServer *server)
  67 {
  68     g_return_val_if_fail (server != NULL, 0);
  69     return server->priv->number;
  70 }
  71
  72 const gchar *
  73 x_server_get_address (XServer *server)
  74 {
  75     g_return_val_if_fail (server != NULL, NULL);
  76
  77     if (!server->priv->address)
  78     {
  79         if (server->priv->hostname)
  80             server->priv->address = g_strdup_printf("%s:%d", server->priv->hostname, server->priv->number);
  81         else
  82             server->priv->address = g_strdup_printf(":%d", server->priv->number);
  83     }
  84
  85     return server->priv->address;
  86 }
  87
  88 void
  89 x_server_set_authority (XServer *server, XAuthority *authority)
  90 {
  91     g_return_if_fail (server != NULL);
  92
  93     if (server->priv->authority)
  94         g_object_unref (server->priv->authority);
  95     if (authority)
  96         server->priv->authority = g_object_ref (authority);
  97     else
  98         server->priv->authority = NULL;
  99 }
 100
 101 XAuthority *
 102 x_server_get_authority (XServer *server)
 103 {
 104     g_return_val_if_fail (server != NULL, NULL);
 105     return server->priv->authority;
 106 }
 107
 108 static gboolean
 109 x_server_get_can_share (DisplayServer *server)
 110 {
 111     return TRUE;
 112 }
 113
 114 static gboolean
 115 x_server_start (DisplayServer *display_server)
 116 {
 117     XServer *server = X_SERVER (display_server);
 118     xcb_auth_info_t *auth = NULL, a;
 119
 120     if (server->priv->authority)
 121     {
 122         a.namelen = strlen (x_authority_get_authorization_name (server->priv->authority));
 123         a.name = (char *) x_authority_get_authorization_name (server->priv->authority);
 124         a.datalen = x_authority_get_authorization_data_length (server->priv->authority);
 125         a.data = (char *) x_authority_get_authorization_data (server->priv->authority);
 126         auth = &a;
 127     }
 128
 129     /* Open connection */
 130     g_debug ("Connecting to XServer %s", x_server_get_address (server));
 131     server->priv->connection = xcb_connect_to_display_with_auth_info (x_server_get_address (server), auth, NULL);
 132     if (xcb_connection_has_error (server->priv->connection))
 133     {
 134         g_debug ("Error connecting to XServer %s", x_server_get_address (server));
 135         return FALSE;
 136     }
 137
 138     return DISPLAY_SERVER_CLASS (x_server_parent_class)->start (display_server);
 139 }
 140
 141 static void
 142 x_server_connect_session (DisplayServer *display_server, Session *session)
 143 {
 144     gint vt;
 145
 146     display_server = session_get_display_server (session);
 147
 148     vt = display_server_get_vt (display_server);
 149     if (vt > 0)
 150     {
 151         gchar *t;
 152
 153         t = g_strdup_printf ("/dev/tty%d", vt);
 154         session_set_tty (session, t);
 155         g_free (t);
 156
 157         t = g_strdup_printf ("%d", vt);
 158         session_set_env (session, "XDG_VTNR", t);
 159         g_free (t);
 160     }
 161
 162     session_set_env (session, "DISPLAY", x_server_get_address (X_SERVER (display_server)));
 163     session_set_tty (session, x_server_get_address (X_SERVER (display_server)));
 164     session_set_xdisplay (session, x_server_get_address (X_SERVER (display_server)));
 165     session_set_remote_host_name (session, x_server_get_hostname (X_SERVER (display_server)));
 166     session_set_x_authority (session,
 167                              x_server_get_authority (X_SERVER (display_server)),
 168                              config_get_boolean (config_get_instance (), "LightDM", "user-authority-in-system-dir"));
 169 }
 170
 171 static void
 172 x_server_disconnect_session (DisplayServer *display_server, Session *session)
 173 {
 174     session_set_tty (session, NULL);
 175     session_unset_env (session, "XDG_VTNR");
 176     session_unset_env (session, "DISPLAY");
 177     session_set_xdisplay (session, NULL);
 178     session_set_remote_host_name (session, NULL);
 179     session_set_x_authority (session, NULL, FALSE);
 180 }
 181
 182 void
 183 x_server_init (XServer *server)
 184 {
 185     server->priv = G_TYPE_INSTANCE_GET_PRIVATE (server, X_SERVER_TYPE, XServerPrivate);
 186 }
 187
 188 static void
 189 x_server_finalize (GObject *object)
 190 {
 191     XServer *self;
 192
 193     self = X_SERVER (object);
 194
 195     g_free (self->priv->hostname);
 196     g_free (self->priv->address);
 197     if (self->priv->authority)
 198         g_object_unref (self->priv->authority);
 199     if (self->priv->connection)
 200         xcb_disconnect (self->priv->connection);
 201
 202     G_OBJECT_CLASS (x_server_parent_class)->finalize (object);
 203 }
 204
 205 static void
 206 x_server_class_init (XServerClass *klass)
 207 {
 208     GObjectClass *object_class = G_OBJECT_CLASS (klass);
 209     DisplayServerClass *display_server_class = DISPLAY_SERVER_CLASS (klass);
 210
 211     display_server_class->get_can_share = x_server_get_can_share;
 212     display_server_class->start = x_server_start;
 213     display_server_class->connect_session = x_server_connect_session;
 214     display_server_class->disconnect_session = x_server_disconnect_session;
 215     object_class->finalize = x_server_finalize;
 216
 217     g_type_class_add_private (klass, sizeof (XServerPrivate));
 218 }