diff options
Diffstat (limited to '')
| -rw-r--r-- | src/gpg/function/GpgKeyOpera.cpp | 191 |
1 files changed, 191 insertions, 0 deletions
diff --git a/src/gpg/function/GpgKeyOpera.cpp b/src/gpg/function/GpgKeyOpera.cpp new file mode 100644 index 00000000..6eed26f1 --- /dev/null +++ b/src/gpg/function/GpgKeyOpera.cpp @@ -0,0 +1,191 @@ +/** + * 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 "gpg/function/GpgKeyOpera.h" +#include "gpg/GpgConstants.h" +#include "gpg/GpgGenKeyInfo.h" +#include "gpg/function/GpgCommandExecutor.h" + +/** + * Delete keys + * @param uidList key ids + */ +void GpgFrontend::GpgKeyOpera::DeleteKeys( + GpgFrontend::KeyIdArgsListPtr uid_list) { + + GpgError err; + gpgme_key_t key; + + for (const auto &tmp : *uid_list) { + + err = gpgme_op_keylist_start(ctx, tmp.c_str(), 0); + assert(gpg_err_code(err) != GPG_ERR_NO_ERROR); + + err = gpgme_op_keylist_next(ctx, &key); + assert(gpg_err_code(err) != GPG_ERR_NO_ERROR); + + err = gpgme_op_keylist_end(ctx); + assert(gpg_err_code(err) != GPG_ERR_NO_ERROR); + + err = gpgme_op_delete(ctx, key, 1); + assert(gpg_err_code(err) != GPG_ERR_NO_ERROR); + } +} + +/** + * Set the expire date and time of a key pair(actually the master key) or subkey + * @param key target key pair + * @param subkey null if master key + * @param expires date and time + * @return if successful + */ +void GpgFrontend::GpgKeyOpera::SetExpire(const GpgKey &key, + std::unique_ptr<GpgSubKey> &subkey, + std::unique_ptr<QDateTime> &expires) { + unsigned long expires_time = 0; + if (expires != nullptr) { + qDebug() << "Expire Datetime" << expires->toString(); + expires_time = QDateTime::currentDateTime().secsTo(*expires); + } + + const char *sub_fprs = nullptr; + + if (subkey != nullptr) + sub_fprs = subkey->fpr().c_str(); + + auto err = + gpgme_op_setexpire(ctx, gpgme_key_t(key), expires_time, sub_fprs, 0); + + assert(gpg_err_code(err) != GPG_ERR_NO_ERROR); +} + +/** + * Generate revoke cert of a key pair + * @param key target key pair + * @param outputFileName out file name(path) + * @return the process doing this job + */ +void GpgFrontend::GpgKeyOpera::GenerateRevokeCert( + const GpgKey &key, const std::string &output_file_name) { + GpgCommandExecutor::GetInstance().Execute( + {"--command-fd", "0", "--status-fd", "1", "-o", output_file_name.c_str(), + "--gen-revoke", key.fpr().c_str()}, + [](QProcess *proc) -> void { + qDebug() << "Function Called" << proc; + // Code From Gpg4Win + while (proc->canReadLine()) { + const QString line = QString::fromUtf8(proc->readLine()).trimmed(); + if (line == QLatin1String("[GNUPG:] GET_BOOL gen_revoke.okay")) { + proc->write("y\n"); + } else if (line == + QLatin1String( + "[GNUPG:] GET_LINE ask_revocation_reason.code")) { + proc->write("0\n"); + } else if (line == + QLatin1String( + "[GNUPG:] GET_LINE ask_revocation_reason.text")) { + proc->write("\n"); + } else if (line == QLatin1String( + "[GNUPG:] GET_BOOL openfile.overwrite.okay")) { + // We asked before + proc->write("y\n"); + } else if (line == + QLatin1String( + "[GNUPG:] GET_BOOL ask_revocation_reason.okay")) { + proc->write("y\n"); + } + } + // Code From Gpg4Win + }); +} + +/** + * Generate a new key pair + * @param params key generation args + * @return error information + */ +GpgFrontend::GpgError +GpgFrontend::GpgKeyOpera::GenerateKey(std::unique_ptr<GenKeyInfo> params) { + + auto userid_utf8 = params->getUserid(); + const char *userid = userid_utf8.c_str(); + auto algo_utf8 = (params->getAlgo() + params->getKeySizeStr()); + const char *algo = algo_utf8.c_str(); + unsigned long expires = + QDateTime::currentDateTime().secsTo(params->getExpired()); + unsigned int flags = 0; + + if (!params->isSubKey()) + flags |= GPGME_CREATE_CERT; + if (params->isAllowEncryption()) + flags |= GPGME_CREATE_ENCR; + if (params->isAllowSigning()) + flags |= GPGME_CREATE_SIGN; + if (params->isAllowAuthentication()) + flags |= GPGME_CREATE_AUTH; + if (params->isNonExpired()) + flags |= GPGME_CREATE_NOEXPIRE; + if (params->isNoPassPhrase()) + flags |= GPGME_CREATE_NOPASSWD; + + auto err = gpgme_op_createkey(ctx, userid, algo, 0, expires, nullptr, flags); + return check_gpg_error(err); +} + +/** + * Generate a new subkey of a certain key pair + * @param key target key pair + * @param params opera args + * @return error info + */ +GpgFrontend::GpgError +GpgFrontend::GpgKeyOpera::GenerateSubkey(const GpgKey &key, + std::unique_ptr<GenKeyInfo> params) { + + if (!params->isSubKey()) + return GPG_ERR_CANCELED; + + auto algo_utf8 = (params->getAlgo() + params->getKeySizeStr()); + const char *algo = algo_utf8.c_str(); + unsigned long expires = + QDateTime::currentDateTime().secsTo(params->getExpired()); + unsigned int flags = 0; + + if (!params->isSubKey()) + flags |= GPGME_CREATE_CERT; + if (params->isAllowEncryption()) + flags |= GPGME_CREATE_ENCR; + if (params->isAllowSigning()) + flags |= GPGME_CREATE_SIGN; + if (params->isAllowAuthentication()) + flags |= GPGME_CREATE_AUTH; + if (params->isNonExpired()) + flags |= GPGME_CREATE_NOEXPIRE; + + flags |= GPGME_CREATE_NOPASSWD; + + auto err = + gpgme_op_createsubkey(ctx, gpgme_key_t(key), algo, 0, expires, flags); + return check_gpg_error(err); +}
\ No newline at end of file |