kde-playground/kgpg/kgpg.cpp

/*  This file is part of the KDE project
    Copyright (C) 2022 Ivailo Monev <xakepa10@gmail.com>

    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Library General Public
    License version 2, as published by the Free Software Foundation.

    This library 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
    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public License
    along with this library; see the file COPYING.LIB.  If not, write to
    the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
    Boston, MA 02110-1301, USA.
*/

#include <QImageWriter>
#include <klocale.h>
#include <kapplication.h>
#include <klocale.h>
#include <kcmdlineargs.h>
#include <kaboutdata.h>
#include <kdebug.h>

#include "kgpg.h"

KGPG::KGPG(QWidget *parent)
    : KMainWindow(parent),
    m_release(false)
{
    m_ui.setupUi(this);

    // required by context
    kDebug() << gpgme_check_version(NULL);

#ifdef kgpg_key_query
    gpgme_error_t gpgresult = gpgme_new(&m_gpgctx);
    if (gpgresult != 0) {
        kWarning() << gpgme_strerror(gpgresult);
        return;
    }
    m_release = true;

    gpgresult = gpgme_set_keylist_mode(m_gpgctx, GPGME_KEYLIST_MODE_LOCAL);
    if (gpgresult != 0) {
        kWarning() << gpgme_strerror(gpgresult);
        return;
    }

    // required by key query
    gpgresult = gpgme_op_keylist_start(m_gpgctx, NULL, false);
    if (gpgresult != 0) {
        kWarning() << gpgme_strerror(gpgresult);
        return;
    }

    char* gpgkeyfpr = NULL; // will either use key or password from input for encryption
    gpgme_key_t gpgkey;
    gpgresult = gpgme_op_keylist_next(m_gpgctx, &gpgkey);
    while (gpgresult == 0) {
        gpgme_user_id_t gpguid = NULL;
        for (gpguid = gpgkey->uids; gpguid; gpguid = gpguid->next) {
            kWarning() << gpguid->email << gpguid->name << gpguid->comment << gpguid->uidhash << gpgkey->fpr;
            ::free(gpgkeyfpr);
            gpgkeyfpr = ::strdup(gpgkey->fpr);
        }

        gpgme_key_unref(gpgkey);
        gpgresult = gpgme_op_keylist_next(m_gpgctx, &gpgkey);
    }
#endif // kgpg_key_query

#ifdef kgpg_encrypt
    gpgme_data_t gpgindata;
    gpgresult = gpgme_data_new_from_file(&gpgindata, "/home/smil3y/gpgme/tests/run-encrypt.c", 1);
    if (gpgresult != 0) {
        kWarning() << gpgme_strerror(gpgresult);
        ::free(gpgkeyfpr);
        return;
    }

    gpgme_key_t gpgencryptkey;
    gpgresult = gpgme_get_key(m_gpgctx, gpgkeyfpr, &gpgencryptkey, 1);
    ::free(gpgkeyfpr);
    if (gpgresult != 0) {
        kWarning() << "gpgme_get_key" << gpgme_strerror(gpgresult);
        gpgme_data_release(gpgindata);
        return;
    }

    gpgme_data_t gpgoutdata;
    gpgresult = gpgme_data_new(&gpgoutdata);
    if (gpgresult != 0) {
        kWarning() << gpgme_strerror(gpgresult);
        gpgme_data_release(gpgindata);
        return;
    }

    gpgme_key_t gpgkeys[2] = { gpgencryptkey, NULL };
    gpgresult = gpgme_op_encrypt(m_gpgctx, gpgkeys, GPGME_ENCRYPT_ALWAYS_TRUST, gpgindata, gpgoutdata);
    if (gpgresult != 0) {
        kWarning() << gpgme_strerror(gpgresult);
        gpgme_data_release(gpgindata);
        gpgme_data_release(gpgoutdata);
        return;
    }
#endif // kgpg_encrypt

#ifdef kgpg_decrypt
    gpgme_data_t gpgdecryptoutdata;
    gpgme_data_new(&gpgdecryptoutdata);
    if (gpgresult != 0) {
        kWarning() << gpgme_strerror(gpgresult);
        gpgme_data_release(gpgindata);
        gpgme_data_release(gpgoutdata);
        return;
    }

    gpgme_data_seek(gpgoutdata, 0, SEEK_SET);
    gpgresult = gpgme_op_decrypt(m_gpgctx, gpgoutdata, gpgdecryptoutdata);
    if (gpgresult != 0) {
        kWarning() << gpgme_strerror(gpgresult);
        gpgme_data_release(gpgindata);
        gpgme_data_release(gpgoutdata);
        gpgme_data_release(gpgdecryptoutdata);
        return;
    }

    size_t gpgbuffersize = 0;
    char* gpgbuffer = gpgme_data_release_and_get_mem(gpgdecryptoutdata, &gpgbuffersize);
    kWarning() << "decryption" << gpgbuffer;
#endif // kgpg_decrypt

    gpgme_free(gpgbuffer);
    gpgme_data_release(gpgindata);
    gpgme_data_release(gpgoutdata);
    
}

KGPG::~KGPG()
{
    // will crash if not initialized
    if (m_release) {
        gpgme_release(m_gpgctx);
    }
}

int main(int argc, char **argv)
{
    KAboutData aboutData("kgpg", 0, ki18n("KGPG"),
        "1.0.0", ki18n("KDE encryption and decryption utility"), KAboutData::License_GPL,
        ki18n("(c) 2022 Ivailo Monev"));
    aboutData.addAuthor(ki18n("Ivailo Monev"), KLocalizedString(), "xakepa10@gmail.com");

    KCmdLineArgs::init(argc, argv, &aboutData);

    KApplication app;

    KGPG* kgpg = new KGPG();
    kgpg->show();

    return app.exec();
}

#include "moc_kgpg.cpp"