/**************************************************************************** ** ** Copyright (C) 2004-2008 Trolltech ASA. All rights reserved. ** ** This file is part of the example classes of the Qt Toolkit. ** ** This file may be used under the terms of the GNU General Public ** License versions 2.0 or 3.0 as published by the Free Software ** Foundation and appearing in the files LICENSE.GPL2 and LICENSE.GPL3 ** included in the packaging of this file. Alternatively you may (at ** your option) use any later version of the GNU General Public ** License if such license has been publicly approved by Trolltech ASA ** (or its successors, if any) and the KDE Free Qt Foundation. In ** addition, as a special exception, Trolltech gives you certain ** additional rights. These rights are described in the Trolltech GPL ** Exception version 1.2, which can be found at ** http://www.trolltech.com/products/qt/gplexception/ and in the file ** GPL_EXCEPTION.txt in this package. ** ** Please review the following information to ensure GNU General ** Public Licensing requirements will be met: ** http://trolltech.com/products/qt/licenses/licensing/opensource/. If ** you are unsure which license is appropriate for your use, please ** review the following information: ** http://trolltech.com/products/qt/licenses/licensing/licensingoverview ** or contact the sales department at sales@trolltech.com. ** ** In addition, as a special exception, Trolltech, as the sole ** copyright holder for Qt Designer, grants users of the Qt/Eclipse ** Integration plug-in the right for the Qt/Eclipse Integration to ** link to functionality provided by Qt Designer and its related ** libraries. ** ** This file is provided "AS IS" with NO WARRANTY OF ANY KIND, ** INCLUDING THE WARRANTIES OF DESIGN, MERCHANTABILITY AND FITNESS FOR ** A PARTICULAR PURPOSE. Trolltech reserves all rights not expressly ** granted herein. ** ** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE ** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. ** ****************************************************************************/ #ifndef TORRENTCLIENT_H #define TORRENTCLIENT_H #include <QBitArray> #include <QHostAddress> #include <QList> class MetaInfo; class PeerWireClient; class TorrentClientPrivate; class TorrentPeer; class TorrentPiece; QT_BEGIN_NAMESPACE class QTimerEvent; QT_END_NAMESPACE class TorrentPeer { public: QHostAddress address; quint16 port; QString id; bool interesting; bool seed; uint lastVisited; uint connectStart; uint connectTime; QBitArray pieces; int numCompletedPieces; inline bool operator==(const TorrentPeer &other) { return port == other.port && address == other.address && id == other.id; } }; class TorrentClient : public QObject { Q_OBJECT public: enum State { Idle, Paused, Stopping, Preparing, Searching, Connecting, WarmingUp, Downloading, Endgame, Seeding }; enum Error { UnknownError, TorrentParseError, InvalidTrackerError, FileError, ServerError }; TorrentClient(QObject *parent = 0); ~TorrentClient(); bool setTorrent(const QString &fileName); bool setTorrent(const QByteArray &torrentData); MetaInfo metaInfo() const; void setMaxConnections(int connections); int maxConnections() const; void setDestinationFolder(const QString &directory); QString destinationFolder() const; void setDumpedState(const QByteArray &dumpedState); QByteArray dumpedState() const; // Progress and stats for download feedback. qint64 progress() const; void setDownloadedBytes(qint64 bytes); qint64 downloadedBytes() const; void setUploadedBytes(qint64 bytes); qint64 uploadedBytes() const; int connectedPeerCount() const; int seedCount() const; // Accessors for the tracker QByteArray peerId() const; QByteArray infoHash() const; quint16 serverPort() const; // State and error. State state() const; QString stateString() const; Error error() const; QString errorString() const; signals: void stateChanged(TorrentClient::State state); void error(TorrentClient::Error error); void downloadCompleted(); void peerInfoUpdated(); void dataSent(int uploadedBytes); void dataReceived(int downloadedBytes); void progressUpdated(int percentProgress); void downloadRateUpdated(int bytesPerSecond); void uploadRateUpdated(int bytesPerSecond); void stopped(); public slots: void start(); void stop(); void setPaused(bool paused); void setupIncomingConnection(PeerWireClient *client); protected slots: void timerEvent(QTimerEvent *event); private slots: // File management void sendToPeer(int readId, int pieceIndex, int begin, const QByteArray &data); void fullVerificationDone(); void pieceVerified(int pieceIndex, bool ok); void handleFileError(); // Connection handling void connectToPeers(); QList<TorrentPeer *> weighedFreePeers() const; void setupOutgoingConnection(); void initializeConnection(PeerWireClient *client); void removeClient(); void peerPiecesAvailable(const QBitArray &pieces); void peerRequestsBlock(int pieceIndex, int begin, int length); void blockReceived(int pieceIndex, int begin, const QByteArray &data); void peerWireBytesWritten(qint64 bytes); void peerWireBytesReceived(qint64 bytes); int blocksLeftForPiece(const TorrentPiece *piece) const; // Scheduling void scheduleUploads(); void scheduleDownloads(); void schedulePieceForClient(PeerWireClient *client); void requestMore(PeerWireClient *client); int requestBlocks(PeerWireClient *client, TorrentPiece *piece, int maxBlocks); void peerChoked(); void peerUnchoked(); // Tracker handling void addToPeerList(const QList<TorrentPeer> &peerList); void trackerStopped(); // Progress void updateProgress(int progress = -1); private: TorrentClientPrivate *d; friend class TorrentClientPrivate; }; #endif