mirror of
https://github.com/fafhrd91/actix-web
synced 2024-11-24 16:32:59 +01:00
cb19ebfe0c
* add rustls support for actix-http and awc * fix features conflict * remove unnecessary duplication * test server with rust-tls * fix * test rustls * awc rustls test * format * tests * fix dependencies * fixes and add changes * remove test-server and Cargo.toml dev-dependencies changes * cargo fmt
87 lines
2.7 KiB
Rust
87 lines
2.7 KiB
Rust
#![cfg(feature = "ssl")]
|
|
use openssl::ssl::{SslAcceptor, SslConnector, SslFiletype, SslMethod, SslVerifyMode};
|
|
|
|
use std::io::Result;
|
|
use std::sync::atomic::{AtomicUsize, Ordering};
|
|
use std::sync::Arc;
|
|
|
|
use actix_codec::{AsyncRead, AsyncWrite};
|
|
use actix_http::HttpService;
|
|
use actix_http_test::TestServer;
|
|
use actix_server::ssl::OpensslAcceptor;
|
|
use actix_service::{service_fn, NewService};
|
|
use actix_web::http::Version;
|
|
use actix_web::{web, App, HttpResponse};
|
|
|
|
fn ssl_acceptor<T: AsyncRead + AsyncWrite>() -> Result<OpensslAcceptor<T, ()>> {
|
|
// load ssl keys
|
|
let mut builder = SslAcceptor::mozilla_intermediate(SslMethod::tls()).unwrap();
|
|
builder
|
|
.set_private_key_file("../tests/key.pem", SslFiletype::PEM)
|
|
.unwrap();
|
|
builder
|
|
.set_certificate_chain_file("../tests/cert.pem")
|
|
.unwrap();
|
|
builder.set_alpn_select_callback(|_, protos| {
|
|
const H2: &[u8] = b"\x02h2";
|
|
if protos.windows(3).any(|window| window == H2) {
|
|
Ok(b"h2")
|
|
} else {
|
|
Err(openssl::ssl::AlpnError::NOACK)
|
|
}
|
|
});
|
|
builder.set_alpn_protos(b"\x02h2")?;
|
|
Ok(actix_server::ssl::OpensslAcceptor::new(builder.build()))
|
|
}
|
|
|
|
#[test]
|
|
fn test_connection_reuse_h2() {
|
|
let openssl = ssl_acceptor().unwrap();
|
|
let num = Arc::new(AtomicUsize::new(0));
|
|
let num2 = num.clone();
|
|
|
|
let mut srv = TestServer::new(move || {
|
|
let num2 = num2.clone();
|
|
service_fn(move |io| {
|
|
num2.fetch_add(1, Ordering::Relaxed);
|
|
Ok(io)
|
|
})
|
|
.and_then(
|
|
openssl
|
|
.clone()
|
|
.map_err(|e| println!("Openssl error: {}", e)),
|
|
)
|
|
.and_then(
|
|
HttpService::build()
|
|
.h2(App::new()
|
|
.service(web::resource("/").route(web::to(|| HttpResponse::Ok()))))
|
|
.map_err(|_| ()),
|
|
)
|
|
});
|
|
|
|
// 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::build()
|
|
.connector(awc::Connector::new().ssl(builder.build()).finish())
|
|
.finish();
|
|
|
|
// req 1
|
|
let request = client.get(srv.surl("/")).send();
|
|
let response = srv.block_on(request).unwrap();
|
|
assert!(response.status().is_success());
|
|
|
|
// req 2
|
|
let req = client.post(srv.surl("/"));
|
|
let response = srv.block_on_fn(move || req.send()).unwrap();
|
|
assert!(response.status().is_success());
|
|
assert_eq!(response.version(), Version::HTTP_2);
|
|
|
|
// one connection
|
|
assert_eq!(num.load(Ordering::Relaxed), 1);
|
|
}
|