]> rtime.felk.cvut.cz Git - sojka/lightdm.git/blob - src/x-server.c
Correctly set XDG_VTNR for unity sessions that are not autologin.
[sojka/lightdm.git] / 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 const gchar *
109 x_server_get_session_type (DisplayServer *server)
110 {
111     return "x";
112 }
113
114 static gboolean
115 x_server_get_can_share (DisplayServer *server)
116 {
117     return TRUE;
118 }
119
120 static gboolean
121 x_server_start (DisplayServer *display_server)
122 {
123     XServer *server = X_SERVER (display_server);
124     xcb_auth_info_t *auth = NULL, a;
125
126     if (server->priv->authority)
127     {
128         a.namelen = strlen (x_authority_get_authorization_name (server->priv->authority));
129         a.name = (char *) x_authority_get_authorization_name (server->priv->authority);
130         a.datalen = x_authority_get_authorization_data_length (server->priv->authority);
131         a.data = (char *) x_authority_get_authorization_data (server->priv->authority);
132         auth = &a;
133     }
134
135     /* Open connection */  
136     g_debug ("Connecting to XServer %s", x_server_get_address (server));
137     server->priv->connection = xcb_connect_to_display_with_auth_info (x_server_get_address (server), auth, NULL);
138     if (xcb_connection_has_error (server->priv->connection))
139     {
140         g_debug ("Error connecting to XServer %s", x_server_get_address (server));
141         return FALSE;
142     }
143
144     return DISPLAY_SERVER_CLASS (x_server_parent_class)->start (display_server);
145 }
146
147 static void
148 x_server_connect_session (DisplayServer *display_server, Session *session)
149 {
150     gint vt;
151
152     display_server = session_get_display_server (session);
153
154     vt = display_server_get_vt (display_server);
155     if (vt > 0)
156     {
157         gchar *t;
158
159         t = g_strdup_printf ("/dev/tty%d", vt);
160         session_set_tty (session, t);
161         g_free (t);
162
163         t = g_strdup_printf ("%d", vt);
164         session_set_env (session, "XDG_VTNR", t);
165         g_free (t);
166     }
167
168     session_set_env (session, "DISPLAY", x_server_get_address (X_SERVER (display_server)));
169     session_set_tty (session, x_server_get_address (X_SERVER (display_server)));
170     session_set_xdisplay (session, x_server_get_address (X_SERVER (display_server)));
171     session_set_remote_host_name (session, x_server_get_hostname (X_SERVER (display_server)));
172     session_set_x_authority (session,
173                              x_server_get_authority (X_SERVER (display_server)),
174                              config_get_boolean (config_get_instance (), "LightDM", "user-authority-in-system-dir"));
175 }
176
177 static void
178 x_server_disconnect_session (DisplayServer *display_server, Session *session)
179 {
180     gint vt;
181
182     vt = display_server_get_vt (display_server);
183     if (vt > 0)
184     {
185         session_set_tty (session, NULL);
186         session_unset_env (session, "XDG_VTNR");
187     }
188     session_unset_env (session, "DISPLAY");
189     session_set_xdisplay (session, NULL);
190     session_set_remote_host_name (session, NULL);
191     session_set_x_authority (session, NULL, FALSE);
192 }
193
194 void
195 x_server_init (XServer *server)
196 {
197     server->priv = G_TYPE_INSTANCE_GET_PRIVATE (server, X_SERVER_TYPE, XServerPrivate);
198 }
199
200 static void
201 x_server_finalize (GObject *object)
202 {
203     XServer *self;
204
205     self = X_SERVER (object);
206
207     g_free (self->priv->hostname);
208     g_free (self->priv->address);
209     if (self->priv->authority)
210         g_object_unref (self->priv->authority);
211     if (self->priv->connection)
212         xcb_disconnect (self->priv->connection);
213
214     G_OBJECT_CLASS (x_server_parent_class)->finalize (object);
215 }
216
217 static void
218 x_server_class_init (XServerClass *klass)
219 {
220     GObjectClass *object_class = G_OBJECT_CLASS (klass);
221     DisplayServerClass *display_server_class = DISPLAY_SERVER_CLASS (klass);
222
223     display_server_class->get_session_type = x_server_get_session_type;    
224     display_server_class->get_can_share = x_server_get_can_share;
225     display_server_class->start = x_server_start;
226     display_server_class->connect_session = x_server_connect_session;
227     display_server_class->disconnect_session = x_server_disconnect_session;
228     object_class->finalize = x_server_finalize;
229
230     g_type_class_add_private (klass, sizeof (XServerPrivate));
231 }