void ChunkDownloader::result(const Protos::Core::GetChunkResult& result)
{
if (result.status() != Protos::Core::GetChunkResult::OK)
{
L_WARN(QString("Status error from GetChunkResult : %1. Download aborted.").arg(result.status()));
if (this->peers.removeOne(this->currentDownloadingPeer))
{
this->linkedPeers.rmLink(this->currentDownloadingPeer);
emit numberOfPeersChanged();
}
this->downloadingEnded();
}
else
{
if (!result.has_chunk_size())
{
L_ERRO(QString("Message 'GetChunkResult' doesn't contain the size of the chunk : %1. Download aborted.").arg(this->chunk->getHash().toStr()));
this->closeTheSocket = true;
this->downloadingEnded();
}
else
{
this->chunkSize = result.chunk_size();
}
}
}
void PeerMessageSocket::onNewMessage(const Common::Message& message)
{
switch (message.getHeader().getType())
{
case Common::MessageHeader::CORE_GET_ENTRIES:
{
if (!this->entriesResultsToReceive.isEmpty())
return;
const Protos::Core::GetEntries& getEntries = message.getMessage<Protos::Core::GetEntries>();
for (int i = 0; i < getEntries.dirs().entry_size(); i++)
{
QSharedPointer<FM::IGetEntriesResult> entriesResult = this->fileManager->getScannedEntries(getEntries.dirs().entry(i));
connect(entriesResult.data(), SIGNAL(result(const Protos::Core::GetEntriesResult::EntryResult&)), this, SLOT(entriesResult(const Protos::Core::GetEntriesResult::EntryResult&)), Qt::DirectConnection);
connect(entriesResult.data(), SIGNAL(timeout()), this, SLOT(entriesResultTimeout()), Qt::DirectConnection);
this->entriesResultsToReceive << entriesResult;
this->entriesResultMessage.add_result();
}
// Add the root directories if asked.
if (getEntries.dirs().entry_size() == 0 || getEntries.get_roots())
this->entriesResultMessage.add_result()->mutable_entries()->CopyFrom(this->fileManager->getEntries());
if (this->entriesResultsToReceive.isEmpty())
this->sendEntriesResultMessage();
else
foreach (QSharedPointer<FM::IGetEntriesResult> entriesResult, this->entriesResultsToReceive)
entriesResult->start();
}
break;
case Common::MessageHeader::CORE_GET_ENTRIES_RESULT:
this->finished();
break;
case Common::MessageHeader::CORE_GET_HASHES:
{
const Protos::Core::GetHashes& getHashes = message.getMessage<Protos::Core::GetHashes>();
this->currentHashesResult = this->fileManager->getHashes(getHashes.file());
connect(this->currentHashesResult.data(), SIGNAL(nextHash(Protos::Core::HashResult)), this, SLOT(nextAskedHash(Protos::Core::HashResult)), Qt::QueuedConnection);
Protos::Core::GetHashesResult res = this->currentHashesResult->start();
this->nbHash = res.nb_hash();
this->send(Common::MessageHeader::CORE_GET_HASHES_RESULT, res);
if (res.status() != Protos::Core::GetHashesResult_Status_OK)
{
this->currentHashesResult.clear();
this->finished();
}
}
break;
case Common::MessageHeader::CORE_GET_HASHES_RESULT:
{
const Protos::Core::GetHashesResult& getHashesResult = message.getMessage<Protos::Core::GetHashesResult>();
this->nbHash = getHashesResult.nb_hash();
}
break;
case Common::MessageHeader::CORE_HASH_RESULT:
{
if (--this->nbHash == 0)
this->finished();
}
break;
case Common::MessageHeader::CORE_GET_CHUNK:
{
const Protos::Core::GetChunk& getChunkMessage = message.getMessage<Protos::Core::GetChunk>();
const Common::Hash hash(getChunkMessage.chunk().hash());
if (hash.isNull())
{
L_WARN("GET_CHUNK: Chunk null");
this->finished(true);
break;
}
// TODO: implements 'GetChunkResult.ALREADY_DOWNLOADING', 'GetChunkResult.TOO_MANY_CONNECTIONS' and 'GetChunkResult.DONT_HAVE_DATA_FROM_OFFSET'
QSharedPointer<FM::IChunk> chunk = this->fileManager->getChunk(hash);
if (chunk.isNull())
{
Protos::Core::GetChunkResult result;
result.set_status(Protos::Core::GetChunkResult::DONT_HAVE);
this->send(Common::MessageHeader::CORE_GET_CHUNK_RESULT, result);
this->finished();
L_WARN(QString("GET_CHUNK: Chunk unknown : %1").arg(hash.toStr()));
}
else
{
Protos::Core::GetChunkResult result;
result.set_status(Protos::Core::GetChunkResult::OK);
result.set_chunk_size(chunk->getKnownBytes());
this->send(Common::MessageHeader::CORE_GET_CHUNK_RESULT, result);
this->stopListening();
emit getChunk(chunk, getChunkMessage.offset(), this);
}
}
break;
default:; // Do nothing.
}
}
void Socket::onNewMessage(Common::MessageHeader::MessageType type, const google::protobuf::Message& message)
{
switch (type)
{
case Common::MessageHeader::CORE_GET_ENTRIES:
{
const Protos::Core::GetEntries& getEntries = static_cast<const Protos::Core::GetEntries&>(message);
Protos::Core::GetEntriesResult result;
for (int i = 0; i < getEntries.dirs().entry_size(); i++)
result.add_entries()->CopyFrom(this->fileManager->getEntries(getEntries.dirs().entry(i)));
// Add the root directories if asked.
if (getEntries.dirs().entry_size() == 0 || getEntries.get_roots())
result.add_entries()->CopyFrom(this->fileManager->getEntries());
this->send(Common::MessageHeader::CORE_GET_ENTRIES_RESULT, result);
this->finished();
}
break;
case Common::MessageHeader::CORE_GET_ENTRIES_RESULT:
this->finished();
break;
case Common::MessageHeader::CORE_GET_HASHES:
{
const Protos::Core::GetHashes& getHashes = static_cast<const Protos::Core::GetHashes&>(message);
this->currentHashesResult = this->fileManager->getHashes(getHashes.file());
connect(this->currentHashesResult.data(), SIGNAL(nextHash(Common::Hash)), this, SLOT(nextAskedHash(Common::Hash)), Qt::QueuedConnection);
Protos::Core::GetHashesResult res = this->currentHashesResult->start();
this->nbHash = res.nb_hash();
this->send(Common::MessageHeader::CORE_GET_HASHES_RESULT, res);
if (res.status() != Protos::Core::GetHashesResult_Status_OK)
{
this->currentHashesResult.clear();
this->finished();
}
}
break;
case Common::MessageHeader::CORE_GET_HASHES_RESULT:
{
const Protos::Core::GetHashesResult& getHashesResult = static_cast<const Protos::Core::GetHashesResult&>(message);
this->nbHash = getHashesResult.nb_hash();
}
break;
case Common::MessageHeader::CORE_HASH:
{
if (--this->nbHash == 0)
this->finished();
}
break;
case Common::MessageHeader::CORE_GET_CHUNK:
{
const Protos::Core::GetChunk& getChunkMessage = static_cast<const Protos::Core::GetChunk&>(message);
const Common::Hash hash(getChunkMessage.chunk().hash());
if (hash.isNull())
{
L_WARN("GET_CHUNK: Chunk null");
this->finished(ISocket::SFS_ERROR);
break;
}
QSharedPointer<FM::IChunk> chunk = this->fileManager->getChunk(hash);
if (chunk.isNull())
{
Protos::Core::GetChunkResult result;
result.set_status(Protos::Core::GetChunkResult_Status_DONT_HAVE);
this->send(Common::MessageHeader::CORE_GET_CHUNK_RESULT, result);
this->finished();
L_WARN(QString("GET_CHUNK: Chunk unknown : %1").arg(hash.toStr()));
}
else
{
Protos::Core::GetChunkResult result;
result.set_status(Protos::Core::GetChunkResult_Status_OK);
result.set_chunk_size(chunk->getKnownBytes());
this->send(Common::MessageHeader::CORE_GET_CHUNK_RESULT, result);
this->stopListening();
emit getChunk(chunk, getChunkMessage.offset(), this);
}
}
break;
default:; // Do nothing.
}
}
