CVCMIServer.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993
  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/mapping/CCampaignHandler.h"
  14. #include "../lib/CThreadHelper.h"
  15. #include "../lib/serializer/Connection.h"
  16. #include "../lib/CModHandler.h"
  17. #include "../lib/CArtHandler.h"
  18. #include "../lib/CGeneralTextHandler.h"
  19. #include "../lib/CHeroHandler.h"
  20. #include "../lib/CTownHandler.h"
  21. #include "../lib/CBuildingHandler.h"
  22. #include "../lib/spells/CSpellHandler.h"
  23. #include "../lib/CCreatureHandler.h"
  24. #include "zlib.h"
  25. #include "CVCMIServer.h"
  26. #include "../lib/StartInfo.h"
  27. #include "../lib/mapping/CMap.h"
  28. #include "../lib/rmg/CMapGenOptions.h"
  29. #ifdef VCMI_ANDROID
  30. #include "lib/CAndroidVMHelper.h"
  31. #elif defined(VCMI_IOS)
  32. // todo ios
  33. #else
  34. #include "../lib/Interprocess.h"
  35. #endif
  36. #include "../lib/VCMI_Lib.h"
  37. #include "../lib/VCMIDirs.h"
  38. #include "CGameHandler.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/CGameState.h"
  54. #if defined(__GNUC__) && !defined(__MINGW32__) && !defined(VCMI_ANDROID) && !defined(VCMI_IOS)
  55. #include <execinfo.h>
  56. #endif
  57. template<typename T> class CApplyOnServer;
  58. class CBaseForServerApply
  59. {
  60. public:
  61. virtual bool applyOnServerBefore(CVCMIServer * srv, void * pack) const =0;
  62. virtual void applyOnServerAfter(CVCMIServer * srv, void * pack) const =0;
  63. virtual ~CBaseForServerApply() {}
  64. template<typename U> static CBaseForServerApply * getApplier(const U * t = nullptr)
  65. {
  66. return new CApplyOnServer<U>();
  67. }
  68. };
  69. template <typename T> class CApplyOnServer : public CBaseForServerApply
  70. {
  71. public:
  72. bool applyOnServerBefore(CVCMIServer * srv, void * pack) const override
  73. {
  74. T * ptr = static_cast<T *>(pack);
  75. if(ptr->checkClientPermissions(srv))
  76. {
  77. boost::unique_lock<boost::mutex> stateLock(srv->stateMutex);
  78. return ptr->applyOnServer(srv);
  79. }
  80. else
  81. return false;
  82. }
  83. void applyOnServerAfter(CVCMIServer * srv, void * pack) const override
  84. {
  85. T * ptr = static_cast<T *>(pack);
  86. ptr->applyOnServerAfterAnnounce(srv);
  87. }
  88. };
  89. template <>
  90. class CApplyOnServer<CPack> : public CBaseForServerApply
  91. {
  92. public:
  93. bool applyOnServerBefore(CVCMIServer * srv, void * pack) const override
  94. {
  95. logGlobal->error("Cannot apply plain CPack!");
  96. assert(0);
  97. return false;
  98. }
  99. void applyOnServerAfter(CVCMIServer * srv, void * pack) const override
  100. {
  101. logGlobal->error("Cannot apply plain CPack!");
  102. assert(0);
  103. }
  104. };
  105. std::string SERVER_NAME_AFFIX = "server";
  106. std::string SERVER_NAME = GameConstants::VCMI_VERSION + std::string(" (") + SERVER_NAME_AFFIX + ')';
  107. CVCMIServer::CVCMIServer(boost::program_options::variables_map & opts)
  108. : port(3030), io(std::make_shared<boost::asio::io_service>()), state(EServerState::LOBBY), cmdLineOptions(opts), currentClientId(1), currentPlayerId(1), restartGameplay(false)
  109. {
  110. uuid = boost::uuids::to_string(boost::uuids::random_generator()());
  111. logNetwork->trace("CVCMIServer created! UUID: %s", uuid);
  112. applier = std::make_shared<CApplier<CBaseForServerApply>>();
  113. registerTypesLobbyPacks(*applier);
  114. if(cmdLineOptions.count("port"))
  115. port = cmdLineOptions["port"].as<ui16>();
  116. logNetwork->info("Port %d will be used", port);
  117. try
  118. {
  119. acceptor = std::make_shared<TAcceptor>(*io, boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), port));
  120. }
  121. catch(...)
  122. {
  123. logNetwork->info("Port %d is busy, trying to use random port instead", port);
  124. if(cmdLineOptions.count("run-by-client") && !cmdLineOptions.count("enable-shm"))
  125. {
  126. logNetwork->error("Cant pass port number to client without shared memory!", port);
  127. exit(0);
  128. }
  129. acceptor = std::make_shared<TAcceptor>(*io, boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), 0));
  130. port = acceptor->local_endpoint().port();
  131. }
  132. logNetwork->info("Listening for connections at port %d", port);
  133. }
  134. CVCMIServer::~CVCMIServer()
  135. {
  136. announceQueue.clear();
  137. if(announceLobbyThread)
  138. announceLobbyThread->join();
  139. }
  140. void CVCMIServer::run()
  141. {
  142. if(!restartGameplay)
  143. {
  144. this->announceLobbyThread = vstd::make_unique<boost::thread>(&CVCMIServer::threadAnnounceLobby, this);
  145. #if !defined(VCMI_ANDROID) && !defined(VCMI_IOS)
  146. if(cmdLineOptions.count("enable-shm"))
  147. {
  148. std::string sharedMemoryName = "vcmi_memory";
  149. if(cmdLineOptions.count("enable-shm-uuid") && cmdLineOptions.count("uuid"))
  150. {
  151. sharedMemoryName += "_" + cmdLineOptions["uuid"].as<std::string>();
  152. }
  153. shm = std::make_shared<SharedMemory>(sharedMemoryName);
  154. }
  155. #endif
  156. startAsyncAccept();
  157. #if !defined(VCMI_ANDROID) && !defined(VCMI_IOS)
  158. if(shm)
  159. {
  160. shm->sr->setToReadyAndNotify(port);
  161. }
  162. #endif
  163. }
  164. while(state == EServerState::LOBBY || state == EServerState::GAMEPLAY_STARTING)
  165. boost::this_thread::sleep(boost::posix_time::milliseconds(50));
  166. logNetwork->info("Thread handling connections ended");
  167. if(state == EServerState::GAMEPLAY)
  168. {
  169. gh->run(si->mode == StartInfo::LOAD_GAME);
  170. }
  171. while(state == EServerState::GAMEPLAY_ENDED)
  172. boost::this_thread::sleep(boost::posix_time::milliseconds(50));
  173. }
  174. void CVCMIServer::threadAnnounceLobby()
  175. {
  176. while(state != EServerState::SHUTDOWN)
  177. {
  178. {
  179. boost::unique_lock<boost::recursive_mutex> myLock(mx);
  180. while(!announceQueue.empty())
  181. {
  182. announcePack(std::move(announceQueue.front()));
  183. announceQueue.pop_front();
  184. }
  185. if(state != EServerState::LOBBY)
  186. {
  187. if(acceptor)
  188. acceptor->close();
  189. }
  190. if(acceptor)
  191. {
  192. io->reset();
  193. io->poll();
  194. }
  195. }
  196. boost::this_thread::sleep(boost::posix_time::milliseconds(50));
  197. }
  198. }
  199. void CVCMIServer::prepareToStartGame()
  200. {
  201. if(state == EServerState::GAMEPLAY)
  202. {
  203. restartGameplay = true;
  204. * si = * gh->gs->initialOpts;
  205. si->seedToBeUsed = si->seedPostInit = 0;
  206. state = EServerState::LOBBY;
  207. // FIXME: dirry hack to make sure old CGameHandler::run is finished
  208. boost::this_thread::sleep(boost::posix_time::milliseconds(1000));
  209. }
  210. state = EServerState::GAMEPLAY_STARTING;
  211. gh = std::make_shared<CGameHandler>(this);
  212. switch(si->mode)
  213. {
  214. case StartInfo::CAMPAIGN:
  215. logNetwork->info("Preparing to start new campaign");
  216. si->campState->currentMap = boost::make_optional(campaignMap);
  217. si->campState->chosenCampaignBonuses[campaignMap] = campaignBonus;
  218. gh->init(si.get());
  219. break;
  220. case StartInfo::NEW_GAME:
  221. logNetwork->info("Preparing to start new game");
  222. gh->init(si.get());
  223. break;
  224. case StartInfo::LOAD_GAME:
  225. logNetwork->info("Preparing to start loaded game");
  226. gh->load(si->mapname);
  227. break;
  228. default:
  229. logNetwork->error("Wrong mode in StartInfo!");
  230. assert(0);
  231. break;
  232. }
  233. }
  234. void CVCMIServer::startGameImmidiately()
  235. {
  236. for(auto c : connections)
  237. c->enterGameplayConnectionMode(gh->gs);
  238. state = EServerState::GAMEPLAY;
  239. }
  240. void CVCMIServer::startAsyncAccept()
  241. {
  242. assert(!upcomingConnection);
  243. assert(acceptor);
  244. #if BOOST_VERSION >= 107000 // Boost version >= 1.70
  245. upcomingConnection = std::make_shared<TSocket>(acceptor->get_executor());
  246. #else
  247. upcomingConnection = std::make_shared<TSocket>(acceptor->get_io_service());
  248. #endif
  249. acceptor->async_accept(*upcomingConnection, std::bind(&CVCMIServer::connectionAccepted, this, _1));
  250. }
  251. void CVCMIServer::connectionAccepted(const boost::system::error_code & ec)
  252. {
  253. if(ec)
  254. {
  255. if(state != EServerState::SHUTDOWN)
  256. logNetwork->info("Something wrong during accepting: %s", ec.message());
  257. return;
  258. }
  259. try
  260. {
  261. logNetwork->info("We got a new connection! :)");
  262. auto c = std::make_shared<CConnection>(upcomingConnection, SERVER_NAME, uuid);
  263. upcomingConnection.reset();
  264. connections.insert(c);
  265. c->handler = std::make_shared<boost::thread>(&CVCMIServer::threadHandleClient, this, c);
  266. }
  267. catch(std::exception & e)
  268. {
  269. logNetwork->error("Failure processing new connection! %s", e.what());
  270. upcomingConnection.reset();
  271. }
  272. startAsyncAccept();
  273. }
  274. void CVCMIServer::threadHandleClient(std::shared_ptr<CConnection> c)
  275. {
  276. setThreadName("CVCMIServer::handleConnection");
  277. c->enterLobbyConnectionMode();
  278. try
  279. {
  280. while(c->connected)
  281. {
  282. CPack * pack = c->retrievePack();
  283. if(auto lobbyPack = dynamic_ptr_cast<CPackForLobby>(pack))
  284. {
  285. handleReceivedPack(std::unique_ptr<CPackForLobby>(lobbyPack));
  286. }
  287. else if(auto serverPack = dynamic_ptr_cast<CPackForServer>(pack))
  288. {
  289. gh->handleReceivedPack(serverPack);
  290. }
  291. }
  292. }
  293. catch(boost::system::system_error & e)
  294. {
  295. (void)e;
  296. if(state != EServerState::LOBBY)
  297. gh->handleClientDisconnection(c);
  298. }
  299. /*
  300. catch(const std::exception & e)
  301. {
  302. (void)e;
  303. boost::unique_lock<boost::recursive_mutex> queueLock(mx);
  304. logNetwork->error("%s dies... \nWhat happened: %s", c->toString(), e.what());
  305. }
  306. catch(...)
  307. {
  308. state = EServerState::SHUTDOWN;
  309. handleException();
  310. throw;
  311. }*/
  312. boost::unique_lock<boost::recursive_mutex> queueLock(mx);
  313. // if(state != ENDING_AND_STARTING_GAME)
  314. if(c->connected)
  315. {
  316. auto lcd = vstd::make_unique<LobbyClientDisconnected>();
  317. lcd->c = c;
  318. lcd->clientId = c->connectionID;
  319. handleReceivedPack(std::move(lcd));
  320. }
  321. logNetwork->info("Thread listening for %s ended", c->toString());
  322. c->handler.reset();
  323. }
  324. void CVCMIServer::handleReceivedPack(std::unique_ptr<CPackForLobby> pack)
  325. {
  326. CBaseForServerApply * apply = applier->getApplier(typeList.getTypeID(pack.get()));
  327. if(apply->applyOnServerBefore(this, pack.get()))
  328. addToAnnounceQueue(std::move(pack));
  329. }
  330. void CVCMIServer::announcePack(std::unique_ptr<CPackForLobby> pack)
  331. {
  332. for(auto c : connections)
  333. {
  334. // FIXME: we need to avoid senting something to client that not yet get answer for LobbyClientConnected
  335. // Until UUID set we only pass LobbyClientConnected to this client
  336. if(c->uuid == uuid && !dynamic_cast<LobbyClientConnected *>(pack.get()))
  337. continue;
  338. c->sendPack(pack.get());
  339. }
  340. applier->getApplier(typeList.getTypeID(pack.get()))->applyOnServerAfter(this, pack.get());
  341. }
  342. void CVCMIServer::announceTxt(const std::string & txt, const std::string & playerName)
  343. {
  344. logNetwork->info("%s says: %s", playerName, txt);
  345. auto cm = vstd::make_unique<LobbyChatMessage>();
  346. cm->playerName = playerName;
  347. cm->message = txt;
  348. addToAnnounceQueue(std::move(cm));
  349. }
  350. void CVCMIServer::addToAnnounceQueue(std::unique_ptr<CPackForLobby> pack)
  351. {
  352. boost::unique_lock<boost::recursive_mutex> queueLock(mx);
  353. announceQueue.push_back(std::move(pack));
  354. }
  355. bool CVCMIServer::passHost(int toConnectionId)
  356. {
  357. for(auto c : connections)
  358. {
  359. if(isClientHost(c->connectionID))
  360. continue;
  361. if(c->connectionID != toConnectionId)
  362. continue;
  363. hostClient = c;
  364. hostClientId = c->connectionID;
  365. announceTxt(boost::str(boost::format("Pass host to connection %d") % toConnectionId));
  366. return true;
  367. }
  368. return false;
  369. }
  370. void CVCMIServer::clientConnected(std::shared_ptr<CConnection> c, std::vector<std::string> & names, std::string uuid, StartInfo::EMode mode)
  371. {
  372. c->connectionID = currentClientId++;
  373. if(!hostClient)
  374. {
  375. hostClient = c;
  376. hostClientId = c->connectionID;
  377. si->mode = mode;
  378. }
  379. logNetwork->info("Connection with client %d established. UUID: %s", c->connectionID, c->uuid);
  380. for(auto & name : names)
  381. {
  382. logNetwork->info("Client %d player: %s", c->connectionID, name);
  383. ui8 id = currentPlayerId++;
  384. ClientPlayer cp;
  385. cp.connection = c->connectionID;
  386. cp.name = name;
  387. playerNames.insert(std::make_pair(id, cp));
  388. announceTxt(boost::str(boost::format("%s (pid %d cid %d) joins the game") % name % id % c->connectionID));
  389. //put new player in first slot with AI
  390. for(auto & elem : si->playerInfos)
  391. {
  392. if(elem.second.isControlledByAI() && !elem.second.compOnly)
  393. {
  394. setPlayerConnectedId(elem.second, id);
  395. break;
  396. }
  397. }
  398. }
  399. }
  400. void CVCMIServer::clientDisconnected(std::shared_ptr<CConnection> c)
  401. {
  402. connections -= c;
  403. for(auto it = playerNames.begin(); it != playerNames.end();)
  404. {
  405. if(it->second.connection != c->connectionID)
  406. {
  407. it++;
  408. continue;
  409. }
  410. int id = it->first;
  411. announceTxt(boost::str(boost::format("%s (pid %d cid %d) left the game") % id % playerNames[id].name % c->connectionID));
  412. playerNames.erase(it++);
  413. // Reset in-game players client used back to AI
  414. if(PlayerSettings * s = si->getPlayersSettings(id))
  415. {
  416. setPlayerConnectedId(*s, PlayerSettings::PLAYER_AI);
  417. }
  418. }
  419. }
  420. void CVCMIServer::setPlayerConnectedId(PlayerSettings & pset, ui8 player) const
  421. {
  422. if(vstd::contains(playerNames, player))
  423. pset.name = playerNames.find(player)->second.name;
  424. else
  425. pset.name = VLC->generaltexth->allTexts[468]; //Computer
  426. pset.connectedPlayerIDs.clear();
  427. if(player != PlayerSettings::PLAYER_AI)
  428. pset.connectedPlayerIDs.insert(player);
  429. }
  430. void CVCMIServer::updateStartInfoOnMapChange(std::shared_ptr<CMapInfo> mapInfo, std::shared_ptr<CMapGenOptions> mapGenOpts)
  431. {
  432. mi = mapInfo;
  433. if(!mi)
  434. return;
  435. auto namesIt = playerNames.cbegin();
  436. si->playerInfos.clear();
  437. if(mi->scenarioOptionsOfSave)
  438. {
  439. si = CMemorySerializer::deepCopy(*mi->scenarioOptionsOfSave);
  440. si->mode = StartInfo::LOAD_GAME;
  441. if(si->campState)
  442. campaignMap = si->campState->currentMap.get();
  443. for(auto & ps : si->playerInfos)
  444. {
  445. if(!ps.second.compOnly && ps.second.connectedPlayerIDs.size() && namesIt != playerNames.cend())
  446. {
  447. setPlayerConnectedId(ps.second, namesIt++->first);
  448. }
  449. else
  450. {
  451. setPlayerConnectedId(ps.second, PlayerSettings::PLAYER_AI);
  452. }
  453. }
  454. }
  455. else if(si->mode == StartInfo::NEW_GAME || si->mode == StartInfo::CAMPAIGN)
  456. {
  457. if(mi->campaignHeader)
  458. return;
  459. for(int i = 0; i < mi->mapHeader->players.size(); i++)
  460. {
  461. const PlayerInfo & pinfo = mi->mapHeader->players[i];
  462. //neither computer nor human can play - no player
  463. if(!(pinfo.canHumanPlay || pinfo.canComputerPlay))
  464. continue;
  465. PlayerSettings & pset = si->playerInfos[PlayerColor(i)];
  466. pset.color = PlayerColor(i);
  467. if(pinfo.canHumanPlay && namesIt != playerNames.cend())
  468. {
  469. setPlayerConnectedId(pset, namesIt++->first);
  470. }
  471. else
  472. {
  473. setPlayerConnectedId(pset, PlayerSettings::PLAYER_AI);
  474. if(!pinfo.canHumanPlay)
  475. {
  476. pset.compOnly = true;
  477. }
  478. }
  479. pset.castle = pinfo.defaultCastle();
  480. pset.hero = pinfo.defaultHero();
  481. if(pset.hero != PlayerSettings::RANDOM && pinfo.hasCustomMainHero())
  482. {
  483. pset.hero = pinfo.mainCustomHeroId;
  484. pset.heroName = pinfo.mainCustomHeroName;
  485. pset.heroPortrait = pinfo.mainCustomHeroPortrait;
  486. }
  487. pset.handicap = PlayerSettings::NO_HANDICAP;
  488. }
  489. if(mi->isRandomMap && mapGenOpts)
  490. si->mapGenOptions = std::shared_ptr<CMapGenOptions>(mapGenOpts);
  491. else
  492. si->mapGenOptions.reset();
  493. }
  494. si->mapname = mi->fileURI;
  495. }
  496. void CVCMIServer::updateAndPropagateLobbyState()
  497. {
  498. boost::unique_lock<boost::mutex> stateLock(stateMutex);
  499. // Update player settings for RMG
  500. // TODO: find appropriate location for this code
  501. if(si->mapGenOptions && si->mode == StartInfo::NEW_GAME)
  502. {
  503. for(const auto & psetPair : si->playerInfos)
  504. {
  505. const auto & pset = psetPair.second;
  506. si->mapGenOptions->setStartingTownForPlayer(pset.color, pset.castle);
  507. if(pset.isControlledByHuman())
  508. {
  509. si->mapGenOptions->setPlayerTypeForStandardPlayer(pset.color, EPlayerType::HUMAN);
  510. }
  511. }
  512. }
  513. auto lus = vstd::make_unique<LobbyUpdateState>();
  514. lus->state = *this;
  515. addToAnnounceQueue(std::move(lus));
  516. }
  517. void CVCMIServer::setPlayer(PlayerColor clickedColor)
  518. {
  519. struct PlayerToRestore
  520. {
  521. PlayerColor color;
  522. int id;
  523. void reset() { id = -1; color = PlayerColor::CANNOT_DETERMINE; }
  524. PlayerToRestore(){ reset(); }
  525. } playerToRestore;
  526. PlayerSettings & clicked = si->playerInfos[clickedColor];
  527. //identify clicked player
  528. int clickedNameID = 0; //number of player - zero means AI, assume it initially
  529. if(clicked.isControlledByHuman())
  530. clickedNameID = *(clicked.connectedPlayerIDs.begin()); //if not AI - set appropiate ID
  531. if(clickedNameID > 0 && playerToRestore.id == clickedNameID) //player to restore is about to being replaced -> put him back to the old place
  532. {
  533. PlayerSettings & restPos = si->playerInfos[playerToRestore.color];
  534. setPlayerConnectedId(restPos, playerToRestore.id);
  535. playerToRestore.reset();
  536. }
  537. int newPlayer; //which player will take clicked position
  538. //who will be put here?
  539. if(!clickedNameID) //AI player clicked -> if possible replace computer with unallocated player
  540. {
  541. newPlayer = getIdOfFirstUnallocatedPlayer();
  542. if(!newPlayer) //no "free" player -> get just first one
  543. newPlayer = playerNames.begin()->first;
  544. }
  545. else //human clicked -> take next
  546. {
  547. auto i = playerNames.find(clickedNameID); //clicked one
  548. i++; //player AFTER clicked one
  549. if(i != playerNames.end())
  550. newPlayer = i->first;
  551. else
  552. newPlayer = 0; //AI if we scrolled through all players
  553. }
  554. setPlayerConnectedId(clicked, newPlayer); //put player
  555. //if that player was somewhere else, we need to replace him with computer
  556. if(newPlayer) //not AI
  557. {
  558. for(auto i = si->playerInfos.begin(); i != si->playerInfos.end(); i++)
  559. {
  560. int curNameID = *(i->second.connectedPlayerIDs.begin());
  561. if(i->first != clickedColor && curNameID == newPlayer)
  562. {
  563. assert(i->second.connectedPlayerIDs.size());
  564. playerToRestore.color = i->first;
  565. playerToRestore.id = newPlayer;
  566. setPlayerConnectedId(i->second, PlayerSettings::PLAYER_AI); //set computer
  567. break;
  568. }
  569. }
  570. }
  571. }
  572. void CVCMIServer::optionNextCastle(PlayerColor player, int dir)
  573. {
  574. PlayerSettings & s = si->playerInfos[player];
  575. si16 & cur = s.castle;
  576. auto & allowed = getPlayerInfo(player.getNum()).allowedFactions;
  577. const bool allowRandomTown = getPlayerInfo(player.getNum()).isFactionRandom;
  578. if(cur == PlayerSettings::NONE) //no change
  579. return;
  580. if(cur == PlayerSettings::RANDOM) //first/last available
  581. {
  582. if(dir > 0)
  583. cur = *allowed.begin(); //id of first town
  584. else
  585. cur = *allowed.rbegin(); //id of last town
  586. }
  587. else // next/previous available
  588. {
  589. if((cur == *allowed.begin() && dir < 0) || (cur == *allowed.rbegin() && dir > 0))
  590. {
  591. if(allowRandomTown)
  592. {
  593. cur = PlayerSettings::RANDOM;
  594. }
  595. else
  596. {
  597. if(dir > 0)
  598. cur = *allowed.begin();
  599. else
  600. cur = *allowed.rbegin();
  601. }
  602. }
  603. else
  604. {
  605. assert(dir >= -1 && dir <= 1); //othervice std::advance may go out of range
  606. auto iter = allowed.find((ui8)cur);
  607. std::advance(iter, dir);
  608. cur = *iter;
  609. }
  610. }
  611. if(s.hero >= 0 && !getPlayerInfo(player.getNum()).hasCustomMainHero()) // remove hero unless it set to fixed one in map editor
  612. {
  613. s.hero = PlayerSettings::RANDOM;
  614. }
  615. if(cur < 0 && s.bonus == PlayerSettings::RESOURCE)
  616. s.bonus = PlayerSettings::RANDOM;
  617. }
  618. void CVCMIServer::setCampaignMap(int mapId)
  619. {
  620. campaignMap = mapId;
  621. si->difficulty = si->campState->camp->scenarios[mapId].difficulty;
  622. campaignBonus = -1;
  623. updateStartInfoOnMapChange(si->campState->getMapInfo(mapId));
  624. }
  625. void CVCMIServer::setCampaignBonus(int bonusId)
  626. {
  627. campaignBonus = bonusId;
  628. const CCampaignScenario & scenario = si->campState->camp->scenarios[campaignMap];
  629. const std::vector<CScenarioTravel::STravelBonus> & bonDescs = scenario.travelOptions.bonusesToChoose;
  630. if(bonDescs[bonusId].type == CScenarioTravel::STravelBonus::HERO)
  631. {
  632. for(auto & elem : si->playerInfos)
  633. {
  634. if(elem.first == PlayerColor(bonDescs[bonusId].info1))
  635. setPlayerConnectedId(elem.second, 1);
  636. else
  637. setPlayerConnectedId(elem.second, PlayerSettings::PLAYER_AI);
  638. }
  639. }
  640. }
  641. void CVCMIServer::optionNextHero(PlayerColor player, int dir)
  642. {
  643. PlayerSettings & s = si->playerInfos[player];
  644. if(s.castle < 0 || s.hero == PlayerSettings::NONE)
  645. return;
  646. if(s.hero == PlayerSettings::RANDOM) // first/last available
  647. {
  648. int max = static_cast<int>(VLC->heroh->size()),
  649. min = 0;
  650. s.hero = nextAllowedHero(player, min, max, 0, dir);
  651. }
  652. else
  653. {
  654. if(dir > 0)
  655. s.hero = nextAllowedHero(player, s.hero, (int)VLC->heroh->size(), 1, dir);
  656. else
  657. s.hero = nextAllowedHero(player, -1, s.hero, 1, dir); // min needs to be -1 -- hero at index 0 would be skipped otherwise
  658. }
  659. }
  660. int CVCMIServer::nextAllowedHero(PlayerColor player, int min, int max, int incl, int dir)
  661. {
  662. if(dir > 0)
  663. {
  664. for(int i = min + incl; i <= max - incl; i++)
  665. if(canUseThisHero(player, i))
  666. return i;
  667. }
  668. else
  669. {
  670. for(int i = max - incl; i >= min + incl; i--)
  671. if(canUseThisHero(player, i))
  672. return i;
  673. }
  674. return -1;
  675. }
  676. void CVCMIServer::optionNextBonus(PlayerColor player, int dir)
  677. {
  678. PlayerSettings & s = si->playerInfos[player];
  679. PlayerSettings::Ebonus & ret = s.bonus = static_cast<PlayerSettings::Ebonus>(static_cast<int>(s.bonus) + dir);
  680. if(s.hero == PlayerSettings::NONE &&
  681. !getPlayerInfo(player.getNum()).heroesNames.size() &&
  682. ret == PlayerSettings::ARTIFACT) //no hero - can't be artifact
  683. {
  684. if(dir < 0)
  685. ret = PlayerSettings::RANDOM;
  686. else
  687. ret = PlayerSettings::GOLD;
  688. }
  689. if(ret > PlayerSettings::RESOURCE)
  690. ret = PlayerSettings::RANDOM;
  691. if(ret < PlayerSettings::RANDOM)
  692. ret = PlayerSettings::RESOURCE;
  693. if(s.castle == PlayerSettings::RANDOM && ret == PlayerSettings::RESOURCE) //random castle - can't be resource
  694. {
  695. if(dir < 0)
  696. ret = PlayerSettings::GOLD;
  697. else
  698. ret = PlayerSettings::RANDOM;
  699. }
  700. }
  701. bool CVCMIServer::canUseThisHero(PlayerColor player, int ID)
  702. {
  703. return VLC->heroh->size() > ID
  704. && si->playerInfos[player].castle == VLC->heroh->objects[ID]->heroClass->faction
  705. && !vstd::contains(getUsedHeroes(), ID)
  706. && mi->mapHeader->allowedHeroes[ID];
  707. }
  708. std::vector<int> CVCMIServer::getUsedHeroes()
  709. {
  710. std::vector<int> heroIds;
  711. for(auto & p : si->playerInfos)
  712. {
  713. const auto & heroes = getPlayerInfo(p.first.getNum()).heroesNames;
  714. for(auto & hero : heroes)
  715. if(hero.heroId >= 0) //in VCMI map format heroId = -1 means random hero
  716. heroIds.push_back(hero.heroId);
  717. if(p.second.hero != PlayerSettings::RANDOM)
  718. heroIds.push_back(p.second.hero);
  719. }
  720. return heroIds;
  721. }
  722. ui8 CVCMIServer::getIdOfFirstUnallocatedPlayer() const
  723. {
  724. for(auto i = playerNames.cbegin(); i != playerNames.cend(); i++)
  725. {
  726. if(!si->getPlayersSettings(i->first))
  727. return i->first;
  728. }
  729. return 0;
  730. }
  731. #if defined(__GNUC__) && !defined(__MINGW32__) && !defined(VCMI_ANDROID) && !defined(VCMI_IOS)
  732. void handleLinuxSignal(int sig)
  733. {
  734. const int STACKTRACE_SIZE = 100;
  735. void * buffer[STACKTRACE_SIZE];
  736. int ptrCount = backtrace(buffer, STACKTRACE_SIZE);
  737. char * * strings;
  738. logGlobal->error("Error: signal %d :", sig);
  739. strings = backtrace_symbols(buffer, ptrCount);
  740. if(strings == nullptr)
  741. {
  742. logGlobal->error("There are no symbols.");
  743. }
  744. else
  745. {
  746. for(int i = 0; i < ptrCount; ++i)
  747. {
  748. logGlobal->error(strings[i]);
  749. }
  750. free(strings);
  751. }
  752. _exit(EXIT_FAILURE);
  753. }
  754. #endif
  755. static void handleCommandOptions(int argc, char * argv[], boost::program_options::variables_map & options)
  756. {
  757. namespace po = boost::program_options;
  758. po::options_description opts("Allowed options");
  759. opts.add_options()
  760. ("help,h", "display help and exit")
  761. ("version,v", "display version information and exit")
  762. ("run-by-client", "indicate that server launched by client on same machine")
  763. ("uuid", po::value<std::string>(), "")
  764. ("enable-shm-uuid", "use UUID for shared memory identifier")
  765. ("enable-shm", "enable usage of shared memory")
  766. ("port", po::value<ui16>(), "port at which server will listen to connections from client");
  767. if(argc > 1)
  768. {
  769. try
  770. {
  771. po::store(po::parse_command_line(argc, argv, opts), options);
  772. }
  773. catch(std::exception & e)
  774. {
  775. std::cerr << "Failure during parsing command-line options:\n" << e.what() << std::endl;
  776. }
  777. }
  778. po::notify(options);
  779. if(options.count("help"))
  780. {
  781. auto time = std::time(0);
  782. printf("%s - A Heroes of Might and Magic 3 clone\n", GameConstants::VCMI_VERSION.c_str());
  783. printf("Copyright (C) 2007-%d VCMI dev team - see AUTHORS file\n", std::localtime(&time)->tm_year + 1900);
  784. printf("This is free software; see the source for copying conditions. There is NO\n");
  785. printf("warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n");
  786. printf("\n");
  787. std::cout << opts;
  788. exit(0);
  789. }
  790. if(options.count("version"))
  791. {
  792. printf("%s\n", GameConstants::VCMI_VERSION.c_str());
  793. std::cout << VCMIDirs::get().genHelpString();
  794. exit(0);
  795. }
  796. }
  797. #ifdef VCMI_IOS
  798. #define main server_main
  799. #endif
  800. int main(int argc, char * argv[])
  801. {
  802. #if !defined(VCMI_ANDROID) && !defined(VCMI_IOS)
  803. // Correct working dir executable folder (not bundle folder) so we can use executable relative paths
  804. boost::filesystem::current_path(boost::filesystem::system_complete(argv[0]).parent_path());
  805. #endif
  806. // Installs a sig sev segmentation violation handler
  807. // to log stacktrace
  808. #if defined(__GNUC__) && !defined(__MINGW32__) && !defined(VCMI_ANDROID) && !defined(VCMI_IOS)
  809. signal(SIGSEGV, handleLinuxSignal);
  810. #endif
  811. // todo ios: double console log
  812. console = new CConsoleHandler();
  813. CBasicLogConfigurator logConfig(VCMIDirs::get().userLogsPath() / "VCMI_Server_log.txt", console);
  814. logConfig.configureDefault();
  815. logGlobal->info(SERVER_NAME);
  816. boost::program_options::variables_map opts;
  817. #ifdef VCMI_IOS
  818. argc = 1;
  819. boost::condition_variable * cond = reinterpret_cast<boost::condition_variable *>(argv[1]);
  820. cond->notify_one();
  821. //#endif
  822. #else
  823. handleCommandOptions(argc, argv, opts);
  824. preinitDLL(console);
  825. settings.init();
  826. logConfig.configure();
  827. loadDLLClasses();
  828. srand((ui32)time(nullptr));
  829. //#ifdef VCMI_IOS
  830. #endif
  831. try
  832. {
  833. boost::asio::io_service io_service;
  834. CVCMIServer server(opts);
  835. try
  836. {
  837. while(server.state != EServerState::SHUTDOWN)
  838. {
  839. server.run();
  840. }
  841. io_service.run();
  842. }
  843. catch(boost::system::system_error & e) //for boost errors just log, not crash - probably client shut down connection
  844. {
  845. logNetwork->error(e.what());
  846. server.state = EServerState::SHUTDOWN;
  847. }
  848. catch(...)
  849. {
  850. handleException();
  851. }
  852. }
  853. catch(boost::system::system_error & e)
  854. {
  855. logNetwork->error(e.what());
  856. //catch any startup errors (e.g. can't access port) errors
  857. //and return non-zero status so client can detect error
  858. throw;
  859. }
  860. #ifdef VCMI_ANDROID
  861. CAndroidVMHelper envHelper;
  862. envHelper.callStaticVoidMethod(CAndroidVMHelper::NATIVE_METHODS_DEFAULT_CLASS, "killServer");
  863. #endif
  864. logConfig.deconfigure();
  865. vstd::clear_pointer(VLC);
  866. return 0;
  867. }
  868. #ifdef VCMI_ANDROID
  869. void CVCMIServer::create()
  870. {
  871. const char * foo[1] = {"android-server"};
  872. main(1, const_cast<char **>(foo));
  873. }
  874. #elif defined(VCMI_IOS)
  875. void CVCMIServer::create(boost::condition_variable * cond)
  876. {
  877. const auto executablePath = VCMIDirs::get().serverPath();
  878. void *argv[] = {const_cast<char *>(executablePath.c_str()), cond};
  879. main(2, reinterpret_cast<char **>(argv));
  880. }
  881. #endif