mimetreeparser/src/messagepart.cpp

/*
   SPDX-FileCopyrightText: 2015 Sandro Knauß <sknauss@kde.org>
 
   SPDX-License-Identifier: LGPL-2.0-or-later
*/
 
#include "messagepart.h"
#include "cryptohelper.h"
#include "job/qgpgmejobexecutor.h"
#include "memento/compositememento.h"
#include "memento/cryptobodypartmemento.h"
#include "memento/decryptverifybodypartmemento.h"
#include "memento/keycachememento.h"
#include "memento/verifydetachedbodypartmemento.h"
#include "memento/verifyopaquebodypartmemento.h"
#include "mimetreeparser_debug.h"
#include "objecttreeparser.h"
 
#include "bodyformatter/utils.h"
 
#include <KMime/Content>
#include <KMime/Types>
#include <Libkleo/Compliance>
#include <Libkleo/KeyCache>
 
#include <QGpgME/DN>
#include <QGpgME/ImportJob>
#include <QGpgME/Protocol>
#include <QGpgME/VerifyDetachedJob>
#include <QGpgME/VerifyOpaqueJob>
 
#include <gpgme++/key.h>
#include <gpgme++/keylistresult.h>
#include <gpgme.h>
 
#include <KLocalizedString>
 
#include <QUrl>
 
using namespace MimeTreeParser;
 
//------MessagePart-----------------------
namespace MimeTreeParser
{
class MessagePartPrivate
{
public:
    MessagePart *mParentPart = nullptr;
    QList<MessagePart::Ptr> mBlocks;
    KMime::Content *mNode = nullptr;
    KMime::Content *mAttachmentNode = nullptr;
    QString mText;
    PartMetaData mMetaData;
    bool mRoot = false;
    bool mIsImage = false;
    bool mNeverDisplayInline = false;
};
}
 
MessagePart::MessagePart(ObjectTreeParser *otp, const QString &text)
    : mOtp(otp)
    , d(new MessagePartPrivate)
{
    d->mText = text;
}
 
MessagePart::~MessagePart() = default;
 
MessagePart *MessagePart::parentPart() const
{
    return d->mParentPart;
}
 
void MessagePart::setParentPart(MessagePart *parentPart)
{
    d->mParentPart = parentPart;
}
 
QString MessagePart::htmlContent() const
{
    return text();
}
 
QString MessagePart::plaintextContent() const
{
    return text();
}
 
PartMetaData *MessagePart::partMetaData() const
{
    return &d->mMetaData;
}
 
Interface::BodyPartMemento *MessagePart::memento() const
{
    return nodeHelper()->bodyPartMemento(content(), "__plugin__");
}
 
void MessagePart::setMemento(Interface::BodyPartMemento *memento)
{
    nodeHelper()->setBodyPartMemento(content(), "__plugin__", memento);
}
 
KMime::Content *MessagePart::content() const
{
    return d->mNode;
}
 
void MessagePart::setContent(KMime::Content *node)
{
    d->mNode = node;
}
 
KMime::Content *MessagePart::attachmentContent() const
{
    return d->mAttachmentNode;
}
 
void MessagePart::setAttachmentContent(KMime::Content *node)
{
    d->mAttachmentNode = node;
}
 
bool MessagePart::isAttachment() const
{
    return d->mAttachmentNode;
}
 
QString MessagePart::attachmentIndex() const
{
    return attachmentContent()->index().toString();
}
 
QString MessagePart::attachmentLink() const
{
    return mOtp->nodeHelper()->asHREF(content(), QStringLiteral("body"));
}
 
QString MessagePart::makeLink(const QString &path) const
{
    // FIXME: use a PRNG for the first arg, instead of a serial number
    static int serial = 0;
    if (path.isEmpty()) {
        return {};
    }
    return QStringLiteral("x-kmail:/bodypart/%1/%2/%3")
        .arg(serial++)
        .arg(mOtp->nodeHelper()->asHREF(content()), QString::fromLatin1(QUrl::toPercentEncoding(path, "/")));
}
 
void MessagePart::setIsRoot(bool root)
{
    d->mRoot = root;
}
 
bool MessagePart::isRoot() const
{
    return d->mRoot;
}
 
QString MessagePart::text() const
{
    return d->mText;
}
 
void MessagePart::setText(const QString &text)
{
    d->mText = text;
}
 
bool MessagePart::isHtml() const
{
    return false;
}
 
Interface::ObjectTreeSource *MessagePart::source() const
{
    Q_ASSERT(mOtp);
    return mOtp->mSource;
}
 
NodeHelper *MessagePart::nodeHelper() const
{
    Q_ASSERT(mOtp);
    return mOtp->nodeHelper();
}
 
void MessagePart::parseInternal(KMime::Content *node, bool onlyOneMimePart)
{
    auto subMessagePart = mOtp->parseObjectTreeInternal(node, onlyOneMimePart);
    d->mRoot = subMessagePart->isRoot();
    const QList<MessagePart::Ptr> subParts = subMessagePart->subParts();
    for (const auto &part : subParts) {
        appendSubPart(part);
    }
}
 
QString MessagePart::renderInternalText() const
{
    QString text;
    const auto subPartsLst = subParts();
    for (const auto &mp : subPartsLst) {
        text += mp->text();
    }
    return text;
}
 
void MessagePart::fix() const
{
    const auto subPartsLst = subParts();
    for (const auto &mp : subPartsLst) {
        const auto m = mp.dynamicCast<MessagePart>();
        if (m) {
            m->fix();
        }
    }
}
 
void MessagePart::appendSubPart(const MessagePart::Ptr &messagePart)
{
    messagePart->setParentPart(this);
    d->mBlocks.append(messagePart);
}
 
const QList<MessagePart::Ptr> &MessagePart::subParts() const
{
    return d->mBlocks;
}
 
bool MessagePart::hasSubParts() const
{
    return !d->mBlocks.isEmpty();
}
 
void MessagePart::clearSubParts()
{
    d->mBlocks.clear();
}
 
bool MessagePart::neverDisplayInline() const
{
    return d->mNeverDisplayInline;
}
 
void MessagePart::setNeverDisplayInline(bool displayInline)
{
    d->mNeverDisplayInline = displayInline;
}
 
bool MessagePart::isImage() const
{
    return d->mIsImage;
}
 
void MessagePart::setIsImage(bool image)
{
    d->mIsImage = image;
}
 
bool MessagePart::hasHeader(const char *headerType) const
{
    Q_UNUSED(headerType)
    return false;
}
 
const KMime::Headers::Base *MimeTreeParser::MessagePart::header(const char *headerType) const
{
    Q_UNUSED(headerType)
    return nullptr;
}
 
QList<KMime::Headers::Base *> MessagePart::headers(const char *headerType) const
{
    Q_UNUSED(headerType)
    return {};
}
 
//-----MessagePartList----------------------
MessagePartList::MessagePartList(ObjectTreeParser *otp)
    : MessagePart(otp, QString())
{
}
 
MessagePartList::~MessagePartList() = default;
 
QString MessagePartList::text() const
{
    return renderInternalText();
}
 
QString MessagePartList::plaintextContent() const
{
    return {};
}
 
QString MessagePartList::htmlContent() const
{
    return {};
}
 
//-----TextMessageBlock----------------------
 
TextMessagePart::TextMessagePart(ObjectTreeParser *otp, KMime::Content *node, bool decryptMessage)
    : MessagePartList(otp)
    , mDecryptMessage(decryptMessage)
{
    if (!node) {
        qCWarning(MIMETREEPARSER_LOG) << "not a valid node";
        return;
    }
 
    setContent(node);
 
    parseContent();
}
 
TextMessagePart::~TextMessagePart() = default;
 
bool TextMessagePart::decryptMessage() const
{
    return mDecryptMessage;
}
 
void TextMessagePart::parseContent()
{
    const auto codecName = mOtp->codecNameFor(content());
    QStringDecoder aCodec(codecName.constData());
    const QString &fromAddress = mOtp->nodeHelper()->fromAsString(content());
    mSignatureState = KMMsgNotSigned;
    mEncryptionState = KMMsgNotEncrypted;
    const auto blocks = prepareMessageForDecryption(content()->decodedContent());
 
    const auto cryptProto = QGpgME::openpgp();
 
    if (!blocks.isEmpty()) {
        /* The (overall) signature/encrypted status is broken
         * if one unencrypted part is at the beginning or in the middle
         * because mailmain adds an unencrypted part at the end this should not break the overall status
         *
         * That's why we first set the tmp status and if one encrypted/signed block comes afterwards, than
         * the status is set to unencrypted
         */
        bool fullySignedOrEncrypted = true;
        bool fullySignedOrEncryptedTmp = true;
 
        int blockIndex = -1;
        for (const auto &block : blocks) {
            blockIndex += 1;
            if (!fullySignedOrEncryptedTmp) {
                fullySignedOrEncrypted = false;
            }
 
            if (block.type() == NoPgpBlock && !block.text().trimmed().isEmpty()) {
                fullySignedOrEncryptedTmp = false;
                aCodec.resetState();
                appendSubPart(MessagePart::Ptr(new MessagePart(mOtp, aCodec.decode(block.text()))));
            } else if (block.type() == PgpMessageBlock) {
                EncryptedMessagePart::Ptr mp(new EncryptedMessagePart(mOtp, QString(), cryptProto, fromAddress, nullptr));
                mp->setDecryptMessage(decryptMessage());
                mp->setIsEncrypted(true);
                mp->setMementoName(mp->mementoName() + "-" + nodeHelper()->asHREF(content(), QString::number(blockIndex)).toLocal8Bit());
                appendSubPart(mp);
                if (!decryptMessage()) {
                    continue;
                }
                mp->startDecryption(block.text(), codecName);
                if (mp->partMetaData()->inProgress) {
                    continue;
                }
            } else if (block.type() == ClearsignedBlock) {
                SignedMessagePart::Ptr mp(new SignedMessagePart(mOtp, QString(), cryptProto, fromAddress, nullptr));
                mp->setMementoName(mp->mementoName() + "-" + nodeHelper()->asHREF(content(), QString::number(blockIndex)).toLocal8Bit());
                appendSubPart(mp);
                mp->startVerification(block.text(), codecName);
            } else {
                continue;
            }
 
            const auto mp = subParts().last().staticCast<MessagePart>();
            const PartMetaData *messagePart(mp->partMetaData());
 
            if (!messagePart->isEncrypted && !messagePart->isSigned && !block.text().trimmed().isEmpty()) {
                aCodec.resetState();
                mp->setText(aCodec.decode(block.text()));
            }
 
            if (messagePart->isEncrypted) {
                mEncryptionState = KMMsgPartiallyEncrypted;
            }
 
            if (messagePart->isSigned) {
                mSignatureState = KMMsgPartiallySigned;
            }
        }
 
        // Do we have an fully Signed/Encrypted Message?
        if (fullySignedOrEncrypted) {
            if (mSignatureState == KMMsgPartiallySigned) {
                mSignatureState = KMMsgFullySigned;
            }
            if (mEncryptionState == KMMsgPartiallyEncrypted) {
                mEncryptionState = KMMsgFullyEncrypted;
            }
        }
    }
}
 
KMMsgEncryptionState TextMessagePart::encryptionState() const
{
    return mEncryptionState;
}
 
KMMsgSignatureState TextMessagePart::signatureState() const
{
    return mSignatureState;
}
 
bool TextMessagePart::showLink() const
{
    return !temporaryFilePath().isEmpty();
}
 
bool TextMessagePart::isFirstTextPart() const
{
    return content()->topLevel()->textContent() == content();
}
 
bool TextMessagePart::hasLabel() const
{
    return !NodeHelper::fileName(content()).isEmpty();
}
 
QString TextMessagePart::label() const
{
    const QString name = content()->contentType()->name();
    QString label = name.isEmpty() ? NodeHelper::fileName(content()) : name;
    if (label.isEmpty()) {
        label = i18nc("display name for an unnamed attachment", "Unnamed");
    }
    return label;
}
 
QString TextMessagePart::comment() const
{
    const QString comment = content()->contentDescription()->asUnicodeString();
    if (comment == label()) {
        return {};
    }
    return comment;
}
 
QString TextMessagePart::temporaryFilePath() const
{
    return nodeHelper()->writeNodeToTempFile(content());
}
 
//-----AttachmentMessageBlock----------------------
 
AttachmentMessagePart::AttachmentMessagePart(ObjectTreeParser *otp, KMime::Content *node, bool decryptMessage)
    : TextMessagePart(otp, node, decryptMessage)
{
}
 
AttachmentMessagePart::~AttachmentMessagePart() = default;
 
//-----HtmlMessageBlock----------------------
 
HtmlMessagePart::HtmlMessagePart(ObjectTreeParser *otp, KMime::Content *node, Interface::ObjectTreeSource *source)
    : MessagePart(otp, QString())
{
    Q_UNUSED(source)
    if (!node) {
        qCWarning(MIMETREEPARSER_LOG) << "not a valid node";
        return;
    }
    setContent(node);
 
    const QByteArray partBody(node->decodedContent());
    mBodyHTML = QStringDecoder(mOtp->codecNameFor(node).constData()).decode(partBody);
    mCharset = NodeHelper::charset(node);
}
 
HtmlMessagePart::~HtmlMessagePart() = default;
 
void HtmlMessagePart::fix() const
{
    mOtp->mHtmlContent += mBodyHTML;
}
 
QString HtmlMessagePart::text() const
{
    return mBodyHTML;
}
 
QString MimeTreeParser::HtmlMessagePart::plaintextContent() const
{
    return {};
}
 
bool HtmlMessagePart::isHtml() const
{
    return true;
}
 
QString HtmlMessagePart::bodyHtml() const
{
    return mBodyHTML;
}
 
//-----MimeMessageBlock----------------------
 
MimeMessagePart::MimeMessagePart(ObjectTreeParser *otp, KMime::Content *node, bool onlyOneMimePart)
    : MessagePart(otp, QString())
    , mOnlyOneMimePart(onlyOneMimePart)
{
    if (!node) {
        qCWarning(MIMETREEPARSER_LOG) << "not a valid node";
        return;
    }
    setContent(node);
 
    parseInternal(node, mOnlyOneMimePart);
}
 
MimeMessagePart::~MimeMessagePart() = default;
 
QString MimeMessagePart::text() const
{
    return renderInternalText();
}
 
QString MimeMessagePart::plaintextContent() const
{
    return {};
}
 
QString MimeMessagePart::htmlContent() const
{
    return {};
}
 
//-----AlternativeMessagePart----------------------
 
AlternativeMessagePart::AlternativeMessagePart(ObjectTreeParser *otp, KMime::Content *node, Util::HtmlMode preferredMode)
    : MessagePart(otp, QString())
    , mPreferredMode(preferredMode)
{
    setContent(node);
    KMime::Content *dataIcal = findTypeInDirectChilds(node, "text/calendar");
    KMime::Content *dataHtml = findTypeInDirectChilds(node, "text/html");
    KMime::Content *dataText = findTypeInDirectChilds(node, "text/plain");
 
    if (!dataHtml) {
        // If we didn't find the HTML part as the first child of the multipart/alternative, it might
        // be that this is a HTML message with images, and text/plain and multipart/related are the
        // immediate children of this multipart/alternative node.
        // In this case, the HTML node is a child of multipart/related.
        dataHtml = findTypeInDirectChilds(node, "multipart/related");
 
        // Still not found? Stupid apple mail actually puts the attachments inside of the
        // multipart/alternative, which is wrong. Therefore we also have to look for multipart/mixed
        // here.
        // Do this only when preferring HTML mail, though, since otherwise the attachments are hidden
        // when displaying plain text.
        if (!dataHtml) {
            dataHtml = findTypeInDirectChilds(node, "multipart/mixed");
        }
    }
 
    if (dataIcal) {
        mChildNodes[Util::MultipartIcal] = dataIcal;
    }
 
    if (dataText) {
        mChildNodes[Util::MultipartPlain] = dataText;
    }
 
    if (dataHtml) {
        mChildNodes[Util::MultipartHtml] = dataHtml;
    }
 
    if (mChildNodes.isEmpty()) {
        qCWarning(MIMETREEPARSER_LOG) << "no valid nodes";
        return;
    }
 
    QMapIterator<Util::HtmlMode, KMime::Content *> i(mChildNodes);
    while (i.hasNext()) {
        i.next();
        mChildParts[i.key()] = MimeMessagePart::Ptr(new MimeMessagePart(mOtp, i.value(), true));
    }
}
 
AlternativeMessagePart::~AlternativeMessagePart() = default;
 
Util::HtmlMode AlternativeMessagePart::preferredMode() const
{
    return mPreferredMode;
}
 
void AlternativeMessagePart::setPreferredMode(Util::HtmlMode preferredMode)
{
    mPreferredMode = preferredMode;
}
 
QList<Util::HtmlMode> AlternativeMessagePart::availableModes()
{
    return mChildParts.keys();
}
 
QString AlternativeMessagePart::text() const
{
    if (mChildParts.contains(Util::MultipartPlain)) {
        return mChildParts[Util::MultipartPlain]->text();
    }
    return {};
}
 
void AlternativeMessagePart::fix() const
{
    if (mChildParts.contains(Util::MultipartPlain)) {
        mChildParts[Util::MultipartPlain]->fix();
    }
 
    const auto mode = preferredMode();
    if (mode != Util::MultipartPlain && mChildParts.contains(mode)) {
        mChildParts[mode]->fix();
    }
}
 
const QMap<Util::HtmlMode, MimeMessagePart::Ptr> &AlternativeMessagePart::childParts() const
{
    return mChildParts;
}
 
bool AlternativeMessagePart::isHtml() const
{
    return mChildParts.contains(Util::MultipartHtml);
}
 
QString AlternativeMessagePart::plaintextContent() const
{
    return text();
}
 
QString AlternativeMessagePart::htmlContent() const
{
    if (mChildParts.contains(Util::MultipartHtml)) {
        return mChildParts[Util::MultipartHtml]->text();
    } else {
        return plaintextContent();
    }
}
 
//-----CertMessageBlock----------------------
 
CertMessagePart::CertMessagePart(ObjectTreeParser *otp, KMime::Content *node, const QGpgME::Protocol *cryptoProto, bool autoImport)
    : MessagePart(otp, QString())
    , mAutoImport(autoImport)
    , mCryptoProto(cryptoProto)
{
    if (!node) {
        qCWarning(MIMETREEPARSER_LOG) << "not a valid node";
        return;
    }
    setContent(node);
 
    if (!mAutoImport) {
        return;
    }
 
    const QByteArray certData = node->decodedContent();
 
    QGpgME::ImportJob *import = mCryptoProto->importJob();
    QGpgMEJobExecutor executor;
    mImportResult = executor.exec(import, certData);
}
 
CertMessagePart::~CertMessagePart() = default;
 
QString CertMessagePart::text() const
{
    return {};
}
 
const GpgME::ImportResult &CertMessagePart::importResult() const
{
    return mImportResult;
}
 
//-----SignedMessageBlock---------------------
SignedMessagePart::SignedMessagePart(ObjectTreeParser *otp,
                                     const QString &text,
                                     const QGpgME::Protocol *cryptoProto,
                                     const QString &fromAddress,
                                     KMime::Content *node)
    : MessagePart(otp, text)
    , mCryptoProto(cryptoProto)
    , mFromAddress(fromAddress)
    , mMementoName("verification")
{
    setContent(node);
    partMetaData()->technicalProblem = (mCryptoProto == nullptr);
    partMetaData()->isSigned = true;
    partMetaData()->isGoodSignature = false;
    partMetaData()->status = i18n("Wrong Crypto Plug-In.");
    partMetaData()->status_code = GPGME_SIG_STAT_NONE;
}
 
SignedMessagePart::~SignedMessagePart() = default;
 
void SignedMessagePart::setIsSigned(bool isSigned)
{
    partMetaData()->isSigned = isSigned;
}
 
bool SignedMessagePart::isSigned() const
{
    return partMetaData()->isSigned;
}
 
QByteArray SignedMessagePart::mementoName() const
{
    return mMementoName;
}
 
void SignedMessagePart::setMementoName(const QByteArray &name)
{
    mMementoName = name;
}
 
static GpgME::Protocol toGpgMeProtocol(const QGpgME::Protocol *protocol)
{
    if (protocol == QGpgME::openpgp()) {
        return GpgME::OpenPGP;
    }
 
    if (protocol == QGpgME::smime()) {
        return GpgME::CMS;
    }
 
    return GpgME::UnknownProtocol;
}
 
bool SignedMessagePart::okVerify(const QByteArray &data, const QByteArray &signature, KMime::Content *textNode)
{
    NodeHelper *nodeHelper = mOtp->nodeHelper();
 
    partMetaData()->isSigned = false;
    partMetaData()->technicalProblem = (mCryptoProto == nullptr);
    partMetaData()->status = i18n("Wrong Crypto Plug-In.");
    partMetaData()->status_code = GPGME_SIG_STAT_NONE;
 
    const QByteArray _mementoName = mementoName();
 
    auto m = dynamic_cast<CompositeMemento *>(nodeHelper->bodyPartMemento(content(), _mementoName));
    Q_ASSERT(!m || mCryptoProto); // No CryptoPlugin and having a bodyPartMemento -> there is something completely wrong
 
    if (!m && mCryptoProto) {
        CryptoBodyPartMemento *newM = nullptr;
        if (!signature.isEmpty()) {
            QGpgME::VerifyDetachedJob *job = mCryptoProto->verifyDetachedJob();
            if (job) {
                newM = new VerifyDetachedBodyPartMemento(job, mCryptoProto->keyListJob(), signature, data);
            }
        } else {
            QGpgME::VerifyOpaqueJob *job = mCryptoProto->verifyOpaqueJob();
            if (job) {
                newM = new VerifyOpaqueBodyPartMemento(job, mCryptoProto->keyListJob(), data);
            }
        }
 
        if (newM) {
            m = new CompositeMemento();
            m->addMemento(newM);
            m->addMemento(new KeyCacheMemento(Kleo::KeyCache::mutableInstance(), toGpgMeProtocol(mCryptoProto)));
        }
 
        if (m) {
            if (mOtp->allowAsync()) {
                QObject::connect(m, &CryptoBodyPartMemento::update, nodeHelper, &NodeHelper::update);
                if (m->start()) {
                    partMetaData()->inProgress = true;
                    mOtp->mHasPendingAsyncJobs = true;
                }
            } else {
                m->exec();
            }
            nodeHelper->setBodyPartMemento(content(), _mementoName, m);
        }
    } else if (m && m->isRunning()) {
        partMetaData()->inProgress = true;
        mOtp->mHasPendingAsyncJobs = true;
    } else {
        partMetaData()->inProgress = false;
        mOtp->mHasPendingAsyncJobs = false;
    }
 
    if (m && !partMetaData()->inProgress) {
        if (!signature.isEmpty()) {
            mVerifiedText = data;
        }
        setVerificationResult(m, textNode);
    }
 
    if (!m && !partMetaData()->inProgress) {
        QString errorMsg;
        QString cryptPlugLibName;
        QString cryptPlugDisplayName;
        if (mCryptoProto) {
            cryptPlugLibName = mCryptoProto->name();
            cryptPlugDisplayName = mCryptoProto->displayName();
        }
 
        if (!mCryptoProto) {
            if (cryptPlugDisplayName.isEmpty()) {
                errorMsg = i18n("No appropriate crypto plug-in was found.");
            } else {
                errorMsg = i18nc("%1 is either 'OpenPGP' or 'S/MIME'", "No %1 plug-in was found.", cryptPlugDisplayName);
            }
        } else {
            errorMsg = i18n("Crypto plug-in \"%1\" cannot verify signatures.", cryptPlugLibName);
        }
        partMetaData()->errorText = i18n(
            "The message is signed, but the "
            "validity of the signature cannot be "
            "verified.<br />"
            "Reason: %1",
            errorMsg);
    }
 
    return partMetaData()->isSigned;
}
 
static QString prettifyDN(const char *uid)
{
    return QGpgME::DN(uid).prettyDN();
}
 
void SignedMessagePart::startVerification(const QByteArray &text, QByteArrayView aCodec)
{
    startVerificationDetached(text, nullptr, QByteArray());
 
    if (!content() && partMetaData()->isSigned) {
        QStringDecoder codec(aCodec.constData());
        setText(codec.decode(mVerifiedText));
    }
}
 
void SignedMessagePart::startVerificationDetached(const QByteArray &text, KMime::Content *textNode, const QByteArray &signature)
{
    partMetaData()->isEncrypted = false;
    partMetaData()->isDecryptable = false;
 
    if (textNode) {
        parseInternal(textNode, false);
    }
 
    if (!okVerify(text, signature, textNode)) {
        partMetaData()->creationTime = QDateTime();
    }
}
 
void SignedMessagePart::setVerificationResult(const CompositeMemento *m, KMime::Content *textNode)
{
    {
        const auto kc = m->memento<KeyCacheMemento>();
        if (kc) {
            mKeyCache = kc->keyCache();
        }
    }
    {
        const auto vm = m->memento<VerifyDetachedBodyPartMemento>();
        if (vm) {
            mSignatures = vm->verifyResult().signatures();
            partMetaData()->verificationResult = vm->verifyResult();
        }
    }
    {
        const auto vm = m->memento<VerifyOpaqueBodyPartMemento>();
        if (vm) {
            mVerifiedText = vm->plainText();
            mSignatures = vm->verifyResult().signatures();
            partMetaData()->verificationResult = vm->verifyResult();
        }
    }
    {
        const auto vm = m->memento<DecryptVerifyBodyPartMemento>();
        if (vm) {
            mVerifiedText = vm->plainText();
            mSignatures = vm->verifyResult().signatures();
            partMetaData()->verificationResult = vm->verifyResult();
        }
    }
    partMetaData()->auditLogError = m->auditLogError();
    partMetaData()->auditLog = m->auditLogAsHtml();
    partMetaData()->isSigned = !mSignatures.empty();
 
    if (partMetaData()->isSigned && content()) {
        mOtp->nodeHelper()->setSignatureState(content(), KMMsgFullySigned);
        if (textNode) {
            return;
        }
 
        mOtp->nodeHelper()->setPartMetaData(content(), *partMetaData());
 
        if (!mVerifiedText.isEmpty()) {
            auto tempNode = new KMime::Content();
            tempNode->setContent(KMime::CRLFtoLF(mVerifiedText.constData()));
            tempNode->parse();
 
            if (!tempNode->head().isEmpty()) {
                tempNode->contentDescription()->from7BitString("signed data");
            }
            mOtp->nodeHelper()->attachExtraContent(content(), tempNode);
 
            parseInternal(tempNode, false);
        }
    }
}
 
QString SignedMessagePart::plaintextContent() const
{
    if (!content()) {
        return MessagePart::text();
    } else {
        return {};
    }
}
 
QString SignedMessagePart::htmlContent() const
{
    if (!content()) {
        return MessagePart::text();
    } else {
        return {};
    }
}
 
const QGpgME::Protocol *SignedMessagePart::cryptoProto() const
{
    return mCryptoProto;
}
 
QString SignedMessagePart::fromAddress() const
{
    return mFromAddress;
}
 
bool SignedMessagePart::hasHeader(const char *headerType) const
{
    if (content()) {
        return content()->hasHeader(headerType);
    }
    return false;
}
 
const KMime::Headers::Base *MimeTreeParser::SignedMessagePart::header(const char *headerType) const
{
    if (content()) {
        return content()->headerByType(headerType);
    }
    return nullptr;
}
 
QList<KMime::Headers::Base *> SignedMessagePart::headers(const char *headerType) const
{
    if (content()) {
        return content()->headersByType(headerType);
    }
    return {};
}
 
//-----CryptMessageBlock---------------------
EncryptedMessagePart::EncryptedMessagePart(ObjectTreeParser *otp,
                                           const QString &text,
                                           const QGpgME::Protocol *cryptoProto,
                                           const QString &fromAddress,
                                           KMime::Content *node)
    : MessagePart(otp, text)
    , mPassphraseError(false)
    , mNoSecKey(false)
    , mDecryptMessage(false)
    , mCryptoProto(cryptoProto)
    , mFromAddress(fromAddress)
    , mMementoName("decryptverify")
{
    setContent(node);
    partMetaData()->technicalProblem = (mCryptoProto == nullptr);
    partMetaData()->isSigned = false;
    partMetaData()->isGoodSignature = false;
    partMetaData()->isEncrypted = false;
    partMetaData()->isDecryptable = false;
    partMetaData()->status = i18n("Wrong Crypto Plug-In.");
    partMetaData()->status_code = GPGME_SIG_STAT_NONE;
}
 
EncryptedMessagePart::~EncryptedMessagePart() = default;
 
void EncryptedMessagePart::setDecryptMessage(bool decrypt)
{
    mDecryptMessage = decrypt;
}
 
bool EncryptedMessagePart::decryptMessage() const
{
    return mDecryptMessage;
}
 
void EncryptedMessagePart::setIsEncrypted(bool encrypted)
{
    partMetaData()->isEncrypted = encrypted;
}
 
bool EncryptedMessagePart::isEncrypted() const
{
    return partMetaData()->isEncrypted;
}
 
bool EncryptedMessagePart::isDecryptable() const
{
    return partMetaData()->isDecryptable;
}
 
bool EncryptedMessagePart::isNoSecKey() const
{
    return mNoSecKey;
}
 
bool EncryptedMessagePart::passphraseError() const
{
    return mPassphraseError;
}
 
QByteArray EncryptedMessagePart::mementoName() const
{
    return mMementoName;
}
 
void EncryptedMessagePart::setMementoName(const QByteArray &name)
{
    mMementoName = name;
}
 
void EncryptedMessagePart::startDecryption(const QByteArray &text, QByteArrayView aCodec)
{
    auto content = new KMime::Content;
    content->setBody(text);
    content->parse();
 
    startDecryption(content);
 
    if (!partMetaData()->inProgress && partMetaData()->isDecryptable) {
        QStringDecoder codec(aCodec.constData());
        if (hasSubParts()) {
            auto _mp = (subParts()[0]).dynamicCast<SignedMessagePart>();
            if (_mp) {
                _mp->setText(codec.decode(mDecryptedData));
            } else {
                setText(codec.decode(mDecryptedData));
            }
        } else {
            setText(codec.decode(mDecryptedData));
        }
    }
    delete content;
}
 
bool EncryptedMessagePart::okDecryptMIME(KMime::Content &data)
{
    mPassphraseError = false;
    partMetaData()->inProgress = false;
    partMetaData()->errorText.clear();
    partMetaData()->auditLogError = GpgME::Error();
    partMetaData()->auditLog.clear();
    bool bDecryptionOk = false;
    bool cannotDecrypt = false;
    NodeHelper *nodeHelper = mOtp->nodeHelper();
 
    Q_ASSERT(decryptMessage());
 
    const QByteArray _mementoName = mementoName();
    // Check whether the memento contains a result from last time:
    const auto *m = dynamic_cast<CompositeMemento *>(nodeHelper->bodyPartMemento(&data, _mementoName));
 
    Q_ASSERT(!m || mCryptoProto); // No CryptoPlugin and having a bodyPartMemento -> there is something completely wrong
 
    if (!m && mCryptoProto) {
        QGpgME::DecryptVerifyJob *job = mCryptoProto->decryptVerifyJob();
        if (!job) {
            cannotDecrypt = true;
        } else {
            const QByteArray ciphertext = data.decodedContent();
            auto newM = new CompositeMemento();
            newM->addMemento(new KeyCacheMemento(Kleo::KeyCache::mutableInstance(), toGpgMeProtocol(mCryptoProto)));
            newM->addMemento(new DecryptVerifyBodyPartMemento(job, ciphertext));
            if (mOtp->allowAsync()) {
                QObject::connect(newM, &CryptoBodyPartMemento::update, nodeHelper, &NodeHelper::update);
                if (newM->start()) {
                    partMetaData()->inProgress = true;
                    mOtp->mHasPendingAsyncJobs = true;
                } else {
                    m = newM;
                }
            } else {
                newM->exec();
                m = newM;
            }
 
            nodeHelper->setBodyPartMemento(&data, _mementoName, newM);
        }
    } else if (m && m->isRunning()) {
        partMetaData()->inProgress = true;
        mOtp->mHasPendingAsyncJobs = true;
        m = nullptr;
    }
 
    if (m) {
        {
            const auto *kcm = m->memento<KeyCacheMemento>();
            if (kcm) {
                mKeyCache = kcm->keyCache();
            }
        }
        auto *decryptMemento = m->memento<DecryptVerifyBodyPartMemento>();
        const QByteArray &plainText = decryptMemento->plainText();
        const GpgME::DecryptionResult &decryptResult = decryptMemento->decryptResult();
        const GpgME::VerificationResult &verifyResult = decryptMemento->verifyResult();
        partMetaData()->isSigned = !verifyResult.signatures().empty();
 
        if (partMetaData()->isSigned) {
            auto subPart = SignedMessagePart::Ptr(new SignedMessagePart(mOtp, MessagePart::text(), mCryptoProto, mFromAddress, content()));
            subPart->setVerificationResult(m, nullptr);
            appendSubPart(subPart);
        }
 
        mDecryptRecipients.clear();
        bDecryptionOk = !decryptResult.error();
 
        //         std::stringstream ss;
        //         ss << decryptResult << '\n' << verifyResult;
        //         qCDebug(MIMETREEPARSER_LOG) << ss.str().c_str();
 
        for (const auto &recipient : decryptResult.recipients()) {
            if (!recipient.status()) {
                bDecryptionOk = true;
            }
            GpgME::Key key;
            key = mKeyCache->findByKeyIDOrFingerprint(recipient.keyID());
            if (key.isNull()) {
                auto ret = mKeyCache->findSubkeysByKeyID({recipient.keyID()});
                if (ret.size() == 1) {
                    key = ret.front().parent();
                }
                if (key.isNull()) {
                    qCDebug(MIMETREEPARSER_LOG) << "Found no Key for KeyID " << recipient.keyID();
                }
            }
            mDecryptRecipients.emplace_back(recipient, key);
        }
 
        if (!bDecryptionOk && partMetaData()->isSigned) {
            // Only a signed part
            partMetaData()->isEncrypted = false;
            bDecryptionOk = true;
            mDecryptedData = plainText;
        } else {
            mPassphraseError = decryptResult.error().isCanceled() || decryptResult.error().code() == GPG_ERR_NO_SECKEY;
            partMetaData()->isEncrypted = bDecryptionOk || decryptResult.error().code() != GPG_ERR_NO_DATA;
 
            if (decryptResult.error().isCanceled()) {
                setDecryptMessage(false);
            }
 
            partMetaData()->errorText = QString::fromLocal8Bit(decryptResult.error().asString());
            if (Kleo::DeVSCompliance::isCompliant()) {
                partMetaData()->isCompliant = decryptResult.isDeVs();
                partMetaData()->compliance = Kleo::DeVSCompliance::name(decryptResult.isDeVs());
            } else {
                partMetaData()->isCompliant = true;
            }
            if (partMetaData()->isEncrypted && decryptResult.numRecipients() > 0) {
                partMetaData()->keyId = decryptResult.recipient(0).keyID();
            }
 
            if (bDecryptionOk) {
                mDecryptedData = plainText;
            } else {
                mNoSecKey = true;
                const auto decryRecipients = decryptResult.recipients();
                for (const GpgME::DecryptionResult::Recipient &recipient : decryRecipients) {
                    mNoSecKey &= (recipient.status().code() == GPG_ERR_NO_SECKEY);
                }
                if (!mPassphraseError && !mNoSecKey) { // GpgME do not detect passphrase error correctly
                    mPassphraseError = true;
                }
            }
        }
    }
 
    if (!bDecryptionOk) {
        QString cryptPlugLibName;
        if (mCryptoProto) {
            cryptPlugLibName = mCryptoProto->name();
        }
 
        if (!mCryptoProto) {
            partMetaData()->errorText = i18n("No appropriate crypto plug-in was found.");
        } else if (cannotDecrypt) {
            partMetaData()->errorText = i18n("Crypto plug-in \"%1\" cannot decrypt messages.", cryptPlugLibName);
        } else if (!passphraseError()) {
            partMetaData()->errorText = i18n("Crypto plug-in \"%1\" could not decrypt the data.", cryptPlugLibName) + QLatin1StringView("<br />")
                + i18n("Error: %1", partMetaData()->errorText);
        }
    }
    return bDecryptionOk;
}
 
void EncryptedMessagePart::startDecryption(KMime::Content *data)
{
    if (!content() && !data) {
        return;
    }
 
    if (!data) {
        data = content();
    }
 
    partMetaData()->isEncrypted = true;
 
    bool bOkDecrypt = okDecryptMIME(*data);
 
    if (partMetaData()->inProgress) {
        return;
    }
    partMetaData()->isDecryptable = bOkDecrypt;
 
    if (!partMetaData()->isDecryptable) {
        setText(QString::fromUtf8(mDecryptedData.constData()));
    }
 
    if (partMetaData()->isEncrypted && !decryptMessage()) {
        partMetaData()->isDecryptable = true;
    }
 
    if (content() && !partMetaData()->isSigned) {
        mOtp->nodeHelper()->setPartMetaData(content(), *partMetaData());
 
        if (decryptMessage()) {
            auto tempNode = new KMime::Content();
            tempNode->setContent(KMime::CRLFtoLF(mDecryptedData.constData()));
            tempNode->parse();
 
            if (!tempNode->head().isEmpty()) {
                tempNode->contentDescription()->from7BitString("encrypted data");
            }
            mOtp->nodeHelper()->attachExtraContent(content(), tempNode);
 
            parseInternal(tempNode, false);
        }
    }
}
 
QString EncryptedMessagePart::plaintextContent() const
{
    if (!content()) {
        return MessagePart::text();
    } else {
        return {};
    }
}
 
QString EncryptedMessagePart::htmlContent() const
{
    if (!content()) {
        return MessagePart::text();
    } else {
        return {};
    }
}
 
QString EncryptedMessagePart::text() const
{
    if (hasSubParts()) {
        auto _mp = (subParts()[0]).dynamicCast<SignedMessagePart>();
        if (_mp) {
            return _mp->text();
        } else {
            return MessagePart::text();
        }
    } else {
        return MessagePart::text();
    }
}
 
const QGpgME::Protocol *EncryptedMessagePart::cryptoProto() const
{
    return mCryptoProto;
}
 
QString EncryptedMessagePart::fromAddress() const
{
    return mFromAddress;
}
 
const std::vector<std::pair<GpgME::DecryptionResult::Recipient, GpgME::Key>> &EncryptedMessagePart::decryptRecipients() const
{
    return mDecryptRecipients;
}
 
bool EncryptedMessagePart::hasHeader(const char *headerType) const
{
    const auto extraContent = mOtp->nodeHelper()->decryptedNodeForContent(content());
    if (extraContent) {
        return nodeHelper()->hasMailHeader(headerType, extraContent);
    }
    return false;
}
 
const KMime::Headers::Base *EncryptedMessagePart::header(const char *headerType) const
{
    const auto extraContent = mOtp->nodeHelper()->decryptedNodeForContent(content());
    if (extraContent) {
        return nodeHelper()->mailHeaderAsBase(headerType, extraContent);
    }
    return nullptr;
}
 
QList<KMime::Headers::Base *> EncryptedMessagePart::headers(const char *headerType) const
{
    const auto extraContent = mOtp->nodeHelper()->decryptedNodeForContent(content());
    if (extraContent) {
        return nodeHelper()->headers(headerType, extraContent);
    }
    return {};
}
 
EncapsulatedRfc822MessagePart::EncapsulatedRfc822MessagePart(ObjectTreeParser *otp, KMime::Content *node, const KMime::Message::Ptr &message)
    : MessagePart(otp, QString())
    , mMessage(message)
{
    setContent(node);
    partMetaData()->isEncrypted = false;
    partMetaData()->isSigned = false;
    partMetaData()->isEncapsulatedRfc822Message = true;
 
    mOtp->nodeHelper()->setNodeDisplayedEmbedded(node, true);
    mOtp->nodeHelper()->setPartMetaData(node, *partMetaData());
 
    if (!mMessage) {
        qCWarning(MIMETREEPARSER_LOG) << "Node is of type message/rfc822 but doesn't have a message!";
        return;
    }
 
    // The link to "Encapsulated message" is clickable, therefore the temp file needs to exists,
    // since the user can click the link and expect to have normal attachment operations there.
    mOtp->nodeHelper()->writeNodeToTempFile(message.data());
 
    parseInternal(message.data(), false);
}
 
EncapsulatedRfc822MessagePart::~EncapsulatedRfc822MessagePart() = default;
 
QString EncapsulatedRfc822MessagePart::text() const
{
    return renderInternalText();
}
 
void EncapsulatedRfc822MessagePart::fix() const
{
}
 
const KMime::Message::Ptr EncapsulatedRfc822MessagePart::message() const
{
    return mMessage;
}
 
#include "moc_messagepart.cpp"