multi_socket_driver.h

Go to the documentation of this file.

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

 
#ifndef PROTO_FUZZER_MULTI_SOCKET_DRIVER_H_
#define PROTO_FUZZER_MULTI_SOCKET_DRIVER_H_
 
#include <curl/curl.h>
 
#include <array>
#include <cstddef>
#include <cstdint>
 
namespace proto_fuzzer {

class MultiSocketDriver {
 public:
  struct DriveResult {
    CURLMcode code = CURLM_OK;
    bool made_progress = false;
  };
 
  MultiSocketDriver();
  ~MultiSocketDriver();
 
  MultiSocketDriver(const MultiSocketDriver&) = delete;
  MultiSocketDriver& operator=(const MultiSocketDriver&) = delete;
 
  bool Install(CURLM* multi);
 
  CURLMcode Start(int* running_handles);
 
  DriveResult DriveReady(int* running_handles);
 
  void ProbeControlApis();

  bool watch_exhausted() const { return watch_exhausted_; }
 
 private:
  struct Watch {
    curl_socket_t socket = CURL_SOCKET_BAD;
    int interest = CURL_POLL_NONE;
    bool active = false;
  };
 
  static int SocketCallback(CURL* easy, curl_socket_t socket, int what, void* user_data, void* socket_data);
 
  static int TimerCallback(CURLM* multi, long timeout_ms, void* user_data);
 
  int UpdateSocket(curl_socket_t socket, int what, void* socket_data);
 
  int UpdateTimer(long timeout_ms);
 
  Watch* FindWatch(curl_socket_t socket);
 
  Watch* FindFreeWatch();

  static constexpr std::size_t kMaxWatches = 16;
 
  CURLM* multi_;
  std::array<Watch, kMaxWatches> watches_;
  long timeout_ms_;
  bool timer_pending_;
  bool watch_exhausted_;
  std::uint64_t generation_;
};
 
}  // namespace proto_fuzzer
 
#endif  // PROTO_FUZZER_MULTI_SOCKET_DRIVER_H_