squidy/qBittorrent
1
0
Fork 0
mirror of https://github.com/qbittorrent/qBittorrent.git synced 2025-12-22 00:17:23 -06:00
Code Issues Packages Projects Releases Wiki Activity

Improve embedded tracker

Browse Source 
Now it conforms to BEPs more closely.
This commit is contained in:
Chocobo1
2019-08-08 23:21:07 +08:00
parent 94f7a095bb
commit 8d0d8e4dcb
10 changed files with 423 additions and 230 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
src/base/bittorrent/tracker.h
View File

@@ -1,5 +1,6 @@
/*
* Bittorrent Client using Qt and libtorrent.
* Copyright (C) 2019 Mike Tzou (Chocobo1)
* Copyright (C) 2015 Vladimir Golovnev <glassez@yandex.ru>
* Copyright (C) 2010 Christophe Dumez <chris@qbittorrent.org>
*
@@ -30,12 +31,15 @@
#ifndef BITTORRENT_TRACKER_H
#define BITTORRENT_TRACKER_H
#include <libtorrent/fwd.hpp>
#include <string>
#include <libtorrent/entry.hpp>
#include <QHash>
#include <QObject>
#include <QHostAddress>
#include <QSet>
#include "base/bittorrent/infohash.h"
#include "base/http/irequesthandler.h"
#include "base/http/responsebuilder.h"
@@ -48,55 +52,58 @@ namespace BitTorrent
{
struct Peer
{
QHostAddress ip;
QByteArray peerId;
int port;
ushort port = 0; // self-claimed by peer, might not be the same as socket port
bool isSeeder = false;
QString uid() const;
lt::entry toEntry(bool noPeerId) const;
// caching precomputed values
lt::entry::string_type address;
lt::entry::string_type endpoint;
QByteArray uniqueID() const;
};
bool operator==(const Peer &left, const Peer &right);
bool operator!=(const Peer &left, const Peer &right);
uint qHash(const Peer &key, uint seed);
struct TrackerAnnounceRequest
{
QByteArray infoHash;
QString event;
int numwant;
Peer peer;
// Extensions
bool noPeerId;
};
typedef QHash<QString, Peer> PeerList;
typedef QHash<QByteArray, PeerList> TorrentList;
/* Basic Bittorrent tracker implementation in Qt */
/* Following http://wiki.theory.org/BitTorrent_Tracker_Protocol */
// *Basic* Bittorrent tracker implementation
// [BEP-3] The BitTorrent Protocol Specification
// also see: https://wiki.theory.org/index.php/BitTorrentSpecification#Tracker_HTTP.2FHTTPS_Protocol
class Tracker : public QObject, public Http::IRequestHandler, private Http::ResponseBuilder
{
Q_OBJECT
Q_DISABLE_COPY(Tracker)
struct TrackerAnnounceRequest;
struct TorrentStats
{
qint64 seeders = 0;
QSet<Peer> peers;
void setPeer(const Peer &peer);
bool removePeer(const Peer &peer);
};
public:
explicit Tracker(QObject *parent = nullptr);
~Tracker();
bool start();
Http::Response processRequest(const Http::Request &request, const Http::Environment &env) override;
private:
void respondToAnnounceRequest();
Http::Response processRequest(const Http::Request &request, const Http::Environment &env) override;
void processAnnounceRequest();
void registerPeer(const TrackerAnnounceRequest &announceReq);
void unregisterPeer(const TrackerAnnounceRequest &announceReq);
void replyWithPeerList(const TrackerAnnounceRequest &announceReq);
void prepareAnnounceResponse(const TrackerAnnounceRequest &announceReq);
Http::Server *m_server;
TorrentList m_torrents;
Http::Request m_request;
Http::Environment m_env;
QHash<InfoHash, TorrentStats> m_torrents;
};
}