Pong.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 "Pong.h"

using namespace Gnutella::Packets;

enum Constants
{
    MinimalPayloadLength = 14   
};


Pong::Pong(quint16 port, const QHostAddress &ipAddress, quint32 sharedFiles, quint32 sharedKilobytes)
    : Packet (PongDescriptor)
{
    d = new Data;

    d->ref.init (1);

    d->port = port;
    d->ipAddress = ipAddress;
    d->sharedFiles = sharedFiles;
    d->sharedKilobytes = sharedKilobytes;
}

Pong::Pong(const QByteArray &rawHeader, const QByteArray &rawPayload)
    : Packet (rawHeader, rawPayload)
{
    d = new Data;

    d->ref.init (1);

    parse();
}

Pong::~Pong()
{
    if (!d->ref.deref())
        delete d;
}


Pong & Pong::operator= (const Pong &other)
{
    Data *x = other.d;
    x->ref.ref();
    x = qAtomicSetPtr(&d, x);
    if (!x->ref.deref())
        delete x;

    Packet::operator= (other);
    return *this;
}

void Pong::invalidatePayload()
{
    if (d->ref != 1) {
        Data *x = new Data;
        x->ref.init (1);

        x->port                 = d->port;
        x->ipAddress            = d->ipAddress;
        x->sharedFiles          = d->sharedFiles;
        x->sharedKilobytes      = d->sharedKilobytes;
        x->ggepBlock            = d->ggepBlock;

        x = qAtomicSetPtr (&d, x);
        if (!x->ref.deref())
            delete x;
    };
    Packet::invalidatePayload();
}

bool Pong::prepareReadPayload (const QByteArray &rawPayload)
{
    if (payloadLength() < MinimalPayloadLength)
        return false;

    const char *extensionRawData = rawPayload.constData() + MinimalPayloadLength;
    int extensionSize = rawPayload.length() - MinimalPayloadLength;
    QByteArray extensionData = QByteArray::fromRawData (extensionRawData, extensionSize);

    d->ggepBlock.prepareRead (extensionData);
    return true;
}

void Pong::readPayload (QDataStream &stream)
{
    stream >> d->port;
    Gnutella::Packets::operator>> (stream, d->ipAddress);
    stream >> d->sharedFiles;
    stream >> d->sharedKilobytes;
    stream >> d->ggepBlock;
}

int Pong::prepareWritePayload() const
{
    return MinimalPayloadLength + d->ggepBlock.prepareWrite();
}

void Pong::writePayload (QDataStream &stream) const
{
    stream << d->port;
    Gnutella::Packets::operator<< (stream, d->ipAddress);
    stream << d->sharedFiles;
    stream << d->sharedKilobytes;
    stream << d->ggepBlock;
}

Copyright © 2006 Peter Dimov.
Powered by Doxygen.