290 lines
11 KiB
C++
290 lines
11 KiB
C++
/*
|
|
* File: KeyEdit.cpp
|
|
* Author: SET - nmset@yandex.com
|
|
* License : GPL v2
|
|
* Copyright SET - © 2019
|
|
*
|
|
* Created on October 25, 2020, 10:38 AM
|
|
*/
|
|
|
|
#include "KeyEdit.h"
|
|
#include <Wt/WText.h>
|
|
#include <Wt/WStandardItemModel.h>
|
|
#include <Wt/WStandardItem.h>
|
|
#include "GpgMEWorker.h"
|
|
#include "Tools.h"
|
|
#include "GpgMELogger.h"
|
|
|
|
using namespace std;
|
|
|
|
KeyEdit::KeyEdit(K7Main * owner)
|
|
: WObject()
|
|
{
|
|
m_owner = owner;
|
|
m_popupCertifyUid = NULL;
|
|
m_popupExpiryTime = NULL;
|
|
m_popupAddUid = NULL;
|
|
m_targetUidValidityKeyFpr = WString::Empty;
|
|
m_expiryEditedKeyFpr = WString::Empty;
|
|
}
|
|
|
|
KeyEdit::~KeyEdit()
|
|
{
|
|
delete m_popupCertifyUid;
|
|
}
|
|
|
|
void KeyEdit::OnOwnerTrustDoubleClicked(WTreeTableNode * keyNode, bool keyHasSecret)
|
|
{
|
|
/*
|
|
* A private key that a user does not manage will have its public part
|
|
* listed in the public WTreeTableNode. The certification trust level must
|
|
* not be editable by anyone.
|
|
*/
|
|
WText * lblFpr = static_cast<WText*> (keyNode->columnWidget(3));
|
|
vector<WString> ourKeys = m_owner->m_config->PrivateKeyIds();
|
|
if (!Tools::IsOurKey(lblFpr->text(), ourKeys)
|
|
&& Tools::KeyHasSecret(lblFpr->text()))
|
|
{
|
|
m_owner->m_tmwMessage->SetText(TR("OwnerTrustReadOnly"));
|
|
return;
|
|
}
|
|
/*
|
|
* We leave a primary key with ultimate trust level for further change.
|
|
* kleopatra does not do that.
|
|
*/
|
|
WComboBox * cmbOwnerTrust = new WComboBox();
|
|
FillOwnerTrustCombo(cmbOwnerTrust, keyHasSecret);
|
|
cmbOwnerTrust->blurred().connect(std::bind(&KeyEdit::OnOwnerTrustBlurred, this, keyNode, keyHasSecret));
|
|
WText * lblOwnerTrust = static_cast<WText*> (keyNode->columnWidget(2));
|
|
cmbOwnerTrust->setCurrentIndex(cmbOwnerTrust->findText(lblOwnerTrust->text()));
|
|
/*
|
|
* Prepare to check for change in combobox item.
|
|
* Change is detected by index, not value.
|
|
*/
|
|
cmbOwnerTrust->setAttributeValue("previousTrustLevel", std::to_string(cmbOwnerTrust->currentIndex()));
|
|
keyNode->setColumnWidget(2, unique_ptr<WComboBox> (cmbOwnerTrust));
|
|
// +++
|
|
cmbOwnerTrust->setFocus();
|
|
}
|
|
|
|
void KeyEdit::OnOwnerTrustBlurred(WTreeTableNode* keyNode, bool keyHasSecret)
|
|
{
|
|
// Get new level (not index)
|
|
WComboBox * cmbOwnerTrust = static_cast<WComboBox*> (keyNode->columnWidget(2));
|
|
shared_ptr<WAbstractItemModel> aiModel = cmbOwnerTrust->model();
|
|
WStandardItemModel * iModel = static_cast<WStandardItemModel*> (aiModel.get());
|
|
WStandardItem * item = iModel->item(cmbOwnerTrust->currentIndex(), 0);
|
|
int newLevel = Tools::ToInt(item->text().toUTF8());
|
|
|
|
WText * lblOwnerTrust = new WText(cmbOwnerTrust->currentText());
|
|
lblOwnerTrust->doubleClicked().connect(std::bind(&KeyEdit::OnOwnerTrustDoubleClicked, this, keyNode, keyHasSecret));
|
|
const WText * lblFpr = static_cast<WText*> (keyNode->columnWidget(3));
|
|
const uint newTrustLevel = cmbOwnerTrust->currentIndex();
|
|
const WString previousTrustLevel = cmbOwnerTrust->attributeValue("previousTrustLevel");
|
|
keyNode->setColumnWidget(2, unique_ptr<WText> (lblOwnerTrust));
|
|
// If nothing was changed, don't go to engine.
|
|
if (WString(std::to_string(newTrustLevel)) == previousTrustLevel)
|
|
return;
|
|
|
|
GpgMEWorker gpgWorker;
|
|
GpgME::Error e = gpgWorker.EditOwnerTrust(lblFpr->text().toUTF8().c_str(), (GpgME::Key::OwnerTrust) newLevel);
|
|
if (e.code() != 0)
|
|
{
|
|
lblOwnerTrust->setText(previousTrustLevel);
|
|
m_owner->m_tmwMessage->SetText(TR("OwnerTrustFailure"));
|
|
LGE(e);
|
|
return;
|
|
}
|
|
m_owner->m_tmwMessage->SetText(TR("OwnerTrustSuccess"));
|
|
}
|
|
|
|
void KeyEdit::FillOwnerTrustCombo(WComboBox * cmb, bool keyHasSecret)
|
|
{
|
|
shared_ptr<WStandardItemModel> siModel = make_shared<WStandardItemModel> ();
|
|
OwnerTrustMap OwnerTrustLevel = m_owner->OwnerTrustLevel;
|
|
vector<unique_ptr < WStandardItem>> colIndex;
|
|
vector<unique_ptr < WStandardItem>> colText;
|
|
colIndex.push_back(cpp14::make_unique<WStandardItem> (std::to_string(UserID::Validity::Unknown)));
|
|
colText.push_back(cpp14::make_unique<WStandardItem> (TR("UidUnknown")));
|
|
if (keyHasSecret)
|
|
{
|
|
colIndex.push_back(cpp14::make_unique<WStandardItem> (std::to_string(UserID::Validity::Ultimate)));
|
|
colText.push_back(cpp14::make_unique<WStandardItem> (TR("UidUltimate")));
|
|
}
|
|
else
|
|
{
|
|
colIndex.push_back(cpp14::make_unique<WStandardItem> (std::to_string(UserID::Validity::Never)));
|
|
colText.push_back(cpp14::make_unique<WStandardItem> (TR("UidNever")));
|
|
colIndex.push_back(cpp14::make_unique<WStandardItem> (std::to_string(UserID::Validity::Marginal)));
|
|
colText.push_back(cpp14::make_unique<WStandardItem> (TR("UidMarginal")));
|
|
colIndex.push_back(cpp14::make_unique<WStandardItem> (std::to_string(UserID::Validity::Full)));
|
|
colText.push_back(cpp14::make_unique<WStandardItem> (TR("UidFull")));
|
|
}
|
|
siModel->appendColumn(std::move(colIndex));
|
|
siModel->appendColumn(std::move(colText));
|
|
cmb->clear();
|
|
cmb->setModel(siModel);
|
|
cmb->setModelColumn(1);
|
|
}
|
|
|
|
void KeyEdit::OnUidValidityClicked(WTreeTableNode* uidNode, vector<WString>& privateKeys, const WString& targetKeyFpr)
|
|
{
|
|
if (targetKeyFpr != m_targetUidValidityKeyFpr)
|
|
{
|
|
bool passwordVisibility = true;
|
|
if (m_popupCertifyUid)
|
|
passwordVisibility = m_popupCertifyUid->IsPasswordVisible();
|
|
delete m_popupCertifyUid;
|
|
WText * lblUidValidity = static_cast<WText*> (uidNode->columnWidget(2));
|
|
m_popupCertifyUid = new PopupCertifyUserId(lblUidValidity, m_owner->m_tmwMessage);
|
|
m_popupCertifyUid->Create(privateKeys, targetKeyFpr);
|
|
m_popupCertifyUid->ShowPassphrase(passwordVisibility);
|
|
m_targetUidValidityKeyFpr = targetKeyFpr;
|
|
m_popupCertifyUid->GetCertifyButton()->clicked().connect(this, &KeyEdit::EditUidValidity);
|
|
}
|
|
m_popupCertifyUid->show();
|
|
}
|
|
|
|
void KeyEdit::EditUidValidity()
|
|
{
|
|
if (!m_popupCertifyUid->Validate())
|
|
{
|
|
m_owner->m_tmwMessage->SetText(TR("InvalidInput"));
|
|
return;
|
|
}
|
|
const WString signingKey = m_popupCertifyUid->GetSelectedKey();
|
|
const WString keyToSign = m_popupCertifyUid->GetKeyToSign();
|
|
GpgMEWorker gpgWorker;
|
|
GpgME::Error e;
|
|
if (m_popupCertifyUid->WhatToDo() == PopupCertifyUserId::CertifyUid)
|
|
{
|
|
vector<uint>& uidsToSign = m_popupCertifyUid->GetUidsToSign();
|
|
int options = m_popupCertifyUid->GetCertifyOptions();
|
|
e = gpgWorker.CertifyKey(signingKey.toUTF8().c_str(),
|
|
keyToSign.toUTF8().c_str(),
|
|
uidsToSign, options,
|
|
m_popupCertifyUid->GetPassphrase());
|
|
}
|
|
else
|
|
{
|
|
vector<GpgME::UserID> uidsToRevoke
|
|
= m_popupCertifyUid->GetUidsToRevokeCertification();
|
|
e = gpgWorker.RevokeKeyCertifications(signingKey.toUTF8().c_str(),
|
|
keyToSign.toUTF8().c_str(),
|
|
uidsToRevoke,
|
|
m_popupCertifyUid->GetPassphrase());
|
|
}
|
|
if (e.code() != 0)
|
|
{
|
|
m_owner->m_tmwMessage->SetText(e.asString());
|
|
m_popupCertifyUid->ShowPassphrase(true);
|
|
LGE(e);
|
|
return;
|
|
}
|
|
if (m_popupCertifyUid->WhatToDo() == PopupCertifyUserId::CertifyUid)
|
|
m_owner->m_tmwMessage->SetText(TR("CertificationSuccess"));
|
|
else
|
|
m_owner->m_tmwMessage->SetText(TR("RevocationSuccess"));
|
|
m_popupCertifyUid->ShowPassphrase(false);
|
|
m_owner->DisplayUids(keyToSign);
|
|
}
|
|
|
|
void KeyEdit::OnExpiryClicked(WTreeTableNode* subkeyNode, const WString& keyFpr,
|
|
const WString& subkeyFpr)
|
|
{
|
|
if (keyFpr != m_expiryEditedKeyFpr)
|
|
{
|
|
delete m_popupExpiryTime;
|
|
WText * lblExpiry = static_cast<WText*> (subkeyNode->columnWidget(2));
|
|
m_popupExpiryTime = new PopupExpiryTime(lblExpiry, m_owner->m_tmwMessage);
|
|
m_popupExpiryTime->Create(keyFpr);
|
|
m_expiryEditedKeyFpr = keyFpr;
|
|
m_popupExpiryTime->GetApplyButton()->clicked().connect(this, &KeyEdit::SetKeyExpiryTime);
|
|
}
|
|
m_popupExpiryTime->SetSubkeyFpr(subkeyFpr);
|
|
m_popupExpiryTime->show();
|
|
}
|
|
|
|
void KeyEdit::SetKeyExpiryTime()
|
|
{
|
|
GpgMEWorker gpgWorker;
|
|
GpgME::Error e;
|
|
const WString keyFpr = m_popupExpiryTime->GetKeyFpr();
|
|
WString subkeyFpr = m_popupExpiryTime->GetSubkeyFpr();
|
|
if (keyFpr == subkeyFpr)
|
|
subkeyFpr = WString::Empty;
|
|
e = gpgWorker.SetKeyExpiryTime(keyFpr.toUTF8().c_str(),
|
|
subkeyFpr.toUTF8().c_str(),
|
|
m_popupExpiryTime->GetPassphrase(),
|
|
m_popupExpiryTime->GetExpiry());
|
|
if (e.code() != 0)
|
|
{
|
|
m_owner->m_tmwMessage->SetText(TR("SetExpirationTimeFailure"));
|
|
m_popupExpiryTime->ShowPassphrase(true);
|
|
LGE(e);
|
|
return;
|
|
}
|
|
m_owner->m_tmwMessage->SetText(TR("SetExpirationTimeSuccess"));
|
|
m_popupExpiryTime->ShowPassphrase(false);
|
|
m_owner->DisplaySubKeys(m_expiryEditedKeyFpr, true);
|
|
}
|
|
|
|
void KeyEdit::OnUidEmailClicked(WTreeTableNode* uidNode, const WString& keyFpr)
|
|
{
|
|
WText * lblEmail = static_cast<WText*> (uidNode->columnWidget(1));
|
|
if (keyFpr != m_addUidKeyFpr)
|
|
{
|
|
delete m_popupAddUid;
|
|
m_popupAddUid = new PopupAddUid(lblEmail, m_owner->m_tmwMessage);
|
|
m_popupAddUid->Create();
|
|
m_addUidKeyFpr = keyFpr;
|
|
m_popupAddUid->GetApplyButton()->clicked().connect(this, &KeyEdit::AddOrRevokeUid);
|
|
}
|
|
WText * lblName = uidNode->label();
|
|
WText * lblComment = static_cast<WText*> (uidNode->columnWidget(3));
|
|
m_popupAddUid->SetNodeIdentity(lblName->text(), lblEmail->text(),
|
|
lblComment->text());
|
|
m_popupAddUid->show();
|
|
}
|
|
|
|
void KeyEdit::AddOrRevokeUid()
|
|
{
|
|
if (!m_popupAddUid->Validate())
|
|
{
|
|
m_owner->m_tmwMessage->SetText(TR("InvalidInput"));
|
|
return;
|
|
}
|
|
const WString name = m_popupAddUid->GetName();
|
|
const WString email = m_popupAddUid->GetEmail();
|
|
const WString comment = m_popupAddUid->GetComment();
|
|
const WString passphrase = m_popupAddUid->GetPassphrase();
|
|
|
|
Error e;
|
|
GpgMEWorker gpgw;
|
|
if (m_popupAddUid->WhatToDo() == PopupAddUid::What::Revoke)
|
|
{
|
|
e = gpgw.RevokeUserID(m_addUidKeyFpr.toUTF8().c_str(),
|
|
passphrase.toUTF8(),
|
|
name.toUTF8(), email.toUTF8(), comment.toUTF8());
|
|
}
|
|
else
|
|
{
|
|
e = gpgw.AddUserID(m_addUidKeyFpr.toUTF8().c_str(),
|
|
passphrase.toUTF8(),
|
|
name.toUTF8(), email.toUTF8(), comment.toUTF8());
|
|
}
|
|
m_popupAddUid->hide();
|
|
if (e.code() != 0)
|
|
{
|
|
m_popupAddUid->ShowPassphrase(true);
|
|
m_owner->m_tmwMessage->SetText(e.asString());
|
|
LGE(e);
|
|
}
|
|
else
|
|
{
|
|
m_popupAddUid->ShowPassphrase(false);
|
|
}
|
|
// Key certifications are not listed on this refresh !
|
|
m_owner->DisplayUids(m_addUidKeyFpr, true);
|
|
}
|