Add set_consume which always takes ownership

[elogind.git] / src / shared / set.c

diff --git a/src/shared/set.c b/src/shared/set.c
index 5f83c50839ae78b831d6f3853eea7ce6c6eb89a4..c338dc3a44c462176a96dce3c3e189c9c9afa9f4 100644 (file)
--- a/src/shared/set.c
+++ b/src/shared/set.c
@@ -49,6 +49,13 @@ int set_put(Set *s, void *value) {
         return hashmap_put(MAKE_HASHMAP(s), value, value);
 }
 
+int set_consume(Set *s, void *value) {
+        int r = set_put(s, value);
+        if (r < 0)
+                free(value);
+        return r;
+}
+
 int set_replace(Set *s, void *value) {
         return hashmap_replace(MAKE_HASHMAP(s), value, value);
 }