Searcher.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 "Searcher.h"
#include "UIs/Searching/SearchModel.h"
#include "Gnutella/LocalPeer.h"
#include "Gnutella/Packets/QueryHits.h"
#include "Gnutella/Packets/Query.h"
#include "Gnutella/Packets/VendorMessages/QueryStatusRequest.h"
#include "Gnutella/Packets/VendorMessages/QueryStatusResponse.h"
#include "Gnutella/PacketProcessing/PacketProcessor.h"

using Gnutella::LocalPeer;
using Gnutella::Packets::QueryHits;
using Gnutella::Packets::VendorMessages::QueryStatusRequest;
using Gnutella::Packets::VendorMessages::QueryStatusResponse;
using Gnutella::Searching::Searcher;
using Gnutella::PacketProcessing::PacketProcessor;
using UIs::Searching::SearchModel;
using UIs::Searching::SearchResult;

class Gnutella::Searching::Searcher::SearcherPrivate {
public:
    LocalPeer                       *localPeer;
    PacketProcessor                 *packetProcessor;
    QHash <QUuid, SearchModel*>     searches;

    SearcherPrivate() : localPeer (0), packetProcessor (0), searches() {}

private:
    REFERENCE_OBJECT (SearcherPrivate)
};

Searcher::Searcher (LocalPeer *localPeer)
 :  p()
{
    p = new SearcherPrivate;
    p->localPeer = localPeer;
    p->packetProcessor = localPeer->packetProcessor();
}

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

void Searcher::startSearch (SearchModel *searchModel, const QString &searchString)
{
    Gnutella::Packets::Query  query(searchString);
    // We will use leaf-guidance for our own searches:
    query.setMinimumSpeed (0x9000); // bit 15 and bit 12 set

    p->searches.insert(query.descriptorId(), searchModel);
    p->packetProcessor->sendPacket (query);
}

void Searcher::stopSearch (SearchModel *searchModel)
{
    QUuid descriptorId = p->searches.key (searchModel);
    stopSearch (descriptorId);
}

void Searcher::stopSearch (const QUuid &descriptorId)
{
    p->searches.remove (descriptorId);

    QueryStatusResponse response (0xFFFF); // \todo Don't use magic values!
    response.setDescriptorId (descriptorId);
    response.setTtl (0); // \todo any trouble with the ttl == 0?
    p->packetProcessor->sendPacket (response);
}

void Searcher::processQueryHits (const QueryHits &queryHits)
{
    SearchModel *searchModel = p->searches.value (queryHits.descriptorId());
    if (searchModel == 0)
        return; // The QueryHits were not for us!
//  qDebug() << queryHits.numberOfHits();
    for (int i=0; i<queryHits.numberOfHits(); i++) {
        QueryHits::Result result = queryHits.result (i);
        searchModel->addItem (new SearchResult (result.fileName(), result.fileExtension(), result.sizeOfFile(), queryHits.speed(), queryHits.ipAddress().toString(), queryHits.serventId()));
    }
}

void Searcher::processQueryStatusRequest (const QueryStatusRequest &request)
{
    SearchModel *searchModel = p->searches.value (request.descriptorId());
    if (searchModel == 0)
        return; // The QueryStatusRequest was not for us!

    QueryStatusResponse response (searchModel->rowCount());
    response.setDescriptorId (request.descriptorId());
    p->packetProcessor->sendPacket (response);
}

Copyright © 2006 Peter Dimov.
Powered by Doxygen.