53 lines
1.7 KiB
Rust
53 lines
1.7 KiB
Rust
use axum::body::Body;
|
|
use http::{Method, Request, Version};
|
|
use rust_proxy::proxy::websocket::{h2_protocol, validate_h2};
|
|
|
|
fn extended_connect(protocol: &'static str) -> Request<Body> {
|
|
let mut request = Request::builder()
|
|
.method(Method::CONNECT)
|
|
.version(Version::HTTP_2)
|
|
.uri("https://example.test/socket")
|
|
.header("sec-websocket-version", "13")
|
|
.body(Body::empty())
|
|
.unwrap();
|
|
request
|
|
.extensions_mut()
|
|
.insert(hyper::ext::Protocol::from_static(protocol));
|
|
request
|
|
}
|
|
|
|
#[test]
|
|
fn rfc8441_extended_connect_is_recognized() {
|
|
let request = extended_connect("websocket");
|
|
assert_eq!(h2_protocol(&request), Some("websocket"));
|
|
let handshake = validate_h2(&request).expect("valid RFC 8441 bootstrap");
|
|
assert!(handshake.h2());
|
|
assert!(handshake.upstream_key().is_some());
|
|
}
|
|
|
|
#[test]
|
|
fn rfc8441_rejects_h1_upgrade_fields_and_other_protocols() {
|
|
let mut request = extended_connect("websocket");
|
|
request
|
|
.headers_mut()
|
|
.insert("upgrade", "websocket".parse().unwrap());
|
|
assert!(validate_h2(&request).is_err());
|
|
|
|
let request = extended_connect("connect-udp");
|
|
assert!(validate_h2(&request).is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn rfc8441_vhost_flag_deserializes_explicitly() {
|
|
let value = serde_json::json!({
|
|
"id": "ws-h2",
|
|
"frontend": {"hosts": ["example.test"], "listener_ids": ["http"]},
|
|
"backends": [{"id": "origin", "url": "http://127.0.0.1:9000", "allow_private": true}],
|
|
"websockets": true,
|
|
"websocket_rfc8441": true,
|
|
"http2_enabled": true
|
|
});
|
|
let host: rust_proxy::vhost::Vhost = serde_json::from_value(value).unwrap();
|
|
assert!(host.websockets && host.websocket_rfc8441 && host.http2_enabled);
|
|
}
|