1
0
mirror of https://github.com/fafhrd91/actix-web synced 2024-12-04 20:31:55 +01:00
actix-web/awc/tests/test_ssl_client.rs

95 lines
2.8 KiB
Rust
Raw Normal View History

2019-11-20 19:35:07 +01:00
#![cfg(feature = "openssl")]
2021-02-07 04:54:58 +01:00
extern crate tls_openssl as openssl;
2023-07-17 03:38:12 +02:00
use std::sync::{
atomic::{AtomicUsize, Ordering},
Arc,
};
use actix_http::HttpService;
2019-12-12 18:08:38 +01:00
use actix_http_test::test_server;
2021-04-19 00:34:51 +02:00
use actix_service::{fn_service, map_config, ServiceFactoryExt};
2021-04-01 16:26:13 +02:00
use actix_utils::future::ok;
2023-07-17 03:38:12 +02:00
use actix_web::{dev::AppConfig, http::Version, web, App, HttpResponse};
use openssl::{
pkey::PKey,
ssl::{SslAcceptor, SslConnector, SslMethod, SslVerifyMode},
x509::X509,
};
fn tls_config() -> SslAcceptor {
2024-05-19 11:12:32 +02:00
let rcgen::CertifiedKey { cert, key_pair } =
rcgen::generate_simple_self_signed(["localhost".to_owned()]).unwrap();
let cert_file = cert.pem();
let key_file = key_pair.serialize_pem();
let cert = X509::from_pem(cert_file.as_bytes()).unwrap();
let key = PKey::private_key_from_pem(key_file.as_bytes()).unwrap();
let mut builder = SslAcceptor::mozilla_intermediate(SslMethod::tls()).unwrap();
builder.set_certificate(&cert).unwrap();
builder.set_private_key(&key).unwrap();
builder.set_alpn_select_callback(|_, protos| {
const H2: &[u8] = b"\x02h2";
if protos.windows(3).any(|window| window == H2) {
Ok(b"h2")
} else {
2021-02-07 04:54:58 +01:00
Err(openssl::ssl::AlpnError::NOACK)
}
});
2019-12-02 12:33:11 +01:00
builder.set_alpn_protos(b"\x02h2").unwrap();
2019-12-02 12:33:11 +01:00
builder.build()
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_connection_reuse_h2() {
let num = Arc::new(AtomicUsize::new(0));
let num2 = num.clone();
2019-12-12 18:08:38 +01:00
let srv = test_server(move || {
2019-11-26 06:25:50 +01:00
let num2 = num2.clone();
2021-04-16 21:28:21 +02:00
fn_service(move |io| {
2019-11-26 06:25:50 +01:00
num2.fetch_add(1, Ordering::Relaxed);
ok(io)
})
.and_then(
HttpService::build()
2019-12-20 12:27:32 +01:00
.h2(map_config(
2021-02-12 00:03:17 +01:00
App::new().service(web::resource("/").route(web::to(HttpResponse::Ok))),
2019-12-20 12:27:32 +01:00
|_| AppConfig::default(),
))
.openssl(tls_config())
2019-11-26 06:25:50 +01:00
.map_err(|_| ()),
)
2020-03-08 08:42:45 +01:00
})
.await;
2019-11-26 06:25:50 +01:00
// disable ssl verification
let mut builder = SslConnector::builder(SslMethod::tls()).unwrap();
builder.set_verify(SslVerifyMode::NONE);
let _ = builder
.set_alpn_protos(b"\x02h2\x08http/1.1")
.map_err(|e| log::error!("Can not set alpn protocol: {:?}", e));
let client = awc::Client::builder()
.connector(awc::Connector::new().openssl(builder.build()))
2019-11-26 06:25:50 +01:00
.finish();
2019-11-26 06:25:50 +01:00
// req 1
let request = client.get(srv.surl("/")).send();
let response = request.await.unwrap();
assert!(response.status().is_success());
2019-11-26 06:25:50 +01:00
// req 2
let req = client.post(srv.surl("/"));
let response = req.send().await.unwrap();
assert!(response.status().is_success());
assert_eq!(response.version(), Version::HTTP_2);
2019-11-26 06:25:50 +01:00
// one connection
assert_eq!(num.load(Ordering::Relaxed), 1);
}