mirror of
https://github.com/SerenityOS/serenity.git
synced 2025-01-24 10:22:05 -05:00
34adf9eeae
We're now sniffing the incoming data to verify the server has sent a full response, instead of passing partial data to our IMAP parser. Our parser really can't handle partial input very well, so for the time being, this heuristic does a much better job of verifying we have full response before parsing. It doesn't yet handle unprompted untagged reponses, nor the "continuation request" responses that start with a `+`, but we don't fully support those yet.
89 lines
3.6 KiB
C++
89 lines
3.6 KiB
C++
/*
|
|
* Copyright (c) 2021, Kyle Pereira <hey@xylepereira.me>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/Function.h>
|
|
#include <LibCore/Promise.h>
|
|
#include <LibIMAP/Parser.h>
|
|
#include <LibTLS/TLSv12.h>
|
|
|
|
namespace IMAP {
|
|
template<typename T>
|
|
using Promise = Core::Promise<T>;
|
|
|
|
class Client {
|
|
AK_MAKE_NONCOPYABLE(Client);
|
|
friend class Parser;
|
|
|
|
public:
|
|
static ErrorOr<NonnullOwnPtr<Client>> connect_tls(StringView host, u16 port);
|
|
static ErrorOr<NonnullOwnPtr<Client>> connect_plaintext(StringView host, u16 port);
|
|
|
|
Client(Client&&);
|
|
|
|
RefPtr<Promise<Empty>> connection_promise()
|
|
{
|
|
return m_connect_pending;
|
|
}
|
|
|
|
ErrorOr<RefPtr<Promise<Optional<Response>>>> send_command(Command&&);
|
|
ErrorOr<RefPtr<Promise<Optional<Response>>>> send_simple_command(CommandType);
|
|
ErrorOr<void> send_raw(StringView data);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> login(StringView username, StringView password);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> list(StringView reference_name, StringView mailbox_name);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> lsub(StringView reference_name, StringView mailbox_name);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> select(StringView string);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> examine(StringView string);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> search(Optional<DeprecatedString> charset, Vector<SearchKey>&& search_keys, bool uid);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> fetch(FetchCommand request, bool uid);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> store(StoreMethod, Sequence, bool silent, Vector<DeprecatedString> const& flags, bool uid);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> copy(Sequence sequence_set, StringView name, bool uid);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> create_mailbox(StringView name);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> delete_mailbox(StringView name);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> subscribe(StringView mailbox);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> unsubscribe(StringView mailbox);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> rename(StringView from, StringView to);
|
|
ErrorOr<RefPtr<Promise<Optional<Response>>>> authenticate(StringView method);
|
|
ErrorOr<RefPtr<Promise<Optional<ContinueRequest>>>> idle();
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> finish_idle();
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> status(StringView mailbox, Vector<StatusItemType> const& types);
|
|
ErrorOr<RefPtr<Promise<Optional<SolidResponse>>>> append(StringView mailbox, Message&& message, Optional<Vector<DeprecatedString>> flags = {}, Optional<Core::DateTime> date_time = {});
|
|
|
|
bool is_open();
|
|
void close();
|
|
|
|
Function<void(ResponseData&&)> unrequested_response_callback;
|
|
|
|
private:
|
|
Client(StringView host, u16 port, NonnullOwnPtr<Core::Socket>);
|
|
void setup_callbacks();
|
|
|
|
ErrorOr<void> on_ready_to_receive();
|
|
bool verify_response_is_complete();
|
|
|
|
ErrorOr<void> handle_parsed_response(ParseStatus&& parse_status);
|
|
ErrorOr<void> send_next_command();
|
|
|
|
StringView m_host;
|
|
u16 m_port;
|
|
|
|
NonnullOwnPtr<Core::Socket> m_socket;
|
|
RefPtr<Promise<Empty>> m_connect_pending {};
|
|
|
|
int m_current_command = 1;
|
|
|
|
// Sent but response not received
|
|
Vector<RefPtr<Promise<Optional<Response>>>> m_pending_promises;
|
|
// Not yet sent
|
|
Vector<Command> m_command_queue {};
|
|
|
|
ByteBuffer m_buffer;
|
|
Parser m_parser {};
|
|
|
|
bool m_expecting_response { false };
|
|
};
|
|
}
|