libchromeos/chromeos/http/http_connection_curl.h

// Copyright 2014 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef LIBCHROMEOS_CHROMEOS_HTTP_HTTP_CONNECTION_CURL_H_
#define LIBCHROMEOS_CHROMEOS_HTTP_HTTP_CONNECTION_CURL_H_

#include <map>
#include <string>
#include <vector>

#include <base/macros.h>
#include <chromeos/chromeos_export.h>
#include <chromeos/http/http_connection.h>
#include <chromeos/http/http_transport_curl.h>
#include <curl/curl.h>

namespace chromeos {
namespace http {
namespace curl {

// This is a libcurl-based implementation of http::Connection.
class CHROMEOS_EXPORT Connection : public http::Connection {
 public:
  Connection(CURL* curl_handle,
             const std::string& method,
             const std::shared_ptr<CurlInterface>& curl_interface,
             const std::shared_ptr<http::Transport>& transport);
  ~Connection() override;

  // Overrides from http::Connection.
  // See http_connection.h for description of these methods.
  bool SendHeaders(const HeaderList& headers,
                   chromeos::ErrorPtr* error) override;
  bool SetRequestData(std::unique_ptr<DataReaderInterface> data_reader,
                      chromeos::ErrorPtr* error) override;
  bool FinishRequest(chromeos::ErrorPtr* error) override;
  int FinishRequestAsync(const SuccessCallback& success_callback,
                         const ErrorCallback& error_callback) override;

  int GetResponseStatusCode() const override;
  std::string GetResponseStatusText() const override;
  std::string GetProtocolVersion() const override;
  std::string GetResponseHeader(const std::string& header_name) const override;
  uint64_t GetResponseDataSize() const override;
  bool ReadResponseData(void* data,
                        size_t buffer_size,
                        size_t* size_read,
                        chromeos::ErrorPtr* error) override;

 protected:
  // Write data callback. Used by CURL when receiving response data.
  CHROMEOS_PRIVATE static size_t write_callback(char* ptr,
                                                size_t size,
                                                size_t num,
                                                void* data);
  // Read data callback. Used by CURL when sending request body data.
  CHROMEOS_PRIVATE static size_t read_callback(char* ptr,
                                               size_t size,
                                               size_t num,
                                               void* data);
  // Write header data callback. Used by CURL when receiving response headers.
  CHROMEOS_PRIVATE static size_t header_callback(char* ptr,
                                                 size_t size,
                                                 size_t num,
                                                 void* data);

  // Helper method to set up the |curl_handle_| with all the parameters
  // pertaining to the current connection.
  CHROMEOS_PRIVATE void PrepareRequest();

  // HTTP request verb, such as "GET", "POST", "PUT", ...
  std::string method_;

  // Binary data for request body.
  std::unique_ptr<DataReaderInterface> request_data_reader_;

  // Received response data.
  std::vector<unsigned char> response_data_;
  size_t response_data_ptr_{0};

  // List of optional request headers provided by the caller.
  // After request has been sent, contains the received response headers.
  std::multimap<std::string, std::string> headers_;

  // HTTP protocol version, such as HTTP/1.1
  std::string protocol_version_;
  // Response status text, such as "OK" for 200, or "Forbidden" for 403
  std::string status_text_;
  // Flag used when parsing response headers to separate the response status
  // from the rest of response headers.
  bool status_text_set_{false};

  CURL* curl_handle_{nullptr};
  curl_slist* header_list_{nullptr};

  std::shared_ptr<CurlInterface> curl_interface_;

 private:
  friend class http::curl::Transport;
  DISALLOW_COPY_AND_ASSIGN(Connection);
};

}  // namespace curl
}  // namespace http
}  // namespace chromeos

#endif  // LIBCHROMEOS_CHROMEOS_HTTP_HTTP_CONNECTION_CURL_H_