1 extern crate proxmox_backup
;
4 use std
::os
::unix
::io
::AsRawFd
;
6 use proxmox_backup
::tools
;
7 use proxmox_backup
::cli
::command
::*;
8 use proxmox_backup
::api
::schema
::*;
9 use proxmox_backup
::api
::router
::*;
10 use proxmox_backup
::client
::http_client
::*;
11 use proxmox_backup
::client
::catar_backup_stream
::*;
12 //use proxmox_backup::backup::chunk_store::*;
13 //use proxmox_backup::backup::image_index::*;
14 //use proxmox_backup::config::datastore;
15 //use proxmox_backup::catar::encoder::*;
16 use proxmox_backup
::backup
::datastore
::*;
18 use serde_json
::{Value}
;
22 fn backup_directory(body
: Body
, store
: &str, archive_name
: &str) -> Result
<(), Error
> {
24 let client
= HttpClient
::new("localhost");
26 let path
= format
!("api3/json/admin/datastore/{}/upload_catar?archive_name={}", store
, archive_name
);
28 client
.upload("application/x-proxmox-backup-catar", body
, &path
)?
;
34 fn backup_image(datastore: &DataStore, file: &std::fs::File, size: usize, target: &str, chunk_size: usize) -> Result<(), Error> {
36 let mut target = PathBuf::from(target);
38 if let Some(ext) = target.extension() {
40 bail!("got wrong file extension - expected '.iidx'");
43 target.set_extension("iidx");
46 let mut index = datastore.create_image_writer(&target, size, chunk_size)?;
48 tools::file_chunker(file, chunk_size, |pos, chunk| {
49 index.add_chunk(pos, chunk)?;
53 index.close()?; // commit changes
59 fn create_backup(param
: Value
, _info
: &ApiMethod
) -> Result
<Value
, Error
> {
61 let filename
= tools
::required_string_param(¶m
, "filename")?
;
62 let store
= tools
::required_string_param(¶m
, "store")?
;
63 let target
= tools
::required_string_param(¶m
, "target")?
;
65 let mut chunk_size
= 4*1024*1024;
67 if let Some(size
) = param
["chunk-size"].as_u64() {
68 static SIZES
: [u64; 7] = [64, 128, 256, 512, 1024, 2048, 4096];
70 if SIZES
.contains(&size
) {
71 chunk_size
= (size
as usize) * 1024;
73 bail
!("Got unsupported chunk size '{}'", size
);
77 let stat
= match nix
::sys
::stat
::stat(filename
) {
79 Err(err
) => bail
!("unable to access '{}' - {}", filename
, err
),
82 if (stat
.st_mode
& libc
::S_IFDIR
) != 0 {
83 println
!("Backup directory '{}' to '{}'", filename
, store
);
85 let stream
= CaTarBackupStream
::open(filename
)?
;
87 let body
= Body
::wrap_stream(stream
);
89 backup_directory(body
, store
, target
)?
;
91 } else if (stat
.st_mode
& (libc
::S_IFREG
|libc
::S_IFBLK
)) != 0 {
92 println
!("Backup image '{}' to '{}'", filename
, store
);
94 if stat
.st_size
<= 0 { bail!("got strange file size '{}'
", stat.st_size); }
95 let size = stat.st_size as usize;
97 panic!("implement me
");
99 //backup_image(&datastore, &file, size, &target, chunk_size)?;
101 // let idx = datastore.open_image_reader(target)?;
105 bail!("unsupported file
type (expected a directory
, file or block device
)");
108 //datastore.garbage_collection()?;
114 pub fn complete_file_name(arg: &str) -> Vec<String> {
116 let mut result = vec![];
118 use nix::fcntl::OFlag;
119 use nix::sys::stat::Mode;
120 use nix::fcntl::AtFlags;
122 let mut dirname = std::path::PathBuf::from(arg);
124 if let Ok(stat) = nix::sys::stat::fstatat(libc::AT_FDCWD, &dirname, AtFlags::empty()) {
127 if let Some(parent) = dirname.parent() {
128 dirname = parent.to_owned();
132 let mut dir = match nix::dir::Dir::openat(libc::AT_FDCWD, &dirname, OFlag::O_DIRECTORY, Mode::empty()) {
139 for item in dir.iter() {
140 if let Ok(entry) = item {
141 if let Ok(name) = entry.file_name().to_str() {
142 if name == "." || name == ".." { continue; }
143 let mut newpath = dirname.clone();
146 if let Ok(stat) = nix::sys::stat::fstatat(libc::AT_FDCWD, &newpath, AtFlags::empty()) {
147 if (stat.st_mode & libc::S_IFMT) == libc::S_IFDIR {
149 if let Some(newpath) = newpath.to_str() {
150 result.push(newpath.to_owned());
155 if let Some(newpath) = newpath.to_str() {
156 result.push(newpath.to_owned());
168 let cmd_def = CliCommand::new(
171 ObjectSchema::new("Create backup
.")
172 .required("filename
", StringSchema::new("Source
name (file or directory name
)"))
173 .required("store
", StringSchema::new("Datastore name
."))
174 .required("target
", StringSchema::new("Target name
."))
177 IntegerSchema::new("Chunk size
in KB
. Must be a power of
2.")
183 .arg_param(vec!["filename
", "target
"])
184 .completion_cb("filename
", complete_file_name)
185 .completion_cb("store
", proxmox_backup::config::datastore::complete_datastore_name);
188 if let Err(err) = run_cli_command(&cmd_def.into()) {
189 eprintln!("Error
: {}
", err);
190 if err.downcast::<UsageError>().is_ok() {
193 std::process::exit(-1);