network/server.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005/06 by The Quassel Team                             *
   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) any later version.                                   *
   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 "util.h"
  22 #include "global.h"
  23 #include "server.h"
  24 #include "cmdcodes.h"
  25 #include "message.h"
  26
  27 #include <QMetaObject>
  28 #include <QDateTime>
  29
  30 Server::Server(QString net) : network(net) {
  31
  32 }
  33
  34 Server::~Server() {
  35
  36 }
  37
  38 void Server::run() {
  39   connect(&socket, SIGNAL(connected()), this, SLOT(socketConnected()));
  40   connect(&socket, SIGNAL(disconnected()), this, SLOT(socketDisconnected()));
  41   connect(&socket, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(socketError(QAbstractSocket::SocketError)));
  42   connect(&socket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), this, SLOT(socketStateChanged(QAbstractSocket::SocketState)));
  43   connect(&socket, SIGNAL(readyRead()), this, SLOT(socketHasData()));
  44
  45   exec();
  46 }
  47
  48 void Server::connectToIrc(QString net) {
  49   if(net != network) return; // not me!
  50   networkSettings = global->getData("Networks").toMap()[net].toMap();
  51   identity = global->getData("Identities").toMap()[networkSettings["Identity"].toString()].toMap();
  52   QList<QVariant> servers = networkSettings["Servers"].toList();
  53   QString host = servers[0].toMap()["Address"].toString();
  54   quint16 port = servers[0].toMap()["Port"].toUInt();
  55   displayStatusMsg(QString("Connecting to %1:%2...").arg(host).arg(port));
  56   socket.connectToHost(host, port);
  57 }
  58
  59 void Server::disconnectFromIrc(QString net) {
  60   if(net != network) return; // not me!
  61   socket.disconnectFromHost();
  62 }
  63
  64 void Server::socketHasData() {
  65   while(socket.canReadLine()) {
  66     QString s = socket.readLine().trimmed();
  67     qDebug() << "Read: " << s;
  68     emit recvRawServerMsg(s);
  69     //Message *msg = Message::createFromServerString(this, s);
  70     handleServerMsg(s);
  71   }
  72 }
  73
  74 void Server::socketError( QAbstractSocket::SocketError err ) {
  75   //qDebug() << "Socket Error!";
  76   //emit error(err);
  77 }
  78
  79 void Server::socketConnected( ) {
  80   putRawLine(QString("NICK :%1").arg(identity["NickList"].toStringList()[0]));
  81   putRawLine(QString("USER %1 8 * :%2").arg(identity["Ident"].toString()).arg(identity["RealName"].toString()));
  82 }
  83
  84 void Server::socketDisconnected( ) {
  85   //qDebug() << "Socket disconnected!";
  86   emit disconnected();
  87 }
  88
  89 void Server::socketStateChanged(QAbstractSocket::SocketState state) {
  90   //qDebug() << "Socket state changed: " << state;
  91 }
  92
  93 QString Server::updateNickFromMask(QString mask) {
  94   QString user = userFromMask(mask);
  95   QString host = hostFromMask(mask);
  96   QString nick = nickFromMask(mask);
  97   if(nicks.contains(nick) && !user.isEmpty() && !host.isEmpty()) {
  98     VarMap n = nicks[nick].toMap();
  99     if(n["User"].toString() != user || n["Host"].toString() != host) {
 100       if(!n["User"].toString().isEmpty() || !n["Host"].toString().isEmpty())
 101         qWarning(QString("Strange: Hostmask for nick %1 has changed!").arg(nick).toAscii());
 102       n["User"] = user; n["Host"] = host;
 103       nicks[nick] = n;
 104       emit nickUpdated(network, nick, n);
 105     }
 106   }
 107   return nick;
 108 }
 109
 110 void Server::userInput(QString net, QString buf, QString msg) {
 111   if(net != network) return; // not me!
 112   msg = msg.trimmed(); // remove whitespace from start and end
 113   if(msg.isEmpty()) return;
 114   if(!msg.startsWith('/')) {
 115     msg = QString("/SAY ") + msg;
 116   }
 117   handleUserMsg(buf, msg);
 118 }
 119
 120 void Server::putRawLine(QString s) {
 121   qDebug() << "SentRaw: " << s;
 122   s += "\r\n";
 123   socket.write(s.toAscii());
 124 }
 125
 126 void Server::putCmd(QString cmd, QStringList params, QString prefix) {
 127   QString m;
 128   if(!prefix.isEmpty()) m += ":" + prefix + " ";
 129   m += cmd.toUpper();
 130   for(int i = 0; i < params.size() - 1; i++) {
 131     m += " " + params[i];
 132   }
 133   if(!params.isEmpty()) m += " :" + params.last();
 134   qDebug() << "SentCmd: " << m;
 135   m += "\r\n";
 136   socket.write(m.toAscii());
 137 }
 138
 139 /** Handle a raw message string sent by the server. We try to find a suitable handler, otherwise we call a default handler. */
 140 void Server::handleServerMsg(QString msg) {
 141   try {
 142     if(msg.isEmpty()) {
 143       qWarning() << "Received empty string from server!";
 144       return;
 145     }
 146     // OK, first we split the raw message into its various parts...
 147     QString prefix;
 148     QString cmd;
 149     QStringList params;
 150     if(msg[0] == ':') {
 151       msg.remove(0,1);
 152       prefix = msg.section(' ', 0, 0);
 153       msg = msg.section(' ', 1);
 154     }
 155     cmd = msg.section(' ', 0, 0).toUpper();
 156     msg = msg.section(' ', 1);
 157     QString left = msg.section(':', 0, 0);
 158     QString trailing = msg.section(':', 1);
 159     if(!left.isEmpty()) {
 160       params << left.split(' ', QString::SkipEmptyParts);
 161     }
 162     if(!trailing.isEmpty()) {
 163       params << trailing;
 164     }
 165     // numeric replies usually have our own nick as first param. Remove this!
 166     // BTW, this behavior is not in the RFC.
 167     uint num = cmd.toUInt();
 168     if(num > 1 && params.count() > 0) {  // 001 sets our nick, so we shouldn't remove anything
 169       if(params[0] == currentNick) params.removeFirst();
 170       else qWarning((QString("First param NOT nick: %1:%2 %3").arg(prefix).arg(cmd).arg(params.join(" "))).toAscii());
 171     }
 172     // Now we try to find a handler for this message. BTW, I do love the Trolltech guys ;-)
 173     QString hname = cmd.toLower();
 174     hname[0] = hname[0].toUpper();
 175     hname = "handleServer" + hname;
 176     if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, prefix), Q_ARG(QStringList, params))) {
 177       // Ok. Default handler it is.
 178       defaultServerHandler(cmd, prefix, params);
 179     }
 180   } catch(Exception e) {
 181     emit displayMsg("", Message(Message::Error, e.msg()));
 182   }
 183 }
 184
 185 void Server::defaultServerHandler(QString cmd, QString prefix, QStringList params) {
 186   uint num = cmd.toUInt();
 187   if(num) {
 188     // A lot of server messages don't really need their own handler because they don't do much.
 189     // Catch and handle these here.
 190     switch(num) {
 191       // Welcome, status, info messages. Just display these.
 192       case 2: case 3: case 4: case 5: case 251: case 252: case 253: case 254: case 255: case 372: case 375:
 193         emit displayMsg("", Message(Message::Server, params.join(" "), prefix));
 194         break;
 195       // Ignore these commands.
 196       case 366: case 376:
 197         break;
 198
 199       // Everything else will be marked in red, so we can add them somewhere.
 200       default:
 201         emit displayMsg("", Message(Message::Error, cmd + " " + params.join(" "), prefix));
 202     }
 203     //qDebug() << prefix <<":"<<cmd<<params;
 204   } else {
 205     emit displayMsg("", Message(Message::Error, QString("Unknown: ") + cmd + " " + params.join(" "), prefix));
 206     //qDebug() << prefix <<":"<<cmd<<params;
 207   }
 208 }
 209
 210 void Server::handleUserMsg(QString bufname, QString usrMsg) {
 211   try {
 212     Buffer *buffer = 0;
 213     if(!bufname.isEmpty()) {
 214       Q_ASSERT(buffers.contains(bufname));
 215       buffer = buffers[bufname];
 216     }
 217     QString cmd = usrMsg.section(' ', 0, 0).remove(0, 1).toUpper();
 218     QString msg = usrMsg.section(' ', 1).trimmed();
 219     QString hname = cmd.toLower();
 220     hname[0] = hname[0].toUpper();
 221     hname = "handleUser" + hname;
 222     if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, msg), Q_ARG(Buffer*, buffer))) {
 223         // Ok. Default handler it is.
 224       defaultUserHandler(cmd, msg, buffer);
 225     }
 226   } catch(Exception e) {
 227     emit displayMsg("", Message(Message::Error, e.msg()));
 228   }
 229 }
 230
 231 void Server::defaultUserHandler(QString cmd, QString msg, Buffer *buf) {
 232   emit displayMsg("", Message(Message::Error, QString("Error: %1 %2").arg(cmd).arg(msg)));
 233
 234 }
 235
 236 /**********************************************************************************/
 237
 238 /*
 239 void Server::handleUser(QString msg, Buffer *buf) {
 240
 241
 242 }
 243 */
 244
 245 void Server::handleUserJoin(QString msg, Buffer *buf) {
 246   putCmd("JOIN", QStringList(msg));
 247
 248 }
 249
 250 void Server::handleUserQuote(QString msg, Buffer *buf) {
 251   putRawLine(msg);
 252 }
 253
 254 void Server::handleUserSay(QString msg, Buffer *buf) {
 255   if(!buf) return;  // server buffer
 256   QStringList params;
 257   params << buf->name() << msg;
 258   putCmd("PRIVMSG", params);
 259   emit displayMsg(params[0], Message(Message::Msg, msg, currentNick, Message::Self));
 260 }
 261
 262 /**********************************************************************************/
 263
 264 void Server::handleServerJoin(QString prefix, QStringList params) {
 265   Q_ASSERT(params.count() == 1);
 266   QString nick = updateNickFromMask(prefix);
 267   if(nick == currentNick) {
 268     Q_ASSERT(!buffers.contains(params[0]));  // cannot join a buffer twice!
 269     Buffer *buf = new Buffer(params[0]);
 270     buffers[params[0]] = buf;
 271   } else {
 272     VarMap n;
 273     if(nicks.contains(nick)) {
 274       n = nicks[nick].toMap();
 275       VarMap chans = n["Channels"].toMap();
 276       // Q_ASSERT(!chans.keys().contains(params[0])); TODO uncomment
 277       chans[params[0]] = VarMap();
 278       n["Channels"] = chans;
 279       nicks[nick] = n;
 280       emit nickUpdated(network, nick, n);
 281     } else {
 282       VarMap chans;
 283       chans[params[0]] = VarMap();
 284       n["Channels"] = chans;
 285       n["User"] = userFromMask(prefix);
 286       n["Host"] = hostFromMask(prefix);
 287       nicks[nick] = n;
 288       emit nickAdded(network, nick, n);
 289     }
 290     emit displayMsg(params[0], Message(Message::Join, params[0], prefix));
 291   }
 292 }
 293
 294 void Server::handleServerKick(QString prefix, QStringList params) {
 295   QString kicker = updateNickFromMask(prefix);
 296   QString nick = params[1];
 297   Q_ASSERT(nicks.contains(nick));
 298   VarMap n = nicks[nick].toMap();
 299   VarMap chans = n["Channels"].toMap();
 300   Q_ASSERT(chans.contains(params[0]));
 301   chans.remove(params[0]);
 302   QString msg = nick;
 303   if(params.count() > 2) msg = QString("%1 %2").arg(msg).arg(params[2]);
 304   emit displayMsg(params[0], Message(Message::Kick, msg, prefix));
 305   if(chans.count() > 0) {
 306     n["Channels"] = chans;
 307     nicks[nick] = n;
 308     emit nickUpdated(network, nick, n);
 309   } else {
 310     nicks.remove(nick);
 311     emit nickRemoved(network, nick);
 312   }
 313 }
 314
 315 void Server::handleServerNick(QString prefix, QStringList params) {
 316   QString oldnick = updateNickFromMask(prefix);
 317   QString newnick = params[0];
 318   QVariant v = nicks.take(oldnick);
 319   nicks[newnick] = v;
 320   VarMap chans = v.toMap()["Channels"].toMap();
 321   foreach(QString c, chans.keys()) {
 322     if(oldnick != currentNick) { emit displayMsg(c, Message(Message::Nick, newnick, prefix)); }
 323     else { emit displayMsg(c, Message(Message::Nick, newnick, newnick)); }
 324   }
 325   emit nickRenamed(network, oldnick, newnick);
 326   if(oldnick == currentNick) {
 327     currentNick = newnick;
 328     emit ownNickSet(network, newnick);
 329   }
 330 }
 331
 332 void Server::handleServerNotice(QString prefix, QStringList params) {
 333   //Message msg(Message::Notice, params[1], prefix);
 334   if(prefix == currentServer) emit displayMsg("", Message(Message::Server, params[1], prefix));
 335   else emit displayMsg("", Message(Message::Notice, params[1], prefix));
 336 }
 337
 338 void Server::handleServerPart(QString prefix, QStringList params) {
 339   QString nick = updateNickFromMask(prefix);
 340   Q_ASSERT(nicks.contains(nick));
 341   VarMap n = nicks[nick].toMap();
 342   VarMap chans = n["Channels"].toMap();
 343   Q_ASSERT(chans.contains(params[0]));
 344   chans.remove(params[0]);
 345   QString msg;
 346   if(params.count() > 1) msg = params[1];
 347   emit displayMsg(params[0], Message(Message::Part, msg, prefix));
 348   if(chans.count() > 0) {
 349     n["Channels"] = chans;
 350     nicks[nick] = n;
 351     emit nickUpdated(network, nick, n);
 352   } else {
 353     nicks.remove(nick);
 354     emit nickRemoved(network, nick);
 355   }
 356 }
 357
 358 void Server::handleServerPing(QString prefix, QStringList params) {
 359   putCmd("PONG", params);
 360 }
 361
 362 void Server::handleServerPrivmsg(QString prefix, QStringList params) {
 363   updateNickFromMask(prefix);
 364   emit displayMsg(params[0], Message(Message::Msg, params[1], prefix));
 365
 366 }
 367
 368 void Server::handleServerQuit(QString prefix, QStringList params) {
 369   QString nick = updateNickFromMask(prefix);
 370   Q_ASSERT(nicks.contains(nick));
 371   VarMap chans = nicks[nick].toMap()["Channels"].toMap();
 372   foreach(QString c, chans.keys()) {
 373     emit displayMsg(c, Message(Message::Quit, params[0], prefix));
 374   }
 375   nicks.remove(nick);
 376   emit nickRemoved(network, nick);
 377 }
 378
 379 /* RPL_WELCOME */
 380 void Server::handleServer001(QString prefix, QStringList params) {
 381   currentServer = prefix;
 382   currentNick = params[0];
 383   VarMap n;
 384   n["Channels"] = VarMap();
 385   nicks[currentNick] = n;
 386   emit ownNickSet(network, currentNick);
 387   emit nickAdded(network, currentNick, VarMap());
 388   emit displayMsg("", Message(Message::Server, params[1], prefix));
 389 }
 390
 391 /* RPL_NOTOPIC */
 392 void Server::handleServer331(QString prefix, QStringList params) {
 393   emit topicSet(network, params[0], "");
 394 }
 395
 396 /* RPL_TOPIC */
 397 void Server::handleServer332(QString prefix, QStringList params) {
 398   emit topicSet(network, params[0], params[1]);
 399   emit displayMsg(params[0], Message(Message::Server, tr("Topic for %1 is \"%2\"").arg(params[0]).arg(params[1])));
 400 }
 401
 402 /* Topic set by... */
 403 void Server::handleServer333(QString prefix, QStringList params) {
 404   emit displayMsg(params[0], Message(Message::Server,
 405                   tr("Topic set by %1 on %2").arg(params[1]).arg(QDateTime::fromTime_t(params[2].toUInt()).toString())));
 406 }
 407
 408 /* RPL_NAMREPLY */
 409 void Server::handleServer353(QString prefix, QStringList params) {
 410   params.removeFirst(); // = or *
 411   QString buf = params.takeFirst();
 412   foreach(QString nick, params[0].split(' ')) {
 413     // TODO: parse more prefix characters! use 005?
 414     QString mode = "";
 415     if(nick.startsWith('@')) { mode = "o"; nick.remove(0,1); }
 416     else if(nick.startsWith('+')) { mode = "v"; nick.remove(0,1); }
 417     VarMap c; c["Mode"] = mode;
 418     if(nicks.contains(nick)) {
 419       VarMap n = nicks[nick].toMap();
 420       VarMap chans = n["Channels"].toMap();
 421       chans[buf] = c;
 422       n["Channels"] = chans;
 423       nicks[nick] = n;
 424       emit nickUpdated(network, nick, n);
 425     } else {
 426       VarMap n; VarMap c; VarMap chans;
 427       c["Mode"] = mode;
 428       chans[buf] = c;
 429       n["Channels"] = chans;
 430       nicks[nick] = n;
 431       emit nickAdded(network, nick, n);
 432     }
 433   }
 434 }
 435 /***********************************************************************************/
 436
 437 /* Exception classes for message handling */
 438 Server::ParseError::ParseError(QString cmd, QString prefix, QStringList params) {
 439   _msg = QString("Command Parse Error: ") + cmd + params.join(" ");
 440
 441 }
 442
 443 Server::UnknownCmdError::UnknownCmdError(QString cmd, QString prefix, QStringList params) {
 444   _msg = QString("Unknown Command: ") + cmd;
 445
 446 }