http_file.h

// Copyright 2020 Google LLC. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd
 
#ifndef PACKAGER_FILE_HTTP_H_
#define PACKAGER_FILE_HTTP_H_
 
#include <cstdint>
#include <memory>
#include <string>
#include <vector>
 
#include <absl/synchronization/notification.h>
 
#include <packager/file.h>
#include <packager/file/io_cache.h>
#include <packager/status.h>
 
typedef void CURL;
struct curl_slist;
 
namespace shaka {
 
enum class HttpMethod {
  kGet,
  kPost,
  kPut,
  kDelete,
};
 
class HttpFile : public File {
 public:
  HttpFile(HttpMethod method, const std::string& url);
  HttpFile(HttpMethod method,
           const std::string& url,
           const std::string& upload_content_type,
           const std::vector<std::string>& headers,
           int32_t timeout_in_seconds);
 
  HttpFile(const HttpFile&) = delete;
  HttpFile& operator=(const HttpFile&) = delete;
 
  static bool Delete(const std::string& url);
 
  Status CloseWithStatus();
 
  bool Close() override;
  int64_t Read(void* buffer, uint64_t length) override;
  int64_t Write(const void* buffer, uint64_t length) override;
  void CloseForWriting() override;
  int64_t Size() override;
  bool Flush() override;
  bool Seek(uint64_t position) override;
  bool Tell(uint64_t* position) override;
  bool Open() override;
 
 protected:
  ~HttpFile() override;
 
 private:
  struct CurlDelete {
    void operator()(CURL* curl);
    void operator()(curl_slist* headers);
  };
 
  void SetupRequest();
  void ThreadMain();
 
  const std::string url_;
  const std::string upload_content_type_;
  const int32_t timeout_in_seconds_;
  const HttpMethod method_;
  const bool isUpload_;
  IoCache download_cache_;
  IoCache upload_cache_;
  std::unique_ptr<CURL, CurlDelete> curl_;
  // The headers need to remain alive for the duration of the request.
  std::unique_ptr<curl_slist, CurlDelete> request_headers_;
  Status status_;
  std::string user_agent_;
  std::string ca_file_;
  std::string client_cert_file_;
  std::string client_cert_private_key_file_;
  std::string client_cert_private_key_password_;
 
  // Signaled when the "curl easy perform" task completes.
  absl::Notification task_exit_event_;
};
 
}  // namespace shaka
 
#endif  // PACKAGER_FILE_HTTP_H_