use std::{iter, sync::Arc};
use async_trait::async_trait;
use bytes::Bytes;
use http::{
HeaderValue, Response, StatusCode, Uri,
header::{self, LOCATION, REFRESH, STRICT_TRANSPORT_SECURITY, VIA},
uri::{Builder, Scheme},
};
use metrics::counter;
use pingora_core::{
ErrorType, OkOrErr, OrErr, apps::http_app::ServeHttp, prelude::HttpPeer,
protocols::http::ServerSession, upstreams::peer::Peer,
};
use pingora_http::{RequestHeader, ResponseHeader};
use pingora_proxy::{ProxyHttp, Session};
use tracing::{debug, info};
use crate::{
RunContext, certificates::{acme::AcmeRuntime, store::CertStore}, config::Backend, metrics::Metrics, proxy::{rewrite_port, router::Router, strip_port}
};
const REDIRECT_BODY: &[u8] = "<html><body>301 Moved Permanently</body></html>".as_bytes();
const TOKEN_NOT_FOUND: &[u8] = "<html><body>ACME token not found in request path</body></html>".as_bytes();
const ACME_HTTP01_PREFIX: &str = "/.well-known/acme-challenge/";
const YEAR_IN_SECS: u64 = 31536000;
fn token_not_found() -> Response<Vec<u8>> {
counter!("vicarian_acme_http01_notfound_total").increment(1);
Response::builder()
.status(StatusCode::NOT_FOUND)
.body(TOKEN_NOT_FOUND.to_vec())
.expect("Failed to send 404 response to token")
}
struct RequestComponents<'a> {
host: &'a str,
path: &'a str,
_query: &'a str,
}
fn to_components(session: &Session) -> pingora_core::Result<RequestComponents<'_>> {
let host = if session.is_http2() {
session.req_header().uri.host()
.or_err(ErrorType::InvalidHTTPHeader, "No Host component in request URI")?
} else {
let host_header = session.req_header().headers.get(header::HOST)
.or_err(ErrorType::InvalidHTTPHeader, "No Host header in request")?
.to_str()
.or_err(ErrorType::InvalidHTTPHeader, "Invalid Host header")?;
strip_port(host_header)
};
let pq = session.req_header().uri.path_and_query();
let (path, _query) = if let Some(pq) = pq {
(pq.path(), pq.query().unwrap_or(""))
} else {
("", "")
};
Ok(RequestComponents{
host, path, _query,
})
}
pub struct CleartextHandler {
acme: Arc<AcmeRuntime>,
port: String,
}
impl CleartextHandler {
pub fn new(acme: Arc<AcmeRuntime>, tls_port: u16) -> Self {
Self {
acme,
port: tls_port.to_string()
}
}
}
impl CleartextHandler {
async fn redirect_to_tls(&self, session: &mut ServerSession) -> Response<Vec<u8>> {
counter!("vicarian_http_redirects_total").increment(1);
let host = session.get_header(header::HOST)
.expect("Failed to get host header on HTTP service")
.to_str()
.expect("Failed to convert host header to str");
let path = session.req_header().uri.clone();
let new_host = rewrite_port(host, &self.port);
let location = Builder::from(path)
.scheme(Scheme::HTTPS)
.authority(new_host)
.build()
.expect("Failed to convert URI to HTTPS");
debug!("Redirect to {location}");
let body = REDIRECT_BODY.to_owned();
Response::builder()
.status(StatusCode::MOVED_PERMANENTLY)
.header(header::CONTENT_TYPE, "text/html")
.header(header::CONTENT_LENGTH, body.len())
.header(header::LOCATION, location.to_string())
.body(body)
.expect("Failed to create HTTP->HTTPS redirect response")
}
async fn acme_challenge(&self, session: &mut ServerSession) -> Response<Vec<u8>> {
counter!("vicarian_acme_http01_endpoint_total").increment(1);
let fqdn = session.get_header(header::HOST)
.expect("Failed to get host header on HTTP service")
.to_str()
.expect("Failed to convert host header to str");
let path = session.req_header().uri.path_and_query()
.expect("Failed to find already matched path?");
let path_token = match path.path().strip_prefix(ACME_HTTP01_PREFIX) {
Some(token) => token,
None => return token_not_found(),
};
let key_auth = if let Some(toks) = self.acme.challenge_tokens(fqdn)
&& toks.token == path_token
{
toks.key_auth
} else {
return token_not_found()
};
let body = key_auth.as_bytes().to_vec();
Response::builder()
.status(StatusCode::OK)
.header(header::CONTENT_TYPE, "text/plain")
.header(header::CONTENT_LENGTH, body.len())
.body(body)
.expect("Failed to create HTTP->HTTPS redirect response")
}
}
#[async_trait]
impl ServeHttp for CleartextHandler {
async fn response(&self, session: &mut ServerSession) -> Response<Vec<u8>> {
counter!("vicarian_http_requests_total").increment(1);
let path_p = session.req_header().uri.path_and_query();
if let Some(pq) = path_p
&& pq.path().starts_with(ACME_HTTP01_PREFIX)
{
info!("Received ACME challenge request: {pq}");
self.acme_challenge(session).await
} else {
self.redirect_to_tls(session).await
}
}
}
pub struct Vicarian {
_context: Arc<RunContext>,
_certstore: Arc<CertStore>,
routes_by_host: papaya::HashMap<String, Arc<Router>>,
}
impl Vicarian {
pub fn new(_certstore: Arc<CertStore>, context: Arc<RunContext>) -> Self {
let routes_by_host = context.config.vhosts.iter()
.flat_map(|vhost| {
let router = Arc::new(Router::new(&vhost.backends));
iter::once(&vhost.hostname)
.chain(vhost.aliases.iter())
.map(move |h| (h.clone(), router.clone()))
})
.collect::<papaya::HashMap<String, Arc<Router>>>();
Self {
_context: context,
_certstore,
routes_by_host,
}
}
async fn metrics_reply(&self, session: &mut Session) -> pingora_core::Result<()> {
counter!("vicarian_metrics_scrape_total").increment(1);
debug!("Replying to metrics endpoint");
let metrics = Metrics::get();
let scraped = metrics.handle.render();
let body = Bytes::copy_from_slice(scraped.as_bytes());
let mut header = ResponseHeader::build(200, Some(body.len()))?;
header.insert_header(header::CONTENT_TYPE, "text/plain")?;
session.write_response_header(Box::new(header), false).await?;
session.write_response_body(Some(body), true).await?;
Ok(())
}
}
const E404: pingora_core::ErrorType = ErrorType::HTTPStatus(StatusCode::NOT_FOUND.as_u16());
const E500: pingora_core::ErrorType = ErrorType::HTTPStatus(StatusCode::INTERNAL_SERVER_ERROR.as_u16());
#[derive(Clone)]
pub struct VicarianCtx {
backend: Arc<Backend>,
}
#[async_trait]
impl ProxyHttp for Vicarian {
type CTX = Option<VicarianCtx>;
fn new_ctx(&self) -> Self::CTX {
None
}
async fn request_filter(&self, session: &mut Session, ctx: &mut Self::CTX) -> pingora_core::Result<bool>
where
Self::CTX: Send + Sync,
{
debug!("Reques: {}", session.req_header().uri);
counter!("vicarian_tls_requests_total").increment(1);
let components = to_components(session)?;
let backend = {
let pinned = self.routes_by_host.pin();
let router = pinned.get(&components.host.to_string())
.or_err(E404, "Hostname not found in backends")?;
router.lookup(components.path)
.or_err(E404, "Path not found in host backends")?
.backend
};
let url = &backend.url;
let scheme = url.scheme_str()
.or_err(E500, "Failed to parse backed URL scheme")?;
match scheme {
"http" | "https" => {
*ctx = Some(VicarianCtx {
backend: backend.clone()
});
Ok(false)
}
"module" => {
let module = url.authority()
.or_err(E404, "Module name not found in host backends")?;
match module.as_str() {
"metrics" => {
self.metrics_reply(session).await?;
}
_ => {
let desc = format!("Unknown URL scheme: {scheme}");
return Err(pingora_core::Error::explain(E500, desc))
}
}
Ok(true)
}
&_ => {
let desc = format!("Unknown URL scheme: {scheme}");
Err(pingora_core::Error::explain(E500, desc))
}
}
}
async fn upstream_peer(&self, _session: &mut Session, ctx: &mut Self::CTX) -> pingora_core::Result<Box<HttpPeer>> {
let backend = ctx.clone()
.or_err(E500, "Request context not initialised; shouldn't happen?")?
.backend;
let url = &backend.url;
let host = url.host()
.or_err(E500, "Backend host lookup failed")?;
let port = url.port() .or_err(E500, "Backend port lookup failed")?
.as_u16();
let tls = url.scheme() == Some(&Scheme::HTTPS);
let mut peer = HttpPeer::new((host, port), tls, host.to_string());
if backend.trust && let Some(opts) = peer.get_mut_peer_options() {
opts.verify_cert = false;
}
debug!("Using peer: {peer:?}");
Ok(Box::new(peer))
}
async fn upstream_request_filter(&self, session: &mut Session,
upstream_request: &mut RequestHeader,
ctx: &mut Self::CTX,)
-> pingora_core::Result<()>
{
let backend = ctx.clone()
.or_err(E500, "Request context not initialised; shouldn't happen?")?
.backend;
if let Some(context) = &backend.context
&& ! context.is_empty() && context != "/"
&& ! backend.url.path().starts_with(context)
{
debug!("Modifying {} for context {context}", upstream_request.uri);
let upath = upstream_request.uri.path()
.strip_prefix(context)
.unwrap_or("/");
let uquery = upstream_request.uri.query()
.map(|s| format!("?{s}"))
.unwrap_or_default();
let upq = format!("{upath}{uquery}");
let uuri = Uri::builder()
.path_and_query(upq)
.build()
.or_err(E500, "Failed to rewrite path")?;
debug!("Modified to {uuri}");
upstream_request.set_uri(uuri);
}
if let Some(sockaddr) = session.client_addr()
&& let Some(inet) = sockaddr.as_inet()
{
let ip = inet.ip().to_string();
upstream_request.insert_header("X-Forwarded-For", &ip)?;
upstream_request.insert_header("X-Real-IP", &ip)?;
}
Ok(())
}
async fn upstream_response_filter(&self, _session: &mut Session,
upstream_response: &mut ResponseHeader,
ctx: &mut Self::CTX)
-> pingora_core::Result<()>
{
let backend = ctx.clone()
.or_err(E500, "Request context not initialised; shouldn't happen?")?
.backend;
if let Some(context) = &backend.context
&& ! context.is_empty() && context != "/"
&& ! backend.url.path().starts_with(context)
{
for headername in [LOCATION, REFRESH] {
let header_p = upstream_response.headers.get(&headername);
if let Some(header) = header_p {
let oldloc = header.to_str()
.or_err(E500, "Failed to rewrite location header")?;
let newloc = HeaderValue::from_str(&format!("{context}{oldloc}"))
.or_err(E500, "Failed to rewrite location header")?;
debug!("Modifying Location to {newloc:?}");
let _old = upstream_response.insert_header(&headername, newloc);
}
}
}
Ok(())
}
async fn response_filter(&self, session: &mut Session,
upstream_response: &mut ResponseHeader,
_ctx: &mut Self::CTX)
-> pingora_core::Result<()>
{
info!("CALLING RESPONSE FILTER");
let hsts = format!("max-age={YEAR_IN_SECS}; includeSubDomains");
upstream_response.insert_header(STRICT_TRANSPORT_SECURITY, hsts)?;
let via = format!("{:?} Vicarian", session.req_header().version);
upstream_response.insert_header(VIA, via)?;
Ok(())
}
}