src/basic/cap-list.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <errno.h>
   4 #include <string.h>
   5
   6 #include "alloc-util.h"
   7 #include "capability-util.h"
   8 #include "cap-list.h"
   9 #include "extract-word.h"
  10 #include "macro.h"
  11 #include "missing.h"
  12 #include "parse-util.h"
  13 #include "util.h"
  14
  15 static const struct capability_name* lookup_capability(register const char *str, register GPERF_LEN_TYPE len);
  16
  17 #include "cap-from-name.h"
  18 #include "cap-to-name.h"
  19
  20 const char *capability_to_name(int id) {
  21
  22         if (id < 0)
  23                 return NULL;
  24
  25         if ((size_t) id >= 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);
  39         if (r >= 0) {
  40                 if (i >= 0 && (size_t) i < ELEMENTSOF(capability_names))
  41                         return i;
  42                 else
  43                         return -EINVAL;
  44         }
  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 }
  53
  54 int capability_list_length(void) {
  55         return (int) ELEMENTSOF(capability_names);
  56 }
  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
  76                         if (!GREEDY_REALLOC(str, allocated, n + add + 2))
  77                                 return -ENOMEM;
  78
  79                         strcpy(mempcpy(str + n, p, add), " ");
  80                         n += add + 1;
  81                 }
  82
  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 */
  87
  88         *s = TAKE_PTR(str);
  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 }