Add a test for rejecting forbidden headers

bytecodealliance · Oct 26, 2023 · fe05c4d · fe05c4d
1 parent 192b047
commit fe05c4d
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/crates/test-programs/src/bin/http_outbound_request_invalid_header.rs b/crates/test-programs/src/bin/http_outbound_request_invalid_header.rs
@@ -0,0 +1,12 @@
+use test_programs::wasi::http::types::{FieldsType, Headers};
+
+fn main() {
+ let hdrs = Headers::new(FieldsType::Request, &[]);
+ assert!(hdrs
+ .set(&"transfer-encoding".to_owned(), &[b"bad".to_vec()])
+ .is_err());
+
+ assert!(hdrs
+ .append(&"tRaNsFeR-eNcOdInG".to_owned(), &b"bad".to_vec())
+ .is_err());
+}
diff --git a/crates/wasi-http/tests/all/async_.rs b/crates/wasi-http/tests/all/async_.rs
@@ -49,6 +49,12 @@ async fn http_outbound_request_invalid_version() -> Result<()> {
  run(HTTP_OUTBOUND_REQUEST_INVALID_VERSION_COMPONENT, &server).await
 }
 
+#[test_log::test(tokio::test(flavor = "multi_thread"))]
+async fn http_outbound_request_invalid_header() -> Result<()> {
+ let server = Server::http2()?;
+ run(HTTP_OUTBOUND_REQUEST_INVALID_HEADER_COMPONENT, &server).await
+}
+
 #[test_log::test(tokio::test(flavor = "multi_thread"))]
 async fn http_outbound_request_unknown_method() -> Result<()> {
  let server = Server::http1()?;

diff --git a/crates/wasi-http/tests/all/sync.rs b/crates/wasi-http/tests/all/sync.rs
@@ -48,6 +48,12 @@ fn http_outbound_request_invalid_version() -> Result<()> {
  run(HTTP_OUTBOUND_REQUEST_INVALID_VERSION_COMPONENT, &server)
 }
 
+#[test_log::test]
+fn http_outbound_request_invalid_header() -> Result<()> {
+ let server = Server::http2()?;
+ run(HTTP_OUTBOUND_REQUEST_INVALID_HEADER_COMPONENT, &server)
+}
+
 #[test_log::test]
 fn http_outbound_request_unknown_method() -> Result<()> {
  let server = Server::http1()?;