| OLD | NEW |
| (Empty) |
| 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 #ifndef CHROME_BROWSER_GOOGLE_APIS_TEST_SERVER_HTTP_REQUEST_H_ | |
| 6 #define CHROME_BROWSER_GOOGLE_APIS_TEST_SERVER_HTTP_REQUEST_H_ | |
| 7 | |
| 8 #include <map> | |
| 9 #include <string> | |
| 10 | |
| 11 #include "base/basictypes.h" | |
| 12 #include "base/memory/scoped_ptr.h" | |
| 13 #include "base/strings/string_piece.h" | |
| 14 | |
| 15 namespace google_apis { | |
| 16 namespace test_server { | |
| 17 | |
| 18 // Methods of HTTP requests supported by the test HTTP server. | |
| 19 enum HttpMethod { | |
| 20 METHOD_UNKNOWN, | |
| 21 METHOD_GET, | |
| 22 METHOD_HEAD, | |
| 23 METHOD_POST, | |
| 24 METHOD_PUT, | |
| 25 METHOD_DELETE, | |
| 26 METHOD_PATCH, | |
| 27 }; | |
| 28 | |
| 29 // Represents a HTTP request. Since it can be big, use scoped_ptr to pass it | |
| 30 // instead of copying. However, the struct is copyable so tests can save and | |
| 31 // examine a HTTP request. | |
| 32 struct HttpRequest { | |
| 33 HttpRequest(); | |
| 34 ~HttpRequest(); | |
| 35 | |
| 36 std::string relative_url; // Starts with '/'. Example: "/test?query=foo" | |
| 37 HttpMethod method; | |
| 38 std::map<std::string, std::string> headers; | |
| 39 std::string content; | |
| 40 bool has_content; | |
| 41 }; | |
| 42 | |
| 43 // Parses the input data and produces a valid HttpRequest object. If there is | |
| 44 // more than one request in one chunk, then only the first one will be parsed. | |
| 45 // The common use is as below: | |
| 46 // HttpRequestParser parser; | |
| 47 // (...) | |
| 48 // void OnDataChunkReceived(Socket* socket, const char* data, int size) { | |
| 49 // parser.ProcessChunk(std::string(data, size)); | |
| 50 // if (parser.ParseRequest() == HttpRequestParser::ACCEPTED) { | |
| 51 // scoped_ptr<HttpRequest> request = parser.GetRequest(); | |
| 52 // (... process the request ...) | |
| 53 // } | |
| 54 class HttpRequestParser { | |
| 55 public: | |
| 56 // Parsing result. | |
| 57 enum ParseResult { | |
| 58 WAITING, // A request is not completed yet, waiting for more data. | |
| 59 ACCEPTED, // A request has been parsed and it is ready to be processed. | |
| 60 }; | |
| 61 | |
| 62 // Parser state. | |
| 63 enum State { | |
| 64 STATE_HEADERS, // Waiting for a request headers. | |
| 65 STATE_CONTENT, // Waiting for content data. | |
| 66 STATE_ACCEPTED, // Request has been parsed. | |
| 67 }; | |
| 68 | |
| 69 HttpRequestParser(); | |
| 70 ~HttpRequestParser(); | |
| 71 | |
| 72 // Adds chunk of data into the internal buffer. | |
| 73 void ProcessChunk(const base::StringPiece& data); | |
| 74 | |
| 75 // Parses the http request (including data - if provided). | |
| 76 // If returns ACCEPTED, then it means that the whole request has been found | |
| 77 // in the internal buffer (and parsed). After calling GetRequest(), it will be | |
| 78 // ready to parse another request. | |
| 79 ParseResult ParseRequest(); | |
| 80 | |
| 81 // Retrieves parsed request. Can be only called, when the parser is in | |
| 82 // STATE_ACCEPTED state. After calling it, the parser is ready to parse | |
| 83 // another request. | |
| 84 scoped_ptr<HttpRequest> GetRequest(); | |
| 85 | |
| 86 private: | |
| 87 HttpMethod GetMethodType(const std::string& token) const; | |
| 88 | |
| 89 // Parses headers and returns ACCEPTED if whole request was parsed. Otherwise | |
| 90 // returns WAITING. | |
| 91 ParseResult ParseHeaders(); | |
| 92 | |
| 93 // Parses request's content data and returns ACCEPTED if all of it have been | |
| 94 // processed. Chunked Transfer Encoding *is not* supported. | |
| 95 ParseResult ParseContent(); | |
| 96 | |
| 97 // Fetches the next line from the buffer. Result does not contain \r\n. | |
| 98 // Returns an empty string for an empty line. It will assert if there is | |
| 99 // no line available. | |
| 100 std::string ShiftLine(); | |
| 101 | |
| 102 scoped_ptr<HttpRequest> http_request_; | |
| 103 std::string buffer_; | |
| 104 size_t buffer_position_; // Current position in the internal buffer. | |
| 105 State state_; | |
| 106 // Content length of the request currently being parsed. | |
| 107 size_t declared_content_length_; | |
| 108 | |
| 109 DISALLOW_COPY_AND_ASSIGN(HttpRequestParser); | |
| 110 }; | |
| 111 | |
| 112 } // namespace test_server | |
| 113 } // namespace google_apis | |
| 114 | |
| 115 #endif // CHROME_BROWSER_GOOGLE_APIS_TEST_SERVER_HTTP_REQUEST_H_ | |
| OLD | NEW |