5d13a15b1d
Double newlines (i.e. one empty lines) are great to structure code. But let's avoid triple newlines (i.e. two empty lines), quadruple newlines, quintuple newlines, …, that's just spurious whitespace. It's an easy way to drop 121 lines of code, and keeps the coding style of our sources a bit tigther.
172 lines
4.2 KiB
C
172 lines
4.2 KiB
C
/* SPDX-License-Identifier: LGPL-2.1+ */
|
|
/***
|
|
This file is part of systemd.
|
|
|
|
Copyright 2015 Lennart Poettering
|
|
***/
|
|
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
|
|
#include "alloc-util.h"
|
|
#include "btrfs-util.h"
|
|
#include "import-util.h"
|
|
#include "log.h"
|
|
#include "macro.h"
|
|
#include "path-util.h"
|
|
#include "string-table.h"
|
|
#include "string-util.h"
|
|
#include "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_info("Set up default quota hierarchy for %s.", path);
|
|
|
|
return 0;
|
|
}
|