Added Dsdv Data transport packet.

Shifted Dsdv address to separate TLV data type.



git-svn-id: http://svn.code.sf.net/p/retroshare/code/branches/v0.5-dhtmods@4679 b45a01b8-16f6-495d-af2f-9b41ad6348cc
This commit is contained in:
drbob 2011-11-21 14:38:13 +00:00
parent 7f30b19f2a
commit 1cfcac7377
6 changed files with 376 additions and 53 deletions

View File

@ -56,7 +56,7 @@ std::ostream &RsDsdvRouteItem::print(std::ostream &out, uint16_t indent)
} }
uint32_t RsDsdvSerialiser::sizeItem(RsDsdvRouteItem *item) uint32_t RsDsdvSerialiser::sizeRoute(RsDsdvRouteItem *item)
{ {
uint32_t s = 8; /* header */ uint32_t s = 8; /* header */
s += item->routes.TlvSize(); s += item->routes.TlvSize();
@ -65,9 +65,9 @@ uint32_t RsDsdvSerialiser::sizeItem(RsDsdvRouteItem *item)
} }
/* serialise the data to the buffer */ /* serialise the data to the buffer */
bool RsDsdvSerialiser::serialiseItem(RsDsdvRouteItem *item, void *data, uint32_t *pktsize) bool RsDsdvSerialiser::serialiseRoute(RsDsdvRouteItem *item, void *data, uint32_t *pktsize)
{ {
uint32_t tlvsize = sizeItem(item); uint32_t tlvsize = sizeRoute(item);
uint32_t offset = 0; uint32_t offset = 0;
if (*pktsize < tlvsize) if (*pktsize < tlvsize)
@ -80,8 +80,8 @@ bool RsDsdvSerialiser::serialiseItem(RsDsdvRouteItem *item, void *data, uint
ok &= setRsItemHeader(data, tlvsize, item->PacketId(), tlvsize); ok &= setRsItemHeader(data, tlvsize, item->PacketId(), tlvsize);
#ifdef RSSERIAL_DEBUG #ifdef RSSERIAL_DEBUG
std::cerr << "RsDsdvSerialiser::serialiseItem() Header: " << ok << std::endl; std::cerr << "RsDsdvSerialiser::serialiseRoute() Header: " << ok << std::endl;
std::cerr << "RsDsdvSerialiser::serialiseItem() Size: " << tlvsize << std::endl; std::cerr << "RsDsdvSerialiser::serialiseRoute() Size: " << tlvsize << std::endl;
#endif #endif
/* skip the header */ /* skip the header */
@ -94,14 +94,14 @@ bool RsDsdvSerialiser::serialiseItem(RsDsdvRouteItem *item, void *data, uint
{ {
ok = false; ok = false;
#ifdef RSSERIAL_DEBUG #ifdef RSSERIAL_DEBUG
std::cerr << "RsDsdvSerialiser::serialiseItem() Size Error! " << std::endl; std::cerr << "RsDsdvSerialiser::serialiseRoute() Size Error! " << std::endl;
#endif #endif
} }
return ok; return ok;
} }
RsDsdvRouteItem *RsDsdvSerialiser::deserialiseItem(void *data, uint32_t *pktsize) RsDsdvRouteItem *RsDsdvSerialiser::deserialiseRoute(void *data, uint32_t *pktsize)
{ {
/* get the type and size */ /* get the type and size */
uint32_t rstype = getRsItemId(data); uint32_t rstype = getRsItemId(data);
@ -151,20 +151,199 @@ RsDsdvRouteItem *RsDsdvSerialiser::deserialiseItem(void *data, uint32_t *pktsize
return item; return item;
} }
/*************************************************************************/
uint32_t RsDsdvSerialiser::size(RsItem *item) RsDsdvDataItem::~RsDsdvDataItem()
{ {
return sizeItem((RsDsdvRouteItem *) item); return;
} }
bool RsDsdvSerialiser::serialise(RsItem *item, void *data, uint32_t *pktsize) void RsDsdvDataItem::clear()
{ {
return serialiseItem((RsDsdvRouteItem *) item, data, pktsize); src.TlvClear();
dest.TlvClear();
ttl = 0;
data.TlvClear();
}
std::ostream &RsDsdvDataItem::print(std::ostream &out, uint16_t indent)
{
printRsItemBase(out, "RsDsdvDataItem", indent);
uint16_t int_Indent = indent + 2;
src.print(out, int_Indent);
dest.print(out, int_Indent);
printIndent(out, int_Indent);
out << "TTL: " << ttl << std::endl;
data.print(out, int_Indent);
printRsItemEnd(out, "RsDsdvDataItem", indent);
return out;
}
uint32_t RsDsdvSerialiser::sizeData(RsDsdvDataItem *item)
{
uint32_t s = 8; /* header */
s += item->src.TlvSize();
s += item->dest.TlvSize();
s += 4;
s += item->data.TlvSize();
return s;
}
/* serialise the data to the buffer */
bool RsDsdvSerialiser::serialiseData(RsDsdvDataItem *item, void *data, uint32_t *pktsize)
{
uint32_t tlvsize = sizeData(item);
uint32_t offset = 0;
if (*pktsize < tlvsize)
return false; /* not enough space */
*pktsize = tlvsize;
bool ok = true;
ok &= setRsItemHeader(data, tlvsize, item->PacketId(), tlvsize);
#ifdef RSSERIAL_DEBUG
std::cerr << "RsDsdvSerialiser::serialiseData() Header: " << ok << std::endl;
std::cerr << "RsDsdvSerialiser::serialiseData() Size: " << tlvsize << std::endl;
#endif
/* skip the header */
offset += 8;
/* add mandatory parts first */
ok &= item->src.SetTlv(data, tlvsize, &offset);
ok &= item->dest.SetTlv(data, tlvsize, &offset);
ok &= setRawUInt32(data, tlvsize, &offset, item->ttl);
ok &= item->data.SetTlv(data, tlvsize, &offset);
if (offset != tlvsize)
{
ok = false;
#ifdef RSSERIAL_DEBUG
std::cerr << "RsDsdvSerialiser::serialiseData() Size Error! " << std::endl;
#endif
}
return ok;
}
RsDsdvDataItem *RsDsdvSerialiser::deserialiseData(void *data, uint32_t *pktsize)
{
/* get the type and size */
uint32_t rstype = getRsItemId(data);
uint32_t tlvsize = getRsItemSize(data);
uint32_t offset = 0;
if ((RS_PKT_VERSION_SERVICE != getRsItemVersion(rstype)) ||
(RS_SERVICE_TYPE_DSDV != getRsItemService(rstype)) ||
(RS_PKT_SUBTYPE_DSDV_DATA != getRsItemSubType(rstype)))
{
return NULL; /* wrong type */
}
if (*pktsize < tlvsize) /* check size */
return NULL; /* not enough data */
/* set the packet length */
*pktsize = tlvsize;
bool ok = true;
/* ready to load */
RsDsdvDataItem *item = new RsDsdvDataItem();
item->clear();
/* skip the header */
offset += 8;
/* add mandatory parts first */
ok &= item->src.GetTlv(data, tlvsize, &offset);
ok &= item->dest.GetTlv(data, tlvsize, &offset);
ok &= getRawUInt32(data, tlvsize, &offset, &(item->ttl));
ok &= item->data.GetTlv(data, tlvsize, &offset);
if (offset != tlvsize)
{
/* error */
delete item;
return NULL;
}
if (!ok)
{
delete item;
return NULL;
}
return item;
}
/*************************************************************************/
uint32_t RsDsdvSerialiser::size(RsItem *i)
{
RsDsdvRouteItem *dri;
RsDsdvDataItem *ddi;
if (NULL != (dri = dynamic_cast<RsDsdvRouteItem *>(i)))
{
return sizeRoute(dri);
}
if (NULL != (ddi = dynamic_cast<RsDsdvDataItem *>(i)))
{
return sizeData(ddi);
}
return 0;
}
bool RsDsdvSerialiser::serialise(RsItem *i, void *data, uint32_t *pktsize)
{
RsDsdvRouteItem *dri;
RsDsdvDataItem *ddi;
if (NULL != (dri = dynamic_cast<RsDsdvRouteItem *>(i)))
{
return serialiseRoute(dri, data, pktsize);
}
if (NULL != (ddi = dynamic_cast<RsDsdvDataItem *>(i)))
{
return serialiseData(ddi, data, pktsize);
}
return false;
} }
RsItem *RsDsdvSerialiser::deserialise(void *data, uint32_t *pktsize) RsItem *RsDsdvSerialiser::deserialise(void *data, uint32_t *pktsize)
{ {
return deserialiseItem(data, pktsize); /* get the type and size */
uint32_t rstype = getRsItemId(data);
if ((RS_PKT_VERSION_SERVICE != getRsItemVersion(rstype)) ||
(RS_SERVICE_TYPE_DSDV != getRsItemService(rstype)))
{
return NULL; /* wrong type */
}
switch(getRsItemSubType(rstype))
{
case RS_PKT_SUBTYPE_DSDV_ROUTE:
return deserialiseRoute(data, pktsize);
break;
case RS_PKT_SUBTYPE_DSDV_DATA:
return deserialiseData(data, pktsize);
break;
default:
return NULL;
break;
}
} }
/*************************************************************************/ /*************************************************************************/

View File

@ -30,11 +30,14 @@
#include "serialiser/rsserviceids.h" #include "serialiser/rsserviceids.h"
#include "serialiser/rsserial.h" #include "serialiser/rsserial.h"
#include "serialiser/rstlvbase.h"
#include "serialiser/rstlvtypes.h"
#include "serialiser/rstlvdsdv.h" #include "serialiser/rstlvdsdv.h"
/**************************************************************************/ /**************************************************************************/
#define RS_PKT_SUBTYPE_DSDV_ROUTE 0x01 #define RS_PKT_SUBTYPE_DSDV_ROUTE 0x01
#define RS_PKT_SUBTYPE_DSDV_DATA 0x02
class RsDsdvRouteItem: public RsItem class RsDsdvRouteItem: public RsItem
{ {
@ -50,6 +53,24 @@ std::ostream &print(std::ostream &out, uint16_t indent = 0);
RsTlvDsdvEntrySet routes; RsTlvDsdvEntrySet routes;
}; };
class RsDsdvDataItem: public RsItem
{
public:
RsDsdvDataItem()
:RsItem(RS_PKT_VERSION_SERVICE, RS_SERVICE_TYPE_DSDV,
RS_PKT_SUBTYPE_DSDV_DATA), data(TLV_TYPE_BIN_GENERIC)
{ return; }
virtual ~RsDsdvDataItem();
virtual void clear();
std::ostream &print(std::ostream &out, uint16_t indent = 0);
RsTlvDsdvEndPoint src;
RsTlvDsdvEndPoint dest;
uint32_t ttl;
RsTlvBinaryData data;
};
class RsDsdvSerialiser: public RsSerialType class RsDsdvSerialiser: public RsSerialType
{ {
public: public:
@ -65,9 +86,13 @@ virtual RsItem * deserialise(void *data, uint32_t *size);
private: private:
virtual uint32_t sizeItem(RsDsdvRouteItem *); virtual uint32_t sizeRoute(RsDsdvRouteItem *);
virtual bool serialiseItem (RsDsdvRouteItem *item, void *data, uint32_t *size); virtual bool serialiseRoute(RsDsdvRouteItem *item, void *data, uint32_t *size);
virtual RsDsdvRouteItem *deserialiseItem(void *data, uint32_t *size); virtual RsDsdvRouteItem *deserialiseRoute(void *data, uint32_t *size);
virtual uint32_t sizeData(RsDsdvDataItem *);
virtual bool serialiseData(RsDsdvDataItem *item, void *data, uint32_t *size);
virtual RsDsdvDataItem *deserialiseData(void *data, uint32_t *size);
}; };

View File

@ -184,6 +184,8 @@ const uint16_t TLV_TYPE_BIN_FILEDATA = 0x0140; /* Used - ACTIVE! */
const uint16_t TLV_TYPE_BIN_SERIALISE = 0x0150; /* Used (Generic - Distrib) */ const uint16_t TLV_TYPE_BIN_SERIALISE = 0x0150; /* Used (Generic - Distrib) */
const uint16_t TLV_TYPE_BIN_GENERIC = 0x0160; /* Used (DSDV Data) */
/**** Compound Types ****/ /**** Compound Types ****/
const uint16_t TLV_TYPE_FILEITEM = 0x1000; const uint16_t TLV_TYPE_FILEITEM = 0x1000;
@ -210,8 +212,9 @@ const uint16_t TLV_TYPE_IMAGE = 0x1060;
const uint16_t TLV_TYPE_ADDRESS_INFO = 0x1070; const uint16_t TLV_TYPE_ADDRESS_INFO = 0x1070;
const uint16_t TLV_TYPE_ADDRESS_SET = 0x1071; const uint16_t TLV_TYPE_ADDRESS_SET = 0x1071;
const uint16_t TLV_TYPE_DSDV_ENTRY = 0x1080; const uint16_t TLV_TYPE_DSDV_ENDPOINT = 0x1080;
const uint16_t TLV_TYPE_DSDV_ENTRY_SET= 0x1081; const uint16_t TLV_TYPE_DSDV_ENTRY = 0x1081;
const uint16_t TLV_TYPE_DSDV_ENTRY_SET= 0x1082;

View File

@ -36,30 +36,146 @@
#include <iostream> #include <iostream>
/************************************* RsTlvDsdvEndPoint ************************************/
RsTlvDsdvEndPoint::RsTlvDsdvEndPoint()
:RsTlvItem(), idType(0)
{
return;
}
void RsTlvDsdvEndPoint::TlvClear()
{
idType = 0;
anonChunk.clear();
serviceId.clear();
}
uint32_t RsTlvDsdvEndPoint::TlvSize()
{
uint32_t s = TLV_HEADER_SIZE; /* header + 4 + str + str */
s += 4; // idType;
s += GetTlvStringSize(anonChunk);
s += GetTlvStringSize(serviceId);
return s;
}
bool RsTlvDsdvEndPoint::SetTlv(void *data, uint32_t size, uint32_t *offset) /* serialise */
{
/* must check sizes */
uint32_t tlvsize = TlvSize();
uint32_t tlvend = *offset + tlvsize;
if (size < tlvend)
return false; /* not enough space */
bool ok = true;
/* start at data[offset] */
/* add mandatory parts first */
ok &= SetTlvBase(data, tlvend, offset, TLV_TYPE_DSDV_ENDPOINT, tlvsize);
ok &= setRawUInt32(data, tlvend, offset, idType);
ok &= SetTlvString(data, tlvend, offset, TLV_TYPE_STR_GENID, anonChunk);
ok &= SetTlvString(data, tlvend, offset, TLV_TYPE_STR_HASH_SHA1, serviceId);
return ok;
}
bool RsTlvDsdvEndPoint::GetTlv(void *data, uint32_t size, uint32_t *offset) /* serialise */
{
if (size < *offset + TLV_HEADER_SIZE)
return false;
uint16_t tlvtype = GetTlvType( &(((uint8_t *) data)[*offset]) );
uint32_t tlvsize = GetTlvSize( &(((uint8_t *) data)[*offset]) );
uint32_t tlvend = *offset + tlvsize;
if (size < tlvend) /* check size */
return false; /* not enough space */
if (tlvtype != TLV_TYPE_DSDV_ENDPOINT) /* check type */
return false;
bool ok = true;
/* ready to load */
TlvClear();
/* skip the header */
(*offset) += TLV_HEADER_SIZE;
ok &= getRawUInt32(data, tlvend, offset, &(idType));
ok &= GetTlvString(data, tlvend, offset, TLV_TYPE_STR_GENID, anonChunk);
ok &= GetTlvString(data, tlvend, offset, TLV_TYPE_STR_HASH_SHA1, serviceId);
/***************************************************************************
* NB: extra components could be added (for future expansion of the type).
* or be present (if this code is reading an extended version).
*
* We must chew up the extra characters to conform with TLV specifications
***************************************************************************/
if (*offset != tlvend)
{
#ifdef TLV_DEBUG
std::cerr << "RsTlvDsdvEndPoint::GetTlv() Warning extra bytes at end of item";
std::cerr << std::endl;
#endif
*offset = tlvend;
}
return ok;
}
std::ostream &RsTlvDsdvEndPoint::print(std::ostream &out, uint16_t indent)
{
printBase(out, "RsTlvDsdvEndPoint", indent);
uint16_t int_Indent = indent + 2;
printIndent(out, int_Indent);
out << "idType:" << idType;
out << std::endl;
printIndent(out, int_Indent);
out << "AnonChunk:" << anonChunk;
out << std::endl;
printIndent(out, int_Indent);
out << "ServiceId:" << serviceId;
out << std::endl;
printEnd(out, "RsTlvDsdvEndPoint", indent);
return out;
}
/************************************* RsTlvDsdvEntry ************************************/ /************************************* RsTlvDsdvEntry ************************************/
RsTlvDsdvEntry::RsTlvDsdvEntry() RsTlvDsdvEntry::RsTlvDsdvEntry()
:RsTlvItem(), idType(0), sequence(0), distance(0) :RsTlvItem(), sequence(0), distance(0)
{ {
return; return;
} }
void RsTlvDsdvEntry::TlvClear() void RsTlvDsdvEntry::TlvClear()
{ {
idType = 0; endPoint.TlvClear();
anonChunk.clear();
serviceId.clear();
sequence = 0; sequence = 0;
distance = 0; distance = 0;
} }
uint32_t RsTlvDsdvEntry::TlvSize() uint32_t RsTlvDsdvEntry::TlvSize()
{ {
uint32_t s = TLV_HEADER_SIZE; /* header + 4 + str + str + 4 + 4 */ uint32_t s = TLV_HEADER_SIZE; /* header + EndPoint.Size + 4 + 4 */
s += 4; // idType; s += endPoint.TlvSize();
s += GetTlvStringSize(anonChunk);
s += GetTlvStringSize(serviceId);
s += 4; // sequence; s += 4; // sequence;
s += 4; // distance; s += 4; // distance;
@ -81,11 +197,9 @@ bool RsTlvDsdvEntry::SetTlv(void *data, uint32_t size, uint32_t *offset) /* ser
/* start at data[offset] */ /* start at data[offset] */
/* add mandatory parts first */ /* add mandatory parts first */
ok &= SetTlvBase(data, tlvend, offset, TLV_TYPE_ADDRESS_INFO, tlvsize); ok &= SetTlvBase(data, tlvend, offset, TLV_TYPE_DSDV_ENTRY, tlvsize);
ok &= setRawUInt32(data, tlvend, offset, idType); ok &= endPoint.SetTlv(data, size, offset);
ok &= SetTlvString(data, tlvend, offset, TLV_TYPE_STR_GENID, anonChunk);
ok &= SetTlvString(data, tlvend, offset, TLV_TYPE_STR_HASH_SHA1, serviceId);
ok &= setRawUInt32(data, tlvend, offset, sequence); ok &= setRawUInt32(data, tlvend, offset, sequence);
ok &= setRawUInt32(data, tlvend, offset, distance); ok &= setRawUInt32(data, tlvend, offset, distance);
@ -117,9 +231,7 @@ bool RsTlvDsdvEntry::GetTlv(void *data, uint32_t size, uint32_t *offset) /* ser
/* skip the header */ /* skip the header */
(*offset) += TLV_HEADER_SIZE; (*offset) += TLV_HEADER_SIZE;
ok &= getRawUInt32(data, tlvend, offset, &(idType)); ok &= endPoint.GetTlv(data, size, offset);
ok &= GetTlvString(data, tlvend, offset, TLV_TYPE_STR_GENID, anonChunk);
ok &= GetTlvString(data, tlvend, offset, TLV_TYPE_STR_HASH_SHA1, serviceId);
ok &= getRawUInt32(data, tlvend, offset, &(sequence)); ok &= getRawUInt32(data, tlvend, offset, &(sequence));
ok &= getRawUInt32(data, tlvend, offset, &(distance)); ok &= getRawUInt32(data, tlvend, offset, &(distance));
@ -149,17 +261,7 @@ std::ostream &RsTlvDsdvEntry::print(std::ostream &out, uint16_t indent)
printBase(out, "RsTlvDsdvEntry", indent); printBase(out, "RsTlvDsdvEntry", indent);
uint16_t int_Indent = indent + 2; uint16_t int_Indent = indent + 2;
printIndent(out, int_Indent); endPoint.print(out, int_Indent);
out << "idType:" << idType;
out << std::endl;
printIndent(out, int_Indent);
out << "AnonChunk:" << anonChunk;
out << std::endl;
printIndent(out, int_Indent);
out << "ServiceId:" << serviceId;
out << std::endl;
printIndent(out, int_Indent); printIndent(out, int_Indent);
out << "Sequence:" << sequence; out << "Sequence:" << sequence;

View File

@ -36,6 +36,22 @@
#define RSDSDV_MAX_ROUTE_TABLE 1000 #define RSDSDV_MAX_ROUTE_TABLE 1000
class RsTlvDsdvEndPoint: public RsTlvItem
{
public:
RsTlvDsdvEndPoint();
virtual ~RsTlvDsdvEndPoint() { return; }
virtual uint32_t TlvSize();
virtual void TlvClear();
virtual bool SetTlv(void *data, uint32_t size, uint32_t *offset); /* serialise */
virtual bool GetTlv(void *data, uint32_t size, uint32_t *offset); /* deserialise */
virtual std::ostream &print(std::ostream &out, uint16_t indent);
uint32_t idType;
std::string anonChunk;
std::string serviceId;
};
class RsTlvDsdvEntry: public RsTlvItem class RsTlvDsdvEntry: public RsTlvItem
{ {
public: public:
@ -47,9 +63,7 @@ virtual bool SetTlv(void *data, uint32_t size, uint32_t *offset); /* seriali
virtual bool GetTlv(void *data, uint32_t size, uint32_t *offset); /* deserialise */ virtual bool GetTlv(void *data, uint32_t size, uint32_t *offset); /* deserialise */
virtual std::ostream &print(std::ostream &out, uint16_t indent); virtual std::ostream &print(std::ostream &out, uint16_t indent);
uint32_t idType; RsTlvDsdvEndPoint endPoint;
std::string anonChunk;
std::string serviceId;
uint32_t sequence; uint32_t sequence;
uint32_t distance; uint32_t distance;
}; };

View File

@ -224,9 +224,9 @@ int p3Dsdv::generateRoutingTable(const std::string &peerId, bool incremental)
RsTlvDsdvEntry entry; RsTlvDsdvEntry entry;
entry.idType = v.mDest.mIdType; entry.endPoint.idType = v.mDest.mIdType;
entry.anonChunk = v.mDest.mAnonChunk; entry.endPoint.anonChunk = v.mDest.mAnonChunk;
entry.serviceId = v.mDest.mHash; entry.endPoint.serviceId = v.mDest.mHash;
entry.sequence = v.mStableRoute.mSequence; entry.sequence = v.mStableRoute.mSequence;
entry.distance = v.mStableRoute.mDistance; entry.distance = v.mStableRoute.mDistance;
@ -286,14 +286,14 @@ int p3Dsdv::handleDSDV(RsDsdvRouteItem *dsdv)
/* find the entry */ /* find the entry */
std::map<std::string, RsDsdvTableEntry>::iterator tit; std::map<std::string, RsDsdvTableEntry>::iterator tit;
tit = mTable.find(entry.serviceId); tit = mTable.find(entry.endPoint.serviceId);
if (tit == mTable.end()) if (tit == mTable.end())
{ {
/* new entry! */ /* new entry! */
RsDsdvTableEntry v; RsDsdvTableEntry v;
v.mDest.mIdType = entry.idType; v.mDest.mIdType = entry.endPoint.idType;
v.mDest.mAnonChunk = entry.anonChunk; v.mDest.mAnonChunk = entry.endPoint.anonChunk;
v.mDest.mHash = entry.serviceId; v.mDest.mHash = entry.endPoint.serviceId;
v.mBestRoute.mNextHop = dsdv->PeerId(); v.mBestRoute.mNextHop = dsdv->PeerId();
v.mBestRoute.mReceived = now; v.mBestRoute.mReceived = now;