smtp_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_SMTP_MOCK_SERVER_H_
#define PROTO_FUZZER_SMTP_MOCK_SERVER_H_
 
#include <cstddef>
#include <memory>
#include <string_view>
 
#include "proto_fuzzer/pingpong_mock_server.h"
 
namespace proto_fuzzer {

class SmtpMockServer : public PingpongMockServer {
 public:
  explicit SmtpMockServer(curl::fuzzer::proto::SmtpProbeMode mode = curl::fuzzer::proto::SMTP_PROBE_DEFAULT);
 
  ~SmtpMockServer() override;
 
  void ObserveState() override;
 
  SmtpMockServer(const SmtpMockServer&) = delete;
  SmtpMockServer& operator=(const SmtpMockServer&) = delete;
 
  // Apply the harness-owned client controls after ordinary scenario options.
  void ConfigureClient(CURL* easy) const;
 
 protected:
  std::unique_ptr<MockConnection> CreateConnection() override;
 
  void HandleLine(std::string_view line) override;
 
  void ResetProtocolState() override;
 
  void QueueShutdownReply() override;
 
 private:
  static constexpr std::string_view kEndOfBody = "\r\n.\r\n";

  static constexpr std::string_view kShutdownReply = "221 mock closing SMTP connection\r\n";
 
  static int FirstReplyCode(std::string_view response);
 
  bool AllowsRecipientFailures() const;
 
  // Return whether the selected probe asks curl to issue STARTTLS.
  bool IsStartTlsMode() const;

  std::size_t recipient_count_;
  std::size_t rcpt_command_count_;
  bool auth_exchange_active_;
  curl::fuzzer::proto::SmtpProbeMode mode_;
  bool tls_upgrade_started_;
};
 
}  // namespace proto_fuzzer
 
#endif  // PROTO_FUZZER_SMTP_MOCK_SERVER_H_