aboutsummaryrefslogtreecommitdiffstats
path: root/src/ui/function/GpgOperaHelper.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/ui/function/GpgOperaHelper.cpp')
-rw-r--r--src/ui/function/GpgOperaHelper.cpp346
1 files changed, 346 insertions, 0 deletions
diff --git a/src/ui/function/GpgOperaHelper.cpp b/src/ui/function/GpgOperaHelper.cpp
new file mode 100644
index 00000000..35c4ba35
--- /dev/null
+++ b/src/ui/function/GpgOperaHelper.cpp
@@ -0,0 +1,346 @@
+/**
+ * Copyright (C) 2021-2024 Saturneric <[email protected]>
+ *
+ * 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 <[email protected]> starting on May 12, 2021.
+ *
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ *
+ */
+
+#include "GpgOperaHelper.h"
+
+#include "core/function/gpg/GpgFileOpera.h"
+#include "core/function/result_analyse/GpgDecryptResultAnalyse.h"
+#include "core/function/result_analyse/GpgEncryptResultAnalyse.h"
+#include "core/function/result_analyse/GpgSignResultAnalyse.h"
+#include "core/model/GpgDecryptResult.h"
+#include "core/model/GpgEncryptResult.h"
+#include "core/model/GpgSignResult.h"
+#include "core/utils/GpgUtils.h"
+#include "ui/dialog/WaitingDialog.h"
+
+namespace GpgFrontend::UI {
+
+void GpgOperaHelper::BuildOperas(QSharedPointer<GpgOperaContextBasement>& base,
+ int category, int channel,
+ const GpgOperaFactory& f) {
+ assert(base != nullptr);
+
+ auto context = GetGpgOperaContextFromBasement(base, category);
+ if (context == nullptr) return;
+
+ assert(context->paths.size() == context->o_paths.size());
+
+ for (int i = 0; i < context->paths.size(); i++) {
+ context->base->operas.push_back(f(context, channel, i));
+ }
+}
+
+template <typename AnalyseType>
+void HandleExtraLogicIfNeeded(const QSharedPointer<GpgOperaContext>& context,
+ const AnalyseType& analyse) {}
+
+template <>
+void HandleExtraLogicIfNeeded<GpgVerifyResultAnalyse>(
+ const QSharedPointer<GpgOperaContext>& context,
+ const GpgVerifyResultAnalyse& analyse) {
+ context->base->unknown_fprs.append(analyse.GetUnknownSignatures());
+}
+
+template <typename ResultType, typename AnalyseType, typename OperaFunc>
+auto GpgOperaHelper::BuildSimpleGpgFileOperasHelper(
+ QSharedPointer<GpgOperaContext>& context, int channel, int index,
+ OperaFunc opera_func) -> OperaWaitingCb {
+ const auto& path = context->paths[index];
+ const auto& o_path = context->o_paths[index];
+ auto& opera_results = context->base->opera_results;
+
+ return [=, &opera_results](const OperaWaitingHd& op_hd) {
+ opera_func(
+ path, o_path,
+ [=, &opera_results](GpgError err, const DataObjectPtr& data_obj) {
+ // stop waiting
+ op_hd();
+
+ if (CheckGpgError(err) == GPG_ERR_USER_1 || data_obj == nullptr ||
+ !data_obj->Check<ResultType>()) {
+ opera_results.append(
+ {-1, "# " + tr("Critical Error"), QFileInfo(path).fileName()});
+ return;
+ }
+
+ auto result = ExtractParams<ResultType>(data_obj, 0);
+ auto result_analyse = AnalyseType(channel, err, result);
+ result_analyse.Analyse();
+
+ HandleExtraLogicIfNeeded(context, result_analyse);
+
+ opera_results.append(
+ {result_analyse.GetStatus(), result_analyse.GetResultReport(),
+ QFileInfo(path.isEmpty() ? o_path : path).fileName()});
+ });
+ };
+}
+
+template <typename ResultTypeA, typename AnalyseTypeA, typename ResultTypeB,
+ typename AnalyseTypeB, typename OperaFunc>
+auto GpgOperaHelper::BuildComplexGpgFileOperasHelper(
+ QSharedPointer<GpgOperaContext>& context, int channel, int index,
+ OperaFunc opera_func) -> OperaWaitingCb {
+ const auto& path = context->paths[index];
+ const auto& o_path = context->o_paths[index];
+ auto& opera_results = context->base->opera_results;
+
+ return [=, &opera_results](const OperaWaitingHd& op_hd) {
+ opera_func(
+ path, o_path,
+ [=, &opera_results](GpgError err, const DataObjectPtr& data_obj) {
+ // stop waiting
+ op_hd();
+
+ if (CheckGpgError(err) == GPG_ERR_USER_1 || data_obj == nullptr ||
+ !data_obj->Check<ResultTypeA, ResultTypeB>()) {
+ opera_results.append(
+ {-1, "# " + tr("Critical Error"), QFileInfo(path).fileName()});
+ return;
+ }
+
+ auto result_1 = ExtractParams<ResultTypeA>(data_obj, 0);
+ auto result_2 = ExtractParams<ResultTypeB>(data_obj, 1);
+
+ auto result_analyse_1 = AnalyseTypeA(channel, err, result_1);
+ result_analyse_1.Analyse();
+
+ HandleExtraLogicIfNeeded(context, result_analyse_1);
+
+ auto result_analyse_2 = AnalyseTypeB(channel, err, result_2);
+ result_analyse_2.Analyse();
+
+ HandleExtraLogicIfNeeded(context, result_analyse_2);
+
+ opera_results.append(
+ {std::min(result_analyse_1.GetStatus(),
+ result_analyse_2.GetStatus()),
+ result_analyse_1.GetResultReport() +
+ result_analyse_2.GetResultReport(),
+ QFileInfo(path.isEmpty() ? o_path : path).fileName()});
+ });
+ };
+}
+
+template <typename EncryptFuncSymmetric, typename EncryptFuncKeys>
+auto BuildOperasEncryptHelper(
+ QSharedPointer<GpgOperaContext>& context, int channel, int index,
+ EncryptFuncSymmetric encrypt_symmetric,
+ EncryptFuncKeys encrypt_with_keys) -> OperaWaitingCb {
+ if (context->base->keys.isEmpty()) {
+ return GpgOperaHelper::BuildSimpleGpgFileOperasHelper<
+ GpgEncryptResult, GpgEncryptResultAnalyse>(
+ context, channel, index,
+ [=](const QString& path, const QString& o_path, const auto& callback) {
+ encrypt_symmetric(path, o_path, callback);
+ });
+ }
+
+ return GpgOperaHelper::BuildSimpleGpgFileOperasHelper<
+ GpgEncryptResult, GpgEncryptResultAnalyse>(
+ context, channel, index,
+ [=](const QString& path, const QString& o_path, const auto& callback) {
+ encrypt_with_keys(path, o_path, callback);
+ });
+}
+
+auto GpgOperaHelper::BuildOperasFileEncrypt(
+ QSharedPointer<GpgOperaContext>& context, int channel,
+ int index) -> OperaWaitingCb {
+ return BuildOperasEncryptHelper(
+ context, channel, index,
+ [context, channel](const QString& path, const QString& o_path,
+ const auto& callback) {
+ GpgFileOpera::GetInstance(channel).EncryptFileSymmetric(
+ path, context->base->ascii, o_path, callback);
+ },
+ [context, channel](const QString& path, const QString& o_path,
+ const auto& callback) {
+ GpgFileOpera::GetInstance(channel).EncryptFile(
+ context->base->keys, path, context->base->ascii, o_path, callback);
+ });
+}
+
+auto GpgOperaHelper::BuildOperasDirectoryEncrypt(
+ QSharedPointer<GpgOperaContext>& context, int channel,
+ int index) -> OperaWaitingCb {
+ return BuildOperasEncryptHelper(
+ context, channel, index,
+ [context, channel](const QString& path, const QString& o_path,
+ const auto& callback) {
+ GpgFileOpera::GetInstance(channel).EncryptDirectorySymmetric(
+ path, context->base->ascii, o_path, callback);
+ },
+ [context, channel](const QString& path, const QString& o_path,
+ const auto& callback) {
+ GpgFileOpera::GetInstance(channel).EncryptDirectory(
+ context->base->keys, path, context->base->ascii, o_path, callback);
+ });
+}
+
+auto GpgOperaHelper::BuildOperasFileDecrypt(
+ QSharedPointer<GpgOperaContext>& context, int channel,
+ int index) -> OperaWaitingCb {
+ return BuildSimpleGpgFileOperasHelper<GpgDecryptResult,
+ GpgDecryptResultAnalyse>(
+ context, channel, index,
+ [channel](const QString& path, const QString& o_path,
+ const auto& callback) {
+ GpgFileOpera::GetInstance(channel).DecryptFile(path, o_path, callback);
+ });
+}
+
+auto GpgOperaHelper::BuildOperasArchiveDecrypt(
+ QSharedPointer<GpgOperaContext>& context, int channel,
+ int index) -> OperaWaitingCb {
+ return BuildSimpleGpgFileOperasHelper<GpgDecryptResult,
+ GpgDecryptResultAnalyse>(
+ context, channel, index,
+ [channel](const QString& path, const QString& o_path,
+ const auto& callback) {
+ GpgFileOpera::GetInstance(channel).DecryptArchive(path, o_path,
+ callback);
+ });
+}
+
+auto GpgOperaHelper::BuildOperasFileSign(
+ QSharedPointer<GpgOperaContext>& context, int channel,
+ int index) -> OperaWaitingCb {
+ return BuildSimpleGpgFileOperasHelper<GpgSignResult, GpgSignResultAnalyse>(
+ context, channel, index,
+ [channel, context](const QString& path, const QString& o_path,
+ const auto& callback) {
+ GpgFileOpera::GetInstance(channel).SignFile(
+ context->base->keys, path, context->base->ascii, o_path, callback);
+ });
+}
+
+auto GpgOperaHelper::BuildOperasFileVerify(
+ QSharedPointer<GpgOperaContext>& context, int channel,
+ int index) -> OperaWaitingCb {
+ return BuildSimpleGpgFileOperasHelper<GpgVerifyResult,
+ GpgVerifyResultAnalyse>(
+ context, channel, index,
+ [channel](const QString& path, const QString& o_path,
+ const auto& callback) {
+ GpgFileOpera::GetInstance(channel).VerifyFile(o_path, path, callback);
+ });
+}
+
+auto GpgOperaHelper::BuildOperasFileEncryptSign(
+ QSharedPointer<GpgOperaContext>& context, int channel,
+ int index) -> OperaWaitingCb {
+ return BuildComplexGpgFileOperasHelper<GpgEncryptResult,
+ GpgEncryptResultAnalyse, GpgSignResult,
+ GpgSignResultAnalyse>(
+ context, channel, index,
+ [channel, context](const QString& path, const QString& o_path,
+ const auto& callback) {
+ GpgFileOpera::GetInstance(channel).EncryptSignFile(
+ context->base->keys, context->base->singer_keys, path,
+ context->base->ascii, o_path, callback);
+ });
+}
+
+auto GpgOperaHelper::BuildOperasDirectoryEncryptSign(
+ QSharedPointer<GpgOperaContext>& context, int channel,
+ int index) -> OperaWaitingCb {
+ return BuildComplexGpgFileOperasHelper<GpgEncryptResult,
+ GpgEncryptResultAnalyse, GpgSignResult,
+ GpgSignResultAnalyse>(
+ context, channel, index,
+ [channel, context](const QString& path, const QString& o_path,
+ const auto& callback) {
+ GpgFileOpera::GetInstance(channel).EncryptSignDirectory(
+ context->base->keys, context->base->singer_keys, path,
+ context->base->ascii, o_path, callback);
+ });
+}
+
+auto GpgOperaHelper::BuildOperasFileDecryptVerify(
+ QSharedPointer<GpgOperaContext>& context, int channel,
+ int index) -> OperaWaitingCb {
+ return BuildComplexGpgFileOperasHelper<
+ GpgDecryptResult, GpgDecryptResultAnalyse, GpgVerifyResult,
+ GpgVerifyResultAnalyse>(
+ context, channel, index,
+ [channel](const QString& path, const QString& o_path,
+ const auto& callback) {
+ GpgFileOpera::GetInstance(channel).DecryptVerifyFile(path, o_path,
+ callback);
+ });
+}
+
+auto GpgOperaHelper::BuildOperasArchiveDecryptVerify(
+ QSharedPointer<GpgOperaContext>& context, int channel,
+ int index) -> OperaWaitingCb {
+ return BuildComplexGpgFileOperasHelper<
+ GpgDecryptResult, GpgDecryptResultAnalyse, GpgVerifyResult,
+ GpgVerifyResultAnalyse>(
+ context, channel, index,
+ [channel](const QString& path, const QString& o_path,
+ const auto& callback) {
+ GpgFileOpera::GetInstance(channel).DecryptVerifyArchive(path, o_path,
+ callback);
+ });
+}
+
+void GpgOperaHelper::WaitForMultipleOperas(
+ QWidget* parent, const QString& title,
+ const QContainer<OperaWaitingCb>& operas) {
+ QEventLoop looper;
+ QPointer<WaitingDialog> const dialog = new WaitingDialog(title, true, parent);
+ connect(dialog, &QDialog::finished, &looper, &QEventLoop::quit);
+ connect(dialog, &QDialog::finished, dialog, &QDialog::deleteLater);
+ dialog->show();
+
+ std::atomic<int> remaining_tasks(static_cast<int>(operas.size()));
+ const auto tasks_count = operas.size();
+
+ for (const auto& opera : operas) {
+ QTimer::singleShot(64, parent, [=, &remaining_tasks]() {
+ opera([dialog, &remaining_tasks, tasks_count]() {
+ if (dialog == nullptr) return;
+
+ const auto pg_value =
+ static_cast<double>(tasks_count - remaining_tasks + 1) * 100.0 /
+ static_cast<double>(tasks_count);
+ emit dialog->SignalUpdateValue(static_cast<int>(pg_value));
+ QCoreApplication::processEvents();
+
+ if (--remaining_tasks == 0) {
+ dialog->close();
+ dialog->accept();
+ }
+ });
+ });
+ }
+
+ looper.exec();
+}
+} // namespace GpgFrontend::UI
generated by cgit v1.2.3 (git 2.25.1) at 2025-04-27 14:50:04 +0000