File indexing completed on 2024-04-21 16:35:04

 /*
  * SPDX-FileCopyrightText: 2017 Elvis Angelaccio <elvis.angelaccio@kde.org>
  *
  * SPDX-License-Identifier: GPL-2.0-or-later
  *
  */
 
 #include "decryptjob.h"
 #include "symmydebug.h"
 
 #include <KIO/CopyJob>
 
 #include <QMimeDatabase>
 #include <QTemporaryFile>
 
 #include <QGpgME/DecryptJob>
 #include <QGpgME/Protocol>
 
 #include <gpgme++/context.h>
 #include <gpgme++/key.h>
 
 using namespace GpgME;
 using namespace QGpgME;
 
 namespace Symmy
 {
 
 DecryptJob::DecryptJob(const QString &passphrase, const QString &ciphertextFilename) : Job {passphrase}
 {
     m_ciphertext = std::make_shared<QFile>(ciphertextFilename);
     m_plaintext = std::make_shared<QTemporaryFile>(plaintextFilename());
 }
 
 DecryptJob::~DecryptJob()
 {
 }
 
 QString DecryptJob::ciphertextFilename() const
 {
     return m_ciphertext->fileName();
 }
 
 QString DecryptJob::plaintextFilename() const
 {
     auto filename = ciphertextFilename();
     const auto pgpSuffixes = QMimeDatabase{}.mimeTypeForName(QStringLiteral("application/pgp-encrypted")).suffixes();
     for (const auto &suffix : pgpSuffixes) {
         if (filename.endsWith(suffix)) {
             filename.chop(suffix.length() + 1); // dot is not included in the suffix
             break;
         }
     }
 
     return filename;
 }
 
 void DecryptJob::doWork()
 {
     if (not m_ciphertext->open(QIODevice::ReadOnly | QIODevice::Unbuffered)) {
         qCDebug(SYMMY) << "Could not open ciphertext file" << ciphertextFilename();
         emitResult();
         return;
     }
 
     if (not m_plaintext->open()) {
         qCDebug(SYMMY) << "Could not open plaintext file" << m_plaintext->fileName();
         emitResult();
         return;
     }
 
     auto decryptJob = openpgp()->decryptJob();
     auto context = QGpgME::Job::context(decryptJob);
     context->setPassphraseProvider(this);
     context->setPinentryMode(GpgME::Context::PinentryLoopback);
 
     setJob(decryptJob);
 
     connect(decryptJob, &QGpgME::DecryptJob::result, this, &DecryptJob::slotResult);
     connect(decryptJob, &QGpgME::Job::progress, this, [this](const QString &, int current, int total) {
         const auto ratio = static_cast<float>(current) / total;
         setPercent(static_cast<unsigned long>(100 * ratio));
     });
 
     decryptJob->start(m_ciphertext, m_plaintext);
 }
 
 void DecryptJob::slotResult(const DecryptionResult &, const QByteArray &, const QString &, const Error &)
 {
     qCDebug(SYMMY) << "Decryption job finished, plaintext size:" << m_plaintext->size();
 
     if (m_plaintext->size() == 0) {
         setError(UserDefinedError);
         emitResult();
         return;
     }
 
     auto plaintextPath = m_plaintext->fileName();
     // Drop the ".XXXXXX" suffix of QTemporaryFile.
     plaintextPath.chop(7);
     qCDebug(SYMMY) << "Moving temporary file" << QUrl::fromLocalFile(m_plaintext->fileName()) << "to" << QUrl::fromLocalFile(plaintextPath);
     auto job = KIO::move(QUrl::fromLocalFile(m_plaintext->fileName()), QUrl::fromLocalFile(plaintextPath));
     connect(job, &KJob::result, this, &DecryptJob::emitResult);
 }
 
 }