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.
356 lines
8.9 KiB
356 lines
8.9 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 <ksocketaddress.h>
|
|
#include <util/file.h>
|
|
#include <util/log.h>
|
|
#include <util/functions.h>
|
|
#include <netinet/in.h>
|
|
#include "kbucket.h"
|
|
#include "kclosestnodessearch.h"
|
|
#include "rpcserver.h"
|
|
#include "node.h"
|
|
|
|
using namespace KNetwork;
|
|
using namespace bt;
|
|
|
|
namespace dht
|
|
{
|
|
KBucketEntry::KBucketEntry()
|
|
{
|
|
last_responded = bt::GetCurrentTime();
|
|
failed_queries = 0;
|
|
questionable_pings = 0;
|
|
}
|
|
|
|
KBucketEntry::KBucketEntry(const KInetSocketAddress & addr,const Key & id)
|
|
: addr(addr),node_id(id)
|
|
{
|
|
last_responded = bt::GetCurrentTime();
|
|
failed_queries = 0;
|
|
questionable_pings = 0;
|
|
}
|
|
|
|
KBucketEntry::KBucketEntry(const KBucketEntry & other)
|
|
: addr(other.addr),node_id(other.node_id),
|
|
last_responded(other.last_responded),failed_queries(other.failed_queries),questionable_pings(other.questionable_pings)
|
|
{}
|
|
|
|
|
|
KBucketEntry::~KBucketEntry()
|
|
{}
|
|
|
|
KBucketEntry & KBucketEntry::operator = (const KBucketEntry & other)
|
|
{
|
|
addr = other.addr;
|
|
node_id = other.node_id;
|
|
last_responded = other.last_responded;
|
|
failed_queries = other.failed_queries;
|
|
questionable_pings = other.questionable_pings;
|
|
return *this;
|
|
}
|
|
|
|
bool KBucketEntry::operator == (const KBucketEntry & entry) const
|
|
{
|
|
return addr == entry.addr && node_id == entry.node_id;
|
|
}
|
|
|
|
bool KBucketEntry::isGood() const
|
|
{
|
|
if (bt::GetCurrentTime() - last_responded > 15 * 60 * 1000)
|
|
return false;
|
|
else
|
|
return true;
|
|
}
|
|
|
|
bool KBucketEntry::isQuestionable() const
|
|
{
|
|
if (bt::GetCurrentTime() - last_responded > 15 * 60 * 1000)
|
|
return true;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
|
|
bool KBucketEntry::isBad() const
|
|
{
|
|
if (isGood())
|
|
return false;
|
|
|
|
return failed_queries > 2 || questionable_pings > 2;
|
|
}
|
|
|
|
void KBucketEntry::hasResponded()
|
|
{
|
|
last_responded = bt::GetCurrentTime();
|
|
failed_queries = 0; // reset failed queries
|
|
questionable_pings = 0;
|
|
}
|
|
|
|
|
|
//////////////////////////////////////////////////////////
|
|
|
|
KBucket::KBucket(Uint32 idx,RPCServer* srv,Node* node)
|
|
: idx(idx),srv(srv),node(node)
|
|
{
|
|
last_modified = bt::GetCurrentTime();
|
|
refresh_task = 0;
|
|
}
|
|
|
|
|
|
KBucket::~KBucket()
|
|
{}
|
|
|
|
void KBucket::insert(const KBucketEntry & entry)
|
|
{
|
|
TQValueList<KBucketEntry>::iterator i = entries.tqfind(entry);
|
|
|
|
// If in the list, move it to the end
|
|
if (i != entries.end())
|
|
{
|
|
KBucketEntry & e = *i;
|
|
e.hasResponded();
|
|
last_modified = bt::GetCurrentTime();
|
|
entries.remove(i);
|
|
entries.append(entry);
|
|
return;
|
|
}
|
|
|
|
// insert if not already in the list and we still have room
|
|
if (i == entries.end() && entries.count() < dht::K)
|
|
{
|
|
entries.append(entry);
|
|
last_modified = bt::GetCurrentTime();
|
|
}
|
|
else if (!replaceBadEntry(entry))
|
|
{
|
|
// ping questionable nodes when replacing a bad one fails
|
|
pingQuestionable(entry);
|
|
}
|
|
}
|
|
|
|
void KBucket::onResponse(RPCCall* c,MsgBase* rsp)
|
|
{
|
|
last_modified = bt::GetCurrentTime();
|
|
|
|
if (!pending_entries_busy_pinging.tqcontains(c))
|
|
return;
|
|
|
|
KBucketEntry entry = pending_entries_busy_pinging[c];
|
|
pending_entries_busy_pinging.erase(c); // call is done so erase it
|
|
|
|
// we have a response so try to find the next bad or questionable node
|
|
// if we do not have room see if we can get rid of some bad peers
|
|
if (!replaceBadEntry(entry)) // if no bad peers ping a questionable one
|
|
pingQuestionable(entry);
|
|
|
|
}
|
|
|
|
|
|
|
|
void KBucket::onTimeout(RPCCall* c)
|
|
{
|
|
if (!pending_entries_busy_pinging.tqcontains(c))
|
|
return;
|
|
|
|
KBucketEntry entry = pending_entries_busy_pinging[c];
|
|
|
|
// replace the entry which timed out
|
|
TQValueList<KBucketEntry>::iterator i;
|
|
for (i = entries.begin();i != entries.end();i++)
|
|
{
|
|
KBucketEntry & e = *i;
|
|
if (e.getAddress() == c->getRequest()->getOrigin())
|
|
{
|
|
last_modified = bt::GetCurrentTime();
|
|
entries.remove(i);
|
|
entries.append(entry);
|
|
break;
|
|
}
|
|
}
|
|
pending_entries_busy_pinging.erase(c); // call is done so erase it
|
|
// see if we can do another pending entry
|
|
if (pending_entries_busy_pinging.count() < 2 && pending_entries.count() > 0)
|
|
{
|
|
KBucketEntry pe = pending_entries.front();
|
|
pending_entries.pop_front();
|
|
if (!replaceBadEntry(pe)) // if no bad peers ping a questionable one
|
|
pingQuestionable(pe);
|
|
}
|
|
}
|
|
|
|
void KBucket::pingQuestionable(const KBucketEntry & replacement_entry)
|
|
{
|
|
if (pending_entries_busy_pinging.count() >= 2)
|
|
{
|
|
pending_entries.append(replacement_entry); // lets not have to many pending_entries calls going on
|
|
return;
|
|
}
|
|
|
|
TQValueList<KBucketEntry>::iterator i;
|
|
// we haven't found any bad ones so try the questionable ones
|
|
for (i = entries.begin();i != entries.end();i++)
|
|
{
|
|
KBucketEntry & e = *i;
|
|
if (e.isQuestionable())
|
|
{
|
|
Out(SYS_DHT|LOG_DEBUG) << "Pinging questionable node : " << e.getAddress().toString() << endl;
|
|
PingReq* p = new PingReq(node->getOurID());
|
|
p->setDestination(e.getAddress());
|
|
RPCCall* c = srv->doCall(p);
|
|
if (c)
|
|
{
|
|
e.onPingQuestionable();
|
|
c->addListener(this);
|
|
// add the pending entry
|
|
pending_entries_busy_pinging.insert(c,replacement_entry);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
bool KBucket::replaceBadEntry(const KBucketEntry & entry)
|
|
{
|
|
TQValueList<KBucketEntry>::iterator i;
|
|
for (i = entries.begin();i != entries.end();i++)
|
|
{
|
|
KBucketEntry & e = *i;
|
|
if (e.isBad())
|
|
{
|
|
// bad one get rid of it
|
|
last_modified = bt::GetCurrentTime();
|
|
entries.remove(i);
|
|
entries.append(entry);
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool KBucket::tqcontains(const KBucketEntry & entry) const
|
|
{
|
|
return entries.tqcontains(entry);
|
|
}
|
|
|
|
void KBucket::findKClosestNodes(KClosestNodesSearch & kns)
|
|
{
|
|
TQValueList<KBucketEntry>::iterator i = entries.begin();
|
|
while (i != entries.end())
|
|
{
|
|
kns.tryInsert(*i);
|
|
i++;
|
|
}
|
|
}
|
|
|
|
bool KBucket::onTimeout(const KInetSocketAddress & addr)
|
|
{
|
|
TQValueList<KBucketEntry>::iterator i;
|
|
|
|
for (i = entries.begin();i != entries.end();i++)
|
|
{
|
|
KBucketEntry & e = *i;
|
|
if (e.getAddress() == addr)
|
|
{
|
|
e.requestTimeout();
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool KBucket::needsToBeRefreshed() const
|
|
{
|
|
bt::TimeStamp now = bt::GetCurrentTime();
|
|
if (last_modified > now)
|
|
{
|
|
last_modified = now;
|
|
return false;
|
|
}
|
|
|
|
return !refresh_task && entries.count() > 0 && (now - last_modified > BUCKET_REFRESH_INTERVAL);
|
|
}
|
|
|
|
void KBucket::updateRefreshTimer()
|
|
{
|
|
last_modified = bt::GetCurrentTime();
|
|
}
|
|
|
|
|
|
|
|
void KBucket::save(bt::File & fptr)
|
|
{
|
|
BucketHeader hdr;
|
|
hdr.magic = BUCKET_MAGIC_NUMBER;
|
|
hdr.index = idx;
|
|
hdr.num_entries = entries.count();
|
|
|
|
fptr.write(&hdr,sizeof(BucketHeader));
|
|
TQValueList<KBucketEntry>::iterator i;
|
|
for (i = entries.begin();i != entries.end();i++)
|
|
{
|
|
KBucketEntry & e = *i;
|
|
const KIpAddress & ip = e.getAddress().ipAddress();
|
|
Uint8 tmp[26];
|
|
bt::WriteUint32(tmp,0,ip.IPv4Addr());
|
|
bt::WriteUint16(tmp,4,e.getAddress().port());
|
|
memcpy(tmp+6,e.getID().getData(),20);
|
|
fptr.write(tmp,26);
|
|
}
|
|
}
|
|
|
|
void KBucket::load(bt::File & fptr,const BucketHeader & hdr)
|
|
{
|
|
if (hdr.num_entries > K)
|
|
return;
|
|
|
|
for (Uint32 i = 0;i < hdr.num_entries;i++)
|
|
{
|
|
Uint8 tmp[26];
|
|
if (fptr.read(tmp,26) != 26)
|
|
return;
|
|
|
|
entries.append(KBucketEntry(
|
|
KInetSocketAddress(
|
|
KIpAddress(bt::ReadUint32(tmp,0)),
|
|
bt::ReadUint16(tmp,4)),
|
|
dht::Key(tmp+6)));
|
|
}
|
|
}
|
|
|
|
void KBucket::onFinished(Task* t)
|
|
{
|
|
if (t == refresh_task)
|
|
refresh_task = 0;
|
|
}
|
|
|
|
void KBucket::setRefreshTask(Task* t)
|
|
{
|
|
refresh_task = t;
|
|
if (refresh_task)
|
|
{
|
|
connect(refresh_task,TQT_SIGNAL(finished( Task* )),
|
|
this,TQT_SLOT(onFinished( Task* )));
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
#include "kbucket.moc"
|