/* * libretroshare/src/serialiser: rsconfigitems.cc * * RetroShare Serialiser. * * Copyright 2007-2008 by Robert Fernie, Chris Parker. * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Library General Public * License Version 2 as published by the Free Software Foundation. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Library General Public License for more details. * * You should have received a copy of the GNU Library General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 * USA. * * Please report all bugs and problems to "retroshare@lunamutt.com". * */ #include "serialiser/rsbaseserial.h" #include "serialiser/rsconfigitems.h" #define RSSERIAL_DEBUG 1 #include /*************************************************************************/ uint32_t RsFileTransferSerialiser::size(RsItem *i) { RsFileTransfer *rft; if (NULL != (rft = dynamic_cast(i))) { return sizeTransfer(rft); } return 0; } /* serialise the data to the buffer */ bool RsFileTransferSerialiser::serialise(RsItem *i, void *data, uint32_t *pktsize) { RsFileTransfer *rft; if (NULL != (rft = dynamic_cast(i))) { return serialiseTransfer(rft, data, pktsize); } return false; } RsItem *RsFileTransferSerialiser::deserialise(void *data, uint32_t *pktsize) { /* get the type and size */ uint32_t rstype = getRsItemId(data); if ((RS_PKT_VERSION1 != getRsItemVersion(rstype)) || (RS_PKT_CLASS_CONFIG != getRsItemClass(rstype)) || (RS_PKT_TYPE_FILE_CONFIG != getRsItemType(rstype))) { return NULL; /* wrong type */ } switch(getRsItemSubType(rstype)) { case RS_PKT_SUBTYPE_DEFAULT: return deserialiseTransfer(data, pktsize); break; default: return NULL; break; } return NULL; } /*************************************************************************/ RsFileTransfer::~RsFileTransfer() { return; } void RsFileTransfer::clear() { file.TlvClear(); allPeerIds.TlvClear(); cPeerId = ""; state = 0; in = false; transferred = 0; crate = 0; trate = 0; lrate = 0; ltransfer = 0; } std::ostream &RsFileTransfer::print(std::ostream &out, uint16_t indent) { printRsItemBase(out, "RsFileTransfer", indent); uint16_t int_Indent = indent + 2; file.print(out, int_Indent); allPeerIds.print(out, int_Indent); printIndent(out, int_Indent); out << "cPeerId: " << cPeerId << std::endl; printIndent(out, int_Indent); out << "State: " << state << std::endl; printIndent(out, int_Indent); out << "In/Out: " << in << std::endl; printIndent(out, int_Indent); out << "Transferred: " << transferred << std::endl; printIndent(out, int_Indent); out << "crate: " << crate << std::endl; printIndent(out, int_Indent); out << "trate: " << trate << std::endl; printIndent(out, int_Indent); out << "lrate: " << lrate << std::endl; printIndent(out, int_Indent); out << "ltransfer: " << ltransfer << std::endl; printRsItemEnd(out, "RsFileTransfer", indent); return out; } uint32_t RsFileTransferSerialiser::sizeTransfer(RsFileTransfer *item) { uint32_t s = 8; /* header */ s += item->file.TlvSize(); s += item->allPeerIds.TlvSize(); s += GetTlvStringSize(item->cPeerId); s += 2; /* state */ s += 2; /* in/out */ s += 8; /* transferred */ s += 4; /* crate */ s += 4; /* trate */ s += 4; /* lrate */ s += 4; /* ltransfer */ return s; } bool RsFileTransferSerialiser::serialiseTransfer(RsFileTransfer *item, void *data, uint32_t *pktsize) { uint32_t tlvsize = sizeTransfer(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); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() Header: " << ok << std::endl; std::cerr << "RsFileTransferSerialiser::serialiseTransfer() Size: " << tlvsize << std::endl; /* skip the header */ offset += 8; /* add mandatory parts first */ ok &= item->file.SetTlv(data, tlvsize, &offset); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() FileItem: " << ok << std::endl; ok &= item->allPeerIds.SetTlv(data, tlvsize, &offset); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() allPeerIds: " << ok << std::endl; ok &= SetTlvString(data, tlvsize, &offset, TLV_TYPE_STR_PEERID, item->cPeerId); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() cId: " << ok << std::endl; ok &= setRawUInt16(data, tlvsize, &offset, item->state); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() State: " << ok << std::endl; ok &= setRawUInt16(data, tlvsize, &offset, item->in); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() In/Out: " << ok << std::endl; ok &= setRawUInt64(data, tlvsize, &offset, item->transferred); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() Transferred: " << ok << std::endl; ok &= setRawUInt32(data, tlvsize, &offset, item->crate); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() crate: " << ok << std::endl; ok &= setRawUInt32(data, tlvsize, &offset, item->trate); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() trate: " << ok << std::endl; ok &= setRawUInt32(data, tlvsize, &offset, item->lrate); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() lrate: " << ok << std::endl; ok &= setRawUInt32(data, tlvsize, &offset, item->ltransfer); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() ltransfer: " << ok << std::endl; if (offset != tlvsize) { ok = false; std::cerr << "RsFileTransferSerialiser::serialiseTransfer() Size Error! " << std::endl; } return ok; } RsFileTransfer *RsFileTransferSerialiser::deserialiseTransfer(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_VERSION1 != getRsItemVersion(rstype)) || (RS_PKT_CLASS_CONFIG != getRsItemClass(rstype)) || (RS_PKT_TYPE_FILE_CONFIG != getRsItemType(rstype)) || (RS_PKT_SUBTYPE_DEFAULT != 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 */ RsFileTransfer *item = new RsFileTransfer(); item->clear(); /* skip the header */ offset += 8; /* get mandatory parts first */ ok &= item->file.GetTlv(data, rssize, &offset); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() FileItem: " << ok << std::endl; ok &= item->allPeerIds.GetTlv(data, rssize, &offset); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() allPeerIds: " << ok << std::endl; /* string */ ok &= GetTlvString(data, rssize, &offset, TLV_TYPE_STR_PEERID, item->cPeerId); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() cId: " << ok << std::endl; /* data */ ok &= getRawUInt16(data, rssize, &offset, &(item->state)); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() State: " << ok << std::endl; ok &= getRawUInt16(data, rssize, &offset, &(item->in)); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() In/Out: " << ok << std::endl; ok &= getRawUInt64(data, rssize, &offset, &(item->transferred)); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() Transferred: " << ok << std::endl; ok &= getRawUInt32(data, rssize, &offset, &(item->crate)); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() crate: " << ok << std::endl; ok &= getRawUInt32(data, rssize, &offset, &(item->trate)); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() trate: " << ok << std::endl; ok &= getRawUInt32(data, rssize, &offset, &(item->lrate)); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() lrate: " << ok << std::endl; ok &= getRawUInt32(data, rssize, &offset, &(item->ltransfer)); std::cerr << "RsFileTransferSerialiser::serialiseTransfer() ltransfer: " << ok << std::endl; if (offset != rssize) { /* error */ delete item; return NULL; } if (!ok) { delete item; return NULL; } return item; } /*************************************************************************/ /*************************************************************************/ /*************************************************************************/ /*************************************************************************/ /*************************************************************************/ RsGeneralConfigSerialiser::~RsGeneralConfigSerialiser() { return; } uint32_t RsGeneralConfigSerialiser::size(RsItem *i) { RsConfigKeyValueSet *kvs; if (NULL != (kvs = dynamic_cast(i))) { return sizeKeyValueSet(kvs); } else if (NULL != (kvs = dynamic_cast(i))) { return sizeKeyValueSet(kvs); } return 0; } /* serialise the data to the buffer */ bool RsGeneralConfigSerialiser::serialise(RsItem *i, void *data, uint32_t *pktsize) { RsConfigKeyValueSet *kvs; /* do reply first - as it is derived from Item */ if (NULL != (kvs = dynamic_cast(i))) { return serialiseKeyValueSet(kvs, data, pktsize); } else if (NULL != (kvs = dynamic_cast(i))) { return serialiseKeyValueSet(kvs, data, pktsize); } return false; } RsItem *RsGeneralConfigSerialiser::deserialise(void *data, uint32_t *pktsize) { /* get the type and size */ uint32_t rstype = getRsItemId(data); if ((RS_PKT_VERSION1 != getRsItemVersion(rstype)) || (RS_PKT_CLASS_CONFIG != getRsItemClass(rstype)) || (RS_PKT_TYPE_GENERAL_CONFIG != getRsItemType(rstype))) { std::cerr << "RsGeneralConfigSerialiser::deserialise() Wrong Type" << std::endl; return NULL; /* wrong type */ } switch(getRsItemSubType(rstype)) { case RS_PKT_SUBTYPE_KEY_VALUE: return deserialiseKeyValueSet(data, pktsize); break; default: return NULL; break; } return NULL; } /*************************************************************************/ RsConfigKeyValueSet::~RsConfigKeyValueSet() { return; } void RsConfigKeyValueSet::clear() { tlvkvs.pairs.clear(); } std::ostream &RsConfigKeyValueSet::print(std::ostream &out, uint16_t indent) { printRsItemBase(out, "RsConfigKeyValueSet", indent); uint16_t int_Indent = indent + 2; tlvkvs.print(out, int_Indent); printRsItemEnd(out, "RsConfigKeyValueSet", indent); return out; } uint32_t RsGeneralConfigSerialiser::sizeKeyValueSet(RsConfigKeyValueSet *item) { uint32_t s = 8; /* header */ s += item->tlvkvs.TlvSize(); return s; } /* serialise the data to the buffer */ bool RsGeneralConfigSerialiser::serialiseKeyValueSet(RsConfigKeyValueSet *item, void *data, uint32_t *pktsize) { uint32_t tlvsize = sizeKeyValueSet(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); std::cerr << "RsGeneralConfigSerialiser::serialiseKeyValueSet() Header: " << ok << std::endl; std::cerr << "RsGeneralConfigSerialiser::serialiseKeyValueSet() Size: " << tlvsize << std::endl; /* skip the header */ offset += 8; /* add mandatory parts first */ ok &= item->tlvkvs.SetTlv(data, tlvsize, &offset); std::cerr << "RsGeneralConfigSerialiser::serialiseKeyValueSet() kvs: " << ok << std::endl; if (offset != tlvsize) { ok = false; std::cerr << "RsGeneralConfigSerialiser::serialiseKeyValueSet() Size Error! " << std::endl; } return ok; } RsConfigKeyValueSet *RsGeneralConfigSerialiser::deserialiseKeyValueSet(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_VERSION1 != getRsItemVersion(rstype)) || (RS_PKT_CLASS_CONFIG != getRsItemClass(rstype)) || (RS_PKT_TYPE_GENERAL_CONFIG != getRsItemType(rstype)) || (RS_PKT_SUBTYPE_KEY_VALUE != getRsItemSubType(rstype))) { std::cerr << "RsGeneralConfigSerialiser::deserialiseKeyValueSet() Wrong Type" << std::endl; return NULL; /* wrong type */ } if (*pktsize < rssize) /* check size */ { std::cerr << "RsGeneralConfigSerialiser::deserialiseKeyValueSet() Not Enough Space" << std::endl; return NULL; /* not enough data */ } /* set the packet length */ *pktsize = rssize; bool ok = true; /* ready to load */ RsConfigKeyValueSet *item = new RsConfigKeyValueSet(); item->clear(); /* skip the header */ offset += 8; /* get mandatory parts first */ ok &= item->tlvkvs.GetTlv(data, rssize, &offset), std::cerr << "RsGeneralConfigSerialiser::deserialiseKeyValueSet() kvs: " << ok << std::endl; if (offset != rssize) { std::cerr << "RsGeneralConfigSerialiser::deserialiseKeyValueSet() offset != rssize" << std::endl; /* error */ delete item; return NULL; } if (!ok) { std::cerr << "RsGeneralConfigSerialiser::deserialiseKeyValueSet() ok = false" << std::endl; delete item; return NULL; } return item; } /*************************************************************************/ RsPeerConfig::~RsPeerConfig() { return; } void RsPeerConfig::clear() { status = 0; lastconn_ts = 0; lastrecv_ts = 0; nextconn_ts = 0; nextconn_period = 0; } std::ostream &RsPeerConfig::print(std::ostream &out, uint16_t indent) { printRsItemBase(out, "RsPeerConfig", indent); uint16_t int_Indent = indent + 2; printIndent(out, int_Indent); out << "status: " << status << std::endl; printIndent(out, int_Indent); out << "lastconn_ts: " << lastconn_ts << std::endl; printIndent(out, int_Indent); out << "lastrecv_ts: " << lastrecv_ts << std::endl; printIndent(out, int_Indent); out << "nextconn_ts: " << nextconn_ts << std::endl; printIndent(out, int_Indent); out << "nextconn_period: " << nextconn_period << std::endl; return out; } /*************************************************************************/ RsPeerConfigSerialiser::~RsPeerConfigSerialiser() { return; } uint32_t RsPeerConfigSerialiser::size(RsItem *i) { uint32_t s = 8; /* header */ s += 2; /* status */ s += 2; /* lastconnect_ts */ s += 2; /* lastrecv_ts */ s += 2; /* nextconn_ts */ s += 2; /* nextperiod */ return s; } bool RsPeerConfigSerialiser::serialise(RsItem *i, void *data, uint32_t *size) { RsPeerConfig *item = (RsPeerConfig *) i; uint32_t tlvsize = RsPeerConfigSerialiser::size(item); uint32_t offset = 0; if(*size < tlvsize) return false; /* not enough space */ *size = tlvsize; bool ok = true; // serialise header ok &= setRsItemHeader(data, tlvsize, item->PacketId(), tlvsize); std::cerr << "RsConfigSerialiser::serialise() Header: " << ok << std::endl; std::cerr << "RsConfigSerialiser::serialise() Header: " << tlvsize << std::endl; /* skip the header */ offset += 8; /* add mandatory parts first */ ok &= setRawUInt32(data, tlvsize, &offset, item->lastconn_ts); /* Mandatory */ std::cerr << "RsPeerConfigSerialiser::serialise() lastconn_ts: " << ok << std::endl; /* WRITE error check */ ok &= setRawUInt32(data, tlvsize, &offset, item->lastrecv_ts); /* Mandatory */ std::cerr << "RsPeerConfigSerialiser::serialise() lastrecv_ts: " << ok << std::endl; ok &= setRawUInt32(data, tlvsize, &offset, item->nextconn_ts); /* Mandatory */ std::cerr << "RsPeerConfigSerialiser::serialise() nextconn_ts: " << ok << std::endl; ok &= setRawUInt32(data, tlvsize, &offset, item->nextconn_period); /* Mandatory */ std::cerr << "RsPeerConfigSerialiser::serialise() nextconn_period: " << ok << std::endl; if(offset != tlvsize) { ok = false; std::cerr << "RsPeerConfigSerialiser::serialise() Size Error! " << std::endl; } return ok; } RsItem *RsPeerConfigSerialiser::deserialise(void *data, uint32_t *size) { /* get the type and size */ uint32_t rstype = getRsItemId(data); uint32_t rssize = getRsItemSize(data); uint32_t offset = 0; if ((RS_PKT_VERSION1 != getRsItemVersion(rstype)) || (RS_PKT_CLASS_CONFIG != getRsItemClass(rstype)) || (RS_PKT_TYPE_PEER_CONFIG != getRsItemType(rstype)) || (RS_PKT_SUBTYPE_DEFAULT != getRsItemSubType(rstype))) { return NULL; /* wrong type */ } if (*size < rssize) /* check size */ return NULL; /* not enough data */ /* set the packet length */ *size = rssize; bool ok = true; RsPeerConfig *item = new RsPeerConfig(); item->clear(); /* skip the header */ offset += 8; /* get mandatory parts first */ ok &= getRawUInt32(data, rssize, &offset, &(item->lastconn_ts)); std::cerr << "RsFileTransferSerialiser::deserilise() lastconn_ts: " << ok << std::endl; ok &= getRawUInt32(data, rssize, &offset, &(item->lastconn_ts)); std::cerr << "RsFileTransferSerialiser::deserilise() lastrecv_ts: " << ok << std::endl; ok &= getRawUInt32(data, rssize, &offset, &(item->lastconn_ts)); std::cerr << "RsFileTransferSerialiser::deserilise() nextconn_ts: " << ok << std::endl; ok &= getRawUInt32(data, rssize, &offset, &(item->lastconn_ts)); std::cerr << "RsFileTransferSerialiser::deserilise() nextconn_period: " << ok << std::endl; if (offset != rssize) { /* error */ delete item; return NULL; } return item; } /****************************************************************************/ RsCacheConfig::~RsCacheConfig() { return; } void RsCacheConfig::clear() { cacheid = 0; path = ""; name = ""; hash = ""; recvd = 0; } std::ostream &RsCacheConfig::print(std::ostream &out, uint16_t indent) { printRsItemBase(out, "RsCacheConfig", indent); // begin 'WRITE' check uint16_t int_Indent = indent + 2; printIndent(out, int_Indent); //indent out << "cacheid: " << cacheid << std::endl; // display value of cacheid printIndent(out, int_Indent); out << "path: " << path << std::endl; // display value of path printIndent(out, int_Indent); out << "name: " << name << std::endl; // display value of name printIndent(out, int_Indent); out << "hash: " << hash << std::endl; // display value of hash printIndent(out, int_Indent); out << "recvd: " << recvd << std::endl; // display value of recvd printRsItemEnd(out, "RsCacheConfig", indent); // end of 'WRITE' check return out; } /**************************************************************************/ RsCacheConfigSerialiser::~RsCacheConfigSerialiser() { return; } uint32_t RsCacheConfigSerialiser::size(RsItem *i) { RsCacheConfig *item = (RsCacheConfig *) i; uint32_t s = 8; // to store calculated size, initiailize with size of header s += 2; /* cacheid */ s += GetTlvStringSize(item->path); s += GetTlvStringSize(item->name); s += GetTlvStringSize(item->hash); s += 2; /* recvd */ return s; } bool RsCacheConfigSerialiser::serialise(RsItem *i, void *data, uint32_t *size) { RsCacheConfig *item = (RsCacheConfig *) i; uint32_t tlvsize = RsCacheConfigSerialiser::size(item); uint32_t offset = 0; if(*size < tlvsize) return false; /* not enough space */ *size = tlvsize; bool ok = true; ok &=setRsItemHeader(data, tlvsize, item->PacketId(), tlvsize); std::cerr << "RsCacheConfigSerialiser::serialise() Header: " << ok << std::endl; std::cerr << "RsCacheConfigSerialiser::serialise() Size: " << size << std::endl; /* skip the header */ offset += 8; /* add the mandatory parts first */ ok &= setRawUInt32(data, tlvsize, &offset, item->cacheid); std::cerr << "RsCacheConfigSerialiser::serialise() cacheid: " << ok << std::endl; ok &= setRawUInt32(data, tlvsize, &offset, item->recvd); std::cerr << "RsCacheConfigSerialiser::serialise() recvd: " << ok << std::endl; ok &= SetTlvString(data, tlvsize, &offset, TLV_TYPE_STR_PATH, item->path); std::cerr << "RsCacheConfigSerialiser::serialise() path: " << ok << std::endl; ok &= SetTlvString(data, tlvsize, &offset, TLV_TYPE_STR_NAME, item->name); std::cerr << "RsCacheConfigSerialiser::serialise() name: " << ok << std::endl; ok &= SetTlvString(data, tlvsize, &offset, TLV_TYPE_STR_HASH_SHA1, item->hash); std::cerr << "RsCacheConfigSerialiser::serialise() hash: " << ok << std::endl; if (offset !=tlvsize) { ok = false; std::cerr << "RsConfigSerialiser::serialisertransfer() Size Error! " << std::endl; } return ok; } RsItem *RsCacheConfigSerialiser::deserialise(void *data, uint32_t *size) {/* get the type and size */ uint32_t rstype = getRsItemId(data); uint32_t rssize = getRsItemSize(data); uint32_t *offset; *offset = 0; uint32_t tlvend = *offset + rssize; if ((RS_PKT_VERSION1 != getRsItemVersion(rstype)) || (RS_PKT_CLASS_CONFIG != getRsItemClass(rstype)) || (RS_PKT_TYPE_CACHE_CONFIG != getRsItemType(rstype)) || (RS_PKT_SUBTYPE_DEFAULT != getRsItemSubType(rstype))) { return NULL; /* wrong type */ } if (*size < rssize) /* check size */ return NULL; /* not enough data */ /* set the packet length */ *size = rssize; bool ok = true; /* ready to load */ RsCacheConfig *item = new RsCacheConfig(); item->clear(); /* skip the header */ offset += 8; /* get mandatory parts first */ ok &= getRawUInt32(data, rssize, offset, &(item->cacheid)); std::cerr << "RsCacheConfigSerialiser::deserialise() cacheid: " << ok << std::endl; ok &= getRawUInt32(data, rssize, offset, &(item->recvd)); std::cerr << "RsCacheConfigSerialiser::deserialise() recvd: " << ok << std::endl; while((*offset) + 2 < tlvend) { /* get the next type */ uint16_t tlvsubtype = GetTlvType( &(((uint8_t *) data)[*offset]) ); switch(tlvsubtype) { case TLV_TYPE_STR_PATH: ok &= GetTlvString(data, tlvend, offset, tlvsubtype, item->path); break; case TLV_TYPE_STR_NAME: ok &= GetTlvString(data, tlvend, offset, tlvsubtype, item->name); break; case TLV_TYPE_STR_HASH_SHA1: ok &= GetTlvString(data, tlvend, offset, tlvsubtype, item->hash); break; default: break; } if (!ok) { return false; } } return item; }