123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281 |
- #include "StdInc.h"
- #include "Connection.h"
- #include "../registerTypes/RegisterTypes.h"
- #include "../mapping/CMap.h"
- #include "../CGameState.h"
- #include <boost/asio.hpp>
- /*
- * Connection.cpp, 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
- *
- */
- using namespace boost;
- using namespace boost::asio::ip;
- #if defined(__hppa__) || \
- defined(__m68k__) || defined(mc68000) || defined(_M_M68K) || \
- (defined(__MIPS__) && defined(__MISPEB__)) || \
- defined(__ppc__) || defined(__POWERPC__) || defined(_M_PPC) || \
- defined(__sparc__)
- #define BIG_ENDIAN
- #else
- #define LIL_ENDIAN
- #endif
- void CConnection::init()
- {
- boost::asio::ip::tcp::no_delay option(true);
- socket->set_option(option);
- enableSmartPointerSerializatoin();
- disableStackSendingByID();
- registerTypes(iser);
- registerTypes(oser);
- #ifdef LIL_ENDIAN
- myEndianess = true;
- #else
- myEndianess = false;
- #endif
- connected = true;
- std::string pom;
- //we got connection
- oser & std::string("Aiya!\n") & name & myEndianess; //identify ourselves
- iser & pom & pom & contactEndianess;
- logNetwork->infoStream() << "Established connection with "<<pom;
- wmx = new boost::mutex;
- rmx = new boost::mutex;
- handler = nullptr;
- receivedStop = sendStop = false;
- static int cid = 1;
- connectionID = cid++;
- iser.fileVersion = SERIALIZATION_VERSION;
- }
- CConnection::CConnection(std::string host, std::string port, std::string Name)
- :iser(this), oser(this), io_service(new asio::io_service), name(Name)
- {
- int i;
- boost::system::error_code error = asio::error::host_not_found;
- socket = new tcp::socket(*io_service);
- tcp::resolver resolver(*io_service);
- tcp::resolver::iterator end, pom, endpoint_iterator = resolver.resolve(tcp::resolver::query(host,port),error);
- if(error)
- {
- logNetwork->errorStream() << "Problem with resolving: \n" << error;
- goto connerror1;
- }
- pom = endpoint_iterator;
- if(pom != end)
- logNetwork->infoStream()<<"Found endpoints:";
- else
- {
- logNetwork->errorStream() << "Critical problem: No endpoints found!";
- goto connerror1;
- }
- i=0;
- while(pom != end)
- {
- logNetwork->infoStream() << "\t" << i << ": " << (boost::asio::ip::tcp::endpoint&)*pom;
- pom++;
- }
- i=0;
- while(endpoint_iterator != end)
- {
- logNetwork->infoStream() << "Trying connection to " << (boost::asio::ip::tcp::endpoint&)*endpoint_iterator << " (" << i++ << ")";
- socket->connect(*endpoint_iterator, error);
- if(!error)
- {
- init();
- return;
- }
- else
- {
- logNetwork->errorStream() << "Problem with connecting: " << error;
- }
- endpoint_iterator++;
- }
- //we shouldn't be here - error handling
- connerror1:
- logNetwork->errorStream() << "Something went wrong... checking for error info";
- if(error)
- logNetwork->errorStream() << error;
- else
- logNetwork->errorStream() << "No error info. ";
- delete io_service;
- //delete socket;
- throw std::runtime_error("Can't establish connection :(");
- }
- CConnection::CConnection(TSocket * Socket, std::string Name )
- :iser(this), oser(this), socket(Socket),io_service(&Socket->get_io_service()), name(Name)//, send(this), rec(this)
- {
- init();
- }
- CConnection::CConnection(TAcceptor * acceptor, boost::asio::io_service *Io_service, std::string Name)
- : iser(this), oser(this), name(Name)//, send(this), rec(this)
- {
- boost::system::error_code error = asio::error::host_not_found;
- socket = new tcp::socket(*io_service);
- acceptor->accept(*socket,error);
- if (error)
- {
- logNetwork->errorStream() << "Error on accepting: " << error;
- delete socket;
- throw std::runtime_error("Can't establish connection :(");
- }
- init();
- }
- int CConnection::write(const void * data, unsigned size)
- {
- try
- {
- int ret;
- ret = asio::write(*socket,asio::const_buffers_1(asio::const_buffer(data,size)));
- return ret;
- }
- catch(...)
- {
- //connection has been lost
- connected = false;
- throw;
- }
- }
- int CConnection::read(void * data, unsigned size)
- {
- try
- {
- int ret = asio::read(*socket,asio::mutable_buffers_1(asio::mutable_buffer(data,size)));
- return ret;
- }
- catch(...)
- {
- //connection has been lost
- connected = false;
- throw;
- }
- }
- CConnection::~CConnection(void)
- {
- if(handler)
- handler->join();
- delete handler;
- close();
- delete io_service;
- delete wmx;
- delete rmx;
- }
- template<class T>
- CConnection & CConnection::operator&(const T &t) {
- // throw std::exception();
- //XXX this is temporaly ? solution to fix gcc (4.3.3, other?) compilation
- // problem for more details contact [email protected] or [email protected]
- // do not remove this exception it shoudnt be called
- return *this;
- }
- void CConnection::close()
- {
- if(socket)
- {
- socket->close();
- delete socket;
- socket = nullptr;
- }
- }
- bool CConnection::isOpen() const
- {
- return socket && connected;
- }
- void CConnection::reportState(CLogger * out)
- {
- out->debugStream() << "CConnection";
- if(socket && socket->is_open())
- {
- out->debugStream() << "\tWe have an open and valid socket";
- out->debugStream() << "\t" << socket->available() <<" bytes awaiting";
- }
- }
- CPack * CConnection::retreivePack()
- {
- CPack *ret = nullptr;
- boost::unique_lock<boost::mutex> lock(*rmx);
- logNetwork->traceStream() << "Listening... ";
- iser & ret;
- logNetwork->traceStream() << "\treceived server message of type " << (ret? typeid(*ret).name() : "nullptr") << ", data: " << ret;
- return ret;
- }
- void CConnection::sendPackToServer(const CPack &pack, PlayerColor player, ui32 requestID)
- {
- boost::unique_lock<boost::mutex> lock(*wmx);
- logNetwork->traceStream() << "Sending to server a pack of type " << typeid(pack).name();
- oser & player & requestID & &pack; //packs has to be sent as polymorphic pointers!
- }
- void CConnection::disableStackSendingByID()
- {
- CSerializer::sendStackInstanceByIds = false;
- }
- void CConnection::enableStackSendingByID()
- {
- CSerializer::sendStackInstanceByIds = true;
- }
- void CConnection::disableSmartPointerSerialization()
- {
- iser.smartPointerSerialization = oser.smartPointerSerialization = false;
- }
- void CConnection::enableSmartPointerSerializatoin()
- {
- iser.smartPointerSerialization = oser.smartPointerSerialization = true;
- }
- void CConnection::prepareForSendingHeroes()
- {
- iser.loadedPointers.clear();
- oser.savedPointers.clear();
- disableSmartVectorMemberSerialization();
- enableSmartPointerSerializatoin();
- disableStackSendingByID();
- }
- void CConnection::enterPregameConnectionMode()
- {
- iser.loadedPointers.clear();
- oser.savedPointers.clear();
- disableSmartVectorMemberSerialization();
- disableSmartPointerSerialization();
- }
- void CConnection::disableSmartVectorMemberSerialization()
- {
- CSerializer::smartVectorMembersSerialization = false;
- }
- void CConnection::enableSmartVectorMemberSerializatoin()
- {
- CSerializer::smartVectorMembersSerialization = true;
- }
- std::ostream & operator<<(std::ostream &str, const CConnection &cpc)
- {
- return str << "Connection with " << cpc.name << " (ID: " << cpc.connectionID << /*", " << (cpc.host ? "host" : "guest") <<*/ ")";
- }
|