src/greeter.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
  14 #include <stdlib.h>
  15 #include <string.h>
  16 #include <errno.h>
  17 #include <fcntl.h>
  18 #include <gcrypt.h>
  19
  20 #include "greeter.h"
  21 #include "configuration.h"
  22
  23 enum {
  24     CONNECTED,
  25     CREATE_SESSION,
  26     START_SESSION,
  27     LAST_SIGNAL
  28 };
  29 static guint signals[LAST_SIGNAL] = { 0 };
  30
  31 struct GreeterPrivate
  32 {
  33     /* PAM service to authenticate with */
  34     gchar *pam_service;
  35     gchar *autologin_pam_service;
  36
  37     /* Buffer for data read from greeter */
  38     guint8 *read_buffer;
  39     gsize n_read;
  40     gboolean use_secure_memory;
  41
  42     /* Hints for the greeter */
  43     GHashTable *hints;
  44
  45     /* Default session to use */
  46     gchar *default_session;
  47
  48     /* Sequence number of current PAM session */
  49     guint32 authentication_sequence_number;
  50
  51     /* Remote session name */
  52     gchar *remote_session;
  53
  54     /* PAM session being constructed by the greeter */
  55     Session *authentication_session;
  56
  57     /* TRUE if a user has been authenticated and the session requested to start */
  58     gboolean start_session;
  59
  60     /* TRUE if can log into guest accounts */
  61     gboolean allow_guest;
  62
  63     /* TRUE if logging into guest session */
  64     gboolean guest_account_authenticated;
  65
  66     /* Communication channels to communicate with */
  67     GIOChannel *to_greeter_channel;
  68     GIOChannel *from_greeter_channel;
  69 };
  70
  71 G_DEFINE_TYPE (Greeter, greeter, SESSION_TYPE);
  72
  73 /* Messages from the greeter to the server */
  74 typedef enum
  75 {
  76     GREETER_MESSAGE_CONNECT = 0,
  77     GREETER_MESSAGE_AUTHENTICATE,
  78     GREETER_MESSAGE_AUTHENTICATE_AS_GUEST,
  79     GREETER_MESSAGE_CONTINUE_AUTHENTICATION,
  80     GREETER_MESSAGE_START_SESSION,
  81     GREETER_MESSAGE_CANCEL_AUTHENTICATION,
  82     GREETER_MESSAGE_SET_LANGUAGE,
  83     GREETER_MESSAGE_AUTHENTICATE_REMOTE
  84 } GreeterMessage;
  85
  86 /* Messages from the server to the greeter */
  87 typedef enum
  88 {
  89     SERVER_MESSAGE_CONNECTED = 0,
  90     SERVER_MESSAGE_PROMPT_AUTHENTICATION,
  91     SERVER_MESSAGE_END_AUTHENTICATION,
  92     SERVER_MESSAGE_SESSION_RESULT
  93 } ServerMessage;
  94
  95 static gboolean read_cb (GIOChannel *source, GIOCondition condition, gpointer data);
  96
  97 void
  98 greeter_set_pam_services (Greeter *greeter, const gchar *pam_service, const gchar *autologin_pam_service)
  99 {
 100     g_free (greeter->priv->pam_service);
 101     greeter->priv->pam_service = g_strdup (pam_service);
 102     g_free (greeter->priv->autologin_pam_service);
 103     greeter->priv->autologin_pam_service = g_strdup (autologin_pam_service);
 104 }
 105
 106 void
 107 greeter_set_allow_guest (Greeter *greeter, gboolean allow_guest)
 108 {
 109     greeter->priv->allow_guest = allow_guest;
 110 }
 111
 112 void
 113 greeter_set_hint (Greeter *greeter, const gchar *name, const gchar *value)
 114 {
 115     g_hash_table_insert (greeter->priv->hints, g_strdup (name), g_strdup (value));
 116 }
 117
 118 static void *
 119 secure_malloc (Greeter *greeter, size_t n)
 120 {
 121     if (greeter->priv->use_secure_memory)
 122         return gcry_malloc_secure (n);
 123     else
 124         return g_malloc (n);
 125 }
 126
 127 static void *
 128 secure_realloc (Greeter *greeter, void *ptr, size_t n)
 129 {
 130     if (greeter->priv->use_secure_memory)
 131         return gcry_realloc (ptr, n);
 132     else
 133         return g_realloc (ptr, n);
 134 }
 135
 136 static void
 137 secure_free (Greeter *greeter, void *ptr)
 138 {
 139     if (greeter->priv->use_secure_memory)
 140         return gcry_free (ptr);
 141     else
 142         return g_free (ptr);
 143 }
 144
 145 static guint32
 146 int_length (void)
 147 {
 148     return 4;
 149 }
 150
 151 #define HEADER_SIZE (sizeof (guint32) * 2)
 152 #define MAX_MESSAGE_LENGTH 1024
 153
 154 static void
 155 write_message (Greeter *greeter, guint8 *message, gsize message_length)
 156 {
 157     GError *error = NULL;
 158
 159     g_io_channel_write_chars (greeter->priv->to_greeter_channel, (gchar *) message, message_length, NULL, &error);
 160     if (error)
 161         g_warning ("Error writing to greeter: %s", error->message);
 162     g_clear_error (&error);
 163     g_io_channel_flush (greeter->priv->to_greeter_channel, NULL);
 164 }
 165
 166 static void
 167 write_int (guint8 *buffer, gint buffer_length, guint32 value, gsize *offset)
 168 {
 169     if (*offset + 4 >= buffer_length)
 170         return;
 171     buffer[*offset] = value >> 24;
 172     buffer[*offset+1] = (value >> 16) & 0xFF;
 173     buffer[*offset+2] = (value >> 8) & 0xFF;
 174     buffer[*offset+3] = value & 0xFF;
 175     *offset += 4;
 176 }
 177
 178 static void
 179 write_string (guint8 *buffer, gint buffer_length, const gchar *value, gsize *offset)
 180 {
 181     gint length;
 182
 183     if (value)
 184         length = strlen (value);
 185     else
 186         length = 0;
 187     write_int (buffer, buffer_length, length, offset);
 188     if (*offset + length >= buffer_length)
 189         return;
 190     if (length > 0)
 191     {
 192         memcpy (buffer + *offset, value, length);
 193         *offset += length;
 194     }
 195 }
 196
 197 static void
 198 write_header (guint8 *buffer, gint buffer_length, guint32 id, guint32 length, gsize *offset)
 199 {
 200     write_int (buffer, buffer_length, id, offset);
 201     write_int (buffer, buffer_length, length, offset);
 202 }
 203
 204 static guint32
 205 string_length (const gchar *value)
 206 {
 207     if (value == NULL)
 208         return int_length ();
 209     else
 210         return int_length () + strlen (value);
 211 }
 212
 213 static void
 214 handle_connect (Greeter *greeter, const gchar *version)
 215 {
 216     guint8 message[MAX_MESSAGE_LENGTH];
 217     gsize offset = 0;
 218     guint32 length;
 219     GHashTableIter iter;
 220     gpointer key, value;
 221
 222     g_debug ("Greeter connected version=%s", version);
 223
 224     length = string_length (VERSION);
 225     g_hash_table_iter_init (&iter, greeter->priv->hints);
 226     while (g_hash_table_iter_next (&iter, &key, &value))
 227         length += string_length (key) + string_length (value);
 228
 229     write_header (message, MAX_MESSAGE_LENGTH, SERVER_MESSAGE_CONNECTED, length, &offset);
 230     write_string (message, MAX_MESSAGE_LENGTH, VERSION, &offset);
 231     g_hash_table_iter_init (&iter, greeter->priv->hints);
 232     while (g_hash_table_iter_next (&iter, &key, &value))
 233     {
 234         write_string (message, MAX_MESSAGE_LENGTH, key, &offset);
 235         write_string (message, MAX_MESSAGE_LENGTH, value, &offset);
 236     }
 237     write_message (greeter, message, offset);
 238
 239     g_signal_emit (greeter, signals[CONNECTED], 0);
 240 }
 241
 242 static void
 243 pam_messages_cb (Session *session, Greeter *greeter)
 244 {
 245     int i;
 246     guint32 size;
 247     guint8 message[MAX_MESSAGE_LENGTH];
 248     const struct pam_message *messages;
 249     int messages_length;
 250     gsize offset = 0;
 251     int n_prompts = 0;
 252
 253     messages = session_get_messages (session);
 254     messages_length = session_get_messages_length (session);
 255
 256     /* Respond to d-bus query with messages */
 257     g_debug ("Prompt greeter with %d message(s)", messages_length);
 258     size = int_length () + string_length (session_get_username (session)) + int_length ();
 259     for (i = 0; i < messages_length; i++)
 260         size += int_length () + string_length (messages[i].msg);
 261
 262     write_header (message, MAX_MESSAGE_LENGTH, SERVER_MESSAGE_PROMPT_AUTHENTICATION, size, &offset);
 263     write_int (message, MAX_MESSAGE_LENGTH, greeter->priv->authentication_sequence_number, &offset);
 264     write_string (message, MAX_MESSAGE_LENGTH, session_get_username (session), &offset);
 265     write_int (message, MAX_MESSAGE_LENGTH, messages_length, &offset);
 266     for (i = 0; i < messages_length; i++)
 267     {
 268         write_int (message, MAX_MESSAGE_LENGTH, messages[i].msg_style, &offset);
 269         write_string (message, MAX_MESSAGE_LENGTH, messages[i].msg, &offset);
 270
 271         if (messages[i].msg_style == PAM_PROMPT_ECHO_OFF || messages[i].msg_style == PAM_PROMPT_ECHO_ON)
 272             n_prompts++;
 273     }
 274     write_message (greeter, message, offset);
 275
 276     /* Continue immediately if nothing to respond with */
 277     // FIXME: Should probably give the greeter a chance to ack the message
 278     if (n_prompts == 0)
 279     {
 280         struct pam_response *response;
 281         response = calloc (messages_length, sizeof (struct pam_response));
 282         session_respond (greeter->priv->authentication_session, response);
 283         free (response);
 284     }
 285 }
 286
 287 static void
 288 send_end_authentication (Greeter *greeter, guint32 sequence_number, const gchar *username, int result)
 289 {
 290     guint8 message[MAX_MESSAGE_LENGTH];
 291     gsize offset = 0;
 292
 293     write_header (message, MAX_MESSAGE_LENGTH, SERVER_MESSAGE_END_AUTHENTICATION, int_length () + string_length (username) + int_length (), &offset);
 294     write_int (message, MAX_MESSAGE_LENGTH, sequence_number, &offset);
 295     write_string (message, MAX_MESSAGE_LENGTH, username, &offset);
 296     write_int (message, MAX_MESSAGE_LENGTH, result, &offset);
 297     write_message (greeter, message, offset);
 298 }
 299
 300 static void
 301 authentication_complete_cb (Session *session, Greeter *greeter)
 302 {
 303     int result;
 304
 305     g_debug ("Authenticate result for user %s: %s", session_get_username (session), session_get_authentication_result_string (session));
 306
 307     result = session_get_authentication_result (session);
 308     if (session_get_is_authenticated (session))
 309     {
 310         if (session_get_user (session))
 311             g_debug ("User %s authorized", session_get_username (session));
 312         else
 313         {
 314             g_debug ("User %s authorized, but no account of that name exists", session_get_username (session));
 315             result = PAM_USER_UNKNOWN;
 316         }
 317     }
 318
 319     send_end_authentication (greeter, greeter->priv->authentication_sequence_number, session_get_username (session), result);
 320 }
 321
 322 static void
 323 reset_session (Greeter *greeter)
 324 {
 325     g_free (greeter->priv->remote_session);
 326     greeter->priv->remote_session = NULL;
 327     if (greeter->priv->authentication_session)
 328     {
 329         g_signal_handlers_disconnect_matched (greeter->priv->authentication_session, G_SIGNAL_MATCH_DATA, 0, 0, NULL, NULL, greeter);
 330         session_stop (greeter->priv->authentication_session);
 331         g_object_unref (greeter->priv->authentication_session);
 332         greeter->priv->authentication_session = NULL;
 333     }
 334
 335     greeter->priv->guest_account_authenticated = FALSE;
 336 }
 337
 338 static void
 339 handle_login (Greeter *greeter, guint32 sequence_number, const gchar *username)
 340 {
 341     const gchar *autologin_username, *service;
 342     gboolean is_interactive;
 343
 344     if (username[0] == '\0')
 345     {
 346         g_debug ("Greeter start authentication");
 347         username = NULL;
 348     }
 349     else
 350         g_debug ("Greeter start authentication for %s", username);
 351
 352     reset_session (greeter);
 353
 354     greeter->priv->authentication_sequence_number = sequence_number;
 355     g_signal_emit (greeter, signals[CREATE_SESSION], 0, &greeter->priv->authentication_session);
 356     if (!greeter->priv->authentication_session)
 357     {
 358         send_end_authentication (greeter, sequence_number, "", PAM_USER_UNKNOWN);
 359         return;
 360     }
 361
 362     g_signal_connect (G_OBJECT (greeter->priv->authentication_session), "got-messages", G_CALLBACK (pam_messages_cb), greeter);
 363     g_signal_connect (G_OBJECT (greeter->priv->authentication_session), "authentication-complete", G_CALLBACK (authentication_complete_cb), greeter);
 364
 365     /* Use non-interactive service for autologin user */
 366     autologin_username = g_hash_table_lookup (greeter->priv->hints, "autologin-user");
 367     if (autologin_username != NULL && g_strcmp0 (username, autologin_username) == 0)
 368     {
 369         service = greeter->priv->autologin_pam_service;
 370         is_interactive = FALSE;
 371     }
 372     else
 373     {
 374         service = greeter->priv->pam_service;
 375         is_interactive = TRUE;
 376     }
 377
 378     /* Run the session process */
 379     session_set_pam_service (greeter->priv->authentication_session, service);
 380     session_set_username (greeter->priv->authentication_session, username);
 381     session_set_do_authenticate (greeter->priv->authentication_session, TRUE);
 382     session_set_is_interactive (greeter->priv->authentication_session, is_interactive);
 383     session_start (greeter->priv->authentication_session);
 384 }
 385
 386 static void
 387 handle_login_as_guest (Greeter *greeter, guint32 sequence_number)
 388 {
 389     g_debug ("Greeter start authentication for guest account");
 390
 391     reset_session (greeter);
 392
 393     if (!greeter->priv->allow_guest)
 394     {
 395         g_debug ("Guest account is disabled");
 396         send_end_authentication (greeter, sequence_number, "", PAM_USER_UNKNOWN);
 397         return;
 398     }
 399
 400     greeter->priv->guest_account_authenticated = TRUE;
 401     send_end_authentication (greeter, sequence_number, "", PAM_SUCCESS);
 402 }
 403
 404 static gchar *
 405 get_remote_session_service (const gchar *session_name)
 406 {
 407     GKeyFile *session_desktop_file;
 408     gboolean result;
 409     const gchar *c;
 410     gchar *remote_sessions_dir, *filename, *path, *service = NULL;
 411     GError *error = NULL;
 412
 413     /* Validate session name doesn't contain directory separators */
 414     for (c = session_name; *c; c++)
 415     {
 416         if (*c == '/')
 417             return NULL;
 418     }
 419
 420     /* Load the session file */
 421     session_desktop_file = g_key_file_new ();
 422     filename = g_strdup_printf ("%s.desktop", session_name);
 423     remote_sessions_dir = config_get_string (config_get_instance (), "LightDM", "remote-sessions-directory");
 424     path = g_build_filename (remote_sessions_dir, filename, NULL);
 425     g_free (remote_sessions_dir);
 426     g_free (filename);
 427     result = g_key_file_load_from_file (session_desktop_file, path, G_KEY_FILE_NONE, &error);
 428     if (error)
 429         g_debug ("Failed to load session file %s: %s", path, error->message);
 430     g_free (path);
 431     g_clear_error (&error);
 432     if (result)
 433         service = g_key_file_get_string (session_desktop_file, G_KEY_FILE_DESKTOP_GROUP, "X-LightDM-PAM-Service", NULL);
 434     g_key_file_free (session_desktop_file);
 435
 436     return service;
 437 }
 438
 439 static void
 440 handle_login_remote (Greeter *greeter, const gchar *session_name, const gchar *username, guint32 sequence_number)
 441 {
 442     gchar *service;
 443
 444     if (username[0] == '\0')
 445     {
 446         g_debug ("Greeter start authentication for remote session %s", session_name);
 447         username = NULL;
 448     }
 449     else
 450         g_debug ("Greeter start authentication for remote session %s as user %s", session_name, username);
 451
 452     reset_session (greeter);
 453
 454     service = get_remote_session_service (session_name);
 455     if (!service)
 456     {
 457         send_end_authentication (greeter, sequence_number, "", PAM_SYSTEM_ERR);
 458         return;
 459     }
 460
 461     greeter->priv->authentication_sequence_number = sequence_number;
 462     greeter->priv->remote_session = g_strdup (session_name);
 463     g_signal_emit (greeter, signals[CREATE_SESSION], 0, &greeter->priv->authentication_session);
 464     if (greeter->priv->authentication_session)
 465     {
 466         g_signal_connect (G_OBJECT (greeter->priv->authentication_session), "got-messages", G_CALLBACK (pam_messages_cb), greeter);
 467         g_signal_connect (G_OBJECT (greeter->priv->authentication_session), "authentication-complete", G_CALLBACK (authentication_complete_cb), greeter);
 468
 469         /* Run the session process */
 470         session_set_pam_service (greeter->priv->authentication_session, service);
 471         session_set_username (greeter->priv->authentication_session, username);
 472         session_set_do_authenticate (greeter->priv->authentication_session, TRUE);
 473         session_set_is_interactive (greeter->priv->authentication_session, TRUE);
 474         session_set_is_guest (greeter->priv->authentication_session, TRUE);
 475         session_start (greeter->priv->authentication_session);
 476     }
 477
 478     g_free (service);
 479
 480     if (!greeter->priv->authentication_session)
 481     {
 482         send_end_authentication (greeter, sequence_number, "", PAM_USER_UNKNOWN);
 483         return;
 484     }
 485 }
 486
 487 static void
 488 handle_continue_authentication (Greeter *greeter, gchar **secrets)
 489 {
 490     int messages_length;
 491     const struct pam_message *messages;
 492     struct pam_response *response;
 493     int i, j, n_prompts = 0;
 494
 495     /* Not in authentication */
 496     if (greeter->priv->authentication_session == NULL)
 497         return;
 498
 499     messages_length = session_get_messages_length (greeter->priv->authentication_session);
 500     messages = session_get_messages (greeter->priv->authentication_session);
 501
 502     /* Check correct number of responses */
 503     for (i = 0; i < messages_length; i++)
 504     {
 505         int msg_style = messages[i].msg_style;
 506         if (msg_style == PAM_PROMPT_ECHO_OFF || msg_style == PAM_PROMPT_ECHO_ON)
 507             n_prompts++;
 508     }
 509     if (g_strv_length (secrets) != n_prompts)
 510     {
 511         session_respond_error (greeter->priv->authentication_session, PAM_CONV_ERR);
 512         return;
 513     }
 514
 515     g_debug ("Continue authentication");
 516
 517     /* Build response */
 518     response = calloc (messages_length, sizeof (struct pam_response));
 519     for (i = 0, j = 0; i < messages_length; i++)
 520     {
 521         int msg_style = messages[i].msg_style;
 522         if (msg_style == PAM_PROMPT_ECHO_OFF || msg_style == PAM_PROMPT_ECHO_ON)
 523         {
 524             size_t secret_length = strlen (secrets[j]) + 1;
 525             response[i].resp = secure_malloc (greeter, secret_length);
 526             memcpy (response[i].resp, secrets[j], secret_length); // FIXME: Need to convert from UTF-8
 527             j++;
 528         }
 529     }
 530
 531     session_respond (greeter->priv->authentication_session, response);
 532
 533     for (i = 0; i < messages_length; i++)
 534         secure_free (greeter, response[i].resp);
 535     free (response);
 536 }
 537
 538 static void
 539 handle_cancel_authentication (Greeter *greeter)
 540 {
 541     /* Not in authentication */
 542     if (greeter->priv->authentication_session == NULL)
 543         return;
 544
 545     g_debug ("Cancel authentication");
 546     reset_session (greeter);
 547 }
 548
 549 static void
 550 handle_start_session (Greeter *greeter, const gchar *session)
 551 {
 552     gboolean result;
 553     guint8 message[MAX_MESSAGE_LENGTH];
 554     gsize offset = 0;
 555     SessionType session_type = SESSION_TYPE_LOCAL;
 556
 557     if (strcmp (session, "") == 0)
 558         session = NULL;
 559
 560     /* Use session type chosen in remote session */
 561     if (greeter->priv->remote_session)
 562     {
 563         session_type = SESSION_TYPE_REMOTE;
 564         session = greeter->priv->remote_session;
 565     }
 566
 567     if (greeter->priv->guest_account_authenticated || session_get_is_authenticated (greeter->priv->authentication_session))
 568     {
 569         if (session)
 570             g_debug ("Greeter requests session %s", session);
 571         else
 572             g_debug ("Greeter requests default session");
 573         greeter->priv->start_session = TRUE;
 574         g_signal_emit (greeter, signals[START_SESSION], 0, session_type, session, &result);
 575     }
 576     else
 577     {
 578         g_debug ("Ignoring start session request, user is not authorized");
 579         result = FALSE;
 580     }
 581
 582     write_header (message, MAX_MESSAGE_LENGTH, SERVER_MESSAGE_SESSION_RESULT, int_length (), &offset);
 583     write_int (message, MAX_MESSAGE_LENGTH, result ? 0 : 1, &offset);
 584     write_message (greeter, message, offset);
 585 }
 586
 587 static void
 588 handle_set_language (Greeter *greeter, const gchar *language)
 589 {
 590     User *user;
 591
 592     if (!greeter->priv->guest_account_authenticated && !session_get_is_authenticated (greeter->priv->authentication_session))
 593     {
 594         g_debug ("Ignoring set language request, user is not authorized");
 595         return;
 596     }
 597
 598     // FIXME: Could use this
 599     if (greeter->priv->guest_account_authenticated)
 600     {
 601         g_debug ("Ignoring set language request for guest user");
 602         return;
 603     }
 604
 605     g_debug ("Greeter sets language %s", language);
 606     user = session_get_user (greeter->priv->authentication_session);
 607     user_set_language (user, language);
 608 }
 609
 610 static guint32
 611 read_int (Greeter *greeter, gsize *offset)
 612 {
 613     guint32 value;
 614     guint8 *buffer;
 615     if (greeter->priv->n_read - *offset < sizeof (guint32))
 616     {
 617         g_warning ("Not enough space for int, need %zu, got %zu", sizeof (guint32), greeter->priv->n_read - *offset);
 618         return 0;
 619     }
 620     buffer = greeter->priv->read_buffer + *offset;
 621     value = buffer[0] << 24 | buffer[1] << 16 | buffer[2] << 8 | buffer[3];
 622     *offset += int_length ();
 623     return value;
 624 }
 625
 626 static int
 627 get_message_length (Greeter *greeter)
 628 {
 629     gsize offset;
 630     int payload_length;
 631
 632     offset = int_length ();
 633     payload_length = read_int (greeter, &offset);
 634
 635     if (HEADER_SIZE + payload_length < HEADER_SIZE)
 636     {
 637         g_warning ("Payload length of %u octets too long", payload_length);
 638         return HEADER_SIZE;
 639     }
 640
 641     return HEADER_SIZE + payload_length;
 642 }
 643
 644 static gchar *
 645 read_string_full (Greeter *greeter, gsize *offset, void* (*alloc_fn)(size_t n))
 646 {
 647     guint32 length;
 648     gchar *value;
 649
 650     length = read_int (greeter, offset);
 651     if (greeter->priv->n_read - *offset < length)
 652     {
 653         g_warning ("Not enough space for string, need %u, got %zu", length, greeter->priv->n_read - *offset);
 654         return g_strdup ("");
 655     }
 656
 657     value = (*alloc_fn) (sizeof (gchar *) * (length + 1));
 658     memcpy (value, greeter->priv->read_buffer + *offset, length);
 659     value[length] = '\0';
 660     *offset += length;
 661
 662     return value;
 663 }
 664
 665 static gchar *
 666 read_string (Greeter *greeter, gsize *offset)
 667 {
 668     return read_string_full (greeter, offset, g_malloc);
 669 }
 670
 671 static gchar *
 672 read_secret (Greeter *greeter, gsize *offset)
 673 {
 674     if (greeter->priv->use_secure_memory)
 675         return read_string_full (greeter, offset, gcry_malloc_secure);
 676     else
 677         return read_string_full (greeter, offset, g_malloc);
 678 }
 679
 680 static gboolean
 681 read_cb (GIOChannel *source, GIOCondition condition, gpointer data)
 682 {
 683     Greeter *greeter = data;
 684     gsize n_to_read, n_read, offset;
 685     GIOStatus status;
 686     int id, i;
 687     guint32 sequence_number, n_secrets, max_secrets;
 688     gchar *version, *username, *session_name, *language;
 689     gchar **secrets;
 690     GError *error = NULL;
 691
 692     if (condition == G_IO_HUP)
 693     {
 694         g_debug ("Greeter closed communication channel");
 695         return FALSE;
 696     }
 697
 698     n_to_read = HEADER_SIZE;
 699     if (greeter->priv->n_read >= HEADER_SIZE)
 700     {
 701         n_to_read = get_message_length (greeter);
 702         if (n_to_read <= HEADER_SIZE)
 703             return FALSE;
 704     }
 705
 706     status = g_io_channel_read_chars (greeter->priv->from_greeter_channel,
 707                                       (gchar *) greeter->priv->read_buffer + greeter->priv->n_read,
 708                                       n_to_read - greeter->priv->n_read,
 709                                       &n_read,
 710                                       &error);
 711     if (error)
 712         g_warning ("Error reading from greeter: %s", error->message);
 713     g_clear_error (&error);
 714     if (status != G_IO_STATUS_NORMAL)
 715         return TRUE;
 716
 717     greeter->priv->n_read += n_read;
 718     if (greeter->priv->n_read != n_to_read)
 719         return TRUE;
 720
 721     /* If have header, rerun for content */
 722     if (greeter->priv->n_read == HEADER_SIZE)
 723     {
 724         n_to_read = get_message_length (greeter);
 725         if (n_to_read > HEADER_SIZE)
 726         {
 727             greeter->priv->read_buffer = secure_realloc (greeter, greeter->priv->read_buffer, n_to_read);
 728             read_cb (source, condition, greeter);
 729             return TRUE;
 730         }
 731     }
 732
 733     offset = 0;
 734     id = read_int (greeter, &offset);
 735     read_int (greeter, &offset);
 736     switch (id)
 737     {
 738     case GREETER_MESSAGE_CONNECT:
 739         version = read_string (greeter, &offset);
 740         handle_connect (greeter, version);
 741         g_free (version);
 742         break;
 743     case GREETER_MESSAGE_AUTHENTICATE:
 744         sequence_number = read_int (greeter, &offset);
 745         username = read_string (greeter, &offset);
 746         handle_login (greeter, sequence_number, username);
 747         g_free (username);
 748         break;
 749     case GREETER_MESSAGE_AUTHENTICATE_AS_GUEST:
 750         sequence_number = read_int (greeter, &offset);
 751         handle_login_as_guest (greeter, sequence_number);
 752         break;
 753     case GREETER_MESSAGE_AUTHENTICATE_REMOTE:
 754         sequence_number = read_int (greeter, &offset);
 755         session_name = read_string (greeter, &offset);
 756         username = read_string (greeter, &offset);
 757         handle_login_remote (greeter, session_name, username, sequence_number);
 758         break;
 759     case GREETER_MESSAGE_CONTINUE_AUTHENTICATION:
 760         n_secrets = read_int (greeter, &offset);
 761         max_secrets = (G_MAXUINT32 - 1) / sizeof (gchar *);
 762         if (n_secrets > max_secrets)
 763         {
 764             g_warning ("Array length of %u elements too long", n_secrets);
 765             return FALSE;
 766         }
 767         secrets = g_malloc (sizeof (gchar *) * (n_secrets + 1));
 768         for (i = 0; i < n_secrets; i++)
 769             secrets[i] = read_secret (greeter, &offset);
 770         secrets[i] = NULL;
 771         handle_continue_authentication (greeter, secrets);
 772         for (i = 0; i < n_secrets; i++)
 773             secure_free (greeter, secrets[i]);
 774         g_free (secrets);
 775         break;
 776     case GREETER_MESSAGE_CANCEL_AUTHENTICATION:
 777         handle_cancel_authentication (greeter);
 778         break;
 779     case GREETER_MESSAGE_START_SESSION:
 780         session_name = read_string (greeter, &offset);
 781         handle_start_session (greeter, session_name);
 782         g_free (session_name);
 783         break;
 784     case GREETER_MESSAGE_SET_LANGUAGE:
 785         language = read_string (greeter, &offset);
 786         handle_set_language (greeter, language);
 787         g_free (language);
 788         break;
 789     default:
 790         g_warning ("Unknown message from greeter: %d", id);
 791         break;
 792     }
 793
 794     greeter->priv->n_read = 0;
 795
 796     return TRUE;
 797 }
 798
 799 gboolean
 800 greeter_get_guest_authenticated (Greeter *greeter)
 801 {
 802     g_return_val_if_fail (greeter != NULL, FALSE);
 803     return greeter->priv->guest_account_authenticated;
 804 }
 805
 806 Session *
 807 greeter_get_authentication_session (Greeter *greeter)
 808 {
 809     g_return_val_if_fail (greeter != NULL, NULL);
 810     return greeter->priv->authentication_session;
 811 }
 812
 813 gboolean
 814 greeter_get_start_session (Greeter *greeter)
 815 {
 816     g_return_val_if_fail (greeter != NULL, FALSE);
 817     return greeter->priv->start_session;
 818 }
 819
 820 static gboolean
 821 greeter_start (Session *session)
 822 {
 823     Greeter *greeter = GREETER (session);
 824     int to_greeter_pipe[2], from_greeter_pipe[2];
 825     gboolean result = FALSE;
 826     gchar *value;
 827
 828     /* Create a pipe to talk with the greeter */
 829     if (pipe (to_greeter_pipe) != 0 || pipe (from_greeter_pipe) != 0)
 830     {
 831         g_warning ("Failed to create pipes: %s", strerror (errno));
 832         return FALSE;
 833     }
 834     greeter->priv->to_greeter_channel = g_io_channel_unix_new (to_greeter_pipe[1]);
 835     g_io_channel_set_encoding (greeter->priv->to_greeter_channel, NULL, NULL);
 836     greeter->priv->from_greeter_channel = g_io_channel_unix_new (from_greeter_pipe[0]);
 837     g_io_channel_set_encoding (greeter->priv->from_greeter_channel, NULL, NULL);
 838     g_io_channel_set_buffered (greeter->priv->from_greeter_channel, FALSE);
 839     g_io_add_watch (greeter->priv->from_greeter_channel, G_IO_IN | G_IO_HUP, read_cb, greeter);
 840
 841     /* Let the greeter session know how to communicate with the daemon */
 842     value = g_strdup_printf ("%d", from_greeter_pipe[1]);
 843     session_set_env (SESSION (greeter), "LIGHTDM_TO_SERVER_FD", value);
 844     g_free (value);
 845     value = g_strdup_printf ("%d", to_greeter_pipe[0]);
 846     session_set_env (SESSION (greeter), "LIGHTDM_FROM_SERVER_FD", value);
 847     g_free (value);
 848
 849     /* Don't allow the daemon end of the pipes to be accessed in child processes */
 850     fcntl (to_greeter_pipe[1], F_SETFD, FD_CLOEXEC);
 851     fcntl (from_greeter_pipe[0], F_SETFD, FD_CLOEXEC);
 852
 853     result = SESSION_CLASS (greeter_parent_class)->start (session);
 854
 855     /* Close the session ends of the pipe */
 856     close (to_greeter_pipe[0]);
 857     close (from_greeter_pipe[1]);
 858
 859     return result;
 860 }
 861
 862 static Session *
 863 greeter_real_create_session (Greeter *greeter)
 864 {
 865     return NULL;
 866 }
 867
 868 static gboolean
 869 greeter_real_start_session (Greeter *greeter, SessionType type, const gchar *session)
 870 {
 871     return FALSE;
 872 }
 873
 874 static void
 875 greeter_stop (Session *session)
 876 {
 877     Greeter *greeter = GREETER (session);
 878
 879     /* Stop any events occurring after we've stopped */
 880     if (greeter->priv->authentication_session)
 881         g_signal_handlers_disconnect_matched (greeter->priv->authentication_session, G_SIGNAL_MATCH_DATA, 0, 0, NULL, NULL, greeter);
 882
 883     SESSION_CLASS (greeter_parent_class)->stop (session);
 884 }
 885
 886 static void
 887 greeter_init (Greeter *greeter)
 888 {
 889     greeter->priv = G_TYPE_INSTANCE_GET_PRIVATE (greeter, GREETER_TYPE, GreeterPrivate);
 890     greeter->priv->read_buffer = secure_malloc (greeter, HEADER_SIZE);
 891     greeter->priv->hints = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, g_free);
 892     greeter->priv->use_secure_memory = config_get_boolean (config_get_instance (), "LightDM", "lock-memory");
 893 }
 894
 895 static void
 896 greeter_finalize (GObject *object)
 897 {
 898     Greeter *self;
 899
 900     self = GREETER (object);
 901
 902     g_free (self->priv->pam_service);
 903     g_free (self->priv->autologin_pam_service);
 904     secure_free (self, self->priv->read_buffer);
 905     g_hash_table_unref (self->priv->hints);
 906     g_free (self->priv->remote_session);
 907     if (self->priv->authentication_session)
 908     {
 909         g_signal_handlers_disconnect_matched (self->priv->authentication_session, G_SIGNAL_MATCH_DATA, 0, 0, NULL, NULL, self);
 910         g_object_unref (self->priv->authentication_session);
 911     }
 912     if (self->priv->to_greeter_channel)
 913         g_io_channel_unref (self->priv->to_greeter_channel);
 914     if (self->priv->from_greeter_channel)
 915         g_io_channel_unref (self->priv->from_greeter_channel);
 916
 917     G_OBJECT_CLASS (greeter_parent_class)->finalize (object);
 918 }
 919
 920 static void
 921 greeter_class_init (GreeterClass *klass)
 922 {
 923     SessionClass *session_class = SESSION_CLASS (klass);
 924     GObjectClass *object_class = G_OBJECT_CLASS (klass);
 925
 926     klass->create_session = greeter_real_create_session;
 927     klass->start_session = greeter_real_start_session;
 928     session_class->start = greeter_start;
 929     session_class->stop = greeter_stop;
 930     object_class->finalize = greeter_finalize;
 931
 932     signals[CONNECTED] =
 933         g_signal_new ("connected",
 934                       G_TYPE_FROM_CLASS (klass),
 935                       G_SIGNAL_RUN_LAST,
 936                       G_STRUCT_OFFSET (GreeterClass, connected),
 937                       NULL, NULL,
 938                       NULL,
 939                       G_TYPE_NONE, 0);
 940
 941     signals[CREATE_SESSION] =
 942         g_signal_new ("create-session",
 943                       G_TYPE_FROM_CLASS (klass),
 944                       G_SIGNAL_RUN_LAST,
 945                       G_STRUCT_OFFSET (GreeterClass, create_session),
 946                       g_signal_accumulator_first_wins,
 947                       NULL,
 948                       NULL,
 949                       SESSION_TYPE, 0);
 950
 951     signals[START_SESSION] =
 952         g_signal_new ("start-session",
 953                       G_TYPE_FROM_CLASS (klass),
 954                       G_SIGNAL_RUN_LAST,
 955                       G_STRUCT_OFFSET (GreeterClass, start_session),
 956                       g_signal_accumulator_true_handled,
 957                       NULL,
 958                       NULL,
 959                       G_TYPE_BOOLEAN, 2, G_TYPE_INT, G_TYPE_STRING);
 960
 961     g_type_class_add_private (klass, sizeof (GreeterPrivate));
 962 }