source: psi/trunk/cutestuff/enctest.cpp@ 80

#include<qdom.h>
#include<qtextstream.h>
#include<stdio.h>

#include"util/cipher.h"
#include"xmlsec/xmlenc.h"

static QCString elemToString(const QDomElement &e)
{
        QString out;
        QTextStream ts(&out, IO_WriteOnly);
        e.save(ts, 1);
        return out.utf8();
}

int main()
{
        QDomDocument doc;
        QDomElement order = doc.createElement("order");
        doc.appendChild(order);

        QDomElement e = doc.createElement("creditcard");
        e.appendChild(doc.createTextNode("1234 5678 9012 3456"));
        order.appendChild(e);
        printf("---- Original\n%s----\n\n", elemToString(order).data());

        Cipher::Type mainKeyType = Cipher::TripleDES;
        Cipher::Type subKeyType = Cipher::AES_256;

        RSAKey rsa = generateRSAKey();
        if(rsa.isNull()) {
                printf("RSA key generation failed!\n");
                return 0;
        }
        bool ok;
        QCString cs = "Hello.";
        printf("cs[%d]\n", cs.size());
        QByteArray result = encryptRSA2(cs, rsa, &ok);
        if(!ok) {
                printf("Unable to RSA encrypt!\n");
                return 0;
        }
        printf("result[%d]\n", result.size());
        QByteArray result2 = decryptRSA2(result, rsa, &ok);
        if(!ok) {
                printf("Unable to RSA decrypt!\n");
                return 0;
        }
        printf("result2[%d]\n", result2.size());
        printf("{%s}\n", result2.data());

        QByteArray iv = Cipher::generateIV(mainKeyType);
        printf("ivsize=[%d]\n", iv.size());
        Cipher::Key key1 = Cipher::generateKey(mainKeyType);
        printf("keysize=[%d]\n", key1.data().size());
        Cipher::Key key = Cipher::generateKey(subKeyType);
        printf("Key: { ");
        for(int n = 0; n < (int)key.data().size(); ++n)
                printf("%02x", (unsigned char)key.data()[n]);
        printf(" }\n\n");

        XmlEnc::Encrypted dat;
        dat.encryptKey(key, key1);
        QDomElement encKey = dat.toXml(&doc);

        //printf("---- Encrypted Key\n%s----\n\n", elemToString(encKey).data());

        XmlEnc::KeyInfo ki;
        ki.setName("My Key");
        //ki.attachEncryptedKey(encKey);
        ki.setValue(key.data());

        dat.setKeyInfo(ki);
        dat.encryptElement(e, key);
        QDomElement enc = dat.toXml(&doc);
        order.replaceChild(enc, e);
        printf("---- Encrypt\n%s----\n\n", elemToString(order).data());

        XmlEnc::Encrypted de;
        if(!de.fromXml(encKey))
                printf("Error: could not read encrypted xml!\n");
        QByteArray r = de.decryptKey(key1);

        Cipher::Key realKey;
        realKey.setData(r);
        realKey.setType(subKeyType);

        if(!de.fromXml(enc))
                printf("Error: could not read encrypted xml!\n");

        QDomElement dec = de.decryptElement(&doc, realKey);
        order.replaceChild(dec, enc);
        printf("---- Decrypt\n%s----\n\n", elemToString(order).data());
}