sockio.h
上传用户:dangjiwu
上传日期:2013-07-19
资源大小:42019k
文件大小:9k
源码类别:

Symbian

开发平台:

Visual C++

  1. /* ***** BEGIN LICENSE BLOCK *****
  2.  * Source last modified: $Id: sockio.h,v 1.1.1.1.50.3 2004/07/09 01:46:56 hubbe Exp $
  3.  * 
  4.  * Portions Copyright (c) 1995-2004 RealNetworks, Inc. All Rights Reserved.
  5.  * 
  6.  * The contents of this file, and the files included with this file,
  7.  * are subject to the current version of the RealNetworks Public
  8.  * Source License (the "RPSL") available at
  9.  * http://www.helixcommunity.org/content/rpsl unless you have licensed
  10.  * the file under the current version of the RealNetworks Community
  11.  * Source License (the "RCSL") available at
  12.  * http://www.helixcommunity.org/content/rcsl, in which case the RCSL
  13.  * will apply. You may also obtain the license terms directly from
  14.  * RealNetworks.  You may not use this file except in compliance with
  15.  * the RPSL or, if you have a valid RCSL with RealNetworks applicable
  16.  * to this file, the RCSL.  Please see the applicable RPSL or RCSL for
  17.  * the rights, obligations and limitations governing use of the
  18.  * contents of the file.
  19.  * 
  20.  * Alternatively, the contents of this file may be used under the
  21.  * terms of the GNU General Public License Version 2 or later (the
  22.  * "GPL") in which case the provisions of the GPL are applicable
  23.  * instead of those above. If you wish to allow use of your version of
  24.  * this file only under the terms of the GPL, and not to allow others
  25.  * to use your version of this file under the terms of either the RPSL
  26.  * or RCSL, indicate your decision by deleting the provisions above
  27.  * and replace them with the notice and other provisions required by
  28.  * the GPL. If you do not delete the provisions above, a recipient may
  29.  * use your version of this file under the terms of any one of the
  30.  * RPSL, the RCSL or the GPL.
  31.  * 
  32.  * This file is part of the Helix DNA Technology. RealNetworks is the
  33.  * developer of the Original Code and owns the copyrights in the
  34.  * portions it created.
  35.  * 
  36.  * This file, and the files included with this file, is distributed
  37.  * and made available on an 'AS IS' basis, WITHOUT WARRANTY OF ANY
  38.  * KIND, EITHER EXPRESS OR IMPLIED, AND REALNETWORKS HEREBY DISCLAIMS
  39.  * ALL SUCH WARRANTIES, INCLUDING WITHOUT LIMITATION, ANY WARRANTIES
  40.  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, QUIET
  41.  * ENJOYMENT OR NON-INFRINGEMENT.
  42.  * 
  43.  * Technology Compatibility Kit Test Suite(s) Location:
  44.  *    http://www.helixcommunity.org/content/tck
  45.  * 
  46.  * Contributor(s):
  47.  * 
  48.  * ***** END LICENSE BLOCK ***** */
  49. #ifndef _SOCKIO_H_
  50. #define _SOCKIO_H_
  51. class TCPIO;
  52. #ifdef _WIN16
  53. #include <memory.h>
  54. #include <winsock.h>
  55. #elif defined _WIN32
  56. #if defined USE_WINSOCK1
  57. #include <winsock.h>
  58. #else
  59. #include <winsock2.h>
  60. #endif //USE_WINSOCK1
  61. #endif
  62. #include "hlxclib/fcntl.h"
  63. #ifndef WIN32_PLATFORM_PSPC
  64. #include "asyncio.h"
  65. #endif
  66. #include "hlxclib/sys/types.h"
  67. #include "hxassert.h"
  68. #define EWOULDBLOCK WSAEWOULDBLOCK
  69. #define EINPROGRESS WSAEINPROGRESS
  70. struct iovec {
  71.     char* iov_base;
  72.     size_t iov_len;
  73. };
  74. typedef struct iovec HX_IOVEC;
  75. #define MAX_IOVECTORS 1000
  76. int inet_aton(register const char *cp, struct in_addr* addr);
  77. class SocketIO 
  78. #ifndef WIN32_PLATFORM_PSPC
  79. : public AsyncIO
  80. #endif
  81. {
  82. public:
  83.     friend class TCPIO;
  84.     friend class UDPIO;
  85. SocketIO::SocketIO();
  86.     virtual      SocketIO::~SocketIO();
  87.     static INT32 create_address(struct sockaddr_in& addr, char* host,
  88.        INT16 port);
  89.     INT32               init(INT32 type, BOOL do_block=TRUE,
  90.                                 BOOL reuse_addr=TRUE, BOOL reuse_port=FALSE);
  91.     INT32 init(INT32 type, UINT32 local_addr, INT16 port,
  92.              BOOL do_block=TRUE, 
  93.          BOOL reuse_addr=TRUE, BOOL reuse_port=FALSE);
  94.     INT32 listen(INT32 backlog);
  95.     INT32 connect(char* host, INT16 port);
  96.     INT32 connect(sockaddr_in *addr);
  97.     TCPIO* accept(sockaddr_in *addr, INT32* addrlen);
  98.     INT32 bind(sockaddr_in*);
  99.     INT32 blocking();
  100.     INT32 nonblocking();
  101.     INT32 getsockname(sockaddr_in* addr, INT32* addr_len);
  102.     INT32 close();
  103.     INT32 read(void* buf, INT32 size);
  104.     INT32 write(const void* buf, INT32 size);
  105.     off_t seek(off_t off, INT32 whence);
  106.     INT16 port();
  107.     INT32 disable();
  108.     INT32 reuse_port(BOOL enable);
  109.     INT32 reuse_addr(BOOL enable);
  110.     INT32 error();
  111.     INT32 flags();
  112.     virtual SOCKET socket();
  113.     off_t file_size();
  114.     static const UINT32 MAX_HOSTNAME_LEN;
  115.     static int gethostname(char* name, int count);
  116. protected:
  117. SocketIO::SocketIO(SOCKET conn);
  118.     INT32 err;
  119.     SOCKET sock;
  120.     INT32 _flags;
  121. };
  122. inline
  123. SocketIO::SocketIO()
  124. {
  125.     err = 0;
  126.     _flags = O_RDWR;
  127.     sock = INVALID_SOCKET;
  128. }
  129. inline
  130. SocketIO::SocketIO(SOCKET conn)
  131. {
  132.     err = 0;
  133.     _flags = O_RDWR;
  134.     sock = conn;
  135. }
  136. inline INT32
  137. SocketIO::error()
  138. {
  139.     return err;
  140. }
  141. inline INT32
  142. SocketIO::flags()
  143. {
  144.     return _flags;
  145. }
  146. inline INT32
  147. SocketIO::disable()
  148. {
  149.     sock = INVALID_SOCKET;
  150.     return 0;
  151. }
  152. inline INT32
  153. SocketIO::close() 
  154. {
  155.     if (sock < 0)
  156.     {
  157. return 0;
  158.     }
  159.     INT32 ret = ::closesocket(sock);
  160.     sock = INVALID_SOCKET;
  161.     if (ret < 0)
  162. err = WSAGetLastError();
  163.     return ret;
  164. }
  165. inline
  166. SocketIO::~SocketIO() 
  167. {
  168.     close();
  169. }
  170. inline INT32
  171. SocketIO::blocking() 
  172. {
  173.     u_long dont_block = 0;
  174.     INT32 ret = ::ioctlsocket(sock, FIONBIO, &dont_block);
  175.     if (ret < 0)
  176. err = WSAGetLastError();
  177.     return ret;
  178. }
  179. inline INT32
  180. SocketIO::nonblocking() 
  181. {
  182.     u_long dont_block = 1;
  183.     INT32 ret = ::ioctlsocket(sock, FIONBIO, &dont_block);
  184.     if (ret < 0)
  185. err = WSAGetLastError();
  186.     return ret;
  187. }
  188. inline INT32
  189. SocketIO::getsockname(sockaddr_in* addr, INT32* addr_len)
  190. {
  191.     if (sock < 0)
  192.     {
  193. err = WSAEBADF;
  194. return -1;
  195.     }
  196.     int temp_addr = HX_SAFEINT(*addr_len);
  197.     int ret = ::getsockname(sock,(sockaddr*) addr, &temp_addr);
  198.     *addr_len = temp_addr;
  199.     if (ret < 0)
  200. err = WSAGetLastError();
  201.     return ret;
  202. }
  203. inline INT32
  204. SocketIO::listen(INT32 backlog) 
  205. {
  206.     if (sock < 0)
  207.     {
  208. err = WSAEBADF;
  209. return -1;
  210.     }
  211.     INT32 ret = ::listen(sock, HX_SAFEINT(backlog));
  212.     if (ret < 0)
  213. err = WSAGetLastError();
  214.     return ret;
  215. }
  216. inline INT32
  217. SocketIO::bind(sockaddr_in* addr) 
  218. {
  219.     if (sock < 0)
  220.     {
  221. err = WSAEBADF;
  222. return -1;
  223.     }
  224.     INT32 ret = ::bind(sock, (sockaddr*)addr, sizeof *addr);
  225.     if (ret < 0)
  226. err = WSAGetLastError();
  227.     return ret;
  228. }
  229. inline INT32
  230. SocketIO::connect(sockaddr_in* addr) 
  231. {
  232.     if (sock < 0)
  233.     {
  234. err = WSAEBADF;
  235. return -1;
  236.     }
  237.     INT32 ret = ::connect(sock, (sockaddr*)addr, sizeof *addr);
  238.     if (ret < 0)
  239. err = WSAGetLastError();
  240.     return ret;
  241. }
  242. inline INT32
  243. SocketIO::read(void * buf, INT32 len) 
  244. {
  245.     if (sock < 0)
  246.     {
  247. err = WSAEBADF;
  248. return -1;
  249.     }
  250.     INT32 ret = ::recv(sock, (char *)buf, HX_SAFEINT(len), 0);
  251.     if (ret < 0)
  252. err = WSAGetLastError();
  253.     return ret;
  254. }
  255. inline INT32
  256. SocketIO::write(const void * buf, INT32 len) 
  257. {    
  258.     if (sock < 0)
  259.     {
  260. err = WSAEBADF;
  261. return -1;
  262.     }
  263.     INT32 ret = ::send(sock, (const char *)buf, HX_SAFEINT(len), 0);
  264.     if (ret < 0)
  265. err = WSAGetLastError();
  266. #if defined SOLARIS2_4
  267.     if (err == EINVAL)    /* This is a hack to get solaris 2.4 to work */
  268.     {
  269. err = 0;
  270. ret = 0;
  271.     }
  272. #endif
  273.     return ret;
  274. }
  275. inline INT16
  276. SocketIO::port()
  277. {
  278.     sockaddr_in addr;
  279.     INT32 addr_len = sizeof addr;
  280.     memset(&addr, 0, HX_SAFESIZE_T(addr_len));
  281.     INT32 ret = getsockname(&addr, &addr_len);
  282.     return (ret < 0) ? (INT16) ret : ntohs(addr.sin_port);
  283. }
  284. inline INT32
  285. SocketIO::reuse_port(BOOL enable)
  286. {
  287. #if defined SO_REUSEPORT
  288.     if (sock < 0)
  289.     {
  290. err = WSAEBADF;
  291. return -1;
  292.     }
  293.     INT32  ret;
  294.     int  opt_val = 0;
  295.     if (enable)
  296.     {
  297. opt_val = 1;
  298.     }
  299.     ret = ::setsockopt(sock, SOL_SOCKET, SO_REUSEPORT, (char*) &opt_val,
  300.        sizeof (int));
  301.     if (ret < 0)
  302.     {
  303. err = WSAGetLastError();
  304.     }
  305.     return ret;
  306. #else
  307.     return 0;
  308. #endif
  309. }
  310. inline INT32
  311. SocketIO::reuse_addr(BOOL enable)
  312.     if (sock < 0)
  313.     {
  314. err = WSAEBADF;
  315. return -1;
  316.     }
  317.     INT32 ret;
  318.     int   opt_val = 0;
  319.     if (enable)
  320.     {
  321. opt_val = 1;
  322.     }
  323.     ret = ::setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char*) &opt_val,
  324.        sizeof (int));
  325.     if (ret < 0)
  326.     {
  327. err = WSAGetLastError();
  328.     }
  329.     return ret;
  330. }
  331. inline off_t
  332. SocketIO::seek(off_t off, INT32 whence)
  333. {
  334.     ASSERT(FALSE);
  335.     return -1;
  336. }
  337. inline int
  338. SocketIO::gethostname(char* name, int count)
  339. {
  340.     int result;
  341.     result = ::gethostname(name, count);
  342.     if (result < 0)
  343.     {
  344. return WSAGetLastError();
  345.     }
  346.     return 0;
  347. }
  348. inline off_t
  349. SocketIO::file_size()
  350. {
  351.     ASSERT(0);
  352.     return -1;
  353. }
  354. inline SOCKET
  355. SocketIO::socket()
  356. {
  357.     return sock;
  358. }
  359. #endif /* _SOCKIO_H_ */