mirror of
https://github.com/systemd/systemd.git
synced 2026-07-25 00:35:55 +00:00
1c0ade2e1f
We nowadays support unprivileged invocation of systemd-nspawn + systemd-vmspawn, but there was no support for discovering suitable disk images (i.e. no per-user counterpart of /var/lib/machines). Add this now, and hook it up everywhere. Instead of hardcoding machined's, importd's, portabled's, sysupdated's image discovery to RUNTIME_SCOPE_SYSTEM I introduced a field that make the scope variable, even if this field is always initialized to RUNTIME_SCOPE_SYSTEM for now. I think these four services should eventually be updated to support a per-user concept too, this is preparation for that, even though it doesn't outright add support for this. This is for the largest part not user visible, except for in nspawn, vmspawn and the dissect tool. For the latter I added a pair of --user/--system switches to select the discovery scope.
306 lines
9.9 KiB
C
306 lines
9.9 KiB
C
/* SPDX-License-Identifier: LGPL-2.1-or-later */
|
|
|
|
#include <getopt.h>
|
|
#include <locale.h>
|
|
|
|
#include "sd-event.h"
|
|
#include "sd-id128.h"
|
|
|
|
#include "alloc-util.h"
|
|
#include "ansi-color.h"
|
|
#include "build.h"
|
|
#include "discover-image.h"
|
|
#include "export-raw.h"
|
|
#include "export-tar.h"
|
|
#include "fd-util.h"
|
|
#include "fs-util.h"
|
|
#include "hostname-util.h"
|
|
#include "import-common.h"
|
|
#include "import-util.h"
|
|
#include "main-func.h"
|
|
#include "signal-util.h"
|
|
#include "string-util.h"
|
|
#include "terminal-util.h"
|
|
#include "verbs.h"
|
|
|
|
static ImportCompressType arg_compress = IMPORT_COMPRESS_UNKNOWN;
|
|
static ImageClass arg_class = IMAGE_MACHINE;
|
|
static RuntimeScope arg_runtime_scope = _RUNTIME_SCOPE_INVALID;
|
|
|
|
static void determine_compression_from_filename(const char *p) {
|
|
|
|
if (arg_compress != IMPORT_COMPRESS_UNKNOWN)
|
|
return;
|
|
|
|
if (!p) {
|
|
arg_compress = IMPORT_COMPRESS_UNCOMPRESSED;
|
|
return;
|
|
}
|
|
|
|
if (endswith(p, ".xz"))
|
|
arg_compress = IMPORT_COMPRESS_XZ;
|
|
else if (endswith(p, ".gz"))
|
|
arg_compress = IMPORT_COMPRESS_GZIP;
|
|
else if (endswith(p, ".bz2"))
|
|
arg_compress = IMPORT_COMPRESS_BZIP2;
|
|
else
|
|
arg_compress = IMPORT_COMPRESS_UNCOMPRESSED;
|
|
}
|
|
|
|
static void on_tar_finished(TarExport *export, int error, void *userdata) {
|
|
sd_event *event = userdata;
|
|
assert(export);
|
|
|
|
if (error == 0)
|
|
log_info("Operation completed successfully.");
|
|
|
|
sd_event_exit(event, abs(error));
|
|
}
|
|
|
|
static int export_tar(int argc, char *argv[], void *userdata) {
|
|
_cleanup_(tar_export_unrefp) TarExport *export = NULL;
|
|
_cleanup_(sd_event_unrefp) sd_event *event = NULL;
|
|
_cleanup_(image_unrefp) Image *image = NULL;
|
|
const char *path = NULL, *local = NULL;
|
|
_cleanup_close_ int open_fd = -EBADF;
|
|
int r, fd;
|
|
|
|
local = argv[1];
|
|
if (image_name_is_valid(local)) {
|
|
r = image_find(arg_runtime_scope, arg_class, local, NULL, &image);
|
|
if (r == -ENOENT)
|
|
return log_error_errno(r, "Image %s not found.", local);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to look for image %s: %m", local);
|
|
|
|
local = image->path;
|
|
} else
|
|
local = argv[1];
|
|
|
|
if (argc >= 3)
|
|
path = argv[2];
|
|
path = empty_or_dash_to_null(path);
|
|
|
|
determine_compression_from_filename(path);
|
|
|
|
if (path) {
|
|
open_fd = open(path, O_WRONLY|O_CREAT|O_TRUNC|O_CLOEXEC|O_NOCTTY, 0666);
|
|
if (open_fd < 0)
|
|
return log_error_errno(errno, "Failed to open tar image for export: %m");
|
|
|
|
fd = open_fd;
|
|
|
|
log_info("Exporting '%s', saving to '%s' with compression '%s'.", local, path, import_compress_type_to_string(arg_compress));
|
|
} else {
|
|
_cleanup_free_ char *pretty = NULL;
|
|
|
|
fd = STDOUT_FILENO;
|
|
|
|
(void) fd_get_path(fd, &pretty);
|
|
log_info("Exporting '%s', saving to '%s' with compression '%s'.", local, strna(pretty), import_compress_type_to_string(arg_compress));
|
|
}
|
|
|
|
r = import_allocate_event_with_signals(&event);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
r = tar_export_new(&export, event, on_tar_finished, event);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to allocate exporter: %m");
|
|
|
|
r = tar_export_start(export, local, fd, arg_compress);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to export image: %m");
|
|
|
|
r = sd_event_loop(event);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to run event loop: %m");
|
|
|
|
log_info("Exiting.");
|
|
return -r;
|
|
}
|
|
|
|
static void on_raw_finished(RawExport *export, int error, void *userdata) {
|
|
sd_event *event = userdata;
|
|
assert(export);
|
|
|
|
if (error == 0)
|
|
log_info("Operation completed successfully.");
|
|
|
|
sd_event_exit(event, abs(error));
|
|
}
|
|
|
|
static int export_raw(int argc, char *argv[], void *userdata) {
|
|
_cleanup_(raw_export_unrefp) RawExport *export = NULL;
|
|
_cleanup_(sd_event_unrefp) sd_event *event = NULL;
|
|
_cleanup_(image_unrefp) Image *image = NULL;
|
|
const char *path = NULL, *local = NULL;
|
|
_cleanup_close_ int open_fd = -EBADF;
|
|
int r, fd;
|
|
|
|
local = argv[1];
|
|
if (image_name_is_valid(local)) {
|
|
r = image_find(arg_runtime_scope, arg_class, local, NULL, &image);
|
|
if (r == -ENOENT)
|
|
return log_error_errno(r, "Image %s not found.", local);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to look for image %s: %m", local);
|
|
|
|
local = image->path;
|
|
} else
|
|
local = argv[1];
|
|
|
|
if (argc >= 3)
|
|
path = argv[2];
|
|
path = empty_or_dash_to_null(path);
|
|
|
|
determine_compression_from_filename(path);
|
|
|
|
if (path) {
|
|
open_fd = open(path, O_WRONLY|O_CREAT|O_TRUNC|O_CLOEXEC|O_NOCTTY, 0666);
|
|
if (open_fd < 0)
|
|
return log_error_errno(errno, "Failed to open raw image for export: %m");
|
|
|
|
fd = open_fd;
|
|
|
|
log_info("Exporting '%s', saving to '%s' with compression '%s'.", local, path, import_compress_type_to_string(arg_compress));
|
|
} else {
|
|
_cleanup_free_ char *pretty = NULL;
|
|
|
|
fd = STDOUT_FILENO;
|
|
|
|
(void) fd_get_path(fd, &pretty);
|
|
log_info("Exporting '%s', saving to '%s' with compression '%s'.", local, strna(pretty), import_compress_type_to_string(arg_compress));
|
|
}
|
|
|
|
r = import_allocate_event_with_signals(&event);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
r = raw_export_new(&export, event, on_raw_finished, event);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to allocate exporter: %m");
|
|
|
|
r = raw_export_start(export, local, fd, arg_compress);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to export image: %m");
|
|
|
|
r = sd_event_loop(event);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to run event loop: %m");
|
|
|
|
log_info("Exiting.");
|
|
return -r;
|
|
}
|
|
|
|
static int help(int argc, char *argv[], void *userdata) {
|
|
printf("%1$s [OPTIONS...] {COMMAND} ...\n"
|
|
"\n%4$sExport disk images.%5$s\n"
|
|
"\n%2$sCommands:%3$s\n"
|
|
" tar NAME [FILE] Export a TAR image\n"
|
|
" raw NAME [FILE] Export a RAW image\n"
|
|
"\n%2$sOptions:%3$s\n"
|
|
" -h --help Show this help\n"
|
|
" --version Show package version\n"
|
|
" --format=FORMAT Select format\n"
|
|
" --class=CLASS Select image class (machine, sysext, confext,\n"
|
|
" portable)\n",
|
|
program_invocation_short_name,
|
|
ansi_underline(),
|
|
ansi_normal(),
|
|
ansi_highlight(),
|
|
ansi_normal());
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int parse_argv(int argc, char *argv[]) {
|
|
|
|
enum {
|
|
ARG_VERSION = 0x100,
|
|
ARG_FORMAT,
|
|
ARG_CLASS,
|
|
};
|
|
|
|
static const struct option options[] = {
|
|
{ "help", no_argument, NULL, 'h' },
|
|
{ "version", no_argument, NULL, ARG_VERSION },
|
|
{ "format", required_argument, NULL, ARG_FORMAT },
|
|
{ "class", required_argument, NULL, ARG_CLASS },
|
|
{}
|
|
};
|
|
|
|
int c;
|
|
|
|
assert(argc >= 0);
|
|
assert(argv);
|
|
|
|
while ((c = getopt_long(argc, argv, "h", options, NULL)) >= 0)
|
|
|
|
switch (c) {
|
|
|
|
case 'h':
|
|
return help(0, NULL, NULL);
|
|
|
|
case ARG_VERSION:
|
|
return version();
|
|
|
|
case ARG_FORMAT:
|
|
if (streq(optarg, "uncompressed"))
|
|
arg_compress = IMPORT_COMPRESS_UNCOMPRESSED;
|
|
else if (streq(optarg, "xz"))
|
|
arg_compress = IMPORT_COMPRESS_XZ;
|
|
else if (streq(optarg, "gzip"))
|
|
arg_compress = IMPORT_COMPRESS_GZIP;
|
|
else if (streq(optarg, "bzip2"))
|
|
arg_compress = IMPORT_COMPRESS_BZIP2;
|
|
else
|
|
return log_error_errno(SYNTHETIC_ERRNO(EINVAL),
|
|
"Unknown format: %s", optarg);
|
|
break;
|
|
|
|
case ARG_CLASS:
|
|
arg_class = image_class_from_string(optarg);
|
|
if (arg_class < 0)
|
|
return log_error_errno(arg_class, "Failed to parse --class= argument: %s", optarg);
|
|
|
|
break;
|
|
|
|
case '?':
|
|
return -EINVAL;
|
|
|
|
default:
|
|
assert_not_reached();
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
static int export_main(int argc, char *argv[]) {
|
|
static const Verb verbs[] = {
|
|
{ "help", VERB_ANY, VERB_ANY, 0, help },
|
|
{ "tar", 2, 3, 0, export_tar },
|
|
{ "raw", 2, 3, 0, export_raw },
|
|
{}
|
|
};
|
|
|
|
return dispatch_verb(argc, argv, verbs, NULL);
|
|
}
|
|
|
|
static int run(int argc, char *argv[]) {
|
|
int r;
|
|
|
|
setlocale(LC_ALL, "");
|
|
log_setup();
|
|
|
|
r = parse_argv(argc, argv);
|
|
if (r <= 0)
|
|
return r;
|
|
|
|
(void) ignore_signals(SIGPIPE);
|
|
|
|
return export_main(argc, argv);
|
|
}
|
|
|
|
DEFINE_MAIN_FUNCTION(run);
|