add O_OBJECT macros to appropriate classes, fix PacketList errors

This commit is contained in:
Stephen Birarda 2015-07-08 22:46:43 -07:00
parent 302e502d84
commit f2a198dfdd
5 changed files with 50 additions and 48 deletions

View file

@ -39,7 +39,7 @@
#include "Node.h"
#include "NLPacket.h"
#include "PacketHeaders.h"
#include "PacketList.h"
#include "NLPacketList.h"
#include "UUIDHasher.h"
const int MAX_PACKET_SIZE = 1450;
@ -69,7 +69,6 @@ Q_DECLARE_METATYPE(SharedNodePointer)
using namespace tbb;
typedef std::pair<QUuid, SharedNodePointer> UUIDNodePair;
typedef concurrent_unordered_map<QUuid, SharedNodePointer, UUIDHasher> NodeHash;
using NLPacketList = PacketList<NLPacket>;
typedef quint8 PingType_t;
namespace PingType {

View file

@ -15,23 +15,24 @@
#include "Packet.h"
class NLPacket : public Packet {
Q_OBJECT
public:
static std::unique_ptr<NLPacket> create(PacketType::Value type, int64_t size = -1);
// Provided for convenience, try to limit use
static std::unique_ptr<NLPacket> createCopy(const std::unique_ptr<NLPacket>& other);
static int64_t localHeaderSize(PacketType::Value type);
static int64_t maxPayloadSize(PacketType::Value type);
virtual qint64 totalHeadersSize() const; // Cumulated size of all the headers
virtual qint64 localHeaderSize() const; // Current level's header size
protected:
NLPacket(PacketType::Value type, int64_t size);
NLPacket(NLPacket& other);
void setSourceUuid(QUuid sourceUuid);
void setConnectionUuid(QUuid connectionUuid);
};
#endif // hifi_NLPacket_h
#endif // hifi_NLPacket_h

View file

@ -1,5 +1,5 @@
//
// PacketList.cpp
// NLPacketList.cpp
// libraries/networking/src
//
// Created by Stephen Birarda on 07/06/15.
@ -9,19 +9,19 @@
// See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
//
#include "PacketList.h"
#include "NLPacketList.h"
#include <QDebug>
template <typename T> PacketList<T>::PacketList(PacketType::Value packetType) :
NLPacketList::NLPacketList(PacketType::Value packetType) :
_packetType(packetType)
{
}
template <typename T> std::unique_ptr<T> PacketList<T>::createPacketWithExtendedHeader() {
std::unique_ptr<NLPacket> NLPacketList::createPacketWithExtendedHeader() {
// use the static create method to create a new packet
auto packet = T::create(_packetType);
auto packet = NLPacket::create(_packetType);
// add the extended header to the front of the packet
if (packet->write(_extendedHeader) == -1) {
@ -32,16 +32,16 @@ template <typename T> std::unique_ptr<T> PacketList<T>::createPacketWithExtended
return packet;
}
template <typename T> qint64 PacketList<T>::writeData(const char* data, qint64 maxSize) {
qint64 NLPacketList::writeData(const char* data, qint64 maxSize) {
if (!_currentPacket) {
// we don't have a current packet, time to set one up
_currentPacket = createPacketWithExtendedHeader();
}
// check if this block of data can fit into the currentPacket
if (maxSize <= _currentPacket.size()) {
if (maxSize <= _currentPacket->bytesAvailable()) {
// it fits, just write it to the current packet
_currentPacket.write(data, maxSize);
_currentPacket->write(data, maxSize);
// return the number of bytes written
return maxSize;
@ -56,9 +56,9 @@ template <typename T> qint64 PacketList<T>::writeData(const char* data, qint64 m
// We need to try and pull the first part of the segment out to our new packet
// check now to see if this is an unsupported write
int numBytesToEnd = _currentPacket.size() - _segmentStartIndex;
int numBytesToEnd = _currentPacket->bytesAvailable();
if ((newPacket.size() - numBytesToEnd) < maxSize) {
if ((newPacket->size() - numBytesToEnd) < maxSize) {
// this is an unsupported case - the segment is bigger than the size of an individual packet
// but the PacketList is not going to be sent ordered
qDebug() << "Error in PacketList::writeData - attempted to write a segment to an unordered packet that is"
@ -67,23 +67,23 @@ template <typename T> qint64 PacketList<T>::writeData(const char* data, qint64 m
}
// copy from currentPacket where the segment started to the beginning of the newPacket
newPacket.write(_currentPacket->getPayload() + _segmentStartIndex, numBytesToEnd);
newPacket->write(_currentPacket->getPayload() + _segmentStartIndex, numBytesToEnd);
// the current segment now starts at the beginning of the new packet
_segmentStartIndex = 0;
// shrink the current payload to the actual size of the packet
_currentPacket.setSizeUsed(_segmentStartIndex);
_currentPacket->setSizeUsed(_segmentStartIndex);
}
// move the current packet to our list of packets
_packets.insert(std::move(_currentPacket));
_packets.push_back(std::move(_currentPacket));
// write the data to the newPacket
newPacket.write(data, maxSize);
newPacket->write(data, maxSize);
// set our current packet to the new packet
_currentPacket = newPacket;
// swap our current packet with the new packet
_currentPacket.swap(newPacket);
// return the number of bytes written to the new packet
return maxSize;
@ -91,11 +91,11 @@ template <typename T> qint64 PacketList<T>::writeData(const char* data, qint64 m
// we're an ordered PacketList - let's fit what we can into the current packet and then put the leftover
// into a new packet
int numBytesToEnd = _currentPacket.size() - _currentPacket.sizeUsed();
int numBytesToEnd = _currentPacket->bytesAvailable();
_currentPacket->write(data, numBytesToEnd);
// move the current packet to our list of packets
_packets.insert(std::move(_currentPacket));
_packets.push_back(std::move(_currentPacket));
// recursively call our writeData method for the remaining data to write to a new packet
return numBytesToEnd + writeData(data + numBytesToEnd, maxSize - numBytesToEnd);
@ -103,8 +103,8 @@ template <typename T> qint64 PacketList<T>::writeData(const char* data, qint64 m
}
}
template <typename T> void PacketList<T>::closeCurrentPacket() {
void NLPacketList::closeCurrentPacket() {
// move the current packet to our list of packets
_packets.insert(std::move(_currentPacket));
_packets.push_back(std::move(_currentPacket));
}

View file

@ -1,5 +1,5 @@
//
// PacketList.h
// NLPacketList.h
// libraries/networking/src
//
// Created by Stephen Birarda on 07/06/15.
@ -9,60 +9,61 @@
// See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
//
#ifndef hifi_PacketList_h
#define hifi_PacketList_h
#ifndef hifi_NLPacketList_h
#define hifi_NLPacketList_h
#include <QIODevice>
#include <QtCore/QIODevice>
#include "PacketHeaders.h"
template <typename T> class PacketList : public QIODevice {
#include "NLPacket.h"
class NLPacketList : public QIODevice {
Q_OBJECT
public:
PacketList(PacketType::Value packetType);
NLPacketList(PacketType::Value packetType);
virtual bool isSequential() const { return true; }
void startSegment() { _segmentStartIndex = _currentPacket->payload().pos(); }
void startSegment() { _segmentStartIndex = _currentPacket->pos(); }
void endSegment() { _segmentStartIndex = -1; }
int getNumPackets() const { return _packets.size() + (_currentPacket ? 1 : 0); }
void getCurrentPacketCapacity() const { return _currentPacket->bytesAvailable(); }
void closeCurrentPacket();
void setExtendedHeader(const QByteArray& extendedHeader) { _extendedHeader = extendedHeader; }
template<typename U> qint64 readPrimitive(U* data);
template<typename U> qint64 writePrimitive(const U& data);
template<typename T> qint64 readPrimitive(T* data);
template<typename T> qint64 writePrimitive(const T& data);
protected:
virtual qint64 writeData(const char* data, qint64 maxSize);
virtual qint64 readData(char* data, qint64 maxSize) { return 0; }
private:
PacketList(const PacketList& other) = delete;
PacketList& operator=(const PacketList& other) = delete;
NLPacketList(const NLPacketList& other) = delete;
NLPacketList& operator=(const NLPacketList& other) = delete;
std::unique_ptr<T> createPacketWithExtendedHeader();
std::unique_ptr<NLPacket> createPacketWithExtendedHeader();
PacketType::Value _packetType;
bool _isOrdered = false;
std::unique_ptr<T> _currentPacket;
std::list<std::unique_ptr<T>> _packets;
std::unique_ptr<NLPacket> _currentPacket;
std::list<std::unique_ptr<NLPacket>> _packets;
int _segmentStartIndex = -1;
QByteArray _extendedHeader;
};
template <typename T> template <typename U> qint64 PacketList<T>::readPrimitive(U* data) {
return QIODevice::read(reinterpret_cast<char*>(data), sizeof(U));
template <typename T> qint64 NLPacketList::readPrimitive(T* data) {
return QIODevice::read(reinterpret_cast<char*>(data), sizeof(T));
}
template <typename T> template <typename U> qint64 PacketList<T>::writePrimitive(const U& data) {
static_assert(!std::is_pointer<U>::value, "U must not be a pointer");
return QIODevice::write(reinterpret_cast<const char*>(&data), sizeof(U));
template <typename T> qint64 NLPacketList::writePrimitive(const T& data) {
static_assert(!std::is_pointer<T>::value, "T must not be a pointer");
return QIODevice::write(reinterpret_cast<const char*>(&data), sizeof(T));
}

View file

@ -19,6 +19,7 @@
#include "PacketHeaders.h"
class Packet : public QIODevice {
Q_OBJECT
public:
using SequenceNumber = uint16_t;