h2_proxy_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_H2_PROXY_MOCK_SERVER_H_
#define PROTO_FUZZER_H2_PROXY_MOCK_SERVER_H_
 
#include "proto_fuzzer/tls_mock_server.h"
 
namespace proto_fuzzer {

enum class H2ProxyMode {
  kTunnel,
  kForward,
};

class H2ProxyMockServer final : public TlsMockServer, private IncomingDataObserver {
 public:
  explicit H2ProxyMockServer(H2ProxyMode mode = H2ProxyMode::kTunnel,
                             curl::fuzzer::proto::ProxyProtocol protocol = curl::fuzzer::proto::PROXY_PROTOCOL_HTTP2);
  ~H2ProxyMockServer() override = default;
 
  H2ProxyMockServer(const H2ProxyMockServer&) = delete;
  H2ProxyMockServer& operator=(const H2ProxyMockServer&) = delete;
 
  void Install(CURL* easy) override;
 
  const std::string& client_transcript() const;
 
 protected:
  std::unique_ptr<MockConnection> CreateConnection() override;
 
 private:
  void ObserveIncomingData(const unsigned char* data, std::size_t size) override;
 
  H2ProxyMode mode_;
  curl::fuzzer::proto::ProxyProtocol protocol_;
  std::string client_transcript_;
};
 
}  // namespace proto_fuzzer
 
#endif  // PROTO_FUZZER_H2_PROXY_MOCK_SERVER_H_