[thirdparty/systemd.git] / src / dbus-unit.c

/*-*- Mode: C; c-basic-offset: 8 -*-*/

/***
  This file is part of systemd.

  Copyright 2010 Lennart Poettering

  systemd is free software; you can redistribute it and/or modify it
  under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  systemd is distributed in the hope that it will be useful, but
  WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with systemd; If not, see <http://www.gnu.org/licenses/>.
***/

#include <errno.h>

#include "dbus.h"
#include "log.h"
#include "dbus-unit.h"
#include "bus-errors.h"

const char bus_unit_interface[] = BUS_UNIT_INTERFACE;

int bus_unit_append_names(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        char *t;
        Iterator j;
        DBusMessageIter sub;
        Unit *u = data;

        if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "s", &sub))
                return -ENOMEM;

        SET_FOREACH(t, u->meta.names, j)
                if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, &t))
                        return -ENOMEM;

        if (!dbus_message_iter_close_container(i, &sub))
                return -ENOMEM;

        return 0;
}

int bus_unit_append_following(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        Unit *u = data, *f;
        const char *d;

        assert(m);
        assert(i);
        assert(property);
        assert(u);

        f = unit_following(u);
        d = f ? f->meta.id : "";

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &d))
                return -ENOMEM;

        return 0;
}

int bus_unit_append_dependencies(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        Unit *u;
        Iterator j;
        DBusMessageIter sub;
        Set *s = data;

        if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "s", &sub))
                return -ENOMEM;

        SET_FOREACH(u, s, j)
                if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, &u->meta.id))
                        return -ENOMEM;

        if (!dbus_message_iter_close_container(i, &sub))
                return -ENOMEM;

        return 0;
}

int bus_unit_append_description(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        Unit *u = data;
        const char *d;

        assert(m);
        assert(i);
        assert(property);
        assert(u);

        d = unit_description(u);

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &d))
                return -ENOMEM;

        return 0;
}

DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_unit_append_load_state, unit_load_state, UnitLoadState);

int bus_unit_append_active_state(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        Unit *u = data;
        const char *state;

        assert(m);
        assert(i);
        assert(property);
        assert(u);

        state = unit_active_state_to_string(unit_active_state(u));

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &state))
                return -ENOMEM;

        return 0;
}

int bus_unit_append_sub_state(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        Unit *u = data;
        const char *state;

        assert(m);
        assert(i);
        assert(property);
        assert(u);

        state = unit_sub_state_to_string(u);

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &state))
                return -ENOMEM;

        return 0;
}

int bus_unit_append_can_start(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        Unit *u = data;
        dbus_bool_t b;

        assert(m);
        assert(i);
        assert(property);
        assert(u);

        b = unit_can_start(u) &&
                !u->meta.only_by_dependency;

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b))
                return -ENOMEM;

        return 0;
}

int bus_unit_append_can_reload(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        Unit *u = data;
        dbus_bool_t b;

        assert(m);
        assert(i);
        assert(property);
        assert(u);

        b = unit_can_reload(u);

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b))
                return -ENOMEM;

        return 0;
}

int bus_unit_append_job(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        Unit *u = data;
        DBusMessageIter sub;
        char *p;

        assert(m);
        assert(i);
        assert(property);
        assert(u);

        if (!dbus_message_iter_open_container(i, DBUS_TYPE_STRUCT, NULL, &sub))
                return -ENOMEM;

        if (u->meta.job) {

                if (!(p = job_dbus_path(u->meta.job)))
                        return -ENOMEM;

                if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_UINT32, &u->meta.job->id) ||
                    !dbus_message_iter_append_basic(&sub, DBUS_TYPE_OBJECT_PATH, &p)) {
                        free(p);
                        return -ENOMEM;
                }
        } else {
                uint32_t id = 0;

                /* No job, so let's fill in some placeholder
                 * data. Since we need to fill in a valid path we
                 * simple point to ourselves. */

                if (!(p = unit_dbus_path(u)))
                        return -ENOMEM;

                if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_UINT32, &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;
}

int bus_unit_append_default_cgroup(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        Unit *u = data;
        char *t;
        CGroupBonding *cgb;
        bool success;

        assert(m);
        assert(i);
        assert(property);
        assert(u);

        if ((cgb = unit_get_default_cgroup(u))) {
                if (!(t = cgroup_bonding_to_string(cgb)))
                        return -ENOMEM;
        } else
                t = (char*) "";

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

        if (cgb)
                free(t);

        return success ? 0 : -ENOMEM;
}

int bus_unit_append_cgroups(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        Unit *u = data;
        CGroupBonding *cgb;
        DBusMessageIter sub;

        if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "s", &sub))
                return -ENOMEM;

        LIST_FOREACH(by_unit, cgb, u->meta.cgroup_bondings) {
                char *t;
                bool success;

                if (!(t = cgroup_bonding_to_string(cgb)))
                        return -ENOMEM;

                success = dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, &t);
                free(t);

                if (!success)
                        return -ENOMEM;
        }

        if (!dbus_message_iter_close_container(i, &sub))
                return -ENOMEM;

        return 0;
}

int bus_unit_append_need_daemon_reload(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        Unit *u = data;
        dbus_bool_t b;

        assert(m);
        assert(i);
        assert(property);
        assert(u);

        b = unit_need_daemon_reload(u);

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b))
                return -ENOMEM;

        return 0;
}

static DBusHandlerResult bus_unit_message_dispatch(Unit *u, DBusConnection *connection, DBusMessage *message) {
        DBusMessage *reply = NULL;
        Manager *m = u->meta.manager;
        DBusError error;
        JobType job_type = _JOB_TYPE_INVALID;
        char *path = NULL;
        bool reload_if_possible = false;

        dbus_error_init(&error);

        if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Start"))
                job_type = JOB_START;
        else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Stop"))
                job_type = JOB_STOP;
        else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Reload"))
                job_type = JOB_RELOAD;
        else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Restart"))
                job_type = JOB_RESTART;
        else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "TryRestart"))
                job_type = JOB_TRY_RESTART;
        else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "ReloadOrRestart")) {
                reload_if_possible = true;
                job_type = JOB_RESTART;
        } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "ReloadOrTryRestart")) {
                reload_if_possible = true;
                job_type = JOB_TRY_RESTART;
        } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "ResetMaintenance")) {

                unit_reset_maintenance(u);

                if (!(reply = dbus_message_new_method_return(message)))
                        goto oom;

        } else if (UNIT_VTABLE(u)->bus_message_handler)
                return UNIT_VTABLE(u)->bus_message_handler(u, connection, message);
        else
                return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;

        if (job_type != _JOB_TYPE_INVALID) {
                const char *smode;
                JobMode mode;
                Job *j;
                int r;

                if (job_type == JOB_START && u->meta.only_by_dependency) {
                        dbus_set_error(&error, BUS_ERROR_ONLY_BY_DEPENDENCY, "Unit may be activated by dependency only.");
                        return bus_send_error_reply(m, connection, message, &error, -EPERM);
                }

                if (!dbus_message_get_args(
                                    message,
                                    &error,
                                    DBUS_TYPE_STRING, &smode,
                                    DBUS_TYPE_INVALID))
                        return bus_send_error_reply(m, connection, message, &error, -EINVAL);

                if (reload_if_possible && unit_can_reload(u)) {
                        if (job_type == JOB_RESTART)
                                job_type = JOB_RELOAD_OR_START;
                        else if (job_type == JOB_TRY_RESTART)
                                job_type = JOB_RELOAD;
                }

                if ((mode = job_mode_from_string(smode)) == _JOB_MODE_INVALID) {
                        dbus_set_error(&error, BUS_ERROR_INVALID_JOB_MODE, "Job mode %s is invalid.", smode);
                        return bus_send_error_reply(m, connection, message, &error, -EINVAL);
                }

                if ((r = manager_add_job(m, job_type, u, mode, true, &error, &j)) < 0)
                        return bus_send_error_reply(m, connection, message, &error, r);

                if (!(reply = dbus_message_new_method_return(message)))
                        goto oom;

                if (!(path = job_dbus_path(j)))
                        goto oom;

                if (!dbus_message_append_args(
                                    reply,
                                    DBUS_TYPE_OBJECT_PATH, &path,
                                    DBUS_TYPE_INVALID))
                        goto oom;
        }

        free(path);

        if (reply) {
                if (!dbus_connection_send(connection, reply, NULL))
                        goto oom;

                dbus_message_unref(reply);
        }

        return DBUS_HANDLER_RESULT_HANDLED;

oom:
        free(path);

        if (reply)
                dbus_message_unref(reply);

        dbus_error_free(&error);

        return DBUS_HANDLER_RESULT_NEED_MEMORY;
}

static DBusHandlerResult bus_unit_message_handler(DBusConnection *connection, DBusMessage  *message, void *data) {
        Manager *m = data;
        Unit *u;
        int r;

        assert(connection);
        assert(message);
        assert(m);

        if ((r = manager_get_unit_from_dbus_path(m, dbus_message_get_path(message), &u)) < 0) {

                if (r == -ENOMEM)
                        return DBUS_HANDLER_RESULT_NEED_MEMORY;

                if (r == -ENOENT)
                        return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;

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

        return bus_unit_message_dispatch(u, connection, message);
}

const DBusObjectPathVTable bus_unit_vtable = {
        .message_function = bus_unit_message_handler
};

void bus_unit_send_change_signal(Unit *u) {
        char *p = NULL;
        DBusMessage *m = NULL;

        assert(u);

        if (u->meta.in_dbus_queue) {
                LIST_REMOVE(Meta, dbus_queue, u->meta.manager->dbus_unit_queue, &u->meta);
                u->meta.in_dbus_queue = false;
        }

        if (!bus_has_subscriber(u->meta.manager)) {
                u->meta.sent_dbus_new_signal = true;
                return;
        }

        if (!(p = unit_dbus_path(u)))
                goto oom;

        if (u->meta.sent_dbus_new_signal) {
                /* Send a change signal */

                if (!(m = dbus_message_new_signal(p, "org.freedesktop.systemd1.Unit", "Changed")))
                        goto oom;
        } else {
                /* Send a new signal */

                if (!(m = dbus_message_new_signal("/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", "UnitNew")))
                        goto oom;

                if (!dbus_message_append_args(m,
                                              DBUS_TYPE_STRING, &u->meta.id,
                                              DBUS_TYPE_OBJECT_PATH, &p,
                                              DBUS_TYPE_INVALID))
                        goto oom;
        }

        if (bus_broadcast(u->meta.manager, m) < 0)
                goto oom;

        free(p);
        dbus_message_unref(m);

        u->meta.sent_dbus_new_signal = true;

        return;

oom:
        free(p);

        if (m)
                dbus_message_unref(m);

        log_error("Failed to allocate unit change/new signal.");
}

void bus_unit_send_removed_signal(Unit *u) {
        char *p = NULL;
        DBusMessage *m = NULL;

        assert(u);

        if (!bus_has_subscriber(u->meta.manager))
                return;

        if (!u->meta.sent_dbus_new_signal)
                bus_unit_send_change_signal(u);

        if (!(p = unit_dbus_path(u)))
                goto oom;

        if (!(m = dbus_message_new_signal("/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", "UnitRemoved")))
                goto oom;

        if (!dbus_message_append_args(m,
                                      DBUS_TYPE_STRING, &u->meta.id,
                                      DBUS_TYPE_OBJECT_PATH, &p,
                                      DBUS_TYPE_INVALID))
                goto oom;

        if (bus_broadcast(u->meta.manager, m) < 0)
                goto oom;

        free(p);
        dbus_message_unref(m);

        return;

oom:
        free(p);

        if (m)
                dbus_message_unref(m);

        log_error("Failed to allocate unit remove signal.");
}
Commit	Line	Data
ea430986 LP	1	/-- Mode: C; c-basic-offset: 8 --/
ea430986 LP	2
a7334b09 LP	3	/***
	4	This file is part of systemd.
	5
	6	Copyright 2010 Lennart Poettering
	7
	8	systemd is free software; you can redistribute it and/or modify it
	9	under the terms of the GNU General Public License as published by
	10	the Free Software Foundation; either version 2 of the License, or
	11	(at your option) any later version.
	12
	13	systemd is distributed in the hope that it will be useful, but
	14	WITHOUT ANY WARRANTY; without even the implied warranty of
	15	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	16	General Public License for more details.
	17
	18	You should have received a copy of the GNU General Public License
	19	along with systemd; If not, see <http://www.gnu.org/licenses/>.
	20	***/
	21
ea430986 LP	22	#include <errno.h>
	23
	24	#include "dbus.h"
	25	#include "log.h"
4139c1b2	26	#include "dbus-unit.h"
398ef8ba	27	#include "bus-errors.h"
ea430986	28
4288f619 LP	29	const char bus_unit_interface[] = BUS_UNIT_INTERFACE;
4288f619 LP	30
4139c1b2 LP	31	int bus_unit_append_names(Manager m, DBusMessageIter i, const char property, void data) {
	32	char *t;
	33	Iterator j;
	34	DBusMessageIter sub;
	35	Unit *u = data;
	36
	37	if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "s", &sub))
	38	return -ENOMEM;
	39
	40	SET_FOREACH(t, u->meta.names, j)
	41	if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, &t))
	42	return -ENOMEM;
	43
	44	if (!dbus_message_iter_close_container(i, &sub))
	45	return -ENOMEM;
	46
	47	return 0;
	48	}
	49
8fe914ec	50	int bus_unit_append_following(Manager m, DBusMessageIter i, const char property, void data) {
a7f241db	51	Unit u = data, f;
8fe914ec LP	52	const char *d;
	53
	54	assert(m);
	55	assert(i);
	56	assert(property);
	57	assert(u);
	58
a7f241db LP	59	f = unit_following(u);
a7f241db LP	60	d = f ? f->meta.id : "";
8fe914ec LP	61
	62	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &d))
	63	return -ENOMEM;
	64
	65	return 0;
	66	}
	67
5301be81 LP	68	int bus_unit_append_dependencies(Manager m, DBusMessageIter i, const char property, void data) {
	69	Unit *u;
	70	Iterator j;
	71	DBusMessageIter sub;
	72	Set *s = data;
	73
	74	if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "s", &sub))
	75	return -ENOMEM;
	76
	77	SET_FOREACH(u, s, j)
	78	if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, &u->meta.id))
	79	return -ENOMEM;
	80
	81	if (!dbus_message_iter_close_container(i, &sub))
	82	return -ENOMEM;
	83
	84	return 0;
	85	}
	86
4139c1b2	87	int bus_unit_append_description(Manager m, DBusMessageIter i, const char property, void data) {
ea430986 LP	88	Unit *u = data;
	89	const char *d;
	90
	91	assert(m);
	92	assert(i);
	93	assert(property);
	94	assert(u);
	95
	96	d = unit_description(u);
	97
	98	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &d))
	99	return -ENOMEM;
	100
	101	return 0;
	102	}
	103
6f4706b7	104	DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_unit_append_load_state, unit_load_state, UnitLoadState);
ea430986	105
4139c1b2	106	int bus_unit_append_active_state(Manager m, DBusMessageIter i, const char property, void data) {
ea430986 LP	107	Unit *u = data;
	108	const char *state;
	109
	110	assert(m);
	111	assert(i);
	112	assert(property);
	113	assert(u);
	114
	115	state = unit_active_state_to_string(unit_active_state(u));
	116
	117	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &state))
	118	return -ENOMEM;
	119
	120	return 0;
	121	}
	122
4139c1b2	123	int bus_unit_append_sub_state(Manager m, DBusMessageIter i, const char property, void data) {
10a94420 LP	124	Unit *u = data;
	125	const char *state;
	126
	127	assert(m);
	128	assert(i);
	129	assert(property);
	130	assert(u);
	131
	132	state = unit_sub_state_to_string(u);
	133
	134	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &state))
	135	return -ENOMEM;
	136
	137	return 0;
	138	}
	139
4139c1b2	140	int bus_unit_append_can_start(Manager m, DBusMessageIter i, const char property, void data) {
38131695 LP	141	Unit *u = data;
	142	dbus_bool_t b;
	143
	144	assert(m);
	145	assert(i);
	146	assert(property);
	147	assert(u);
	148
064f51fa LP	149	b = unit_can_start(u) &&
064f51fa LP	150	!u->meta.only_by_dependency;
38131695 LP	151
	152	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b))
	153	return -ENOMEM;
	154
	155	return 0;
	156	}
	157
4139c1b2	158	int bus_unit_append_can_reload(Manager m, DBusMessageIter i, const char property, void data) {
38131695 LP	159	Unit *u = data;
	160	dbus_bool_t b;
	161
	162	assert(m);
	163	assert(i);
	164	assert(property);
	165	assert(u);
	166
4139c1b2	167	b = unit_can_reload(u);
38131695 LP	168
	169	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b))
	170	return -ENOMEM;
	171
	172	return 0;
	173	}
	174
4139c1b2	175	int bus_unit_append_job(Manager m, DBusMessageIter i, const char property, void data) {
38131695 LP	176	Unit *u = data;
	177	DBusMessageIter sub;
	178	char *p;
	179
	180	assert(m);
	181	assert(i);
	182	assert(property);
	183	assert(u);
	184
	185	if (!dbus_message_iter_open_container(i, DBUS_TYPE_STRUCT, NULL, &sub))
	186	return -ENOMEM;
	187
	188	if (u->meta.job) {
	189
	190	if (!(p = job_dbus_path(u->meta.job)))
	191	return -ENOMEM;
	192
	193	if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_UINT32, &u->meta.job->id) \|\|
86ad3bc1	194	!dbus_message_iter_append_basic(&sub, DBUS_TYPE_OBJECT_PATH, &p)) {
38131695 LP	195	free(p);
	196	return -ENOMEM;
	197	}
	198	} else {
	199	uint32_t id = 0;
	200
	201	/* No job, so let's fill in some placeholder
	202	* data. Since we need to fill in a valid path we
	203	* simple point to ourselves. */
	204
	205	if (!(p = unit_dbus_path(u)))
	206	return -ENOMEM;
	207
	208	if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_UINT32, &id) \|\|
86ad3bc1	209	!dbus_message_iter_append_basic(&sub, DBUS_TYPE_OBJECT_PATH, &p)) {
38131695 LP	210	free(p);
	211	return -ENOMEM;
	212	}
	213	}
	214
	215	free(p);
	216
	217	if (!dbus_message_iter_close_container(i, &sub))
	218	return -ENOMEM;
	219
	220	return 0;
	221	}
	222
4139c1b2 LP	223	int bus_unit_append_default_cgroup(Manager m, DBusMessageIter i, const char property, void data) {
	224	Unit *u = data;
	225	char *t;
	226	CGroupBonding *cgb;
	227	bool success;
ea430986	228
4139c1b2 LP	229	assert(m);
	230	assert(i);
	231	assert(property);
	232	assert(u);
	233
	234	if ((cgb = unit_get_default_cgroup(u))) {
	235	if (!(t = cgroup_bonding_to_string(cgb)))
	236	return -ENOMEM;
	237	} else
	238	t = (char*) "";
ea430986	239
4139c1b2 LP	240	success = dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &t);
	241
	242	if (cgb)
	243	free(t);
	244
	245	return success ? 0 : -ENOMEM;
	246	}
	247
	248	int bus_unit_append_cgroups(Manager m, DBusMessageIter i, const char property, void data) {
	249	Unit *u = data;
	250	CGroupBonding *cgb;
	251	DBusMessageIter sub;
	252
	253	if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "s", &sub))
	254	return -ENOMEM;
	255
	256	LIST_FOREACH(by_unit, cgb, u->meta.cgroup_bondings) {
	257	char *t;
	258	bool success;
	259
	260	if (!(t = cgroup_bonding_to_string(cgb)))
	261	return -ENOMEM;
	262
	263	success = dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, &t);
	264	free(t);
	265
	266	if (!success)
	267	return -ENOMEM;
	268	}
	269
	270	if (!dbus_message_iter_close_container(i, &sub))
	271	return -ENOMEM;
	272
	273	return 0;
	274	}
	275
45fb0699 LP	276	int bus_unit_append_need_daemon_reload(Manager m, DBusMessageIter i, const char property, void data) {
	277	Unit *u = data;
	278	dbus_bool_t b;
	279
	280	assert(m);
	281	assert(i);
	282	assert(property);
	283	assert(u);
	284
	285	b = unit_need_daemon_reload(u);
	286
	287	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b))
	288	return -ENOMEM;
	289
	290	return 0;
	291	}
	292
5e8d1c9a	293	static DBusHandlerResult bus_unit_message_dispatch(Unit u, DBusConnection connection, DBusMessage *message) {
b548631a LP	294	DBusMessage *reply = NULL;
	295	Manager *m = u->meta.manager;
	296	DBusError error;
	297	JobType job_type = _JOB_TYPE_INVALID;
c87eba54	298	char *path = NULL;
6f28c033	299	bool reload_if_possible = false;
b548631a LP	300
	301	dbus_error_init(&error);
	302
	303	if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Start"))
	304	job_type = JOB_START;
	305	else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Stop"))
	306	job_type = JOB_STOP;
	307	else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Reload"))
	308	job_type = JOB_RELOAD;
	309	else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Restart"))
	310	job_type = JOB_RESTART;
9a1ac7b9 LP	311	else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "TryRestart"))
9a1ac7b9 LP	312	job_type = JOB_TRY_RESTART;
6f28c033 LP	313	else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "ReloadOrRestart")) {
	314	reload_if_possible = true;
	315	job_type = JOB_RESTART;
	316	} else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "ReloadOrTryRestart")) {
	317	reload_if_possible = true;
	318	job_type = JOB_TRY_RESTART;
5632e374 LP	319	} else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "ResetMaintenance")) {
	320
	321	unit_reset_maintenance(u);
	322
	323	if (!(reply = dbus_message_new_method_return(message)))
	324	goto oom;
	325
6f28c033	326	} else if (UNIT_VTABLE(u)->bus_message_handler)
5e8d1c9a	327	return UNIT_VTABLE(u)->bus_message_handler(u, connection, message);
b548631a	328	else
4139c1b2	329	return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
b548631a LP	330
	331	if (job_type != _JOB_TYPE_INVALID) {
	332	const char *smode;
	333	JobMode mode;
	334	Job *j;
	335	int r;
b548631a	336
398ef8ba LP	337	if (job_type == JOB_START && u->meta.only_by_dependency) {
	338	dbus_set_error(&error, BUS_ERROR_ONLY_BY_DEPENDENCY, "Unit may be activated by dependency only.");
	339	return bus_send_error_reply(m, connection, message, &error, -EPERM);
	340	}
bc0f8771	341
b548631a LP	342	if (!dbus_message_get_args(
	343	message,
	344	&error,
	345	DBUS_TYPE_STRING, &smode,
	346	DBUS_TYPE_INVALID))
5e8d1c9a	347	return bus_send_error_reply(m, connection, message, &error, -EINVAL);
b548631a	348
6f28c033 LP	349	if (reload_if_possible && unit_can_reload(u)) {
	350	if (job_type == JOB_RESTART)
	351	job_type = JOB_RELOAD_OR_START;
	352	else if (job_type == JOB_TRY_RESTART)
	353	job_type = JOB_RELOAD;
	354	}
	355
398ef8ba LP	356	if ((mode = job_mode_from_string(smode)) == _JOB_MODE_INVALID) {
	357	dbus_set_error(&error, BUS_ERROR_INVALID_JOB_MODE, "Job mode %s is invalid.", smode);
	358	return bus_send_error_reply(m, connection, message, &error, -EINVAL);
	359	}
b548631a	360
398ef8ba LP	361	if ((r = manager_add_job(m, job_type, u, mode, true, &error, &j)) < 0)
398ef8ba LP	362	return bus_send_error_reply(m, connection, message, &error, r);
b548631a LP	363
	364	if (!(reply = dbus_message_new_method_return(message)))
	365	goto oom;
	366
	367	if (!(path = job_dbus_path(j)))
	368	goto oom;
	369
	370	if (!dbus_message_append_args(
	371	reply,
	372	DBUS_TYPE_OBJECT_PATH, &path,
	373	DBUS_TYPE_INVALID))
	374	goto oom;
	375	}
	376
c87eba54 LP	377	free(path);
c87eba54 LP	378
b548631a	379	if (reply) {
5e8d1c9a	380	if (!dbus_connection_send(connection, reply, NULL))
b548631a LP	381	goto oom;
	382
	383	dbus_message_unref(reply);
	384	}
	385
	386	return DBUS_HANDLER_RESULT_HANDLED;
	387
	388	oom:
c87eba54 LP	389	free(path);
c87eba54 LP	390
b548631a LP	391	if (reply)
	392	dbus_message_unref(reply);
	393
	394	dbus_error_free(&error);
	395
	396	return DBUS_HANDLER_RESULT_NEED_MEMORY;
ea430986 LP	397	}
ea430986 LP	398
5e8d1c9a	399	static DBusHandlerResult bus_unit_message_handler(DBusConnection connection, DBusMessage message, void *data) {
ea430986 LP	400	Manager *m = data;
	401	Unit *u;
	402	int r;
	403
	404	assert(connection);
	405	assert(message);
	406	assert(m);
	407
ea430986 LP	408	if ((r = manager_get_unit_from_dbus_path(m, dbus_message_get_path(message), &u)) < 0) {
	409
	410	if (r == -ENOMEM)
	411	return DBUS_HANDLER_RESULT_NEED_MEMORY;
	412
	413	if (r == -ENOENT)
	414	return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
	415
5e8d1c9a	416	return bus_send_error_reply(m, connection, message, NULL, r);
ea430986 LP	417	}
ea430986 LP	418
5e8d1c9a	419	return bus_unit_message_dispatch(u, connection, message);
ea430986 LP	420	}
	421
	422	const DBusObjectPathVTable bus_unit_vtable = {
	423	.message_function = bus_unit_message_handler
	424	};
c1e1601e LP	425
	426	void bus_unit_send_change_signal(Unit *u) {
	427	char *p = NULL;
	428	DBusMessage *m = NULL;
	429
	430	assert(u);
c1e1601e	431
c0bd0cf7 LP	432	if (u->meta.in_dbus_queue) {
	433	LIST_REMOVE(Meta, dbus_queue, u->meta.manager->dbus_unit_queue, &u->meta);
	434	u->meta.in_dbus_queue = false;
	435	}
c1e1601e	436
a567261a	437	if (!bus_has_subscriber(u->meta.manager)) {
94b6dfa2	438	u->meta.sent_dbus_new_signal = true;
c1e1601e	439	return;
94b6dfa2	440	}
c1e1601e LP	441
	442	if (!(p = unit_dbus_path(u)))
	443	goto oom;
	444
	445	if (u->meta.sent_dbus_new_signal) {
	446	/* Send a change signal */
	447
	448	if (!(m = dbus_message_new_signal(p, "org.freedesktop.systemd1.Unit", "Changed")))
	449	goto oom;
	450	} else {
c1e1601e LP	451	/* Send a new signal */
c1e1601e LP	452
701cc384	453	if (!(m = dbus_message_new_signal("/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", "UnitNew")))
c1e1601e LP	454	goto oom;
c1e1601e LP	455
c1e1601e	456	if (!dbus_message_append_args(m,
9e2f7c11	457	DBUS_TYPE_STRING, &u->meta.id,
c1e1601e LP	458	DBUS_TYPE_OBJECT_PATH, &p,
	459	DBUS_TYPE_INVALID))
	460	goto oom;
	461	}
	462
5e8d1c9a	463	if (bus_broadcast(u->meta.manager, m) < 0)
c1e1601e LP	464	goto oom;
	465
	466	free(p);
	467	dbus_message_unref(m);
	468
	469	u->meta.sent_dbus_new_signal = true;
	470
	471	return;
	472
	473	oom:
	474	free(p);
	475
	476	if (m)
	477	dbus_message_unref(m);
	478
	479	log_error("Failed to allocate unit change/new signal.");
	480	}
	481
	482	void bus_unit_send_removed_signal(Unit *u) {
	483	char *p = NULL;
	484	DBusMessage *m = NULL;
c1e1601e LP	485
	486	assert(u);
	487
a567261a	488	if (!bus_has_subscriber(u->meta.manager))
c1e1601e LP	489	return;
c1e1601e LP	490
7535cc78 LP	491	if (!u->meta.sent_dbus_new_signal)
	492	bus_unit_send_change_signal(u);
	493
c1e1601e LP	494	if (!(p = unit_dbus_path(u)))
	495	goto oom;
	496
701cc384	497	if (!(m = dbus_message_new_signal("/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", "UnitRemoved")))
c1e1601e LP	498	goto oom;
c1e1601e LP	499
c1e1601e	500	if (!dbus_message_append_args(m,
9e2f7c11	501	DBUS_TYPE_STRING, &u->meta.id,
c1e1601e LP	502	DBUS_TYPE_OBJECT_PATH, &p,
	503	DBUS_TYPE_INVALID))
	504	goto oom;
	505
5e8d1c9a	506	if (bus_broadcast(u->meta.manager, m) < 0)
c1e1601e LP	507	goto oom;
	508
	509	free(p);
	510	dbus_message_unref(m);
	511
	512	return;
	513
	514	oom:
	515	free(p);
	516
	517	if (m)
	518	dbus_message_unref(m);
	519
	520	log_error("Failed to allocate unit remove signal.");
	521	}