sockio.h
上传用户:zhongxx05
上传日期:2007-06-06
资源大小:33641k
文件大小:10k
- /* ***** BEGIN LICENSE BLOCK *****
- * Version: RCSL 1.0/RPSL 1.0
- *
- * Portions Copyright (c) 1995-2002 RealNetworks, Inc. All Rights Reserved.
- *
- * The contents of this file, and the files included with this file, are
- * subject to the current version of the RealNetworks Public Source License
- * Version 1.0 (the "RPSL") available at
- * http://www.helixcommunity.org/content/rpsl unless you have licensed
- * the file under the RealNetworks Community Source License Version 1.0
- * (the "RCSL") available at http://www.helixcommunity.org/content/rcsl,
- * in which case the RCSL will apply. You may also obtain the license terms
- * directly from RealNetworks. You may not use this file except in
- * compliance with the RPSL or, if you have a valid RCSL with RealNetworks
- * applicable to this file, the RCSL. Please see the applicable RPSL or
- * RCSL for the rights, obligations and limitations governing use of the
- * contents of the file.
- *
- * This file is part of the Helix DNA Technology. RealNetworks is the
- * developer of the Original Code and owns the copyrights in the portions
- * it created.
- *
- * This file, and the files included with this file, is distributed and made
- * available on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
- * EXPRESS OR IMPLIED, AND REALNETWORKS HEREBY DISCLAIMS ALL SUCH WARRANTIES,
- * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS
- * FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
- *
- * Technology Compatibility Kit Test Suite(s) Location:
- * http://www.helixcommunity.org/content/tck
- *
- * Contributor(s):
- *
- * ***** END LICENSE BLOCK ***** */
- #ifndef _SOCKIO_H_
- #define _SOCKIO_H_
- class TCPIO;
- #ifdef _SOLARIS
- #include <sys/systeminfo.h>
- #include <sys/filio.h>
- #endif /* _SOLARIS */
- #include <errno.h>
- #include <sys/ioctl.h>
- #include <unistd.h>
- #include <sys/types.h>
- #include <sys/socket.h>
- #include <netinet/in.h>
- #ifndef _BEOS
- #include <arpa/inet.h>
- #endif
- #ifndef _VXWORKS
- #include <netdb.h>
- #else
- #include <ioLib.h>
- #ifndef fcntl
- #define fcntl ioctl
- #endif
- #include <sockLib.h>
- #endif
- #include <string.h>
- #include "hxassert.h"
- #include "hxstrutl.h"
- #include "asyncio.h"
- #if defined PAULM_SOCKTIMING
- #include "sockettimer.h"
- extern SocketTimer g_SocketTimer;
- #endif
- #if defined (_AIX42)
- #define HX_SOCKLEN_T size_t
- #elif defined(_AIX43) || defined(_FREEBSD5) || defined(_FREEBSD4) || defined(_OPENBSD) ||defined(_NETBSD) || defined(_LINUX) || (defined (__GNUC__) && defined(_SOLARIS))
- #define HX_SOCKLEN_T socklen_t
- #else
- #define HX_SOCKLEN_T int
- #endif
- #ifndef HX_SOCKADDR_T
- #if defined _HPUX
- #define HX_SOCKADDR_T void
- #else
- #define HX_SOCKADDR_T sockaddr
- #endif
- #endif
- #ifndef INADDR_NONE
- #define INADDR_NONE (-1)
- #endif
- #include <sys/uio.h>
- #include <unistd.h>
- #define HX_IOVEC struct iovec
- #ifdef IOV_MAX
- #define MAX_IOVECTORS IOV_MAX
- #else
- #define MAX_IOVECTORS 1000
- #endif
- class SocketIO: public AsyncIO
- {
- public:
- friend class TCPIO;
- friend class UDPIO;
- SocketIO();
- virtual ~SocketIO();
- static INT32 create_address(struct sockaddr_in& addr, char* host,
- INT16 port);
- INT32 init(INT32 type, BOOL do_block=TRUE,
- BOOL reuse_addr=TRUE, BOOL reuse_port=FALSE);
- INT32 init(INT32 type, UINT32 local_addr, INT16 port,
- BOOL do_block=TRUE,
- BOOL reuse_addr=TRUE, BOOL reuse_port=FALSE);
- virtual INT32 listen(INT32 backlog);
- virtual INT32 connect(sockaddr_in *addr);
- virtual INT32 bind(sockaddr_in*);
- virtual INT32 blocking();
- virtual INT32 nonblocking();
- virtual INT32 getsockname(sockaddr_in* addr, INT32* addr_len);
- virtual INT32 close();
- virtual INT32 read(void* buf, INT32 size);
- virtual INT32 write(const void* buf, INT32 size);
- virtual off_t seek(off_t off, INT32 whence);
- virtual INT16 port();
- virtual INT32 disable();
- virtual INT32 reuse_port(BOOL enable);
- virtual INT32 reuse_addr(BOOL enable);
- virtual INT32 error();
- virtual INT32 flags();
- virtual int fd();
- virtual void fd(int s);
- #if defined PAULM_SOCKTIMING
- void fd(int s, int c);
- #endif
- virtual off_t file_size();
- static const UINT32 MAX_HOSTNAME_LEN;
- static int gethostname(char* name, int count);
- protected:
- SocketIO(int conn);
- INT32 err;
- int sock;
- INT32 _flags;
- };
- inline
- SocketIO::SocketIO()
- {
- _flags = O_RDWR;
- sock = -1;
- err = 0;
- }
- inline
- SocketIO::SocketIO(int conn)
- {
- _flags = O_RDWR;
- sock = conn;
- err = 0;
- }
- inline INT32
- SocketIO::error()
- {
- return err;
- }
- inline INT32
- SocketIO::flags()
- {
- return _flags;
- }
- inline int
- SocketIO::fd()
- {
- return sock;
- }
- inline void
- SocketIO::fd(int s)
- {
- #if defined PAULM_SOCKTIMING
- sockaddr_in a;
- int size = sizeof(a);
- if (!::getsockname(s, (sockaddr*)&a, &size))
- {
- g_SocketTimer.Add(s, ntohs(a.sin_port));
- }
- #endif
- sock = s;
- }
- #if defined PAULM_SOCKTIMING
- inline void
- SocketIO::fd(int s, int c)
- {
- sockaddr_in a;
- int size = sizeof(a);
- if (!::getsockname(s, (sockaddr*)&a, &size))
- {
- g_SocketTimer.Add(s, ntohs(a.sin_port), c);
- }
- sock = s;
- }
- #endif
- inline INT32
- SocketIO::disable()
- {
- sock = -2;
- return 0;
- }
- inline INT32
- SocketIO::close()
- {
- if (sock < 0)
- {
- return 0;
- }
- #if defined PAULM_SOCKTIMING
- g_SocketTimer.Remove(sock);
- #endif
- INT32 ret = ::close(sock);
- sock = -1;
- if (ret < 0)
- err = errno;
- return ret;
- }
- inline
- SocketIO::~SocketIO()
- {
- close();
- }
- inline INT32
- SocketIO::blocking()
- {
- #ifdef FIONBIO
- INT32 dont_block = 0;
- INT32 ret = ::ioctl(sock, FIONBIO, (char*)&dont_block);
- if (ret < 0)
- err = errno;
- #elif SO_NONBLOCK
- char dont_block=0;
- INT32 ret = ::setsockopt(sock,SOL_SOCKET,SO_NONBLOCK,&dont_block,1);
- #else
- INT32 ret = ::fcntl(sock, F_SETFL, ::fcntl(sock, F_GETFL, 0) & ~O_NONBLOCK);
- #endif
- return ret;
- }
- inline INT32
- SocketIO::nonblocking()
- {
- #ifdef FIONBIO
- INT32 dont_block = 1;
- INT32 ret = ::ioctl(sock, FIONBIO, (char*)&dont_block);
- if (ret < 0)
- err = errno;
- #elif SO_NONBLOCK
- char dont_block=1;
- INT32 ret = ::setsockopt(sock,SOL_SOCKET,SO_NONBLOCK,&dont_block,1);
- #else
- INT32 ret = ::fcntl(sock, F_SETFL, ::fcntl(sock, F_GETFL, 0) | O_NONBLOCK);
- #endif
- return ret;
- }
- inline INT32
- SocketIO::getsockname(sockaddr_in* addr, INT32* addr_len)
- {
- if (sock < 0)
- {
- err = EBADF;
- return -1;
- }
- HX_SOCKLEN_T temp_addr = *addr_len;
- int ret = ::getsockname(sock, (HX_SOCKADDR_T*)addr, (HX_SOCKLEN_T *)&temp_addr);
- *addr_len = temp_addr;
- #ifdef _LINUX
- if (!ret && ((sockaddr_in*)addr)->sin_family == AF_UNIX)
- {
- ((sockaddr_in*)addr)->sin_addr.s_addr = inet_addr("127.0.0.1");
- }
- #elif _SUN
- if (!ret && !((sockaddr_in*)addr)->sin_addr.s_addr)
- {
- ((sockaddr_in*)addr)->sin_addr.s_addr = inet_addr("127.0.0.1");
- }
- #endif /* _LINUX */
- if (ret < 0)
- err = errno;
- return ret;
- }
- inline INT32
- SocketIO::listen(INT32 backlog)
- {
- if (sock < 0)
- {
- err = EBADF;
- return -1;
- }
- INT32 ret = ::listen(sock, backlog);
- if (ret < 0)
- err = errno;
- return ret;
- }
- inline INT32
- SocketIO::bind(sockaddr_in* addr)
- {
- if (sock < 0)
- {
- err = EBADF;
- return -1;
- }
- INT32 ret = ::bind(sock, (HX_SOCKADDR_T*)addr, sizeof *addr);
- if (ret < 0)
- err = errno;
- return ret;
- }
- inline INT32
- SocketIO::connect(sockaddr_in* addr)
- {
- if (sock < 0)
- {
- err = EBADF;
- return -1;
- }
- INT32 ret = ::connect(sock, (HX_SOCKADDR_T*)addr, sizeof *addr);
- if (ret < 0)
- err = errno;
- return ret;
- }
- inline INT32
- SocketIO::read(void * buf, INT32 len)
- {
- if (sock < 0)
- {
- err = EBADF;
- return -1;
- }
- INT32 ret = ::recv(sock, (char*)buf, len, 0);
- err = errno;
- while (ret < 0 && err == EINTR)
- {
- ret = ::recv(sock, (char*)buf, len, 0);
- err = errno;
- }
- return ret;
- }
- inline INT32
- SocketIO::write(const void * buf, INT32 len)
- {
- if (sock < 0)
- {
- err = EBADF;
- return -1;
- }
- INT32 ret = ::send(sock, (const char *)buf, len, 0);
- if (ret < 0)
- err = errno;
- #if defined SOLARIS2_4
- if (err == EINVAL) /* This is a hack to get solaris 2.4 to work */
- {
- err = 0;
- ret = 0;
- }
- #endif
- return ret;
- }
- inline INT16
- SocketIO::port()
- {
- sockaddr_in addr;
- INT32 addr_len = sizeof addr;
- memset(&addr, 0, addr_len);
- INT32 ret = getsockname(&addr, &addr_len);
- return (ret < 0) ? ret : ntohs(addr.sin_port);
- }
- inline INT32
- SocketIO::reuse_port(BOOL enable)
- {
- #if defined(SO_REUSEPORT) && !defined(__QNXNTO__)
- if (sock < 0)
- {
- err = EBADF;
- return -1;
- }
- INT32 ret;
- int opt_val = 0;
- if (enable)
- {
- opt_val = 1;
- }
- ret = ::setsockopt(sock, SOL_SOCKET, SO_REUSEPORT, (char*) &opt_val,
- sizeof (int));
- if (ret < 0)
- {
- err = errno;
- }
- return ret;
- #else
- return 0;
- #endif
- }
- inline INT32
- SocketIO::reuse_addr(BOOL enable)
- {
- if (sock < 0)
- {
- err = EBADF;
- return -1;
- }
- INT32 ret;
- int opt_val = 0;
- if (enable)
- {
- opt_val = 1;
- }
- ret = ::setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char*) &opt_val,
- sizeof (int));
- if (ret < 0)
- {
- err = errno;
- }
- return ret;
- }
- inline off_t
- SocketIO::seek(off_t off, INT32 whence)
- {
- ASSERT(FALSE);
- return -1;
- }
- inline int
- SocketIO::gethostname(char* name, int count)
- {
- LONG32 result;
- #ifdef _SOLARIS
- result = sysinfo(SI_HOSTNAME, name, count);
- #elif defined _VXWORKS
- SafeStrCpy(name, "VXWORKS", count);
- #else
- result = ::gethostname(name, count);
- #endif /* _SOLARIS */
- if (result < 0)
- {
- return errno;
- }
- return 0;
- }
- inline off_t
- SocketIO::file_size()
- {
- ASSERT(0);
- return -1;
- }
- #endif /* _SOCKIO_H_ */