encrypt.cc

简单加密工具升了下级

// head files
#include <QFile>
#include <QByteArray>
#include <QTime>
#include "encrypt.h"
#include "aes.h"


// function implementations

ZT_Encrypt::ZT_Encrypt(QWidget* parent): QWidget(parent)
{
    // initialize
    initUI();
    initBinding();
}

ZT_Encrypt::~ZT_Encrypt()
{
    delete chkDelete;
    delete dlgSource;
    delete dlgDestination;
    delete lblSource;
    delete lblDestination;
    delete lblPassword;
    delete txtSource;
    delete txtDestination;
    delete txtPassword;
    delete msgMessage;
    delete btnEncrypt;
    delete btnAbout;
    delete btnDecrypt;
    delete btnBrowseSource;
    delete btnBrowseDestination;
}

void ZT_Encrypt::initUI()
{
    setWindowTitle(QObject::tr("ZT-Encrypt"));
    setFixedSize(480, 210);

    lblSource = new QLabel(QObject::tr("Source"), this);
    lblSource->setGeometry(QRect(20, 20, 100, 25));

    lblDestination = new QLabel(QObject::tr("Destination"), this);
    lblDestination->setGeometry(QRect(20, 55, 100, 25));

    lblPassword = new QLabel(QObject::tr("Password"), this);
    lblPassword->setGeometry(QRect(20, 90, 100, 25));

    chkDelete = new QCheckBox(QObject::tr("Delete"), this);
    chkDelete->setGeometry(QRect(20, 125, 200, 25));

    btnEncrypt = new QPushButton(QObject::tr("Encrypt"), this);
    btnEncrypt->setGeometry(QRect(40, 160, 80, 30));

    btnAbout = new QPushButton(QObject::tr("About"), this);
    btnAbout->setGeometry(QRect(280, 160, 80, 30));

    btnDecrypt = new QPushButton(QObject::tr("Decrypt"), this);
    btnDecrypt->setGeometry(QRect(160, 160, 80, 30));

    btnBrowseSource = new QPushButton(QObject::tr("Browse"), this);
    btnBrowseSource->setGeometry(QRect(380, 20, 80, 25));

    btnBrowseDestination = new QPushButton(QObject::tr("Browse"), this);
    btnBrowseDestination->setGeometry(QRect(380, 55, 80, 25));

    txtSource = new QLineEdit(this);
    txtSource->setGeometry(QRect(120, 20, 240, 25));

    txtDestination = new QLineEdit(this);
    txtDestination->setGeometry(QRect(120, 55, 240, 25));

    txtPassword = new QLineEdit(this);
    txtPassword->setGeometry(QRect(120, 90, 240, 25));

    msgMessage = new QMessageBox();
    msgMessage->setWindowTitle(QObject::tr("Info"));
    msgMessage->setWindowModality(Qt::ApplicationModal);

    dlgSource = new QFileDialog();
    dlgSource->setWindowTitle(QObject::tr("Open"));
    dlgSource->setWindowModality(Qt::ApplicationModal);

    dlgDestination = new QFileDialog();
    dlgDestination->setWindowModality(Qt::ApplicationModal);
}

void ZT_Encrypt::initBinding()
{
    QObject::connect(btnBrowseSource, SIGNAL(clicked()),
                     this, SLOT(showBrowseSource()));

    QObject::connect(btnBrowseDestination, SIGNAL(clicked()),
                     this, SLOT(showBrowseDestination()));

    QObject::connect(dlgSource, SIGNAL(filesSelected(const QStringList&)),
                     this, SLOT(setSource(const QStringList&)));

    QObject::connect(dlgDestination, SIGNAL(filesSelected(const QStringList&)),
                     this, SLOT(setDestination(const QStringList&)));

    QObject::connect(btnEncrypt, SIGNAL(clicked()),
                     this, SLOT(encrypt()));

    QObject::connect(btnDecrypt, SIGNAL(clicked()),
                     this, SLOT(decrypt()));
}

void ZT_Encrypt::showBrowseSource()
{
    dlgSource->show();
}

void ZT_Encrypt::showBrowseDestination()
{
    dlgDestination->show();
}

void ZT_Encrypt::setSource(const QStringList& path)
{
    txtSource->setText(path.at(0));
}

void ZT_Encrypt::setDestination(const QStringList& path)
{
    txtDestination->setText(path.at(0));
}

void ZT_Encrypt::encrypt()
{
    doEncryption(0);
}

void ZT_Encrypt::decrypt()
{
    doEncryption(1);
}

void ZT_Encrypt::doEncryption(int op)
{
    // check if op is valid
    if (op != 0 && op != 1)
    {
        return ;
    }

    // check if we have the password
    if (txtPassword->text().length() == 0)
    {
        return ;
    }

    // open source file
    QFile src(txtSource->text());
    if (!src.exists())
    {
        // the specified source file doesn't exist
        return ;
    }
    src.open(QIODevice::ReadWrite);

    // open destination file
    QFile dst(txtDestination->text());
    if (dst.exists())
    {
        // warn the user that the specified destination file being overwritten
    }
    dst.open(QIODevice::WriteOnly);

    // prepare the key and iv
    QByteArray tmp_pass = txtPassword->text().toLatin1();
    char* data = tmp_pass.data();
    int len = tmp_pass.length();
    int temp = 0;

    AES_KEY aes_key;
    unsigned char key[16] = {0};
    for (int i = 0; i < len; i++)
    {
        temp += (data[i] << i);
    }
    qsrand(temp);
    for (int i = 0; i < 16; i++)
    {
        key[i] = qrand();
    }

    char iv[16] = {0};
    temp = 0;
    for (int i = len - 1; i >= 0; i--)
    {
        temp += (data[i] << i);
    }
    qsrand(temp);
    for (int i = 0; i < 16; i++)
    {
        iv[i] = qrand();
    }

    // do the encryption
    char in[512];
    char out[512];
    qint64 size = src.size();
    qint64 read = 0;
    qint64 tmp = 0;
    while (read < size)
    {
        tmp = src.read(in, 512);
        if (op == 0)
        {
            // encrypt
            AES_set_encrypt_key(key, 128, &aes_key);
            AES_cbc_encrypt((unsigned char*)in, (unsigned char*)out, tmp, &aes_key, (unsigned char*)iv, AES_ENCRYPT);
        }
        else
        {
            // decrypt
            AES_set_decrypt_key(key, 128, &aes_key);
            AES_cbc_encrypt((unsigned char*)in, (unsigned char*)out, tmp, &aes_key, (unsigned char*)iv, AES_DECRYPT);
        }
        dst.write(out, tmp);
        read += tmp;
    }

    // delete source if checked
    if (chkDelete->isChecked())
    {
        qint64 size = src.size();
        char buffer[4096];
        for (int i = 0; i < 4; i++)
        {
            qsrand(QTime::currentTime().msec());
            for (int k = 0; k < 4096; k++)
            {
                buffer[k] = qrand();
            }
            int j = 0;
            while (j < size)
            {
                src.write(buffer, 1024);
                j += 1024;
            }
        }
        src.resize(qrand());
        src.remove();
    }

    // show the message of job done
    if (op == 0)
    {
        msgMessage->setText(QObject::tr("EncryptComplete"));
    }
    else
    {
        msgMessage->setText(QObject::tr("DecryptComplete"));
    }
    msgMessage->show();
}