/**
 * Copyright (C) 2021-2024 Saturneric <eric@bktus.com>
 *
 * This file is part of GpgFrontend.
 *
 * GpgFrontend is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * GpgFrontend is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with GpgFrontend. If not, see <https://www.gnu.org/licenses/>.
 *
 * The initial version of the source code is inherited from
 * the gpg4usb project, which is under GPL-3.0-or-later.
 *
 * All the source code of GpgFrontend was modified and released by
 * Saturneric <eric@bktus.com> starting on May 12, 2021.
 *
 * SPDX-License-Identifier: GPL-3.0-or-later
 *
 */

#include "EMailHelper.h"

#include <QCryptographicHash>
#include <QFile>
#include <QFileInfo>
#include <QJsonArray>
#include <QJsonDocument>
#include <QJsonObject>
#include <QLocale>
#include <QRegularExpression>
#include <QTimeZone>
#include <QUrl>
#include <algorithm>
#include <limits>

namespace {
MimeLogFn g_mime_log_sink = nullptr;
}  // namespace

void SetMimeLogSink(MimeLogFn fn) { g_mime_log_sink = fn; }

void MimeLog(const QString& message) {
  if (g_mime_log_sink != nullptr) g_mime_log_sink(message);
}

static const QRegularExpression kNameEmailStringRegex{
    R"(^\s*(.*)\s*<\s*([^<>@\s]+@[^<>@\s]+)\s*>\s*$)"};

auto IsValidMicalgFormat(const QString& prm_micalg_value) -> bool {
  QRegularExpression regex("^pgp-(\\w+)$");
  QRegularExpressionMatch match = regex.match(prm_micalg_value);
  return match.hasMatch();
}

auto WorseStatus(int a, int b) -> int { return std::min(a, b); }

auto FormatMailBox(const std::shared_ptr<vmime::mailbox>& m) -> QString {
  if (!m) return {"Unknown"};

  QString name = Q_SC(m->getName().getConvertedText(vmime::charsets::UTF_8));
  QString address =
      Q_SC(m->getEmail().toText().getConvertedText(vmime::charsets::UTF_8));

  if (!name.isEmpty()) {
    return QString("%1 <%2>").arg(name, address);
  }
  return address;
}

auto TryGetField(const vmime::shared_ptr<vmime::header>& header,
                 const QString& field_name)
    -> vmime::shared_ptr<vmime::headerField> {
  // Both header::getField() overloads create and insert the field when it is
  // absent, so reading an optional header through them silently adds an empty
  // one to the message -- which then gets signed, or sent. Ask first.
  const auto name = field_name.toStdString();
  if (!header->hasField(name)) return nullptr;
  return header->getField(name);
}

auto ExtractFieldValue(const vmime::shared_ptr<vmime::header>& header,
                       const QString& field_name) -> QString {
  auto field = TryGetField(header, field_name);
  if (!field) {
    MimeLog(QString("cannot get '%1' field from header").arg(field_name));
    return {};
  }

  auto field_value = field->getValue();
  if (!field_value) {
    MimeLog(QString("cannot get '%1' field value from header").arg(field_name));
    return {};
  }

  return Q_SC(field_value->generate());
}

auto ExtractFieldValueMailBox(const vmime::shared_ptr<vmime::header>& header,
                              const QString& field_name) -> QString {
  auto field = TryGetField(header, field_name);
  if (!field) {
    MimeLog(QString("cannot get '%1' field from header").arg(field_name));
    return {};
  }

  auto field_value = field->getValue<vmime::mailbox>();
  if (!field_value) {
    MimeLog(QString("cannot get '%1' field value from header").arg(field_name));
    return {};
  }

  return FormatMailBox(field_value);
}

auto ExtractFieldValueAddressList(
    const vmime::shared_ptr<vmime::header>& header, const QString& field_name)
    -> QString {
  auto field = TryGetField(header, field_name);
  if (!field) {
    MimeLog(QString("cannot get '%1' field from header").arg(field_name));
    return {};
  }

  auto field_value = field->getValue<vmime::addressList>();
  if (!field_value) {
    MimeLog(QString("cannot get '%1' field value from header").arg(field_name));
    return {};
  }

  return ExtractFieldValueAddressListItems(header, field_name).join(", ");
}

auto ExtractFieldValueAddressListItems(
    const vmime::shared_ptr<vmime::header>& header, const QString& field_name)
    -> QStringList {
  auto field = TryGetField(header, field_name);
  if (!field) return {};

  auto field_value = field->getValue<vmime::addressList>();
  if (!field_value) return {};

  // Returned as a list rather than a joined string: a display name may contain
  // a comma ("Doe, John"), and splitting a joined list back apart on "," tears
  // such a name in half and invents a recipient that was never addressed.
  QStringList mailboxes;
  for (const auto& mailbox : field_value->toMailboxList()->getMailboxList()) {
    mailboxes.append(FormatMailBox(mailbox));
  }
  return mailboxes;
}

auto ExtractFieldValueText(const vmime::shared_ptr<vmime::header>& header,
                           const QString& field_name) -> QString {
  if (!header->hasField(field_name.toStdString())) return {};

  auto field = TryGetField(header, field_name);
  if (!field) {
    MimeLog(QString("cannot get '%1' field from header").arg(field_name));
    return {};
  }

  auto field_value = field->getValue<vmime::text>();
  if (!field_value) {
    MimeLog(QString("cannot get '%1' field value from header").arg(field_name));
    return {};
  }

  return Q_SC(field_value->getConvertedText(vmime::charsets::UTF_8));
}

auto ExtractFieldValueDateTime(const vmime::shared_ptr<vmime::header>& header,
                               const QString& field_name) -> QDateTime {
  auto field = TryGetField(header, field_name);
  if (!field) {
    MimeLog(QString("cannot get '%1' field from header").arg(field_name));
    return {};
  }

  auto field_value = field->getValue<vmime::datetime>();
  if (!field_value) {
    MimeLog(QString("cannot get '%1' field value from header").arg(field_name));
    return {};
  }

  QDate date(field_value->getYear(), field_value->getMonth(),
             field_value->getDay());
  QTime time(field_value->getHour(), field_value->getMinute(),
             field_value->getSecond());
  field_value->getZone();

  auto zone = field_value->getZone();
  QDateTime datetime(date, time);

  int offset_sec = zone * 60;

#if QT_VERSION >= QT_VERSION_CHECK(6, 6, 0)
  QTimeZone tz(offset_sec);
  datetime.setTimeZone(tz);
#else
  datetime.setOffsetFromUtc(offset_sec);
#endif

  return datetime;
}

auto ParseEmailString(const QString& input, QString& name, QString& email)
    -> bool {
  QRegularExpressionMatch match = kNameEmailStringRegex.match(input);

  if (match.hasMatch()) {
    name = match.captured(1).trimmed();
    email = match.captured(2).trimmed();
    return true;
  }

  return false;
}

auto EncodeBase64WithLineBreaks(const QByteArray& data, int line_length)
    -> QString {
  // Get the Base64 encoded data
  QByteArray base64_data = data.toBase64();

  // Split the base64 data into lines of the given line length
  QStringList lines;
  for (int i = 0; i < base64_data.size(); i += line_length) {
    lines.append(base64_data.mid(i, line_length));
  }

  // Join lines with CRLF
  return lines.join("\r\n");
}

auto CheckIfEMLMessage(const QByteArray& data,
                       vmime::shared_ptr<vmime::message>& message) -> bool {
  if (data.size() > kMaxParseInputBytes) {
    MimeLog(QString("refusing to parse %1 bytes: over the input ceiling")
                .arg(data.size()));
    return false;
  }

  vmime::string vmime_data(data.constData(), data.size());

  message = vmime::make_shared<vmime::message>();
  try {
    // A parsing context of our own rather than the shared default, with the
    // nesting depth held far below what vmime would otherwise allow.
    //
    // Two things are bounded by this, and the second is the reason it is this
    // low. Descending is recursion, so depth is stack depth, and a few
    // thousand levels -- a message well under 1 MB -- terminates the process.
    // And each level rescans the buffer for its boundary, so the work is
    // depth x size: at vmime's own default a 3 MB message took over a minute.
    //
    // EMailParseLimits::max_depth is 8, so nothing legitimate comes close;
    // this only has to be generous enough that the module's own limit is
    // always the one that reports the refusal.
    vmime::parsingContext ctx;
    ctx.setMaxNestingDepth(kMaxParseNestingDepth);

    message->parse(ctx, vmime_data);
    return message->getParsedLength() != 0 && !message->getHeader()->isEmpty();
  } catch (const vmime::exception& e) {
    MimeLog(QString("error parsing vmime data: %1").arg(e.what()));
    return false;
  }
}

auto BuildMimeEML(const EMailMetaData& meta_data, const QByteArray& body_data,
                  const QList<EMailAttachment>& attachments,
                  QByteArray& eml_data) -> int {
  auto from = meta_data.from;
  auto recipient_list = meta_data.to;
  auto cc_list = meta_data.cc;
  auto subject = meta_data.subject;

  QString name;
  QString email;

  try {
    vmime::messageBuilder plaintext_msg_builder;

    if (ParseEmailString(from, name, email)) {
      plaintext_msg_builder.setExpeditor(
          vmime::mailbox(Q_TEXT(name), email.toStdString()));
    } else {
      plaintext_msg_builder.setExpeditor(vmime::mailbox(from.toStdString()));
    }

    for (const QString& recipient : recipient_list) {
      auto trimmed_recipient = recipient.trimmed();
      if (ParseEmailString(trimmed_recipient, name, email)) {
        plaintext_msg_builder.getRecipients().appendAddress(
            vmime::make_shared<vmime::mailbox>(Q_TEXT(name),
                                               email.toStdString()));
      } else {
        plaintext_msg_builder.getRecipients().appendAddress(
            vmime::make_shared<vmime::mailbox>(
                trimmed_recipient.toStdString()));
      }
    }

    for (const QString& recipient : cc_list) {
      auto trimmed_recipient = recipient.trimmed();
      if (ParseEmailString(trimmed_recipient, name, email)) {
        plaintext_msg_builder.getCopyRecipients().appendAddress(
            vmime::make_shared<vmime::mailbox>(Q_TEXT(name),
                                               email.toStdString()));
      } else {
        plaintext_msg_builder.getCopyRecipients().appendAddress(
            vmime::make_shared<vmime::mailbox>(
                trimmed_recipient.toStdString()));
      }
    }

    // Blind recipients are deliberately NOT handed to the builder.
    //
    // vmime::messageBuilder turns blind copy recipients into a real Bcc:
    // header, and this function produces the bytes the user saves and sends.
    // Writing that header tells every recipient exactly who was blind-copied,
    // which is the one thing BCC exists to prevent. Blind recipients belong to
    // EMailComposeState, beside the document rather than inside it; they
    // select encryption recipients and never reach the message.

    plaintext_msg_builder.setSubject(Q_TEXT(subject));

    // vmime::messageBuilder refuses to construct a message with no recipient.
    // That is a rule about sending, and this function is also what a draft is
    // serialized through while it is being written -- an unaddressed draft
    // still has to round-trip, or the headers and attachments the user has
    // already entered are lost the moment anything reads the document back.
    // Blind recipients no longer count towards being addressed: since no Bcc
    // header is written, a message with only blind recipients really does
    // reach the builder with no addressee, and must take the draft path.
    const bool addressed = !recipient_list.isEmpty() || !cc_list.isEmpty();

    vmime::shared_ptr<vmime::message> plaintext_msg;
    if (addressed) {
      plaintext_msg = plaintext_msg_builder.construct();
    } else {
      plaintext_msg = vmime::make_shared<vmime::message>();
      auto draft_header = plaintext_msg->getHeader();
      draft_header->Subject()->setValue(Q_TEXT(subject));
      if (!from.trimmed().isEmpty()) {
        if (ParseEmailString(from, name, email)) {
          draft_header->From()->setValue(
              vmime::mailbox(Q_TEXT(name), email.toStdString()));
        } else {
          draft_header->From()->setValue(vmime::mailbox(from.toStdString()));
        }
      }
      draft_header->Date()->setValue(vmime::datetime::now());
      draft_header->MimeVersion()->setValue(vmime::SUPPORTED_MIME_VERSION);
    }

    auto plaintext_msg_header = plaintext_msg->getHeader();

    // Threading headers, when the caller supplied them. Written here rather
    // than by the builder, which has no notion of them, and only when present
    // so an ordinary new message does not gain empty fields.
    if (!meta_data.in_reply_to.trimmed().isEmpty()) {
      plaintext_msg_header->InReplyTo()->setValue(
          meta_data.in_reply_to.trimmed().toStdString());
    }
    if (!meta_data.references.isEmpty()) {
      plaintext_msg_header->References()->setValue(
          meta_data.references.join(" ").toStdString());
    }

    // Message-ID, and only when one was supplied. This function is called for
    // every draft save and every reserialization, so minting an identifier
    // here would give the message a different identity each time it was
    // written -- which would in turn make it impossible to look the sent copy
    // up afterwards. Whoever is about to send decides the identity; this just
    // renders it. See FreezeOutgoing().
    if (!meta_data.message_id.trimmed().isEmpty()) {
      auto id = meta_data.message_id.trimmed();
      if (id.startsWith('<')) id.remove(0, 1);
      if (id.endsWith('>')) id.chop(1);

      const auto at = id.indexOf('@');
      if (at > 0) {
        plaintext_msg_header->MessageId()->setValue(vmime::messageId(
            id.left(at).toStdString(), id.mid(at + 1).toStdString()));
      }
    }

    auto plaintext_msg_content_type_header_field =
        plaintext_msg_header->getField<vmime::contentTypeField>(
            vmime::fields::CONTENT_TYPE);
    plaintext_msg_content_type_header_field->setValue("text/plain");
    plaintext_msg_content_type_header_field->appendParameter(
        vmime::make_shared<vmime::parameter>("charset", "UTF-8"));
    plaintext_msg_content_type_header_field->appendParameter(
        vmime::make_shared<vmime::parameter>("format", "flowed"));

    auto plaintext_msg_content_trans_encode_field =
        plaintext_msg_header->getField(
            vmime::fields::CONTENT_TRANSFER_ENCODING);
    plaintext_msg_content_trans_encode_field->setValue("base64");

    auto plaintext_msg_body = plaintext_msg->getBody();

    auto mime_part_body_content =
        vmime::make_shared<vmime::stringContentHandler>();
    mime_part_body_content->setData(body_data.toStdString());
    plaintext_msg_body->setContents(mime_part_body_content);

    // With nothing attached the message stays exactly what it has always been:
    // a single text/plain part. Attachments turn it into multipart/mixed, with
    // that same text part as the first child.
    for (const auto& att : attachments) {
      auto content = vmime::make_shared<vmime::stringContentHandler>();
      content->setData(std::string(att.data.constData(),
                                   static_cast<size_t>(att.data.size())));

      const auto name =
          att.filename.isEmpty() ? QString("attachment") : att.filename;
      const auto type = att.mime_type.isEmpty()
                            ? QString("application/octet-stream")
                            : att.mime_type;

      auto attachment = vmime::make_shared<vmime::defaultAttachment>(
          content, vmime::encoding("base64"),
          vmime::mediaType(type.toStdString()),
          vmime::text(att.description.toStdString(), vmime::charsets::UTF_8));

      vmime::attachmentHelper::addAttachment(plaintext_msg, attachment);

      // addAttachment does not carry a filename of its own, so set it on the
      // part it just appended. Without this the recipient sees a nameless blob.
      auto body = plaintext_msg->getBody();
      auto part = body->getPartAt(body->getPartCount() - 1);
      auto disposition =
          part->getHeader()->getField<vmime::contentDispositionField>(
              vmime::fields::CONTENT_DISPOSITION);
      disposition->setValue("attachment");
      disposition->setFilename(
          vmime::word(name.toStdString(), vmime::charsets::UTF_8));
    }

    // Bytes, not text: the generated message is what will be signed or sent,
    // and a UTF-8 round trip here would not survive an 8bit part.
    eml_data = QByteArray::fromStdString(
        plaintext_msg->generate(vmime::lineLengthLimits::convenient));
    return 0;

  } catch (const vmime::exception& e) {
    eml_data = QByteArray("VMIME Error: ") + e.what();
    return -1;
  }
}

auto BuildPlainTextEML(const EMailMetaData& meta_data,
                       const QByteArray& body_data, QByteArray& eml_data)
    -> int {
  return BuildMimeEML(meta_data, body_data, {}, eml_data);
}

auto GetMetaData(QByteArray& data, EMailMetaData& meta_data) {}

auto GetEMLMetaData(vmime::shared_ptr<vmime::message>& message,
                    EMailMetaData& meta_data) -> int {
  auto header = message->getHeader();

  meta_data.from = ExtractFieldValueMailBox(header, vmime::fields::FROM);
  meta_data.to = ExtractFieldValueAddressListItems(header, vmime::fields::TO);
  meta_data.cc = ExtractFieldValueAddressListItems(header, vmime::fields::CC);
  meta_data.bcc_header =
      ExtractFieldValueAddressListItems(header, vmime::fields::BCC);
  meta_data.message_id = ExtractFieldValue(header, vmime::fields::MESSAGE_ID);
  meta_data.in_reply_to = ExtractFieldValue(header, vmime::fields::IN_REPLY_TO);
  const auto references =
      ExtractFieldValue(header, vmime::fields::REFERENCES).trimmed();
  if (!references.isEmpty()) {
    meta_data.references =
        references.split(QRegularExpression("\\s+"), Qt::SkipEmptyParts);
  }
  meta_data.subject = ExtractFieldValueText(header, vmime::fields::SUBJECT);
  meta_data.datetime = ExtractFieldValueDateTime(header, vmime::fields::DATE);
  // Reply-To is an address list, not a single mailbox (RFC 5322 allows several,
  // and vmime registers it as addressList). Reading it as a mailbox made
  // getValue<mailbox>() return null every time, so this field was silently
  // always empty -- which meant replies ignored it and nothing could notice a
  // Reply-To pointing somewhere unrelated.
  meta_data.reply_to =
      ExtractFieldValueAddressList(header, vmime::fields::REPLY_TO);
  meta_data.organization =
      ExtractFieldValueText(header, vmime::fields::ORGANIZATION);

  return 0;
}
namespace {

// Windows refuses these as filenames whatever extension follows, and a file
// named after one can be a nuisance on other systems too.
/// Extensions this program will hand to the desktop to open.
///
/// Documents, images, audio, video and plain text: file types whose registered
/// handler views them. Nothing that is itself a program, a script, a shortcut,
/// an installer or an archive -- an archive because what comes out of it is
/// unknown until it is out, and by then nobody has been asked anything. See
/// IsSafeToOpenAttachment for why this is an allow-list and not a blocklist.
/// How much of a subject becomes a file name. Long enough to tell two messages
/// apart, short enough that the dialog shows the whole thing.
constexpr int kSuggestedNameLength = 60;

const QStringList kOpenableExtensions = {
    // documents
    "pdf", "txt", "text", "log", "md", "markdown", "csv", "tsv", "rtf", "odt",
    "ods", "odp", "odg", "doc", "docx", "xls", "xlsx", "ppt", "pptx", "epub",
    // images
    "png", "jpg", "jpeg", "gif", "bmp", "tif", "tiff", "webp", "heic", "avif",
    "ico",
    // audio and video
    "mp3", "wav", "flac", "ogg", "oga", "opus", "m4a", "aac", "mp4", "m4v",
    "webm", "mkv", "mov", "avi",
    // Structured text that is read rather than run. Deliberately no "svg": it
    // carries script, and a viewer that runs that is a viewer running a
    // stranger's code.
    "json", "xml", "yaml", "yml", "ini", "conf", "eml", "ics", "vcf"};

const QStringList kReservedDeviceNames = {
    "CON",  "PRN",  "AUX",  "NUL",  "COM1", "COM2", "COM3", "COM4",
    "COM5", "COM6", "COM7", "COM8", "COM9", "LPT1", "LPT2", "LPT3",
    "LPT4", "LPT5", "LPT6", "LPT7", "LPT8", "LPT9"};

// Long enough for real filenames, short enough to leave room for a directory
// and a " (12)" suffix inside the usual 255-byte limit.
constexpr int kMaxFileNameLength = 180;

auto ExtensionForMimeType(const QString& mime_type) -> QString {
  static const QMap<QString, QString> kMap = {
      {"text/plain", "txt"},           {"text/html", "html"},
      {"application/pdf", "pdf"},      {"image/png", "png"},
      {"image/jpeg", "jpg"},           {"image/gif", "gif"},
      {"application/pgp-keys", "asc"}, {"application/zip", "zip"},
      {"message/rfc822", "eml"}};
  return kMap.value(mime_type.trimmed().toLower());
}

// Splits "name.ext" so a truncation or a " (2)" can be applied to the stem
// without destroying the extension.
auto SplitExtension(const QString& name) -> QPair<QString, QString> {
  const auto dot = name.lastIndexOf('.');
  if (dot <= 0 || dot == name.size() - 1) return {name, {}};
  return {name.left(dot), name.mid(dot)};
}

}  // namespace

auto SuggestedEMailFileName(const QString& subject) -> QString {
  const auto trimmed = subject.trimmed();
  if (trimmed.isEmpty()) return QStringLiteral("untitled.eml");

  // Capped before sanitizing rather than after: the sanitizer's own length cap
  // is a filesystem limit, and a name that long is unusable as a suggestion
  // long before it is illegal.
  auto name = SanitizeAttachmentFileName(trimmed.left(kSuggestedNameLength),
                                         QStringLiteral("message/rfc822"));

  // The sanitizer guarantees a non-empty name but not this extension -- a
  // subject that was entirely path separators comes back as "attachment".
  if (!name.endsWith(QStringLiteral(".eml"), Qt::CaseInsensitive)) {
    name += QStringLiteral(".eml");
  }
  return name;
}

auto IsSafeToOpenAttachment(const EMailAttachment& att) -> bool {
  // The name as it will actually reach the desktop, so the decision is made
  // about the same string the system will dispatch on.
  const auto name = SanitizeAttachmentFileName(att.filename, att.mime_type);

  const auto dot = name.lastIndexOf('.');
  // No extension at all: nothing to recognise, so nothing to allow. Most
  // desktops will not open it either, and the ones that guess are exactly the
  // case worth not relying on.
  if (dot < 0 || dot == name.size() - 1) return false;

  return kOpenableExtensions.contains(name.mid(dot + 1).toLower());
}

auto SanitizeAttachmentFileName(const QString& raw, const QString& mime_type)
    -> QString {
  QString name = raw;

  // Take only the last component, then drop any separator that survives. A
  // sender controls this string, so "../../etc/passwd" and "C:\\evil" both have
  // to come out as an ordinary name in the directory the user picked.
  name.replace('\\', '/');
  const auto slash = name.lastIndexOf('/');
  if (slash >= 0) name = name.mid(slash + 1);

  QString cleaned;
  cleaned.reserve(name.size());
  for (const auto ch : name) {
    // Control characters, NUL and the characters Windows forbids outright.
    if (ch.unicode() < 0x20 || ch == QChar(0x7F)) continue;
    if (ch == ':' || ch == '*' || ch == '?' || ch == '"' || ch == '<' ||
        ch == '>' || ch == '|' || ch == '/' || ch == '\\') {
      continue;
    }
    cleaned.append(ch);
  }

  // A name of nothing but dots and spaces (".", "..", "...") is not a name.
  while (cleaned.startsWith('.') || cleaned.startsWith(' '))
    cleaned.remove(0, 1);
  while (cleaned.endsWith('.') || cleaned.endsWith(' ')) cleaned.chop(1);
  cleaned = cleaned.trimmed();

  if (cleaned.isEmpty()) {
    const auto ext = ExtensionForMimeType(mime_type);
    cleaned = ext.isEmpty() ? QString("attachment")
                            : QString("attachment.%1").arg(ext);
  }

  auto [stem, ext] = SplitExtension(cleaned);

  // Reserved names are rejected on their stem, so "NUL.txt" is caught too.
  if (kReservedDeviceNames.contains(stem.toUpper())) {
    stem = QString("%1_").arg(stem);
  }

  if (stem.size() + ext.size() > kMaxFileNameLength) {
    stem = stem.left(qMax(1, kMaxFileNameLength - ext.size()));
  }

  return stem + ext;
}

auto UniqueAttachmentFileNames(const QList<EMailAttachment>& attachments,
                               const QStringList& taken) -> QStringList {
  QSet<QString> used;
  for (const auto& t : taken) used.insert(t.toLower());

  QStringList names;
  names.reserve(attachments.size());

  for (const auto& att : attachments) {
    const auto base = SanitizeAttachmentFileName(att.filename, att.mime_type);
    auto [stem, ext] = SplitExtension(base);

    auto candidate = base;
    int counter = 2;
    while (used.contains(candidate.toLower())) {
      candidate = QString("%1 (%2)%3").arg(stem).arg(counter++).arg(ext);
    }

    used.insert(candidate.toLower());
    names.append(candidate);
  }

  return names;
}

namespace {

auto PartContentType(const vmime::shared_ptr<const vmime::bodyPart>& part)
    -> QString {
  auto field = part->getHeader()->findField(vmime::fields::CONTENT_TYPE);
  if (!field) return "text/plain";  // the RFC 2045 default
  auto value = field->getValue<vmime::mediaType>();
  if (!value) return "text/plain";
  return Q_SC(value->generate()).trimmed().toLower();
}

auto PartDisposition(const vmime::shared_ptr<const vmime::bodyPart>& part)
    -> QString {
  auto field = part->getHeader()->findField(vmime::fields::CONTENT_DISPOSITION);
  if (!field) return {};
  auto value = field->getValue<vmime::contentDisposition>();
  if (!value) return {};
  return Q_SC(value->generate()).trimmed().toLower();
}

auto PartFileName(const vmime::shared_ptr<const vmime::bodyPart>& part)
    -> QString {
  // Content-Disposition filename= wins, with Content-Type name= as the older
  // fallback some senders still use.
  if (auto field =
          part->getHeader()->findField(vmime::fields::CONTENT_DISPOSITION)) {
    auto disp = vmime::dynamicCast<const vmime::contentDispositionField>(field);
    if (disp && disp->hasParameter("filename")) {
      return Q_SC(disp->getFilename().getBuffer());
    }
  }

  if (auto field = part->getHeader()->findField(vmime::fields::CONTENT_TYPE)) {
    auto ct = vmime::dynamicCast<const vmime::contentTypeField>(field);
    // findParameter, not getParameter: the latter is non-const and would
    // insert the parameter when it is missing.
    if (ct) {
      if (auto name = ct->findParameter("name")) {
        return Q_SC(name->getValue().getBuffer());
      }
    }
  }

  return {};
}

auto DecodePart(const vmime::shared_ptr<const vmime::bodyPart>& part)
    -> QByteArray {
  auto contents = part->getBody()->getContents();
  if (!contents) return {};

  std::ostringstream oss;
  vmime::utility::outputStreamAdapter osa(oss);
  contents->extract(osa);
  osa.flush();
  return QByteArray::fromStdString(oss.str());
}

// The OpenPGP control parts of RFC 3156 are structure, not content: showing
// them to the user as attachments would be noise.
auto IsProtocolPart(const QString& content_type) -> bool {
  return content_type == "application/pgp-encrypted" ||
         content_type == "application/pgp-signature";
}

}  // namespace

namespace {

// Parameter of a Content-Type field, without vmime's insert-on-read behaviour.
auto ContentTypeParam(const vmime::shared_ptr<const vmime::bodyPart>& part,
                      const std::string& name) -> QString {
  auto field = part->getHeader()->findField(vmime::fields::CONTENT_TYPE);
  if (!field) return {};
  auto ct = vmime::dynamicCast<const vmime::contentTypeField>(field);
  if (!ct) return {};
  auto param = ct->findParameter(name);
  if (!param) return {};
  return Q_SC(param->getValue().getBuffer()).trimmed();
}

auto PartTransferEncoding(const vmime::shared_ptr<const vmime::bodyPart>& part)
    -> QString {
  auto field =
      part->getHeader()->findField(vmime::fields::CONTENT_TRANSFER_ENCODING);
  if (!field) return {};
  auto value = field->getValue<vmime::encoding>();
  if (!value) return {};
  return Q_SC(value->generate()).trimmed().toLower();
}

// Content-ID without the angle brackets, which is the form an HTML "cid:"
// reference actually uses.
auto PartContentId(const vmime::shared_ptr<const vmime::bodyPart>& part)
    -> QString {
  auto field = part->getHeader()->findField("Content-Id");
  if (!field) return {};
  auto value = field->getValue();
  if (!value) return {};
  auto id = Q_SC(value->generate()).trimmed();
  if (id.startsWith('<') && id.endsWith('>')) id = id.mid(1, id.size() - 2);
  return id;
}

// Every header field of a part, decoded, in the order it was written. The
// undecoded truth stays available through RawHeaderBlock().
auto PartHeaderFields(const vmime::shared_ptr<const vmime::bodyPart>& part)
    -> QList<QPair<QString, QString>> {
  QList<QPair<QString, QString>> fields;
  for (const auto& field : part->getHeader()->getFieldList()) {
    auto name = Q_SC(field->getName());
    QString value;

    // Text fields carry RFC 2047 encoded-words; anything else is generated
    // as-is. getValue<text>() simply returns null when the field is not text.
    if (auto text_value = field->getValue<vmime::text>()) {
      value = Q_SC(text_value->getConvertedText(vmime::charsets::UTF_8));
    } else if (auto raw_value = field->getValue()) {
      value = Q_SC(raw_value->generate());
    }

    fields.append({name, value.trimmed()});
  }
  return fields;
}

struct TreeState {
  int parts_seen{0};
  qint64 bytes_seen{0};
  bool limit_hit{false};
  int next_index{0};
  int next_region{0};
  int next_alt_group{0};
};

void BuildNode(const vmime::shared_ptr<const vmime::bodyPart>& part,
               EMailPart& node, const EMailParseLimits& limits,
               TreeState& state, QList<EMailSignatureRegion>& regions,
               int depth, const QList<int>& covering, int alt_group) {
  if (state.limit_hit) return;

  if (depth > limits.max_depth || ++state.parts_seen > limits.max_parts) {
    state.limit_hit = true;
    MimeLog(QString("message tree exceeds parse limits at depth %1, part %2")
                .arg(depth)
                .arg(state.parts_seen));
    return;
  }

  node.index = state.next_index++;
  node.depth = depth;
  node.content_type = PartContentType(part);
  node.charset = ContentTypeParam(part, "charset");
  node.disposition = PartDisposition(part);
  node.filename = PartFileName(part);
  node.content_id = PartContentId(part);
  node.transfer_encoding = PartTransferEncoding(part);
  node.header_fields = PartHeaderFields(part);
  node.covered_by_regions = covering;
  node.alternative_group = alt_group;
  node.is_protocol_part = IsProtocolPart(node.content_type);
  node.is_openpgp_key = node.content_type == "application/pgp-keys";

  node.raw_offset = static_cast<qint64>(part->getParsedOffset());
  node.raw_length = static_cast<qint64>(part->getParsedLength());
  if (auto body = part->getBody()) {
    node.body_offset = static_cast<qint64>(body->getParsedOffset());
    node.body_length = static_cast<qint64>(body->getParsedLength());
  }

  const auto sub_parts = part->getBody()->getPartCount();
  node.is_multipart = sub_parts > 0;

  if (!node.is_multipart) {
    node.data = DecodePart(part);
    node.decoded_size = node.data.size();

    // Protocol parts stay out of the size budget: they are structure, and the
    // limit exists to bound the content a message can make us hold. Counting
    // them would also change the limit behaviour this walk has to preserve.
    if (!node.is_protocol_part) {
      state.bytes_seen += node.decoded_size;
      if (state.bytes_seen > limits.max_total_bytes) {
        state.limit_hit = true;
        MimeLog(QString("message exceeds the decoded size limit (%1 bytes)")
                    .arg(state.bytes_seen));
        return;
      }
    }
    return;
  }

  // A multipart/signed opens a region covering its FIRST part in full,
  // headers included -- that is the entity RFC 3156 signs, and the same slice
  // the verify path hashes. The signature part itself is not covered.
  auto child_covering = covering;
  const bool opens_region = node.content_type == "multipart/signed";
  int region_id = -1;
  int region_slot = -1;
  if (opens_region) {
    EMailSignatureRegion region;
    region.region_id = state.next_region++;
    region.nesting_depth = covering.size();
    region.declared_micalg = ContentTypeParam(part, "micalg");

    auto signed_entity = part->getBody()->getPartAt(0);
    region.raw_offset = static_cast<qint64>(signed_entity->getParsedOffset());
    region.raw_length = static_cast<qint64>(signed_entity->getParsedLength());

    region_id = region.region_id;
    region_slot = static_cast<int>(regions.size());
    regions.append(region);
  }

  // One alternative group per multipart/alternative, so siblings can be told
  // apart from alternatives further down the tree.
  const bool opens_alternative = node.content_type == "multipart/alternative";
  const int child_alt_group =
      opens_alternative ? state.next_alt_group++ : alt_group;

  for (size_t i = 0; i < sub_parts; ++i) {
    EMailPart child;
    auto covering_for_child = child_covering;
    if (opens_region && i == 0) covering_for_child.append(region_id);

    BuildNode(part->getBody()->getPartAt(i), child, limits, state, regions,
              depth + 1, covering_for_child, child_alt_group);
    if (state.limit_hit) return;

    if (region_slot >= 0) {
      // Note where the two halves of this region ended up, so a nested region
      // can later be verified on its own bytes rather than the outermost
      // ones. The signature is the first pgp-signature part beside the entity.
      if (i == 0) {
        regions[region_slot].signed_part_index = child.index;
        // Encrypt-then-sign: the entity under this signature is the encrypted
        // blob, so the signature authenticates ciphertext and not the message
        // that will be read. Recorded here, where the shape is still visible.
        regions[region_slot].covers_ciphertext_only =
            child.content_type == "multipart/encrypted";
      } else if (regions[region_slot].signature_part_index < 0 &&
                 child.content_type == "application/pgp-signature") {
        regions[region_slot].signature_part_index = child.index;
      }
    }

    node.children.append(child);
  }
}

// Body candidacy, matching the rule ExtractParts() has always used.
auto IsBodyCandidate(const EMailPart& part) -> bool {
  return !part.is_multipart && !part.is_protocol_part &&
         part.filename.isEmpty() &&
         !part.disposition.startsWith("attachment") &&
         (part.content_type == "text/plain" ||
          part.content_type == "text/html");
}

void CollectFlat(const EMailPart& node, QList<const EMailPart*>& out) {
  out.append(&node);
  for (const auto& child : node.children) CollectFlat(child, out);
}

}  // namespace

auto ParseMimeTree(const vmime::shared_ptr<vmime::message>& message,
                   const QByteArray& raw, EMailPart& root,
                   QList<EMailSignatureRegion>& regions,
                   const EMailParseLimits& limits) -> int {
  Q_UNUSED(raw)
  if (!message) return -1;

  TreeState state;
  try {
    BuildNode(message, root, limits, state, regions, 0, {}, -1);
  } catch (const vmime::exception& e) {
    MimeLog(QString("error parsing message tree: %1").arg(e.what()));
    return -1;
  }

  return state.limit_hit ? -1 : 0;
}

auto FlattenMimeTree(const EMailPart& root) -> QList<const EMailPart*> {
  QList<const EMailPart*> out;
  CollectFlat(root, out);
  return out;
}

auto RawHeaderBlock(const EMailPart& part, const QByteArray& raw)
    -> QByteArray {
  if (part.raw_offset < 0 || part.body_offset < 0) return {};
  if (part.body_offset < part.raw_offset) return {};
  if (part.body_offset > raw.size()) return {};
  return raw.mid(static_cast<int>(part.raw_offset),
                 static_cast<int>(part.body_offset - part.raw_offset));
}

namespace {

/// Whether @p line continues the header field above it (RFC 5322 folding).
auto IsFoldedContinuation(const QByteArray& line) -> bool {
  return !line.isEmpty() && (line.at(0) == ' ' || line.at(0) == '\t');
}

/// Splits @p block into lines, keeping each line's own terminator with it so a
/// caller can rebuild the block byte-for-byte. CR LF and bare LF both end a
/// line; neither is rewritten as the other.
auto SplitKeepingEndings(const QByteArray& block) -> QList<QByteArray> {
  QList<QByteArray> lines;
  int start = 0;
  for (int i = 0; i < block.size(); ++i) {
    if (block.at(i) != '\n') continue;
    lines.append(block.mid(start, i - start + 1));
    start = i + 1;
  }
  if (start < block.size()) lines.append(block.mid(start));
  return lines;
}

/// @p line without its trailing CR LF or LF.
auto WithoutEnding(const QByteArray& line) -> QByteArray {
  auto out = line;
  while (!out.isEmpty() && (out.endsWith('\n') || out.endsWith('\r'))) {
    out.chop(1);
  }
  return out;
}

}  // namespace

auto SplitRawHeaderFields(const QByteArray& block)
    -> QList<EMailRawHeaderField> {
  QList<EMailRawHeaderField> fields;

  for (const auto& line : SplitKeepingEndings(block)) {
    const auto bare = WithoutEnding(line);

    // The blank line that ends the header block. Nothing after it belongs to
    // the headers, and a block handed in with its terminator still attached
    // must not grow an empty trailing field.
    if (bare.isEmpty()) break;

    if (IsFoldedContinuation(line) && !fields.isEmpty()) {
      // Joined with the fold left in: the value is being shown to someone
      // looking for exactly this kind of detail, so it is not re-wrapped or
      // collapsed.
      fields.back().value += bare;
      fields.back().raw_line += line;
      continue;
    }

    const int colon = bare.indexOf(':');
    EMailRawHeaderField field;
    field.raw_line = line;
    if (colon <= 0) {
      // Neither a continuation nor `name: value`. Kept, nameless, because a
      // malformed header is worth seeing rather than silently dropping.
      field.value = bare;
    } else {
      field.name = QString::fromLatin1(bare.left(colon)).trimmed();
      field.value = bare.mid(colon + 1).trimmed();
    }
    fields.append(field);
  }

  return fields;
}

auto UnwrapProtectedLayer(const EMailPart& root, const QByteArray& raw,
                          QByteArray& out_eml) -> EMailUnwrapResult {
  // The outermost signature is the one being removed, and it is not
  // necessarily the message itself: the ordinary shape of a signed mail with
  // an attachment is multipart/mixed wrapping the multipart/signed.
  const auto flat = FlattenMimeTree(root);

  const EMailPart* signed_part = nullptr;
  bool has_encrypted = false;
  for (const auto* part : flat) {
    if (part->content_type == "multipart/encrypted") has_encrypted = true;
    if (signed_part == nullptr && part->content_type == "multipart/signed") {
      signed_part = part;
    }
  }

  if (signed_part == nullptr) {
    // A signature may well exist inside the ciphertext, but it is not visible
    // from here and guessing would be worse than saying so.
    return has_encrypted ? EMailUnwrapResult::kNOT_SUPPORTED
                         : EMailUnwrapResult::kNOT_PROTECTED;
  }

  // RFC 3156: exactly the signed entity followed by its detached signature.
  if (signed_part->children.size() != 2) return EMailUnwrapResult::kMALFORMED;

  const auto& entity = signed_part->children.at(0);
  const auto& signature = signed_part->children.at(1);
  if (signature.content_type != "application/pgp-signature") {
    return EMailUnwrapResult::kMALFORMED;
  }
  if (entity.raw_offset < 0 || entity.raw_length <= 0 ||
      entity.raw_offset + entity.raw_length > raw.size()) {
    return EMailUnwrapResult::kMALFORMED;
  }

  const auto entity_bytes = raw.mid(static_cast<int>(entity.raw_offset),
                                    static_cast<int>(entity.raw_length));

  if (signed_part != &root) {
    // A nested part is headers plus body, and so is the entity inside it, so
    // the entity simply takes its place: the entity's own Content-* fields
    // become the part's, which is exactly what the wrapper was describing.
    if (signed_part->raw_offset < 0 || signed_part->raw_length <= 0 ||
        signed_part->raw_offset + signed_part->raw_length > raw.size()) {
      return EMailUnwrapResult::kMALFORMED;
    }

    const auto tail_offset = signed_part->raw_offset + signed_part->raw_length;
    out_eml = raw.left(static_cast<int>(signed_part->raw_offset)) +
              entity_bytes + raw.mid(static_cast<int>(tail_offset));
    return EMailUnwrapResult::kOK;
  }

  // At the top level the part's headers are the MESSAGE's headers. They say
  // who sent this and about what, and have to survive; only the fields
  // describing the wrapper are dropped.
  const auto outer = RawHeaderBlock(*signed_part, raw);
  if (outer.isEmpty()) return EMailUnwrapResult::kMALFORMED;

  // Dropping a field means dropping its folded continuation lines with it --
  // a surviving `boundary=` line would reparse as garbage.
  QByteArray kept;
  bool dropping = false;
  for (const auto& line : SplitKeepingEndings(outer)) {
    if (WithoutEnding(line).isEmpty()) break;  // the header terminator

    if (IsFoldedContinuation(line)) {
      if (!dropping) kept += line;
      continue;
    }

    const auto bare = WithoutEnding(line);
    const int colon = bare.indexOf(':');
    const auto name = colon > 0 ? bare.left(colon) : bare;
    dropping = name.toLower().startsWith("content-");
    if (!dropping) kept += line;
  }

  if (kept.isEmpty()) return EMailUnwrapResult::kMALFORMED;

  // The entity brings its own headers, so the two blocks are simply adjacent:
  // the outer block's terminating blank line was never copied above.
  out_eml = kept + entity_bytes;
  return EMailUnwrapResult::kOK;
}

auto SelectBodyPart(const EMailPart& root, bool prefer_html)
    -> const EMailPart* {
  const auto flat = FlattenMimeTree(root);

  // Within one multipart/alternative the members are the same content in
  // different forms, so exactly one of them is the body. Picking whichever is
  // walked first -- what the flat rule does -- lands on an arbitrary member
  // and hides the rest.
  const QString wanted = prefer_html ? "text/html" : "text/plain";
  const QString fallback = prefer_html ? "text/plain" : "text/html";

  const EMailPart* best = nullptr;
  const EMailPart* best_fallback = nullptr;

  for (const auto* part : flat) {
    if (!IsBodyCandidate(*part)) continue;
    if (part->content_type == wanted) {
      if (best == nullptr) best = part;
    } else if (part->content_type == fallback) {
      if (best_fallback == nullptr) best_fallback = part;
    }
  }

  return best != nullptr ? best : best_fallback;
}

auto ClassifyOpenPGPStructure(const EMailPart& root,
                              const QList<EMailSignatureRegion>& regions)
    -> EMailSecurityState {
  const auto flat = FlattenMimeTree(root);

  bool has_signed = false;
  bool has_encrypted = false;
  bool malformed = false;

  for (const auto* part : flat) {
    if (part->content_type == "multipart/signed") {
      has_signed = true;
      // RFC 3156 is exact about this: two parts, and a signature part.
      if (part->children.size() != 2) malformed = true;
    } else if (part->content_type == "multipart/encrypted") {
      has_encrypted = true;
      if (part->children.size() != 2) malformed = true;
    }
  }

  // A signed subtree that produced no region means the structure did not hold
  // up well enough to say what is being signed.
  if (has_signed && regions.isEmpty()) malformed = true;

  if (malformed) return EMailSecurityState::kMALFORMED_PGP;
  if (has_signed && has_encrypted) return EMailSecurityState::kSIGNED_ENCRYPTED;
  if (has_encrypted) return EMailSecurityState::kENCRYPTED;
  if (has_signed) return EMailSecurityState::kSIGNED;
  return EMailSecurityState::kPLAIN;
}

auto CheckBeforeExport(const QByteArray& source) -> EMailExportCheck {
  EMailExportCheck check;

  vmime::shared_ptr<vmime::message> parsed;
  if (!CheckIfEMLMessage(source, parsed)) return check;

  EMailPart root;
  QList<EMailSignatureRegion> regions;
  if (ParseMimeTree(parsed, source, root, regions) != 0) return check;

  EMailMetaData meta;
  GetEMLMetaData(parsed, meta);

  for (const auto& finding : PreflightMessage(meta, root, regions, {})) {
    const auto line = QString("%1 - %2").arg(finding.title, finding.detail);
    if (finding.level == EMailFindingLevel::kRISK) {
      check.risks.append(line);
    } else if (finding.level == EMailFindingLevel::kWARN) {
      check.notes.append(line);
    }
  }

  // Notes alone are not worth interrupting a save for; the Security tab
  // already carries them.
  check.needs_confirmation = !check.risks.isEmpty();
  return check;
}

namespace {

/// Mirrors GpgSigValidity, grouped by what it means for the person reading.
///
/// 1 is the one worth naming. GpgVerifyResultAnalyse sets kVALID_WITH_ISSUES
/// exactly when gpgme reports GPGME_SIGSUM_RED, which means the signature is
/// BAD -- and 0 is kFULLY_VALID, 2 is kVALID_NOT_FULLY_TRUSTED, so 1 sits
/// between two genuine outcomes and reads like a near miss. It has been taken
/// for one twice: once by a `validity <= 2` test that painted it like a clean
/// signature, and again by the label "valid, with issues" beside it. The
/// grouping here is the single answer both of those needed.
auto BadgeForValidity(int validity) -> EMailBadgeState {
  switch (validity) {
    case 0:  // fully valid
    case 2:  // valid, key not fully trusted
      return EMailBadgeState::kSIGNED_GOOD;
    case 4:  // public key missing
      return EMailBadgeState::kSIGNED_UNKNOWN_KEY;
    case 6:  // signature expired
    case 7:  // signing key expired
      return EMailBadgeState::kSIGNED_EXPIRED;
    case 1:  // GPGME_SIGSUM_RED -- a bad signature
    case 3:  // invalid
    case 5:  // signing key revoked
      return EMailBadgeState::kSIGNED_BAD;
    default:
      // -1 is the default in EMailSignatureResult, chosen so that an absent
      // or unparseable field does not read as fully valid. Anything else is
      // a value this build does not know, which is not a reason to reassure.
      return EMailBadgeState::kSIGNED_UNKNOWN_KEY;
  }
}

}  // namespace

auto SignerMatchesAddress(const EMailSignatureResult& result,
                          const QString& address) -> bool {
  const auto wanted = AddressOfUid(address);
  const auto signer = AddressOfUid(result.uid);

  // Nothing to disagree with. A result carrying no UID is the unknown-key
  // case, which the badge already reports on its own; calling it a mismatch
  // as well would say the same thing twice and in stronger words.
  if (wanted.isEmpty() || signer.isEmpty()) return true;

  return wanted.compare(signer, Qt::CaseInsensitive) == 0;
}

auto BadgeForSignature(const EMailSignatureResult& result, const QString& from)
    -> EMailBadgeState {
  const auto by_validity = BadgeForValidity(result.validity);
  if (by_validity != EMailBadgeState::kSIGNED_GOOD) return by_validity;

  // Only now. Identity is a question about a signature that verified; asking
  // it of a broken one answers nothing and buries the real problem.
  return SignerMatchesAddress(result, from) ? EMailBadgeState::kSIGNED_GOOD
                                            : EMailBadgeState::kSIGNED_MISMATCH;
}

auto DeriveSecurityBadge(EMailSecurityState structure,
                         EMailVerifyState verify_state,
                         const QList<EMailSignatureResult>& results,
                         const QString& from) -> EMailBadgeState {
  switch (structure) {
    case EMailSecurityState::kMALFORMED_PGP:
      return EMailBadgeState::kMALFORMED;
    case EMailSecurityState::kPLAIN:
      return EMailBadgeState::kNOT_PROTECTED;
    case EMailSecurityState::kENCRYPTED:
      return EMailBadgeState::kENCRYPTED_ONLY;
    case EMailSecurityState::kSIGNED:
    case EMailSecurityState::kSIGNED_ENCRYPTED:
      break;
  }

  // Structurally signed. Everything below turns on what verification found,
  // and until it has run the honest answer is that nobody has looked.
  if (verify_state != EMailVerifyState::kVERIFIED || results.isEmpty()) {
    return EMailBadgeState::kSIGNED_UNVERIFIED;
  }

  // The worst signature decides. A message carrying one good signature and
  // one bad one is not a good message, and reporting the best of them is how
  // an attacker gets a second attempt.
  auto worst = EMailBadgeState::kSIGNED_GOOD;
  for (const auto& result : results) {
    worst = std::max(worst, BadgeForSignature(result, from));
  }
  return worst;
}

auto AggregateVerification(EMailSecurityState structure,
                           const QList<EMailSignatureRegion>& regions,
                           const QList<EMailRegionVerdict>& verdicts)
    -> EMailBadgeState {
  // Nothing to combine. What the message IS decides, which is the structural
  // question DeriveSecurityBadge() answers for an unverified message too.
  if (regions.isEmpty()) {
    switch (structure) {
      case EMailSecurityState::kPLAIN:
        return EMailBadgeState::kNOT_PROTECTED;
      case EMailSecurityState::kENCRYPTED:
        return EMailBadgeState::kENCRYPTED_ONLY;
      case EMailSecurityState::kMALFORMED_PGP:
      case EMailSecurityState::kSIGNED:
      case EMailSecurityState::kSIGNED_ENCRYPTED:
        // It said it was signed and not one region resolved. That is the
        // structure failing to hold up, not a message with no signature.
        return EMailBadgeState::kMALFORMED;
    }
    return EMailBadgeState::kNOT_PROTECTED;
  }

  // The candidate depth: the shallowest region that covers something other
  // than ciphertext. Anything deeper is content this message CARRIES rather
  // than content it vouches for.
  //
  // Taken from the REGIONS, which are what the message structurally has, not
  // from the verdicts, which are only what was looked at. Reading it off the
  // verdicts made a message nothing had verified yet indistinguishable from
  // one whose every signature was over ciphertext.
  auto candidate_depth = std::numeric_limits<int>::max();
  for (const auto& region : regions) {
    if (region.covers_ciphertext_only) continue;
    candidate_depth = std::min(candidate_depth, region.nesting_depth);
  }

  if (candidate_depth == std::numeric_limits<int>::max()) {
    // Every signature is over ciphertext. Nothing signed the plaintext the
    // user ends up reading, and saying "signed" here would claim otherwise.
    // The signatures are still reported; they are just not this.
    return EMailBadgeState::kENCRYPTED_ONLY;
  }

  // Worst-wins across the candidate set. Siblings at that depth are combined;
  // nested regions are not, and neither are the ciphertext-only ones.
  auto worst = EMailBadgeState::kSIGNED_GOOD;
  auto any = false;
  for (const auto& verdict : verdicts) {
    if (verdict.covers_ciphertext_only) continue;
    if (verdict.nesting_depth != candidate_depth) continue;
    worst = std::max(worst, verdict.verdict);
    any = true;
  }

  // A candidate region exists in the parse but nothing looked at it -- the
  // walk stopped on its ceiling, say. Nobody has checked, and that is the
  // honest word for it.
  return any ? worst : EMailBadgeState::kSIGNED_UNVERIFIED;
}

auto VerificationMatchesSource(const EMailVerificationResult& result,
                               const QByteArray& source) -> bool {
  // A result that recorded nothing cannot vouch for anything. Treated as a
  // mismatch on purpose: the alternative is accepting an unanchored verdict.
  if (result.source_sha256.isEmpty()) return false;
  if (result.source_length != static_cast<qint64>(source.size())) return false;

  return result.source_sha256 ==
         QCryptographicHash::hash(source, QCryptographicHash::Sha256);
}

auto ToneForBadge(EMailBadgeState state) -> EMailBadgeTone {
  switch (state) {
    case EMailBadgeState::kSIGNED_GOOD:
      return EMailBadgeTone::kGOOD;

    case EMailBadgeState::kSIGNED_BAD:
    case EMailBadgeState::kMALFORMED:
      return EMailBadgeTone::kDANGER;

    case EMailBadgeState::kSIGNED_MISMATCH:
    case EMailBadgeState::kSIGNED_EXPIRED:
    case EMailBadgeState::kSIGNED_UNKNOWN_KEY:
    case EMailBadgeState::kSIGNED_UNVERIFIED:
      return EMailBadgeTone::kWARN;

    case EMailBadgeState::kSIGNED_ERROR:
      // A check that could not run is not evidence of anything, so it is not
      // drawn as one. It is louder than "not checked yet" only in wording.
      return EMailBadgeTone::kWARN;

    case EMailBadgeState::kNOT_PROTECTED:
    case EMailBadgeState::kENCRYPTED_ONLY:
      // Neither is a fault. An unprotected message is the ordinary case, and
      // painting it as a warning trains people to ignore the warnings that
      // matter. Encryption with no signature is exactly what it says.
      return EMailBadgeTone::kMUTED;
  }
  return EMailBadgeTone::kMUTED;
}

auto PlanVerifyAfterDecrypt(const QByteArray& plaintext)
    -> EMailPostDecryptPlan {
  if (plaintext.trimmed().isEmpty()) return EMailPostDecryptPlan::kUNREADABLE;

  vmime::shared_ptr<vmime::message> message;
  if (!CheckIfEMLMessage(plaintext, message) || !message) {
    return EMailPostDecryptPlan::kUNREADABLE;
  }

  // The outermost layer and nothing else. VerifyEMLData() refuses anything
  // whose top-level content type is not multipart/signed, so that single
  // question decides whether handing it over can produce an answer at all.
  //
  // A message that CLAIMS multipart/signed and is malformed in some other way
  // -- a missing protocol parameter, an unusable micalg -- still goes to the
  // verify on purpose: the refusal it returns names what is wrong, and that
  // diagnosis is worth more than silently calling the message unsigned.
  if (PartContentType(message) == "multipart/signed") {
    return EMailPostDecryptPlan::kVERIFY;
  }

  return EMailPostDecryptPlan::kNOT_SIGNED;
}

void MergeVerifiedMetaData(EMailMetaData& decrypted,
                           const EMailMetaData& verified) {
  // Only the verify produces these; there is nothing on the decrypt side to
  // weigh them against.
  decrypted.micalg = verified.micalg;
  decrypted.signed_entity_digest = verified.signed_entity_digest;
  decrypted.signed_entity_digest_algo = verified.signed_entity_digest_algo;
  decrypted.signed_entity_non_canonical = verified.signed_entity_non_canonical;
  decrypted.signature_regions = verified.signature_regions;
  if (!verified.public_keys.isEmpty()) {
    decrypted.public_keys = verified.public_keys;
  }

  // Headers prefer the verified side where it has them: those are the headers
  // that travelled inside the ciphertext, and they are the ones an outer
  // envelope cannot have rewritten.
  const auto prefer = [](QString& into, const QString& from) {
    if (!from.isEmpty()) into = from;
  };
  const auto prefer_list = [](QStringList& into, const QStringList& from) {
    if (!from.isEmpty()) into = from;
  };

  prefer(decrypted.from, verified.from);
  prefer(decrypted.subject, verified.subject);
  prefer(decrypted.reply_to, verified.reply_to);
  prefer(decrypted.organization, verified.organization);
  prefer(decrypted.message_id, verified.message_id);
  prefer(decrypted.in_reply_to, verified.in_reply_to);
  prefer_list(decrypted.to, verified.to);
  prefer_list(decrypted.cc, verified.cc);
  prefer_list(decrypted.bcc_header, verified.bcc_header);
  prefer_list(decrypted.references, verified.references);
  if (verified.datetime.isValid()) decrypted.datetime = verified.datetime;

  // Content is NEVER concatenated. Both sides walked the same plaintext, so
  // appending one to the other is what listed every attachment twice; the
  // verified side is consulted only where the decrypt found nothing at all.
  if (decrypted.body.isEmpty() && !verified.body.isEmpty()) {
    decrypted.body = verified.body;
    decrypted.body_content_type = verified.body_content_type;
  }
  if (decrypted.attachments.isEmpty() && !verified.attachments.isEmpty()) {
    decrypted.attachments = verified.attachments;
  }
}

auto AddressOfUid(const QString& uid) -> QString {
  QString name;
  QString email;
  if (ParseEmailString(uid, name, email)) return email.trimmed().toLower();

  // A bare address with no display name never reaches the angle-bracket form.
  const auto trimmed = uid.trimmed();
  if (trimmed.contains('@') && !trimmed.contains(' ')) return trimmed.toLower();
  return {};
}

namespace {

auto InfoRoot(const QByteArray& info_json) -> QJsonObject {
  const auto doc = QJsonDocument::fromJson(info_json);
  return doc.isObject() ? doc.object() : QJsonObject{};
}

// A recipient the sender asked the engine not to name. GnuPG reports this as
// an all-zero key id; it means "deliberately anonymous", not "unknown".
auto IsHiddenKeyId(const QString& key_id) -> bool {
  if (key_id.isEmpty()) return false;
  for (const auto c : key_id) {
    if (c != '0') return false;
  }
  return true;
}

}  // namespace

auto ParseSignatureResults(const QByteArray& info_json, int region_id)
    -> QList<EMailSignatureResult> {
  QList<EMailSignatureResult> results;

  const auto signatures = InfoRoot(info_json).value("signatures").toArray();
  for (const auto& entry : signatures) {
    const auto obj = entry.toObject();

    EMailSignatureResult result;
    // Stamped from the call site, which knows which region's bytes were just
    // verified. Nothing here depends on the order results arrive in.
    result.region_id = region_id;
    result.fingerprint = obj.value("fingerprint").toString();
    result.pubkey_algo = obj.value("pubkeyAlgo").toString();
    result.hash_algo = obj.value("hashAlgo").toString();
    result.uid = obj.value("uid").toString();
    // -1, not 0: QJsonValue::toInt() yields 0 for a missing or non-numeric
    // field, and 0 is kFULLY_VALID. Absence must not read as a good signature.
    result.validity = obj.value("validity").toInt(-1);

    const auto sign_time = obj.value("signTime").toString();
    if (!sign_time.isEmpty()) {
      result.sign_time = QDateTime::fromString(sign_time, Qt::ISODate);
    }

    for (const auto& warning : obj.value("warnings").toArray()) {
      result.warnings.append(warning.toString());
    }

    results.append(result);
  }

  return results;
}

auto ParseRecipientInfos(const QByteArray& info_json)
    -> QList<EMailRecipientInfo> {
  QList<EMailRecipientInfo> recipients;

  const auto entries = InfoRoot(info_json).value("recipients").toArray();
  for (const auto& entry : entries) {
    const auto obj = entry.toObject();

    EMailRecipientInfo info;
    info.uid = obj.value("uid").toString();
    info.fingerprint = obj.value("fingerprint").toString();
    info.key_id = obj.value("keyId").toString();
    info.pubkey_algo = obj.value("pubkeyAlgo").toString();
    info.key_found = obj.value("keyFound").toBool();
    info.algo_is_primary_key = obj.value("algoIsPrimaryKey").toBool();
    info.hidden = IsHiddenKeyId(info.key_id);

    recipients.append(info);
  }

  return recipients;
}

void CheckMicalgAgreement(QList<EMailSignatureResult>& results,
                          const EMailSignatureRegion& region) {
  // "pgp-sha256" in the header against "SHA256" from the signature: compare
  // the part that carries the meaning, not the spelling.
  auto normalize = [](QString value) {
    value = value.trimmed().toLower();
    if (value.startsWith("pgp-")) value = value.mid(4);
    value.remove('-');
    return value;
  };

  const auto declared = normalize(region.declared_micalg);
  if (declared.isEmpty()) return;

  for (auto& result : results) {
    if (result.region_id != region.region_id) continue;
    const auto actual = normalize(result.hash_algo);
    if (actual.isEmpty()) continue;
    result.micalg_mismatch = actual != declared;
  }
}

auto MatchRecipients(const QStringList& to, const QStringList& cc,
                     const QStringList& bcc,
                     const QList<EMailRecipientInfo>& encrypted)
    -> QList<EMailRecipientRow> {
  QList<EMailRecipientRow> rows;
  QList<bool> claimed;
  claimed.reserve(encrypted.size());
  for (int i = 0; i < encrypted.size(); ++i) claimed.append(false);

  const auto match_address = [&](const QString& address, const QString& field) {
    const auto wanted = AddressOfUid(address);

    EMailRecipientRow row;
    row.address = address;
    row.header_field = field;

    for (int i = 0; i < encrypted.size(); ++i) {
      if (claimed[i]) continue;
      // Only a recipient whose key is actually in the keyring can be tied to
      // an address; an unknown key id names nobody.
      if (!encrypted[i].key_found) continue;
      if (wanted.isEmpty()) continue;
      if (AddressOfUid(encrypted[i].uid) != wanted) continue;

      claimed[i] = true;
      row.match = RecipientMatch::kMATCHED;
      row.info = encrypted[i];
      rows.append(row);
      return;
    }

    // Addressed but not encrypted to: they were told this message is for them
    // and they cannot open it. This is the one case that warrants a warning.
    row.match = RecipientMatch::kADDRESSED_NOT_ENCRYPTED;
    rows.append(row);
  };

  for (const auto& address : to) match_address(address, "To");
  for (const auto& address : cc) match_address(address, "Cc");
  for (const auto& address : bcc) match_address(address, "Bcc");

  for (int i = 0; i < encrypted.size(); ++i) {
    if (claimed[i]) continue;

    EMailRecipientRow row;
    row.info = encrypted[i];
    // Not a finding. A message is routinely encrypted to keys that never
    // appear in a header, and a withheld key id is a choice the sender made
    // rather than something going wrong.
    row.match = encrypted[i].hidden ? RecipientMatch::kHIDDEN_RECIPIENT
                                    : RecipientMatch::kENCRYPTED_NOT_ADDRESSED;
    rows.append(row);
  }

  return rows;
}

namespace {

// "Re: " / "Fwd: " only once, however many times a message has been round the
// houses. Stacking prefixes is the classic way a subject line becomes
// unreadable after three exchanges.
auto PrefixSubject(const QString& subject, const QString& prefix) -> QString {
  const auto trimmed = subject.trimmed();
  if (trimmed.startsWith(prefix, Qt::CaseInsensitive)) return trimmed;
  return prefix + trimmed;
}

auto NormalizedAddresses(const QStringList& addresses) -> QStringList {
  QStringList out;
  for (const auto& address : addresses) {
    const auto trimmed = address.trimmed();
    if (!trimmed.isEmpty()) out.append(trimmed);
  }
  return out;
}

// Deduplicate by address rather than by display name: the same person may
// appear once bare and once with a name, and they should be addressed once.
auto WithoutDuplicatesOrSelf(const QStringList& addresses,
                             const QString& self_address, QStringList& seen)
    -> QStringList {
  const auto self = AddressOfUid(self_address);
  QStringList out;
  for (const auto& address : addresses) {
    const auto key = AddressOfUid(address);
    if (key.isEmpty()) continue;
    if (!self.isEmpty() && key == self) continue;
    if (seen.contains(key)) continue;
    seen.append(key);
    out.append(address);
  }
  return out;
}

}  // namespace

void BuildDerivedMetaData(const EMailMetaData& source, EMailReplyMode mode,
                          const QString& self_address, EMailMetaData& out) {
  out = EMailMetaData{};

  // The user's own identity on the new message, as far as it can be known
  // from the old one: whoever the original was addressed to.
  out.from = self_address;

  if (mode == EMailReplyMode::kFORWARD) {
    out.subject = PrefixSubject(source.subject, "Fwd: ");
    // No recipients: a forward is addressed by the person forwarding it, and
    // guessing here is how mail goes to the wrong people.
    out.attachments = source.attachments;
  } else {
    out.subject = PrefixSubject(source.subject, "Re: ");

    // Reply-To exists precisely so the sender can redirect answers; honouring
    // From instead would send the reply somewhere the sender asked it not to
    // go.
    const auto reply_target =
        source.reply_to.trimmed().isEmpty() ? source.from : source.reply_to;

    QStringList seen;
    out.to = WithoutDuplicatesOrSelf(NormalizedAddresses({reply_target}),
                                     self_address, seen);

    if (mode == EMailReplyMode::kREPLY_ALL) {
      // Everyone who was visibly addressed, minus whoever is already in To and
      // minus the user. Bcc is deliberately not carried: those recipients were
      // blind, and a reply-all that exposes them defeats the original choice.
      auto others = NormalizedAddresses(source.to);
      others += NormalizedAddresses(source.cc);
      out.cc = WithoutDuplicatesOrSelf(others, self_address, seen);
    }
  }

  // Threading headers, so the answer attaches to what it answers. References
  // accumulates the chain and In-Reply-To names the immediate parent.
  if (!source.message_id.isEmpty()) {
    out.in_reply_to = source.message_id;
    out.references = source.references;
    if (!out.references.contains(source.message_id)) {
      out.references.append(source.message_id);
    }
  }
}

auto FindEncryptedCiphertextPart(const EMailPart& root) -> const EMailPart* {
  // RFC 3156 §4: multipart/encrypted with exactly two parts, the control part
  // first and the ciphertext second. The same shape DecryptEMLData insists on
  // before it hands anything to the engine -- anything looser is not a message
  // whose recipients can be read with any confidence.
  if (root.content_type == "multipart/encrypted" && root.children.size() == 2) {
    const auto& body = root.children.at(1);
    if (body.content_type == "application/octet-stream" &&
        !body.data.isEmpty()) {
      return &body;
    }
  }

  for (const auto& child : root.children) {
    if (const auto* found = FindEncryptedCiphertextPart(child)) return found;
  }

  return nullptr;
}

auto DescribeDecryptCapability(const QList<EMailEncRecipient>& recipients)
    -> EMailDecryptCapability {
  EMailDecryptCapability out;

  // Nothing was readable out of the message. Saying "you cannot open this"
  // here would be inventing an answer to a question that was never asked.
  if (recipients.isEmpty()) return out;

  for (const auto& recipient : recipients) {
    if (recipient.hidden) {
      out.has_hidden = true;
      continue;
    }
    if (!recipient.has_secret) continue;

    // Name the key the way the user would recognise it, falling back to the
    // identifier the message itself carried.
    out.holding.append(recipient.uid.isEmpty() ? recipient.key_id
                                               : recipient.uid);
  }

  if (!out.holding.isEmpty()) {
    out.holding.removeDuplicates();
    out.verdict = EMailDecryptVerdict::kCAN_OPEN;
    return out;
  }

  // A withheld recipient may perfectly well be this user, so a message
  // carrying one is never reported as unopenable on the strength of the
  // recipients that could be read.
  out.verdict = out.has_hidden ? EMailDecryptVerdict::kUNKNOWN
                               : EMailDecryptVerdict::kCANNOT_OPEN;
  return out;
}

auto TabTitleForSubject(const QString& subject) -> QString {
  auto title = subject;

  // A Subject is a header field: it may arrive folded across lines, and a
  // decoded encoded-word can carry control characters. Either would be drawn
  // into the tab bar verbatim.
  title.replace(QRegularExpression("\\s+"), " ");
  title.remove(QRegularExpression("[\\x00-\\x1F\\x7F]"));
  title = title.trimmed();

  // The tab bar spells "unsaved" with a leading asterisk, so a subject that
  // starts with one would claim the tab is dirty before anything is typed.
  while (title.startsWith('*')) {
    title.remove(0, 1);
    title = title.trimmed();
  }

  if (title.isEmpty()) return {};

  // Long enough to tell two replies apart, short enough not to push every
  // other tab off the bar.
  constexpr int kMaxTabTitleChars = 40;
  if (title.size() > kMaxTabTitleChars) {
    title =
        title.left(kMaxTabTitleChars - 1).trimmed() + QString(QChar(0x2026));
  }

  return title;
}

auto BuildQuotedBody(const EMailMetaData& source, EMailReplyMode mode)
    -> QByteArray {
  const auto text = QString::fromUtf8(source.body);

  if (mode == EMailReplyMode::kFORWARD) {
    QString out = "\n\n---------- Forwarded message ----------\n";
    if (!source.from.isEmpty()) out += "From: " + source.from + "\n";
    if (!source.to.isEmpty()) out += "To: " + source.to.join("; ") + "\n";
    if (!source.cc.isEmpty()) out += "Cc: " + source.cc.join("; ") + "\n";
    if (source.datetime.isValid()) {
      // RFC 2822/5322 date-time, not ISO 8601: this block reproduces header
      // syntax, and a Date: line is defined to carry that form.
      out += "Date: " + source.datetime.toString(Qt::RFC2822Date) + "\n";
    }
    if (!source.subject.isEmpty()) out += "Subject: " + source.subject + "\n";
    out += "\n" + text;
    return out.toUtf8();
  }

  QString out;
  if (!source.from.isEmpty()) {
    // The attribution line is prose the reader sees, not a header field, so it
    // is translated and the date is rendered in the reader's own locale. An
    // ISO timestamp here was simply unreadable ("2026-09-14T10:33:00"), and an
    // RFC 5322 date would have forced English month and day names into every
    // translation.
    if (source.datetime.isValid()) {
      const auto when = QLocale::system().toString(
          source.datetime.toLocalTime(), QLocale::ShortFormat);
      out +=
          "\n" + QObject::tr("On %1, %2 wrote:").arg(when, source.from) + "\n";
    } else {
      out += "\n" + QObject::tr("%1 wrote:").arg(source.from) + "\n";
    }
  } else {
    out += "\n";
  }

  // Quote every line, empty ones included: a blank line left unquoted reads as
  // the end of the quotation.
  const auto lines = text.split('\n');
  for (const auto& line : lines) {
    out += line.isEmpty() ? QString(">\n") : QString("> %1\n").arg(line);
  }

  return out.toUtf8();
}

namespace {

// Fields that RFC 5322 allows at most once. A second copy is not merely
// untidy: readers disagree about which one wins, so a message carrying two
// can show one thing to the person and another to whatever checked it.
auto SingletonFieldNames() -> QStringList {
  return {"From",    "Sender", "Reply-To",   "To",          "Cc",        "Bcc",
          "Subject", "Date",   "Message-ID", "In-Reply-To", "References"};
}

// Does this text mix scripts inside one label? Latin plus Cyrillic in one
// domain is the classic homograph construction and essentially never
// legitimate.
auto MixesScripts(const QString& text) -> bool {
  bool latin = false;
  bool other = false;
  for (const auto c : text) {
    if (!c.isLetter()) continue;
    const auto script = c.script();
    if (script == QChar::Script_Latin) {
      latin = true;
    } else if (script != QChar::Script_Common &&
               script != QChar::Script_Inherited) {
      other = true;
    }
  }
  return latin && other;
}

auto DomainOf(const QString& address) -> QString {
  const auto email = AddressOfUid(address);
  const auto at = email.lastIndexOf('@');
  return at < 0 ? QString() : email.mid(at + 1);
}

}  // namespace

auto LooksLikeSpoofedAddress(const QString& address) -> bool {
  const auto domain = DomainOf(address);
  if (domain.isEmpty()) return false;

  if (MixesScripts(domain)) return true;

  // A domain that is not plain ASCII has a punycode form that differs from
  // what is displayed; that gap is exactly what a homograph attack lives in.
  const auto ace = QString::fromLatin1(QUrl::toAce(domain));
  return !ace.isEmpty() && ace.compare(domain, Qt::CaseInsensitive) != 0;
}

auto HasBareLineFeeds(const QByteArray& bytes) -> bool {
  for (qsizetype i = 0; i < bytes.size(); ++i) {
    if (bytes[i] != '\n') continue;
    if (i == 0 || bytes[i - 1] != '\r') return true;
  }
  return false;
}

auto InspectMessage(const EMailMetaData& meta, const EMailPart& root,
                    const QList<EMailSignatureRegion>& regions,
                    const QList<EMailRegionVerdict>& verdicts)
    -> QList<EMailFinding> {
  QList<EMailFinding> findings;

  // --- the signed bytes are still the signed bytes -------------------------

  // REPORTED, not re-derived. This used to scan the document itself, which
  // meant two places decided whether a signature's bytes were canonical -- and
  // two places that can disagree about why a signature failed will eventually
  // do so. The verifier decides it, on the exact slice it handed the engine,
  // and this says what the verifier found.
  for (const auto& verdict : verdicts) {
    if (!verdict.signed_bytes_non_canonical) continue;

    findings.append(
        {EMailFindingLevel::kWARN,
         QObject::tr("Signed part no longer in canonical form"),
         QObject::tr(
             "The signed part contains line endings that are not CRLF, which "
             "is not the form the signature was computed over. Something "
             "rewrote this message after it was signed, usually a program "
             "that changed line endings while saving or copying it, not an "
             "attack. The signature cannot verify against these bytes, and "
             "importing the sender's key will not change that. Checking it "
             "needs the original, unmodified message.")});
  }

  // --- header integrity ----------------------------------------------------

  QMap<QString, int> counts;
  for (const auto& field : root.header_fields) {
    counts[field.first.toLower()] += 1;
  }

  for (const auto& name : SingletonFieldNames()) {
    const auto count = counts.value(name.toLower(), 0);
    if (count <= 1) continue;
    findings.append(
        {EMailFindingLevel::kRISK, QObject::tr("Duplicate %1 header").arg(name),
         QObject::tr("This message carries %1 copies of a header that may "
                     "appear only once. Different mail programs pick "
                     "different copies, so what you see here may not be what "
                     "another reader sees.")
             .arg(count)});
  }

  // --- identity ------------------------------------------------------------

  for (const auto& address : QStringList{meta.from} + meta.to + meta.cc) {
    if (address.trimmed().isEmpty()) continue;
    if (!LooksLikeSpoofedAddress(address)) continue;
    findings.append(
        {EMailFindingLevel::kRISK, QObject::tr("Address may be disguised"),
         QObject::tr("The domain in \"%1\" uses characters that can be made "
                     "to look like a different, familiar address.")
             .arg(address)});
  }

  // A Reply-To on another domain is ordinary for mailing lists and support
  // systems, so this is a note rather than an accusation -- but it is the
  // mechanism behind most reply-hijacking, and it is invisible until you look.
  if (!meta.reply_to.trimmed().isEmpty() && !meta.from.trimmed().isEmpty()) {
    const auto reply_domain = DomainOf(meta.reply_to);
    const auto from_domain = DomainOf(meta.from);
    if (!reply_domain.isEmpty() && !from_domain.isEmpty() &&
        reply_domain.compare(from_domain, Qt::CaseInsensitive) != 0) {
      findings.append(
          {EMailFindingLevel::kWARN, QObject::tr("Replies go somewhere else"),
           QObject::tr("This message is from \"%1\", but replies would be "
                       "sent to \"%2\", which is a different domain.")
               .arg(from_domain, reply_domain)});
    }
  }

  // --- structure -----------------------------------------------------------

  const auto flat = FlattenMimeTree(root);

  for (const auto* part : flat) {
    if (part->content_type == "multipart/signed" &&
        part->children.size() != 2) {
      findings.append(
          {EMailFindingLevel::kRISK, QObject::tr("Malformed signed section"),
           QObject::tr("A signed section must contain exactly two parts; this "
                       "one contains %1. It may not verify, and what it "
                       "covers is ambiguous.")
               .arg(part->children.size())});
    }
    if (part->content_type == "multipart/encrypted" &&
        part->children.size() != 2) {
      findings.append(
          {EMailFindingLevel::kRISK, QObject::tr("Malformed encrypted section"),
           QObject::tr("An encrypted section must contain exactly two parts; "
                       "this one contains %1.")
               .arg(part->children.size())});
    }
  }

  // Content smuggled in beside a signature is the reason coverage is tracked
  // per part rather than per message.
  if (!regions.isEmpty()) {
    int uncovered = 0;
    for (const auto* part : flat) {
      if (part->is_multipart || part->is_protocol_part) continue;
      if (part->covered_by_regions.isEmpty()) ++uncovered;
    }
    if (uncovered > 0) {
      findings.append(
          {EMailFindingLevel::kWARN, QObject::tr("Not everything is signed"),
           QObject::tr("%1 part(s) of this message sit outside the signature. "
                       "They arrived unauthenticated and could have been "
                       "added or changed by anyone in the path.")
               .arg(uncovered)});
    }
  }

  // A signature over ciphertext is the case most easily misread as "this
  // message is signed by X". It is not: signing someone else's encrypted blob
  // takes no key of theirs and no knowledge of what is inside it.
  for (const auto& region : regions) {
    if (!region.covers_ciphertext_only) continue;
    findings.append(
        {EMailFindingLevel::kWARN,
         QObject::tr("Signature covers the encrypted data only"),
         QObject::tr("This signature was made over the encrypted block, not "
                     "over the message inside it. It shows who sent the "
                     "ciphertext along; it does not say who wrote what you "
                     "are reading, and anyone could have signed a copy of "
                     "this same block.")});
  }

  // --- OpenPGP material carried as attachments -----------------------------

  int attached_keys = 0;
  int detached_signatures = 0;
  for (const auto* part : flat) {
    if (part->is_openpgp_key) ++attached_keys;
    // A pgp-signature part that is NOT the second half of a multipart/signed
    // is a detached signature riding along as a file, which is a different
    // thing and is not verified by opening the message.
    if (part->content_type == "application/pgp-signature" &&
        part->covered_by_regions.isEmpty()) {
      bool belongs_to_a_region = false;
      for (const auto& region : regions) {
        if (region.signature_part_index == part->index) {
          belongs_to_a_region = true;
          break;
        }
      }
      if (!belongs_to_a_region) ++detached_signatures;
    }
  }

  if (attached_keys > 0) {
    findings.append(
        {EMailFindingLevel::kNOTE, QObject::tr("Public key attached"),
         QObject::tr("This message carries %1 OpenPGP public key(s). A key "
                     "arriving in a message proves nothing about who sent it: "
                     "anyone can attach any key, including one they made for "
                     "the name on the From line.")
             .arg(attached_keys)});
  }

  if (detached_signatures > 0) {
    findings.append(
        {EMailFindingLevel::kNOTE, QObject::tr("Detached signature attached"),
         QObject::tr("This message carries %1 signature file(s) that are not "
                     "part of its own signed structure. Opening the message "
                     "does not check them; they sign something else.")
             .arg(detached_signatures)});
  }

  // A Bcc header on a received message means the sender's tooling disclosed
  // recipients who were meant to be blind.
  if (!meta.bcc_header.isEmpty()) {
    findings.append(
        {EMailFindingLevel::kWARN, QObject::tr("Blind recipients are visible"),
         QObject::tr("This message carries a Bcc header naming %1 "
                     "recipient(s). Anyone who received it can see who was "
                     "blind-copied.")
             .arg(meta.bcc_header.size())});
  }

  return findings;
}

auto PreflightMessage(const EMailMetaData& meta, const EMailPart& root,
                      const QList<EMailSignatureRegion>& regions,
                      const QStringList& compose_bcc) -> QList<EMailFinding> {
  QList<EMailFinding> findings;

  if (regions.isEmpty()) {
    findings.append(
        {EMailFindingLevel::kNOTE, QObject::tr("Nothing is signed"),
         QObject::tr("This message carries no signature, so the recipient "
                     "cannot tell that it came from you or that it arrived "
                     "unchanged.")});
  }

  const auto flat = FlattenMimeTree(root);
  int unsigned_parts = 0;
  for (const auto* part : flat) {
    if (part->is_multipart || part->is_protocol_part) continue;
    if (!regions.isEmpty() && part->covered_by_regions.isEmpty()) {
      ++unsigned_parts;
    }
  }

  if (unsigned_parts > 0) {
    findings.append(
        {EMailFindingLevel::kWARN, QObject::tr("Some parts are not signed"),
         QObject::tr("%1 part(s) of this message would go out without the "
                     "signature covering them.")
             .arg(unsigned_parts)});
  }

  // The blind recipients are about to become an encryption decision, and the
  // one thing that must never happen is for them to become a header.
  if (!compose_bcc.isEmpty()) {
    findings.append(
        {EMailFindingLevel::kNOTE, QObject::tr("Blind recipients"),
         QObject::tr("%1 blind recipient(s) will be included in encryption "
                     "but will not appear anywhere in the message.")
             .arg(compose_bcc.size())});
  }

  if (!meta.bcc_header.isEmpty()) {
    findings.append(
        {EMailFindingLevel::kRISK, QObject::tr("Bcc header present"),
         QObject::tr("This message carries a Bcc header. Sending it would "
                     "tell every recipient who was blind-copied.")});
  }

  for (const auto& address : QStringList{meta.from} + meta.to + meta.cc) {
    if (address.trimmed().isEmpty()) continue;
    if (!LooksLikeSpoofedAddress(address)) continue;
    findings.append(
        {EMailFindingLevel::kRISK, QObject::tr("Recipient may be disguised"),
         QObject::tr("The domain in \"%1\" uses characters that can be made "
                     "to look like a different address. Check it before "
                     "sending.")
             .arg(address)});
  }

  if (meta.to.isEmpty() && meta.cc.isEmpty() && compose_bcc.isEmpty()) {
    findings.append({EMailFindingLevel::kWARN, QObject::tr("No recipients"),
                     QObject::tr("This message is not addressed to anyone.")});
  }

  return findings;
}

auto ExtractParts(const vmime::shared_ptr<vmime::message>& message,
                  EMailMetaData& meta_data, const EMailParseLimits& limits)
    -> int {
  if (!message) return -1;

  EMailPart root;
  QList<EMailSignatureRegion> regions;
  if (ParseMimeTree(message, {}, root, regions, limits) != 0) return -1;

  meta_data.signature_regions = regions;

  // The flattening rule is kept exactly as it was, deliberately. This function
  // feeds the crypto paths and the result cards, and the tree was introduced
  // underneath it rather than to change it -- the alternative-aware selection
  // lives in SelectBodyPart(), which the views use. A differential test pins
  // the two implementations together across the corpus.
  for (const auto* part : FlattenMimeTree(root)) {
    if (part->is_multipart || part->is_protocol_part) continue;

    const bool is_body = meta_data.body.isEmpty() && part->filename.isEmpty() &&
                         part->content_type == "text/plain" &&
                         !part->disposition.startsWith("attachment");
    if (is_body) {
      meta_data.body = part->data;
      meta_data.body_content_type = part->content_type;
      continue;
    }

    EMailAttachment att;
    att.filename = part->filename;
    att.mime_type = part->content_type;
    att.disposition = part->disposition;
    att.data = part->data;
    att.is_openpgp_key = part->is_openpgp_key;
    // Being covered by any region is exactly what "inside the signed part"
    // meant before regions existed.
    att.inside_signed_part = !part->covered_by_regions.isEmpty();
    meta_data.attachments.append(att);
  }

  return 0;
}

auto ReadFileWithin(const QString& path, qint64 max_bytes, QByteArray& out,
                    QString& error, qint64& size) -> EMailFileAdmission {
  out.clear();
  error.clear();
  size = 0;

  const QFileInfo info(path);

  // A file that is not there is a different answer from one of the wrong
  // kind, and the system has a better sentence for it than this code does.
  if (!info.exists()) {
    QFile probe(path);
    probe.open(QIODevice::ReadOnly);
    error = probe.errorString();
    return EMailFileAdmission::kUNREADABLE;
  }

  // First, and not as an optimisation. Everything below assumes the size it
  // is told means something, and for a FIFO or a device node it does not.
  if (!info.isFile()) return EMailFileAdmission::kNOT_REGULAR;

  size = info.size();
  if (size > max_bytes) return EMailFileAdmission::kTOO_LARGE;

  QFile file(path);
  // NOT QIODevice::Text: line endings are content here. A PGP/MIME signature
  // covers the exact octets of a message in canonical CRLF form, and an
  // attachment is whatever bytes the user chose.
  if (!file.open(QIODevice::ReadOnly)) {
    error = file.errorString();
    return EMailFileAdmission::kUNREADABLE;
  }

  // Again, on the handle this time. The path was looked at a moment ago and
  // may not name the same file now.
  size = file.size();
  if (size > max_bytes) return EMailFileAdmission::kTOO_LARGE;

  // Bounded rather than readAll(): the size above is what the filesystem says,
  // and a file being appended to while it is read would otherwise have no
  // limit at all.
  out = file.read(max_bytes);
  if (out.size() < size) {
    error = file.errorString();
    out.clear();
    return EMailFileAdmission::kUNREADABLE;
  }

  return EMailFileAdmission::kOK;
}

auto DocumentUnchangedOnFailure(const QByteArray& original) -> QByteArray {
  return original;
}

auto DecodePartContent(const vmime::shared_ptr<const vmime::bodyPart>& part)
    -> QByteArray {
  if (!part) return {};
  return DecodePart(part);
}

auto BuildInnerPartHeader(const vmime::shared_ptr<vmime::header>& source)
    -> QByteArray {
  // Copied rather than hand-picked field by field. The old code listed five
  // headers explicitly and rebuilt the part from them, which silently dropped
  // Content-Transfer-Encoding -- and since the body is carried over as a raw,
  // still-encoded slice, a base64 or quoted-printable body then arrived at the
  // recipient with nothing to say how to decode it. Every message with an
  // attachment has such a body.
  static const QStringList kDescriptiveFields = {
      vmime::fields::MIME_VERSION, vmime::fields::FROM,
      vmime::fields::TO,           vmime::fields::CC,
      vmime::fields::REPLY_TO,     vmime::fields::SUBJECT,
      vmime::fields::DATE,         vmime::fields::MESSAGE_ID};

  auto header = vmime::make_shared<vmime::header>();

  for (const auto& field : source->getFieldList()) {
    const auto name = Q_SC(field->getName());

    // Everything describing how to read the body has to travel with it.
    const bool is_content_field =
        name.startsWith("Content-", Qt::CaseInsensitive);
    const bool is_descriptive =
        std::any_of(kDescriptiveFields.cbegin(), kDescriptiveFields.cend(),
                    [&name](const QString& f) {
                      return name.compare(f, Qt::CaseInsensitive) == 0;
                    });

    if (!is_content_field && !is_descriptive) continue;

    header->appendField(vmime::dynamicCast<vmime::headerField>(field->clone()));
  }

  // Bytes: this is concatenated with a raw body slice to form the octets a
  // signature will cover, so it must not go through a text round trip.
  return QByteArray::fromStdString(
      header->generate(vmime::lineLengthLimits::convenient));
}

namespace {

/// The Content-Description that SignEMLData() writes on the key part it
/// attaches on its own initiative. It is the only thing distinguishing that
/// part from a key the USER chose to attach, and silently removing the user's
/// attachment would be worse than leaving a stale one behind.
constexpr auto kSignerKeyDescription = "OpenPGP public key";

auto PartDescription(const vmime::shared_ptr<const vmime::bodyPart>& part)
    -> QString {
  auto field = part->getHeader()->findField(vmime::fields::CONTENT_DESCRIPTION);
  if (!field) return {};
  auto value = field->getValue();
  if (!value) return {};
  return Q_SC(value->generate()).trimmed();
}

/// Whether @p part is the signer's key as Sign attached it, rather than a key
/// the user attached themselves.
auto IsSignerPublicKeyPart(const vmime::shared_ptr<vmime::bodyPart>& part)
    -> bool {
  return PartContentType(part) == "application/pgp-keys" &&
         PartDescription(part) == QLatin1String(kSignerKeyDescription);
}

/// Replaces every Content-* field of @p dst with the ones @p src carries.
void AdoptContentFields(const vmime::shared_ptr<vmime::header>& dst,
                        const vmime::shared_ptr<vmime::header>& src) {
  const auto is_content = [](const vmime::shared_ptr<vmime::headerField>& f) {
    return Q_SC(f->getName()).startsWith("Content-", Qt::CaseInsensitive);
  };

  // Collected before removing any: removeField() mutates the list being walked.
  QList<vmime::shared_ptr<vmime::headerField>> stale;
  for (const auto& field : dst->getFieldList()) {
    if (is_content(field)) stale.append(field);
  }
  for (const auto& field : stale) dst->removeField(field);

  for (const auto& field : src->getFieldList()) {
    if (!is_content(field)) continue;
    dst->appendField(vmime::dynamicCast<vmime::headerField>(field->clone()));
  }
}

/// Makes @p part the whole message: its content headers and its body become
/// the message's, and everything else about the message is left alone.
void PromotePartToMessage(const vmime::shared_ptr<vmime::message>& message,
                          const vmime::shared_ptr<vmime::bodyPart>& part) {
  // Cloned, and the fields adopted, while @p part is still owned by the body
  // that is about to be replaced.
  auto body = vmime::dynamicCast<vmime::body>(part->getBody()->clone());
  AdoptContentFields(message->getHeader(), part->getHeader());
  message->setBody(body);
}

/// Takes one layer of a previous Sign off @p message. Returns whether it found
/// one; see StripPreviousSignature() for why this runs to a fixpoint.
auto StripOneSignatureLayer(const vmime::shared_ptr<vmime::message>& message)
    -> bool {
  if (PartContentType(message) == "multipart/signed" &&
      ContentTypeParam(message, "protocol") == "application/pgp-signature" &&
      message->getBody()->getPartCount() == 2) {
    PromotePartToMessage(message, message->getBody()->getPartAt(0));
    return true;
  }

  if (!PartContentType(message).startsWith("multipart/")) return false;

  auto body = message->getBody();
  QList<vmime::shared_ptr<vmime::bodyPart>> stale;
  for (size_t i = 0; i < body->getPartCount(); ++i) {
    auto part = body->getPartAt(i);
    if (IsSignerPublicKeyPart(part)) stale.append(part);
  }
  if (stale.isEmpty()) return false;

  for (const auto& part : stale) body->removePart(part);

  // A container that existed only to hold the body beside the key Sign
  // attached is not a container any more. Only collapsed when one of those
  // keys really was just removed: a single-part multipart/mixed the user built
  // themselves is theirs to keep.
  if (body->getPartCount() == 1 &&
      PartContentType(message) == "multipart/mixed") {
    PromotePartToMessage(message, body->getPartAt(0));
  }

  return true;
}

}  // namespace

auto StripPreviousSignature(const vmime::shared_ptr<vmime::message>& message)
    -> bool {
  if (!message) return false;

  // Run to a fixpoint rather than once. Each re-sign left a wrapper AND a key
  // part behind, and they alternate as the layers come off: unwrapping a
  // signature exposes the container holding the previous signer's key, and
  // collapsing that container exposes the signature under it.
  //
  // Bounded so that a message built to nest forever cannot spin here. Eight
  // layers is four re-signs deep, well past anything that arrives by accident.
  constexpr int kMaxLayers = 16;

  bool changed = false;
  for (int i = 0; i < kMaxLayers; ++i) {
    if (!StripOneSignatureLayer(message)) break;
    changed = true;
  }
  return changed;
}