mirror of
https://github.com/RetroShare/RetroShare.git
synced 2024-12-28 08:59:37 -05:00
fixed memory leak and potential crash due to not copying items passed to config saving in messages. Added save/load for map between GRouter and Msg ids which fixes the never removed outgoing messages
git-svn-id: http://svn.code.sf.net/p/retroshare/code/trunk@8184 b45a01b8-16f6-495d-af2f-9b41ad6348cc
This commit is contained in:
parent
9ad35ad03f
commit
90d6ac6697
@ -762,6 +762,144 @@ RsMsgSrcId* RsMsgSerialiser::deserialiseMsgSrcIdItem(void* data, uint32_t* pktsi
|
|||||||
|
|
||||||
/************************* end of definition of msgSrcId serialisation functions ************************/
|
/************************* end of definition of msgSrcId serialisation functions ************************/
|
||||||
|
|
||||||
|
std::ostream& RsMsgGRouterMap::print(std::ostream& out, uint16_t indent)
|
||||||
|
{
|
||||||
|
printRsItemBase(out, "RsMsgGRouterMap", indent);
|
||||||
|
uint16_t int_Indent = indent + 2;
|
||||||
|
|
||||||
|
for(std::map<GRouterMsgPropagationId,uint32_t>::const_iterator it(ongoing_msgs.begin());it!=ongoing_msgs.end();++it)
|
||||||
|
{
|
||||||
|
printIndent(out, int_Indent);
|
||||||
|
out << " " << std::hex << it->first << std::dec << " : " << it->second << std::endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
printRsItemEnd(out, "RsMsgGRouterMap", indent);
|
||||||
|
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
void RsMsgGRouterMap::clear()
|
||||||
|
{
|
||||||
|
ongoing_msgs.clear() ;
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t RsMsgGRouterMap::serial_size(bool)
|
||||||
|
{
|
||||||
|
uint32_t s = 8; /* header */
|
||||||
|
|
||||||
|
s += 4; // number of entries
|
||||||
|
s += (8+4)*ongoing_msgs.size(); // entries
|
||||||
|
|
||||||
|
return s;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool RsMsgGRouterMap::serialise(void *data, uint32_t& pktsize,bool config)
|
||||||
|
{
|
||||||
|
uint32_t tlvsize = serial_size( config) ;
|
||||||
|
uint32_t offset = 0;
|
||||||
|
|
||||||
|
if (pktsize < tlvsize)
|
||||||
|
return false; /* not enough space */
|
||||||
|
|
||||||
|
pktsize = tlvsize;
|
||||||
|
|
||||||
|
bool ok = true;
|
||||||
|
|
||||||
|
ok &= setRsItemHeader(data, tlvsize, PacketId(), tlvsize);
|
||||||
|
|
||||||
|
#ifdef RSSERIAL_DEBUG
|
||||||
|
std::cerr << "RsMsgSerialiser::serialiseMsgParentIdItem() Header: " << ok << std::endl;
|
||||||
|
std::cerr << "RsMsgSerialiser::serialiseMsgParentIdItem() Size: " << tlvsize << std::endl;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/* skip the header */
|
||||||
|
offset += 8;
|
||||||
|
|
||||||
|
ok &= setRawUInt32(data, tlvsize, &offset, ongoing_msgs.size());
|
||||||
|
|
||||||
|
for(std::map<GRouterMsgPropagationId,uint32_t>::const_iterator it=ongoing_msgs.begin();ok && it!=ongoing_msgs.end();++it)
|
||||||
|
{
|
||||||
|
ok &= setRawUInt64(data, tlvsize, &offset, it->first);
|
||||||
|
ok &= setRawUInt32(data, tlvsize, &offset, it->second);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (offset != tlvsize)
|
||||||
|
{
|
||||||
|
ok = false;
|
||||||
|
std::cerr << "RsMsgSerialiser::serialiseMsgGRouterMap() Size Error! " << std::endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
return ok;
|
||||||
|
}
|
||||||
|
|
||||||
|
RsMsgGRouterMap* RsMsgSerialiser::deserialiseMsgGRouterMap(void* data, uint32_t* pktsize)
|
||||||
|
{
|
||||||
|
/* get the type and size */
|
||||||
|
uint32_t rstype = getRsItemId(data);
|
||||||
|
uint32_t rssize = getRsItemSize(data);
|
||||||
|
|
||||||
|
uint32_t offset = 0;
|
||||||
|
|
||||||
|
|
||||||
|
if ((RS_PKT_VERSION_SERVICE != getRsItemVersion(rstype)) ||
|
||||||
|
(RS_SERVICE_TYPE_MSG != getRsItemService(rstype)) ||
|
||||||
|
(RS_PKT_SUBTYPE_MSG_GROUTER_MAP != getRsItemSubType(rstype)))
|
||||||
|
{
|
||||||
|
return NULL; /* wrong type */
|
||||||
|
}
|
||||||
|
|
||||||
|
if (*pktsize < rssize) /* check size */
|
||||||
|
return NULL; /* not enough data */
|
||||||
|
|
||||||
|
/* set the packet length */
|
||||||
|
*pktsize = rssize;
|
||||||
|
|
||||||
|
bool ok = true;
|
||||||
|
|
||||||
|
/* ready to load */
|
||||||
|
RsMsgGRouterMap *item = new RsMsgGRouterMap();
|
||||||
|
item->clear();
|
||||||
|
|
||||||
|
/* skip the header */
|
||||||
|
offset += 8;
|
||||||
|
|
||||||
|
uint32_t s=0 ;
|
||||||
|
|
||||||
|
/* get mandatory parts first */
|
||||||
|
ok &= getRawUInt32(data, rssize, &offset, &s);
|
||||||
|
|
||||||
|
for(uint32_t i=0;i<s && ok;++i)
|
||||||
|
{
|
||||||
|
uint32_t mid;
|
||||||
|
uint64_t routing_id;
|
||||||
|
|
||||||
|
ok &= getRawUInt64(data, rssize, &offset, &routing_id);
|
||||||
|
ok &= getRawUInt32(data, rssize, &offset, &mid);
|
||||||
|
|
||||||
|
item->ongoing_msgs.insert(std::make_pair(routing_id,mid)) ;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (offset != rssize)
|
||||||
|
{
|
||||||
|
/* error */
|
||||||
|
delete item;
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!ok)
|
||||||
|
{
|
||||||
|
delete item;
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
return item;
|
||||||
|
}
|
||||||
|
|
||||||
|
/************************* end of definition of msgGRouterMap serialisation functions ************************/
|
||||||
|
|
||||||
|
|
||||||
/************************************** Message ParentId **********************/
|
/************************************** Message ParentId **********************/
|
||||||
|
|
||||||
std::ostream& RsMsgParentId::print(std::ostream& out, uint16_t indent)
|
std::ostream& RsMsgParentId::print(std::ostream& out, uint16_t indent)
|
||||||
@ -922,6 +1060,9 @@ RsItem* RsMsgSerialiser::deserialise(void *data, uint32_t *pktsize)
|
|||||||
case RS_PKT_SUBTYPE_MSG_TAGS:
|
case RS_PKT_SUBTYPE_MSG_TAGS:
|
||||||
return deserialiseMsgTagItem(data, pktsize);
|
return deserialiseMsgTagItem(data, pktsize);
|
||||||
break;
|
break;
|
||||||
|
case RS_PKT_SUBTYPE_MSG_GROUTER_MAP:
|
||||||
|
return deserialiseMsgGRouterMap(data, pktsize);
|
||||||
|
break;
|
||||||
default:
|
default:
|
||||||
return NULL;
|
return NULL;
|
||||||
break;
|
break;
|
||||||
|
@ -35,6 +35,7 @@
|
|||||||
|
|
||||||
#include "serialiser/rstlvidset.h"
|
#include "serialiser/rstlvidset.h"
|
||||||
#include "serialiser/rstlvfileitem.h"
|
#include "serialiser/rstlvfileitem.h"
|
||||||
|
#include "grouter/grouteritems.h"
|
||||||
|
|
||||||
#if 0
|
#if 0
|
||||||
#include "serialiser/rstlvtypes.h"
|
#include "serialiser/rstlvtypes.h"
|
||||||
@ -49,6 +50,7 @@ const uint8_t RS_PKT_SUBTYPE_MSG_TAGS = 0x04;
|
|||||||
const uint8_t RS_PKT_SUBTYPE_MSG_SRC_TAG = 0x05;
|
const uint8_t RS_PKT_SUBTYPE_MSG_SRC_TAG = 0x05;
|
||||||
const uint8_t RS_PKT_SUBTYPE_MSG_PARENT_TAG = 0x06;
|
const uint8_t RS_PKT_SUBTYPE_MSG_PARENT_TAG = 0x06;
|
||||||
const uint8_t RS_PKT_SUBTYPE_MSG_INVITE = 0x07;
|
const uint8_t RS_PKT_SUBTYPE_MSG_INVITE = 0x07;
|
||||||
|
const uint8_t RS_PKT_SUBTYPE_MSG_GROUTER_MAP = 0x08;
|
||||||
|
|
||||||
|
|
||||||
/**************************************************************************/
|
/**************************************************************************/
|
||||||
@ -207,6 +209,23 @@ class RsPublicMsgInviteConfigItem : public RsMessageItem
|
|||||||
time_t time_stamp ;
|
time_t time_stamp ;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
class RsMsgGRouterMap : public RsMessageItem
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
RsMsgGRouterMap() : RsMessageItem(RS_PKT_SUBTYPE_MSG_GROUTER_MAP) {}
|
||||||
|
|
||||||
|
std::ostream &print(std::ostream &out, uint16_t indent = 0);
|
||||||
|
|
||||||
|
virtual bool serialise(void *data,uint32_t& size,bool config) ;
|
||||||
|
virtual uint32_t serial_size(bool config) ;
|
||||||
|
|
||||||
|
virtual ~RsMsgGRouterMap() {}
|
||||||
|
virtual void clear();
|
||||||
|
|
||||||
|
// ----------- Specific fields ------------- //
|
||||||
|
//
|
||||||
|
std::map<GRouterMsgPropagationId,uint32_t> ongoing_msgs ;
|
||||||
|
};
|
||||||
|
|
||||||
class RsMsgParentId : public RsMessageItem
|
class RsMsgParentId : public RsMessageItem
|
||||||
{
|
{
|
||||||
@ -256,6 +275,7 @@ class RsMsgSerialiser: public RsSerialType
|
|||||||
virtual RsMsgSrcId *deserialiseMsgSrcIdItem(void *data, uint32_t *size);
|
virtual RsMsgSrcId *deserialiseMsgSrcIdItem(void *data, uint32_t *size);
|
||||||
virtual RsMsgParentId *deserialiseMsgParentIdItem(void *data, uint32_t *size);
|
virtual RsMsgParentId *deserialiseMsgParentIdItem(void *data, uint32_t *size);
|
||||||
virtual RsPublicMsgInviteConfigItem *deserialisePublicMsgInviteConfigItem(void *data, uint32_t *size);
|
virtual RsPublicMsgInviteConfigItem *deserialisePublicMsgInviteConfigItem(void *data, uint32_t *size);
|
||||||
|
virtual RsMsgGRouterMap *deserialiseMsgGRouterMap(void *data, uint32_t *size);
|
||||||
|
|
||||||
bool m_bConfiguration; // is set to true for saving configuration (enables serialising msgId)
|
bool m_bConfiguration; // is set to true for saving configuration (enables serialising msgId)
|
||||||
};
|
};
|
||||||
|
@ -428,28 +428,33 @@ bool p3MsgService::saveList(bool& cleanup, std::list<RsItem*>& itemList)
|
|||||||
|
|
||||||
MsgTagType stdTags;
|
MsgTagType stdTags;
|
||||||
|
|
||||||
cleanup = false;
|
cleanup = true;
|
||||||
|
|
||||||
mMsgMtx.lock();
|
mMsgMtx.lock();
|
||||||
|
|
||||||
for(mit = imsg.begin(); mit != imsg.end(); ++mit)
|
for(mit = imsg.begin(); mit != imsg.end(); ++mit)
|
||||||
itemList.push_back(mit->second);
|
itemList.push_back(new RsMsgItem(*mit->second));
|
||||||
|
|
||||||
for(lit = mSrcIds.begin(); lit != mSrcIds.end(); ++lit)
|
for(lit = mSrcIds.begin(); lit != mSrcIds.end(); ++lit)
|
||||||
itemList.push_back(lit->second);
|
itemList.push_back(new RsMsgSrcId(*lit->second));
|
||||||
|
|
||||||
|
|
||||||
for(mit = msgOutgoing.begin(); mit != msgOutgoing.end(); ++mit)
|
for(mit = msgOutgoing.begin(); mit != msgOutgoing.end(); ++mit)
|
||||||
itemList.push_back(mit->second) ;
|
itemList.push_back(new RsMsgItem(*mit->second)) ;
|
||||||
|
|
||||||
for(mit2 = mTags.begin(); mit2 != mTags.end(); ++mit2)
|
for(mit2 = mTags.begin(); mit2 != mTags.end(); ++mit2)
|
||||||
itemList.push_back(mit2->second);
|
itemList.push_back(new RsMsgTagType(*mit2->second));
|
||||||
|
|
||||||
for(mit3 = mMsgTags.begin(); mit3 != mMsgTags.end(); ++mit3)
|
for(mit3 = mMsgTags.begin(); mit3 != mMsgTags.end(); ++mit3)
|
||||||
itemList.push_back(mit3->second);
|
itemList.push_back(new RsMsgTags(*mit3->second));
|
||||||
|
|
||||||
for(mit4 = mParentId.begin(); mit4 != mParentId.end(); ++mit4)
|
for(mit4 = mParentId.begin(); mit4 != mParentId.end(); ++mit4)
|
||||||
itemList.push_back(mit4->second);
|
itemList.push_back(new RsMsgParentId(*mit4->second));
|
||||||
|
|
||||||
|
RsMsgGRouterMap *grmap = new RsMsgGRouterMap ;
|
||||||
|
grmap->ongoing_msgs = _ongoing_messages ;
|
||||||
|
|
||||||
|
itemList.push_back(grmap) ;
|
||||||
|
|
||||||
RsConfigKeyValueSet *vitem = new RsConfigKeyValueSet ;
|
RsConfigKeyValueSet *vitem = new RsConfigKeyValueSet ;
|
||||||
RsTlvKeyValue kv;
|
RsTlvKeyValue kv;
|
||||||
@ -526,6 +531,7 @@ bool p3MsgService::loadList(std::list<RsItem*>& load)
|
|||||||
RsMsgTags* mti;
|
RsMsgTags* mti;
|
||||||
RsMsgSrcId* msi;
|
RsMsgSrcId* msi;
|
||||||
RsMsgParentId* msp;
|
RsMsgParentId* msp;
|
||||||
|
RsMsgGRouterMap* grm;
|
||||||
// RsPublicMsgInviteConfigItem* msv;
|
// RsPublicMsgInviteConfigItem* msv;
|
||||||
|
|
||||||
std::list<RsMsgItem*> items;
|
std::list<RsMsgItem*> items;
|
||||||
@ -545,6 +551,12 @@ bool p3MsgService::loadList(std::list<RsItem*>& load)
|
|||||||
mMsgUniqueId = mitem->msgId + 1;
|
mMsgUniqueId = mitem->msgId + 1;
|
||||||
}
|
}
|
||||||
items.push_back(mitem);
|
items.push_back(mitem);
|
||||||
|
}
|
||||||
|
else if (NULL != (grm = dynamic_cast<RsMsgGRouterMap *>(*it)))
|
||||||
|
{
|
||||||
|
// merge.
|
||||||
|
for(std::map<GRouterMsgPropagationId,uint32_t>::const_iterator it(grm->ongoing_msgs.begin());it!=grm->ongoing_msgs.end();++it)
|
||||||
|
_ongoing_messages.insert(*it) ;
|
||||||
}
|
}
|
||||||
else if(NULL != (mtt = dynamic_cast<RsMsgTagType *>(*it)))
|
else if(NULL != (mtt = dynamic_cast<RsMsgTagType *>(*it)))
|
||||||
{
|
{
|
||||||
|
Loading…
Reference in New Issue
Block a user