Skip to content

Commit 68d4e4a

Browse files
noxnox
authored
feat(client): allow HTTP/0.9 responses behind a flag (#2473)
Fixes #2468
1 parent 51ed71b commit 68d4e4a

File tree

8 files changed

+158
-4
lines changed

8 files changed

+158
-4
lines changed

‎src/client/client.rs

+8
Original file line numberDiff line numberDiff line change
@@ -972,6 +972,14 @@ impl Builder {
972972
self
973973
}
974974

975+
/// Set whether HTTP/0.9 responses should be tolerated.
976+
///
977+
/// Default is false.
978+
pub fn http09_responses(&mut self, val: bool) -> &mut Self {
979+
self.conn_builder.h09_responses(val);
980+
self
981+
}
982+
975983
/// Set whether the connection **must** use HTTP/2.
976984
///
977985
/// The destination must either allow HTTP2 Prior Knowledge, or the

‎src/client/conn.rs

+10
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,7 @@ where
122122
#[derive(Clone, Debug)]
123123
pub struct Builder {
124124
pub(super) exec: Exec,
125+
h09_responses: bool,
125126
h1_title_case_headers: bool,
126127
h1_read_buf_exact_size: Option<usize>,
127128
h1_max_buf_size: Option<usize>,
@@ -493,6 +494,7 @@ impl Builder {
493494
pub fn new() -> Builder {
494495
Builder {
495496
exec: Exec::Default,
497+
h09_responses: false,
496498
h1_read_buf_exact_size: None,
497499
h1_title_case_headers: false,
498500
h1_max_buf_size: None,
@@ -514,6 +516,11 @@ impl Builder {
514516
self
515517
}
516518

519+
pub(super) fn h09_responses(&mut self, enabled: bool) -> &mut Builder {
520+
self.h09_responses = enabled;
521+
self
522+
}
523+
517524
pub(super) fn h1_title_case_headers(&mut self, enabled: bool) -> &mut Builder {
518525
self.h1_title_case_headers = enabled;
519526
self
@@ -700,6 +707,9 @@ impl Builder {
700707
if opts.h1_title_case_headers {
701708
conn.set_title_case_headers();
702709
}
710+
if opts.h09_responses {
711+
conn.set_h09_responses();
712+
}
703713
if let Some(sz) = opts.h1_read_buf_exact_size {
704714
conn.set_read_buf_exact_size(sz);
705715
}

‎src/lib.rs

-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
#![doc(html_root_url = "https://docs.rs/hyper/0.14.4")]
21
#![deny(missing_docs)]
32
#![deny(missing_debug_implementations)]
43
#![cfg_attr(test, deny(rust_2018_idioms))]

‎src/proto/h1/conn.rs

+11
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ where
4747
#[cfg(feature = "ffi")]
4848
preserve_header_case: false,
4949
title_case_headers: false,
50+
h09_responses: false,
5051
notify_read: false,
5152
reading: Reading::Init,
5253
writing: Writing::Init,
@@ -78,6 +79,11 @@ where
7879
self.state.title_case_headers = true;
7980
}
8081

82+
#[cfg(feature = "client")]
83+
pub(crate) fn set_h09_responses(&mut self) {
84+
self.state.h09_responses = true;
85+
}
86+
8187
#[cfg(feature = "server")]
8288
pub(crate) fn set_allow_half_close(&mut self) {
8389
self.state.allow_half_close = true;
@@ -146,6 +152,7 @@ where
146152
req_method: &mut self.state.method,
147153
#[cfg(feature = "ffi")]
148154
preserve_header_case: self.state.preserve_header_case,
155+
h09_responses: self.state.h09_responses,
149156
}
150157
)) {
151158
Ok(msg) => msg,
@@ -157,6 +164,9 @@ where
157164

158165
debug!("incoming body is {}", msg.decode);
159166

167+
// Prevent accepting HTTP/0.9 responses after the initial one, if any.
168+
self.state.h09_responses = false;
169+
160170
self.state.busy();
161171
self.state.keep_alive &= msg.keep_alive;
162172
self.state.version = msg.head.version;
@@ -753,6 +763,7 @@ struct State {
753763
#[cfg(feature = "ffi")]
754764
preserve_header_case: bool,
755765
title_case_headers: bool,
766+
h09_responses: bool,
756767
/// Set to true when the Dispatcher should poll read operations
757768
/// again. See the `maybe_notify` method for more.
758769
notify_read: bool,

‎src/proto/h1/io.rs

+2
Original file line numberDiff line numberDiff line change
@@ -161,6 +161,7 @@ where
161161
req_method: parse_ctx.req_method,
162162
#[cfg(feature = "ffi")]
163163
preserve_header_case: parse_ctx.preserve_header_case,
164+
h09_responses: parse_ctx.h09_responses,
164165
},
165166
)? {
166167
Some(msg) => {
@@ -640,6 +641,7 @@ mod tests {
640641
req_method: &mut None,
641642
#[cfg(feature = "ffi")]
642643
preserve_header_case: false,
644+
h09_responses: false,
643645
};
644646
assert!(buffered
645647
.parse::<ClientTransaction>(cx, parse_ctx)

‎src/proto/h1/mod.rs

+1
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ pub(crate) struct ParseContext<'a> {
7272
req_method: &'a mut Option<Method>,
7373
#[cfg(feature = "ffi")]
7474
preserve_header_case: bool,
75+
h09_responses: bool,
7576
}
7677

7778
/// Passed to Http1Transaction::encode

‎src/proto/h1/role.rs

+60-3
Original file line numberDiff line numberDiff line change
@@ -683,8 +683,8 @@ impl Http1Transaction for Client {
683683
);
684684
let mut res = httparse::Response::new(&mut headers);
685685
let bytes = buf.as_ref();
686-
match res.parse(bytes)? {
687-
httparse::Status::Complete(len) => {
686+
match res.parse(bytes) {
687+
Ok(httparse::Status::Complete(len)) => {
688688
trace!("Response.parse Complete({})", len);
689689
let status = StatusCode::from_u16(res.code.unwrap())?;
690690

@@ -710,7 +710,18 @@ impl Http1Transaction for Client {
710710
let headers_len = res.headers.len();
711711
(len, status, reason, version, headers_len)
712712
}
713-
httparse::Status::Partial => return Ok(None),
713+
Ok(httparse::Status::Partial) => return Ok(None),
714+
Err(httparse::Error::Version) if ctx.h09_responses => {
715+
trace!("Response.parse accepted HTTP/0.9 response");
716+
717+
#[cfg(not(feature = "ffi"))]
718+
let reason = ();
719+
#[cfg(feature = "ffi")]
720+
let reason = None;
721+
722+
(0, StatusCode::OK, reason, Version::HTTP_09, 0)
723+
}
724+
Err(e) => return Err(e.into()),
714725
}
715726
};
716727

@@ -1222,6 +1233,7 @@ mod tests {
12221233
req_method: &mut method,
12231234
#[cfg(feature = "ffi")]
12241235
preserve_header_case: false,
1236+
h09_responses: false,
12251237
},
12261238
)
12271239
.unwrap()
@@ -1244,6 +1256,7 @@ mod tests {
12441256
req_method: &mut Some(crate::Method::GET),
12451257
#[cfg(feature = "ffi")]
12461258
preserve_header_case: false,
1259+
h09_responses: false,
12471260
};
12481261
let msg = Client::parse(&mut raw, ctx).unwrap().unwrap();
12491262
assert_eq!(raw.len(), 0);
@@ -1261,10 +1274,46 @@ mod tests {
12611274
req_method: &mut None,
12621275
#[cfg(feature = "ffi")]
12631276
preserve_header_case: false,
1277+
h09_responses: false,
12641278
};
12651279
Server::parse(&mut raw, ctx).unwrap_err();
12661280
}
12671281

1282+
const H09_RESPONSE: &'static str = "Baguettes are super delicious, don't you agree?";
1283+
1284+
#[test]
1285+
fn test_parse_response_h09_allowed() {
1286+
let _ = pretty_env_logger::try_init();
1287+
let mut raw = BytesMut::from(H09_RESPONSE);
1288+
let ctx = ParseContext {
1289+
cached_headers: &mut None,
1290+
req_method: &mut Some(crate::Method::GET),
1291+
#[cfg(feature = "ffi")]
1292+
preserve_header_case: false,
1293+
h09_responses: true,
1294+
};
1295+
let msg = Client::parse(&mut raw, ctx).unwrap().unwrap();
1296+
assert_eq!(raw, H09_RESPONSE);
1297+
assert_eq!(msg.head.subject, crate::StatusCode::OK);
1298+
assert_eq!(msg.head.version, crate::Version::HTTP_09);
1299+
assert_eq!(msg.head.headers.len(), 0);
1300+
}
1301+
1302+
#[test]
1303+
fn test_parse_response_h09_rejected() {
1304+
let _ = pretty_env_logger::try_init();
1305+
let mut raw = BytesMut::from(H09_RESPONSE);
1306+
let ctx = ParseContext {
1307+
cached_headers: &mut None,
1308+
req_method: &mut Some(crate::Method::GET),
1309+
#[cfg(feature = "ffi")]
1310+
preserve_header_case: false,
1311+
h09_responses: false,
1312+
};
1313+
Client::parse(&mut raw, ctx).unwrap_err();
1314+
assert_eq!(raw, H09_RESPONSE);
1315+
}
1316+
12681317
#[test]
12691318
fn test_decoder_request() {
12701319
fn parse(s: &str) -> ParsedMessage<RequestLine> {
@@ -1276,6 +1325,7 @@ mod tests {
12761325
req_method: &mut None,
12771326
#[cfg(feature = "ffi")]
12781327
preserve_header_case: false,
1328+
h09_responses: false,
12791329
},
12801330
)
12811331
.expect("parse ok")
@@ -1291,6 +1341,7 @@ mod tests {
12911341
req_method: &mut None,
12921342
#[cfg(feature = "ffi")]
12931343
preserve_header_case: false,
1344+
h09_responses: false,
12941345
},
12951346
)
12961347
.expect_err(comment)
@@ -1505,6 +1556,7 @@ mod tests {
15051556
req_method: &mut Some(Method::GET),
15061557
#[cfg(feature = "ffi")]
15071558
preserve_header_case: false,
1559+
h09_responses: false,
15081560
}
15091561
)
15101562
.expect("parse ok")
@@ -1520,6 +1572,7 @@ mod tests {
15201572
req_method: &mut Some(m),
15211573
#[cfg(feature = "ffi")]
15221574
preserve_header_case: false,
1575+
h09_responses: false,
15231576
},
15241577
)
15251578
.expect("parse ok")
@@ -1535,6 +1588,7 @@ mod tests {
15351588
req_method: &mut Some(Method::GET),
15361589
#[cfg(feature = "ffi")]
15371590
preserve_header_case: false,
1591+
h09_responses: false,
15381592
},
15391593
)
15401594
.expect_err("parse should err")
@@ -1850,6 +1904,7 @@ mod tests {
18501904
req_method: &mut Some(Method::GET),
18511905
#[cfg(feature = "ffi")]
18521906
preserve_header_case: false,
1907+
h09_responses: false,
18531908
},
18541909
)
18551910
.expect("parse ok")
@@ -1931,6 +1986,7 @@ mod tests {
19311986
req_method: &mut None,
19321987
#[cfg(feature = "ffi")]
19331988
preserve_header_case: false,
1989+
h09_responses: false,
19341990
},
19351991
)
19361992
.unwrap()
@@ -1966,6 +2022,7 @@ mod tests {
19662022
req_method: &mut None,
19672023
#[cfg(feature = "ffi")]
19682024
preserve_header_case: false,
2025+
h09_responses: false,
19692026
},
19702027
)
19712028
.unwrap()

‎tests/client.rs

+66
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,43 @@ macro_rules! test {
112112
headers: { $($response_header_name:expr => $response_header_val:expr,)* },
113113
body: $response_body:expr,
114114
) => (
115+
test! {
116+
name: $name,
117+
server:
118+
expected: $server_expected,
119+
reply: $server_reply,
120+
client:
121+
set_host: $set_host,
122+
title_case_headers: $title_case_headers,
123+
allow_h09_responses: false,
124+
request: {$(
125+
$c_req_prop: $c_req_val,
126+
)*},
127+
128+
response:
129+
status: $client_status,
130+
headers: { $($response_header_name => $response_header_val,)* },
131+
body: $response_body,
132+
}
133+
);
134+
(
135+
name: $name:ident,
136+
server:
137+
expected: $server_expected:expr,
138+
reply: $server_reply:expr,
139+
client:
140+
set_host: $set_host:expr,
141+
title_case_headers: $title_case_headers:expr,
142+
allow_h09_responses: $allow_h09_responses:expr,
143+
request: {$(
144+
$c_req_prop:ident: $c_req_val:tt,
145+
)*},
146+
147+
response:
148+
status: $client_status:ident,
149+
headers: { $($response_header_name:expr => $response_header_val:expr,)* },
150+
body: $response_body:expr,
151+
) => (
115152
#[test]
116153
fn $name() {
117154
let _ = pretty_env_logger::try_init();
@@ -127,6 +164,7 @@ macro_rules! test {
127164
client:
128165
set_host: $set_host,
129166
title_case_headers: $title_case_headers,
167+
allow_h09_responses: $allow_h09_responses,
130168
request: {$(
131169
$c_req_prop: $c_req_val,
132170
)*},
@@ -181,6 +219,7 @@ macro_rules! test {
181219
client:
182220
set_host: true,
183221
title_case_headers: false,
222+
allow_h09_responses: false,
184223
request: {$(
185224
$c_req_prop: $c_req_val,
186225
)*},
@@ -205,6 +244,7 @@ macro_rules! test {
205244
client:
206245
set_host: $set_host:expr,
207246
title_case_headers: $title_case_headers:expr,
247+
allow_h09_responses: $allow_h09_responses:expr,
208248
request: {$(
209249
$c_req_prop:ident: $c_req_val:tt,
210250
)*},
@@ -217,6 +257,7 @@ macro_rules! test {
217257
let client = Client::builder()
218258
.set_host($set_host)
219259
.http1_title_case_headers($title_case_headers)
260+
.http09_responses($allow_h09_responses)
220261
.build(connector);
221262

222263
#[allow(unused_assignments, unused_mut)]
@@ -1067,6 +1108,31 @@ test! {
10671108
body: &b"abc"[..],
10681109
}
10691110

1111+
test! {
1112+
name: client_allows_http09_when_requested,
1113+
1114+
server:
1115+
expected: "\
1116+
GET / HTTP/1.1\r\n\
1117+
Host: {addr}\r\n\
1118+
\r\n\
1119+
",
1120+
reply: "Mmmmh, baguettes.",
1121+
1122+
client:
1123+
set_host: true,
1124+
title_case_headers: true,
1125+
allow_h09_responses: true,
1126+
request: {
1127+
method: GET,
1128+
url: "http://{addr}/",
1129+
},
1130+
response:
1131+
status: OK,
1132+
headers: {},
1133+
body: &b"Mmmmh, baguettes."[..],
1134+
}
1135+
10701136
mod dispatch_impl {
10711137
use super::*;
10721138
use std::io::{self, Read, Write};

0 commit comments

Comments
 (0)