/**************************************************************************
**
** This file is part of Qt Creator
**
** Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
**
** Contact: Nokia Corporation (qt-info@nokia.com)
**
** Commercial Usage
**
** Licensees holding valid Qt Commercial licenses may use this file in
** accordance with the Qt Commercial License Agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and Nokia.
**
** GNU Lesser General Public License Usage
**
** Alternatively, this file may be used under the terms of the GNU Lesser
** General Public License version 2.1 as published by the Free Software
** Foundation and appearing in the file LICENSE.LGPL included in the
** packaging of this file.  Please review the following information to
** ensure the GNU Lesser General Public License version 2.1 requirements
** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
**
** If you are unsure which license is appropriate for your use, please
** contact the sales department at http://www.qtsoftware.com/contact.
**
**************************************************************************/

#include "trkutils.h"

#include <QtCore/QCoreApplication>
#include <QtCore/QFile>

#include <QtNetwork/QLocalServer>
#include <QtNetwork/QLocalSocket>


#ifdef Q_OS_UNIX

#include <signal.h>

void signalHandler(int)
{
    qApp->exit(1);
}

#endif

using namespace trk;

// Format of the replay source is something like
// ---IDE------------------------------------------------------
//  Command: 0x05 Support Mask
// [05 02]
//---TRK------------------------------------------------------
//  Command: 0x80 Acknowledge
//    Error: 0x00
// [80 02 00 7E 00 4F 5F 01 00 00 00 0F 1F 00 00 00
//  00 00 00 01 00 11 00 03 00 00 00 00 00 03 00 00...]

struct Inferior
{
    Inferior();
    uint pid;
    uint tid;
    uint codeseg;
    uint dataseg;
};

Inferior::Inferior()
{
    pid = 0x000008F5;
    tid = 0x000008F6;
    codeseg = 0x786A4000;
    dataseg = 0x00400000;
}

class TrkServer : public QObject
{
    Q_OBJECT

public:
    TrkServer();
    ~TrkServer();

    void setServerName(const QString &name) { m_serverName = name; }
    void setMemoryDumpName(const QString &source) { m_memoryDumpName = source; }
    void startServer();

private slots:
    void handleConnection();
    void readFromAdapter();
    void writeToAdapter(byte command, byte token, const QByteArray &data);

    void handleAdapterMessage(const TrkResult &result);

private:
    void logMessage(const QString &msg);
    byte nextNotificationToken();

    QString m_serverName;
    QString m_memoryDumpName;

    QByteArray m_adapterReadBuffer;

    QByteArray m_memoryData;
    QLocalServer m_server;
    int m_lastSent;
    QLocalSocket *m_adapterConnection;
    Inferior m_inferior;
    byte m_notificationToken;
};

TrkServer::TrkServer()
{
    m_adapterConnection = 0;
    m_notificationToken = 0;
}

TrkServer::~TrkServer()
{
    logMessage("Shutting down.\n");
    m_server.close();
}

void TrkServer::startServer()
{
    QFile file(m_memoryDumpName);
    file.open(QIODevice::ReadOnly);
    m_memoryData = file.readAll();
    file.close();

    logMessage(QString("Read %1 bytes of data from %2")
        .arg(m_memoryData.size()).arg(m_memoryDumpName));

    m_lastSent = 0;
    if (!m_server.listen(m_serverName)) {
        logMessage(QString("Error: Unable to start the TRK server %1: %2.")
            .arg(m_serverName).arg(m_server.errorString()));
        return;
    }

    logMessage("The TRK server is running. Run the adapter now.");
    connect(&m_server, SIGNAL(newConnection()), this, SLOT(handleConnection()));
}

void TrkServer::logMessage(const QString &msg)
{
    qDebug() << "TRKSERVER: " << qPrintable(msg);
}

void TrkServer::handleConnection()
{
    m_adapterConnection = m_server.nextPendingConnection();
    connect(m_adapterConnection, SIGNAL(disconnected()),
            m_adapterConnection, SLOT(deleteLater()));
    connect(m_adapterConnection, SIGNAL(readyRead()),
            this, SLOT(readFromAdapter()));
}

void TrkServer::readFromAdapter()
{
    QByteArray packet = m_adapterConnection->readAll();
    m_adapterReadBuffer.append(packet);

    logMessage("trk: -> " + stringFromArray(packet));

    if (packet != m_adapterReadBuffer)
        logMessage("buffer: " + stringFromArray(m_adapterReadBuffer));

    while (!m_adapterReadBuffer.isEmpty())
        handleAdapterMessage(extractResult(&m_adapterReadBuffer));
}

void TrkServer::writeToAdapter(byte command, byte token, const QByteArray &data)
{
    QByteArray msg = frameMessage(command, token, data);
    logMessage("trk: <- " + stringFromArray(msg));
    m_adapterConnection->write(msg);
}

void TrkServer::handleAdapterMessage(const TrkResult &result)
{
    QByteArray data;
    data.append(char(0x00));  // No error
    switch (result.code) {
        case 0x00: { // Ping
            m_notificationToken = 0;
            writeToAdapter(0x80, 0x00, data);
            break;
        }
        case 0x01: { // Connect
            writeToAdapter(0x80, result.token, data);
            break;
        }
        case 0x10: { // Read Memory
            const char *p = result.data.data();
            byte option = p[0];
            Q_UNUSED(option);
            ushort len = extractShort(p + 1);
            uint addr = extractInt(p + 3); 
            //qDebug() << "MESSAGE: " << result.data.toHex();
            qDebug() << "ADDR: " << hexNumber(addr) << " " << hexNumber(len);
            if (addr < m_inferior.codeseg
                || addr + len >= m_inferior.codeseg + m_memoryData.size()) {
                qDebug() << "ADDRESS OUTSIDE CODESEG: " << hexNumber(addr)
                    << hexNumber(m_inferior.codeseg); 
                for (int i = 0; i != len / 4; ++i)
                    appendInt(&data, 0xDEADBEEF);
            }
            for (int i = 0; i != len; ++i)
                appendByte(&data, m_memoryData[addr - m_inferior.codeseg + i]);
            writeToAdapter(0x80, result.token, data);
            break;
        }
        case 0x12: { // Read Registers
            appendByte(&data, 0x00);
            appendByte(&data, 0x00);
            appendByte(&data, 0x00);
            appendInt(&data, 0xC92D7FBC, BigEndian);
            appendInt(&data, 0x00000000, BigEndian);
            appendInt(&data, 0x00600000, BigEndian);
            appendInt(&data, 0x00000000, BigEndian);
            appendInt(&data, 0x786A7970, BigEndian);
            appendInt(&data, 0x00000000, BigEndian);
            appendInt(&data, 0x00000000, BigEndian);
            appendInt(&data, 0x00000012, BigEndian);
            appendInt(&data, 0x00000040, BigEndian);
            appendInt(&data, 0xC82AF210, BigEndian);
            appendInt(&data, 0x00000000, BigEndian);
            appendInt(&data, 0xC8000548, BigEndian);
            appendInt(&data, 0x00403ED0, BigEndian);
            appendInt(&data, 0x786A6BD8, BigEndian);
            appendInt(&data, 0x786A4CC8, BigEndian);
            appendInt(&data, 0x68000010, BigEndian);
            writeToAdapter(0x80, result.token, data);
            break;
        }
        case 0x18: { // Continue
            writeToAdapter(0x80, result.token, data); // ACK Package

            if (0) { // Fake "Stop" 
                QByteArray note;
                appendInt(&note, 0); // FIXME: use proper address
                appendInt(&note, m_inferior.pid);
                appendInt(&note, m_inferior.tid);
                appendByte(&note, 0x00);
                appendByte(&note, 0x00);
                writeToAdapter(0x90, nextNotificationToken(), note);
            }
            
            break;
        }
        case 0x40: { // Create Item
            appendInt(&data, m_inferior.pid, BigEndian);
            appendInt(&data, m_inferior.tid, BigEndian);
            appendInt(&data, m_inferior.codeseg, BigEndian);
            appendInt(&data, m_inferior.dataseg, BigEndian);
            writeToAdapter(0x80, result.token, data);
            break;
        }
        case 0x41: { // Delete Item
            writeToAdapter(0x80, result.token, data);

            // A Process?
            // Command: 0xA1 Notify Deleted
            //[A1 02 00 00 00 00 00 00 00 00 01 B5]
            QByteArray note; // FIXME
            appendByte(&note, 0);
            appendByte(&note, 0);
            appendInt(&note, 0);
            appendInt(&note, m_inferior.pid);
            writeToAdapter(0xA1, nextNotificationToken(), note);
            break;
        }
        default:
            data[0] = 0x10; // Command not supported
            writeToAdapter(0xff, result.token, data);
            break;
    }

}

byte TrkServer::nextNotificationToken()
{
    ++m_notificationToken;
    if (m_notificationToken == 0)
        ++m_notificationToken;
    return m_notificationToken;
}

int main(int argc, char *argv[])
{
    if (argc < 3) {
        qDebug() << "Usage: " << argv[0] << " <trkservername>"
            << " <replaysource>";
        return 1;
    }

#ifdef Q_OS_UNIX
    signal(SIGUSR1, signalHandler);
#endif

    QCoreApplication app(argc, argv);

    TrkServer server;
    server.setServerName(argv[1]);
    server.setMemoryDumpName(argv[2]);
    server.startServer();

    return app.exec();
}

#include "trkserver.moc"