Function read_proxy_header

gateway/src/pp.rs:98–125

(mut stream: I)

}

async fn read_proxy_header<I>(mut stream: I) -> Result<(I, ProxyHeader)>
where
    I: AsyncRead + Unpin,
{
    let mut buffer = [0; READ_BUFFER_LEN];
    let mut dynamic_buffer = None;

    stream.read_exact(&mut buffer[..V1_PREFIX_LEN]).await?;

    if &buffer[..V1_PREFIX_LEN] == V1_PROTOCOL_PREFIX.as_bytes() {
        read_v1_header(&mut stream, &mut buffer).await?;
    } else {
        stream
            .read_exact(&mut buffer[V1_PREFIX_LEN..V2_MINIMUM_LEN])
            .await?;
        if &buffer[..V2_PREFIX_LEN] == V2_PROTOCOL_PREFIX {
            dynamic_buffer = read_v2_header(&mut stream, &mut buffer).await?;
        } else {
            bail!("no valid proxy protocol header detected");
        }
    }

    let mut buffer = dynamic_buffer.as_deref().unwrap_or(&buffer[..]);

    let header =
        proxy_protocol::parse(&mut buffer).context("failed to parse proxy protocol header")?;
    Ok((stream, header))
}

async fn read_v2_header<I>(
    mut stream: I,