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.
155 lines
4.1 KiB
155 lines
4.1 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. *
|
|
***************************************************************************/
|
|
#include <kurl.h>
|
|
#include <tqhostaddress.h>
|
|
#include <util/log.h>
|
|
#include <util/functions.h>
|
|
#include <torrent/globals.h>
|
|
#include <torrent/server.h>
|
|
#include <torrent/peermanager.h>
|
|
#include <interfaces/torrentinterface.h>
|
|
#include "dhttrackerbackend.h"
|
|
#include "dht.h"
|
|
#include "announcetask.h"
|
|
|
|
using namespace bt;
|
|
|
|
namespace dht
|
|
{
|
|
|
|
DHTTrackerBackend::DHTTrackerBackend(DHTBase & dh_table,kt::TorrentInterface* tor)
|
|
: dh_table(dh_table),curr_task(0),tor(tor)
|
|
{
|
|
connect(&timer,TQ_SIGNAL(timeout()),this,TQ_SLOT(onTimeout()));
|
|
connect(&dh_table,TQ_SIGNAL(started()),this,TQ_SLOT(manualUpdate()));
|
|
connect(&dh_table,TQ_SIGNAL(stopped()),this,TQ_SLOT(dhtStopped()));
|
|
started = false;
|
|
}
|
|
|
|
|
|
DHTTrackerBackend::~DHTTrackerBackend()
|
|
{
|
|
if (curr_task)
|
|
curr_task->kill();
|
|
}
|
|
|
|
void DHTTrackerBackend::start()
|
|
{
|
|
started = true;
|
|
if (dh_table.isRunning())
|
|
doRequest();
|
|
}
|
|
|
|
void DHTTrackerBackend::dhtStopped()
|
|
{
|
|
stop(0);
|
|
curr_task = 0;
|
|
}
|
|
|
|
void DHTTrackerBackend::stop(bt::WaitJob*)
|
|
{
|
|
started = false;
|
|
if (curr_task)
|
|
{
|
|
curr_task->kill();
|
|
timer.stop();
|
|
}
|
|
}
|
|
|
|
void DHTTrackerBackend::manualUpdate()
|
|
{
|
|
if (dh_table.isRunning() && started)
|
|
doRequest();
|
|
}
|
|
|
|
|
|
bool DHTTrackerBackend::doRequest()
|
|
{
|
|
if (!dh_table.isRunning())
|
|
return false;
|
|
|
|
if (curr_task)
|
|
return true;
|
|
|
|
const SHA1Hash & info_hash = tor->getInfoHash();
|
|
Uint16 port = bt::Globals::instance().getServer().getPortInUse();
|
|
curr_task = dh_table.announce(info_hash,port);
|
|
if (curr_task)
|
|
{
|
|
for (Uint32 i = 0;i < tor->getNumDHTNodes();i++)
|
|
{
|
|
const kt::DHTNode & n = tor->getDHTNode(i);
|
|
curr_task->addDHTNode(n.ip,n.port);
|
|
}
|
|
connect(curr_task,TQ_SIGNAL(dataReady( Task* )),this,TQ_SLOT(onDataReady( Task* )));
|
|
connect(curr_task,TQ_SIGNAL(finished( Task* )),this,TQ_SLOT(onFinished( Task* )));
|
|
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void DHTTrackerBackend::onFinished(Task* t)
|
|
{
|
|
if (curr_task == t)
|
|
{
|
|
onDataReady(curr_task);
|
|
curr_task = 0;
|
|
// do another announce in 5 minutes or so
|
|
timer.start(5 * 60 * 1000,true);
|
|
}
|
|
}
|
|
|
|
void DHTTrackerBackend::onDataReady(Task* t)
|
|
{
|
|
if (curr_task == t)
|
|
{
|
|
Uint32 cnt = 0;
|
|
DBItem item;
|
|
while (curr_task->takeItem(item))
|
|
{
|
|
Uint16 port = bt::ReadUint16(item.getData(),4);
|
|
TQString ip = TQHostAddress(ReadUint32(item.getData(),0)).toString();
|
|
|
|
addPeer(ip,port);
|
|
cnt++;
|
|
}
|
|
|
|
if (cnt)
|
|
{
|
|
Out(SYS_DHT|LOG_NOTICE) <<
|
|
TQString("DHT: Got %1 potential peers for torrent %2")
|
|
.arg(cnt).arg(tor->getStats().torrent_name) << endl;
|
|
peersReady(this);
|
|
}
|
|
}
|
|
}
|
|
|
|
void DHTTrackerBackend::onTimeout()
|
|
{
|
|
if (dh_table.isRunning() && started)
|
|
doRequest();
|
|
}
|
|
|
|
}
|
|
|
|
#include "dhttrackerbackend.moc"
|