You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ktorrent/libktorrent/torrent/peersourcemanager.h

184 lines
4.8 KiB

/***************************************************************************
* Copyright (C) 2005 by Joris Guisson *
* joris.guisson@gmail.com *
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program; if not, write to the *
* Free Software Foundation, Inc., *
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. *
***************************************************************************/
#ifndef BTPEERSOURCEMANAGER_H
#define BTPEERSOURCEMANAGER_H
#include <tqtimer.h>
#include <tqdatetime.h>
#include <tqptrlist.h>
#include <util/ptrmap.h>
#include <interfaces/trackerslist.h>
namespace kt
{
class PeerSource;
}
namespace dht
{
class DHTTrackerBackend;
}
namespace bt
{
class Tracker;
class PeerManager;
class Torrent;
class TorrentControl;
/**
* @author Joris Guisson <joris.guisson@gmail.com>
*
* This class manages all PeerSources.
*/
class PeerSourceManager : public TQObject, public kt::TrackersList
{
TQ_OBJECT
TorrentControl* tor;
PeerManager* pman;
PtrMap<KURL,Tracker> trackers;
TQPtrList<kt::PeerSource> additional;
Tracker* curr;
dht::DHTTrackerBackend* m_dht;
bool started;
bool pending;
KURL::List custom_trackers;
TQDateTime request_time;
TQTimer timer;
Uint32 failures;
bool no_save_custom_trackers;
public:
PeerSourceManager(TorrentControl* tor,PeerManager* pman);
virtual ~PeerSourceManager();
/**
* Add a PeerSource, the difference between PeerSource and Tracker
* is that only one Tracker can be used at the same time,
* PeerSource can always be used.
* @param ps The PeerSource
*/
void addPeerSource(kt::PeerSource* ps);
/**
* See if the PeerSourceManager has been started
*/
bool isStarted() const {return started;}
/**
* Start gathering peers
*/
void start();
/**
* Stop gathering peers
* @param wjob WaitJob to wait at exit for the completion of stopped events to the trackers
*/
void stop(WaitJob* wjob = 0);
/**
* Notify peersources and trackrs that the download is complete.
*/
void completed();
/**
* Do a manual update on all peer sources and trackers.
*/
void manualUpdate();
/**
* Remove a Tracker or PeerSource.
* @param ps
*/
void removePeerSource(kt::PeerSource* ps);
virtual KURL getTrackerURL() const;
virtual KURL::List getTrackerURLs();
virtual void addTracker(KURL url, bool custom = true,int tier = 1);
virtual bool removeTracker(KURL url);
virtual void setTracker(KURL url);
virtual void restoreDefault();
/**
* Get the time to the next tracker update.
* @return The time in seconds
*/
Uint32 getTimeToNextUpdate() const;
/// Get the number of potential seeders
Uint32 getNumSeeders() const;
/// Get the number of potential leechers
Uint32 getNumLeechers() const;
/// Get the number of failures
Uint32 getNumFailures() const {return failures;}
///Adds DHT as PeerSource for this torrent
void addDHT();
///Removes DHT from PeerSourceManager for this torrent.
void removeDHT();
///Checks if DHT is enabled
bool dhtStarted();
private slots:
/**
* The an error happened contacting the tracker.
* @param err The error
*/
void onTrackerError(const TQString & err);
/**
* Tracker update was OK.
* @param
*/
void onTrackerOK();
/**
* Tracker is doing a request.
*/
void onTrackerRequestPending();
/**
* Update the current tracker manually
*/
void updateCurrentManually();
signals:
/**
* Status has changed of the tracker.
* @param ns The new status
*/
void statusChanged(const TQString & ns);
private:
void saveCustomURLs();
void loadCustomURLs();
void addTracker(Tracker* trk);
void switchTracker(Tracker* trk);
Tracker* selectTracker();
};
}
#endif