66 lines
2.2 KiB
Rust
66 lines
2.2 KiB
Rust
//! This examples shows how you can combine `hyper-rustls` and `tonic` to
|
|
//! provide a custom `ClientConfig` for the tls configuration.
|
|
|
|
pub mod pb {
|
|
tonic::include_proto!("/grpc.examples.unaryecho");
|
|
}
|
|
|
|
use hyper::{client::HttpConnector, Uri};
|
|
use pb::{echo_client::EchoClient, EchoRequest};
|
|
use tokio_rustls::rustls::{ClientConfig, RootCertStore};
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let fd = std::fs::File::open("examples/data/tls/ca.pem")?;
|
|
|
|
let mut roots = RootCertStore::empty();
|
|
|
|
let mut buf = std::io::BufReader::new(&fd);
|
|
let certs = rustls_pemfile::certs(&mut buf)?;
|
|
roots.add_parsable_certificates(&certs);
|
|
|
|
let tls = ClientConfig::builder()
|
|
.with_safe_defaults()
|
|
.with_root_certificates(roots)
|
|
.with_no_client_auth();
|
|
|
|
let mut http = HttpConnector::new();
|
|
http.enforce_http(false);
|
|
|
|
// We have to do some wrapping here to map the request type from
|
|
// `https://example.com` -> `https://[::1]:50051` because `rustls`
|
|
// doesn't accept ip's as `ServerName`.
|
|
let connector = tower::ServiceBuilder::new()
|
|
.layer_fn(move |s| {
|
|
let tls = tls.clone();
|
|
|
|
hyper_rustls::HttpsConnectorBuilder::new()
|
|
.with_tls_config(tls)
|
|
.https_or_http()
|
|
.enable_http2()
|
|
.wrap_connector(s)
|
|
})
|
|
// Since our cert is signed with `example.com` but we actually want to connect
|
|
// to a local server we will override the Uri passed from the `HttpsConnector`
|
|
// and map it to the correct `Uri` that will connect us directly to the local server.
|
|
.map_request(|_| Uri::from_static("https://[::1]:50051"))
|
|
.service(http);
|
|
|
|
let client = hyper::Client::builder().build(connector);
|
|
|
|
// Using `with_origin` will let the codegenerated client set the `scheme` and
|
|
// `authority` from the porvided `Uri`.
|
|
let uri = Uri::from_static("https://example.com");
|
|
let mut client = EchoClient::with_origin(client, uri);
|
|
|
|
let request = tonic::Request::new(EchoRequest {
|
|
message: "hello".into(),
|
|
});
|
|
|
|
let response = client.unary_echo(request).await?;
|
|
|
|
println!("RESPONSE={:?}", response);
|
|
|
|
Ok(())
|
|
}
|