src/core/corenetwork.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-09 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  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include "corenetwork.h"
  22
  23 #include "core.h"
  24 #include "coresession.h"
  25 #include "coreidentity.h"
  26
  27 #include "ircserverhandler.h"
  28 #include "userinputhandler.h"
  29 #include "ctcphandler.h"
  30
  31 CoreNetwork::CoreNetwork(const NetworkId &networkid, CoreSession *session)
  32   : Network(networkid, session),
  33     _coreSession(session),
  34     _ircServerHandler(new IrcServerHandler(this)),
  35     _userInputHandler(new UserInputHandler(this)),
  36     _ctcpHandler(new CtcpHandler(this)),
  37     _autoReconnectCount(0),
  38     _quitRequested(false),
  39
  40     _previousConnectionAttemptFailed(false),
  41     _lastUsedServerIndex(0),
  42
  43     _lastPingTime(0),
  44     _maxPingCount(3),
  45     _pingCount(0),
  46
  47     // TODO make autowho configurable (possibly per-network)
  48     _autoWhoEnabled(true),
  49     _autoWhoInterval(90),
  50     _autoWhoNickLimit(0), // unlimited
  51     _autoWhoDelay(5)
  52 {
  53   _autoReconnectTimer.setSingleShot(true);
  54   _socketCloseTimer.setSingleShot(true);
  55   connect(&_socketCloseTimer, SIGNAL(timeout()), this, SLOT(socketCloseTimeout()));
  56
  57   _pingTimer.setInterval(30000);
  58   connect(&_pingTimer, SIGNAL(timeout()), this, SLOT(sendPing()));
  59
  60   _autoWhoTimer.setInterval(_autoWhoDelay * 1000);
  61   _autoWhoCycleTimer.setInterval(_autoWhoInterval * 1000);
  62
  63   QHash<QString, QString> channels = coreSession()->persistentChannels(networkId());
  64   foreach(QString chan, channels.keys()) {
  65     _channelKeys[chan.toLower()] = channels[chan];
  66   }
  67
  68   connect(&_autoReconnectTimer, SIGNAL(timeout()), this, SLOT(doAutoReconnect()));
  69   connect(&_autoWhoTimer, SIGNAL(timeout()), this, SLOT(sendAutoWho()));
  70   connect(&_autoWhoCycleTimer, SIGNAL(timeout()), this, SLOT(startAutoWhoCycle()));
  71   connect(&_tokenBucketTimer, SIGNAL(timeout()), this, SLOT(fillBucketAndProcessQueue()));
  72   connect(this, SIGNAL(connectRequested()), this, SLOT(connectToIrc()));
  73
  74
  75   connect(&socket, SIGNAL(connected()), this, SLOT(socketInitialized()));
  76   connect(&socket, SIGNAL(disconnected()), this, SLOT(socketDisconnected()));
  77   connect(&socket, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(socketError(QAbstractSocket::SocketError)));
  78   connect(&socket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), this, SLOT(socketStateChanged(QAbstractSocket::SocketState)));
  79   connect(&socket, SIGNAL(readyRead()), this, SLOT(socketHasData()));
  80 #ifdef HAVE_SSL
  81   connect(&socket, SIGNAL(encrypted()), this, SLOT(socketInitialized()));
  82   connect(&socket, SIGNAL(sslErrors(const QList<QSslError> &)), this, SLOT(sslErrors(const QList<QSslError> &)));
  83 #endif
  84 }
  85
  86 CoreNetwork::~CoreNetwork() {
  87   if(connectionState() != Disconnected && connectionState() != Network::Reconnecting)
  88     disconnectFromIrc(false);      // clean up, but this does not count as requested disconnect!
  89   disconnect(&socket, 0, this, 0); // this keeps the socket from triggering events during clean up
  90   delete _ircServerHandler;
  91   delete _userInputHandler;
  92   delete _ctcpHandler;
  93 }
  94
  95 QString CoreNetwork::channelDecode(const QString &bufferName, const QByteArray &string) const {
  96   if(!bufferName.isEmpty()) {
  97     IrcChannel *channel = ircChannel(bufferName);
  98     if(channel)
  99       return channel->decodeString(string);
 100   }
 101   return decodeString(string);
 102 }
 103
 104 QString CoreNetwork::userDecode(const QString &userNick, const QByteArray &string) const {
 105   IrcUser *user = ircUser(userNick);
 106   if(user)
 107     return user->decodeString(string);
 108   return decodeString(string);
 109 }
 110
 111 QByteArray CoreNetwork::channelEncode(const QString &bufferName, const QString &string) const {
 112   if(!bufferName.isEmpty()) {
 113     IrcChannel *channel = ircChannel(bufferName);
 114     if(channel)
 115       return channel->encodeString(string);
 116   }
 117   return encodeString(string);
 118 }
 119
 120 QByteArray CoreNetwork::userEncode(const QString &userNick, const QString &string) const {
 121   IrcUser *user = ircUser(userNick);
 122   if(user)
 123     return user->encodeString(string);
 124   return encodeString(string);
 125 }
 126
 127 void CoreNetwork::connectToIrc(bool reconnecting) {
 128   if(!reconnecting && useAutoReconnect() && _autoReconnectCount == 0) {
 129     _autoReconnectTimer.setInterval(autoReconnectInterval() * 1000);
 130     if(unlimitedReconnectRetries())
 131       _autoReconnectCount = -1;
 132     else
 133       _autoReconnectCount = autoReconnectRetries();
 134   }
 135   if(serverList().isEmpty()) {
 136     qWarning() << "Server list empty, ignoring connect request!";
 137     return;
 138   }
 139   CoreIdentity *identity = identityPtr();
 140   if(!identity) {
 141     qWarning() << "Invalid identity configures, ignoring connect request!";
 142     return;
 143   }
 144
 145   // cleaning up old quit reason
 146   _quitReason.clear();
 147
 148   // use a random server?
 149   if(useRandomServer()) {
 150     _lastUsedServerIndex = qrand() % serverList().size();
 151   } else if(_previousConnectionAttemptFailed) {
 152     // cycle to next server if previous connection attempt failed
 153     displayMsg(Message::Server, BufferInfo::StatusBuffer, "", tr("Connection failed. Cycling to next Server"));
 154     if(++_lastUsedServerIndex >= serverList().size()) {
 155       _lastUsedServerIndex = 0;
 156     }
 157   }
 158   _previousConnectionAttemptFailed = false;
 159
 160   Server server = usedServer();
 161   displayStatusMsg(tr("Connecting to %1:%2...").arg(server.host).arg(server.port));
 162   displayMsg(Message::Server, BufferInfo::StatusBuffer, "", tr("Connecting to %1:%2...").arg(server.host).arg(server.port));
 163
 164   if(server.useProxy) {
 165     QNetworkProxy proxy((QNetworkProxy::ProxyType)server.proxyType, server.proxyHost, server.proxyPort, server.proxyUser, server.proxyPass);
 166     socket.setProxy(proxy);
 167   } else {
 168     socket.setProxy(QNetworkProxy::NoProxy);
 169   }
 170
 171 #ifdef HAVE_SSL
 172   socket.setProtocol((QSsl::SslProtocol)server.sslVersion);
 173   if(server.useSsl) {
 174     CoreIdentity *identity = identityPtr();
 175     if(identity) {
 176       socket.setLocalCertificate(identity->sslCert());
 177       socket.setPrivateKey(identity->sslKey());
 178     }
 179     socket.connectToHostEncrypted(server.host, server.port);
 180   } else {
 181     socket.connectToHost(server.host, server.port);
 182   }
 183 #else
 184   socket.connectToHost(server.host, server.port);
 185 #endif
 186 }
 187
 188 void CoreNetwork::disconnectFromIrc(bool requested, const QString &reason, bool withReconnect) {
 189   _quitRequested = requested; // see socketDisconnected();
 190   if(!withReconnect) {
 191     _autoReconnectTimer.stop();
 192     _autoReconnectCount = 0; // prohibiting auto reconnect
 193   }
 194   disablePingTimeout();
 195   _msgQueue.clear();
 196
 197   IrcUser *me_ = me();
 198   if(me_) {
 199     QString awayMsg;
 200     if(me_->isAway())
 201       awayMsg = me_->awayMessage();
 202     Core::setAwayMessage(userId(), networkId(), awayMsg);
 203     Core::setUserModes(userId(), networkId(), me_->userModes());
 204   }
 205
 206   if(reason.isEmpty() && identityPtr())
 207     _quitReason = identityPtr()->quitReason();
 208   else
 209     _quitReason = reason;
 210
 211   displayMsg(Message::Server, BufferInfo::StatusBuffer, "", tr("Disconnecting. (%1)").arg((!requested && !withReconnect) ? tr("Core Shutdown") : _quitReason));
 212   switch(socket.state()) {
 213   case QAbstractSocket::ConnectedState:
 214     userInputHandler()->issueQuit(_quitReason);
 215     if(requested || withReconnect) {
 216       // the irc server has 10 seconds to close the socket
 217       _socketCloseTimer.start(10000);
 218       break;
 219     }
 220   default:
 221     socket.close();
 222     socketDisconnected();
 223   }
 224 }
 225
 226 void CoreNetwork::userInput(BufferInfo buf, QString msg) {
 227   userInputHandler()->handleUserInput(buf, msg);
 228 }
 229
 230 void CoreNetwork::putRawLine(QByteArray s) {
 231   if(_tokenBucket > 0)
 232     writeToSocket(s);
 233   else
 234     _msgQueue.append(s);
 235 }
 236
 237 void CoreNetwork::putCmd(const QString &cmd, const QList<QByteArray> &params, const QByteArray &prefix) {
 238   QByteArray msg;
 239
 240   if(!prefix.isEmpty())
 241     msg += ":" + prefix + " ";
 242   msg += cmd.toUpper().toAscii();
 243
 244   for(int i = 0; i < params.size() - 1; i++) {
 245     msg += " " + params[i];
 246   }
 247   if(!params.isEmpty())
 248     msg += " :" + params.last();
 249
 250   putRawLine(msg);
 251 }
 252
 253 void CoreNetwork::setChannelJoined(const QString &channel) {
 254   _autoWhoQueue.prepend(channel.toLower()); // prepend so this new chan is the first to be checked
 255
 256   Core::setChannelPersistent(userId(), networkId(), channel, true);
 257   Core::setPersistentChannelKey(userId(), networkId(), channel, _channelKeys[channel.toLower()]);
 258 }
 259
 260 void CoreNetwork::setChannelParted(const QString &channel) {
 261   removeChannelKey(channel);
 262   _autoWhoQueue.removeAll(channel.toLower());
 263   _autoWhoPending.remove(channel.toLower());
 264
 265   Core::setChannelPersistent(userId(), networkId(), channel, false);
 266 }
 267
 268 void CoreNetwork::addChannelKey(const QString &channel, const QString &key) {
 269   if(key.isEmpty()) {
 270     removeChannelKey(channel);
 271   } else {
 272     _channelKeys[channel.toLower()] = key;
 273   }
 274 }
 275
 276 void CoreNetwork::removeChannelKey(const QString &channel) {
 277   _channelKeys.remove(channel.toLower());
 278 }
 279
 280 bool CoreNetwork::setAutoWhoDone(const QString &channel) {
 281   QString chan = channel.toLower();
 282   if(_autoWhoPending.value(chan, 0) <= 0)
 283     return false;
 284   if(--_autoWhoPending[chan] <= 0)
 285     _autoWhoPending.remove(chan);
 286   return true;
 287 }
 288
 289 void CoreNetwork::setMyNick(const QString &mynick) {
 290   Network::setMyNick(mynick);
 291   if(connectionState() == Network::Initializing)
 292     networkInitialized();
 293 }
 294
 295 void CoreNetwork::socketHasData() {
 296   while(socket.canReadLine()) {
 297     QByteArray s = socket.readLine().trimmed();
 298     ircServerHandler()->handleServerMsg(s);
 299   }
 300 }
 301
 302 void CoreNetwork::socketError(QAbstractSocket::SocketError error) {
 303   if(_quitRequested && error == QAbstractSocket::RemoteHostClosedError)
 304     return;
 305
 306   _previousConnectionAttemptFailed = true;
 307   qWarning() << qPrintable(tr("Could not connect to %1 (%2)").arg(networkName(), socket.errorString()));
 308   emit connectionError(socket.errorString());
 309   displayMsg(Message::Error, BufferInfo::StatusBuffer, "", tr("Connection failure: %1").arg(socket.errorString()));
 310   emitConnectionError(socket.errorString());
 311   if(socket.state() < QAbstractSocket::ConnectedState) {
 312     socketDisconnected();
 313   }
 314 }
 315
 316 void CoreNetwork::socketInitialized() {
 317   Server server = usedServer();
 318 #ifdef HAVE_SSL
 319   if(server.useSsl && !socket.isEncrypted())
 320     return;
 321 #endif
 322
 323   CoreIdentity *identity = identityPtr();
 324   if(!identity) {
 325     qCritical() << "Identity invalid!";
 326     disconnectFromIrc();
 327     return;
 328   }
 329
 330   // TokenBucket to avoid sending too much at once
 331   _messageDelay = 2200;    // this seems to be a safe value (2.2 seconds delay)
 332   _burstSize = 5;
 333   _tokenBucket = _burstSize; // init with a full bucket
 334   _tokenBucketTimer.start(_messageDelay);
 335
 336   if(!server.password.isEmpty()) {
 337     putRawLine(serverEncode(QString("PASS %1").arg(server.password)));
 338   }
 339   QString nick;
 340   if(identity->nicks().isEmpty()) {
 341     nick = "quassel";
 342     qWarning() << "CoreNetwork::socketInitialized(): no nicks supplied for identity Id" << identity->id();
 343   } else {
 344     nick = identity->nicks()[0];
 345   }
 346   putRawLine(serverEncode(QString("NICK :%1").arg(nick)));
 347   putRawLine(serverEncode(QString("USER %1 8 * :%2").arg(identity->ident(), identity->realName())));
 348 }
 349
 350 void CoreNetwork::socketDisconnected() {
 351   disablePingTimeout();
 352   _msgQueue.clear();
 353
 354   _autoWhoCycleTimer.stop();
 355   _autoWhoTimer.stop();
 356   _autoWhoQueue.clear();
 357   _autoWhoPending.clear();
 358
 359   _socketCloseTimer.stop();
 360
 361   _tokenBucketTimer.stop();
 362
 363   IrcUser *me_ = me();
 364   if(me_) {
 365     foreach(QString channel, me_->channels())
 366       displayMsg(Message::Quit, BufferInfo::ChannelBuffer, channel, _quitReason, me_->hostmask());
 367   }
 368
 369   setConnected(false);
 370   emit disconnected(networkId());
 371   if(_quitRequested) {
 372     setConnectionState(Network::Disconnected);
 373     Core::setNetworkConnected(userId(), networkId(), false);
 374   } else if(_autoReconnectCount != 0) {
 375     setConnectionState(Network::Reconnecting);
 376     if(_autoReconnectCount == autoReconnectRetries())
 377       doAutoReconnect(); // first try is immediate
 378     else
 379       _autoReconnectTimer.start();
 380   }
 381 }
 382
 383 void CoreNetwork::socketStateChanged(QAbstractSocket::SocketState socketState) {
 384   Network::ConnectionState state;
 385   switch(socketState) {
 386     case QAbstractSocket::UnconnectedState:
 387       state = Network::Disconnected;
 388       break;
 389     case QAbstractSocket::HostLookupState:
 390     case QAbstractSocket::ConnectingState:
 391       state = Network::Connecting;
 392       break;
 393     case QAbstractSocket::ConnectedState:
 394       state = Network::Initializing;
 395       break;
 396     case QAbstractSocket::ClosingState:
 397       state = Network::Disconnecting;
 398       break;
 399     default:
 400       state = Network::Disconnected;
 401   }
 402   setConnectionState(state);
 403 }
 404
 405 void CoreNetwork::networkInitialized() {
 406   setConnectionState(Network::Initialized);
 407   setConnected(true);
 408   _quitRequested = false;
 409
 410   if(useAutoReconnect()) {
 411     // reset counter
 412     _autoReconnectCount = autoReconnectRetries();
 413   }
 414
 415   // restore away state
 416   QString awayMsg = Core::awayMessage(userId(), networkId());
 417   if(!awayMsg.isEmpty())
 418     userInputHandler()->handleAway(BufferInfo(), Core::awayMessage(userId(), networkId()));
 419
 420   // restore old user modes if server default mode is set.
 421   IrcUser *me_ = me();
 422   if(me_) {
 423     if(!me_->userModes().isEmpty()) {
 424       restoreUserModes();
 425     } else {
 426       connect(me_, SIGNAL(userModesSet(QString)), this, SLOT(restoreUserModes()));
 427       connect(me_, SIGNAL(userModesAdded(QString)), this, SLOT(restoreUserModes()));
 428     }
 429   }
 430
 431   sendPerform();
 432
 433   enablePingTimeout();
 434
 435   if(_autoWhoEnabled) {
 436     _autoWhoCycleTimer.start();
 437     _autoWhoTimer.start();
 438     startAutoWhoCycle();  // FIXME wait for autojoin to be completed
 439   }
 440
 441   Core::bufferInfo(userId(), networkId(), BufferInfo::StatusBuffer); // create status buffer
 442   Core::setNetworkConnected(userId(), networkId(), true);
 443 }
 444
 445 void CoreNetwork::sendPerform() {
 446   BufferInfo statusBuf = BufferInfo::fakeStatusBuffer(networkId());
 447
 448   // do auto identify
 449   if(useAutoIdentify() && !autoIdentifyService().isEmpty() && !autoIdentifyPassword().isEmpty()) {
 450     userInputHandler()->handleMsg(statusBuf, QString("%1 IDENTIFY %2").arg(autoIdentifyService(), autoIdentifyPassword()));
 451   }
 452
 453   // send perform list
 454   foreach(QString line, perform()) {
 455     if(!line.isEmpty()) userInput(statusBuf, line);
 456   }
 457
 458   // rejoin channels we've been in
 459   if(rejoinChannels()) {
 460     QStringList channels, keys;
 461     foreach(QString chan, coreSession()->persistentChannels(networkId()).keys()) {
 462       QString key = channelKey(chan);
 463       if(!key.isEmpty()) {
 464         channels.prepend(chan);
 465         keys.prepend(key);
 466       } else {
 467         channels.append(chan);
 468       }
 469     }
 470     QString joinString = QString("%1 %2").arg(channels.join(",")).arg(keys.join(",")).trimmed();
 471     if(!joinString.isEmpty())
 472       userInputHandler()->handleJoin(statusBuf, joinString);
 473   }
 474 }
 475
 476 void CoreNetwork::restoreUserModes() {
 477   IrcUser *me_ = me();
 478   Q_ASSERT(me_);
 479
 480   disconnect(me_, SIGNAL(userModesSet(QString)), this, SLOT(restoreUserModes()));
 481   disconnect(me_, SIGNAL(userModesAdded(QString)), this, SLOT(restoreUserModes()));
 482
 483   QString removeModes;
 484   QString addModes = Core::userModes(userId(), networkId());
 485   QString currentModes = me_->userModes();
 486
 487   removeModes = currentModes;
 488   removeModes.remove(QRegExp(QString("[%1]").arg(addModes)));
 489   addModes.remove(QRegExp(QString("[%1]").arg(currentModes)));
 490
 491   removeModes = QString("%1 -%2").arg(me_->nick(), removeModes);
 492   addModes = QString("%1 +%2").arg(me_->nick(), addModes);
 493   userInputHandler()->handleMode(BufferInfo(), removeModes);
 494   userInputHandler()->handleMode(BufferInfo(), addModes);
 495 }
 496
 497 void CoreNetwork::setUseAutoReconnect(bool use) {
 498   Network::setUseAutoReconnect(use);
 499   if(!use)
 500     _autoReconnectTimer.stop();
 501 }
 502
 503 void CoreNetwork::setAutoReconnectInterval(quint32 interval) {
 504   Network::setAutoReconnectInterval(interval);
 505   _autoReconnectTimer.setInterval(interval * 1000);
 506 }
 507
 508 void CoreNetwork::setAutoReconnectRetries(quint16 retries) {
 509   Network::setAutoReconnectRetries(retries);
 510   if(_autoReconnectCount != 0) {
 511     if(unlimitedReconnectRetries())
 512       _autoReconnectCount = -1;
 513     else
 514       _autoReconnectCount = autoReconnectRetries();
 515   }
 516 }
 517
 518 void CoreNetwork::doAutoReconnect() {
 519   if(connectionState() != Network::Disconnected && connectionState() != Network::Reconnecting) {
 520     qWarning() << "CoreNetwork::doAutoReconnect(): Cannot reconnect while not being disconnected!";
 521     return;
 522   }
 523   if(_autoReconnectCount > 0)
 524     _autoReconnectCount--;
 525   connectToIrc(true);
 526 }
 527
 528 void CoreNetwork::sendPing() {
 529   uint now = QDateTime::currentDateTime().toTime_t();
 530   if(_pingCount != 0) {
 531     qDebug() << "UserId:" << userId() << "Network:" << networkName() << "missed" << _pingCount << "pings."
 532              << "BA:" << socket.bytesAvailable() << "BTW:" << socket.bytesToWrite();
 533   }
 534   if(_pingCount >= _maxPingCount && now - _lastPingTime <= (uint)(_pingTimer.interval() / 1000) + 1) {
 535     // the second check compares the actual elapsed time since the last ping and the pingTimer interval
 536     // if the interval is shorter then the actual elapsed time it means that this thread was somehow blocked
 537     // and unable to even handle a ping answer. So we ignore those misses.
 538     disconnectFromIrc(false, QString("No Ping reply in %1 seconds.").arg(_maxPingCount * _pingTimer.interval() / 1000), true /* withReconnect */);
 539   } else {
 540     _lastPingTime = now;
 541     _pingCount++;
 542     userInputHandler()->handlePing(BufferInfo(), QString());
 543   }
 544 }
 545
 546 void CoreNetwork::enablePingTimeout() {
 547   resetPingTimeout();
 548   _pingTimer.start();
 549 }
 550
 551 void CoreNetwork::disablePingTimeout() {
 552   _pingTimer.stop();
 553   resetPingTimeout();
 554 }
 555
 556 void CoreNetwork::sendAutoWho() {
 557   // Don't send autowho if there are still some pending
 558   if(_autoWhoPending.count())
 559     return;
 560
 561   while(!_autoWhoQueue.isEmpty()) {
 562     QString chan = _autoWhoQueue.takeFirst();
 563     IrcChannel *ircchan = ircChannel(chan);
 564     if(!ircchan) continue;
 565     if(_autoWhoNickLimit > 0 && ircchan->ircUsers().count() > _autoWhoNickLimit) continue;
 566     _autoWhoPending[chan]++;
 567     putRawLine("WHO " + serverEncode(chan));
 568     if(_autoWhoQueue.isEmpty() && _autoWhoEnabled && !_autoWhoCycleTimer.isActive()) {
 569       // Timer was stopped, means a new cycle is due immediately
 570       _autoWhoCycleTimer.start();
 571       startAutoWhoCycle();
 572     }
 573     break;
 574   }
 575 }
 576
 577 void CoreNetwork::startAutoWhoCycle() {
 578   if(!_autoWhoQueue.isEmpty()) {
 579     _autoWhoCycleTimer.stop();
 580     return;
 581   }
 582   _autoWhoQueue = channels();
 583 }
 584
 585 #ifdef HAVE_SSL
 586 void CoreNetwork::sslErrors(const QList<QSslError> &sslErrors) {
 587   Q_UNUSED(sslErrors)
 588   socket.ignoreSslErrors();
 589   // TODO errorhandling
 590 }
 591 #endif  // HAVE_SSL
 592
 593 void CoreNetwork::fillBucketAndProcessQueue() {
 594   if(_tokenBucket < _burstSize) {
 595     _tokenBucket++;
 596   }
 597
 598   while(_msgQueue.size() > 0 && _tokenBucket > 0) {
 599     writeToSocket(_msgQueue.takeFirst());
 600   }
 601 }
 602
 603 void CoreNetwork::writeToSocket(const QByteArray &data) {
 604   socket.write(data);
 605   socket.write("\r\n");
 606   _tokenBucket--;
 607 }
 608
 609 Network::Server CoreNetwork::usedServer() const {
 610   if(_lastUsedServerIndex < serverList().count())
 611     return serverList()[_lastUsedServerIndex];
 612
 613   if(!serverList().isEmpty())
 614     return serverList()[0];
 615
 616   return Network::Server();
 617 }
 618
 619 void CoreNetwork::requestConnect() const {
 620   if(connectionState() != Disconnected) {
 621     qWarning() << "Requesting connect while already being connected!";
 622     return;
 623   }
 624   Network::requestConnect();
 625 }
 626
 627 void CoreNetwork::requestDisconnect() const {
 628   if(connectionState() == Disconnected) {
 629     qWarning() << "Requesting disconnect while not being connected!";
 630     return;
 631   }
 632   userInputHandler()->handleQuit(BufferInfo(), QString());
 633 }
 634
 635 void CoreNetwork::requestSetNetworkInfo(const NetworkInfo &info) {
 636   Network::Server currentServer = usedServer();
 637   setNetworkInfo(info);
 638   Core::updateNetwork(coreSession()->user(), info);
 639
 640   // the order of the servers might have changed,
 641   // so we try to find the previously used server
 642   _lastUsedServerIndex = 0;
 643   for(int i = 0; i < serverList().count(); i++) {
 644     Network::Server server = serverList()[i];
 645     if(server.host == currentServer.host && server.port == currentServer.port) {
 646       _lastUsedServerIndex = i;
 647       break;
 648     }
 649   }
 650 }