// Copyright 2018 Google LLC. All Rights Reserved. This file and proprietary
// source code may only be used and distributed under the Widevine License
// Agreement.

#ifndef CDM_TEST_HTTP_SOCKET_H_
#define CDM_TEST_HTTP_SOCKET_H_

#include <stdlib.h>

#include <ctime>
#include <string>

#include <gtest/gtest_prod.h>
#include <openssl/ssl.h>

#include "disallow_copy_and_assign.h"

namespace wvcdm {

// Provides basic Linux based TCP socket interface.
class HttpSocket {
 public:
  // A scheme (http:// or https://) is required for the URL.
  explicit HttpSocket(const std::string& url);
  ~HttpSocket();

  bool ConnectAndLogErrors(int timeout_in_ms);
  void CloseSocket();

  const std::string& scheme() const { return scheme_; }
  bool secure_connect() const { return secure_connect_; }
  const std::string& domain_name() const { return domain_name_; }
  int port() const { return atoi(port_.c_str()); }
  const std::string& resource_path() const { return resource_path_; }
  const std::string& url() const { return url_; }

  int ReadAndLogErrors(char* data, int len, int timeout_in_ms);
  int WriteAndLogErrors(const char* data, int len, int timeout_in_ms);

 private:
  static bool ParseUrl(const std::string& url, std::string* scheme,
                       bool* secure_connect, std::string* domain_name,
                       std::string* port, std::string* path);
  // The following three functions do the work without logging errors.
  bool Connect(int timeout_in_ms);
  int Read(char* data, int len, int timeout_in_ms);
  int Write(const char* data, int len, int timeout_in_ms);
  // Log times with a note as an error.
  void LogTime(const char* note, const std::time_t& start,
               const std::time_t& finish);
  // Wait for a socket to be ready for reading or writing.
  // Establishing a connection counts as "ready for write".
  // Returns false on select error or timeout.
  // Returns true when the socket is ready.
  bool Wait(bool for_read, int timeout_in_ms);
  FRIEND_TEST(HttpSocketTest, ParseUrlTest);

  std::string url_;
  std::string scheme_;
  bool secure_connect_;
  std::string domain_name_;
  std::string port_;
  std::string resource_path_;
  bool valid_url_;

  int socket_fd_;
  SSL* ssl_;
  SSL_CTX* ssl_ctx_;

  // When the socket was created. Logged on error to help debug flaky
  // tests. e.g. b/186031735
  std::time_t create_time_;

  CORE_DISALLOW_COPY_AND_ASSIGN(HttpSocket);
};

}  // namespace wvcdm

#endif  // CDM_TEST_HTTP_SOCKET_H_