tls_websocket_mock_server.cc

Go to the documentation of this file.

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

 
#include "proto_fuzzer/tls_websocket_mock_server.h"
 
namespace proto_fuzzer {

TlsWebSocketMockServer::TlsWebSocketMockServer(curl::fuzzer::proto::TlsCertificateChainProfile certificate_chain)
    : transport_(TlsApplicationProtocol::kHttp11, certificate_chain) {}
 
TlsWebSocketMockServer::~TlsWebSocketMockServer() {
  // The connection borrows transport_'s server context. Release it before the
  // derived member is destroyed rather than relying on OpenSSL ref-counting.
  connection_.reset();
}

void TlsWebSocketMockServer::Install(CURL* easy) {
  WebSocketMockServer::Install(easy);
  transport_.Install(easy);
}

std::unique_ptr<MockConnection> TlsWebSocketMockServer::CreateConnection() { return transport_.CreateConnection(); }

std::size_t TlsWebSocketMockServer::DrainHandshakeData() {
  return connection_ == nullptr ? 0 : connection_->DrainIncoming();
}

void TlsWebSocketMockServer::FlushTransport() {
  if (connection_ != nullptr) {
    (void)connection_->DrainIncoming();
  }
}

std::size_t TlsWebSocketMockServer::completed_handshake_count() const { return transport_.completed_handshake_count(); }

void TlsWebSocketMockServer::ObserveState() {
#if defined(PROTO_FUZZER_OBSERVE_STATE)
  WebSocketMockServer::ObserveState();
  const volatile std::size_t tls_websocket_observation = completed_handshake_count();
  (void)tls_websocket_observation;
#endif
}
 
}  // namespace proto_fuzzer