CWebsocketServer.cpp 61 KB

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