imap_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_IMAP_MOCK_SERVER_H_
#define PROTO_FUZZER_IMAP_MOCK_SERVER_H_
 
#include <cstddef>
#include <string>
#include <string_view>
 
#include "proto_fuzzer/pingpong_mock_server.h"
 
namespace proto_fuzzer {

class ImapMockServer : public PingpongMockServer {
 public:
  ImapMockServer();
 
  ~ImapMockServer() override;
 
  void ObserveState() override;
 
  ImapMockServer(const ImapMockServer&) = delete;
  ImapMockServer& operator=(const ImapMockServer&) = delete;
 
 protected:
  void HandleLine(std::string_view line) override;
 
  void ResetProtocolState() override;
 
  void QueueShutdownReply() override;
 
 private:
  static constexpr std::string_view kShutdownTail = " OK LOGOUT completed\r\n";
 
  // Resolve a scripted reply into wire bytes.
  std::string RewriteReply(std::string_view response) const;
 
  // Whether a complete line needs the current command tag prepended.
  static bool StatusLineNeedsTag(std::string_view line);
 
  // Whether a scripted reply contains a terminal tag-relative status line.
  bool ResponseHasTaggedCompletion(std::string_view response) const;
 
  // Whether a scripted reply contains an IMAP continuation line.
  static bool ResponseHasContinuation(std::string_view response);
 
  // Literal byte count for a trailing `{N}` marker, or 0 when absent.
  static std::size_t LiteralSize(std::string_view line);
 
  // Literal size announced by an APPEND command, or max size_t when absent.
  static std::size_t AppendLiteralSize(std::string_view command);
 
  // Tag curl will use for the next command, derived from the current tag.
  std::string NextCommandTag() const;

  std::string current_tag_;

  bool authentication_continuation_active_ = false;
};
 
}  // namespace proto_fuzzer
 
#endif  // PROTO_FUZZER_IMAP_MOCK_SERVER_H_