151 lines
4.6 KiB
C++
151 lines
4.6 KiB
C++
//
|
|
// The developer of the original code and/or files is Tripwire, Inc.
|
|
// Portions created by Tripwire, Inc. are copyright (C) 2000-2017 Tripwire,
|
|
// Inc. Tripwire is a registered trademark of Tripwire, Inc. All rights
|
|
// reserved.
|
|
//
|
|
// This program is free software. The contents of this file are subject
|
|
// to the terms of the GNU General Public License as published by the
|
|
// Free Software Foundation; either version 2 of the License, or (at your
|
|
// option) any later version. You may redistribute it and/or modify it
|
|
// only in compliance with the GNU General Public License.
|
|
//
|
|
// This program is distributed in the hope that it will be useful.
|
|
// However, this program is distributed AS-IS WITHOUT ANY
|
|
// WARRANTY; INCLUDING THE IMPLIED WARRANTY OF MERCHANTABILITY OR FITNESS
|
|
// FOR A PARTICULAR PURPOSE. Please see the GNU General Public License
|
|
// for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program; if not, write to the Free Software
|
|
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
|
|
// USA.
|
|
//
|
|
// Nothing in the GNU General Public License or any other license to use
|
|
// the code or files shall permit you to use Tripwire's trademarks,
|
|
// service marks, or other intellectual property without Tripwire's
|
|
// prior written consent.
|
|
//
|
|
// If you have any questions, please contact Tripwire, Inc. at either
|
|
// info@tripwire.org or www.tripwire.org.
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// srefcountobj_t.cpp
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "core/stdcore.h"
|
|
#include "core/serializerimpl.h"
|
|
|
|
#include "core/errorgeneral.h"
|
|
#include "core/debug.h"
|
|
#include "core/archive.h"
|
|
#include "core/srefcountobj.h"
|
|
#include "twtest/test.h"
|
|
|
|
class cSerRefCountObjTest : public iSerRefCountObj
|
|
{
|
|
DECLARE_SERREFCOUNT()
|
|
|
|
public:
|
|
cSerRefCountObjTest();
|
|
|
|
protected:
|
|
~cSerRefCountObjTest();
|
|
|
|
// iSerializable interface
|
|
virtual void Read (iSerializer* pSerializer, int32 version = 0); // throw (eSerializer, eArchive)
|
|
virtual void Write(iSerializer* pSerializer) const; // throw (eSerializer, eArchive)
|
|
|
|
private:
|
|
int mDummyData;
|
|
|
|
};
|
|
|
|
IMPLEMENT_SERREFCOUNT(cSerRefCountObjTest, _T("cSerRefCountObjTest"), 0, 1);
|
|
|
|
cSerRefCountObjTest::cSerRefCountObjTest()
|
|
{
|
|
static int counter;
|
|
mDummyData = ++counter;
|
|
}
|
|
|
|
cSerRefCountObjTest::~cSerRefCountObjTest()
|
|
{
|
|
}
|
|
|
|
|
|
void cSerRefCountObjTest::Read(iSerializer* pSerializer, int32 version)
|
|
{
|
|
if (version > Version())
|
|
ThrowAndAssert(eSerializerVersionMismatch(_T("cSerRefCountObjTest::Read()")));
|
|
|
|
int32 dummy;
|
|
pSerializer->ReadInt32(dummy);
|
|
mDummyData = dummy;
|
|
}
|
|
|
|
void cSerRefCountObjTest::Write(iSerializer* pSerializer) const
|
|
{
|
|
pSerializer->WriteInt32(mDummyData);
|
|
}
|
|
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void TestSerRefCountObj()
|
|
{
|
|
// first, we need to register the object with the serializer class...
|
|
cSerializerImpl::RegisterSerializableRefCt(CLASS_TYPE(cSerRefCountObjTest), cSerRefCountObjTest::Create);
|
|
|
|
cSerRefCountObjTest* pObj1 = new cSerRefCountObjTest;
|
|
cSerRefCountObjTest* pObj2 = new cSerRefCountObjTest;
|
|
pObj1->AddRef();
|
|
cSerRefCountObjTest* pObj3 = pObj1;
|
|
pObj1->AddRef();
|
|
cSerRefCountObjTest* pObj4 = pObj1;
|
|
|
|
cMemoryArchive memarchive;
|
|
{
|
|
cSerializerImpl serializer(memarchive, cSerializerImpl::S_WRITE);
|
|
serializer.Init();
|
|
serializer.WriteObjectDynCreate(pObj1);
|
|
serializer.WriteObjectDynCreate(pObj2);
|
|
serializer.WriteObjectDynCreate(pObj3);
|
|
serializer.WriteObjectDynCreate(pObj4);
|
|
serializer.Finit();
|
|
}
|
|
|
|
pObj1->Release();
|
|
pObj2->Release();
|
|
pObj3->Release();
|
|
pObj4->Release();
|
|
|
|
{
|
|
memarchive.Seek(0, cBidirArchive::BEGINNING);
|
|
cSerializerImpl serializer(memarchive, cSerializerImpl::S_READ);
|
|
serializer.Init();
|
|
pObj1 = static_cast<cSerRefCountObjTest*>(serializer.ReadObjectDynCreate());
|
|
pObj2 = static_cast<cSerRefCountObjTest*>(serializer.ReadObjectDynCreate());
|
|
pObj3 = static_cast<cSerRefCountObjTest*>(serializer.ReadObjectDynCreate());
|
|
pObj4 = static_cast<cSerRefCountObjTest*>(serializer.ReadObjectDynCreate());
|
|
serializer.Finit();
|
|
}
|
|
|
|
TEST(pObj1 == pObj3);
|
|
TEST(pObj1 == pObj4);
|
|
TEST(pObj1 != pObj2);
|
|
|
|
pObj1->Release();
|
|
pObj2->Release();
|
|
pObj3->Release();
|
|
pObj4->Release();
|
|
|
|
return;
|
|
}
|
|
|
|
void RegisterSuite_SerRefCountObj()
|
|
{
|
|
RegisterTest("SerRefCountObj", "Basic", TestSerRefCountObj);
|
|
}
|