mirror of
https://github.com/RetroShare/RetroShare.git
synced 2025-02-17 21:34:10 -05:00
Corrected two harmful bugs in tunnel connexion code:
- added a destructor to RsTunnelDataItem, to remove a memory leak on all data packets. - apparently the code would allow data items of zero size, hence calling malloc(0), which has undetermined behavior. git-svn-id: http://svn.code.sf.net/p/retroshare/code/trunk@3191 b45a01b8-16f6-495d-af2f-9b41ad6348cc
This commit is contained in:
parent
13671d03ea
commit
5045e720b2
@ -82,6 +82,12 @@ RsItem *RsTunnelSerialiser::deserialise(void *data, uint32_t *size)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
RsTunnelDataItem::~RsTunnelDataItem()
|
||||||
|
{
|
||||||
|
if(encoded_data != NULL)
|
||||||
|
free(encoded_data) ;
|
||||||
|
}
|
||||||
|
|
||||||
bool RsTunnelDataItem::serialize(void *data,uint32_t& pktsize)
|
bool RsTunnelDataItem::serialize(void *data,uint32_t& pktsize)
|
||||||
{
|
{
|
||||||
#ifdef P3TUNNEL_DEBUG
|
#ifdef P3TUNNEL_DEBUG
|
||||||
@ -112,6 +118,8 @@ bool RsTunnelDataItem::serialize(void *data,uint32_t& pktsize)
|
|||||||
ok &= SetTlvString(data, tlvsize, &offset, TLV_TYPE_STR_VALUE, destPeerId);
|
ok &= SetTlvString(data, tlvsize, &offset, TLV_TYPE_STR_VALUE, destPeerId);
|
||||||
|
|
||||||
ok &= setRawUInt32(data, tlvsize, &offset, encoded_data_len) ;
|
ok &= setRawUInt32(data, tlvsize, &offset, encoded_data_len) ;
|
||||||
|
|
||||||
|
if(encoded_data != NULL)
|
||||||
memcpy((void*)((unsigned char*)data+offset),encoded_data,encoded_data_len) ;
|
memcpy((void*)((unsigned char*)data+offset),encoded_data,encoded_data_len) ;
|
||||||
|
|
||||||
#ifdef P3TUNNEL_DEBUG
|
#ifdef P3TUNNEL_DEBUG
|
||||||
@ -215,11 +223,16 @@ RsTunnelDataItem *RsTunnelDataItem::deserialise(void *data,uint32_t pktsize)
|
|||||||
|
|
||||||
ok &= getRawUInt32(data, pktsize, &offset, &item->encoded_data_len) ;
|
ok &= getRawUInt32(data, pktsize, &offset, &item->encoded_data_len) ;
|
||||||
|
|
||||||
if(!ok) // return early to avoid calling malloc with 0 size
|
if(!ok) // return early to avoid calling malloc with invalid size
|
||||||
return NULL ;
|
return NULL ;
|
||||||
|
|
||||||
|
if(item->encoded_data_len > 0)
|
||||||
|
{
|
||||||
item->encoded_data = (void*)malloc(item->encoded_data_len) ;
|
item->encoded_data = (void*)malloc(item->encoded_data_len) ;
|
||||||
memcpy(item->encoded_data, (void*)((unsigned char*)data+offset), item->encoded_data_len);
|
memcpy(item->encoded_data, (void*)((unsigned char*)data+offset), item->encoded_data_len);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
item->encoded_data = NULL ;
|
||||||
|
|
||||||
if ((offset + item->encoded_data_len) != rssize)
|
if ((offset + item->encoded_data_len) != rssize)
|
||||||
{
|
{
|
||||||
|
@ -47,6 +47,8 @@ class RsTunnelItem: public RsItem
|
|||||||
public:
|
public:
|
||||||
RsTunnelItem(uint8_t tunnel_subtype) : RsItem(RS_PKT_VERSION_SERVICE,RS_SERVICE_TYPE_TUNNEL,tunnel_subtype) {}
|
RsTunnelItem(uint8_t tunnel_subtype) : RsItem(RS_PKT_VERSION_SERVICE,RS_SERVICE_TYPE_TUNNEL,tunnel_subtype) {}
|
||||||
|
|
||||||
|
virtual ~RsTunnelItem() {}
|
||||||
|
|
||||||
virtual bool serialize(void *data,uint32_t& size) = 0 ; // Isn't it better that items can serialize themselves ?
|
virtual bool serialize(void *data,uint32_t& size) = 0 ; // Isn't it better that items can serialize themselves ?
|
||||||
virtual uint32_t serial_size() { return 0;}
|
virtual uint32_t serial_size() { return 0;}
|
||||||
|
|
||||||
@ -56,8 +58,13 @@ class RsTunnelItem: public RsItem
|
|||||||
class RsTunnelDataItem: public RsTunnelItem
|
class RsTunnelDataItem: public RsTunnelItem
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
RsTunnelDataItem() : RsTunnelItem(RS_TUNNEL_SUBTYPE_DATA) {}
|
RsTunnelDataItem() : RsTunnelItem(RS_TUNNEL_SUBTYPE_DATA)
|
||||||
RsTunnelDataItem(void *data,uint32_t size) ; // deserialization
|
{
|
||||||
|
encoded_data = NULL ; // To comply with the destructor below.
|
||||||
|
}
|
||||||
|
virtual ~RsTunnelDataItem() ;
|
||||||
|
|
||||||
|
static RsTunnelDataItem *deserialise(void *data,uint32_t size) ; // deserialization
|
||||||
|
|
||||||
uint32_t encoded_data_len;
|
uint32_t encoded_data_len;
|
||||||
void *encoded_data;
|
void *encoded_data;
|
||||||
@ -76,7 +83,9 @@ class RsTunnelHandshakeItem: public RsTunnelItem
|
|||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
RsTunnelHandshakeItem() : RsTunnelItem(RS_TUNNEL_SUBTYPE_HANDSHAKE) {}
|
RsTunnelHandshakeItem() : RsTunnelItem(RS_TUNNEL_SUBTYPE_HANDSHAKE) {}
|
||||||
RsTunnelHandshakeItem(void *data,uint32_t size) ; // deserialization
|
virtual ~RsTunnelHandshakeItem() {}
|
||||||
|
|
||||||
|
static RsTunnelHandshakeItem *deserialise(void *data,uint32_t size) ; // deserialization
|
||||||
|
|
||||||
std::string sourcePeerId ;
|
std::string sourcePeerId ;
|
||||||
std::string relayPeerId ;
|
std::string relayPeerId ;
|
||||||
|
@ -139,8 +139,11 @@ void p3tunnel::sendTunnelDataPrivate(std::string to, std::string sourcePeerId, s
|
|||||||
rdi->relayPeerId = relayPeerId;
|
rdi->relayPeerId = relayPeerId;
|
||||||
rdi->encoded_data_len = data_length;
|
rdi->encoded_data_len = data_length;
|
||||||
|
|
||||||
|
if(data_length > 0)
|
||||||
|
{
|
||||||
rdi->encoded_data = (void*)malloc(data_length);
|
rdi->encoded_data = (void*)malloc(data_length);
|
||||||
memcpy(rdi->encoded_data, data, data_length);
|
memcpy(rdi->encoded_data, data, data_length);
|
||||||
|
}
|
||||||
|
|
||||||
rdi->PeerId(to);
|
rdi->PeerId(to);
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user