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

using namespace Gnutella::Packets;

enum Constants
{
    MinimalPayloadLength = 3,   
    DefaultQueryTtl = 4         
};

Query::Query (const QByteArray &rawHeader, const QByteArray &rawPayload)
    : Packet (rawHeader, rawPayload)
{
    parse();
}

Query::Query (const QString &searchCriteria, quint16 minimumSpeed)
    : Packet (QueryDescriptor)
{
    setTtl (DefaultQueryTtl);   // May become obsolete with dynamic quering.
    setSearchCriteria (searchCriteria);
    setMinimumSpeed (minimumSpeed);
}

Query::~Query()
{
}

bool Query::prepareReadPayload (const QByteArray &rawPayload)
{
    int payloadLength = rawPayload.length();
    if (payloadLength < MinimalPayloadLength)
        return false;

    int i = 0;
    for (i = 2; i < payloadLength; i++)
        if (rawPayload.at (i) == 0)
            break;
    if (i == payloadLength) // No '\0' was found.
        return false;

    p.rawSearchCriteria.resize (i + 1 - 2); // include the zero, but not the min speed field.

    // Everything after the '\0' is query data.
    int rawQueryDataSize = payloadLength - i - 1;
    // Some servents may terminate the extension block with a zero.
    if (rawPayload.at (payloadLength - 1) == 0)
        rawQueryDataSize--;

    QByteArray rawQueryData = QByteArray::fromRawData (rawPayload.data() + i + 1, rawQueryDataSize);
    p.queryData.prepareRead (rawQueryData);
    return true;
}

void Query::readPayload (QDataStream &stream)
{
    stream.setByteOrder (QDataStream::BigEndian);
    stream >> p.minimumSpeed;
    stream.setByteOrder (QDataStream::LittleEndian);

    stream.readRawData (p.rawSearchCriteria.data(), p.rawSearchCriteria.size());
    p.searchCriteria = QString::fromLatin1 (p.rawSearchCriteria);
    p.rawSearchCriteria = QByteArray();

    p.queryData.read (stream);
}

int Query::prepareWritePayload() const
{
    p.rawSearchCriteria = p.searchCriteria.toLatin1(); // The latin1 encoded string is NOT zero-terminated.
    return 2 + p.rawSearchCriteria.length() + 1 + p.queryData.prepareWrite();
    // \todo Limewire puts an additional zero after all extensions, is such are there.
}

void Query::writePayload (QDataStream &stream) const
{
    stream.setByteOrder (QDataStream::BigEndian);
    stream << p.minimumSpeed;
    stream.setByteOrder (QDataStream::LittleEndian);

    stream.writeRawData (p.rawSearchCriteria.data(), p.rawSearchCriteria.length());
    stream << uchar (0);
    p.queryData.write (stream);

    p.rawSearchCriteria = QByteArray();
}

quint16 Query::minimumSpeed() const
{
    return p.minimumSpeed;
}

QString Query::searchCriteria() const
{
    return p.searchCriteria;
}

/*Query::QueryData Query::queryData() const
{
    return p.queryData;
}*/

void Query::setMinimumSpeed (const quint16 &minimumSpeed)
{
    p.minimumSpeed = minimumSpeed;
    invalidatePayload();
}

void Query::setSearchCriteria (const QString &searchCriteria)
{
    p.searchCriteria = searchCriteria;
    invalidatePayload();
}

Copyright © 2006 Peter Dimov.
Powered by Doxygen.