cd46a2daf5
Update files using old header.LGPL3 to header.LGPL Update files using old FDL template to use new one Update files using old BSD template to use new one Change-Id: I36a78272516f9953d02956522f285b40adfc8915 Reviewed-by: Lars Knoll <lars.knoll@theqtcompany.com>
253 lines
8.6 KiB
C++
253 lines
8.6 KiB
C++
/****************************************************************************
|
|
**
|
|
** Copyright (C) 2016 The Qt Company Ltd.
|
|
** Contact: https://www.qt.io/licensing/
|
|
**
|
|
** This file is part of the examples of the Qt Toolkit.
|
|
**
|
|
** $QT_BEGIN_LICENSE:BSD$
|
|
** Commercial License Usage
|
|
** Licensees holding valid commercial Qt licenses may use this file in
|
|
** accordance with the commercial license agreement provided with the
|
|
** Software or, alternatively, in accordance with the terms contained in
|
|
** a written agreement between you and The Qt Company. For licensing terms
|
|
** and conditions see https://www.qt.io/terms-conditions. For further
|
|
** information use the contact form at https://www.qt.io/contact-us.
|
|
**
|
|
** BSD License Usage
|
|
** Alternatively, you may use this file under the terms of the BSD license
|
|
** as follows:
|
|
**
|
|
** "Redistribution and use in source and binary forms, with or without
|
|
** modification, are permitted provided that the following conditions are
|
|
** met:
|
|
** * Redistributions of source code must retain the above copyright
|
|
** notice, this list of conditions and the following disclaimer.
|
|
** * Redistributions in binary form must reproduce the above copyright
|
|
** notice, this list of conditions and the following disclaimer in
|
|
** the documentation and/or other materials provided with the
|
|
** distribution.
|
|
** * Neither the name of The Qt Company Ltd nor the names of its
|
|
** contributors may be used to endorse or promote products derived
|
|
** from this software without specific prior written permission.
|
|
**
|
|
**
|
|
** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
|
|
**
|
|
** $QT_END_LICENSE$
|
|
**
|
|
****************************************************************************/
|
|
|
|
#include "bencodeparser.h"
|
|
#include "connectionmanager.h"
|
|
#include "torrentclient.h"
|
|
#include "torrentserver.h"
|
|
#include "trackerclient.h"
|
|
|
|
#include <QtCore>
|
|
#include <QNetworkRequest>
|
|
|
|
TrackerClient::TrackerClient(TorrentClient *downloader, QObject *parent)
|
|
: QObject(parent), torrentDownloader(downloader)
|
|
{
|
|
length = 0;
|
|
requestInterval = 5 * 60;
|
|
requestIntervalTimer = -1;
|
|
firstTrackerRequest = true;
|
|
lastTrackerRequest = false;
|
|
firstSeeding = true;
|
|
|
|
connect(&http, SIGNAL(finished(QNetworkReply*)), this, SLOT(httpRequestDone(QNetworkReply*)));
|
|
}
|
|
|
|
void TrackerClient::start(const MetaInfo &info)
|
|
{
|
|
metaInfo = info;
|
|
QTimer::singleShot(0, this, SLOT(fetchPeerList()));
|
|
|
|
if (metaInfo.fileForm() == MetaInfo::SingleFileForm) {
|
|
length = metaInfo.singleFile().length;
|
|
} else {
|
|
QList<MetaInfoMultiFile> files = metaInfo.multiFiles();
|
|
for (int i = 0; i < files.size(); ++i)
|
|
length += files.at(i).length;
|
|
}
|
|
}
|
|
|
|
void TrackerClient::startSeeding()
|
|
{
|
|
firstSeeding = true;
|
|
fetchPeerList();
|
|
}
|
|
|
|
void TrackerClient::stop()
|
|
{
|
|
lastTrackerRequest = true;
|
|
fetchPeerList();
|
|
}
|
|
|
|
void TrackerClient::timerEvent(QTimerEvent *event)
|
|
{
|
|
if (event->timerId() == requestIntervalTimer) {
|
|
fetchPeerList();
|
|
} else {
|
|
QObject::timerEvent(event);
|
|
}
|
|
}
|
|
|
|
void TrackerClient::fetchPeerList()
|
|
{
|
|
QUrl url(metaInfo.announceUrl());
|
|
|
|
// Base the query on announce url to include a passkey (if any)
|
|
QUrlQuery query(url);
|
|
|
|
// Percent encode the hash
|
|
QByteArray infoHash = torrentDownloader->infoHash();
|
|
QByteArray encodedSum;
|
|
for (int i = 0; i < infoHash.size(); ++i) {
|
|
encodedSum += '%';
|
|
encodedSum += QByteArray::number(infoHash[i], 16).right(2).rightJustified(2, '0');
|
|
}
|
|
|
|
bool seeding = (torrentDownloader->state() == TorrentClient::Seeding);
|
|
|
|
query.addQueryItem("info_hash", encodedSum);
|
|
query.addQueryItem("peer_id", ConnectionManager::instance()->clientId());
|
|
query.addQueryItem("port", QByteArray::number(TorrentServer::instance()->serverPort()));
|
|
query.addQueryItem("compact", "1");
|
|
query.addQueryItem("uploaded", QByteArray::number(torrentDownloader->uploadedBytes()));
|
|
|
|
if (!firstSeeding) {
|
|
query.addQueryItem("downloaded", "0");
|
|
query.addQueryItem("left", "0");
|
|
} else {
|
|
query.addQueryItem("downloaded",
|
|
QByteArray::number(torrentDownloader->downloadedBytes()));
|
|
int left = qMax<int>(0, metaInfo.totalSize() - torrentDownloader->downloadedBytes());
|
|
query.addQueryItem("left", QByteArray::number(seeding ? 0 : left));
|
|
}
|
|
|
|
if (seeding && firstSeeding) {
|
|
query.addQueryItem("event", "completed");
|
|
firstSeeding = false;
|
|
} else if (firstTrackerRequest) {
|
|
firstTrackerRequest = false;
|
|
query.addQueryItem("event", "started");
|
|
} else if(lastTrackerRequest) {
|
|
query.addQueryItem("event", "stopped");
|
|
}
|
|
|
|
if (!trackerId.isEmpty())
|
|
query.addQueryItem("trackerid", trackerId);
|
|
|
|
url.setQuery(query);
|
|
|
|
QNetworkRequest req(url);
|
|
if (!url.userName().isEmpty()) {
|
|
uname = url.userName();
|
|
pwd = url.password();
|
|
connect(&http, SIGNAL(authenticationRequired(QNetworkReply*,QAuthenticator*)),
|
|
this, SLOT(provideAuthentication(QNetworkReply*,QAuthenticator*)));
|
|
}
|
|
http.get(req);
|
|
}
|
|
|
|
void TrackerClient::httpRequestDone(QNetworkReply *reply)
|
|
{
|
|
reply->deleteLater();
|
|
if (lastTrackerRequest) {
|
|
emit stopped();
|
|
return;
|
|
}
|
|
|
|
if (reply->error() != QNetworkReply::NoError) {
|
|
emit connectionError(reply->error());
|
|
return;
|
|
}
|
|
|
|
QByteArray response = reply->readAll();
|
|
reply->abort();
|
|
|
|
BencodeParser parser;
|
|
if (!parser.parse(response)) {
|
|
qWarning("Error parsing bencode response from tracker: %s",
|
|
qPrintable(parser.errorString()));
|
|
return;
|
|
}
|
|
|
|
QMap<QByteArray, QVariant> dict = parser.dictionary();
|
|
|
|
if (dict.contains("failure reason")) {
|
|
// no other items are present
|
|
emit failure(QString::fromUtf8(dict.value("failure reason").toByteArray()));
|
|
return;
|
|
}
|
|
|
|
if (dict.contains("warning message")) {
|
|
// continue processing
|
|
emit warning(QString::fromUtf8(dict.value("warning message").toByteArray()));
|
|
}
|
|
|
|
if (dict.contains("tracker id")) {
|
|
// store it
|
|
trackerId = dict.value("tracker id").toByteArray();
|
|
}
|
|
|
|
if (dict.contains("interval")) {
|
|
// Mandatory item
|
|
if (requestIntervalTimer != -1)
|
|
killTimer(requestIntervalTimer);
|
|
requestIntervalTimer = startTimer(dict.value("interval").toInt() * 1000);
|
|
}
|
|
|
|
if (dict.contains("peers")) {
|
|
// store it
|
|
peers.clear();
|
|
QVariant peerEntry = dict.value("peers");
|
|
if (peerEntry.type() == QVariant::List) {
|
|
QList<QVariant> peerTmp = peerEntry.toList();
|
|
for (int i = 0; i < peerTmp.size(); ++i) {
|
|
TorrentPeer tmp;
|
|
QMap<QByteArray, QVariant> peer = qvariant_cast<QMap<QByteArray, QVariant> >(peerTmp.at(i));
|
|
tmp.id = QString::fromUtf8(peer.value("peer id").toByteArray());
|
|
tmp.address.setAddress(QString::fromUtf8(peer.value("ip").toByteArray()));
|
|
tmp.port = peer.value("port").toInt();
|
|
peers << tmp;
|
|
}
|
|
} else {
|
|
QByteArray peerTmp = peerEntry.toByteArray();
|
|
for (int i = 0; i < peerTmp.size(); i += 6) {
|
|
TorrentPeer tmp;
|
|
uchar *data = (uchar *)peerTmp.constData() + i;
|
|
tmp.port = (int(data[4]) << 8) + data[5];
|
|
uint ipAddress = 0;
|
|
ipAddress += uint(data[0]) << 24;
|
|
ipAddress += uint(data[1]) << 16;
|
|
ipAddress += uint(data[2]) << 8;
|
|
ipAddress += uint(data[3]);
|
|
tmp.address.setAddress(ipAddress);
|
|
peers << tmp;
|
|
}
|
|
}
|
|
emit peerListUpdated(peers);
|
|
}
|
|
}
|
|
|
|
void TrackerClient::provideAuthentication(QNetworkReply *reply, QAuthenticator *auth)
|
|
{
|
|
Q_UNUSED(reply);
|
|
auth->setUser(uname);
|
|
auth->setPassword(pwd);
|
|
}
|