b3ce4470ae
Task-number: QTBUG-22750 Change-Id: I161fad772bfb26797e6ee9d69da925b6747c371f Reviewed-by: Lars Knoll <lars.knoll@nokia.com>
246 lines
8.2 KiB
C++
246 lines
8.2 KiB
C++
/****************************************************************************
|
|
**
|
|
** Copyright (C) 2012 Nokia Corporation and/or its subsidiary(-ies).
|
|
** All rights reserved.
|
|
** Contact: Nokia Corporation (qt-info@nokia.com)
|
|
**
|
|
** This file is part of the examples of the Qt Toolkit.
|
|
**
|
|
** $QT_BEGIN_LICENSE:BSD$
|
|
** 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 Nokia Corporation and its Subsidiary(-ies) 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()
|
|
{
|
|
// Prepare connection details
|
|
QString fullUrl = metaInfo.announceUrl();
|
|
QUrl url(fullUrl);
|
|
QString passkey = "?";
|
|
if (fullUrl.contains("?passkey")) {
|
|
passkey = metaInfo.announceUrl().mid(fullUrl.indexOf("?passkey"), -1);
|
|
passkey += '&';
|
|
}
|
|
|
|
// Percent encode the hash
|
|
QByteArray infoHash = torrentDownloader->infoHash();
|
|
QString encodedSum;
|
|
for (int i = 0; i < infoHash.size(); ++i) {
|
|
encodedSum += '%';
|
|
encodedSum += QString::number(infoHash[i], 16).right(2).rightJustified(2, '0');
|
|
}
|
|
|
|
bool seeding = (torrentDownloader->state() == TorrentClient::Seeding);
|
|
QByteArray query;
|
|
query += url.path().toLatin1();
|
|
query += passkey;
|
|
query += "info_hash=" + encodedSum;
|
|
query += "&peer_id=" + ConnectionManager::instance()->clientId();
|
|
query += "&port=" + QByteArray::number(TorrentServer::instance()->serverPort());
|
|
query += "&compact=1";
|
|
query += "&uploaded=" + QByteArray::number(torrentDownloader->uploadedBytes());
|
|
|
|
if (!firstSeeding) {
|
|
query += "&downloaded=0";
|
|
query += "&left=0";
|
|
} else {
|
|
query += "&downloaded=" + QByteArray::number(
|
|
torrentDownloader->downloadedBytes());
|
|
int left = qMax<int>(0, metaInfo.totalSize() - torrentDownloader->downloadedBytes());
|
|
query += "&left=" + QByteArray::number(seeding ? 0 : left);
|
|
}
|
|
|
|
if (seeding && firstSeeding) {
|
|
query += "&event=completed";
|
|
firstSeeding = false;
|
|
} else if (firstTrackerRequest) {
|
|
firstTrackerRequest = false;
|
|
query += "&event=started";
|
|
} else if(lastTrackerRequest) {
|
|
query += "&event=stopped";
|
|
}
|
|
|
|
if (!trackerId.isEmpty())
|
|
query += "&trackerid=" + trackerId;
|
|
|
|
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);
|
|
}
|