123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740 |
- /*
- * 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 <type_traits>
- #include <boost/mpl/eval_if.hpp>
- #include <boost/mpl/equal_to.hpp>
- #include <boost/mpl/int.hpp>
- #include <boost/mpl/identity.hpp>
- #include <boost/mpl/for_each.hpp>
- #include <boost/any.hpp>
- #include "ConstTransitivePtr.h"
- #include "CCreatureSet.h" //for CStackInstance
- #include "mapObjects/CGHeroInstance.h"
- #include "mapping/CCampaignHandler.h" //for CCampaignState
- #include "rmg/CMapGenerator.h" // for CMapGenOptions
- const ui32 version = 755;
- const ui32 minSupportedVersion = 753;
- class CISer;
- class COSer;
- 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;
- const std::string SAVEGAME_MAGIC = "VCMISVG";
- 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);
- }
- };
- struct IPointerCaster
- {
- virtual boost::any castRawPtr(const boost::any &ptr) const = 0; // takes From*, performs dynamic cast, returns To*
- virtual boost::any castSharedPtr(const boost::any &ptr) const = 0; // takes std::shared_ptr<From>, performs dynamic cast, returns std::shared_ptr<To>
- virtual boost::any castWeakPtr(const boost::any &ptr) const = 0; // takes std::weak_ptr<From>, performs dynamic cast, returns std::weak_ptr<To>. The object under poitner must live.
- //virtual boost::any castUniquePtr(const boost::any &ptr) const = 0; // takes std::unique_ptr<From>, performs dynamic cast, returns std::unique_ptr<To>
- };
- template <typename From, typename To>
- struct PointerCaster : IPointerCaster
- {
- virtual boost::any castRawPtr(const boost::any &ptr) const override // takes void* pointing to From object, performs dynamic cast, returns void* pointing to To object
- {
- From * from = (From*)boost::any_cast<void*>(ptr);
- To * ret = dynamic_cast<To*>(from);
- return (void*)ret;
- }
- // Helper function performing casts between smart pointers using dynamic_pointer_cast
- template<typename SmartPt>
- boost::any castSmartPtr(const boost::any &ptr) const
- {
- try
- {
- auto from = boost::any_cast<SmartPt>(ptr);
- auto ret = std::dynamic_pointer_cast<To>(from);
- return ret;
- }
- catch(std::exception &e)
- {
- THROW_FORMAT("Failed cast %s -> %s. Given argument was %s. Error message: %s", typeid(From).name() % typeid(To).name() % ptr.type().name() % e.what());
- }
- }
- virtual boost::any castSharedPtr(const boost::any &ptr) const override
- {
- return castSmartPtr<std::shared_ptr<From>>(ptr);
- }
- virtual boost::any castWeakPtr(const boost::any &ptr) const override
- {
- auto from = boost::any_cast<std::weak_ptr<From>>(ptr);
- return castSmartPtr<std::shared_ptr<From>>(from.lock());
- }
- // virtual boost::any castUniquePtr(const boost::any &ptr) const override
- // {
- // return castSmartPtr<std::unique_ptr<From>>(ptr);
- // }
- };
- class DLL_LINKAGE CTypeList
- {
- public:
- struct TypeDescriptor;
- typedef std::shared_ptr<TypeDescriptor> TypeInfoPtr;
- struct TypeDescriptor
- {
- ui16 typeID;
- const char *name;
- std::vector<TypeInfoPtr> children, parents;
- };
- private:
- std::map<const std::type_info *, TypeInfoPtr, TypeComparer> typeInfos;
- std::map<std::pair<TypeInfoPtr, TypeInfoPtr>, std::unique_ptr<const IPointerCaster>> casters; //for each pair <Base, Der> we provide a caster (each registered relations creates a single entry here)
- CTypeList(CTypeList &)
- {
- // This type is non-copyable.
- // Unfortunately on Windows it is required for DLL_EXPORT-ed type to provide copy c-tor, so we can't =delete it.
- assert(0);
- }
- CTypeList &operator=(CTypeList &)
- {
- // As above.
- assert(0);
- return *this;
- }
- public:
- CTypeList();
- TypeInfoPtr registerType(const std::type_info *type);
- template <typename Base, typename Derived>
- void registerType(const Base * b = nullptr, const Derived * d = nullptr)
- {
- static_assert(std::is_base_of<Base, Derived>::value, "First registerType template parameter needs to ba a base class of the second one.");
- static_assert(std::has_virtual_destructor<Base>::value, "Base class needs to have a virtual destructor.");
- static_assert(!std::is_same<Base, Derived>::value, "Parameters of registerTypes should be two diffrenet types.");
- auto bt = getTypeInfo(b), dt = getTypeInfo(d); //obtain std::type_info
- auto bti = registerType(bt), dti = registerType(dt); //obtain our TypeDescriptor
- // register the relation between classes
- bti->children.push_back(dti);
- dti->parents.push_back(bti);
- casters[std::make_pair(bti, dti)] = make_unique<const PointerCaster<Base, Derived>>();
- casters[std::make_pair(dti, bti)] = make_unique<const PointerCaster<Derived, Base>>();
- }
- ui16 getTypeID(const std::type_info *type);
- TypeInfoPtr getTypeDescriptor(const std::type_info *type, bool throws = true); //if not throws, failure returns nullptr
- template <typename T>
- ui16 getTypeID(const T * t = nullptr)
- {
- return getTypeID(getTypeInfo(t));
- }
- // Returns sequence of types starting from "from" and ending on "to". Every next type is derived from the previous.
- // Throws if there is no link registered.
- std::vector<TypeInfoPtr> castSequence(TypeInfoPtr from, TypeInfoPtr to);
- std::vector<TypeInfoPtr> castSequence(const std::type_info *from, const std::type_info *to);
- template<boost::any(IPointerCaster::*CastingFunction)(const boost::any &) const>
- boost::any castHelper(boost::any inputPtr, const std::type_info *fromArg, const std::type_info *toArg)
- {
- auto typesSequence = castSequence(fromArg, toArg);
- boost::any ptr = inputPtr;
- for(int i = 0; i < (int)typesSequence.size() - 1; i++)
- {
- auto &from = typesSequence[i];
- auto &to = typesSequence[i + 1];
- auto castingPair = std::make_pair(from, to);
- if(!casters.count(castingPair))
- THROW_FORMAT("Cannot find caster for conversion %s -> %s which is needed to cast %s -> %s", from->name % to->name % fromArg->name() % toArg->name());
- auto &caster = casters.at(castingPair);
- ptr = (*caster.*CastingFunction)(ptr); //Why does unique_ptr does not have operator->* ..?
- }
- return ptr;
- }
- template<typename TInput>
- void *castToMostDerived(const TInput *inputPtr)
- {
- auto &baseType = typeid(typename std::remove_cv<TInput>::type);
- auto derivedType = getTypeInfo(inputPtr);
- if(baseType == *derivedType)
- return (void*)inputPtr;
- return boost::any_cast<void*>(castHelper<&IPointerCaster::castRawPtr>((void*)inputPtr, &baseType, derivedType));
- }
- template<typename TInput>
- boost::any castSharedToMostDerived(const std::shared_ptr<TInput> inputPtr)
- {
- auto &baseType = typeid(typename std::remove_cv<TInput>::type);
- auto derivedType = getTypeInfo(inputPtr.get());
- if(baseType == *derivedType)
- return inputPtr;
- return castHelper<&IPointerCaster::castSharedPtr>(inputPtr, &baseType, derivedType);
- }
- void* castRaw(void *inputPtr, const std::type_info *from, const std::type_info *to)
- {
- return boost::any_cast<void*>(castHelper<&IPointerCaster::castRawPtr>(inputPtr, from, to));
- }
- boost::any castShared(boost::any inputPtr, const std::type_info *from, const std::type_info *to)
- {
- return castHelper<&IPointerCaster::castSharedPtr>(inputPtr, from, to);
- }
- template <typename T> const std::type_info * getTypeInfo(const T * t = nullptr)
- {
- if(t)
- return &typeid(*t);
- else
- return &typeid(T);
- }
- };
- extern DLL_LINKAGE CTypeList typeList;
- template<typename Variant, typename Source>
- struct VariantLoaderHelper
- {
- Source & source;
- std::vector<std::function<Variant()>> funcs;
- VariantLoaderHelper(Source & source):
- source(source)
- {
- mpl::for_each<typename Variant::types>(std::ref(*this));
- }
- template<typename Type>
- void operator()(Type)
- {
- funcs.push_back([&]() -> Variant
- {
- Type obj;
- source >> obj;
- return Variant(obj);
- });
- }
- };
- 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 ObjType, typename IdType>
- struct VectorisedObjectInfo
- {
- const std::vector<ConstTransitivePtr<ObjType> > *vector; //pointer to the appropriate vector
- std::function<IdType(const ObjType &)> idRetriever;
- //const IdType ObjType::*idPtr; //pointer to the field representing the position in the vector
- VectorisedObjectInfo(const std::vector< ConstTransitivePtr<ObjType> > *Vector, std::function<IdType(const ObjType &)> IdGetter)
- :vector(Vector), idRetriever(IdGetter)
- {
- }
- };
- 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, typename NT>
- NT idToNumber(const BaseForID<T, NT> &t)
- {
- return t.getNum();
- }
- /// 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 std::function<U(const T&)> &idRetriever)
- {
- vectors[&typeid(T)] = VectorisedObjectInfo<T, U>(Vector, idRetriever);
- }
- template <typename T, typename U>
- void registerVectoredType(const std::vector<ConstTransitivePtr<T> > *Vector, const std::function<U(const T&)> &idRetriever)
- {
- vectors[&typeid(T)] = VectorisedObjectInfo<T, U>(Vector, idRetriever);
- }
- template <typename T, typename U>
- const VectorisedObjectInfo<T, U> *getVectorisedTypeInfo()
- {
- const std::type_info *myType = nullptr;
- //
- // 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 nullptr;
- 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 nullptr;*/
- si32 idAsNumber = idToNumber(id);
- assert(oInfo.vector);
- assert(static_cast<si32>(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 oInfo.idRetriever(*obj);
- }
- void addStdVecItems(CGameState *gs, LibClasses *lib = VLC);
- };
- class IBinaryWriter : public virtual CSerializer
- {
- public:
- virtual int write(const void * data, unsigned size) = 0;
- };
- class DLL_LINKAGE CSaverBase
- {
- protected:
- IBinaryWriter * writer;
- public:
- CSaverBase(IBinaryWriter * w): writer(w){};
-
- inline int write(const void * data, unsigned size)
- {
- return writer->write(data, size);
- };
- };
- class CBasicPointerSaver
- {
- public:
- virtual void savePtr(CSaverBase &ar, const void *data) const =0;
- virtual ~CBasicPointerSaver(){}
- };
- template <typename T> //metafunction returning CGObjectInstance if T is its derivate or T elsewise
- struct VectorisedTypeFor
- {
- typedef typename
- //if
- mpl::eval_if<boost::is_same<CGHeroInstance,T>,
- mpl::identity<CGHeroInstance>,
- //else 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<CHero,U>,
- mpl::identity<HeroTypeID>,
- //else if
- mpl::eval_if<boost::is_same<CArtifactInstance,U>,
- mpl::identity<ArtifactInstanceID>,
- //else if
- mpl::eval_if<boost::is_same<CGHeroInstance,U>,
- mpl::identity<HeroTypeID>,
- //else if
- mpl::eval_if<boost::is_base_of<CGObjectInstance,U>,
- mpl::identity<ObjectInstanceID>,
- //else
- mpl::identity<si32>
- > > > > > >::type type;
- };
- 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);
- SlotID slot;
- if(data->getNodeType() == CBonusSystemNode::COMMANDER)
- slot = SlotID::COMMANDER_SLOT_PLACEHOLDER;
- else
- slot = data->armyObj->findStack(data);
- assert(slot != SlotID());
- 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;
- SlotID slot;
- s >> armedObj >> slot;
- if(slot != SlotID::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.
- class DLL_LINKAGE COSer : public CSaverBase
- {
- public:
-
- struct SaveBoolean
- {
- static void invoke(COSer &s, const bool &data)
- {
- s.saveBoolean(data);
- }
- };
-
- struct SaveBooleanVector
- {
- static void invoke(COSer &s, const std::vector<bool> &data)
- {
- s.saveBooleanVector(data);
- }
- };
- template<typename T>
- struct SavePrimitive
- {
- static void invoke(COSer &s, const T &data)
- {
- s.savePrimitive(data);
- }
- };
- template<typename T>
- struct SaveSerializable
- {
- static void invoke(COSer &s, const T &data)
- {
- s.saveSerializable(data);
- }
- };
- template<typename T>
- struct SaveEnum
- {
- static void invoke(COSer &s, const T &data)
- {
- s.saveEnum(data);
- }
- };
-
- template<typename T>
- struct SavePointer
- {
- static void invoke(COSer &s, const T &data)
- {
- s.savePointer(data);
- }
- };
-
- template<typename T>
- struct SaveArray
- {
- static void invoke(COSer &s, const T &data)
- {
- s.saveArray(data);
- }
- };
- template<typename T>
- struct SaveWrong
- {
- static void invoke(COSer &s, const T &data)
- {
- throw std::runtime_error("Wrong save serialization call!");
- }
- };
-
- template <typename T>
- class CPointerSaver : public CBasicPointerSaver
- {
- public:
- void savePtr(CSaverBase &ar, const void *data) const override
- {
- COSer &s = static_cast<COSer&>(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);
- }
- };
-
- bool saving;
- std::map<ui16,CBasicPointerSaver*> savers; // typeID => CPointerSaver<serializer,type>
- std::map<const void*, ui32> savedPointers;
- bool smartPointerSerialization;
- COSer(IBinaryWriter * w): CSaverBase(w)
- {
- 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 addSaver(const T * t = nullptr)
- {
- auto ID = typeList.getTypeID(t);
- if(!savers.count(ID))
- savers[ID] = new CPointerSaver<T>;
- }
- template<typename Base, typename Derived> void registerType(const Base * b = nullptr, const Derived * d = nullptr)
- {
- typeList.registerType(b, d);
- addSaver(b);
- addSaver(d);
- }
- template<class T>
- COSer & operator<<(const T &t)
- {
- this->save(t);
- return * this;
- }
- template<class T>
- COSer & operator&(const T & t)
- {
- return * this << t;
- }
- template <typename T>
- void savePrimitive(const T &data)
- {
- this->write(&data,sizeof(data));
- }
- template <typename T>
- void savePointer(const T &data)
- {
- //write if pointer is not nullptr
- ui8 hlp = (data!=nullptr);
- *this << hlp;
- //if pointer is nullptr then we don't need anything more...
- if(!hlp)
- return;
- if(writer->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 = writer->getVectorisedTypeInfo<VType, IDType>())
- {
- IDType id = writer->getIdFromVectorItem<VType>(*info, data);
- *this << id;
- if(id != IDType(-1)) //vector id is enough
- return;
- }
- }
- if(writer->sendStackInstanceByIds)
- {
- const bool gotSaved = SaveIfStackInstance<COSer,T>::invoke(*this, data);
- if(gotSaved)
- return;
- }
- if(smartPointerSerialization)
- {
- // We might have an object that has multiple inheritance and store it via the non-first base pointer.
- // Therefore, all pointers need to be normalized to the actual object address.
- auto actualPointer = typeList.castToMostDerived(data);
- std::map<const void*,ui32>::iterator i = savedPointers.find(actualPointer);
- 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[actualPointer] = 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, typeList.castToMostDerived(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>,
- //else if
- typename mpl::eval_if< mpl::equal_to<SerializationLevel<T>,mpl::int_<BooleanVector> >,
- mpl::identity<SaveBooleanVector>,
- //else if
- typename mpl::eval_if< mpl::equal_to<SerializationLevel<T>,mpl::int_<Primitive> >,
- mpl::identity<SavePrimitive<T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Enum> >,
- mpl::identity<SaveEnum<T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Pointer> >,
- mpl::identity<SavePointer<T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Array> >,
- mpl::identity<SaveArray<T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Serializable> >,
- mpl::identity<SaveSerializable<T> >,
- //else
- mpl::identity<SaveWrong<T> >
- >
- >
- >
- >
- >
- >
- >::type typex;
- typex::invoke(* 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, size_t N>
- void saveSerializable(const std::array<T, N> &data)
- {
- for(ui32 i=0; i < N; 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 std::unordered_set<T, U> &data)
- {
- std::unordered_set<T, U> &d = const_cast<std::unordered_set<T, U> &>(data);
- ui32 length = d.size();
- *this << length;
- for(typename std::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->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 <typename T1, typename T2>
- void saveSerializable(const std::multimap<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<COSer> visitor(*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 IBinaryReader : public virtual CSerializer
- {
- public:
- virtual int read(void * data, unsigned size) = 0;
- };
- class DLL_LINKAGE CLoaderBase
- {
- protected:
- IBinaryReader * reader;
- public:
- CLoaderBase(IBinaryReader * r): reader(r){};
-
- inline int read(void * data, unsigned size)
- {
- return reader->read(data, size);
- };
- };
- class CBasicPointerLoader
- {
- public:
- virtual const std::type_info * loadPtr(CLoaderBase &ar, void *data, ui32 pid) const =0; //data is pointer to the ACTUAL POINTER
- virtual ~CBasicPointerLoader(){}
- };
- template <typename T, typename Enable = void>
- struct ClassObjectCreator
- {
- static T *invoke()
- {
- static_assert(!std::is_abstract<T>::value, "Cannot call new upon abstract classes!");
- return new T();
- }
- };
- template<typename T>
- struct ClassObjectCreator<T, typename std::enable_if<std::is_abstract<T>::value>::type>
- {
- static T *invoke()
- {
- throw std::runtime_error("Something went really wrong during deserialization. Attempted creating an object of an abstract class " + std::string(typeid(T).name()));
- }
- };
- /// The class which manages loading of objects.
- class DLL_LINKAGE CISer : public CLoaderBase
- {
- public:
- struct LoadBoolean
- {
- static void invoke(CISer &s, bool &data)
- {
- s.loadBoolean(data);
- }
- };
-
- struct LoadBooleanVector
- {
- static void invoke(CISer &s, std::vector<bool> &data)
- {
- s.loadBooleanVector(data);
- }
- };
-
- template<typename T>
- struct LoadEnum
- {
- static void invoke(CISer &s, T &data)
- {
- s.loadEnum(data);
- }
- };
- template<typename T>
- struct LoadPrimitive
- {
- static void invoke(CISer &s, T &data)
- {
- s.loadPrimitive(data);
- }
- };
- template<typename T>
- struct LoadPointer
- {
- static void invoke(CISer &s, T &data)
- {
- s.loadPointer(data);
- }
- };
-
- template<typename T>
- struct LoadArray
- {
- static void invoke(CISer &s, T &data)
- {
- s.loadArray(data);
- }
- };
- template<typename T>
- struct LoadSerializable
- {
- static void invoke(CISer &s, T &data)
- {
- s.loadSerializable(data);
- }
- };
- template<typename T>
- struct LoadWrong
- {
- static void invoke(CISer &s, const T &data)
- {
- throw std::runtime_error("Wrong load serialization call!");
- }
- };
-
- template <typename T> class CPointerLoader : public CBasicPointerLoader
- {
- public:
- const std::type_info * loadPtr(CLoaderBase &ar, void *data, ui32 pid) const override //data is pointer to the ACTUAL POINTER
- {
- CISer &s = static_cast<CISer&>(ar);
- T *&ptr = *static_cast<T**>(data);
- //create new object under pointer
- typedef typename boost::remove_pointer<T>::type npT;
- ptr = ClassObjectCreator<npT>::invoke(); //does new npT or throws for abstract classes
- s.ptrAllocated(ptr, pid);
- //T is most derived known type, it's time to call actual serialize
- ptr->serialize(s,version);
- return &typeid(T);
- }
- };
-
- bool saving;
- std::map<ui16,CBasicPointerLoader*> loaders; // typeID => CPointerSaver<serializer,type>
- si32 fileVersion;
- bool reverseEndianess; //if source has different endianness than us, we reverse bytes
- std::map<ui32, void*> loadedPointers;
- std::map<ui32, const std::type_info*> loadedPointersTypes;
- std::map<const void*, boost::any> loadedSharedPointers;
- bool smartPointerSerialization;
- CISer(IBinaryReader * r): CLoaderBase(r)
- {
- 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 addLoader(const T * t = nullptr)
- {
- auto ID = typeList.getTypeID(t);
- if(!loaders.count(ID))
- loaders[ID] = new CPointerLoader<T>;
- }
- template<typename Base, typename Derived> void registerType(const Base * b = nullptr, const Derived * d = nullptr)
- {
- typeList.registerType(b, d);
- addLoader(b);
- addLoader(d);
- }
- template<class T>
- CISer & operator>>(T &t)
- {
- this->load(t);
- return * this;
- }
- template<class T>
- CISer & operator&(T & t)
- {
- return * 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>,
- //else if
- typename mpl::eval_if< mpl::equal_to<SerializationLevel<T>,mpl::int_<BooleanVector> >,
- mpl::identity<LoadBooleanVector>,
- //else if
- typename mpl::eval_if< mpl::equal_to<SerializationLevel<T>,mpl::int_<Primitive> >,
- mpl::identity<LoadPrimitive<T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Enum> >,
- mpl::identity<LoadEnum<T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Pointer> >,
- mpl::identity<LoadPointer<T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Array> >,
- mpl::identity<LoadArray<T> >,
- //else if
- typename mpl::eval_if<mpl::equal_to<SerializationLevel<T>,mpl::int_<Serializable> >,
- mpl::identity<LoadSerializable<T> >,
- //else
- mpl::identity<LoadWrong<T> >
- >
- >
- >
- >
- >
- >
- >::type typex;
- typex::invoke(* this, data);
- }
- template <typename T>
- void loadPrimitive(T &data)
- {
- if(0) //for testing #989
- {
- this->read(&data,sizeof(data));
- }
- else
- {
- unsigned length = sizeof(data);
- char* dataPtr = (char*)&data;
- 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 = nullptr;
- return;
- }
- if(reader->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 = reader->getVectorisedTypeInfo<VType, IDType>())
- {
- IDType id;
- *this >> id;
- if(id != IDType(-1))
- {
- data = static_cast<T>(reader->getVectorItemFromId<VType, IDType>(*info, id));
- return;
- }
- }
- }
- if(reader->sendStackInstanceByIds)
- {
- bool gotLoaded = LoadIfStackInstance<CISer,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
- // Cast it in case we are loading it to a non-first base pointer
- assert(loadedPointersTypes.count(pid));
- data = reinterpret_cast<T>(typeList.castRaw(i->second, loadedPointersTypes.at(pid), &typeid(typename boost::remove_const<typename boost::remove_pointer<T>::type>::type)));
- 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 = ClassObjectCreator<ncpT>::invoke();
- ptrAllocated(data, pid);
- *this >> *data;
- }
- else
- {
- auto typeInfo = loaders[tid]->loadPtr(*this,&data, pid);
- data = reinterpret_cast<T>(typeList.castRaw((void*)data, typeInfo, &typeid(typename boost::remove_const<typename boost::remove_pointer<T>::type>::type)));
- }
- }
- template <typename T>
- void ptrAllocated(const T *ptr, ui32 pid)
- {
- if(smartPointerSerialization && pid != 0xffffffff)
- {
- loadedPointersTypes[pid] = &typeid(T);
- 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) \
- { \
- logGlobal->warnStream() << "Warning: very big length: " << length;\
- reader->reportState(logGlobal); \
- };
- template <typename T>
- void loadSerializable(shared_ptr<T> &data)
- {
- typedef typename boost::remove_const<T>::type NonConstT;
- NonConstT *internalPtr;
- *this >> internalPtr;
- void *internalPtrDerived = typeList.castToMostDerived(internalPtr);
- if(internalPtr)
- {
- auto itr = loadedSharedPointers.find(internalPtrDerived);
- if(itr != loadedSharedPointers.end())
- {
- // This pointers is already loaded. The "data" needs to be pointed to it,
- // so their shared state is actually shared.
- try
- {
- auto actualType = typeList.getTypeInfo(internalPtr);
- auto typeWeNeedToReturn = typeList.getTypeInfo<T>();
- if(*actualType == *typeWeNeedToReturn)
- {
- // No casting needed, just unpack already stored shared_ptr and return it
- data = boost::any_cast<std::shared_ptr<T>>(itr->second);
- }
- else
- {
- // We need to perform series of casts
- auto ret = typeList.castShared(itr->second, actualType, typeWeNeedToReturn);
- data = boost::any_cast<std::shared_ptr<T>>(ret);
- }
- }
- catch(std::exception &e)
- {
- logGlobal->errorStream() << e.what();
- logGlobal->errorStream() << boost::format("Failed to cast stored shared ptr. Real type: %s. Needed type %s. FIXME FIXME FIXME")
- % itr->second.type().name() % typeid(std::shared_ptr<T>).name();
- //TODO scenario with inheritance -> we can have stored ptr to base and load ptr to derived (or vice versa)
- assert(0);
- }
- }
- else
- {
- auto hlp = std::shared_ptr<NonConstT>(internalPtr);
- data = hlp; //possibly adds const
- loadedSharedPointers[internalPtrDerived] = typeList.castSharedToMostDerived(hlp);
- }
- }
- else
- data.reset();
- }
- 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, size_t N>
- void loadSerializable(std::array<T, N> &data)
- {
- for(ui32 i = 0; i < N; i++)
- *this >> data[i];
- }
- template <typename T>
- void loadSerializable(std::set<T> &data)
- {
- READ_CHECK_U32(length);
- data.clear();
- T ins;
- for(ui32 i=0;i<length;i++)
- {
- *this >> ins;
- data.insert(ins);
- }
- }
- template <typename T, typename U>
- void loadSerializable(std::unordered_set<T, U> &data)
- {
- READ_CHECK_U32(length);
- data.clear();
- 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);
- data.clear();
- 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);
- data.clear();
- T1 key;
- T2 value;
- for(ui32 i=0;i<length;i++)
- {
- *this >> key >> value;
- data.insert(std::pair<T1, T2>(std::move(key), std::move(value)));
- }
- }
- template <typename T1, typename T2>
- void loadSerializable(std::multimap<T1, T2> &data)
- {
- READ_CHECK_U32(length);
- data.clear();
- T1 key;
- T2 value;
- for(ui32 i = 0; i < length; i++)
- {
- *this >> key >> value;
- data.insert(std::pair<T1, T2>(std::move(key), std::move(value)));
- }
- }
- void loadSerializable(std::string &data)
- {
- READ_CHECK_U32(length);
- data.resize(length);
- this->read((void*)data.c_str(),length);
- }
- template <BOOST_VARIANT_ENUM_PARAMS(typename T)>
- void loadSerializable(boost::variant<BOOST_VARIANT_ENUM_PARAMS(T)> &data)
- {
- typedef boost::variant<BOOST_VARIANT_ENUM_PARAMS(T)> TVariant;
- VariantLoaderHelper<TVariant, CISer> loader(*this);
- si32 which;
- *this >> which;
- assert(which < loader.funcs.size());
- data = loader.funcs.at(which)();
- }
- 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;
- // SlotID 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);
- convData.resize(data.size());
- range::copy(convData, data.begin());
- }
- };
- class DLL_LINKAGE CSaveFile
- :public IBinaryWriter
- {
- public:
-
- COSer serializer;
-
- std::string fName;
- unique_ptr<std::ofstream> sfile;
- CSaveFile(const std::string &fname); //throws!
- ~CSaveFile();
- int write(const void * data, unsigned size) override;
- void openNextFile(const std::string &fname); //throws!
- void clear();
- void reportState(CLogger * out) override;
- void putMagicBytes(const std::string &text);
-
- template<class T>
- CSaveFile & operator<<(const T &t)
- {
- serializer << t;
- return * this;
- }
- };
- class DLL_LINKAGE CLoadFile
- : public IBinaryReader
- {
- public:
- CISer serializer;
-
- std::string fName;
- unique_ptr<boost::filesystem::ifstream> sfile;
- CLoadFile(const boost::filesystem::path & fname, int minimalVersion = version); //throws!
- ~CLoadFile();
- int read(void * data, unsigned size) override; //throws!
- void openNextFile(const boost::filesystem::path & fname, int minimalVersion); //throws!
- void clear();
- void reportState(CLogger * out) override;
- void checkMagicBytes(const std::string & text);
-
- template<class T>
- CLoadFile & operator>>(T &t)
- {
- serializer >> t;
- return * this;
- }
- };
- class DLL_LINKAGE CLoadIntegrityValidator
- : public IBinaryReader
- {
- public:
- CISer serializer;
- unique_ptr<CLoadFile> primaryFile, controlFile;
- bool foundDesync;
- CLoadIntegrityValidator(const std::string &primaryFileName, const std::string &controlFileName, int minimalVersion = version); //throws!
- int read( void * data, unsigned size) override; //throws!
- void checkMagicBytes(const std::string &text);
- unique_ptr<CLoadFile> decay(); //returns primary file. CLoadIntegrityValidator stops being usable anymore
- };
- 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 IBinaryReader, public IBinaryWriter
- {
- //CGameState *gs;
- CConnection(void);
- void init();
- void reportState(CLogger * out) override;
- public:
- CISer iser;
- COSer oser;
-
- boost::mutex *rmx, *wmx; // read/write mutexes
- TSocket * socket;
- bool logging;
- bool connected;
- bool myEndianess, contactEndianess; //true if little endian, if endianness 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) override;
- int read(void * data, unsigned size) override;
- 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, PlayerColor player, ui32 requestID);
- void disableStackSendingByID();
- void enableStackSendingByID();
- void disableSmartPointerSerialization();
- void enableSmartPointerSerializatoin();
- void disableSmartVectorMemberSerialization();
- void enableSmartVectorMemberSerializatoin();
- void prepareForSendingHeroes(); //disables sending vectorised, enables smart pointer serialization, clears saved/loaded ptr cache
- void enterPregameConnectionMode();
-
- template<class T>
- CConnection & operator>>(T &t)
- {
- iser >> t;
- return * this;
- }
-
- template<class T>
- CConnection & operator<<(const T &t)
- {
- oser << t;
- return * this;
- }
- };
- DLL_LINKAGE std::ostream &operator<<(std::ostream &str, const CConnection &cpc);
- // Serializer that stores objects in the dynamic buffer. Allows performing deep object copies.
- class DLL_LINKAGE CMemorySerializer
- : public IBinaryReader, public IBinaryWriter
- {
- std::vector<ui8> buffer;
- size_t readPos; //index of the next byte to be read
- public:
- CISer iser;
- COSer oser;
-
- int read(void * data, unsigned size) override; //throws!
- int write(const void * data, unsigned size) override;
- CMemorySerializer();
- template <typename T>
- static unique_ptr<T> deepCopy(const T &data)
- {
- CMemorySerializer mem;
- mem.oser << &data;
- unique_ptr<T> ret;
- mem.iser >> ret;
- return ret;
- }
- };
- 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 RegisteredType>
- void addApplier(ui16 ID)
- {
- if(!apps.count(ID))
- {
- RegisteredType * rtype = nullptr;
- apps[ID] = T::getApplier(rtype);
- }
- }
- template<typename Base, typename Derived>
- void registerType(const Base * b = nullptr, const Derived * d = nullptr)
- {
- typeList.registerType(b, d);
- addApplier<Base>(typeList.getTypeID(b));
- addApplier<Derived>(typeList.getTypeID(d));
- }
- };
|