revert(server): "fix(server): Drain requests on drop."
This commit is contained in:
		
							
								
								
									
										32
									
								
								src/http.rs
									
									
									
									
									
								
							
							
						
						
									
										32
									
								
								src/http.rs
									
									
									
									
									
								
							| @@ -3,10 +3,7 @@ use std::borrow::Cow::{Borrowed, Owned}; | |||||||
| use std::borrow::IntoCow; | use std::borrow::IntoCow; | ||||||
| use std::cmp::min; | use std::cmp::min; | ||||||
| use std::old_io::{self, Reader, IoResult, BufWriter}; | use std::old_io::{self, Reader, IoResult, BufWriter}; | ||||||
| use std::old_io::util as io_util; |  | ||||||
| use std::mem; |  | ||||||
| use std::num::from_u16; | use std::num::from_u16; | ||||||
| use std::ptr; |  | ||||||
| use std::str; | use std::str; | ||||||
| use std::string::CowString; | use std::string::CowString; | ||||||
|  |  | ||||||
| @@ -23,7 +20,7 @@ use HttpError::{HttpHeaderError, HttpIoError, HttpMethodError, HttpStatusError, | |||||||
|                 HttpUriError, HttpVersionError}; |                 HttpUriError, HttpVersionError}; | ||||||
| use HttpResult; | use HttpResult; | ||||||
|  |  | ||||||
| use self::HttpReader::{SizedReader, ChunkedReader, EofReader}; | use self::HttpReader::{SizedReader, ChunkedReader, EofReader, EmptyReader}; | ||||||
| use self::HttpWriter::{ThroughWriter, ChunkedWriter, SizedWriter, EmptyWriter}; | use self::HttpWriter::{ThroughWriter, ChunkedWriter, SizedWriter, EmptyWriter}; | ||||||
|  |  | ||||||
| /// Readers to handle different Transfer-Encodings. | /// Readers to handle different Transfer-Encodings. | ||||||
| @@ -50,31 +47,23 @@ pub enum HttpReader<R> { | |||||||
|     /// > reliably; the server MUST respond with the 400 (Bad Request) |     /// > reliably; the server MUST respond with the 400 (Bad Request) | ||||||
|     /// > status code and then close the connection. |     /// > status code and then close the connection. | ||||||
|     EofReader(R), |     EofReader(R), | ||||||
|  |     /// A Reader used for messages that should never have a body. | ||||||
|  |     /// | ||||||
|  |     /// See https://tools.ietf.org/html/rfc7230#section-3.3.3 | ||||||
|  |     EmptyReader(R), | ||||||
| } | } | ||||||
|  |  | ||||||
| impl<R: Reader> HttpReader<R> { | impl<R: Reader> HttpReader<R> { | ||||||
|  |  | ||||||
|     /// Unwraps this HttpReader and returns the underlying Reader. |     /// Unwraps this HttpReader and returns the underlying Reader. | ||||||
|     #[inline] |  | ||||||
|     pub fn unwrap(self) -> R { |     pub fn unwrap(self) -> R { | ||||||
|         let r = unsafe { |         match self { | ||||||
|             ptr::read(match self { |             SizedReader(r, _) => r, | ||||||
|                 SizedReader(ref r, _) => r, |             ChunkedReader(r, _) => r, | ||||||
|                 ChunkedReader(ref r, _) => r, |             EofReader(r) => r, | ||||||
|                 EofReader(ref r) => r, |             EmptyReader(r) => r, | ||||||
|             }) |  | ||||||
|         }; |  | ||||||
|         unsafe { mem::forget(self); } |  | ||||||
|         r |  | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
|  |  | ||||||
| #[unsafe_destructor] |  | ||||||
| impl<R: Reader> Drop for HttpReader<R> { |  | ||||||
|     #[inline] |  | ||||||
|     fn drop(&mut self) { |  | ||||||
|         let _cant_use = io_util::copy(self, &mut io_util::NullWriter); |  | ||||||
|     } |  | ||||||
| } | } | ||||||
|  |  | ||||||
| impl<R: Reader> Reader for HttpReader<R> { | impl<R: Reader> Reader for HttpReader<R> { | ||||||
| @@ -127,6 +116,7 @@ impl<R: Reader> Reader for HttpReader<R> { | |||||||
|             EofReader(ref mut body) => { |             EofReader(ref mut body) => { | ||||||
|                 body.read(buf) |                 body.read(buf) | ||||||
|             }, |             }, | ||||||
|  |             EmptyReader(_) => Err(old_io::standard_error(old_io::EndOfFile)) | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,5 +1,5 @@ | |||||||
| #![feature(core, collections, hash, io, os, path, std_misc, | #![feature(core, collections, hash, io, os, path, std_misc, | ||||||
|            slicing_syntax, box_syntax, unsafe_destructor)] |            slicing_syntax, box_syntax)] | ||||||
| #![deny(missing_docs)] | #![deny(missing_docs)] | ||||||
| #![cfg_attr(test, deny(warnings))] | #![cfg_attr(test, deny(warnings))] | ||||||
| #![cfg_attr(test, feature(alloc, test))] | #![cfg_attr(test, feature(alloc, test))] | ||||||
|   | |||||||
| @@ -2,7 +2,7 @@ | |||||||
| //! | //! | ||||||
| //! These are requests that a `hyper::Server` receives, and include its method, | //! These are requests that a `hyper::Server` receives, and include its method, | ||||||
| //! target URI, headers, and message body. | //! target URI, headers, and message body. | ||||||
| use std::old_io::{self, IoResult}; | use std::old_io::IoResult; | ||||||
| use std::old_io::net::ip::SocketAddr; | use std::old_io::net::ip::SocketAddr; | ||||||
|  |  | ||||||
| use {HttpResult}; | use {HttpResult}; | ||||||
| @@ -11,7 +11,7 @@ use method::Method::{self, Get, Head}; | |||||||
| use header::{Headers, ContentLength, TransferEncoding}; | use header::{Headers, ContentLength, TransferEncoding}; | ||||||
| use http::{read_request_line}; | use http::{read_request_line}; | ||||||
| use http::HttpReader; | use http::HttpReader; | ||||||
| use http::HttpReader::{SizedReader, ChunkedReader}; | use http::HttpReader::{SizedReader, ChunkedReader, EmptyReader}; | ||||||
| use uri::RequestUri; | use uri::RequestUri; | ||||||
|  |  | ||||||
| /// A request bundles several parts of an incoming `NetworkStream`, given to a `Handler`. | /// A request bundles several parts of an incoming `NetworkStream`, given to a `Handler`. | ||||||
| @@ -26,7 +26,7 @@ pub struct Request<'a> { | |||||||
|     pub uri: RequestUri, |     pub uri: RequestUri, | ||||||
|     /// The version of HTTP for this request. |     /// The version of HTTP for this request. | ||||||
|     pub version: HttpVersion, |     pub version: HttpVersion, | ||||||
|     body: Body<HttpReader<&'a mut (Reader + 'a)>> |     body: HttpReader<&'a mut (Reader + 'a)> | ||||||
| } | } | ||||||
|  |  | ||||||
|  |  | ||||||
| @@ -39,19 +39,18 @@ impl<'a> Request<'a> { | |||||||
|         let headers = try!(Headers::from_raw(&mut stream)); |         let headers = try!(Headers::from_raw(&mut stream)); | ||||||
|         debug!("{:?}", headers); |         debug!("{:?}", headers); | ||||||
|  |  | ||||||
|         let body = if let Some(len) = headers.get::<ContentLength>() { |         let body = if method == Get || method == Head { | ||||||
|             SizedReader(stream, **len) |             EmptyReader(stream) | ||||||
|  |         } else if headers.has::<ContentLength>() { | ||||||
|  |             match headers.get::<ContentLength>() { | ||||||
|  |                 Some(&ContentLength(len)) => SizedReader(stream, len), | ||||||
|  |                 None => unreachable!() | ||||||
|  |             } | ||||||
|         } else if headers.has::<TransferEncoding>() { |         } else if headers.has::<TransferEncoding>() { | ||||||
|             todo!("check for Transfer-Encoding: chunked"); |             todo!("check for Transfer-Encoding: chunked"); | ||||||
|             ChunkedReader(stream, None) |             ChunkedReader(stream, None) | ||||||
|         } else { |         } else { | ||||||
|             SizedReader(stream, 0) |             EmptyReader(stream) | ||||||
|         }; |  | ||||||
|  |  | ||||||
|         let body = if method == Get || method == Head { |  | ||||||
|             Body::Empty(body) |  | ||||||
|         } else { |  | ||||||
|             Body::NonEmpty(body) |  | ||||||
|         }; |         }; | ||||||
|  |  | ||||||
|         Ok(Request { |         Ok(Request { | ||||||
| @@ -69,31 +68,13 @@ impl<'a> Request<'a> { | |||||||
|                                  RequestUri, HttpVersion, |                                  RequestUri, HttpVersion, | ||||||
|                                  HttpReader<&'a mut (Reader + 'a)>,) { |                                  HttpReader<&'a mut (Reader + 'a)>,) { | ||||||
|         (self.remote_addr, self.method, self.headers, |         (self.remote_addr, self.method, self.headers, | ||||||
|          self.uri, self.version, self.body.into_inner()) |          self.uri, self.version, self.body) | ||||||
|     } |     } | ||||||
| } | } | ||||||
|  |  | ||||||
| impl<'a> Reader for Request<'a> { | impl<'a> Reader for Request<'a> { | ||||||
|     #[inline] |  | ||||||
|     fn read(&mut self, buf: &mut [u8]) -> IoResult<usize> { |     fn read(&mut self, buf: &mut [u8]) -> IoResult<usize> { | ||||||
|         match self.body { |         self.body.read(buf) | ||||||
|             Body::Empty(..) => Err(old_io::standard_error(old_io::EndOfFile)), |  | ||||||
|             Body::NonEmpty(ref mut r) => r.read(buf) |  | ||||||
|         } |  | ||||||
|     } |  | ||||||
| } |  | ||||||
|  |  | ||||||
| enum Body<R> { |  | ||||||
|     Empty(R), |  | ||||||
|     NonEmpty(R), |  | ||||||
| } |  | ||||||
|  |  | ||||||
| impl<R> Body<R> { |  | ||||||
|     fn into_inner(self) -> R { |  | ||||||
|         match self { |  | ||||||
|             Body::Empty(r) => r, |  | ||||||
|             Body::NonEmpty(r) => r |  | ||||||
|         } |  | ||||||
|     } |     } | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -114,9 +95,8 @@ mod tests { | |||||||
|         let mut stream = MockStream::with_input(b"\ |         let mut stream = MockStream::with_input(b"\ | ||||||
|             GET / HTTP/1.1\r\n\ |             GET / HTTP/1.1\r\n\ | ||||||
|             Host: example.domain\r\n\ |             Host: example.domain\r\n\ | ||||||
|             Content-Length: 18\r\n\ |  | ||||||
|             \r\n\ |             \r\n\ | ||||||
|             I'm a bad request.\ |             I'm a bad request.\r\n\ | ||||||
|         "); |         "); | ||||||
|  |  | ||||||
|         let mut req = Request::new(&mut stream, sock("127.0.0.1:80")).unwrap(); |         let mut req = Request::new(&mut stream, sock("127.0.0.1:80")).unwrap(); | ||||||
| @@ -128,20 +108,6 @@ mod tests { | |||||||
|         let mut stream = MockStream::with_input(b"\ |         let mut stream = MockStream::with_input(b"\ | ||||||
|             HEAD / HTTP/1.1\r\n\ |             HEAD / HTTP/1.1\r\n\ | ||||||
|             Host: example.domain\r\n\ |             Host: example.domain\r\n\ | ||||||
|             Content-Length: 18\r\n\ |  | ||||||
|             \r\n\ |  | ||||||
|             I'm a bad request.\ |  | ||||||
|         "); |  | ||||||
|  |  | ||||||
|         let mut req = Request::new(&mut stream, sock("127.0.0.1:80")).unwrap(); |  | ||||||
|         assert_eq!(req.read_to_string(), Ok("".to_string())); |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     #[test] |  | ||||||
|     fn test_post_body_with_no_content_length() { |  | ||||||
|         let mut stream = MockStream::with_input(b"\ |  | ||||||
|             POST / HTTP/1.1\r\n\ |  | ||||||
|             Host: example.domain\r\n\ |  | ||||||
|             \r\n\ |             \r\n\ | ||||||
|             I'm a bad request.\r\n\ |             I'm a bad request.\r\n\ | ||||||
|         "); |         "); | ||||||
| @@ -151,17 +117,16 @@ mod tests { | |||||||
|     } |     } | ||||||
|  |  | ||||||
|     #[test] |     #[test] | ||||||
|     fn test_unexpected_body_drains_upon_drop() { |     fn test_post_empty_body() { | ||||||
|         let mut stream = MockStream::with_input(b"\ |         let mut stream = MockStream::with_input(b"\ | ||||||
|             GET / HTTP/1.1\r\n\ |             POST / HTTP/1.1\r\n\ | ||||||
|             Host: example.domain\r\n\ |             Host: example.domain\r\n\ | ||||||
|             Content-Length: 18\r\n\ |  | ||||||
|             \r\n\ |             \r\n\ | ||||||
|             I'm a bad request.\ |             I'm a bad request.\r\n\ | ||||||
|         "); |         "); | ||||||
|  |  | ||||||
|         Request::new(&mut stream, sock("127.0.0.1:80")).unwrap().read_to_string().unwrap(); |         let mut req = Request::new(&mut stream, sock("127.0.0.1:80")).unwrap(); | ||||||
|         assert!(stream.read.eof()); |         assert_eq!(req.read_to_string(), Ok("".to_string())); | ||||||
|     } |     } | ||||||
|  |  | ||||||
|     #[test] |     #[test] | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user