src/client/http_client.rs

   1 use failure::*;
   2
   3 use http::Uri;
   4 use hyper::Body;
   5 use hyper::client::Client;
   6 use xdg::BaseDirectories;
   7 use chrono::{DateTime, Utc};
   8 use std::collections::HashSet;
   9 use std::sync::{Arc, Mutex};
  10 use std::io::Write;
  11
  12 use http::{Request, Response};
  13 use http::header::HeaderValue;
  14
  15 use futures::*;
  16 use futures::stream::Stream;
  17 use std::sync::atomic::{AtomicUsize, Ordering};
  18 use tokio::sync::mpsc;
  19 use openssl::ssl::{SslConnector, SslMethod};
  20
  21 use serde_json::{json, Value};
  22 use url::percent_encoding::{percent_encode,  DEFAULT_ENCODE_SET};
  23
  24 use crate::tools::{self, BroadcastFuture, tty};
  25 use crate::tools::futures::{cancellable, Canceller};
  26 use super::pipe_to_stream::*;
  27 use super::merge_known_chunks::*;
  28
  29 use crate::backup::*;
  30
  31 #[derive(Clone)]
  32 struct AuthInfo {
  33     username: String,
  34     ticket: String,
  35     token: String,
  36 }
  37
  38 /// HTTP(S) API client
  39 pub struct HttpClient {
  40     client: Client<hyper_openssl::HttpsConnector<hyper::client::HttpConnector>>,
  41     server: String,
  42     auth: BroadcastFuture<AuthInfo>,
  43 }
  44
  45 fn store_ticket_info(server: &str, username: &str, ticket: &str, token: &str) -> Result<(), Error> {
  46
  47     let base = BaseDirectories::with_prefix("proxmox-backup")?;
  48
  49     // usually /run/user/<uid>/...
  50     let path = base.place_runtime_file("tickets")?;
  51
  52     let mode = nix::sys::stat::Mode::from_bits_truncate(0o0600);
  53
  54     let mut data = tools::file_get_json(&path, Some(json!({})))?;
  55
  56     let now = Utc::now().timestamp();
  57
  58     data[server][username] = json!({ "timestamp": now, "ticket": ticket, "token": token});
  59
  60     let mut new_data = json!({});
  61
  62     let ticket_lifetime = tools::ticket::TICKET_LIFETIME - 60;
  63
  64     let empty = serde_json::map::Map::new();
  65     for (server, info) in data.as_object().unwrap_or(&empty) {
  66         for (_user, uinfo) in info.as_object().unwrap_or(&empty) {
  67             if let Some(timestamp) = uinfo["timestamp"].as_i64() {
  68                 let age = now - timestamp;
  69                 if age < ticket_lifetime {
  70                     new_data[server][username] = uinfo.clone();
  71                 }
  72             }
  73         }
  74     }
  75
  76     tools::file_set_contents(path, new_data.to_string().as_bytes(), Some(mode))?;
  77
  78     Ok(())
  79 }
  80
  81 fn load_ticket_info(server: &str, username: &str) -> Option<(String, String)> {
  82     let base = match BaseDirectories::with_prefix("proxmox-backup") {
  83         Ok(b) => b,
  84         _ => return None,
  85     };
  86
  87     // usually /run/user/<uid>/...
  88     let path = match base.place_runtime_file("tickets") {
  89         Ok(p) => p,
  90         _ => return None,
  91     };
  92
  93     let data = match tools::file_get_json(&path, None) {
  94         Ok(v) => v,
  95         _ => return None,
  96     };
  97
  98     let now = Utc::now().timestamp();
  99
 100     let ticket_lifetime = tools::ticket::TICKET_LIFETIME - 60;
 101
 102     if let Some(uinfo) = data[server][username].as_object() {
 103         if let Some(timestamp) = uinfo["timestamp"].as_i64() {
 104             let age = now - timestamp;
 105             if age < ticket_lifetime {
 106                 let ticket = match uinfo["ticket"].as_str() {
 107                     Some(t) => t,
 108                     None => return None,
 109                 };
 110                 let token = match uinfo["token"].as_str() {
 111                     Some(t) => t,
 112                     None => return None,
 113                 };
 114                 return Some((ticket.to_owned(), token.to_owned()));
 115             }
 116         }
 117     }
 118
 119     None
 120 }
 121
 122 impl HttpClient {
 123
 124     pub fn new(server: &str, username: &str) -> Result<Self, Error> {
 125         let client = Self::build_client();
 126
 127         let password = if let Some((ticket, _token)) = load_ticket_info(server, username) {
 128             ticket
 129         } else {
 130             Self::get_password(&username)?
 131         };
 132
 133         let login = Self::credentials(client.clone(), server.to_owned(), username.to_owned(), password);
 134
 135         Ok(Self {
 136             client,
 137             server: String::from(server),
 138             auth: BroadcastFuture::new(login),
 139         })
 140     }
 141
 142     fn get_password(_username: &str) -> Result<String, Error> {
 143         use std::env::VarError::*;
 144         match std::env::var("PBS_PASSWORD") {
 145             Ok(p) => return Ok(p),
 146             Err(NotUnicode(_)) => bail!("PBS_PASSWORD contains bad characters"),
 147             Err(NotPresent) => {
 148                 // Try another method
 149             }
 150         }
 151
 152         // If we're on a TTY, query the user for a password
 153         if tty::stdin_isatty() {
 154             return Ok(String::from_utf8(tty::read_password("Password: ")?)?);
 155         }
 156
 157         bail!("no password input mechanism available");
 158     }
 159
 160     fn build_client() -> Client<hyper_openssl::HttpsConnector<hyper::client::HttpConnector>> {
 161
 162         let mut ssl_connector_builder = SslConnector::builder(SslMethod::tls()).unwrap();
 163
 164         ssl_connector_builder.set_verify(openssl::ssl::SslVerifyMode::NONE); // fixme!
 165
 166         let mut httpc = hyper::client::HttpConnector::new(1);
 167         httpc.set_nodelay(true); // important for h2 download performance!
 168         httpc.set_recv_buffer_size(Some(1024*1024)); //important for h2 download performance!
 169         httpc.enforce_http(false); // we want https...
 170
 171         let https = hyper_openssl::HttpsConnector::with_connector(httpc,  ssl_connector_builder).unwrap();
 172
 173         Client::builder()
 174         //.http2_initial_stream_window_size( (1 << 31) - 2)
 175         //.http2_initial_connection_window_size( (1 << 31) - 2)
 176             .build::<_, Body>(https)
 177     }
 178
 179     pub fn request(&self, mut req: Request<Body>) -> impl Future<Item=Value, Error=Error>  {
 180
 181         let login = self.auth.listen();
 182
 183         let client = self.client.clone();
 184
 185         login.and_then(move |auth| {
 186
 187             let enc_ticket = format!("PBSAuthCookie={}", percent_encode(auth.ticket.as_bytes(), DEFAULT_ENCODE_SET));
 188             req.headers_mut().insert("Cookie", HeaderValue::from_str(&enc_ticket).unwrap());
 189             req.headers_mut().insert("CSRFPreventionToken", HeaderValue::from_str(&auth.token).unwrap());
 190
 191             let request = Self::api_request(client, req);
 192
 193             request
 194         })
 195     }
 196
 197     pub fn get(&self, path: &str, data: Option<Value>) -> impl Future<Item=Value, Error=Error> {
 198
 199         let req = Self::request_builder(&self.server, "GET", path, data).unwrap();
 200         self.request(req)
 201     }
 202
 203     pub fn delete(&mut self, path: &str, data: Option<Value>) -> impl Future<Item=Value, Error=Error> {
 204
 205         let req = Self::request_builder(&self.server, "DELETE", path, data).unwrap();
 206         self.request(req)
 207     }
 208
 209     pub fn post(&mut self, path: &str, data: Option<Value>) -> impl Future<Item=Value, Error=Error> {
 210
 211         let req = Self::request_builder(&self.server, "POST", path, data).unwrap();
 212         self.request(req)
 213     }
 214
 215     pub fn download<W: Write>(&mut self, path: &str, output: W) ->  impl Future<Item=W, Error=Error> {
 216
 217         let mut req = Self::request_builder(&self.server, "GET", path, None).unwrap();
 218
 219         let login = self.auth.listen();
 220
 221         let client = self.client.clone();
 222
 223         login.and_then(move |auth| {
 224
 225             let enc_ticket = format!("PBSAuthCookie={}", percent_encode(auth.ticket.as_bytes(), DEFAULT_ENCODE_SET));
 226             req.headers_mut().insert("Cookie", HeaderValue::from_str(&enc_ticket).unwrap());
 227
 228             client.request(req)
 229                 .map_err(Error::from)
 230                 .and_then(|resp| {
 231                     let status = resp.status();
 232                     if !status.is_success() {
 233                         future::Either::A(
 234                             HttpClient::api_response(resp)
 235                                 .and_then(|_| { bail!("unknown error"); })
 236                         )
 237                     } else {
 238                         future::Either::B(
 239                             resp.into_body()
 240                                 .map_err(Error::from)
 241                                 .fold(output, move |mut acc, chunk| {
 242                                     acc.write_all(&chunk)?;
 243                                     Ok::<_, Error>(acc)
 244                                 })
 245                         )
 246                     }
 247                 })
 248         })
 249     }
 250
 251     pub fn upload(
 252         &mut self,
 253         content_type: &str,
 254         body: Body,
 255         path: &str,
 256         data: Option<Value>,
 257     ) -> impl Future<Item=Value, Error=Error> {
 258
 259         let path = path.trim_matches('/');
 260         let mut url = format!("https://{}:8007/{}", &self.server, path);
 261
 262         if let Some(data) = data {
 263             let query = tools::json_object_to_query(data).unwrap();
 264             url.push('?');
 265             url.push_str(&query);
 266         }
 267
 268         let url: Uri = url.parse().unwrap();
 269
 270         let req = Request::builder()
 271             .method("POST")
 272             .uri(url)
 273             .header("User-Agent", "proxmox-backup-client/1.0")
 274             .header("Content-Type", content_type)
 275             .body(body).unwrap();
 276
 277         self.request(req)
 278     }
 279
 280     pub fn start_backup(
 281         &self,
 282         datastore: &str,
 283         backup_type: &str,
 284         backup_id: &str,
 285         backup_time: DateTime<Utc>,
 286         debug: bool,
 287     ) -> impl Future<Item=Arc<BackupClient>, Error=Error> {
 288
 289         let param = json!({
 290             "backup-type": backup_type,
 291             "backup-id": backup_id,
 292             "backup-time": backup_time.timestamp(),
 293             "store": datastore,
 294             "debug": debug
 295         });
 296
 297         let req = Self::request_builder(&self.server, "GET", "/api2/json/backup", Some(param)).unwrap();
 298
 299         self.start_h2_connection(req, String::from(PROXMOX_BACKUP_PROTOCOL_ID_V1!()))
 300             .map(|(h2, canceller)| BackupClient::new(h2, canceller))
 301     }
 302
 303     pub fn start_backup_reader(
 304         &self,
 305         datastore: &str,
 306         backup_type: &str,
 307         backup_id: &str,
 308         backup_time: DateTime<Utc>,
 309         debug: bool,
 310     ) -> impl Future<Item=Arc<BackupReader>, Error=Error> {
 311
 312         let param = json!({
 313             "backup-type": backup_type,
 314             "backup-id": backup_id,
 315             "backup-time": backup_time.timestamp(),
 316             "store": datastore,
 317             "debug": debug,
 318         });
 319         let req = Self::request_builder(&self.server, "GET", "/api2/json/reader", Some(param)).unwrap();
 320
 321         self.start_h2_connection(req, String::from(PROXMOX_BACKUP_READER_PROTOCOL_ID_V1!()))
 322             .map(|(h2, canceller)| BackupReader::new(h2, canceller))
 323     }
 324
 325     pub fn start_h2_connection(
 326         &self,
 327         mut req: Request<Body>,
 328         protocol_name: String,
 329     ) -> impl Future<Item=(H2Client, Canceller), Error=Error> {
 330
 331         let login = self.auth.listen();
 332         let client = self.client.clone();
 333
 334         login.and_then(move |auth| {
 335
 336             let enc_ticket = format!("PBSAuthCookie={}", percent_encode(auth.ticket.as_bytes(), DEFAULT_ENCODE_SET));
 337             req.headers_mut().insert("Cookie", HeaderValue::from_str(&enc_ticket).unwrap());
 338             req.headers_mut().insert("UPGRADE", HeaderValue::from_str(&protocol_name).unwrap());
 339
 340             client.request(req)
 341                 .map_err(Error::from)
 342                 .and_then(|resp| {
 343
 344                     let status = resp.status();
 345                     if status != http::StatusCode::SWITCHING_PROTOCOLS {
 346                         future::Either::A(Self::api_response(resp).and_then(|_| { bail!("unknown error"); }))
 347                     } else {
 348                         future::Either::B(resp.into_body().on_upgrade().map_err(Error::from))
 349                     }
 350                 })
 351                 .and_then(|upgraded| {
 352                    let max_window_size = (1 << 31) - 2;
 353
 354                     h2::client::Builder::new()
 355                         .initial_connection_window_size(max_window_size)
 356                         .initial_window_size(max_window_size)
 357                         .max_frame_size(4*1024*1024)
 358                         .handshake(upgraded)
 359                         .map_err(Error::from)
 360                 })
 361                 .and_then(|(h2, connection)| {
 362                     let connection = connection
 363                         .map_err(|_| panic!("HTTP/2.0 connection failed"));
 364
 365                     let (connection, canceller) = cancellable(connection)?;
 366                     // A cancellable future returns an Option which is None when cancelled and
 367                     // Some when it finished instead, since we don't care about the return type we
 368                     // need to map it away:
 369                     let connection = connection.map(|_| ());
 370
 371                     // Spawn a new task to drive the connection state
 372                     hyper::rt::spawn(connection);
 373
 374                     // Wait until the `SendRequest` handle has available capacity.
 375                     Ok(h2.ready()
 376                        .map(move |c| (H2Client::new(c), canceller))
 377                        .map_err(Error::from))
 378                 })
 379                 .flatten()
 380         })
 381     }
 382
 383     fn credentials(
 384         client: Client<hyper_openssl::HttpsConnector<hyper::client::HttpConnector>>,
 385         server: String,
 386         username: String,
 387         password: String,
 388     ) -> Box<dyn Future<Item=AuthInfo, Error=Error> + Send> {
 389
 390         let server2 = server.clone();
 391
 392         let create_request = futures::future::lazy(move || {
 393             let data = json!({ "username": username, "password": password });
 394             let req = Self::request_builder(&server, "POST", "/api2/json/access/ticket", Some(data)).unwrap();
 395             Self::api_request(client, req)
 396         });
 397
 398         let login_future = create_request
 399             .and_then(move |cred| {
 400                 let auth = AuthInfo {
 401                     username: cred["data"]["username"].as_str().unwrap().to_owned(),
 402                     ticket: cred["data"]["ticket"].as_str().unwrap().to_owned(),
 403                     token: cred["data"]["CSRFPreventionToken"].as_str().unwrap().to_owned(),
 404                 };
 405
 406                 let _ = store_ticket_info(&server2, &auth.username, &auth.ticket, &auth.token);
 407
 408                 Ok(auth)
 409             });
 410
 411         Box::new(login_future)
 412     }
 413
 414     fn api_response(response: Response<Body>) -> impl Future<Item=Value, Error=Error> {
 415
 416         let status = response.status();
 417
 418         response
 419             .into_body()
 420             .concat2()
 421             .map_err(Error::from)
 422             .and_then(move |data| {
 423
 424                 let text = String::from_utf8(data.to_vec()).unwrap();
 425                 if status.is_success() {
 426                     if text.len() > 0 {
 427                         let value: Value = serde_json::from_str(&text)?;
 428                         Ok(value)
 429                     } else {
 430                         Ok(Value::Null)
 431                     }
 432                 } else {
 433                     bail!("HTTP Error {}: {}", status, text);
 434                 }
 435             })
 436     }
 437
 438     fn api_request(
 439         client: Client<hyper_openssl::HttpsConnector<hyper::client::HttpConnector>>,
 440         req: Request<Body>
 441     ) -> impl Future<Item=Value, Error=Error> {
 442
 443         client.request(req)
 444             .map_err(Error::from)
 445             .and_then(Self::api_response)
 446     }
 447
 448     pub fn request_builder(server: &str, method: &str, path: &str, data: Option<Value>) -> Result<Request<Body>, Error> {
 449         let path = path.trim_matches('/');
 450         let url: Uri = format!("https://{}:8007/{}", server, path).parse()?;
 451
 452         if let Some(data) = data {
 453             if method == "POST" {
 454                 let request = Request::builder()
 455                     .method(method)
 456                     .uri(url)
 457                     .header("User-Agent", "proxmox-backup-client/1.0")
 458                     .header(hyper::header::CONTENT_TYPE, "application/json")
 459                     .body(Body::from(data.to_string()))?;
 460                 return Ok(request);
 461             } else {
 462                 let query = tools::json_object_to_query(data)?;
 463                 let url: Uri = format!("https://{}:8007/{}?{}", server, path, query).parse()?;
 464                 let request = Request::builder()
 465                     .method(method)
 466                     .uri(url)
 467                     .header("User-Agent", "proxmox-backup-client/1.0")
 468                     .header(hyper::header::CONTENT_TYPE, "application/x-www-form-urlencoded")
 469                     .body(Body::empty())?;
 470                 return Ok(request);
 471             }
 472         }
 473
 474         let request = Request::builder()
 475             .method(method)
 476             .uri(url)
 477             .header("User-Agent", "proxmox-backup-client/1.0")
 478             .header(hyper::header::CONTENT_TYPE, "application/x-www-form-urlencoded")
 479             .body(Body::empty())?;
 480
 481         Ok(request)
 482     }
 483 }
 484
 485
 486 pub struct BackupReader {
 487     h2: H2Client,
 488     canceller: Canceller,
 489 }
 490
 491 impl Drop for BackupReader {
 492
 493     fn drop(&mut self) {
 494         self.canceller.cancel();
 495     }
 496 }
 497
 498 impl BackupReader {
 499
 500     pub fn new(h2: H2Client, canceller: Canceller) -> Arc<Self> {
 501         Arc::new(Self { h2, canceller: canceller })
 502     }
 503
 504     pub fn get(&self, path: &str, param: Option<Value>) -> impl Future<Item=Value, Error=Error> {
 505         self.h2.get(path, param)
 506     }
 507
 508     pub fn put(&self, path: &str, param: Option<Value>) -> impl Future<Item=Value, Error=Error> {
 509         self.h2.put(path, param)
 510     }
 511
 512     pub fn post(&self, path: &str, param: Option<Value>) -> impl Future<Item=Value, Error=Error> {
 513         self.h2.post(path, param)
 514     }
 515
 516     pub fn download<W: Write>(
 517         &self,
 518         file_name: &str,
 519         output: W,
 520     ) -> impl Future<Item=W, Error=Error> {
 521         let path = "download";
 522         let param = json!({ "file-name": file_name });
 523         self.h2.download(path, Some(param), output)
 524     }
 525
 526     pub fn speedtest<W: Write>(
 527         &self,
 528         output: W,
 529     ) -> impl Future<Item=W, Error=Error> {
 530         self.h2.download("speedtest", None, output)
 531     }
 532
 533     pub fn download_chunk<W: Write>(
 534         &self,
 535         digest: &[u8; 32],
 536         output: W,
 537     ) -> impl Future<Item=W, Error=Error> {
 538         let path = "chunk";
 539         let param = json!({ "digest": proxmox::tools::digest_to_hex(digest) });
 540         self.h2.download(path, Some(param), output)
 541     }
 542
 543     pub fn force_close(self) {
 544         self.canceller.cancel();
 545     }
 546 }
 547
 548 pub struct BackupClient {
 549     h2: H2Client,
 550     canceller: Canceller,
 551 }
 552
 553 impl Drop for BackupClient {
 554
 555     fn drop(&mut self) {
 556         self.canceller.cancel();
 557     }
 558 }
 559
 560 impl BackupClient {
 561
 562     pub fn new(h2: H2Client, canceller: Canceller) -> Arc<Self> {
 563         Arc::new(Self { h2, canceller })
 564     }
 565
 566     pub fn get(&self, path: &str, param: Option<Value>) -> impl Future<Item=Value, Error=Error> {
 567         self.h2.get(path, param)
 568     }
 569
 570     pub fn put(&self, path: &str, param: Option<Value>) -> impl Future<Item=Value, Error=Error> {
 571         self.h2.put(path, param)
 572     }
 573
 574     pub fn post(&self, path: &str, param: Option<Value>) -> impl Future<Item=Value, Error=Error> {
 575         self.h2.post(path, param)
 576     }
 577
 578     pub fn finish(self: Arc<Self>) -> impl Future<Item=(), Error=Error> {
 579         self.h2.clone()
 580             .post("finish", None)
 581             .map(move |_| {
 582                 self.canceller.cancel();
 583             })
 584     }
 585
 586     pub fn force_close(self) {
 587         self.canceller.cancel();
 588     }
 589
 590     pub fn upload_blob_from_data(
 591         &self,
 592         data: Vec<u8>,
 593         file_name: &str,
 594         crypt_config: Option<Arc<CryptConfig>>,
 595         compress: bool,
 596      ) -> impl Future<Item=(), Error=Error> {
 597
 598         let h2 = self.h2.clone();
 599         let file_name = file_name.to_owned();
 600
 601         futures::future::ok(())
 602             .and_then(move |_| {
 603                 let blob = if let Some(ref crypt_config) = crypt_config {
 604                     DataBlob::encode(&data, Some(crypt_config), compress)?
 605                 } else {
 606                     DataBlob::encode(&data, None, compress)?
 607                 };
 608
 609                 let raw_data = blob.into_inner();
 610                 Ok(raw_data)
 611             })
 612             .and_then(move |raw_data| {
 613                 let param = json!({"encoded-size": raw_data.len(), "file-name": file_name });
 614                 h2.upload("blob", Some(param), raw_data)
 615                     .map(|_| {})
 616             })
 617     }
 618
 619     pub fn upload_blob_from_file<P: AsRef<std::path::Path>>(
 620         &self,
 621         src_path: P,
 622         file_name: &str,
 623         crypt_config: Option<Arc<CryptConfig>>,
 624         compress: bool,
 625      ) -> impl Future<Item=(), Error=Error> {
 626
 627         let h2 = self.h2.clone();
 628         let file_name = file_name.to_owned();
 629         let src_path = src_path.as_ref().to_owned();
 630
 631         let task = tokio::fs::File::open(src_path.clone())
 632             .map_err(move |err| format_err!("unable to open file {:?} - {}", src_path, err))
 633             .and_then(move |file| {
 634                 let contents = vec![];
 635                 tokio::io::read_to_end(file, contents)
 636                     .map_err(Error::from)
 637                     .and_then(move |(_, contents)| {
 638                         let blob = if let Some(ref crypt_config) = crypt_config {
 639                             DataBlob::encode(&contents, Some(crypt_config), compress)?
 640                         } else {
 641                             DataBlob::encode(&contents, None, compress)?
 642                         };
 643                         let raw_data = blob.into_inner();
 644                         Ok(raw_data)
 645                     })
 646                     .and_then(move |raw_data| {
 647                         let param = json!({"encoded-size": raw_data.len(), "file-name": file_name });
 648                         h2.upload("blob", Some(param), raw_data)
 649                             .map(|_| {})
 650                     })
 651             });
 652
 653         task
 654     }
 655
 656     pub fn upload_stream(
 657         &self,
 658         archive_name: &str,
 659         stream: impl Stream<Item=bytes::BytesMut, Error=Error>,
 660         prefix: &str,
 661         fixed_size: Option<u64>,
 662         crypt_config: Option<Arc<CryptConfig>>,
 663     ) -> impl Future<Item=(), Error=Error> {
 664
 665         let known_chunks = Arc::new(Mutex::new(HashSet::new()));
 666
 667         let h2 = self.h2.clone();
 668         let h2_2 = self.h2.clone();
 669         let h2_3 = self.h2.clone();
 670         let h2_4 = self.h2.clone();
 671
 672         let mut param = json!({ "archive-name": archive_name });
 673         if let Some(size) = fixed_size {
 674             param["size"] = size.into();
 675         }
 676
 677         let index_path = format!("{}_index", prefix);
 678         let close_path = format!("{}_close", prefix);
 679
 680         let prefix = prefix.to_owned();
 681
 682         Self::download_chunk_list(h2, &index_path, archive_name, known_chunks.clone())
 683             .and_then(move |_| {
 684                 h2_2.post(&index_path, Some(param))
 685             })
 686             .and_then(move |res| {
 687                 let wid = res.as_u64().unwrap();
 688                 Self::upload_chunk_info_stream(h2_3, wid, stream, &prefix, known_chunks.clone(), crypt_config)
 689                      .and_then(move |(chunk_count, size, _speed)| {
 690                         let param = json!({
 691                             "wid": wid ,
 692                             "chunk-count": chunk_count,
 693                             "size": size,
 694                         });
 695                         h2_4.post(&close_path, Some(param))
 696                      })
 697                     .map(|_| ())
 698             })
 699     }
 700
 701     fn response_queue() -> (
 702         mpsc::Sender<h2::client::ResponseFuture>,
 703         sync::oneshot::Receiver<Result<(), Error>>
 704     ) {
 705         let (verify_queue_tx, verify_queue_rx) = mpsc::channel(100);
 706         let (verify_result_tx, verify_result_rx) = sync::oneshot::channel();
 707
 708         hyper::rt::spawn(
 709             verify_queue_rx
 710                 .map_err(Error::from)
 711                 .for_each(|response: h2::client::ResponseFuture| {
 712                     response
 713                         .map_err(Error::from)
 714                         .and_then(H2Client::h2api_response)
 715                         .and_then(|result| {
 716                             println!("RESPONSE: {:?}", result);
 717                             Ok(())
 718                         })
 719                         .map_err(|err| format_err!("pipelined request failed: {}", err))
 720                 })
 721                 .then(|result|
 722                       verify_result_tx.send(result)
 723                 )
 724                 .map_err(|_| { /* ignore closed channel */ })
 725         );
 726
 727         (verify_queue_tx, verify_result_rx)
 728     }
 729
 730     fn append_chunk_queue(h2: H2Client, wid: u64, path: String) -> (
 731         mpsc::Sender<(MergedChunkInfo, Option<h2::client::ResponseFuture>)>,
 732         sync::oneshot::Receiver<Result<(), Error>>
 733     ) {
 734         let (verify_queue_tx, verify_queue_rx) = mpsc::channel(64);
 735         let (verify_result_tx, verify_result_rx) = sync::oneshot::channel();
 736
 737         let h2_2 = h2.clone();
 738
 739         hyper::rt::spawn(
 740             verify_queue_rx
 741                 .map_err(Error::from)
 742                 .and_then(move |(merged_chunk_info, response): (MergedChunkInfo, Option<h2::client::ResponseFuture>)| {
 743                     match (response, merged_chunk_info) {
 744                         (Some(response), MergedChunkInfo::Known(list)) => {
 745                             future::Either::A(
 746                                 response
 747                                     .map_err(Error::from)
 748                                     .and_then(H2Client::h2api_response)
 749                                     .and_then(move |_result| {
 750                                         Ok(MergedChunkInfo::Known(list))
 751                                     })
 752                             )
 753                         }
 754                         (None, MergedChunkInfo::Known(list)) => {
 755                             future::Either::B(future::ok(MergedChunkInfo::Known(list)))
 756                         }
 757                         _ => unreachable!(),
 758                     }
 759                 })
 760                 .merge_known_chunks()
 761                 .and_then(move |merged_chunk_info| {
 762                     match merged_chunk_info {
 763                         MergedChunkInfo::Known(chunk_list) => {
 764                             let mut digest_list = vec![];
 765                             let mut offset_list = vec![];
 766                             for (offset, digest) in chunk_list {
 767                                 //println!("append chunk {} (offset {})", proxmox::tools::digest_to_hex(&digest), offset);
 768                                 digest_list.push(proxmox::tools::digest_to_hex(&digest));
 769                                 offset_list.push(offset);
 770                             }
 771                             println!("append chunks list len ({})", digest_list.len());
 772                             let param = json!({ "wid": wid, "digest-list": digest_list, "offset-list": offset_list });
 773                             let mut request = H2Client::request_builder("localhost", "PUT", &path, None).unwrap();
 774                             request.headers_mut().insert(hyper::header::CONTENT_TYPE,  HeaderValue::from_static("application/json"));
 775                             let param_data = bytes::Bytes::from(param.to_string().as_bytes());
 776                             let upload_data = Some(param_data);
 777                             h2_2.send_request(request, upload_data)
 778                                 .and_then(move |response| {
 779                                     response
 780                                         .map_err(Error::from)
 781                                         .and_then(H2Client::h2api_response)
 782                                         .and_then(|_| Ok(()))
 783                                 })
 784                                 .map_err(|err| format_err!("pipelined request failed: {}", err))
 785                         }
 786                         _ => unreachable!(),
 787                     }
 788                 })
 789                 .for_each(|_| Ok(()))
 790                 .then(|result|
 791                       verify_result_tx.send(result)
 792                 )
 793                 .map_err(|_| { /* ignore closed channel */ })
 794         );
 795
 796         (verify_queue_tx, verify_result_rx)
 797     }
 798
 799     fn download_chunk_list(
 800         h2: H2Client,
 801         path: &str,
 802         archive_name: &str,
 803         known_chunks: Arc<Mutex<HashSet<[u8;32]>>>,
 804     ) -> impl Future<Item=(), Error=Error> {
 805
 806         let param = json!({ "archive-name": archive_name });
 807         let request = H2Client::request_builder("localhost", "GET", path, Some(param)).unwrap();
 808
 809         h2.send_request(request, None)
 810             .and_then(move |response| {
 811                 response
 812                     .map_err(Error::from)
 813                     .and_then(move |resp| {
 814                         let status = resp.status();
 815
 816                         if !status.is_success() {
 817                             future::Either::A(H2Client::h2api_response(resp).and_then(|_| { bail!("unknown error"); }))
 818                         } else {
 819                             future::Either::B(future::ok(resp.into_body()))
 820                         }
 821                     })
 822                     .and_then(move |mut body| {
 823
 824                         let mut release_capacity = body.release_capacity().clone();
 825
 826                         DigestListDecoder::new(body.map_err(Error::from))
 827                             .for_each(move |chunk| {
 828                                 let _ = release_capacity.release_capacity(chunk.len());
 829                                 println!("GOT DOWNLOAD {}", proxmox::tools::digest_to_hex(&chunk));
 830                                 known_chunks.lock().unwrap().insert(chunk);
 831                                 Ok(())
 832                             })
 833                        })
 834             })
 835     }
 836
 837     fn upload_chunk_info_stream(
 838         h2: H2Client,
 839         wid: u64,
 840         stream: impl Stream<Item=bytes::BytesMut, Error=Error>,
 841         prefix: &str,
 842         known_chunks: Arc<Mutex<HashSet<[u8;32]>>>,
 843         crypt_config: Option<Arc<CryptConfig>>,
 844     ) -> impl Future<Item=(usize, usize, usize), Error=Error> {
 845
 846         let repeat = std::sync::Arc::new(AtomicUsize::new(0));
 847         let repeat2 = repeat.clone();
 848
 849         let stream_len = std::sync::Arc::new(AtomicUsize::new(0));
 850         let stream_len2 = stream_len.clone();
 851
 852         let append_chunk_path = format!("{}_index", prefix);
 853         let upload_chunk_path = format!("{}_chunk", prefix);
 854
 855         let (upload_queue, upload_result) = Self::append_chunk_queue(h2.clone(), wid, append_chunk_path.to_owned());
 856
 857         let start_time = std::time::Instant::now();
 858
 859         stream
 860             .and_then(move |data| {
 861
 862                 let chunk_len = data.len();
 863
 864                 repeat.fetch_add(1, Ordering::SeqCst);
 865                 let offset = stream_len.fetch_add(chunk_len, Ordering::SeqCst) as u64;
 866
 867                 let mut chunk_builder = DataChunkBuilder::new(data.as_ref())
 868                     .compress(true);
 869
 870                 if let Some(ref crypt_config) = crypt_config {
 871                     chunk_builder = chunk_builder.crypt_config(crypt_config);
 872                 }
 873
 874                 let mut known_chunks = known_chunks.lock().unwrap();
 875                 let digest = chunk_builder.digest();
 876                 let chunk_is_known = known_chunks.contains(digest);
 877                 if chunk_is_known {
 878                     Ok(MergedChunkInfo::Known(vec![(offset, *digest)]))
 879                 } else {
 880                     known_chunks.insert(*digest);
 881                     let chunk = chunk_builder.build()?;
 882                     Ok(MergedChunkInfo::New(ChunkInfo { chunk, chunk_len: chunk_len as u64, offset }))
 883                 }
 884             })
 885             .merge_known_chunks()
 886             .for_each(move |merged_chunk_info| {
 887
 888                 if let MergedChunkInfo::New(chunk_info) = merged_chunk_info {
 889                     let offset = chunk_info.offset;
 890                     let digest = *chunk_info.chunk.digest();
 891                     let digest_str = proxmox::tools::digest_to_hex(&digest);
 892                     let upload_queue = upload_queue.clone();
 893
 894                     println!("upload new chunk {} ({} bytes, offset {})", digest_str,
 895                              chunk_info.chunk_len, offset);
 896
 897                     let chunk_data = chunk_info.chunk.raw_data();
 898                     let param = json!({
 899                         "wid": wid,
 900                         "digest": digest_str,
 901                         "size": chunk_info.chunk_len,
 902                         "encoded-size": chunk_data.len(),
 903                     });
 904
 905                     let request = H2Client::request_builder("localhost", "POST", &upload_chunk_path, Some(param)).unwrap();
 906                     let upload_data = Some(bytes::Bytes::from(chunk_data));
 907
 908                     let new_info = MergedChunkInfo::Known(vec![(offset, digest)]);
 909
 910                     future::Either::A(
 911                         h2.send_request(request, upload_data)
 912                             .and_then(move |response| {
 913                                 upload_queue.clone().send((new_info, Some(response)))
 914                                     .map(|_| ()).map_err(Error::from)
 915                             })
 916                     )
 917                 } else {
 918
 919                     future::Either::B(
 920                         upload_queue.clone().send((merged_chunk_info, None))
 921                             .map(|_| ()).map_err(Error::from)
 922                     )
 923                 }
 924             })
 925             .then(move |result| {
 926                 //println!("RESULT {:?}", result);
 927                 upload_result.map_err(Error::from).and_then(|upload1_result| {
 928                     Ok(upload1_result.and(result))
 929                 })
 930             })
 931             .flatten()
 932             .and_then(move |_| {
 933                 let repeat = repeat2.load(Ordering::SeqCst);
 934                 let stream_len = stream_len2.load(Ordering::SeqCst);
 935                 let speed = ((stream_len*1000000)/(1024*1024))/(start_time.elapsed().as_micros() as usize);
 936                 println!("Uploaded {} chunks in {} seconds ({} MB/s).", repeat, start_time.elapsed().as_secs(), speed);
 937                 if repeat > 0 {
 938                     println!("Average chunk size was {} bytes.", stream_len/repeat);
 939                     println!("Time per request: {} microseconds.", (start_time.elapsed().as_micros())/(repeat as u128));
 940                 }
 941                 Ok((repeat, stream_len, speed))
 942             })
 943     }
 944
 945     pub fn upload_speedtest(&self) -> impl Future<Item=usize, Error=Error> {
 946
 947         let mut data = vec![];
 948         // generate pseudo random byte sequence
 949         for i in 0..1024*1024 {
 950             for j in 0..4 {
 951                 let byte = ((i >> (j<<3))&0xff) as u8;
 952                 data.push(byte);
 953             }
 954         }
 955
 956         let item_len = data.len();
 957
 958         let repeat = std::sync::Arc::new(AtomicUsize::new(0));
 959         let repeat2 = repeat.clone();
 960
 961         let (upload_queue, upload_result) = Self::response_queue();
 962
 963         let start_time = std::time::Instant::now();
 964
 965         let h2 = self.h2.clone();
 966
 967         futures::stream::repeat(data)
 968             .take_while(move |_| {
 969                 repeat.fetch_add(1, Ordering::SeqCst);
 970                 Ok(start_time.elapsed().as_secs() < 5)
 971             })
 972             .for_each(move |data| {
 973                 let h2 = h2.clone();
 974
 975                 let upload_queue = upload_queue.clone();
 976
 977                 println!("send test data ({} bytes)", data.len());
 978                 let request = H2Client::request_builder("localhost", "POST", "speedtest", None).unwrap();
 979                 h2.send_request(request, Some(bytes::Bytes::from(data)))
 980                     .and_then(move |response| {
 981                         upload_queue.send(response)
 982                             .map(|_| ()).map_err(Error::from)
 983                     })
 984             })
 985             .then(move |result| {
 986                 println!("RESULT {:?}", result);
 987                 upload_result.map_err(Error::from).and_then(|upload1_result| {
 988                     Ok(upload1_result.and(result))
 989                 })
 990             })
 991             .flatten()
 992             .and_then(move |_| {
 993                 let repeat = repeat2.load(Ordering::SeqCst);
 994                 println!("Uploaded {} chunks in {} seconds.", repeat, start_time.elapsed().as_secs());
 995                 let speed = ((item_len*1000000*(repeat as usize))/(1024*1024))/(start_time.elapsed().as_micros() as usize);
 996                 if repeat > 0 {
 997                     println!("Time per request: {} microseconds.", (start_time.elapsed().as_micros())/(repeat as u128));
 998                 }
 999                 Ok(speed)
1000             })
1001     }
1002 }
1003
1004 #[derive(Clone)]
1005 pub struct H2Client {
1006     h2: h2::client::SendRequest<bytes::Bytes>,
1007 }
1008
1009 impl H2Client {
1010
1011     pub fn new(h2: h2::client::SendRequest<bytes::Bytes>) -> Self {
1012         Self { h2 }
1013     }
1014
1015     pub fn get(&self, path: &str, param: Option<Value>) -> impl Future<Item=Value, Error=Error> {
1016         let req = Self::request_builder("localhost", "GET", path, param).unwrap();
1017         self.request(req)
1018     }
1019
1020     pub fn put(&self, path: &str, param: Option<Value>) -> impl Future<Item=Value, Error=Error> {
1021         let req = Self::request_builder("localhost", "PUT", path, param).unwrap();
1022         self.request(req)
1023     }
1024
1025     pub fn post(&self, path: &str, param: Option<Value>) -> impl Future<Item=Value, Error=Error> {
1026         let req = Self::request_builder("localhost", "POST", path, param).unwrap();
1027         self.request(req)
1028     }
1029
1030     pub fn download<W: Write>(&self, path: &str, param: Option<Value>, output: W) -> impl Future<Item=W, Error=Error> {
1031         let request = Self::request_builder("localhost", "GET", path, param).unwrap();
1032
1033         self.send_request(request, None)
1034             .and_then(move |response| {
1035                 response
1036                     .map_err(Error::from)
1037                     .and_then(move |resp| {
1038                         let status = resp.status();
1039                         if !status.is_success() {
1040                             future::Either::A(
1041                                 H2Client::h2api_response(resp)
1042                                     .and_then(|_| { bail!("unknown error"); })
1043                             )
1044                         } else {
1045                             let mut body = resp.into_body();
1046                             let mut release_capacity = body.release_capacity().clone();
1047
1048                             future::Either::B(
1049                                 body
1050                                     .map_err(Error::from)
1051                                     .fold(output, move |mut acc, chunk| {
1052                                         let _ = release_capacity.release_capacity(chunk.len());
1053                                         acc.write_all(&chunk)?;
1054                                         Ok::<_, Error>(acc)
1055                                     })
1056                             )
1057                         }
1058                     })
1059             })
1060     }
1061
1062     pub fn upload(&self, path: &str, param: Option<Value>, data: Vec<u8>) -> impl Future<Item=Value, Error=Error> {
1063         let request = Self::request_builder("localhost", "POST", path, param).unwrap();
1064
1065         self.h2.clone()
1066             .ready()
1067             .map_err(Error::from)
1068             .and_then(move |mut send_request| {
1069                 let (response, stream) = send_request.send_request(request, false).unwrap();
1070                 PipeToSendStream::new(bytes::Bytes::from(data), stream)
1071                     .and_then(|_| {
1072                         response
1073                             .map_err(Error::from)
1074                             .and_then(Self::h2api_response)
1075                     })
1076             })
1077     }
1078
1079     fn request(
1080         &self,
1081         request: Request<()>,
1082     ) -> impl Future<Item=Value, Error=Error> {
1083
1084         self.send_request(request, None)
1085             .and_then(move |response| {
1086                 response
1087                     .map_err(Error::from)
1088                     .and_then(Self::h2api_response)
1089             })
1090     }
1091
1092     fn send_request(
1093         &self,
1094         request: Request<()>,
1095         data: Option<bytes::Bytes>,
1096     ) -> impl Future<Item=h2::client::ResponseFuture, Error=Error> {
1097
1098         self.h2.clone()
1099             .ready()
1100             .map_err(Error::from)
1101             .and_then(move |mut send_request| {
1102                 if let Some(data) = data {
1103                     let (response, stream) = send_request.send_request(request, false).unwrap();
1104                     future::Either::A(PipeToSendStream::new(data, stream)
1105                         .and_then(move |_| {
1106                             future::ok(response)
1107                         }))
1108                 } else {
1109                     let (response, _stream) = send_request.send_request(request, true).unwrap();
1110                     future::Either::B(future::ok(response))
1111                 }
1112             })
1113     }
1114
1115     fn h2api_response(response: Response<h2::RecvStream>) -> impl Future<Item=Value, Error=Error> {
1116
1117         let status = response.status();
1118
1119         let (_head, mut body) = response.into_parts();
1120
1121         // The `release_capacity` handle allows the caller to manage
1122         // flow control.
1123         //
1124         // Whenever data is received, the caller is responsible for
1125         // releasing capacity back to the server once it has freed
1126         // the data from memory.
1127         let mut release_capacity = body.release_capacity().clone();
1128
1129         body
1130             .map(move |chunk| {
1131                 // Let the server send more data.
1132                 let _ = release_capacity.release_capacity(chunk.len());
1133                 chunk
1134             })
1135             .concat2()
1136             .map_err(Error::from)
1137             .and_then(move |data| {
1138                 let text = String::from_utf8(data.to_vec()).unwrap();
1139                 if status.is_success() {
1140                     if text.len() > 0 {
1141                         let mut value: Value = serde_json::from_str(&text)?;
1142                         if let Some(map) = value.as_object_mut() {
1143                             if let Some(data) = map.remove("data") {
1144                                 return Ok(data);
1145                             }
1146                         }
1147                         bail!("got result without data property");
1148                     } else {
1149                         Ok(Value::Null)
1150                     }
1151                 } else {
1152                     bail!("HTTP Error {}: {}", status, text);
1153                 }
1154             })
1155     }
1156
1157     // Note: We always encode parameters with the url
1158     pub fn request_builder(server: &str, method: &str, path: &str, data: Option<Value>) -> Result<Request<()>, Error> {
1159         let path = path.trim_matches('/');
1160
1161         if let Some(data) = data {
1162             let query = tools::json_object_to_query(data)?;
1163             // We detected problem with hyper around 6000 characters - seo we try to keep on the safe side
1164             if query.len() > 4096 { bail!("h2 query data too large ({} bytes) - please encode data inside body", query.len()); }
1165             let url: Uri = format!("https://{}:8007/{}?{}", server, path, query).parse()?;
1166              let request = Request::builder()
1167                 .method(method)
1168                 .uri(url)
1169                 .header("User-Agent", "proxmox-backup-client/1.0")
1170                 .header(hyper::header::CONTENT_TYPE, "application/x-www-form-urlencoded")
1171                 .body(())?;
1172             return Ok(request);
1173         } else {
1174             let url: Uri = format!("https://{}:8007/{}", server, path).parse()?;
1175             let request = Request::builder()
1176                 .method(method)
1177                 .uri(url)
1178                 .header("User-Agent", "proxmox-backup-client/1.0")
1179                 .header(hyper::header::CONTENT_TYPE, "application/x-www-form-urlencoded")
1180                 .body(())?;
1181
1182             Ok(request)
1183         }
1184     }
1185 }