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