|  | Home · All Classes · Main Classes · Grouped Classes · Modules · Functions |

# torrentserver.cpp Example File
network/torrent/torrentserver.cpp

        /****************************************************************************
        **
        ** Copyright (C) 2004-2006 Trolltech ASA. All rights reserved.
        **
        ** This file is part of the documentation of the Qt Toolkit.
        **
        ** This file may be used under the terms of the GNU General Public
        ** License version 2.0 as published by the Free Software Foundation
        ** and appearing in the file LICENSE.GPL included in the packaging of
        ** this file.  Please review the following information to ensure GNU
        ** General Public Licensing requirements will be met:
        ** http://www.trolltech.com/products/qt/opensource.html
        **
        ** If you are unsure which license is appropriate for your use, please
        ** review the following information:
        ** http://www.trolltech.com/products/qt/licensing.html or contact the
        ** sales department at sales@trolltech.com.
        **
        ** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
        ** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
        **
        ****************************************************************************/

        #include "connectionmanager.h"
        #include "peerwireclient.h"
        #include "ratecontroller.h"
        #include "torrentclient.h"
        #include "torrentserver.h"

        Q_GLOBAL_STATIC(TorrentServer, torrentServer)

        TorrentServer *TorrentServer::instance()
        {
            return torrentServer();
        }

        void TorrentServer::addClient(TorrentClient *client)
        {
            clients << client;
        }

        void TorrentServer::removeClient(TorrentClient *client)
        {
            clients.removeAll(client);
        }

        void TorrentServer::incomingConnection(int socketDescriptor)
        {
            PeerWireClient *client = new PeerWireClient(ConnectionManager::instance()->clientId(), this);
            if (clients.isEmpty())
                client->abort();

            if (ConnectionManager::instance()->canAddConnection()) {
                if (client->setSocketDescriptor(socketDescriptor)) {
                    connect(client, SIGNAL(infoHashReceived(const QByteArray &)),
                            this, SLOT(processInfoHash(const QByteArray &)));
                    connect(client, SIGNAL(error(QAbstractSocket::SocketError)),
                            this, SLOT(removeClient()));
                    RateController::instance()->addSocket(client);
                    ConnectionManager::instance()->addConnection(client);
                    if (clients.size() == 1) {
                        client->disconnect(client, 0, this, 0);
                        clients.first()->setupIncomingConnection(client);
                    }
                    return;
                }
            }

            delete client;
        }

        void TorrentServer::removeClient()
        {
            PeerWireClient *peer = qobject_cast<PeerWireClient *>(sender());
            peer->deleteLater();
            RateController::instance()->removeSocket(peer);
            ConnectionManager::instance()->removeConnection(peer);
        }

        void TorrentServer::processInfoHash(const QByteArray &infoHash)
        {
            PeerWireClient *peer = qobject_cast<PeerWireClient *>(sender());
            foreach (TorrentClient *client, clients) {
                if (client->state() >= TorrentClient::Searching && client->infoHash() == infoHash) {
                    peer->disconnect(peer, 0, this, 0);
                    client->setupIncomingConnection(peer);
                    return;
                }
            }
            removeClient();
        }