/****************************************************************************
**
** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
** All rights reserved.
** Contact: Nokia Corporation (qt-info@nokia.com)
**
** This file is part of the examples of the Qt Toolkit.
**
** $QT_BEGIN_LICENSE:LGPL$
** No Commercial Usage
** This file contains pre-release code and may not be distributed.
** You may use this file in accordance with the terms and conditions
** contained in the Technology Preview License Agreement accompanying
** this package.
**
** 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.
**
** In addition, as a special exception, Nokia gives you certain additional
** rights.  These rights are described in the Nokia Qt LGPL Exception
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
**
** If you have questions regarding the use of this file, please contact
** Nokia at qt-info@nokia.com.
**
**
**
**
**
**
**
**
** $QT_END_LICENSE$
**
****************************************************************************/

#include "ftpserver.h"

#define FTP_CMND_PORT 21
#define FTP_DATA_PORT 9801

ftpServer::ftpServer( QObject *parent  )
         : QTcpServer( parent )
{
    ftpSocket = NULL;
    dataSocket = NULL;

    tcpServer = new QTcpServer( /* this */ );
    not_done = true;
    bool result = tcpServer->listen(QHostAddress::Any, FTP_CMND_PORT );
    if ( !result )
    {
        delete tcpServer;
        close();
        not_done = false;
        return;
    }
    connect(tcpServer, SIGNAL(newConnection()), this, SLOT(handleNewConnection()));

/*
    dataServer = new QTcpServer( );
    result = dataServer->listen(QHostAddress::Any, FTP_DATA_PORT );
    if ( !result )
    {
        delete tcpServer;
        close();
        not_done = false;

        delete dataServer;
        close();
        return;
    }

    connect(dataServer, SIGNAL(newConnection()), this, SLOT(handleNewDataConnection()));
*/
}

ftpServer::~ftpServer()
{
    if ( ftpSocket )
    {
        ftpSocket->disconnect(SIGNAL(readyRead()));
        ftpSocket->disconnect(SIGNAL(error(QAbstractSocket::SocketError)));
        ftpSocket->close();
        ftpSocket->abort();
        ftpSocket->deleteLater();
        ftpSocket = NULL;
    }
}

bool ftpServer::isDone()
{
    return not_done;
}
void ftpServer::handleNewConnection()
{
    QTcpSocket *clientConnection = tcpServer->nextPendingConnection();
    clientConnection->setTextModeEnabled(true);

    if ( !ftpSocket )
    {
        ftpSocket = clientConnection;
        connect( ftpSocket,
            SIGNAL(readyRead()),
            this,
            SLOT(readConnection()));
        connect(ftpSocket, SIGNAL(error(QAbstractSocket::SocketError)),
             this, SLOT(displayConnectionError(QAbstractSocket::SocketError)));

        QString greet("220 Service ready\n");
        ftpSocket->write(qPrintable(greet));

    }
    else if ( !dataSocket )
    {
        dataSocket = clientConnection;
        connect( dataSocket,
            SIGNAL(readyRead()),
            this,
            SLOT(readConnection()));
        connect(dataSocket, SIGNAL(error(QAbstractSocket::SocketError)),
             this, SLOT(displayConnectionError(QAbstractSocket::SocketError)));

        if ( dataSocket->waitForConnected() )
        {
        //qDebug() << "Data connection accepted.\n";
        }


    }
    else
    {
        // We already are busy with open connection?!
        // qDebug() << "Server busy.\n";
        clientConnection->disconnectFromHost();
    }
}

void ftpServer::handleNewDataConnection()
{
    QTcpSocket *dataConnection = tcpServer->nextPendingConnection();
    dataConnection->setTextModeEnabled(true);

    if ( !dataSocket )
    {
        dataSocket = dataConnection;
        connect( dataSocket,
            SIGNAL(readyRead()),
            this,
            SLOT(readConnection()));
        connect(dataSocket, SIGNAL(error(QAbstractSocket::SocketError)),
             this, SLOT(displayConnectionError(QAbstractSocket::SocketError)));

        QString list(".\n..\nasdf.txt\nqwerty.txt\n");
        ftpSocket->write(qPrintable(list));

    }
    else
    {
        // We already are busy with open connection?!
        //qDebug() << "Data Server busy.\n";
        dataConnection->disconnectFromHost();
    }
}

void ftpServer::displayConnectionError(QAbstractSocket::SocketError socketError)
{
		Q_UNUSED(socketError);
    ftpSocket->disconnect(SIGNAL(readyRead()));
    ftpSocket->disconnect(SIGNAL(error(QAbstractSocket::SocketError)));
    ftpSocket->close();
    ftpSocket->abort();
    ftpSocket->deleteLater();
    ftpSocket = NULL;
}

void ftpServer::displayDataConnectionError(QAbstractSocket::SocketError socketError)
{
		Q_UNUSED(socketError);
    dataSocket->disconnect(SIGNAL(readyRead()));
    dataSocket->disconnect(SIGNAL(error(QAbstractSocket::SocketError)));
    dataSocket->close();
    dataSocket->abort();
    dataSocket->deleteLater();
    dataSocket = NULL;
}

void ftpServer::readDataConnection()
{
    while (dataSocket->canReadLine())
    {
        dataSocket->readAll();
    }
}

void ftpServer::readConnection()
{
    while (ftpSocket->canReadLine())
    {
        char buf[256];
        qint64 lineLength = ftpSocket->readLine(buf, sizeof(buf));
        if (lineLength != -1)
        {
            QString tmpCommand(buf);
            tmpCommand = tmpCommand.simplified();

            // qDebug() << "Received: '" << buf << "'\n";
            if ( tmpCommand.left(5) == QString("USER ") )
            {
                QString acceptuser("331 User name ok\n");
                ftpSocket->write(qPrintable(acceptuser));
            }
            else if ( tmpCommand.left(5) == QString("PASS ") )
            {
                QString acceptCommand("230 User logged in\n");
                ftpSocket->write(qPrintable(acceptCommand));
            }
            else if ( tmpCommand.left(5) == QString("TYPE ") )
            {
                QString acceptCommand("200 Command OK\n");
                // QString acceptCommand("227 Entering Passive Mode. 21,21,21,21,21,21\n");
                ftpSocket->write(qPrintable(acceptCommand));
            }
            else if ( tmpCommand.left(4) == QString("PASV") )
            {
                QString acceptCommand("227 Entering Passive Mode. (127,0,0,1,0,21)\n");
                // QString acceptCommand("200 Command OK\n");
                ftpSocket->write(qPrintable(acceptCommand));
            }
            else if ( tmpCommand.left(4) == QString("LIST") )
            {
                // QString acceptuser("502 Command not implemented\n");
                // QString acceptuser("-rw-r--r--    1 vogel    user     1209041 May  1 13:55 IMPJET24.CAS\n-rw-r--r--    1 vogel    user     10259798 May  1 20:57 IMPJET24.DAT\n");
            // qDebug() << "sending 200...\n";
                QString acceptCommand("200 Command OK\n");
                ftpSocket->write(qPrintable(acceptCommand));
                // qDebug() << "sending 150...\n";
                QString port;
                port.setNum(dataSocket->localPort());
                QString dataCommand("150 ASCII data connection for /bin/ls (127.0.0.1,");
                dataCommand.append( port );
                dataCommand.append(") (0 bytes).");
                ftpSocket->write(qPrintable(dataCommand));
                // not_done = false;

                //
                //QString list(".\n..\nasdf.txt\nqwerty.txt\n");
                QString list("07.12.2009\t13:49\t<DIR>\tinclude\n03.12.2009\t14:01\t721\tINSTALL\n");


                dataSocket->write(qPrintable(list));
                dataSocket->flush();
                dataSocket->disconnectFromHost();
                dataSocket = NULL;

                // qDebug() << "sending 226...\n";
                QString finishCommand("226 Closing data connection, file transfer successful.\n");
                ftpSocket->write(qPrintable(finishCommand));

                not_done = false;

            }
            else
            {
                // qDebug() << "Received unknown:" << buf << "\n";
            }
        }
    }
}