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"
26 #include "signal-util.h"
27 #include "hostname-util.h"
30 #include "machine-image.h"
31 #include "import-util.h"
32 #include "export-tar.h"
33 #include "export-raw.h"
35 static ImportCompressType arg_compress
= IMPORT_COMPRESS_UNKNOWN
;
37 static void determine_compression_from_filename(const char *p
) {
39 if (arg_compress
!= IMPORT_COMPRESS_UNKNOWN
)
43 arg_compress
= IMPORT_COMPRESS_UNCOMPRESSED
;
47 if (endswith(p
, ".xz"))
48 arg_compress
= IMPORT_COMPRESS_XZ
;
49 else if (endswith(p
, ".gz"))
50 arg_compress
= IMPORT_COMPRESS_GZIP
;
51 else if (endswith(p
, ".bz2"))
52 arg_compress
= IMPORT_COMPRESS_BZIP2
;
54 arg_compress
= IMPORT_COMPRESS_UNCOMPRESSED
;
57 static int interrupt_signal_handler(sd_event_source
*s
, const struct signalfd_siginfo
*si
, void *userdata
) {
58 log_notice("Transfer aborted.");
59 sd_event_exit(sd_event_source_get_event(s
), EINTR
);
63 static void on_tar_finished(TarExport
*export
, int error
, void *userdata
) {
64 sd_event
*event
= userdata
;
68 log_info("Operation completed successfully.");
70 sd_event_exit(event
, abs(error
));
73 static int export_tar(int argc
, char *argv
[], void *userdata
) {
74 _cleanup_(tar_export_unrefp
) TarExport
*export
= NULL
;
75 _cleanup_event_unref_ sd_event
*event
= NULL
;
76 _cleanup_(image_unrefp
) Image
*image
= NULL
;
77 const char *path
= NULL
, *local
= NULL
;
78 _cleanup_close_
int open_fd
= -1;
81 if (machine_name_is_valid(argv
[1])) {
82 r
= image_find(argv
[1], &image
);
84 return log_error_errno(r
, "Failed to look for machine %s: %m", argv
[1]);
86 log_error("Machine image %s not found.", argv
[1]);
96 if (isempty(path
) || streq(path
, "-"))
99 determine_compression_from_filename(path
);
102 open_fd
= open(path
, O_WRONLY
|O_CREAT
|O_TRUNC
|O_CLOEXEC
|O_NOCTTY
, 0666);
104 return log_error_errno(errno
, "Failed to open tar image for export: %m");
108 log_info("Exporting '%s', saving to '%s' with compression '%s'.", local
, path
, import_compress_type_to_string(arg_compress
));
110 _cleanup_free_
char *pretty
= NULL
;
114 (void) readlink_malloc("/proc/self/fd/1", &pretty
);
115 log_info("Exporting '%s', saving to '%s' with compression '%s'.", local
, strna(pretty
), import_compress_type_to_string(arg_compress
));
118 r
= sd_event_default(&event
);
120 return log_error_errno(r
, "Failed to allocate event loop: %m");
122 assert_se(sigprocmask_many(SIG_BLOCK
, NULL
, SIGTERM
, SIGINT
, -1) >= 0);
123 (void) sd_event_add_signal(event
, NULL
, SIGTERM
, interrupt_signal_handler
, NULL
);
124 (void) sd_event_add_signal(event
, NULL
, SIGINT
, interrupt_signal_handler
, NULL
);
126 r
= tar_export_new(&export
, event
, on_tar_finished
, event
);
128 return log_error_errno(r
, "Failed to allocate exporter: %m");
130 r
= tar_export_start(export
, local
, fd
, arg_compress
);
132 return log_error_errno(r
, "Failed to export image: %m");
134 r
= sd_event_loop(event
);
136 return log_error_errno(r
, "Failed to run event loop: %m");
138 log_info("Exiting.");
142 static void on_raw_finished(RawExport
*export
, int error
, void *userdata
) {
143 sd_event
*event
= userdata
;
147 log_info("Operation completed successfully.");
149 sd_event_exit(event
, abs(error
));
152 static int export_raw(int argc
, char *argv
[], void *userdata
) {
153 _cleanup_(raw_export_unrefp
) RawExport
*export
= NULL
;
154 _cleanup_event_unref_ sd_event
*event
= NULL
;
155 _cleanup_(image_unrefp
) Image
*image
= NULL
;
156 const char *path
= NULL
, *local
= NULL
;
157 _cleanup_close_
int open_fd
= -1;
160 if (machine_name_is_valid(argv
[1])) {
161 r
= image_find(argv
[1], &image
);
163 return log_error_errno(r
, "Failed to look for machine %s: %m", argv
[1]);
165 log_error("Machine image %s not found.", argv
[1]);
175 if (isempty(path
) || streq(path
, "-"))
178 determine_compression_from_filename(path
);
181 open_fd
= open(path
, O_WRONLY
|O_CREAT
|O_TRUNC
|O_CLOEXEC
|O_NOCTTY
, 0666);
183 return log_error_errno(errno
, "Failed to open raw image for export: %m");
187 log_info("Exporting '%s', saving to '%s' with compression '%s'.", local
, path
, import_compress_type_to_string(arg_compress
));
189 _cleanup_free_
char *pretty
= NULL
;
193 (void) readlink_malloc("/proc/self/fd/1", &pretty
);
194 log_info("Exporting '%s', saving to '%s' with compression '%s'.", local
, strna(pretty
), import_compress_type_to_string(arg_compress
));
197 r
= sd_event_default(&event
);
199 return log_error_errno(r
, "Failed to allocate event loop: %m");
201 assert_se(sigprocmask_many(SIG_BLOCK
, NULL
, SIGTERM
, SIGINT
, -1) >= 0);
202 (void) sd_event_add_signal(event
, NULL
, SIGTERM
, interrupt_signal_handler
, NULL
);
203 (void) sd_event_add_signal(event
, NULL
, SIGINT
, interrupt_signal_handler
, NULL
);
205 r
= raw_export_new(&export
, event
, on_raw_finished
, event
);
207 return log_error_errno(r
, "Failed to allocate exporter: %m");
209 r
= raw_export_start(export
, local
, fd
, arg_compress
);
211 return log_error_errno(r
, "Failed to export image: %m");
213 r
= sd_event_loop(event
);
215 return log_error_errno(r
, "Failed to run event loop: %m");
217 log_info("Exiting.");
221 static int help(int argc
, char *argv
[], void *userdata
) {
223 printf("%s [OPTIONS...] {COMMAND} ...\n\n"
224 "Export container or virtual machine images.\n\n"
225 " -h --help Show this help\n"
226 " --version Show package version\n"
227 " --format=FORMAT Select format\n\n"
229 " tar NAME [FILE] Export a TAR image\n"
230 " raw NAME [FILE] Export a RAW image\n",
231 program_invocation_short_name
);
236 static int parse_argv(int argc
, char *argv
[]) {
243 static const struct option options
[] = {
244 { "help", no_argument
, NULL
, 'h' },
245 { "version", no_argument
, NULL
, ARG_VERSION
},
246 { "format", required_argument
, NULL
, ARG_FORMAT
},
255 while ((c
= getopt_long(argc
, argv
, "h", options
, NULL
)) >= 0)
260 return help(0, NULL
, NULL
);
263 puts(PACKAGE_STRING
);
264 puts(SYSTEMD_FEATURES
);
268 if (streq(optarg
, "uncompressed"))
269 arg_compress
= IMPORT_COMPRESS_UNCOMPRESSED
;
270 else if (streq(optarg
, "xz"))
271 arg_compress
= IMPORT_COMPRESS_XZ
;
272 else if (streq(optarg
, "gzip"))
273 arg_compress
= IMPORT_COMPRESS_GZIP
;
274 else if (streq(optarg
, "bzip2"))
275 arg_compress
= IMPORT_COMPRESS_BZIP2
;
277 log_error("Unknown format: %s", optarg
);
286 assert_not_reached("Unhandled option");
292 static int export_main(int argc
, char *argv
[]) {
294 static const Verb verbs
[] = {
295 { "help", VERB_ANY
, VERB_ANY
, 0, help
},
296 { "tar", 2, 3, 0, export_tar
},
297 { "raw", 2, 3, 0, export_raw
},
301 return dispatch_verb(argc
, argv
, verbs
, NULL
);
304 int main(int argc
, char *argv
[]) {
307 setlocale(LC_ALL
, "");
308 log_parse_environment();
311 r
= parse_argv(argc
, argv
);
315 (void) ignore_signals(SIGPIPE
, -1);
317 r
= export_main(argc
, argv
);
320 return r
< 0 ? EXIT_FAILURE
: EXIT_SUCCESS
;