123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860 |
- /*
- * Copyright (c) 2021, Kyle Pereira <hey@xylepereira.me>
- *
- * SPDX-License-Identifier: BSD-2-Clause
- */
- #include <AK/CharacterTypes.h>
- #include <LibIMAP/Parser.h>
- namespace IMAP {
- ParseStatus Parser::parse(ByteBuffer&& buffer, bool expecting_tag)
- {
- if (m_incomplete) {
- m_buffer += buffer;
- m_incomplete = false;
- } else {
- m_buffer = move(buffer);
- position = 0;
- m_response = SolidResponse();
- }
- if (try_consume("+"sv)) {
- consume(" "sv);
- auto data = consume_until_end_of_line();
- consume("\r\n"sv);
- return { true, { ContinueRequest { data } } };
- }
- while (try_consume("*"sv)) {
- parse_untagged();
- }
- if (expecting_tag) {
- if (at_end()) {
- m_incomplete = true;
- return { true, {} };
- }
- parse_response_done();
- }
- if (m_parsing_failed) {
- return { false, {} };
- } else {
- return { true, { { move(m_response) } } };
- }
- }
- bool Parser::try_consume(StringView x)
- {
- size_t i = 0;
- auto previous_position = position;
- while (i < x.length() && !at_end() && to_ascii_lowercase(x[i]) == to_ascii_lowercase(m_buffer[position])) {
- i++;
- position++;
- }
- if (i != x.length()) {
- // We didn't match the full string.
- position = previous_position;
- return false;
- }
- return true;
- }
- void Parser::parse_response_done()
- {
- consume("A"sv);
- auto tag = parse_number();
- consume(" "sv);
- ResponseStatus status = parse_status();
- consume(" "sv);
- m_response.m_tag = tag;
- m_response.m_status = status;
- StringBuilder response_data;
- while (!at_end() && m_buffer[position] != '\r') {
- response_data.append((char)m_buffer[position]);
- position += 1;
- }
- consume("\r\n"sv);
- m_response.m_response_text = response_data.to_deprecated_string();
- }
- void Parser::consume(StringView x)
- {
- if (!try_consume(x)) {
- dbgln("{} not matched at {}, buffer: {}", x, position, StringView(m_buffer.data(), m_buffer.size()));
- m_parsing_failed = true;
- }
- }
- Optional<unsigned> Parser::try_parse_number()
- {
- auto number_matched = 0;
- while (!at_end() && 0 <= m_buffer[position] - '0' && m_buffer[position] - '0' <= 9) {
- number_matched++;
- position++;
- }
- if (number_matched == 0)
- return {};
- auto number = StringView(m_buffer.data() + position - number_matched, number_matched);
- return number.to_uint();
- }
- unsigned Parser::parse_number()
- {
- auto number = try_parse_number();
- if (!number.has_value()) {
- m_parsing_failed = true;
- return -1;
- }
- return number.value();
- }
- void Parser::parse_untagged()
- {
- consume(" "sv);
- // Certain messages begin with a number like:
- // * 15 EXISTS
- auto number = try_parse_number();
- if (number.has_value()) {
- consume(" "sv);
- auto data_type = parse_atom();
- if (data_type == "EXISTS"sv) {
- m_response.data().set_exists(number.value());
- consume("\r\n"sv);
- } else if (data_type == "RECENT"sv) {
- m_response.data().set_recent(number.value());
- consume("\r\n"sv);
- } else if (data_type == "FETCH"sv) {
- auto fetch_response = parse_fetch_response();
- m_response.data().add_fetch_response(number.value(), move(fetch_response));
- } else if (data_type == "EXPUNGE"sv) {
- m_response.data().add_expunged(number.value());
- consume("\r\n"sv);
- }
- return;
- }
- if (try_consume("CAPABILITY"sv)) {
- parse_capability_response();
- } else if (try_consume("LIST"sv)) {
- auto item = parse_list_item();
- m_response.data().add_list_item(move(item));
- } else if (try_consume("LSUB"sv)) {
- auto item = parse_list_item();
- m_response.data().add_lsub_item(move(item));
- } else if (try_consume("FLAGS"sv)) {
- consume(" "sv);
- auto flags = parse_list(+[](StringView x) { return DeprecatedString(x); });
- m_response.data().set_flags(move(flags));
- consume("\r\n"sv);
- } else if (try_consume("OK"sv)) {
- consume(" "sv);
- if (try_consume("["sv)) {
- auto actual_type = parse_atom();
- if (actual_type == "CLOSED"sv) {
- // No-op.
- } else if (actual_type == "UIDNEXT"sv) {
- consume(" "sv);
- auto n = parse_number();
- m_response.data().set_uid_next(n);
- } else if (actual_type == "UIDVALIDITY"sv) {
- consume(" "sv);
- auto n = parse_number();
- m_response.data().set_uid_validity(n);
- } else if (actual_type == "UNSEEN"sv) {
- consume(" "sv);
- auto n = parse_number();
- m_response.data().set_unseen(n);
- } else if (actual_type == "PERMANENTFLAGS"sv) {
- consume(" "sv);
- auto flags = parse_list(+[](StringView x) { return DeprecatedString(x); });
- m_response.data().set_permanent_flags(move(flags));
- } else if (actual_type == "HIGHESTMODSEQ"sv) {
- consume(" "sv);
- parse_number();
- // No-op for now.
- } else {
- dbgln("Unknown: {}", actual_type);
- consume_while([](u8 x) { return x != ']'; });
- }
- consume("]"sv);
- }
- consume_until_end_of_line();
- consume("\r\n"sv);
- } else if (try_consume("SEARCH"sv)) {
- Vector<unsigned> ids;
- while (!try_consume("\r\n"sv)) {
- consume(" "sv);
- auto id = parse_number();
- ids.append(id);
- }
- m_response.data().set_search_results(move(ids));
- } else if (try_consume("BYE"sv)) {
- auto message = consume_until_end_of_line();
- consume("\r\n"sv);
- m_response.data().set_bye(message.is_empty() ? Optional<DeprecatedString>() : Optional<DeprecatedString>(message));
- } else if (try_consume("STATUS"sv)) {
- consume(" "sv);
- auto mailbox = parse_astring();
- consume(" ("sv);
- auto status_item = StatusItem();
- status_item.set_mailbox(mailbox);
- while (!try_consume(")"sv)) {
- auto status_att = parse_atom();
- consume(" "sv);
- auto value = parse_number();
- auto type = StatusItemType::Recent;
- if (status_att == "MESSAGES"sv) {
- type = StatusItemType::Messages;
- } else if (status_att == "UNSEEN"sv) {
- type = StatusItemType::Unseen;
- } else if (status_att == "UIDNEXT"sv) {
- type = StatusItemType::UIDNext;
- } else if (status_att == "UIDVALIDITY"sv) {
- type = StatusItemType::UIDValidity;
- } else if (status_att == "RECENT"sv) {
- type = StatusItemType::Recent;
- } else {
- dbgln("Unmatched status attribute: {}", status_att);
- m_parsing_failed = true;
- }
- status_item.set(type, value);
- if (!at_end() && m_buffer[position] != ')')
- consume(" "sv);
- }
- m_response.data().set_status(move(status_item));
- try_consume(" "sv); // Not in the spec but the Outlook server sends a space for some reason.
- consume("\r\n"sv);
- } else {
- auto x = consume_until_end_of_line();
- consume("\r\n"sv);
- dbgln("ignored {}", x);
- }
- }
- StringView Parser::parse_quoted_string()
- {
- auto str = consume_while([](u8 x) { return x != '"'; });
- consume("\""sv);
- return str;
- }
- StringView Parser::parse_string()
- {
- if (try_consume("\""sv)) {
- return parse_quoted_string();
- } else {
- return parse_literal_string();
- }
- }
- Optional<StringView> Parser::parse_nstring()
- {
- if (try_consume("NIL"sv))
- return {};
- else
- return { parse_string() };
- }
- FetchResponseData Parser::parse_fetch_response()
- {
- consume(" ("sv);
- auto fetch_response = FetchResponseData();
- while (!try_consume(")"sv)) {
- auto data_item = parse_fetch_data_item();
- switch (data_item.type) {
- case FetchCommand::DataItemType::BodyStructure: {
- consume(" ("sv);
- auto structure = parse_body_structure();
- fetch_response.set_body_structure(move(structure));
- break;
- }
- case FetchCommand::DataItemType::Envelope: {
- fetch_response.set_envelope(parse_envelope());
- break;
- }
- case FetchCommand::DataItemType::Flags: {
- consume(" "sv);
- auto flags = parse_list(+[](StringView x) { return DeprecatedString(x); });
- fetch_response.set_flags(move(flags));
- break;
- }
- case FetchCommand::DataItemType::InternalDate: {
- consume(" \""sv);
- auto date_view = consume_while([](u8 x) { return x != '"'; });
- consume("\""sv);
- auto date = Core::DateTime::parse("%d-%b-%Y %H:%M:%S %z"sv, date_view).value();
- fetch_response.set_internal_date(date);
- break;
- }
- case FetchCommand::DataItemType::UID: {
- consume(" "sv);
- fetch_response.set_uid(parse_number());
- break;
- }
- case FetchCommand::DataItemType::PeekBody:
- // Spec doesn't allow for this in a response.
- m_parsing_failed = true;
- break;
- case FetchCommand::DataItemType::BodySection: {
- auto body = parse_nstring();
- fetch_response.add_body_data(move(data_item), Optional<DeprecatedString>(move(body)));
- break;
- }
- }
- if (!at_end() && m_buffer[position] != ')')
- consume(" "sv);
- }
- consume("\r\n"sv);
- return fetch_response;
- }
- Envelope Parser::parse_envelope()
- {
- consume(" ("sv);
- auto date = parse_nstring();
- consume(" "sv);
- auto subject = parse_nstring();
- consume(" "sv);
- auto from = parse_address_list();
- consume(" "sv);
- auto sender = parse_address_list();
- consume(" "sv);
- auto reply_to = parse_address_list();
- consume(" "sv);
- auto to = parse_address_list();
- consume(" "sv);
- auto cc = parse_address_list();
- consume(" "sv);
- auto bcc = parse_address_list();
- consume(" "sv);
- auto in_reply_to = parse_nstring();
- consume(" "sv);
- auto message_id = parse_nstring();
- consume(")"sv);
- Envelope envelope = {
- date.has_value() ? AK::Optional<DeprecatedString>(date.value()) : AK::Optional<DeprecatedString>(),
- subject.has_value() ? AK::Optional<DeprecatedString>(subject.value()) : AK::Optional<DeprecatedString>(),
- from,
- sender,
- reply_to,
- to,
- cc,
- bcc,
- in_reply_to.has_value() ? AK::Optional<DeprecatedString>(in_reply_to.value()) : AK::Optional<DeprecatedString>(),
- message_id.has_value() ? AK::Optional<DeprecatedString>(message_id.value()) : AK::Optional<DeprecatedString>(),
- };
- return envelope;
- }
- BodyStructure Parser::parse_body_structure()
- {
- if (!at_end() && m_buffer[position] == '(') {
- auto data = MultiPartBodyStructureData();
- while (try_consume("("sv)) {
- auto child = parse_body_structure();
- data.bodies.append(make<BodyStructure>(move(child)));
- }
- consume(" "sv);
- data.media_type = parse_string();
- if (!try_consume(")"sv)) {
- consume(" "sv);
- data.params = try_consume("NIL"sv) ? Optional<HashMap<DeprecatedString, DeprecatedString>>() : parse_body_fields_params();
- if (!try_consume(")"sv)) {
- consume(" "sv);
- if (!try_consume("NIL"sv)) {
- data.disposition = { parse_disposition() };
- }
- if (!try_consume(")"sv)) {
- consume(" "sv);
- if (!try_consume("NIL"sv)) {
- data.langs = { parse_langs() };
- }
- if (!try_consume(")"sv)) {
- consume(" "sv);
- data.location = try_consume("NIL"sv) ? Optional<DeprecatedString>() : Optional<DeprecatedString>(parse_string());
- if (!try_consume(")"sv)) {
- consume(" "sv);
- Vector<BodyExtension> extensions;
- while (!try_consume(")"sv)) {
- extensions.append(parse_body_extension());
- try_consume(" "sv);
- }
- data.extensions = { move(extensions) };
- }
- }
- }
- }
- }
- return BodyStructure(move(data));
- } else {
- return parse_one_part_body();
- }
- }
- BodyStructure Parser::parse_one_part_body()
- {
- auto type = parse_string();
- consume(" "sv);
- auto subtype = parse_string();
- consume(" "sv);
- if (type.equals_ignoring_ascii_case("TEXT"sv)) {
- // body-type-text
- auto params = parse_body_fields_params();
- consume(" "sv);
- auto id = parse_nstring();
- consume(" "sv);
- auto description = parse_nstring();
- consume(" "sv);
- auto encoding = parse_string();
- consume(" "sv);
- auto num_octets = parse_number();
- consume(" "sv);
- auto num_lines = parse_number();
- auto data = BodyStructureData {
- type,
- subtype,
- Optional<DeprecatedString>(move(id)),
- Optional<DeprecatedString>(move(description)),
- encoding,
- params,
- num_octets,
- num_lines,
- {}
- };
- if (!try_consume(")"sv)) {
- consume(" "sv);
- auto md5 = parse_nstring();
- if (md5.has_value())
- data.md5 = { md5.value() };
- if (!try_consume(")"sv)) {
- consume(" "sv);
- if (!try_consume("NIL"sv)) {
- auto disposition = parse_disposition();
- data.disposition = { disposition };
- }
- if (!try_consume(")"sv)) {
- consume(" "sv);
- if (!try_consume("NIL"sv)) {
- data.langs = { parse_langs() };
- }
- if (!try_consume(")"sv)) {
- consume(" "sv);
- auto location = parse_nstring();
- if (location.has_value())
- data.location = { location.value() };
- Vector<BodyExtension> extensions;
- while (!try_consume(")"sv)) {
- extensions.append(parse_body_extension());
- try_consume(" "sv);
- }
- data.extensions = { move(extensions) };
- }
- }
- }
- }
- return BodyStructure(move(data));
- } else if (type.equals_ignoring_ascii_case("MESSAGE"sv) && subtype.equals_ignoring_ascii_case("RFC822"sv)) {
- // body-type-message
- auto params = parse_body_fields_params();
- consume(" "sv);
- auto id = parse_nstring();
- consume(" "sv);
- auto description = parse_nstring();
- consume(" "sv);
- auto encoding = parse_string();
- consume(" "sv);
- auto num_octets = parse_number();
- consume(" "sv);
- auto envelope = parse_envelope();
- BodyStructureData data {
- type,
- subtype,
- Optional<DeprecatedString>(move(id)),
- Optional<DeprecatedString>(move(description)),
- encoding,
- params,
- num_octets,
- 0,
- envelope
- };
- return BodyStructure(move(data));
- } else {
- // body-type-basic
- auto params = parse_body_fields_params();
- consume(" "sv);
- auto id = parse_nstring();
- consume(" "sv);
- auto description = parse_nstring();
- consume(" "sv);
- auto encoding = parse_string();
- consume(" "sv);
- auto num_octets = parse_number();
- consume(" "sv);
- BodyStructureData data {
- type,
- subtype,
- Optional<DeprecatedString>(move(id)),
- Optional<DeprecatedString>(move(description)),
- encoding,
- params,
- num_octets,
- 0,
- {}
- };
- return BodyStructure(move(data));
- }
- }
- Vector<DeprecatedString> Parser::parse_langs()
- {
- AK::Vector<DeprecatedString> langs;
- if (!try_consume("("sv)) {
- langs.append(parse_string());
- } else {
- while (!try_consume(")"sv)) {
- langs.append(parse_string());
- try_consume(" "sv);
- }
- }
- return langs;
- }
- Tuple<DeprecatedString, HashMap<DeprecatedString, DeprecatedString>> Parser::parse_disposition()
- {
- auto disposition_type = parse_string();
- consume(" "sv);
- auto disposition_vals = parse_body_fields_params();
- consume(")"sv);
- return { move(disposition_type), move(disposition_vals) };
- }
- StringView Parser::parse_literal_string()
- {
- consume("{"sv);
- auto num_bytes = parse_number();
- consume("}\r\n"sv);
- if (m_buffer.size() < position + num_bytes) {
- m_parsing_failed = true;
- return ""sv;
- }
- position += num_bytes;
- return StringView(m_buffer.data() + position - num_bytes, num_bytes);
- }
- ListItem Parser::parse_list_item()
- {
- consume(" "sv);
- auto flags_vec = parse_list(parse_mailbox_flag);
- unsigned flags = 0;
- for (auto flag : flags_vec) {
- flags |= static_cast<unsigned>(flag);
- }
- consume(" \""sv);
- auto reference = consume_while([](u8 x) { return x != '"'; });
- consume("\" "sv);
- auto mailbox = parse_astring();
- consume("\r\n"sv);
- return ListItem { flags, DeprecatedString(reference), DeprecatedString(mailbox) };
- }
- void Parser::parse_capability_response()
- {
- auto capability = AK::Vector<DeprecatedString>();
- while (!try_consume("\r\n"sv)) {
- consume(" "sv);
- auto x = DeprecatedString(parse_atom());
- capability.append(x);
- }
- m_response.data().add_capabilities(move(capability));
- }
- StringView Parser::parse_atom()
- {
- auto is_non_atom_char = [](u8 x) {
- auto non_atom_chars = { '(', ')', '{', ' ', '%', '*', '"', '\\', ']' };
- return AK::find(non_atom_chars.begin(), non_atom_chars.end(), x) != non_atom_chars.end();
- };
- auto start = position;
- auto count = 0;
- while (!at_end() && !is_ascii_control(m_buffer[position]) && !is_non_atom_char(m_buffer[position])) {
- count++;
- position++;
- }
- return StringView(m_buffer.data() + start, count);
- }
- ResponseStatus Parser::parse_status()
- {
- auto atom = parse_atom();
- if (atom == "OK"sv) {
- return ResponseStatus::OK;
- } else if (atom == "BAD"sv) {
- return ResponseStatus::Bad;
- } else if (atom == "NO"sv) {
- return ResponseStatus::No;
- }
- m_parsing_failed = true;
- return ResponseStatus::Bad;
- }
- template<typename T>
- Vector<T> Parser::parse_list(T converter(StringView))
- {
- consume("("sv);
- Vector<T> x;
- bool first = true;
- while (!try_consume(")"sv)) {
- if (!first)
- consume(" "sv);
- auto item = consume_while([](u8 x) {
- return x != ' ' && x != ')';
- });
- x.append(converter(item));
- first = false;
- }
- return x;
- }
- MailboxFlag Parser::parse_mailbox_flag(StringView s)
- {
- if (s == "\\All"sv)
- return MailboxFlag::All;
- if (s == "\\Drafts"sv)
- return MailboxFlag::Drafts;
- if (s == "\\Flagged"sv)
- return MailboxFlag::Flagged;
- if (s == "\\HasChildren"sv)
- return MailboxFlag::HasChildren;
- if (s == "\\HasNoChildren"sv)
- return MailboxFlag::HasNoChildren;
- if (s == "\\Important"sv)
- return MailboxFlag::Important;
- if (s == "\\Junk"sv)
- return MailboxFlag::Junk;
- if (s == "\\Marked"sv)
- return MailboxFlag::Marked;
- if (s == "\\Noinferiors"sv)
- return MailboxFlag::NoInferiors;
- if (s == "\\Noselect"sv)
- return MailboxFlag::NoSelect;
- if (s == "\\Sent"sv)
- return MailboxFlag::Sent;
- if (s == "\\Trash"sv)
- return MailboxFlag::Trash;
- if (s == "\\Unmarked"sv)
- return MailboxFlag::Unmarked;
- dbgln("Unrecognized mailbox flag {}", s);
- return MailboxFlag::Unknown;
- }
- StringView Parser::consume_while(Function<bool(u8)> should_consume)
- {
- int chars = 0;
- while (!at_end() && should_consume(m_buffer[position])) {
- position++;
- chars++;
- }
- return StringView(m_buffer.data() + position - chars, chars);
- }
- StringView Parser::consume_until_end_of_line()
- {
- return consume_while([](u8 x) { return x != '\r'; });
- }
- FetchCommand::DataItem Parser::parse_fetch_data_item()
- {
- auto msg_attr = consume_while([](u8 x) { return is_ascii_alpha(x) != 0; });
- if (msg_attr.equals_ignoring_ascii_case("BODY"sv) && try_consume("["sv)) {
- auto data_item = FetchCommand::DataItem {
- .type = FetchCommand::DataItemType::BodySection,
- .section = { {} }
- };
- auto section_type = consume_while([](u8 x) { return x != ']' && x != ' '; });
- if (section_type.equals_ignoring_ascii_case("HEADER.FIELDS"sv)) {
- data_item.section->type = FetchCommand::DataItem::SectionType::HeaderFields;
- data_item.section->headers = Vector<DeprecatedString>();
- consume(" "sv);
- auto headers = parse_list(+[](StringView x) { return x; });
- for (auto& header : headers) {
- data_item.section->headers->append(header);
- }
- consume("]"sv);
- } else if (section_type.equals_ignoring_ascii_case("HEADER.FIELDS.NOT"sv)) {
- data_item.section->type = FetchCommand::DataItem::SectionType::HeaderFieldsNot;
- data_item.section->headers = Vector<DeprecatedString>();
- consume(" ("sv);
- auto headers = parse_list(+[](StringView x) { return x; });
- for (auto& header : headers) {
- data_item.section->headers->append(header);
- }
- consume("]"sv);
- } else if (is_ascii_digit(section_type[0])) {
- data_item.section->type = FetchCommand::DataItem::SectionType::Parts;
- data_item.section->parts = Vector<unsigned>();
- while (!try_consume("]"sv)) {
- auto num = try_parse_number();
- if (num.has_value()) {
- data_item.section->parts->append(num.value());
- continue;
- }
- auto atom = parse_atom();
- if (atom.equals_ignoring_ascii_case("MIME"sv)) {
- data_item.section->ends_with_mime = true;
- continue;
- }
- }
- } else if (section_type.equals_ignoring_ascii_case("TEXT"sv)) {
- data_item.section->type = FetchCommand::DataItem::SectionType::Text;
- } else if (section_type.equals_ignoring_ascii_case("HEADER"sv)) {
- data_item.section->type = FetchCommand::DataItem::SectionType::Header;
- } else {
- dbgln("Unmatched section type {}", section_type);
- m_parsing_failed = true;
- }
- if (try_consume("<"sv)) {
- auto start = parse_number();
- data_item.partial_fetch = true;
- data_item.start = (int)start;
- consume(">"sv);
- }
- try_consume(" "sv);
- return data_item;
- } else if (msg_attr.equals_ignoring_ascii_case("FLAGS"sv)) {
- return FetchCommand::DataItem {
- .type = FetchCommand::DataItemType::Flags
- };
- } else if (msg_attr.equals_ignoring_ascii_case("UID"sv)) {
- return FetchCommand::DataItem {
- .type = FetchCommand::DataItemType::UID
- };
- } else if (msg_attr.equals_ignoring_ascii_case("INTERNALDATE"sv)) {
- return FetchCommand::DataItem {
- .type = FetchCommand::DataItemType::InternalDate
- };
- } else if (msg_attr.equals_ignoring_ascii_case("ENVELOPE"sv)) {
- return FetchCommand::DataItem {
- .type = FetchCommand::DataItemType::Envelope
- };
- } else if (msg_attr.equals_ignoring_ascii_case("BODY"sv) || msg_attr.equals_ignoring_ascii_case("BODYSTRUCTURE"sv)) {
- return FetchCommand::DataItem {
- .type = FetchCommand::DataItemType::BodyStructure
- };
- } else {
- dbgln("msg_attr not matched: {}", msg_attr);
- m_parsing_failed = true;
- return FetchCommand::DataItem {};
- }
- }
- Optional<Vector<Address>> Parser::parse_address_list()
- {
- if (try_consume("NIL"sv))
- return {};
- auto addresses = Vector<Address>();
- consume("("sv);
- while (!try_consume(")"sv)) {
- addresses.append(parse_address());
- if (!at_end() && m_buffer[position] != ')')
- consume(" "sv);
- }
- return { addresses };
- }
- Address Parser::parse_address()
- {
- consume("("sv);
- auto address = Address();
- auto name = parse_nstring();
- address.name = Optional<DeprecatedString>(move(name));
- consume(" "sv);
- auto source_route = parse_nstring();
- address.source_route = Optional<DeprecatedString>(move(source_route));
- consume(" "sv);
- auto mailbox = parse_nstring();
- address.mailbox = Optional<DeprecatedString>(move(mailbox));
- consume(" "sv);
- auto host = parse_nstring();
- address.host = Optional<DeprecatedString>(move(host));
- consume(")"sv);
- return address;
- }
- StringView Parser::parse_astring()
- {
- if (!at_end() && (m_buffer[position] == '{' || m_buffer[position] == '"'))
- return parse_string();
- else
- return parse_atom();
- }
- HashMap<DeprecatedString, DeprecatedString> Parser::parse_body_fields_params()
- {
- if (try_consume("NIL"sv))
- return {};
- HashMap<DeprecatedString, DeprecatedString> fields;
- consume("("sv);
- while (!try_consume(")"sv)) {
- auto key = parse_string();
- consume(" "sv);
- auto value = parse_string();
- fields.set(key, value);
- try_consume(" "sv);
- }
- return fields;
- }
- BodyExtension Parser::parse_body_extension()
- {
- if (try_consume("NIL"sv)) {
- return BodyExtension { Optional<DeprecatedString> {} };
- } else if (try_consume("("sv)) {
- Vector<OwnPtr<BodyExtension>> extensions;
- while (!try_consume(")"sv)) {
- extensions.append(make<BodyExtension>(parse_body_extension()));
- try_consume(" "sv);
- }
- return BodyExtension { move(extensions) };
- } else if (!at_end() && (m_buffer[position] == '"' || m_buffer[position] == '{')) {
- return BodyExtension { { parse_string() } };
- } else {
- return BodyExtension { parse_number() };
- }
- }
- }
|