180 lines
4.5 KiB
C
180 lines
4.5 KiB
C
/* SPDX-License-Identifier: LGPL-2.1-or-later */
|
|
|
|
#include <errno.h>
|
|
|
|
#include "alloc-util.h"
|
|
#include "btrfs-util.h"
|
|
#include "chattr-util.h"
|
|
#include "errno-util.h"
|
|
#include "import-util.h"
|
|
#include "log.h"
|
|
#include "macro.h"
|
|
#include "nulstr-util.h"
|
|
#include "path-util.h"
|
|
#include "string-table.h"
|
|
#include "string-util.h"
|
|
|
|
int import_url_last_component(const char *url, char **ret) {
|
|
const char *e, *p;
|
|
char *s;
|
|
|
|
e = strchrnul(url, '?');
|
|
|
|
while (e > url && e[-1] == '/')
|
|
e--;
|
|
|
|
p = e;
|
|
while (p > url && p[-1] != '/')
|
|
p--;
|
|
|
|
if (e <= p)
|
|
return -EINVAL;
|
|
|
|
s = strndup(p, e - p);
|
|
if (!s)
|
|
return -ENOMEM;
|
|
|
|
*ret = s;
|
|
return 0;
|
|
}
|
|
|
|
int import_url_change_last_component(const char *url, const char *suffix, char **ret) {
|
|
const char *e;
|
|
char *s;
|
|
|
|
assert(url);
|
|
assert(ret);
|
|
|
|
e = strchrnul(url, '?');
|
|
|
|
while (e > url && e[-1] == '/')
|
|
e--;
|
|
|
|
while (e > url && e[-1] != '/')
|
|
e--;
|
|
|
|
if (e <= url)
|
|
return -EINVAL;
|
|
|
|
s = new(char, (e - url) + strlen(suffix) + 1);
|
|
if (!s)
|
|
return -ENOMEM;
|
|
|
|
strcpy(mempcpy(s, url, e - url), suffix);
|
|
*ret = s;
|
|
return 0;
|
|
}
|
|
|
|
static const char* const import_verify_table[_IMPORT_VERIFY_MAX] = {
|
|
[IMPORT_VERIFY_NO] = "no",
|
|
[IMPORT_VERIFY_CHECKSUM] = "checksum",
|
|
[IMPORT_VERIFY_SIGNATURE] = "signature",
|
|
};
|
|
|
|
DEFINE_STRING_TABLE_LOOKUP(import_verify, ImportVerify);
|
|
|
|
int tar_strip_suffixes(const char *name, char **ret) {
|
|
const char *e;
|
|
char *s;
|
|
|
|
e = endswith(name, ".tar");
|
|
if (!e)
|
|
e = endswith(name, ".tar.xz");
|
|
if (!e)
|
|
e = endswith(name, ".tar.gz");
|
|
if (!e)
|
|
e = endswith(name, ".tar.bz2");
|
|
if (!e)
|
|
e = endswith(name, ".tgz");
|
|
if (!e)
|
|
e = strchr(name, 0);
|
|
|
|
if (e <= name)
|
|
return -EINVAL;
|
|
|
|
s = strndup(name, e - name);
|
|
if (!s)
|
|
return -ENOMEM;
|
|
|
|
*ret = s;
|
|
return 0;
|
|
}
|
|
|
|
int raw_strip_suffixes(const char *p, char **ret) {
|
|
|
|
static const char suffixes[] =
|
|
".xz\0"
|
|
".gz\0"
|
|
".bz2\0"
|
|
".raw\0"
|
|
".qcow2\0"
|
|
".img\0"
|
|
".bin\0";
|
|
|
|
_cleanup_free_ char *q = NULL;
|
|
|
|
q = strdup(p);
|
|
if (!q)
|
|
return -ENOMEM;
|
|
|
|
for (;;) {
|
|
const char *sfx;
|
|
bool changed = false;
|
|
|
|
NULSTR_FOREACH(sfx, suffixes) {
|
|
char *e;
|
|
|
|
e = endswith(q, sfx);
|
|
if (e) {
|
|
*e = 0;
|
|
changed = true;
|
|
}
|
|
}
|
|
|
|
if (!changed)
|
|
break;
|
|
}
|
|
|
|
*ret = TAKE_PTR(q);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int import_assign_pool_quota_and_warn(const char *path) {
|
|
int r;
|
|
|
|
r = btrfs_subvol_auto_qgroup("/var/lib/machines", 0, true);
|
|
if (r == -ENOTTY) {
|
|
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.");
|
|
return 0;
|
|
}
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to set up default quota hierarchy for /var/lib/machines: %m");
|
|
if (r > 0)
|
|
log_info("Set up default quota hierarchy for /var/lib/machines.");
|
|
|
|
r = btrfs_subvol_auto_qgroup(path, 0, true);
|
|
if (r == -ENOTTY) {
|
|
log_debug_errno(r, "Failed to set up quota hierarchy for %s, as directory is not on btrfs or not a subvolume. Ignoring.", path);
|
|
return 0;
|
|
}
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to set up default quota hierarchy for %s: %m", path);
|
|
if (r > 0)
|
|
log_debug("Set up default quota hierarchy for %s.", path);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int import_set_nocow_and_log(int fd, const char *path) {
|
|
int r;
|
|
|
|
r = chattr_fd(fd, FS_NOCOW_FL, FS_NOCOW_FL, NULL);
|
|
if (r < 0)
|
|
return log_full_errno(
|
|
ERRNO_IS_NOT_SUPPORTED(r) ? LOG_DEBUG : LOG_WARNING,
|
|
r, "Failed to set file attributes on %s: %m", path);
|
|
|
|
return 0;
|
|
}
|