1
0
Fork 0
mirror of https://github.com/actix/actix-web.git synced 2024-05-19 16:58:14 +00:00
actix-web/actix-web/src/server.rs

817 lines
29 KiB
Rust
Raw Normal View History

use std::{
any::Any,
2021-12-04 19:40:47 +00:00
cmp, fmt, io,
marker::PhantomData,
net,
sync::{Arc, Mutex},
2022-01-31 17:30:34 +00:00
time::Duration,
};
2023-07-17 01:38:12 +00:00
#[cfg(any(feature = "openssl", feature = "rustls"))]
use actix_http::TlsAcceptorConfig;
2021-06-17 16:57:58 +00:00
use actix_http::{body::MessageBody, Extensions, HttpService, KeepAlive, Request, Response};
2019-03-05 00:29:03 +00:00
use actix_server::{Server, ServerBuilder};
2021-06-17 16:57:58 +00:00
use actix_service::{
map_config, IntoServiceFactory, Service, ServiceFactory, ServiceFactoryExt as _,
};
2019-11-20 17:33:22 +00:00
#[cfg(feature = "openssl")]
2021-11-30 14:12:04 +00:00
use actix_tls::accept::openssl::reexports::{AlpnError, SslAcceptor, SslAcceptorBuilder};
2019-11-20 17:33:22 +00:00
#[cfg(feature = "rustls")]
2021-11-30 14:12:04 +00:00
use actix_tls::accept::rustls::reexports::ServerConfig as RustlsServerConfig;
2019-03-05 00:29:03 +00:00
2021-06-17 16:57:58 +00:00
use crate::{config::AppConfig, Error};
2019-03-05 00:29:03 +00:00
struct Socket {
scheme: &'static str,
addr: net::SocketAddr,
}
struct Config {
host: Option<String>,
2019-03-05 00:29:03 +00:00
keep_alive: KeepAlive,
2022-01-31 17:30:34 +00:00
client_request_timeout: Duration,
client_disconnect_timeout: Duration,
#[cfg(any(feature = "openssl", feature = "rustls"))]
tls_handshake_timeout: Option<Duration>,
2019-03-05 00:29:03 +00:00
}
/// An HTTP Server.
///
2021-02-11 22:39:54 +00:00
/// Create new HTTP server with application factory.
2019-03-05 00:29:03 +00:00
///
/// # Automatic HTTP Version Selection
///
/// There are two ways to select the HTTP version of an incoming connection:
///
/// - One is to rely on the ALPN information that is provided when using a TLS (HTTPS); both
/// versions are supported automatically when using either of the `.bind_rustls()` or
/// `.bind_openssl()` methods.
/// - The other is to read the first few bytes of the TCP stream. This is the only viable approach
/// for supporting H2C, which allows the HTTP/2 protocol to work over plaintext connections. Use
/// the `.bind_auto_h2c()` method to enable this behavior.
2022-07-31 20:58:15 +00:00
///
/// # Examples
///
/// ```no_run
/// use actix_web::{web, App, HttpResponse, HttpServer};
2019-03-05 00:29:03 +00:00
///
2022-07-31 20:58:15 +00:00
/// #[actix_web::main]
/// async fn main() -> std::io::Result<()> {
2022-07-31 20:58:15 +00:00
/// HttpServer::new(|| {
/// App::new()
/// .service(web::resource("/").to(|| async { "hello world" }))
/// })
/// .bind(("127.0.0.1", 8080))?
/// .run()
/// .await
2019-03-05 00:29:03 +00:00
/// }
/// ```
pub struct HttpServer<F, I, S, B>
where
F: Fn() -> I + Send + Clone + 'static,
I: IntoServiceFactory<S, Request>,
S: ServiceFactory<Request, Config = AppConfig>,
2019-04-05 23:46:44 +00:00
S::Error: Into<Error>,
S::InitError: fmt::Debug,
2019-03-05 00:29:03 +00:00
S::Response: Into<Response<B>>,
B: MessageBody,
{
pub(super) factory: F,
config: Arc<Mutex<Config>>,
backlog: u32,
2019-03-05 00:29:03 +00:00
sockets: Vec<Socket>,
builder: ServerBuilder,
2021-12-25 04:53:51 +00:00
#[allow(clippy::type_complexity)]
on_connect_fn: Option<Arc<dyn Fn(&dyn Any, &mut Extensions) + Send + Sync>>,
2021-01-04 00:49:02 +00:00
_phantom: PhantomData<(S, B)>,
2019-03-05 00:29:03 +00:00
}
impl<F, I, S, B> HttpServer<F, I, S, B>
where
F: Fn() -> I + Send + Clone + 'static,
I: IntoServiceFactory<S, Request>,
S: ServiceFactory<Request, Config = AppConfig> + 'static,
2019-11-20 17:33:22 +00:00
S::Error: Into<Error> + 'static,
2019-04-05 23:46:44 +00:00
S::InitError: fmt::Debug,
2019-11-20 17:33:22 +00:00
S::Response: Into<Response<B>> + 'static,
<S::Service as Service<Request>>::Future: 'static,
S::Service: 'static,
2021-12-04 19:40:47 +00:00
2019-03-07 07:06:14 +00:00
B: MessageBody + 'static,
2019-03-05 00:29:03 +00:00
{
2021-02-11 22:39:54 +00:00
/// Create new HTTP server with application factory
2019-03-05 00:29:03 +00:00
pub fn new(factory: F) -> Self {
HttpServer {
factory,
config: Arc::new(Mutex::new(Config {
host: None,
2022-01-31 17:30:34 +00:00
keep_alive: KeepAlive::default(),
client_request_timeout: Duration::from_secs(5),
client_disconnect_timeout: Duration::from_secs(1),
#[cfg(any(feature = "rustls", feature = "openssl"))]
tls_handshake_timeout: None,
2019-03-05 00:29:03 +00:00
})),
2019-03-11 22:26:05 +00:00
backlog: 1024,
2019-03-05 00:29:03 +00:00
sockets: Vec::new(),
builder: ServerBuilder::default(),
on_connect_fn: None,
2021-01-04 00:49:02 +00:00
_phantom: PhantomData,
}
}
2022-07-31 20:58:15 +00:00
/// Sets number of workers to start (per bind address).
2019-03-05 00:29:03 +00:00
///
2022-02-08 06:58:26 +00:00
/// By default, the number of available physical CPUs is used as the worker count.
2019-03-05 00:29:03 +00:00
pub fn workers(mut self, num: usize) -> Self {
self.builder = self.builder.workers(num);
2019-03-05 00:29:03 +00:00
self
}
/// Sets server keep-alive preference.
///
/// By default keep-alive is set to 5 seconds.
pub fn keep_alive<T: Into<KeepAlive>>(self, val: T) -> Self {
self.config.lock().unwrap().keep_alive = val.into();
self
}
2022-07-31 20:58:15 +00:00
/// Sets the maximum number of pending connections.
2019-03-05 00:29:03 +00:00
///
/// This refers to the number of clients that can be waiting to be served. Exceeding this number
/// results in the client getting an error when attempting to connect. It should only affect
/// servers under significant load.
2019-03-05 00:29:03 +00:00
///
/// Generally set in the 642048 range. Default value is 2048.
2019-03-05 00:29:03 +00:00
///
/// This method will have no effect if called after a `bind()`.
pub fn backlog(mut self, backlog: u32) -> Self {
2019-03-11 22:26:05 +00:00
self.backlog = backlog;
self.builder = self.builder.backlog(backlog);
2019-03-05 00:29:03 +00:00
self
}
2022-07-31 20:58:15 +00:00
/// Sets the per-worker maximum number of concurrent connections.
2019-03-05 00:29:03 +00:00
///
2020-09-09 08:20:54 +00:00
/// All socket listeners will stop accepting connections when this limit is reached for
/// each worker.
2019-03-05 00:29:03 +00:00
///
/// By default max connections is set to a 25k.
2020-09-09 08:20:54 +00:00
pub fn max_connections(mut self, num: usize) -> Self {
2021-11-15 04:03:33 +00:00
self.builder = self.builder.max_concurrent_connections(num);
2019-03-05 00:29:03 +00:00
self
}
2022-07-31 20:58:15 +00:00
/// Sets the per-worker maximum concurrent TLS connection limit.
2019-03-05 00:29:03 +00:00
///
2020-09-09 08:20:54 +00:00
/// All listeners will stop accepting connections when this limit is reached. It can be used to
/// limit the global TLS CPU usage.
2019-03-05 00:29:03 +00:00
///
/// By default max connections is set to a 256.
#[allow(unused_variables)]
2020-09-09 08:20:54 +00:00
pub fn max_connection_rate(self, num: usize) -> Self {
#[cfg(any(feature = "rustls", feature = "openssl"))]
actix_tls::accept::max_concurrent_tls_connect(num);
2019-03-05 00:29:03 +00:00
self
}
2022-07-31 20:58:15 +00:00
/// Sets max number of threads for each worker's blocking task thread pool.
///
/// One thread pool is set up **per worker**; not shared across workers.
///
/// By default set to 512 divided by the number of workers.
pub fn worker_max_blocking_threads(mut self, num: usize) -> Self {
self.builder = self.builder.worker_max_blocking_threads(num);
self
}
2022-07-31 20:58:15 +00:00
/// Sets server client timeout for first request.
2019-03-05 00:29:03 +00:00
///
2022-07-31 20:58:15 +00:00
/// Defines a timeout for reading client request head. If a client does not transmit the entire
/// set headers within this time, the request is terminated with a 408 (Request Timeout) error.
2019-03-05 00:29:03 +00:00
///
/// To disable timeout set value to 0.
///
/// By default client timeout is set to 5000 milliseconds.
2022-01-31 17:30:34 +00:00
pub fn client_request_timeout(self, dur: Duration) -> Self {
self.config.lock().unwrap().client_request_timeout = dur;
2019-03-05 00:29:03 +00:00
self
}
2022-01-31 17:30:34 +00:00
#[doc(hidden)]
#[deprecated(since = "4.0.0", note = "Renamed to `client_request_timeout`.")]
pub fn client_timeout(self, dur: Duration) -> Self {
self.client_request_timeout(dur)
}
2022-07-31 20:58:15 +00:00
/// Sets server connection shutdown timeout.
2019-03-05 00:29:03 +00:00
///
2022-07-31 20:58:15 +00:00
/// Defines a timeout for connection shutdown. If a shutdown procedure does not complete within
/// this time, the request is dropped.
2019-03-05 00:29:03 +00:00
///
/// To disable timeout set value to 0.
///
/// By default client timeout is set to 5000 milliseconds.
2022-01-31 17:30:34 +00:00
pub fn client_disconnect_timeout(self, dur: Duration) -> Self {
self.config.lock().unwrap().client_disconnect_timeout = dur;
2019-03-05 00:29:03 +00:00
self
}
2022-07-31 20:58:15 +00:00
/// Sets TLS handshake timeout.
///
2022-07-31 20:58:15 +00:00
/// Defines a timeout for TLS handshake. If the TLS handshake does not complete within this
/// time, the connection is closed.
///
/// By default handshake timeout is set to 3000 milliseconds.
#[cfg(any(feature = "openssl", feature = "rustls"))]
pub fn tls_handshake_timeout(self, dur: Duration) -> Self {
self.config
.lock()
.unwrap()
.tls_handshake_timeout
.replace(dur);
self
}
2022-01-31 17:30:34 +00:00
#[doc(hidden)]
#[deprecated(since = "4.0.0", note = "Renamed to `client_disconnect_timeout`.")]
2022-01-31 17:30:34 +00:00
pub fn client_shutdown(self, dur: u64) -> Self {
self.client_disconnect_timeout(Duration::from_millis(dur))
}
/// Sets function that will be called once before each connection is handled.
///
/// It will receive a `&std::any::Any`, which contains underlying connection type and an
/// [Extensions] container so that connection data can be accessed in middleware and handlers.
///
/// # Connection Types
/// - `actix_tls::accept::openssl::TlsStream<actix_web::rt::net::TcpStream>` when using OpenSSL.
/// - `actix_tls::accept::rustls::TlsStream<actix_web::rt::net::TcpStream>` when using Rustls.
/// - `actix_web::rt::net::TcpStream` when no encryption is used.
///
/// See the `on_connect` example for additional details.
pub fn on_connect<CB>(self, f: CB) -> HttpServer<F, I, S, B>
where
CB: Fn(&dyn Any, &mut Extensions) + Send + Sync + 'static,
{
HttpServer {
factory: self.factory,
config: self.config,
backlog: self.backlog,
sockets: self.sockets,
builder: self.builder,
on_connect_fn: Some(Arc::new(f)),
_phantom: PhantomData,
}
}
2022-07-31 20:58:15 +00:00
/// Sets server host name.
2019-03-05 00:29:03 +00:00
///
2022-07-31 20:58:15 +00:00
/// Host name is used by application router as a hostname for url generation. Check
/// [`ConnectionInfo`](crate::dev::ConnectionInfo::host()) docs for more info.
///
2022-07-31 20:58:15 +00:00
/// By default, hostname is set to "localhost".
pub fn server_hostname<T: AsRef<str>>(self, val: T) -> Self {
self.config.lock().unwrap().host = Some(val.as_ref().to_owned());
2019-03-05 00:29:03 +00:00
self
}
/// Flags the `System` to exit after server shutdown.
2022-07-31 20:58:15 +00:00
///
/// Does nothing when running under `#[tokio::main]` runtime.
2019-03-05 00:29:03 +00:00
pub fn system_exit(mut self) -> Self {
self.builder = self.builder.system_exit();
2019-03-05 00:29:03 +00:00
self
}
2022-07-31 20:58:15 +00:00
/// Disables signal handling.
2019-03-05 00:29:03 +00:00
pub fn disable_signals(mut self) -> Self {
self.builder = self.builder.disable_signals();
2019-03-05 00:29:03 +00:00
self
}
2022-07-31 20:58:15 +00:00
/// Sets timeout for graceful worker shutdown of workers.
2019-03-05 00:29:03 +00:00
///
2022-07-31 20:58:15 +00:00
/// After receiving a stop signal, workers have this much time to finish serving requests.
/// Workers still alive after the timeout are force dropped.
2019-03-05 00:29:03 +00:00
///
/// By default shutdown timeout sets to 30 seconds.
pub fn shutdown_timeout(mut self, sec: u64) -> Self {
self.builder = self.builder.shutdown_timeout(sec);
2019-03-05 00:29:03 +00:00
self
}
2022-07-31 20:58:15 +00:00
/// Returns addresses of bound sockets.
2019-03-05 00:29:03 +00:00
pub fn addrs(&self) -> Vec<net::SocketAddr> {
self.sockets.iter().map(|s| s.addr).collect()
}
2022-07-31 20:58:15 +00:00
/// Returns addresses of bound sockets and the scheme for it.
2019-03-05 00:29:03 +00:00
///
2022-07-31 20:58:15 +00:00
/// This is useful when the server is bound from different sources with some sockets listening
/// on HTTP and some listening on HTTPS and the user should be presented with an enumeration of
/// which socket requires which protocol.
2019-03-05 00:29:03 +00:00
pub fn addrs_with_scheme(&self) -> Vec<(net::SocketAddr, &str)> {
self.sockets.iter().map(|s| (s.addr, s.scheme)).collect()
}
/// Resolves socket address(es) and binds server to created listener(s).
///
/// # Hostname Resolution
/// When `addr` includes a hostname, it is possible for this method to bind to both the IPv4 and
/// IPv6 addresses that result from a DNS lookup. You can test this by passing `localhost:8080`
/// and noting that the server binds to `127.0.0.1:8080` _and_ `[::1]:8080`. To bind additional
/// addresses, call this method multiple times.
///
/// Note that, if a DNS lookup is required, resolving hostnames is a blocking operation.
///
/// # Typical Usage
/// In general, use `127.0.0.1:<port>` when testing locally and `0.0.0.0:<port>` when deploying
/// (with or without a reverse proxy or load balancer) so that the server is accessible.
///
/// # Errors
/// Returns an `io::Error` if:
/// - `addrs` cannot be resolved into one or more socket addresses;
/// - all the resolved socket addresses are already bound.
///
/// # Example
/// ```
/// # use actix_web::{App, HttpServer};
/// # fn inner() -> std::io::Result<()> {
/// HttpServer::new(|| App::new())
/// .bind(("127.0.0.1", 8080))?
/// .bind("[::1]:9000")?
/// # ; Ok(()) }
/// ```
pub fn bind<A: net::ToSocketAddrs>(mut self, addrs: A) -> io::Result<Self> {
2023-03-11 22:17:52 +00:00
let sockets = bind_addrs(addrs, self.backlog)?;
for lst in sockets {
self = self.listen(lst)?;
}
Ok(self)
}
/// Resolves socket address(es) and binds server to created listener(s) for plaintext HTTP/1.x
/// or HTTP/2 connections.
pub fn bind_auto_h2c<A: net::ToSocketAddrs>(mut self, addrs: A) -> io::Result<Self> {
let sockets = bind_addrs(addrs, self.backlog)?;
for lst in sockets {
self = self.listen_auto_h2c(lst)?;
}
Ok(self)
}
/// Resolves socket address(es) and binds server to created listener(s) for TLS connections
/// using Rustls.
///
/// See [`bind()`](Self::bind) for more details on `addrs` argument.
///
/// ALPN protocols "h2" and "http/1.1" are added to any configured ones.
#[cfg(feature = "rustls")]
pub fn bind_rustls<A: net::ToSocketAddrs>(
mut self,
addrs: A,
config: RustlsServerConfig,
) -> io::Result<Self> {
2023-03-11 22:17:52 +00:00
let sockets = bind_addrs(addrs, self.backlog)?;
for lst in sockets {
self = self.listen_rustls_inner(lst, config.clone())?;
}
Ok(self)
}
/// Resolves socket address(es) and binds server to created listener(s) for TLS connections
/// using OpenSSL.
///
/// See [`bind()`](Self::bind) for more details on `addrs` argument.
///
/// ALPN protocols "h2" and "http/1.1" are added to any configured ones.
#[cfg(feature = "openssl")]
pub fn bind_openssl<A>(mut self, addrs: A, builder: SslAcceptorBuilder) -> io::Result<Self>
where
A: net::ToSocketAddrs,
{
2023-03-11 22:17:52 +00:00
let sockets = bind_addrs(addrs, self.backlog)?;
let acceptor = openssl_acceptor(builder)?;
for lst in sockets {
self = self.listen_openssl_inner(lst, acceptor.clone())?;
}
Ok(self)
}
2022-07-31 20:58:15 +00:00
/// Binds to existing listener for accepting incoming connection requests.
2019-03-05 00:29:03 +00:00
///
2022-07-31 20:58:15 +00:00
/// No changes are made to `lst`'s configuration. Ensure it is configured properly before
/// passing ownership to `listen()`.
pub fn listen(mut self, lst: net::TcpListener) -> io::Result<Self> {
2019-03-05 00:29:03 +00:00
let cfg = self.config.clone();
let factory = self.factory.clone();
let addr = lst.local_addr().unwrap();
2022-07-31 20:58:15 +00:00
2019-03-05 00:29:03 +00:00
self.sockets.push(Socket {
addr,
scheme: "http",
});
2022-07-31 20:58:15 +00:00
let on_connect_fn = self.on_connect_fn.clone();
2019-03-05 00:29:03 +00:00
2021-02-11 23:03:17 +00:00
self.builder =
self.builder
.listen(format!("actix-web-service-{}", addr), lst, move || {
let cfg = cfg.lock().unwrap();
let host = cfg.host.clone().unwrap_or_else(|| format!("{}", addr));
let mut svc = HttpService::build()
.keep_alive(cfg.keep_alive)
.client_request_timeout(cfg.client_request_timeout)
.client_disconnect_timeout(cfg.client_disconnect_timeout)
2021-02-11 23:03:17 +00:00
.local_addr(addr);
if let Some(handler) = on_connect_fn.clone() {
2023-07-17 01:38:12 +00:00
svc =
svc.on_connect_ext(move |io: &_, ext: _| (handler)(io as &dyn Any, ext))
2021-02-11 23:03:17 +00:00
};
2021-06-17 16:57:58 +00:00
let fac = factory()
.into_factory()
.map_err(|err| err.into().error_response());
svc.finish(map_config(fac, move |_| {
AppConfig::new(false, host.clone(), addr)
2021-02-11 23:03:17 +00:00
}))
.tcp()
})?;
Ok(self)
}
/// Binds to existing listener for accepting incoming plaintext HTTP/1.x or HTTP/2 connections.
pub fn listen_auto_h2c(mut self, lst: net::TcpListener) -> io::Result<Self> {
let cfg = self.config.clone();
let factory = self.factory.clone();
let addr = lst.local_addr().unwrap();
self.sockets.push(Socket {
addr,
scheme: "http",
});
let on_connect_fn = self.on_connect_fn.clone();
self.builder =
self.builder
.listen(format!("actix-web-service-{}", addr), lst, move || {
let cfg = cfg.lock().unwrap();
let host = cfg.host.clone().unwrap_or_else(|| format!("{}", addr));
let mut svc = HttpService::build()
.keep_alive(cfg.keep_alive)
.client_request_timeout(cfg.client_request_timeout)
.client_disconnect_timeout(cfg.client_disconnect_timeout)
.local_addr(addr);
if let Some(handler) = on_connect_fn.clone() {
2023-07-17 01:38:12 +00:00
svc =
svc.on_connect_ext(move |io: &_, ext: _| (handler)(io as &dyn Any, ext))
};
let fac = factory()
.into_factory()
.map_err(|err| err.into().error_response());
svc.finish(map_config(fac, move |_| {
AppConfig::new(false, host.clone(), addr)
}))
.tcp_auto_h2c()
})?;
Ok(self)
2019-03-05 00:29:03 +00:00
}
/// Binds to existing listener for accepting incoming TLS connection requests using Rustls.
2022-07-31 20:58:15 +00:00
///
/// See [`listen()`](Self::listen) for more details on the `lst` argument.
2019-03-05 00:29:03 +00:00
///
2022-07-31 20:58:15 +00:00
/// ALPN protocols "h2" and "http/1.1" are added to any configured ones.
#[cfg(feature = "rustls")]
pub fn listen_rustls(
self,
2019-03-05 00:29:03 +00:00
lst: net::TcpListener,
config: RustlsServerConfig,
2019-03-05 00:29:03 +00:00
) -> io::Result<Self> {
self.listen_rustls_inner(lst, config)
2019-03-05 00:29:03 +00:00
}
#[cfg(feature = "rustls")]
fn listen_rustls_inner(
mut self,
lst: net::TcpListener,
config: RustlsServerConfig,
) -> io::Result<Self> {
2019-03-05 00:29:03 +00:00
let factory = self.factory.clone();
let cfg = self.config.clone();
let addr = lst.local_addr().unwrap();
self.sockets.push(Socket {
addr,
2019-03-29 23:28:19 +00:00
scheme: "https",
2019-03-05 00:29:03 +00:00
});
let on_connect_fn = self.on_connect_fn.clone();
2021-02-11 23:03:17 +00:00
self.builder =
self.builder
.listen(format!("actix-web-service-{}", addr), lst, move || {
let c = cfg.lock().unwrap();
let host = c.host.clone().unwrap_or_else(|| format!("{}", addr));
2021-02-11 23:03:17 +00:00
let svc = HttpService::build()
.keep_alive(c.keep_alive)
2022-01-31 17:30:34 +00:00
.client_request_timeout(c.client_request_timeout)
.client_disconnect_timeout(c.client_disconnect_timeout);
2021-02-11 23:03:17 +00:00
let svc = if let Some(handler) = on_connect_fn.clone() {
2022-03-10 03:12:29 +00:00
svc.on_connect_ext(move |io: &_, ext: _| (handler)(io as &dyn Any, ext))
2021-02-11 23:03:17 +00:00
} else {
svc
};
2021-06-17 16:57:58 +00:00
let fac = factory()
.into_factory()
.map_err(|err| err.into().error_response());
let acceptor_config = match c.tls_handshake_timeout {
Some(dur) => TlsAcceptorConfig::default().handshake_timeout(dur),
None => TlsAcceptorConfig::default(),
};
2021-06-17 16:57:58 +00:00
svc.finish(map_config(fac, move |_| {
AppConfig::new(true, host.clone(), addr)
2021-02-11 23:03:17 +00:00
}))
.rustls_with_config(config.clone(), acceptor_config)
2021-02-11 23:03:17 +00:00
})?;
Ok(self)
2019-03-05 00:29:03 +00:00
}
/// Binds to existing listener for accepting incoming TLS connection requests using OpenSSL.
2022-07-31 20:58:15 +00:00
///
/// See [`listen()`](Self::listen) for more details on the `lst` argument.
2019-03-05 00:29:03 +00:00
///
2022-07-31 20:58:15 +00:00
/// ALPN protocols "h2" and "http/1.1" are added to any configured ones.
#[cfg(feature = "openssl")]
pub fn listen_openssl(
self,
2019-03-29 23:28:19 +00:00
lst: net::TcpListener,
builder: SslAcceptorBuilder,
2019-03-29 23:28:19 +00:00
) -> io::Result<Self> {
self.listen_openssl_inner(lst, openssl_acceptor(builder)?)
2019-03-29 23:28:19 +00:00
}
2019-03-05 00:29:03 +00:00
#[cfg(feature = "openssl")]
fn listen_openssl_inner(
mut self,
2019-03-29 23:28:19 +00:00
lst: net::TcpListener,
acceptor: SslAcceptor,
) -> io::Result<Self> {
2019-03-29 23:28:19 +00:00
let factory = self.factory.clone();
let cfg = self.config.clone();
let addr = lst.local_addr().unwrap();
self.sockets.push(Socket {
addr,
scheme: "https",
});
let on_connect_fn = self.on_connect_fn.clone();
2021-02-11 23:03:17 +00:00
self.builder =
self.builder
.listen(format!("actix-web-service-{}", addr), lst, move || {
let c = cfg.lock().unwrap();
let host = c.host.clone().unwrap_or_else(|| format!("{}", addr));
2021-02-11 23:03:17 +00:00
let svc = HttpService::build()
.keep_alive(c.keep_alive)
2022-01-31 17:30:34 +00:00
.client_request_timeout(c.client_request_timeout)
.client_disconnect_timeout(c.client_disconnect_timeout)
.local_addr(addr);
2021-02-11 23:03:17 +00:00
let svc = if let Some(handler) = on_connect_fn.clone() {
svc.on_connect_ext(move |io: &_, ext: _| (handler)(io as &dyn Any, ext))
} else {
svc
};
2021-06-17 16:57:58 +00:00
let fac = factory()
.into_factory()
.map_err(|err| err.into().error_response());
// false positive lint (?)
#[allow(clippy::significant_drop_in_scrutinee)]
let acceptor_config = match c.tls_handshake_timeout {
Some(dur) => TlsAcceptorConfig::default().handshake_timeout(dur),
None => TlsAcceptorConfig::default(),
};
2021-06-17 16:57:58 +00:00
svc.finish(map_config(fac, move |_| {
AppConfig::new(true, host.clone(), addr)
2021-02-11 23:03:17 +00:00
}))
.openssl_with_config(acceptor.clone(), acceptor_config)
2021-02-11 23:03:17 +00:00
})?;
Ok(self)
2019-03-05 00:29:03 +00:00
}
/// Opens Unix Domain Socket (UDS) from `uds` path and binds server to created listener.
#[cfg(unix)]
pub fn bind_uds<A>(mut self, uds_path: A) -> io::Result<Self>
where
A: AsRef<std::path::Path>,
{
use actix_http::Protocol;
use actix_rt::net::UnixStream;
use actix_service::{fn_service, ServiceFactoryExt as _};
2020-09-09 08:20:54 +00:00
let cfg = self.config.clone();
let factory = self.factory.clone();
let socket_addr =
net::SocketAddr::new(net::IpAddr::V4(net::Ipv4Addr::new(127, 0, 0, 1)), 8080);
2019-03-05 00:29:03 +00:00
self.sockets.push(Socket {
scheme: "http",
addr: socket_addr,
});
2019-03-05 00:29:03 +00:00
self.builder = self.builder.bind_uds(
format!("actix-web-service-{:?}", uds_path.as_ref()),
uds_path,
move || {
let c = cfg.lock().unwrap();
let config = AppConfig::new(
false,
c.host.clone().unwrap_or_else(|| format!("{}", socket_addr)),
socket_addr,
);
2019-03-05 00:29:03 +00:00
let fac = factory()
.into_factory()
.map_err(|err| err.into().error_response());
2019-03-05 00:29:03 +00:00
fn_service(|io: UnixStream| async { Ok((io, Protocol::Http1, None)) }).and_then(
HttpService::build()
.keep_alive(c.keep_alive)
.client_request_timeout(c.client_request_timeout)
.client_disconnect_timeout(c.client_disconnect_timeout)
.finish(map_config(fac, move |_| config.clone())),
)
},
)?;
2019-03-05 00:29:03 +00:00
2019-03-29 23:28:19 +00:00
Ok(self)
2019-03-05 00:29:03 +00:00
}
/// Binds to existing Unix Domain Socket (UDS) listener.
2022-07-31 20:58:15 +00:00
#[cfg(unix)]
2021-02-11 23:03:17 +00:00
pub fn listen_uds(mut self, lst: std::os::unix::net::UnixListener) -> io::Result<Self> {
use actix_http::Protocol;
2019-12-02 11:33:11 +00:00
use actix_rt::net::UnixStream;
2021-04-16 19:28:21 +00:00
use actix_service::{fn_service, ServiceFactoryExt as _};
2019-12-02 11:33:11 +00:00
let cfg = self.config.clone();
let factory = self.factory.clone();
2021-02-11 23:03:17 +00:00
let socket_addr =
net::SocketAddr::new(net::IpAddr::V4(net::Ipv4Addr::new(127, 0, 0, 1)), 8080);
self.sockets.push(Socket {
scheme: "http",
addr: socket_addr,
});
2021-07-12 15:55:24 +00:00
let addr = lst.local_addr()?;
let name = format!("actix-web-service-{:?}", addr);
let on_connect_fn = self.on_connect_fn.clone();
2021-07-12 15:55:24 +00:00
self.builder = self.builder.listen_uds(name, lst, move || {
2019-12-10 03:00:51 +00:00
let c = cfg.lock().unwrap();
let config = AppConfig::new(
false,
c.host.clone().unwrap_or_else(|| format!("{}", socket_addr)),
socket_addr,
);
2021-04-16 19:28:21 +00:00
fn_service(|io: UnixStream| async { Ok((io, Protocol::Http1, None)) }).and_then({
let mut svc = HttpService::build()
2021-04-16 19:28:21 +00:00
.keep_alive(c.keep_alive)
2022-01-31 17:30:34 +00:00
.client_request_timeout(c.client_request_timeout)
.client_disconnect_timeout(c.client_disconnect_timeout);
if let Some(handler) = on_connect_fn.clone() {
2023-07-17 01:38:12 +00:00
svc = svc.on_connect_ext(move |io: &_, ext: _| (handler)(io as &dyn Any, ext));
}
2021-06-17 16:57:58 +00:00
let fac = factory()
.into_factory()
.map_err(|err| err.into().error_response());
svc.finish(map_config(fac, move |_| config.clone()))
2021-04-16 19:28:21 +00:00
})
})?;
Ok(self)
}
2019-03-05 00:29:03 +00:00
}
impl<F, I, S, B> HttpServer<F, I, S, B>
where
F: Fn() -> I + Send + Clone + 'static,
I: IntoServiceFactory<S, Request>,
S: ServiceFactory<Request, Config = AppConfig>,
2019-04-05 23:46:44 +00:00
S::Error: Into<Error>,
S::InitError: fmt::Debug,
2019-03-05 00:29:03 +00:00
S::Response: Into<Response<B>>,
S::Service: 'static,
B: MessageBody,
{
/// Start listening for incoming connections.
///
2022-07-31 20:58:15 +00:00
/// # Workers
/// This method starts a number of HTTP workers in separate threads. The number of workers in a
/// set is defined by [`workers()`](Self::workers) or, by default, the number of the machine's
/// physical cores. One worker set is created for each socket address to be bound. For example,
/// if workers is set to 4, and there are 2 addresses to bind, then 8 worker threads will be
/// spawned.
2019-03-05 00:29:03 +00:00
///
2022-07-31 20:58:15 +00:00
/// # Panics
/// This methods panics if no socket addresses were successfully bound or if no Tokio runtime
/// is set up.
pub fn run(self) -> Server {
self.builder.run()
2019-03-05 00:29:03 +00:00
}
}
2023-03-11 22:17:52 +00:00
/// Bind TCP listeners to socket addresses resolved from `addrs` with options.
2023-07-17 01:38:12 +00:00
fn bind_addrs(addrs: impl net::ToSocketAddrs, backlog: u32) -> io::Result<Vec<net::TcpListener>> {
2023-03-11 22:17:52 +00:00
let mut err = None;
let mut success = false;
let mut sockets = Vec::new();
for addr in addrs.to_socket_addrs()? {
match create_tcp_listener(addr, backlog) {
Ok(lst) => {
success = true;
sockets.push(lst);
}
Err(e) => err = Some(e),
}
}
if success {
Ok(sockets)
} else if let Some(err) = err.take() {
Err(err)
} else {
Err(io::Error::new(
io::ErrorKind::Other,
"Can not bind to address.",
))
}
}
/// Creates a TCP listener from socket address and options.
2021-02-11 23:03:17 +00:00
fn create_tcp_listener(addr: net::SocketAddr, backlog: u32) -> io::Result<net::TcpListener> {
use socket2::{Domain, Protocol, Socket, Type};
2021-03-19 12:17:06 +00:00
let domain = Domain::for_address(addr);
let socket = Socket::new(domain, Type::STREAM, Some(Protocol::TCP))?;
socket.set_reuse_address(true)?;
socket.bind(&addr.into())?;
// clamp backlog to max u32 that fits in i32 range
2021-01-04 00:49:02 +00:00
let backlog = cmp::min(backlog, i32::MAX as u32) as i32;
socket.listen(backlog)?;
2021-03-19 12:17:06 +00:00
Ok(net::TcpListener::from(socket))
2019-03-05 00:29:03 +00:00
}
2023-03-11 22:17:52 +00:00
/// Configures OpenSSL acceptor `builder` with ALPN protocols.
2021-12-04 19:40:47 +00:00
#[cfg(feature = "openssl")]
2019-03-05 00:29:03 +00:00
fn openssl_acceptor(mut builder: SslAcceptorBuilder) -> io::Result<SslAcceptor> {
2020-09-09 08:20:54 +00:00
builder.set_alpn_select_callback(|_, protocols| {
2019-03-05 00:29:03 +00:00
const H2: &[u8] = b"\x02h2";
2019-12-02 11:33:11 +00:00
const H11: &[u8] = b"\x08http/1.1";
2020-09-09 08:20:54 +00:00
if protocols.windows(3).any(|window| window == H2) {
2019-03-05 00:29:03 +00:00
Ok(b"h2")
2020-09-09 08:20:54 +00:00
} else if protocols.windows(9).any(|window| window == H11) {
2019-12-02 11:33:11 +00:00
Ok(b"http/1.1")
2019-03-05 00:29:03 +00:00
} else {
Err(AlpnError::NOACK)
}
});
2020-09-09 08:20:54 +00:00
2019-03-05 00:29:03 +00:00
builder.set_alpn_protos(b"\x08http/1.1\x02h2")?;
Ok(builder.build())
}