Connection.cpp 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281
  1. #include "StdInc.h"
  2. #include "Connection.h"
  3. #include "../registerTypes/RegisterTypes.h"
  4. #include "../mapping/CMap.h"
  5. #include "../CGameState.h"
  6. #include <boost/asio.hpp>
  7. /*
  8. * Connection.cpp, part of VCMI engine
  9. *
  10. * Authors: listed in file AUTHORS in main folder
  11. *
  12. * License: GNU General Public License v2.0 or later
  13. * Full text of license available in license.txt file, in main folder
  14. *
  15. */
  16. using namespace boost;
  17. using namespace boost::asio::ip;
  18. #if defined(__hppa__) || \
  19. defined(__m68k__) || defined(mc68000) || defined(_M_M68K) || \
  20. (defined(__MIPS__) && defined(__MISPEB__)) || \
  21. defined(__ppc__) || defined(__POWERPC__) || defined(_M_PPC) || \
  22. defined(__sparc__)
  23. #define BIG_ENDIAN
  24. #else
  25. #define LIL_ENDIAN
  26. #endif
  27. void CConnection::init()
  28. {
  29. boost::asio::ip::tcp::no_delay option(true);
  30. socket->set_option(option);
  31. enableSmartPointerSerializatoin();
  32. disableStackSendingByID();
  33. registerTypes(iser);
  34. registerTypes(oser);
  35. #ifdef LIL_ENDIAN
  36. myEndianess = true;
  37. #else
  38. myEndianess = false;
  39. #endif
  40. connected = true;
  41. std::string pom;
  42. //we got connection
  43. oser & std::string("Aiya!\n") & name & myEndianess; //identify ourselves
  44. iser & pom & pom & contactEndianess;
  45. logNetwork->infoStream() << "Established connection with "<<pom;
  46. wmx = new boost::mutex;
  47. rmx = new boost::mutex;
  48. handler = nullptr;
  49. receivedStop = sendStop = false;
  50. static int cid = 1;
  51. connectionID = cid++;
  52. iser.fileVersion = SERIALIZATION_VERSION;
  53. }
  54. CConnection::CConnection(std::string host, std::string port, std::string Name)
  55. :iser(this), oser(this), io_service(new asio::io_service), name(Name)
  56. {
  57. int i;
  58. boost::system::error_code error = asio::error::host_not_found;
  59. socket = new tcp::socket(*io_service);
  60. tcp::resolver resolver(*io_service);
  61. tcp::resolver::iterator end, pom, endpoint_iterator = resolver.resolve(tcp::resolver::query(host,port),error);
  62. if(error)
  63. {
  64. logNetwork->errorStream() << "Problem with resolving: \n" << error;
  65. goto connerror1;
  66. }
  67. pom = endpoint_iterator;
  68. if(pom != end)
  69. logNetwork->infoStream()<<"Found endpoints:";
  70. else
  71. {
  72. logNetwork->errorStream() << "Critical problem: No endpoints found!";
  73. goto connerror1;
  74. }
  75. i=0;
  76. while(pom != end)
  77. {
  78. logNetwork->infoStream() << "\t" << i << ": " << (boost::asio::ip::tcp::endpoint&)*pom;
  79. pom++;
  80. }
  81. i=0;
  82. while(endpoint_iterator != end)
  83. {
  84. logNetwork->infoStream() << "Trying connection to " << (boost::asio::ip::tcp::endpoint&)*endpoint_iterator << " (" << i++ << ")";
  85. socket->connect(*endpoint_iterator, error);
  86. if(!error)
  87. {
  88. init();
  89. return;
  90. }
  91. else
  92. {
  93. logNetwork->errorStream() << "Problem with connecting: " << error;
  94. }
  95. endpoint_iterator++;
  96. }
  97. //we shouldn't be here - error handling
  98. connerror1:
  99. logNetwork->errorStream() << "Something went wrong... checking for error info";
  100. if(error)
  101. logNetwork->errorStream() << error;
  102. else
  103. logNetwork->errorStream() << "No error info. ";
  104. delete io_service;
  105. //delete socket;
  106. throw std::runtime_error("Can't establish connection :(");
  107. }
  108. CConnection::CConnection(TSocket * Socket, std::string Name )
  109. :iser(this), oser(this), socket(Socket),io_service(&Socket->get_io_service()), name(Name)//, send(this), rec(this)
  110. {
  111. init();
  112. }
  113. CConnection::CConnection(TAcceptor * acceptor, boost::asio::io_service *Io_service, std::string Name)
  114. : iser(this), oser(this), name(Name)//, send(this), rec(this)
  115. {
  116. boost::system::error_code error = asio::error::host_not_found;
  117. socket = new tcp::socket(*io_service);
  118. acceptor->accept(*socket,error);
  119. if (error)
  120. {
  121. logNetwork->errorStream() << "Error on accepting: " << error;
  122. delete socket;
  123. throw std::runtime_error("Can't establish connection :(");
  124. }
  125. init();
  126. }
  127. int CConnection::write(const void * data, unsigned size)
  128. {
  129. try
  130. {
  131. int ret;
  132. ret = asio::write(*socket,asio::const_buffers_1(asio::const_buffer(data,size)));
  133. return ret;
  134. }
  135. catch(...)
  136. {
  137. //connection has been lost
  138. connected = false;
  139. throw;
  140. }
  141. }
  142. int CConnection::read(void * data, unsigned size)
  143. {
  144. try
  145. {
  146. int ret = asio::read(*socket,asio::mutable_buffers_1(asio::mutable_buffer(data,size)));
  147. return ret;
  148. }
  149. catch(...)
  150. {
  151. //connection has been lost
  152. connected = false;
  153. throw;
  154. }
  155. }
  156. CConnection::~CConnection(void)
  157. {
  158. if(handler)
  159. handler->join();
  160. delete handler;
  161. close();
  162. delete io_service;
  163. delete wmx;
  164. delete rmx;
  165. }
  166. template<class T>
  167. CConnection & CConnection::operator&(const T &t) {
  168. // throw std::exception();
  169. //XXX this is temporaly ? solution to fix gcc (4.3.3, other?) compilation
  170. // problem for more details contact [email protected] or [email protected]
  171. // do not remove this exception it shoudnt be called
  172. return *this;
  173. }
  174. void CConnection::close()
  175. {
  176. if(socket)
  177. {
  178. socket->close();
  179. delete socket;
  180. socket = nullptr;
  181. }
  182. }
  183. bool CConnection::isOpen() const
  184. {
  185. return socket && connected;
  186. }
  187. void CConnection::reportState(CLogger * out)
  188. {
  189. out->debugStream() << "CConnection";
  190. if(socket && socket->is_open())
  191. {
  192. out->debugStream() << "\tWe have an open and valid socket";
  193. out->debugStream() << "\t" << socket->available() <<" bytes awaiting";
  194. }
  195. }
  196. CPack * CConnection::retreivePack()
  197. {
  198. CPack *ret = nullptr;
  199. boost::unique_lock<boost::mutex> lock(*rmx);
  200. logNetwork->traceStream() << "Listening... ";
  201. iser & ret;
  202. logNetwork->traceStream() << "\treceived server message of type " << (ret? typeid(*ret).name() : "nullptr") << ", data: " << ret;
  203. return ret;
  204. }
  205. void CConnection::sendPackToServer(const CPack &pack, PlayerColor player, ui32 requestID)
  206. {
  207. boost::unique_lock<boost::mutex> lock(*wmx);
  208. logNetwork->traceStream() << "Sending to server a pack of type " << typeid(pack).name();
  209. oser & player & requestID & &pack; //packs has to be sent as polymorphic pointers!
  210. }
  211. void CConnection::disableStackSendingByID()
  212. {
  213. CSerializer::sendStackInstanceByIds = false;
  214. }
  215. void CConnection::enableStackSendingByID()
  216. {
  217. CSerializer::sendStackInstanceByIds = true;
  218. }
  219. void CConnection::disableSmartPointerSerialization()
  220. {
  221. iser.smartPointerSerialization = oser.smartPointerSerialization = false;
  222. }
  223. void CConnection::enableSmartPointerSerializatoin()
  224. {
  225. iser.smartPointerSerialization = oser.smartPointerSerialization = true;
  226. }
  227. void CConnection::prepareForSendingHeroes()
  228. {
  229. iser.loadedPointers.clear();
  230. oser.savedPointers.clear();
  231. disableSmartVectorMemberSerialization();
  232. enableSmartPointerSerializatoin();
  233. disableStackSendingByID();
  234. }
  235. void CConnection::enterPregameConnectionMode()
  236. {
  237. iser.loadedPointers.clear();
  238. oser.savedPointers.clear();
  239. disableSmartVectorMemberSerialization();
  240. disableSmartPointerSerialization();
  241. }
  242. void CConnection::disableSmartVectorMemberSerialization()
  243. {
  244. CSerializer::smartVectorMembersSerialization = false;
  245. }
  246. void CConnection::enableSmartVectorMemberSerializatoin()
  247. {
  248. CSerializer::smartVectorMembersSerialization = true;
  249. }
  250. std::ostream & operator<<(std::ostream &str, const CConnection &cpc)
  251. {
  252. return str << "Connection with " << cpc.name << " (ID: " << cpc.connectionID << /*", " << (cpc.host ? "host" : "guest") <<*/ ")";
  253. }