/**********************************************************************
** Copyright (C) 2000-2002 Trolltech AS.  All rights reserved.
**
** This file is part of the Qtopia Environment.
**
** This file may be distributed and/or modified under the terms of the
** GNU General Public License version 2 as published by the Free Software
** Foundation and appearing in the file LICENSE.GPL included in the
** packaging of this file.
**
** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
**
** See http://www.trolltech.com/gpl/ for GPL licensing information.
**
** Contact info@trolltech.com if any conditions of this licensing are
** not clear to you.
**
**********************************************************************/

#include "qcopbridge.h"
#include "transferserver.h"

/* OPIE */
#include <qpe/qcopenvelope_qws.h>
#include <qpe/qpeapplication.h>
#include <qpe/version.h>
#include <opie2/odebug.h>

/* QT */
#include <qtextstream.h>
#ifdef QWS
#include <qcopchannel_qws.h>
#endif

/* STD */
#define _XOPEN_SOURCE
#include <pwd.h>
#include <sys/types.h>
#include <unistd.h>

#if defined(_OS_LINUX_)
#include <shadow.h>
#endif

//#define INSECURE

const int block_size = 51200;

QCopBridge::QCopBridge( Q_UINT16 port, QObject *parent ,
        const char* name )
    : QServerSocket( port, 1, parent, name ),
      desktopChannel( 0 ),
      cardChannel( 0 )
{
    if ( !ok() )
    owarn << "Failed to bind to port " << port << "" << oendl;
    else {
#ifndef QT_NO_COP
    desktopChannel = new QCopChannel( "QPE/Desktop", this );
    connect( desktopChannel, SIGNAL(received(const QCString&,const QByteArray&)),
         this, SLOT(desktopMessage(const QCString&,const QByteArray&)) );
    cardChannel = new QCopChannel( "QPE/Card", this );
    connect( cardChannel, SIGNAL(received(const QCString&,const QByteArray&)),
         this, SLOT(desktopMessage(const QCString&,const QByteArray&)) );
#endif
    }
    sendSync = FALSE;
}

QCopBridge::~QCopBridge()
{
#ifndef QT_NO_COP
    delete desktopChannel;
#endif
}

void QCopBridge::newConnection( int socket )
{
    QCopBridgePI *pi = new QCopBridgePI( socket, this );
    openConnections.append( pi );
    connect ( pi, SIGNAL( connectionClosed(QCopBridgePI*) ), this, SLOT( connectionClosed(QCopBridgePI*) ) );
#ifndef QT_NO_COP
    QCopEnvelope( "QPE/System", "setScreenSaverMode(int)" ) << QPEApplication::DisableSuspend;
#endif

    if ( sendSync ) {
    pi ->startSync();
    sendSync = FALSE;
    }
}

void QCopBridge::connectionClosed( QCopBridgePI *pi )
{
    openConnections.remove( pi );
    if ( openConnections.count() == 0 ) {
#ifndef QT_NO_COP
    QCopEnvelope( "QPE/System", "setScreenSaverMode(int)" ) << QPEApplication::Enable;
#endif
    }
}

void QCopBridge::closeOpenConnections()
{
    QCopBridgePI *pi;
    for ( pi = openConnections.first(); pi != 0; pi = openConnections.next() )
    pi->close();
}


void QCopBridge::desktopMessage( const QCString &command, const QByteArray &args )
{
    command.stripWhiteSpace();

    int paren = command.find( "(" );
    if ( paren <= 0 ) {
    odebug << "DesktopMessage: bad qcop syntax" << oendl;
    return;
    }

    QString params = command.mid( paren + 1 );
    if ( params[params.length()-1] != ')' ) {
    odebug << "DesktopMessage: bad qcop syntax" << oendl;
    return;
    }

    params.truncate( params.length()-1 );

    QStringList paramList = QStringList::split( ",", params );
    QString data;
    if ( paramList.count() ) {
    QDataStream stream( args, IO_ReadOnly );
    for ( QStringList::Iterator it = paramList.begin(); it != paramList.end(); ++it ) {
        QString str;
        if ( *it == "QString" ) {
        stream >> str;
        } else if ( *it == "QCString" ) {
        QCString cstr;
        stream >> cstr;
        str = QString::fromLocal8Bit( cstr );
        } else if ( *it == "int" ) {
        int i;
        stream >> i;
        str = QString::number( i );
        } else if ( *it == "bool" ) {
        int i;
        stream >> i;
        str = QString::number( i );
        } else {
        odebug << " cannot route the argument type " << (*it) << " throught the qcop bridge" << oendl;
        return;
        }
        QString estr;
        for (int i=0; i<(int)str.length(); i++) {
        QChar ch = str[i];
        if ( ch.row() )
            goto quick;
        switch (ch.cell()) {
            case '&':
            estr.append( "&amp;" );
            break;
            case ' ':
            estr.append( "&0x20;" );
            break;
            case '\n':
            estr.append( "&0x0d;" );
            break;
            case '\r':
            estr.append( "&0x0a;" );
            break;
            default: quick:
            estr.append(ch);
        }
        }
        data += " " + estr;
    }
    }
    QString sendCommand = QString(command.data()) + data;
    // send the command to all open connections
    if ( command == "startSync()" ) {
    // we need to buffer it a bit
    sendSync = TRUE;
    startTimer( 20000 );
    }

    QCopBridgePI *pi;
    for ( pi = openConnections.first(); pi != 0; pi = openConnections.next() ) {
    pi->sendDesktopMessage( sendCommand );
    }
}

void QCopBridge::timerEvent( QTimerEvent * )
{
    sendSync = FALSE;
    killTimers();
}


QCopBridgePI::QCopBridgePI( int socket, QObject *parent , const char* name )
    : QSocket( parent, name )
{
    setSocket( socket );

    peerport = peerPort();
    peeraddress = peerAddress();

#ifndef INSECURE
    if ( !SyncAuthentication::isAuthorized(peeraddress) ) {
    state = Forbidden;
    startTimer( 0 );
    } else
#endif
    {
    state = Connected;
    sendSync = FALSE;
    connect( this, SIGNAL( readyRead() ), SLOT( read() ) );
    connect( this, SIGNAL( connectionClosed() ), SLOT( connectionClosed() ) );

    QString intro="220 Qtopia ";
    intro += QPE_VERSION; intro += ";";
    intro += "challenge="; intro += SyncAuthentication::serverId(); intro += ";";
    intro += "loginname="; intro += SyncAuthentication::loginName(); intro += ";";
    intro += "displayname="; intro += SyncAuthentication::ownerName(); intro += ";";
    send( intro );
    state = Wait_USER;

    // idle timer to close connections when not used anymore
    startTimer( 60000 );
    connected = TRUE;
    }
}


QCopBridgePI::~QCopBridgePI()
{

}

void QCopBridgePI::connectionClosed()
{
    emit connectionClosed( this );
    // odebug << "Debug: Connection closed" << oendl;
    delete this;
}

void QCopBridgePI::sendDesktopMessage( const QString &msg )
{
    QString str = "CALL QPE/Desktop " + msg;
    send ( str );
}


void QCopBridgePI::send( const QString& msg )
{
    QTextStream os( this );
    os << msg << endl;
    //odebug << "sending qcop message: " << msg << "" << oendl;
}

void QCopBridgePI::read()
{
    while ( canReadLine() )
    process( readLine().stripWhiteSpace() );
}

void QCopBridgePI::process( const QString& message )
{
    //odebug << "Command: " << message << "" << oendl;

    // split message using "," as separator
    QStringList msg = QStringList::split( " ", message );
    if ( msg.isEmpty() ) return;

    // command token
    QString cmd = msg[0].upper();

    // argument token
    QString arg;
    if ( msg.count() >= 2 )
    arg = msg[1];

    // we always respond to QUIT, regardless of state
    if ( cmd == "QUIT" ) {
    send( "211 Have a nice day!" );
    delete this;
    return;
    }

    // connected to client
    if ( Connected == state )
    return;

    // waiting for user name
    if ( Wait_USER == state ) {

    if ( cmd != "USER" || msg.count() < 2 || !SyncAuthentication::checkUser( arg ) ) {
        send( "530 Please login with USER and PASS" );
        return;
    }
    send( "331 User name ok, need password" );
    state = Wait_PASS;
    return;
    }

    // waiting for password
    if ( Wait_PASS == state ) {

    if ( cmd != "PASS" || !SyncAuthentication::checkPassword( arg ) ) {
        send( "530 Please login with USER and PASS" );
        return;
    }
    send( "230 User logged in, proceed" );
    state = Ready;
    if ( sendSync ) {
        sendDesktopMessage( "startSync()" );
        sendSync = FALSE;
    }
    return;
    }

    // noop (NOOP)
    else if ( cmd == "NOOP" ) {
    connected = TRUE;
    send( "200 Command okay" );
    }

    // call (CALL)
    else if ( cmd == "CALL" ) {

    // example: call QPE/System execute(QString) addressbook

    if ( msg.count() < 3 ) {
        send( "500 Syntax error, command unrecognized" );
    }
    else {

        QString channel = msg[1];
        QString command = msg[2];

        command.stripWhiteSpace();

        int paren = command.find( "(" );
        if ( paren <= 0 ) {
        send( "500 Syntax error, command unrecognized" );
        return;
        }

        QString params = command.mid( paren + 1 );
        if ( params[params.length()-1] != ')' ) {
        send( "500 Syntax error, command unrecognized" );
        return;
        }

        params.truncate( params.length()-1 );
        QByteArray buffer;
        QDataStream ds( buffer, IO_WriteOnly );

        int msgId = 3;

        QStringList paramList = QStringList::split( ",", params );
        if ( paramList.count() > msg.count() - 3 ) {
        send( "500 Syntax error, command unrecognized" );
        return;
        }

        for ( QStringList::Iterator it = paramList.begin(); it != paramList.end(); ++it ) {

        QString arg = msg[msgId];
        arg.replace( QRegExp("&0x20;"), " " );
        arg.replace( QRegExp("&amp;"), "&" );
        arg.replace( QRegExp("&0x0d;"), "\n" );
        arg.replace( QRegExp("&0x0a;"), "\r" );
        if ( *it == "QString" )
            ds << arg;
        else if ( *it == "QCString" )
            ds << arg.local8Bit();
        else if ( *it == "int" )
            ds << arg.toInt();
        else if ( *it == "bool" )
            ds << arg.toInt();
        else {
            send( "500 Syntax error, command unrecognized" );
            return;
        }
        msgId++;
        }

#ifndef QT_NO_COP
        if ( !QCopChannel::isRegistered( channel.latin1() ) ) {
        // send message back about it
        QString answer = "599 ChannelNotRegistered " + channel;
        send( answer );
        return;
        }
#endif

#ifndef QT_NO_COP
        if ( paramList.count() )
        QCopChannel::send( channel.latin1(), command.latin1(), buffer );
        else
        QCopChannel::send( channel.latin1(), command.latin1() );

        send( "200 Command okay" );
#endif
    }
    }
    // not implemented
    else
    send( "502 Command not implemented" );
}



void QCopBridgePI::timerEvent( QTimerEvent * )
{
    if ( connected )
    connected = FALSE;
    else
    connectionClosed();
}