#include <errno.h>

#include <string.h>

#include "logind.h"

#include "logind-user.h"

#include "dbus-common.h"

#define BUS_USER_INTERFACE \

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

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

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

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

" </method>\n" \

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

" <property name=\"GID\" type=\"u\" 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=\"RuntimePath\" type=\"s\" access=\"read\"/>\n" \

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

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

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

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

" <property name=\"Sessions\" type=\"a(so)\" 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_USER_INTERFACE \

BUS_PROPERTIES_INTERFACE \

BUS_PEER_INTERFACE \

BUS_INTROSPECTABLE_INTERFACE \

"</node>\n"

#define INTERFACES_LIST \

BUS_GENERIC_INTERFACES_LIST \

"org.freedesktop.login1.User\0"

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

DBusMessageIter sub;

User *u = data;

const char *id, *path;

char *p = NULL;

assert(i);

assert(property);

assert(u);

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

return -ENOMEM;

if (u->display) {

id = u->display->id;

path = p = session_bus_path(u->display);

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_user_append_state(DBusMessageIter *i, const char *property, void *data) {

User *u = data;

const char *state;

assert(i);

assert(property);

assert(u);

state = user_state_to_string(user_get_state(u));

if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &state))

return -ENOMEM;

return 0;

}

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

DBusMessageIter sub, sub2;

User *u = data;

Session *session;

assert(i);

assert(property);

assert(u);

if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "(so)", &sub))

return -ENOMEM;

LIST_FOREACH(sessions_by_user, session, u->sessions) {

char *p;

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

return -ENOMEM;

p = session_bus_path(session);

if (!p)

return -ENOMEM;

if (!dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &session->id) ||

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

free(p);

return -ENOMEM;

}

free(p);

if (!dbus_message_iter_close_container(&sub, &sub2))

return -ENOMEM;

}

if (!dbus_message_iter_close_container(i, &sub))

return -ENOMEM;

return 0;

}

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

User *u = data;

dbus_bool_t b;

assert(i);

assert(property);

assert(u);

b = user_get_idle_hint(u, NULL) > 0;

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

return -ENOMEM;

return 0;

}

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

User *u = data;

dual_timestamp t;

uint64_t k;

assert(i);

assert(property);

assert(u);

user_get_idle_hint(u, &t);

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

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

return -ENOMEM;

return 0;

}

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

User *u = data;

char *t;

int r;

bool success;

assert(i);

assert(property);

assert(u);

r = cg_join_spec(SYSTEMD_CGROUP_CONTROLLER, u->cgroup_path, &t);

if (r < 0)

return r;

success = dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &t);

free(t);

return success ? 0 : -ENOMEM;

}

static int get_user_for_path(Manager *m, const char *path, User **_u) {

User *u;

unsigned long lu;

int r;

assert(m);

assert(path);

assert(_u);

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

return -EINVAL;

r = safe_atolu(path + 29, &lu);

if (r < 0)

return r;

u = hashmap_get(m->users, ULONG_TO_PTR(lu));

if (!u)

return -ENOENT;

*_u = u;

return 0;

}

static const BusProperty bus_login_user_properties[] = {

{ "UID", bus_property_append_uid, "u", offsetof(User, uid) },

{ "GID", bus_property_append_gid, "u", offsetof(User, gid) },

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

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

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

{ "RuntimePath", bus_property_append_string, "s", offsetof(User, runtime_path), true },

{ "DefaultControlGroup", bus_user_append_default_cgroup, "s", 0 },

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

{ "Display", bus_user_append_display, "(so)", 0 },

{ "State", bus_user_append_state, "s", 0 },

{ "Sessions", bus_user_append_sessions, "a(so)", 0 },

{ "IdleHint", bus_user_append_idle_hint, "b", 0 },

{ "IdleSinceHint", bus_user_append_idle_hint_since, "t", 0 },

{ "IdleSinceHintMonotonic", bus_user_append_idle_hint_since, "t", 0 },

{ NULL, }

};

static DBusHandlerResult user_message_dispatch(

User *u,

DBusConnection *connection,

DBusMessage *message) {

DBusError error;

DBusMessage *reply = NULL;

int r;

assert(u);

assert(connection);

assert(message);

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

r = user_stop(u);

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.User", "Kill")) {

int32_t signo;

if (!dbus_message_get_args(

message,

&error,

DBUS_TYPE_INT32, &signo,

DBUS_TYPE_INVALID))

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

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

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

r = user_kill(u, 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.User", bus_login_user_properties, u },

{ 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 user_message_handler(

DBusConnection *connection,

DBusMessage *message,

void *userdata) {

Manager *m = userdata;

User *u;

int r;

r = get_user_for_path(m, dbus_message_get_path(message), &u);

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 user");

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

}

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

}

return user_message_dispatch(u, connection, message);

}

const DBusObjectPathVTable bus_user_vtable = {

.message_function = user_message_handler

};

char *user_bus_path(User *u) {

char *s;

assert(u);

if (asprintf(&s, "/org/freedesktop/login1/user/%llu", (unsigned long long) u->uid) < 0)

return NULL;

return s;

}

int user_send_signal(User *u, bool new_user) {

DBusMessage *m;

int r = -ENOMEM;

char *p = NULL;

uint32_t uid;

assert(u);

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

"org.freedesktop.login1.Manager",

new_user ? "UserNew" : "UserRemoved");

if (!m)

return -ENOMEM;

p = user_bus_path(u);

if (!p)

goto finish;

uid = u->uid;

if (!dbus_message_append_args(

m,

DBUS_TYPE_UINT32, &uid,

DBUS_TYPE_OBJECT_PATH, &p,

DBUS_TYPE_INVALID))

goto finish;

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

goto finish;

r = 0;

finish:

dbus_message_unref(m);

free(p);

return r;

}

int user_send_changed(User *u, const char *properties) {

DBusMessage *m;

int r = -ENOMEM;

char *p = NULL;

assert(u);

if (!u->started)

return 0;

p = user_bus_path(u);

if (!p)

return -ENOMEM;

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

if (!m)

goto finish;

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

goto finish;

r = 0;

finish:

if (m)

dbus_message_unref(m);

free(p);

return r;

}