summaryrefslogtreecommitdiffstats
path: root/third_party/rust/neqo-http3/tests/httpconn.rs
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
commit36d22d82aa202bb199967e9512281e9a53db42c9 (patch)
tree105e8c98ddea1c1e4784a60a5a6410fa416be2de /third_party/rust/neqo-http3/tests/httpconn.rs
parentInitial commit. (diff)
downloadfirefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz
firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip
Adding upstream version 115.7.0esr.upstream/115.7.0esrupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/rust/neqo-http3/tests/httpconn.rs')
-rw-r--r--third_party/rust/neqo-http3/tests/httpconn.rs460
1 files changed, 460 insertions, 0 deletions
diff --git a/third_party/rust/neqo-http3/tests/httpconn.rs b/third_party/rust/neqo-http3/tests/httpconn.rs
new file mode 100644
index 0000000000..c78b3f0be8
--- /dev/null
+++ b/third_party/rust/neqo-http3/tests/httpconn.rs
@@ -0,0 +1,460 @@
+// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
+// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
+// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
+// option. This file may not be copied, modified, or distributed
+// except according to those terms.
+
+#![allow(unused_assignments)]
+
+use neqo_common::{event::Provider, qtrace, Datagram};
+use neqo_crypto::{AuthenticationStatus, ResumptionToken};
+use neqo_http3::{
+ Header, Http3Client, Http3ClientEvent, Http3OrWebTransportStream, Http3Parameters, Http3Server,
+ Http3ServerEvent, Http3State, Priority,
+};
+use neqo_transport::{ConnectionError, ConnectionParameters, Error, Output, StreamType};
+use std::mem;
+use std::time::{Duration, Instant};
+use test_fixture::*;
+
+const RESPONSE_DATA: &[u8] = &[0x61, 0x62, 0x63];
+
+fn receive_request(server: &mut Http3Server) -> Option<Http3OrWebTransportStream> {
+ while let Some(event) = server.next_event() {
+ if let Http3ServerEvent::Headers {
+ stream,
+ headers,
+ fin,
+ } = event
+ {
+ assert_eq!(
+ &headers,
+ &[
+ Header::new(":method", "GET"),
+ Header::new(":scheme", "https"),
+ Header::new(":authority", "something.com"),
+ Header::new(":path", "/")
+ ]
+ );
+ assert!(fin);
+
+ return Some(stream);
+ }
+ }
+ None
+}
+
+fn set_response(request: &mut Http3OrWebTransportStream) {
+ request
+ .send_headers(&[
+ Header::new(":status", "200"),
+ Header::new("content-length", "3"),
+ ])
+ .unwrap();
+ request.send_data(RESPONSE_DATA).unwrap();
+ request.stream_close_send().unwrap();
+}
+
+fn process_server_events(server: &mut Http3Server) {
+ let mut request = receive_request(server).unwrap();
+ set_response(&mut request);
+}
+
+fn process_client_events(conn: &mut Http3Client) {
+ let mut response_header_found = false;
+ let mut response_data_found = false;
+ while let Some(event) = conn.next_event() {
+ match event {
+ Http3ClientEvent::HeaderReady { headers, fin, .. } => {
+ assert_eq!(
+ &headers,
+ &[
+ Header::new(":status", "200"),
+ Header::new("content-length", "3"),
+ ]
+ );
+ assert!(!fin);
+ response_header_found = true;
+ }
+ Http3ClientEvent::DataReadable { stream_id } => {
+ let mut buf = [0u8; 100];
+ let (amount, fin) = conn.read_data(now(), stream_id, &mut buf).unwrap();
+ assert!(fin);
+ assert_eq!(amount, RESPONSE_DATA.len());
+ assert_eq!(&buf[..RESPONSE_DATA.len()], RESPONSE_DATA);
+ response_data_found = true;
+ }
+ _ => {}
+ }
+ }
+ assert!(response_header_found);
+ assert!(response_data_found);
+}
+
+fn connect_peers(hconn_c: &mut Http3Client, hconn_s: &mut Http3Server) -> Option<Datagram> {
+ assert_eq!(hconn_c.state(), Http3State::Initializing);
+ let out = hconn_c.process(None, now()); // Initial
+ let out = hconn_s.process(out.dgram(), now()); // Initial + Handshake
+ let out = hconn_c.process(out.dgram(), now()); // ACK
+ mem::drop(hconn_s.process(out.dgram(), now())); //consume ACK
+ let authentication_needed = |e| matches!(e, Http3ClientEvent::AuthenticationNeeded);
+ assert!(hconn_c.events().any(authentication_needed));
+ hconn_c.authenticated(AuthenticationStatus::Ok, now());
+ let out = hconn_c.process(None, now()); // Handshake
+ assert_eq!(hconn_c.state(), Http3State::Connected);
+ let out = hconn_s.process(out.dgram(), now()); // Handshake
+ let out = hconn_c.process(out.dgram(), now());
+ let out = hconn_s.process(out.dgram(), now());
+ // assert!(hconn_s.settings_received);
+ let out = hconn_c.process(out.dgram(), now());
+ // assert!(hconn_c.settings_received);
+
+ out.dgram()
+}
+
+fn connect_peers_with_network_propagation_delay(
+ hconn_c: &mut Http3Client,
+ hconn_s: &mut Http3Server,
+ net_delay: u64,
+) -> (Option<Datagram>, Instant) {
+ let net_delay = Duration::from_millis(net_delay);
+ assert_eq!(hconn_c.state(), Http3State::Initializing);
+ let mut now = now();
+ let out = hconn_c.process(None, now); // Initial
+ now += net_delay;
+ let out = hconn_s.process(out.dgram(), now); // Initial + Handshake
+ now += net_delay;
+ let out = hconn_c.process(out.dgram(), now); // ACK
+ now += net_delay;
+ let out = hconn_s.process(out.dgram(), now); //consume ACK
+ assert!(out.dgram().is_none());
+ let authentication_needed = |e| matches!(e, Http3ClientEvent::AuthenticationNeeded);
+ assert!(hconn_c.events().any(authentication_needed));
+ now += net_delay;
+ hconn_c.authenticated(AuthenticationStatus::Ok, now);
+ let out = hconn_c.process(None, now); // Handshake
+ assert_eq!(hconn_c.state(), Http3State::Connected);
+ now += net_delay;
+ let out = hconn_s.process(out.dgram(), now); // HANDSHAKE_DONE
+ now += net_delay;
+ let out = hconn_c.process(out.dgram(), now); // Consume HANDSHAKE_DONE, send control streams.
+ now += net_delay;
+ let out = hconn_s.process(out.dgram(), now); // consume and send control streams.
+ now += net_delay;
+ let out = hconn_c.process(out.dgram(), now); // consume control streams.
+ (out.dgram(), now)
+}
+
+fn connect() -> (Http3Client, Http3Server, Option<Datagram>) {
+ let mut hconn_c = default_http3_client();
+ let mut hconn_s = default_http3_server();
+
+ let out = connect_peers(&mut hconn_c, &mut hconn_s);
+ (hconn_c, hconn_s, out)
+}
+
+fn exchange_packets(client: &mut Http3Client, server: &mut Http3Server, out_ex: Option<Datagram>) {
+ let mut out = out_ex;
+ loop {
+ out = client.process(out, now()).dgram();
+ out = server.process(out, now()).dgram();
+ if out.is_none() {
+ break;
+ }
+ }
+}
+
+#[test]
+fn test_connect() {
+ let (_hconn_c, _hconn_s, _d) = connect();
+}
+
+#[test]
+fn test_fetch() {
+ let (mut hconn_c, mut hconn_s, dgram) = connect();
+
+ qtrace!("-----client");
+ let req = hconn_c
+ .fetch(
+ now(),
+ "GET",
+ &("https", "something.com", "/"),
+ &[],
+ Priority::default(),
+ )
+ .unwrap();
+ assert_eq!(req, 0);
+ hconn_c.stream_close_send(req).unwrap();
+ let out = hconn_c.process(dgram, now());
+ qtrace!("-----server");
+ let out = hconn_s.process(out.dgram(), now());
+ mem::drop(hconn_c.process(out.dgram(), now()));
+ process_server_events(&mut hconn_s);
+ let out = hconn_s.process(None, now());
+
+ qtrace!("-----client");
+ mem::drop(hconn_c.process(out.dgram(), now()));
+ let out = hconn_s.process(None, now());
+ mem::drop(hconn_c.process(out.dgram(), now()));
+ process_client_events(&mut hconn_c);
+}
+
+#[test]
+fn test_103_response() {
+ let (mut hconn_c, mut hconn_s, dgram) = connect();
+
+ let req = hconn_c
+ .fetch(
+ now(),
+ "GET",
+ &("https", "something.com", "/"),
+ &[],
+ Priority::default(),
+ )
+ .unwrap();
+ assert_eq!(req, 0);
+ hconn_c.stream_close_send(req).unwrap();
+ let out = hconn_c.process(dgram, now());
+
+ let out = hconn_s.process(out.dgram(), now());
+ mem::drop(hconn_c.process(out.dgram(), now()));
+ let mut request = receive_request(&mut hconn_s).unwrap();
+
+ let info_headers = [
+ Header::new(":status", "103"),
+ Header::new("link", "</style.css>; rel=preload; as=style"),
+ ];
+ // Send 103
+ request.send_headers(&info_headers).unwrap();
+ let out = hconn_s.process(None, now());
+
+ mem::drop(hconn_c.process(out.dgram(), now()));
+
+ let info_headers_event = |e| {
+ matches!(e, Http3ClientEvent::HeaderReady { headers,
+ interim,
+ fin, .. } if !fin && interim && headers.as_ref() == info_headers)
+ };
+ assert!(hconn_c.events().any(info_headers_event));
+
+ set_response(&mut request);
+ let out = hconn_s.process(None, now());
+ mem::drop(hconn_c.process(out.dgram(), now()));
+ process_client_events(&mut hconn_c)
+}
+
+#[test]
+fn test_data_writable_events() {
+ const STREAM_LIMIT: u64 = 5000;
+ const DATA_AMOUNT: usize = 10000;
+
+ let mut hconn_c = http3_client_with_params(Http3Parameters::default().connection_parameters(
+ ConnectionParameters::default().max_stream_data(StreamType::BiDi, false, STREAM_LIMIT),
+ ));
+ let mut hconn_s = default_http3_server();
+
+ mem::drop(connect_peers(&mut hconn_c, &mut hconn_s));
+
+ // Create a request.
+ let req = hconn_c
+ .fetch(
+ now(),
+ "GET",
+ &("https", "something.com", "/"),
+ &[],
+ Priority::default(),
+ )
+ .unwrap();
+ hconn_c.stream_close_send(req).unwrap();
+ exchange_packets(&mut hconn_c, &mut hconn_s, None);
+
+ let mut request = receive_request(&mut hconn_s).unwrap();
+
+ request
+ .send_headers(&[
+ Header::new(":status", "200"),
+ Header::new("content-length", DATA_AMOUNT.to_string()),
+ ])
+ .unwrap();
+
+ // Send a lot of data
+ let buf = &[1; DATA_AMOUNT];
+ let mut sent = request.send_data(buf).unwrap();
+ assert!(sent < DATA_AMOUNT);
+
+ // Exchange packets and read the data on the client side.
+ exchange_packets(&mut hconn_c, &mut hconn_s, None);
+ let stream_id = request.stream_id();
+ let mut recv_buf = [0_u8; DATA_AMOUNT];
+ let (mut recvd, _) = hconn_c.read_data(now(), stream_id, &mut recv_buf).unwrap();
+ assert_eq!(sent, recvd);
+ exchange_packets(&mut hconn_c, &mut hconn_s, None);
+
+ let data_writable = |e| {
+ matches!(
+ e,
+ Http3ServerEvent::DataWritable {
+ stream
+ } if stream.stream_id() == stream_id
+ )
+ };
+ // Make sure we have a DataWritable event.
+ assert!(hconn_s.events().any(data_writable));
+ // Data can be sent again.
+ let s = request.send_data(&buf[sent..]).unwrap();
+ assert!(s > 0);
+ sent += s;
+
+ // Exchange packets and read the data on the client side.
+ exchange_packets(&mut hconn_c, &mut hconn_s, None);
+ let (r, _) = hconn_c
+ .read_data(now(), stream_id, &mut recv_buf[recvd..])
+ .unwrap();
+ recvd += r;
+ exchange_packets(&mut hconn_c, &mut hconn_s, None);
+ assert_eq!(sent, recvd);
+
+ // One more DataWritable event.
+ assert!(hconn_s.events().any(data_writable));
+ // Send more data.
+ let s = request.send_data(&buf[sent..]).unwrap();
+ assert!(s > 0);
+ sent += s;
+ assert_eq!(sent, DATA_AMOUNT);
+
+ exchange_packets(&mut hconn_c, &mut hconn_s, None);
+ let (r, _) = hconn_c
+ .read_data(now(), stream_id, &mut recv_buf[recvd..])
+ .unwrap();
+ recvd += r;
+
+ // Make sure all data is received by the client.
+ assert_eq!(recvd, DATA_AMOUNT);
+ assert_eq!(&recv_buf, buf);
+}
+
+fn get_token(client: &mut Http3Client) -> ResumptionToken {
+ assert_eq!(client.state(), Http3State::Connected);
+ client
+ .events()
+ .find_map(|e| {
+ if let Http3ClientEvent::ResumptionToken(token) = e {
+ Some(token)
+ } else {
+ None
+ }
+ })
+ .unwrap()
+}
+
+#[test]
+fn zerortt() {
+ let (mut hconn_c, _, dgram) = connect();
+ let token = get_token(&mut hconn_c);
+
+ // Create a new connection with a resumption token.
+ let mut hconn_c = default_http3_client();
+ hconn_c
+ .enable_resumption(now(), &token)
+ .expect("Set resumption token.");
+ let mut hconn_s = default_http3_server();
+
+ // Create a request.
+ let req = hconn_c
+ .fetch(
+ now(),
+ "GET",
+ &("https", "something.com", "/"),
+ &[],
+ Priority::default(),
+ )
+ .unwrap();
+ hconn_c.stream_close_send(req).unwrap();
+
+ let out = hconn_c.process(dgram, now());
+ let out = hconn_s.process(out.dgram(), now());
+
+ let mut request_stream = None;
+ let mut zerortt_state_change = false;
+ while let Some(event) = hconn_s.next_event() {
+ match event {
+ Http3ServerEvent::Headers {
+ stream,
+ headers,
+ fin,
+ } => {
+ assert_eq!(
+ &headers,
+ &[
+ Header::new(":method", "GET"),
+ Header::new(":scheme", "https"),
+ Header::new(":authority", "something.com"),
+ Header::new(":path", "/")
+ ]
+ );
+ assert!(fin);
+
+ request_stream = Some(stream);
+ }
+ Http3ServerEvent::StateChange { state, .. } => {
+ assert_eq!(state, Http3State::ZeroRtt);
+ zerortt_state_change = true;
+ }
+ _ => {}
+ }
+ }
+ assert!(zerortt_state_change);
+ let mut request_stream = request_stream.unwrap();
+
+ // Send a response
+ set_response(&mut request_stream);
+
+ // Receive the response
+ exchange_packets(&mut hconn_c, &mut hconn_s, out.dgram());
+ process_client_events(&mut hconn_c);
+}
+
+#[test]
+/// When a client has an outstanding fetch, it will send keepalives.
+/// Test that it will successfully run until the connection times out.
+fn fetch_noresponse_will_idletimeout() {
+ let mut hconn_c = default_http3_client();
+ let mut hconn_s = default_http3_server();
+
+ let (dgram, mut now) =
+ connect_peers_with_network_propagation_delay(&mut hconn_c, &mut hconn_s, 10);
+
+ qtrace!("-----client");
+ let req = hconn_c
+ .fetch(
+ now,
+ "GET",
+ &("https", "something.com", "/"),
+ &[],
+ Priority::default(),
+ )
+ .unwrap();
+ assert_eq!(req, 0);
+ hconn_c.stream_close_send(req).unwrap();
+ let _out = hconn_c.process(dgram, now);
+ qtrace!("-----server");
+
+ let mut done = false;
+ while !done {
+ while let Some(event) = hconn_c.next_event() {
+ if let Http3ClientEvent::StateChange(state) = event {
+ match state {
+ Http3State::Closing(error_code) | Http3State::Closed(error_code) => {
+ assert_eq!(error_code, ConnectionError::Transport(Error::IdleTimeout));
+ done = true;
+ }
+ _ => {}
+ }
+ }
+ }
+
+ if let Output::Callback(t) = hconn_c.process_output(now) {
+ now += t;
+ }
+ }
+}