QrtWriter.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 "QrtWriter.h"
#include "QueryRoutingTable.h"
#include "Imports.cpp"

using namespace Gnutella::PacketProcessing::QueryRouting;

namespace Gnutella {
namespace PacketProcessing {
namespace QueryRouting {

enum Constants
{
    PatchSize = 1024    
};

class QrtWriterPrivate
{
    REFERENCE_OBJECT (QrtWriterPrivate)

public:
    PacketSession       *session;
    QueryRoutingTable   oldTable;
    quint8              patchEntryBits;

    QrtWriterPrivate() :    session (0), oldTable(), patchEntryBits (0) {}
};

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

QrtWriter::QrtWriter (PacketSession *session, quint8 patchEntryBits)
 :  d (new QrtWriterPrivate)
{
    d->session = session;
    d->patchEntryBits = patchEntryBits;
}

QrtWriter::~QrtWriter()
{
    delete d;
}

void QrtWriter::write (const QueryRoutingTable &newTable)
{
    if (d->oldTable.size() != newTable.size()) {
        QueryRoutingReset reset (newTable.size(), 2);
        d->session->sendPacket (reset);
        d->oldTable.reset (newTable.size());
    }
    QByteArray patchBuffer = d->oldTable.makePatchTo (newTable,
                                                      d->patchEntryBits);
    d->oldTable = newTable;
    patchBuffer = compress (patchBuffer);

    int patchSize = PatchSize;
    int seqSize = patchBuffer.size() / patchSize;
    if (patchBuffer.size() % patchSize != 0)
        seqSize++;

    char *patchStart = patchBuffer.data();
    char *patchEnd = patchStart + patchBuffer.size();
    int seqNo;
    for (seqNo = 0; seqNo < seqSize && patchStart + patchSize <= patchEnd;
        seqNo++, patchStart += patchSize) {
        QByteArray patchPiece (patchStart, patchSize);
        QueryRoutingPatch patch (seqNo, seqSize, QueryRoutingPatch::ZLIB,
                                 d->patchEntryBits, patchPiece);
        d->session->sendPacket (patch);
    }
    if (patchStart != patchEnd) {
        QByteArray patchPiece (patchStart, patchEnd - patchStart);
        QueryRoutingPatch patch (seqNo, seqSize, QueryRoutingPatch::ZLIB,
                                 d->patchEntryBits, patchPiece);
        d->session->sendPacket (patch);
    }
}

Copyright © 2006 Peter Dimov.
Powered by Doxygen.