Branch data Line data Source code
1 : : /* SPDX-License-Identifier: LGPL-2.1+ */
2 : :
3 : : #include <errno.h>
4 : : #include <string.h>
5 : :
6 : : #include "alloc-util.h"
7 : : #include "btrfs-util.h"
8 : : #include "import-util.h"
9 : : #include "log.h"
10 : : #include "macro.h"
11 : : #include "nulstr-util.h"
12 : : #include "path-util.h"
13 : : #include "string-table.h"
14 : : #include "string-util.h"
15 : :
16 : 0 : int import_url_last_component(const char *url, char **ret) {
17 : : const char *e, *p;
18 : : char *s;
19 : :
20 : 0 : e = strchrnul(url, '?');
21 : :
22 [ # # # # ]: 0 : while (e > url && e[-1] == '/')
23 : 0 : e--;
24 : :
25 : 0 : p = e;
26 [ # # # # ]: 0 : while (p > url && p[-1] != '/')
27 : 0 : p--;
28 : :
29 [ # # ]: 0 : if (e <= p)
30 : 0 : return -EINVAL;
31 : :
32 : 0 : s = strndup(p, e - p);
33 [ # # ]: 0 : if (!s)
34 : 0 : return -ENOMEM;
35 : :
36 : 0 : *ret = s;
37 : 0 : return 0;
38 : : }
39 : :
40 : 0 : int import_url_change_last_component(const char *url, const char *suffix, char **ret) {
41 : : const char *e;
42 : : char *s;
43 : :
44 [ # # ]: 0 : assert(url);
45 [ # # ]: 0 : assert(ret);
46 : :
47 : 0 : e = strchrnul(url, '?');
48 : :
49 [ # # # # ]: 0 : while (e > url && e[-1] == '/')
50 : 0 : e--;
51 : :
52 [ # # # # ]: 0 : while (e > url && e[-1] != '/')
53 : 0 : e--;
54 : :
55 [ # # ]: 0 : if (e <= url)
56 : 0 : return -EINVAL;
57 : :
58 : 0 : s = new(char, (e - url) + strlen(suffix) + 1);
59 [ # # ]: 0 : if (!s)
60 : 0 : return -ENOMEM;
61 : :
62 : 0 : strcpy(mempcpy(s, url, e - url), suffix);
63 : 0 : *ret = s;
64 : 0 : return 0;
65 : : }
66 : :
67 : : static const char* const import_verify_table[_IMPORT_VERIFY_MAX] = {
68 : : [IMPORT_VERIFY_NO] = "no",
69 : : [IMPORT_VERIFY_CHECKSUM] = "checksum",
70 : : [IMPORT_VERIFY_SIGNATURE] = "signature",
71 : : };
72 : :
73 [ + + + + ]: 40 : DEFINE_STRING_TABLE_LOOKUP(import_verify, ImportVerify);
74 : :
75 : 0 : int tar_strip_suffixes(const char *name, char **ret) {
76 : : const char *e;
77 : : char *s;
78 : :
79 : 0 : e = endswith(name, ".tar");
80 [ # # ]: 0 : if (!e)
81 : 0 : e = endswith(name, ".tar.xz");
82 [ # # ]: 0 : if (!e)
83 : 0 : e = endswith(name, ".tar.gz");
84 [ # # ]: 0 : if (!e)
85 : 0 : e = endswith(name, ".tar.bz2");
86 [ # # ]: 0 : if (!e)
87 : 0 : e = endswith(name, ".tgz");
88 [ # # ]: 0 : if (!e)
89 : 0 : e = strchr(name, 0);
90 : :
91 [ # # ]: 0 : if (e <= name)
92 : 0 : return -EINVAL;
93 : :
94 : 0 : s = strndup(name, e - name);
95 [ # # ]: 0 : if (!s)
96 : 0 : return -ENOMEM;
97 : :
98 : 0 : *ret = s;
99 : 0 : return 0;
100 : : }
101 : :
102 : 0 : int raw_strip_suffixes(const char *p, char **ret) {
103 : :
104 : : static const char suffixes[] =
105 : : ".xz\0"
106 : : ".gz\0"
107 : : ".bz2\0"
108 : : ".raw\0"
109 : : ".qcow2\0"
110 : : ".img\0"
111 : : ".bin\0";
112 : :
113 : 0 : _cleanup_free_ char *q = NULL;
114 : :
115 : 0 : q = strdup(p);
116 [ # # ]: 0 : if (!q)
117 : 0 : return -ENOMEM;
118 : :
119 : 0 : for (;;) {
120 : : const char *sfx;
121 : 0 : bool changed = false;
122 : :
123 [ # # # # ]: 0 : NULSTR_FOREACH(sfx, suffixes) {
124 : : char *e;
125 : :
126 : 0 : e = endswith(q, sfx);
127 [ # # ]: 0 : if (e) {
128 : 0 : *e = 0;
129 : 0 : changed = true;
130 : : }
131 : : }
132 : :
133 [ # # ]: 0 : if (!changed)
134 : 0 : break;
135 : : }
136 : :
137 : 0 : *ret = TAKE_PTR(q);
138 : :
139 : 0 : return 0;
140 : : }
141 : :
142 : 0 : int import_assign_pool_quota_and_warn(const char *path) {
143 : : int r;
144 : :
145 : 0 : r = btrfs_subvol_auto_qgroup("/var/lib/machines", 0, true);
146 [ # # ]: 0 : if (r == -ENOTTY) {
147 [ # # ]: 0 : log_debug_errno(r, "Failed to set up default quota hierarchy for /var/lib/machines, as directory is not on btrfs or not a subvolume. Ignoring.");
148 : 0 : return 0;
149 : : }
150 [ # # ]: 0 : if (r < 0)
151 [ # # ]: 0 : return log_error_errno(r, "Failed to set up default quota hierarchy for /var/lib/machines: %m");
152 [ # # ]: 0 : if (r > 0)
153 [ # # ]: 0 : log_info("Set up default quota hierarchy for /var/lib/machines.");
154 : :
155 : 0 : r = btrfs_subvol_auto_qgroup(path, 0, true);
156 [ # # ]: 0 : if (r == -ENOTTY) {
157 [ # # ]: 0 : log_debug_errno(r, "Failed to set up quota hierarchy for %s, as directory is not on btrfs or not a subvolume. Ignoring.", path);
158 : 0 : return 0;
159 : : }
160 [ # # ]: 0 : if (r < 0)
161 [ # # ]: 0 : return log_error_errno(r, "Failed to set up default quota hierarchy for %s: %m", path);
162 [ # # ]: 0 : if (r > 0)
163 [ # # ]: 0 : log_debug("Set up default quota hierarchy for %s.", path);
164 : :
165 : 0 : return 0;
166 : : }
|
