1 files changed, 238 insertions, 0 deletions
diff --git a/src/ui/main_window/MainWindowServerSlotFunction.cpp b/src/ui/main_window/MainWindowServerSlotFunction.cpp
new file mode 100644
index 00000000..3a7e9f71
--- /dev/null
+++ b/src/ui/main_window/MainWindowServerSlotFunction.cpp
@@ -0,0 +1,238 @@
+/**
+ * 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.
+ *
+ * Foobar 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 Foobar.  If not, see <https://www.gnu.org/licenses/>.
+ *
+ * The initial version of the source code is inherited from gpg4usb-team.
+ * Their source code version also complies with GNU General Public License.
+ *
+ * The source code version of this software was modified and released
+ * by Saturneric<[email protected]> starting on May 12, 2021.
+ *
+ */
+
+#include "MainWindow.h"
+#include "server/ComUtils.h"
+#include "ui/ShowCopyDialog.h"
+
+#include "rapidjson/document.h"
+#include "rapidjson/prettywriter.h"
+
+/**
+ * get full size crypt text from server using short crypto text
+ * @param shortenCryptoText short crypto text([GpgFrontend_ShortCrypto]://)
+ * @return
+ */
+QString MainWindow::getCryptText(const QString &shortenCryptoText) {
+
+    QString ownKeyId = settings.value("general/ownKeyId").toString();
+
+    GpgKey key = mCtx->getKeyById(ownKeyId);
+    if (!key.good) {
+        QMessageBox::critical(this, tr("Invalid Own Key"),
+                              tr("Own Key can not be use to do any operation. "
+                                 "Please go to the setting interface to select an OwnKey and get a ServiceToken."));
+        return {};
+    }
+
+    auto utils = new ComUtils(this);
+
+    QString serviceToken = settings.value("general/serviceToken").toString();
+    if (serviceToken.isEmpty() || !utils->checkServiceTokenFormat(serviceToken)) {
+        QMessageBox::critical(this, tr("Error"),
+                              tr("Please obtain a Service Token from the server in the settings."));
+        return {};
+    }
+
+    QUrl reqUrl(utils->getUrl(ComUtils::GetFullCryptText));
+    QNetworkRequest request(reqUrl);
+    request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
+
+    // Sign Shorten Text
+    auto outSignTextBase64 = ComUtils::getSignStringBase64(mCtx, shortenCryptoText, key);
+
+    rapidjson::Document doc;
+    doc.SetObject();
+
+    rapidjson::Value s, t;
+
+    // Signature
+    s.SetString(outSignTextBase64.constData(), outSignTextBase64.count());
+    // Service Token
+    const auto t_byte_array = serviceToken.toUtf8();
+    t.SetString(t_byte_array.constData(), t_byte_array.count());
+
+    rapidjson::Document::AllocatorType &allocator = doc.GetAllocator();
+
+    doc.AddMember("signature", s, allocator);
+    doc.AddMember("serviceToken", t, allocator);
+
+    rapidjson::StringBuffer sb;
+    rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(sb);
+    doc.Accept(writer);
+
+    QByteArray postData(sb.GetString());
+    qDebug() << "postData" << QString::fromUtf8(postData);
+
+    QNetworkReply *reply = utils->getNetworkManager().post(request, postData);
+
+    auto dialog = new WaitingDialog(tr("Getting Cpt From Server"), this);
+    dialog->show();
+
+    while (reply->isRunning()) QApplication::processEvents();
+
+    dialog->close();
+
+    QByteArray replyData = reply->readAll().constData();
+    if (utils->checkServerReply(replyData)) {
+        /**
+         * {
+         *      "cryptoText" : ...
+         *      "sha": ...
+         *      "serviceToken": ...
+         *      "date": ...
+         * }
+         */
+
+        if (!utils->checkDataValueStr("cryptoText")
+            || !utils->checkDataValueStr("sha")
+            || !utils->checkDataValueStr("serviceToken")) {
+            QMessageBox::critical(this, tr("Error"),
+                                  tr("The communication content with the server does not meet the requirements"));
+            return {};
+        }
+
+        auto cryptoText = utils->getDataValueStr("cryptoText");
+        auto sha = utils->getDataValueStr("sha");
+        auto serviceTokenFromServer = utils->getDataValueStr("serviceToken");
+
+        QCryptographicHash sha_generator(QCryptographicHash::Sha256);
+        sha_generator.addData(cryptoText.toUtf8());
+
+        if (sha_generator.result().toHex() == sha && serviceToken == serviceTokenFromServer) {
+            return cryptoText;
+        } else QMessageBox::critical(this, tr("Error"), tr("Invalid short ciphertext"));
+
+        return {};
+    }
+
+    return {};
+}
+
+void MainWindow::shortenCryptText() {
+
+    // gather information
+    QString serviceToken = settings.value("general/serviceToken").toString();
+    QString ownKeyId = settings.value("general/ownKeyId").toString();
+    QByteArray cryptoText = edit->curTextPage()->toPlainText().toUtf8();
+
+    auto utils = new ComUtils(this);
+
+    if (serviceToken.isEmpty() || !utils->checkServiceTokenFormat(serviceToken)) {
+        QMessageBox::critical(this, tr("Invalid Service Token"),
+                              tr("Please go to the setting interface to get a ServiceToken."));
+        return;
+    }
+
+    QUrl reqUrl(utils->getUrl(ComUtils::ShortenCryptText));
+    QNetworkRequest request(reqUrl);
+    request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
+
+    GpgKey key = mCtx->getKeyById(ownKeyId);
+    if (!key.good) {
+        QMessageBox::critical(this, tr("Invalid Own Key"), tr("Own Key can not be use to do any operation."));
+        return;
+    }
+
+    QCryptographicHash ch(QCryptographicHash::Md5);
+    ch.addData(cryptoText);
+    QString md5 = ch.result().toHex();
+
+    qDebug() << "md5" << md5;
+
+    QByteArray signText = QString("[%1][%2]").arg(serviceToken, md5).toUtf8();
+
+    QCryptographicHash sha(QCryptographicHash::Sha256);
+    sha.addData(signText);
+    QString shaText = sha.result().toHex();
+
+    qDebug() << "shaText" << shaText;
+
+    QByteArray outSignTextBase64 = ComUtils::getSignStringBase64(mCtx, signText, key);
+
+    rapidjson::Value c, s, m, t;
+
+    rapidjson::Document doc;
+    doc.SetObject();
+
+    c.SetString(cryptoText.constData(), cryptoText.count());
+    auto m_byte_array = shaText.toUtf8();
+    m.SetString(m_byte_array.constData(), m_byte_array.count());
+    s.SetString(outSignTextBase64.constData(), outSignTextBase64.count());
+    auto t_byte_array = serviceToken.toUtf8();
+    t.SetString(t_byte_array.constData(), t_byte_array.count());
+
+    rapidjson::Document::AllocatorType &allocator = doc.GetAllocator();
+
+    doc.AddMember("cryptoText", c, allocator);
+    doc.AddMember("sha", m, allocator);
+    doc.AddMember("sign", s, allocator);
+    doc.AddMember("serviceToken", t, allocator);
+
+    rapidjson::StringBuffer sb;
+    rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(sb);
+    doc.Accept(writer);
+
+    QByteArray postData(sb.GetString());
+    qDebug() << "postData" << QString::fromUtf8(postData);
+
+    QNetworkReply *reply = networkAccessManager->post(request, postData);
+
+    auto *dialog = new WaitingDialog(tr("Getting Scpt From Server"), this);
+    dialog->show();
+    while (reply->isRunning()) QApplication::processEvents();
+    dialog->close();
+
+    if (utils->checkServerReply(reply->readAll().constData())) {
+
+        /**
+         * {
+         *      "shortenText" : ...
+         *      "md5": ...
+         * }
+         */
+
+        if (!utils->checkDataValueStr("shortenText") || !utils->checkDataValueStr("md5")) {
+            QMessageBox::critical(this, tr("Error"),
+                                  tr("The communication content with the server does not meet the requirements"));
+            return;
+        }
+
+        QString shortenText = utils->getDataValueStr("shortenText");
+
+        QCryptographicHash md5_generator(QCryptographicHash::Md5);
+        md5_generator.addData(shortenText.toUtf8());
+        if (md5_generator.result().toHex() == utils->getDataValueStr("md5")) {
+            auto *dialog = new ShowCopyDialog(shortenText,
+                                              tr("Notice: Use Decrypt & Verify operation to decrypt this short crypto text."),
+                                              this);
+            dialog->show();
+        } else {
+            QMessageBox::critical(this, tr("Error"), tr("There is a problem with the communication with the server"));
+            return;
+        }
+    }
+
+}
+