chiark / gitweb /
Implement SocketUser= and SocketGroup= for [Socket]
[elogind.git] / src / core / swap.c
index b4f53b724862ceae6a2caf57242b9c63dff43035..97145a9974a404ca1bfac7ad6748b078310fd3a7 100644 (file)
@@ -55,6 +55,7 @@ static const UnitActiveState state_translation_table[_SWAP_STATE_MAX] = {
 
 static void swap_unset_proc_swaps(Swap *s) {
         Swap *first;
+        Hashmap *swaps;
 
         assert(s);
 
@@ -63,14 +64,17 @@ static void swap_unset_proc_swaps(Swap *s) {
 
         /* Remove this unit from the chain of swaps which share the
          * same kernel swap device. */
-
-        first = hashmap_get(UNIT(s)->manager->swaps_by_proc_swaps, s->parameters_proc_swaps.what);
+        swaps = UNIT(s)->manager->swaps_by_proc_swaps;
+        first = hashmap_get(swaps, s->parameters_proc_swaps.what);
         LIST_REMOVE(Swap, same_proc_swaps, first, s);
 
         if (first)
-                hashmap_remove_and_replace(UNIT(s)->manager->swaps_by_proc_swaps, s->parameters_proc_swaps.what, first->parameters_proc_swaps.what, first);
+                hashmap_remove_and_replace(swaps,
+                                           s->parameters_proc_swaps.what,
+                                           first->parameters_proc_swaps.what,
+                                           first);
         else
-                hashmap_remove(UNIT(s)->manager->swaps_by_proc_swaps, s->parameters_proc_swaps.what);
+                hashmap_remove(swaps, s->parameters_proc_swaps.what);
 
         free(s->parameters_proc_swaps.what);
         s->parameters_proc_swaps.what = NULL;
@@ -146,7 +150,8 @@ int swap_add_one_mount_link(Swap *s, Mount *m) {
         if (!path_startswith(s->what, m->where))
                 return 0;
 
-        if ((r = unit_add_two_dependencies(UNIT(s), UNIT_AFTER, UNIT_REQUIRES, UNIT(m), true)) < 0)
+        r = unit_add_two_dependencies(UNIT(s), UNIT_AFTER, UNIT_REQUIRES, UNIT(m), true);
+        if (r < 0)
                 return r;
 
         return 0;
@@ -165,22 +170,6 @@ static int swap_add_mount_links(Swap *s) {
         return 0;
 }
 
-static int swap_add_target_links(Swap *s) {
-        Unit *tu;
-        int r;
-
-        assert(s);
-
-        if (!s->from_fragment)
-                return 0;
-
-        r = manager_load_unit(UNIT(s)->manager, SPECIAL_SWAP_TARGET, NULL, NULL, &tu);
-        if (r < 0)
-                return r;
-
-        return unit_add_dependency(UNIT(s), UNIT_BEFORE, tu, true);
-}
-
 static int swap_add_device_links(Swap *s) {
         SwapParameters *p;
 
@@ -225,24 +214,23 @@ static int swap_add_default_dependencies(Swap *s) {
 
 static int swap_verify(Swap *s) {
         bool b;
-        char *e;
+        char _cleanup_free_ *e = NULL;
 
         if (UNIT(s)->load_state != UNIT_LOADED)
                   return 0;
 
-        if (!(e = unit_name_from_path(s->what, ".swap")))
-                  return -ENOMEM;
+        e = unit_name_from_path(s->what, ".swap");
+        if (e == NULL)
+                return log_oom();
 
         b = unit_has_name(UNIT(s), e);
-        free(e);
-
         if (!b) {
-                log_error("%s: Value of \"What\" and unit name do not match, not loading.\n", UNIT(s)->id);
+                log_error("%s: Value of \"What\" and unit name do not match, not loading.", UNIT(s)->id);
                 return -EINVAL;
         }
 
         if (s->exec_context.pam_name && s->kill_context.kill_mode != KILL_CONTROL_GROUP) {
-                log_error("%s has PAM enabled. Kill mode must be set to 'control-group'. Refusing.", UNIT(s)->id);
+                log_error("%s has PAM enabled. Kill mode must be set to 'control-group'. Refusing to load.", UNIT(s)->id);
                 return -EINVAL;
         }
 
@@ -257,11 +245,13 @@ static int swap_load(Unit *u) {
         assert(u->load_state == UNIT_STUB);
 
         /* Load a .swap file */
-        if ((r = unit_load_fragment_and_dropin_optional(u)) < 0)
+        r = unit_load_fragment_and_dropin_optional(u);
+        if (r < 0)
                 return r;
 
         if (u->load_state == UNIT_LOADED) {
-                if ((r = unit_add_exec_dependencies(u, &s->exec_context)) < 0)
+                r = unit_add_exec_dependencies(u, &s->exec_context);
+                if (r < 0)
                         return r;
 
                 if (UNIT(s)->fragment_path)
@@ -285,21 +275,23 @@ static int swap_load(Unit *u) {
                         if ((r = unit_set_description(u, s->what)) < 0)
                                 return r;
 
-                if ((r = swap_add_device_links(s)) < 0)
-                        return r;
-
-                if ((r = swap_add_mount_links(s)) < 0)
+                r = swap_add_device_links(s);
+                if (r < 0)
                         return r;
 
-                if ((r = swap_add_target_links(s)) < 0)
+                r = swap_add_mount_links(s);
+                if (r < 0)
                         return r;
 
-                if ((r = unit_add_default_cgroups(u)) < 0)
+                r = unit_add_default_cgroups(u);
+                if (r < 0)
                         return r;
 
-                if (UNIT(s)->default_dependencies)
-                        if ((r = swap_add_default_dependencies(s)) < 0)
+                if (UNIT(s)->default_dependencies) {
+                        r = swap_add_default_dependencies(s);
+                        if (r < 0)
                                 return r;
+                }
 
                 r = unit_exec_context_defaults(u, &s->exec_context);
                 if (r < 0)
@@ -319,7 +311,8 @@ static int swap_add_one(
                 bool set_flags) {
 
         Unit *u = NULL;
-        char *e = NULL, *wp = NULL;
+        char _cleanup_free_ *e = NULL;
+        char *wp = NULL;
         bool delete = false;
         int r;
         SwapParameters *p;
@@ -331,7 +324,7 @@ static int swap_add_one(
 
         e = unit_name_from_path(what, ".swap");
         if (!e)
-                return -ENOMEM;
+                return log_oom();
 
         u = manager_get_unit(m, e);
 
@@ -344,10 +337,8 @@ static int swap_add_one(
                 delete = true;
 
                 u = unit_new(m, sizeof(Swap));
-                if (!u) {
-                        free(e);
-                        return -ENOMEM;
-                }
+                if (!u)
+                        return log_oom();
 
                 r = unit_add_name(u, e);
                 if (r < 0)
@@ -355,7 +346,7 @@ static int swap_add_one(
 
                 SWAP(u)->what = strdup(what);
                 if (!SWAP(u)->what) {
-                        r = -ENOMEM;
+                        r = log_oom();
                         goto fail;
                 }
 
@@ -366,16 +357,19 @@ static int swap_add_one(
         p = &SWAP(u)->parameters_proc_swaps;
 
         if (!p->what) {
-                if (!(wp = strdup(what_proc_swaps))) {
-                        r = -ENOMEM;
+                wp = strdup(what_proc_swaps);
+                if (!wp) {
+                        r = log_oom();
                         goto fail;
                 }
 
-                if (!m->swaps_by_proc_swaps)
-                        if (!(m->swaps_by_proc_swaps = hashmap_new(string_hash_func, string_compare_func))) {
-                                r = -ENOMEM;
+                if (!m->swaps_by_proc_swaps) {
+                        m->swaps_by_proc_swaps = hashmap_new(string_hash_func, string_compare_func);
+                        if (!m->swaps_by_proc_swaps) {
+                                r = log_oom();
                                 goto fail;
                         }
+                }
 
                 free(p->what);
                 p->what = wp;
@@ -383,7 +377,8 @@ static int swap_add_one(
                 first = hashmap_get(m->swaps_by_proc_swaps, wp);
                 LIST_PREPEND(Swap, same_proc_swaps, first, SWAP(u));
 
-                if ((r = hashmap_replace(m->swaps_by_proc_swaps, wp, first)) < 0)
+                r = hashmap_replace(m->swaps_by_proc_swaps, wp, first);
+                if (r < 0)
                         goto fail;
         }
 
@@ -400,15 +395,12 @@ static int swap_add_one(
 
         unit_add_to_dbus_queue(u);
 
-        free(e);
-
         return 0;
 
 fail:
         log_warning("Failed to load swap unit: %s", strerror(-r));
 
         free(wp);
-        free(e);
 
         if (delete && u)
                 unit_free(u);
@@ -430,11 +422,13 @@ static int swap_process_new_swap(Manager *m, const char *device, int prio, bool
                 /* So this is a proper swap device. Create swap units
                  * for all names this swap device is known under */
 
-                if (!(d = udev_device_new_from_devnum(m->udev, 'b', st.st_rdev)))
-                        return -ENOMEM;
+                d = udev_device_new_from_devnum(m->udev, 'b', st.st_rdev);
+                if (!d)
+                        return log_oom();
 
                 dn = udev_device_get_devnode(d);
-                if (dn)
+                /* Skip dn==device, since that case will be handled below */
+                if (dn && !streq(dn, device))
                         r = swap_add_one(m, dn, device, prio, false, false, set_flags);
 
                 /* Add additional units for all symlinks */
@@ -449,7 +443,8 @@ static int swap_process_new_swap(Manager *m, const char *device, int prio, bool
                                 continue;
 
                         if (stat(p, &st) >= 0)
-                                if ((!S_ISBLK(st.st_mode)) || st.st_rdev != udev_device_get_devnum(d))
+                                if ((!S_ISBLK(st.st_mode)) ||
+                                    st.st_rdev != udev_device_get_devnum(d))
                                         continue;
 
                         k = swap_add_one(m, p, device, prio, false, false, set_flags);
@@ -493,7 +488,8 @@ static void swap_set_state(Swap *s, SwapState state) {
                           swap_state_to_string(old_state),
                           swap_state_to_string(state));
 
-        unit_notify(UNIT(s), state_translation_table[old_state], state_translation_table[state], true);
+        unit_notify(UNIT(s), state_translation_table[old_state],
+                    state_translation_table[state], true);
 }
 
 static int swap_coldplug(Unit *u) {
@@ -521,10 +517,12 @@ static int swap_coldplug(Unit *u) {
                         if (s->control_pid <= 0)
                                 return -EBADMSG;
 
-                        if ((r = unit_watch_pid(UNIT(s), s->control_pid)) < 0)
+                        r = unit_watch_pid(UNIT(s), s->control_pid);
+                        if (r < 0)
                                 return r;
 
-                        if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0)
+                        r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch);
+                        if (r < 0)
                                 return r;
                 }
 
@@ -586,27 +584,30 @@ static int swap_spawn(Swap *s, ExecCommand *c, pid_t *_pid) {
         assert(c);
         assert(_pid);
 
-        if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0)
+        r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch);
+        if (r < 0)
                 goto fail;
 
-        if ((r = exec_spawn(c,
-                            NULL,
-                            &s->exec_context,
-                            NULL, 0,
-                            UNIT(s)->manager->environment,
-                            true,
-                            true,
-                            true,
-                            UNIT(s)->manager->confirm_spawn,
-                            UNIT(s)->cgroup_bondings,
-                            UNIT(s)->cgroup_attributes,
-                            NULL,
-                            UNIT(s)->id,
-                            NULL,
-                            &pid)) < 0)
+        r = exec_spawn(c,
+                       NULL,
+                       &s->exec_context,
+                       NULL, 0,
+                       UNIT(s)->manager->environment,
+                       true,
+                       true,
+                       true,
+                       UNIT(s)->manager->confirm_spawn,
+                       UNIT(s)->cgroup_bondings,
+                       UNIT(s)->cgroup_attributes,
+                       NULL,
+                       UNIT(s)->id,
+                       NULL,
+                       &pid);
+        if (r < 0)
                 goto fail;
 
-        if ((r = unit_watch_pid(UNIT(s), pid)) < 0)
+        r = unit_watch_pid(UNIT(s), pid);
+        if (r < 0)
                 /* FIXME: we need to do something here */
                 goto fail;
 
@@ -662,15 +663,18 @@ static void swap_enter_signal(Swap *s, SwapState state, SwapResult f) {
 
                 if (s->kill_context.kill_mode == KILL_CONTROL_GROUP) {
 
-                        if (!(pid_set = set_new(trivial_hash_func, trivial_compare_func))) {
-                                r = -ENOMEM;
+                        pid_set = set_new(trivial_hash_func, trivial_compare_func);
+                        if (!pid_set) {
+                                r = log_oom();
                                 goto fail;
                         }
 
                         /* Exclude the control pid from being killed via the cgroup */
-                        if (s->control_pid > 0)
-                                if ((r = set_put(pid_set, LONG_TO_PTR(s->control_pid))) < 0)
+                        if (s->control_pid > 0) {
+                                r = set_put(pid_set, LONG_TO_PTR(s->control_pid));
+                                if (r < 0)
                                         goto fail;
+                        }
 
                         r = cgroup_bonding_kill_list(UNIT(s)->cgroup_bondings, sig, true, false, pid_set, NULL);
                         if (r < 0) {
@@ -685,7 +689,8 @@ static void swap_enter_signal(Swap *s, SwapState state, SwapResult f) {
         }
 
         if (wait_for_exit) {
-                if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0)
+                r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch);
+                if (r < 0)
                         goto fail;
 
                 swap_set_state(s, state);
@@ -741,7 +746,8 @@ static void swap_enter_activating(Swap *s) {
 
         swap_unwatch_control_pid(s);
 
-        if ((r = swap_spawn(s, s->control_command, &s->control_pid)) < 0)
+        r = swap_spawn(s, s->control_command, &s->control_pid);
+        if (r < 0)
                 goto fail;
 
         swap_set_state(s, SWAP_ACTIVATING);
@@ -761,16 +767,17 @@ static void swap_enter_deactivating(Swap *s) {
         s->control_command_id = SWAP_EXEC_DEACTIVATE;
         s->control_command = s->exec_command + SWAP_EXEC_DEACTIVATE;
 
-        if ((r = exec_command_set(
-                             s->control_command,
+        r = exec_command_set(s->control_command,
                              "/sbin/swapoff",
                              s->what,
-                             NULL)) < 0)
+                             NULL);
+        if (r < 0)
                 goto fail;
 
         swap_unwatch_control_pid(s);
 
-        if ((r = swap_spawn(s, s->control_command, &s->control_pid)) < 0)
+        r = swap_spawn(s, s->control_command, &s->control_pid);
+        if (r < 0)
                 goto fail;
 
         swap_set_state(s, SWAP_DEACTIVATING);
@@ -860,7 +867,8 @@ static int swap_deserialize_item(Unit *u, const char *key, const char *value, FD
         if (streq(key, "state")) {
                 SwapState state;
 
-                if ((state = swap_state_from_string(value)) < 0)
+                state = swap_state_from_string(value);
+                if (state < 0)
                         log_debug("Failed to parse state value %s", value);
                 else
                         s->deserialized_state = state;
@@ -883,7 +891,8 @@ static int swap_deserialize_item(Unit *u, const char *key, const char *value, FD
         } else if (streq(key, "control-command")) {
                 SwapExecCommand id;
 
-                if ((id = swap_exec_command_from_string(value)) < 0)
+                id = swap_exec_command_from_string(value);
+                if (id < 0)
                         log_debug("Failed to parse exec-command value %s", value);
                 else {
                         s->control_command_id = id;
@@ -1050,18 +1059,18 @@ static int swap_load_proc_swaps(Manager *m, bool set_flags) {
                 char *dev = NULL, *d;
                 int prio = 0, k;
 
-                if ((k = fscanf(m->proc_swaps,
-                                "%ms "  /* device/file */
-                                "%*s "  /* type of swap */
-                                "%*s "  /* swap size */
-                                "%*s "  /* used */
-                                "%i\n", /* priority */
-                                &dev, &prio)) != 2) {
-
+                k = fscanf(m->proc_swaps,
+                           "%ms "  /* device/file */
+                           "%*s "  /* type of swap */
+                           "%*s "  /* swap size */
+                           "%*s "  /* used */
+                           "%i\n", /* priority */
+                           &dev, &prio);
+                if (k != 2) {
                         if (k == EOF)
                                 break;
 
-                        log_warning("Failed to parse /proc/swaps:%u.", i);
+                        log_warning("Failed to parse /proc/swaps:%u", i);
                         free(dev);
                         continue;
                 }
@@ -1100,7 +1109,8 @@ int swap_fd_event(Manager *m, int events) {
         assert(m);
         assert(events & EPOLLPRI);
 
-        if ((r = swap_load_proc_swaps(m, true)) < 0) {
+        r = swap_load_proc_swaps(m, true);
+        if (r < 0) {
                 log_error("Failed to reread /proc/swaps: %s", strerror(-r));
 
                 /* Reset flags, just in case, for late calls */
@@ -1240,7 +1250,8 @@ static int swap_enumerate(Manager *m) {
         assert(m);
 
         if (!m->proc_swaps) {
-                if (!(m->proc_swaps = fopen("/proc/swaps", "re")))
+                m->proc_swaps = fopen("/proc/swaps", "re");
+                if (!m->proc_swaps)
                         return (errno == ENOENT) ? 0 : -errno;
 
                 m->swap_watch.type = WATCH_SWAP;
@@ -1254,7 +1265,8 @@ static int swap_enumerate(Manager *m) {
                         return -errno;
         }
 
-        if ((r = swap_load_proc_swaps(m, false)) < 0)
+        r = swap_load_proc_swaps(m, false);
+        if (r < 0)
                 swap_shutdown(m);
 
         return r;