src/dashserver/server.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2006 by David Cuadrado                                  *
   3  *   krawek@gmail.com                                                      *
   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 "server.h"
  22 #include "connection.h"
  23
  24 #include <QHostInfo>
  25 #include <QTimer>
  26 #include <QQueue>
  27 #include <QHostAddress>
  28
  29 #include <dcore/debug.h>
  30
  31 #include "packagehandlerbase.h"
  32 #include "defaultpackagehandler.h"
  33
  34 #include "bans/banmanager.h"
  35 #include "backups/backupmanager.h"
  36 #include "users/usermanager.h"
  37 #include "comunications/comunicationmanager.h"
  38 #include "registers/regmanager.h"
  39 #include "project/projectmanager.h"
  40
  41 #include "package.h"
  42 #include "observer.h"
  43 #include "logger.h"
  44
  45 #include <dash/network/socket.h>
  46
  47 using namespace Dash::Network;
  48
  49 namespace DashServer {
  50
  51 class TcpServer::Private
  52 {
  53         public:
  54                 QList<DashServer::Connection *> connections;
  55                 QList<DashServer::Connection *> admins;
  56
  57                 Backups::Manager *backupManager;
  58                 Bans::Manager *banManager;
  59                 Users::Manager *userManager;
  60                 Comunications::Manager *comunicationManager;
  61
  62                 QList<DashServer::Observer *> observers;
  63 };
  64
  65 TcpServer::TcpServer(QObject *parent) : QTcpServer(parent), d(new Private)
  66 {
  67         D_INIT;
  68         d->backupManager = new Backups::Manager;
  69         d->observers << d->backupManager;
  70
  71         d->banManager = new Bans::Manager;
  72         d->observers << d->banManager;
  73
  74         d->userManager = new Users::Manager;
  75         d->observers << d->userManager;
  76
  77         d->comunicationManager = new Comunications::Manager;
  78         d->observers << d->comunicationManager;
  79
  80         addObserver(new DashServer::Project::Manager);
  81
  82         addObserver(new DashServer::Registers::Manager);
  83 }
  84
  85
  86 TcpServer::~TcpServer()
  87 {
  88         D_END;
  89
  90         DashServer::Logger::self()->info("DashServer finished");
  91
  92         qDebug("=> Deleting observers");
  93         qDeleteAll(d->observers);
  94
  95         qDebug("=> Deleting connections");
  96         foreach(DashServer::Connection *cnn, d->connections)
  97         {
  98                 removeConnection(cnn);
  99         }
 100
 101         delete d;
 102 }
 103
 104 bool TcpServer::openConnection(const QString &host, int port)
 105 {
 106         DashServer::Logger::self()->info(QObject::tr("Initialized server on %1:%2").arg(host).arg(port));
 107
 108         QList<QHostAddress> addrs = QHostInfo::fromName(host).addresses();
 109         if ( !addrs.isEmpty() )
 110         {
 111                 if(! listen(QHostAddress(addrs[0]), port) )
 112                 {
 113                         dError() << "Can't connect to " << host<<":"<<port<< " error was: " << errorString();
 114                         return false;
 115                 }
 116         }
 117         else
 118         {
 119                 dError() << "Error while try to resolve " << host;
 120                 return false;
 121         }
 122         return true;
 123 }
 124
 125 void TcpServer::addAdmin(DashServer::Connection *cnx)
 126 {
 127         d->admins << cnx;
 128 }
 129
 130 Backups::Manager *TcpServer::backupManager() const
 131 {
 132         return d->backupManager;
 133 }
 134
 135 Bans::Manager *TcpServer::banManager() const
 136 {
 137         return d->banManager;
 138 }
 139
 140 Users::Manager *TcpServer::userManager() const
 141 {
 142         return d->userManager;
 143 }
 144
 145 void TcpServer::addObserver(DashServer::Observer *observer)
 146 {
 147         d->observers << observer;
 148 }
 149
 150 bool TcpServer::removeObserver(DashServer::Observer *observer)
 151 {
 152         return d->observers.removeAll(observer) > 0;
 153 }
 154
 155 void TcpServer::incomingConnection(int socketDescriptor)
 156 {
 157         D_SHOW_VAR(d->connections.count());
 158         D_SHOW_VAR(socketDescriptor);
 159
 160         DashServer::Connection *newConnection = new DashServer::Connection(socketDescriptor,this);
 161         QString ip = newConnection->client()->peerAddress().toString();
 162         d->banManager->initialize(ip);
 163         if ( !d->banManager->isBanned(ip) )
 164         {
 165                 d->banManager->unban(ip);
 166                 handle(newConnection);
 167                 d->connections << newConnection;
 168         }
 169         else
 170         {
 171                 newConnection->sendError(tr("You're banned, Please contact to server administrator if you think is an error!"), Dash::Network::Package::Error::Err);
 172                 newConnection->close();
 173         }
 174 }
 175
 176 void TcpServer::handle(DashServer::Connection *cnx)
 177 {
 178         connect(cnx, SIGNAL(requestSendToAll( const QString& )), this, SLOT(sendToAll( const QString& )));
 179
 180         connect(cnx, SIGNAL(packageReaded(DashServer::Connection*, const QString&, const QString&)), this, SLOT(handlePackage(DashServer::Connection*, const QString&, const QString&)));
 181
 182         connect(cnx, SIGNAL(connectionClosed(DashServer::Connection*)), this, SLOT(removeConnection(DashServer::Connection*)));
 183 }
 184
 185
 186 void TcpServer::sendToAll(const QString &msg)
 187 {
 188         foreach(DashServer::Connection *connection, d->connections)
 189         {
 190                 connection->sendToClient(msg, false);
 191         }
 192 }
 193
 194 void TcpServer::sendToAll(const QDomDocument &pkg)
 195 {
 196         D_FUNCINFO;
 197         sendToAll(pkg.toString(0));
 198 }
 199
 200 void TcpServer::sendToAdmins(const QString &str)
 201 {
 202         foreach(DashServer::Connection *cnn, d->admins)
 203         {
 204                 cnn->sendToClient(str, false);
 205         }
 206 }
 207
 208 void TcpServer::removeConnection(DashServer::Connection *cnx)
 209 {
 210         D_FUNCINFO;
 211
 212         if( ! cnx ) return;
 213
 214         d->connections.removeAll(cnx);
 215         d->admins.removeAll(cnx);
 216
 217         cnx->blockSignals(true);
 218         cnx->close();
 219         cnx->blockSignals(false);
 220
 221
 222         foreach(DashServer::Observer *observer, d->observers)
 223         {
 224                 observer->connectionClosed(cnx);
 225         }
 226
 227         cnx->deleteLater();
 228 }
 229
 230
 231 void TcpServer::handlePackage(DashServer::Connection* client, const QString &root, const QString&package)
 232 {
 233         DashServer::Package *pkg = new DashServer::Package(root, package, client);
 234
 235         foreach(DashServer::Observer *observer, d->observers)
 236         {
 237                 observer->handlePackage(pkg);
 238
 239                 if ( pkg->isAccepted() )
 240                         break;
 241         }
 242
 243         if( ! pkg->isAccepted() )
 244                 qWarning() << "NOT ACCEPTED: " << pkg->root();
 245
 246         delete pkg;
 247
 248         if( !client->isRunning() )
 249                 client->start();
 250 }
 251
 252
 253 }
 254
 255