/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the QtDBus module 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 <qdebug.h>

#include <qcoreapplication.h>

#include <qstringlist.h>

#include "qdbusconnection.h"

#include "qdbusconnectioninterface.h"

#include "qdbuserror.h"

#include "qdbusmessage.h"

#include "qdbusmessage_p.h"

#include "qdbusconnection_p.h"

#include "qdbusinterface_p.h"

#include "qdbusutil_p.h"

#include "qdbusthreaddebug_p.h"

QT_BEGIN_NAMESPACE

class QDBusConnectionManager

{

public:

    QDBusConnectionManager() {}

    ~QDBusConnectionManager();

    QDBusConnectionPrivate *connection(const QString &name) const;

    void removeConnection(const QString &name);

    void setConnection(const QString &name, QDBusConnectionPrivate *c);

    QDBusConnectionPrivate *sender() const;

    void setSender(const QDBusConnectionPrivate *s);

    mutable QMutex mutex;

private:

    QHash<QString, QDBusConnectionPrivate *> connectionHash;

    mutable QMutex senderMutex;

    QString senderName; // internal; will probably change

};

Q_GLOBAL_STATIC(QDBusConnectionManager, _q_manager)

QDBusConnectionPrivate *QDBusConnectionManager::sender() const

{

    QMutexLocker locker(&senderMutex);

    return connection(senderName);

}

void QDBusConnectionManager::setSender(const QDBusConnectionPrivate *s)

{

    QMutexLocker locker(&senderMutex);

    senderName = (s ? s->name : QString());

}

QDBusConnectionPrivate *QDBusConnectionManager::connection(const QString &name) const

{

    return connectionHash.value(name, 0);

}

void QDBusConnectionManager::removeConnection(const QString &name)

{

    QDBusConnectionPrivate *d = 0;

    d = connectionHash.take(name);

    if (d && !d->ref.deref())

        d->deleteYourself();

    // Static objects may be keeping the connection open.

    // However, it is harmless to have outstanding references to a connection that is

    // closing as long as those references will be soon dropped without being used.

    // ### Output a warning if connections are being used after they have been removed.

}

QDBusConnectionManager::~QDBusConnectionManager()

{

    for (QHash<QString, QDBusConnectionPrivate *>::const_iterator it = connectionHash.constBegin();

         it != connectionHash.constEnd(); ++it) {

        QDBusConnectionPrivate *d = it.value();

        if (!d->ref.deref())

            d->deleteYourself();

        else

            d->closeConnection();

    }

    connectionHash.clear();

}

QDBUS_EXPORT void qDBusBindToApplication();

void qDBusBindToApplication()

{

}

void QDBusConnectionManager::setConnection(const QString &name, QDBusConnectionPrivate *c)

{

    connectionHash[name] = c;

    c->name = name;

}

/*!

    \fn QDBusConnection &QDBusConnection::sessionBus()

    \relates QDBusConnection

    Returns a QDBusConnection object opened with the session bus. The object reference returned

    by this function is valid until the QCoreApplication's destructor is run, when the

    connection will be closed and the object, deleted.

*/

/*!

    \fn QDBusConnection &QDBusConnection::systemBus()

    \relates QDBusConnection

    Returns a QDBusConnection object opened with the system bus. The object reference returned

    by this function is valid until the QCoreApplication's destructor is run, when the

    connection will be closed and the object, deleted.

*/

/*!

    \class QDBusConnection

    \inmodule QtDBus

    \since 4.2

    \brief The QDBusConnection class represents a connection to the D-Bus bus daemon.

    This class is the initial point in a D-Bus session. Using it, you

    can get access to remote objects, interfaces; connect remote

    signals to your object's slots; register objects, etc.

    D-Bus connections are created using the connectToBus() function,

    which opens a connection to the server daemon and does the initial

    handshaking, associating that connection with a name. Further

    attempts to connect using the same name will return the same

    connection.

    The connection is then torn down using the disconnectFromBus()

    function.

    As a convenience for the two most common connection types, the

    sessionBus() and systemBus() functions return open connections to

    the session server daemon and the system server daemon,

    respectively. Those connections are opened when first used and are

    closed when the QCoreApplication destructor is run.

    D-Bus also supports peer-to-peer connections, without the need for

    a bus server daemon. Using this facility, two applications can

    talk to each other and exchange messages. This can be achieved by

    passing an address to connectToBus() function, which was opened by

    another D-Bus application using QDBusServer.

*/

/*!

    \enum QDBusConnection::BusType

    Specifies the type of the bus connection. The valid bus types are:

    \value SessionBus           the session bus, associated with the running desktop session

    \value SystemBus            the system bus, used to communicate with system-wide processes

    \value ActivationBus        the activation bus, the "alias" for the bus that started the

                                service

    On the Session Bus, one can find other applications by the same user that are sharing the same

    desktop session (hence the name). On the System Bus, however, processes shared for the whole

    system are usually found.

*/

/*!

    \enum QDBusConnection::RegisterOption

    Specifies the options for registering objects with the connection. The possible values are:

    \value ExportAdaptors                       export the contents of adaptors found in this object

    \value ExportScriptableSlots                export this object's scriptable slots

    \value ExportScriptableSignals              export this object's scriptable signals

    \value ExportScriptableProperties           export this object's scriptable properties

    \value ExportScriptableContents             shorthand form for ExportScriptableSlots |

                                                ExportScriptableSignals |

                                                ExportScriptableProperties

    \value ExportNonScriptableSlots             export this object's non-scriptable slots

    \value ExportNonScriptableSignals           export this object's non-scriptable signals

    \value ExportNonScriptableProperties        export this object's non-scriptable properties

    \value ExportNonScriptableContents          shorthand form for ExportNonScriptableSlots |

                                                ExportNonScriptableSignals |

                                                ExportNonScriptableProperties

    \value ExportAllSlots                       export all of this object's slots

    \value ExportAllSignals                     export all of this object's signals

    \value ExportAllProperties                  export all of this object's properties

    \value ExportAllContents                    export all of this object's contents

    \value ExportChildObjects                   export this object's child objects

    \sa registerObject(), QDBusAbstractAdaptor, {usingadaptors.html}{Using adaptors}

*/

/*!

    \enum QDBusConnection::UnregisterMode

    The mode for unregistering an object path:

    \value UnregisterNode       unregister this node only: do not unregister child objects

    \value UnregisterTree       unregister this node and all its sub-tree

    Note, however, if this object was registered with the ExportChildObjects option, UnregisterNode

    will unregister the child objects too.

*/

/*!

    Creates a QDBusConnection object attached to the connection with name \a name.

    This does not open the connection. You have to call connectToBus() to open it.

*/

QDBusConnection::QDBusConnection(const QString &name)

{

    if (name.isEmpty()) {

        d = 0;

    } else {

        QMutexLocker locker(&_q_manager()->mutex);

        d = _q_manager()->connection(name);

        if (d)

            d->ref.ref();

    }

}

/*!

    Creates a copy of the \a other connection.

*/

QDBusConnection::QDBusConnection(const QDBusConnection &other)

{

    d = other.d;

    if (d)

        d->ref.ref();

}

/*!

  \internal

   Creates a connection object with the given \a dd as private object.

*/

QDBusConnection::QDBusConnection(QDBusConnectionPrivate *dd)

{

    d = dd;

    if (d)

        d->ref.ref();

}

/*!

    Disposes of this object. This does not close the connection: you

    have to call disconnectFromBus() to do that.

*/

QDBusConnection::~QDBusConnection()

{

    if (d && !d->ref.deref())

        d->deleteYourself();

}

/*!

    Creates a copy of the connection \a other in this object. Note

    that the connection this object referenced before the copy, is not

    spontaneously disconnected.

    \sa disconnectFromBus()

*/

QDBusConnection &QDBusConnection::operator=(const QDBusConnection &other)

{

    if (other.d)

        other.d->ref.ref();

    if (d && !d->ref.deref())

        d->deleteYourself();

    d = other.d;

    return *this;

}

/*!

    Opens a connection of type \a type to one of the known busses and

    associate with it the connection name \a name. Returns a

    QDBusConnection object associated with that connection.

*/

QDBusConnection QDBusConnection::connectToBus(BusType type, const QString &name)

{

//    Q_ASSERT_X(QCoreApplication::instance(), "QDBusConnection::addConnection",

//               "Cannot create connection without a Q[Core]Application instance");

    if (!qdbus_loadLibDBus()) {

        QDBusConnectionPrivate *d = 0;

        return QDBusConnection(d);

    }

    QMutexLocker locker(&_q_manager()->mutex);

    QDBusConnectionPrivate *d = _q_manager()->connection(name);

    if (d || name.isEmpty())

        return QDBusConnection(d);

    d = new QDBusConnectionPrivate;

    DBusConnection *c = 0;

    QDBusErrorInternal error;

    switch (type) {

        case SystemBus:

            c = q_dbus_bus_get_private(DBUS_BUS_SYSTEM, error);

            break;

        case SessionBus:

            c = q_dbus_bus_get_private(DBUS_BUS_SESSION, error);

            break;

        case ActivationBus:

            c = q_dbus_bus_get_private(DBUS_BUS_STARTER, error);

            break;

    }

    d->setConnection(c, error); //setConnection does the error handling for us

    _q_manager()->setConnection(name, d);

    QDBusConnection retval(d);

    // create the bus service

    // will lock in QDBusConnectionPrivate::connectRelay()

    d->setBusService(retval);

    return retval;

}

/*!

    Opens a peer-to-peer connection on address \a address and associate with it the

    connection name \a name. Returns a QDBusConnection object associated with that connection.

*/

QDBusConnection QDBusConnection::connectToBus(const QString &address,

                                              const QString &name)

{

//    Q_ASSERT_X(QCoreApplication::instance(), "QDBusConnection::addConnection",

//               "Cannot create connection without a Q[Core]Application instance");

    if (!qdbus_loadLibDBus()){

        QDBusConnectionPrivate *d = 0;

        return QDBusConnection(d);

    }

    QMutexLocker locker(&_q_manager()->mutex);

    QDBusConnectionPrivate *d = _q_manager()->connection(name);

    if (d || name.isEmpty())

        return QDBusConnection(d);

    d = new QDBusConnectionPrivate;

    // setConnection does the error handling for us

    QDBusErrorInternal error;

    DBusConnection *c = q_dbus_connection_open_private(address.toUtf8().constData(), error);

    if (c) {

        if (!q_dbus_bus_register(c, error)) {

            q_dbus_connection_unref(c);

            c = 0;

        }

    }

    d->setConnection(c, error);

    _q_manager()->setConnection(name, d);

    QDBusConnection retval(d);

    // create the bus service

    // will lock in QDBusConnectionPrivate::connectRelay()

    d->setBusService(retval);

    return retval;

}

/*!

    Closes the connection of name \a name.

    Note that if there are still QDBusConnection objects associated

    with the same connection, the connection will not be closed until

    all references are dropped. However, no further references can be

    created using the QDBusConnection constructor.

*/

void QDBusConnection::disconnectFromBus(const QString &name)

{

    if (_q_manager()) {

        QMutexLocker locker(&_q_manager()->mutex);

        _q_manager()->removeConnection(name);

    }

}

/*!

    Sends the \a message over this connection, without waiting for a

    reply. This is suitable for errors, signals, and return values as

    well as calls whose return values are not necessary.

    Returns true if the message was queued successfully, false otherwise.

*/

bool QDBusConnection::send(const QDBusMessage &message) const

{

    if (!d || !d->connection) {

        QDBusError err = QDBusError(QDBusError::Disconnected,

                                    QLatin1String("Not connected to D-BUS server"));

        if (d)

            d->lastError = err;

        return false;

    }

    return d->send(message) != 0;

}

/*!

    Sends the \a message over this connection and returns immediately.

    When the reply is received, the method \a returnMethod is called in

    the \a receiver object. If an error occurs, the method \a errorMethod

    will be called instead.

    If no reply is received within \a timeout milliseconds, an automatic

    error will be delivered indicating the expiration of the call.

    The default \a timeout is -1, which will be replaced with an

    implementation-defined value that is suitable for inter-process

    communications (generally, 25 seconds).

    This function is suitable for method calls only. It is guaranteed

    that the slot will be called exactly once with the reply, as long

    as the parameter types match and no error occurs.

    Returns true if the message was sent, or false if the message could

    not be sent.

*/

bool QDBusConnection::callWithCallback(const QDBusMessage &message, QObject *receiver,

                                       const char *returnMethod, const char *errorMethod,

                                       int timeout) const

{

    if (!d || !d->connection) {

        QDBusError err = QDBusError(QDBusError::Disconnected,

                                    QLatin1String("Not connected to D-BUS server"));

        if (d)

            d->lastError = err;

        return false;

    }

    return d->sendWithReplyAsync(message, receiver, returnMethod, errorMethod, timeout) != 0;

}

/*!

    \overload

    \deprecated

    Sends the \a message over this connection and returns immediately.

    When the reply is received, the method \a returnMethod is called in

    the \a receiver object.

    This function is suitable for method calls only. It is guaranteed

    that the slot will be called exactly once with the reply, as long

    as the parameter types match and no error occurs.

    This function is dangerous because it cannot report errors, including

    the expiration of the timeout.

    Returns true if the message was sent, or false if the message could

    not be sent.

*/

bool QDBusConnection::callWithCallback(const QDBusMessage &message, QObject *receiver,

                                       const char *returnMethod, int timeout) const

{

    return callWithCallback(message, receiver, returnMethod, 0, timeout);

}

/*!

    Sends the \a message over this connection and blocks, waiting for

    a reply, for at most \a timeout milliseconds. This function is

    suitable for method calls only. It returns the reply message as

    its return value, which will be either of type

    QDBusMessage::ReplyMessage or QDBusMessage::ErrorMessage.

    See the QDBusInterface::call() function for a more friendly way

    of placing calls.

    \warning If \a mode is QDBus::BlockWithGui, this function will

             reenter the Qt event loop in order to wait for the

             reply. During the wait, it may deliver signals and other

             method calls to your application. Therefore, it must be

             prepared to handle a reentrancy whenever a call is

             placed with call().

*/

QDBusMessage QDBusConnection::call(const QDBusMessage &message, QDBus::CallMode mode, int timeout) const

{

    if (!d || !d->connection) {

        QDBusError err = QDBusError(QDBusError::Disconnected,

                                    QLatin1String("Not connected to D-Bus server"));

        if (d)

            d->lastError = err;

        return QDBusMessage::createError(err);

    }

    if (mode != QDBus::NoBlock)

        return d->sendWithReply(message, mode, timeout);

    d->send(message);

    QDBusMessage retval;

    retval << QVariant(); // add one argument (to avoid .at(0) problems)

    return retval;

}