File indexing completed on 2024-04-21 05:50:43

 /*
     SPDX-FileCopyrightText: 2008, 2009, 2010, 2011, 2012, 2013 Rolf Eike Beer <kde@opensource.sf-tec.de>
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #include "kgpgtextorfiletransaction.h"
 #include "kgpg_general_debug.h"
 
 #include "gpgproc.h"
 
 #include <KIO/JobTracker>
 #include <KIO/StoredTransferJob>
 #include <KIO/FileCopyJob>
 
 #include <QRegularExpression>
 #include <QTemporaryFile>
 
 KGpgTextOrFileTransaction::KGpgTextOrFileTransaction(QObject *parent, const QString &text, const bool allowChaining)
     : KGpgTransaction(parent, allowChaining)
 {
     setText(text);
 }
 
 KGpgTextOrFileTransaction::KGpgTextOrFileTransaction(QObject *parent, const QList<QUrl> &files, const bool allowChaining)
     : KGpgTransaction(parent, allowChaining)
 {
     setUrls(files);
 }
 
 KGpgTextOrFileTransaction::~KGpgTextOrFileTransaction()
 {
     cleanUrls();
 }
 
 void
 KGpgTextOrFileTransaction::setText(const QString &text)
 {
     m_text = text;
     cleanUrls();
 
     int begin = text.indexOf(QRegularExpression(QStringLiteral("^(.*\n)?-----BEGIN PGP [A-Z ]*-----\r?\n")));
     if (begin < 0)
         return;
 
     // find the end of the BEGIN PGP ... line
     static const QChar lf = QLatin1Char('\n');
     begin = text.indexOf(lf, begin);
     Q_ASSERT(begin > 0);
 
     // now loop until either an empty line is found (end of header) or
     // a line beginning with Charset is found. If the latter, use the
     // charset found there as hint for the following operation
     int nextlf;
     begin++;
     while ((nextlf = text.indexOf(lf, begin)) > 0) {
         static const QChar cr = QLatin1Char('\r');
         if ((nextlf == begin) || ((nextlf == begin + 1) && (text[begin] == cr)))
             break;
 
         const QString charset = QLatin1String("Charset: ");
         if (QStringView(text).mid(begin, charset.length()) == charset) {
             QString cs = text.mid(begin + charset.length(), nextlf - begin - charset.length());
             if (!getProcess()->setCodec(cs.toLatin1()))
                 qCDebug(KGPG_LOG_GENERAL) << "unsupported charset found in header" << cs;
             break;
         }
         begin = nextlf + 1;
     }
 
 
 }
 
 void
 KGpgTextOrFileTransaction::setUrls(const QList<QUrl> &files)
 {
     m_text.clear();
     m_inpfiles = files;
 }
 
 bool
 KGpgTextOrFileTransaction::preStart()
 {
     QStringList locfiles;
 
     for (const QUrl &url : std::as_const(m_inpfiles)) {
         if (url.isLocalFile()) {
             locfiles.append(url.toLocalFile());
         } else {
             QTemporaryFile tmpFile;
             tmpFile.open();
 
             auto copyJob = KIO::file_copy(url, QUrl::fromLocalFile(tmpFile.fileName()));
             copyJob->exec();
             if (!copyJob->error()) {
                 tmpFile.setAutoRemove(false);
                 m_tempfiles.append(tmpFile.fileName());
             } else {
                 m_messages.append(copyJob->errorString());
                 cleanUrls();
                 setSuccess(TS_KIO_FAILED);
                 return false;
             }
         }
     }
 
     if (locfiles.isEmpty() && m_tempfiles.isEmpty() && m_text.isEmpty() && !hasInputTransaction()) {
         setSuccess(TS_MSG_SEQUENCE);
         return false;
     }
 
     QStringList args(QLatin1String("--status-fd=1"));
 
     args << command();
     // if the input is not stdin set command-fd so GnuPG
     // can ask if e.g. the file already exists
     bool hasCFd = !args.contains(QLatin1String("--command-fd=0"));
     if (!locfiles.isEmpty() || !m_tempfiles.isEmpty()) {
         if (!hasCFd)
             args << QLatin1String("--command-fd=0");
         m_closeInput = false;
     } else if (closeInputAfterText()) {
         m_closeInput = true;
     } else {
         m_closeInput = hasCFd;
     }
     if (locfiles.count() + m_tempfiles.count() > 1)
         args << QLatin1String("--multifile");
     args << locfiles << m_tempfiles;
     addArguments(args);
 
     return true;
 }
 
 void
 KGpgTextOrFileTransaction::postStart()
 {
     if (!m_text.isEmpty()){
         GPGProc *proc = getProcess();
         proc->write(m_text.toUtf8());
         if (m_closeInput)
             proc->closeWriteChannel();
     }
 }
 
 bool
 KGpgTextOrFileTransaction::nextLine(const QString &line)
 {
     m_messages.append(line);
 
     return false;
 }
 
 void
 KGpgTextOrFileTransaction::finish()
 {
     if (getProcess()->exitCode() != 0) {
         setSuccess(TS_MSG_SEQUENCE);
     }
 }
 
 const QStringList &
 KGpgTextOrFileTransaction::getMessages() const
 {
     return m_messages;
 }
 
 void
 KGpgTextOrFileTransaction::cleanUrls()
 {
     for (const QString &u : std::as_const(m_tempfiles))
         QFile::remove(u);
 
     m_tempfiles.clear();
     m_locfiles.clear();
     m_inpfiles.clear();
 }
 
 const QList<QUrl> &
 KGpgTextOrFileTransaction::getInputFiles() const
 {
     return m_inpfiles;
 }