diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/dbus-manager.c | 7 | ||||
-rw-r--r-- | src/main.c | 7 | ||||
-rw-r--r-- | src/manager.c | 19 | ||||
-rw-r--r-- | src/manager.h | 2 | ||||
-rw-r--r-- | src/system.conf | 1 | ||||
-rw-r--r-- | src/unit.c | 66 |
6 files changed, 70 insertions, 32 deletions
diff --git a/src/dbus-manager.c b/src/dbus-manager.c index d1d3b4784e..28986e5426 100644 --- a/src/dbus-manager.c +++ b/src/dbus-manager.c @@ -167,7 +167,9 @@ " <property name=\"NotifySocket\" type=\"s\" access=\"read\"/>\n" \ " <property name=\"ControlGroupHierarchy\" type=\"s\" access=\"read\"/>\n" \ " <property name=\"MountAuto\" type=\"b\" access=\"read\"/>\n" \ - " <property name=\"SwapAuto\" type=\"b\" access=\"read\"/>\n" + " <property name=\"SwapAuto\" type=\"b\" access=\"read\"/>\n" \ + " <property name=\"DefaultControllers\" type=\"as\" access=\"read\"/>\n" + \ #ifdef HAVE_SYSV_COMPAT #define BUS_MANAGER_INTERFACE_PROPERTIES_SYSV \ @@ -319,7 +321,8 @@ static DBusHandlerResult bus_manager_message_handler(DBusConnection *connection, { "org.freedesktop.systemd1.Manager", "NotifySocket", bus_property_append_string, "s", m->notify_socket }, { "org.freedesktop.systemd1.Manager", "ControlGroupHierarchy", bus_property_append_string, "s", m->cgroup_hierarchy }, { "org.freedesktop.systemd1.Manager", "MountAuto", bus_property_append_bool, "b", &m->mount_auto }, - { "org.freedesktop.systemd1.Manager", "SwapAuto", bus_property_append_bool, "b", &m->swap_auto }, + { "org.freedesktop.systemd1.Manager", "SwapAuto", bus_property_append_bool, "b", &m->swap_auto }, + { "org.freedesktop.systemd1.Manager", "DefaultControllers", bus_property_append_strv, "as", m->default_controllers }, #ifdef HAVE_SYSV_COMPAT { "org.freedesktop.systemd1.Manager", "SysVConsole", bus_property_append_bool, "b", &m->sysv_console }, { "org.freedesktop.systemd1.Manager", "SysVInitPath", bus_property_append_strv, "as", m->lookup_paths.sysvinit_path }, diff --git a/src/main.c b/src/main.c index 4bdc6763fa..99e277c67b 100644 --- a/src/main.c +++ b/src/main.c @@ -49,6 +49,7 @@ #include "missing.h" #include "label.h" #include "build.h" +#include "strv.h" static enum { ACTION_RUN, @@ -72,6 +73,7 @@ static bool arg_sysv_console = true; static bool arg_mount_auto = true; static bool arg_swap_auto = true; static char *arg_console = NULL; +static char **arg_default_controllers = NULL; static FILE* serialization = NULL; @@ -502,6 +504,7 @@ static int parse_config_file(void) { { "CPUAffinity", config_parse_cpu_affinity, NULL, "Manager" }, { "MountAuto", config_parse_bool, &arg_mount_auto, "Manager" }, { "SwapAuto", config_parse_bool, &arg_swap_auto, "Manager" }, + { "DefaultControllers", config_parse_strv, &arg_default_controllers, "Manager" }, { NULL, NULL, NULL, NULL } }; @@ -1089,6 +1092,9 @@ int main(int argc, char *argv[]) { if (arg_console) manager_set_console(m, arg_console); + if (arg_default_controllers) + manager_set_default_controllers(m, arg_default_controllers); + if ((r = manager_startup(m, serialization, fds)) < 0) log_error("Failed to fully start up daemon: %s", strerror(-r)); @@ -1211,6 +1217,7 @@ finish: free(arg_default_unit); free(arg_console); + strv_free(arg_default_controllers); dbus_shutdown(); diff --git a/src/manager.c b/src/manager.c index 827e9937c6..62847061a2 100644 --- a/src/manager.c +++ b/src/manager.c @@ -222,6 +222,9 @@ int manager_new(ManagerRunningAs running_as, Manager **_m) { if (!(m->environment = strv_copy(environ))) goto fail; + if (!(m->default_controllers = strv_new("cpu", NULL))) + goto fail; + if (!(m->units = hashmap_new(string_hash_func, string_compare_func))) goto fail; @@ -461,6 +464,8 @@ void manager_free(Manager *m) { lookup_paths_free(&m->lookup_paths); strv_free(m->environment); + strv_free(m->default_controllers); + hashmap_free(m->cgroup_bondings); set_free_free(m->unit_path_cache); @@ -2988,6 +2993,20 @@ void manager_undo_generators(Manager *m) { m->generator_unit_path = NULL; } +int manager_set_default_controllers(Manager *m, char **controllers) { + char **l; + + assert(m); + + if (!(l = strv_copy(controllers))) + return -ENOMEM; + + strv_free(m->default_controllers); + m->default_controllers = l; + + return 0; +} + static const char* const manager_running_as_table[_MANAGER_RUNNING_AS_MAX] = { [MANAGER_SYSTEM] = "system", [MANAGER_USER] = "user" diff --git a/src/manager.h b/src/manager.h index c7ace2dd5e..f9c769e42b 100644 --- a/src/manager.h +++ b/src/manager.h @@ -142,6 +142,7 @@ struct Manager { Set *unit_path_cache; char **environment; + char **default_controllers; dual_timestamp initrd_timestamp; dual_timestamp startup_timestamp; @@ -256,6 +257,7 @@ unsigned manager_dispatch_run_queue(Manager *m); unsigned manager_dispatch_dbus_queue(Manager *m); int manager_set_console(Manager *m, const char *console); +int manager_set_default_controllers(Manager *m, char **controllers); int manager_loop(Manager *m); diff --git a/src/system.conf b/src/system.conf index c6a7bf3df8..c94cf3601f 100644 --- a/src/system.conf +++ b/src/system.conf @@ -20,3 +20,4 @@ #CPUAffinity=1 2 #MountAuto=yes #SwapAuto=yes +#DefaultControllers=cpu diff --git a/src/unit.c b/src/unit.c index bfb1dd644e..7d673e138d 100644 --- a/src/unit.c +++ b/src/unit.c @@ -1785,55 +1785,61 @@ fail: return r; } -int unit_add_default_cgroups(Unit *u) { +static int unit_add_one_default_cgroup(Unit *u, const char *controller) { CGroupBonding *b = NULL; int r = -ENOMEM; - const char * const default_controllers[] = { - SYSTEMD_CGROUP_CONTROLLER, - "cpu", - NULL - }; - const char * const*c; assert(u); - /* Adds in the default cgroups, if it wasn't specified yet */ - - STRV_FOREACH(c, default_controllers) { - - if (cgroup_bonding_find_list(u->meta.cgroup_bondings, *c)) - continue; + if (!controller) + controller = SYSTEMD_CGROUP_CONTROLLER; - if (!(b = new0(CGroupBonding, 1))) - return -ENOMEM; + if (cgroup_bonding_find_list(u->meta.cgroup_bondings, controller)) + return 0; - if (!(b->path = default_cgroup_path(u))) - goto fail; + if (!(b = new0(CGroupBonding, 1))) + return -ENOMEM; - if (!(b->controller = strdup(*c))) - goto fail; + if (!(b->controller = strdup(controller))) + goto fail; - b->ours = true; - b->essential = c == default_controllers; /* the first one is essential */ + if (!(b->path = default_cgroup_path(u))) + goto fail; - if ((r = unit_add_cgroup(u, b)) < 0) - goto fail; + b->ours = true; + b->essential = streq(controller, SYSTEMD_CGROUP_CONTROLLER); - b = NULL; - } + if ((r = unit_add_cgroup(u, b)) < 0) + goto fail; return 0; fail: - if (b) { - free(b->path); - free(b->controller); - free(b); - } + free(b->path); + free(b->controller); + free(b); return r; } +int unit_add_default_cgroups(Unit *u) { + char **c; + int r; + assert(u); + + /* Adds in the default cgroups, if they weren't specified + * otherwise. */ + + if ((r = unit_add_one_default_cgroup(u, NULL)) < 0) + return r; + + STRV_FOREACH(c, u->meta.manager->default_controllers) + if ((r = unit_add_one_default_cgroup(u, *c)) < 0) + return r; + + return 0; +} + CGroupBonding* unit_get_default_cgroup(Unit *u) { assert(u); |