#include <errno.h>

#include <string.h>

#include "logind.h"

#include "logind-session.h"

#include "dbus-common.h"

#include "util.h"

#define BUS_SESSION_INTERFACE \

" <interface name=\"org.freedesktop.login1.Session\">\n" \

" <method name=\"Terminate\"/>\n" \

" <method name=\"Activate\"/>\n" \

" <method name=\"Lock\"/>\n" \

" <method name=\"Unlock\"/>\n" \

" <method name=\"SetIdleHint\">\n" \

" <arg name=\"b\" type=\"b\"/>\n" \

" </method>\n" \

" <method name=\"Kill\">\n" \

" <arg name=\"who\" type=\"s\"/>\n" \

" <arg name=\"signal\" type=\"s\"/>\n" \

" </method>\n" \

" <property name=\"Id\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"User\" type=\"(uo)\" access=\"read\"/>\n" \

" <property name=\"Name\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"Timestamp\" type=\"t\" access=\"read\"/>\n" \

" <property name=\"TimestampMonotonic\" type=\"t\" access=\"read\"/>\n" \

" <property name=\"ControlGroupPath\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"VTNr\" type=\"u\" access=\"read\"/>\n" \

" <property name=\"Seat\" type=\"(so)\" access=\"read\"/>\n" \

" <property name=\"TTY\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"Display\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"Remote\" type=\"b\" access=\"read\"/>\n" \

" <property name=\"RemoteHost\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"RemoteUser\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"Service\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"Leader\" type=\"u\" access=\"read\"/>\n" \

" <property name=\"Audit\" type=\"u\" access=\"read\"/>\n" \

" <property name=\"Type\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"Active\" type=\"b\" access=\"read\"/>\n" \

" <property name=\"Controllers\" type=\"as\" access=\"read\"/>\n" \

" <property name=\"ResetControllers\" type=\"as\" access=\"read\"/>\n" \

" <property name=\"KillProcesses\" type=\"b\" access=\"read\"/>\n" \

" <property name=\"IdleHint\" type=\"b\" access=\"read\"/>\n" \

" <property name=\"IdleSinceHint\" type=\"t\" access=\"read\"/>\n" \

" <property name=\"IdleSinceHintMonotonic\" type=\"t\" access=\"read\"/>\n" \

" </interface>\n"

#define INTROSPECTION \

DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE \

"<node>\n" \

BUS_SESSION_INTERFACE \

BUS_PROPERTIES_INTERFACE \

BUS_PEER_INTERFACE \

BUS_INTROSPECTABLE_INTERFACE \

"</node>\n"

#define INTERFACES_LIST \

BUS_GENERIC_INTERFACES_LIST \

"org.freedesktop.login1.Session\0"

static int bus_session_append_seat(DBusMessageIter *i, const char *property, void *data) {

DBusMessageIter sub;

Session *s = data;

const char *id, *path;

char *p = NULL;

assert(i);

assert(property);

assert(s);

if (!dbus_message_iter_open_container(i, DBUS_TYPE_STRUCT, NULL, &sub))

return -ENOMEM;

if (s->seat) {

id = s->seat->id;

path = p = seat_bus_path(s->seat);

if (!p)

return -ENOMEM;

} else {

id = "";

path = "/";

}

if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, &id) ||

!dbus_message_iter_append_basic(&sub, DBUS_TYPE_OBJECT_PATH, &path)) {

free(p);

return -ENOMEM;

}

free(p);

if (!dbus_message_iter_close_container(i, &sub))

return -ENOMEM;

return 0;

}

static int bus_session_append_user(DBusMessageIter *i, const char *property, void *data) {

DBusMessageIter sub;

User *u = data;

char *p = NULL;

assert(i);

assert(property);

assert(u);

if (!dbus_message_iter_open_container(i, DBUS_TYPE_STRUCT, NULL, &sub))

return -ENOMEM;

p = user_bus_path(u);

if (!p)

return -ENOMEM;

if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_UINT32, &u->uid) ||

!dbus_message_iter_append_basic(&sub, DBUS_TYPE_OBJECT_PATH, &p)) {

free(p);

return -ENOMEM;

}

free(p);

if (!dbus_message_iter_close_container(i, &sub))

return -ENOMEM;

return 0;

}

static int bus_session_append_active(DBusMessageIter *i, const char *property, void *data) {

Session *s = data;

dbus_bool_t b;

assert(i);

assert(property);

assert(s);

b = session_is_active(s);

if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b))

return -ENOMEM;

return 0;

}

static int bus_session_append_idle_hint(DBusMessageIter *i, const char *property, void *data) {

Session *s = data;

int b;

assert(i);

assert(property);

assert(s);

b = session_get_idle_hint(s, NULL) > 0;

if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b))

return -ENOMEM;

return 0;

}

static int bus_session_append_idle_hint_since(DBusMessageIter *i, const char *property, void *data) {

Session *s = data;

dual_timestamp t;

uint64_t u;

assert(i);

assert(property);

assert(s);

session_get_idle_hint(s, &t);

u = streq(property, "IdleSinceHint") ? t.realtime : t.monotonic;

if (!dbus_message_iter_append_basic(i, DBUS_TYPE_UINT64, &u))

return -ENOMEM;

return 0;

}

static DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_session_append_type, session_type, SessionType);

static int get_session_for_path(Manager *m, const char *path, Session **_s) {

Session *s;

char *id;

assert(m);

assert(path);

assert(_s);

if (!startswith(path, "/org/freedesktop/login1/session/"))

return -EINVAL;

id = bus_path_unescape(path + 32);

if (!id)

return -ENOMEM;

s = hashmap_get(m->sessions, id);

free(id);

if (!s)

return -ENOENT;

*_s = s;

return 0;

}

static const BusProperty bus_login_session_properties[] = {

{ "Id", bus_property_append_string, "s", offsetof(Session, id), true },

{ "Timestamp", bus_property_append_usec, "t", offsetof(Session, timestamp.realtime) },

{ "TimestampMonotonic", bus_property_append_usec, "t", offsetof(Session, timestamp.monotonic) },

{ "ControlGroupPath", bus_property_append_string, "s", offsetof(Session, cgroup_path), true },

{ "VTNr", bus_property_append_uint32, "u", offsetof(Session, vtnr) },

{ "Seat", bus_session_append_seat, "(so)", 0 },

{ "TTY", bus_property_append_string, "s", offsetof(Session, tty), true },

{ "Display", bus_property_append_string, "s", offsetof(Session, display), true },

{ "Remote", bus_property_append_bool, "b", offsetof(Session, remote) },

{ "RemoteUser", bus_property_append_string, "s", offsetof(Session, remote_user), true },

{ "RemoteHost", bus_property_append_string, "s", offsetof(Session, remote_host), true },

{ "Service", bus_property_append_string, "s", offsetof(Session, service), true },

{ "Leader", bus_property_append_pid, "u", offsetof(Session, leader) },

{ "Audit", bus_property_append_uint32, "u", offsetof(Session, audit_id) },

{ "Type", bus_session_append_type, "s", offsetof(Session, type) },

{ "Active", bus_session_append_active, "b", 0 },

{ "Controllers", bus_property_append_strv, "as", offsetof(Session, controllers), true },

{ "ResetControllers", bus_property_append_strv, "as", offsetof(Session, reset_controllers), true },

{ "KillProcesses", bus_property_append_bool, "b", offsetof(Session, kill_processes) },

{ "IdleHint", bus_session_append_idle_hint, "b", 0 },

{ "IdleSinceHint", bus_session_append_idle_hint_since, "t", 0 },

{ "IdleSinceHintMonotonic", bus_session_append_idle_hint_since, "t", 0 },

{ NULL, }

};

static const BusProperty bus_login_session_user_properties[] = {

{ "User", bus_session_append_user, "(uo)", 0 },

{ "Name", bus_property_append_string, "s", offsetof(User, name), true },

};

static DBusHandlerResult session_message_dispatch(

Session *s,

DBusConnection *connection,

DBusMessage *message) {

DBusError error;

DBusMessage *reply = NULL;

int r;

assert(s);

assert(connection);

assert(message);

dbus_error_init(&error);

if (dbus_message_is_method_call(message, "org.freedesktop.login1.Session", "Terminate")) {

r = session_stop(s);

if (r < 0)

return bus_send_error_reply(connection, message, NULL, r);

reply = dbus_message_new_method_return(message);

if (!reply)

goto oom;

} else if (dbus_message_is_method_call(message, "org.freedesktop.login1.Session", "Activate")) {

r = session_activate(s);

if (r < 0)

return bus_send_error_reply(connection, message, NULL, r);

reply = dbus_message_new_method_return(message);

if (!reply)

goto oom;

} else if (dbus_message_is_method_call(message, "org.freedesktop.login1.Session", "Lock") ||

dbus_message_is_method_call(message, "org.freedesktop.login1.Session", "Unlock")) {

if (session_send_lock(s, streq(dbus_message_get_member(message), "Lock")) < 0)

goto oom;

reply = dbus_message_new_method_return(message);

if (!reply)

goto oom;

} else if (dbus_message_is_method_call(message, "org.freedesktop.login1.Session", "SetIdleHint")) {

dbus_bool_t b;

unsigned long ul;

if (!dbus_message_get_args(

message,

&error,

DBUS_TYPE_BOOLEAN, &b,

DBUS_TYPE_INVALID))

return bus_send_error_reply(connection, message, &error, -EINVAL);

ul = dbus_bus_get_unix_user(connection, dbus_message_get_sender(message), &error);

if (ul == (unsigned long) -1)

return bus_send_error_reply(connection, message, &error, -EIO);

if (ul != 0 && ul != s->user->uid)

return bus_send_error_reply(connection, message, NULL, -EPERM);

session_set_idle_hint(s, b);

reply = dbus_message_new_method_return(message);

if (!reply)

goto oom;

} else if (dbus_message_is_method_call(message, "org.freedesktop.login1.Session", "Kill")) {

const char *swho;

int32_t signo;

KillWho who;

if (!dbus_message_get_args(

message,

&error,

DBUS_TYPE_STRING, &swho,

DBUS_TYPE_INT32, &signo,

DBUS_TYPE_INVALID))

return bus_send_error_reply(connection, message, &error, -EINVAL);

if (isempty(swho))

who = KILL_ALL;

else {

who = kill_who_from_string(swho);

if (who < 0)

return bus_send_error_reply(connection, message, &error, -EINVAL);

}

if (signo <= 0 || signo >= _NSIG)

return bus_send_error_reply(connection, message, &error, -EINVAL);

r = session_kill(s, who, signo);

if (r < 0)

return bus_send_error_reply(connection, message, NULL, r);

reply = dbus_message_new_method_return(message);

if (!reply)

goto oom;

} else {

const BusBoundProperties bps[] = {

{ "org.freedesktop.login1.Session", bus_login_session_properties, s },

{ "org.freedesktop.login1.Session", bus_login_session_user_properties, s->user },

{ NULL, }

};

return bus_default_message_handler(connection, message, INTROSPECTION, INTERFACES_LIST, bps);

}

if (reply) {

if (!dbus_connection_send(connection, reply, NULL))

goto oom;

dbus_message_unref(reply);

}

return DBUS_HANDLER_RESULT_HANDLED;

oom:

if (reply)

dbus_message_unref(reply);

dbus_error_free(&error);

return DBUS_HANDLER_RESULT_NEED_MEMORY;

}

static DBusHandlerResult session_message_handler(

DBusConnection *connection,

DBusMessage *message,

void *userdata) {

Manager *m = userdata;

Session *s;

int r;

r = get_session_for_path(m, dbus_message_get_path(message), &s);

if (r < 0) {

if (r == -ENOMEM)

return DBUS_HANDLER_RESULT_NEED_MEMORY;

if (r == -ENOENT) {

DBusError e;

dbus_error_init(&e);

dbus_set_error_const(&e, DBUS_ERROR_UNKNOWN_OBJECT, "Unknown session");

return bus_send_error_reply(connection, message, &e, r);

}

return bus_send_error_reply(connection, message, NULL, r);

}

return session_message_dispatch(s, connection, message);

}

const DBusObjectPathVTable bus_session_vtable = {

.message_function = session_message_handler

};

char *session_bus_path(Session *s) {

char *t, *r;

assert(s);

t = bus_path_escape(s->id);

if (!t)

return NULL;

r = strappend("/org/freedesktop/login1/session/", t);

free(t);

return r;

}

int session_send_signal(Session *s, bool new_session) {

DBusMessage *m;

int r = -ENOMEM;

char *p = NULL;

assert(s);

m = dbus_message_new_signal("/org/freedesktop/login1",

"org.freedesktop.login1.Manager",

new_session ? "SessionNew" : "SessionRemoved");

if (!m)

return -ENOMEM;

p = session_bus_path(s);

if (!p)

goto finish;

if (!dbus_message_append_args(

m,

DBUS_TYPE_STRING, &s->id,

DBUS_TYPE_OBJECT_PATH, &p,

DBUS_TYPE_INVALID))

goto finish;

if (!dbus_connection_send(s->manager->bus, m, NULL))

goto finish;

r = 0;

finish:

dbus_message_unref(m);

free(p);

return r;

}

int session_send_changed(Session *s, const char *properties) {

DBusMessage *m;

int r = -ENOMEM;

char *p = NULL;

assert(s);

if (!s->started)

return 0;

p = session_bus_path(s);

if (!p)

return -ENOMEM;

m = bus_properties_changed_new(p, "org.freedesktop.login1.Session", properties);

if (!m)

goto finish;

if (!dbus_connection_send(s->manager->bus, m, NULL))

goto finish;

r = 0;

finish:

if (m)

dbus_message_unref(m);

free(p);

return r;

}

int session_send_lock(Session *s, bool lock) {

DBusMessage *m;

bool b;

char *p;

assert(s);

p = session_bus_path(s);

if (!p)

return -ENOMEM;

m = dbus_message_new_signal(p, "org.freedesktop.login1.Session", lock ? "Lock" : "Unlock");

free(p);

if (!m)

return -ENOMEM;

b = dbus_connection_send(s->manager->bus, m, NULL);

dbus_message_unref(m);

if (!b)

return -ENOMEM;

return 0;

}