1 files changed, 182 insertions, 0 deletions
diff --git a/src/core/function/gpg/GpgFileOpera.cpp b/src/core/function/gpg/GpgFileOpera.cpp
new file mode 100644
index 00000000..7044353b
--- /dev/null
+++ b/src/core/function/gpg/GpgFileOpera.cpp
@@ -0,0 +1,182 @@
+/**
+ * Copyright (C) 2021 Saturneric
+ *
+ * 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 "GpgFileOpera.h"
+
+#include <memory>
+#include <string>
+
+#include "GpgConstants.h"
+#include "GpgBasicOperator.h"
+#include "function/FileOperator.h"
+
+GpgFrontend::GpgError GpgFrontend::GpgFileOpera::EncryptFile(
+    KeyListPtr keys, const std::string& in_path, const std::string& out_path,
+    GpgEncrResult& result, int _channel) {
+
+  std::string in_buffer;
+  if(!FileOperator::ReadFileStd(in_path, in_buffer)) {
+    throw std::runtime_error("read file error");
+  }
+  std::unique_ptr<std::string> out_buffer = nullptr;
+
+  auto err = GpgBasicOperator::GetInstance(_channel).Encrypt(
+      std::move(keys), in_buffer, out_buffer, result);
+
+  if (check_gpg_error_2_err_code(err) == GPG_ERR_NO_ERROR)
+    if (!FileOperator::WriteFileStd(out_path, *out_buffer)) {
+      throw std::runtime_error("write_buffer_to_file error");
+    };
+
+  return err;
+}
+
+GpgFrontend::GpgError GpgFrontend::GpgFileOpera::DecryptFile(
+    const std::string& in_path, const std::string& out_path,
+    GpgDecrResult& result) {
+  std::string in_buffer;
+  if(!FileOperator::ReadFileStd(in_path, in_buffer)) {
+    throw std::runtime_error("read file error");
+  }
+  std::unique_ptr<std::string> out_buffer;
+
+  auto err =
+      GpgBasicOperator::GetInstance().Decrypt(in_buffer, out_buffer, result);
+
+  assert(check_gpg_error_2_err_code(err) == GPG_ERR_NO_ERROR);
+
+  if (check_gpg_error_2_err_code(err) == GPG_ERR_NO_ERROR)
+    if (!FileOperator::WriteFileStd(out_path, *out_buffer)) {
+      throw std::runtime_error("write_buffer_to_file error");
+    };
+
+  return err;
+}
+
+gpgme_error_t GpgFrontend::GpgFileOpera::SignFile(KeyListPtr keys,
+                                                  const std::string& in_path,
+                                                  const std::string& out_path,
+                                                  GpgSignResult& result,
+                                                  int _channel) {
+  std::string in_buffer;
+  if(!FileOperator::ReadFileStd(in_path, in_buffer)) {
+    throw std::runtime_error("read file error");
+  }
+  std::unique_ptr<std::string> out_buffer;
+
+  auto err = GpgBasicOperator::GetInstance(_channel).Sign(
+      std::move(keys), in_buffer, out_buffer, GPGME_SIG_MODE_DETACH, result);
+
+  if (check_gpg_error_2_err_code(err) == GPG_ERR_NO_ERROR)
+    if (!FileOperator::WriteFileStd(out_path, *out_buffer)) {
+      throw std::runtime_error("write_buffer_to_file error");
+    };
+
+  return err;
+}
+
+gpgme_error_t GpgFrontend::GpgFileOpera::VerifyFile(
+    const std::string& data_path, const std::string& sign_path,
+    GpgVerifyResult& result, int _channel) {
+  std::string in_buffer;
+  if(!FileOperator::ReadFileStd(data_path, in_buffer)) {
+    throw std::runtime_error("read file error");
+  }
+  std::unique_ptr<std::string> sign_buffer = nullptr;
+  if (!sign_path.empty()) {
+    std::string sign_buffer_str;
+    if (!FileOperator::ReadFileStd(sign_path, sign_buffer_str)) {
+      throw std::runtime_error("read file error");
+    }
+    sign_buffer =
+        std::make_unique<std::string>(sign_buffer_str);
+  }
+  auto err = GpgBasicOperator::GetInstance(_channel).Verify(in_buffer, sign_buffer,
+                                                         result);
+  return err;
+}
+
+gpg_error_t GpgFrontend::GpgFileOpera::EncryptSignFile(
+    KeyListPtr keys, KeyListPtr signer_keys, const std::string& in_path,
+    const std::string& out_path, GpgEncrResult& encr_res,
+    GpgSignResult& sign_res, int _channel) {
+  std::string in_buffer;
+  if(!FileOperator::ReadFileStd(in_path, in_buffer)) {
+    throw std::runtime_error("read file error");
+  }
+  std::unique_ptr<std::string> out_buffer = nullptr;
+
+  auto err = GpgBasicOperator::GetInstance(_channel).EncryptSign(
+      std::move(keys), std::move(signer_keys), in_buffer, out_buffer, encr_res,
+      sign_res);
+
+  if (check_gpg_error_2_err_code(err) == GPG_ERR_NO_ERROR)
+    if (!FileOperator::WriteFileStd(out_path, *out_buffer)) {
+      throw std::runtime_error("write_buffer_to_file error");
+    };
+
+  return err;
+}
+
+gpg_error_t GpgFrontend::GpgFileOpera::DecryptVerifyFile(
+    const std::string& in_path, const std::string& out_path,
+    GpgDecrResult& decr_res, GpgVerifyResult& verify_res) {
+  std::string in_buffer;
+  if(!FileOperator::ReadFileStd(in_path, in_buffer)) {
+    throw std::runtime_error("read file error");
+  }
+
+  std::unique_ptr<std::string> out_buffer = nullptr;
+  auto err = GpgBasicOperator::GetInstance().DecryptVerify(in_buffer, out_buffer,
+                                                        decr_res, verify_res);
+
+  if (check_gpg_error_2_err_code(err) == GPG_ERR_NO_ERROR)
+    if (!FileOperator::WriteFileStd(out_path, *out_buffer)) {
+      throw std::runtime_error("write file error");
+    };
+
+  return err;
+}
+unsigned int GpgFrontend::GpgFileOpera::EncryptFileSymmetric(
+    const std::string& in_path, const std::string& out_path,
+    GpgFrontend::GpgEncrResult& result, int _channel) {
+  std::string in_buffer;
+  if(!FileOperator::ReadFileStd(in_path, in_buffer)) {
+    throw std::runtime_error("read file error");
+  }
+
+  std::unique_ptr<std::string> out_buffer;
+  auto err = GpgBasicOperator::GetInstance(_channel).EncryptSymmetric(
+      in_buffer, out_buffer, result);
+
+  if (check_gpg_error_2_err_code(err) == GPG_ERR_NO_ERROR)
+    if (!FileOperator::WriteFileStd(out_path, *out_buffer)) {
+      throw std::runtime_error("write_buffer_to_file error");
+    };
+
+  return err;
+}