src/dbus-execute.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   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>
  24 #include <sys/prctl.h>
  25
  26 #include "dbus-execute.h"
  27 #include "missing.h"
  28 #include "ioprio.h"
  29 #include "strv.h"
  30
  31 DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_execute_append_kill_mode, kill_mode, KillMode);
  32
  33 DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_execute_append_input, exec_input, ExecInput);
  34 DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_execute_append_output, exec_output, ExecOutput);
  35
  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
  69 int bus_execute_append_oom_score_adjust(Manager *m, DBusMessageIter *i, const char *property, void *data) {
  70         ExecContext *c = data;
  71         int32_t n;
  72
  73         assert(m);
  74         assert(i);
  75         assert(property);
  76         assert(c);
  77
  78         if (c->oom_score_adjust_set)
  79                 n = c->oom_score_adjust;
  80         else {
  81                 char *t;
  82
  83                 n = 0;
  84                 if (read_one_line_file("/proc/self/oom_score_adj", &t) >= 0) {
  85                         safe_atoi(t, &n);
  86                         free(t);
  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;
  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
 200         if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "y", &sub))
 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
 217 int bus_execute_append_timer_slack_nsec(Manager *m, DBusMessageIter *i, const char *property, void *data) {
 218         ExecContext *c = data;
 219         uint64_t u;
 220
 221         assert(m);
 222         assert(i);
 223         assert(property);
 224         assert(c);
 225
 226         if (c->timer_slack_nsec_set)
 227                 u = (uint64_t) c->timer_slack_nsec;
 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
 253         if (!s)
 254                 return -ENOMEM;
 255
 256         b = dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &s);
 257
 258         if (t)
 259                 cap_free(t);
 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 }
 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
 304         if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "(sasbttuii)", &sub))
 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) ||
 329                     !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_BOOLEAN, &c->ignore) ||
 330                     !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_UINT64, &c->exec_status.start_timestamp.realtime) ||
 331                     !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_UINT64, &c->exec_status.exit_timestamp.realtime) ||
 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))
 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 }