mirror of
https://github.com/HifiExperiments/overte.git
synced 2025-07-16 03:36:40 +02:00
102 lines
3.7 KiB
C++
102 lines
3.7 KiB
C++
//
|
|
// PacketReceiver.h
|
|
// libraries/networking/src
|
|
//
|
|
// Created by Stephen Birarda on 1/23/2014.
|
|
// Update by Ryan Huffman on 7/8/2015.
|
|
// Copyright 2014 High Fidelity, Inc.
|
|
//
|
|
// Distributed under the Apache License, Version 2.0.
|
|
// See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
|
|
//
|
|
|
|
#ifndef hifi_PacketReceiver_h
|
|
#define hifi_PacketReceiver_h
|
|
|
|
#include <vector>
|
|
#include <unordered_map>
|
|
|
|
#include <QtCore/QMap>
|
|
#include <QtCore/QMetaMethod>
|
|
#include <QtCore/QMutex>
|
|
#include <QtCore/QObject>
|
|
#include <QtCore/QPointer>
|
|
#include <QtCore/QSet>
|
|
|
|
#include "NLPacket.h"
|
|
#include "NLPacketList.h"
|
|
#include "ReceivedMessage.h"
|
|
#include "udt/PacketHeaders.h"
|
|
|
|
class EntityEditPacketSender;
|
|
class OctreePacketProcessor;
|
|
|
|
namespace std {
|
|
template <>
|
|
struct hash<std::pair<HifiSockAddr, udt::Packet::MessageNumber>> {
|
|
size_t operator()(const std::pair<HifiSockAddr, udt::Packet::MessageNumber>& pair) const {
|
|
return hash<HifiSockAddr>()(pair.first) ^ hash<udt::Packet::MessageNumber>()(pair.second);
|
|
}
|
|
};
|
|
}
|
|
|
|
class PacketReceiver : public QObject {
|
|
Q_OBJECT
|
|
public:
|
|
using PacketTypeList = std::vector<PacketType>;
|
|
|
|
PacketReceiver(QObject* parent = 0);
|
|
PacketReceiver(const PacketReceiver&) = delete;
|
|
|
|
PacketReceiver& operator=(const PacketReceiver&) = delete;
|
|
|
|
int getInPacketCount() const { return _inPacketCount; }
|
|
int getInByteCount() const { return _inByteCount; }
|
|
|
|
void setShouldDropPackets(bool shouldDropPackets) { _shouldDropPackets = shouldDropPackets; }
|
|
|
|
void resetCounters() { _inPacketCount = 0; _inByteCount = 0; }
|
|
|
|
// If deliverPending is false, ReceivedMessage will only be delivered once all packets for the message have
|
|
// been received. If deliverPending is true, ReceivedMessage will be delivered as soon as the first packet
|
|
// for the message is received.
|
|
bool registerListener(PacketType type, QObject* listener, const char* slot, bool deliverPending = false);
|
|
bool registerListenerForTypes(PacketTypeList types, QObject* listener, const char* slot);
|
|
void unregisterListener(QObject* listener);
|
|
|
|
void handleVerifiedPacket(std::unique_ptr<udt::Packet> packet);
|
|
void handleVerifiedMessagePacket(std::unique_ptr<udt::Packet> message);
|
|
void handleMessageFailure(HifiSockAddr from, udt::Packet::MessageNumber messageNumber);
|
|
|
|
private:
|
|
struct Listener {
|
|
QPointer<QObject> object;
|
|
QMetaMethod method;
|
|
bool deliverPending;
|
|
};
|
|
|
|
void handleVerifiedMessage(QSharedPointer<ReceivedMessage> message, bool justReceived);
|
|
|
|
// these are brutal hacks for now - ideally GenericThread / ReceivedPacketProcessor
|
|
// should be changed to have a true event loop and be able to handle our QMetaMethod::invoke
|
|
void registerDirectListenerForTypes(PacketTypeList types, QObject* listener, const char* slot);
|
|
void registerDirectListener(PacketType type, QObject* listener, const char* slot);
|
|
|
|
QMetaMethod matchingMethodForListener(PacketType type, QObject* object, const char* slot) const;
|
|
void registerVerifiedListener(PacketType type, QObject* listener, const QMetaMethod& slot, bool deliverPending = false);
|
|
|
|
QMutex _packetListenerLock;
|
|
QHash<PacketType, Listener> _messageListenerMap;
|
|
int _inPacketCount = 0;
|
|
int _inByteCount = 0;
|
|
bool _shouldDropPackets = false;
|
|
QMutex _directConnectSetMutex;
|
|
QSet<QObject*> _directlyConnectedObjects;
|
|
|
|
std::unordered_map<std::pair<HifiSockAddr, udt::Packet::MessageNumber>, QSharedPointer<ReceivedMessage>> _pendingMessages;
|
|
|
|
friend class EntityEditPacketSender;
|
|
friend class OctreePacketProcessor;
|
|
};
|
|
|
|
#endif // hifi_PacketReceiver_h
|