| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315 |
- /*
- * Connection.h, part of VCMI engine
- *
- * Authors: listed in file AUTHORS in main folder
- *
- * License: GNU General Public License v2.0 or later
- * Full text of license available in license.txt file, in main folder
- *
- */
- #pragma once
- #include <typeinfo> //XXX this is in namespace std if you want w/o use typeinfo.h?
- #include <boost/type_traits/is_fundamental.hpp>
- #include <boost/type_traits/is_enum.hpp>
- #include <boost/type_traits/is_pointer.hpp>
- #include <boost/type_traits/is_class.hpp>
- #include <boost/type_traits/is_base_of.hpp>
- #include <boost/type_traits/is_array.hpp>
- #include <boost/type_traits/remove_pointer.hpp>
- #include <boost/type_traits/remove_const.hpp>
- #include <boost/variant.hpp>
- #include <boost/mpl/eval_if.hpp>
- #include <boost/mpl/equal_to.hpp>
- #include <boost/mpl/int.hpp>
- #include <boost/mpl/identity.hpp>
- #include <boost/any.hpp>
- #include "ConstTransitivePtr.h"
- #include "CCreatureSet.h" //for CStackInstance
- #include "CObjectHandler.h" //for CArmedInstance
- #include "Mapping/CCampaignHandler.h" //for CCampaignState
- const ui32 version = 737;
- const TSlot COMMANDER_SLOT_PLACEHOLDER = -2;
- class CConnection;
- class CGObjectInstance;
- class CStackInstance;
- class CGameState;
- class CCreature;
- class LibClasses;
- class CHero;
- struct CPack;
- extern DLL_LINKAGE LibClasses * VLC;
- namespace mpl = boost::mpl;
- namespace boost
- {
- namespace asio
- {
- namespace ip
- {
- class tcp;
- }
- class io_service;
- template <typename Protocol> class stream_socket_service;
- template <typename Protocol,typename StreamSocketService>
- class basic_stream_socket;
- template <typename Protocol> class socket_acceptor_service;
- template <typename Protocol,typename SocketAcceptorService>
- class basic_socket_acceptor;
- }
- class mutex;
- }
- enum SerializationLvl
- {
- Wrong=0,
- Boolean,
- Primitive,
- Array,
- Pointer,
- Enum,
- Serializable,
- BooleanVector
- };
- struct TypeComparer
- {
- bool operator()(const std::type_info *a, const std::type_info *b) const
- {
- return a->before(*b);
- }
- };
- class DLL_LINKAGE CTypeList
- {
- typedef std::multimap<const std::type_info *,ui16,TypeComparer> TTypeMap;
- TTypeMap types;
- public:
- CTypeList();
- ui16 registerType(const std::type_info *type);
- template <typename T> ui16 registerType(const T * t = NULL)
- {
- return registerType(getTypeInfo(t));
- }
- ui16 getTypeID(const std::type_info *type);
- template <typename T> ui16 getTypeID(const T * t = NULL)
- {
- return getTypeID(getTypeInfo(t));
- }
- template <typename T> const std::type_info * getTypeInfo(const T * t = NULL)
- {
- if(t)
- return &typeid(*t);
- else
- return &typeid(T);
- }
- };
- extern DLL_LINKAGE CTypeList typeList;
- template<typename Ser>
- struct SaveBoolean
- {
- static void invoke(Ser &s, const bool &data)
- {
- s.saveBoolean(data);
- }
- };
- template<typename Ser>
- struct LoadBoolean
- {
- static void invoke(Ser &s, bool &data)
- {
- s.loadBoolean(data);
- }
- };
- template<typename Ser>
- struct SaveBooleanVector
- {
- static void invoke(Ser &s, const std::vector<bool> &data)
- {
- s.saveBooleanVector(data);
- }
- };
- template<typename Ser>
- struct LoadBooleanVector
- {
- static void invoke(Ser &s, std::vector<bool> &data)
- {
- s.loadBooleanVector(data);
- }
- };
- template<typename Ser,typename T>
- struct SavePrimitive
- {
- static void invoke(Ser &s, const T &data)
- {
- s.savePrimitive(data);
- }
- };
- template<typename Ser,typename T>
- struct SaveSerializable
- {
- static void invoke(Ser &s, const T &data)
- {
- s.saveSerializable(data);
- }
- };
- template<typename Ser,typename T>
- struct SaveEnum
- {
- static void invoke(Ser &s, const T &data)
- {
- s.saveEnum(data);
- }
- };
- template<typename Ser,typename T>
- struct LoadEnum
- {
- static void invoke(Ser &s, T &data)
- {
- s.loadEnum(data);
- }
- };
- template<typename Ser,typename T>
- struct LoadPrimitive
- {
- static void invoke(Ser &s, T &data)
- {
- s.loadPrimitive(data);
- }
- };
- template<typename Ser,typename T>
- struct SavePointer
- {
- static void invoke(Ser &s, const T &data)
- {
- s.savePointer(data);
- }
- };
- template<typename Ser,typename T>
- struct LoadPointer
- {
- static void invoke(Ser &s, T &data)
- {
- s.loadPointer(data);
- }
- };
- template<typename Ser,typename T>
- struct SaveArray
- {
- static void invoke(Ser &s, const T &data)
- {
- s.saveArray(data);
- }
- };
- template<typename Ser,typename T>
- struct LoadArray
- {
- static void invoke(Ser &s, T &data)
- {
- s.loadArray(data);
- }
- };
- template<typename Ser,typename T>
- struct LoadSerializable
- {
- static void invoke(Ser &s, T &data)
- {
- s.loadSerializable(data);
- }
- };
- template<typename Ser,typename T>
- struct SaveWrong
- {
- static void invoke(Ser &s, const T &data)
- {
- throw std::runtime_error("Wrong save serialization call!");
- }
- };
- template<typename Ser,typename T>
- struct LoadWrong
- {
- static void invoke(Ser &s, const T &data)
- {
- throw std::runtime_error("Wrong load serialization call!");
- }
- };
- template<typename T>
- struct SerializationLevel
- {
- typedef mpl::integral_c_tag tag;
- typedef
- typename mpl::eval_if<
- boost::is_same<T, bool>,
- mpl::int_<Boolean>,
- //else
- typename mpl::eval_if<
- boost::is_same<T, std::vector<bool> >,
- mpl::int_<BooleanVector>,
- //else
- typename mpl::eval_if<
- boost::is_fundamental<T>,
- mpl::int_<Primitive>,
- //else
- typename mpl::eval_if<
- boost::is_enum<T>,
- mpl::int_<Enum>,
- //else
- typename mpl::eval_if<
- boost::is_class<T>,
- mpl::int_<Serializable>,
- //else
- typename mpl::eval_if<
- boost::is_array<T>,
- mpl::int_<Array>,
- //else
- typename mpl::eval_if<
- boost::is_pointer<T>,
- mpl::int_<Pointer>,
- //else
- typename mpl::eval_if<
- boost::is_enum<T>,
- mpl::int_<Primitive>,
- //else
- mpl::int_<Wrong>
- >
- >
- >
- >
- >
- >
- >
- >::type type;
- static const int value = SerializationLevel::type::value;
- };
- template <typename T, typename U>
- struct VectorisedObjectInfo
- {
- const std::vector<ConstTransitivePtr<T> > *vector; //pointer to the appropriate vector
- const U T::*idPtr; //pointer to the field representing the position in the vector
- VectorisedObjectInfo(const std::vector< ConstTransitivePtr<T> > *Vector, const U T::*IdPtr)
- :vector(Vector), idPtr(IdPtr)
- {
- }
- };
- /// Class which is responsible for storing and loading data.
- class DLL_LINKAGE CSerializer
- {
- public:
- typedef std::map<const std::type_info *, boost::any, TypeComparer> TTypeVecMap;
- TTypeVecMap vectors; //entry must be a pointer to vector containing pointers to the objects of key type
- bool smartVectorMembersSerialization;
- bool sendStackInstanceByIds;
- CSerializer();
- ~CSerializer();
- virtual void reportState(CLogger &out){};
- template <typename T, typename U>
- void registerVectoredType(const std::vector<T*> *Vector, const U T::*IdPtr)
- {
- vectors[&typeid(T)] = VectorisedObjectInfo<T, U>(Vector, IdPtr);
- }
- template <typename T, typename U>
- void registerVectoredType(const std::vector<ConstTransitivePtr<T> > *Vector, const U T::*IdPtr)
- {
- vectors[&typeid(T)] = VectorisedObjectInfo<T, U>(Vector, IdPtr);
- }
- template <typename T, typename U>
- const VectorisedObjectInfo<T, U> *getVectorisedTypeInfo()
- {
- const std::type_info *myType = NULL;
- //
- // if(boost::is_base_of<CGObjectInstance, T>::value) //ugly workaround to support also types derived from CGObjectInstance -> if we encounter one, treat it aas CGObj..
- // myType = &typeid(CGObjectInstance);
- // else
- myType = &typeid(T);
- TTypeVecMap::iterator i = vectors.find(myType);
- if(i == vectors.end())
- return NULL;
- else
- {
- assert(!i->second.empty());
- assert(i->second.type() == typeid(VectorisedObjectInfo<T, U>));
- VectorisedObjectInfo<T, U> *ret = &(boost::any_cast<VectorisedObjectInfo<T, U>&>(i->second));
- return ret;
- }
- }
- template <typename T, typename U>
- T* getVectorItemFromId(const VectorisedObjectInfo<T, U> &oInfo, U id) const
- {
- /* if(id < 0)
- return NULL;*/
- si32 idAsNumber = idToNumber(id);
- assert(oInfo.vector);
- assert(oInfo.vector->size() > idAsNumber);
- return const_cast<T*>((*oInfo.vector)[idAsNumber].get());
- }
- template <typename T, typename U>
- U getIdFromVectorItem(const VectorisedObjectInfo<T, U> &oInfo, const T* obj) const
- {
- if(!obj)
- return U(-1);
- return obj->*oInfo.idPtr;
- }
- void addStdVecItems(CGameState *gs, LibClasses *lib = VLC);
- };
- class DLL_LINKAGE CSaverBase : public virtual CSerializer
- {
- };
- class CBasicPointerSaver
- {
- public:
- virtual void savePtr(CSaverBase &ar, const void *data) const =0;
- virtual ~CBasicPointerSaver(){}
- };
- template <typename Serializer, typename T> class CPointerSaver : public CBasicPointerSaver
- {
- public:
- void savePtr(CSaverBase &ar, const void *data) const
- {
- Serializer &s = static_cast<Serializer&>(ar);
- const T *ptr = static_cast<const T*>(data);
- //T is most derived known type, it's time to call actual serialize
- const_cast<T&>(*ptr).serialize(s,version);
- }
- };
- template <typename T> //metafunction returning CGObjectInstance if T is its derivate or T elsewise
- struct VectorisedTypeFor
- {
- typedef typename
- //if
- mpl::eval_if<boost::is_base_of<CGObjectInstance,T>,
- mpl::identity<CGObjectInstance>,
- //else
- mpl::identity<T>
- >::type type;
- };
- template <typename U>
- struct VectorizedIDType
- {
- typedef typename
- //if
- mpl::eval_if<boost::is_same<CArtifact,U>,
- mpl::identity<ArtifactID>,
- //else if
- mpl::eval_if<boost::is_same<CCreature,U>,
- mpl::identity<CreatureID>,
- //else if
- mpl::eval_if<boost::is_same<CArtifactInstance,U>,
- mpl::identity<ArtifactInstanceID>,
- //else if
- mpl::eval_if<boost::is_base_of<CGObjectInstance,U>,
- mpl::identity<ObjectInstanceID>,
- //else
- mpl::identity<si32>
- > > > >::type type;
- };
- template<typename T>
- si32 idToNumber(const T &t, typename boost::enable_if<boost::is_convertible<T,si32> >::type * dummy = 0)
- {
- return t;
- }
- template<typename T>
- si32 idToNumber(const BaseForID<T> &t)
- {
- return t.getNum();
- }
- template <typename Handler>
- struct VariantVisitorSaver : boost::static_visitor<>
- {
- Handler &h;
- VariantVisitorSaver(Handler &H):h(H)
- {
- }
- template <typename T>
- void operator()(const T &t)
- {
- h << t;
- }
- };
- template<typename Ser,typename T>
- struct SaveIfStackInstance
- {
- static bool invoke(Ser &s, const T &data)
- {
- return false;
- }
- };
- template<typename Ser>
- struct SaveIfStackInstance<Ser, CStackInstance *>
- {
- static bool invoke(Ser &s, const CStackInstance* const &data)
- {
- assert(data->armyObj);
- TSlot slot = -1;
- if(data->getNodeType() == CBonusSystemNode::COMMANDER)
- slot = COMMANDER_SLOT_PLACEHOLDER;
- else
- slot = data->armyObj->findStack(data);
- assert(slot != -1);
- s << data->armyObj << slot;
- return true;
- }
- };
- template<typename Ser,typename T>
- struct LoadIfStackInstance
- {
- static bool invoke(Ser &s, T &data)
- {
- return false;
- }
- };
- template<typename Ser>
- struct LoadIfStackInstance<Ser, CStackInstance *>
- {
- static bool invoke(Ser &s, CStackInstance* &data)
- {
- CArmedInstance *armedObj;
- TSlot slot;
- s >> armedObj >> slot;
- if(slot != COMMANDER_SLOT_PLACEHOLDER)
- {
- assert(armedObj->hasStackAtSlot(slot));
- data = armedObj->stacks[slot];
- }
- else
- {
- auto hero = dynamic_cast<CGHeroInstance *>(armedObj);
- assert(hero);
- assert(hero->commander);
- data = hero->commander;
- }
- return true;
- }
- };
- /// The class which manages saving objects.
- template <typename Serializer> class DLL_LINKAGE COSer : public CSaverBase
- {
- public:
- bool saving;
- std::map<ui16,CBasicPointerSaver*> savers; // typeID => CPointerSaver<serializer,type>
- std::map<const void*, ui32> savedPointers;
- bool smartPointerSerialization;
- COSer()
- {
- saving=true;
- smartPointerSerialization = true;
- }
- ~COSer()
- {
- std::map<ui16,CBasicPointerSaver*>::iterator iter;
- for(iter = savers.begin(); iter != savers.end(); iter++)
- delete iter->second;
- }
- template<typename T> void registerType(const T * t=NULL)
- {
- ui16 ID = typeList.registerType(t);
- savers[ID] = new CPointerSaver<COSer<Serializer>,T>;
- }
- Serializer * This()
- {
- return static_cast<Serializer*>(this);
- }
- template<class T>
- Serializer & operator<<(const T &t)
- {
- this->This()->save(t);
- return * this->This();
- }
- template<class T>
- COSer & operator&(const T & t)
- {
- return * this->This() << t;
- }
- int write(const void * data, unsigned size);
- template <typename T>
- void savePrimitive(const T &data)
- {
- this->This()->write(&data,sizeof(data));
- }
- template <typename T>
- void savePointer(const T &data)
- {
- //write if pointer is not NULL
- ui8 hlp = (data!=NULL);
- *this << hlp;
- //if pointer is NULL then we don't need anything more...
- if(!hlp)
- return;
- if(smartVectorMembersSerialization)
- {
- typedef typename boost::remove_const<typename boost::remove_pointer<T>::type>::type TObjectType;
- typedef typename VectorisedTypeFor<TObjectType>::type VType;
- typedef typename VectorizedIDType<TObjectType>::type IDType;
- if(const auto *info = getVectorisedTypeInfo<VType, IDType>())
- {
- IDType id = getIdFromVectorItem<VType>(*info, data);
- *this << id;
- if(id != IDType(-1)) //vector id is enough
- return;
- }
- }
- if(sendStackInstanceByIds)
- {
- const bool gotSaved = SaveIfStackInstance<Serializer,T>::invoke(*This(), data);
- if(gotSaved)
- return;
- }
- if(smartPointerSerialization)
- {
- std::map<const void*,ui32>::iterator i = savedPointers.find(data);
- if(i != savedPointers.end())
- {
- //this pointer has been already serialized - write only it's id
- *this << i->second;
- return;
- }
- //give id to this pointer
- ui32 pid = (ui32)savedPointers.size();
- savedPointers[data] = pid;
- *this << pid;
- }
- //write type identifier
- ui16 tid = typeList.getTypeID(data);
- *this << tid;
- This()->savePointerHlp(tid, data);
- }
- //that part of ptr serialization was extracted to allow customization of its behavior in derived classes
- template <typename T>
- void savePointerHlp(ui16 tid, const T &data)
- {
- if(!tid)
- *this << *data; //if type is unregistered simply write all data in a standard way
- else
- savers[tid]->savePtr(*this,data); //call serializer specific for our real type
- }
- template <typename T>
- void saveArray(const T &data)
- {
- ui32 size = ARRAY_COUNT(data);
- for(ui32 i=0; i < size; i++)
- *this << data[i];
- }
- template <typename T>
- void save(const T &data)
- {
- typedef
- //if
- typename mpl::eval_if< mpl::equal_to<SerializationLevel<T>,mpl::int_<Boolean> >,
- mpl::identity<SaveBoolean<Serializer> >,
- //else if
- typename mpl::eval_if< mpl::equal_to<SerializationLevel<T>,mpl::int_<BooleanVector> >,
- mpl::identity<SaveBooleanVector<Serializer> >,
- //else if
- typename mpl::eval_if< mpl::equal_to<SerializationLevel<T>,mpl::int_<Primitive> >,
- mpl::identity<SavePrimitive<Serializer,T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Enum> >,
- mpl::identity<SaveEnum<Serializer,T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Pointer> >,
- mpl::identity<SavePointer<Serializer,T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Array> >,
- mpl::identity<SaveArray<Serializer,T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Serializable> >,
- mpl::identity<SaveSerializable<Serializer,T> >,
- //else
- mpl::identity<SaveWrong<Serializer,T> >
- >
- >
- >
- >
- >
- >
- >::type typex;
- typex::invoke(* this->This(), data);
- }
- template <typename T>
- void saveSerializable(const T &data)
- {
- const_cast<T&>(data).serialize(*this,version);
- }
- template <typename T>
- void saveSerializable(const shared_ptr<T> &data)
- {
- T *internalPtr = data.get();
- *this << internalPtr;
- }
- template <typename T>
- void saveSerializable(const unique_ptr<T> &data)
- {
- T *internalPtr = data.get();
- *this << internalPtr;
- }
- template <typename T>
- void saveSerializable(const std::vector<T> &data)
- {
- ui32 length = data.size();
- *this << length;
- for(ui32 i=0;i<length;i++)
- *this << data[i];
- }
- template <typename T>
- void saveSerializable(const std::set<T> &data)
- {
- std::set<T> &d = const_cast<std::set<T> &>(data);
- ui32 length = d.size();
- *this << length;
- for(typename std::set<T>::iterator i=d.begin();i!=d.end();i++)
- *this << *i;
- }
- template <typename T, typename U>
- void saveSerializable(const boost::unordered_set<T, U> &data)
- {
- boost::unordered_set<T, U> &d = const_cast<boost::unordered_set<T, U> &>(data);
- ui32 length = d.size();
- *this << length;
- for(typename boost::unordered_set<T, U>::iterator i=d.begin();i!=d.end();i++)
- *this << *i;
- }
- template <typename T>
- void saveSerializable(const std::list<T> &data)
- {
- std::list<T> &d = const_cast<std::list<T> &>(data);
- ui32 length = d.size();
- *this << length;
- for(typename std::list<T>::iterator i=d.begin();i!=d.end();i++)
- *this << *i;
- }
- void saveSerializable(const std::string &data)
- {
- *this << ui32(data.length());
- this->This()->write(data.c_str(),data.size());
- }
- template <typename T1, typename T2>
- void saveSerializable(const std::pair<T1,T2> &data)
- {
- *this << data.first << data.second;
- }
- template <typename T1, typename T2>
- void saveSerializable(const std::map<T1,T2> &data)
- {
- *this << ui32(data.size());
- for(typename std::map<T1,T2>::const_iterator i=data.begin();i!=data.end();i++)
- *this << i->first << i->second;
- }
- template <BOOST_VARIANT_ENUM_PARAMS(typename T)>
- void saveSerializable(const boost::variant<BOOST_VARIANT_ENUM_PARAMS(T)> &data)
- {
- si32 which = data.which();
- *this << which;
- VariantVisitorSaver<Serializer> visitor(*this->This());
- boost::apply_visitor(visitor, data);
- }
- template <typename T>
- void saveSerializable(const boost::optional<T> &data)
- {
- if(data)
- {
- *this << (ui8)1;
- *this << *data;
- }
- else
- {
- *this << (ui8)0;
- }
- }
- template <typename E>
- void saveEnum(const E &data)
- {
- si32 writ = static_cast<si32>(data);
- *this << writ;
- }
- void saveBoolean(const bool & data)
- {
- ui8 writ = static_cast<ui8>(data);
- *this << writ;
- }
- void saveBooleanVector(const std::vector<bool> & data)
- {
- std::vector<ui8> convData;
- std::copy(data.begin(), data.end(), std::back_inserter(convData));
- saveSerializable(convData);
- }
- };
- class DLL_LINKAGE CLoaderBase : public virtual CSerializer
- {};
- class CBasicPointerLoader
- {
- public:
- virtual void loadPtr(CLoaderBase &ar, void *data, ui32 pid) const =0; //data is pointer to the ACTUAL POINTER
- virtual ~CBasicPointerLoader(){}
- };
- template <typename Serializer, typename T> class CPointerLoader : public CBasicPointerLoader
- {
- public:
- void loadPtr(CLoaderBase &ar, void *data, ui32 pid) const //data is pointer to the ACTUAL POINTER
- {
- Serializer &s = static_cast<Serializer&>(ar);
- T *&ptr = *static_cast<T**>(data);
- //create new object under pointer
- typedef typename boost::remove_pointer<T>::type npT;
- ptr = new npT;
- s.ptrAllocated(ptr, pid);
- //T is most derived known type, it's time to call actual serialize
- ptr->serialize(s,version);
- }
- };
- /// The class which manages loading of objects.
- template <typename Serializer> class DLL_LINKAGE CISer : public CLoaderBase
- {
- public:
- bool saving;
- std::map<ui16,CBasicPointerLoader*> loaders; // typeID => CPointerSaver<serializer,type>
- ui32 fileVersion;
- bool reverseEndianess; //if source has different endianess than us, we reverse bytes
- std::map<ui32, void*> loadedPointers;
- bool smartPointerSerialization;
- CISer()
- {
- saving = false;
- fileVersion = 0;
- smartPointerSerialization = true;
- reverseEndianess = false;
- }
- ~CISer()
- {
- std::map<ui16,CBasicPointerLoader*>::iterator iter;
- for(iter = loaders.begin(); iter != loaders.end(); iter++)
- delete iter->second;
- }
- template<typename T> void registerType(const T * t=NULL)
- {
- ui16 ID = typeList.registerType(t);
- loaders[ID] = new CPointerLoader<CISer<Serializer>,T>;
- }
- Serializer * This()
- {
- return static_cast<Serializer*>(this);
- }
- template<class T>
- Serializer & operator>>(T &t)
- {
- this->This()->load(t);
- return * this->This();
- }
- template<class T>
- CISer & operator&(T & t)
- {
- return * this->This() >> t;
- }
- int write(const void * data, unsigned size);
- template <typename T>
- void load(T &data)
- {
- typedef
- //if
- typename mpl::eval_if< mpl::equal_to<SerializationLevel<T>,mpl::int_<Boolean> >,
- mpl::identity<LoadBoolean<Serializer> >,
- //else if
- typename mpl::eval_if< mpl::equal_to<SerializationLevel<T>,mpl::int_<BooleanVector> >,
- mpl::identity<LoadBooleanVector<Serializer> >,
- //else if
- typename mpl::eval_if< mpl::equal_to<SerializationLevel<T>,mpl::int_<Primitive> >,
- mpl::identity<LoadPrimitive<Serializer,T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Enum> >,
- mpl::identity<LoadEnum<Serializer,T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Pointer> >,
- mpl::identity<LoadPointer<Serializer,T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Array> >,
- mpl::identity<LoadArray<Serializer,T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Serializable> >,
- mpl::identity<LoadSerializable<Serializer,T> >,
- //else
- mpl::identity<LoadWrong<Serializer,T> >
- >
- >
- >
- >
- >
- >
- >::type typex;
- typex::invoke(* this->This(), data);
- }
- template <typename T>
- void loadPrimitive(T &data)
- {
- if(0) //for testing #989
- {
- this->This()->read(&data,sizeof(data));
- }
- else
- {
- unsigned length = sizeof(data);
- char* dataPtr = (char*)&data;
- this->This()->read(dataPtr,length);
- if(reverseEndianess)
- std::reverse(dataPtr, dataPtr + length);
- }
- }
- template <typename T>
- void loadSerializableBySerializeCall(T &data)
- {
- ////that const cast is evil because it allows to implicitly overwrite const objects when deserializing
- typedef typename boost::remove_const<T>::type nonConstT;
- nonConstT &hlp = const_cast<nonConstT&>(data);
- hlp.serialize(*this,fileVersion);
- //data.serialize(*this,myVersion);
- }
- template <typename T>
- void loadSerializable(T &data)
- {
- loadSerializableBySerializeCall(data);
- }
- template <typename T>
- void loadArray(T &data)
- {
- ui32 size = ARRAY_COUNT(data);
- for(ui32 i = 0; i < size; i++)
- *this >> data[i];
- }
- template <typename T>
- void loadPointer(T &data)
- {
- ui8 hlp;
- *this >> hlp;
- if(!hlp)
- {
- data = NULL;
- return;
- }
- if(smartVectorMembersSerialization)
- {
- typedef typename boost::remove_const<typename boost::remove_pointer<T>::type>::type TObjectType; //eg: const CGHeroInstance * => CGHeroInstance
- typedef typename VectorisedTypeFor<TObjectType>::type VType; //eg: CGHeroInstance -> CGobjectInstance
- typedef typename VectorizedIDType<TObjectType>::type IDType;
- if(const auto *info = getVectorisedTypeInfo<VType, IDType>())
- {
- IDType id;
- *this >> id;
- if(id != IDType(-1))
- {
- data = static_cast<T>(getVectorItemFromId<VType, IDType>(*info, id));
- return;
- }
- }
- }
- if(sendStackInstanceByIds)
- {
- bool gotLoaded = LoadIfStackInstance<Serializer,T>::invoke(*This(), data);
- if(gotLoaded)
- return;
- }
- ui32 pid = 0xffffffff; //pointer id (or maybe rather pointee id)
- if(smartPointerSerialization)
- {
- *this >> pid; //get the id
- std::map<ui32, void*>::iterator i = loadedPointers.find(pid); //lookup
- if(i != loadedPointers.end())
- {
- //we already got this pointer
- data = static_cast<T>(i->second);
- return;
- }
- }
- //get type id
- ui16 tid;
- *this >> tid;
- This()->loadPointerHlp(tid, data, pid);
- }
- //that part of ptr deserialization was extracted to allow customization of its behavior in derived classes
- template <typename T>
- void loadPointerHlp( ui16 tid, T & data, ui32 pid )
- {
- if(!tid)
- {
- typedef typename boost::remove_pointer<T>::type npT;
- typedef typename boost::remove_const<npT>::type ncpT;
- data = new ncpT;
- ptrAllocated(data, pid);
- *this >> *data;
- }
- else
- {
- loaders[tid]->loadPtr(*this,&data, pid);
- }
- }
- template <typename T>
- void ptrAllocated(const T *ptr, ui32 pid)
- {
- if(smartPointerSerialization && pid != 0xffffffff)
- loadedPointers[pid] = (void*)ptr; //add loaded pointer to our lookup map; cast is to avoid errors with const T* pt
- }
- #define READ_CHECK_U32(x) \
- ui32 length; \
- *this >> length; \
- if(length > 500000) \
- { \
- tlog2 << "Warning: very big length: " << length << "\n" ;\
- reportState(tlog2); \
- };
- template <typename T>
- void loadSerializable(shared_ptr<T> &data)
- {
- T *internalPtr;
- *this >> internalPtr;
- data.reset(internalPtr);
- }
- template <typename T>
- void loadSerializable(unique_ptr<T> &data)
- {
- T *internalPtr;
- *this >> internalPtr;
- data.reset(internalPtr);
- }
- template <typename T>
- void loadSerializable(std::vector<T> &data)
- {
- READ_CHECK_U32(length);
- data.resize(length);
- for(ui32 i=0;i<length;i++)
- *this >> data[i];
- }
- template <typename T>
- void loadSerializable(std::set<T> &data)
- {
- READ_CHECK_U32(length);
- T ins;
- for(ui32 i=0;i<length;i++)
- {
- *this >> ins;
- data.insert(ins);
- }
- }
- template <typename T, typename U>
- void loadSerializable(boost::unordered_set<T, U> &data)
- {
- READ_CHECK_U32(length);
- T ins;
- for(ui32 i=0;i<length;i++)
- {
- *this >> ins;
- data.insert(ins);
- }
- }
- template <typename T>
- void loadSerializable(std::list<T> &data)
- {
- READ_CHECK_U32(length);
- T ins;
- for(ui32 i=0;i<length;i++)
- {
- *this >> ins;
- data.push_back(ins);
- }
- }
- template <typename T1, typename T2>
- void loadSerializable(std::pair<T1,T2> &data)
- {
- *this >> data.first >> data.second;
- }
- template <typename T1, typename T2>
- void loadSerializable(std::map<T1,T2> &data)
- {
- READ_CHECK_U32(length);
- T1 t;
- for(ui32 i=0;i<length;i++)
- {
- *this >> t;
- *this >> data[t];
- }
- }
- void loadSerializable(std::string &data)
- {
- READ_CHECK_U32(length);
- data.resize(length);
- this->This()->read((void*)data.c_str(),length);
- }
- template <BOOST_VARIANT_ENUM_PARAMS(typename T)>
- void loadSerializable(boost::variant<BOOST_VARIANT_ENUM_PARAMS(T)> &data)
- {
- si32 which;
- *this >> which;
- if(which == 0)
- {
- T0 obj;
- *this >> obj;
- data = obj;
- }
- else if(which == 1)
- {
- T1 obj;
- *this >> obj;
- data = obj;
- }
- else
- assert(0);
- //TODO write more if needed, general solution would be much longer
- }
- template <typename T>
- void loadSerializable(boost::optional<T> & data)
- {
- ui8 present;
- *this >> present;
- if(present)
- {
- T t;
- *this >> t;
- data = t;
- }
- else
- {
- data = boost::optional<T>();
- }
- }
- void loadSerializable(CStackInstance *&s)
- {
- if(sendStackInstanceByIds)
- {
- CArmedInstance *armed;
- TSlot slot;
- *this >> armed >> slot;
- assert(armed->hasStackAtSlot(slot));
- s = armed->stacks[slot];
- }
- else
- loadSerializableBySerializeCall(s);
- }
- template <typename E>
- void loadEnum(E &data)
- {
- si32 read;
- *this >> read;
- data = static_cast<E>(read);
- }
- void loadBoolean(bool &data)
- {
- ui8 read;
- *this >> read;
- data = static_cast<bool>(read);
- }
- void loadBooleanVector(std::vector<bool> & data)
- {
- std::vector<ui8> convData;
- loadSerializable(convData);
- std::copy(convData.begin(), convData.end(), std::back_inserter(data));
- }
- };
- class DLL_LINKAGE CSaveFile
- : public COSer<CSaveFile>
- {
- void dummyMagicFunction()
- {
- *this << std::string("This function makes stuff working.");
- }
- public:
- std::string fName;
- unique_ptr<std::ofstream> sfile;
- CSaveFile(const std::string &fname); //throws!
- ~CSaveFile();
- int write(const void * data, unsigned size);
- void openNextFile(const std::string &fname); //throws!
- void clear();
- void reportState(CLogger &out);
- };
- class DLL_LINKAGE CLoadFile
- : public CISer<CLoadFile>
- {
- void dummyMagicFunction()
- {
- std::string dummy = "This function makes stuff working.";
- *this >> dummy;
- }
- public:
- std::string fName;
- unique_ptr<std::ifstream> sfile;
- CLoadFile(const std::string &fname, int minimalVersion = version); //throws!
- ~CLoadFile();
- int read(const void * data, unsigned size); //throws!
- void openNextFile(const std::string &fname, int minimalVersion); //throws!
- void clear();
- void reportState(CLogger &out);
- };
- typedef boost::asio::basic_stream_socket < boost::asio::ip::tcp , boost::asio::stream_socket_service<boost::asio::ip::tcp> > TSocket;
- typedef boost::asio::basic_socket_acceptor<boost::asio::ip::tcp, boost::asio::socket_acceptor_service<boost::asio::ip::tcp> > TAcceptor;
- class DLL_LINKAGE CConnection
- :public CISer<CConnection>, public COSer<CConnection>
- {
- //CGameState *gs;
- CConnection(void);
- void init();
- void reportState(CLogger &out);
- public:
- boost::mutex *rmx, *wmx; // read/write mutexes
- TSocket * socket;
- bool logging;
- bool connected;
- bool myEndianess, contactEndianess; //true if little endian, if endianess is different we'll have to revert received multi-byte vars
- boost::asio::io_service *io_service;
- std::string name; //who uses this connection
- int connectionID;
- boost::thread *handler;
- bool receivedStop, sendStop;
- CConnection(std::string host, std::string port, std::string Name);
- CConnection(TAcceptor * acceptor, boost::asio::io_service *Io_service, std::string Name);
- CConnection(TSocket * Socket, std::string Name); //use immediately after accepting connection into socket
- int write(const void * data, unsigned size);
- int read(void * data, unsigned size);
- void close();
- bool isOpen() const;
- template<class T>
- CConnection &operator&(const T&);
- virtual ~CConnection(void);
- CPack *retreivePack(); //gets from server next pack (allocates it with new)
- void sendPackToServer(const CPack &pack, TPlayerColor player, ui32 requestID);
- void disableStackSendingByID();
- void enableStackSendingByID();
- void disableSmartPointerSerialization();
- void enableSmartPointerSerializatoin();
- void prepareForSendingHeroes(); //disables sending vectorised, enables smart pointer serialization, clears saved/loaded ptr cache
- };
- DLL_LINKAGE std::ostream &operator<<(std::ostream &str, const CConnection &cpc);
- template<typename T>
- class CApplier
- {
- public:
- std::map<ui16,T*> apps;
- ~CApplier()
- {
- typename std::map<ui16, T*>::iterator iter;
- for(iter = apps.begin(); iter != apps.end(); iter++)
- delete iter->second;
- }
- template<typename U> void registerType(const U * t=NULL)
- {
- ui16 ID = typeList.registerType(t);
- apps[ID] = T::getApplier(t);
- }
- };
|