From d1d1db673bb3d51dd1abf275338b15543abeb64b Mon Sep 17 00:00:00 2001 From: teodor Date: Wed, 29 Sep 2004 17:07:01 +0000 Subject: [PATCH] poll<=select --- tcp.c | 24 +++++++++--------------- 1 file changed, 9 insertions(+), 15 deletions(-) diff --git a/tcp.c b/tcp.c index 7867f6d..6833be2 100644 --- a/tcp.c +++ b/tcp.c @@ -214,38 +214,32 @@ TC_ServerInitConnect( TC_Connection *cs ) { u_int32_t TC_ServerConnect( TC_Connection *cs ) { - fd_set fdwr, fdexcept; + struct pollfd pfd; int ret; - struct timeval timeout; if ( cs->state != CS_INPROCESS ) return cs->state; - FD_ZERO( &fdwr ); - FD_ZERO( &fdexcept ); - - FD_SET(cs->fd, &fdwr); - FD_SET(cs->fd, &fdexcept); - - memset( &timeout, 0, sizeof(struct timeval)); - - ret = select( cs->fd+1, NULL, &fdwr, &fdexcept, &timeout ) ; + pfd.fd = cs->fd; + pfd.events = POLLOUT; + pfd.revents = 0; + ret = poll( &pfd, 1, 0 ); if ( ret<0 ) { - tlog( TL_CRIT, "TC_ServerConnect: select: %s", + tlog( TL_CRIT, "TC_ServerConnect: poll: %s", strerror(errno)); cs->state = CS_ERROR; return CS_ERROR; } else if ( ret == 0 ) return CS_INPROCESS; - if ( FD_ISSET(cs->fd, &fdexcept) ) { - tlog( TL_CRIT, "TC_ServerConnect: select return connect error for %s:%d", + if ( (pfd.revents & (POLLHUP | POLLNVAL | POLLERR)) ) { + tlog( TL_CRIT, "TC_ServerConnect: poll return connect error for %s:%d", inet_ntoa(cs->serv_addr.sin_addr), ntohs(cs->serv_addr.sin_port)); cs->state = CS_ERROR; return CS_ERROR; } - if ( ! FD_ISSET(cs->fd, &fdwr) ) + if ( ! (pfd.revents & POLLOUT) ) return CS_INPROCESS; -- 2.37.3