]>
Commit | Line | Data |
---|---|---|
53e1b683 | 1 | /* SPDX-License-Identifier: LGPL-2.1+ */ |
2822da4f | 2 | |
11c3a366 | 3 | #include <errno.h> |
2822da4f LP |
4 | #include <string.h> |
5 | ||
dd1f5bd0 YW |
6 | #include "alloc-util.h" |
7 | #include "capability-util.h" | |
2822da4f | 8 | #include "cap-list.h" |
dd1f5bd0 | 9 | #include "extract-word.h" |
93cc7779 | 10 | #include "macro.h" |
2822da4f | 11 | #include "missing.h" |
6bedfcbb LP |
12 | #include "parse-util.h" |
13 | #include "util.h" | |
2822da4f | 14 | |
c9f7b4d3 | 15 | static const struct capability_name* lookup_capability(register const char *str, register GPERF_LEN_TYPE len); |
2822da4f | 16 | |
2822da4f | 17 | #include "cap-from-name.h" |
cf0fbc49 | 18 | #include "cap-to-name.h" |
2822da4f LP |
19 | |
20 | const char *capability_to_name(int id) { | |
21 | ||
22 | if (id < 0) | |
23 | return NULL; | |
24 | ||
25 | if (id >= (int) ELEMENTSOF(capability_names)) | |
26 | return NULL; | |
27 | ||
28 | return capability_names[id]; | |
29 | } | |
30 | ||
31 | int capability_from_name(const char *name) { | |
32 | const struct capability_name *sc; | |
33 | int r, i; | |
34 | ||
35 | assert(name); | |
36 | ||
37 | /* Try to parse numeric capability */ | |
38 | r = safe_atoi(name, &i); | |
db4bd5bd YW |
39 | if (r >= 0) { |
40 | if (i >= 0 && i < (int) ELEMENTSOF(capability_names)) | |
41 | return i; | |
42 | else | |
43 | return -EINVAL; | |
44 | } | |
2822da4f LP |
45 | |
46 | /* Try to parse string capability */ | |
47 | sc = lookup_capability(name, strlen(name)); | |
48 | if (!sc) | |
49 | return -EINVAL; | |
50 | ||
51 | return sc->id; | |
52 | } | |
097df453 FB |
53 | |
54 | int capability_list_length(void) { | |
55 | return (int) ELEMENTSOF(capability_names); | |
56 | } | |
dd1f5bd0 YW |
57 | |
58 | int capability_set_to_string_alloc(uint64_t set, char **s) { | |
59 | _cleanup_free_ char *str = NULL; | |
60 | unsigned long i; | |
61 | size_t allocated = 0, n = 0; | |
62 | ||
63 | assert(s); | |
64 | ||
65 | for (i = 0; i < cap_last_cap(); i++) | |
66 | if (set & (UINT64_C(1) << i)) { | |
67 | const char *p; | |
68 | size_t add; | |
69 | ||
70 | p = capability_to_name(i); | |
71 | if (!p) | |
72 | return -EINVAL; | |
73 | ||
74 | add = strlen(p); | |
75 | ||
6088cefb | 76 | if (!GREEDY_REALLOC(str, allocated, n + add + 2)) |
dd1f5bd0 YW |
77 | return -ENOMEM; |
78 | ||
79 | strcpy(mempcpy(str + n, p, add), " "); | |
80 | n += add + 1; | |
81 | } | |
82 | ||
6088cefb ZJS |
83 | if (!GREEDY_REALLOC(str, allocated, n + 1)) |
84 | return -ENOMEM; | |
85 | ||
86 | str[n > 0 ? n - 1 : 0] = '\0'; /* truncate the last space, if it's there */ | |
dd1f5bd0 | 87 | |
ae2a15bc | 88 | *s = TAKE_PTR(str); |
dd1f5bd0 YW |
89 | |
90 | return 0; | |
91 | } | |
92 | ||
93 | int capability_set_from_string(const char *s, uint64_t *set) { | |
94 | uint64_t val = 0; | |
95 | const char *p; | |
96 | ||
97 | assert(set); | |
98 | ||
99 | for (p = s;;) { | |
100 | _cleanup_free_ char *word = NULL; | |
101 | int r; | |
102 | ||
103 | r = extract_first_word(&p, &word, NULL, EXTRACT_QUOTES); | |
104 | if (r == -ENOMEM) | |
105 | return r; | |
106 | if (r <= 0) | |
107 | break; | |
108 | ||
109 | r = capability_from_name(word); | |
110 | if (r < 0) | |
111 | continue; | |
112 | ||
113 | val |= ((uint64_t) UINT64_C(1)) << (uint64_t) r; | |
114 | } | |
115 | ||
116 | *set = val; | |
117 | ||
118 | return 0; | |
119 | } |