happy_eyeballs_mock_server.h

Go to the documentation of this file.

/*
 * Copyright (C) Max Dymond, <cmeister2@gmail.com>, et al.
 *
 * SPDX-License-Identifier: curl
 */

 
#ifndef PROTO_FUZZER_HAPPY_EYEBALLS_MOCK_SERVER_H_
#define PROTO_FUZZER_HAPPY_EYEBALLS_MOCK_SERVER_H_
 
#include <curl/curl.h>
 
#include <cstddef>
#include <string>
#include <vector>
 
#include "curl_fuzzer.pb.h"
#include "proto_fuzzer/curl_raii.h"
#include "proto_fuzzer/mock_server_base.h"
 
namespace proto_fuzzer {

class HappyEyeballsMockServer : public MockServerBase {
 public:
  HappyEyeballsMockServer();
  ~HappyEyeballsMockServer() override;
 
  HappyEyeballsMockServer(const HappyEyeballsMockServer&) = delete;
  HappyEyeballsMockServer& operator=(const HappyEyeballsMockServer&) = delete;
 
  void Install(CURL* easy) override;
 
  void ConfigureRequestData(ScenarioRequestData* request_data) override;

  std::size_t opened_socket_count() const { return opened_socket_count_; }

  std::size_t accepted_peer_count() const { return accepted_peer_count_; }

  int listener_port() const { return listener_port_; }
 
 protected:
  curl_socket_t HandleOpenSocket(curlsocktype purpose, struct curl_sockaddr* address) override;
 
  SocketSetupDisposition GetSocketSetupDisposition(curl_socket_t curlfd, curlsocktype purpose) const override;
 
  void RunLoop(CURLM* multi, CURL* easy, const curl::fuzzer::proto::Scenario& scenario) override;
 
 private:
  struct Peer {
    int fd = -1;
    const curl::fuzzer::proto::Connection* script = nullptr;
    std::size_t next_chunk = 0;
    std::string pending;
    std::size_t pending_offset = 0;
    bool sent_final = false;
  };
 
  bool OpenListeners();
 
  void CloseListeners();
 
  bool AcceptPending();
 
  bool ServicePeers();
 
  void ClosePeer(Peer* peer);
 
  const curl::fuzzer::proto::Connection* ScriptForAccept(std::size_t index) const;
 
  bool IsOwnedDestination(const struct curl_sockaddr* address) const;
 
  CURL* easy_;

  int race_fd_;
  int serve_fd_;
  int listener_port_;
  std::string serve_address_;
  std::vector<int> dummy_fds_;
 
  std::size_t opened_socket_count_;
  std::size_t accepted_peer_count_;

  bool setup_ok_ = false;

  const curl::fuzzer::proto::Scenario* scenario_;
 
  std::vector<Peer> peers_;

  CurlSlistPtr resolve_list_;
};
 
}  // namespace proto_fuzzer
 
#endif  // PROTO_FUZZER_HAPPY_EYEBALLS_MOCK_SERVER_H_