1208 lines
24 KiB
C
1208 lines
24 KiB
C
/*
|
|
* $Id$
|
|
*
|
|
Copyright 2006-2012 Chung, Hyung-Hwan.
|
|
This file is part of QSE.
|
|
|
|
QSE is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Lesser General Public License as
|
|
published by the Free Software Foundation, either version 3 of
|
|
the License, or (at your option) any later version.
|
|
|
|
QSE is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with QSE. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <qse/cmn/nwio.h>
|
|
#include <qse/cmn/time.h>
|
|
#include "mem.h"
|
|
|
|
#if defined(_WIN32)
|
|
# include <winsock2.h>
|
|
# include <ws2tcpip.h> /* sockaddr_in6 */
|
|
# include <windows.h>
|
|
#elif defined(__OS2__)
|
|
# include <types.h>
|
|
# include <sys/socket.h>
|
|
# include <netinet/in.h>
|
|
# include <tcpustd.h>
|
|
# include <sys/ioctl.h>
|
|
# include <nerrno.h>
|
|
# pragma library("tcpip32.lib")
|
|
#elif defined(__DOS__)
|
|
/* TODO: consider watt-32 */
|
|
#else
|
|
# include "syscall.h"
|
|
# include <sys/socket.h>
|
|
# include <netinet/in.h>
|
|
# include <sys/time.h>
|
|
#endif
|
|
|
|
QSE_IMPLEMENT_COMMON_FUNCTIONS (nwio)
|
|
|
|
enum
|
|
{
|
|
STATUS_UDP_CONNECT = (1 << 0)
|
|
};
|
|
|
|
static qse_ssize_t socket_output (
|
|
qse_tio_t* tio, qse_tio_cmd_t cmd, void* buf, qse_size_t size);
|
|
static qse_ssize_t socket_input (
|
|
qse_tio_t* tio, qse_tio_cmd_t cmd, void* buf, qse_size_t size);
|
|
|
|
#if defined(AF_INET)
|
|
union sockaddr_t
|
|
{
|
|
struct sockaddr_in in4;
|
|
#if defined(AF_INET6) && !defined(__OS2__)
|
|
struct sockaddr_in6 in6;
|
|
#endif
|
|
};
|
|
#endif
|
|
|
|
static int nwad_to_sockaddr (const qse_nwad_t* nwad, int* family, void* addr)
|
|
{
|
|
int addrsize = -1;
|
|
|
|
switch (nwad->type)
|
|
{
|
|
case QSE_NWAD_IN4:
|
|
{
|
|
#if defined(AF_INET)
|
|
struct sockaddr_in* in;
|
|
|
|
in = (struct sockaddr_in*)addr;
|
|
addrsize = QSE_SIZEOF(*in);
|
|
QSE_MEMSET (in, 0, addrsize);
|
|
|
|
*family = AF_INET;
|
|
in->sin_family = AF_INET;
|
|
in->sin_addr.s_addr = nwad->u.in4.addr.value;
|
|
in->sin_port = nwad->u.in4.port;
|
|
#endif
|
|
break;
|
|
}
|
|
|
|
case QSE_NWAD_IN6:
|
|
{
|
|
#if defined(AF_INET6) && !defined(__OS2__)
|
|
struct sockaddr_in6* in;
|
|
|
|
in = (struct sockaddr_in6*)addr;
|
|
addrsize = QSE_SIZEOF(*in);
|
|
QSE_MEMSET (in, 0, addrsize);
|
|
|
|
*family = AF_INET6;
|
|
in->sin6_family = AF_INET6;
|
|
memcpy (&in->sin6_addr, &nwad->u.in6.addr, QSE_SIZEOF(nwad->u.in6.addr));
|
|
in->sin6_scope_id = nwad->u.in6.scope;
|
|
in->sin6_port = nwad->u.in6.port;
|
|
#endif
|
|
break;
|
|
}
|
|
}
|
|
|
|
return addrsize;
|
|
}
|
|
|
|
#if defined(_WIN32)
|
|
static qse_nwio_errnum_t syserr_to_errnum (DWORD e)
|
|
{
|
|
switch (e)
|
|
{
|
|
case WSA_NOT_ENOUGH_MEMORY:
|
|
return QSE_NWIO_ENOMEM;
|
|
|
|
case WSA_INVALID_PARAMETER:
|
|
case WSA_INVALID_HANDLE:
|
|
return QSE_NWIO_EINVAL;
|
|
|
|
case WSAEACCES:
|
|
return QSE_NWIO_EACCES;
|
|
|
|
case WSAEINTR:
|
|
return QSE_NWIO_EINTR;
|
|
|
|
case WSAECONNREFUSED:
|
|
case WSAENETUNREACH:
|
|
case WSAEHOSTUNREACH:
|
|
case WSAEHOSTDOWN:
|
|
return QSE_NWIO_ECONN;
|
|
|
|
default:
|
|
return QSE_NWIO_ESYSERR;
|
|
}
|
|
}
|
|
#elif defined(__OS2__)
|
|
static qse_nwio_errnum_t syserr_to_errnum (int e)
|
|
{
|
|
switch (e)
|
|
{
|
|
case SOCENOMEM:
|
|
return QSE_NWIO_ENOMEM;
|
|
|
|
case SOCEINVAL:
|
|
return QSE_NWIO_EINVAL;
|
|
|
|
case SOCEACCES:
|
|
return QSE_NWIO_EACCES;
|
|
|
|
case SOCENOENT:
|
|
return QSE_NWIO_ENOENT;
|
|
|
|
case SOCEEXIST:
|
|
return QSE_NWIO_EEXIST;
|
|
|
|
case SOCEINTR:
|
|
return QSE_NWIO_EINTR;
|
|
|
|
case SOCEPIPE:
|
|
return QSE_NWIO_EPIPE;
|
|
|
|
case SOCECONNREFUSED:
|
|
case SOCENETUNREACH:
|
|
case SOCEHOSTUNREACH:
|
|
case SOCEHOSTDOWN:
|
|
return QSE_NWIO_ECONN;
|
|
|
|
default:
|
|
return QSE_NWIO_ESYSERR;
|
|
}
|
|
}
|
|
|
|
#elif defined(__DOS__)
|
|
static qse_nwio_errnum_t syserr_to_errnum (int e)
|
|
{
|
|
/* TODO: */
|
|
return QSE_NWIO_ESYSERR;
|
|
}
|
|
#else
|
|
static qse_nwio_errnum_t syserr_to_errnum (int e)
|
|
{
|
|
switch (e)
|
|
{
|
|
case ENOMEM:
|
|
return QSE_NWIO_ENOMEM;
|
|
|
|
case EINVAL:
|
|
return QSE_NWIO_EINVAL;
|
|
|
|
case EACCES:
|
|
return QSE_NWIO_EACCES;
|
|
|
|
case ENOENT:
|
|
return QSE_NWIO_ENOENT;
|
|
|
|
case EEXIST:
|
|
return QSE_NWIO_EEXIST;
|
|
|
|
case EINTR:
|
|
return QSE_NWIO_EINTR;
|
|
|
|
case EPIPE:
|
|
return QSE_NWIO_EPIPE;
|
|
|
|
#if defined(ECONNREFUSED) || defined(ENETUNREACH) || defined(EHOSTUNREACH) || defined(EHOSTDOWN)
|
|
#if defined(ECONNREFUSED)
|
|
case ECONNREFUSED:
|
|
#endif
|
|
#if defined(ENETUNREACH)
|
|
case ENETUNREACH:
|
|
#endif
|
|
#if defined(EHOSTUNREACH)
|
|
case EHOSTUNREACH:
|
|
#endif
|
|
#if defined(EHOSTDOWN)
|
|
case EHOSTDOWN:
|
|
#endif
|
|
return QSE_NWIO_ECONN;
|
|
#endif
|
|
|
|
default:
|
|
return QSE_NWIO_ESYSERR;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
static qse_nwio_errnum_t tio_errnum_to_nwio_errnum (qse_tio_t* tio)
|
|
{
|
|
switch (tio->errnum)
|
|
{
|
|
case QSE_TIO_ENOMEM:
|
|
return QSE_NWIO_ENOMEM;
|
|
case QSE_TIO_EINVAL:
|
|
return QSE_NWIO_EINVAL;
|
|
case QSE_TIO_EACCES:
|
|
return QSE_NWIO_EACCES;
|
|
case QSE_TIO_ENOENT:
|
|
return QSE_NWIO_ENOENT;
|
|
case QSE_TIO_EILSEQ:
|
|
return QSE_NWIO_EILSEQ;
|
|
case QSE_TIO_EICSEQ:
|
|
return QSE_NWIO_EICSEQ;
|
|
case QSE_TIO_EILCHR:
|
|
return QSE_NWIO_EILCHR;
|
|
default:
|
|
return QSE_NWIO_EOTHER;
|
|
}
|
|
}
|
|
|
|
static int wait_for_data (qse_nwio_t* nwio, int tmout, int what)
|
|
{
|
|
int xret;
|
|
|
|
#if defined(_WIN32)
|
|
fd_set fds[2];
|
|
struct timeval tv;
|
|
|
|
FD_ZERO (&fds[0]);
|
|
FD_ZERO (&fds[1]);
|
|
|
|
FD_SET (nwio->handle, &fds[what]);
|
|
|
|
tv.tv_sec = tmout / QSE_MSECS_PER_SEC;
|
|
tv.tv_usec = (tmout % QSE_MSECS_PER_SEC) * QSE_USECS_PER_MSEC;
|
|
|
|
xret = select (nwio->handle + 1, &fds[0], &fds[1], QSE_NULL, &tv);
|
|
if (xret == SOCKET_ERROR)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
return -1;
|
|
}
|
|
else if (xret == 0)
|
|
{
|
|
nwio->errnum = QSE_NWIO_ETMOUT;
|
|
return -1;
|
|
}
|
|
#elif defined(__OS2__)
|
|
int count[2] = { 0, 0 };
|
|
|
|
count[what]++;
|
|
|
|
xret = os2_select (&nwio->handle, count[0], count[1], 0, tmout);
|
|
if (xret <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (sock_errno());
|
|
return -1;
|
|
}
|
|
else if (xret == 0)
|
|
{
|
|
nwio->errnum = QSE_NWIO_ETMOUT;
|
|
return -1;
|
|
}
|
|
|
|
#elif defined(__DOS__)
|
|
nwio->errnum = QSE_NWIO_ENOIMPL;
|
|
return -1;
|
|
|
|
#else
|
|
fd_set fds[2];
|
|
struct timeval tv;
|
|
|
|
FD_ZERO (&fds[0]);
|
|
FD_ZERO (&fds[1]);
|
|
|
|
FD_SET (nwio->handle, &fds[what]);
|
|
|
|
tv.tv_sec = tmout / QSE_MSECS_PER_SEC;
|
|
tv.tv_usec = (tmout % QSE_MSECS_PER_SEC) * QSE_USECS_PER_MSEC;
|
|
|
|
xret = select (nwio->handle + 1, &fds[0], &fds[1], QSE_NULL, &tv);
|
|
if (xret <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
return -1;
|
|
}
|
|
else if (xret == 0)
|
|
{
|
|
nwio->errnum = QSE_NWIO_ETMOUT;
|
|
return -1;
|
|
}
|
|
#endif
|
|
|
|
return 0;
|
|
}
|
|
|
|
qse_nwio_t* qse_nwio_open (
|
|
qse_mmgr_t* mmgr, qse_size_t xtnsize, const qse_nwad_t* nwad,
|
|
int flags, const qse_nwio_tmout_t* tmout)
|
|
{
|
|
qse_nwio_t* nwio;
|
|
|
|
nwio = QSE_MMGR_ALLOC (mmgr, QSE_SIZEOF(qse_nwio_t) + xtnsize);
|
|
if (nwio == QSE_NULL) return QSE_NULL;
|
|
|
|
if (qse_nwio_init (nwio, mmgr, nwad, flags, tmout) <= -1)
|
|
{
|
|
QSE_MMGR_FREE (mmgr, nwio);
|
|
return QSE_NULL;
|
|
}
|
|
|
|
return nwio;
|
|
}
|
|
|
|
void qse_nwio_close (qse_nwio_t* nwio)
|
|
{
|
|
qse_nwio_fini (nwio);
|
|
QSE_MMGR_FREE (nwio->mmgr, nwio);
|
|
}
|
|
|
|
int qse_nwio_init (
|
|
qse_nwio_t* nwio, qse_mmgr_t* mmgr, const qse_nwad_t* nwad,
|
|
int flags, const qse_nwio_tmout_t* tmout)
|
|
{
|
|
#if defined(AF_INET)
|
|
union sockaddr_t addr;
|
|
#endif
|
|
#if defined(HAVE_SOCKLEN_T)
|
|
socklen_t addrlen;
|
|
#else
|
|
int addrlen;
|
|
#endif
|
|
int family, type;
|
|
int tmp;
|
|
|
|
QSE_MEMSET (nwio, 0, QSE_SIZEOF(*nwio));
|
|
nwio->mmgr = mmgr;
|
|
nwio->flags = flags;
|
|
nwio->errnum = QSE_NWIO_ENOERR;
|
|
if (tmout) nwio->tmout = *tmout;
|
|
else
|
|
{
|
|
nwio->tmout.r = -1;
|
|
nwio->tmout.w = -1;
|
|
nwio->tmout.c = -1;
|
|
nwio->tmout.a = -1;
|
|
}
|
|
|
|
#if defined(AF_INET)
|
|
tmp = nwad_to_sockaddr (nwad, &family, &addr);
|
|
if (tmp <= -1)
|
|
{
|
|
nwio->errnum = QSE_NWIO_EINVAL;
|
|
return -1;
|
|
}
|
|
addrlen = tmp;
|
|
|
|
if (flags & QSE_NWIO_TCP) type = SOCK_STREAM;
|
|
else if (flags & QSE_NWIO_UDP) type = SOCK_DGRAM;
|
|
else
|
|
#endif
|
|
{
|
|
nwio->errnum = QSE_NWIO_EINVAL;
|
|
return -1;
|
|
}
|
|
|
|
#if defined(_WIN32)
|
|
nwio->handle = socket (family, type, 0);
|
|
if (nwio->handle == INVALID_SOCKET)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
goto oops;
|
|
}
|
|
|
|
if ((flags & QSE_NWIO_TCP) && (flags & QSE_NWIO_KEEPALIVE))
|
|
{
|
|
int optval = 1;
|
|
setsockopt (nwio->handle, SOL_SOCKET, SO_KEEPALIVE, (void*)&optval, QSE_SIZEOF(optval));
|
|
}
|
|
|
|
if (flags & QSE_NWIO_PASSIVE)
|
|
{
|
|
qse_nwio_hnd_t handle;
|
|
|
|
if (flags & QSE_NWIO_REUSEADDR)
|
|
{
|
|
int optval = 1;
|
|
setsockopt (nwio->handle, SOL_SOCKET, SO_REUSEADDR, (void*)&optval, QSE_SIZEOF(optval));
|
|
}
|
|
|
|
if (bind (nwio->handle, (struct sockaddr*)&addr, addrlen) == SOCKET_ERROR)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
goto oops;
|
|
}
|
|
|
|
if (flags & QSE_NWIO_TCP)
|
|
{
|
|
if (listen (nwio->handle, 10) == SOCKET_ERROR)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
goto oops;
|
|
}
|
|
|
|
if (nwio->tmout.a >= 0 &&
|
|
wait_for_data (nwio, nwio->tmout.a, 0) <= -1) goto oops;
|
|
|
|
handle = accept (nwio->handle, (struct sockaddr*)&addr, &addrlen);
|
|
if (handle == INVALID_SOCKET)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
goto oops;
|
|
}
|
|
|
|
closesocket (nwio->handle);
|
|
nwio->handle = handle;
|
|
}
|
|
else if (flags & QSE_NWIO_UDP)
|
|
{
|
|
nwio->status |= STATUS_UDP_CONNECT;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
int xret;
|
|
|
|
if (nwio->tmout.c >= 0 && (flags & QSE_NWIO_TCP))
|
|
{
|
|
unsigned long cmd = 1;
|
|
|
|
if (ioctlsocket(nwio->handle, FIONBIO, &cmd) == SOCKET_ERROR)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
goto oops;
|
|
}
|
|
}
|
|
|
|
xret = connect (nwio->handle, (struct sockaddr*)&addr, addrlen);
|
|
|
|
if (nwio->tmout.c >= 0 && (flags & QSE_NWIO_TCP))
|
|
{
|
|
unsigned long cmd = 0;
|
|
|
|
if ((xret == SOCKET_ERROR && WSAGetLastError() != WSAEWOULDBLOCK) ||
|
|
ioctlsocket (nwio->handle, FIONBIO, &cmd) == SOCKET_ERROR)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
goto oops;
|
|
}
|
|
|
|
if (wait_for_data (nwio, nwio->tmout.c, 1) <= -1) goto oops;
|
|
else
|
|
{
|
|
int xlen;
|
|
DWORD xerr;
|
|
|
|
xlen = QSE_SIZEOF(xerr);
|
|
if (getsockopt (nwio->handle, SOL_SOCKET, SO_ERROR, (char*)&xerr, &xlen) == SOCKET_ERROR)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
goto oops;
|
|
}
|
|
else if (xerr != 0)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (xerr);
|
|
goto oops;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (xret == SOCKET_ERROR)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
goto oops;
|
|
}
|
|
}
|
|
}
|
|
|
|
#elif defined(__OS2__)
|
|
nwio->handle = socket (family, type, 0);
|
|
if (nwio->handle <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (sock_errno());
|
|
goto oops;
|
|
}
|
|
|
|
if ((flags & QSE_NWIO_TCP) && (flags & QSE_NWIO_KEEPALIVE))
|
|
{
|
|
int optval = 1;
|
|
setsockopt (nwio->handle, SOL_SOCKET, SO_KEEPALIVE, (void*)&optval, QSE_SIZEOF(optval));
|
|
}
|
|
|
|
if (flags & QSE_NWIO_PASSIVE)
|
|
{
|
|
qse_nwio_hnd_t handle;
|
|
|
|
if (flags & QSE_NWIO_REUSEADDR)
|
|
{
|
|
int optval = 1;
|
|
setsockopt (nwio->handle, SOL_SOCKET, SO_REUSEADDR, (void*)&optval, QSE_SIZEOF(optval));
|
|
}
|
|
|
|
if (bind (nwio->handle, (struct sockaddr*)&addr, addrlen) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (sock_errno());
|
|
goto oops;
|
|
}
|
|
|
|
if (flags & QSE_NWIO_TCP)
|
|
{
|
|
if (listen (nwio->handle, 10) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (sock_errno());
|
|
goto oops;
|
|
}
|
|
|
|
if (nwio->tmout.a >= 0 &&
|
|
wait_for_data (nwio, nwio->tmout.a, 0) <= -1) goto oops;
|
|
|
|
handle = accept (nwio->handle, (struct sockaddr*)&addr, &addrlen);
|
|
if (handle <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (sock_errno());
|
|
goto oops;
|
|
}
|
|
|
|
soclose (nwio->handle);
|
|
nwio->handle = handle;
|
|
}
|
|
else if (flags & QSE_NWIO_UDP)
|
|
{
|
|
nwio->status |= STATUS_UDP_CONNECT;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
int xret;
|
|
|
|
if (nwio->tmout.c >= 0 && (flags & QSE_NWIO_TCP))
|
|
{
|
|
int noblk = 1;
|
|
|
|
if (ioctl (nwio->handle, FIONBIO, &noblk, QSE_SIZEOF(noblk)) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (sock_errno());
|
|
goto oops;
|
|
}
|
|
}
|
|
|
|
xret = connect (nwio->handle, (struct sockaddr*)&addr, addrlen);
|
|
|
|
if (nwio->tmout.c >= 0 && (flags & QSE_NWIO_TCP))
|
|
{
|
|
int noblk = 0;
|
|
|
|
if ((xret <= -1 && sock_errno() != SOCEINPROGRESS) ||
|
|
ioctl (nwio->handle, FIONBIO, &noblk, QSE_SIZEOF(noblk)) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (sock_errno());
|
|
goto oops;
|
|
}
|
|
|
|
if (wait_for_data (nwio, nwio->tmout.c, 1) <= -1) goto oops;
|
|
else
|
|
{
|
|
int xlen, xerr;
|
|
|
|
xlen = QSE_SIZEOF(xerr);
|
|
if (getsockopt (nwio->handle, SOL_SOCKET, SO_ERROR, (char*)&xerr, &xlen) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (sock_errno());
|
|
goto oops;
|
|
}
|
|
else if (xerr != 0)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (xerr);
|
|
goto oops;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (xret <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (sock_errno());
|
|
goto oops;
|
|
}
|
|
}
|
|
}
|
|
|
|
#elif defined(__DOS__)
|
|
nwio->errnum = QSE_NWIO_ENOIMPL;
|
|
return -1;
|
|
|
|
#else
|
|
#if defined(SOCK_CLOEXEC)
|
|
nwio->handle = socket (family, type | SOCK_CLOEXEC, 0);
|
|
#else
|
|
nwio->handle = socket (family, type, 0);
|
|
#endif
|
|
if (nwio->handle <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
goto oops;
|
|
}
|
|
|
|
#if !defined(SOCK_CLOEXEC) && defined(FD_CLOEXEC)
|
|
{
|
|
int tmp = fcntl (nwio->handle, F_GETFD);
|
|
if (tmp >= 0) fcntl (nwio->handle, F_SETFD, tmp | FD_CLOEXEC);
|
|
}
|
|
#endif
|
|
|
|
if ((flags & QSE_NWIO_TCP) && (flags & QSE_NWIO_KEEPALIVE))
|
|
{
|
|
int optval = 1;
|
|
setsockopt (nwio->handle, SOL_SOCKET, SO_KEEPALIVE, (void*)&optval, QSE_SIZEOF(optval));
|
|
}
|
|
|
|
if (flags & QSE_NWIO_PASSIVE)
|
|
{
|
|
qse_nwio_hnd_t handle;
|
|
|
|
#if defined(SO_REUSEADDR)
|
|
if (flags & QSE_NWIO_REUSEADDR)
|
|
{
|
|
int optval = 1;
|
|
setsockopt (nwio->handle, SOL_SOCKET, SO_REUSEADDR, (void*)&optval, QSE_SIZEOF(optval));
|
|
}
|
|
#endif
|
|
|
|
if (bind (nwio->handle, (struct sockaddr*)&addr, addrlen) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
goto oops;
|
|
}
|
|
|
|
if (flags & QSE_NWIO_TCP)
|
|
{
|
|
if (listen (nwio->handle, 10) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
goto oops;
|
|
}
|
|
|
|
if (nwio->tmout.a >= 0 &&
|
|
wait_for_data (nwio, nwio->tmout.a, 0) <= -1) goto oops;
|
|
|
|
handle = accept (nwio->handle, (struct sockaddr*)&addr, &addrlen);
|
|
if (handle <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
goto oops;
|
|
}
|
|
|
|
QSE_CLOSE (nwio->handle);
|
|
nwio->handle = handle;
|
|
}
|
|
else if (flags & QSE_NWIO_UDP)
|
|
{
|
|
nwio->status |= STATUS_UDP_CONNECT;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
int orgfl, xret;
|
|
|
|
if (nwio->tmout.c >= 0 && (flags & QSE_NWIO_TCP))
|
|
{
|
|
orgfl = fcntl (nwio->handle, F_GETFL, 0);
|
|
if (orgfl <= -1 ||
|
|
fcntl (nwio->handle, F_SETFL, orgfl | O_NONBLOCK) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
goto oops;
|
|
}
|
|
}
|
|
|
|
xret = connect (nwio->handle, (struct sockaddr*)&addr, addrlen);
|
|
|
|
if (nwio->tmout.c >= 0 && (flags & QSE_NWIO_TCP))
|
|
{
|
|
if ((xret <= -1 && errno != EINPROGRESS) ||
|
|
fcntl (nwio->handle, F_SETFL, orgfl) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
goto oops;
|
|
}
|
|
|
|
if (wait_for_data (nwio, nwio->tmout.c, 1) <= -1) goto oops;
|
|
else
|
|
{
|
|
#if defined(HAVE_SOCKLEN_T)
|
|
socklen_t xlen;
|
|
#else
|
|
int xlen;
|
|
#endif
|
|
xlen = QSE_SIZEOF(xret);
|
|
if (getsockopt (nwio->handle, SOL_SOCKET, SO_ERROR, (char*)&xret, &xlen) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
goto oops;
|
|
}
|
|
else if (xret != 0)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (xret);
|
|
goto oops;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (xret <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
goto oops;
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
|
|
if (flags & QSE_NWIO_TEXT)
|
|
{
|
|
int topt = 0;
|
|
|
|
if (flags & QSE_NWIO_IGNOREMBWCERR) topt |= QSE_TIO_IGNOREMBWCERR;
|
|
if (flags & QSE_NWIO_NOAUTOFLUSH) topt |= QSE_TIO_NOAUTOFLUSH;
|
|
|
|
nwio->tio = qse_tio_open (mmgr, QSE_SIZEOF(qse_nwio_t*), topt);
|
|
if (nwio->tio == QSE_NULL)
|
|
{
|
|
nwio->errnum = QSE_NWIO_ENOMEM;
|
|
goto oops;
|
|
}
|
|
|
|
/* store the back-reference to nwio in the extension area.*/
|
|
*(qse_nwio_t**)QSE_XTN(nwio->tio) = nwio;
|
|
|
|
if (qse_tio_attachin (nwio->tio, socket_input, QSE_NULL, 4096) <= -1 ||
|
|
qse_tio_attachout (nwio->tio, socket_output, QSE_NULL, 4096) <= -1)
|
|
{
|
|
if (nwio->errnum == QSE_NWIO_ENOERR)
|
|
nwio->errnum = tio_errnum_to_nwio_errnum (nwio->tio);
|
|
goto oops;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
|
|
oops:
|
|
if (nwio->tio)
|
|
{
|
|
qse_tio_close (nwio->tio);
|
|
nwio->tio = QSE_NULL;
|
|
}
|
|
|
|
#if defined(_WIN32)
|
|
if (nwio->handle != INVALID_SOCKET) closesocket (nwio->handle);
|
|
|
|
#elif defined(__OS2__)
|
|
if (nwio->handle >= 0) soclose (nwio->handle);
|
|
|
|
#elif defined(__DOS__)
|
|
/* TODO: */
|
|
|
|
#else
|
|
if (nwio->handle >= 0) QSE_CLOSE (nwio->handle);
|
|
#endif
|
|
return -1;
|
|
}
|
|
|
|
void qse_nwio_fini (qse_nwio_t* nwio)
|
|
{
|
|
/*if (qse_nwio_flush (nwio) <= -1) return -1;*/
|
|
qse_nwio_flush (nwio);
|
|
if (nwio->tio)
|
|
{
|
|
qse_tio_close (nwio->tio);
|
|
nwio->tio = QSE_NULL;
|
|
}
|
|
|
|
#if defined(_WIN32)
|
|
closesocket (nwio->handle);
|
|
#elif defined(__OS2__)
|
|
/* TODO: */
|
|
#elif defined(__DOS__)
|
|
/* TODO: */
|
|
#else
|
|
QSE_CLOSE (nwio->handle);
|
|
#endif
|
|
}
|
|
|
|
qse_nwio_errnum_t qse_nwio_geterrnum (const qse_nwio_t* nwio)
|
|
{
|
|
return nwio->errnum;
|
|
}
|
|
|
|
qse_cmgr_t* qse_nwio_getcmgr (qse_nwio_t* nwio)
|
|
{
|
|
return nwio->tio? qse_tio_getcmgr (nwio->tio): QSE_NULL;
|
|
}
|
|
|
|
void qse_nwio_setcmgr (qse_nwio_t* nwio, qse_cmgr_t* cmgr)
|
|
{
|
|
if (nwio->tio) qse_tio_setcmgr (nwio->tio, cmgr);
|
|
}
|
|
|
|
qse_nwio_hnd_t qse_nwio_gethandle (const qse_nwio_t* nwio)
|
|
{
|
|
return nwio->handle;
|
|
}
|
|
|
|
qse_ubi_t qse_nwio_gethandleasubi (const qse_nwio_t* nwio)
|
|
{
|
|
qse_ubi_t ubi;
|
|
|
|
#if defined(_WIN32)
|
|
ubi.intptr = nwio->handle;
|
|
#elif defined(__OS2__)
|
|
ubi.i = nwio->handle;
|
|
#elif defined(__DOS__)
|
|
ubi.i = nwio->handle;
|
|
#else
|
|
ubi.i = nwio->handle;
|
|
#endif
|
|
return ubi;
|
|
}
|
|
|
|
qse_ssize_t qse_nwio_flush (qse_nwio_t* nwio)
|
|
{
|
|
qse_ssize_t n;
|
|
|
|
if (nwio->tio)
|
|
{
|
|
nwio->errnum = QSE_NWIO_ENOERR;
|
|
n = qse_tio_flush (nwio->tio);
|
|
if (n <= -1 && nwio->errnum == QSE_NWIO_ENOERR)
|
|
nwio->errnum = tio_errnum_to_nwio_errnum (nwio->tio);
|
|
}
|
|
else n = 0;
|
|
|
|
return n;
|
|
}
|
|
|
|
void qse_nwio_purge (qse_nwio_t* nwio)
|
|
{
|
|
if (nwio->tio) qse_tio_purge (nwio->tio);
|
|
}
|
|
|
|
/* ---------------------------------------------------------- */
|
|
|
|
static qse_ssize_t nwio_read (qse_nwio_t* nwio, void* buf, qse_size_t size)
|
|
{
|
|
#if defined(_WIN32)
|
|
int count;
|
|
#elif defined(__OS2__)
|
|
int n;
|
|
#elif defined(__DOS__)
|
|
int n;
|
|
#else
|
|
qse_ssize_t n;
|
|
#endif
|
|
|
|
#if defined(_WIN32)
|
|
if (size > (QSE_TYPE_MAX(qse_ssize_t) & QSE_TYPE_MAX(int)))
|
|
size = QSE_TYPE_MAX(qse_ssize_t) & QSE_TYPE_MAX(int);
|
|
|
|
if (nwio->status & STATUS_UDP_CONNECT)
|
|
{
|
|
union sockaddr_t addr;
|
|
int addrlen;
|
|
|
|
addrlen = QSE_SIZEOF(addr);
|
|
|
|
if (nwio->tmout.a >= 0 &&
|
|
wait_for_data (nwio, nwio->tmout.a, 0) <= -1) return -1;
|
|
|
|
count = recvfrom (
|
|
nwio->handle, buf, size, 0,
|
|
(struct sockaddr*)&addr, &addrlen);
|
|
if (count == SOCKET_ERROR)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
}
|
|
else if (count >= 1)
|
|
{
|
|
/* for udp, it just creates a stream with the
|
|
* first sender */
|
|
if (connect (nwio->handle, (struct sockaddr*)&addr, addrlen) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
return -1;
|
|
}
|
|
nwio->status &= ~STATUS_UDP_CONNECT;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (nwio->tmout.r >= 0 &&
|
|
wait_for_data (nwio, nwio->tmout.r, 0) <= -1) return -1;
|
|
|
|
count = recv (nwio->handle, buf, size, 0);
|
|
if (count == SOCKET_ERROR) nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
}
|
|
|
|
return count;
|
|
|
|
#elif defined(__OS2__)
|
|
if (size > (QSE_TYPE_MAX(qse_ssize_t) & QSE_TYPE_MAX(int)))
|
|
size = QSE_TYPE_MAX(qse_ssize_t) & QSE_TYPE_MAX(int);
|
|
|
|
if (nwio->status & STATUS_UDP_CONNECT)
|
|
{
|
|
union sockaddr_t addr;
|
|
int addrlen;
|
|
|
|
addrlen = QSE_SIZEOF(addr);
|
|
|
|
if (nwio->tmout.a >= 0 &&
|
|
wait_for_data (nwio, nwio->tmout.a, 0) <= -1) return -1;
|
|
|
|
n = recvfrom (
|
|
nwio->handle, buf, size, 0,
|
|
(struct sockaddr*)&addr, &addrlen);
|
|
if (n <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (sock_errno());
|
|
}
|
|
else if (n >= 1)
|
|
{
|
|
/* for udp, it just creates a stream with the
|
|
* first sender */
|
|
if (connect (nwio->handle, (struct sockaddr*)&addr, addrlen) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (sock_errno());
|
|
return -1;
|
|
}
|
|
nwio->status &= ~STATUS_UDP_CONNECT;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (nwio->tmout.r >= 0 &&
|
|
wait_for_data (nwio, nwio->tmout.r, 0) <= -1) return -1;
|
|
|
|
n = recv (nwio->handle, buf, size, 0);
|
|
if (n <= -1) nwio->errnum = syserr_to_errnum (sock_errno());
|
|
}
|
|
|
|
return n;
|
|
|
|
#elif defined(__DOS__)
|
|
|
|
nwio->errnum = QSE_NWIO_ENOIMPL;
|
|
return -1;
|
|
|
|
#else
|
|
|
|
if (size > (QSE_TYPE_MAX(qse_ssize_t) & QSE_TYPE_MAX(size_t)))
|
|
size = QSE_TYPE_MAX(qse_ssize_t) & QSE_TYPE_MAX(size_t);
|
|
|
|
reread:
|
|
if (nwio->status & STATUS_UDP_CONNECT)
|
|
{
|
|
union sockaddr_t addr;
|
|
#if defined(HAVE_SOCKLEN_T)
|
|
socklen_t addrlen;
|
|
#else
|
|
int addrlen;
|
|
#endif
|
|
|
|
addrlen = QSE_SIZEOF(addr);
|
|
|
|
/* it's similar to accept for tcp because i'm expecting
|
|
* the first sender and call connect() to it below just
|
|
* like the 'nc' utility does.
|
|
* so i treat this recvfrom() as if it is accept().
|
|
*/
|
|
if (nwio->tmout.a >= 0 &&
|
|
wait_for_data (nwio, nwio->tmout.a, 0) <= -1) return -1;
|
|
|
|
n = recvfrom (
|
|
nwio->handle, buf, size, 0,
|
|
(struct sockaddr*)&addr, &addrlen);
|
|
if (n <= -1)
|
|
{
|
|
if (errno == EINTR)
|
|
{
|
|
if (nwio->flags & QSE_NWIO_READNORETRY)
|
|
nwio->errnum = QSE_NWIO_EINTR;
|
|
else goto reread;
|
|
}
|
|
else
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
}
|
|
}
|
|
else if (n >= 1)
|
|
{
|
|
/* for udp, it just creates a stream with the
|
|
* first sender */
|
|
if (connect (nwio->handle, (struct sockaddr*)&addr, addrlen) <= -1)
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
return -1;
|
|
}
|
|
nwio->status &= ~STATUS_UDP_CONNECT;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (nwio->tmout.r >= 0 &&
|
|
wait_for_data (nwio, nwio->tmout.r, 0) <= -1) return -1;
|
|
|
|
n = recv (nwio->handle, buf, size, 0);
|
|
if (n <= -1)
|
|
{
|
|
if (errno == EINTR)
|
|
{
|
|
if (nwio->flags & QSE_NWIO_READNORETRY)
|
|
nwio->errnum = QSE_NWIO_EINTR;
|
|
else goto reread;
|
|
}
|
|
else
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
}
|
|
}
|
|
}
|
|
|
|
return n;
|
|
#endif
|
|
}
|
|
|
|
qse_ssize_t qse_nwio_read (qse_nwio_t* nwio, void* buf, qse_size_t size)
|
|
{
|
|
if (nwio->tio == QSE_NULL)
|
|
return nwio_read (nwio, buf, size);
|
|
else
|
|
{
|
|
qse_ssize_t n;
|
|
|
|
nwio->errnum = QSE_NWIO_ENOERR;
|
|
n = qse_tio_read (nwio->tio, buf, size);
|
|
if (n <= -1 && nwio->errnum == QSE_NWIO_ENOERR)
|
|
nwio->errnum = tio_errnum_to_nwio_errnum (nwio->tio);
|
|
|
|
return n;
|
|
}
|
|
}
|
|
|
|
static qse_ssize_t nwio_write (qse_nwio_t* nwio, const void* data, qse_size_t size)
|
|
{
|
|
#if defined(_WIN32)
|
|
int count;
|
|
#elif defined(__OS2__)
|
|
int n;
|
|
#elif defined(__DOS__)
|
|
int n;
|
|
#else
|
|
qse_ssize_t n;
|
|
#endif
|
|
|
|
#if defined(_WIN32)
|
|
|
|
if (size > (QSE_TYPE_MAX(qse_ssize_t) & QSE_TYPE_MAX(int)))
|
|
size = QSE_TYPE_MAX(qse_ssize_t) & QSE_TYPE_MAX(int);
|
|
|
|
if (nwio->tmout.w >= 0 &&
|
|
wait_for_data (nwio, nwio->tmout.w, 1) <= -1) return -1;
|
|
|
|
count = send (nwio->handle, data, size, 0);
|
|
if (count == SOCKET_ERROR) nwio->errnum = syserr_to_errnum (WSAGetLastError());
|
|
return count;
|
|
|
|
#elif defined(__OS2__)
|
|
|
|
if (size > (QSE_TYPE_MAX(qse_ssize_t) & QSE_TYPE_MAX(int)))
|
|
size = QSE_TYPE_MAX(qse_ssize_t) & QSE_TYPE_MAX(int);
|
|
|
|
if (nwio->tmout.w >= 0 &&
|
|
wait_for_data (nwio, nwio->tmout.w, 1) <= -1) return -1;
|
|
|
|
n = send (nwio->handle, data, size, 0);
|
|
if (n <= -1) nwio->errnum = syserr_to_errnum (sock_errno());
|
|
return n;
|
|
|
|
#elif defined(__DOS__)
|
|
|
|
nwio->errnum = QSE_NWIO_ENOIMPL;
|
|
return -1;
|
|
|
|
#else
|
|
|
|
if (size > (QSE_TYPE_MAX(qse_ssize_t) & QSE_TYPE_MAX(size_t)))
|
|
size = QSE_TYPE_MAX(qse_ssize_t) & QSE_TYPE_MAX(size_t);
|
|
|
|
rewrite:
|
|
if (nwio->tmout.w >= 0 &&
|
|
wait_for_data (nwio, nwio->tmout.w, 1) <= -1) return -1;
|
|
|
|
n = send (nwio->handle, data, size, 0);
|
|
if (n <= -1)
|
|
{
|
|
if (errno == EINTR)
|
|
{
|
|
if (nwio->flags & QSE_NWIO_WRITENORETRY)
|
|
nwio->errnum = QSE_NWIO_EINTR;
|
|
else goto rewrite;
|
|
}
|
|
else
|
|
{
|
|
nwio->errnum = syserr_to_errnum (errno);
|
|
}
|
|
}
|
|
return n;
|
|
|
|
#endif
|
|
}
|
|
|
|
qse_ssize_t qse_nwio_write (qse_nwio_t* nwio, const void* data, qse_size_t size)
|
|
{
|
|
if (nwio->tio == QSE_NULL)
|
|
{
|
|
return nwio_write (nwio, data, size);
|
|
}
|
|
else
|
|
{
|
|
qse_ssize_t n;
|
|
|
|
nwio->errnum = QSE_NWIO_ENOERR;
|
|
n = qse_tio_write (nwio->tio, data, size);
|
|
if (n <= -1 && nwio->errnum == QSE_NWIO_ENOERR)
|
|
nwio->errnum = tio_errnum_to_nwio_errnum (nwio->tio);
|
|
|
|
return n;
|
|
}
|
|
}
|
|
|
|
/* ---------------------------------------------------------- */
|
|
|
|
static qse_ssize_t socket_input (
|
|
qse_tio_t* tio, qse_tio_cmd_t cmd, void* buf, qse_size_t size)
|
|
{
|
|
if (cmd == QSE_TIO_DATA)
|
|
{
|
|
qse_nwio_t* nwio;
|
|
|
|
nwio = *(qse_nwio_t**)QSE_XTN(tio);
|
|
QSE_ASSERT (nwio != QSE_NULL);
|
|
|
|
return nwio_read (nwio, buf, size);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static qse_ssize_t socket_output (
|
|
qse_tio_t* tio, qse_tio_cmd_t cmd, void* buf, qse_size_t size)
|
|
{
|
|
if (cmd == QSE_TIO_DATA)
|
|
{
|
|
qse_nwio_t* nwio;
|
|
|
|
nwio = *(qse_nwio_t**)QSE_XTN(tio);
|
|
QSE_ASSERT (nwio != QSE_NULL);
|
|
|
|
return nwio_write (nwio, buf, size);
|
|
}
|
|
|
|
return 0;
|
|
}
|