#include <errno.h>

#include "dbus.h"

#include "log.h"

#include "dbus-job.h"

#include "dbus-common.h"

#define BUS_JOB_INTERFACE \

" <interface name=\"org.freedesktop.systemd1.Job\">\n" \

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

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

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

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

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

" </interface>\n"

#define INTROSPECTION \

DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE \

"<node>\n" \

BUS_JOB_INTERFACE \

BUS_PROPERTIES_INTERFACE \

BUS_PEER_INTERFACE \

BUS_INTROSPECTABLE_INTERFACE \

"</node>\n"

const char bus_job_interface[] _introspect_("Job") = BUS_JOB_INTERFACE;

#define INTERFACES_LIST \

BUS_GENERIC_INTERFACES_LIST \

"org.freedesktop.systemd1.Job\0"

#define INVALIDATING_PROPERTIES \

"State\0"

static DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_job_append_state, job_state, JobState);

static DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_job_append_type, job_type, JobType);

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

Job *j = data;

DBusMessageIter sub;

char *p;

assert(i);

assert(property);

assert(j);

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

return -ENOMEM;

if (!(p = unit_dbus_path(j->unit)))

return -ENOMEM;

if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, &j->unit->id) ||

!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 const BusProperty bus_job_properties[] = {

{ "Id", bus_property_append_uint32, "u", offsetof(Job, id) },

{ "State", bus_job_append_state, "s", offsetof(Job, state) },

{ "JobType", bus_job_append_type, "s", offsetof(Job, type) },

{ "Unit", bus_job_append_unit, "(so)", 0 },

{ NULL, }

};

static DBusHandlerResult bus_job_message_dispatch(Job *j, DBusConnection *connection, DBusMessage *message) {

DBusMessage *reply = NULL;

if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Job", "Cancel")) {

if (!(reply = dbus_message_new_method_return(message)))

goto oom;

job_finish_and_invalidate(j, JOB_CANCELED, true);

} else {

const BusBoundProperties bps[] = {

{ "org.freedesktop.systemd1.Job", bus_job_properties, j },

{ 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);

return DBUS_HANDLER_RESULT_NEED_MEMORY;

}

static DBusHandlerResult bus_job_message_handler(DBusConnection *connection, DBusMessage *message, void *data) {

Manager *m = data;

Job *j;

int r;

DBusMessage *reply;

assert(connection);

assert(message);

assert(m);

if (streq(dbus_message_get_path(message), "/org/freedesktop/systemd1/job")) {

/* Be nice to gdbus and return introspection data for our mid-level paths */

if (dbus_message_is_method_call(message, "org.freedesktop.DBus.Introspectable", "Introspect")) {

char *introspection = NULL;

FILE *f;

Iterator i;

size_t size;

if (!(reply = dbus_message_new_method_return(message)))

goto oom;

/* We roll our own introspection code here, instead of

if (!(f = open_memstream(&introspection, &size)))

goto oom;

fputs(DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE

"<node>\n", f);

fputs(BUS_INTROSPECTABLE_INTERFACE, f);

fputs(BUS_PEER_INTERFACE, f);

HASHMAP_FOREACH(j, m->jobs, i)

fprintf(f, "<node name=\"job/%lu\"/>", (unsigned long) j->id);

fputs("</node>\n", f);

if (ferror(f)) {

fclose(f);

free(introspection);

goto oom;

}

fclose(f);

if (!introspection)

goto oom;

if (!dbus_message_append_args(reply, DBUS_TYPE_STRING, &introspection, DBUS_TYPE_INVALID)) {

free(introspection);

goto oom;

}

free(introspection);

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

goto oom;

dbus_message_unref(reply);

return DBUS_HANDLER_RESULT_HANDLED;

}

return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;

}

if ((r = manager_get_job_from_dbus_path(m, dbus_message_get_path(message), &j)) < 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 job");

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

}

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

}

return bus_job_message_dispatch(j, connection, message);

oom:

if (reply)

dbus_message_unref(reply);

return DBUS_HANDLER_RESULT_NEED_MEMORY;

}

const DBusObjectPathVTable bus_job_vtable = {

.message_function = bus_job_message_handler

};

static int job_send_message(Job *j, DBusMessage* (*new_message)(Job *j)) {

DBusMessage *m = NULL;

int r;

assert(j);

assert(new_message);

if (bus_has_subscriber(j->manager) || j->forgot_bus_clients) {

m = new_message(j);

if (!m)

goto oom;

r = bus_broadcast(j->manager, m);

dbus_message_unref(m);

if (r < 0)

return r;

} else {

/* If nobody is subscribed, we just send the message

JobBusClient *cl;

assert(j->bus_client_list);

LIST_FOREACH(client, cl, j->bus_client_list) {

assert(cl->bus);

m = new_message(j);

if (!m)

goto oom;

if (!dbus_message_set_destination(m, cl->name))

goto oom;

if (!dbus_connection_send(cl->bus, m, NULL))

goto oom;

dbus_message_unref(m);

m = NULL;

}

}

return 0;

oom:

if (m)

dbus_message_unref(m);

return -ENOMEM;

}

static DBusMessage* new_change_signal_message(Job *j) {

DBusMessage *m = NULL;

char *p = NULL;

p = job_dbus_path(j);

if (!p)

goto oom;

if (j->sent_dbus_new_signal) {

/* Send a properties changed signal */

m = bus_properties_changed_new(p, "org.freedesktop.systemd1.Job", INVALIDATING_PROPERTIES);

if (!m)

goto oom;

} else {

/* Send a new signal */

m = dbus_message_new_signal("/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", "JobNew");

if (!m)

goto oom;

if (!dbus_message_append_args(m,

DBUS_TYPE_UINT32, &j->id,

DBUS_TYPE_OBJECT_PATH, &p,

DBUS_TYPE_INVALID))

goto oom;

}

return m;

oom:

if (m)

dbus_message_unref(m);

free(p);

return NULL;

}

static DBusMessage* new_removed_signal_message(Job *j) {

DBusMessage *m = NULL;

char *p = NULL;

const char *r;

p = job_dbus_path(j);

if (!p)

goto oom;

m = dbus_message_new_signal("/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", "JobRemoved");

if (!m)

goto oom;

r = job_result_to_string(j->result);

if (!dbus_message_append_args(m,

DBUS_TYPE_UINT32, &j->id,

DBUS_TYPE_OBJECT_PATH, &p,

DBUS_TYPE_STRING, &r,

DBUS_TYPE_INVALID))

goto oom;

return m;

oom:

if (m)

dbus_message_unref(m);

free(p);

return NULL;

}

void bus_job_send_change_signal(Job *j) {

assert(j);

if (j->in_dbus_queue) {

LIST_REMOVE(Job, dbus_queue, j->manager->dbus_job_queue, j);

j->in_dbus_queue = false;

}

if (!bus_has_subscriber(j->manager) && !j->bus_client_list && !j->forgot_bus_clients) {

j->sent_dbus_new_signal = true;

return;

}

if (job_send_message(j, new_change_signal_message) < 0)

goto oom;

j->sent_dbus_new_signal = true;

return;

oom:

log_error("Failed to allocate job change signal.");

}

void bus_job_send_removed_signal(Job *j) {

assert(j);

if (!bus_has_subscriber(j->manager) && !j->bus_client_list && !j->forgot_bus_clients)

return;

if (!j->sent_dbus_new_signal)

bus_job_send_change_signal(j);

if (job_send_message(j, new_removed_signal_message) < 0)

goto oom;

return;

oom:

log_error("Failed to allocate job remove signal.");

}