src/common/protocols/datastream/datastreampeer.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-2016 by the Quassel Project                        *
   3  *   devel@quassel-irc.org                                                 *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) version 3.                                           *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  19  ***************************************************************************/
  20
  21 #include <QtEndian>
  22 #include <QDataStream>
  23 #include <QHostAddress>
  24 #include <QTcpSocket>
  25
  26 #include "datastreampeer.h"
  27
  28 using namespace Protocol;
  29
  30 DataStreamPeer::DataStreamPeer(::AuthHandler *authHandler, QTcpSocket *socket, quint16 features, Compressor::CompressionLevel level, QObject *parent)
  31     : RemotePeer(authHandler, socket, level, parent)
  32 {
  33     Q_UNUSED(features);
  34 }
  35
  36
  37 quint16 DataStreamPeer::supportedFeatures()
  38 {
  39     return 0;
  40 }
  41
  42
  43 bool DataStreamPeer::acceptsFeatures(quint16 peerFeatures)
  44 {
  45     Q_UNUSED(peerFeatures);
  46     return true;
  47 }
  48
  49
  50 quint16 DataStreamPeer::enabledFeatures() const
  51 {
  52     return 0;
  53 }
  54
  55
  56 void DataStreamPeer::processMessage(const QByteArray &msg)
  57 {
  58     QDataStream stream(msg);
  59     stream.setVersion(QDataStream::Qt_4_2);
  60     QVariantList list;
  61     stream >> list;
  62     if (stream.status() != QDataStream::Ok) {
  63         close("Peer sent corrupt data, closing down!");
  64         return;
  65     }
  66
  67     // if no sigproxy is set, we're in handshake mode
  68     if (!signalProxy())
  69         handleHandshakeMessage(list);
  70     else
  71         handlePackedFunc(list);
  72 }
  73
  74
  75 void DataStreamPeer::writeMessage(const QVariantMap &handshakeMsg)
  76 {
  77     QVariantList list;
  78     QVariantMap::const_iterator it = handshakeMsg.begin();
  79     while (it != handshakeMsg.end()) {
  80         list << it.key().toUtf8() << it.value();
  81         ++it;
  82     }
  83
  84     writeMessage(list);
  85 }
  86
  87
  88 void DataStreamPeer::writeMessage(const QVariantList &sigProxyMsg)
  89 {
  90     QByteArray data;
  91     QDataStream msgStream(&data, QIODevice::WriteOnly);
  92     msgStream.setVersion(QDataStream::Qt_4_2);
  93     msgStream << sigProxyMsg;
  94
  95     writeMessage(data);
  96 }
  97
  98
  99 /*** Handshake messages ***/
 100
 101 /* These messages are transmitted during handshake phase, which in case of the legacy protocol means they have
 102  * a structure different from those being used after the handshake.
 103  * Also, the legacy handshake does not fully match the redesigned one, so we'll have to do various mappings here.
 104  */
 105
 106 void DataStreamPeer::handleHandshakeMessage(const QVariantList &mapData)
 107 {
 108     QVariantMap m;
 109     for (int i = 0; i < mapData.count()/2; ++i)
 110         m[QString::fromUtf8(mapData[2*i].toByteArray())] = mapData[2*i+1];
 111
 112     QString msgType = m["MsgType"].toString();
 113     if (msgType.isEmpty()) {
 114         emit protocolError(tr("Invalid handshake message!"));
 115         return;
 116     }
 117
 118     if (msgType == "ClientInit") {
 119         handle(RegisterClient(m["ClientVersion"].toString(), m["ClientDate"].toString(), false, m["Features"].toInt())); // UseSsl obsolete
 120     }
 121
 122     else if (msgType == "ClientInitReject") {
 123         handle(ClientDenied(m["Error"].toString()));
 124     }
 125
 126     else if (msgType == "ClientInitAck") {
 127         handle(ClientRegistered(m["CoreFeatures"].toUInt(), m["Configured"].toBool(), m["StorageBackends"].toList(), false, m["Authenticators"].toList())); // SupportsSsl obsolete
 128     }
 129
 130     else if (msgType == "CoreSetupData") {
 131         QVariantMap map = m["SetupData"].toMap();
 132         handle(SetupData(map["AdminUser"].toString(), map["AdminPasswd"].toString(), map["Backend"].toString(), map["ConnectionProperties"].toMap(), map["Authenticator"].toString(), map["AuthProperties"].toMap()));
 133     }
 134
 135     else if (msgType == "CoreSetupReject") {
 136         handle(SetupFailed(m["Error"].toString()));
 137     }
 138
 139     else if (msgType == "CoreSetupAck") {
 140         handle(SetupDone());
 141     }
 142
 143     else if (msgType == "ClientLogin") {
 144         handle(Login(m["User"].toString(), m["Password"].toString()));
 145     }
 146
 147     else if (msgType == "ClientLoginReject") {
 148         handle(LoginFailed(m["Error"].toString()));
 149     }
 150
 151     else if (msgType == "ClientLoginAck") {
 152         handle(LoginSuccess());
 153     }
 154
 155     else if (msgType == "SessionInit") {
 156         QVariantMap map = m["SessionState"].toMap();
 157         handle(SessionState(map["Identities"].toList(), map["BufferInfos"].toList(), map["NetworkIds"].toList()));
 158     }
 159
 160     else {
 161         emit protocolError(tr("Unknown protocol message of type %1").arg(msgType));
 162     }
 163 }
 164
 165
 166 void DataStreamPeer::dispatch(const RegisterClient &msg) {
 167     QVariantMap m;
 168     m["MsgType"] = "ClientInit";
 169     m["ClientVersion"] = msg.clientVersion;
 170     m["ClientDate"] = msg.buildDate;
 171     m["Features"] = msg.clientFeatures;
 172
 173     writeMessage(m);
 174 }
 175
 176
 177 void DataStreamPeer::dispatch(const ClientDenied &msg) {
 178     QVariantMap m;
 179     m["MsgType"] = "ClientInitReject";
 180     m["Error"] = msg.errorString;
 181
 182     writeMessage(m);
 183 }
 184
 185
 186 void DataStreamPeer::dispatch(const ClientRegistered &msg) {
 187     QVariantMap m;
 188     m["MsgType"] = "ClientInitAck";
 189     m["CoreFeatures"] = msg.coreFeatures;
 190     m["StorageBackends"] = msg.backendInfo;
 191     m["Authenticators"] = msg.authenticatorInfo;
 192     m["LoginEnabled"] = m["Configured"] = msg.coreConfigured;
 193
 194     writeMessage(m);
 195 }
 196
 197
 198 void DataStreamPeer::dispatch(const SetupData &msg)
 199 {
 200     QVariantMap map;
 201     map["AdminUser"] = msg.adminUser;
 202     map["AdminPasswd"] = msg.adminPassword;
 203     map["Backend"] = msg.backend;
 204     map["ConnectionProperties"] = msg.setupData;
 205
 206     // Auth backend properties.
 207     map["Authenticator"] = msg.authenticator;
 208     map["AuthProperties"] = msg.authSetupData;
 209
 210     QVariantMap m;
 211     m["MsgType"] = "CoreSetupData";
 212     m["SetupData"] = map;
 213
 214     writeMessage(m);
 215 }
 216
 217
 218 void DataStreamPeer::dispatch(const SetupFailed &msg)
 219 {
 220     QVariantMap m;
 221     m["MsgType"] = "CoreSetupReject";
 222     m["Error"] = msg.errorString;
 223
 224     writeMessage(m);
 225 }
 226
 227
 228 void DataStreamPeer::dispatch(const SetupDone &msg)
 229 {
 230     Q_UNUSED(msg)
 231
 232     QVariantMap m;
 233     m["MsgType"] = "CoreSetupAck";
 234
 235     writeMessage(m);
 236 }
 237
 238
 239 void DataStreamPeer::dispatch(const Login &msg)
 240 {
 241     QVariantMap m;
 242     m["MsgType"] = "ClientLogin";
 243     m["User"] = msg.user;
 244     m["Password"] = msg.password;
 245
 246     writeMessage(m);
 247 }
 248
 249
 250 void DataStreamPeer::dispatch(const LoginFailed &msg)
 251 {
 252     QVariantMap m;
 253     m["MsgType"] = "ClientLoginReject";
 254     m["Error"] = msg.errorString;
 255
 256     writeMessage(m);
 257 }
 258
 259
 260 void DataStreamPeer::dispatch(const LoginSuccess &msg)
 261 {
 262     Q_UNUSED(msg)
 263
 264     QVariantMap m;
 265     m["MsgType"] = "ClientLoginAck";
 266
 267     writeMessage(m);
 268 }
 269
 270
 271 void DataStreamPeer::dispatch(const SessionState &msg)
 272 {
 273     QVariantMap m;
 274     m["MsgType"] = "SessionInit";
 275
 276     QVariantMap map;
 277     map["BufferInfos"] = msg.bufferInfos;
 278     map["NetworkIds"] = msg.networkIds;
 279     map["Identities"] = msg.identities;
 280     m["SessionState"] = map;
 281
 282     writeMessage(m);
 283 }
 284
 285
 286 /*** Standard messages ***/
 287
 288 void DataStreamPeer::handlePackedFunc(const QVariantList &packedFunc)
 289 {
 290     QVariantList params(packedFunc);
 291
 292     if (params.isEmpty()) {
 293         qWarning() << Q_FUNC_INFO << "Received incompatible data:" << packedFunc;
 294         return;
 295     }
 296
 297     // TODO: make sure that this is a valid request type
 298     RequestType requestType = (RequestType)params.takeFirst().value<qint16>();
 299     switch (requestType) {
 300         case Sync: {
 301             if (params.count() < 3) {
 302                 qWarning() << Q_FUNC_INFO << "Received invalid sync call:" << params;
 303                 return;
 304             }
 305             QByteArray className = params.takeFirst().toByteArray();
 306             QString objectName = QString::fromUtf8(params.takeFirst().toByteArray());
 307             QByteArray slotName = params.takeFirst().toByteArray();
 308             handle(Protocol::SyncMessage(className, objectName, slotName, params));
 309             break;
 310         }
 311         case RpcCall: {
 312             if (params.empty()) {
 313                 qWarning() << Q_FUNC_INFO << "Received empty RPC call!";
 314                 return;
 315             }
 316             QByteArray slotName = params.takeFirst().toByteArray();
 317             handle(Protocol::RpcCall(slotName, params));
 318             break;
 319         }
 320         case InitRequest: {
 321             if (params.count() != 2) {
 322                 qWarning() << Q_FUNC_INFO << "Received invalid InitRequest:" << params;
 323                 return;
 324             }
 325             QByteArray className = params[0].toByteArray();
 326             QString objectName = QString::fromUtf8(params[1].toByteArray());
 327             handle(Protocol::InitRequest(className, objectName));
 328             break;
 329         }
 330         case InitData: {
 331             if (params.count() < 2) {
 332                 qWarning() << Q_FUNC_INFO << "Received invalid InitData:" << params;
 333                 return;
 334             }
 335             QByteArray className = params.takeFirst().toByteArray();
 336             QString objectName = QString::fromUtf8(params.takeFirst().toByteArray());
 337             QVariantMap initData;
 338             for (int i = 0; i < params.count()/2; ++i)
 339                 initData[QString::fromUtf8(params[2*i].toByteArray())] = params[2*i+1];
 340             handle(Protocol::InitData(className, objectName, initData));
 341             break;
 342         }
 343         case HeartBeat: {
 344             if (params.count() != 1) {
 345                 qWarning() << Q_FUNC_INFO << "Received invalid HeartBeat:" << params;
 346                 return;
 347             }
 348             // Note: QDateTime instead of QTime as in the legacy protocol!
 349             handle(Protocol::HeartBeat(params[0].toDateTime()));
 350             break;
 351         }
 352         case HeartBeatReply: {
 353             if (params.count() != 1) {
 354                 qWarning() << Q_FUNC_INFO << "Received invalid HeartBeat:" << params;
 355                 return;
 356             }
 357             // Note: QDateTime instead of QTime as in the legacy protocol!
 358             handle(Protocol::HeartBeatReply(params[0].toDateTime()));
 359             break;
 360         }
 361
 362     }
 363 }
 364
 365
 366 void DataStreamPeer::dispatch(const Protocol::SyncMessage &msg)
 367 {
 368     dispatchPackedFunc(QVariantList() << (qint16)Sync << msg.className << msg.objectName.toUtf8() << msg.slotName << msg.params);
 369 }
 370
 371
 372 void DataStreamPeer::dispatch(const Protocol::RpcCall &msg)
 373 {
 374     dispatchPackedFunc(QVariantList() << (qint16)RpcCall << msg.slotName << msg.params);
 375 }
 376
 377
 378 void DataStreamPeer::dispatch(const Protocol::InitRequest &msg)
 379 {
 380     dispatchPackedFunc(QVariantList() << (qint16)InitRequest << msg.className << msg.objectName.toUtf8());
 381 }
 382
 383
 384 void DataStreamPeer::dispatch(const Protocol::InitData &msg)
 385 {
 386     QVariantList initData;
 387     QVariantMap::const_iterator it = msg.initData.begin();
 388     while (it != msg.initData.end()) {
 389         initData << it.key().toUtf8() << it.value();
 390         ++it;
 391     }
 392     dispatchPackedFunc(QVariantList() << (qint16)InitData << msg.className << msg.objectName.toUtf8() << initData);
 393 }
 394
 395
 396 void DataStreamPeer::dispatch(const Protocol::HeartBeat &msg)
 397 {
 398     dispatchPackedFunc(QVariantList() << (qint16)HeartBeat << msg.timestamp);
 399 }
 400
 401
 402 void DataStreamPeer::dispatch(const Protocol::HeartBeatReply &msg)
 403 {
 404     dispatchPackedFunc(QVariantList() << (qint16)HeartBeatReply << msg.timestamp);
 405 }
 406
 407
 408 void DataStreamPeer::dispatchPackedFunc(const QVariantList &packedFunc)
 409 {
 410     writeMessage(packedFunc);
 411 }