1
0
mirror of https://github.com/actix/examples synced 2024-12-01 01:24:35 +01:00
examples/middleware/http-to-https/src/main.rs

69 lines
2.5 KiB
Rust
Raw Normal View History

2022-02-18 03:18:44 +01:00
use std::{fs::File, io::BufReader};
2022-02-18 03:18:44 +01:00
use actix_web::{dev::Service, get, http, App, HttpResponse, HttpServer};
2022-02-22 13:12:17 +01:00
use futures_util::future::{self, Either, FutureExt};
2024-05-25 06:36:36 +02:00
use rustls::{pki_types::PrivateKeyDer, ServerConfig};
use rustls_pemfile::{certs, pkcs8_private_keys};
#[get("/")]
async fn index() -> String {
2022-02-18 03:44:02 +01:00
String::from("<html><head><title>FOO BAR</title></head><body><h1>FOO BAR</h1></body></html>")
}
#[actix_web::main]
async fn main() -> std::io::Result<()> {
2022-02-22 13:12:17 +01:00
env_logger::init_from_env(env_logger::Env::new().default_filter_or("info"));
2024-05-25 06:36:36 +02:00
rustls::crypto::aws_lc_rs::default_provider()
.install_default()
.unwrap();
let cert_file = &mut BufReader::new(File::open("cert.pem").unwrap());
let key_file = &mut BufReader::new(File::open("key.pem").unwrap());
2024-05-25 06:36:36 +02:00
let cert_chain = certs(cert_file).collect::<Result<Vec<_>, _>>().unwrap();
let mut keys = pkcs8_private_keys(key_file)
.map(|key| key.map(PrivateKeyDer::Pkcs8))
.collect::<Result<Vec<_>, _>>()
.unwrap();
let config = ServerConfig::builder()
.with_no_client_auth()
.with_single_cert(cert_chain, keys.remove(0))
.unwrap();
2024-05-25 06:36:36 +02:00
log::info!(
"starting HTTP server at http://localhost:80 and HTTPS server on http://localhost:443"
);
2022-02-22 13:12:17 +01:00
HttpServer::new(|| {
App::new()
.wrap_fn(|sreq, srv| {
let host = sreq.connection_info().host().to_owned();
let uri = sreq.uri().to_owned();
let url = format!("https://{host}{uri}");
// If the scheme is "https" then it will let other services below this wrap_fn
// handle the request and if it's "http" then a response with redirect status code
// will be sent whose "location" header will be same as before, with just "http"
// changed to "https"
2023-02-26 22:01:15 +01:00
if sreq.connection_info().scheme() == "https" {
Either::Left(srv.call(sreq).map(|res| res))
} else {
println!("An http request has arrived here, i will redirect it to use https");
return Either::Right(future::ready(Ok(sreq.into_response(
HttpResponse::MovedPermanently()
.append_header((http::header::LOCATION, url))
.finish(),
))));
}
})
.service(index)
})
2022-03-06 01:41:32 +01:00
.bind(("127.0.0.1", 80))? // HTTP port
2024-05-25 06:36:36 +02:00
.bind_rustls_0_23(("127.0.0.1", 443), config)? // HTTPS port
.run()
.await
}