websocket_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_WEBSOCKET_MOCK_SERVER_H_
#define PROTO_FUZZER_WEBSOCKET_MOCK_SERVER_H_
 
#include <curl/curl.h>
 
#include <cstddef>
#include <memory>
#include <string>
#include <vector>
 
#include "curl_fuzzer.pb.h"
#include "proto_fuzzer/mock_server.h"
#include "proto_fuzzer/mock_server_base.h"
 
namespace proto_fuzzer {
 
bool ScenarioRequestsManualWsDrive(const curl::fuzzer::proto::Scenario& scenario);

class WebSocketMockServer : public MockServerBase, public IncomingDataObserver {
 public:
  WebSocketMockServer();
  ~WebSocketMockServer() override;
 
  void ObserveState() override;
 
  void Install(CURL* easy) override;
 
  void SetFrames(std::vector<std::string> frames);
 
  void SetManualDelivery(bool manual);
  bool manual_delivery() const;
 
  bool TryAdvanceHandshake(bool* made_progress = nullptr);
  bool handshake_sent() const;
 
  bool DeliverNextChunk();
  bool has_more_chunks() const;
 
  std::size_t remaining_chunks() const;
  const std::string& PeekChunk(std::size_t index) const;
  void ConsumeChunk();
  bool PushRawBytes(const unsigned char* data, std::size_t size);
 
 protected:
  virtual std::unique_ptr<MockConnection> CreateConnection();
 
  virtual std::size_t DrainHandshakeData();
 
  virtual void FlushTransport();
 
  void ObserveIncomingData(const unsigned char* data, std::size_t size) override;
 
  curl_socket_t HandleOpenSocket(curlsocktype purpose = CURLSOCKTYPE_IPCXN,
                                 struct curl_sockaddr* address = nullptr) override;
  void RunLoop(CURLM* multi, CURL* easy, const curl::fuzzer::proto::Scenario& scenario) override;
 
 public:
  bool ws_probe_fired() const;
  void MarkWsProbeFired();
  CURL* easy_handle() const;
  std::size_t manual_received_bytes() const;
 
 private:
  std::vector<std::string> frames_;
  std::size_t next_chunk_;
  bool manual_delivery_;
  bool handshake_sent_;
  bool ws_probe_fired_;
  CURL* easy_handle_;
  std::size_t manual_received_bytes_;
  std::string ws_request_buffer_;
};
 
}  // namespace proto_fuzzer
 
#endif  // PROTO_FUZZER_WEBSOCKET_MOCK_SERVER_H_