1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2015 Lennart Poettering
8 systemd is free software; you can redistribute it and/or modify it
9 under the terms of the GNU Lesser General Public License as published by
10 the Free Software Foundation; either version 2.1 of the License, or
11 (at your option) any later version.
13 systemd is distributed in the hope that it will be useful, but
14 WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 Lesser General Public License for more details.
18 You should have received a copy of the GNU Lesser General Public License
19 along with systemd; If not, see <http://www.gnu.org/licenses/>.
25 #include "event-util.h"
28 #include "machine-image.h"
29 #include "import-util.h"
30 #include "export-tar.h"
31 #include "export-raw.h"
33 static ImportCompressType arg_compress = IMPORT_COMPRESS_UNKNOWN;
35 static void determine_compression_from_filename(const char *p) {
37 if (arg_compress != IMPORT_COMPRESS_UNKNOWN)
41 arg_compress = IMPORT_COMPRESS_UNCOMPRESSED;
45 if (endswith(p, ".xz"))
46 arg_compress = IMPORT_COMPRESS_XZ;
47 else if (endswith(p, ".gz"))
48 arg_compress = IMPORT_COMPRESS_GZIP;
49 else if (endswith(p, ".bz2"))
50 arg_compress = IMPORT_COMPRESS_BZIP2;
52 arg_compress = IMPORT_COMPRESS_UNCOMPRESSED;
55 static int interrupt_signal_handler(sd_event_source *s, const struct signalfd_siginfo *si, void *userdata) {
56 log_notice("Transfer aborted.");
57 sd_event_exit(sd_event_source_get_event(s), EINTR);
61 static void on_tar_finished(TarExport *export, int error, void *userdata) {
62 sd_event *event = userdata;
66 log_info("Operation completed successfully.");
68 sd_event_exit(event, abs(error));
71 static int export_tar(int argc, char *argv[], void *userdata) {
72 _cleanup_(tar_export_unrefp) TarExport *export = NULL;
73 _cleanup_event_unref_ sd_event *event = NULL;
74 _cleanup_(image_unrefp) Image *image = NULL;
75 const char *path = NULL, *local = NULL;
76 _cleanup_close_ int open_fd = -1;
79 if (machine_name_is_valid(argv[1])) {
80 r = image_find(argv[1], &image);
82 return log_error_errno(r, "Failed to look for machine %s: %m", argv[1]);
84 log_error("Machine image %s not found.", argv[1]);
94 if (isempty(path) || streq(path, "-"))
97 determine_compression_from_filename(path);
100 open_fd = open(path, O_WRONLY|O_CREAT|O_TRUNC|O_CLOEXEC|O_NOCTTY, 0666);
102 return log_error_errno(errno, "Failed to open tar image for export: %m");
106 log_info("Exporting '%s', saving to '%s' with compression '%s'.", local, path, import_compress_type_to_string(arg_compress));
108 _cleanup_free_ char *pretty = NULL;
112 (void) readlink_malloc("/proc/self/fd/1", &pretty);
113 log_info("Exporting '%s', saving to '%s' with compression '%s'.", local, strna(pretty), import_compress_type_to_string(arg_compress));
116 r = sd_event_default(&event);
118 return log_error_errno(r, "Failed to allocate event loop: %m");
120 assert_se(sigprocmask_many(SIG_BLOCK, SIGTERM, SIGINT, -1) == 0);
121 sd_event_add_signal(event, NULL, SIGTERM, interrupt_signal_handler, NULL);
122 sd_event_add_signal(event, NULL, SIGINT, interrupt_signal_handler, NULL);
124 r = tar_export_new(&export, event, on_tar_finished, event);
126 return log_error_errno(r, "Failed to allocate exporter: %m");
128 r = tar_export_start(export, local, fd, arg_compress);
130 return log_error_errno(r, "Failed to export image: %m");
132 r = sd_event_loop(event);
134 return log_error_errno(r, "Failed to run event loop: %m");
136 log_info("Exiting.");
140 static void on_raw_finished(RawExport *export, int error, void *userdata) {
141 sd_event *event = userdata;
145 log_info("Operation completed successfully.");
147 sd_event_exit(event, abs(error));
150 static int export_raw(int argc, char *argv[], void *userdata) {
151 _cleanup_(raw_export_unrefp) RawExport *export = NULL;
152 _cleanup_event_unref_ sd_event *event = NULL;
153 _cleanup_(image_unrefp) Image *image = NULL;
154 const char *path = NULL, *local = NULL;
155 _cleanup_close_ int open_fd = -1;
158 if (machine_name_is_valid(argv[1])) {
159 r = image_find(argv[1], &image);
161 return log_error_errno(r, "Failed to look for machine %s: %m", argv[1]);
163 log_error("Machine image %s not found.", argv[1]);
173 if (isempty(path) || streq(path, "-"))
176 determine_compression_from_filename(path);
179 open_fd = open(path, O_WRONLY|O_CREAT|O_TRUNC|O_CLOEXEC|O_NOCTTY, 0666);
181 return log_error_errno(errno, "Failed to open raw image for export: %m");
185 log_info("Exporting '%s', saving to '%s' with compression '%s'.", local, path, import_compress_type_to_string(arg_compress));
187 _cleanup_free_ char *pretty = NULL;
191 (void) readlink_malloc("/proc/self/fd/1", &pretty);
192 log_info("Exporting '%s', saving to '%s' with compression '%s'.", local, strna(pretty), import_compress_type_to_string(arg_compress));
195 r = sd_event_default(&event);
197 return log_error_errno(r, "Failed to allocate event loop: %m");
199 assert_se(sigprocmask_many(SIG_BLOCK, SIGTERM, SIGINT, -1) == 0);
200 sd_event_add_signal(event, NULL, SIGTERM, interrupt_signal_handler, NULL);
201 sd_event_add_signal(event, NULL, SIGINT, interrupt_signal_handler, NULL);
203 r = raw_export_new(&export, event, on_raw_finished, event);
205 return log_error_errno(r, "Failed to allocate exporter: %m");
207 r = raw_export_start(export, local, fd, arg_compress);
209 return log_error_errno(r, "Failed to export image: %m");
211 r = sd_event_loop(event);
213 return log_error_errno(r, "Failed to run event loop: %m");
215 log_info("Exiting.");
219 static int help(int argc, char *argv[], void *userdata) {
221 printf("%s [OPTIONS...] {COMMAND} ...\n\n"
222 "Export container or virtual machine images.\n\n"
223 " -h --help Show this help\n"
224 " --version Show package version\n"
225 " --format=FORMAT Select format\n\n"
227 " tar NAME [FILE] Export a TAR image\n"
228 " raw NAME [FILE] Export a RAW image\n",
229 program_invocation_short_name);
234 static int parse_argv(int argc, char *argv[]) {
241 static const struct option options[] = {
242 { "help", no_argument, NULL, 'h' },
243 { "version", no_argument, NULL, ARG_VERSION },
244 { "format", required_argument, NULL, ARG_FORMAT },
253 while ((c = getopt_long(argc, argv, "h", options, NULL)) >= 0)
258 return help(0, NULL, NULL);
261 puts(PACKAGE_STRING);
262 puts(SYSTEMD_FEATURES);
266 if (streq(optarg, "uncompressed"))
267 arg_compress = IMPORT_COMPRESS_UNCOMPRESSED;
268 else if (streq(optarg, "xz"))
269 arg_compress = IMPORT_COMPRESS_XZ;
270 else if (streq(optarg, "gzip"))
271 arg_compress = IMPORT_COMPRESS_GZIP;
272 else if (streq(optarg, "bzip2"))
273 arg_compress = IMPORT_COMPRESS_BZIP2;
275 log_error("Unknown format: %s", optarg);
284 assert_not_reached("Unhandled option");
290 static int export_main(int argc, char *argv[]) {
292 static const Verb verbs[] = {
293 { "help", VERB_ANY, VERB_ANY, 0, help },
294 { "tar", 2, 3, 0, export_tar },
295 { "raw", 2, 3, 0, export_raw },
299 return dispatch_verb(argc, argv, verbs, NULL);
302 int main(int argc, char *argv[]) {
305 setlocale(LC_ALL, "");
306 log_parse_environment();
309 r = parse_argv(argc, argv);
313 ignore_signals(SIGPIPE, -1);
315 r = export_main(argc, argv);
318 return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;