pgiticlog.cpp

#include "pgiticlog.h"
#include <QSqlDatabase>
#include <QSqlError>
#include <QSqlQuery>
#include <QVariant>
#include <QDateTime>
#include <aes.h>
#include <QCryptographicHash>
#include <QFile>
#include <QTextStream>
#include <statics.h>

const uint8_t iv[] = { 0xf0, 0xe1, 0xd2, 0xc3, 0xb4, 0xa5, 0x96,
                       0x87, 0x78, 0x69, 0x5a, 0x4b, 0x3c, 0x2d, 0x5e, 0xaf };

inline int getAlignedSize(int currSize, int alignment) {
    int padding = (alignment - currSize % alignment) % alignment;
    return currSize + padding;
}

pgiticlog::pgiticlog(QObject *parent) :
    QObject(parent)
{

}

QString encodeText(const QString &rawText, const QString &key) {
        QCryptographicHash hash(QCryptographicHash::Md5);
        hash.addData(key.toUtf8());
        QByteArray keyData = hash.result();

        const ushort *rawData = rawText.utf16();
        void *rawDataVoid = (void*)rawData;
        const char *rawDataChar = static_cast<char*>(rawDataVoid);
        QByteArray inputData;
        // ushort is 2*uint8_t + 1 byte for '\0'
        inputData.append(rawDataChar, rawText.size() * 2 + 1);

        const int length = inputData.size();
        int encryptionLength = getAlignedSize(length, 16);

        QByteArray encodingBuffer(encryptionLength, 0);
        inputData.resize(encryptionLength);

        AES128_CBC_encrypt_buffer((uint8_t*)encodingBuffer.data(), (uint8_t*)inputData.data(),
           encryptionLength, (const uint8_t*)keyData.data(), iv);

        QByteArray data(encodingBuffer.data(), encryptionLength);
        QString hex = QString::fromLatin1(data.toHex());
        return hex;
    }

QString decodeText(const QString &hexEncodedText, const QString &key)
{
        QCryptographicHash hash(QCryptographicHash::Md5);
        hash.addData(key.toUtf8());
        QByteArray keyData = hash.result();

        const int length = hexEncodedText.size();
        int encryptionLength = getAlignedSize(length, 16);

        QByteArray encodingBuffer(encryptionLength, 0);

        QByteArray encodedText = QByteArray::fromHex(hexEncodedText.toLatin1());
        encodedText.resize(encryptionLength);

        AES128_CBC_decrypt_buffer((uint8_t*)encodingBuffer.data(), (uint8_t*)encodedText.data(),
          encryptionLength, (const uint8_t*)keyData.data(), iv);

        void *data = encodingBuffer.data();
        const ushort *decodedData = static_cast<ushort*>(data);
        QString result = QString::fromUtf16(decodedData);
        return result;
}

bool pgiticlog::save_config()
{
    QSqlQuery qry;

    std::string command;

    std::string _joyx = (joyx)?"1":"0";
    std::string _joyy = (joyy)?"1":"0";
    std::string _rememberme = (rememberme)?"1":"0";
    std::string _logout_idle = (logout_idle)?"1":"0";
    std::string _cmdloop = (cmd_loop)?"1":"0";
    std::string _loopvalue = QString::number(loop_value).toStdString().c_str();
    std::string _remoteip = "'" + remote_ip + "'";
    std::string _remoteport = "'" + remote_port + "'";
    std::string _admin_pass = "'" + admin_pass + "'";
    std::string _user_pass = "'" + user_pass + "'";
    std::string _controller_model = "'" + controller_model + "'";
    std::string _camera_model = "'" + camera_model + "'";
    std::string _camera_speed =  QString::number(camera_speed).toStdString().c_str();
    std::string _speaker_volume =  QString::number(speaker_volume).toStdString().c_str();

    std::string _serial1_name = "'" + serial1_name + "'";
    std::string _serial2_name = "'" + serial2_name + "'";
    std::string _serial3_name = "'" + serial3_name + "'";

    std::string _serial1_baud = QString::number(serial1_baud).toStdString().c_str();
    std::string _serial2_baud = QString::number(serial2_baud).toStdString().c_str();
    std::string _serial3_baud = QString::number(serial3_baud).toStdString().c_str();


    command = "UPDATE config SET";
    command.append("  joyx = ")              ; command.append(_joyx);
    command.append(", joyy = ")              ; command.append(_joyy);
    command.append(", cmd_loop = ")          ; command.append(_cmdloop);
    command.append(", logout_idle= ")        ; command.append(_logout_idle);
    command.append(", loop_value = ")        ; command.append(_loopvalue);
    command.append(", remote_ip = ")         ; command.append(_remoteip );
    command.append(", remote_port = ")       ; command.append(_remoteport);
    command.append(", admin_pass = ")        ; command.append(_admin_pass);
    command.append(", user_pass = ")         ; command.append(_user_pass);
    command.append(", controller_model = ")  ; command.append(_controller_model);
    command.append(", camera_model = ")      ; command.append(_camera_model);
    command.append(", cameraspeed = ")       ; command.append(_camera_speed);
    command.append(", volume = ")            ; command.append(_speaker_volume);
    command.append(", rememberme = ")        ; command.append(_rememberme);
    command.append(", serial1name = ")        ; command.append(_serial1_name);
    command.append(", serial2name = ")        ; command.append(_serial2_name);
    command.append(", serial3name = ")        ; command.append(_serial3_name);
    command.append(", serial1baudrate = ")    ; command.append(_serial1_baud);
    command.append(", serial2baudrate = ")    ; command.append(_serial2_baud);
    command.append(", serial3baudrate = ")    ; command.append(_serial3_baud);

    command.append(" WHERE id = 1" );

    std::cout<<command<<std::endl;

    // mtlog->insert_log("pgiticlog",command.c_str(),"DEBUG");

    qry.prepare(command.c_str());


    if( !qry.exec() )
    {
        mtlog->insert_log("pgiticlog",qry.lastError().text(),"ERROR"); return false;
    }
    else
    {
        mtlog->insert_log("pgiticlog","Updated","DEBUG"); return true;
    }

}

void pgiticlog::load_licenses()
{
    _key  = "pgitickeyabcdefg";
    QSqlQuery query;
    if(query.exec("SELECT * FROM license"))
    {
        if(query.next())
        {
            int id = query.value(0).toInt();
            QString _id = QString::number(id);

            _license1 = query.value(1).toString();
            _license2 = query.value(2).toString();
            _license3 = query.value(3).toString();
            _license4 = query.value(4).toString();

            if ( _license1 != "none")
            {
            _license1 = decodeText(_license1,_key);

                            if ( _license1 == (HID + "a").c_str())
                            {
                               _lic1 = true;
                            }
                            else
                            {
                               _lic1 = false;
                            }
            }

            if ( _license2 != "none")
            {
            _license2 = decodeText(_license2,_key);

                            if ( _license2 == (HID + "b").c_str())
                            {
                                 _lic2 = true;
                            }
                            else
                            {
                                  _lic2 = false;
                            }
            }

            if ( _license3 != "none")
            {
            _license3 = decodeText(_license3,_key);

                            if ( _license3 == (HID + "c").c_str())
                            {
                                _lic3 = true;
                            }
                            else
                            {
                               _lic3 = false;
                            }
            }

            if ( _license4 != "none")
            {
            _license4 = decodeText(_license4,_key);

                            if ( _license4 == (HID + "d").c_str())
                            {
                                  _lic4 = true;
                            }
                            else
                            {
                                  _lic4 = false;
                            }
            }


           mtlog->insert_log("pgiticlog","Licenses loaded done","DEBUG");
        }

    }
    else
    {
    }



}

void pgiticlog::load_config()
{
    QSqlQuery query;
    if(query.exec("SELECT * FROM config"))
    {
        if(query.next())
        {
            int id = query.value(0).toInt();
            QString _id = QString::number(id);
            joyx = query.value(1).toBool();
            joyy = query.value(2).toBool();
            cmd_loop = query.value(3).toBool();
            logout_idle = query.value(4).toBool();
            loop_value = query.value(5).toInt();
            remote_ip = query.value(6).toString().toStdString();
            remote_port = query.value(7).toString().toStdString();
            admin_pass = query.value(8).toString().toStdString();
            user_pass = query.value(9).toString().toStdString();
            controller_model = query.value(10).toString().toStdString();
            camera_model = query.value(11).toString().toStdString();
            camera_speed = query.value(12).toInt();
            speaker_volume = query.value(13).toInt();
            rememberme = query.value(14).toBool();
            superuser_pass = query.value(15).toString().toStdString();
            serial1_name = query.value(16).toString().toStdString();
            serial1_baud = query.value(17).toInt();
            serial2_name = query.value(18).toString().toStdString();
            serial2_baud = query.value(19).toInt();
            serial3_name = query.value(20).toString().toStdString();
            serial3_baud = query.value(21).toInt();
            mtlog->insert_log("pgiticlog","Config loaded done","DEBUG");


        }

    }
    else
    {
    }
}

void pgiticlog::deletealllogs()
{
    QSqlQuery qry;
    qry.prepare( "DELETE FROM log" );
    if( !qry.exec() )
        mtlog->insert_log("pgiticlog",qry.lastError().text(),"ERROR");
    else
       mtlog->insert_log("pgiticlog","All logs deleted","DEBUG");
}

void pgiticlog::deleteallhardlogs()
{
    QSqlQuery qry;
    qry.prepare( "DELETE FROM hardlog" );
    if( !qry.exec() )
        mtlog->insert_log("pgiticlog",qry.lastError().text(),"ERROR");
    else
       mtlog->insert_log("pgiticlog","All hard logs deleted","DEBUG");
}

std::vector<std::string>  pgiticlog::get_log()
{
    std::vector<std::string> list;
    QSqlQuery query;
    if(query.exec("SELECT * FROM log order by id DESC limit 100"))
    {
        while(query.next())
        {
            int id = query.value(0).toInt();
            QString _id = QString::number(id);
            QDateTime datetime = query.value(1).toDateTime();
            QString sender = query.value(2).toString();
            QString type = query.value(3).toString();
            QString info = query.value(4).toString();

            std::string itemm = "(" + datetime.toString("yyyy-MM-dd HH:mm:ss").toStdString() + ") [" + sender.toStdString() + "] (" + type.toStdString() + ") => " + info.toStdString();
            list.push_back(itemm);

        }
        return list;
    }
    else
    {
    }
}

std::vector<std::string>  pgiticlog::get_hard_log()
{
    std::vector<std::string> list;
    QSqlQuery query;
    if(query.exec("SELECT * FROM hardlog order by id DESC limit 100"))
    {
        while(query.next())
        {
            int id = query.value(0).toInt();
            QString _id = QString::number(id);
            QDateTime datetime = query.value(1).toDateTime();
            QString sender = query.value(2).toString();
            QString type = query.value(3).toString();
            QString info = query.value(4).toString();
            QString from = query.value(5).toString();

            std::string itemm = "(" + datetime.toString("yyyy-MM-dd HH:mm:ss").toStdString() + ") [" + sender.toStdString() + "] (" + type.toStdString() + ")" + " (" + from.toStdString() + ")" + " => " + info.toStdString();
            list.push_back(itemm);

        }
        return list;
    }
    else
    {
    }
}

std::vector<std::string>  pgiticlog::get_log_query(std::string cmd)
{
    std::vector<std::string> list;
    QSqlQuery query;
    if(query.exec(cmd.c_str()))
    {
        while(query.next())
        {
            int id = query.value(0).toInt();
            QString _id = QString::number(id);
            QDateTime datetime = query.value(1).toDateTime();
            QString sender = query.value(2).toString();
            QString type = query.value(3).toString();
            QString info = query.value(4).toString();

            std::string itemm = "(" + datetime.toString("yyyy-MM-dd HH:mm:ss").toStdString() + ") [" + sender.toStdString() + "] (" + type.toStdString() + ") => " + info.toStdString();
            list.push_back(itemm);

        }
        return list;
    }
    else
    {
    }


}

std::vector<std::string>  pgiticlog::get_hard_log_query(std::string cmd)
{
    std::vector<std::string> list;
    QSqlQuery query;
    if(query.exec(cmd.c_str()))
    {
        while(query.next())
        {
            int id = query.value(0).toInt();
            QString _id = QString::number(id);
            QDateTime datetime = query.value(1).toDateTime();
            QString sender = query.value(2).toString();
            QString type = query.value(3).toString();
            QString info = query.value(4).toString();
            QString from = query.value(5).toString();

            std::string itemm = "(" + datetime.toString("yyyy-MM-dd HH:mm:ss").toStdString() + ") [" + sender.toStdString() + "] (" + type.toStdString() + ")" + " (" + from.toStdString() + ")" + " => " + info.toStdString();
            list.push_back(itemm);

        }
        return list;
    }
    else
    {
    }


}

void  pgiticlog::insert_log(QString sender,QString info,QString type)
{
    QSqlQuery qry;
    std::string command;
    command = "INSERT INTO log (sender, info, type, timestamp) VALUES ('" +
            sender.toStdString() + "','" +
            info.toStdString()   + "','" +
            type.toStdString()   + "'," +
            "DateTime('now'))";

    qry.prepare(command.c_str());


    if( !qry.exec() )
         mtlog->insert_log("pgiticlog",qry.lastError().text(),"ERROR");

}

void  pgiticlog::insert_hard_log(QString sender,QString info,QString type,QString from)
{
    QSqlQuery qry;
    std::string command;
    command = "INSERT INTO hardlog (sender, info, type, source, timestamp) VALUES ('" +

               sender.toStdString() + "','" +
               info.toStdString()   + "','" +
               type.toStdString()   + "','" +
               from.toStdString()   + "',"  +
               "DateTime('now'))";

    qry.prepare(command.c_str());


    if( !qry.exec() )
    {
         mtlog->insert_log("pgiticlog",qry.lastError().text(),"ERROR");

    }


}

void pgiticlog::open()
{
    db = QSqlDatabase::addDatabase("QSQLITE");
    db.setDatabaseName("/home/pi/database/data.db");
    bool result = db.open();
    if ( result )
    mtlog->insert_log("pgiticlog","Database opened","DEBUG");
    else
    mtlog->insert_log("pgiticlog","Database read error","ERROR");

    QSqlQuery qry;
    qry.exec("VACUUM");
}

void pgiticlog::close()
{
    db.close();
}

void pgiticlog::start()
{
    mtlog->insert_log("pgiticlog","Serial Interface Started","INFO");

    //Read Hardware ID
    QFile file("/proc/cpuinfo");
    if(!file.open(QIODevice::ReadOnly)) {
         mtlog->insert_log("pgiticlog","CPU ID read error","ERROR");
    }

    QTextStream in(&file);
    QString line = in.readAll();
    int index = line.lastIndexOf("Serial		:");
    QString sub = line.mid(index,line.length() - index);
    sub = sub.replace("Serial		:","");
    sub = sub.trimmed();
    HID = sub.toStdString();
    file.close();
}