fix: solve lower/upper case name conflict

1 files changed, 589 insertions, 0 deletions

diff --git a/src/ui/dialog/key_generate/KeyGenerateDialog.cpp b/src/ui/dialog/key_generate/KeyGenerateDialog.cpp
new file mode 100644
index 00000000..49425e5b
--- /dev/null
+++ b/src/ui/dialog/key_generate/KeyGenerateDialog.cpp
@@ -0,0 +1,589 @@
+/**
+ * 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 "KeyGenerateDialog.h"
+
+#include "core/GpgModel.h"
+#include "core/function/GlobalSettingStation.h"
+#include "core/function/gpg/GpgKeyOpera.h"
+#include "core/typedef/GpgTypedef.h"
+#include "core/utils/CacheUtils.h"
+#include "core/utils/GpgUtils.h"
+#include "ui/UISignalStation.h"
+#include "ui/UserInterfaceUtils.h"
+#include "ui/function/GpgOperaHelper.h"
+
+//
+#include "ui_KeyGenDialog.h"
+
+namespace GpgFrontend::UI {
+
+auto SearchAlgoByName(const QString& name,
+                      const QContainer<KeyAlgo>& algos) -> QContainer<KeyAlgo> {
+  QContainer<KeyAlgo> res;
+
+  for (const auto& algo : algos) {
+    if (algo.Name() != name) continue;
+    res.append(algo);
+  }
+
+  return res;
+}
+
+auto GetAlgoByNameAndKeyLength(const QString& name, int key_length,
+                               const QContainer<KeyAlgo>& algos)
+    -> std::tuple<bool, KeyAlgo> {
+  for (const auto& algo : algos) {
+    if (algo.Name() != name) continue;
+    if (algo.KeyLength() != key_length) continue;
+    return {true, algo};
+  }
+
+  return {};
+}
+
+auto GetAlgoByName(const QString& name, const QContainer<KeyAlgo>& algos)
+    -> std::tuple<bool, KeyAlgo> {
+  for (const auto& algo : algos) {
+    if (algo.Name() != name) continue;
+    return {true, algo};
+  }
+
+  return {};
+}
+
+void SetKeyLengthComboxBoxByAlgo(QComboBox* combo,
+                                 const QContainer<KeyAlgo>& algos) {
+  combo->clear();
+
+  QContainer<KeyAlgo> sorted_algos(algos.begin(), algos.end());
+  std::sort(sorted_algos.begin(), sorted_algos.end(),
+            [](const KeyAlgo& a, const KeyAlgo& b) {
+              return a.KeyLength() < b.KeyLength();
+            });
+
+  QStringList key_lengths;
+  for (const auto& algo : sorted_algos) {
+    key_lengths.append(QString::number(algo.KeyLength()));
+  }
+
+  combo->addItems(key_lengths);
+}
+
+KeyGenDialog::KeyGenDialog(int channel, QWidget* parent)
+    : GeneralDialog(typeid(KeyGenDialog).name(), parent),
+      ui_(QSharedPointer<Ui_KeyGenDialog>::create()),
+      gen_key_info_(QSharedPointer<GenKeyInfo>::create()),
+      gen_subkey_info_(nullptr),
+      supported_primary_key_algos_(GenKeyInfo::GetSupportedKeyAlgo()),
+      supported_subkey_algos_(GenKeyInfo::GetSupportedSubkeyAlgo()),
+      channel_(channel) {
+  ui_->setupUi(this);
+
+  for (const auto& key_db : GetGpgKeyDatabaseInfos()) {
+    ui_->keyDBIndexComboBox->insertItem(
+        key_db.channel, QString("%1: %2").arg(key_db.channel).arg(key_db.name));
+  }
+
+  ui_->easyAlgoComboBox->addItems({
+      tr("Custom"),
+      "RSA",
+      "DSA",
+      "ECC (25519)",
+  });
+
+  ui_->easyValidityPeriodComboBox->addItems({
+      tr("Custom"),
+      tr("3 Months"),
+      tr("6 Months"),
+      tr("1 Year"),
+      tr("2 Years"),
+      tr("5 Years"),
+      tr("10 Years"),
+      tr("Non Expired"),
+  });
+
+  QSet<QString> p_algo_set;
+  for (const auto& algo : supported_primary_key_algos_) {
+    p_algo_set.insert(algo.Name());
+  }
+  ui_->pAlgoComboBox->addItems(
+      QStringList(p_algo_set.cbegin(), p_algo_set.cend()));
+
+  QSet<QString> s_algo_set;
+  for (const auto& algo : supported_subkey_algos_) {
+    s_algo_set.insert(algo.Name());
+  }
+  ui_->sAlgoComboBox->addItem(tr("None"));
+  ui_->sAlgoComboBox->addItems(
+      QStringList(s_algo_set.cbegin(), s_algo_set.cend()));
+
+  ui_->easyAlgoComboBox->setCurrentText("RSA");
+  ui_->easyValidityPeriodComboBox->setCurrentText(tr("2 Years"));
+
+  set_signal_slot_config();
+
+  slot_easy_mode_changed("RSA");
+  slot_easy_valid_date_changed(tr("2 Years"));
+
+  this->setWindowTitle(tr("Generate Key"));
+  this->setAttribute(Qt::WA_DeleteOnClose);
+  this->setModal(true);
+}
+
+void KeyGenDialog::slot_key_gen_accept() {
+  QString buffer;
+  QTextStream error_stream(&buffer);
+
+  if (ui_->nameEdit->text().size() < 5) {
+    error_stream << " -> " << tr("Name must contain at least five characters.")
+                 << Qt::endl;
+  }
+  if (ui_->emailEdit->text().isEmpty() ||
+      !check_email_address(ui_->emailEdit->text())) {
+    error_stream << " -> " << tr("Please give a valid email address.")
+                 << Qt::endl;
+  }
+
+  const auto err_string = error_stream.readAll();
+  if (!err_string.isEmpty()) {
+    ui_->statusPlainTextEdit->clear();
+    ui_->statusPlainTextEdit->appendPlainText(err_string);
+    return;
+  }
+
+  gen_key_info_->SetName(ui_->nameEdit->text());
+  gen_key_info_->SetEmail(ui_->emailEdit->text());
+  gen_key_info_->SetComment(ui_->commentEdit->text());
+
+  if (ui_->noPassphraseCheckBox->checkState() != 0U) {
+    gen_key_info_->SetNonPassPhrase(true);
+    if (gen_subkey_info_ != nullptr) {
+      gen_subkey_info_->SetNonPassPhrase(true);
+    }
+  }
+
+  if (ui_->pExpireCheckBox->checkState() != 0U) {
+    gen_key_info_->SetNonExpired(true);
+    if (gen_subkey_info_ != nullptr) gen_subkey_info_->SetNonExpired(true);
+  } else {
+    gen_key_info_->SetExpireTime(ui_->pValidityPeriodDateTimeEdit->dateTime());
+    if (gen_subkey_info_ != nullptr) {
+      gen_subkey_info_->SetExpireTime(
+          ui_->sValidityPeriodDateTimeEdit->dateTime());
+    }
+  }
+
+  if (!GetSettings()
+           .value("gnupg/use_pinentry_as_password_input_dialog",
+                  QString::fromLocal8Bit(qgetenv("container")) != "flatpak")
+           .toBool() &&
+      !ui_->noPassphraseCheckBox->isChecked()) {
+    SetCacheValue("PinentryContext", "NEW_PASSPHRASE");
+  }
+
+  LOG_D() << "try to generate key at gpg context channel: " << channel_;
+
+  GpgOperaHelper::WaitForOpera(
+      this, tr("Generating"),
+      [this, gen_key_info = this->gen_key_info_](const OperaWaitingHd& hd) {
+        GpgKeyOpera::GetInstance(channel_).GenerateKeyWithSubkey(
+            gen_key_info, gen_subkey_info_,
+            [this, hd](GpgError err, const DataObjectPtr&) {
+              // stop showing waiting dialog
+              hd();
+
+              if (CheckGpgError(err) == GPG_ERR_USER_1) {
+                QMessageBox::critical(this, tr("Error"),
+                                      tr("Unknown error occurred"));
+                return;
+              }
+
+              CommonUtils::RaiseMessageBox(
+                  this->parentWidget() != nullptr ? this->parentWidget() : this,
+                  err);
+              if (CheckGpgError(err) == GPG_ERR_NO_ERROR) {
+                emit SignalKeyGenerated();
+              }
+            });
+      });
+
+  this->done(0);
+}
+
+void KeyGenDialog::refresh_widgets_state() {
+  ui_->pAlgoComboBox->blockSignals(true);
+  ui_->pAlgoComboBox->setCurrentText(gen_key_info_->GetAlgo().Name());
+  ui_->pAlgoComboBox->blockSignals(false);
+
+  ui_->pKeyLengthComboBox->blockSignals(true);
+  SetKeyLengthComboxBoxByAlgo(
+      ui_->pKeyLengthComboBox,
+      SearchAlgoByName(ui_->pAlgoComboBox->currentText(),
+                       supported_primary_key_algos_));
+  ui_->pKeyLengthComboBox->setCurrentText(
+      QString::number(gen_key_info_->GetKeyLength()));
+  ui_->pKeyLengthComboBox->blockSignals(false);
+
+  ui_->pEncrCheckBox->blockSignals(true);
+  ui_->pEncrCheckBox->setCheckState(
+      gen_key_info_->IsAllowEncryption() ? Qt::Checked : Qt::Unchecked);
+  ui_->pEncrCheckBox->setEnabled(gen_key_info_->IsAllowChangeEncryption());
+  ui_->pEncrCheckBox->blockSignals(false);
+
+  ui_->pSignCheckBox->blockSignals(true);
+  ui_->pSignCheckBox->setCheckState(
+      gen_key_info_->IsAllowSigning() ? Qt::Checked : Qt::Unchecked);
+  ui_->pSignCheckBox->setEnabled(gen_key_info_->IsAllowChangeSigning());
+  ui_->pSignCheckBox->blockSignals(false);
+
+  ui_->pAuthCheckBox->blockSignals(true);
+  ui_->pAuthCheckBox->setCheckState(
+      gen_key_info_->IsAllowAuthentication() ? Qt::Checked : Qt::Unchecked);
+  ui_->pAuthCheckBox->setEnabled(gen_key_info_->IsAllowChangeAuthentication());
+  ui_->pAuthCheckBox->blockSignals(false);
+
+  ui_->noPassphraseCheckBox->setEnabled(gen_key_info_->IsAllowNoPassPhrase());
+
+  ui_->pValidityPeriodDateTimeEdit->blockSignals(true);
+  ui_->pValidityPeriodDateTimeEdit->setDateTime(gen_key_info_->GetExpireTime());
+  ui_->pValidityPeriodDateTimeEdit->setDisabled(gen_key_info_->IsNonExpired());
+  ui_->pValidityPeriodDateTimeEdit->blockSignals(false);
+
+  ui_->pExpireCheckBox->blockSignals(true);
+  ui_->pExpireCheckBox->setChecked(gen_key_info_->IsNonExpired());
+  ui_->pExpireCheckBox->blockSignals(false);
+
+  if (gen_subkey_info_ == nullptr) {
+    ui_->tab_3->setDisabled(true);
+
+    ui_->sAlgoComboBox->blockSignals(true);
+    ui_->sAlgoComboBox->setCurrentText(tr("None"));
+    ui_->sAlgoComboBox->blockSignals(false);
+
+    ui_->sKeyLengthComboBox->blockSignals(true);
+    ui_->sKeyLengthComboBox->clear();
+    ui_->sKeyLengthComboBox->blockSignals(false);
+
+    ui_->sEncrCheckBox->blockSignals(true);
+    ui_->sEncrCheckBox->setCheckState(Qt::Unchecked);
+    ui_->sEncrCheckBox->blockSignals(false);
+
+    ui_->sSignCheckBox->blockSignals(true);
+    ui_->sSignCheckBox->setCheckState(Qt::Unchecked);
+    ui_->sSignCheckBox->blockSignals(false);
+
+    ui_->sAuthCheckBox->blockSignals(true);
+    ui_->sAuthCheckBox->setCheckState(Qt::Unchecked);
+    ui_->sAuthCheckBox->blockSignals(false);
+
+    ui_->sValidityPeriodDateTimeEdit->blockSignals(true);
+    ui_->sValidityPeriodDateTimeEdit->setDateTime(QDateTime::currentDateTime());
+    ui_->sValidityPeriodDateTimeEdit->setDisabled(true);
+    ui_->sValidityPeriodDateTimeEdit->blockSignals(false);
+
+    ui_->sExpireCheckBox->blockSignals(true);
+    ui_->sExpireCheckBox->setCheckState(Qt::Unchecked);
+    ui_->sExpireCheckBox->blockSignals(false);
+    return;
+  }
+
+  ui_->tab_3->setDisabled(false);
+
+  ui_->sAlgoComboBox->blockSignals(true);
+  ui_->sAlgoComboBox->setCurrentText(gen_subkey_info_->GetAlgo().Name());
+  ui_->sAlgoComboBox->blockSignals(false);
+
+  ui_->sKeyLengthComboBox->blockSignals(true);
+  SetKeyLengthComboxBoxByAlgo(
+      ui_->sKeyLengthComboBox,
+      SearchAlgoByName(ui_->sAlgoComboBox->currentText(),
+                       supported_subkey_algos_));
+  ui_->sKeyLengthComboBox->setCurrentText(
+      QString::number(gen_subkey_info_->GetKeyLength()));
+  ui_->sKeyLengthComboBox->blockSignals(false);
+
+  ui_->sEncrCheckBox->blockSignals(true);
+  ui_->sEncrCheckBox->setCheckState(
+      gen_subkey_info_->IsAllowEncryption() ? Qt::Checked : Qt::Unchecked);
+  ui_->sEncrCheckBox->setEnabled(gen_subkey_info_->IsAllowChangeEncryption());
+  ui_->sEncrCheckBox->blockSignals(false);
+
+  ui_->sSignCheckBox->blockSignals(true);
+  ui_->sSignCheckBox->setCheckState(
+      gen_subkey_info_->IsAllowSigning() ? Qt::Checked : Qt::Unchecked);
+  ui_->sSignCheckBox->setEnabled(gen_subkey_info_->IsAllowChangeSigning());
+  ui_->sSignCheckBox->blockSignals(false);
+
+  ui_->sAuthCheckBox->blockSignals(true);
+  ui_->sAuthCheckBox->setCheckState(
+      gen_subkey_info_->IsAllowAuthentication() ? Qt::Checked : Qt::Unchecked);
+  ui_->sAuthCheckBox->setEnabled(
+      gen_subkey_info_->IsAllowChangeAuthentication());
+  ui_->sAuthCheckBox->blockSignals(false);
+
+  ui_->sValidityPeriodDateTimeEdit->blockSignals(true);
+  ui_->sValidityPeriodDateTimeEdit->setDateTime(
+      gen_subkey_info_->GetExpireTime());
+  ui_->sValidityPeriodDateTimeEdit->setDisabled(
+      gen_subkey_info_->IsNonExpired());
+  ui_->sValidityPeriodDateTimeEdit->blockSignals(false);
+
+  ui_->sExpireCheckBox->blockSignals(true);
+  ui_->sExpireCheckBox->setChecked(gen_subkey_info_->IsNonExpired());
+  ui_->sExpireCheckBox->blockSignals(false);
+}
+
+void KeyGenDialog::set_signal_slot_config() {
+  connect(ui_->generateButton, &QPushButton::clicked, this,
+          &KeyGenDialog::slot_key_gen_accept);
+
+  connect(ui_->pExpireCheckBox, &QCheckBox::stateChanged, this,
+          [this](int state) {
+            ui_->pValidityPeriodDateTimeEdit->setDisabled(state == Qt::Checked);
+
+            slot_set_easy_valid_date_2_custom();
+          });
+  connect(ui_->sExpireCheckBox, &QCheckBox::stateChanged, this,
+          [this](int state) {
+            ui_->sValidityPeriodDateTimeEdit->setDisabled(state == Qt::Checked);
+
+            slot_set_easy_valid_date_2_custom();
+          });
+
+  connect(ui_->pEncrCheckBox, &QCheckBox::stateChanged, this,
+          [this](int state) {
+            gen_key_info_->SetAllowEncryption(state == Qt::Checked);
+          });
+  connect(ui_->pSignCheckBox, &QCheckBox::stateChanged, this,
+          [this](int state) {
+            gen_key_info_->SetAllowSigning(state == Qt::Checked);
+          });
+  connect(ui_->pAuthCheckBox, &QCheckBox::stateChanged, this,
+          [this](int state) {
+            gen_key_info_->SetAllowAuthentication(state == Qt::Checked);
+          });
+
+  connect(ui_->sEncrCheckBox, &QCheckBox::stateChanged, this,
+          [this](int state) {
+            gen_subkey_info_->SetAllowEncryption(state == Qt::Checked);
+          });
+  connect(ui_->sSignCheckBox, &QCheckBox::stateChanged, this,
+          [this](int state) {
+            gen_subkey_info_->SetAllowSigning(state == Qt::Checked);
+          });
+  connect(ui_->sAuthCheckBox, &QCheckBox::stateChanged, this,
+          [this](int state) {
+            gen_subkey_info_->SetAllowAuthentication(state == Qt::Checked);
+          });
+
+  connect(ui_->noPassphraseCheckBox, &QCheckBox::stateChanged, this,
+          [this](int state) -> void {
+            gen_key_info_->SetNonPassPhrase(state != 0);
+            if (gen_subkey_info_ != nullptr) {
+              gen_subkey_info_->SetNonPassPhrase(state != 0);
+            }
+          });
+
+  connect(ui_->pAlgoComboBox, &QComboBox::currentTextChanged, this,
+          [=](const QString&) {
+            sync_gen_key_info();
+            slot_set_easy_key_algo_2_custom();
+            refresh_widgets_state();
+          });
+
+  connect(ui_->sAlgoComboBox, &QComboBox::currentTextChanged, this,
+          [=](const QString&) {
+            sync_gen_subkey_info();
+            slot_set_easy_key_algo_2_custom();
+            refresh_widgets_state();
+          });
+
+  connect(ui_->easyAlgoComboBox, &QComboBox::currentTextChanged, this,
+          &KeyGenDialog::slot_easy_mode_changed);
+
+  connect(ui_->easyValidityPeriodComboBox, &QComboBox::currentTextChanged, this,
+          &KeyGenDialog::slot_easy_valid_date_changed);
+
+  connect(ui_->pValidityPeriodDateTimeEdit, &QDateTimeEdit::dateTimeChanged,
+          this, [=](const QDateTime& dt) {
+            gen_key_info_->SetExpireTime(dt);
+
+            slot_set_easy_valid_date_2_custom();
+          });
+
+  connect(ui_->sValidityPeriodDateTimeEdit, &QDateTimeEdit::dateTimeChanged,
+          this, [=](const QDateTime& dt) {
+            gen_subkey_info_->SetExpireTime(dt);
+
+            slot_set_easy_valid_date_2_custom();
+          });
+
+  connect(ui_->keyDBIndexComboBox, &QComboBox::currentIndexChanged, this,
+          [=](int index) { channel_ = index; });
+
+  connect(this, &KeyGenDialog::SignalKeyGenerated,
+          UISignalStation::GetInstance(),
+          &UISignalStation::SignalKeyDatabaseRefresh);
+}
+
+auto KeyGenDialog::check_email_address(const QString& str) -> bool {
+  return re_email_.match(str).hasMatch();
+}
+
+void KeyGenDialog::sync_gen_key_info() {
+  auto [found, algo] = GetAlgoByName(ui_->pAlgoComboBox->currentText(),
+
+                                     supported_primary_key_algos_);
+
+  ui_->generateButton->setDisabled(!found);
+
+  if (found) {
+    gen_key_info_->SetAlgo(algo);
+  }
+}
+
+void KeyGenDialog::sync_gen_subkey_info() {
+  if (gen_subkey_info_ != nullptr) {
+    auto [s_found, algo] = GetAlgoByName(ui_->sAlgoComboBox->currentText(),
+                                         supported_subkey_algos_);
+
+    ui_->generateButton->setDisabled(!s_found);
+    if (s_found) gen_subkey_info_->SetAlgo(algo);
+  }
+}
+
+void KeyGenDialog::slot_easy_mode_changed(const QString& mode) {
+  if (mode == "RSA") {
+    auto [found, algo] = GenKeyInfo::SearchPrimaryKeyAlgo("rsa2048");
+    if (found) gen_key_info_->SetAlgo(algo);
+
+    gen_subkey_info_ = nullptr;
+  }
+
+  else if (mode == "DSA") {
+    auto [found, algo] = GenKeyInfo::SearchPrimaryKeyAlgo("dsa2048");
+    if (found) gen_key_info_->SetAlgo(algo);
+
+    if (gen_subkey_info_ == nullptr) {
+      gen_subkey_info_ = QSharedPointer<GenKeyInfo>::create(true);
+    }
+
+    auto [s_found, s_algo] = GenKeyInfo::SearchSubKeyAlgo("elg2048");
+    if (s_found) gen_subkey_info_->SetAlgo(s_algo);
+  }
+
+  else if (mode == "ECC (25519)") {
+    auto [found, algo] = GenKeyInfo::SearchPrimaryKeyAlgo("ed25519");
+    if (found) gen_key_info_->SetAlgo(algo);
+
+    if (gen_subkey_info_ == nullptr) {
+      gen_subkey_info_ = QSharedPointer<GenKeyInfo>::create(true);
+    }
+
+    auto [s_found, s_algo] = GenKeyInfo::SearchSubKeyAlgo("cv25519");
+    if (s_found) gen_subkey_info_->SetAlgo(s_algo);
+  }
+
+  else {
+    auto [found, algo] = GenKeyInfo::SearchPrimaryKeyAlgo("rsa2048");
+    if (found) gen_key_info_->SetAlgo(algo);
+
+    if (gen_subkey_info_ == nullptr) {
+      gen_subkey_info_ = QSharedPointer<GenKeyInfo>::create(true);
+    }
+
+    auto [s_found, s_algo] = GenKeyInfo::SearchSubKeyAlgo("rsa2048");
+    if (s_found) gen_subkey_info_->SetAlgo(s_algo);
+  }
+
+  refresh_widgets_state();
+}
+
+void KeyGenDialog::slot_easy_valid_date_changed(const QString& mode) {
+  if (mode == tr("3 Months")) {
+    gen_key_info_->SetNonExpired(false);
+    gen_key_info_->SetExpireTime(QDateTime::currentDateTime().addMonths(3));
+  }
+
+  else if (mode == tr("6 Months")) {
+    gen_key_info_->SetNonExpired(false);
+    gen_key_info_->SetExpireTime(QDateTime::currentDateTime().addMonths(6));
+  }
+
+  else if (mode == tr("1 Year")) {
+    gen_key_info_->SetNonExpired(false);
+    gen_key_info_->SetExpireTime(QDateTime::currentDateTime().addYears(1));
+  }
+
+  else if (mode == tr("2 Years")) {
+    gen_key_info_->SetNonExpired(false);
+    gen_key_info_->SetExpireTime(QDateTime::currentDateTime().addYears(2));
+  }
+
+  else if (mode == tr("5 Years")) {
+    gen_key_info_->SetNonExpired(false);
+    gen_key_info_->SetExpireTime(QDateTime::currentDateTime().addYears(5));
+  }
+
+  else if (mode == tr("10 Years")) {
+    gen_key_info_->SetNonExpired(false);
+    gen_key_info_->SetExpireTime(QDateTime::currentDateTime().addYears(10));
+
+  }
+
+  else if (mode == tr("Non Expired")) {
+    gen_key_info_->SetNonExpired(true);
+    gen_key_info_->SetExpireTime(QDateTime::currentDateTime());
+  }
+
+  else {
+    gen_key_info_->SetNonExpired(false);
+    gen_key_info_->SetExpireTime(QDateTime::currentDateTime().addYears(2));
+  }
+
+  if (gen_subkey_info_ != nullptr) {
+    gen_subkey_info_->SetExpireTime(gen_key_info_->GetExpireTime());
+    gen_subkey_info_->SetNonExpired(gen_key_info_->IsNonExpired());
+  }
+
+  refresh_widgets_state();
+}
+
+void KeyGenDialog::slot_set_easy_valid_date_2_custom() {
+  ui_->easyValidityPeriodComboBox->blockSignals(true);
+  ui_->easyValidityPeriodComboBox->setCurrentText(tr("Custom"));
+  ui_->easyValidityPeriodComboBox->blockSignals(false);
+}
+
+void KeyGenDialog::slot_set_easy_key_algo_2_custom() {
+  ui_->easyAlgoComboBox->blockSignals(true);
+  ui_->easyAlgoComboBox->setCurrentText(tr("Custom"));
+  ui_->easyAlgoComboBox->blockSignals(false);
+}
+}  // namespace GpgFrontend::UI