source: qca/trunk/examples/ciphertest/ciphertest.cpp@ 58

#include"qca.h"
#include<stdio.h>

static QCString arrayToCString(const QByteArray &);
static QByteArray cstringToArray(const QCString &);
static void doDynTest(QCA::Cipher *c, const QString &name, const QCString &cs);

int main(int argc, char **argv)
{
        QCA::init();
        QCString cs = (argc >= 2) ? argv[1] : "hello";

        // AES128 test
        if(!QCA::isSupported(QCA::CAP_AES128))
                printf("AES128 not supported!\n");
        else {
                // encrypt
                QByteArray key = QCA::AES128::generateKey();
                QByteArray iv = QCA::AES128::generateIV();
                printf("aes128:key:%s\n", QCA::arrayToHex(key).latin1());
                printf("aes128:iv:%s\n", QCA::arrayToHex(iv).latin1());
                QCA::AES128 c(QCA::Encrypt, QCA::CBC, key, iv);
                c.update(cstringToArray(cs));
                QByteArray f = c.final();
                QString result = QCA::arrayToHex(f);
                printf(">aes128(\"%s\") = [%s]\n", cs.data(), result.latin1());

                // decrypt
                QCA::AES128 d(QCA::Decrypt, QCA::CBC, key, iv);
                d.update(f);
                QCString dec = arrayToCString(d.final());
                printf("<aes128(\"%s\") = [%s]\n", result.latin1(), dec.data());
        }

        // BlowFish, TripleDES, and AES256 tested dynamically
        if(!QCA::isSupported(QCA::CAP_BlowFish))
                printf("BlowFish not supported!\n");
        else
                doDynTest(new QCA::BlowFish, "bfish", cs);

        if(!QCA::isSupported(QCA::CAP_TripleDES))
                printf("TripleDES not supported!\n");
        else
                doDynTest(new QCA::TripleDES, "3des", cs);

        if(!QCA::isSupported(QCA::CAP_AES256))
                printf("AES256 not supported!\n");
        else
                doDynTest(new QCA::AES256, "aes256", cs);

        return 0;
}

QCString arrayToCString(const QByteArray &a)
{
        QCString cs;
        cs.resize(a.size()+1);
        memcpy(cs.data(), a.data(), a.size());
        return cs;
}

QByteArray cstringToArray(const QCString &cs)
{
        QByteArray a(cs.length());
        memcpy(a.data(), cs.data(), a.size());
        return a;
}

void doDynTest(QCA::Cipher *c, const QString &name, const QCString &cs)
{
        // encrypt
        QByteArray key = c->dyn_generateKey();
        QByteArray iv = c->dyn_generateIV();
        printf("%s:key:%s\n", name.latin1(), QCA::arrayToHex(key).latin1());
        printf("%s:iv:%s\n", name.latin1(), QCA::arrayToHex(iv).latin1());
        c->reset(QCA::Encrypt, QCA::CBC, key, iv);
        c->update(cstringToArray(cs));
        QByteArray f = c->final();
        QString result = QCA::arrayToHex(f);
        printf(">%s(\"%s\") = [%s]\n", name.latin1(), cs.data(), result.latin1());

        // decrypt
        c->reset(QCA::Decrypt, QCA::CBC, key, iv);
        c->update(f);
        QCString dec = arrayToCString(c->final());
        printf("<%s(\"%s\") = [%s]\n", name.latin1(), result.latin1(), dec.data());
        delete c;
}