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

Symbian

开发平台:

Visual C++

  1. /* ***** BEGIN LICENSE BLOCK *****
  2.  * Source last modified: $Id: sockio.h,v 1.7.2.4 2004/07/09 01:46:54 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 _SOLARIS
  53. #include <sys/systeminfo.h>
  54. #include <sys/filio.h>
  55. #endif /* _SOLARIS */
  56. #include <errno.h>
  57. #include <sys/ioctl.h>
  58. #include <unistd.h>
  59. #include <sys/types.h>
  60. #include <sys/socket.h>
  61. #include <netinet/in.h>
  62. #ifndef _BEOS
  63. #include <arpa/inet.h>
  64. #endif
  65. #ifndef _VXWORKS
  66. #include <netdb.h>
  67. #else
  68. #include <ioLib.h>
  69. #ifndef fcntl
  70. #define fcntl ioctl
  71. #endif
  72. #include <sockLib.h>
  73. #endif
  74. #include <string.h>
  75. #include "hxassert.h"
  76. #include "hxstrutl.h"
  77. #include "asyncio.h"
  78. #if defined PAULM_SOCKTIMING 
  79. #include "sockettimer.h"
  80. extern SocketTimer g_SocketTimer;
  81. #endif
  82. #if defined (_AIX42)
  83. #define HX_SOCKLEN_T size_t
  84. #elif defined(_AIX43) || defined(_FREEBSD5) || defined(_FREEBSD4) || defined(_OPENBSD) ||defined(_NETBSD) || defined(_LINUX) || (defined (__GNUC__) && defined(_SOLARIS))
  85. #define HX_SOCKLEN_T socklen_t
  86. #else
  87. #define HX_SOCKLEN_T int
  88. #endif
  89. #ifndef HX_SOCKADDR_T
  90. #if defined _HPUX
  91. #define HX_SOCKADDR_T void
  92. #else
  93. #define HX_SOCKADDR_T sockaddr
  94. #endif
  95. #endif
  96. #ifndef INADDR_NONE
  97. #define INADDR_NONE (-1)
  98. #endif
  99. #include <sys/uio.h>
  100. #include <unistd.h>
  101. #define HX_IOVEC struct iovec
  102. #ifdef IOV_MAX
  103. #define MAX_IOVECTORS IOV_MAX
  104. #else
  105. #define MAX_IOVECTORS 1000
  106. #endif
  107. class SocketIO: public AsyncIO
  108. {
  109. public:
  110.     friend class TCPIO;
  111.     friend class UDPIO;
  112. SocketIO();
  113.     virtual      ~SocketIO();
  114.     static INT32 create_address(struct sockaddr_in& addr, char* host,
  115.        INT16 port);
  116.     INT32               init(INT32 type, BOOL do_block=TRUE,
  117.                                BOOL reuse_addr=TRUE, BOOL reuse_port=FALSE);
  118.     INT32 init(INT32 type, UINT32 local_addr, INT16 port, 
  119.      BOOL do_block=TRUE, 
  120.      BOOL reuse_addr=TRUE, BOOL reuse_port=FALSE);
  121.     virtual INT32 listen(INT32 backlog);
  122.     virtual INT32 connect(sockaddr_in *addr);
  123.     virtual INT32 bind(sockaddr_in*);
  124.     virtual INT32 blocking();
  125.     virtual INT32 nonblocking();
  126.     virtual INT32 getsockname(sockaddr_in* addr, INT32* addr_len);
  127.     virtual INT32 close();
  128.     virtual INT32 read(void* buf, INT32 size);
  129.     virtual INT32 write(const void* buf, INT32 size);
  130.     virtual off_t seek(off_t off, INT32 whence);
  131.     virtual INT16 port();
  132.     virtual INT32 disable();
  133.     virtual INT32 reuse_port(BOOL enable);
  134.     virtual INT32 reuse_addr(BOOL enable);
  135.     virtual INT32 error();
  136.     virtual INT32 flags();
  137.     virtual int fd();
  138.     virtual void fd(int s);
  139. #if defined PAULM_SOCKTIMING 
  140.     void fd(int s, int c);
  141. #endif
  142.     virtual off_t file_size();
  143.     static const UINT32 MAX_HOSTNAME_LEN;
  144.     static int gethostname(char* name, int count);
  145. protected:
  146. SocketIO(int conn);
  147.     INT32 err;
  148.     int sock;
  149.     INT32 _flags;
  150. };
  151. inline
  152. SocketIO::SocketIO()
  153. {
  154.     _flags = O_RDWR;
  155.     sock = -1;
  156.     err = 0;
  157. }
  158. inline
  159. SocketIO::SocketIO(int conn)
  160. {
  161.     _flags = O_RDWR;
  162.     sock = conn;
  163.     err = 0;
  164. }
  165. inline INT32
  166. SocketIO::error()
  167. {
  168.     return err;
  169. }
  170. inline INT32
  171. SocketIO::flags()
  172. {
  173.     return _flags;
  174. }
  175. inline int
  176. SocketIO::fd()
  177. {
  178.     return sock;
  179. }
  180. inline void
  181. SocketIO::fd(int s)
  182. {
  183. #if defined PAULM_SOCKTIMING 
  184.     sockaddr_in a;
  185.     int size = sizeof(a);
  186.     if (!::getsockname(s, (sockaddr*)&a, &size))
  187.     {
  188. g_SocketTimer.Add(s, ntohs(a.sin_port));
  189.     }
  190. #endif
  191.     sock = s;
  192. }
  193. #if defined PAULM_SOCKTIMING 
  194. inline void
  195. SocketIO::fd(int s, int c)
  196. {
  197.     sockaddr_in a;
  198.     int size = sizeof(a);
  199.     if (!::getsockname(s, (sockaddr*)&a, &size))
  200.     {
  201. g_SocketTimer.Add(s, ntohs(a.sin_port), c);
  202.     }
  203.     sock  = s;
  204. }
  205. #endif
  206. inline INT32
  207. SocketIO::disable()
  208. {
  209.     sock = -2;
  210.     return 0;
  211. }
  212. inline INT32
  213. SocketIO::close() 
  214. {
  215.     if (sock < 0)
  216.     {
  217. return 0;
  218.     }
  219. #if defined PAULM_SOCKTIMING 
  220.     g_SocketTimer.Remove(sock);
  221. #endif
  222.     INT32 ret = ::close(sock);
  223.     sock = -1;
  224.     if (ret < 0)
  225. err = errno;
  226.     return ret;
  227. }
  228. inline
  229. SocketIO::~SocketIO() 
  230. {
  231.     close();
  232. }
  233. inline INT32
  234. SocketIO::blocking() 
  235. {
  236. #ifdef FIONBIO
  237.     INT32 dont_block = 0;
  238.     INT32 ret = ::ioctl(sock, FIONBIO, (char*)&dont_block);
  239.     if (ret < 0)
  240. err = errno;
  241. #elif SO_NONBLOCK
  242.     char dont_block=0;
  243.     INT32 ret = ::setsockopt(sock,SOL_SOCKET,SO_NONBLOCK,&dont_block,1);
  244. #else
  245.     INT32 ret = ::fcntl(sock, F_SETFL, ::fcntl(sock, F_GETFL, 0) & ~O_NONBLOCK);
  246. #endif
  247.     return ret;
  248. }
  249. inline INT32
  250. SocketIO::nonblocking() 
  251. {
  252. #ifdef FIONBIO
  253.     INT32 dont_block = 1;
  254.     INT32 ret = ::ioctl(sock, FIONBIO, (char*)&dont_block);
  255.     if (ret < 0)
  256. err = errno;
  257. #elif SO_NONBLOCK
  258.     char dont_block=1;
  259.     INT32 ret = ::setsockopt(sock,SOL_SOCKET,SO_NONBLOCK,&dont_block,1);
  260. #else
  261.     INT32 ret = ::fcntl(sock, F_SETFL, ::fcntl(sock, F_GETFL, 0) | O_NONBLOCK);
  262. #endif
  263.     return ret;
  264. }
  265. inline INT32
  266. SocketIO::getsockname(sockaddr_in* addr, INT32* addr_len)
  267. {
  268.     if (sock < 0)
  269.     {
  270. err = EBADF;
  271. return -1;
  272.     }
  273.     HX_SOCKLEN_T temp_addr = *addr_len;
  274.     int ret = ::getsockname(sock, (HX_SOCKADDR_T*)addr, (HX_SOCKLEN_T *)&temp_addr);
  275.     *addr_len = temp_addr;
  276. #ifdef _LINUX
  277.     if (!ret && ((sockaddr_in*)addr)->sin_family == AF_UNIX) 
  278.     { 
  279. ((sockaddr_in*)addr)->sin_addr.s_addr = inet_addr("127.0.0.1"); 
  280.     }
  281. #elif _SUN
  282.     if (!ret && !((sockaddr_in*)addr)->sin_addr.s_addr)
  283.     {
  284. ((sockaddr_in*)addr)->sin_addr.s_addr = inet_addr("127.0.0.1");
  285.     }
  286. #endif /* _LINUX */
  287.     if (ret < 0)
  288. err = errno;
  289.     return ret;
  290. }
  291. inline INT32
  292. SocketIO::listen(INT32 backlog) 
  293. {
  294.     if (sock < 0)
  295.     {
  296. err = EBADF;
  297. return -1;
  298.     }
  299.     INT32 ret = ::listen(sock, backlog);
  300.     if (ret < 0)
  301. err = errno;
  302.     return ret;
  303. }
  304. inline INT32
  305. SocketIO::bind(sockaddr_in* addr) 
  306. {
  307.     if (sock < 0)
  308.     {
  309. err = EBADF;
  310. return -1;
  311.     }
  312.     INT32 ret = ::bind(sock, (HX_SOCKADDR_T*)addr, sizeof *addr);
  313.     if (ret < 0)
  314. err = errno;
  315.     return ret;
  316. }
  317. inline INT32
  318. SocketIO::connect(sockaddr_in* addr) 
  319. {
  320.     if (sock < 0)
  321.     {
  322. err = EBADF;
  323. return -1;
  324.     }
  325.     INT32 ret = ::connect(sock, (HX_SOCKADDR_T*)addr, sizeof *addr);
  326.     if (ret < 0)
  327. err = errno;
  328.     return ret;
  329. }
  330. inline INT32
  331. SocketIO::read(void * buf, INT32 len) 
  332. {
  333.     if (sock < 0)
  334.     {
  335. err = EBADF;
  336. return -1;
  337.     }
  338.     INT32 ret = ::recv(sock, (char*)buf, len, 0);
  339.     err = errno;
  340.     while (ret < 0 && err == EINTR)
  341.     {
  342. ret = ::recv(sock, (char*)buf, len, 0);
  343. err = errno;
  344.     }
  345.     return ret;
  346. }
  347. inline INT32
  348. SocketIO::write(const void * buf, INT32 len) 
  349. {    
  350.     if (sock < 0)
  351.     {
  352. err = EBADF;
  353. return -1;
  354.     }
  355.     INT32 ret = ::send(sock, (const char *)buf, len, 0);
  356.     if (ret < 0)
  357. err = errno;
  358. #if defined SOLARIS2_4
  359.     if (err == EINVAL)    /* This is a hack to get solaris 2.4 to work */
  360.     {
  361. err = 0;
  362. ret = 0;
  363.     }
  364. #endif
  365.     return ret;
  366. }
  367. inline INT16
  368. SocketIO::port()
  369. {
  370.     sockaddr_in addr;
  371.     INT32 addr_len = sizeof addr;
  372.     memset(&addr, 0, addr_len);
  373.     INT32 ret = getsockname(&addr, &addr_len);
  374.     return (ret < 0) ? ret : ntohs(addr.sin_port);
  375. }
  376. inline INT32
  377. SocketIO::reuse_port(BOOL enable)
  378. {
  379. #if defined(SO_REUSEPORT) && !defined(__QNXNTO__)
  380.     if (sock < 0)
  381.     {
  382. err = EBADF;
  383. return -1;
  384.     }
  385.     INT32  ret;
  386.     int  opt_val = 0;
  387.     if (enable)
  388.     {
  389. opt_val = 1;
  390.     }
  391.     ret = ::setsockopt(sock, SOL_SOCKET, SO_REUSEPORT, (char*) &opt_val,
  392.        sizeof (int));
  393.     if (ret < 0)
  394.     {
  395. err = errno;
  396.     }
  397.     return ret;
  398. #else
  399.     return 0;
  400. #endif
  401. }
  402. inline INT32
  403. SocketIO::reuse_addr(BOOL enable)
  404.     if (sock < 0)
  405.     {
  406. err = EBADF;
  407. return -1;
  408.     }
  409.     INT32 ret;
  410.     int   opt_val = 0;
  411.     if (enable)
  412.     {
  413. opt_val = 1;
  414.     }
  415.     ret = ::setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char*) &opt_val,
  416.        sizeof (int));
  417.     if (ret < 0)
  418.     {
  419. err = errno;
  420.     }
  421.     return ret;
  422. }
  423. inline off_t
  424. SocketIO::seek(off_t off, INT32 whence)
  425. {
  426.     ASSERT(FALSE);
  427.     return -1;
  428. }
  429. inline int
  430. SocketIO::gethostname(char* name, int count)
  431. {
  432.     LONG32 result;
  433. #ifdef _SOLARIS
  434.     result = sysinfo(SI_HOSTNAME, name, count);
  435. #elif defined _VXWORKS
  436.     SafeStrCpy(name,  "VXWORKS", count);
  437. #else
  438.     result = ::gethostname(name, count);
  439. #endif /* _SOLARIS */
  440.     if (result < 0)
  441.     {
  442. return errno;
  443.     }
  444.     return 0;
  445. }
  446. inline off_t
  447. SocketIO::file_size()
  448. {
  449.     ASSERT(0);
  450.     return -1;
  451. }
  452. #endif /* _SOCKIO_H_ */