开源的socket服务端客户端,支持C# C++

SocketHelper.h 18KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613
  1. /*
  2. * Copyright: JessMA Open Source (ldcsaa@gmail.com)
  3. *
  4. * Version : 4.1.3
  5. * Author : Bruce Liang
  6. * Website : http://www.jessma.org
  7. * Project : https://github.com/ldcsaa
  8. * Blog : http://www.cnblogs.com/ldcsaa
  9. * Wiki : http://www.oschina.net/p/hp-socket
  10. * QQ Group : 75375912
  11. *
  12. * Licensed under the Apache License, Version 2.0 (the "License");
  13. * you may not use this file except in compliance with the License.
  14. * You may obtain a copy of the License at
  15. *
  16. * http://www.apache.org/licenses/LICENSE-2.0
  17. *
  18. * Unless required by applicable law or agreed to in writing, software
  19. * distributed under the License is distributed on an "AS IS" BASIS,
  20. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  21. * See the License for the specific language governing permissions and
  22. * limitations under the License.
  23. */
  24. #pragma once
  25. #include <ws2tcpip.h>
  26. #include <mswsock.h>
  27. #include <malloc.h>
  28. #include <atlfile.h>
  29. #include "SocketInterface.h"
  30. #include "../../Common/Src/WaitFor.h"
  31. #include "../../Common/Src/bufferpool.h"
  32. #include "../../Common/Src/RingBuffer.h"
  33. /************************************************************************
  34. 名称:全局常量
  35. 描述:声明组件的公共全局常量
  36. ************************************************************************/
  37. #define HP_VERSION_MAJOR 4
  38. #define HP_VERSION_MINOR 1
  39. #define HP_VERSION_REVISE 3
  40. #define HP_VERSION_BUILD 1
  41. /* IOCP 最大工作线程数 */
  42. extern const DWORD MAX_WORKER_THREAD_COUNT;
  43. /* IOCP Socket 缓冲区最小值 */
  44. extern const DWORD MIN_SOCKET_BUFFER_SIZE;
  45. /* 小文件最大字节数 */
  46. extern const DWORD MAX_SMALL_FILE_SIZE;
  47. /* 最大连接时长 */
  48. extern const DWORD MAX_CONNECTION_PERIOD;
  49. /* Server/Agent 默认最大连接数 */
  50. extern const DWORD DEFAULT_MAX_CONNECTION_COUNT;
  51. /* Server/Agent 默认 IOCP 工作线程数 */
  52. extern const DWORD DEFAULT_WORKER_THREAD_COUNT;
  53. /* Server/Agent 默认 Socket 缓存对象锁定时间 */
  54. extern const DWORD DEFAULT_FREE_SOCKETOBJ_LOCK_TIME;
  55. /* Server/Agent 默认 Socket 缓存池大小 */
  56. extern const DWORD DEFAULT_FREE_SOCKETOBJ_POOL;
  57. /* Server/Agent 默认 Socket 缓存池回收阀值 */
  58. extern const DWORD DEFAULT_FREE_SOCKETOBJ_HOLD;
  59. /* Server/Agent 默认内存块缓存池大小 */
  60. extern const DWORD DEFAULT_FREE_BUFFEROBJ_POOL;
  61. /* Server/Agent 默认内存块缓存池回收阀值 */
  62. extern const DWORD DEFAULT_FREE_BUFFEROBJ_HOLD;
  63. /* Client 默认内存块缓存池大小 */
  64. extern const DWORD DEFAULT_CLIENT_FREE_BUFFER_POOL_SIZE;
  65. /* Client 默认内存块缓存池回收阀值 */
  66. extern const DWORD DEFAULT_CLIENT_FREE_BUFFER_POOL_HOLD;
  67. /* Agent 默认绑定地址 */
  68. extern LPCTSTR DEFAULT_BIND_ADDRESS;
  69. /* TCP 默认通信数据缓冲区大小 */
  70. extern const DWORD DEFAULT_TCP_SOCKET_BUFFER_SIZE;
  71. /* TCP 默认心跳包间隔 */
  72. extern const DWORD DEFALUT_TCP_KEEPALIVE_TIME;
  73. /* TCP 默认心跳确认包检测间隔 */
  74. extern const DWORD DEFALUT_TCP_KEEPALIVE_INTERVAL;
  75. /* TCP Server 默认 Listen 队列大小 */
  76. extern const DWORD DEFAULT_TCP_SERVER_SOCKET_LISTEN_QUEUE;
  77. /* TCP Server 默认预投递 Accept 数量 */
  78. extern const DWORD DEFAULT_TCP_SERVER_ACCEPT_SOCKET_COUNT;
  79. /* UDP 默认数据报文最大长度 */
  80. extern const DWORD DEFAULT_UDP_MAX_DATAGRAM_SIZE;
  81. /* UDP 默认 Receive 预投递数量 */
  82. extern const DWORD DEFAULT_UDP_POST_RECEIVE_COUNT;
  83. /* UDP 默认监测包尝试次数 */
  84. extern const DWORD DEFAULT_UDP_DETECT_ATTEMPTS;
  85. /* UDP 默认监测包发送间隔 */
  86. extern const DWORD DEFAULT_UDP_DETECT_INTERVAL;
  87. /* TCP Pack 包长度位数 */
  88. extern const DWORD TCP_PACK_LENGTH_BITS;
  89. /* TCP Pack 包长度掩码 */
  90. extern const DWORD TCP_PACK_LENGTH_MASK;
  91. /* TCP Pack 包最大长度硬限制 */
  92. extern const DWORD TCP_PACK_MAX_SIZE_LIMIT;
  93. /* TCP Pack 包默认最大长度 */
  94. extern const DWORD TCP_PACK_DEFAULT_MAX_SIZE;
  95. /* TCP Pack 包头标识值硬限制 */
  96. extern const USHORT TCP_PACK_HEADER_FLAG_LIMIT;
  97. /* TCP Pack 包头默认标识值 */
  98. extern const USHORT TCP_PACK_DEFAULT_HEADER_FLAG;
  99. /************************************************************************
  100. 名称:Windows Socket 组件初始化类
  101. 描述:自动加载和卸载 Windows Socket 组件
  102. ************************************************************************/
  103. class CInitSocket
  104. {
  105. public:
  106. CInitSocket(LPWSADATA lpWSAData = nullptr, BYTE minorVersion = 2, BYTE majorVersion = 2)
  107. {
  108. LPWSADATA lpTemp = lpWSAData;
  109. if(!lpTemp)
  110. lpTemp = (LPWSADATA)_alloca(sizeof(WSADATA));
  111. m_iResult = ::WSAStartup(MAKEWORD(majorVersion, minorVersion), lpTemp);
  112. }
  113. ~CInitSocket()
  114. {
  115. if(IsValid())
  116. ::WSACleanup();
  117. }
  118. int GetResult() {return m_iResult;}
  119. BOOL IsValid() {return m_iResult == 0;}
  120. private:
  121. int m_iResult;
  122. };
  123. /* Server 组件和 Agent 组件内部使用的事件处理结果常量 */
  124. // 连接已关闭
  125. #define HR_CLOSED 0xFF
  126. /* 关闭连接标识 */
  127. enum EnSocketCloseFlag
  128. {
  129. SCF_NONE = 0, // 不触发事件
  130. SCF_CLOSE = 1, // 触发 正常关闭 OnClose 事件
  131. SCF_ERROR = 2 // 触发 异常关闭 OnClose 事件
  132. };
  133. /* 数据缓冲区基础结构 */
  134. template<class T> struct TBufferObjBase
  135. {
  136. WSAOVERLAPPED ov;
  137. EnSocketOperation operation;
  138. WSABUF buff;
  139. int capacity;
  140. CPrivateHeap& heap;
  141. T* next;
  142. T* last;
  143. static T* Construct(CPrivateHeap& heap, DWORD dwCapacity)
  144. {
  145. T* pBufferObj = (T*)heap.Alloc(sizeof(T) + dwCapacity);
  146. ASSERT(pBufferObj);
  147. ZeroMemory(pBufferObj, sizeof(T));
  148. pBufferObj->TBufferObjBase::TBufferObjBase(heap, dwCapacity);
  149. pBufferObj->buff.buf = ((char*)pBufferObj) + sizeof(T);
  150. return pBufferObj;
  151. }
  152. static void Destruct(T* pBufferObj)
  153. {
  154. ASSERT(pBufferObj);
  155. pBufferObj->heap.Free(pBufferObj);
  156. }
  157. TBufferObjBase(CPrivateHeap& hp, DWORD dwCapacity)
  158. : heap(hp)
  159. , capacity((int)dwCapacity)
  160. {
  161. ASSERT(capacity > 0);
  162. }
  163. int Cat(const BYTE* pData, int length)
  164. {
  165. ASSERT(pData != nullptr && length > 0);
  166. int cat = min(Remain(), length);
  167. if(cat > 0)
  168. {
  169. memcpy(buff.buf + buff.len, pData, cat);
  170. buff.len += cat;
  171. }
  172. return cat;
  173. }
  174. void Reset() {buff.len = 0;}
  175. int Remain() {return capacity - buff.len;}
  176. BOOL IsFull() {return buff.len == capacity;}
  177. };
  178. /* 数据缓冲区结构 */
  179. struct TBufferObj : public TBufferObjBase<TBufferObj>
  180. {
  181. SOCKET client;
  182. };
  183. /* UDP 数据缓冲区结构 */
  184. struct TUdpBufferObj : public TBufferObjBase<TUdpBufferObj>
  185. {
  186. SOCKADDR_IN remoteAddr;
  187. int addrLen;
  188. };
  189. /* 数据缓冲区链表模板 */
  190. template<class T> struct TBufferObjListT : public TSimpleList<T>
  191. {
  192. public:
  193. int Cat(const BYTE* pData, int length)
  194. {
  195. ASSERT(pData != nullptr && length > 0);
  196. int remain = length;
  197. while(remain > 0)
  198. {
  199. T* pItem = Back();
  200. if(pItem == nullptr || pItem->IsFull())
  201. pItem = PushBack(bfPool.PickFreeItem());
  202. int cat = pItem->Cat(pData, remain);
  203. pData += cat;
  204. remain -= cat;
  205. }
  206. return length;
  207. }
  208. T* PushTail(const BYTE* pData, int length)
  209. {
  210. ASSERT(pData != nullptr && length > 0 && length <= (int)bfPool.GetItemCapacity());
  211. T* pItem = PushBack(bfPool.PickFreeItem());
  212. pItem->Cat(pData, length);
  213. return pItem;
  214. }
  215. void Release()
  216. {
  217. bfPool.PutFreeItem(*this);
  218. }
  219. public:
  220. TBufferObjListT(CNodePoolT<T>& pool) : bfPool(pool)
  221. {
  222. }
  223. private:
  224. CNodePoolT<T>& bfPool;
  225. };
  226. /* 数据缓冲区对象池 */
  227. typedef CNodePoolT<TBufferObj> CBufferObjPool;
  228. /* UDP 数据缓冲区对象池 */
  229. typedef CNodePoolT<TUdpBufferObj> CUdpBufferObjPool;
  230. /* 数据缓冲区链表模板 */
  231. typedef TBufferObjListT<TBufferObj> TBufferObjList;
  232. /* UDP 数据缓冲区链表模板 */
  233. typedef TBufferObjListT<TUdpBufferObj> TUdpBufferObjList;
  234. /* 数据缓冲区结构链表 */
  235. typedef CRingPool<TBufferObj> TBufferObjPtrList;
  236. /* Udp 数据缓冲区结构链表 */
  237. typedef CRingPool<TUdpBufferObj> TUdpBufferObjPtrList;
  238. /* Socket 缓冲区基础结构 */
  239. struct TSocketObjBase
  240. {
  241. CONNID connID;
  242. SOCKADDR_IN remoteAddr;
  243. PVOID extra;
  244. PVOID reserved;
  245. PVOID reserved2;
  246. BOOL valid;
  247. union
  248. {
  249. DWORD freeTime;
  250. DWORD connTime;
  251. };
  252. DWORD activeTime;
  253. CCriSec csSend;
  254. volatile BOOL smooth;
  255. volatile long pending;
  256. volatile long sndCount;
  257. CReentrantSpinGuard csRecv;
  258. static BOOL IsExist(TSocketObjBase* pSocketObj)
  259. {return pSocketObj != nullptr;}
  260. static BOOL IsValid(TSocketObjBase* pSocketObj)
  261. {return pSocketObj != nullptr && pSocketObj->valid;}
  262. static void Invalid(TSocketObjBase* pSocketObj)
  263. {ASSERT(IsExist(pSocketObj)); pSocketObj->valid = FALSE;}
  264. static BOOL IsSmooth(TSocketObjBase* pSocketObj)
  265. {ASSERT(IsExist(pSocketObj)); return pSocketObj->valid && pSocketObj->smooth;}
  266. static BOOL IsPending(TSocketObjBase* pSocketObj)
  267. {ASSERT(IsExist(pSocketObj)); return pSocketObj->valid && pSocketObj->pending > 0;}
  268. static void Release(TSocketObjBase* pSocketObj)
  269. {
  270. ASSERT(IsExist(pSocketObj));
  271. pSocketObj->freeTime = ::TimeGetTime();
  272. }
  273. int Pending() {return pending;}
  274. void Reset(CONNID dwConnID)
  275. {
  276. connID = dwConnID;
  277. valid = TRUE;
  278. smooth = TRUE;
  279. pending = 0;
  280. sndCount = 0;
  281. extra = nullptr;
  282. reserved = nullptr;
  283. reserved2= nullptr;
  284. }
  285. };
  286. /* 数据缓冲区结构 */
  287. struct TSocketObj : public TSocketObjBase
  288. {
  289. SOCKET socket;
  290. CStringA host;
  291. TBufferObjList sndBuff;
  292. TSocketObj(CBufferObjPool& bfPool)
  293. : sndBuff(bfPool)
  294. {
  295. }
  296. static void Release(TSocketObj* pSocketObj)
  297. {
  298. __super::Release(pSocketObj);
  299. pSocketObj->sndBuff.Release();
  300. }
  301. void Reset(CONNID dwConnID, SOCKET soClient)
  302. {
  303. __super::Reset(dwConnID);
  304. host.Empty();
  305. socket = soClient;
  306. }
  307. };
  308. /* UDP 数据缓冲区结构 */
  309. struct TUdpSocketObj : public TSocketObjBase
  310. {
  311. TUdpBufferObjList sndBuff;
  312. volatile DWORD detectFails;
  313. TUdpSocketObj(CUdpBufferObjPool& bfPool)
  314. : sndBuff(bfPool)
  315. {
  316. }
  317. static void Release(TUdpSocketObj* pSocketObj)
  318. {
  319. __super::Release(pSocketObj);
  320. pSocketObj->sndBuff.Release();
  321. }
  322. void Reset(CONNID dwConnID)
  323. {
  324. __super::Reset(dwConnID);
  325. detectFails = 0;
  326. }
  327. };
  328. /* 有效 TSocketObj 缓存 */
  329. typedef CRingCache2<TSocketObj, CONNID, true> TSocketObjPtrPool;
  330. /* 失效 TSocketObj 缓存 */
  331. typedef CRingPool<TSocketObj> TSocketObjPtrList;
  332. /* 失效 TSocketObj 垃圾回收结构链表 */
  333. typedef CCASQueue<TSocketObj> TSocketObjPtrQueue;
  334. /* 有效 TUdpSocketObj 缓存 */
  335. typedef CRingCache2<TUdpSocketObj, CONNID, true> TUdpSocketObjPtrPool;
  336. /* 失效 TUdpSocketObj 缓存 */
  337. typedef CRingPool<TUdpSocketObj> TUdpSocketObjPtrList;
  338. /* 失效 TUdpSocketObj 垃圾回收结构链表 */
  339. typedef CCASQueue<TUdpSocketObj> TUdpSocketObjPtrQueue;
  340. /* SOCKADDR_IN 比较器 */
  341. struct sockaddr_func
  342. {
  343. struct hash
  344. {
  345. size_t operator() (const SOCKADDR_IN* pA) const
  346. {
  347. return ((pA->sin_family << 16) | ntohs(pA->sin_port)) ^ pA->sin_addr.s_addr;
  348. }
  349. };
  350. struct equal_to
  351. {
  352. bool operator () (const SOCKADDR_IN* pA, const SOCKADDR_IN* pB) const
  353. {
  354. return memcmp(pA, pB, offsetof(SOCKADDR_IN, sin_zero)) == 0;
  355. }
  356. };
  357. };
  358. /* 地址-连接 ID 哈希表 */
  359. typedef unordered_map<SOCKADDR_IN*, CONNID, sockaddr_func::hash, sockaddr_func::equal_to>
  360. TSockAddrMap;
  361. /* 地址-连接 ID 哈希表迭代器 */
  362. typedef TSockAddrMap::iterator TSockAddrMapI;
  363. /* 地址-连接 ID 哈希表 const 迭代器 */
  364. typedef TSockAddrMap::const_iterator TSockAddrMapCI;
  365. /* IClient 组件关闭上下文 */
  366. struct TClientCloseContext
  367. {
  368. BOOL bFireOnClose;
  369. EnSocketOperation enOperation;
  370. int iErrorCode;
  371. TClientCloseContext(BOOL bFire = TRUE, EnSocketOperation enOp = SO_CLOSE, int iCode = SE_OK)
  372. {
  373. Reset(bFire, enOp, iCode);
  374. }
  375. void Reset(BOOL bFire = TRUE, EnSocketOperation enOp = SO_CLOSE, int iCode = SE_OK)
  376. {
  377. bFireOnClose = bFire;
  378. enOperation = enOp;
  379. iErrorCode = iCode;
  380. }
  381. };
  382. /*****************************************************************************************************/
  383. /******************************************** 公共帮助方法 ********************************************/
  384. /*****************************************************************************************************/
  385. // 获取 HPSocket 版本号(4 个字节分别为:主版本号,子版本号,修正版本号,构建编号)
  386. DWORD GetHPSocketVersion();
  387. /* 获取错误描述文本 */
  388. LPCTSTR GetSocketErrorDesc(EnSocketError enCode);
  389. /* IPv4 字符串地址转换为整数 */
  390. ULONG GetIPv4InAddr(LPCTSTR lpszAddress);
  391. /* 检查字符串是否符合 IP 地址格式 */
  392. BOOL IsIPAddress(LPCTSTR lpszAddress);
  393. /* 通过主机名获取 IP 地址 */
  394. BOOL GetIPAddress(LPCTSTR lpszHost, __out LPTSTR lpszIP, __inout int& iIPLenth);
  395. /* 通过主机名获取最优的 IP 地址 */
  396. BOOL GetOptimalIPByHostName(LPCTSTR lpszHost, __out IN_ADDR& addr);
  397. /* 获取 IN_ADDR 结构的 IP 地址 */
  398. BOOL IN_ADDR_2_IP(const IN_ADDR& addr, __out LPTSTR lpszAddress, __inout int& iAddressLen);
  399. /* 把 SOCKADDR_IN 结构转换为地址数据 */
  400. BOOL sockaddr_IN_2_A(const SOCKADDR_IN& addr, __out ADDRESS_FAMILY& usFamily, __out LPTSTR lpszAddress, __inout int& iAddressLen, __out USHORT& usPort);
  401. /* 把地址数据转换为 SOCKADDR_IN 结构 */
  402. BOOL sockaddr_A_2_IN(ADDRESS_FAMILY usFamily, LPCTSTR lpszAddress, USHORT usPort, __out SOCKADDR_IN& addr);
  403. /* 获取 Socket 的本地或远程地址信息 */
  404. BOOL GetSocketAddress(SOCKET socket, __out LPTSTR lpszAddress, __inout int& iAddressLen, __out USHORT& usPort, BOOL bLocal = TRUE);
  405. /* 获取 Socket 的本地地址信息 */
  406. BOOL GetSocketLocalAddress(SOCKET socket, __out LPTSTR lpszAddress, __inout int& iAddressLen, __out USHORT& usPort);
  407. /* 获取 Socket 的远程地址信息 */
  408. BOOL GetSocketRemoteAddress(SOCKET socket, __out LPTSTR lpszAddress, __inout int& iAddressLen, __out USHORT& usPort);
  409. /* 64 位网络字节序转主机字节序 */
  410. ULONGLONG NToH64(ULONGLONG value);
  411. /* 64 位主机字节序转网络字节序 */
  412. ULONGLONG HToN64(ULONGLONG value);
  413. /* 获取 Socket 的某个扩展函数的指针 */
  414. PVOID GetExtensionFuncPtr (SOCKET sock, GUID guid);
  415. /* 获取 AcceptEx 扩展函数指针 */
  416. LPFN_ACCEPTEX Get_AcceptEx_FuncPtr (SOCKET sock);
  417. /* 获取 GetAcceptExSockaddrs 扩展函数指针 */
  418. LPFN_GETACCEPTEXSOCKADDRS Get_GetAcceptExSockaddrs_FuncPtr(SOCKET sock);
  419. /* 获取 ConnectEx 扩展函数指针 */
  420. LPFN_CONNECTEX Get_ConnectEx_FuncPtr (SOCKET sock);
  421. /* 获取 TransmitFile 扩展函数指针 */
  422. LPFN_TRANSMITFILE Get_TransmitFile_FuncPtr (SOCKET sock);
  423. /* 获取 DisconnectEx 扩展函数指针 */
  424. LPFN_DISCONNECTEX Get_DisconnectEx_FuncPtr (SOCKET sock);
  425. HRESULT ReadSmallFile(LPCTSTR lpszFileName, CAtlFile& file, CAtlFileMapping<>& fmap, DWORD dwMaxFileSize = MAX_SMALL_FILE_SIZE);
  426. HRESULT MakeSmallFilePackage(LPCTSTR lpszFileName, CAtlFile& file, CAtlFileMapping<>& fmap, WSABUF szBuf[3], const LPWSABUF pHead = nullptr, const LPWSABUF pTail = nullptr);
  427. /************************************************************************
  428. 名称:IOCP 指令投递帮助方法
  429. 描述:简化 IOCP 指令投递
  430. ************************************************************************/
  431. /* IOCP 命令 */
  432. enum EnIocpCommand
  433. {
  434. IOCP_CMD_EXIT = 0x00000000, // 退出程序
  435. IOCP_CMD_ACCEPT = 0xFFFFFFF1, // 接受连接
  436. IOCP_CMD_DISCONNECT = 0xFFFFFFF2, // 断开连接
  437. IOCP_CMD_SEND = 0xFFFFFFF3 // 发送数据
  438. };
  439. /* IOCP 命令处理动作 */
  440. enum EnIocpAction
  441. {
  442. IOCP_ACT_GOON = 0, // 继续执行
  443. IOCP_ACT_CONTINUE = 1, // 重新执行
  444. IOCP_ACT_BREAK = 2 // 中断执行
  445. };
  446. BOOL PostIocpCommand(HANDLE hIOCP, EnIocpCommand enCmd, ULONG_PTR ulParam);
  447. BOOL PostIocpExit(HANDLE hIOCP);
  448. BOOL PostIocpAccept(HANDLE hIOCP);
  449. BOOL PostIocpDisconnect(HANDLE hIOCP, CONNID dwConnID);
  450. BOOL PostIocpSend(HANDLE hIOCP, CONNID dwConnID);
  451. BOOL PostIocpClose(HANDLE hIOCP, CONNID dwConnID, int iErrorCode);
  452. /************************************************************************
  453. 名称:setsockopt() 帮助方法
  454. 描述:简化常用的 setsockopt() 调用
  455. ************************************************************************/
  456. int SSO_SetSocketOption (SOCKET sock, int level, int name, LPVOID val, int len);
  457. int SSO_GetSocketOption (SOCKET sock, int level, int name, LPVOID val, int* len);
  458. int SSO_IoctlSocket (SOCKET sock, long cmd, u_long* arg);
  459. int SSO_WSAIoctl (SOCKET sock, DWORD dwIoControlCode, LPVOID lpvInBuffer, DWORD cbInBuffer, LPVOID lpvOutBuffer, DWORD cbOutBuffer, LPDWORD lpcbBytesReturned);
  460. int SSO_UpdateAcceptContext (SOCKET soClient, SOCKET soBind);
  461. int SSO_UpdateConnectContext(SOCKET soClient, int iOption);
  462. int SSO_NoDelay (SOCKET sock, BOOL bNoDelay = TRUE);
  463. int SSO_DontLinger (SOCKET sock, BOOL bDont = TRUE);
  464. int SSO_Linger (SOCKET sock, USHORT l_onoff, USHORT l_linger);
  465. int SSO_KeepAlive (SOCKET sock, BOOL bKeepAlive = TRUE);
  466. int SSO_KeepAliveVals (SOCKET sock, u_long onoff, u_long time, u_long interval);
  467. int SSO_RecvBuffSize (SOCKET sock, int size);
  468. int SSO_SendBuffSize (SOCKET sock, int size);
  469. int SSO_ReuseAddress (SOCKET sock, BOOL bReuse = TRUE);
  470. int SSO_UDP_ConnReset (SOCKET sock, BOOL bNewBehavior = TRUE);
  471. /************************************************************************
  472. 名称:Socket 操作方法
  473. 描述:Socket 操作包装方法
  474. ************************************************************************/
  475. /* 检测 IOCP 操作返回值:NO_ERROR 则返回 TRUE */
  476. #define IOCP_NO_ERROR(result) (result == NO_ERROR)
  477. /* 检测 IOCP 操作返回值:WSA_IO_PENDING 则返回 TRUE */
  478. #define IOCP_PENDING(result) (result == WSA_IO_PENDING)
  479. /* 检测 IOCP 操作返回值:NO_ERROR 或 WSA_IO_PENDING 则返回 TRUE */
  480. #define IOCP_SUCCESS(result) (IOCP_NO_ERROR(result) || IOCP_PENDING(result))
  481. /* 生成 Connection ID */
  482. CONNID GenerateConnectionID ();
  483. /* 关闭 Socket */
  484. int ManualCloseSocket (SOCKET sock, int iShutdownFlag = 0xFF, BOOL bGraceful = TRUE, BOOL bReuseAddress = FALSE);
  485. /* 投递 AccceptEx(),并把 WSA_IO_PENDING 转换为 NO_ERROR */
  486. int PostAccept (LPFN_ACCEPTEX pfnAcceptEx, SOCKET soListen, SOCKET soClient, TBufferObj* pBufferObj);
  487. /* 投递 AccceptEx() */
  488. int PostAcceptNotCheck (LPFN_ACCEPTEX pfnAcceptEx, SOCKET soListen, SOCKET soClient, TBufferObj* pBufferObj);
  489. /* 投递 ConnectEx(),并把 WSA_IO_PENDING 转换为 NO_ERROR */
  490. int PostConnect (LPFN_CONNECTEX pfnConnectEx, SOCKET soClient, SOCKADDR_IN& soAddrIN, TBufferObj* pBufferObj);
  491. /* 投递 ConnectEx() */
  492. int PostConnectNotCheck (LPFN_CONNECTEX pfnConnectEx, SOCKET soClient, SOCKADDR_IN& soAddrIN, TBufferObj* pBufferObj);
  493. /* 投递 WSASend(),并把 WSA_IO_PENDING 转换为 NO_ERROR */
  494. int PostSend (TSocketObj* pSocketObj, TBufferObj* pBufferObj);
  495. /* 投递 WSASend() */
  496. int PostSendNotCheck (TSocketObj* pSocketObj, TBufferObj* pBufferObj);
  497. /* 投递 WSARecv() ,并把 WSA_IO_PENDING 转换为 NO_ERROR*/
  498. int PostReceive (TSocketObj* pSocketObj, TBufferObj* pBufferObj);
  499. /* 投递 WSARecv() */
  500. int PostReceiveNotCheck (TSocketObj* pSocketObj, TBufferObj* pBufferObj);
  501. /* 投递 WSASendTo() ,并把 WSA_IO_PENDING 转换为 NO_ERROR*/
  502. int PostSendTo (SOCKET sock, TUdpBufferObj* pBufferObj);
  503. /* 投递 WSASendTo() */
  504. int PostSendToNotCheck (SOCKET sock, TUdpBufferObj* pBufferObj);
  505. /* 投递 WSARecvFrom() ,并把 WSA_IO_PENDING 转换为 NO_ERROR*/
  506. int PostReceiveFrom (SOCKET sock, TUdpBufferObj* pBufferObj);
  507. /* 投递 WSARecvFrom() */
  508. int PostReceiveFromNotCheck (SOCKET sock, TUdpBufferObj* pBufferObj);