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

/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/

/***
  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/dbus.h>
#include <sys/prctl.h>

#include "dbus-execute.h"
#include "missing.h"
#include "ioprio.h"
#include "strv.h"

DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_execute_append_kill_mode, kill_mode, KillMode);

DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_execute_append_input, exec_input, ExecInput);
DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_execute_append_output, exec_output, ExecOutput);

int bus_execute_append_env_files(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        char **env_files = data, **j;
        DBusMessageIter sub, sub2;

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

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

        STRV_FOREACH(j, env_files) {
                dbus_bool_t b = false;
                char *fn = *j;

                if (fn[0] == '-') {
                        b = true;
                        fn++;
                }

                if (!dbus_message_iter_open_container(&sub, DBUS_TYPE_STRUCT, NULL, &sub2) ||
                    !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &fn) ||
                    !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_BOOLEAN, &b) ||
                    !dbus_message_iter_close_container(&sub, &sub2))
                        return -ENOMEM;
        }

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

        return 0;
}

int bus_execute_append_oom_score_adjust(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        ExecContext *c = data;
        int32_t n;

        assert(m);
        assert(i);
        assert(property);
        assert(c);

        if (c->oom_score_adjust_set)
                n = c->oom_score_adjust;
        else {
                char *t;

                n = 0;
                if (read_one_line_file("/proc/self/oom_score_adj", &t) >= 0) {
                        safe_atoi(t, &n);
                        free(t);
                } else if (read_one_line_file("/proc/self/oom_adj", &t) >= 0) {
                        safe_atoi(t, &n);
                        free(t);

                        if (n == OOM_ADJUST_MAX)
                                n = OOM_SCORE_ADJ_MAX;
                        else
                                n = (n * OOM_SCORE_ADJ_MAX) / -OOM_DISABLE;
                }
        }

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_INT32, &n))
                return -ENOMEM;

        return 0;
}

int bus_execute_append_nice(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        ExecContext *c = data;
        int32_t n;

        assert(m);
        assert(i);
        assert(property);
        assert(c);

        if (c->nice_set)
                n = c->nice;
        else
                n = getpriority(PRIO_PROCESS, 0);

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_INT32, &n))
                return -ENOMEM;

        return 0;
}

int bus_execute_append_ioprio(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        ExecContext *c = data;
        int32_t n;

        assert(m);
        assert(i);
        assert(property);
        assert(c);

        if (c->ioprio_set)
                n = c->ioprio;
        else
                n = ioprio_get(IOPRIO_WHO_PROCESS, 0);

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_INT32, &n))
                return -ENOMEM;

        return 0;
}

int bus_execute_append_cpu_sched_policy(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        ExecContext *c = data;
        int32_t n;

        assert(m);
        assert(i);
        assert(property);
        assert(c);

        if (c->cpu_sched_set)
                n = c->cpu_sched_policy;
        else
                n = sched_getscheduler(0);

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_INT32, &n))
                return -ENOMEM;

        return 0;
}

int bus_execute_append_cpu_sched_priority(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        ExecContext *c = data;
        int32_t n;

        assert(m);
        assert(i);
        assert(property);
        assert(c);

        if (c->cpu_sched_set)
                n = c->cpu_sched_priority;
        else {
                struct sched_param p;
                n = 0;

                zero(p);
                if (sched_getparam(0, &p) >= 0)
                        n = p.sched_priority;
        }

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_INT32, &n))
                return -ENOMEM;

        return 0;
}

int bus_execute_append_affinity(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        ExecContext *c = data;
        dbus_bool_t b;
        DBusMessageIter sub;

        assert(m);
        assert(i);
        assert(property);
        assert(c);

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

        if (c->cpuset)
                b = dbus_message_iter_append_fixed_array(&sub, DBUS_TYPE_BYTE, &c->cpuset, CPU_ALLOC_SIZE(c->cpuset_ncpus));
        else
                b = dbus_message_iter_append_fixed_array(&sub, DBUS_TYPE_BYTE, &c->cpuset, 0);

        if (!b)
                return -ENOMEM;

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

        return 0;
}

int bus_execute_append_timer_slack_nsec(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        ExecContext *c = data;
        uint64_t u;

        assert(m);
        assert(i);
        assert(property);
        assert(c);

        if (c->timer_slack_nsec_set)
                u = (uint64_t) c->timer_slack_nsec;
        else
                u = (uint64_t) prctl(PR_GET_TIMERSLACK);

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_UINT64, &u))
                return -ENOMEM;

        return 0;
}

int bus_execute_append_capabilities(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        ExecContext *c = data;
        char *t = NULL;
        const char *s;
        dbus_bool_t b;

        assert(m);
        assert(i);
        assert(property);
        assert(c);

        if (c->capabilities)
                s = t = cap_to_text(c->capabilities, NULL);
        else
                s = "";

        if (!s)
                return -ENOMEM;

        b = dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &s);

        if (t)
                cap_free(t);

        if (!b)
                return -ENOMEM;

        return 0;
}

int bus_execute_append_rlimits(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        ExecContext *c = data;
        int r;
        uint64_t u;

        assert(m);
        assert(i);
        assert(property);
        assert(c);

        assert_se((r = rlimit_from_string(property)) >= 0);

        if (c->rlimit[r])
                u = (uint64_t) c->rlimit[r]->rlim_max;
        else {
                struct rlimit rl;

                zero(rl);
                getrlimit(r, &rl);

                u = (uint64_t) rl.rlim_max;
        }

        if (!dbus_message_iter_append_basic(i, DBUS_TYPE_UINT64, &u))
                return -ENOMEM;

        return 0;
}

int bus_execute_append_command(Manager *m, DBusMessageIter *i, const char *property, void *data) {
        ExecCommand *c = data;
        DBusMessageIter sub, sub2, sub3;

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

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

        LIST_FOREACH(command, c, c) {
                char **l;
                uint32_t pid;
                int32_t code, status;

                if (!c->path)
                        continue;

                if (!dbus_message_iter_open_container(&sub, DBUS_TYPE_STRUCT, NULL, &sub2) ||
                    !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &c->path) ||
                    !dbus_message_iter_open_container(&sub2, DBUS_TYPE_ARRAY, "s", &sub3))
                        return -ENOMEM;

                STRV_FOREACH(l, c->argv)
                        if (!dbus_message_iter_append_basic(&sub3, DBUS_TYPE_STRING, l))
                                return -ENOMEM;

                pid = (uint32_t) c->exec_status.pid;
                code = (int32_t) c->exec_status.code;
                status = (int32_t) c->exec_status.status;

                if (!dbus_message_iter_close_container(&sub2, &sub3) ||
                    !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_BOOLEAN, &c->ignore) ||
                    !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_UINT64, &c->exec_status.start_timestamp.realtime) ||
                    !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_UINT64, &c->exec_status.exit_timestamp.realtime) ||
                    !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_UINT32, &pid) ||
                    !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_INT32, &code) ||
                    !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_INT32, &status))
                        return -ENOMEM;

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

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

        return 0;
}
Commit	Line	Data
d6c9574f	1	/-- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil --/
4139c1b2 LP	2
	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
	22	#include <errno.h>
	23	#include <dbus/dbus.h>
82c121a4	24	#include <sys/prctl.h>
4139c1b2 LP	25
4139c1b2 LP	26	#include "dbus-execute.h"
82c121a4 LP	27	#include "missing.h"
82c121a4 LP	28	#include "ioprio.h"
fe68089d	29	#include "strv.h"
4139c1b2	30
2e22afe9 LP	31	DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_execute_append_kill_mode, kill_mode, KillMode);
2e22afe9 LP	32
4139c1b2 LP	33	DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_execute_append_input, exec_input, ExecInput);
4139c1b2 LP	34	DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_execute_append_output, exec_output, ExecOutput);
82c121a4	35
8c7be95e LP	36	int bus_execute_append_env_files(Manager m, DBusMessageIter i, const char property, void data) {
	37	char env_files = data, j;
	38	DBusMessageIter sub, sub2;
	39
	40	assert(m);
	41	assert(i);
	42	assert(property);
	43
	44	if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "(sb)", &sub))
	45	return -ENOMEM;
	46
	47	STRV_FOREACH(j, env_files) {
	48	dbus_bool_t b = false;
	49	char fn = j;
	50
	51	if (fn[0] == '-') {
	52	b = true;
	53	fn++;
	54	}
	55
	56	if (!dbus_message_iter_open_container(&sub, DBUS_TYPE_STRUCT, NULL, &sub2) \|\|
	57	!dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &fn) \|\|
	58	!dbus_message_iter_append_basic(&sub2, DBUS_TYPE_BOOLEAN, &b) \|\|
	59	!dbus_message_iter_close_container(&sub, &sub2))
	60	return -ENOMEM;
	61	}
	62
	63	if (!dbus_message_iter_close_container(i, &sub))
	64	return -ENOMEM;
	65
	66	return 0;
	67	}
	68
dd6c17b1	69	int bus_execute_append_oom_score_adjust(Manager m, DBusMessageIter i, const char property, void data) {
82c121a4 LP	70	ExecContext *c = data;
	71	int32_t n;
	72
	73	assert(m);
	74	assert(i);
	75	assert(property);
	76	assert(c);
	77
dd6c17b1 LP	78	if (c->oom_score_adjust_set)
dd6c17b1 LP	79	n = c->oom_score_adjust;
82c121a4 LP	80	else {
	81	char *t;
	82
	83	n = 0;
dd6c17b1	84	if (read_one_line_file("/proc/self/oom_score_adj", &t) >= 0) {
82c121a4 LP	85	safe_atoi(t, &n);
82c121a4 LP	86	free(t);
dd6c17b1 LP	87	} else if (read_one_line_file("/proc/self/oom_adj", &t) >= 0) {
	88	safe_atoi(t, &n);
	89	free(t);
	90
	91	if (n == OOM_ADJUST_MAX)
	92	n = OOM_SCORE_ADJ_MAX;
	93	else
	94	n = (n * OOM_SCORE_ADJ_MAX) / -OOM_DISABLE;
82c121a4 LP	95	}
	96	}
	97
	98	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_INT32, &n))
	99	return -ENOMEM;
	100
	101	return 0;
	102	}
	103
	104	int bus_execute_append_nice(Manager m, DBusMessageIter i, const char property, void data) {
	105	ExecContext *c = data;
	106	int32_t n;
	107
	108	assert(m);
	109	assert(i);
	110	assert(property);
	111	assert(c);
	112
	113	if (c->nice_set)
	114	n = c->nice;
	115	else
	116	n = getpriority(PRIO_PROCESS, 0);
	117
	118	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_INT32, &n))
	119	return -ENOMEM;
	120
	121	return 0;
	122	}
	123
	124	int bus_execute_append_ioprio(Manager m, DBusMessageIter i, const char property, void data) {
	125	ExecContext *c = data;
	126	int32_t n;
	127
	128	assert(m);
	129	assert(i);
	130	assert(property);
	131	assert(c);
	132
	133	if (c->ioprio_set)
	134	n = c->ioprio;
	135	else
	136	n = ioprio_get(IOPRIO_WHO_PROCESS, 0);
	137
	138	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_INT32, &n))
	139	return -ENOMEM;
	140
	141	return 0;
	142	}
	143
	144	int bus_execute_append_cpu_sched_policy(Manager m, DBusMessageIter i, const char property, void data) {
	145	ExecContext *c = data;
	146	int32_t n;
	147
	148	assert(m);
	149	assert(i);
	150	assert(property);
	151	assert(c);
	152
	153	if (c->cpu_sched_set)
	154	n = c->cpu_sched_policy;
	155	else
	156	n = sched_getscheduler(0);
	157
	158	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_INT32, &n))
159	return -ENOMEM;
160
161	return 0;
162	}
163
164	int bus_execute_append_cpu_sched_priority(Manager m, DBusMessageIter i, const char property, void data) {
165	ExecContext *c = data;
166	int32_t n;
167
168	assert(m);
169	assert(i);
170	assert(property);
171	assert(c);
172
173	if (c->cpu_sched_set)
174	n = c->cpu_sched_priority;
175	else {
176	struct sched_param p;
177	n = 0;
178
179	zero(p);
180	if (sched_getparam(0, &p) >= 0)
181	n = p.sched_priority;
182	}
183
184	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_INT32, &n))
185	return -ENOMEM;
186
187	return 0;
188	}
189
190	int bus_execute_append_affinity(Manager m, DBusMessageIter i, const char property, void data) {
191	ExecContext *c = data;
192	dbus_bool_t b;
193	DBusMessageIter sub;
194
195	assert(m);
196	assert(i);
197	assert(property);
198	assert(c);
199
fe68089d	200	if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "y", &sub))
82c121a4 LP	201	return -ENOMEM;
	202
	203	if (c->cpuset)
	204	b = dbus_message_iter_append_fixed_array(&sub, DBUS_TYPE_BYTE, &c->cpuset, CPU_ALLOC_SIZE(c->cpuset_ncpus));
	205	else
	206	b = dbus_message_iter_append_fixed_array(&sub, DBUS_TYPE_BYTE, &c->cpuset, 0);
	207
	208	if (!b)
	209	return -ENOMEM;
	210
	211	if (!dbus_message_iter_close_container(i, &sub))
	212	return -ENOMEM;
	213
	214	return 0;
	215	}
	216
03fae018	217	int bus_execute_append_timer_slack_nsec(Manager m, DBusMessageIter i, const char property, void data) {
82c121a4 LP	218	ExecContext *c = data;
	219	uint64_t u;
	220
	221	assert(m);
	222	assert(i);
	223	assert(property);
	224	assert(c);
	225
03fae018 LP	226	if (c->timer_slack_nsec_set)
03fae018 LP	227	u = (uint64_t) c->timer_slack_nsec;
82c121a4 LP	228	else
	229	u = (uint64_t) prctl(PR_GET_TIMERSLACK);
	230
	231	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_UINT64, &u))
	232	return -ENOMEM;
	233
	234	return 0;
	235	}
	236
	237	int bus_execute_append_capabilities(Manager m, DBusMessageIter i, const char property, void data) {
	238	ExecContext *c = data;
	239	char *t = NULL;
	240	const char *s;
	241	dbus_bool_t b;
	242
	243	assert(m);
	244	assert(i);
	245	assert(property);
	246	assert(c);
	247
	248	if (c->capabilities)
	249	s = t = cap_to_text(c->capabilities, NULL);
	250	else
	251	s = "";
	252
674cdd19	253	if (!s)
82c121a4 LP	254	return -ENOMEM;
	255
	256	b = dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &s);
674cdd19 LP	257
	258	if (t)
	259	cap_free(t);
82c121a4 LP	260
	261	if (!b)
	262	return -ENOMEM;
	263
	264	return 0;
	265	}
	266
	267	int bus_execute_append_rlimits(Manager m, DBusMessageIter i, const char property, void data) {
	268	ExecContext *c = data;
	269	int r;
	270	uint64_t u;
	271
	272	assert(m);
	273	assert(i);
	274	assert(property);
	275	assert(c);
	276
	277	assert_se((r = rlimit_from_string(property)) >= 0);
	278
	279	if (c->rlimit[r])
	280	u = (uint64_t) c->rlimit[r]->rlim_max;
	281	else {
	282	struct rlimit rl;
	283
	284	zero(rl);
	285	getrlimit(r, &rl);
	286
	287	u = (uint64_t) rl.rlim_max;
	288	}
	289
	290	if (!dbus_message_iter_append_basic(i, DBUS_TYPE_UINT64, &u))
	291	return -ENOMEM;
	292
	293	return 0;
	294	}
fe68089d LP	295
	296	int bus_execute_append_command(Manager m, DBusMessageIter i, const char property, void data) {
	297	ExecCommand *c = data;
	298	DBusMessageIter sub, sub2, sub3;
	299
	300	assert(m);
	301	assert(i);
	302	assert(property);
	303
b708e7ce	304	if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "(sasbttuii)", &sub))
fe68089d LP	305	return -ENOMEM;
	306
	307	LIST_FOREACH(command, c, c) {
	308	char **l;
	309	uint32_t pid;
	310	int32_t code, status;
	311
	312	if (!c->path)
	313	continue;
	314
	315	if (!dbus_message_iter_open_container(&sub, DBUS_TYPE_STRUCT, NULL, &sub2) \|\|
	316	!dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &c->path) \|\|
	317	!dbus_message_iter_open_container(&sub2, DBUS_TYPE_ARRAY, "s", &sub3))
	318	return -ENOMEM;
	319
	320	STRV_FOREACH(l, c->argv)
	321	if (!dbus_message_iter_append_basic(&sub3, DBUS_TYPE_STRING, l))
	322	return -ENOMEM;
	323
	324	pid = (uint32_t) c->exec_status.pid;
	325	code = (int32_t) c->exec_status.code;
	326	status = (int32_t) c->exec_status.status;
	327
	328	if (!dbus_message_iter_close_container(&sub2, &sub3) \|\|
b708e7ce	329	!dbus_message_iter_append_basic(&sub2, DBUS_TYPE_BOOLEAN, &c->ignore) \|\|
fe68089d LP	330	!dbus_message_iter_append_basic(&sub2, DBUS_TYPE_UINT64, &c->exec_status.start_timestamp.realtime) \|\|
fe68089d LP	331	!dbus_message_iter_append_basic(&sub2, DBUS_TYPE_UINT64, &c->exec_status.exit_timestamp.realtime) \|\|
e364ad06 LP	332	!dbus_message_iter_append_basic(&sub2, DBUS_TYPE_UINT32, &pid) \|\|
	333	!dbus_message_iter_append_basic(&sub2, DBUS_TYPE_INT32, &code) \|\|
	334	!dbus_message_iter_append_basic(&sub2, DBUS_TYPE_INT32, &status))
fe68089d LP	335	return -ENOMEM;
	336
	337	if (!dbus_message_iter_close_container(&sub, &sub2))
	338	return -ENOMEM;
	339	}
	340
	341	if (!dbus_message_iter_close_container(i, &sub))
	342	return -ENOMEM;
	343
	344	return 0;
	345	}