////  PacketTests.cpp//  tests/networking/src////  Created by Stephen Birarda on 07/14/15.//  Copyright 2015 High Fidelity, Inc.//  Copyright 2021 Vircadia contributors.////  Distributed under the Apache License, Version 2.0.//  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html// #include "PacketTests.h"#include <test-utils/QTestExtensions.h> #include <NLPacket.h> QTEST_MAIN(PacketTests) std::unique_ptr<NLPacket> copyToReadPacket(std::unique_ptr<NLPacket>& packet) {    auto size = packet->getDataSize();    auto data = std::unique_ptr<char[]>(new char[size]);    memcpy(data.get(), packet->getData(), size);    return NLPacket::fromReceivedPacket(std::move(data), size, SockAddr());} void PacketTests::emptyPacketTest() {    auto packet = NLPacket::create(PacketType::Unknown);     QCOMPARE(packet->getType(), PacketType::Unknown);    QCOMPARE(packet->getPayloadSize(), 0);    QCOMPARE(packet->getDataSize(), NLPacket::totalHeaderSize(packet->getType()));    QCOMPARE(packet->bytesLeftToRead(), 0);    QCOMPARE(packet->bytesAvailableForWrite(), packet->getPayloadCapacity());} void PacketTests::packetTypeTest() {    auto packet = NLPacket::create(PacketType::EntityAdd);     QCOMPARE(packet->getType(), PacketType::EntityAdd);     packet->setType(PacketType::MixedAudio);    QCOMPARE(packet->getType(), PacketType::MixedAudio);     packet->setType(PacketType::MuteEnvironment);    QCOMPARE(packet->getType(), PacketType::MuteEnvironment);} void PacketTests::writeTest() {    auto packet = NLPacket::create(PacketType::Unknown);     QCOMPARE(packet->getPayloadSize(), 0);     QCOMPARE(packet->write("somedata"), 8);    QCOMPARE(packet->getPayloadSize(), 8);     QCOMPARE(packet->write("moredata"), 8);    QCOMPARE(packet->getPayloadSize(), 16);     COMPARE_DATA(packet->getPayload(), "somedatamoredata", 16);} void PacketTests::readTest() {    // Test reads for several different size packets    for (int i = 1; i < 4; i++) {        auto packet = NLPacket::create(PacketType::Unknown);         auto size = packet->getPayloadCapacity();        size /= i;         char* data = new char[size];         // Fill with "random" data        for (qint64 i = 0; i < size; i++) {            data[i] = i;        }         QCOMPARE(packet->write(data, size), size);         auto readPacket = copyToReadPacket(packet);         QCOMPARE(readPacket->getDataSize(), packet->getDataSize());        QCOMPARE(readPacket->getPayloadSize(), packet->getPayloadSize());        COMPARE_DATA(readPacket->getData(), packet->getData(), packet->getDataSize());        COMPARE_DATA(readPacket->getPayload(), packet->getPayload(), packet->getPayloadSize());         auto payloadSize = readPacket->getPayloadSize();        char* readData = new char[payloadSize];        QCOMPARE(readPacket->read(readData, payloadSize), payloadSize);        COMPARE_DATA(readData, data, payloadSize);    }} void PacketTests::writePastCapacityTest() {    #ifndef QT_NO_DEBUG    QSKIP("This test triggers an assertion when built in debug mode");    #else    auto packet = NLPacket::create(PacketType::Unknown);     auto size = packet->getPayloadCapacity();    char* data = new char[size];     // Fill with "random" data    for (qint64 i = 0; i < size; i++) {        data[i] = i;    }     QCOMPARE(packet->getPayloadSize(), 0);     QCOMPARE(packet->write(data, size), size);     QCOMPARE(packet->getPayloadSize(), size);    COMPARE_DATA(packet->getPayload(), data, size);     QCOMPARE(packet->bytesAvailableForWrite(), 0);    QCOMPARE(packet->getPayloadSize(), size);     QCOMPARE(NLPacket::PACKET_WRITE_ERROR, packet->write("data")); // asserts in DEBUG     // NLPacket::write() shouldn't allow the caller to write if no space is left    QCOMPARE(packet->getPayloadSize(), size);    #endif} void PacketTests::primitiveTest() {    auto packet = NLPacket::create(PacketType::Unknown);     int value1 = 5;    char value2 = 10;    bool value3 = true;    qint64 value4 = -93404;     packet->writePrimitive(value1);    packet->writePrimitive(value2);    packet->writePrimitive(value3);    packet->writePrimitive(value4);     auto recvPacket = copyToReadPacket(packet);     // Peek & read first value    {        int peekValue = 0;        QCOMPARE(recvPacket->peekPrimitive(&peekValue), (int)sizeof(peekValue));        QCOMPARE(peekValue, value1);        int readValue = 0;        QCOMPARE(recvPacket->readPrimitive(&readValue), (int)sizeof(readValue));        QCOMPARE(readValue, value1);    }     // Peek & read second value    {        char peekValue = 0;        QCOMPARE(recvPacket->peekPrimitive(&peekValue), (int)sizeof(peekValue));        QCOMPARE(peekValue, value2);         char readValue = 0;        QCOMPARE(recvPacket->readPrimitive(&readValue), (int)sizeof(readValue));        QCOMPARE(readValue, value2);    }     // Peek & read third value    {        bool peekValue = 0;        QCOMPARE(recvPacket->peekPrimitive(&peekValue), (int)sizeof(peekValue));        QCOMPARE(peekValue, value3);         bool readValue = 0;        QCOMPARE(recvPacket->readPrimitive(&readValue), (int)sizeof(readValue));        QCOMPARE(readValue, value3);    }     // Peek & read fourth value    {        qint64 peekValue = 0;        QCOMPARE(recvPacket->peekPrimitive(&peekValue), (int)sizeof(peekValue));        QCOMPARE(peekValue, value4);         qint64 readValue = 0;        QCOMPARE(recvPacket->readPrimitive(&readValue), (int)sizeof(readValue));        QCOMPARE(readValue, value4);    }     // We should be at the end of the packet, with no more to peek or read    int noValue;    QCOMPARE(recvPacket->peekPrimitive(&noValue), 0);    QCOMPARE(recvPacket->readPrimitive(&noValue), 0);}