1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134
use std::fmt; use std; use http_parser::{HttpParser, HttpParserType}; use token::HttpToken; use parser_handler::{ParserHandler, ParserType}; #[derive(Clone, Debug, PartialEq, Eq)] pub struct ParserError { pub error: String, } impl fmt::Display for ParserError { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { write!(f, "{}", self.error) } } impl std::error::Error for ParserError { fn description(&self) -> &str { &self.error[..] } } pub struct Parser { parser: HttpParser, handler: ParserHandler, } impl Parser { pub fn request() -> Parser { Parser { parser: HttpParser::new(HttpParserType::Request), handler: ParserHandler::new(ParserType::Request), } } pub fn response() -> Parser { Parser { parser: HttpParser::new(HttpParserType::Response), handler: ParserHandler::new(ParserType::Response), } } pub fn next_token(&mut self, data: Option<&[u8]>) -> (Result<Option<HttpToken>, ParserError>, usize) { let mut nparsed = 0; if self.parser.errno.is_none() { if let Some(data) = data { if data.len() > 0 { nparsed = self.parser.execute(&mut self.handler, data); } } } if let Some(token) = self.handler.tokens.pop_front() { return (Ok(Some(token)), nparsed); } else { if let Some(ref e) = self.parser.errno { return (Err(ParserError { error: format!("{}", e) }), nparsed); } else { return (Ok(None), nparsed); } } } } #[cfg(test)] mod tests { use super::*; use token::HttpToken; #[test] fn simple() { let mut parser = Parser::request(); let mut tokens = Vec::new(); let mut http_request: Vec<_> = b"GET /te".iter() .cloned().collect(); loop { let (t, nparsed) = parser.next_token(Some(&http_request)); http_request.drain(..nparsed); if let Ok(Some(t)) = t { tokens.push(t); } else { break; } } http_request.extend(b"st HTTP/1.0\r\n\ Cont".iter().cloned()); loop { let (t, nparsed) = parser.next_token(Some(&http_request)); http_request.drain(..nparsed); if let Ok(Some(t)) = t { tokens.push(t); } else { break; } } http_request.extend(b"ent-Length: 00".iter().cloned()); loop { let (t, nparsed) = parser.next_token(Some(&http_request)); http_request.drain(..nparsed); if let Ok(Some(t)) = t { tokens.push(t); } else { break; } } http_request.extend(b"00\r\n\r\n".iter().cloned()); loop { let (t, nparsed) = parser.next_token(Some(&http_request)); http_request.drain(..nparsed); if let Ok(Some(t)) = t { tokens.push(t); } else { break; } } assert_eq!(tokens, [HttpToken::Method("GET".to_string()), HttpToken::Url("/test".to_string()), HttpToken::Field("Content-Length".to_string(), "0000".to_string()), HttpToken::EndOfMessage]); assert_eq!(parser.next_token(None), (Ok(None), 0)); } }