mirror of
https://github.com/RetroShare/RetroShare.git
synced 2024-12-10 00:14:49 -05:00
247 lines
6.7 KiB
C
247 lines
6.7 KiB
C
|
#ifndef SUPPORT_H_
|
||
|
#define SUPPORT_H_
|
||
|
|
||
|
/*
|
||
|
* libretroshare/src/tests/serialiser:
|
||
|
*
|
||
|
* RetroShare Serialiser tests.
|
||
|
*
|
||
|
* Copyright 2007-2008 by Christopher Evi-Parker, Cyril Soler
|
||
|
*
|
||
|
* 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 <gtest/gtest.h>
|
||
|
|
||
|
#include <string>
|
||
|
#include <stdint.h>
|
||
|
#include <iostream>
|
||
|
|
||
|
#include "serialiser/rsserial.h"
|
||
|
#include "serialiser/rstlvkeys.h"
|
||
|
#include "serialiser/rstlvbinary.h"
|
||
|
#include "serialiser/rstlvfileitem.h"
|
||
|
#include "serialiser/rstlvidset.h"
|
||
|
#include "serialiser/rstlvimage.h"
|
||
|
|
||
|
#include "rstlvutil.h"
|
||
|
|
||
|
|
||
|
/**
|
||
|
* This contains functions that may be useful for testing throughout the
|
||
|
* retroshare's serialiser
|
||
|
* package, if you find a function you keep using everywhere, might be a good idea to add it here
|
||
|
*/
|
||
|
|
||
|
|
||
|
#define SHORT_STR 100
|
||
|
#define LARGE_STR 1000
|
||
|
|
||
|
void randString(const uint32_t, std::string&);
|
||
|
void randString(const uint32_t, std::wstring&);
|
||
|
|
||
|
|
||
|
/* for testing compound tlv items */
|
||
|
|
||
|
void init_item(RsTlvSecurityKey&);
|
||
|
void init_item(RsTlvSecurityKeySet&);
|
||
|
void init_item(RsTlvKeySignature&);
|
||
|
void init_item(RsTlvKeySignatureSet&);
|
||
|
void init_item(RsTlvBinaryData&);
|
||
|
void init_item(RsTlvFileItem&);
|
||
|
void init_item(RsTlvFileSet&);
|
||
|
void init_item(RsTlvHashSet&);
|
||
|
void init_item(RsTlvPeerIdSet&);
|
||
|
void init_item(RsTlvImage&);
|
||
|
void init_item(RsTlvPeerIdSet&);
|
||
|
|
||
|
bool operator==(const RsTlvSecurityKey&, const RsTlvSecurityKey& );
|
||
|
bool operator==(const RsTlvSecurityKeySet&, const RsTlvSecurityKeySet& );
|
||
|
bool operator==(const RsTlvKeySignature&, const RsTlvKeySignature& );
|
||
|
bool operator==(const RsTlvBinaryData&, const RsTlvBinaryData&);
|
||
|
bool operator==(const RsTlvFileItem&, const RsTlvFileItem&);
|
||
|
bool operator==(const RsTlvFileSet&, const RsTlvFileSet& );
|
||
|
bool operator==(const RsTlvHashSet&, const RsTlvHashSet&);
|
||
|
bool operator==(const RsTlvImage&, const RsTlvImage& );
|
||
|
bool operator==(const RsTlvPeerIdSet& , const RsTlvPeerIdSet& );
|
||
|
bool operator==(const RsTlvKeySignatureSet& , const RsTlvKeySignatureSet& );
|
||
|
|
||
|
|
||
|
|
||
|
/*!
|
||
|
* This templated test function which allows you to test
|
||
|
* retroshare serialiser items (except compound tlv items)
|
||
|
* you the function must implement a function
|
||
|
*
|
||
|
* 'RsSerialType* init_item(YourRsItem& rs_item)'
|
||
|
* which returns valid serialiser that
|
||
|
* can serialiser rs_item. You also need to implement an operator
|
||
|
*
|
||
|
* Also need to implement a function
|
||
|
* 'bool operator =(YourRsItem& rs_itemL, YourRsItem& rs_temR)'
|
||
|
* which allows this function to test for equality between both parameters.
|
||
|
* rs_temR is the result of deserialising the left operand (1st parameter).
|
||
|
* not YourRsItem in specifier in about functions should be a derived type from RsItem
|
||
|
*
|
||
|
* @param T the item you want to test
|
||
|
*/
|
||
|
|
||
|
template<class T> int test_RsItem()
|
||
|
{
|
||
|
/* make a serialisable RsTurtleItem */
|
||
|
|
||
|
RsSerialiser srl;
|
||
|
|
||
|
/* initialise */
|
||
|
T rsfi ;
|
||
|
RsSerialType *rsfis = init_item(rsfi) ;
|
||
|
|
||
|
/* attempt to serialise it before we add it to the serialiser */
|
||
|
|
||
|
EXPECT_TRUE(0 == srl.size(&rsfi));
|
||
|
|
||
|
static const uint32_t MAX_BUFSIZE = 22000 ;
|
||
|
|
||
|
char *buffer = new char[MAX_BUFSIZE];
|
||
|
uint32_t sersize = MAX_BUFSIZE;
|
||
|
|
||
|
EXPECT_TRUE(false == srl.serialise(&rsfi, (void *) buffer, &sersize));
|
||
|
|
||
|
/* now add to serialiser */
|
||
|
|
||
|
srl.addSerialType(rsfis);
|
||
|
|
||
|
uint32_t size = srl.size(&rsfi);
|
||
|
bool done = srl.serialise(&rsfi, (void *) buffer, &sersize);
|
||
|
|
||
|
std::cerr << "test_Item() size: " << size << std::endl;
|
||
|
std::cerr << "test_Item() done: " << done << std::endl;
|
||
|
std::cerr << "test_Item() sersize: " << sersize << std::endl;
|
||
|
|
||
|
std::cerr << "test_Item() serialised:" << std::endl;
|
||
|
//displayRawPacket(std::cerr, (void *) buffer, sersize);
|
||
|
|
||
|
EXPECT_TRUE(done == true);
|
||
|
|
||
|
uint32_t sersize2 = sersize;
|
||
|
RsItem *output = srl.deserialise((void *) buffer, &sersize2);
|
||
|
|
||
|
EXPECT_TRUE(output != NULL);
|
||
|
if (!output)
|
||
|
{
|
||
|
return 1;
|
||
|
}
|
||
|
|
||
|
EXPECT_TRUE(sersize2 == sersize);
|
||
|
|
||
|
T *outfi = dynamic_cast<T *>(output);
|
||
|
|
||
|
EXPECT_TRUE(outfi != NULL);
|
||
|
if (!outfi)
|
||
|
{
|
||
|
return 1;
|
||
|
}
|
||
|
|
||
|
if(!(*outfi == rsfi))
|
||
|
{
|
||
|
std::cerr << "Items differ: "<< std::endl;
|
||
|
outfi->print(std::cerr,0) ;
|
||
|
rsfi.print(std::cerr,0) ;
|
||
|
}
|
||
|
if (outfi)
|
||
|
EXPECT_TRUE(*outfi == rsfi) ;
|
||
|
|
||
|
|
||
|
sersize2 = MAX_BUFSIZE;
|
||
|
bool done2 = srl.serialise(outfi, (void *) &(buffer[16*8]), &sersize2);
|
||
|
|
||
|
EXPECT_TRUE(done2) ;
|
||
|
EXPECT_TRUE(sersize2 == sersize);
|
||
|
|
||
|
// displayRawPacket(std::cerr, (void *) buffer, 16 * 8 + sersize2);
|
||
|
|
||
|
delete[] buffer ;
|
||
|
//delete rsfis;
|
||
|
|
||
|
return 1;
|
||
|
}
|
||
|
|
||
|
template<class T> int test_RsItem(uint16_t servtype)
|
||
|
{
|
||
|
/* make a serialisable RsTurtleItem */
|
||
|
|
||
|
RsSerialiser srl;
|
||
|
|
||
|
/* initialise */
|
||
|
T rsfi(servtype) ;
|
||
|
RsSerialType *rsfis = init_item(rsfi) ; // deleted on destruction of srl
|
||
|
|
||
|
/* attempt to serialise it before we add it to the serialiser */
|
||
|
|
||
|
EXPECT_TRUE(0 == srl.size(&rsfi));
|
||
|
|
||
|
static const uint32_t MAX_BUFSIZE = 22000 ;
|
||
|
|
||
|
char *buffer = new char[MAX_BUFSIZE];
|
||
|
uint32_t sersize = MAX_BUFSIZE;
|
||
|
|
||
|
EXPECT_TRUE(false == srl.serialise(&rsfi, (void *) buffer, &sersize));
|
||
|
|
||
|
/* now add to serialiser */
|
||
|
|
||
|
srl.addSerialType(rsfis);
|
||
|
|
||
|
uint32_t size = srl.size(&rsfi);
|
||
|
bool done = srl.serialise(&rsfi, (void *) buffer, &sersize);
|
||
|
|
||
|
std::cerr << "test_Item() size: " << size << std::endl;
|
||
|
std::cerr << "test_Item() done: " << done << std::endl;
|
||
|
std::cerr << "test_Item() sersize: " << sersize << std::endl;
|
||
|
|
||
|
std::cerr << "test_Item() serialised:" << std::endl;
|
||
|
//displayRawPacket(std::cerr, (void *) buffer, sersize);
|
||
|
|
||
|
EXPECT_TRUE(done == true);
|
||
|
|
||
|
uint32_t sersize2 = sersize;
|
||
|
RsItem *output = srl.deserialise((void *) buffer, &sersize2);
|
||
|
|
||
|
EXPECT_TRUE(output != NULL);
|
||
|
EXPECT_TRUE(sersize2 == sersize);
|
||
|
|
||
|
T *outfi = dynamic_cast<T *>(output);
|
||
|
|
||
|
EXPECT_TRUE(outfi != NULL);
|
||
|
|
||
|
if (outfi)
|
||
|
EXPECT_TRUE(*outfi == rsfi) ;
|
||
|
|
||
|
sersize2 = MAX_BUFSIZE;
|
||
|
bool done2 = srl.serialise(outfi, (void *) &(buffer[16*8]), &sersize2);
|
||
|
|
||
|
EXPECT_TRUE(done2) ;
|
||
|
EXPECT_TRUE(sersize2 == sersize);
|
||
|
|
||
|
// displayRawPacket(std::cerr, (void *) buffer, 16 * 8 + sersize2);
|
||
|
|
||
|
delete[] buffer ;
|
||
|
|
||
|
return 1;
|
||
|
}
|
||
|
#endif /* SUPPORT_H_ */
|