git.sur5r.net Git - openldap/blob - libraries/libldap/os-ip.c

   1 /* os-ip.c -- platform-specific TCP & UDP related code */
   2 /* $OpenLDAP$ */
   3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
   4  *
   5  * Copyright 1998-2004 The OpenLDAP Foundation.
   6  * Portions Copyright 1999 Lars Uffmann.
   7  * All rights reserved.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted only as authorized by the OpenLDAP
  11  * Public License.
  12  *
  13  * A copy of this license is available in the file LICENSE in the
  14  * top-level directory of the distribution or, alternatively, at
  15  * <http://www.OpenLDAP.org/license.html>.
  16  */
  17 /* Portions Copyright (c) 1995 Regents of the University of Michigan.
  18  * All rights reserved.
  19  */
  20 /* Significant additional contributors include:
  21  *    Lars Uffman
  22  */
  23
  24 #include "portable.h"
  25
  26 #include <stdio.h>
  27
  28 #include <ac/stdlib.h>
  29
  30 #include <ac/errno.h>
  31 #include <ac/socket.h>
  32 #include <ac/string.h>
  33 #include <ac/time.h>
  34 #include <ac/unistd.h>
  35
  36 #ifdef HAVE_IO_H
  37 #include <io.h>
  38 #endif /* HAVE_IO_H */
  39
  40 #include "ldap-int.h"
  41
  42 #if defined( HAVE_GETADDRINFO ) && defined( HAVE_INET_NTOP )
  43 #  ifdef LDAP_PF_INET6
  44 int ldap_int_inet4or6 = AF_UNSPEC;
  45 #  else
  46 int ldap_int_inet4or6 = AF_INET;
  47 #  endif
  48 #endif
  49
  50 #ifdef LDAP_DEBUG
  51
  52 #define osip_debug(ld,fmt,arg1,arg2,arg3) \
  53 do { \
  54         ldap_log_printf(NULL, LDAP_DEBUG_TRACE, fmt, arg1, arg2, arg3); \
  55 } while(0)
  56
  57 #else
  58
  59 #define osip_debug(ld,fmt,arg1,arg2,arg3) ((void)0)
  60
  61 #endif /* LDAP_DEBUG */
  62
  63 static void
  64 ldap_pvt_set_errno(int err)
  65 {
  66         errno = err;
  67 }
  68
  69 int
  70 ldap_int_timeval_dup( struct timeval **dest, const struct timeval *src )
  71 {
  72         struct timeval *new;
  73
  74         assert( dest != NULL );
  75
  76         if (src == NULL) {
  77                 *dest = NULL;
  78                 return 0;
  79         }
  80
  81         new = (struct timeval *) LDAP_MALLOC(sizeof(struct timeval));
  82
  83         if( new == NULL ) {
  84                 *dest = NULL;
  85                 return 1;
  86         }
  87
  88         AC_MEMCPY( (char *) new, (const char *) src, sizeof(struct timeval));
  89
  90         *dest = new;
  91         return 0;
  92 }
  93
  94 static int
  95 ldap_pvt_ndelay_on(LDAP *ld, int fd)
  96 {
  97         osip_debug(ld, "ldap_ndelay_on: %d\n",fd,0,0);
  98         return ber_pvt_socket_set_nonblock( fd, 1 );
  99 }
 100
 101 static int
 102 ldap_pvt_ndelay_off(LDAP *ld, int fd)
 103 {
 104         osip_debug(ld, "ldap_ndelay_off: %d\n",fd,0,0);
 105         return ber_pvt_socket_set_nonblock( fd, 0 );
 106 }
 107
 108 static ber_socket_t
 109 ldap_int_socket(LDAP *ld, int family, int type )
 110 {
 111         ber_socket_t s = socket(family, type, 0);
 112         osip_debug(ld, "ldap_new_socket: %d\n",s,0,0);
 113         return ( s );
 114 }
 115
 116 static int
 117 ldap_pvt_close_socket(LDAP *ld, int s)
 118 {
 119         osip_debug(ld, "ldap_close_socket: %d\n",s,0,0);
 120         return tcp_close(s);
 121 }
 122
 123 static int
 124 ldap_int_prepare_socket(LDAP *ld, int s, int proto )
 125 {
 126         osip_debug(ld, "ldap_prepare_socket: %d\n", s,0,0);
 127
 128 #ifdef TCP_NODELAY
 129         if( proto == LDAP_PROTO_TCP ) {
 130                 int dummy = 1;
 131                 if ( setsockopt( s, IPPROTO_TCP, TCP_NODELAY,
 132                         (char*) &dummy, sizeof(dummy) ) == AC_SOCKET_ERROR )
 133                 {
 134                         osip_debug(ld, "ldap_prepare_socket: "
 135                                 "setsockopt(%d, TCP_NODELAY) failed (ignored).\n",
 136                                 s, 0, 0);
 137                 }
 138         }
 139 #endif
 140
 141         return 0;
 142 }
 143
 144 #ifndef HAVE_WINSOCK
 145
 146 #undef TRACE
 147 #define TRACE do { \
 148         osip_debug(ld, \
 149                 "ldap_is_socket_ready: error on socket %d: errno: %d (%s)\n", \
 150                 s, \
 151                 errno, \
 152                 sock_errstr(errno) ); \
 153 } while( 0 )
 154
 155 /*
 156  * check the socket for errors after select returned.
 157  */
 158 static int
 159 ldap_pvt_is_socket_ready(LDAP *ld, int s)
 160 {
 161         osip_debug(ld, "ldap_is_sock_ready: %d\n",s,0,0);
 162
 163 #if defined( notyet ) /* && defined( SO_ERROR ) */
 164 {
 165         int so_errno;
 166         socklen_t dummy = sizeof(so_errno);
 167         if ( getsockopt( s, SOL_SOCKET, SO_ERROR, &so_errno, &dummy )
 168                 == AC_SOCKET_ERROR )
 169         {
 170                 return -1;
 171         }
 172         if ( so_errno ) {
 173                 ldap_pvt_set_errno(so_errno);
 174                 TRACE;
 175                 return -1;
 176         }
 177         return 0;
 178 }
 179 #else
 180 {
 181         /* error slippery */
 182 #ifdef LDAP_PF_INET6
 183         struct sockaddr_storage sin;
 184 #else
 185         struct sockaddr_in sin;
 186 #endif
 187         char ch;
 188         socklen_t dummy = sizeof(sin);
 189         if ( getpeername( s, (struct sockaddr *) &sin, &dummy )
 190                 == AC_SOCKET_ERROR )
 191         {
 192                 /* XXX: needs to be replace with ber_stream_read() */
 193                 read(s, &ch, 1);
 194                 TRACE;
 195                 return -1;
 196         }
 197         return 0;
 198 }
 199 #endif
 200         return -1;
 201 }
 202 #undef TRACE
 203
 204 #endif /* HAVE_WINSOCK */
 205
 206 static int
 207 ldap_pvt_connect(LDAP *ld, ber_socket_t s,
 208         struct sockaddr *sin, socklen_t addrlen,
 209         int async)
 210 {
 211         int rc;
 212         struct timeval  tv, *opt_tv=NULL;
 213
 214 #ifdef LDAP_CONNECTIONLESS
 215         /* We could do a connect() but that would interfere with
 216          * attempts to poll a broadcast address
 217          */
 218         if (LDAP_IS_UDP(ld)) {
 219                 if (ld->ld_options.ldo_peer)
 220                         ldap_memfree(ld->ld_options.ldo_peer);
 221                 ld->ld_options.ldo_peer=ldap_memalloc(sizeof(struct sockaddr));
 222                 AC_MEMCPY(ld->ld_options.ldo_peer,sin,sizeof(struct sockaddr));
 223                 return ( 0 );
 224         }
 225 #endif
 226         if ( (opt_tv = ld->ld_options.ldo_tm_net) != NULL ) {
 227                 tv.tv_usec = opt_tv->tv_usec;
 228                 tv.tv_sec = opt_tv->tv_sec;
 229         }
 230
 231         osip_debug(ld, "ldap_connect_timeout: fd: %d tm: %ld async: %d\n",
 232                         s, opt_tv ? tv.tv_sec : -1L, async);
 233
 234         if ( ldap_pvt_ndelay_on(ld, s) == -1 )
 235                 return ( -1 );
 236
 237         if ( connect(s, sin, addrlen) != AC_SOCKET_ERROR ) {
 238                 if ( ldap_pvt_ndelay_off(ld, s) == -1 )
 239                         return ( -1 );
 240                 return ( 0 );
 241         }
 242
 243 #ifdef HAVE_WINSOCK
 244         ldap_pvt_set_errno( WSAGetLastError() );
 245 #endif
 246
 247         if ( errno != EINPROGRESS && errno != EWOULDBLOCK ) {
 248                 return ( -1 );
 249         }
 250
 251 #ifdef notyet
 252         if ( async ) return ( -2 );
 253 #endif
 254
 255 #ifdef HAVE_POLL
 256         {
 257                 struct pollfd fd;
 258                 int timeout = INFTIM;
 259
 260                 if( opt_tv != NULL ) timeout = TV2MILLISEC( &tv );
 261
 262                 fd.fd = s;
 263                 fd.events = POLLOUT;
 264
 265                 do {
 266                         fd.revents = 0;
 267                         rc = poll( &fd, 1, timeout );
 268                 } while( rc == AC_SOCKET_ERROR && errno == EINTR &&
 269                         LDAP_BOOL_GET(&ld->ld_options, LDAP_BOOL_RESTART ));
 270
 271                 if( rc == AC_SOCKET_ERROR ) return rc;
 272
 273                 if( fd.revents & POLLOUT ) {
 274                         if ( ldap_pvt_is_socket_ready(ld, s) == -1 ) return -1;
 275                         if ( ldap_pvt_ndelay_off(ld, s) == -1 ) return -1;
 276                         return ( 0 );
 277                 }
 278         }
 279 #else
 280         {
 281                 fd_set          wfds, *z=NULL;
 282 #ifdef HAVE_WINSOCK
 283                 fd_set          efds;
 284 #endif
 285                 FD_ZERO(&wfds);
 286                 FD_SET(s, &wfds );
 287
 288 #ifdef HAVE_WINSOCK
 289                 FD_ZERO(&efds);
 290                 FD_SET(s, &efds );
 291 #endif
 292
 293                 do {
 294                         rc = select(ldap_int_tblsize, z, &wfds,
 295 #ifdef HAVE_WINSOCK
 296                                 &efds,
 297 #else
 298                                 z,
 299 #endif
 300                                 opt_tv ? &tv : NULL);
 301                 } while( rc == AC_SOCKET_ERROR && errno == EINTR &&
 302                         LDAP_BOOL_GET(&ld->ld_options, LDAP_BOOL_RESTART ));
 303
 304                 if( rc == AC_SOCKET_ERROR ) return rc;
 305
 306 #ifdef HAVE_WINSOCK
 307                 /* This means the connection failed */
 308                 if ( FD_ISSET(s, &efds) ) {
 309                     int so_errno;
 310                     int dummy = sizeof(so_errno);
 311                     if ( getsockopt( s, SOL_SOCKET, SO_ERROR,
 312                                 (char *) &so_errno, &dummy ) == AC_SOCKET_ERROR || !so_errno )
 313                     {
 314                         /* impossible */
 315                         so_errno = WSAGetLastError();
 316                     }
 317                     ldap_pvt_set_errno(so_errno);
 318                     osip_debug(ld, "ldap_pvt_connect: error on socket %d: "
 319                                "errno: %d (%s)\n", s, errno, sock_errstr(errno));
 320                     return -1;
 321                 }
 322 #endif
 323                 if ( FD_ISSET(s, &wfds) ) {
 324 #ifndef HAVE_WINSOCK
 325                         if ( ldap_pvt_is_socket_ready(ld, s) == -1 ) return -1;
 326 #endif
 327                         if ( ldap_pvt_ndelay_off(ld, s) == -1 ) return -1;
 328                         return 0;
 329                 }
 330         }
 331 #endif
 332
 333         osip_debug(ld, "ldap_connect_timeout: timed out\n",0,0,0);
 334         ldap_pvt_set_errno( ETIMEDOUT );
 335         return -1;
 336 }
 337
 338 #ifndef HAVE_INET_ATON
 339 int
 340 ldap_pvt_inet_aton( const char *host, struct in_addr *in)
 341 {
 342         unsigned long u = inet_addr( host );
 343         if ( u != 0xffffffff || u != (unsigned long) -1 ) {
 344                 in->s_addr = u;
 345                 return 1;
 346         }
 347         return 0;
 348 }
 349 #endif
 350
 351
 352 int
 353 ldap_connect_to_host(LDAP *ld, Sockbuf *sb,
 354         int proto,
 355         const char *host, int port,
 356         int async )
 357 {
 358         int     rc;
 359         int     socktype;
 360         ber_socket_t            s = AC_SOCKET_INVALID;
 361
 362 #if defined( HAVE_GETADDRINFO ) && defined( HAVE_INET_NTOP )
 363         char serv[7];
 364         int err;
 365         struct addrinfo hints, *res, *sai;
 366 #else
 367         int i;
 368         int use_hp = 0;
 369         struct hostent *hp = NULL;
 370         struct hostent he_buf;
 371         struct in_addr in;
 372         char *ha_buf=NULL;
 373 #endif
 374
 375         if( host == NULL ) host = "localhost";
 376
 377         switch(proto) {
 378         case LDAP_PROTO_TCP: socktype = SOCK_STREAM;
 379                 osip_debug( ld,
 380                         "ldap_connect_to_host: TCP %s:%d\n",
 381                         host, port, 0);
 382                 break;
 383         case LDAP_PROTO_UDP: socktype = SOCK_DGRAM;
 384                 osip_debug( ld,
 385                         "ldap_connect_to_host: UDP %s:%d\n",
 386                         host, port, 0);
 387                 break;
 388         default:
 389                 osip_debug( ld, "ldap_connect_to_host: unknown proto: %d\n",
 390                         proto, 0, 0 );
 391                 return -1;
 392         }
 393
 394 #if defined( HAVE_GETADDRINFO ) && defined( HAVE_INET_NTOP )
 395         memset( &hints, '\0', sizeof(hints) );
 396         hints.ai_family = ldap_int_inet4or6;
 397         hints.ai_socktype = socktype;
 398         snprintf(serv, sizeof serv, "%d", port );
 399
 400 #ifdef LDAP_R_COMPILE
 401         /* most getaddrinfo(3) use non-threadsafe resolver libraries */
 402         ldap_pvt_thread_mutex_lock(&ldap_int_resolv_mutex);
 403 #endif
 404
 405         err = getaddrinfo( host, serv, &hints, &res );
 406
 407 #ifdef LDAP_R_COMPILE
 408         ldap_pvt_thread_mutex_unlock(&ldap_int_resolv_mutex);
 409 #endif
 410
 411         if ( err != 0 ) {
 412                 osip_debug(ld, "ldap_connect_to_host: getaddrinfo failed: %s\n",
 413                         AC_GAI_STRERROR(err), 0, 0);
 414                 return -1;
 415         }
 416         rc = -1;
 417
 418         for( sai=res; sai != NULL; sai=sai->ai_next) {
 419                 if( sai->ai_addr == NULL ) {
 420                         osip_debug(ld, "ldap_connect_to_host: getaddrinfo "
 421                                 "ai_addr is NULL?\n", 0, 0, 0);
 422                         continue;
 423                 }
 424
 425                 /* we assume AF_x and PF_x are equal for all x */
 426                 s = ldap_int_socket( ld, sai->ai_family, socktype );
 427                 if ( s == AC_SOCKET_INVALID ) {
 428                         continue;
 429                 }
 430
 431                 if ( ldap_int_prepare_socket(ld, s, proto ) == -1 ) {
 432                         ldap_pvt_close_socket(ld, s);
 433                         break;
 434                 }
 435
 436                 switch (sai->ai_family) {
 437 #ifdef LDAP_PF_INET6
 438                         case AF_INET6: {
 439                                 char addr[INET6_ADDRSTRLEN];
 440                                 inet_ntop( AF_INET6,
 441                                         &((struct sockaddr_in6 *)sai->ai_addr)->sin6_addr,
 442                                         addr, sizeof addr);
 443                                 osip_debug(ld, "ldap_connect_to_host: Trying %s %s\n",
 444                                         addr, serv, 0);
 445                         } break;
 446 #endif
 447                         case AF_INET: {
 448                                 char addr[INET_ADDRSTRLEN];
 449                                 inet_ntop( AF_INET,
 450                                         &((struct sockaddr_in *)sai->ai_addr)->sin_addr,
 451                                         addr, sizeof addr);
 452                                 osip_debug(ld, "ldap_connect_to_host: Trying %s:%s\n",
 453                                         addr, serv, 0);
 454                         } break;
 455                 }
 456
 457                 rc = ldap_pvt_connect( ld, s,
 458                         sai->ai_addr, sai->ai_addrlen, async );
 459                 if ( (rc == 0) || (rc == -2) ) {
 460                         ber_sockbuf_ctrl( sb, LBER_SB_OPT_SET_FD, &s );
 461                         break;
 462                 }
 463                 ldap_pvt_close_socket(ld, s);
 464         }
 465         freeaddrinfo(res);
 466
 467 #else
 468         if (! inet_aton( host, &in ) ) {
 469                 int local_h_errno;
 470                 rc = ldap_pvt_gethostbyname_a( host, &he_buf, &ha_buf,
 471                         &hp, &local_h_errno );
 472
 473                 if ( (rc < 0) || (hp == NULL) ) {
 474 #ifdef HAVE_WINSOCK
 475                         ldap_pvt_set_errno( WSAGetLastError() );
 476 #else
 477                         /* not exactly right, but... */
 478                         ldap_pvt_set_errno( EHOSTUNREACH );
 479 #endif
 480                         if (ha_buf) LDAP_FREE(ha_buf);
 481                         return -1;
 482                 }
 483
 484                 use_hp = 1;
 485         }
 486
 487         rc = s = -1;
 488         for ( i = 0; !use_hp || (hp->h_addr_list[i] != 0); ++i, rc = -1 ) {
 489                 struct sockaddr_in      sin;
 490
 491                 s = ldap_int_socket( ld, PF_INET, socktype );
 492                 if ( s == AC_SOCKET_INVALID ) {
 493                         /* use_hp ? continue : break; */
 494                         break;
 495                 }
 496
 497                 if ( ldap_int_prepare_socket( ld, s, proto ) == -1 ) {
 498                         ldap_pvt_close_socket(ld, s);
 499                         break;
 500                 }
 501
 502                 (void)memset((char *)&sin, '\0', sizeof sin);
 503                 sin.sin_family = AF_INET;
 504                 sin.sin_port = htons((short) port);
 505
 506                 if( use_hp ) {
 507                         AC_MEMCPY( &sin.sin_addr, hp->h_addr_list[i],
 508                                 sizeof(sin.sin_addr) );
 509                 } else {
 510                         AC_MEMCPY( &sin.sin_addr, &in.s_addr,
 511                                 sizeof(sin.sin_addr) );
 512                 }
 513
 514                 osip_debug(ld, "ldap_connect_to_host: Trying %s:%d\n",
 515                         inet_ntoa(sin.sin_addr), port, 0);
 516
 517                 rc = ldap_pvt_connect(ld, s,
 518                         (struct sockaddr *)&sin, sizeof(sin),
 519                         async);
 520
 521                 if ( (rc == 0) || (rc == -2) ) {
 522                         ber_sockbuf_ctrl( sb, LBER_SB_OPT_SET_FD, &s );
 523                         break;
 524                 }
 525
 526                 ldap_pvt_close_socket(ld, s);
 527
 528                 if (!use_hp) break;
 529         }
 530         if (ha_buf) LDAP_FREE(ha_buf);
 531 #endif
 532
 533         return rc;
 534 }
 535
 536 #if defined( LDAP_API_FEATURE_X_OPENLDAP_V2_KBIND ) || \
 537         defined( HAVE_CYRUS_SASL )
 538 char *
 539 ldap_host_connected_to( Sockbuf *sb, const char *host )
 540 {
 541         socklen_t               len;
 542 #ifdef LDAP_PF_INET6
 543         struct sockaddr_storage sabuf;
 544 #else
 545         struct sockaddr sabuf;
 546 #endif
 547         struct sockaddr *sa = (struct sockaddr *) &sabuf;
 548         ber_socket_t    sd;
 549
 550         (void)memset( (char *)sa, '\0', sizeof sabuf );
 551         len = sizeof sabuf;
 552
 553         ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_FD, &sd );
 554         if ( getpeername( sd, sa, &len ) == -1 ) {
 555                 return( NULL );
 556         }
 557
 558         /*
 559          * do a reverse lookup on the addr to get the official hostname.
 560          * this is necessary for kerberos to work right, since the official
 561          * hostname is used as the kerberos instance.
 562          */
 563
 564         switch (sa->sa_family) {
 565 #ifdef LDAP_PF_LOCAL
 566         case AF_LOCAL:
 567                 return LDAP_STRDUP( ldap_int_hostname );
 568 #endif
 569 #ifdef LDAP_PF_INET6
 570         case AF_INET6:
 571                 {
 572                         struct in6_addr localhost = IN6ADDR_LOOPBACK_INIT;
 573                         if( memcmp ( &((struct sockaddr_in6 *)sa)->sin6_addr,
 574                                 &localhost, sizeof(localhost)) == 0 )
 575                         {
 576                                 return LDAP_STRDUP( ldap_int_hostname );
 577                         }
 578                 }
 579                 break;
 580 #endif
 581         case AF_INET:
 582                 {
 583                         struct in_addr localhost;
 584                         localhost.s_addr = htonl( INADDR_ANY );
 585
 586                         if( memcmp ( &((struct sockaddr_in *)sa)->sin_addr,
 587                                 &localhost, sizeof(localhost) ) == 0 )
 588                         {
 589                                 return LDAP_STRDUP( ldap_int_hostname );
 590                         }
 591
 592 #ifdef INADDR_LOOPBACK
 593                         localhost.s_addr = htonl( INADDR_LOOPBACK );
 594
 595                         if( memcmp ( &((struct sockaddr_in *)sa)->sin_addr,
 596                                 &localhost, sizeof(localhost) ) == 0 )
 597                         {
 598                                 return LDAP_STRDUP( ldap_int_hostname );
 599                         }
 600 #endif
 601                 }
 602                 break;
 603
 604         default:
 605                 return( NULL );
 606                 break;
 607         }
 608
 609         {
 610                 char *herr;
 611 #ifdef NI_MAXHOST
 612                 char hbuf[NI_MAXHOST];
 613 #elif defined( MAXHOSTNAMELEN
 614                 char hbuf[MAXHOSTNAMELEN];
 615 #else
 616                 char hbuf[256];
 617 #endif
 618                 hbuf[0] = 0;
 619
 620                 if (ldap_pvt_get_hname( sa, len, hbuf, sizeof(hbuf), &herr ) == 0
 621                         && hbuf[0] )
 622                 {
 623                         return LDAP_STRDUP( hbuf );
 624                 }
 625         }
 626
 627         return host ? LDAP_STRDUP( host ) : NULL;
 628 }
 629 #endif
 630
 631
 632 #ifdef HAVE_POLL
 633 /* for UNIX poll(2) */
 634         /* ??? */
 635 #else
 636 /* for UNIX select(2) */
 637 struct selectinfo {
 638         fd_set  si_readfds;
 639         fd_set  si_writefds;
 640         fd_set  si_use_readfds;
 641         fd_set  si_use_writefds;
 642 };
 643 #endif
 644
 645 void
 646 ldap_mark_select_write( LDAP *ld, Sockbuf *sb )
 647 {
 648         struct selectinfo       *sip;
 649         ber_socket_t            sd;
 650
 651         sip = (struct selectinfo *)ld->ld_selectinfo;
 652
 653         ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_FD, &sd );
 654         if ( !FD_ISSET( sd, &sip->si_writefds )) {
 655                 FD_SET( sd, &sip->si_writefds );
 656         }
 657 }
 658
 659
 660 void
 661 ldap_mark_select_read( LDAP *ld, Sockbuf *sb )
 662 {
 663         struct selectinfo       *sip;
 664         ber_socket_t            sd;
 665
 666         sip = (struct selectinfo *)ld->ld_selectinfo;
 667
 668         ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_FD, &sd );
 669         if ( !FD_ISSET( sd, &sip->si_readfds )) {
 670                 FD_SET( sd, &sip->si_readfds );
 671         }
 672 }
 673
 674
 675 void
 676 ldap_mark_select_clear( LDAP *ld, Sockbuf *sb )
 677 {
 678         struct selectinfo       *sip;
 679         ber_socket_t            sd;
 680
 681         sip = (struct selectinfo *)ld->ld_selectinfo;
 682
 683         ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_FD, &sd );
 684         FD_CLR( sd, &sip->si_writefds );
 685         FD_CLR( sd, &sip->si_readfds );
 686 }
 687
 688
 689 int
 690 ldap_is_write_ready( LDAP *ld, Sockbuf *sb )
 691 {
 692         struct selectinfo       *sip;
 693         ber_socket_t            sd;
 694
 695         sip = (struct selectinfo *)ld->ld_selectinfo;
 696
 697         ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_FD, &sd );
 698         return( FD_ISSET( sd, &sip->si_use_writefds ));
 699 }
 700
 701
 702 int
 703 ldap_is_read_ready( LDAP *ld, Sockbuf *sb )
 704 {
 705         struct selectinfo       *sip;
 706         ber_socket_t            sd;
 707
 708         sip = (struct selectinfo *)ld->ld_selectinfo;
 709
 710         ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_FD, &sd );
 711         return( FD_ISSET( sd, &sip->si_use_readfds ));
 712 }
 713
 714
 715 void *
 716 ldap_new_select_info( void )
 717 {
 718         struct selectinfo       *sip;
 719
 720         if (( sip = (struct selectinfo *)LDAP_CALLOC( 1,
 721             sizeof( struct selectinfo ))) != NULL ) {
 722                 FD_ZERO( &sip->si_readfds );
 723                 FD_ZERO( &sip->si_writefds );
 724         }
 725
 726         return( (void *)sip );
 727 }
 728
 729
 730 void
 731 ldap_free_select_info( void *sip )
 732 {
 733         LDAP_FREE( sip );
 734 }
 735
 736
 737 #ifndef HAVE_POLL
 738 int ldap_int_tblsize = 0;
 739
 740 void
 741 ldap_int_ip_init( void )
 742 {
 743         int tblsize;
 744
 745 #if defined( HAVE_SYSCONF )
 746         tblsize = sysconf( _SC_OPEN_MAX );
 747 #elif defined( HAVE_GETDTABLESIZE )
 748         tblsize = getdtablesize();
 749 #else
 750         tblsize = FD_SETSIZE;
 751 #endif /* !USE_SYSCONF */
 752
 753 #ifdef FD_SETSIZE
 754         if( tblsize > FD_SETSIZE ) tblsize = FD_SETSIZE;
 755 #endif  /* FD_SETSIZE*/
 756
 757         ldap_int_tblsize = tblsize;
 758 }
 759 #endif
 760
 761
 762 int
 763 ldap_int_select( LDAP *ld, struct timeval *timeout )
 764 {
 765         struct selectinfo       *sip;
 766
 767 #ifdef NEW_LOGGING
 768         LDAP_LOG ( CONNECTION, ENTRY, "ldap_int_select\n", 0, 0, 0 );
 769 #else
 770         Debug( LDAP_DEBUG_TRACE, "ldap_int_select\n", 0, 0, 0 );
 771 #endif
 772
 773 #ifndef HAVE_POLL
 774         if ( ldap_int_tblsize == 0 ) ldap_int_ip_init();
 775 #endif
 776
 777         sip = (struct selectinfo *)ld->ld_selectinfo;
 778         sip->si_use_readfds = sip->si_readfds;
 779         sip->si_use_writefds = sip->si_writefds;
 780
 781 #ifdef HAVE_POLL
 782         assert(0);
 783         return -1;
 784 #else
 785         return( select( ldap_int_tblsize,
 786                 &sip->si_use_readfds, &sip->si_use_writefds,
 787                 NULL, timeout ));
 788 #endif
 789 }