pop3_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/pop3_mock_server.h"
 
#include <string>
#include <string_view>
 
#include "proto_fuzzer/allocation_fault_injection.h"
 
namespace proto_fuzzer {

Pop3MockServer::Pop3MockServer() = default;

Pop3MockServer::~Pop3MockServer() = default;

void Pop3MockServer::HandleLine(std::string_view line) {
  if (scenario() == nullptr) {
    return;
  }
  const std::string* response = NextControlReply();
  if (response == nullptr || !WriteControlBytes(*response)) {
    return;
  }
  if (!scenario()->pop3_starttls_filter_allocation_failure()) {
    return;
  }
  const bool starttls_accepted = TokenEquals(FirstToken(line), "STLS") && TokenEquals(FirstToken(*response), "+OK");
  if (starttls_accepted) {
    allocation_fault_injection::ArmNextAllocationFailure();
  }
}

void Pop3MockServer::QueueShutdownReply() { (void)WriteControlBytes(kShutdownReply); }

void Pop3MockServer::ObserveState() {
#if defined(PROTO_FUZZER_OBSERVE_STATE)
  PingpongMockServer::ObserveState();
  const volatile std::size_t pop3_observation[] = {control_transcript().size(), uploaded_data().size()};
  (void)pop3_observation;
#endif
}
 
}  // namespace proto_fuzzer