tftp_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_TFTP_MOCK_SERVER_H_
#define PROTO_FUZZER_TFTP_MOCK_SERVER_H_
 
#include <netinet/in.h>
 
#include <array>
#include <cstddef>
#include <cstdint>
#include <string>
#include <vector>
 
#include "curl_fuzzer.pb.h"
#include "proto_fuzzer/mock_server_base.h"
#include "proto_fuzzer/scenario_limits.h"
 
namespace proto_fuzzer {

enum class TftpSocketRole {
  kRequest,   
  kTransfer,  
};

struct TftpReceivedDatagram {
  TftpSocketRole received_on;  
  std::uint16_t source_port;   
  std::string bytes;           
};

class TftpMockServer final : public MockServerBase {
 public:
  TftpMockServer();
  ~TftpMockServer() override;
 
  void ObserveState() override;
 
  TftpMockServer(const TftpMockServer&) = delete;
  TftpMockServer& operator=(const TftpMockServer&) = delete;
 
  const std::vector<TftpReceivedDatagram>& received_datagrams() const;
 
  std::uint16_t request_port() const;
 
  std::uint16_t transfer_port() const;
 
 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:
  static constexpr std::size_t kMaxCapturedDatagrams = scenario_limits::kMaxResponseChunks + 2;
 
  void PrepareScript(const curl::fuzzer::proto::Connection& connection);
 
  void ResetPeer();
 
  static bool ConfigureSocket(int fd);
 
  static int OpenLoopbackSocket(struct sockaddr_in* bound_address);
 
  static bool RewriteDestination(struct curl_sockaddr* address, const struct sockaddr_in& destination);
 
  std::size_t DrainSocket(int fd, TftpSocketRole role);
 
  std::size_t DrainClientDatagrams();
 
  void RememberClientAddress(const struct sockaddr_in& address, socklen_t length);
 
  bool SendNextDatagram();
 
  std::array<const std::string*, scenario_limits::kMaxResponseChunks + 1> response_datagrams_;
  std::size_t response_datagram_count_;
  std::size_t next_response_datagram_;
 
  int request_fd_;
  int transfer_fd_;
  struct sockaddr_in client_address_;
  socklen_t client_address_length_;
  bool has_client_address_;
  bool socket_opened_;
  std::uint16_t request_port_;
  std::uint16_t transfer_port_;
  std::vector<TftpReceivedDatagram> received_datagrams_;
};
 
}  // namespace proto_fuzzer
 
#endif  // PROTO_FUZZER_TFTP_MOCK_SERVER_H_