CWebsocketServer.cpp 69 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827
  1. #include "stdafx.h"
  2. #if (defined _WIN32 || defined _WIN64)
  3. #else
  4. #include <future>
  5. #include <thread>
  6. #endif
  7. #include "CWebsocketServer.h"
  8. #include "baseEx.h"
  9. #include "CModTools.h"
  10. #include <vector>
  11. #include "MessageType.h"
  12. #include <boost/thread/thread.hpp>
  13. #include <boost/thread/lock_guard.hpp>
  14. #include <iostream>
  15. #include <locale>
  16. #include <codecvt>
  17. #include "../mod_guiconsole/GUIConsole_msg_g.h"
  18. #define DEFAULT_SERVER_PORT 9002
  19. #define DEFAULT_SERVER_WS_SM2_PORT 9003
  20. #define DEFUALT_SERVER_WSS_PORT 9004
  21. namespace Chromium {
  22. server m_wsserver, m_ws_sm2_server;
  23. server_wss m_server_wss;
  24. CWebsocketServer::CWebsocketServer(const char* strPath, CEntityBase* pEntity)
  25. : m_ios(), m_serializer(NULL), m_socket(NULL), m_pEntity(pEntity), m_initSuccess(false)
  26. {
  27. this->m_pEntity = pEntity;
  28. m_socket = new CSocketClient(m_ios, "127.0.0.1", "4504", pEntity, 0);
  29. m_socket->SetMessageHandler(this);
  30. // Initialize serializer
  31. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Initialize serializer");
  32. this->m_serializer = CWSCodec::getInstance();
  33. this->m_serializer->setEntityHandler(pEntity->GetFunction());
  34. this->m_serializer->init(strPath);
  35. while (Error_Succeed != m_socket->Connect())
  36. Sleep(100);
  37. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("init Entity Session Manager");
  38. myTest();
  39. try {
  40. m_esm = new EntitySessionManager();
  41. }
  42. catch (const std::bad_alloc& e)
  43. {
  44. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer::CWebsocketServer Memory allocation failed: %s", e.what());
  45. }
  46. catch (const std::exception& e)
  47. {
  48. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer::CWebsocketServer Error: %s", e.what());
  49. }
  50. init_websocket();
  51. init_entity_sessions();
  52. }
  53. void CWebsocketServer::updateMsgPool(std::string entityName, std::string& payload, websocketpp::connection_hdl hdl)
  54. {
  55. //存储的消息应具备时效性,保证一个hdl只有一个记录的消息
  56. auto it = ConfigManager::getInstance().m_msg_pool.find(entityName);
  57. if (it != ConfigManager::getInstance().m_msg_pool.end())
  58. {
  59. auto& curArr = it->second;
  60. for (auto i = curArr.begin(); i != curArr.end(); i++)
  61. {
  62. if (&hdl == &i->first)
  63. {
  64. curArr.erase(i);
  65. break;
  66. }
  67. }
  68. it->second.emplace_back(std::make_pair(hdl, payload));
  69. }
  70. else
  71. {
  72. std::vector<std::pair<websocketpp::connection_hdl, std::string>> t_saveMsgs;
  73. t_saveMsgs.emplace_back(std::make_pair(hdl, payload));
  74. ConfigManager::getInstance().m_msg_pool.insert(std::make_pair(entityName, t_saveMsgs));
  75. }
  76. }
  77. void CWebsocketServer::updateNotifyPool(unsigned hdl, unsigned transId)
  78. {
  79. auto it = ConfigManager::getInstance().m_notifyPool.find(hdl);
  80. if (it != ConfigManager::getInstance().m_notifyPool.end())
  81. {
  82. if (transId == 0)
  83. ConfigManager::getInstance().m_notifyPool.erase(it);
  84. else
  85. it->second = transId;
  86. }
  87. else if (transId != 0)
  88. ConfigManager::getInstance().m_notifyPool.insert(std::make_pair(hdl, transId));
  89. }
  90. void CWebsocketServer::storeEntityWithCLass(std::string entityName, std::string entityClass)
  91. {
  92. if (ConfigManager::getInstance().m_entityAndClass.end() == ConfigManager::getInstance().m_entityAndClass.find(entityName)) {
  93. ConfigManager::getInstance().m_entityAndClass.insert(std::make_pair(entityName, entityClass)); //not exist
  94. //DbgEx("storeEntityWithCLass, %s:%s", entityName.c_str(), entityClass.c_str());
  95. }
  96. }
  97. std::pair<bool, std::string> CWebsocketServer::getEntityClass(std::string entityName)
  98. {
  99. auto it = ConfigManager::getInstance().m_entityAndClass.find(entityName);
  100. if (ConfigManager::getInstance().m_entityAndClass.end() == it)
  101. return std::make_pair(false, "");
  102. else
  103. return std::make_pair(true, it->second);
  104. }
  105. void CWebsocketServer::do_sendJsonStartSession(std::string entityName, std::string entityClass)
  106. {
  107. auto startSessionReq = m_esm->GetStartSessionRequest(entityName, entityClass);
  108. ws_msgInfo cur(startSessionReq.second, 0, 0);
  109. auto sessionBuf = this->m_serializer->JsonToBuffer(startSessionReq.second, cur).second;
  110. if (nullptr == sessionBuf)
  111. {
  112. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("JsonToBuffer return NULL !");
  113. return;
  114. }
  115. m_esm->StoreSessionReq(startSessionReq.first, entityName);
  116. m_esm->MakeNewTransID(sessionBuf, 0);
  117. //保存msg,先建立session
  118. if (m_esm->checkBeginSession(entityName)) {
  119. m_esm->updateBeginSessionTime(entityName);
  120. WriteToFramework(sessionBuf);
  121. LogManager::getInstance().logWebSocketBeginSession(-1, -1, cur.transId, cur.entityName, cur.className, 0, "auto_send", cur.payLoad);
  122. }
  123. else
  124. {
  125. LogManager::getInstance().logWebSocketBeginSession(-1, -1, cur.transId, cur.entityName, cur.className, 1, "none", cur.payLoad,
  126. Error_Duplication, "already begin session in 5 seconds");
  127. }
  128. }
  129. void CWebsocketServer::WriteToFramework(CMessage* msg)
  130. {
  131. m_socket->Write(msg);
  132. }
  133. void CWebsocketServer::deal_webchromium_msg(std::string& payload, websocketpp::connection_hdl hdl, int messageType)
  134. {
  135. switch (messageType)
  136. {
  137. case RegisterNotify:
  138. {
  139. std::shared_ptr<cJSON> pJson(cJSON_Parse(payload.c_str()), [](cJSON* p) {
  140. if (nullptr != p)
  141. cJSON_Delete(p);
  142. });
  143. const char* transId_nodeName = "transId";
  144. auto transid = json_deal::getIntergerFromCjsonObj(pJson.get(), transId_nodeName);
  145. if (transid.first == false)
  146. {
  147. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("deal_webchromium_msg err, transIdJson == null");
  148. return;
  149. }
  150. updateNotifyPool((long)hdl.lock().get(), transid.second);
  151. }
  152. break;
  153. case UnRegisterNotify:
  154. updateNotifyPool((long)hdl.lock().get(), 0);
  155. break;
  156. default:
  157. break;
  158. }
  159. }
  160. void CWebsocketServer::deal_logMsg(std::string& payload, websocketpp::connection_hdl hdl, int messageType)
  161. {
  162. std::shared_ptr<cJSON> pJson(cJSON_Parse(payload.c_str()), [](cJSON* p) {
  163. if (nullptr != p)
  164. cJSON_Delete(p);
  165. });
  166. if (nullptr == pJson.get()) {
  167. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer -> deal_logMsg, jsonErr:%s", payload.c_str());
  168. return;
  169. }
  170. auto changeMessageTypeToLogLevel = [](int messageType) ->LOG_LEVEL_E {
  171. switch (messageType)
  172. {
  173. case METHOD_SYSTEM_LOG_DEBUG:
  174. case METHOD_SYSTEM_DETAIL_LOG_DEBUG:
  175. return LOG_LEVEL_DEBUG;
  176. case METHOD_SYSTEM_LOG_INFO:
  177. case METHOD_SYSTEM_DETAIL_LOG_INFO:
  178. return LOG_LEVEL_INFO;
  179. case METHOD_SYSTEM_LOG_WARN:
  180. case METHOD_SYSTEM_DETAIL_LOG_WARN:
  181. return LOG_LEVEL_WARN;
  182. case METHOD_SYSTEM_LOG_ERROR:
  183. case METHOD_SYSTEM_DETAIL_LOG_ERROR:
  184. return LOG_LEVEL_ERROR;
  185. default:
  186. return LOG_LEVEL_DEBUG;
  187. }
  188. };
  189. switch (messageType)
  190. {
  191. case METHOD_SYSTEM_LOG_DEBUG:
  192. case METHOD_SYSTEM_LOG_INFO:
  193. case METHOD_SYSTEM_LOG_WARN:
  194. case METHOD_SYSTEM_LOG_ERROR:
  195. {
  196. const char* msg_nodeName = "msg";
  197. auto msg = json_deal::getStringFromCjsonObj(pJson.get(), msg_nodeName);
  198. if (!msg.first)
  199. {
  200. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do not has node %s", msg_nodeName);
  201. break;
  202. }
  203. DbgWithLink(changeMessageTypeToLogLevel(messageType), LOG_TYPE_BUSINESS_SYSTEM)
  204. .withLogProducer(ConfigManager::getInstance().getLogProducer()).setAPI("logMsg").withExtendLog(false).setResultMsg(msg.second.GetData())();
  205. break;
  206. }
  207. case METHOD_SYSTEM_DETAIL_LOG_DEBUG:
  208. case METHOD_SYSTEM_DETAIL_LOG_INFO:
  209. case METHOD_SYSTEM_DETAIL_LOG_WARN:
  210. case METHOD_SYSTEM_DETAIL_LOG_ERROR:
  211. {
  212. const char* msg_nodeName = "ResultMsg";
  213. const char* msg_EntityName = "EntityName";
  214. const char* msg_logType = "LogType";
  215. const char* msg_CostTime = "CostTime";
  216. const char* msg_ResultCode = "ResultCode";
  217. const char* msg_LogCode = "LogCode";
  218. const char* msg_API = "API";
  219. const char* msg_SourceType = "SourceType";
  220. const char* msg_BussID = "BussID";
  221. const char* msg_TipMsg = "TipMsg";
  222. const char* msg_BeginTime = "BeginTime";
  223. const char* msg_EndTime = "EndTime";
  224. const char* msg_CmptId = "CmptId";
  225. const char* msg_CmptName = "CmptName";
  226. auto msg = json_deal::getStringFromCjsonObj(pJson.get(), msg_nodeName);
  227. if (!msg.first)//must have
  228. {
  229. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do not has node %s", msg_nodeName);
  230. break;
  231. }
  232. auto t_entityName = json_deal::getStringFromCjsonObj(pJson.get(), msg_EntityName);
  233. if(!t_entityName.first)
  234. {
  235. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do not has node %s", msg_EntityName);
  236. break;
  237. }
  238. //create the logProduer if not exist
  239. auto &t_logProducerArr = ConfigManager::getInstance().getLogProducerArr();
  240. if (t_logProducerArr.find(t_entityName.second.GetData()) == t_logProducerArr.end())//can not find the logProducer
  241. {
  242. auto t_CmptId = json_deal::getStringFromCjsonObj(pJson.get(), msg_CmptId);
  243. auto t_CmptName = json_deal::getStringFromCjsonObj(pJson.get(), msg_CmptName);
  244. void *t_producer = create_log_producer_storage(t_entityName.second.GetData(), "0", "", t_CmptId.second.GetData(), t_CmptName.second.GetData());
  245. if (!t_producer)
  246. {
  247. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("create producer failed, %s", t_entityName.second.GetData());
  248. break;
  249. }
  250. else
  251. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("create producer sueccess, entityName:%s, CmptId:%s, t_CmptName:%s"
  252. , t_entityName.second.GetData(), t_CmptId.second.GetData(), t_CmptName.second.GetData());
  253. t_logProducerArr.insert(std::make_pair(t_entityName.second.GetData(), t_producer));
  254. }
  255. auto t_logProducer = t_logProducerArr[t_entityName.second.GetData()];
  256. auto logType = json_deal::getStringFromCjsonObj(pJson.get(), msg_logType);//it can be equal to Sys/User/VTMWeb
  257. LOG_TYPE_E t_type = LOG_TYPE_BUSINESS_SYSTEM;
  258. if (logType.second.Compare("User", true) == 0)
  259. t_type = LOG_TYPE_BUSINESS_USER;
  260. else if (logType.second.Compare("VTMWeb", true) == 0)
  261. t_type = LOG_TYPE_VTMWEB;
  262. else if (logType.second.Compare("Sys", true) == 0)
  263. t_type = LOG_TYPE_BUSINESS_SYSTEM;
  264. else if (logType.second.Compare("VTMSys", true) == 0)
  265. t_type = LOG_TYPE_SYSTEM;
  266. else if (logType.second.Compare("VTMUser", true) == 0)
  267. t_type = LOG_TYPE_USER;
  268. //通过变量方式进行处理
  269. DbgWithLink t_obj(changeMessageTypeToLogLevel(messageType), t_type);
  270. t_obj.withExtendLog(false);
  271. t_obj.withLogProducer(t_logProducer);
  272. t_obj.setResultMsg(msg.second.GetData());
  273. auto t_costTime = json_deal::getIntergerFromCjsonObj(pJson.get(), msg_CostTime);
  274. if (t_costTime.first)
  275. t_obj.setCostTime(t_costTime.second);
  276. auto t_resultCode = json_deal::getStringFromCjsonObj(pJson.get(), msg_ResultCode);
  277. if (t_resultCode.first)
  278. t_obj.setResultCode(t_resultCode.second.GetData());
  279. auto t_logcode = json_deal::getStringFromCjsonObj(pJson.get(), msg_LogCode);
  280. if (t_logcode.first)
  281. t_obj.setLogCode(t_logcode.second.GetData());
  282. auto t_api = json_deal::getStringFromCjsonObj(pJson.get(), msg_API);
  283. if (t_api.first)
  284. t_obj.setAPI(t_api.second.GetData());
  285. auto t_sourceType = json_deal::getStringFromCjsonObj(pJson.get(), msg_SourceType);
  286. if (t_sourceType.first)
  287. t_obj.setSourceType(t_sourceType.second.GetData());
  288. auto t_BussID = json_deal::getStringFromCjsonObj(pJson.get(), msg_BussID);
  289. if (t_BussID.first)
  290. t_obj.setBussID(t_BussID.second.GetData());
  291. auto t_TipMsg = json_deal::getStringFromCjsonObj(pJson.get(), msg_TipMsg);
  292. if (t_TipMsg.first)
  293. t_obj.setTipMsg(t_TipMsg.second.GetData());
  294. auto t_BeginTime = json_deal::getIntergerFromCjsonObj(pJson.get(), msg_BeginTime);
  295. if (t_BeginTime.first)
  296. t_obj.setBeginTime(t_BeginTime.second);
  297. auto t_EndTime = json_deal::getIntergerFromCjsonObj(pJson.get(), msg_EndTime);
  298. if (t_EndTime.first)
  299. t_obj.setEndTime(t_EndTime.second);
  300. t_obj();
  301. break;
  302. }
  303. case METHOD_BEIDOU_LOG:
  304. {
  305. //DbgEx("receive beidou msg:%s", payload.c_str());
  306. const char* BusinessId_nodeName = "BusinessId";
  307. const char* TraceId_nodeName = "TraceId";
  308. const char* SpanId_nodeName = "SpanId";
  309. const char* ParentSpanId_nodeName = "ParentSpanId";
  310. const char* Timestamp_nodeName = "Timestamp";
  311. const char* Host_nodeName = "Host";
  312. const char* CallStack_nodeName = "CallStack";
  313. const char* DbStack_nodeName = "DbStack";
  314. const char* ReturnCode_nodeName = "ReturnCode";
  315. const char* Tags_nodeName = "Tags";
  316. const char* API_nodeName = "API";
  317. const char* deployUnitId_nodeName = "deployUnitId";
  318. const char* serviceUintId_nodeName = "serviceUintId";
  319. auto BusinessId = json_deal::getStringFromCjsonObj(pJson.get(), BusinessId_nodeName);
  320. if (!BusinessId.first)
  321. {
  322. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do not has node %s", BusinessId_nodeName);
  323. break;
  324. }
  325. auto TraceId = json_deal::getStringFromCjsonObj(pJson.get(), TraceId_nodeName);
  326. if (!TraceId.first)
  327. {
  328. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do not has node %s", TraceId_nodeName);
  329. break;
  330. }
  331. auto SpanId = json_deal::getStringFromCjsonObj(pJson.get(), SpanId_nodeName);
  332. if (!SpanId.first)
  333. {
  334. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do not has node %s", SpanId_nodeName);
  335. break;
  336. }
  337. auto ParentSpanId = json_deal::getStringFromCjsonObj(pJson.get(), ParentSpanId_nodeName);
  338. if (!ParentSpanId.first)
  339. {
  340. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do not has node %s", ParentSpanId_nodeName);
  341. break;
  342. }
  343. linkContext cur(BusinessId.second.GetData(), TraceId.second.GetData(), SpanId.second.GetData(), ParentSpanId.second.GetData());
  344. auto API = json_deal::getStringFromCjsonObj(pJson.get(), API_nodeName);
  345. if (!API.first)
  346. API.second = "";
  347. DbgToBeidou obj(cur, API.second.GetData());
  348. auto Timestamp = json_deal::getStringFromCjsonObj(pJson.get(), Timestamp_nodeName);
  349. if (!Timestamp.first)
  350. Timestamp.second = "";
  351. else if(Timestamp.second.GetLength() > 0)
  352. obj.setResponseTime(Timestamp.second);
  353. auto Host = json_deal::getStringFromCjsonObj(pJson.get(), Host_nodeName);
  354. if (!Host.first)
  355. Host.second = "";
  356. else if (Host.second.GetLength() > 0)
  357. obj.setHost(Host.second);
  358. auto CallStack = json_deal::getStringFromCjsonObj(pJson.get(), CallStack_nodeName);
  359. if (!CallStack.first)
  360. CallStack.second = "";
  361. else if (CallStack.second.GetLength() > 0)
  362. obj.setCallStack(CallStack.second);
  363. auto DbStack = json_deal::getStringFromCjsonObj(pJson.get(), DbStack_nodeName);
  364. if (!DbStack.first)
  365. DbStack.second = "";
  366. else if (DbStack.second.GetLength() > 0)
  367. obj.setCallStack(DbStack.second);
  368. auto ReturnCode = json_deal::getStringFromCjsonObj(pJson.get(), ReturnCode_nodeName);
  369. if (!ReturnCode.first)
  370. ReturnCode.second = "";
  371. else if (ReturnCode.second.GetLength() > 0)
  372. obj.setReturnCode(ReturnCode.second);
  373. auto Tags = json_deal::getStringFromCjsonObj(pJson.get(), Tags_nodeName);
  374. if (!Tags.first)
  375. Tags.second = "";
  376. else if (Tags.second.GetLength() > 0)
  377. obj.setTags(Tags.second);
  378. auto deployUnitId = json_deal::getStringFromCjsonObj(pJson.get(), deployUnitId_nodeName);
  379. if (!deployUnitId.first)
  380. deployUnitId.second = "";
  381. else if (deployUnitId.second.GetLength() > 0)
  382. obj.set_deployUnitId(deployUnitId.second);
  383. auto serviceUintId = json_deal::getStringFromCjsonObj(pJson.get(), serviceUintId_nodeName);
  384. if (!serviceUintId.first)
  385. serviceUintId.second = "";
  386. else if (serviceUintId.second.GetLength() > 0)
  387. obj.set_serviceUnitId(serviceUintId.second);
  388. obj();
  389. break;
  390. }
  391. default:
  392. break;
  393. }
  394. }
  395. void CWebsocketServer::deal_sessionBreakMsg(std::string& payload, websocketpp::connection_hdl hdl)
  396. {
  397. std::shared_ptr<cJSON> pJson(cJSON_Parse(payload.c_str()), [](cJSON* p) {
  398. if (nullptr != p)
  399. cJSON_Delete(p);
  400. });
  401. if (nullptr == pJson.get()) {
  402. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer -> deal_sessionBreakMsg, jsonErr:%s", payload.c_str());
  403. return;
  404. }
  405. auto transidNum = json_deal::getIntergerFromCjsonObj(pJson.get(), "transId");
  406. if (transidNum.first == false)
  407. {
  408. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer::deal_sessionBreakMs get transId failed");
  409. return;
  410. }
  411. std::string sessionId = std::to_string((LONGLONG)-1);
  412. std::string transid = std::to_string((LONGLONG)transidNum.second);
  413. /*
  414. std::string js = "{\"messageType\":5,\"errorCode\":\"1537\",\"errorMsg\":\"session break\",\"sessionID\":";
  415. js.append(sessionId.c_str());
  416. js.append(",\"transID\":");
  417. js.append(transid.c_str());
  418. js.append("}");
  419. */
  420. std::string js = CSimpleString::Format("{\"messageType\":5,\"errorCode\":\"%d\",\"errorMsg\":\"1537 session break\",\"sessionID\":-1,\"transID\":125}", Error_DevNotAvailable).GetData();
  421. auto dst_js = restroreTransId(js);
  422. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("deal_sessionBreakMsg, len:%d, srcPayLoad:%s, ret:%s", payload.length()
  423. , payload.length() > 800 ? payload.substr(0, 800).append("...").c_str() : payload.c_str(), dst_js.second.c_str());
  424. do_send_msg(hdl, js);
  425. }
  426. void CWebsocketServer::deal_msg(std::string& payload, websocketpp::connection_hdl hdl, int srcTransId)
  427. {
  428. boost::lock_guard<boost::mutex> lock(m_dealMsgLock); //在buffer和json处理时,deal_msg会调用多次,导致transId存在重复可能
  429. ws_msgInfo cur(payload, 0, 0);
  430. cur.hdlId = (long)hdl.lock().get();
  431. cur.transId = srcTransId;
  432. auto ret = this->m_serializer->JsonToBuffer(payload, cur);
  433. CMessage* p = ret.second;
  434. //DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("deal_msg:%s", payload.c_str());
  435. if (ret.first == MessageType::Broadcast && nullptr == p)
  436. {
  437. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("JsonToBuffer return NULL, perhaps an event happened");
  438. do_sendJsonBroadcast(payload);
  439. return;
  440. }
  441. else if(ret.first == MessageType::Request || ret.first == MessageType::Info)
  442. {
  443. if(nullptr == p)
  444. {
  445. //can not find the method
  446. std::shared_ptr<cJSON> pJson(cJSON_Parse(payload.c_str()), [](cJSON* p) {
  447. if (nullptr != p)
  448. cJSON_Delete(p);
  449. });
  450. auto transIdJson = json_deal::getIntergerFromCjsonObj(pJson.get(), "transId");
  451. if (transIdJson.first)
  452. {
  453. std::string errRetJs = "{\"messageType\":4,\"transID\":" + std::to_string((LONGLONG)transIdJson.second) + ",\"errorCode\":103,\"errorMsg\":\"can not find entity or class!\"}";
  454. auto dst_js = restroreTransId(errRetJs);
  455. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("deal_msg %s get null msg, ret:%s", ret.first == Request ? "Request" : "Info", errRetJs.c_str());
  456. do_send_msg(hdl, dst_js.second);
  457. }
  458. else
  459. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer::deal_msg get transIdJson failed");
  460. return;
  461. }
  462. }
  463. else if (ret.first == MessageType::GetSession)
  464. {
  465. std::shared_ptr<cJSON> pJson(cJSON_Parse(payload.c_str()), [](cJSON* p) {
  466. if (nullptr != p)
  467. cJSON_Delete(p);
  468. });
  469. auto transIdJson = json_deal::getIntergerFromCjsonObj(pJson.get(), "transId");
  470. if (transIdJson.first)
  471. {
  472. int transid = transIdJson.second;
  473. CSystemRunInfo sysruninfo;
  474. m_pEntity->GetFunction()->GetSystemRunInfo(sysruninfo);
  475. auto sessionRet = m_esm->GetAllSessionRequest(transid, sysruninfo.bBasicCfgWork);
  476. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("sessionJson:%s", sessionRet.second.c_str());
  477. if (sessionRet.first)
  478. {
  479. auto dst_js = restroreTransId(sessionRet.second);
  480. do_send_msg(hdl, dst_js.second);
  481. }
  482. }
  483. else
  484. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer::deal_msg get transIdJson failed");
  485. return;
  486. }
  487. else if (ret.first == MessageType::WEB_GET_HTTP_HEADER)
  488. {
  489. CSystemStaticInfo staticInfo;
  490. m_pEntity->GetFunction()->GetSystemStaticInfo(staticInfo);
  491. std::string dst_js = "";
  492. if (staticInfo.strTerminalID.GetLength() == 0) // terminalno is empty
  493. dst_js = "";
  494. else
  495. dst_js = m_esm->Get_getconfig_ack(staticInfo.strTerminalID.GetData()).second;
  496. do_send_msg(hdl, dst_js);
  497. return;
  498. }
  499. else if (ret.first == MessageType::BeginSession)
  500. storeEntityWithCLass(m_serializer->GetEntityName(payload), m_serializer->GetClassName(payload));
  501. // Try to manage entity session
  502. RequestProcessType processType = this->m_esm->RequestProcess(p, m_serializer->GetEntityName(payload), (long)hdl.lock().get(), cur);
  503. switch (processType)
  504. {
  505. case Chromium::PROCESS_NOTHING:
  506. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do PROCESS_NOTHING");
  507. return;
  508. case Chromium::PROCESS_SEND:
  509. if (nullptr != p)
  510. WriteToFramework(p);
  511. break;
  512. case Chromium::PROCESS_STARTSESSION:
  513. {
  514. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do PROCESS_STARTSESSION");
  515. auto entityName = m_serializer->GetEntityName(payload);
  516. if (entityName == "Chromium")
  517. {
  518. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("discard, don't make session with chromium");
  519. break;
  520. }
  521. updateMsgPool(entityName, payload, hdl);
  522. do_sendJsonStartSession(entityName, m_serializer->GetClassName(payload));
  523. }
  524. break;
  525. case Chromium::PROCESS_FINDSESSION:
  526. {
  527. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do PROCESS_FINDSESSION");
  528. std::string js = m_esm->GetStartSessionAck(p, m_serializer->GetEntityName(payload));
  529. auto dst_js = restroreTransId(js);
  530. do_send_msg(hdl, dst_js.second);
  531. }
  532. break;
  533. case Chromium::PROCESS_RECORDMSG:
  534. {
  535. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do PROCESS_RECORDMSG");
  536. auto entityName = m_serializer->GetEntityName(payload);
  537. updateMsgPool(entityName, payload, hdl);
  538. }
  539. break;
  540. default:
  541. break;
  542. }
  543. }
  544. void CWebsocketServer::message_handler_ws_sm2(websocketpp::connection_hdl hdl, server::message_ptr msg) {
  545. long curHdl = (long)hdl.lock().get();
  546. std::string payload = msg->get_payload();
  547. #if (defined _WIN32 || defined _WIN64)
  548. payload = utf8_to_string(payload);//windows web utf8, terminal gbk
  549. #endif
  550. std::shared_ptr<cJSON> pJson(cJSON_Parse(payload.c_str()), [](cJSON* p) {
  551. if (nullptr != p)
  552. cJSON_Delete(p);
  553. });
  554. if (nullptr == pJson.get()) {
  555. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer -> message_handler_ws_sm2, jsonErr:%s", payload.c_str());
  556. return;
  557. }
  558. switch (ConfigManager::getInstance().m_ws_sm2_hdls_manager[curHdl]->getState())
  559. {
  560. case SM2ProtocolState::WAIT_KEY_EXCHANGE:
  561. {
  562. auto pubKey = json_deal::getStringFromCjsonObj(pJson.get(), "pubKey");
  563. if (pubKey.first == false)
  564. {
  565. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("pubKey == null");
  566. return;
  567. }
  568. ConfigManager::getInstance().m_ws_sm2_hdls_manager[curHdl]->setOpposite(pubKey.second.GetData());
  569. auto encPub = ConfigManager::getInstance().m_ws_sm2_hdls_manager[curHdl]->GenerateEncPubKey();
  570. if (!encPub.first)
  571. {
  572. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer -> message_handler_ws_sm2 err:%s", encPub.second.c_str());
  573. break;
  574. }
  575. std::string js = CSimpleString::Format("{\"encPub\":\"%s\"}", encPub.second.c_str()).GetData();
  576. //DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer -> message_handler_ws_sm2 send js:%s", js.c_str());
  577. do_send_msg(hdl, js);
  578. ConfigManager::getInstance().m_ws_sm2_hdls_manager[curHdl]->setState(SM2ProtocolState::DATA_EXCHANGE);
  579. }
  580. break;
  581. case SM2ProtocolState::DATA_EXCHANGE:
  582. {
  583. auto sign = json_deal::getStringFromCjsonObj(pJson.get(), "sign");
  584. if (sign.first == false)
  585. {
  586. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("sign == null, full json:%s", payload.c_str());
  587. return;
  588. }
  589. auto msgData = json_deal::getStringFromCjsonObj(pJson.get(), "msg");
  590. if (msgData.first == false)
  591. {
  592. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("msgData == null, full json:%s", payload.c_str());
  593. return;
  594. }
  595. auto decMsg = ConfigManager::getInstance().m_ws_sm2_hdls_manager[curHdl]->DecryptMsg(sign.second.GetData(), msgData.second.GetData());
  596. if(decMsg.first == false)
  597. {
  598. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("DecryptMsg err:%s", decMsg.second.c_str());
  599. return;
  600. }
  601. msg->set_payload(decMsg.second);
  602. message_handler(hdl, msg);
  603. }
  604. break;
  605. default:
  606. return;
  607. }
  608. }
  609. // websocket message handler
  610. void CWebsocketServer::message_handler(websocketpp::connection_hdl hdl, server::message_ptr msg) {
  611. /*
  612. static bool isTest = true;
  613. if (isTest)
  614. {
  615. msg->set_payload("{\"messageType\":13,\"transID\" : 11111,\"name\" : \"UIState\",\"value\" : \"M\"}");
  616. isTest = false;
  617. }
  618. else
  619. return;
  620. */
  621. static int pos = 0;
  622. static std::map<int, int> t_hdlArr;
  623. int hdlPos = (long)hdl.lock().get();
  624. if (t_hdlArr.end() == t_hdlArr.find(hdlPos))
  625. t_hdlArr[hdlPos] = pos++;
  626. auto msgHandleFun = [&]() {
  627. //DbgEx("CWebsocketServer -> message_handler");
  628. //DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("create thread:%s", __FUNCTION__);
  629. std::string payload = msg->get_payload();
  630. // proto convert here
  631. #if (defined _WIN32 || defined _WIN64)
  632. payload = utf8_to_string(payload);//windows web utf8, terminal gbk
  633. #endif
  634. std::shared_ptr<cJSON> pJson(cJSON_Parse(payload.c_str()), [](cJSON* p) {
  635. if (nullptr != p)
  636. cJSON_Delete(p);
  637. });
  638. if (nullptr == pJson.get()) {
  639. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer -> message_handler, jsonErr:%s", payload.c_str());
  640. return;
  641. }
  642. auto transIdJson = cJSON_GetObjectItem(pJson.get(), "transId");//不做相关改造,因为这里还使用了set
  643. int transid = 0;
  644. if (transIdJson != nullptr)
  645. {
  646. transid = transIdJson->valueint;
  647. int modifyT = t_hdlArr[hdlPos] << 24;
  648. int dstTransId = transid ^ modifyT;
  649. cJSON_SetIntValue(transIdJson, dstTransId);
  650. }
  651. auto messageTypeJson = json_deal::getIntergerFromCjsonObj(pJson.get(), "messageType");
  652. if (messageTypeJson.first == false)
  653. {
  654. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer -> messageTypeJson == null");
  655. return;
  656. }
  657. char *unformateStr = cJSON_PrintUnformatted(pJson.get());
  658. std::string dstPayLoad = unformateStr;
  659. delete[]unformateStr;
  660. auto messageType = messageTypeJson.second;
  661. if (messageType < MessageType::WEB_SPECICAL_MANAGE_END)
  662. {
  663. /*
  664. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("deal_msg, messageType:%d(%s), len:%d, payload:%s", messageType, GetMessageTypeString(messageType).c_str(),
  665. payload.length(), payload.length() > 800 ? payload.substr(0, 800).append("...").c_str() : payload.c_str());
  666. */
  667. deal_msg(dstPayLoad, hdl, transid);
  668. }
  669. else if (messageType > WEB_CHROMIUM_MSG_BEGIN && messageType < WEB_CHROMIUM_MSG_END)
  670. {
  671. /*
  672. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("deal_webchromium_msg, messageType:%d(%s), len:%d, payload:%s", messageType, GetMessageTypeString(messageType).c_str(),
  673. payload.length(), payload.length() > 800 ? payload.substr(0, 800).append("...").c_str() : payload.c_str());
  674. */
  675. deal_webchromium_msg(dstPayLoad, hdl, messageType);
  676. }
  677. else if (messageType > METHOD_SYSTEM_START && messageType < METHOD_SYSTEM_END)
  678. deal_logMsg(dstPayLoad, hdl, messageType);
  679. else if (messageType == METHOD_BEIDOU_LOG)
  680. deal_logMsg(dstPayLoad, hdl, messageType);
  681. };
  682. boost::thread dealMsgThread(msgHandleFun);
  683. dealMsgThread.join();
  684. }
  685. std::pair<int, std::string> CWebsocketServer::restroreTransId(std::string payLoad)
  686. {
  687. std::shared_ptr<cJSON> pJson(cJSON_Parse(payLoad.c_str()), [](cJSON* p) {
  688. if (nullptr != p)
  689. cJSON_Delete(p);
  690. });
  691. auto transIdJson = cJSON_GetObjectItem(pJson.get(), "transId");//不改动,因为需要set
  692. int dstTransId = 0;
  693. if (transIdJson != nullptr)
  694. {
  695. int transid = transIdJson->valueint;
  696. dstTransId = transid & 0x00FFFFFF;
  697. cJSON_SetIntValue(transIdJson, dstTransId);
  698. }
  699. char* unformateStr = cJSON_PrintUnformatted(pJson.get());
  700. std::string dstPayLoad = unformateStr;
  701. delete[]unformateStr;
  702. return std::make_pair(dstTransId, dstPayLoad);
  703. }
  704. std::string url_decode(const std::string& str) {
  705. std::string decoded_str;
  706. for (size_t i = 0; i < str.length(); ++i) {
  707. if (str[i] == '%') {
  708. if (i + 2 < str.length()) {
  709. std::string hex_str = str.substr(i + 1, 2);
  710. try {
  711. int hex_val = std::stoi(hex_str, nullptr, 16);
  712. decoded_str += static_cast<char>(hex_val);
  713. i += 2;
  714. }
  715. catch (const std::invalid_argument& e) {
  716. // 处理无效的编码
  717. decoded_str += '%';
  718. }
  719. catch (const std::out_of_range& e) {
  720. decoded_str += '%';
  721. }
  722. }
  723. else {
  724. decoded_str += '%';
  725. }
  726. }
  727. else if (str[i] == '+') {
  728. decoded_str += ' ';
  729. }
  730. else {
  731. decoded_str += str[i];
  732. }
  733. }
  734. return decoded_str;
  735. }
  736. // 解析 query string 的函数
  737. std::map<std::string, std::string> parse_query_string(const std::string& query_string) {
  738. std::map<std::string, std::string> params;
  739. std::stringstream ss(query_string);
  740. std::string pair;
  741. while (std::getline(ss, pair, '&')) {
  742. size_t equal_pos = pair.find('=');
  743. if (equal_pos != std::string::npos) {
  744. std::string key = pair.substr(0, equal_pos);
  745. std::string value = pair.substr(equal_pos + 1);
  746. params[url_decode(key)] = url_decode(value); // 解码 key 和 value
  747. }
  748. else if (!pair.empty()) {
  749. params[url_decode(pair)] = ""; // 处理只有 key 没有 value 的情况
  750. }
  751. }
  752. return params;
  753. }
  754. void CWebsocketServer::open_handler(websocketpp::connection_hdl hdl) {
  755. // hand shake here
  756. server::connection_ptr con = m_wsserver.get_con_from_hdl(hdl);
  757. std::string query_str = con->get_uri()->get_query();
  758. auto params = parse_query_string(query_str);
  759. if (params.find("name") != params.end())
  760. ConfigManager::getInstance().m_ws_infoArr[(long)hdl.lock().get()] = websocket_info(params["name"], false);
  761. LogManager::getInstance().logWebSocketBuild((uint64_t)hdl.lock().get(), con->get_uri()->str(), "unsafe");
  762. ConfigManager::getInstance().m_connection_hdls.insert(std::pair<unsigned int, websocketpp::connection_hdl>((long)hdl.lock().get(), hdl));
  763. }
  764. void CWebsocketServer::open_handler_wss(websocketpp::connection_hdl hdl) {
  765. // hand shake here
  766. auto con = m_server_wss.get_con_from_hdl(hdl);
  767. std::string query_str = con->get_uri()->get_query();
  768. auto params = parse_query_string(query_str);
  769. if (params.find("name") != params.end())
  770. ConfigManager::getInstance().m_ws_infoArr[(long)hdl.lock().get()] = websocket_info(params["name"], true);
  771. unsigned int dstHdl = (long)hdl.lock().get();//start from 0, be different from ws
  772. LogManager::getInstance().logWebSocketBuild((uint64_t)hdl.lock().get(), con->get_uri()->str(), "wss");
  773. ConfigManager::getInstance().m_connection_wss_hdls.insert(std::pair<unsigned int, websocketpp::connection_hdl>((long)dstHdl, hdl));
  774. }
  775. void CWebsocketServer::open_handler_ws_sm2(websocketpp::connection_hdl hdl) {
  776. // hand shake here
  777. auto con = m_ws_sm2_server.get_con_from_hdl(hdl);
  778. unsigned int dstHdl = (long)hdl.lock().get();//start from 0, be different from ws
  779. std::string query_str = con->get_uri()->get_query();
  780. auto params = parse_query_string(query_str);
  781. if (params.find("name") != params.end())
  782. ConfigManager::getInstance().m_ws_infoArr[(long)hdl.lock().get()] = websocket_info(params["name"], true);
  783. LogManager::getInstance().logWebSocketBuild((uint64_t)hdl.lock().get(), con->get_uri()->str(), "ws_sm2");
  784. ConfigManager::getInstance().m_connection_ws_sm2_hdls.insert(std::pair<unsigned int, websocketpp::connection_hdl>(dstHdl, hdl));
  785. ConfigManager::getInstance().m_ws_sm2_hdls_manager.insert(std::make_pair(dstHdl, new SM2_Encrypt_Manager()));
  786. }
  787. void CWebsocketServer::close_handler(websocketpp::connection_hdl hdl) {
  788. // hand shake here
  789. auto connectionIter = ConfigManager::getInstance().m_connection_hdls.find((long)hdl.lock().get());
  790. if (ConfigManager::getInstance().m_connection_hdls.end() != connectionIter)
  791. {
  792. LogManager::getInstance().logWebSocketClose((uint64_t)hdl.lock().get());
  793. //DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("ws connection with ws server closed : %u", hdl.lock().get());
  794. ConfigManager::getInstance().m_connection_hdls.erase(connectionIter);
  795. }
  796. else
  797. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("ws connection close erase failed : %u", hdl.lock().get());
  798. }
  799. void CWebsocketServer::close_handler_wss(websocketpp::connection_hdl hdl) {
  800. // hand shake here
  801. auto connectionIter = ConfigManager::getInstance().m_connection_wss_hdls.find((long)hdl.lock().get());
  802. if (ConfigManager::getInstance().m_connection_wss_hdls.end() != connectionIter)
  803. {
  804. LogManager::getInstance().logWebSocketClose((uint64_t)hdl.lock().get());
  805. //DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("wss connection with ws server closed : %u", hdl.lock().get());
  806. ConfigManager::getInstance().m_connection_wss_hdls.erase(connectionIter);
  807. }
  808. else
  809. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("wss connection close erase failed : %u", hdl.lock().get());
  810. }
  811. void CWebsocketServer::close_handler_ws_sm2(websocketpp::connection_hdl hdl)
  812. {
  813. // hand shake here
  814. auto connectionIter = ConfigManager::getInstance().m_connection_ws_sm2_hdls.find((long)hdl.lock().get());
  815. if (ConfigManager::getInstance().m_connection_ws_sm2_hdls.end() != connectionIter)
  816. {
  817. LogManager::getInstance().logWebSocketClose((uint64_t)hdl.lock().get());
  818. //DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("ws sm2 connection with ws server closed : %u", hdl.lock().get());
  819. ConfigManager::getInstance().m_connection_ws_sm2_hdls.erase(connectionIter);
  820. ConfigManager::getInstance().m_ws_sm2_hdls_manager.erase((long)hdl.lock().get());
  821. }
  822. else
  823. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("ws sm2 connection close erase failed : %u", hdl.lock().get());
  824. }
  825. void CWebsocketServer::do_run() {
  826. //DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("create thread:%s", __FUNCTION__);
  827. // Start the Asio io_service run loop
  828. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer -> do_run, Start the Asio io_service run loop");
  829. while (!boost::this_thread::interruption_requested())
  830. {
  831. try {
  832. m_ios.poll();
  833. Sleep(2);
  834. }
  835. catch (const std::exception& e) {
  836. // 输出标准异常信息
  837. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Exception in ios poll: %s", e.what());
  838. }
  839. catch (...) {
  840. // 输出其他异常信息
  841. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Other exception in ios poll");
  842. }
  843. }
  844. }
  845. void CWebsocketServer::do_relink()
  846. {
  847. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do_relink Enter");
  848. auto checkEntityIsNoStart = [&](std::string entityName) -> bool
  849. {
  850. CSmartPointer<IEntityFunction> spFunc = m_pEntity->GetFunction();
  851. LOG_ASSERT(spFunc != NULL);
  852. CEntityStaticInfo StaticInfo;
  853. CEntityRunInfo RunInfo;
  854. do {
  855. if (Error_Succeed != spFunc->GetEntityStaticInfo(entityName.c_str(), StaticInfo))
  856. break;
  857. if (Error_Succeed != spFunc->GetEntityRunInfo(entityName.c_str(), RunInfo))
  858. break;
  859. return RunInfo.eState == EntityState_NoStart;
  860. } while (false);
  861. return true;
  862. };
  863. while (!boost::this_thread::interruption_requested())
  864. {
  865. try
  866. {
  867. boost::this_thread::sleep_for(boost::chrono::seconds(10));
  868. if (boost::this_thread::interruption_requested())
  869. break;
  870. auto unlinkArr = m_esm->queryUnLinkSession();
  871. boost::lock_guard<boost::mutex> lock(m_dealMsgLock);
  872. std::string breakEntityStr = "";
  873. std::string noStartEntity = "";
  874. std::string noServerEntity = "";
  875. #if (defined _WIN32 || defined _WIN64)
  876. for each (auto it in unlinkArr)
  877. #else
  878. for (auto it : unlinkArr)
  879. #endif
  880. {
  881. if (checkEntityIsNoStart(it))
  882. {
  883. if (noStartEntity.empty())
  884. noStartEntity.append(it);
  885. else
  886. noStartEntity.append("|").append(it);
  887. continue;
  888. }
  889. if (!CWSCodec::getInstance()->checkEntityHasService(it))
  890. {
  891. if (noServerEntity.empty())
  892. noServerEntity.append(it);
  893. else
  894. noServerEntity.append("|").append(it);
  895. continue;
  896. }
  897. auto ret = getEntityClass(it);
  898. if (ret.first)
  899. {
  900. if (breakEntityStr.empty())
  901. breakEntityStr.append(it);
  902. else
  903. breakEntityStr.append("|").append(it);
  904. do_sendJsonStartSession(it, ret.second);
  905. }
  906. else
  907. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("unable find class of entity %s", it.c_str());
  908. }
  909. if (!breakEntityStr.empty())
  910. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("try to relink entity : %s, noStartEntity: %s, noServerEntity: %s", breakEntityStr.c_str(), noStartEntity.c_str(), noServerEntity.c_str());
  911. }
  912. catch (...)
  913. {
  914. std::exception_ptr p = std::current_exception();
  915. if (p) {
  916. try {
  917. std::rethrow_exception(p);
  918. }
  919. catch (const boost::thread_interrupted& e) {
  920. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("thread interrupted");
  921. }
  922. catch (const std::exception& e) {
  923. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("exception: %s", e.what());
  924. }
  925. }
  926. }
  927. }
  928. }
  929. void CWebsocketServer::run() {
  930. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("CWebsocketServer -> run");
  931. m_doRunThread = boost::thread(boost::bind(&CWebsocketServer::do_run, this));
  932. m_doReLinkThread = boost::thread(boost::bind(&CWebsocketServer::do_relink, this));
  933. }
  934. template <typename T, typename Y>
  935. bool CWebsocketServer::sendBroadCast(T &connections, Y &senders, std::string msg)
  936. {
  937. for (auto it = connections.begin(); it != connections.end(); it++)
  938. {
  939. try
  940. {
  941. websocketpp::connection_hdl hdl = it->second;
  942. websocketpp::lib::error_code dstExption;
  943. senders.send(hdl, msg, websocketpp::frame::opcode::TEXT, dstExption);
  944. }
  945. catch (const websocketpp::lib::error_code& e)
  946. {
  947. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("senders send crash : error message=%s", e.message());
  948. return false;
  949. }
  950. catch (const std::exception& e) {
  951. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("std exception : %s", e.what());
  952. return false;
  953. }
  954. catch (...) {
  955. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("other exception");
  956. return false;
  957. }
  958. }
  959. return true;
  960. }
  961. void CWebsocketServer::do_sendJsonBroadcast(std::string js)
  962. {
  963. auto dst_js = restroreTransId(js);
  964. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("message broadcast : json = %s", dst_js.second.c_str());
  965. if (dst_js.second.empty())
  966. {
  967. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("string empty");
  968. return;
  969. }
  970. auto formatMapToString = [](const std::map<unsigned int, websocketpp::connection_hdl>& mapArr) -> std::string {
  971. std::stringstream ss;
  972. ss << "[";
  973. for (auto it : mapArr)
  974. ss << "(" << it.first << ", " << (long)it.second.lock().get() << ")";
  975. ss << "]";
  976. return ss.str();
  977. };
  978. if (!sendBroadCast(ConfigManager::getInstance().m_connection_hdls, m_wsserver, dst_js.second))
  979. {
  980. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("ws server broadcast error");
  981. }
  982. if (!sendBroadCast(ConfigManager::getInstance().m_connection_ws_sm2_hdls, m_ws_sm2_server, dst_js.second))
  983. {
  984. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("ws_sm2 server broadcast error");
  985. }
  986. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("do_sendJsonBroadcast: broadcasts have already send to these terminals, no_security:%s, security:%s",
  987. formatMapToString(ConfigManager::getInstance().m_connection_hdls).c_str(), formatMapToString(ConfigManager::getInstance().m_connection_ws_sm2_hdls).c_str());
  988. }
  989. void CWebsocketServer::do_send_msg(int hdl, std::string msg)
  990. {
  991. //send ws
  992. try
  993. {
  994. std::map<unsigned int, websocketpp::connection_hdl>::iterator it = ConfigManager::getInstance().m_connection_hdls.find(hdl);
  995. if (ConfigManager::getInstance().m_connection_hdls.end() != it)
  996. {
  997. do_send_msg(it->second, msg);
  998. return;
  999. }
  1000. }
  1001. catch (const websocketpp::lib::error_code& e)
  1002. {
  1003. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("ws send crash : error message=%s", e.message());
  1004. }
  1005. catch (const std::exception& e) {
  1006. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("ws send std exception : %s", e.what());
  1007. }
  1008. catch (...) {
  1009. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("ws send other exception");
  1010. }
  1011. //send ws sm2
  1012. try
  1013. {
  1014. std::map<unsigned int, websocketpp::connection_hdl>::iterator it = ConfigManager::getInstance().m_connection_ws_sm2_hdls.find(hdl);
  1015. if (ConfigManager::getInstance().m_connection_ws_sm2_hdls.end() != it)
  1016. {
  1017. //DbgEx("Send....");
  1018. do_send_msg(it->second, msg);
  1019. }
  1020. return;
  1021. }
  1022. catch (const websocketpp::exception& e)
  1023. {
  1024. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("ws sm2 send crash : error message=%s", e.what());
  1025. }
  1026. catch (const std::exception& e) {
  1027. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("ws sm2 send std exception : %s", e.what());
  1028. }
  1029. catch (...) {
  1030. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("ws sm2 send other exception");
  1031. }
  1032. //send wss
  1033. /*
  1034. try
  1035. {
  1036. //DbgEx("do_send_notifyMsg Enter...");
  1037. std::map<unsigned int, websocketpp::connection_hdl>::iterator it = ConfigManager::getInstance().m_connection_hdls.find(hdlID);
  1038. if (ConfigManager::getInstance().m_connection_hdls.end() != it)
  1039. {
  1040. //DbgEx("Send....");
  1041. websocketpp::connection_hdl hdl = it->second;
  1042. websocketpp::lib::error_code dstExption;
  1043. m_wsserver.send(hdl, notifyMsg.GetData(), websocketpp::frame::opcode::TEXT, dstExption);
  1044. }
  1045. //DbgEx("do_send_notifyMsg End...");
  1046. }
  1047. catch (const websocketpp::lib::error_code& e)
  1048. {
  1049. DbgEx("m_wsserver send crash : error message=%s", e.message());
  1050. }
  1051. catch (const std::exception& e) {
  1052. DbgEx("std exception : %s", e.what());
  1053. }
  1054. catch (...) {
  1055. DbgEx("other exception");
  1056. }
  1057. */
  1058. return;
  1059. }
  1060. void CWebsocketServer::do_send_msg(websocketpp::connection_hdl hdl, std::string msg)
  1061. {
  1062. unsigned int dstHdl = (long)hdl.lock().get();
  1063. websocketpp::lib::error_code dstExption;
  1064. if (ConfigManager::getInstance().m_connection_hdls.find(dstHdl) != ConfigManager::getInstance().m_connection_hdls.end())
  1065. {
  1066. m_wsserver.send(hdl, msg, websocketpp::frame::opcode::TEXT, dstExption);
  1067. }
  1068. else if (ConfigManager::getInstance().m_connection_ws_sm2_hdls.find(dstHdl) != ConfigManager::getInstance().m_connection_ws_sm2_hdls.end())
  1069. {
  1070. if(ConfigManager::getInstance().m_ws_sm2_hdls_manager[dstHdl]->getState() == SM2ProtocolState::WAIT_KEY_EXCHANGE)
  1071. m_ws_sm2_server.send(ConfigManager::getInstance().m_connection_ws_sm2_hdls[dstHdl], msg, websocketpp::frame::opcode::TEXT, dstExption);
  1072. else
  1073. {
  1074. auto dst = ConfigManager::getInstance().m_ws_sm2_hdls_manager[dstHdl]->EncryptMsg(msg);
  1075. std::string js = CSimpleString::Format("{\"sign\":\"%s\",\"msg\":\"%s\"}", dst.first.c_str(), dst.second.c_str()).GetData();
  1076. m_ws_sm2_server.send(ConfigManager::getInstance().m_connection_ws_sm2_hdls[dstHdl], js, websocketpp::frame::opcode::TEXT, dstExption);
  1077. }
  1078. }
  1079. /*
  1080. else
  1081. m_server_wss.send(hdl, msg, websocketpp::frame::opcode::TEXT, dstExption);
  1082. */
  1083. }
  1084. void CWebsocketServer::do_send_notifyMsg(unsigned hdlID, unsigned transId, const std::string& reason, const std::string& errmsg, const std::string& rebootTime, const DWORD& dwSysError, const DWORD& dwUserCode) {
  1085. #if (defined _WIN32 || defined _WIN64)
  1086. auto notifyMsg = CSimpleStringA::Format("{\"messageType\":%d,\"transID:%d\", \"reason\":%s, \"errmsg\":%s, \"rebootTime\":%s}", RegisterNotify, transId, reason, errmsg, rebootTime);
  1087. #else
  1088. auto notifyMsg = CSimpleStringA::Format(R"({"messageType":%d,"transID:%d", "reason":%s, "errmsg":%s, "rebootTime":%s})", RegisterNotify, transId, reason, errmsg, rebootTime);
  1089. #endif
  1090. do_send_msg(hdlID, notifyMsg.GetData());
  1091. }
  1092. void CWebsocketServer::do_sendJson(std::string js, int messageType, int hdlID, unsigned int id, socket_msgInfo& msgInfo)
  1093. {
  1094. if (js.length() == 0)
  1095. return;
  1096. auto dst_js = restroreTransId(js);
  1097. js = dst_js.second;
  1098. if (messageType == MessageType::RequestAck)
  1099. {
  1100. #if(defined _WIN32 || defined _WIN64)
  1101. js = string_to_utf8(js);
  1102. #endif
  1103. if (js.empty())
  1104. {
  1105. LogManager::getInstance().logVtmRequestAck(hdlID, dst_js.first, msgInfo.inner_transID, msgInfo.sessionId,
  1106. msgInfo.entityName, msgInfo.functionName, 2, "none", msgInfo.payLoad, msgInfo.msgBytes, Error_Null, "string empty");
  1107. return;
  1108. }
  1109. if (ConfigManager::getInstance().m_connection_hdls.empty() && ConfigManager::getInstance().m_connection_ws_sm2_hdls.empty())
  1110. LogManager::getInstance().logVtmRequestAck(hdlID, dst_js.first, msgInfo.inner_transID, msgInfo.sessionId,
  1111. msgInfo.entityName, msgInfo.functionName, 2, "none", msgInfo.payLoad, msgInfo.msgBytes, Error_Null, "no websocket client connection");
  1112. else
  1113. {
  1114. if(msgInfo.userCodeVal == 0)
  1115. LogManager::getInstance().logVtmRequestAck(hdlID, dst_js.first, msgInfo.inner_transID, msgInfo.sessionId,
  1116. msgInfo.entityName, msgInfo.functionName, 0, "send", msgInfo.payLoad, msgInfo.msgBytes);
  1117. else
  1118. LogManager::getInstance().logVtmRequestAck(hdlID, dst_js.first, msgInfo.inner_transID, msgInfo.sessionId,
  1119. msgInfo.entityName, msgInfo.functionName, 2, "send", msgInfo.payLoad, msgInfo.msgBytes, msgInfo.userCodeVal, msgInfo.warnMsg);
  1120. do_send_msg(hdlID, js);
  1121. }
  1122. }
  1123. else if (messageType == MessageType::SessionAck)
  1124. {
  1125. if (js.empty())
  1126. {
  1127. LogManager::getInstance().logVtmSessionAck(hdlID, dst_js.first, msgInfo.inner_transID, msgInfo.sessionId, msgInfo.entityName, 2, "none",
  1128. msgInfo.payLoad, msgInfo.msgBytes, Error_Null, "string empty");
  1129. return;
  1130. }
  1131. if (ConfigManager::getInstance().m_connection_hdls.empty() && ConfigManager::getInstance().m_connection_ws_sm2_hdls.empty())
  1132. LogManager::getInstance().logVtmSessionAck(hdlID, dst_js.first, msgInfo.inner_transID, msgInfo.sessionId,
  1133. msgInfo.entityName, 2, "none", msgInfo.payLoad, msgInfo.msgBytes, Error_Null, "no websocket client connection");
  1134. else
  1135. {
  1136. if(msgInfo.userCodeVal == 0)
  1137. LogManager::getInstance().logVtmSessionAck(hdlID, dst_js.first, msgInfo.inner_transID, msgInfo.sessionId,
  1138. msgInfo.entityName, 0, "send", msgInfo.payLoad, msgInfo.msgBytes);
  1139. else
  1140. LogManager::getInstance().logVtmSessionAck(hdlID, dst_js.first, msgInfo.inner_transID, msgInfo.sessionId,
  1141. msgInfo.entityName, 2, "send", msgInfo.payLoad, msgInfo.msgBytes, msgInfo.userCodeVal, msgInfo.warnMsg);
  1142. do_send_msg(hdlID, js);
  1143. }
  1144. }
  1145. else
  1146. {
  1147. /*
  1148. if (ConfigManager::getInstance().m_withLinkLog)
  1149. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("WebSocket Search message_from_socket : json = %s", js.c_str());
  1150. else if (messageType != MessageType::RequestAck && messageType != MessageType::Event)//do not upload message which messageType equals to RequestAck or Event
  1151. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("WebSocket Search message_from_socket : json = %s", js.c_str());
  1152. */
  1153. if (js.empty())
  1154. {
  1155. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("string empty");
  1156. return;
  1157. }
  1158. if (ConfigManager::getInstance().m_connection_hdls.empty() && ConfigManager::getInstance().m_connection_ws_sm2_hdls.empty() && messageType != MessageType::Event)
  1159. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)
  1160. ("message_from_socket : no websocket client connection, messageType:%s", GetMessageTypeString(messageType));
  1161. else {
  1162. #if(defined _WIN32 || defined _WIN64)
  1163. js = string_to_utf8(js);
  1164. #endif
  1165. if (js.empty())
  1166. {
  1167. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("string empty 2");
  1168. return;
  1169. }
  1170. do_send_msg(hdlID, js);
  1171. }
  1172. }
  1173. }
  1174. // socket message handler
  1175. void CWebsocketServer::message_from_socket(CMessage& msg, unsigned int id) {
  1176. //DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("create thread:%s", __FUNCTION__);
  1177. // get message from socket and deserialize
  1178. // then send back to the web client
  1179. auto bufferLength = msg.getBufferLength();
  1180. /*
  1181. if (bufferLength > MAX_TRANSFER_LEN)
  1182. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("WebSocket Search message_from_socket : buffer len = %d, buffer pre50:%s", msg.getBufferLength(), msg.printfHEX(50).c_str());
  1183. else
  1184. receivehexdump(msg.getPayload(), msg.getLength());
  1185. */
  1186. //else
  1187. // DbgEx("WebSocket Search message_from_socket : buffer len = %d", msg.getBufferLength());
  1188. int replaceTransId = 0;
  1189. if (msg.getLength() > 16 && m_esm != NULL)
  1190. {//this is return buffer, it would not send out again, so I exchange the transId and sessionId place
  1191. msg.exchangeSessionIdAndTransId();
  1192. auto tmpReplace = m_esm->getSrcTransID(msg.getTransID());
  1193. replaceTransId = tmpReplace.first == true ? tmpReplace.second : 0;
  1194. }
  1195. // 处理session ack,获取hdlID
  1196. socket_msgInfo cur_from_socket(replaceTransId, msg.getTransID(), msg.getSessionID());
  1197. unsigned int hdlID = 0;
  1198. std::vector<std::pair<int, int>> sendArr;
  1199. if (MessageType::Event == msg.getMessageType())
  1200. {
  1201. auto formatVectorToString = [](const std::vector<std::pair<int, int>>& vec) -> std::string {
  1202. std::stringstream ss;
  1203. ss << "[";
  1204. for (size_t i = 0; i < vec.size(); ++i) {
  1205. ss << "(" << vec[i].first << ", " << vec[i].second << ")";
  1206. if (i != vec.size() - 1) {
  1207. ss << ", ";
  1208. }
  1209. }
  1210. ss << "]";
  1211. return ss.str();
  1212. };
  1213. m_esm->AskProcessEvent(&msg, sendArr);
  1214. if (sendArr.size() == 0)
  1215. m_esm->AskProcessEvent(&msg, sendArr);
  1216. std::string unaccurate_js;
  1217. for (auto i = sendArr.begin(); i != sendArr.end(); i++)
  1218. {
  1219. msg.setTransID(i->first);
  1220. cur_from_socket.hdlId = i->second;
  1221. std::string js = this->m_serializer->BufferToJson(msg, boost::bind(&EntitySessionManager::doWithErrorCode, m_esm, _1, _2), cur_from_socket, 0);
  1222. do_sendJson(js, msg.getMessageType(), i->second, id, cur_from_socket);
  1223. unaccurate_js = js;
  1224. }
  1225. auto signatureID = msg.getSignatureID();
  1226. auto sessionId = cur_from_socket.sessionId;
  1227. auto sessionEntityName = cur_from_socket.entityName;
  1228. if (sendArr.size() == 0)
  1229. LogManager::getInstance().logVtmEvent(-1, -1, sessionId, sessionEntityName, 0, "send", sendArr,
  1230. cur_from_socket.payLoad, cur_from_socket.msgBytes, Error_NotExist, "message_from_socket event sendArr.size() == 0");
  1231. else if (signatureID != eMsgSig_LogInfo && signatureID != eMsgSig_EntityStatus && signatureID != eMsgSig_PerformanceList)//fiter guiconsole msg
  1232. {
  1233. LogManager::getInstance().logVtmEvent(-1, -1, sessionId, sessionEntityName, 0, "send", sendArr,
  1234. cur_from_socket.payLoad, cur_from_socket.msgBytes);
  1235. /*
  1236. #ifdef DEVOPS_ON_ST
  1237. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("message_from_socket event %s send to arr:%s"
  1238. , unaccurate_js.c_str(), formatVectorToString(sendArr).c_str());
  1239. #elif defined(DEVOPS_ON_UAT)
  1240. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("message_from_socket event %s send to arr:%s"
  1241. , unaccurate_js.c_str(), formatVectorToString(sendArr).c_str());
  1242. #else
  1243. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("message_from_socket event send to arr:%s", formatVectorToString(sendArr).c_str());
  1244. #endif
  1245. */
  1246. }
  1247. }
  1248. else if (MessageType::EndSession == msg.getMessageType()) //session end
  1249. {
  1250. int sessionId = msg.getTransID();
  1251. auto ret = m_esm->DoSessionRemove(sessionId);
  1252. if (ret.first)
  1253. LogManager::getInstance().logVtmEndSession(-1, ret.second, sessionId, 0, "clean", "", Error_Succeed, "remove session success");
  1254. else
  1255. LogManager::getInstance().logVtmEndSession(-1, ret.second, sessionId, 1, "clean", "", Error_NotExist, "detect session lost!remove failed");
  1256. }
  1257. else if (MessageType::SessionAck == msg.getMessageType())
  1258. {//sessionAck
  1259. auto ret = m_esm->AskProcessSession(&msg, hdlID);
  1260. auto sessionId = msg.getSessionID();
  1261. if (ACKPROCESS_NOTHING == ret.first)
  1262. {
  1263. LogManager::getInstance().logVtmSessionAck(hdlID, -1, -1, sessionId, cur_from_socket.entityName, 2, "none", cur_from_socket.payLoad, cur_from_socket.msgBytes,
  1264. Error_NotExist, "can not find save session, process noting");
  1265. return;
  1266. }
  1267. if (0 != hdlID)
  1268. {//hdlId为0时,说明本地发起的session
  1269. cur_from_socket.hdlId = hdlID;
  1270. std::string js = this->m_serializer->BufferToJson(msg, boost::bind(&EntitySessionManager::doWithErrorCode, m_esm, _1, _2), cur_from_socket, replaceTransId);
  1271. do_sendJson(js, msg.getMessageType(), hdlID, id, cur_from_socket); //发给首记录的CMessage
  1272. }
  1273. for (auto cur = ConfigManager::getInstance().m_msg_pool.begin(); cur != ConfigManager::getInstance().m_msg_pool.end(); cur++)
  1274. {//sessionId为-1时需处理,否则会引发消息风暴
  1275. if (cur->first == ret.second)
  1276. {
  1277. std::vector<std::pair<websocketpp::connection_hdl, std::string>> msgArr(cur->second);
  1278. ConfigManager::getInstance().m_msg_pool.erase(cur); //为防止处理消息时引发消息风暴,拷贝并删除原msg_pool
  1279. if (-1 != sessionId) {
  1280. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Session with %s Make, deal with record Msg:%d", ret.second.c_str(), msgArr.size());
  1281. for (auto msg = msgArr.begin(); msg != msgArr.end(); msg++)
  1282. deal_msg(msg->second, msg->first, 1);
  1283. break;
  1284. }
  1285. else
  1286. {
  1287. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Session with %s Make Failed, deal with record Msg:%d", ret.second.c_str(), msgArr.size());
  1288. #if (defined _WIN32 || defined _WIN64)
  1289. for each (auto msg in msgArr)
  1290. #else
  1291. for (auto msg : msgArr)
  1292. #endif
  1293. deal_sessionBreakMsg(msg.second, msg.first);
  1294. break;
  1295. }
  1296. }
  1297. }
  1298. }
  1299. else
  1300. {
  1301. m_esm->AckProcess(&msg, hdlID); //替换真实的transId,update session map
  1302. cur_from_socket.hdlId = hdlID;
  1303. std::string js = this->m_serializer->BufferToJson(msg, boost::bind(&EntitySessionManager::doWithErrorCode, m_esm, _1, _2), cur_from_socket, replaceTransId);
  1304. do_sendJson(js, msg.getMessageType(), hdlID, id, cur_from_socket);
  1305. }
  1306. }
  1307. std::string get_password() {
  1308. return "test";
  1309. }
  1310. // See https://wiki.mozilla.org/Security/Server_Side_TLS for more details about
  1311. // the TLS modes. The code below demonstrates how to implement both the modern
  1312. enum tls_mode {
  1313. MOZILLA_INTERMEDIATE = 1,
  1314. MOZILLA_MODERN = 2
  1315. };
  1316. const char* certificate_content =
  1317. "-----BEGIN CERTIFICATE-----\n"
  1318. "MIICtDCCAZygAwIBAgIUbdvfhsKgSv4zdyF1Y8OjFOFiCiMwDQYJKoZIhvcNAQEL\n"
  1319. "BQAwFDESMBAGA1UEAwwJbG9jYWxob3N0MB4XDTI0MDEyOTAyMDIyNloXDTI1MDEy\n"
  1320. "ODAyMDIyNlowFDESMBAGA1UEAwwJbG9jYWxob3N0MIIBIjANBgkqhkiG9w0BAQEF\n"
  1321. "AAOCAQ8AMIIBCgKCAQEAr+2L/iVAFrtMsqMFaI6UgTLhxZCAPkH6LsQMMYOzMz77\n"
  1322. "pRhOKv0RzNu4zSAt14/U29R8deYa/244XNyHgj2TniFotxR1J3Q0Kzaf22g4WqIR\n"
  1323. "qZKJ8y1ry4g70zgyPHUNmcmxrXvq1Y4k8J/g3Jqhlqqcw1+Sy8xQZy+oXhTsjh2+\n"
  1324. "TGeVG9l2m5jdyWJSw7jE9rT0iJhXiuaTXQ8osjsspdvPhIg5agKvP2T6JpG/8jLY\n"
  1325. "4Kkvz38lNiRwahmaet7pU9ggtHyXR0vUGN7M9Z14c17v2veuBwbNDLWeGSTaFA8b\n"
  1326. "M9BOntzMhM6tItZvf2TGR+TAIkGPLcdCpXoW7PFlZQIDAQABMA0GCSqGSIb3DQEB\n"
  1327. "CwUAA4IBAQA6V3yl+Zdpimyi/bNeOwnC9Ml2yDzapsQBDiXlG97zj++bT/lwPvlV\n"
  1328. "/Oti2EWrVCO59eaarQmdOBAfgpipJlxzsWGeya26D04MHGwMI5aejpLIlyETyMzf\n"
  1329. "VzTIufMTtG3w37CSqUIyodhH4VhHGWPazQg3QKHiPryL0k2q9013mc8glSNLI//4\n"
  1330. "tOk5ZRzHbb8uLmgGp54Cr4449DwtltAwQ74Y+NY/0Fasu7jvjLXm4o37twfPGiAy\n"
  1331. "LRjbBuEV6/ab8dTmOQA6mPvNOGHqJg40Y6d3XCiQ0H2ovqBvMz+83tFrmRwcydKP\n"
  1332. "mi9O1t8a62FV2WI38+gIleRJ/iNkpZhI\n"
  1333. "-----END CERTIFICATE-----";
  1334. const char* dh_content =
  1335. "-----BEGIN DH PARAMETERS-----\n"
  1336. "MIIBCAKCAQEAzg8TpgTyX/im9EqTq1tD1fx1IoxE/u6CwE+tXbHL58KRrJStOiGw\n"
  1337. "TyZwqOKIqgC0wbJztjpMYq+X8cfDzkOE7GoSaqYAB5r9AafUiIV6cpvJ+ilWRxZK\n"
  1338. "wo2Qjydz9F+xGcIXnQf/vSpK5Hws9h9OpnyCz413i4PDQXzQnKyHOTCq4t7UEU4m\n"
  1339. "oZ9Weum8zGHomBFubyuinBkIyZYRg4NPnqirTgChfLVpbME9menEWGF4YPtEc3WC\n"
  1340. "AIXtUfKLgAIF/2MNaiTzmm6JCcrfof1PED6VackHimHUwwV/miW50MfnlIery4g8\n"
  1341. "B72avGZXhTQ+49qDaPPfLqlvhQ6MOxc0BwIBAg==\n"
  1342. "-----END DH PARAMETERS-----";
  1343. const char* private_key_content =
  1344. "-----BEGIN RSA PRIVATE KEY-----\n"
  1345. "MIIEpAIBAAKCAQEAr+2L/iVAFrtMsqMFaI6UgTLhxZCAPkH6LsQMMYOzMz77pRhO\n"
  1346. "Kv0RzNu4zSAt14/U29R8deYa/244XNyHgj2TniFotxR1J3Q0Kzaf22g4WqIRqZKJ\n"
  1347. "8y1ry4g70zgyPHUNmcmxrXvq1Y4k8J/g3Jqhlqqcw1+Sy8xQZy+oXhTsjh2+TGeV\n"
  1348. "G9l2m5jdyWJSw7jE9rT0iJhXiuaTXQ8osjsspdvPhIg5agKvP2T6JpG/8jLY4Kkv\n"
  1349. "z38lNiRwahmaet7pU9ggtHyXR0vUGN7M9Z14c17v2veuBwbNDLWeGSTaFA8bM9BO\n"
  1350. "ntzMhM6tItZvf2TGR+TAIkGPLcdCpXoW7PFlZQIDAQABAoIBAAMiJLpp0O6mttq+\n"
  1351. "pw/B7Fixvo4tgO867xkKoln2achpoIND/85ps/m1VyEGJ/LdK7IfonjaGJSAks3q\n"
  1352. "mppB/QFVwH70RRLnEa/MWQQIIurQVKazxrO3VSJqooAkUda8UfnxevHnUVss3TEQ\n"
  1353. "8+kR8vtTPhgHuX5aPPtdgK7uSflM/DnBMQUW2y9t85ahAosRLfRO1LPsKC+cSmGY\n"
  1354. "tELVnBgtKE64Am5rkRododcEkLVrQkf4jenqL0yusuwjSFUK1AUhOOS0Pp47oKCC\n"
  1355. "gXo4oBWmM7Zyj9+/OCkJ4HVBzCERGx5/DJEFRd9KF2HRpOG4cWH5u0c7Ab6WzIul\n"
  1356. "w6JM2JkCgYEA6E8gdZTueoK1RteyZgosJWA5aBrDSK8PQvykh1XgWdsBoD6wdDV6\n"
  1357. "xEpwX+VL0YaVLMaxxqqMMOx9x2ONK/wa11osrOTLqs+yzSqFK2RerpGpm/bbPVg4\n"
  1358. "mCVC9CBHN40TSVLLCrdlcaJvDXLsVNJR1Qc7UPoLf5ZwnEBzMnaYj/0CgYEAwd57\n"
  1359. "UM/sZ6pSgZkBh0XdU/r6UAGE+5wUSVQPpOqOfRXp7KCEO1LoMaWs1xVEBHcjI4Zm\n"
  1360. "oXYTT8XXN2Y8fxeYk5oabDTEUU3RiEySxDmMzJAdGIhSGlAmHnob+pjpnNrSU+ml\n"
  1361. "ktf8EQPT4Uh/9vTwfJfC39cqRyrgMDGHWgDL44kCgYEAnYT5x4DC68MxU+XyC4qe\n"
  1362. "QIGsq1BqViCNwqg4j6PFSmhcA+I+F64jnnGzHPMHo/0TrZbU+JawFIRnDQoNsWNy\n"
  1363. "d8+nN1y0VRAcd017mt6l7MzMKvsJ0eC+DzxE1/ADGkHIcBF5p7yPRqwTjRT99s3d\n"
  1364. "uwr6R4Akx9Ckbu6sre0IqakCgYEAur7tuG1hAoadjTrurvhMd63l2pVsHNY+8Fep\n"
  1365. "7ikG8lTejVIdretxwQkfEdvpNgfw8DJxwAw2E/y2ECZKFf6LuXqTAzJc/RhhwtJH\n"
  1366. "+f6Zsx8K9+uwcqpL3cWwF9eeCLr8KVqCZI3qwBUYzwSAR0mwMcQaMnZXwk/5vlKn\n"
  1367. "V7o2rEkCgYBfQyd4bdSPNkVayTEBK2L8jG4VuXGYuqdBtaHtNrrvAUsQEhv6RFv9\n"
  1368. "b3g6xZrHI2WZ2yoR2xPJ8kuQLmTB04eg/VuVn50N/rdeZyKkmpb9k1fZftQ/2dDE\n"
  1369. "kgJQP3/xqeULzdKboekyUsM0uT5/jbVPaelVtOIpqfcdtLtQc6/4eQ==\n"
  1370. "-----END RSA PRIVATE KEY-----";
  1371. context_ptr on_tls_init(tls_mode mode, websocketpp::connection_hdl hdl) {
  1372. namespace asio = websocketpp::lib::asio;
  1373. std::cout << "on_tls_init called with hdl: " << hdl.lock().get() << std::endl;
  1374. std::cout << "using TLS mode: " << (mode == MOZILLA_MODERN ? "Mozilla Modern" : "Mozilla Intermediate") << std::endl;
  1375. context_ptr ctx = websocketpp::lib::make_shared<asio::ssl::context>(asio::ssl::context::sslv23);
  1376. try {
  1377. if (mode == MOZILLA_MODERN) {
  1378. // Modern disables TLSv1
  1379. ctx->set_options(asio::ssl::context::default_workarounds |
  1380. asio::ssl::context::no_sslv2 |
  1381. asio::ssl::context::no_sslv3 |
  1382. asio::ssl::context::no_tlsv1 |
  1383. asio::ssl::context::single_dh_use);
  1384. }
  1385. else {
  1386. ctx->set_options(asio::ssl::context::default_workarounds |
  1387. asio::ssl::context::no_sslv2 |
  1388. asio::ssl::context::no_sslv3 |
  1389. asio::ssl::context::single_dh_use);
  1390. }
  1391. ctx->set_password_callback(bind(&get_password));
  1392. ctx->use_certificate_chain(asio::buffer(certificate_content, std::strlen(certificate_content)));
  1393. ctx->use_private_key(asio::buffer(private_key_content, std::strlen(private_key_content)), asio::ssl::context::pem);
  1394. // Example method of generating this file:
  1395. // `openssl dhparam -out dh.pem 2048`
  1396. // Mozilla Intermediate suggests 1024 as the minimum size to use
  1397. // Mozilla Modern suggests 2048 as the minimum size to use.
  1398. ctx->use_tmp_dh(asio::buffer(dh_content, std::strlen(dh_content)));
  1399. std::string ciphers;
  1400. if (mode == MOZILLA_MODERN) {
  1401. ciphers = "ECDHE-RSA-AES128-GCM-SHA256:ECDHE-ECDSA-AES128-GCM-SHA256:ECDHE-RSA-AES256-GCM-SHA384:ECDHE-ECDSA-AES256-GCM-SHA384:DHE-RSA-AES128-GCM-SHA256:DHE-DSS-AES128-GCM-SHA256:kEDH+AESGCM:ECDHE-RSA-AES128-SHA256:ECDHE-ECDSA-AES128-SHA256:ECDHE-RSA-AES128-SHA:ECDHE-ECDSA-AES128-SHA:ECDHE-RSA-AES256-SHA384:ECDHE-ECDSA-AES256-SHA384:ECDHE-RSA-AES256-SHA:ECDHE-ECDSA-AES256-SHA:DHE-RSA-AES128-SHA256:DHE-RSA-AES128-SHA:DHE-DSS-AES128-SHA256:DHE-RSA-AES256-SHA256:DHE-DSS-AES256-SHA:DHE-RSA-AES256-SHA:!aNULL:!eNULL:!EXPORT:!DES:!RC4:!3DES:!MD5:!PSK";
  1402. }
  1403. else {
  1404. ciphers = "ECDHE-RSA-AES128-GCM-SHA256:ECDHE-ECDSA-AES128-GCM-SHA256:ECDHE-RSA-AES256-GCM-SHA384:ECDHE-ECDSA-AES256-GCM-SHA384:DHE-RSA-AES128-GCM-SHA256:DHE-DSS-AES128-GCM-SHA256:kEDH+AESGCM:ECDHE-RSA-AES128-SHA256:ECDHE-ECDSA-AES128-SHA256:ECDHE-RSA-AES128-SHA:ECDHE-ECDSA-AES128-SHA:ECDHE-RSA-AES256-SHA384:ECDHE-ECDSA-AES256-SHA384:ECDHE-RSA-AES256-SHA:ECDHE-ECDSA-AES256-SHA:DHE-RSA-AES128-SHA256:DHE-RSA-AES128-SHA:DHE-DSS-AES128-SHA256:DHE-RSA-AES256-SHA256:DHE-DSS-AES256-SHA:DHE-RSA-AES256-SHA:AES128-GCM-SHA256:AES256-GCM-SHA384:AES128-SHA256:AES256-SHA256:AES128-SHA:AES256-SHA:AES:CAMELLIA:DES-CBC3-SHA:!aNULL:!eNULL:!EXPORT:!DES:!RC4:!MD5:!PSK:!aECDH:!EDH-DSS-DES-CBC3-SHA:!EDH-RSA-DES-CBC3-SHA:!KRB5-DES-CBC3-SHA";
  1405. }
  1406. if (SSL_CTX_set_cipher_list(ctx->native_handle(), ciphers.c_str()) != 1) {
  1407. std::cout << "Error setting cipher list" << std::endl;
  1408. }
  1409. }
  1410. catch (std::exception& e) {
  1411. std::cout << "Exception: " << e.what() << std::endl;
  1412. }
  1413. return ctx;
  1414. }
  1415. void CWebsocketServer::init_websocket() {
  1416. auto checkPortExist = []() -> bool {
  1417. boost::asio::io_service ioService;
  1418. boost::asio::ip::tcp::socket* pSockTcp = NULL;
  1419. bool bSockUseError = false;
  1420. try {
  1421. auto tcpEndpoint = boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), DEFAULT_SERVER_PORT);
  1422. pSockTcp = new boost::asio::ip::tcp::socket(ioService, tcpEndpoint);
  1423. bSockUseError = false;
  1424. }
  1425. catch (...)
  1426. {
  1427. bSockUseError = true;
  1428. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("%s sock %d error", __FUNCTION__, DEFAULT_SERVER_PORT);
  1429. }
  1430. //释放
  1431. if (!bSockUseError && pSockTcp != NULL)
  1432. {
  1433. delete pSockTcp;
  1434. pSockTcp = NULL;
  1435. }
  1436. ioService.stop();
  1437. return bSockUseError;
  1438. };
  1439. /*
  1440. while (true) {
  1441. if (!checkPortExist())
  1442. break;
  1443. else
  1444. DbgEx("checkPortExist failed");
  1445. std::this_thread::sleep_for(std::chrono::seconds(5));
  1446. }
  1447. */
  1448. try
  1449. {
  1450. //ws
  1451. // Set logging settings
  1452. m_wsserver.set_error_channels(websocketpp::log::elevel::all);
  1453. m_wsserver.set_access_channels(websocketpp::log::alevel::all ^ websocketpp::log::alevel::frame_payload);
  1454. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("m_wsserver.get_max_message_size = %d", m_wsserver.get_max_message_size());
  1455. m_wsserver.set_max_message_size(MAX_TRANSFER_LEN);
  1456. m_wsserver.set_close_handshake_timeout(3000);
  1457. m_wsserver.set_pong_timeout(3000);
  1458. // Initialize Asio
  1459. m_wsserver.init_asio(&m_ios);
  1460. m_wsserver.set_reuse_addr(true);
  1461. // Set the default message handler to the echo handler
  1462. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Set the default message handler to the echo handler");
  1463. m_wsserver.set_message_handler(websocketpp::lib::bind(&CWebsocketServer::message_handler, this, websocketpp::lib::placeholders::_1, websocketpp::lib::placeholders::_2));
  1464. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Set set_open_handler");
  1465. m_wsserver.set_open_handler(websocketpp::lib::bind(&CWebsocketServer::open_handler, this, websocketpp::lib::placeholders::_1));
  1466. m_wsserver.set_close_handler(websocketpp::lib::bind(&CWebsocketServer::close_handler, this, websocketpp::lib::placeholders::_1));
  1467. // Listen on port 9002, add m_acceptor->set_option(lib::asio::socket_base::reuse_address(true), bec);
  1468. m_wsserver.listen(DEFAULT_SERVER_PORT);
  1469. // Queues a connection accept operation
  1470. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Queues a connection accept operation");
  1471. m_wsserver.start_accept();
  1472. //ws sm2
  1473. // Set logging settings
  1474. m_ws_sm2_server.set_error_channels(websocketpp::log::elevel::all);
  1475. m_ws_sm2_server.set_access_channels(websocketpp::log::alevel::all ^ websocketpp::log::alevel::frame_payload);
  1476. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("m_ws_sm2_server.get_max_message_size = %d", m_ws_sm2_server.get_max_message_size());
  1477. m_ws_sm2_server.set_max_message_size(MAX_TRANSFER_LEN);
  1478. m_ws_sm2_server.set_close_handshake_timeout(3000);
  1479. m_ws_sm2_server.set_pong_timeout(3000);
  1480. // Initialize Asio
  1481. m_ws_sm2_server.init_asio(&m_ios);
  1482. // Set the default message handler to the echo handler
  1483. m_ws_sm2_server.set_message_handler(websocketpp::lib::bind(&CWebsocketServer::message_handler_ws_sm2, this, websocketpp::lib::placeholders::_1, websocketpp::lib::placeholders::_2));
  1484. m_ws_sm2_server.set_open_handler(websocketpp::lib::bind(&CWebsocketServer::open_handler_ws_sm2, this, websocketpp::lib::placeholders::_1));
  1485. m_ws_sm2_server.set_close_handler(websocketpp::lib::bind(&CWebsocketServer::close_handler_ws_sm2, this, websocketpp::lib::placeholders::_1));
  1486. // Listen on port 9003, add m_acceptor->set_option(lib::asio::socket_base::reuse_address(true), bec);
  1487. m_ws_sm2_server.listen(DEFAULT_SERVER_WS_SM2_PORT);
  1488. // Queues a connection accept operation
  1489. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Queues a connection accept operation");
  1490. m_ws_sm2_server.start_accept();
  1491. /*
  1492. // wss
  1493. m_server_wss.set_error_channels(websocketpp::log::elevel::all);
  1494. m_server_wss.set_access_channels(websocketpp::log::alevel::all ^ websocketpp::log::alevel::frame_payload);
  1495. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("m_server_wss.get_max_message_size = %d", m_server_wss.get_max_message_size());
  1496. m_server_wss.set_max_message_size(MAX_TRANSFER_LEN);
  1497. m_server_wss.set_close_handshake_timeout(3000);
  1498. m_server_wss.set_pong_timeout(3000);
  1499. // Initialize Asio
  1500. m_server_wss.init_asio(&m_ios);
  1501. // Set the default message handler to the echo handler
  1502. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Set the default message handler to the echo handler");
  1503. m_server_wss.set_message_handler(websocketpp::lib::bind(&CWebsocketServer::message_handler, this, websocketpp::lib::placeholders::_1, websocketpp::lib::placeholders::_2));
  1504. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Set handles");
  1505. m_server_wss.set_open_handler(websocketpp::lib::bind(&CWebsocketServer::open_handler_wss, this, websocketpp::lib::placeholders::_1));
  1506. m_server_wss.set_close_handler(websocketpp::lib::bind(&CWebsocketServer::close_handler_wss, this, websocketpp::lib::placeholders::_1));
  1507. m_server_wss.set_tls_init_handler(bind(&on_tls_init, MOZILLA_INTERMEDIATE, websocketpp::lib::placeholders::_1));
  1508. // Listen on port 9002, add m_acceptor->set_option(lib::asio::socket_base::reuse_address(true), bec);
  1509. m_server_wss.listen(DEFUALT_SERVER_WSS_PORT);
  1510. // Queues a connection accept operation
  1511. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Queues a connection accept operation");
  1512. m_server_wss.start_accept();
  1513. */
  1514. m_initSuccess = true;
  1515. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("do_run end");
  1516. }
  1517. catch (websocketpp::exception const& e) {
  1518. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("websocketpp exception %s ", e.what());
  1519. m_initSuccess = false;
  1520. }
  1521. catch (...) {
  1522. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("other exception");
  1523. m_initSuccess = false;
  1524. }
  1525. }
  1526. void CWebsocketServer::myTest()
  1527. {
  1528. }
  1529. #if(defined _WIN32 || defined _WIN64)
  1530. std::string CWebsocketServer::utf8_to_string(const std::string& str) {
  1531. int len = MultiByteToWideChar(CP_UTF8, 0, str.c_str(), -1, NULL, 0);
  1532. wchar_t* wszGBK = new wchar_t[len + 1];
  1533. wmemset(wszGBK, 0, len + 1);
  1534. MultiByteToWideChar(CP_UTF8, 0, (LPCTSTR)str.c_str(), -1, wszGBK, len);
  1535. len = WideCharToMultiByte(CP_ACP, 0, wszGBK, -1, NULL, 0, NULL, NULL);
  1536. char* szGBK = new char[len + 1];
  1537. memset(szGBK, 0, len + 1);
  1538. WideCharToMultiByte(CP_ACP, 0, wszGBK, -1, szGBK, len, NULL, NULL);
  1539. std::string strTemp(szGBK);
  1540. delete[] szGBK;
  1541. delete[] wszGBK;
  1542. return strTemp;
  1543. }
  1544. std::string CWebsocketServer::string_to_utf8(const std::string& str) {
  1545. int wcLen = MultiByteToWideChar(CP_ACP, 0, str.c_str(), -1, NULL, 0);
  1546. if (wcLen > 0) {
  1547. wchar_t* pwBuf = new wchar_t[wcLen + 1];
  1548. if (pwBuf == NULL) {
  1549. return std::string();
  1550. }
  1551. memset(pwBuf, 0, sizeof(wchar_t) * (wcLen + 1));
  1552. wcLen = MultiByteToWideChar(CP_ACP, 0, str.c_str(), -1, pwBuf, wcLen);
  1553. if (wcLen <= 0) {
  1554. delete[] pwBuf;
  1555. return std::string();
  1556. }
  1557. std::wstring modifyStr = pwBuf;
  1558. if (modifyBySpecialStr(modifyStr))
  1559. {
  1560. memset(pwBuf, 0, sizeof(wchar_t) * (wcLen + 1));
  1561. memcpy(pwBuf, modifyStr.c_str(), sizeof(wchar_t) * (modifyStr.length()));
  1562. }
  1563. int ucLen = WideCharToMultiByte(CP_UTF8, 0, pwBuf, -1, NULL, 0, NULL, NULL);
  1564. //DbgEx("ucLen = %d", ucLen);
  1565. if (ucLen < 0) {
  1566. delete[] pwBuf;
  1567. return std::string();
  1568. }
  1569. char* pBuf = new char[ucLen + 1];
  1570. if (pBuf == NULL) {
  1571. delete []pwBuf;
  1572. return std::string();
  1573. }
  1574. memset(pBuf, 0, sizeof(char) * (ucLen + 1));
  1575. ucLen = WideCharToMultiByte(CP_UTF8, 0, pwBuf, -1, pBuf, ucLen, NULL, NULL);
  1576. if (ucLen <= 0) {
  1577. delete[] pwBuf;
  1578. delete[] pBuf;
  1579. return std::string();
  1580. }
  1581. std::string retStr(pBuf);
  1582. //DbgEx("string_to_utf8 return: %s", retStr.c_str());
  1583. if (pwBuf) {
  1584. delete[] pwBuf;
  1585. pwBuf = NULL;
  1586. }
  1587. if (pBuf) {
  1588. delete[] pBuf;
  1589. pBuf = NULL;
  1590. }
  1591. return retStr;
  1592. }
  1593. return std::string();
  1594. }
  1595. #endif
  1596. void CWebsocketServer::init_entity_sessions() {
  1597. return;
  1598. }
  1599. }