1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "net/test/embedded_test_server/http_request.h" 6 7 #include <algorithm> 8 9 #include "base/logging.h" 10 #include "base/strings/string_util.h" 11 #include "base/strings/string_number_conversions.h" 12 #include "base/strings/string_split.h" 13 14 namespace net { 15 namespace test_server { 16 17 namespace { 18 19 size_t kRequestSizeLimit = 64 * 1024 * 1024; // 64 mb. 20 21 // Helper function used to trim tokens in http request headers. 22 std::string Trim(const std::string& value) { 23 std::string result; 24 base::TrimString(value, " \t", &result); 25 return result; 26 } 27 28 } // namespace 29 30 HttpRequest::HttpRequest() : method(METHOD_UNKNOWN), 31 has_content(false) { 32 } 33 34 HttpRequest::~HttpRequest() { 35 } 36 37 HttpRequestParser::HttpRequestParser() 38 : http_request_(new HttpRequest()), 39 buffer_position_(0), 40 state_(STATE_HEADERS), 41 declared_content_length_(0) { 42 } 43 44 HttpRequestParser::~HttpRequestParser() { 45 } 46 47 void HttpRequestParser::ProcessChunk(const base::StringPiece& data) { 48 data.AppendToString(&buffer_); 49 DCHECK_LE(buffer_.size() + data.size(), kRequestSizeLimit) << 50 "The HTTP request is too large."; 51 } 52 53 std::string HttpRequestParser::ShiftLine() { 54 size_t eoln_position = buffer_.find("\r\n", buffer_position_); 55 DCHECK_NE(std::string::npos, eoln_position); 56 const int line_length = eoln_position - buffer_position_; 57 std::string result = buffer_.substr(buffer_position_, line_length); 58 buffer_position_ += line_length + 2; 59 return result; 60 } 61 62 HttpRequestParser::ParseResult HttpRequestParser::ParseRequest() { 63 DCHECK_NE(STATE_ACCEPTED, state_); 64 // Parse the request from beginning. However, entire request may not be 65 // available in the buffer. 66 if (state_ == STATE_HEADERS) { 67 if (ParseHeaders() == ACCEPTED) 68 return ACCEPTED; 69 } 70 // This should not be 'else if' of the previous block, as |state_| can be 71 // changed in ParseHeaders(). 72 if (state_ == STATE_CONTENT) { 73 if (ParseContent() == ACCEPTED) 74 return ACCEPTED; 75 } 76 return WAITING; 77 } 78 79 HttpRequestParser::ParseResult HttpRequestParser::ParseHeaders() { 80 // Check if the all request headers are available. 81 if (buffer_.find("\r\n\r\n", buffer_position_) == std::string::npos) 82 return WAITING; 83 84 // Parse request's the first header line. 85 // Request main main header, eg. GET /foobar.html HTTP/1.1 86 { 87 const std::string header_line = ShiftLine(); 88 std::vector<std::string> header_line_tokens; 89 base::SplitString(header_line, ' ', &header_line_tokens); 90 DCHECK_EQ(3u, header_line_tokens.size()); 91 // Method. 92 http_request_->method = GetMethodType(StringToLowerASCII( 93 header_line_tokens[0])); 94 // Address. 95 // Don't build an absolute URL as the parser does not know (should not 96 // know) anything about the server address. 97 http_request_->relative_url = header_line_tokens[1]; 98 // Protocol. 99 const std::string protocol = StringToLowerASCII(header_line_tokens[2]); 100 CHECK(protocol == "http/1.0" || protocol == "http/1.1") << 101 "Protocol not supported: " << protocol; 102 } 103 104 // Parse further headers. 105 { 106 std::string header_name; 107 while (true) { 108 std::string header_line = ShiftLine(); 109 if (header_line.empty()) 110 break; 111 112 if (header_line[0] == ' ' || header_line[0] == '\t') { 113 // Continuation of the previous multi-line header. 114 std::string header_value = 115 Trim(header_line.substr(1, header_line.size() - 1)); 116 http_request_->headers[header_name] += " " + header_value; 117 } else { 118 // New header. 119 size_t delimiter_pos = header_line.find(":"); 120 DCHECK_NE(std::string::npos, delimiter_pos) << "Syntax error."; 121 header_name = Trim(header_line.substr(0, delimiter_pos)); 122 std::string header_value = Trim(header_line.substr( 123 delimiter_pos + 1, 124 header_line.size() - delimiter_pos - 1)); 125 http_request_->headers[header_name] = header_value; 126 } 127 } 128 } 129 130 // Headers done. Is any content data attached to the request? 131 declared_content_length_ = 0; 132 if (http_request_->headers.count("Content-Length") > 0) { 133 http_request_->has_content = true; 134 const bool success = base::StringToSizeT( 135 http_request_->headers["Content-Length"], 136 &declared_content_length_); 137 DCHECK(success) << "Malformed Content-Length header's value."; 138 } 139 if (declared_content_length_ == 0) { 140 // No content data, so parsing is finished. 141 state_ = STATE_ACCEPTED; 142 return ACCEPTED; 143 } 144 145 // The request has not yet been parsed yet, content data is still to be 146 // processed. 147 state_ = STATE_CONTENT; 148 return WAITING; 149 } 150 151 HttpRequestParser::ParseResult HttpRequestParser::ParseContent() { 152 const size_t available_bytes = buffer_.size() - buffer_position_; 153 const size_t fetch_bytes = std::min( 154 available_bytes, 155 declared_content_length_ - http_request_->content.size()); 156 http_request_->content.append(buffer_.data() + buffer_position_, 157 fetch_bytes); 158 buffer_position_ += fetch_bytes; 159 160 if (declared_content_length_ == http_request_->content.size()) { 161 state_ = STATE_ACCEPTED; 162 return ACCEPTED; 163 } 164 165 state_ = STATE_CONTENT; 166 return WAITING; 167 } 168 169 scoped_ptr<HttpRequest> HttpRequestParser::GetRequest() { 170 DCHECK_EQ(STATE_ACCEPTED, state_); 171 scoped_ptr<HttpRequest> result = http_request_.Pass(); 172 173 // Prepare for parsing a new request. 174 state_ = STATE_HEADERS; 175 http_request_.reset(new HttpRequest()); 176 buffer_.clear(); 177 buffer_position_ = 0; 178 declared_content_length_ = 0; 179 180 return result.Pass(); 181 } 182 183 HttpMethod HttpRequestParser::GetMethodType(const std::string& token) const { 184 if (token == "get") { 185 return METHOD_GET; 186 } else if (token == "head") { 187 return METHOD_HEAD; 188 } else if (token == "post") { 189 return METHOD_POST; 190 } else if (token == "put") { 191 return METHOD_PUT; 192 } else if (token == "delete") { 193 return METHOD_DELETE; 194 } else if (token == "patch") { 195 return METHOD_PATCH; 196 } 197 NOTREACHED() << "Method not implemented: " << token; 198 return METHOD_UNKNOWN; 199 } 200 201 } // namespace test_server 202 } // namespace net 203