void UDPListener::processPendingMulticastDatagrams() { while (this->multicastSocket.hasPendingDatagrams()) { QHostAddress peerAddress; const Common::MessageHeader& header = UDPListener::readDatagramToBuffer(this->multicastSocket, peerAddress); if (header.isNull()) continue; switch (header.getType()) { case Common::MessageHeader::CORE_IM_ALIVE: { Protos::Core::IMAlive IMAliveMessage; const bool readOk = IMAliveMessage.ParseFromArray(this->bodyBuffer, header.getSize()); if (!readOk) { L_WARN(QString("Unable to read the IMAlive message from peer %1 %2").arg(header.getSenderID().toStr()).arg(peerAddress.toString())); break; } else if (IMAliveMessage.version() != PROTOCOL_VERSION) // If the protocol version doesn't match we don't add the peer. { L_WARN( QString("The peer %1 %2 %3 doesn't have the same protocol version (%4) as us (%5). It will be ignored.") .arg(Common::ProtoHelper::getStr(IMAliveMessage, &Protos::Core::IMAlive::nick)) .arg(header.getSenderID().toStr()) .arg(peerAddress.toString()) .arg(IMAliveMessage.version()) .arg(PROTOCOL_VERSION) ); break; } this->peerManager->updatePeer( header.getSenderID(), peerAddress, IMAliveMessage.port(), Common::ProtoHelper::getStr(IMAliveMessage, &Protos::Core::IMAlive::nick), IMAliveMessage.amount(), Common::ProtoHelper::getStr(IMAliveMessage, &Protos::Core::IMAlive::core_version) ); if (IMAliveMessage.chunk_size() > 0) { QList<Common::Hash> hashes; hashes.reserve(IMAliveMessage.chunk_size()); for (int i = 0; i < IMAliveMessage.chunk_size(); i++) hashes << IMAliveMessage.chunk(i).hash(); QBitArray bitArray = this->fileManager->haveChunks(hashes); if (!bitArray.isNull()) // If we own at least one chunk we reply with a CHUNKS_OWNED message. { Protos::Core::ChunksOwned chunkOwnedMessage; chunkOwnedMessage.set_tag(IMAliveMessage.tag()); chunkOwnedMessage.mutable_chunk_state()->Reserve(bitArray.size()); for (int i = 0; i < bitArray.size(); i++) chunkOwnedMessage.add_chunk_state(bitArray[i]); this->send(Common::MessageHeader::CORE_CHUNKS_OWNED, header.getSenderID(), chunkOwnedMessage); } } } break; case Common::MessageHeader::CORE_CHAT_MESSAGE: { Protos::Core::ChatMessage chatMessage; chatMessage.ParseFromArray(this->bodyBuffer, header.getSize()); emit newChatMessage(header.getSenderID(), chatMessage); } break; case Common::MessageHeader::CORE_FIND: // Find. { Protos::Core::Find findMessage; findMessage.ParseFromArray(this->bodyBuffer, header.getSize()); QList<Protos::Common::FindResult> results = this->fileManager->find( Common::ProtoHelper::getStr(findMessage, &Protos::Core::Find::pattern), SETTINGS.get<quint32>("max_number_of_search_result_to_send"), SETTINGS.get<quint32>("max_udp_datagram_size") - Common::MessageHeader::HEADER_SIZE ); for (QMutableListIterator<Protos::Common::FindResult> i(results); i.hasNext();) { Protos::Common::FindResult& result = i.next(); result.set_tag(findMessage.tag()); this->send(Common::MessageHeader::CORE_FIND_RESULT, header.getSenderID(), result); } } break; default: L_WARN(QString("Unkown header type from multicast socket : %1").arg(header.getType(), 0, 16)); } } }
void UDPListener::processPendingMulticastDatagrams() { while (this->multicastSocket.hasPendingDatagrams()) { QHostAddress peerAddress; const Common::MessageHeader& header = this->readDatagramToBuffer(this->multicastSocket, peerAddress); if (header.isNull()) continue; try { const Common::Message& message = Common::Message::readMessageBody(header, this->bodyBuffer); switch (header.getType()) { case Common::MessageHeader::CORE_IM_ALIVE: { const Protos::Core::IMAlive& IMAliveMessage = message.getMessage<Protos::Core::IMAlive>(); if (IMAliveMessage.version() != PROTOCOL_VERSION) // If the protocol version doesn't match we don't add the peer. { L_WARN( QString("The peer %1 %2 %3 doesn't have the same protocol version (%4) as us (%5). It will be ignored.") .arg(Common::ProtoHelper::getStr(IMAliveMessage, &Protos::Core::IMAlive::nick)) .arg(header.getSenderID().toStr()) .arg(peerAddress.toString()) .arg(IMAliveMessage.version()) .arg(PROTOCOL_VERSION) ); break; } this->peerManager->updatePeer( header.getSenderID(), peerAddress, IMAliveMessage.port(), Common::ProtoHelper::getStr(IMAliveMessage, &Protos::Core::IMAlive::nick), IMAliveMessage.amount(), Common::ProtoHelper::getStr(IMAliveMessage, &Protos::Core::IMAlive::core_version), IMAliveMessage.download_rate(), IMAliveMessage.upload_rate() ); if (IMAliveMessage.chunk_size() > 0) { QList<Common::Hash> hashes; hashes.reserve(IMAliveMessage.chunk_size()); for (int i = 0; i < IMAliveMessage.chunk_size(); i++) hashes << IMAliveMessage.chunk(i).hash(); const QBitArray& bitArray = this->fileManager->haveChunks(hashes); if (!bitArray.isNull()) // If we own at least one chunk we reply with a CHUNKS_OWNED message. { Protos::Core::ChunksOwned chunkOwnedMessage; chunkOwnedMessage.set_tag(IMAliveMessage.tag()); chunkOwnedMessage.mutable_chunk_state()->Reserve(bitArray.size()); for (int i = 0; i < bitArray.size(); i++) chunkOwnedMessage.add_chunk_state(bitArray[i]); this->send(Common::MessageHeader::CORE_CHUNKS_OWNED, chunkOwnedMessage, header.getSenderID()); } } } break; case Common::MessageHeader::CORE_GOODBYE: this->peerManager->removePeer(header.getSenderID(), peerAddress); break; case Common::MessageHeader::CORE_FIND: { const Protos::Core::Find& findMessage = message.getMessage<Protos::Core::Find>(); QList<Protos::Common::FindResult> results = this->fileManager->find( Common::ProtoHelper::getStr(findMessage, &Protos::Core::Find::pattern), SETTINGS.get<quint32>("max_number_of_search_result_to_send"), this->MAX_UDP_DATAGRAM_PAYLOAD_SIZE - Common::MessageHeader::HEADER_SIZE ); for (QMutableListIterator<Protos::Common::FindResult> i(results); i.hasNext();) { Protos::Common::FindResult& result = i.next(); result.set_tag(findMessage.tag()); this->send(Common::MessageHeader::CORE_FIND_RESULT, result, header.getSenderID()); } } break; default:; // Ignore other messages. } emit received(message); } catch(Common::ReadErrorException&) { L_WARN(QString("Unable to read a multicast message from peer %1 %2").arg(header.getSenderID().toStr()).arg(peerAddress.toString())); } } }