#ifndef _NORM_MESSAGE
#define _NORM_MESSAGE
#include "protokit.h"
#include <string.h>
#include <math.h>
#include <stdlib.h>
#ifdef _WIN32_WCE
#include <stdio.h>
#else
#include <sys/types.h>
#endif
#ifdef SIMULATE
#define SIM_PAYLOAD_MAX (36+8)
#endif
const UINT8 NORM_PROTOCOL_VERSION = 1;
const double NORM_TICK_MIN = 0.100;
inline double UniformRand(double max)
{return (max * ((double)rand() / (double)RAND_MAX));}
inline double ExponentialRand(double max, double groupSize)
{
double lambda = log(groupSize) + 1;
double x = UniformRand(lambda/max)+lambda/(max*(exp(lambda)-1));
return ((max/lambda)*log(x*(exp(lambda)-1)*(max/lambda)));
}
const double NORM_GRTT_MIN = 0.001; const double NORM_GRTT_MAX = 15.0; const double NORM_RTT_MIN = 1.0e-06;
const double NORM_RTT_MAX = 1000.0;
extern const double NORM_RTT[];
inline double NormUnquantizeRtt(UINT8 qrtt)
{return NORM_RTT[qrtt];}
UINT8 NormQuantizeRtt(double rtt);
extern const double NORM_GSIZE[];
inline double NormUnquantizeGroupSize(UINT8 gsize)
{return NORM_GSIZE[gsize];}
UINT8 NormQuantizeGroupSize(double gsize);
inline UINT16 NormQuantizeLoss(double lossFraction)
{
lossFraction = MAX(lossFraction, 0.0);
lossFraction = lossFraction*65535.0 + 0.5;
lossFraction = MIN(lossFraction, 65535.0);
return (UINT16)lossFraction;
} inline double NormUnquantizeLoss(UINT16 lossQuantized)
{
return (((double)lossQuantized) / 65535.0);
}
inline UINT32 NormQuantizeLoss32(double lossFraction)
{
const double MAX_SCALE = (double)((unsigned int)0xffffffff);
lossFraction = MAX(lossFraction, 0.0);
lossFraction = lossFraction*MAX_SCALE + 0.5;
lossFraction = MIN(lossFraction, MAX_SCALE);
return (UINT32)lossFraction;
} inline double NormUnquantizeLoss32(UINT32 lossQuantized)
{
const double MAX_SCALE = (double)((unsigned int)0xffffffff);
return (((double)lossQuantized) / MAX_SCALE);
}
inline UINT16 NormQuantizeRate(double rate)
{
if (rate <= 0.0) return 0x01; UINT16 exponent = (UINT16)log10(rate);
UINT16 mantissa = (UINT16)((4096.0/10.0) * (rate / pow(10.0, (double)exponent)) + 0.5);
return ((mantissa << 4) | exponent);
}
inline double NormUnquantizeRate(UINT16 rate)
{
double mantissa = ((double)(rate >> 4)) * (10.0/4096.0);
double exponent = (double)(rate & 0x000f);
return mantissa * pow(10.0, exponent);
}
class NormObjectSize
{
public:
#ifdef WIN32
#define _FILE_OFFSET_BITS 64
typedef __int64 Offset;
#else
typedef off_t Offset;
#endif NormObjectSize() : size(0) {}
NormObjectSize(Offset theSize) : size(theSize) {}
NormObjectSize(UINT16 msb, UINT32 lsb)
{
size = (Offset)lsb;
#if (_FILE_OFFSET_BITS > 32) && !defined(ANDROID)
size |= ((Offset)msb) << 32;
#endif
}
Offset GetOffset() const {return size;}
#if (_FILE_OFFSET_BITS > 32) && !defined(ANDROID)
UINT16 MSB() const {return ((UINT16)((size >> 32) & 0x0000ffff));}
#else
UINT16 MSB() const {return 0;}
#endif
UINT32 LSB() const {return ((UINT32)(size & 0xffffffff));}
bool operator==(const NormObjectSize& b) const
{return (b.size == size);}
bool operator!=(const NormObjectSize& b) const
{return (b.size != size);}
NormObjectSize operator+(const NormObjectSize& b) const
{
NormObjectSize result(size);
result.size += b.size;
return result;
}
void operator+=(const NormObjectSize& b)
{size += b.size;}
NormObjectSize operator-(const NormObjectSize& b) const
{
NormObjectSize result(size);
result.size -= b.size;
return result;
}
void operator-=(const NormObjectSize& b)
{size -= b.size;}
void operator+=(Offset increment)
{size += increment;}
bool operator>(const NormObjectSize& b) const
{return (size > b.size);}
NormObjectSize operator*(const NormObjectSize& b) const
{
NormObjectSize result(size);
result.size *= b.size;
return result;
}
NormObjectSize operator/(const NormObjectSize& b) const
{
NormObjectSize result(size);
result.size /= b.size;
result.size = ((result.size * b.size) < size) ? result.size + 1 : result.size;
return result;
}
private:
Offset size;
};
#ifndef _NORM_API
typedef UINT32 NormNodeId;
const NormNodeId NORM_NODE_NONE = 0x00000000;
const NormNodeId NORM_NODE_ANY = 0xffffffff;
#endif
class NormObjectId
{
public:
NormObjectId() : value(0) {};
NormObjectId(UINT16 id) {value = id;}
NormObjectId(const NormObjectId& id) {value = id.value;}
operator UINT16() const {return value;}
bool operator<(const NormObjectId& id) const
{
UINT16 diff = value - id.value;
return ((diff > 0x8000) || ((0x8000 == diff) && (value > id.value)));
}
bool operator>(const NormObjectId& id) const
{
UINT16 diff = id.value - value;
return ((diff > 0x8000) || ((0x8000 == diff) && (id.value > value)));
}
bool operator<=(const NormObjectId& id) const
{return ((value == id.value) || (*this < id));}
bool operator>=(const NormObjectId& id) const
{return ((value == id.value) || (*this > id));}
bool operator==(const NormObjectId& id) const
{return (value == id.value);}
bool operator!=(const NormObjectId& id) const
{return (value != id.value);}
void operator-=(UINT16 delta)
{value -= delta;}
const char* GetValuePtr() const
{return (const char*)(&value);}
NormObjectId& operator++(int) {value++; return *this;}
NormObjectId& operator--(int) {value--; return *this;}
private:
UINT16 value;
};
class NormBlockId
{
public:
NormBlockId() {};
NormBlockId(UINT32 id) : value(id) {}
UINT32 GetValue() const
{return value;}
const char* GetValuePtr() const
{return ((const char*)&value);}
bool operator==(const NormBlockId& id) const
{return (value == id.value);}
bool operator!=(const NormBlockId& id) const
{return (value != id.value);}
static INT32 Difference(const NormBlockId& a, const NormBlockId& b, UINT32 mask)
{
if (mask)
{
UINT32 sign = (mask ^ (mask >> 1));
UINT32 result = a.value - b.value;
if (0 == (result & sign))
return (INT32)(result & mask);
else if ((result != sign) || (a.value < b.value))
return (INT32)(result | ~mask);
else
return (INT32)(result & mask);
}
else
{
return ((INT32)(a.value - b.value));
}
}
static int Compare(const NormBlockId& a, const NormBlockId& b, UINT32 mask)
{
if (mask)
{
INT32 delta = Difference(a, b, mask);
if (delta < 0)
return -1;
else if (0 == delta)
return 0;
else return 1;
}
else if (a.value < b.value)
{
return -1;
}
else if (a.value == b.value)
{
return 0;
}
else {
return 1;
}
}
void Increment(UINT32 i, UINT32 mask)
{
value = value + i;
if (mask) value &= mask;
}
void Decrement(UINT32 i, UINT32 mask)
{
if (mask && (value < i))
value = (mask - (i - value) + 1);
else
value -= i;
}
private:
UINT32 value;
};
typedef UINT16 NormSymbolId;
typedef NormSymbolId NormSegmentId;
class NormHeaderExtension
{
public:
enum Type
{
INVALID = 0,
FTI = 64, CC_FEEDBACK = 3, CC_RATE = 128, APP_ACK = 65 };
NormHeaderExtension();
virtual ~NormHeaderExtension() {}
virtual void Init(UINT32* theBuffer, UINT16 numBytes)
{
AttachBuffer(theBuffer, numBytes);
SetType(INVALID);
SetWords(0);
}
void SetType(Type type)
{((UINT8*)buffer)[TYPE_OFFSET] = (UINT8)type;}
void SetWords(UINT8 words)
{((UINT8*)buffer)[LENGTH_OFFSET] = words;}
void AttachBuffer(const UINT32* theBuffer, UINT16 bufferLength)
{
buffer = (UINT32*)theBuffer;
buffer_length = bufferLength;
}
const UINT32* GetBuffer()
{return buffer;}
Type GetType() const
{return buffer ? (Type)(((UINT8*)buffer)[TYPE_OFFSET]) : INVALID;}
UINT16 GetLength() const
{
return (buffer ?
((GetType() < 128) ?
((((UINT8*)buffer)[LENGTH_OFFSET]) << 2) : 4) :
0);
}
const char* GetContent()
{return (((const char*)buffer) + CONTENT_OFFSET);}
UINT16 GetContentLength() const
{
UINT16 totalLen = GetLength();
return ((totalLen > CONTENT_OFFSET) ? (totalLen - CONTENT_OFFSET) : 0);
}
static UINT16 GetContentOffset()
{return (UINT16)CONTENT_OFFSET;}
protected:
enum
{
TYPE_OFFSET = 0, LENGTH_OFFSET = TYPE_OFFSET + 1, CONTENT_OFFSET = LENGTH_OFFSET + 1 };
UINT32* buffer;
UINT16 buffer_length;
};
class NormPayloadId
{
public:
enum FecType
{
RS = 2, RS8 = 5, SB = 129 };
static bool IsValid(UINT8 fecId)
{
switch (fecId)
{
case 2:
case 5:
case 129:
return true;
default:
return false;
}
}
NormPayloadId(UINT8 fecId, UINT8 m, UINT32* theBuffer)
: fec_id(fecId), fec_m(m), buffer(theBuffer) {}
NormPayloadId(UINT8 fecId, UINT8 m, const UINT32* theBuffer)
: fec_id(fecId), fec_m(m), cbuffer(theBuffer) {}
static UINT16 GetLength(UINT8 fecId)
{
switch (fecId)
{
case 2:
case 5:
return 4;
case 129:
return 8;
default:
return 0;
}
}
static UINT32 GetFecBlockMask(UINT8 fecId, UINT8 fecM)
{
switch (fecId)
{
case 2:
if (8 == fecM)
return 0x00ffffff; else return 0x0000ffff; case 5:
return 0x00ffffff; case 129:
return 0xffffffff; default:
return 0x00000000; }
}
void SetFecPayloadId(UINT32 blockId, UINT16 symbolId, UINT16 blockLen)
{
switch (fec_id)
{
case 2:
if (8 == fec_m)
{
blockId = (blockId << 8) | (symbolId & 0x00ff);
*buffer = htonl(blockId); }
else {
UINT16* payloadId = (UINT16*)buffer;
payloadId[0] = htons(blockId); payloadId[1] = htons(symbolId); }
break;
case 5:
blockId = (blockId << 8) | (symbolId & 0x00ff);
*buffer = htonl(blockId); break;
case 129:
*buffer = htonl(blockId); UINT16* ptr = (UINT16*)(buffer + 1);
ptr[0] = htons(blockLen); ptr[1] = htons(symbolId); break;
}
}
NormBlockId GetFecBlockId() const
{
switch (fec_id)
{
case 2:
if (8 == fec_m)
{
UINT32 blockId = ntohl(*cbuffer);
return (0x00ffffff & (blockId >> 8));
}
else {
UINT16* blockId = (UINT16*)cbuffer;
return ntohs(*blockId);
}
case 5:
{
UINT32 blockId = ntohl(*cbuffer);
return (0x00ffffff & (blockId >> 8));
}
case 129:
return ntohl(*cbuffer);
default:
ASSERT(0);
return 0;
}
}
UINT16 GetFecSymbolId() const
{
switch (fec_id)
{
case 2:
if (8 == fec_m)
{
UINT32 payloadId = ntohl(*cbuffer);
return (0x000000ff & payloadId); }
else {
UINT16* payloadId = (UINT16*)cbuffer;
return ntohs(payloadId[1]);
}
case 5:
{
UINT32 payloadId = ntohl(*cbuffer);
return (0x000000ff & payloadId); }
case 129:
{
UINT16* ptr = (UINT16*)(cbuffer + 1);
return ntohs(ptr[1]);
}
default:
ASSERT(0);
return 0;
}
}
UINT16 GetFecBlockLength() const
{
if (129 == fec_id)
{
UINT16* blockLen = (UINT16*)(cbuffer + 1);
return ntohs(*blockLen);
}
else
{
return 0;
}
}
private:
UINT8 fec_id;
UINT8 fec_m;
union
{
UINT32* buffer;
const UINT32* cbuffer;
};
};
class NormMsg
{
friend class NormMessageQueue;
public:
enum Type
{
INVALID = 0,
INFO = 1,
DATA = 2,
CMD = 3,
NACK = 4,
ACK = 5,
REPORT = 6
};
enum {MAX_SIZE = 65536};
NormMsg();
void SetVersion(UINT8 version)
{
((UINT8*)buffer)[VERSION_OFFSET] =
(((UINT8*)buffer)[VERSION_OFFSET] & 0x0f) | (version << 4);
}
void SetType(NormMsg::Type type)
{
((UINT8*)buffer)[TYPE_OFFSET] =
(((UINT8*)buffer)[VERSION_OFFSET] & 0xf0) | (type & 0x0f);
}
void SetSequence(UINT16 sequence)
{
((UINT16*)buffer)[SEQUENCE_OFFSET] = htons(sequence);
}
void SetSourceId(NormNodeId sourceId)
{
buffer[SOURCE_ID_OFFSET] = htonl(sourceId);
}
void SetDestination(const ProtoAddress& dst) {addr = dst;}
void SetSource(const ProtoAddress& src) {addr = src;}
void AttachExtension(NormHeaderExtension& extension)
{
extension.Init(buffer+(header_length/4), MAX_SIZE - header_length);
ExtendHeaderLength(extension.GetLength());
}
void PackExtension(NormHeaderExtension& extension)
{
ExtendHeaderLength(2 + extension.GetContentLength());
}
bool InitFromBuffer(UINT16 msgLength);
bool CopyFromBuffer(const char* theBuffer, unsigned int theLength)
{
if (theLength > MAX_SIZE) return false;
memcpy(buffer, theBuffer, theLength);
return InitFromBuffer(theLength);
}
UINT8 GetVersion() const
{return (((UINT8*)buffer)[VERSION_OFFSET] >> 4);}
NormMsg::Type GetType() const
{return (Type)(((UINT8*)buffer)[TYPE_OFFSET] & 0x0f);}
UINT16 GetHeaderLength() const
{return ((UINT8*)buffer)[HDR_LEN_OFFSET] << 2;}
UINT16 GetBaseHeaderLength()
{return header_length_base;}
UINT16 GetSequence() const
{
return (ntohs((((UINT16*)buffer)[SEQUENCE_OFFSET])));
}
NormNodeId GetSourceId() const
{
return (ntohl(buffer[SOURCE_ID_OFFSET]));
}
const ProtoAddress& GetDestination() const {return addr;}
const ProtoAddress& GetSource() const {return addr;}
const char* GetBuffer() const {return ((char*)buffer);}
UINT16 GetLength() const {return length;}
void Display() const;
bool HasExtensions() const {return (header_length > header_length_base);}
bool HasExtension(NormHeaderExtension::Type extType);
bool GetNextExtension(NormHeaderExtension& ext) const
{
const UINT32* currentBuffer = ext.GetBuffer();
UINT16 nextOffset =
(UINT16)(currentBuffer ? (currentBuffer - buffer + (ext.GetLength()/4)) :
(header_length_base/4));
bool result = HasExtensions() ? (nextOffset < (header_length/4)) : false;
if (result)
{
UINT16 nextLength = ((UINT8*)(buffer + nextOffset))[1] << 2;
ext.AttachBuffer(buffer+nextOffset, nextLength);
}
else
{
ext.AttachBuffer((UINT32*)NULL, 0);
}
return result;
}
char* AccessBuffer() {return ((char*)buffer);}
ProtoAddress& AccessAddress() {return addr;}
NormMsg* GetNext() {return next;}
protected:
enum
{
VERSION_OFFSET = 0,
TYPE_OFFSET = VERSION_OFFSET,
HDR_LEN_OFFSET = VERSION_OFFSET+1,
SEQUENCE_OFFSET = (HDR_LEN_OFFSET+1)/2,
SOURCE_ID_OFFSET = ((SEQUENCE_OFFSET*2)+2)/4,
MSG_OFFSET = (SOURCE_ID_OFFSET*4)+4
};
void SetBaseHeaderLength(UINT16 len)
{
((UINT8*)buffer)[HDR_LEN_OFFSET] = len >> 2;
length = header_length_base = header_length = len;
}
void ExtendHeaderLength(UINT16 len)
{
header_length += len;
length += len;
((UINT8*)buffer)[HDR_LEN_OFFSET] = header_length >> 2;
}
UINT32 buffer[MAX_SIZE / sizeof(UINT32)];
UINT16 length; UINT16 header_length;
UINT16 header_length_base;
ProtoAddress addr;
NormMsg* prev;
NormMsg* next;
};
class NormObjectMsg : public NormMsg
{
friend class NormMsg;
public:
enum Flag
{
FLAG_REPAIR = 0x01,
FLAG_EXPLICIT = 0x02,
FLAG_INFO = 0x04,
FLAG_UNRELIABLE = 0x08,
FLAG_FILE = 0x10,
FLAG_STREAM = 0x20,
FLAG_SYN = 0x40
};
UINT16 GetInstanceId() const
{return (ntohs(((UINT16*)buffer)[INSTANCE_ID_OFFSET]));}
UINT8 GetGrtt() const
{return ((UINT8*)buffer)[GRTT_OFFSET];}
UINT8 GetBackoffFactor() const
{return ((((UINT8*)buffer)[GSIZE_OFFSET] >> 4) & 0x0f);}
UINT8 GetGroupSize() const
{return (((UINT8*)buffer)[GSIZE_OFFSET] & 0x0f);}
bool FlagIsSet(NormObjectMsg::Flag flag) const
{return (0 != (flag & ((UINT8*)buffer)[FLAGS_OFFSET]));}
bool IsStream() const
{return FlagIsSet(FLAG_STREAM);}
UINT8 GetFecId() const
{return ((UINT8*)buffer)[FEC_ID_OFFSET];}
NormObjectId GetObjectId() const
{return (ntohs(((UINT16*)buffer)[OBJ_ID_OFFSET]));}
void SetInstanceId(UINT16 instanceId)
{((UINT16*)buffer)[INSTANCE_ID_OFFSET] = htons(instanceId);}
void SetGrtt(UINT8 grtt) {((UINT8*)buffer)[GRTT_OFFSET] = grtt;}
void SetBackoffFactor(UINT8 backoff)
{((UINT8*)buffer)[BACKOFF_OFFSET] = (((UINT8*)buffer)[GSIZE_OFFSET] & 0x0f) | (backoff << 4);}
void SetGroupSize(UINT8 gsize)
{((UINT8*)buffer)[GSIZE_OFFSET] = (((UINT8*)buffer)[GSIZE_OFFSET] & 0xf0) | gsize;}
void ResetFlags() {((UINT8*)buffer)[FLAGS_OFFSET] = 0;}
void SetFlag(NormObjectMsg::Flag flag) {((UINT8*)buffer)[FLAGS_OFFSET] |= flag;}
void SetObjectId(const NormObjectId& objectId)
{((UINT16*)buffer)[OBJ_ID_OFFSET] = htons((UINT16)objectId);}
protected:
enum
{
INSTANCE_ID_OFFSET = MSG_OFFSET/2,
GRTT_OFFSET = (INSTANCE_ID_OFFSET*2)+2,
BACKOFF_OFFSET = GRTT_OFFSET+1,
GSIZE_OFFSET = BACKOFF_OFFSET,
FLAGS_OFFSET = GSIZE_OFFSET+1,
FEC_ID_OFFSET = FLAGS_OFFSET+1,
OBJ_ID_OFFSET = (FEC_ID_OFFSET+1)/2,
OBJ_MSG_OFFSET = (OBJ_ID_OFFSET*2)+2
};
};
class NormFtiExtension2 : public NormHeaderExtension
{
public:
virtual void Init(UINT32* theBuffer, UINT16 numBytes)
{
AttachBuffer(theBuffer, numBytes);
SetType(FTI); SetWords(4);
}
void SetObjectSize(const NormObjectSize& objectSize)
{
((UINT16*)buffer)[OBJ_SIZE_MSB_OFFSET] = htons(objectSize.MSB());
buffer[OBJ_SIZE_LSB_OFFSET] = htonl(objectSize.LSB());
}
void SetFecFieldSize(UINT8 numBits)
{((UINT8*)buffer)[FEC_M_OFFSET] = numBits;} void SetFecGroupSize(UINT8 symbolsPerPkt)
{((UINT8*)buffer)[FEC_G_OFFSET] = symbolsPerPkt;} void SetSegmentSize(UINT16 segmentSize)
{((UINT16*)buffer)[SEG_SIZE_OFFSET] = htons(segmentSize);}
void SetFecMaxBlockLen(UINT16 ndata)
{((UINT16*)buffer)[FEC_NDATA_OFFSET] = htons(ndata);}
void SetFecNumParity(UINT16 nparity)
{((UINT16*)buffer)[FEC_NPARITY_OFFSET] = htons(nparity);}
NormObjectSize GetObjectSize() const
{
return NormObjectSize(ntohs(((UINT16*)buffer)[OBJ_SIZE_MSB_OFFSET]),
ntohl(buffer[OBJ_SIZE_LSB_OFFSET]));
}
UINT8 GetFecFieldSize() const
{return ((UINT8*)buffer)[FEC_M_OFFSET];} UINT8 GetFecGroupSize() const
{return ((UINT8*)buffer)[FEC_G_OFFSET];} UINT16 GetSegmentSize() const
{return (ntohs(((UINT16*)buffer)[SEG_SIZE_OFFSET]));}
UINT16 GetFecMaxBlockLen() const
{return (ntohs(((UINT16*)buffer)[FEC_NDATA_OFFSET]));}
UINT16 GetFecNumParity() const
{return (ntohs(((UINT16*)buffer)[FEC_NPARITY_OFFSET]));}
private:
enum
{
OBJ_SIZE_MSB_OFFSET = (LENGTH_OFFSET + 1)/2,
OBJ_SIZE_LSB_OFFSET = ((OBJ_SIZE_MSB_OFFSET*2)+2)/4,
FEC_M_OFFSET = ((OBJ_SIZE_LSB_OFFSET*4)+4),
FEC_G_OFFSET = FEC_M_OFFSET + 1,
SEG_SIZE_OFFSET = (FEC_G_OFFSET+1)/2,
FEC_NDATA_OFFSET = ((SEG_SIZE_OFFSET*2)+2)/2,
FEC_NPARITY_OFFSET = ((FEC_NDATA_OFFSET*2)+2)/2
};
};
class NormFtiData
{
public:
NormFtiData()
: object_size(NormObjectSize(0)), segment_size(0),
num_data(0), num_parity(0), fec_m(0), instance_id(0) {}
~NormFtiData() {}
bool IsValid() const
{return (0 != segment_size);}
void Invalidate()
{
object_size = 0;
segment_size = num_data = num_parity = instance_id = 0;
fec_m = 0;
}
void SetObjectSize(const NormObjectSize& objectSize)
{object_size = objectSize;}
void SetSegmentSize(UINT16 segmentSize)
{segment_size = segmentSize;}
void SetFecMaxBlockLen(UINT16 numData)
{num_data = numData;}
void SetFecNumParity(UINT16 numParity)
{num_parity = numParity;}
void SetFecFieldSize(UINT8 fecM)
{fec_m = fecM;}
void SetFecInstanceId(UINT16 instanceId)
{instance_id = instanceId;}
const NormObjectSize& GetObjectSize() const
{return object_size;}
UINT16 GetSegmentSize() const
{return segment_size;}
UINT16 GetFecMaxBlockLen() const
{return num_data;}
UINT16 GetFecNumParity() const
{return num_parity;}
UINT8 GetFecFieldSize() const
{return fec_m;}
UINT16 GetFecInstanceId() const
{return instance_id;}
private:
NormObjectSize object_size;
UINT16 segment_size;
UINT16 num_data;
UINT16 num_parity;
UINT8 fec_m;
UINT16 instance_id;
};
class NormFtiExtension5 : public NormHeaderExtension
{
public:
virtual void Init(UINT32* theBuffer, UINT16 numBytes)
{
AttachBuffer(theBuffer, numBytes);
SetType(FTI); SetWords(3);
}
void SetObjectSize(const NormObjectSize& objectSize)
{
((UINT16*)buffer)[OBJ_SIZE_MSB_OFFSET] = htons(objectSize.MSB());
buffer[OBJ_SIZE_LSB_OFFSET] = htonl(objectSize.LSB());
}
void SetSegmentSize(UINT16 segmentSize)
{((UINT16*)buffer)[SEG_SIZE_OFFSET] = htons(segmentSize);}
void SetFecMaxBlockLen(UINT8 ndata)
{((UINT8*)buffer)[FEC_NDATA_OFFSET] = ndata;}
void SetFecNumParity(UINT8 nparity)
{((UINT8*)buffer)[FEC_NPARITY_OFFSET] = nparity;}
NormObjectSize GetObjectSize() const
{
return NormObjectSize(ntohs(((UINT16*)buffer)[OBJ_SIZE_MSB_OFFSET]),
ntohl(buffer[OBJ_SIZE_LSB_OFFSET]));
}
UINT16 GetSegmentSize() const
{return (ntohs(((UINT16*)buffer)[SEG_SIZE_OFFSET]));}
UINT8 GetFecMaxBlockLen() const
{return (((UINT8*)buffer)[FEC_NDATA_OFFSET]);}
UINT8 GetFecNumParity() const
{return (((UINT8*)buffer)[FEC_NPARITY_OFFSET]);}
private:
enum
{
OBJ_SIZE_MSB_OFFSET = (CONTENT_OFFSET)/2, OBJ_SIZE_LSB_OFFSET = ((OBJ_SIZE_MSB_OFFSET*2)+2)/4, SEG_SIZE_OFFSET = ((OBJ_SIZE_LSB_OFFSET*4)+4)/2, FEC_NDATA_OFFSET = ((SEG_SIZE_OFFSET+1)*2), FEC_NPARITY_OFFSET = (FEC_NDATA_OFFSET+1) };
};
class NormAppAckExtension : public NormHeaderExtension
{
public:
virtual void Init(UINT32* theBuffer, UINT16 numBytes)
{
AttachBuffer(theBuffer, numBytes);
SetType(APP_ACK);
SetWords(0);
}
bool SetContent(const char* data, UINT16 dataLen)
{
if (dataLen > (buffer_length - CONTENT_OFFSET)) return false;
memcpy(((char*)buffer) + CONTENT_OFFSET, data, dataLen);
if (dataLen > 2)
{
dataLen += CONTENT_OFFSET;
UINT16 padLen = dataLen % 4;
if (padLen)
padLen = 4 - padLen;
dataLen += padLen;
SetWords(dataLen/4);
}
else
{
SetWords(1);
}
return true;
}
};
class NormFtiExtension129 : public NormHeaderExtension
{
public:
virtual void Init(UINT32* theBuffer, UINT16 numBytes)
{
AttachBuffer(theBuffer, numBytes);
SetType(FTI);
SetWords(4);
}
void SetFecInstanceId(UINT16 instanceId)
{ ((UINT16*)buffer)[FEC_INSTANCE_OFFSET] = htons(instanceId);}
void SetFecMaxBlockLen(UINT16 ndata)
{((UINT16*)buffer)[FEC_NDATA_OFFSET] = htons(ndata);}
void SetFecNumParity(UINT16 nparity)
{((UINT16*)buffer)[FEC_NPARITY_OFFSET] = htons(nparity);}
void SetSegmentSize(UINT16 segmentSize)
{((UINT16*)buffer)[SEG_SIZE_OFFSET] = htons(segmentSize);}
void SetObjectSize(const NormObjectSize& objectSize)
{
((UINT16*)buffer)[OBJ_SIZE_MSB_OFFSET] = htons(objectSize.MSB());
buffer[OBJ_SIZE_LSB_OFFSET] = htonl(objectSize.LSB());
}
UINT16 GetFecInstanceId() const
{
return (ntohs(((UINT16*)buffer)[FEC_INSTANCE_OFFSET]));
}
UINT16 GetFecMaxBlockLen() const
{return (ntohs(((UINT16*)buffer)[FEC_NDATA_OFFSET]));}
UINT16 GetFecNumParity() const
{return (ntohs(((UINT16*)buffer)[FEC_NPARITY_OFFSET]));}
UINT16 GetSegmentSize() const
{return (ntohs(((UINT16*)buffer)[SEG_SIZE_OFFSET]));}
NormObjectSize GetObjectSize() const
{
return NormObjectSize(ntohs(((UINT16*)buffer)[OBJ_SIZE_MSB_OFFSET]),
ntohl(buffer[OBJ_SIZE_LSB_OFFSET]));
}
private:
enum
{
OBJ_SIZE_MSB_OFFSET = (LENGTH_OFFSET + 1)/2,
OBJ_SIZE_LSB_OFFSET = ((OBJ_SIZE_MSB_OFFSET*2)+2)/4,
FEC_INSTANCE_OFFSET = ((OBJ_SIZE_LSB_OFFSET*4)+4)/2,
SEG_SIZE_OFFSET = ((FEC_INSTANCE_OFFSET*2)+2)/2,
FEC_NDATA_OFFSET = ((SEG_SIZE_OFFSET*2)+2)/2,
FEC_NPARITY_OFFSET = ((FEC_NDATA_OFFSET*2)+2)/2
};
};
class NormInfoMsg : public NormObjectMsg
{
public:
void Init()
{
SetType(INFO);
SetBaseHeaderLength(INFO_HEADER_LEN);
ResetFlags();
}
UINT16 GetInfoLen() const
{return (length - header_length);}
const char* GetInfo() const
{return (((char*)buffer) + header_length);}
void SetFecId(UINT8 fecId)
{((UINT8*)buffer)[FEC_ID_OFFSET] = fecId;}
void SetInfo(const char* data, UINT16 size)
{
memcpy(((char*)buffer)+header_length, data, size);
length = size + header_length;
}
private:
enum {INFO_HEADER_LEN = OBJ_MSG_OFFSET};
};
class NormDataMsg : public NormObjectMsg
{
public:
void Init()
{
SetType(DATA);
ResetFlags();
}
void SetFecId(UINT8 fecId)
{
((UINT8*)buffer)[FEC_ID_OFFSET] = fecId;
SetBaseHeaderLength(OBJ_MSG_OFFSET + NormPayloadId::GetLength(fecId));
}
void SetFecPayloadId(UINT8 fecId, UINT32 blockId, UINT16 symbolId, UINT16 blockLen, UINT8 m)
{
NormPayloadId payloadId(fecId, m, buffer + FEC_PAYLOAD_ID_OFFSET);
payloadId.SetFecPayloadId(blockId, symbolId, blockLen);
}
char* AccessPayload() {return (((char*)buffer)+header_length);}
void SetPayloadLength(UINT16 payloadLength)
{length = header_length + payloadLength;}
void SetPayload(char* payload, UINT16 payloadLength)
{
memcpy(((char*)buffer)+header_length, payload, payloadLength);
length = header_length + payloadLength;
}
char* AccessPayloadData()
{
UINT16 payloadIndex = IsStream() ? header_length+PAYLOAD_DATA_OFFSET : header_length;
return (((char*)buffer)+payloadIndex);
}
NormBlockId GetFecBlockId(UINT8 m) const
{
NormPayloadId payloadId(GetFecId(), m, buffer + FEC_PAYLOAD_ID_OFFSET);
return payloadId.GetFecBlockId();
}
UINT16 GetFecSymbolId(UINT8 m) const
{
NormPayloadId payloadId(GetFecId(), m, buffer + FEC_PAYLOAD_ID_OFFSET);
return payloadId.GetFecSymbolId();
}
UINT16 GetFecBlockLength() const
{
NormPayloadId payloadId(GetFecId(), 8, buffer + FEC_PAYLOAD_ID_OFFSET);
return payloadId.GetFecBlockLength();
}
const char* GetPayload()
const {return (((char*)buffer)+header_length);}
UINT16 GetPayloadLength()
const {return (length - header_length);}
const char* GetPayloadData() const
{
UINT16 dataIndex = IsStream() ? header_length+PAYLOAD_DATA_OFFSET : header_length;
return (((char*)buffer)+dataIndex);
}
UINT16 GetPayloadDataLength() const
{
UINT16 dataIndex = IsStream() ? header_length+PAYLOAD_DATA_OFFSET : header_length;
return (length - dataIndex);
}
static UINT16 GetStreamPayloadHeaderLength()
{return (PAYLOAD_DATA_OFFSET);}
static void WriteStreamPayloadLength(char* payload, UINT16 len)
{
UINT16 temp16 = htons(len);
memcpy(payload+PAYLOAD_LENGTH_OFFSET, &temp16, 2);
}
static void WriteStreamPayloadMsgStart(char* payload, UINT16 msgStartOffset)
{
UINT16 temp16 = htons(msgStartOffset);
memcpy(payload+PAYLOAD_MSG_START_OFFSET, &temp16, 2);
}
static void WriteStreamPayloadOffset(char* payload, UINT32 offset)
{
UINT32 temp32 = htonl(offset);
memcpy(payload+PAYLOAD_OFFSET_OFFSET, &temp32, 4);
}
static UINT16 ReadStreamPayloadLength(const char* payload)
{
UINT16 temp16;
memcpy(&temp16, payload+PAYLOAD_LENGTH_OFFSET, 2);
return (ntohs(temp16));
}
static UINT16 ReadStreamPayloadMsgStart(const char* payload)
{
UINT16 temp16;
memcpy(&temp16, payload+PAYLOAD_MSG_START_OFFSET, 2);
return (ntohs(temp16));
}
static UINT32 ReadStreamPayloadOffset(const char* payload)
{
UINT32 temp32;
memcpy(&temp32, payload+PAYLOAD_OFFSET_OFFSET, 4);
return (ntohl(temp32));
}
private:
enum
{
FEC_PAYLOAD_ID_OFFSET = OBJ_MSG_OFFSET/4
};
enum
{
PAYLOAD_LENGTH_OFFSET = 0,
PAYLOAD_MSG_START_OFFSET = PAYLOAD_LENGTH_OFFSET+2,
PAYLOAD_OFFSET_OFFSET = PAYLOAD_MSG_START_OFFSET+2,
PAYLOAD_DATA_OFFSET = PAYLOAD_OFFSET_OFFSET+4
};
};
class NormCmdMsg : public NormMsg
{
public:
enum Flavor
{
INVALID = 0,
FLUSH = 1,
EOT = 2,
SQUELCH = 3,
CC = 4,
REPAIR_ADV = 5,
ACK_REQ = 6,
APPLICATION = 7
};
void SetInstanceId(UINT16 instanceId)
{((UINT16*)buffer)[INSTANCE_ID_OFFSET] = htons(instanceId);}
void SetGrtt(UINT8 quantizedGrtt)
{((UINT8*)buffer)[GRTT_OFFSET] = quantizedGrtt;}
void SetBackoffFactor(UINT8 backoff)
{((UINT8*)buffer)[BACKOFF_OFFSET] = (((UINT8*)buffer)[GSIZE_OFFSET] & 0x0f) | (backoff << 4);}
void SetGroupSize(UINT8 gsize)
{((UINT8*)buffer)[GSIZE_OFFSET] = (((UINT8*)buffer)[GSIZE_OFFSET] & 0xf0) | gsize;}
void SetFlavor(NormCmdMsg::Flavor flavor)
{((UINT8*)buffer)[FLAVOR_OFFSET] = (UINT8)flavor;}
UINT16 GetInstanceId() const
{return (ntohs(((UINT16*)buffer)[INSTANCE_ID_OFFSET]));}
UINT8 GetGrtt() const
{return ((UINT8*)buffer)[GRTT_OFFSET];}
UINT8 GetBackoffFactor() const
{return ((((UINT8*)buffer)[GSIZE_OFFSET] >> 4) & 0x0f);}
UINT8 GetGroupSize() const
{return (((UINT8*)buffer)[GSIZE_OFFSET] & 0x0f);}
NormCmdMsg::Flavor GetFlavor() const
{return (Flavor)((UINT8*)buffer)[FLAVOR_OFFSET];}
protected:
friend class NormMsg;
enum
{
INSTANCE_ID_OFFSET = MSG_OFFSET/2,
GRTT_OFFSET = (INSTANCE_ID_OFFSET+1)*2,
BACKOFF_OFFSET = GRTT_OFFSET + 1,
GSIZE_OFFSET = BACKOFF_OFFSET,
FLAVOR_OFFSET = GSIZE_OFFSET + 1
};
};
class NormCmdFlushMsg : public NormCmdMsg
{
friend class NormMsg;
public:
void Init()
{
SetType(CMD);
SetFlavor(FLUSH);
}
void SetFecId(UINT8 fecId)
{((UINT8*)buffer)[FEC_ID_OFFSET] = fecId;}
void SetObjectId(const NormObjectId& objectId)
{((UINT16*)buffer)[OBJ_ID_OFFSET] = htons((UINT16)objectId);}
void SetFecPayloadId(UINT8 fecId, UINT32 blockId, UINT16 symbolId, UINT16 blockLen, UINT8 m)
{
SetFecId(fecId);
SetBaseHeaderLength(4*FEC_PAYLOAD_ID_OFFSET + NormPayloadId::GetLength(fecId));
NormPayloadId payloadId(fecId, m, buffer + FEC_PAYLOAD_ID_OFFSET);
payloadId.SetFecPayloadId(blockId, symbolId, blockLen);
ResetAckingNodeList();
}
void ResetAckingNodeList()
{length = header_length;}
bool AppendAckingNode(NormNodeId nodeId, UINT16 segmentSize)
{
if ((length-header_length + 4) > segmentSize) return false;
buffer[length/4] = htonl((UINT32)nodeId);
length += 4;
return true;
}
UINT8 GetFecId() const
{return ((UINT8*)buffer)[FEC_ID_OFFSET];}
NormObjectId GetObjectId() const
{ return ntohs(((UINT16*)buffer)[OBJ_ID_OFFSET]);}
NormBlockId GetFecBlockId(UINT8 m) const
{
NormPayloadId payloadId(GetFecId(), m, buffer + FEC_PAYLOAD_ID_OFFSET);
return payloadId.GetFecBlockId();
}
UINT16 GetFecSymbolId(UINT8 m) const
{
NormPayloadId payloadId(GetFecId(), m, buffer + FEC_PAYLOAD_ID_OFFSET);
return payloadId.GetFecSymbolId();
}
UINT16 GetFecBlockLength() const
{
NormPayloadId payloadId(GetFecId(), 8, buffer + FEC_PAYLOAD_ID_OFFSET);
return payloadId.GetFecBlockLength();
}
UINT16 GetAckingNodeCount() const
{return ((length - header_length) >> 2);}
const UINT32* GetAckingNodeList() const
{return (buffer+(header_length/4));}
NormNodeId GetAckingNodeId(UINT16 index) const
{return (ntohl(buffer[(header_length/4)+index]));}
private:
enum
{
FEC_ID_OFFSET = FLAVOR_OFFSET + 1,
OBJ_ID_OFFSET = (FEC_ID_OFFSET + 1)/2,
FEC_PAYLOAD_ID_OFFSET = ((OBJ_ID_OFFSET+1)*2)/4
};
};
class NormCmdEotMsg : public NormCmdMsg
{
public:
void Init()
{
SetType(CMD);
SetFlavor(EOT);
SetBaseHeaderLength(EOT_HEADER_LEN);
memset(((char*)buffer)+RESERVED_OFFSET, 0, 3);
}
private:
enum
{
RESERVED_OFFSET = FLAVOR_OFFSET + 1,
EOT_HEADER_LEN = RESERVED_OFFSET + 3
};
};
class NormCmdSquelchMsg : public NormCmdMsg
{
public:
void Init(UINT8 fecId)
{
SetType(CMD);
SetFlavor(SQUELCH);
SetFecId(fecId); SetBaseHeaderLength(4*FEC_PAYLOAD_ID_OFFSET + NormPayloadId::GetLength(fecId));
}
void SetFecId(UINT8 fecId)
{((UINT8*)buffer)[FEC_ID_OFFSET] = fecId;}
void SetObjectId(const NormObjectId& objectId)
{((UINT16*)buffer)[OBJ_ID_OFFSET] = htons((UINT16)objectId);}
void SetFecPayloadId(UINT8 fecId, UINT32 blockId, UINT16 symbolId, UINT16 blockLen, UINT8 m)
{
SetFecId(fecId);
SetBaseHeaderLength(4*FEC_PAYLOAD_ID_OFFSET + NormPayloadId::GetLength(fecId));
NormPayloadId payloadId(fecId, m, buffer + FEC_PAYLOAD_ID_OFFSET);
payloadId.SetFecPayloadId(blockId, symbolId, blockLen);
ResetInvalidObjectList();
}
void ResetInvalidObjectList()
{length = header_length;}
bool AppendInvalidObject(NormObjectId objectId, UINT16 segmentSize)
{
if ((length-header_length+2) > segmentSize) return false;
((UINT16*)buffer)[length/2] = htons((UINT16)objectId);
length += 2;
return true;
}
UINT8 GetFecId() const
{return ((UINT8*)buffer)[FEC_ID_OFFSET];}
NormObjectId GetObjectId() const
{return (ntohs(((UINT16*)buffer)[OBJ_ID_OFFSET]));}
NormBlockId GetFecBlockId(UINT8 m) const
{
NormPayloadId payloadId(GetFecId(), m, buffer + FEC_PAYLOAD_ID_OFFSET);
return payloadId.GetFecBlockId();
}
UINT16 GetFecSymbolId(UINT8 m) const
{
NormPayloadId payloadId(GetFecId(), m, buffer + FEC_PAYLOAD_ID_OFFSET);
return payloadId.GetFecSymbolId();
}
UINT16 GetFecBlockLength() const
{
NormPayloadId payloadId(GetFecId(), 8, buffer + FEC_PAYLOAD_ID_OFFSET);
return payloadId.GetFecBlockLength();
}
UINT16 GetInvalidObjectCount() const
{return ((length - header_length) >> 1);}
UINT16* GetInvalidObjectList() const
{return (UINT16*)(buffer+header_length);}
NormObjectId GetInvalidObjectId(UINT16 index) const
{return (ntohs(((UINT16*)buffer)[(header_length/2)+index]));}
private:
enum
{
FEC_ID_OFFSET = FLAVOR_OFFSET + 1,
OBJ_ID_OFFSET = (FEC_ID_OFFSET + 1)/2,
FEC_PAYLOAD_ID_OFFSET = ((OBJ_ID_OFFSET+1)*2)/4
};
};
class NormCC
{
public:
enum Flag
{
CLR = 0x01,
PLR = 0x02,
RTT = 0x04,
START = 0x08,
LEAVE = 0x10,
LIMIT = 0x20 }; };
class NormCmdCCMsg : public NormCmdMsg
{
public:
void Init()
{
SetType(CMD);
SetFlavor(CC);
SetBaseHeaderLength(CC_HEADER_LEN);
((UINT8*)buffer)[RESERVED_OFFSET] = 0;
}
void SetCCSequence(UINT16 ccSequence)
{
((UINT16*)buffer)[CC_SEQUENCE_OFFSET] = htons(ccSequence);
}
void SetSendTime(const struct timeval& sendTime)
{
buffer[SEND_TIME_SEC_OFFSET] = htonl(sendTime.tv_sec);
buffer[SEND_TIME_USEC_OFFSET] = htonl(sendTime.tv_usec);
}
UINT16 GetCCSequence() const
{return (ntohs(((UINT16*)buffer)[CC_SEQUENCE_OFFSET]));}
void GetSendTime(struct timeval& sendTime) const
{
sendTime.tv_sec = ntohl(buffer[SEND_TIME_SEC_OFFSET]);
sendTime.tv_usec = ntohl(buffer[SEND_TIME_USEC_OFFSET]);
}
bool AppendCCNode(UINT16 segMax, NormNodeId nodeId, UINT8 flags,
UINT8 rtt, UINT16 rate)
{
if ((length-header_length+CC_ITEM_SIZE)> segMax) return false;
UINT32* ptr = buffer + length/4;
ptr[CC_NODE_ID_OFFSET] = htonl(nodeId);
((UINT8*)ptr)[CC_FLAGS_OFFSET] = flags;
((UINT8*)ptr)[CC_RTT_OFFSET] = rtt;
((UINT16*)ptr)[CC_RATE_OFFSET] = htons(rate);
length += CC_ITEM_SIZE;
return true;
}
bool GetCCNode(NormNodeId nodeId, UINT8& flags, UINT8& rtt, UINT16& rate) const;
enum {FLAG_SYN = 0x01};
bool SynIsSet() const
{return (0 != (FLAG_SYN & ((UINT8*)buffer)[RESERVED_OFFSET]));}
void SetSyn()
{((UINT8*)buffer)[RESERVED_OFFSET] = FLAG_SYN;}
class Iterator;
friend class Iterator;
class Iterator
{
public:
Iterator(const NormCmdCCMsg& msg);
void Reset() {offset = 0;}
bool GetNextNode(NormNodeId& nodeId, UINT8& flags, UINT8& rtt, UINT16& rate);
private:
const NormCmdCCMsg& cc_cmd;
UINT16 offset;
};
private:
enum
{
RESERVED_OFFSET = FLAVOR_OFFSET + 1,
CC_SEQUENCE_OFFSET = (RESERVED_OFFSET + 1)/2,
SEND_TIME_SEC_OFFSET = ((CC_SEQUENCE_OFFSET*2)+2)/4,
SEND_TIME_USEC_OFFSET = ((SEND_TIME_SEC_OFFSET*4)+4)/4,
CC_HEADER_LEN = (SEND_TIME_USEC_OFFSET*4)+4
};
enum
{
CC_NODE_ID_OFFSET = 0,
CC_FLAGS_OFFSET = CC_NODE_ID_OFFSET + 4,
CC_RTT_OFFSET = CC_FLAGS_OFFSET + 1,
CC_RATE_OFFSET = (CC_RTT_OFFSET + 1)/2,
CC_ITEM_SIZE = (CC_RATE_OFFSET*2)+2
};
};
class NormCCRateExtension : public NormHeaderExtension
{
public:
virtual void Init(UINT32* theBuffer, UINT16 numBytes)
{
AttachBuffer(theBuffer, numBytes);
SetType(CC_RATE);
((UINT8*)buffer)[RESERVED_OFFSET] = 0;
}
void SetSendRate(UINT16 sendRate)
{((UINT16*)buffer)[SEND_RATE_OFFSET] = htons(sendRate);}
UINT16 GetSendRate()
{return (ntohs(((UINT16*)buffer)[SEND_RATE_OFFSET]));}
private:
enum
{
RESERVED_OFFSET = TYPE_OFFSET + 1,
SEND_RATE_OFFSET = (RESERVED_OFFSET + 1)/2
};
};
class NormRepairRequest
{
public:
class Iterator;
friend class NormRepairRequest::Iterator;
enum Form
{
INVALID = 0,
ITEMS = 1,
RANGES = 2,
ERASURES = 3
};
enum Flag
{
SEGMENT = 0x01,
BLOCK = 0x02,
INFO = 0x04,
OBJECT = 0x08
};
NormRepairRequest();
void Init(UINT32* bufferPtr, UINT16 bufferLen)
{
buffer = bufferPtr;
buffer_len = bufferLen;
length = 0;
}
static UINT16 RepairItemLength(UINT8 fecId)
{return (4 + NormPayloadId::GetLength(fecId));}
static UINT16 RepairRangeLength(UINT8 fecId)
{return (2 * RepairItemLength(fecId));}
static UINT16 ErasureItemLength(UINT8 fecId)
{return RepairItemLength(fecId);}
void SetForm(NormRepairRequest::Form theForm)
{form = theForm;}
void ResetFlags()
{flags = 0;}
void SetFlag(NormRepairRequest::Flag theFlag)
{flags |= theFlag;}
void ClearFlag(NormRepairRequest::Flag theFlag)
{flags &= ~theFlag;}
void SetFlags(int theFlags)
{flags = theFlags;}
bool AppendRepairItem(UINT8 fecId,
UINT8 fecM,
const NormObjectId& objectId,
const NormBlockId& blockId,
UINT16 blockLen,
UINT16 symbolId);
bool AppendRepairRange(UINT8 fecId,
UINT8 fecM,
const NormObjectId& startObjectId,
const NormBlockId& startBlockId,
UINT16 startBlockLen,
UINT16 startSymbolId,
const NormObjectId& endObjectId,
const NormBlockId& endBlockId,
UINT16 endBlockLen,
UINT16 endSymbolId);
bool AppendErasureCount(UINT8 fecId,
UINT8 fecM,
const NormObjectId& objectId,
const NormBlockId& blockId,
UINT16 blockLen,
UINT16 erasureCount);
UINT16 Pack();
UINT16 Unpack(const UINT32* bufferPtr, UINT16 bufferLen);
NormRepairRequest::Form GetForm() const
{return form;}
bool FlagIsSet(NormRepairRequest::Flag theFlag) const
{return (0 != (theFlag & flags));}
int GetFlags() const
{return flags;}
UINT16 GetLength() const
{return (ITEM_LIST_OFFSET + length);}
UINT32* GetBuffer() const
{return buffer;}
void Log(UINT8 fecId, UINT8 fecM) const;
class Iterator
{
public:
Iterator(const NormRepairRequest& theRequest, UINT8 fecId, UINT8 fecM);
void Reset() {offset = 0;}
UINT16 NextRepairItem(NormObjectId* objectId,
NormBlockId* blockId,
UINT16* blockLen,
UINT16* symbolId);
private:
const NormRepairRequest& request;
UINT8 fec_id;
UINT8 fec_m;
UINT16 offset;
};
private:
UINT16 RetrieveRepairItem(UINT8 fecM,
UINT16 offset,
UINT8* fecId,
NormObjectId* objectId,
NormBlockId* blockId,
UINT16* blockLen,
UINT16* symbolId) const;
enum
{
FORM_OFFSET = 0,
FLAGS_OFFSET = FORM_OFFSET + 1,
LENGTH_OFFSET = (FLAGS_OFFSET + 1)/2,
ITEM_LIST_OFFSET = (LENGTH_OFFSET*2)+2
};
enum
{
FEC_ID_OFFSET = 0,
RESERVED_OFFSET = FEC_ID_OFFSET + 1,
OBJ_ID_OFFSET = (RESERVED_OFFSET + 1)/2,
FEC_PAYLOAD_ID_OFFSET = ((OBJ_ID_OFFSET+1)*2)/4
};
Form form;
int flags;
UINT16 length; UINT32* buffer;
UINT16 buffer_len;
};
class NormCmdRepairAdvMsg : public NormCmdMsg
{
public:
enum Flag {NORM_REPAIR_ADV_FLAG_LIMIT = 0x01};
void Init()
{
SetType(CMD);
SetFlavor(REPAIR_ADV);
SetBaseHeaderLength(REPAIR_ADV_HEADER_LEN);
ResetFlags();
((UINT16*)buffer)[RESERVED_OFFSET] = 0;
}
void ResetFlags() {((UINT8*)buffer)[FLAGS_OFFSET] = 0;}
void SetFlag(NormCmdRepairAdvMsg::Flag flag)
{((UINT8*)buffer)[FLAGS_OFFSET] |= (UINT8)flag;}
void AttachRepairRequest(NormRepairRequest& request,
UINT16 segmentMax)
{
int buflen = segmentMax - (length - header_length);
buflen = (buflen>0) ? buflen : 0;
request.Init(buffer+length/4, buflen);
}
UINT16 PackRepairRequest(NormRepairRequest& request)
{
UINT16 requestLength = request.Pack();
length += requestLength;
return requestLength;
}
bool FlagIsSet(NormCmdRepairAdvMsg::Flag flag) const
{return (0 != ((UINT8)flag | ((UINT8*)buffer)[FLAGS_OFFSET]));}
const UINT32* GetRepairContent() const
{return (buffer + header_length/4);}
UINT16 GetRepairContentLength() const
{return (length - header_length);}
private:
enum
{
FLAGS_OFFSET = FLAVOR_OFFSET + 1,
RESERVED_OFFSET = FLAGS_OFFSET + 1,
REPAIR_ADV_HEADER_LEN = RESERVED_OFFSET + 2
};
};
class NormCCFeedbackExtension : public NormHeaderExtension
{
public:
virtual void Init(UINT32* theBuffer, UINT16 numBytes)
{
AttachBuffer(theBuffer, numBytes);
SetType(CC_FEEDBACK);
SetWords(3);
((UINT8*)buffer)[CC_FLAGS_OFFSET] = 0;
}
void SetCCSequence(UINT16 ccSequence)
{((UINT16*)buffer)[CC_SEQUENCE_OFFSET] = htons(ccSequence);}
void ResetCCFlags()
{((UINT8*)buffer)[CC_FLAGS_OFFSET] = 0;}
void SetCCFlag(NormCC::Flag flag)
{((UINT8*)buffer)[CC_FLAGS_OFFSET] |= (UINT8)flag;}
void SetCCRtt(UINT8 ccRtt)
{((UINT8*)buffer)[CC_RTT_OFFSET] = ccRtt;}
void SetCCRate(UINT16 ccRate)
{((UINT16*)buffer)[CC_RATE_OFFSET] = htons(ccRate);}
void SetCCLoss32(UINT32 ccLoss)
{
ccLoss = htonl(ccLoss);
UINT16* ptr = (UINT16*)&ccLoss;
((UINT16*)buffer)[CC_LOSS_OFFSET] = ptr[0]; ((UINT16*)buffer)[CC_LOSS_EX_OFFSET] = ptr[1]; }
UINT16 GetCCSequence() const
{return (ntohs(((UINT16*)buffer)[CC_SEQUENCE_OFFSET]));}
UINT8 GetCCFlags()
{return ((UINT8*)buffer)[CC_FLAGS_OFFSET];}
bool CCFlagIsSet(NormCC::Flag flag) const
{return (0 != ((UINT8)flag & ((UINT8*)buffer)[CC_FLAGS_OFFSET]));}
UINT8 GetCCRtt() const
{return ((UINT8*)buffer)[CC_RTT_OFFSET];}
UINT16 GetCCRate()const
{return (ntohs(((UINT16*)buffer)[CC_RATE_OFFSET]));}
UINT32 GetCCLoss32() const
{
UINT32 lossQuantized;
UINT16* ptr = (UINT16*)&lossQuantized;
ptr[0] = ((UINT16*)buffer)[CC_LOSS_OFFSET]; ptr[1] = ((UINT16*)buffer)[CC_LOSS_EX_OFFSET]; return ntohl(lossQuantized); }
private:
enum
{
CC_SEQUENCE_OFFSET = (LENGTH_OFFSET+1)/2,
CC_FLAGS_OFFSET = (CC_SEQUENCE_OFFSET*2)+2,
CC_RTT_OFFSET = CC_FLAGS_OFFSET + 1,
CC_LOSS_OFFSET = (CC_RTT_OFFSET + 1)/2,
CC_RATE_OFFSET = ((CC_LOSS_OFFSET*2)+2)/2,
CC_LOSS_EX_OFFSET = ((CC_RATE_OFFSET*2)+2)/2 };
};
class NormAck
{
public:
enum Type
{
INVALID = 0,
CC = 1,
FLUSH = 2,
APP_BASE = 16
};
};
class NormCmdAckReqMsg : public NormCmdMsg
{
public:
void Init()
{
SetType(CMD);
SetFlavor(ACK_REQ);
SetBaseHeaderLength(ACK_REQ_HEADER_LEN);
((UINT8*)buffer)[RESERVED_OFFSET] = 0;
}
void SetAckType(NormAck::Type ackType)
{((UINT8*)buffer)[ACK_TYPE_OFFSET] = (UINT8)ackType;}
void SetAckId(UINT8 ackId)
{((UINT8*)buffer)[ACK_ID_OFFSET] = ackId;}
void ResetAckingNodeList()
{length = header_length;}
bool AppendAckingNode(NormNodeId nodeId, UINT16 segmentSize)
{
if ((length - header_length + 4) > segmentSize) return false;
buffer[length/4] = htonl(nodeId);
length += 4;
return true;
}
NormAck::Type GetAckType() const
{return (NormAck::Type)(((UINT8*)buffer)[ACK_TYPE_OFFSET]);}
UINT8 GetAckId() const
{return ((UINT8*)buffer)[ACK_ID_OFFSET];}
UINT16 GetAckingNodeCount() const
{return ((length - header_length) >> 2);}
NormNodeId GetAckingNodeId(UINT16 index) const
{return (ntohl(buffer[(header_length/4)+index]));}
private:
enum
{
RESERVED_OFFSET = FLAVOR_OFFSET + 1,
ACK_TYPE_OFFSET = RESERVED_OFFSET + 1,
ACK_ID_OFFSET = ACK_TYPE_OFFSET + 1,
ACK_REQ_HEADER_LEN = ACK_ID_OFFSET + 1
};
};
class NormCmdAppMsg : public NormCmdMsg
{
public:
void Init()
{
SetType(CMD);
SetFlavor(APPLICATION);
SetBaseHeaderLength(APPLICATION_HEADER_LEN);
memset(((UINT8*)buffer)+RESERVED_OFFSET, 0, 3);
}
bool SetContent(const char* content, UINT16 contentLen, UINT16 segmentSize)
{
UINT16 len = MIN(contentLen, segmentSize);
memcpy(((char*)buffer)+header_length, content, len);
length = header_length + len;
return (contentLen <= segmentSize);
}
UINT16 GetContentLength() const
{return (length - header_length);}
const char* GetContent() const
{return (((char*)buffer)+header_length);}
private:
enum
{
RESERVED_OFFSET = FLAVOR_OFFSET + 1,
APPLICATION_HEADER_LEN = RESERVED_OFFSET + 3
};
};
class NormNackMsg : public NormMsg
{
public:
enum {DEFAULT_LENGTH_MAX = 40};
void Init()
{
SetType(NACK);
((UINT16*)buffer)[RESERVED_OFFSET] = 0;
SetBaseHeaderLength(NACK_HEADER_LEN);
}
void SetSenderId(NormNodeId senderId)
{buffer[SENDER_ID_OFFSET] = htonl(senderId);}
void SetInstanceId(UINT16 instanceId)
{((UINT16*)buffer)[INSTANCE_ID_OFFSET] = htons(instanceId);}
void SetGrttResponse(const struct timeval& grttResponse)
{
buffer[GRTT_RESPONSE_SEC_OFFSET] = htonl(grttResponse.tv_sec);
buffer[GRTT_RESPONSE_USEC_OFFSET] = htonl(grttResponse.tv_usec);
}
void AttachRepairRequest(NormRepairRequest& request,
UINT16 segmentMax)
{
int buflen = segmentMax - (length - header_length);
buflen = (buflen>0) ? buflen : 0;
request.Init(buffer+length/4, buflen);
}
UINT16 PackRepairRequest(NormRepairRequest& request)
{
UINT16 requestLength = request.Pack();
length += requestLength;
return requestLength;
}
void InitFrom(NormNackMsg nack)
{
memcpy(buffer, nack.buffer, nack.GetHeaderLength());
header_length_base = nack.header_length_base;
length = header_length = nack.GetHeaderLength();
}
void AppendRepairRequest(const NormRepairRequest request)
{
memcpy(buffer+length/4, request.GetBuffer(), request.GetLength());
length += request.GetLength();
}
void ResetPayload()
{length = GetHeaderLength();}
NormNodeId GetSenderId() const
{return (ntohl(buffer[SENDER_ID_OFFSET]));}
UINT16 GetInstanceId() const
{return (ntohs(((UINT16*)buffer)[INSTANCE_ID_OFFSET]));}
void GetGrttResponse(struct timeval& grttResponse) const
{
grttResponse.tv_sec = ntohl(buffer[GRTT_RESPONSE_SEC_OFFSET]);
grttResponse.tv_usec = ntohl(buffer[GRTT_RESPONSE_USEC_OFFSET]);
}
const UINT32* GetRepairContent() const
{return (buffer + header_length/4);}
UINT16 GetRepairContentLength() const
{return ((length > header_length) ? length - header_length : 0);}
UINT16 UnpackRepairRequest(NormRepairRequest& request,
UINT16 requestOffset)
{
int buflen = length - header_length - requestOffset;
buflen = (buflen > 0) ? buflen : 0;
return request.Unpack(buffer+(header_length+requestOffset)/4, buflen);
}
private:
enum
{
SENDER_ID_OFFSET = MSG_OFFSET/4,
INSTANCE_ID_OFFSET = ((SENDER_ID_OFFSET*4)+4)/2,
RESERVED_OFFSET = ((INSTANCE_ID_OFFSET*2)+2)/2,
GRTT_RESPONSE_SEC_OFFSET = ((RESERVED_OFFSET*2)+2)/4,
GRTT_RESPONSE_USEC_OFFSET = ((GRTT_RESPONSE_SEC_OFFSET*4)+4)/4,
NACK_HEADER_LEN = (GRTT_RESPONSE_USEC_OFFSET*4)+4
};
};
class NormAckMsg : public NormAck, public NormMsg
{
public:
void Init()
{
SetType(ACK);
SetBaseHeaderLength(ACK_HEADER_LEN);
SetAckType(NormAck::INVALID);
}
void SetSenderId(NormNodeId senderId)
{buffer[SENDER_ID_OFFSET] = htonl(senderId);}
void SetInstanceId(UINT16 instanceId)
{((UINT16*)buffer)[INSTANCE_ID_OFFSET] = htons(instanceId);}
void SetAckType(NormAck::Type ackType)
{((UINT8*)buffer)[ACK_TYPE_OFFSET] = (UINT8)ackType;}
void SetAckId(UINT8 ackId)
{((UINT8*)buffer)[ACK_ID_OFFSET] = ackId;}
void SetGrttResponse(const struct timeval& grttResponse)
{
buffer[GRTT_RESPONSE_SEC_OFFSET] = htonl(grttResponse.tv_sec);
buffer[GRTT_RESPONSE_USEC_OFFSET] = htonl(grttResponse.tv_usec);
}
bool SetAckPayload(const char* payload, UINT16 payloadLen, UINT16 segmentSize)
{
UINT16 len = MIN(payloadLen, segmentSize);
memcpy(((char*)buffer)+header_length, payload, len);
length += len;
return (payloadLen <= segmentSize);
}
NormNodeId GetSenderId() const
{return (ntohl(buffer[SENDER_ID_OFFSET]));}
UINT16 GetInstanceId() const
{return (ntohs(((UINT16*)buffer)[INSTANCE_ID_OFFSET]));}
void GetGrttResponse(struct timeval& grttResponse) const
{
grttResponse.tv_sec = ntohl(buffer[GRTT_RESPONSE_SEC_OFFSET]);
grttResponse.tv_usec = ntohl(buffer[GRTT_RESPONSE_USEC_OFFSET]);
}
NormAck::Type GetAckType() const
{return (NormAck::Type)((UINT8*)buffer)[ACK_TYPE_OFFSET];}
UINT8 GetAckId() const
{return ((UINT8*)buffer)[ACK_ID_OFFSET];}
UINT16 GetPayloadLength() const
{return (length - header_length);}
const char* GetPayload() const
{return (((char*)buffer) + header_length);}
protected:
enum
{
SENDER_ID_OFFSET = MSG_OFFSET/4,
INSTANCE_ID_OFFSET = ((SENDER_ID_OFFSET*4)+4)/2,
ACK_TYPE_OFFSET = (INSTANCE_ID_OFFSET*2)+2,
ACK_ID_OFFSET = ACK_TYPE_OFFSET + 1,
GRTT_RESPONSE_SEC_OFFSET = (ACK_ID_OFFSET + 1)/4,
GRTT_RESPONSE_USEC_OFFSET = ((GRTT_RESPONSE_SEC_OFFSET+1)*4)/4,
ACK_HEADER_LEN = (GRTT_RESPONSE_USEC_OFFSET+1)*4
};
};
class NormAckFlushMsg : public NormAckMsg
{
public:
void Init()
{
SetType(ACK);
SetBaseHeaderLength(ACK_HEADER_LEN);
SetAckType(NormAck::FLUSH);
((UINT8*)buffer)[RESERVED_OFFSET] = 0;
}
void SetFecId(UINT8 fecId)
{((UINT8*)buffer)[header_length+FEC_ID_OFFSET] = fecId;}
void SetObjectId(NormObjectId objectId)
{((UINT16*)buffer)[(header_length/2)+OBJ_ID_OFFSET] = htons((UINT16)objectId);}
void SetFecPayloadId(UINT8 fecId, UINT32 blockId, UINT16 symbolId, UINT16 blockLen, UINT8 m)
{
SetFecId(fecId);
((UINT8*)buffer)[header_length+RESERVED_OFFSET] = 0;
NormPayloadId payloadId(fecId, m, buffer + header_length/4 + FEC_PAYLOAD_ID_OFFSET);
payloadId.SetFecPayloadId(blockId, symbolId, blockLen);
length = header_length + 4*FEC_PAYLOAD_ID_OFFSET + NormPayloadId::GetLength(fecId);
}
UINT8 GetFecId() const
{return ((UINT8*)buffer)[header_length+FEC_ID_OFFSET];}
NormObjectId GetObjectId() const
{return ntohs(((UINT16*)buffer)[(header_length/2)+OBJ_ID_OFFSET]);}
NormBlockId GetFecBlockId(UINT8 m) const
{
NormPayloadId payloadId(GetFecId(), m, buffer + header_length/4 + FEC_PAYLOAD_ID_OFFSET);
return payloadId.GetFecBlockId();
}
UINT16 GetFecSymbolId(UINT8 m) const
{
NormPayloadId payloadId(GetFecId(), m, buffer + header_length/4 + FEC_PAYLOAD_ID_OFFSET);
return payloadId.GetFecSymbolId();
}
UINT16 GetFecBlockLength() const
{
NormPayloadId payloadId(GetFecId(), 8, buffer + header_length/4 + FEC_PAYLOAD_ID_OFFSET);
return payloadId.GetFecBlockLength();
}
private:
enum
{
FEC_ID_OFFSET = 0,
RESERVED_OFFSET = FEC_ID_OFFSET + 1,
OBJ_ID_OFFSET = (RESERVED_OFFSET+1)/2,
FEC_PAYLOAD_ID_OFFSET = ((OBJ_ID_OFFSET*2)+2)/4
};
};
class NormReportMsg : public NormMsg
{
};
class NormMessageQueue
{
public:
NormMessageQueue();
~NormMessageQueue();
void Destroy();
void Prepend(NormMsg* msg);
void Append(NormMsg* msg);
void Remove(NormMsg* msg);
NormMsg* RemoveHead();
NormMsg* RemoveTail();
NormMsg* GetHead() {return head;}
bool IsEmpty() {return ((NormMsg*)NULL == head);}
private:
NormMsg* head;
NormMsg* tail;
};
void LogRepairContent(const UINT32* buffer, UINT16 bufferLen, UINT8 fecId, UINT8 fecM);
#endif