QrtExchanger.cpp

Go to the documentation of this file.

/*

Copyright (C) 2005-2007 by Peter Dimov.

This file is part of Calitko (http://www.calitko.org).

Calitko 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.

Calitko 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 Calitko; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

*/

#include "Qt.h"
#include "QrtExchanger.h"
#include "QueryRoutingTable.h"
#include "Imports.cpp"

using namespace Gnutella::PacketProcessing::QueryRouting;

namespace Gnutella {
namespace PacketProcessing {
namespace QueryRouting {

enum Constants
{
    DefaultQrtSize  = 65536
};

class QrtExchangerPrivate
{
    REFERENCE_OBJECT (QrtExchangerPrivate)

public:
    typedef QList <PacketSession *> Sessions;

    Sessions            leafSessions;
    Sessions            peerSessions;
    QueryRoutingTable   ownTable;
    QueryRoutingTable   combinedTable;

    QrtExchangerPrivate()
     :  leafSessions(), peerSessions(), ownTable(), combinedTable()
    {}
};

} // QueryRouting
} // PacketProcessing
} // Gnutella

QrtExchanger::QrtExchanger()
 :  p (new QrtExchangerPrivate)
{
    p->ownTable.reset (DefaultQrtSize);
    p->combinedTable.reset (DefaultQrtSize);
}

QrtExchanger::~QrtExchanger()
{
    delete p;
}

void QrtExchanger::addSession (PacketSession *session)
{
    NodeType nodeType = session->nodeInfo().type;
    if (nodeType == Gnutella::TypePeer)
        p->peerSessions.append (session);
    else if (nodeType == Gnutella::TypeLeaf)
        p->leafSessions.append (session);
    else if (nodeType == Gnutella::TypeUltrapeer)
        ; // \todo p->ultrapeerSessions.append (session);
    else
        Q_ASSERT (false);
}

void QrtExchanger::removeSession (PacketSession *session)
{
    NodeType nodeType = session->nodeInfo().type;
    if (nodeType == Gnutella::TypePeer)
        p->peerSessions.removeAll (session);
    else if (nodeType == Gnutella::TypeLeaf)
        p->leafSessions.removeAll (session);
    else if (nodeType == Gnutella::TypeUltrapeer)
        ; // \todo p->leafSessions.removeAll (session);
    else
        Q_ASSERT (false);
}

void QrtExchanger::qrtRead (PacketSession *session)
{
    NodeType nodeType = session->nodeInfo().type;
    if (nodeType == TypeLeaf) {
        p->combinedTable = p->ownTable;
        foreach (PacketSession *leafSession, p->leafSessions)
            p->combinedTable.addTable (leafSession->queryRoutingTable());
        foreach (PacketSession *peerSession, p->peerSessions)
            peerSession->sendQrt (p->combinedTable);
    }
}

Copyright © 2006 Peter Dimov.
Powered by Doxygen.