HTTP server for testing Google APIs

chrome/browser/google_apis/test_server/http_request.cc

+// Copyright (c) 2012 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/google_apis/test_server/http_request.h"
+
+#include <algorithm>
+#include <map>
+#include <string>
+
+#include "base/basictypes.h"
+#include "base/logging.h"
+#include "base/string_number_conversions.h"
+#include "base/string_split.h"
+#include "base/string_util.h"
+#include "googleurl/src/gurl.h"
+
+
+namespace drive {
+namespace test_server {
+
+namespace {
+size_t kRequestSizeLimit = 64 * 1024 * 1024;  // 64 mb.
+
+// Helper function used to trim tokens in http request headers.
+std::string Trim(const std::string& value) {
+  std::string result;
+  TrimString(value, " \t", &result);
+  return result;
+}
+
+}  // namespace
+
+HttpRequest::HttpRequest() : method(UNKNOWN),
+                             has_content(false) {
+}
+
+HttpRequest::~HttpRequest() {
+}
+
+HttpRequestParser::HttpRequestParser()
+    : http_request_(new HttpRequest()),
+      buffer_position_(0),
+      state_(STATE_HEADERS),
+      declared_content_length_(0) {
+}
+
+HttpRequestParser::~HttpRequestParser() {
+}
+
+void HttpRequestParser::ProcessChunk(const base::StringPiece& data) {
+  data.AppendToString(&buffer_);
+  DCHECK_LE(buffer_.size() + data.size(), kRequestSizeLimit) <<
+      "The HTTP request is too large.";
+}
+
+std::string HttpRequestParser::ShiftLine() {
+  size_t eoln_position = buffer_.find("\r\n", buffer_position_);
+  DCHECK_NE(std::string::npos, eoln_position);
+  const int line_length = eoln_position - buffer_position_;
+  std::string result = buffer_.substr(buffer_position_, line_length);
+  buffer_position_ += line_length + 2;
+  return result;
+}
+
+HttpRequestParser::ParseResult HttpRequestParser::ParseRequest() {
+  DCHECK_NE(STATE_ACCEPTED, state_);
+  // Parse the request from beginning. However, entire request may not be
+  // available in the buffer.
+  if (state_ == STATE_HEADERS) {
+    if (ParseHeaders() == ACCEPTED)
+      return ACCEPTED;
+  }
+  // This should not be 'else if' of the previous block, as |state_| can be
+  // changed in ParseHeaders().
+  if (state_ == STATE_CONTENT) {
+    if (ParseContent() == ACCEPTED)
+      return ACCEPTED;
+  }
+  return WAITING;
+}
+
+HttpRequestParser::ParseResult HttpRequestParser::ParseHeaders() {
+  // Check if the all request headers are available.
+  if (buffer_.find("\r\n\r\n", buffer_position_) == std::string::npos)
+    return WAITING;
+
+  // Parse request's the first header line.
+  // Request main main header, eg. GET /foobar.html HTTP/1.1
+  {
+    const std::string header_line = ShiftLine();
+    std::vector<std::string> header_line_tokens;
+    base::SplitString(header_line, ' ', &header_line_tokens);
+    DCHECK_EQ(3u, header_line_tokens.size());
+    // Method.
+    http_request_->method = GetMethodType(StringToLowerASCII(
+        header_line_tokens[0]));
+    // Address.
+    const GURL host = GURL("http://localhost/");
+    http_request_->url = host.Resolve(header_line_tokens[1]);
+    // Protocol.
+    const std::string protocol = StringToLowerASCII(header_line_tokens[2]);
+    CHECK(protocol == "http/1.0" || protocol == "http/1.1") <<
+        "Protocol not supported: " << protocol;
+  }
+
+  // Parse further headers.
+  {
+    std::string header_name;
+    while (true) {
+      std::string header_line = ShiftLine();
+      if (header_line.empty())
+        break;
+
+      if (header_line[0] == ' ' || header_line[0] == '\t') {
+        // Continuation of the previous multi-line header.
+        std::string header_value =
+            Trim(header_line.substr(1, header_line.size() - 1));
+        http_request_->headers[header_name] += " " + header_value;
+      } else {
+        // New header.
+        size_t delimiter_pos = header_line.find(":");
+        DCHECK_NE(std::string::npos, delimiter_pos) << "Syntax error.";
+        header_name = Trim(header_line.substr(0, delimiter_pos));
+        std::string header_value = Trim(header_line.substr(
+            delimiter_pos + 1,
+            header_line.size() - delimiter_pos - 1));
+        http_request_->headers[header_name] = header_value;
+      }
+    }
+  }
+
+  // Headers done. Is any content data attached to the request?
+  declared_content_length_ = 0;
+  if (http_request_->headers.count("Content-Length") > 0) {
+    http_request_->has_content = true;
+    const bool success = base::StringToSizeT(
+        http_request_->headers["Content-Length"],
+        &declared_content_length_);
+    DCHECK(success) << "Malformed Content-Length header's value.";
+  }
+  if (declared_content_length_ == 0) {
+    // No content data, so parsing is finished.
+    state_ = STATE_ACCEPTED;
+    return ACCEPTED;
+  }
+
+  // The request has not yet been parsed yet, content data is still to be
+  // processed.
+  state_ = STATE_CONTENT;
+  return WAITING;
+}
+
+HttpRequestParser::ParseResult HttpRequestParser::ParseContent() {
+  const size_t available_bytes = buffer_.size() - buffer_position_;
+  const size_t fetch_bytes = std::min(
+      available_bytes,
+      declared_content_length_ - http_request_->content.size());
+  http_request_->content.append(buffer_.data() + buffer_position_,
+                                fetch_bytes);
+  buffer_position_ += fetch_bytes;
+
+  if (declared_content_length_ == http_request_->content.size()) {
+    state_ = STATE_ACCEPTED;
+    return ACCEPTED;
+  }
+
+  state_ = STATE_CONTENT;
+  return WAITING;
+}
+
+scoped_ptr<HttpRequest> HttpRequestParser::GetRequest() {
+  DCHECK_EQ(STATE_ACCEPTED, state_);
+  scoped_ptr<HttpRequest> result = http_request_.Pass();
+
+  // Prepare for parsing a new request.
+  state_ = STATE_HEADERS;
+  http_request_.reset(new HttpRequest());
+  buffer_.clear();
+  buffer_position_ = 0;
+  declared_content_length_ = 0;
+
+  return result.Pass();
+}
+
+HttpMethod HttpRequestParser::GetMethodType(const std::string& token) const {
+  if (token == "get") {
+    return GET;
+  } else if (token == "head") {
+    return HEAD;
+  } else if (token == "post") {
+    return POST;
+  } else if (token == "put") {
+    return PUT;
+  } else if (token == "delete") {
+    return DELETE;
+  }
+  NOTREACHED() << "Method not implemented: " << token;
+  return UNKNOWN;
+}
+
+}  // namespace test_server
+}  // namespace drive