You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
3.8 KiB
105 lines
3.8 KiB
4 months ago
|
// 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 LIBBRILLO_BRILLO_HTTP_HTTP_CONNECTION_CURL_H_
|
||
|
|
#define LIBBRILLO_BRILLO_HTTP_HTTP_CONNECTION_CURL_H_
|
||
|
|
|
||
|
|
#include <map>
|
||
|
|
#include <memory>
|
||
|
|
#include <string>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
#include <base/macros.h>
|
||
|
|
#include <brillo/brillo_export.h>
|
||
|
|
#include <brillo/http/http_connection.h>
|
||
|
|
#include <brillo/http/http_transport_curl.h>
|
||
|
|
#include <curl/curl.h>
|
||
|
|
|
||
|
|
namespace brillo {
|
||
|
|
namespace http {
|
||
|
|
namespace curl {
|
||
|
|
|
||
|
|
// This is a libcurl-based implementation of http::Connection.
|
||
|
|
class BRILLO_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, brillo::ErrorPtr* error) override;
|
||
|
|
bool SetRequestData(StreamPtr stream, brillo::ErrorPtr* error) override;
|
||
|
|
void SetResponseData(StreamPtr stream) override;
|
||
|
|
bool FinishRequest(brillo::ErrorPtr* error) override;
|
||
|
|
RequestID 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;
|
||
|
|
StreamPtr ExtractDataStream(brillo::ErrorPtr* error) override;
|
||
|
|
|
||
|
|
protected:
|
||
|
|
// Write data callback. Used by CURL when receiving response data.
|
||
|
|
BRILLO_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.
|
||
|
|
BRILLO_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.
|
||
|
|
BRILLO_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.
|
||
|
|
BRILLO_PRIVATE void PrepareRequest();
|
||
|
|
|
||
|
|
// HTTP request verb, such as "GET", "POST", "PUT", ...
|
||
|
|
std::string method_;
|
||
|
|
|
||
|
|
// Binary data for request body.
|
||
|
|
StreamPtr request_data_stream_;
|
||
|
|
|
||
|
|
// Received response data.
|
||
|
|
StreamPtr response_data_stream_;
|
||
|
|
|
||
|
|
// 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 brillo
|
||
|
|
|
||
|
|
#endif // LIBBRILLO_BRILLO_HTTP_HTTP_CONNECTION_CURL_H_
|