ArqHelper.h 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791
  1. /*
  2. * Copyright: JessMA Open Source (ldcsaa@gmail.com)
  3. *
  4. * Author : Bruce Liang
  5. * Website : https://github.com/ldcsaa
  6. * Project : https://github.com/ldcsaa/HP-Socket
  7. * Blog : http://www.cnblogs.com/ldcsaa
  8. * Wiki : http://www.oschina.net/p/hp-socket
  9. * QQ Group : 44636872, 75375912
  10. *
  11. * Licensed under the Apache License, Version 2.0 (the "License");
  12. * you may not use this file except in compliance with the License.
  13. * You may obtain a copy of the License at
  14. *
  15. * http://www.apache.org/licenses/LICENSE-2.0
  16. *
  17. * Unless required by applicable law or agreed to in writing, software
  18. * distributed under the License is distributed on an "AS IS" BASIS,
  19. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  20. * See the License for the specific language governing permissions and
  21. * limitations under the License.
  22. */
  23. #pragma once
  24. #include "hpsocket/HPTypeDef.h"
  25. #ifdef _UDP_SUPPORT
  26. #include "SocketHelper.h"
  27. #include "common/FuncHelper.h"
  28. #include "common/BufferPool.h"
  29. #include "common/IODispatcher.h"
  30. #include "common/kcp/ikcp.h"
  31. #include <memory>
  32. using namespace std;
  33. #define DEFAULT_ARQ_NO_DELAY FALSE
  34. #define DEFAULT_ARQ_TURNOFF_NC FALSE
  35. #define DEFAULT_ARQ_FLUSH_INTERVAL 60
  36. #define DEFAULT_ARQ_RESEND_BY_ACKS 0
  37. #define DEFAULT_ARQ_SEND_WND_SIZE 128
  38. #define DEFAULT_ARQ_RECV_WND_SIZE 512
  39. #define DEFAULT_ARQ_MIN_RTO 30
  40. #define DEFAULT_ARQ_FAST_LIMIT 5
  41. #define DEFAULT_ARQ_MAX_TRANS_UNIT DEFAULT_UDP_MAX_DATAGRAM_SIZE
  42. #define DEFAULT_ARQ_MAX_MSG_SIZE DEFAULT_BUFFER_CACHE_CAPACITY
  43. #define DEFAULT_ARQ_HANND_SHAKE_TIMEOUT 5000
  44. #define KCP_HEADER_SIZE 24
  45. #define KCP_MIN_RECV_WND 128
  46. #define ARQ_MAX_HANDSHAKE_INTERVAL 2000
  47. using Fn_ArqOutputProc = int (*)(const char* pBuffer, int iLength, IKCPCB* kcp, LPVOID pv);
  48. DWORD GenerateConversationID();
  49. /************************************************************************
  50. 名称:ARQ 握手状态
  51. 描述:标识当前连接的 ARQ 握手状态
  52. ************************************************************************/
  53. enum EnArqHandShakeStatus
  54. {
  55. ARQ_HSS_INIT = 0, // 初始状态
  56. ARQ_HSS_PROC = 1, // 正在握手
  57. ARQ_HSS_SUCC = 2, // 握手成功
  58. };
  59. struct TArqCmd
  60. {
  61. public:
  62. static const UINT16 MAGIC = 0xBB4F;
  63. static const UINT8 CMD_HANDSHAKE = 0x01;
  64. static const UINT8 FLAG_COMPLETE = 0x01;
  65. static const int PACKAGE_LENGTH = 12;
  66. public:
  67. UINT16 magic;
  68. UINT8 cmd;
  69. UINT8 flag;
  70. DWORD selfID;
  71. DWORD peerID;
  72. public:
  73. static BYTE* MakePackage(UINT8 cmd, UINT8 flag, DWORD selfID, DWORD peerID, UINT16 magic = MAGIC)
  74. {
  75. BYTE* buff = new BYTE[PACKAGE_LENGTH];
  76. *((UINT16*)(buff + 0)) = magic;
  77. *((UINT8*)(buff + 2)) = cmd;
  78. *((UINT8*)(buff + 3)) = flag;
  79. *((DWORD*)(buff + 4)) = selfID;
  80. *((DWORD*)(buff + 8)) = peerID;
  81. return buff;
  82. }
  83. BYTE* MakePackage()
  84. {
  85. return MakePackage(cmd, flag, selfID, peerID, magic);
  86. }
  87. BOOL Parse(const BYTE buff[PACKAGE_LENGTH])
  88. {
  89. magic = *((UINT16*)(buff + 0));
  90. cmd = *((UINT8*) (buff + 2));
  91. flag = *((UINT8*) (buff + 3));
  92. selfID = *((DWORD*) (buff + 4));
  93. peerID = *((DWORD*) (buff + 8));
  94. return IsValid();
  95. }
  96. BOOL IsValid() {return (magic == MAGIC && cmd == CMD_HANDSHAKE && (flag & 0xFE) == 0);}
  97. public:
  98. TArqCmd()
  99. {
  100. ::ZeroMemory(this, sizeof(TArqCmd));
  101. }
  102. TArqCmd(UINT8 c, UINT8 f, DWORD sid, DWORD pid)
  103. : magic (MAGIC)
  104. , cmd (c)
  105. , flag (f)
  106. , selfID(sid)
  107. , peerID(pid)
  108. {
  109. }
  110. };
  111. struct TArqAttr
  112. {
  113. BOOL bNoDelay;
  114. BOOL bTurnoffNc;
  115. DWORD dwResendByAcks;
  116. DWORD dwFlushInterval;
  117. DWORD dwSendWndSize;
  118. DWORD dwRecvWndSize;
  119. DWORD dwMinRto;
  120. DWORD dwMtu;
  121. DWORD dwFastLimit;
  122. DWORD dwMaxMessageSize;
  123. DWORD dwHandShakeTimeout;
  124. public:
  125. TArqAttr( BOOL no_delay = DEFAULT_ARQ_NO_DELAY
  126. , BOOL turnoff_nc = DEFAULT_ARQ_TURNOFF_NC
  127. , DWORD resend_by_acks = DEFAULT_ARQ_RESEND_BY_ACKS
  128. , DWORD flush_interval = DEFAULT_ARQ_FLUSH_INTERVAL
  129. , DWORD send_wnd_size = DEFAULT_ARQ_SEND_WND_SIZE
  130. , DWORD recv_wnd_size = DEFAULT_ARQ_RECV_WND_SIZE
  131. , DWORD min_rto = DEFAULT_ARQ_MIN_RTO
  132. , DWORD mtu = DEFAULT_ARQ_MAX_TRANS_UNIT
  133. , DWORD fast_limit = DEFAULT_ARQ_FAST_LIMIT
  134. , DWORD max_msg_size = DEFAULT_ARQ_MAX_MSG_SIZE
  135. , DWORD hand_shake_timeout = DEFAULT_ARQ_HANND_SHAKE_TIMEOUT
  136. )
  137. : bNoDelay (no_delay)
  138. , bTurnoffNc (turnoff_nc)
  139. , dwResendByAcks (resend_by_acks)
  140. , dwFlushInterval (flush_interval)
  141. , dwSendWndSize (send_wnd_size)
  142. , dwRecvWndSize (recv_wnd_size)
  143. , dwMinRto (min_rto)
  144. , dwMtu (mtu)
  145. , dwFastLimit (fast_limit)
  146. , dwMaxMessageSize (max_msg_size)
  147. , dwHandShakeTimeout(hand_shake_timeout)
  148. {
  149. ASSERT(IsValid());
  150. }
  151. BOOL IsValid() const
  152. {
  153. return ((int)dwResendByAcks >= 0) &&
  154. ((int)dwFlushInterval > 0) &&
  155. ((int)dwSendWndSize > 0) &&
  156. ((int)dwRecvWndSize > 0) &&
  157. ((int)dwMinRto > 0) &&
  158. ((int)dwFastLimit >= 0) &&
  159. ((int)dwHandShakeTimeout > 2 * (int)dwMinRto) &&
  160. ((int)dwMtu >= 3 * KCP_HEADER_SIZE && dwMtu <= MAXIMUM_UDP_MAX_DATAGRAM_SIZE) &&
  161. ((int)dwMaxMessageSize > 0 && dwMaxMessageSize < ((KCP_MIN_RECV_WND - 1) * (dwMtu - KCP_HEADER_SIZE))) ;
  162. }
  163. };
  164. template<class T, class S> class CArqSessionT
  165. {
  166. public:
  167. CArqSessionT* Renew(T* pContext, S* pSocket, const TArqAttr& attr, DWORD dwPeerConvID = 0)
  168. {
  169. m_pContext = pContext;
  170. m_pSocket = pSocket;
  171. m_dwSelfConvID = ::GenerateConversationID();
  172. DoRenew(attr, dwPeerConvID);
  173. RenewExtra(attr);
  174. m_dwCreateTime = ::TimeGetTime();
  175. m_dwHSNextTime = m_dwCreateTime;
  176. m_dwHSSndCount = 0;
  177. m_bHSComplete = FALSE;
  178. m_enStatus = ARQ_HSS_PROC;
  179. Check();
  180. return this;
  181. }
  182. BOOL Reset()
  183. {
  184. if(!IsValid())
  185. return FALSE;
  186. {
  187. CReentrantCriSecLock recvlock(m_csRecv);
  188. CReentrantCriSecLock sendlock(m_csSend);
  189. if(!IsValid())
  190. return FALSE;
  191. m_enStatus = ARQ_HSS_INIT;
  192. DoReset();
  193. }
  194. ResetExtra();
  195. return TRUE;
  196. }
  197. BOOL Check()
  198. {
  199. if(IsReady())
  200. {
  201. if(m_bHSComplete || DoHandShake())
  202. return Flush();
  203. else
  204. return FALSE;
  205. }
  206. else if(IsHandShaking())
  207. return DoHandShake();
  208. else
  209. {
  210. ::SetLastError(ERROR_INVALID_STATE);
  211. return FALSE;
  212. }
  213. }
  214. BOOL DoHandShake()
  215. {
  216. if(!IsValid())
  217. {
  218. ::SetLastError(ERROR_INVALID_STATE);
  219. return FALSE;
  220. }
  221. unique_ptr<BYTE[]> bufCmdPtr;
  222. {
  223. CReentrantCriSecLock recvlock(m_csRecv);
  224. if(!IsValid())
  225. {
  226. ::SetLastError(ERROR_INVALID_STATE);
  227. return FALSE;
  228. }
  229. DWORD dwCurrent = ::TimeGetTime();
  230. if(::GetTimeGap32(m_dwCreateTime, dwCurrent) > m_pContext->GetHandShakeTimeout())
  231. {
  232. ::SetLastError(ERROR_TIMEOUT);
  233. return FALSE;
  234. }
  235. if((int)(::GetTimeGap32(m_dwHSNextTime, dwCurrent)) < 0)
  236. return TRUE;
  237. m_dwHSNextTime = dwCurrent + MIN(m_kcp->interval * (++m_dwHSSndCount), ARQ_MAX_HANDSHAKE_INTERVAL);
  238. UINT8 iFlag = IsReady() ? TArqCmd::FLAG_COMPLETE : 0;
  239. bufCmdPtr.reset(TArqCmd::MakePackage(TArqCmd::CMD_HANDSHAKE, iFlag, m_dwSelfConvID, m_dwPeerConvID));
  240. }
  241. return m_pContext->DoSend(m_pSocket, bufCmdPtr.get(), TArqCmd::PACKAGE_LENGTH);
  242. }
  243. BOOL Flush(BOOL bForce = FALSE)
  244. {
  245. if(!IsReady())
  246. {
  247. ::SetLastError(ERROR_INVALID_STATE);
  248. return FALSE;
  249. }
  250. {
  251. CReentrantCriSecTryLock recvlock(m_csRecv);
  252. if(recvlock.IsValid())
  253. {
  254. CReentrantCriSecTryLock sendlock(m_csSend);
  255. if(sendlock.IsValid())
  256. {
  257. if(!IsReady())
  258. {
  259. ::SetLastError(ERROR_INVALID_STATE);
  260. return FALSE;
  261. }
  262. if(bForce)
  263. ::ikcp_flush(m_kcp);
  264. else
  265. ::ikcp_update(m_kcp, ::TimeGetTime());
  266. }
  267. }
  268. }
  269. return TRUE;
  270. }
  271. int Send(const BYTE* pBuffer, int iLength)
  272. {
  273. if(!IsReady())
  274. return ERROR_INVALID_STATE;
  275. int rs = NO_ERROR;
  276. {
  277. CReentrantCriSecLock sendlock(m_csSend);
  278. if(!IsReady())
  279. return ERROR_INVALID_STATE;
  280. rs = ::ikcp_send(m_kcp, (const char*)pBuffer, iLength);
  281. if(rs < 0) rs = ERROR_INCORRECT_SIZE;
  282. }
  283. if(rs == NO_ERROR)
  284. Flush(TRUE);
  285. return rs;
  286. }
  287. int GetWaitingSend()
  288. {
  289. if(!IsValid())
  290. {
  291. ::SetLastError(ERROR_INVALID_STATE);
  292. return -1;
  293. }
  294. CReentrantCriSecLock sendlock(m_csSend);
  295. if(!IsValid())
  296. {
  297. ::SetLastError(ERROR_INVALID_STATE);
  298. return -1;
  299. }
  300. return ::ikcp_waitsnd(m_kcp);
  301. }
  302. EnHandleResult Receive(const BYTE* pData, int iLength, BYTE* pBuffer, int iCapacity)
  303. {
  304. if(iLength >= KCP_HEADER_SIZE)
  305. return ReceiveArq(pData, iLength, pBuffer, iCapacity);
  306. else if(iLength == TArqCmd::PACKAGE_LENGTH)
  307. return ReceiveHandShake(pData);
  308. else
  309. {
  310. ::WSASetLastError(ERROR_INVALID_DATA);
  311. return HR_ERROR;
  312. }
  313. }
  314. EnHandleResult ReceiveHandShake(const BYTE* pBuffer)
  315. {
  316. TArqCmd cmd;
  317. cmd.Parse(pBuffer);
  318. if(!cmd.IsValid())
  319. {
  320. ::WSASetLastError(ERROR_INVALID_DATA);
  321. return HR_ERROR;
  322. }
  323. {
  324. CReentrantCriSecLock recvlock(m_csRecv);
  325. if(!IsValid())
  326. {
  327. ::WSASetLastError(ERROR_INVALID_STATE);
  328. return HR_ERROR;
  329. }
  330. if(IsReady())
  331. {
  332. BOOL bReset = FALSE;
  333. if(cmd.selfID != m_dwPeerConvID)
  334. bReset = TRUE;
  335. else if(cmd.peerID != m_dwSelfConvID)
  336. {
  337. if(cmd.peerID != 0)
  338. bReset = TRUE;
  339. else
  340. {
  341. if(::GetTimeGap32(m_dwCreateTime) > 2 * m_pContext->GetHandShakeTimeout())
  342. bReset = TRUE;
  343. }
  344. }
  345. if(bReset)
  346. {
  347. ::WSASetLastError(ERROR_CONNRESET);
  348. return HR_ERROR;
  349. }
  350. }
  351. else
  352. {
  353. if(m_dwPeerConvID == 0)
  354. {
  355. m_dwPeerConvID = cmd.selfID;
  356. m_dwHSNextTime = ::TimeGetTime();
  357. m_dwHSSndCount = 0;
  358. }
  359. else if(cmd.selfID != m_dwPeerConvID)
  360. {
  361. ::WSASetLastError(ERROR_CONNRESET);
  362. return HR_ERROR;
  363. }
  364. if(cmd.peerID == m_dwSelfConvID)
  365. {
  366. m_enStatus = ARQ_HSS_SUCC;
  367. return m_pContext->DoFireHandShake(m_pSocket);
  368. }
  369. else if(cmd.peerID != 0)
  370. {
  371. ::WSASetLastError(ERROR_CONNRESET);
  372. return HR_ERROR;
  373. }
  374. }
  375. }
  376. if(!m_bHSComplete && cmd.flag == TArqCmd::FLAG_COMPLETE)
  377. m_bHSComplete = TRUE;
  378. DoHandShake();
  379. return HR_OK;
  380. }
  381. EnHandleResult ReceiveArq(const BYTE* pData, int iLength, BYTE* pBuffer, int iCapacity)
  382. {
  383. if(!IsReady()) return HR_IGNORE;
  384. {
  385. CReentrantCriSecLock recvlock(m_csRecv);
  386. if(!IsReady())
  387. {
  388. ::WSASetLastError(ERROR_INVALID_STATE);
  389. return HR_ERROR;
  390. }
  391. if(iLength < KCP_HEADER_SIZE)
  392. {
  393. ::WSASetLastError(ERROR_INVALID_DATA);
  394. return HR_ERROR;
  395. }
  396. int rs = ::ikcp_input(m_kcp, (const char*)pData, iLength);
  397. if(rs != NO_ERROR)
  398. {
  399. ::WSASetLastError(ERROR_INVALID_DATA);
  400. return HR_ERROR;
  401. }
  402. while(TRUE)
  403. {
  404. int iRead = ::ikcp_recv(m_kcp, (char*)pBuffer, iCapacity);
  405. if(iRead >= 0)
  406. {
  407. EnHandleResult result = m_pContext->DoFireReceive(m_pSocket, pBuffer, iRead);
  408. if(result == HR_ERROR)
  409. return result;
  410. }
  411. else if(iRead == -3)
  412. {
  413. ::WSASetLastError(ERROR_INCORRECT_SIZE);
  414. return HR_ERROR;
  415. }
  416. else
  417. break;
  418. }
  419. }
  420. Flush(TRUE);
  421. return HR_OK;
  422. }
  423. private:
  424. void DoRenew(const TArqAttr& attr, DWORD dwPeerConvID = 0)
  425. {
  426. ASSERT(attr.IsValid());
  427. DoReset();
  428. m_dwPeerConvID = dwPeerConvID;
  429. m_kcp = ::ikcp_create(m_dwSelfConvID, m_pSocket);
  430. ::ikcp_nodelay(m_kcp, attr.bNoDelay ? 1 : 0, (int)attr.dwFlushInterval, (int)attr.dwResendByAcks, attr.bTurnoffNc ? 1 : 0);
  431. ::ikcp_wndsize(m_kcp, (int)attr.dwSendWndSize, (int)attr.dwRecvWndSize);
  432. ::ikcp_setmtu(m_kcp, attr.dwMtu);
  433. m_kcp->rx_minrto = (int)attr.dwMinRto;
  434. m_kcp->fastlimit = (int)attr.dwFastLimit;
  435. m_kcp->output = m_pContext->GetArqOutputProc();
  436. }
  437. void DoReset()
  438. {
  439. if(m_kcp != nullptr)
  440. {
  441. ::ikcp_release(m_kcp);
  442. m_kcp = nullptr;
  443. }
  444. }
  445. public:
  446. BOOL IsValid() const {return GetStatus() != ARQ_HSS_INIT;}
  447. BOOL IsHandShaking() const {return GetStatus() == ARQ_HSS_PROC;}
  448. BOOL IsReady() const {return GetStatus() == ARQ_HSS_SUCC;}
  449. IKCPCB* GetKcp() {return m_kcp;}
  450. DWORD GetConvID() const {if(!IsValid()) return 0; return m_kcp->conv;}
  451. DWORD GetSelfConvID() const {return m_dwSelfConvID;}
  452. DWORD GetPeerConvID() const {return m_dwPeerConvID;}
  453. EnArqHandShakeStatus GetStatus() const {return m_enStatus;}
  454. protected:
  455. virtual void RenewExtra(const TArqAttr& attr) {}
  456. virtual void ResetExtra() {}
  457. public:
  458. CArqSessionT()
  459. : m_pContext (nullptr)
  460. , m_pSocket (nullptr)
  461. , m_kcp (nullptr)
  462. , m_enStatus (ARQ_HSS_INIT)
  463. , m_dwSelfConvID(0)
  464. , m_dwPeerConvID(0)
  465. , m_dwCreateTime(0)
  466. , m_dwHSNextTime(0)
  467. , m_dwHSSndCount(0)
  468. , m_bHSComplete (FALSE)
  469. {
  470. }
  471. virtual ~CArqSessionT()
  472. {
  473. Reset();
  474. }
  475. static CArqSessionT* Construct()
  476. {return new CArqSessionT();}
  477. static void Destruct(CArqSessionT* pSession)
  478. {if(pSession) delete pSession;}
  479. protected:
  480. T* m_pContext;
  481. S* m_pSocket;
  482. private:
  483. BOOL m_bHSComplete;
  484. DWORD m_dwHSNextTime;
  485. DWORD m_dwHSSndCount;
  486. DWORD m_dwCreateTime;
  487. DWORD m_dwSelfConvID;
  488. DWORD m_dwPeerConvID;
  489. EnArqHandShakeStatus m_enStatus;
  490. CReentrantCriSec m_csRecv;
  491. CReentrantCriSec m_csSend;
  492. IKCPCB* m_kcp;
  493. };
  494. template<class T, class S> class CArqSessionPoolT;
  495. template<class T, class S> class CArqSessionExT : public CArqSessionT<T, S>, public CSafeCounter
  496. {
  497. using __super = CArqSessionT<T, S>;
  498. using __super::Reset;
  499. friend class CArqSessionPoolT<T, S>;
  500. public:
  501. DWORD GetFreeTime () const {return m_dwFreeTime;}
  502. FD GetTimer () const {return m_fdTimer;}
  503. protected:
  504. virtual void RenewExtra(const TArqAttr& attr)
  505. {
  506. ResetCount();
  507. m_fdTimer = m_ioDispatcher.AddTimer(attr.dwFlushInterval, this);
  508. ASSERT(IS_VALID_FD(m_fdTimer));
  509. }
  510. virtual void ResetExtra()
  511. {
  512. m_ioDispatcher.DelTimer(m_fdTimer);
  513. m_dwFreeTime = ::TimeGetTime();
  514. m_fdTimer = INVALID_FD;
  515. }
  516. public:
  517. CArqSessionExT(CIODispatcher& ioDispatcher)
  518. : m_ioDispatcher(ioDispatcher)
  519. , m_fdTimer (INVALID_FD)
  520. , m_dwFreeTime (0)
  521. {
  522. }
  523. virtual ~CArqSessionExT()
  524. {
  525. Reset();
  526. }
  527. static CArqSessionExT* Construct(CIODispatcher& ioDispatcher)
  528. {return new CArqSessionExT(ioDispatcher);}
  529. static void Destruct(CArqSessionExT* pSession)
  530. {if(pSession) delete pSession;}
  531. private:
  532. CIODispatcher& m_ioDispatcher;
  533. FD m_fdTimer;
  534. DWORD m_dwFreeTime;
  535. };
  536. template<class T, class S> class CArqSessionPoolT : private CIOHandler
  537. {
  538. using CArqSessionEx = CArqSessionExT<T, S>;
  539. using TArqSessionList = CRingPool<CArqSessionEx>;
  540. using TArqSessionQueue = CCASQueue<CArqSessionEx>;
  541. public:
  542. CArqSessionEx* PickFreeSession(T* pContext, S* pSocket, const TArqAttr& attr)
  543. {
  544. DWORD dwIndex;
  545. CArqSessionEx* pSession = nullptr;
  546. if(m_lsFreeSession.TryLock(&pSession, dwIndex))
  547. {
  548. if(::GetTimeGap32(pSession->GetFreeTime()) >= m_dwSessionLockTime)
  549. ENSURE(m_lsFreeSession.ReleaseLock(nullptr, dwIndex));
  550. else
  551. {
  552. ENSURE(m_lsFreeSession.ReleaseLock(pSession, dwIndex));
  553. pSession = nullptr;
  554. }
  555. }
  556. if(!pSession) pSession = CArqSessionEx::Construct(m_ioDispatcher);
  557. ASSERT(pSession);
  558. return (CArqSessionEx*)pSession->Renew(pContext, pSocket, attr);
  559. }
  560. void PutFreeSession(CArqSessionEx* pSession)
  561. {
  562. if(pSession->Reset())
  563. {
  564. #ifndef USE_EXTERNAL_GC
  565. ReleaseGCSession();
  566. #endif
  567. if(!m_lsFreeSession.TryPut(pSession))
  568. m_lsGCSession.PushBack(pSession);
  569. }
  570. }
  571. void Prepare()
  572. {
  573. m_lsFreeSession.Reset(m_dwSessionPoolSize);
  574. m_ioDispatcher.Start(this, m_pContext->GetPostReceiveCount(), m_pContext->GetWorkerThreadCount());
  575. }
  576. void Clear()
  577. {
  578. m_ioDispatcher.Stop();
  579. m_lsFreeSession.Clear();
  580. ReleaseGCSession(TRUE);
  581. ENSURE(m_lsGCSession.IsEmpty());
  582. }
  583. void ReleaseGCSession(BOOL bForce = FALSE)
  584. {
  585. ::ReleaseGCObj(m_lsGCSession, m_dwSessionLockTime, bForce);
  586. }
  587. virtual BOOL OnReadyRead(const TDispContext* pContext, PVOID pv, UINT events) override
  588. {
  589. if(events & _EPOLL_ALL_ERROR_EVENTS)
  590. return FALSE;
  591. CArqSessionEx* pSession = (CArqSessionEx*)pv;
  592. CLocalSafeCounter localcounter(*pSession);
  593. if(!pSession->Check() && pSession->IsValid() && TUdpSocketObj::IsValid(pSession->m_pSocket))
  594. pSession->m_pContext->Disconnect(pSession->m_pSocket->connID);
  595. ::ReadTimer(pSession->GetTimer());
  596. return TRUE;
  597. }
  598. public:
  599. void SetSessionLockTime (DWORD dwSessionLockTime) {m_dwSessionLockTime = dwSessionLockTime;}
  600. void SetSessionPoolSize (DWORD dwSessionPoolSize) {m_dwSessionPoolSize = dwSessionPoolSize;}
  601. void SetSessionPoolHold (DWORD dwSessionPoolHold) {m_dwSessionPoolHold = dwSessionPoolHold;}
  602. DWORD GetSessionLockTime() {return m_dwSessionLockTime;}
  603. DWORD GetSessionPoolSize() {return m_dwSessionPoolSize;}
  604. DWORD GetSessionPoolHold() {return m_dwSessionPoolHold;}
  605. public:
  606. CArqSessionPoolT(T* pContext,
  607. DWORD dwPoolSize = DEFAULT_SESSION_POOL_SIZE,
  608. DWORD dwPoolHold = DEFAULT_SESSION_POOL_HOLD,
  609. DWORD dwLockTime = DEFAULT_SESSION_LOCK_TIME)
  610. : m_pContext(pContext)
  611. , m_dwSessionPoolSize(dwPoolSize)
  612. , m_dwSessionPoolHold(dwPoolHold)
  613. , m_dwSessionLockTime(dwLockTime)
  614. {
  615. }
  616. ~CArqSessionPoolT() {Clear();}
  617. DECLARE_NO_COPY_CLASS(CArqSessionPoolT)
  618. public:
  619. static const DWORD DEFAULT_SESSION_LOCK_TIME;
  620. static const DWORD DEFAULT_SESSION_POOL_SIZE;
  621. static const DWORD DEFAULT_SESSION_POOL_HOLD;
  622. private:
  623. T* m_pContext;
  624. DWORD m_dwSessionLockTime;
  625. DWORD m_dwSessionPoolSize;
  626. DWORD m_dwSessionPoolHold;
  627. TArqSessionList m_lsFreeSession;
  628. TArqSessionQueue m_lsGCSession;
  629. CIODispatcher m_ioDispatcher;
  630. };
  631. template<class T, class S> const DWORD CArqSessionPoolT<T, S>::DEFAULT_SESSION_LOCK_TIME = DEFAULT_OBJECT_CACHE_LOCK_TIME;
  632. template<class T, class S> const DWORD CArqSessionPoolT<T, S>::DEFAULT_SESSION_POOL_SIZE = DEFAULT_OBJECT_CACHE_POOL_SIZE;
  633. template<class T, class S> const DWORD CArqSessionPoolT<T, S>::DEFAULT_SESSION_POOL_HOLD = DEFAULT_OBJECT_CACHE_POOL_HOLD;
  634. #endif