1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
2
3 #include "fileio.h"
4 #include "ordered-set.h"
5 #include "strv.h"
6
_ordered_set_ensure_allocated(OrderedSet ** s,const struct hash_ops * ops HASHMAP_DEBUG_PARAMS)7 int _ordered_set_ensure_allocated(OrderedSet **s, const struct hash_ops *ops HASHMAP_DEBUG_PARAMS) {
8 if (*s)
9 return 0;
10
11 *s = _ordered_set_new(ops HASHMAP_DEBUG_PASS_ARGS);
12 if (!*s)
13 return -ENOMEM;
14
15 return 0;
16 }
17
_ordered_set_ensure_put(OrderedSet ** s,const struct hash_ops * ops,void * p HASHMAP_DEBUG_PARAMS)18 int _ordered_set_ensure_put(OrderedSet **s, const struct hash_ops *ops, void *p HASHMAP_DEBUG_PARAMS) {
19 int r;
20
21 r = _ordered_set_ensure_allocated(s, ops HASHMAP_DEBUG_PASS_ARGS);
22 if (r < 0)
23 return r;
24
25 return ordered_set_put(*s, p);
26 }
27
ordered_set_consume(OrderedSet * s,void * p)28 int ordered_set_consume(OrderedSet *s, void *p) {
29 int r;
30
31 r = ordered_set_put(s, p);
32 if (r <= 0)
33 free(p);
34
35 return r;
36 }
37
_ordered_set_put_strdup(OrderedSet ** s,const char * p HASHMAP_DEBUG_PARAMS)38 int _ordered_set_put_strdup(OrderedSet **s, const char *p HASHMAP_DEBUG_PARAMS) {
39 char *c;
40 int r;
41
42 assert(s);
43 assert(p);
44
45 r = _ordered_set_ensure_allocated(s, &string_hash_ops_free HASHMAP_DEBUG_PASS_ARGS);
46 if (r < 0)
47 return r;
48
49 if (ordered_set_contains(*s, p))
50 return 0;
51
52 c = strdup(p);
53 if (!c)
54 return -ENOMEM;
55
56 return ordered_set_consume(*s, c);
57 }
58
_ordered_set_put_strdupv(OrderedSet ** s,char ** l HASHMAP_DEBUG_PARAMS)59 int _ordered_set_put_strdupv(OrderedSet **s, char **l HASHMAP_DEBUG_PARAMS) {
60 int n = 0, r;
61
62 STRV_FOREACH(i, l) {
63 r = _ordered_set_put_strdup(s, *i HASHMAP_DEBUG_PASS_ARGS);
64 if (r < 0)
65 return r;
66
67 n += r;
68 }
69
70 return n;
71 }
72
ordered_set_put_string_set(OrderedSet ** s,OrderedSet * l)73 int ordered_set_put_string_set(OrderedSet **s, OrderedSet *l) {
74 int n = 0, r;
75 char *p;
76
77 /* Like ordered_set_put_strv, but for an OrderedSet of strings */
78
79 ORDERED_SET_FOREACH(p, l) {
80 r = ordered_set_put_strdup(s, p);
81 if (r < 0)
82 return r;
83
84 n += r;
85 }
86
87 return n;
88 }
89
ordered_set_print(FILE * f,const char * field,OrderedSet * s)90 void ordered_set_print(FILE *f, const char *field, OrderedSet *s) {
91 bool space = false;
92 char *p;
93
94 if (ordered_set_isempty(s))
95 return;
96
97 fputs(field, f);
98
99 ORDERED_SET_FOREACH(p, s)
100 fputs_with_space(f, p, NULL, &space);
101
102 fputc('\n', f);
103 }
104