#![deny(warnings)]
extern crate futures;
extern crate hyper;
extern crate hyper_rustls;
extern crate rustls;
extern crate tokio;
use futures::{Future, Stream};
use hyper::{client, Uri};
use std::str::FromStr;
use std::{env, fs, io};
fn main() {
if let Err(e) = run_client() {
eprintln!("FAILED: {}", e);
std::process::exit(1);
}
}
fn error(err: String) -> io::Error {
io::Error::new(io::ErrorKind::Other, err)
}
fn run_client() -> io::Result<()> {
let url = match env::args().nth(1) {
Some(ref url) => {
Uri::from_str(url).map_err(|e| error(format!("{}", e)))?
}
None => {
println!("Usage: client <url> <ca_store>");
return Ok(());
}
};
let mut ca = match env::args().nth(2) {
Some(ref path) => {
let f = fs::File::open(path).map_err(|e| {
error(format!("failed to open {}: {}", path, e))
})?;
let rd = io::BufReader::new(f);
Some(rd)
}
None => None,
};
let https = match ca {
Some(ref mut rd) => {
let mut http = client::HttpConnector::new(4);
http.enforce_http(false);
let mut tls = rustls::ClientConfig::new();
tls.root_store.add_pem_file(rd).map_err(|_| {
error("failed to load custom CA store".into())
})?;
hyper_rustls::HttpsConnector::from((http, tls))
}
None => hyper_rustls::HttpsConnector::new(4),
};
let client: client::Client<_, hyper::Body> = client::Client::builder().build(https);
let fut = client
.get(url)
.inspect(|res| {
println!("Status:\n{}", res.status());
println!("Headers:\n{:#?}", res.headers());
})
.and_then(|res| res.into_body().concat2())
.inspect(|body| {
println!("Body:\n{}", String::from_utf8_lossy(&body));
});
let mut rt = tokio::runtime::Runtime::new()?;
rt.block_on(fut)
.map_err(|e| error(format!("{}", e)))?;
Ok(())
}