packet.h

#ifndef PRRT_FRAME_H
#define PRRT_FRAME_H

#include <stdint.h>
#include "../../util/pipe.h"
#include "codingParams.h"
#include <stdbool.h>
#include <netinet/in.h>


typedef enum {
    PACKET_TYPE_DATA = 0,
    PACKET_TYPE_REPETITION = 1,
    PACKET_TYPE_REDUNDANCY = 2,
    PACKET_TYPE_FEEDBACK = 3,
    PACKET_TYPE_PRESENT_REDUNDANCY = 4,
    PACKET_TYPE_CHANNEL_FEEDBACK = 5
} prrtPacketTypeEnum_t;

typedef uint16_t prrtSequenceNumber_t;
typedef uint8_t prrtIndex_t;

typedef uint32_t prrtTimestamp_t; // microsecond
typedef atomic_uint_fast32_t prrtAtomicTimestamp_t;

typedef uint32_t prrtTimedelta_t; // microsecond
typedef atomic_uint_fast32_t prrtAtomicTimedelta_t;

typedef int32_t prrtTimeDifference_t; // microsecond
typedef int64_t prrtTimeDiff_t; // microsecond
typedef uint32_t prrtPacketLength_t; // bytes
typedef float prrtPacketLossRate_t;

typedef uint32_t prrtDeliveryRate_t; // bits per second
typedef atomic_uint_fast32_t prrtAtomicDeliveryRate_t;

typedef uint32_t prrtByteCount_t;
typedef uint8_t prrtPacketType_t;

typedef uint16_t prrtRefcount_t;

typedef struct prrtIncompleteBlock {
    prrtSequenceNumber_t sequenceNumberBase;
    uint16_t repairCycleIndex;
} PrrtIncompleteBlock;

typedef struct prrtPacket {
    ListNode asListNode;
    uint8_t type_priority;
    prrtIndex_t index;
    prrtSequenceNumber_t sequenceNumber;
    void *payload;
    struct timespec channelReceive;
    prrtPacketLength_t payloadLength;

    // Packet State field that are not put into the packet on the wire
    prrtByteCount_t delivered;
    prrtTimestamp_t delivered_time;
    prrtTimestamp_t first_sent_time;
    bool is_app_limited;
    prrtTimestamp_t sent_time;
    prrtTimedelta_t rtt;

    struct sockaddr_in sender_addr;

    prrtRefcount_t refcount;
} PrrtPacket;
#define PRRT_PACKET_GENERAL_HEADER_SIZE 8
#define PRRT_PACKET_ENCODED_GENERAL_HEADER_LENGTH 4

typedef struct prrtPacketDataPayload {
    prrtPacketLength_t dataLength;
    prrtTimestamp_t timestamp;
    prrtTimedelta_t RTprop_us;
    prrtTimestamp_t packetTimeout_us;
    prrtDeliveryRate_t btl_datarate;
    prrtTimedelta_t btl_pace;
} __attribute__((packed)) PrrtPacketDataPayload;

#define PRRT_PACKET_DATA_HEADER_SIZE (\
        sizeof(prrtPacketLength_t) + \
        sizeof(prrtTimestamp_t) + \
        sizeof(prrtTimedelta_t) + \
        sizeof(prrtTimestamp_t) + \
        sizeof(prrtDeliveryRate_t) + \
        sizeof(prrtTimedelta_t))

typedef struct prrtPacketRedundancyPayload {
    prrtSequenceNumber_t baseSequenceNumber;
    prrtTimestamp_t timestamp;
    prrtTimedelta_t btl_pace;
    uint8_t n;
    uint8_t k;
} __attribute__((packed)) PrrtPacketRedundancyPayload;

#define PRRT_PACKET_REDUNDANCY_HEADER_SIZE (\
        sizeof(prrtSequenceNumber_t) + \
        sizeof(prrtTimestamp_t) + \
        sizeof(prrtTimedelta_t) + \
        sizeof(uint8_t) + \
        sizeof(uint8_t) )

typedef struct prrtPacketFeedbackPayload {
    prrtTimestamp_t forwardTripTimestamp_us;
    prrtTimedelta_t btl_pace;
    prrtSequenceNumber_t erasureCount;
    prrtSequenceNumber_t packetCount;
    prrtSequenceNumber_t gapLength;
    prrtSequenceNumber_t gapCount;
    prrtSequenceNumber_t burstLength;
    prrtSequenceNumber_t burstCount;
    prrtSequenceNumber_t ackSequenceNumber;
    prrtPacketType_t ackPacketType;
    PrrtIncompleteBlock *incompleteBlocks;
} __attribute__((packed)) PrrtPacketFeedbackPayload;

#define PRRT_PACKET_FEEDBACK_HEADER_SIZE (\
        sizeof(prrtTimedelta_t) + \
        sizeof(prrtTimestamp_t) + \
        sizeof(prrtTimedelta_t) + \
        sizeof(prrtSequenceNumber_t) + \
        sizeof(prrtSequenceNumber_t) + \
        sizeof(prrtSequenceNumber_t) + \
        sizeof(prrtSequenceNumber_t) + \
        sizeof(prrtSequenceNumber_t) + \
        sizeof(prrtSequenceNumber_t) + \
        sizeof(prrtSequenceNumber_t) + \
        sizeof(prrtPacketType_t) )


#define PrrtPacket_type(packet_ptr) ((prrtPacketType_t) (((packet_ptr)->type_priority >> 4) & 0x0F))

#define PrrtPacket_priority(packet_ptr) ((uint8_t) (packet_ptr->type_priority & 0x0F))

prrtPacketLength_t PrrtPacket_size(PrrtPacket *packet_ptr);

int PrrtPacket_print(PrrtPacket *packet_ptr);

PrrtPacket *PrrtPacket_copy(PrrtPacket *original);

PrrtPacket *PrrtPacket_deepcopy(PrrtPacket *original);

PrrtPacket *PrrtPacket_create_data_packet(uint8_t priority, const void *payloadPointer,
                                          prrtPacketLength_t dataLength, prrtSequenceNumber_t sequenceNumber,
                                          prrtTimedelta_t targetDelay);

PrrtPacket *PrrtPacket_reconstruct_data_packet(PrrtPacketDataPayload *payload, prrtIndex_t index,
                                               prrtSequenceNumber_t sequenceNumber);

PrrtPacket *
PrrtPacket_create_feedback_packet(uint8_t priority, prrtSequenceNumber_t sequenceNumber,
                                  prrtSequenceNumber_t gapLength, prrtSequenceNumber_t gapCount,
                                  prrtSequenceNumber_t burstLength, prrtSequenceNumber_t burstCount,
                                  prrtTimestamp_t forwardTripTime, prrtSequenceNumber_t erasureCount,
                                  prrtSequenceNumber_t packetCount, prrtSequenceNumber_t ackSequenceNumber,
                                  prrtPacketType_t ackPacketType, prrtTimedelta_t btl_pace);

PrrtPacket *PrrtPacket_create_redundancy_packet(uint8_t priority, void *payloadPointer,
                                                prrtPacketLength_t payloadLength,
                                                prrtSequenceNumber_t sequenceNumber, uint8_t index,
                                                prrtSequenceNumber_t baseSequenceNumber,
                                                PrrtCodingConfiguration *codingParams);

bool PrrtPacket_decode(void *srcBuffer, uint16_t srcBufferSize, PrrtPacket *targetPacket);
bool PrrtPacket_decode_payload(void *srcBuffer, PrrtPacket *targetPacket);

bool PrrtPacket_encode(void *buf_ptr, uint16_t buf_size, PrrtPacket *packet_ptr);
bool PrrtPacket_encode_payload(void *buf_ptr, PrrtPacket *packet_ptr);

int PrrtPacket_destroy(PrrtPacket *packet);

#define PrrtPacket_copy_payload_to_buffer(dst, packet, header_size) memcpy(dst, (packet)->payload + (header_size), (packet)->payloadLength - (header_size));
#define PrrtPacket_copy_buffer_to_payload(packet, src, header_size) memcpy((packet)->payload + (header_size), src, (packet)->payloadLength - (header_size));

#define PrrtPacket_get_data_timestamp(packet) ((PrrtPacketDataPayload*) packet->payload)->timestamp;
#define PrrtPacket_get_redundancy_timestamp(packet) ((PrrtPacketRedundancyPayload*) packet->payload)->timestamp;

#define PrrtPacket_byListNode(lnptr) ((PrrtPacket *) (lnptr))

#define PrrtDataPacket_packet_timeout(packet) ((PrrtPacketDataPayload*) (packet)->payload)->packetTimeout_us

#endif //PRRT_FRAME_H