CVCMIServer.cpp 33 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213
  1. /*
  2. * CVCMIServer.cpp, part of VCMI engine
  3. *
  4. * Authors: listed in file AUTHORS in main folder
  5. *
  6. * License: GNU General Public License v2.0 or later
  7. * Full text of license available in license.txt file, in main folder
  8. *
  9. */
  10. #include "StdInc.h"
  11. #include <boost/asio.hpp>
  12. #include "../lib/filesystem/Filesystem.h"
  13. #include "../lib/campaign/CampaignState.h"
  14. #include "../lib/CThreadHelper.h"
  15. #include "../lib/serializer/Connection.h"
  16. #include "../lib/CArtHandler.h"
  17. #include "../lib/CGeneralTextHandler.h"
  18. #include "../lib/CHeroHandler.h"
  19. #include "../lib/CTownHandler.h"
  20. #include "../lib/CBuildingHandler.h"
  21. #include "../lib/spells/CSpellHandler.h"
  22. #include "../lib/CCreatureHandler.h"
  23. #include "zlib.h"
  24. #include "CVCMIServer.h"
  25. #include "../lib/StartInfo.h"
  26. #include "../lib/mapping/CMapHeader.h"
  27. #include "../lib/rmg/CMapGenOptions.h"
  28. #include "../lib/NetPackVisitor.h"
  29. #include "LobbyNetPackVisitors.h"
  30. #ifdef VCMI_ANDROID
  31. #include <jni.h>
  32. #include <android/log.h>
  33. #include "lib/CAndroidVMHelper.h"
  34. #endif
  35. #include "../lib/VCMI_Lib.h"
  36. #include "../lib/VCMIDirs.h"
  37. #include "CGameHandler.h"
  38. #include "PlayerMessageProcessor.h"
  39. #include "../lib/mapping/CMapInfo.h"
  40. #include "../lib/GameConstants.h"
  41. #include "../lib/logging/CBasicLogConfigurator.h"
  42. #include "../lib/CConfigHandler.h"
  43. #include "../lib/ScopeGuard.h"
  44. #include "../lib/serializer/CMemorySerializer.h"
  45. #include "../lib/serializer/Cast.h"
  46. #include "../lib/UnlockGuard.h"
  47. // for applier
  48. #include "../lib/registerTypes/RegisterTypes.h"
  49. // UUID generation
  50. #include <boost/uuid/uuid.hpp>
  51. #include <boost/uuid/uuid_io.hpp>
  52. #include <boost/uuid/uuid_generators.hpp>
  53. #include "../lib/gameState/CGameState.h"
  54. template<typename T> class CApplyOnServer;
  55. class CBaseForServerApply
  56. {
  57. public:
  58. virtual bool applyOnServerBefore(CVCMIServer * srv, void * pack) const =0;
  59. virtual void applyOnServerAfter(CVCMIServer * srv, void * pack) const =0;
  60. virtual ~CBaseForServerApply() {}
  61. template<typename U> static CBaseForServerApply * getApplier(const U * t = nullptr)
  62. {
  63. return new CApplyOnServer<U>();
  64. }
  65. };
  66. template <typename T> class CApplyOnServer : public CBaseForServerApply
  67. {
  68. public:
  69. bool applyOnServerBefore(CVCMIServer * srv, void * pack) const override
  70. {
  71. T * ptr = static_cast<T *>(pack);
  72. ClientPermissionsCheckerNetPackVisitor checker(*srv);
  73. ptr->visit(checker);
  74. if(checker.getResult())
  75. {
  76. boost::unique_lock<boost::mutex> stateLock(srv->stateMutex);
  77. ApplyOnServerNetPackVisitor applier(*srv);
  78. ptr->visit(applier);
  79. return applier.getResult();
  80. }
  81. else
  82. return false;
  83. }
  84. void applyOnServerAfter(CVCMIServer * srv, void * pack) const override
  85. {
  86. T * ptr = static_cast<T *>(pack);
  87. ApplyOnServerAfterAnnounceNetPackVisitor applier(*srv);
  88. ptr->visit(applier);
  89. }
  90. };
  91. template <>
  92. class CApplyOnServer<CPack> : public CBaseForServerApply
  93. {
  94. public:
  95. bool applyOnServerBefore(CVCMIServer * srv, void * pack) const override
  96. {
  97. logGlobal->error("Cannot apply plain CPack!");
  98. assert(0);
  99. return false;
  100. }
  101. void applyOnServerAfter(CVCMIServer * srv, void * pack) const override
  102. {
  103. logGlobal->error("Cannot apply plain CPack!");
  104. assert(0);
  105. }
  106. };
  107. std::string SERVER_NAME_AFFIX = "server";
  108. std::string SERVER_NAME = GameConstants::VCMI_VERSION + std::string(" (") + SERVER_NAME_AFFIX + ')';
  109. CVCMIServer::CVCMIServer(boost::program_options::variables_map & opts)
  110. : port(3030), io(std::make_shared<boost::asio::io_service>()), state(EServerState::LOBBY), cmdLineOptions(opts), currentClientId(1), currentPlayerId(1), restartGameplay(false)
  111. {
  112. uuid = boost::uuids::to_string(boost::uuids::random_generator()());
  113. logNetwork->trace("CVCMIServer created! UUID: %s", uuid);
  114. applier = std::make_shared<CApplier<CBaseForServerApply>>();
  115. registerTypesLobbyPacks(*applier);
  116. if(cmdLineOptions.count("port"))
  117. port = cmdLineOptions["port"].as<ui16>();
  118. logNetwork->info("Port %d will be used", port);
  119. try
  120. {
  121. acceptor = std::make_shared<TAcceptor>(*io, boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), port));
  122. }
  123. catch(...)
  124. {
  125. logNetwork->info("Port %d is busy, trying to use random port instead", port);
  126. if(cmdLineOptions.count("run-by-client"))
  127. {
  128. logNetwork->error("Port must be specified when run-by-client is used!!");
  129. exit(0);
  130. }
  131. acceptor = std::make_shared<TAcceptor>(*io, boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), 0));
  132. port = acceptor->local_endpoint().port();
  133. }
  134. logNetwork->info("Listening for connections at port %d", port);
  135. }
  136. CVCMIServer::~CVCMIServer()
  137. {
  138. announceQueue.clear();
  139. if(announceLobbyThread)
  140. announceLobbyThread->join();
  141. }
  142. void CVCMIServer::run()
  143. {
  144. if(!restartGameplay)
  145. {
  146. this->announceLobbyThread = std::make_unique<boost::thread>(&CVCMIServer::threadAnnounceLobby, this);
  147. startAsyncAccept();
  148. if(!remoteConnectionsThread && cmdLineOptions.count("lobby"))
  149. {
  150. remoteConnectionsThread = std::make_unique<boost::thread>(&CVCMIServer::establishRemoteConnections, this);
  151. }
  152. #if defined(VCMI_ANDROID)
  153. #ifndef SINGLE_PROCESS_APP
  154. CAndroidVMHelper vmHelper;
  155. vmHelper.callStaticVoidMethod(CAndroidVMHelper::NATIVE_METHODS_DEFAULT_CLASS, "onServerReady");
  156. #endif
  157. #endif
  158. }
  159. while(state == EServerState::LOBBY || state == EServerState::GAMEPLAY_STARTING)
  160. boost::this_thread::sleep(boost::posix_time::milliseconds(50));
  161. logNetwork->info("Thread handling connections ended");
  162. if(state == EServerState::GAMEPLAY)
  163. {
  164. gh->run(si->mode == StartInfo::LOAD_GAME);
  165. }
  166. while(state == EServerState::GAMEPLAY_ENDED)
  167. boost::this_thread::sleep(boost::posix_time::milliseconds(50));
  168. }
  169. void CVCMIServer::establishRemoteConnections()
  170. {
  171. //wait for host connection
  172. while(connections.empty())
  173. boost::this_thread::sleep(boost::posix_time::milliseconds(50));
  174. uuid = cmdLineOptions["lobby-uuid"].as<std::string>();
  175. int numOfConnections = cmdLineOptions["connections"].as<ui16>();
  176. auto address = cmdLineOptions["lobby"].as<std::string>();
  177. int port = cmdLineOptions["lobby-port"].as<ui16>();
  178. logGlobal->info("Server is connecting to remote at %s:%d with uuid %s %d times", address, port, uuid, numOfConnections);
  179. for(int i = 0; i < numOfConnections; ++i)
  180. connectToRemote(address, port);
  181. }
  182. void CVCMIServer::connectToRemote(const std::string & addr, int port)
  183. {
  184. std::shared_ptr<CConnection> c;
  185. try
  186. {
  187. logNetwork->info("Establishing connection...");
  188. c = std::make_shared<CConnection>(addr, port, SERVER_NAME, uuid);
  189. }
  190. catch(...)
  191. {
  192. logNetwork->error("\nCannot establish remote connection!");
  193. }
  194. if(c)
  195. {
  196. connections.insert(c);
  197. c->handler = std::make_shared<boost::thread>(&CVCMIServer::threadHandleClient, this, c);
  198. }
  199. }
  200. void CVCMIServer::threadAnnounceLobby()
  201. {
  202. while(state != EServerState::SHUTDOWN)
  203. {
  204. {
  205. boost::unique_lock<boost::recursive_mutex> myLock(mx);
  206. while(!announceQueue.empty())
  207. {
  208. announcePack(std::move(announceQueue.front()));
  209. announceQueue.pop_front();
  210. }
  211. if(acceptor)
  212. {
  213. io->reset();
  214. io->poll();
  215. }
  216. }
  217. boost::this_thread::sleep(boost::posix_time::milliseconds(50));
  218. }
  219. }
  220. void CVCMIServer::prepareToRestart()
  221. {
  222. if(state == EServerState::GAMEPLAY)
  223. {
  224. restartGameplay = true;
  225. * si = * gh->gs->initialOpts;
  226. si->seedToBeUsed = si->seedPostInit = 0;
  227. state = EServerState::LOBBY;
  228. if (si->campState)
  229. {
  230. assert(si->campState->currentScenario().has_value());
  231. campaignMap = si->campState->currentScenario().value_or(CampaignScenarioID(0));
  232. campaignBonus = si->campState->getBonusID(campaignMap).value_or(-1);
  233. }
  234. // FIXME: dirry hack to make sure old CGameHandler::run is finished
  235. boost::this_thread::sleep(boost::posix_time::milliseconds(1000));
  236. }
  237. for(auto c : connections)
  238. {
  239. c->enterLobbyConnectionMode();
  240. c->disableStackSendingByID();
  241. }
  242. boost::unique_lock<boost::recursive_mutex> queueLock(mx);
  243. gh = nullptr;
  244. }
  245. bool CVCMIServer::prepareToStartGame()
  246. {
  247. Load::Progress * progressTracking = nullptr;
  248. bool progressTrackingFinished = false;
  249. std::thread progressTrackingThread([this, &progressTracking, &progressTrackingFinished](){
  250. while(!progressTrackingFinished)
  251. {
  252. if(progressTracking)
  253. {
  254. boost::unique_lock<boost::recursive_mutex> queueLock(mx);
  255. std::unique_ptr<LobbyLoadProgress> loadProgress(new LobbyLoadProgress);
  256. loadProgress->progress = progressTracking->get();
  257. addToAnnounceQueue(std::move(loadProgress));
  258. }
  259. boost::this_thread::sleep(boost::posix_time::milliseconds(200));
  260. }
  261. progressTrackingFinished = false;
  262. });
  263. progressTrackingThread.detach();
  264. gh = std::make_shared<CGameHandler>(this);
  265. switch(si->mode)
  266. {
  267. case StartInfo::CAMPAIGN:
  268. logNetwork->info("Preparing to start new campaign");
  269. si->campState->setCurrentMap(campaignMap);
  270. si->campState->setCurrentMapBonus(campaignBonus);
  271. gh->init(si.get(), progressTracking);
  272. break;
  273. case StartInfo::NEW_GAME:
  274. logNetwork->info("Preparing to start new game");
  275. gh->init(si.get(), progressTracking);
  276. break;
  277. case StartInfo::LOAD_GAME:
  278. logNetwork->info("Preparing to start loaded game");
  279. if(!gh->load(si->mapname))
  280. return false;
  281. break;
  282. default:
  283. logNetwork->error("Wrong mode in StartInfo!");
  284. assert(0);
  285. break;
  286. }
  287. progressTrackingFinished = true;
  288. while(progressTrackingFinished)
  289. continue;
  290. state = EServerState::GAMEPLAY_STARTING;
  291. return true;
  292. }
  293. void CVCMIServer::startGameImmidiately()
  294. {
  295. for(auto c : connections)
  296. c->enterGameplayConnectionMode(gh->gs);
  297. state = EServerState::GAMEPLAY;
  298. }
  299. void CVCMIServer::startAsyncAccept()
  300. {
  301. assert(!upcomingConnection);
  302. assert(acceptor);
  303. #if BOOST_VERSION >= 107000 // Boost version >= 1.70
  304. upcomingConnection = std::make_shared<TSocket>(acceptor->get_executor());
  305. #else
  306. upcomingConnection = std::make_shared<TSocket>(acceptor->get_io_service());
  307. #endif
  308. acceptor->async_accept(*upcomingConnection, std::bind(&CVCMIServer::connectionAccepted, this, _1));
  309. }
  310. void CVCMIServer::connectionAccepted(const boost::system::error_code & ec)
  311. {
  312. if(ec)
  313. {
  314. if(state != EServerState::SHUTDOWN)
  315. logNetwork->info("Something wrong during accepting: %s", ec.message());
  316. return;
  317. }
  318. try
  319. {
  320. if(state == EServerState::LOBBY || !hangingConnections.empty())
  321. {
  322. logNetwork->info("We got a new connection! :)");
  323. auto c = std::make_shared<CConnection>(upcomingConnection, SERVER_NAME, uuid);
  324. upcomingConnection.reset();
  325. connections.insert(c);
  326. c->handler = std::make_shared<boost::thread>(&CVCMIServer::threadHandleClient, this, c);
  327. }
  328. }
  329. catch(std::exception & e)
  330. {
  331. logNetwork->error("Failure processing new connection! %s", e.what());
  332. upcomingConnection.reset();
  333. }
  334. startAsyncAccept();
  335. }
  336. class CVCMIServerPackVisitor : public VCMI_LIB_WRAP_NAMESPACE(ICPackVisitor)
  337. {
  338. private:
  339. CVCMIServer & handler;
  340. std::shared_ptr<CGameHandler> gh;
  341. public:
  342. CVCMIServerPackVisitor(CVCMIServer & handler, std::shared_ptr<CGameHandler> gh)
  343. :handler(handler), gh(gh)
  344. {
  345. }
  346. virtual bool callTyped() override { return false; }
  347. virtual void visitForLobby(CPackForLobby & packForLobby) override
  348. {
  349. handler.handleReceivedPack(std::unique_ptr<CPackForLobby>(&packForLobby));
  350. }
  351. virtual void visitForServer(CPackForServer & serverPack) override
  352. {
  353. if (gh)
  354. gh->handleReceivedPack(&serverPack);
  355. else
  356. logNetwork->error("Received pack for game server while in lobby!");
  357. }
  358. virtual void visitForClient(CPackForClient & clientPack) override
  359. {
  360. }
  361. };
  362. void CVCMIServer::threadHandleClient(std::shared_ptr<CConnection> c)
  363. {
  364. setThreadName("CVCMIServer::handleConnection");
  365. c->enterLobbyConnectionMode();
  366. while(c->connected)
  367. {
  368. CPack * pack;
  369. try
  370. {
  371. pack = c->retrievePack();
  372. }
  373. catch(boost::system::system_error & e)
  374. {
  375. if (e.code() == boost::asio::error::eof)
  376. logNetwork->error("Network error receiving a pack. Connection has been closed");
  377. else
  378. logNetwork->error("Network error receiving a pack. Connection %s dies. What happened: %s", c->toString(), e.what());
  379. hangingConnections.insert(c);
  380. connections.erase(c);
  381. if(connections.empty() || hostClient == c)
  382. state = EServerState::SHUTDOWN;
  383. if(gh && state == EServerState::GAMEPLAY)
  384. {
  385. gh->handleClientDisconnection(c);
  386. }
  387. break;
  388. }
  389. CVCMIServerPackVisitor visitor(*this, this->gh);
  390. pack->visit(visitor);
  391. }
  392. boost::unique_lock<boost::recursive_mutex> queueLock(mx);
  393. if(c->connected)
  394. {
  395. auto lcd = std::make_unique<LobbyClientDisconnected>();
  396. lcd->c = c;
  397. lcd->clientId = c->connectionID;
  398. handleReceivedPack(std::move(lcd));
  399. }
  400. logNetwork->info("Thread listening for %s ended", c->toString());
  401. c->handler.reset();
  402. }
  403. void CVCMIServer::handleReceivedPack(std::unique_ptr<CPackForLobby> pack)
  404. {
  405. CBaseForServerApply * apply = applier->getApplier(typeList.getTypeID(pack.get()));
  406. if(apply->applyOnServerBefore(this, pack.get()))
  407. addToAnnounceQueue(std::move(pack));
  408. }
  409. void CVCMIServer::announcePack(std::unique_ptr<CPackForLobby> pack)
  410. {
  411. for(auto c : connections)
  412. {
  413. // FIXME: we need to avoid sending something to client that not yet get answer for LobbyClientConnected
  414. // Until UUID set we only pass LobbyClientConnected to this client
  415. if(c->uuid == uuid && !dynamic_cast<LobbyClientConnected *>(pack.get()))
  416. continue;
  417. c->sendPack(pack.get());
  418. }
  419. applier->getApplier(typeList.getTypeID(pack.get()))->applyOnServerAfter(this, pack.get());
  420. }
  421. void CVCMIServer::announceMessage(const std::string & txt)
  422. {
  423. logNetwork->info("Show message: %s", txt);
  424. auto cm = std::make_unique<LobbyShowMessage>();
  425. cm->message = txt;
  426. addToAnnounceQueue(std::move(cm));
  427. }
  428. void CVCMIServer::announceTxt(const std::string & txt, const std::string & playerName)
  429. {
  430. logNetwork->info("%s says: %s", playerName, txt);
  431. auto cm = std::make_unique<LobbyChatMessage>();
  432. cm->playerName = playerName;
  433. cm->message = txt;
  434. addToAnnounceQueue(std::move(cm));
  435. }
  436. void CVCMIServer::addToAnnounceQueue(std::unique_ptr<CPackForLobby> pack)
  437. {
  438. boost::unique_lock<boost::recursive_mutex> queueLock(mx);
  439. announceQueue.push_back(std::move(pack));
  440. }
  441. bool CVCMIServer::passHost(int toConnectionId)
  442. {
  443. for(auto c : connections)
  444. {
  445. if(isClientHost(c->connectionID))
  446. continue;
  447. if(c->connectionID != toConnectionId)
  448. continue;
  449. hostClient = c;
  450. hostClientId = c->connectionID;
  451. announceTxt(boost::str(boost::format("Pass host to connection %d") % toConnectionId));
  452. return true;
  453. }
  454. return false;
  455. }
  456. void CVCMIServer::clientConnected(std::shared_ptr<CConnection> c, std::vector<std::string> & names, std::string uuid, StartInfo::EMode mode)
  457. {
  458. if(state == EServerState::LOBBY)
  459. c->connectionID = currentClientId++;
  460. if(!hostClient)
  461. {
  462. hostClient = c;
  463. hostClientId = c->connectionID;
  464. si->mode = mode;
  465. }
  466. logNetwork->info("Connection with client %d established. UUID: %s", c->connectionID, c->uuid);
  467. if(state == EServerState::LOBBY)
  468. {
  469. for(auto & name : names)
  470. {
  471. logNetwork->info("Client %d player: %s", c->connectionID, name);
  472. ui8 id = currentPlayerId++;
  473. ClientPlayer cp;
  474. cp.connection = c->connectionID;
  475. cp.name = name;
  476. playerNames.insert(std::make_pair(id, cp));
  477. announceTxt(boost::str(boost::format("%s (pid %d cid %d) joins the game") % name % id % c->connectionID));
  478. //put new player in first slot with AI
  479. for(auto & elem : si->playerInfos)
  480. {
  481. if(elem.second.isControlledByAI() && !elem.second.compOnly)
  482. {
  483. setPlayerConnectedId(elem.second, id);
  484. break;
  485. }
  486. }
  487. }
  488. }
  489. }
  490. void CVCMIServer::clientDisconnected(std::shared_ptr<CConnection> c)
  491. {
  492. connections -= c;
  493. if(connections.empty() || hostClient == c)
  494. {
  495. state = EServerState::SHUTDOWN;
  496. return;
  497. }
  498. PlayerReinitInterface startAiPack;
  499. startAiPack.playerConnectionId = PlayerSettings::PLAYER_AI;
  500. for(auto it = playerNames.begin(); it != playerNames.end();)
  501. {
  502. if(it->second.connection != c->connectionID)
  503. {
  504. ++it;
  505. continue;
  506. }
  507. int id = it->first;
  508. std::string playerLeftMsgText = boost::str(boost::format("%s (pid %d cid %d) left the game") % id % playerNames[id].name % c->connectionID);
  509. announceTxt(playerLeftMsgText); //send lobby text, it will be ignored for non-lobby clients
  510. auto * playerSettings = si->getPlayersSettings(id);
  511. if(!playerSettings)
  512. {
  513. ++it;
  514. continue;
  515. }
  516. it = playerNames.erase(it);
  517. setPlayerConnectedId(*playerSettings, PlayerSettings::PLAYER_AI);
  518. if(gh && si && state == EServerState::GAMEPLAY)
  519. {
  520. gh->playerMessages->broadcastMessage(playerSettings->color, playerLeftMsgText);
  521. gh->connections[playerSettings->color].insert(hostClient);
  522. startAiPack.players.push_back(playerSettings->color);
  523. }
  524. }
  525. if(!startAiPack.players.empty())
  526. gh->sendAndApply(&startAiPack);
  527. }
  528. void CVCMIServer::reconnectPlayer(int connId)
  529. {
  530. PlayerReinitInterface startAiPack;
  531. startAiPack.playerConnectionId = connId;
  532. if(gh && si && state == EServerState::GAMEPLAY)
  533. {
  534. for(auto it = playerNames.begin(); it != playerNames.end(); ++it)
  535. {
  536. if(it->second.connection != connId)
  537. continue;
  538. int id = it->first;
  539. auto * playerSettings = si->getPlayersSettings(id);
  540. if(!playerSettings)
  541. continue;
  542. std::string messageText = boost::str(boost::format("%s (cid %d) is connected") % playerSettings->name % connId);
  543. gh->playerMessages->broadcastMessage(playerSettings->color, messageText);
  544. startAiPack.players.push_back(playerSettings->color);
  545. }
  546. if(!startAiPack.players.empty())
  547. gh->sendAndApply(&startAiPack);
  548. }
  549. }
  550. void CVCMIServer::setPlayerConnectedId(PlayerSettings & pset, ui8 player) const
  551. {
  552. if(vstd::contains(playerNames, player))
  553. pset.name = playerNames.find(player)->second.name;
  554. else
  555. pset.name = VLC->generaltexth->allTexts[468]; //Computer
  556. pset.connectedPlayerIDs.clear();
  557. if(player != PlayerSettings::PLAYER_AI)
  558. pset.connectedPlayerIDs.insert(player);
  559. }
  560. void CVCMIServer::updateStartInfoOnMapChange(std::shared_ptr<CMapInfo> mapInfo, std::shared_ptr<CMapGenOptions> mapGenOpts)
  561. {
  562. mi = mapInfo;
  563. if(!mi)
  564. return;
  565. auto namesIt = playerNames.cbegin();
  566. si->playerInfos.clear();
  567. if(mi->scenarioOptionsOfSave)
  568. {
  569. si = CMemorySerializer::deepCopy(*mi->scenarioOptionsOfSave);
  570. si->mode = StartInfo::LOAD_GAME;
  571. if(si->campState)
  572. campaignMap = si->campState->currentScenario().value();
  573. for(auto & ps : si->playerInfos)
  574. {
  575. if(!ps.second.compOnly && ps.second.connectedPlayerIDs.size() && namesIt != playerNames.cend())
  576. {
  577. setPlayerConnectedId(ps.second, namesIt++->first);
  578. }
  579. else
  580. {
  581. setPlayerConnectedId(ps.second, PlayerSettings::PLAYER_AI);
  582. }
  583. }
  584. }
  585. else if(si->mode == StartInfo::NEW_GAME || si->mode == StartInfo::CAMPAIGN)
  586. {
  587. if(mi->campaign)
  588. return;
  589. for(int i = 0; i < mi->mapHeader->players.size(); i++)
  590. {
  591. const PlayerInfo & pinfo = mi->mapHeader->players[i];
  592. //neither computer nor human can play - no player
  593. if(!(pinfo.canHumanPlay || pinfo.canComputerPlay))
  594. continue;
  595. PlayerSettings & pset = si->playerInfos[PlayerColor(i)];
  596. pset.color = PlayerColor(i);
  597. if(pinfo.canHumanPlay && namesIt != playerNames.cend())
  598. {
  599. setPlayerConnectedId(pset, namesIt++->first);
  600. }
  601. else
  602. {
  603. setPlayerConnectedId(pset, PlayerSettings::PLAYER_AI);
  604. if(!pinfo.canHumanPlay)
  605. {
  606. pset.compOnly = true;
  607. }
  608. }
  609. pset.castle = pinfo.defaultCastle();
  610. pset.hero = pinfo.defaultHero();
  611. if(pset.hero != PlayerSettings::RANDOM && pinfo.hasCustomMainHero())
  612. {
  613. pset.hero = pinfo.mainCustomHeroId;
  614. pset.heroName = pinfo.mainCustomHeroName;
  615. pset.heroPortrait = pinfo.mainCustomHeroPortrait;
  616. }
  617. pset.handicap = PlayerSettings::NO_HANDICAP;
  618. }
  619. if(mi->isRandomMap && mapGenOpts)
  620. si->mapGenOptions = std::shared_ptr<CMapGenOptions>(mapGenOpts);
  621. else
  622. si->mapGenOptions.reset();
  623. }
  624. si->mapname = mi->fileURI;
  625. }
  626. void CVCMIServer::updateAndPropagateLobbyState()
  627. {
  628. boost::unique_lock<boost::mutex> stateLock(stateMutex);
  629. // Update player settings for RMG
  630. // TODO: find appropriate location for this code
  631. if(si->mapGenOptions && si->mode == StartInfo::NEW_GAME)
  632. {
  633. for(const auto & psetPair : si->playerInfos)
  634. {
  635. const auto & pset = psetPair.second;
  636. si->mapGenOptions->setStartingTownForPlayer(pset.color, pset.castle);
  637. if(pset.isControlledByHuman())
  638. {
  639. si->mapGenOptions->setPlayerTypeForStandardPlayer(pset.color, EPlayerType::HUMAN);
  640. }
  641. }
  642. }
  643. auto lus = std::make_unique<LobbyUpdateState>();
  644. lus->state = *this;
  645. addToAnnounceQueue(std::move(lus));
  646. }
  647. void CVCMIServer::setPlayer(PlayerColor clickedColor)
  648. {
  649. struct PlayerToRestore
  650. {
  651. PlayerColor color;
  652. int id;
  653. void reset() { id = -1; color = PlayerColor::CANNOT_DETERMINE; }
  654. PlayerToRestore(){ reset(); }
  655. } playerToRestore;
  656. PlayerSettings & clicked = si->playerInfos[clickedColor];
  657. //identify clicked player
  658. int clickedNameID = 0; //number of player - zero means AI, assume it initially
  659. if(clicked.isControlledByHuman())
  660. clickedNameID = *(clicked.connectedPlayerIDs.begin()); //if not AI - set appropiate ID
  661. if(clickedNameID > 0 && playerToRestore.id == clickedNameID) //player to restore is about to being replaced -> put him back to the old place
  662. {
  663. PlayerSettings & restPos = si->playerInfos[playerToRestore.color];
  664. setPlayerConnectedId(restPos, playerToRestore.id);
  665. playerToRestore.reset();
  666. }
  667. int newPlayer; //which player will take clicked position
  668. //who will be put here?
  669. if(!clickedNameID) //AI player clicked -> if possible replace computer with unallocated player
  670. {
  671. newPlayer = getIdOfFirstUnallocatedPlayer();
  672. if(!newPlayer) //no "free" player -> get just first one
  673. newPlayer = playerNames.begin()->first;
  674. }
  675. else //human clicked -> take next
  676. {
  677. auto i = playerNames.find(clickedNameID); //clicked one
  678. i++; //player AFTER clicked one
  679. if(i != playerNames.end())
  680. newPlayer = i->first;
  681. else
  682. newPlayer = 0; //AI if we scrolled through all players
  683. }
  684. setPlayerConnectedId(clicked, newPlayer); //put player
  685. //if that player was somewhere else, we need to replace him with computer
  686. if(newPlayer) //not AI
  687. {
  688. for(auto i = si->playerInfos.begin(); i != si->playerInfos.end(); i++)
  689. {
  690. int curNameID = *(i->second.connectedPlayerIDs.begin());
  691. if(i->first != clickedColor && curNameID == newPlayer)
  692. {
  693. assert(i->second.connectedPlayerIDs.size());
  694. playerToRestore.color = i->first;
  695. playerToRestore.id = newPlayer;
  696. setPlayerConnectedId(i->second, PlayerSettings::PLAYER_AI); //set computer
  697. break;
  698. }
  699. }
  700. }
  701. }
  702. void CVCMIServer::optionNextCastle(PlayerColor player, int dir)
  703. {
  704. PlayerSettings & s = si->playerInfos[player];
  705. FactionID & cur = s.castle;
  706. auto & allowed = getPlayerInfo(player.getNum()).allowedFactions;
  707. const bool allowRandomTown = getPlayerInfo(player.getNum()).isFactionRandom;
  708. if(cur == PlayerSettings::NONE) //no change
  709. return;
  710. if(cur == PlayerSettings::RANDOM) //first/last available
  711. {
  712. if(dir > 0)
  713. cur = *allowed.begin(); //id of first town
  714. else
  715. cur = *allowed.rbegin(); //id of last town
  716. }
  717. else // next/previous available
  718. {
  719. if((cur == *allowed.begin() && dir < 0) || (cur == *allowed.rbegin() && dir > 0))
  720. {
  721. if(allowRandomTown)
  722. {
  723. cur = PlayerSettings::RANDOM;
  724. }
  725. else
  726. {
  727. if(dir > 0)
  728. cur = *allowed.begin();
  729. else
  730. cur = *allowed.rbegin();
  731. }
  732. }
  733. else
  734. {
  735. assert(dir >= -1 && dir <= 1); //othervice std::advance may go out of range
  736. auto iter = allowed.find(cur);
  737. std::advance(iter, dir);
  738. cur = *iter;
  739. }
  740. }
  741. if(s.hero >= 0 && !getPlayerInfo(player.getNum()).hasCustomMainHero()) // remove hero unless it set to fixed one in map editor
  742. {
  743. s.hero = PlayerSettings::RANDOM;
  744. }
  745. if(cur < 0 && s.bonus == PlayerSettings::RESOURCE)
  746. s.bonus = PlayerSettings::RANDOM;
  747. }
  748. void CVCMIServer::optionSetCastle(PlayerColor player, int id)
  749. {
  750. PlayerSettings & s = si->playerInfos[player];
  751. FactionID & cur = s.castle;
  752. auto & allowed = getPlayerInfo(player.getNum()).allowedFactions;
  753. if(cur == PlayerSettings::NONE) //no change
  754. return;
  755. if(allowed.find(static_cast<FactionID>(id)) == allowed.end() && id != PlayerSettings::RANDOM) // valid id
  756. return;
  757. cur = static_cast<FactionID>(id);
  758. if(s.hero >= 0 && !getPlayerInfo(player.getNum()).hasCustomMainHero()) // remove hero unless it set to fixed one in map editor
  759. {
  760. s.hero = PlayerSettings::RANDOM;
  761. }
  762. if(cur < 0 && s.bonus == PlayerSettings::RESOURCE)
  763. s.bonus = PlayerSettings::RANDOM;
  764. }
  765. void CVCMIServer::setCampaignMap(CampaignScenarioID mapId)
  766. {
  767. campaignMap = mapId;
  768. si->difficulty = si->campState->scenario(mapId).difficulty;
  769. campaignBonus = -1;
  770. updateStartInfoOnMapChange(si->campState->getMapInfo(mapId));
  771. }
  772. void CVCMIServer::setCampaignBonus(int bonusId)
  773. {
  774. campaignBonus = bonusId;
  775. const CampaignScenario & scenario = si->campState->scenario(campaignMap);
  776. const std::vector<CampaignBonus> & bonDescs = scenario.travelOptions.bonusesToChoose;
  777. if(bonDescs[bonusId].type == CampaignBonusType::HERO)
  778. {
  779. for(auto & elem : si->playerInfos)
  780. {
  781. if(elem.first == PlayerColor(bonDescs[bonusId].info1))
  782. setPlayerConnectedId(elem.second, 1);
  783. else
  784. setPlayerConnectedId(elem.second, PlayerSettings::PLAYER_AI);
  785. }
  786. }
  787. }
  788. void CVCMIServer::optionNextHero(PlayerColor player, int dir)
  789. {
  790. PlayerSettings & s = si->playerInfos[player];
  791. if(s.castle < 0 || s.hero == PlayerSettings::NONE)
  792. return;
  793. if(s.hero == PlayerSettings::RANDOM) // first/last available
  794. {
  795. int max = static_cast<int>(VLC->heroh->size()),
  796. min = 0;
  797. s.hero = nextAllowedHero(player, min, max, 0, dir);
  798. }
  799. else
  800. {
  801. if(dir > 0)
  802. s.hero = nextAllowedHero(player, s.hero, (int)VLC->heroh->size(), 1, dir);
  803. else
  804. s.hero = nextAllowedHero(player, -1, s.hero, 1, dir); // min needs to be -1 -- hero at index 0 would be skipped otherwise
  805. }
  806. }
  807. void CVCMIServer::optionSetHero(PlayerColor player, int id)
  808. {
  809. PlayerSettings & s = si->playerInfos[player];
  810. if(s.castle < 0 || s.hero == PlayerSettings::NONE)
  811. return;
  812. if(id == PlayerSettings::RANDOM)
  813. {
  814. s.hero = PlayerSettings::RANDOM;
  815. }
  816. if(canUseThisHero(player, id))
  817. s.hero = static_cast<HeroTypeID>(id);
  818. }
  819. int CVCMIServer::nextAllowedHero(PlayerColor player, int min, int max, int incl, int dir)
  820. {
  821. if(dir > 0)
  822. {
  823. for(int i = min + incl; i <= max - incl; i++)
  824. if(canUseThisHero(player, i))
  825. return i;
  826. }
  827. else
  828. {
  829. for(int i = max - incl; i >= min + incl; i--)
  830. if(canUseThisHero(player, i))
  831. return i;
  832. }
  833. return -1;
  834. }
  835. void CVCMIServer::optionNextBonus(PlayerColor player, int dir)
  836. {
  837. PlayerSettings & s = si->playerInfos[player];
  838. PlayerSettings::Ebonus & ret = s.bonus = static_cast<PlayerSettings::Ebonus>(static_cast<int>(s.bonus) + dir);
  839. if(s.hero == PlayerSettings::NONE &&
  840. !getPlayerInfo(player.getNum()).heroesNames.size() &&
  841. ret == PlayerSettings::ARTIFACT) //no hero - can't be artifact
  842. {
  843. if(dir < 0)
  844. ret = PlayerSettings::RANDOM;
  845. else
  846. ret = PlayerSettings::GOLD;
  847. }
  848. if(ret > PlayerSettings::RESOURCE)
  849. ret = PlayerSettings::RANDOM;
  850. if(ret < PlayerSettings::RANDOM)
  851. ret = PlayerSettings::RESOURCE;
  852. if(s.castle == PlayerSettings::RANDOM && ret == PlayerSettings::RESOURCE) //random castle - can't be resource
  853. {
  854. if(dir < 0)
  855. ret = PlayerSettings::GOLD;
  856. else
  857. ret = PlayerSettings::RANDOM;
  858. }
  859. }
  860. void CVCMIServer::optionSetBonus(PlayerColor player, int id)
  861. {
  862. PlayerSettings & s = si->playerInfos[player];
  863. if(s.hero == PlayerSettings::NONE &&
  864. !getPlayerInfo(player.getNum()).heroesNames.size() &&
  865. id == PlayerSettings::ARTIFACT) //no hero - can't be artifact
  866. return;
  867. if(id > PlayerSettings::RESOURCE)
  868. return;
  869. if(id < PlayerSettings::RANDOM)
  870. return;
  871. if(s.castle == PlayerSettings::RANDOM && id == PlayerSettings::RESOURCE) //random castle - can't be resource
  872. return;
  873. s.bonus = static_cast<PlayerSettings::Ebonus>(static_cast<int>(id));
  874. }
  875. bool CVCMIServer::canUseThisHero(PlayerColor player, int ID)
  876. {
  877. return VLC->heroh->size() > ID
  878. && si->playerInfos[player].castle == VLC->heroh->objects[ID]->heroClass->faction
  879. && !vstd::contains(getUsedHeroes(), ID)
  880. && mi->mapHeader->allowedHeroes[ID];
  881. }
  882. std::vector<int> CVCMIServer::getUsedHeroes()
  883. {
  884. std::vector<int> heroIds;
  885. for(auto & p : si->playerInfos)
  886. {
  887. const auto & heroes = getPlayerInfo(p.first.getNum()).heroesNames;
  888. for(auto & hero : heroes)
  889. if(hero.heroId >= 0) //in VCMI map format heroId = -1 means random hero
  890. heroIds.push_back(hero.heroId);
  891. if(p.second.hero != PlayerSettings::RANDOM)
  892. heroIds.push_back(p.second.hero);
  893. }
  894. return heroIds;
  895. }
  896. ui8 CVCMIServer::getIdOfFirstUnallocatedPlayer() const
  897. {
  898. for(auto i = playerNames.cbegin(); i != playerNames.cend(); i++)
  899. {
  900. if(!si->getPlayersSettings(i->first))
  901. return i->first;
  902. }
  903. return 0;
  904. }
  905. static void handleCommandOptions(int argc, const char * argv[], boost::program_options::variables_map & options)
  906. {
  907. namespace po = boost::program_options;
  908. po::options_description opts("Allowed options");
  909. opts.add_options()
  910. ("help,h", "display help and exit")
  911. ("version,v", "display version information and exit")
  912. ("run-by-client", "indicate that server launched by client on same machine")
  913. ("uuid", po::value<std::string>(), "")
  914. ("port", po::value<ui16>(), "port at which server will listen to connections from client")
  915. ("lobby", po::value<std::string>(), "address to remote lobby")
  916. ("lobby-port", po::value<ui16>(), "port at which server connect to remote lobby")
  917. ("lobby-uuid", po::value<std::string>(), "")
  918. ("connections", po::value<ui16>(), "amount of connections to remote lobby");
  919. if(argc > 1)
  920. {
  921. try
  922. {
  923. po::store(po::parse_command_line(argc, argv, opts), options);
  924. }
  925. catch(boost::program_options::error & e)
  926. {
  927. std::cerr << "Failure during parsing command-line options:\n" << e.what() << std::endl;
  928. }
  929. }
  930. #ifdef SINGLE_PROCESS_APP
  931. options.emplace("run-by-client", po::variable_value{true, true});
  932. #endif
  933. po::notify(options);
  934. #ifndef SINGLE_PROCESS_APP
  935. if(options.count("help"))
  936. {
  937. auto time = std::time(0);
  938. printf("%s - A Heroes of Might and Magic 3 clone\n", GameConstants::VCMI_VERSION.c_str());
  939. printf("Copyright (C) 2007-%d VCMI dev team - see AUTHORS file\n", std::localtime(&time)->tm_year + 1900);
  940. printf("This is free software; see the source for copying conditions. There is NO\n");
  941. printf("warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n");
  942. printf("\n");
  943. std::cout << opts;
  944. exit(0);
  945. }
  946. if(options.count("version"))
  947. {
  948. printf("%s\n", GameConstants::VCMI_VERSION.c_str());
  949. std::cout << VCMIDirs::get().genHelpString();
  950. exit(0);
  951. }
  952. #endif
  953. }
  954. #ifdef SINGLE_PROCESS_APP
  955. #define main server_main
  956. #endif
  957. #if VCMI_ANDROID_DUAL_PROCESS
  958. void CVCMIServer::create()
  959. {
  960. const int argc = 1;
  961. const char * argv[argc] = { "android-server" };
  962. #else
  963. int main(int argc, const char * argv[])
  964. {
  965. #endif
  966. #if !defined(VCMI_ANDROID) && !defined(SINGLE_PROCESS_APP)
  967. // Correct working dir executable folder (not bundle folder) so we can use executable relative paths
  968. boost::filesystem::current_path(boost::filesystem::system_complete(argv[0]).parent_path());
  969. #endif
  970. #ifndef VCMI_IOS
  971. console = new CConsoleHandler();
  972. #endif
  973. CBasicLogConfigurator logConfig(VCMIDirs::get().userLogsPath() / "VCMI_Server_log.txt", console);
  974. logConfig.configureDefault();
  975. logGlobal->info(SERVER_NAME);
  976. boost::program_options::variables_map opts;
  977. handleCommandOptions(argc, argv, opts);
  978. preinitDLL(console);
  979. logConfig.configure();
  980. loadDLLClasses();
  981. srand((ui32)time(nullptr));
  982. #ifdef SINGLE_PROCESS_APP
  983. boost::condition_variable * cond = reinterpret_cast<boost::condition_variable *>(const_cast<char *>(argv[0]));
  984. cond->notify_one();
  985. #endif
  986. try
  987. {
  988. boost::asio::io_service io_service;
  989. CVCMIServer server(opts);
  990. try
  991. {
  992. while(server.state != EServerState::SHUTDOWN)
  993. {
  994. server.run();
  995. }
  996. io_service.run();
  997. }
  998. catch(boost::system::system_error & e) //for boost errors just log, not crash - probably client shut down connection
  999. {
  1000. logNetwork->error(e.what());
  1001. server.state = EServerState::SHUTDOWN;
  1002. }
  1003. }
  1004. catch(boost::system::system_error & e)
  1005. {
  1006. logNetwork->error(e.what());
  1007. //catch any startup errors (e.g. can't access port) errors
  1008. //and return non-zero status so client can detect error
  1009. throw;
  1010. }
  1011. #if VCMI_ANDROID_DUAL_PROCESS
  1012. CAndroidVMHelper envHelper;
  1013. envHelper.callStaticVoidMethod(CAndroidVMHelper::NATIVE_METHODS_DEFAULT_CLASS, "killServer");
  1014. #endif
  1015. logConfig.deconfigure();
  1016. vstd::clear_pointer(VLC);
  1017. #if !VCMI_ANDROID_DUAL_PROCESS
  1018. return 0;
  1019. #endif
  1020. }
  1021. #if VCMI_ANDROID_DUAL_PROCESS
  1022. extern "C" JNIEXPORT void JNICALL Java_eu_vcmi_vcmi_NativeMethods_createServer(JNIEnv * env, jclass cls)
  1023. {
  1024. __android_log_write(ANDROID_LOG_INFO, "VCMI", "Got jni call to init server");
  1025. CAndroidVMHelper::cacheVM(env);
  1026. CVCMIServer::create();
  1027. }
  1028. extern "C" JNIEXPORT void JNICALL Java_eu_vcmi_vcmi_NativeMethods_initClassloader(JNIEnv * baseEnv, jclass cls)
  1029. {
  1030. CAndroidVMHelper::initClassloader(baseEnv);
  1031. }
  1032. #elif defined(SINGLE_PROCESS_APP)
  1033. void CVCMIServer::create(boost::condition_variable * cond, const std::vector<std::string> & args)
  1034. {
  1035. std::vector<const void *> argv = {cond};
  1036. for(auto & a : args)
  1037. argv.push_back(a.c_str());
  1038. main(argv.size(), reinterpret_cast<const char **>(&*argv.begin()));
  1039. }
  1040. #ifdef VCMI_ANDROID
  1041. void CVCMIServer::reuseClientJNIEnv(void * jniEnv)
  1042. {
  1043. CAndroidVMHelper::initClassloader(jniEnv);
  1044. CAndroidVMHelper::alwaysUseLoadedClass = true;
  1045. }
  1046. #endif // VCMI_ANDROID
  1047. #endif // VCMI_ANDROID_DUAL_PROCESS