Some fixes to connection manager.
[jenirok] / src / common / connectionmanager.cpp
index 14c0b0a..f7b0f32 100644 (file)
  *
  */
 
-#include <QDebug>
-#include <QApplication>
+#include <QtCore/QDebug>
 #include <QtCore/QTimerEvent>
 #include <QtCore/QVariant>
+#include <QtCore/QStringList>
+#include <QtGui/QApplication>
+#include <QtDBus/QDBusArgument>
+#include <QtDBus/QDBusConnection>
 #include <icd/dbus_api.h>
 #include "connectionmanager.h"
 
+bool ConnectionManager::connected_ = false;
 
 ConnectionManager::ConnectionManager(QObject* parent): QObject(parent),
-ready_(false), connected_(false), timeout_(false), ignoreStateChanges_(false),
-timer_(0)
+stateReady_(false), connectionReady_(false), scanReady_(false),
+timeout_(false), numberOfConnections_(0),
+scannedConnections_(0), timer_(0), searchType_(NO_TYPE), error_(NO_ERROR), connections_(0)
 {
     QDBusConnection systemBus = QDBusConnection::systemBus();
 
@@ -42,29 +47,183 @@ timer_(0)
                       ICD_DBUS_API_INTERFACE, ICD_DBUS_API_CONNECT_SIG,
                       this, SLOT(connectionChange(const QDBusMessage&)));
 
+    systemBus.connect(ICD_DBUS_API_INTERFACE, ICD_DBUS_API_PATH,
+                      ICD_DBUS_API_INTERFACE, ICD_DBUS_API_SCAN_SIG,
+                      this, SLOT(scanResult(const QDBusMessage&)));
 
 }
 
+ConnectionManager::~ConnectionManager()
+{
+    QDBusConnection systemBus = QDBusConnection::systemBus();
+
+    systemBus.disconnect(ICD_DBUS_API_INTERFACE, ICD_DBUS_API_PATH,
+                         ICD_DBUS_API_INTERFACE, ICD_DBUS_API_STATE_SIG,
+                         this, SLOT(stateChange(const QDBusMessage&)));
+
+    systemBus.disconnect(ICD_DBUS_API_INTERFACE, ICD_DBUS_API_PATH,
+                         ICD_DBUS_API_INTERFACE, ICD_DBUS_API_CONNECT_SIG,
+                         this, SLOT(connectionChange(const QDBusMessage&)));
+
+    systemBus.disconnect(ICD_DBUS_API_INTERFACE, ICD_DBUS_API_PATH,
+                         ICD_DBUS_API_INTERFACE, ICD_DBUS_API_SCAN_SIG,
+                         this, SLOT(scanResult(const QDBusMessage&)));
+}
+
 bool ConnectionManager::connect()
 {
-    ready_ = false;
-    ignoreStateChanges_ = true;
+    connectionReady_ = false;
     unsigned int flags = static_cast<unsigned int>(ICD_CONNECTION_FLAG_USER_EVENT);
     icd2interface_->call(ICD_DBUS_API_CONNECT_REQ, QVariant(flags));
-    waitSignal();
+
+    waitSignal(&connectionReady_);
+
+    if(connected_)
+    {
+        sleep(WAIT_AFTER_CONNECT);
+    }
+
     return connected_;
 }
 
-bool ConnectionManager::disconnect()
+bool ConnectionManager::connect(ConnectionManager::Connection const& connection)
 {
-    if(!connected_)
+    return connect(connection.id);
+}
+
+bool ConnectionManager::connect(QString const& id)
+{
+    QDBusMessage msg = QDBusMessage::createMethodCall("com.nokia.icd",
+                                                      "/com/nokia/icd",
+                                                      "com.nokia.icd",
+                                                      "connect");
+    QList<QVariant> arguments;
+
+    arguments.append(QVariant(id));
+
+    unsigned int val = 0;
+
+    arguments.append(QVariant(val));
+
+    msg.setArguments(arguments);
+
+    QDBusMessage rep = QDBusConnection::systemBus().call(msg);
+
+    if(rep.type() == QDBusMessage::ErrorMessage)
     {
+        if(rep.errorName() == "com.nokia.icd.error.invalid_iap")
+        {
+            error_ = INVALID_IAP;
+        }
+        else
+        {
+            error_ = UNKNOWN_ERROR;
+        }
+
+        connected_ = false;
+
         return false;
     }
 
-    ready_ = false;
-    ignoreStateChanges_ = false;
-    unsigned int flags = static_cast<unsigned int>(ICD_CONNECTION_FLAG_USER_EVENT);
+    connected_ = true;
+
+    sleep(WAIT_AFTER_CONNECT);
+
+    return true;
+}
+
+bool ConnectionManager::getBestConnection(Connection& connection, ConnectionType type)
+{
+    QList<Connection> connections;
+
+    if(!scanConnections(connections))
+    {
+        qDebug() << "Unable to scan connections";
+        return false;
+    }
+
+    if(connections.size() == 0)
+    {
+        error_ = NO_AVAILABLE_CONNECTIONS;
+        return false;
+    }
+
+    int biggestWlan = -1;
+    int biggestGprs = -1;
+    int bestWlan = -1;
+    int bestGprs = -1;
+
+    for(int i = 0; i < connections.size(); i++)
+    {
+        switch(connections.at(i).type)
+        {
+        case WLAN:
+            if(type != GPRS && connections.at(i).strength > biggestWlan)
+            {
+                biggestWlan = connections.at(i).strength;
+                bestWlan = i;
+            }
+            break;
+
+        case GPRS:
+            if(type != WLAN && connections.at(i).strength > biggestGprs)
+            {
+                biggestGprs = connections.at(i).strength;
+                bestGprs = i;
+            }
+            break;
+
+        default:
+            qDebug() << "Unknown connection type";
+        }
+    }
+
+    if(bestWlan >= 0)
+    {
+        connection = connections.at(bestWlan);
+        return true;
+    }
+    else if(bestGprs >= 0)
+    {
+        connection = connections.at(bestGprs);
+        return true;
+    }
+    else
+    {
+        error_ = NO_AVAILABLE_CONNECTIONS;
+        return false;
+    }
+
+}
+
+bool ConnectionManager::disconnect(bool force)
+{
+    if(force)
+    {
+        QDBusMessage msg = QDBusMessage::createSignal("/com/nokia/icd_ui",
+                                                      "com.nokia.icd_ui",
+                                                      "disconnect");
+
+        QList<QVariant> arguments;
+        bool val = true;
+        arguments.append(QVariant(val));
+        msg.setArguments(arguments);
+
+        bool ret = QDBusConnection::systemBus().send(msg);
+
+        if(ret)
+        {
+            connected_ = false;
+        }
+
+        return ret;
+    }
+
+    connectionReady_ = false;
+    unsigned int flags;
+
+    flags = static_cast<unsigned int>(ICD_CONNECTION_FLAG_USER_EVENT);
+
     icd2interface_->call(ICD_DBUS_API_DISCONNECT_REQ, QVariant(flags));
     connected_ = false;
     return true;
@@ -72,8 +231,7 @@ bool ConnectionManager::disconnect()
 
 bool ConnectionManager::isConnected()
 {
-    ready_ = false;
-    ignoreStateChanges_ = false;
+    stateReady_ = false;
     QDBusMessage rep = icd2interface_->call(ICD_DBUS_API_STATE_REQ);
 
     unsigned int numOfReplies = rep.arguments().value(0).value<unsigned int>();
@@ -83,55 +241,73 @@ bool ConnectionManager::isConnected()
         return false;
     }
 
-    waitSignal();
+    waitSignal(&stateReady_);
     return connected_;
 }
 
-void ConnectionManager::stateChange(const QDBusMessage& rep)
+bool ConnectionManager::scanConnections(QList<ConnectionManager::Connection>& connections,
+                                        ConnectionManager::ConnectionType type)
 {
-    if(ignoreStateChanges_)
+    unsigned int flags = static_cast<unsigned int>(ICD_SCAN_REQUEST_ACTIVE_SAVED);
+    scanReady_ = false;
+    scannedConnections_ = 0;
+    connections_ = &connections;
+    searchType_ = type;
+
+    QDBusMessage rep = icd2interface_->call(ICD_DBUS_API_SCAN_REQ,
+                                            QVariant(flags));
+
+    numberOfConnections_ = rep.arguments().value(0).toList().size();
+
+    // For some reason, during call icd2 doesn't return any connections
+    // it is going to scan. However, it still scans them so use default value
+    // of 2.
+    if(numberOfConnections_ == 0)
     {
-        return;
+        numberOfConnections_ = 2;
     }
 
+    bool ret = waitSignal(&scanReady_);
+    connections_ = 0;
+    return ret;
+}
+
+ConnectionManager::Error ConnectionManager::error() const
+{
+    return error_;
+}
+
+void ConnectionManager::stateChange(const QDBusMessage& rep)
+{
     unsigned int status = rep.arguments().value(7).value<unsigned int>();
 
     switch(status)
     {
     case ICD_STATE_CONNECTING:
-        qDebug() << "Connecting";
         break;
     case ICD_STATE_CONNECTED:
         connected_ = true;
-        ready_ = true;
-        qDebug() << "Connected";
+        stateReady_ = true;
         break;
     case ICD_STATE_DISCONNECTING:
-        qDebug() << "Disconnecting";
         break;
     case ICD_STATE_DISCONNECTED:
         connected_ = false;
-        ready_ = true;
-        qDebug() << "Disconnected";
+        stateReady_ = true;
         break;
     case ICD_STATE_LIMITED_CONN_ENABLED:
         connected_ = true;
-        ready_ = true;
-        qDebug() << "Limited connection enabled";
+        stateReady_ = true;
         break;
     case ICD_STATE_LIMITED_CONN_DISABLED:
         connected_ = false;
-        ready_ = true;
-        qDebug() << "Limited connection disabled";
+        stateReady_ = true;
         break;
     case ICD_STATE_SEARCH_START:
-        qDebug() << "Search start";
         break;
     case ICD_STATE_SEARCH_STOP:
-        qDebug() << "Search stop";
         break;
     case ICD_STATE_INTERNAL_ADDRESS_ACQUIRED:
-        qDebug() << "Internal address acquired";
         break;
     default:
         qDebug() << "Unknown connection status";
@@ -148,38 +324,130 @@ void ConnectionManager::connectionChange(const QDBusMessage& rep)
     {
     case ICD_CONNECTION_SUCCESSFUL:
         connected_ = true;
-        ready_ = true;
-        qDebug() << "Connection successful";
+        connectionReady_ = true;
         break;
     case ICD_CONNECTION_NOT_CONNECTED:
         connected_ = false;
-        ready_ = true;
-        qDebug() << "Connection not connected";
+        connectionReady_ = true;
         break;
     case ICD_CONNECTION_DISCONNECTED:
         connected_ = false;
-        ready_ = true;
-        qDebug() << "Connection disconnected";
+        connectionReady_ = true;
         break;
     default:
         qDebug() << "Unknown connection status";
         break;
     }
+
 }
 
-bool ConnectionManager::waitSignal()
+void ConnectionManager::scanResult(const QDBusMessage& rep)
+{
+    if(!connections_)
+    {
+        return;
+    }
+
+    QList<QVariant> args = rep.arguments();
+
+    unsigned int status = args.value(0).value<unsigned int>();
+
+    if(status == ICD_SCAN_COMPLETE)
+    {
+        scannedConnections_++;
+    }
+
+    if(scannedConnections_ >= numberOfConnections_)
+    {
+        scanReady_ = true;
+        connections_ = 0;
+        return;
+    }
+
+    if(status != ICD_SCAN_NEW && status != ICD_SCAN_NOTIFY)
+    {
+        return;
+    }
+
+    Connection connection;
+
+    QString type = args.value(7).toString();
+
+    if(type == "GPRS")
+    {
+        if(searchType_ == WLAN)
+        {
+            return;
+        }
+
+        connection.type = GPRS;
+    }
+    else if(type == "WLAN_INFRA" || type == "WLAN_ADHOC")
+    {
+        if(searchType_ == GPRS)
+        {
+            return;
+        }
+
+        connection.type = WLAN;
+    }
+    else
+    {
+        qDebug() << "Unknown connection type: " << type;
+        return;
+    }
+
+    connection.id = QString(args.value(10).toByteArray());
+    connection.name = args.value(8).toString();
+    connection.strength = args.value(11).toInt();
+
+    for(int i = 0; i < connections_->size(); i++)
+    {
+        if(connections_->at(i).id == connection.id)
+        {
+            if(status == ICD_SCAN_NEW)
+            {
+                connections_->replace(i, connection);
+            }
+
+            return;
+        }
+    }
+
+    connections_->push_back(connection);
+}
+
+bool ConnectionManager::waitSignal(bool* ready)
 {
     timeout_ = false;
     timer_ = startTimer(TIMEOUT);
 
-    while(!ready_ && !timeout_)
+    while(!*ready && !timeout_)
     {
         QCoreApplication::processEvents(QEventLoop::WaitForMoreEvents);
     }
 
     killTimer(timer_);
 
-    return ready_ || !timeout_;
+    if(timeout_)
+    {
+        qDebug() << "Connection request timed out";
+    }
+
+    return *ready || !timeout_;
+}
+
+void ConnectionManager::sleep(unsigned int ms)
+{
+    timeout_ = false;
+    timer_ = startTimer(ms);
+
+    while(!timeout_)
+    {
+        QCoreApplication::processEvents(QEventLoop::WaitForMoreEvents);
+    }
+
+    killTimer(timer_);
 }
 
 void ConnectionManager::timerEvent(QTimerEvent* event)
@@ -188,6 +456,4 @@ void ConnectionManager::timerEvent(QTimerEvent* event)
     killTimer(timer_);
     timeout_ = true;
     timer_ = 0;
-
-    qDebug() << "Connection request timed out";
 }