mirror of
https://github.com/reactos/reactos.git
synced 2024-10-30 03:27:31 +00:00
138 lines
2.6 KiB
C
138 lines
2.6 KiB
C
#include "syshdrs.h"
|
|
|
|
#ifndef NO_SIGNALS
|
|
extern volatile Sjmp_buf gNetTimeoutJmp;
|
|
extern volatile Sjmp_buf gPipeJmp;
|
|
#endif
|
|
|
|
#ifndef NO_SIGNALS
|
|
|
|
int
|
|
SRecvmsg(int sfd, void *const msg, int fl, int tlen)
|
|
{
|
|
int nread, tleft;
|
|
vsio_sigproc_t sigalrm, sigpipe;
|
|
time_t done, now;
|
|
|
|
if (tlen < 0) {
|
|
errno = 0;
|
|
for (;;) {
|
|
nread = recvmsg(sfd, (struct msghdr *) msg, fl);
|
|
if ((nread >= 0) || (errno != EINTR))
|
|
return (nread);
|
|
}
|
|
}
|
|
|
|
if (SSetjmp(gNetTimeoutJmp) != 0) {
|
|
alarm(0);
|
|
(void) SSignal(SIGALRM, (sio_sigproc_t) sigalrm);
|
|
(void) SSignal(SIGPIPE, (sio_sigproc_t) sigpipe);
|
|
errno = ETIMEDOUT;
|
|
return (kTimeoutErr);
|
|
}
|
|
|
|
if (SSetjmp(gPipeJmp) != 0) {
|
|
alarm(0);
|
|
(void) SSignal(SIGALRM, (sio_sigproc_t) sigalrm);
|
|
(void) SSignal(SIGPIPE, (sio_sigproc_t) sigpipe);
|
|
errno = EPIPE;
|
|
return (kBrokenPipeErr);
|
|
}
|
|
|
|
sigalrm = (vsio_sigproc_t) SSignal(SIGALRM, SIOHandler);
|
|
sigpipe = (vsio_sigproc_t) SSignal(SIGPIPE, SIOHandler);
|
|
|
|
time(&now);
|
|
done = now + tlen;
|
|
tleft = (int) (done - now);
|
|
forever {
|
|
(void) alarm((unsigned int) tleft);
|
|
nread = recvmsg(sfd, (struct msghdr *) msg, fl);
|
|
(void) alarm(0);
|
|
if (nread >= 0)
|
|
break;
|
|
if (errno != EINTR)
|
|
break; /* Fatal error. */
|
|
errno = 0;
|
|
time(&now);
|
|
tleft = (int) (done - now);
|
|
if (tleft < 1) {
|
|
nread = kTimeoutErr;
|
|
errno = ETIMEDOUT;
|
|
break;
|
|
}
|
|
}
|
|
|
|
(void) SSignal(SIGALRM, (sio_sigproc_t) sigalrm);
|
|
(void) SSignal(SIGPIPE, (sio_sigproc_t) sigpipe);
|
|
|
|
return (nread);
|
|
} /* SRecvmsg */
|
|
|
|
#elif defined(HAVE_RECVMSG)
|
|
|
|
int
|
|
SRecvmsg(int sfd, void *const msg, int fl, int tlen)
|
|
{
|
|
int nread, tleft;
|
|
time_t done, now;
|
|
fd_set ss;
|
|
struct timeval tv;
|
|
int result;
|
|
|
|
if (tlen < 0) {
|
|
errno = 0;
|
|
for (;;) {
|
|
nread = recvmsg(sfd, (struct msghdr *) msg, fl);
|
|
if ((nread >= 0) || (errno != EINTR))
|
|
return (nread);
|
|
}
|
|
}
|
|
|
|
time(&now);
|
|
done = now + tlen;
|
|
tleft = (int) (done - now);
|
|
forever {
|
|
|
|
for (;;) {
|
|
errno = 0;
|
|
FD_ZERO(&ss);
|
|
FD_SET(sfd, &ss);
|
|
tv.tv_sec = tleft;
|
|
tv.tv_usec = 0;
|
|
result = select(sfd + 1, SELECT_TYPE_ARG234 &ss, NULL, NULL, SELECT_TYPE_ARG5 &tv);
|
|
if (result == 1) {
|
|
/* ready */
|
|
break;
|
|
} else if (result == 0) {
|
|
/* timeout */
|
|
errno = ETIMEDOUT;
|
|
SETWSATIMEOUTERR
|
|
return (kTimeoutErr);
|
|
} else if (errno != EINTR) {
|
|
return (-1);
|
|
}
|
|
}
|
|
|
|
nread = recvmsg(sfd, (struct msghdr *) msg, fl);
|
|
|
|
if (nread >= 0)
|
|
break;
|
|
if (errno != EINTR)
|
|
break; /* Fatal error. */
|
|
errno = 0;
|
|
time(&now);
|
|
tleft = (int) (done - now);
|
|
if (tleft < 1) {
|
|
nread = kTimeoutErr;
|
|
errno = ETIMEDOUT;
|
|
SETWSATIMEOUTERR
|
|
break;
|
|
}
|
|
}
|
|
|
|
return (nread);
|
|
} /* SRecvmsg */
|
|
|
|
#endif
|