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

   1 /*
   2  * Copyright 1998-1999 The OpenLDAP Foundation, All Rights Reserved.
   3  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
   4  */
   5 /*  Portions
   6  *  Copyright (c) 1995 Regents of the University of Michigan.
   7  *  All rights reserved.
   8  *
   9  *  request.c - sending of ldap requests; handling of referrals
  10  */
  11
  12 #include "portable.h"
  13
  14 #include <stdio.h>
  15 #include <stdlib.h>
  16
  17 #include <ac/errno.h>
  18 #include <ac/socket.h>
  19 #include <ac/string.h>
  20 #include <ac/time.h>
  21 #include <ac/unistd.h>
  22
  23 #include "ldap-int.h"
  24 #include "lber.h"
  25
  26 #if defined( LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS ) || defined( LDAP_API_FEATURE_X_OPENLDAP_V2_DNS )
  27 static LDAPConn *find_connection LDAP_P(( LDAP *ld, LDAPServer *srv, int any ));
  28 static void use_connection LDAP_P(( LDAP *ld, LDAPConn *lc ));
  29 static void free_servers LDAP_P(( LDAPServer *srvlist ));
  30 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS || LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
  31
  32
  33 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_DNS
  34 static LDAPServer *dn2servers LDAP_P(( LDAP *ld, char *dn ));
  35 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
  36
  37 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS
  38 static BerElement *re_encode_request LDAP_P(( LDAP *ld, BerElement *origber,
  39     int msgid, char **dnp ));
  40 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS */
  41
  42
  43 BerElement *
  44 ldap_alloc_ber_with_options( LDAP *ld )
  45 {
  46         BerElement      *ber;
  47
  48     if (( ber = ber_alloc_t( ld->ld_lberoptions )) == NULLBER ) {
  49                 ld->ld_errno = LDAP_NO_MEMORY;
  50 #ifdef STR_TRANSLATION
  51         } else {
  52                 ldap_set_ber_options( ld, ber );
  53 #endif /* STR_TRANSLATION */
  54         }
  55
  56         return( ber );
  57 }
  58
  59
  60 void
  61 ldap_set_ber_options( LDAP *ld, BerElement *ber )
  62 {
  63         ber->ber_options = ld->ld_lberoptions;
  64 #ifdef STR_TRANSLATION
  65         if (( ld->ld_lberoptions & LBER_TRANSLATE_STRINGS ) != 0 ) {
  66                 ber_set_string_translators( ber,
  67                     ld->ld_lber_encode_translate_proc,
  68                     ld->ld_lber_decode_translate_proc );
  69         }
  70 #endif /* STR_TRANSLATION */
  71 }
  72
  73
  74 int
  75 ldap_send_initial_request(
  76         LDAP *ld,
  77         unsigned long msgtype,
  78         const char *dn,
  79         BerElement *ber )
  80 {
  81 #if defined( LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS ) || defined( LDAP_API_FEATURE_X_OPENLDAP_V2_DNS )
  82         LDAPServer      *servers, *srv;
  83 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS || LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
  84
  85         Debug( LDAP_DEBUG_TRACE, "ldap_send_initial_request\n", 0, 0, 0 );
  86
  87         if ( ! ber_pvt_sb_in_use(&ld->ld_sb ) ) {
  88                 /* not connected yet */
  89
  90 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS
  91                 if (( srv = (LDAPServer *)calloc( 1, sizeof( LDAPServer ))) ==
  92                     NULL || ( ld->ld_defhost != NULL && ( srv->lsrv_host =
  93                     strdup( ld->ld_defhost )) == NULL ))
  94                 {
  95                         if (srv != NULL) free( srv );
  96                         ber_free( ber, 1 );
  97                         ld->ld_errno = LDAP_NO_MEMORY;
  98                         return( -1 );
  99                 }
 100
 101                 srv->lsrv_port = ld->ld_defport;
 102
 103                 if (( ld->ld_defconn = ldap_new_connection( ld, &srv, 1,1,0 ))
 104                         == NULL )
 105                 {
 106                         if ( ld->ld_defhost != NULL ) free( srv->lsrv_host );
 107                         free( (char *)srv );
 108                         ber_free( ber, 1 );
 109                         ld->ld_errno = LDAP_SERVER_DOWN;
 110                         return( -1 );
 111                 }
 112                 ++ld->ld_defconn->lconn_refcnt; /* so it never gets closed/freed */
 113
 114 #else /* LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS */
 115                 if ( open_ldap_connection( ld, &ld->ld_sb, ld->ld_defhost,
 116                     ld->ld_defport, &ld->ld_host, 0 ) < 0 )
 117                 {
 118                         ldap_ld_free( ld, 0 );
 119                         ber_free( ber, 1 );
 120                         ld->ld_errno = LDAP_SERVER_DOWN;
 121                         return( -1 );
 122                 }
 123 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS */
 124
 125                 Debug( LDAP_DEBUG_TRACE,
 126                         "ldap_delayed_open successful, ld_host is %s\n",
 127                         ( ld->ld_host == NULL ) ? "(null)" : ld->ld_host, 0, 0 );
 128         }
 129
 130
 131 #if !defined( LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS ) && !defined( LDAP_API_FEATURE_X_OPENLDAP_V2_DNS )
 132
 133         if ( ber_flush( &ld->ld_sb, ber, 1 ) != 0 ) {
 134                 ld->ld_errno = LDAP_SERVER_DOWN;
 135                 return( -1 );
 136         }
 137
 138         ld->ld_errno = LDAP_SUCCESS;
 139         return( ld->ld_msgid );
 140 #else /* !LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS && !LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
 141
 142 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_DNS
 143         if (( LDAP_BOOL_GET(&ld->ld_options, LDAP_BOOL_DNS ) == LDAP_OPT_ON )
 144                 && ldap_is_dns_dn( dn ) )
 145         {
 146                 if (( servers = dn2servers( ld, dn )) == NULL ) {
 147                         ber_free( ber, 1 );
 148                         return( -1 );
 149                 }
 150
 151 #ifdef LDAP_DEBUG
 152                 if ( ldap_debug & LDAP_DEBUG_TRACE ) {
 153                         LDAPServer      *srv;
 154
 155                         for ( srv = servers; srv != NULL;
 156                             srv = srv->lsrv_next ) {
 157                                 fprintf( stderr,
 158                                     "LDAP server %s:  dn %s, port %d\n",
 159                                     srv->lsrv_host, ( srv->lsrv_dn == NULL ) ?
 160                                     "(default)" : srv->lsrv_dn,
 161                                     srv->lsrv_port );
 162                         }
 163                 }
 164 #endif /* LDAP_DEBUG */
 165         } else {
 166 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
 167                 /*
 168                  * use of DNS is turned off or this is an X.500 DN...
 169                  * use our default connection
 170                  */
 171                 servers = NULL;
 172 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_DNS
 173         }
 174 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
 175
 176         return( ldap_send_server_request( ld, ber, ld->ld_msgid, NULL, servers,
 177             NULL, 0 ));
 178 #endif /* !LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS && !LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
 179 }
 180
 181
 182
 183 #if defined( LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS ) || defined( LDAP_API_FEATURE_X_OPENLDAP_V2_DNS )
 184 int
 185 ldap_send_server_request( LDAP *ld, BerElement *ber, int msgid, LDAPRequest
 186         *parentreq, LDAPServer *srvlist, LDAPConn *lc, int bind )
 187 {
 188         LDAPRequest     *lr;
 189         int incparent;
 190
 191         Debug( LDAP_DEBUG_TRACE, "ldap_send_server_request\n", 0, 0, 0 );
 192
 193         incparent = 0;
 194         ld->ld_errno = LDAP_SUCCESS;    /* optimistic */
 195
 196         if ( lc == NULL ) {
 197                 if ( srvlist == NULL ) {
 198                         lc = ld->ld_defconn;
 199                 } else {
 200                         if (( lc = find_connection( ld, srvlist, 1 )) ==
 201                             NULL ) {
 202                                 if ( bind && (parentreq != NULL) ) {
 203                                         /* Remember the bind in the parent */
 204                                         incparent = 1;
 205                                         ++parentreq->lr_outrefcnt;
 206                                 }
 207                                 lc = ldap_new_connection( ld, &srvlist, 0, 1, bind );
 208                         }
 209                         free_servers( srvlist );
 210                 }
 211         }
 212
 213         if ( lc == NULL || lc->lconn_status != LDAP_CONNST_CONNECTED ) {
 214                 ber_free( ber, 1 );
 215                 if ( ld->ld_errno == LDAP_SUCCESS ) {
 216                         ld->ld_errno = LDAP_SERVER_DOWN;
 217                 }
 218                 if ( incparent ) {
 219                         /* Forget about the bind */
 220                         --parentreq->lr_outrefcnt;
 221                 }
 222                 return( -1 );
 223         }
 224
 225         use_connection( ld, lc );
 226         if (( lr = (LDAPRequest *)calloc( 1, sizeof( LDAPRequest ))) ==
 227             NULL ) {
 228                 ld->ld_errno = LDAP_NO_MEMORY;
 229                 ldap_free_connection( ld, lc, 0, 0 );
 230                 ber_free( ber, 1 );
 231                 if ( incparent ) {
 232                         /* Forget about the bind */
 233                         --parentreq->lr_outrefcnt;
 234                 }
 235                 return( -1 );
 236         }
 237         lr->lr_msgid = msgid;
 238         lr->lr_status = LDAP_REQST_INPROGRESS;
 239         lr->lr_res_errno = LDAP_SUCCESS;        /* optimistic */
 240         lr->lr_ber = ber;
 241         lr->lr_conn = lc;
 242         if ( parentreq != NULL ) {      /* sub-request */
 243                 if ( !incparent ) {
 244                         /* Increment if we didn't do it before the bind */
 245                         ++parentreq->lr_outrefcnt;
 246                 }
 247                 lr->lr_origid = parentreq->lr_origid;
 248                 lr->lr_parentcnt = parentreq->lr_parentcnt + 1;
 249                 lr->lr_parent = parentreq;
 250                 lr->lr_refnext = parentreq->lr_refnext;
 251                 parentreq->lr_refnext = lr;
 252         } else {                        /* original request */
 253                 lr->lr_origid = lr->lr_msgid;
 254         }
 255
 256         if (( lr->lr_next = ld->ld_requests ) != NULL ) {
 257                 lr->lr_next->lr_prev = lr;
 258         }
 259         ld->ld_requests = lr;
 260         lr->lr_prev = NULL;
 261
 262         if ( ber_flush( lc->lconn_sb, ber, 0 ) != 0 ) {
 263 #ifdef notyet
 264                 if ( errno == EWOULDBLOCK ) {
 265                         /* need to continue write later */
 266                         lr->lr_status = LDAP_REQST_WRITING;
 267                         ldap_mark_select_write( ld, lc->lconn_sb );
 268                 } else {
 269 #else /* notyet */
 270                         ld->ld_errno = LDAP_SERVER_DOWN;
 271                         ldap_free_request( ld, lr );
 272                         ldap_free_connection( ld, lc, 0, 0 );
 273                         return( -1 );
 274 #endif /* notyet */
 275 #ifdef notyet
 276                 }
 277 #endif /* notyet */
 278         } else {
 279                 if ( parentreq == NULL ) {
 280                         ber->ber_end = ber->ber_ptr;
 281                         ber->ber_ptr = ber->ber_buf;
 282                 }
 283
 284                 /* sent -- waiting for a response */
 285                 ldap_mark_select_read( ld, lc->lconn_sb );
 286         }
 287
 288         ld->ld_errno = LDAP_SUCCESS;
 289         return( msgid );
 290 }
 291
 292
 293 LDAPConn *
 294 ldap_new_connection( LDAP *ld, LDAPServer **srvlistp, int use_ldsb,
 295         int connect, int bind )
 296 {
 297         LDAPConn        *lc;
 298         LDAPServer      *prevsrv, *srv;
 299         Sockbuf         *sb;
 300
 301         /*
 302          * make a new LDAP server connection
 303          * XXX open connection synchronously for now
 304          */
 305         if (( lc = (LDAPConn *)calloc( 1, sizeof( LDAPConn ))) == NULL ||
 306             ( !use_ldsb && ( (sb = ber_sockbuf_alloc()) == NULL ))) {
 307                 if ( lc != NULL ) {
 308                         free( (char *)lc );
 309                 }
 310                 ld->ld_errno = LDAP_NO_MEMORY;
 311                 return( NULL );
 312         }
 313
 314         lc->lconn_sb = ( use_ldsb ) ? &ld->ld_sb : sb;
 315
 316         if ( connect ) {
 317                 prevsrv = NULL;
 318
 319                 for ( srv = *srvlistp; srv != NULL; srv = srv->lsrv_next ) {
 320                         if ( open_ldap_connection( ld, lc->lconn_sb,
 321                             srv->lsrv_host, srv->lsrv_port,
 322                             &lc->lconn_krbinstance, 0 ) != -1 ) {
 323                                 break;
 324                         }
 325                         prevsrv = srv;
 326                 }
 327
 328                 if ( srv == NULL ) {
 329                     if ( !use_ldsb ) {
 330                         ber_sockbuf_free( lc->lconn_sb );
 331                     }
 332                     free( (char *)lc );
 333                     ld->ld_errno = LDAP_SERVER_DOWN;
 334                     return( NULL );
 335                 }
 336
 337                 if ( prevsrv == NULL ) {
 338                     *srvlistp = srv->lsrv_next;
 339                 } else {
 340                     prevsrv->lsrv_next = srv->lsrv_next;
 341                 }
 342                 lc->lconn_server = srv;
 343         }
 344
 345         lc->lconn_status = LDAP_CONNST_CONNECTED;
 346         lc->lconn_next = ld->ld_conns;
 347         ld->ld_conns = lc;
 348
 349         /*
 350          * XXX for now, we always do a synchronous bind.  This will have
 351          * to change in the long run...
 352          */
 353         if ( bind ) {
 354                 int             err, freepasswd, authmethod;
 355                 char            *binddn, *passwd;
 356                 LDAPConn        *savedefconn;
 357
 358                 freepasswd = err = 0;
 359
 360                 if ( ld->ld_rebindproc == 0 ) {
 361                         binddn = passwd = "";
 362                         authmethod = LDAP_AUTH_SIMPLE;
 363                 } else {
 364                         if (( err = (*ld->ld_rebindproc)( ld, &binddn, &passwd,
 365                             &authmethod, 0 )) == LDAP_SUCCESS ) {
 366                                 freepasswd = 1;
 367                         } else {
 368                                 ld->ld_errno = err;
 369                                 err = -1;
 370                         }
 371                 }
 372
 373
 374                 if ( err == 0 ) {
 375                         savedefconn = ld->ld_defconn;
 376                         ld->ld_defconn = lc;
 377                         ++lc->lconn_refcnt;     /* avoid premature free */
 378
 379                         if ( ldap_bind_s( ld, binddn, passwd, authmethod ) !=
 380                             LDAP_SUCCESS ) {
 381                                 err = -1;
 382                         }
 383                         --lc->lconn_refcnt;
 384                         ld->ld_defconn = savedefconn;
 385                 }
 386
 387                 if ( freepasswd ) {
 388                         (*ld->ld_rebindproc)( ld, &binddn, &passwd,
 389                                 &authmethod, 1 );
 390                 }
 391
 392                 if ( err != 0 ) {
 393                         ldap_free_connection( ld, lc, 1, 0 );
 394                         lc = NULL;
 395                 }
 396         }
 397
 398         return( lc );
 399 }
 400
 401
 402 static LDAPConn *
 403 find_connection( LDAP *ld, LDAPServer *srv, int any )
 404 /*
 405  * return an existing connection (if any) to the server srv
 406  * if "any" is non-zero, check for any server in the "srv" chain
 407  */
 408 {
 409         LDAPConn        *lc;
 410         LDAPServer      *ls;
 411
 412         for ( lc = ld->ld_conns; lc != NULL; lc = lc->lconn_next ) {
 413                 for ( ls = srv; ls != NULL; ls = ls->lsrv_next ) {
 414                         if ( lc->lconn_server->lsrv_host != NULL &&
 415                             ls->lsrv_host != NULL && strcasecmp(
 416                             ls->lsrv_host, lc->lconn_server->lsrv_host ) == 0
 417                             && ls->lsrv_port == lc->lconn_server->lsrv_port ) {
 418                                 return( lc );
 419                         }
 420                         if ( !any ) {
 421                                 break;
 422                         }
 423                 }
 424         }
 425
 426         return( NULL );
 427 }
 428
 429
 430
 431 static void
 432 use_connection( LDAP *ld, LDAPConn *lc )
 433 {
 434         ++lc->lconn_refcnt;
 435         lc->lconn_lastused = time( 0 );
 436 }
 437
 438
 439 void
 440 ldap_free_connection( LDAP *ld, LDAPConn *lc, int force, int unbind )
 441 {
 442         LDAPConn        *tmplc, *prevlc;
 443
 444         Debug( LDAP_DEBUG_TRACE, "ldap_free_connection\n", 0, 0, 0 );
 445
 446         if ( force || --lc->lconn_refcnt <= 0 ) {
 447                 if ( lc->lconn_status == LDAP_CONNST_CONNECTED ) {
 448                         ldap_mark_select_clear( ld, lc->lconn_sb );
 449                         if ( unbind ) {
 450                                 ldap_send_unbind( ld, lc->lconn_sb, NULL, NULL );
 451                         }
 452                         ldap_close_connection( lc->lconn_sb );
 453                         ber_pvt_sb_destroy( lc->lconn_sb );
 454                         ber_clear( &lc->lconn_ber, 1 );
 455                 }
 456                 prevlc = NULL;
 457                 for ( tmplc = ld->ld_conns; tmplc != NULL;
 458                     tmplc = tmplc->lconn_next ) {
 459                         if ( tmplc == lc ) {
 460                                 if ( prevlc == NULL ) {
 461                                     ld->ld_conns = tmplc->lconn_next;
 462                                 } else {
 463                                     prevlc->lconn_next = tmplc->lconn_next;
 464                                 }
 465                                 break;
 466                         }
 467                         prevlc = tmplc;
 468                 }
 469                 free_servers( lc->lconn_server );
 470                 if ( lc->lconn_krbinstance != NULL ) {
 471                         free( lc->lconn_krbinstance );
 472                 }
 473                 if ( lc->lconn_sb != &ld->ld_sb ) {
 474                         ber_sockbuf_free( lc->lconn_sb );
 475                 }
 476                 free( lc );
 477                 Debug( LDAP_DEBUG_TRACE, "ldap_free_connection: actually freed\n",
 478                     0, 0, 0 );
 479         } else {
 480                 lc->lconn_lastused = time( 0 );
 481                 Debug( LDAP_DEBUG_TRACE, "ldap_free_connection: refcnt %d\n",
 482                     lc->lconn_refcnt, 0, 0 );
 483         }
 484 }
 485
 486
 487 #ifdef LDAP_DEBUG
 488 void
 489 ldap_dump_connection( LDAP *ld, LDAPConn *lconns, int all )
 490 {
 491         LDAPConn        *lc;
 492         char            timebuf[32];
 493
 494         fprintf( stderr, "** Connection%s:\n", all ? "s" : "" );
 495         for ( lc = lconns; lc != NULL; lc = lc->lconn_next ) {
 496                 if ( lc->lconn_server != NULL ) {
 497                         fprintf( stderr, "* host: %s  port: %d%s\n",
 498                             ( lc->lconn_server->lsrv_host == NULL ) ? "(null)"
 499                             : lc->lconn_server->lsrv_host,
 500                             lc->lconn_server->lsrv_port, ( lc->lconn_sb ==
 501                             &ld->ld_sb ) ? "  (default)" : "" );
 502                 }
 503                 fprintf( stderr, "  refcnt: %d  status: %s\n", lc->lconn_refcnt,
 504                     ( lc->lconn_status == LDAP_CONNST_NEEDSOCKET ) ?
 505                     "NeedSocket" : ( lc->lconn_status ==
 506                     LDAP_CONNST_CONNECTING ) ? "Connecting" : "Connected" );
 507                 fprintf( stderr, "  last used: %s\n",
 508                     ldap_pvt_ctime( &lc->lconn_lastused, timebuf ));
 509                 if ( !all ) {
 510                         break;
 511                 }
 512         }
 513 }
 514
 515
 516 void
 517 ldap_dump_requests_and_responses( LDAP *ld )
 518 {
 519         LDAPRequest     *lr;
 520         LDAPMessage     *lm, *l;
 521
 522         fprintf( stderr, "** Outstanding Requests:\n" );
 523         if (( lr = ld->ld_requests ) == NULL ) {
 524                 fprintf( stderr, "   Empty\n" );
 525         }
 526         for ( ; lr != NULL; lr = lr->lr_next ) {
 527             fprintf( stderr, " * msgid %d,  origid %d, status %s\n",
 528                 lr->lr_msgid, lr->lr_origid, ( lr->lr_status ==
 529                 LDAP_REQST_INPROGRESS ) ? "InProgress" :
 530                 ( lr->lr_status == LDAP_REQST_CHASINGREFS ) ? "ChasingRefs" :
 531                 ( lr->lr_status == LDAP_REQST_NOTCONNECTED ) ? "NotConnected" :
 532                 "Writing" );
 533             fprintf( stderr, "   outstanding referrals %d, parent count %d\n",
 534                     lr->lr_outrefcnt, lr->lr_parentcnt );
 535         }
 536
 537         fprintf( stderr, "** Response Queue:\n" );
 538         if (( lm = ld->ld_responses ) == NULLMSG ) {
 539                 fprintf( stderr, "   Empty\n" );
 540         }
 541         for ( ; lm != NULLMSG; lm = lm->lm_next ) {
 542                 fprintf( stderr, " * msgid %d,  type %d\n",
 543                     lm->lm_msgid, lm->lm_msgtype );
 544                 if (( l = lm->lm_chain ) != NULL ) {
 545                         fprintf( stderr, "   chained responses:\n" );
 546                         for ( ; l != NULLMSG; l = l->lm_chain ) {
 547                                 fprintf( stderr,
 548                                     "  * msgid %d,  type %d\n",
 549                                     l->lm_msgid, l->lm_msgtype );
 550                         }
 551                 }
 552         }
 553 }
 554 #endif /* LDAP_DEBUG */
 555
 556
 557 void
 558 ldap_free_request( LDAP *ld, LDAPRequest *lr )
 559 {
 560         LDAPRequest     *tmplr, *nextlr;
 561
 562         Debug( LDAP_DEBUG_TRACE, "ldap_free_request (origid %d, msgid %d)\n",
 563                 lr->lr_origid, lr->lr_msgid, 0 );
 564
 565         if ( lr->lr_parent != NULL ) {
 566                 --lr->lr_parent->lr_outrefcnt;
 567         } else {
 568                 /* free all referrals (child requests) */
 569                 for ( tmplr = lr->lr_refnext; tmplr != NULL; tmplr = nextlr ) {
 570                         nextlr = tmplr->lr_refnext;
 571                         ldap_free_request( ld, tmplr );
 572                 }
 573         }
 574
 575         if ( lr->lr_prev == NULL ) {
 576                 ld->ld_requests = lr->lr_next;
 577         } else {
 578                 lr->lr_prev->lr_next = lr->lr_next;
 579         }
 580
 581         if ( lr->lr_next != NULL ) {
 582                 lr->lr_next->lr_prev = lr->lr_prev;
 583         }
 584
 585         if ( lr->lr_ber != NULL ) {
 586                 ber_free( lr->lr_ber, 1 );
 587         }
 588
 589         if ( lr->lr_res_error != NULL ) {
 590                 free( lr->lr_res_error );
 591         }
 592
 593         if ( lr->lr_res_matched != NULL ) {
 594                 free( lr->lr_res_matched );
 595         }
 596
 597         free( lr );
 598 }
 599
 600
 601 static void
 602 free_servers( LDAPServer *srvlist )
 603 {
 604     LDAPServer  *nextsrv;
 605
 606     while ( srvlist != NULL ) {
 607         nextsrv = srvlist->lsrv_next;
 608         if ( srvlist->lsrv_dn != NULL ) {
 609                 free( srvlist->lsrv_dn );
 610         }
 611         if ( srvlist->lsrv_host != NULL ) {
 612                 free( srvlist->lsrv_host );
 613         }
 614         free( srvlist );
 615         srvlist = nextsrv;
 616     }
 617 }
 618 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS || LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
 619
 620
 621 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS
 622 /*
 623  * XXX merging of errors in this routine needs to be improved
 624  */
 625 int
 626 ldap_chase_referrals( LDAP *ld, LDAPRequest *lr, char **errstrp, int *hadrefp )
 627 {
 628         int             rc, count, len, newdn;
 629 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_DNS
 630         int             ldapref;
 631 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
 632         char            *p, *ports, *ref, *tmpref, *refdn, *unfollowed;
 633         LDAPRequest     *origreq;
 634         LDAPServer      *srv;
 635         BerElement      *ber;
 636
 637         Debug( LDAP_DEBUG_TRACE, "ldap_chase_referrals\n", 0, 0, 0 );
 638
 639         ld->ld_errno = LDAP_SUCCESS;    /* optimistic */
 640         *hadrefp = 0;
 641
 642         if ( *errstrp == NULL ) {
 643                 return( 0 );
 644         }
 645
 646         len = strlen( *errstrp );
 647         for ( p = *errstrp; len >= LDAP_REF_STR_LEN; ++p, --len ) {
 648                 if (( *p == 'R' || *p == 'r' ) && strncasecmp( p,
 649                     LDAP_REF_STR, LDAP_REF_STR_LEN ) == 0 ) {
 650                         *p = '\0';
 651                         p += LDAP_REF_STR_LEN;
 652                         break;
 653                 }
 654         }
 655
 656         if ( len < LDAP_REF_STR_LEN ) {
 657                 return( 0 );
 658         }
 659
 660         if ( lr->lr_parentcnt >= ld->ld_refhoplimit ) {
 661                 Debug( LDAP_DEBUG_ANY,
 662                     "more than %d referral hops (dropping)\n",
 663                     ld->ld_refhoplimit, 0, 0 );
 664                     /* XXX report as error in ld->ld_errno? */
 665                     return( 0 );
 666         }
 667
 668         /* find original request */
 669         for ( origreq = lr; origreq->lr_parent != NULL;
 670              origreq = origreq->lr_parent ) {
 671                 ;
 672         }
 673
 674         unfollowed = NULL;
 675         rc = count = 0;
 676
 677         /* parse out & follow referrals */
 678         for ( ref = p; rc == 0 && ref != NULL; ref = p ) {
 679 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_DNS
 680                 ldapref = 0;
 681 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
 682
 683                 if (( p = strchr( ref, '\n' )) != NULL ) {
 684                         *p++ = '\0';
 685                 } else {
 686                         p = NULL;
 687                 }
 688
 689                 len = strlen( ref );
 690                 if ( len > LDAP_LDAP_REF_STR_LEN && strncasecmp( ref,
 691                     LDAP_LDAP_REF_STR, LDAP_LDAP_REF_STR_LEN ) == 0 ) {
 692                         Debug( LDAP_DEBUG_TRACE,
 693                             "chasing LDAP referral: <%s>\n", ref, 0, 0 );
 694 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_DNS
 695                         ldapref = 1;
 696 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
 697                         tmpref = ref + LDAP_LDAP_REF_STR_LEN;
 698 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_DNS
 699                 } else if ( len > LDAP_DX_REF_STR_LEN && strncasecmp( ref,
 700                     LDAP_DX_REF_STR, LDAP_DX_REF_STR_LEN ) == 0 ) {
 701                         Debug( LDAP_DEBUG_TRACE,
 702                             "chasing DX referral: <%s>\n", ref, 0, 0 );
 703                         tmpref = ref + LDAP_DX_REF_STR_LEN;
 704 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
 705                 } else {
 706                         Debug( LDAP_DEBUG_TRACE,
 707                             "ignoring unknown referral <%s>\n", ref, 0, 0 );
 708                         rc = ldap_append_referral( ld, &unfollowed, ref );
 709                         *hadrefp = 1;
 710                         continue;
 711                 }
 712
 713                 *hadrefp = 1;
 714                 if (( refdn = strchr( tmpref, '/' )) != NULL ) {
 715                         *refdn++ = '\0';
 716                         newdn = 1;
 717                 } else {
 718                         newdn = 0;
 719                 }
 720
 721                 if (( ber = re_encode_request( ld, origreq->lr_ber,
 722                     ++ld->ld_msgid, &refdn )) == NULL ) {
 723                         return( -1 );
 724                 }
 725
 726 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_DNS
 727                 if ( ldapref ) {
 728 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
 729                         if (( srv = (LDAPServer *)calloc( 1,
 730                             sizeof( LDAPServer ))) == NULL ) {
 731                                 ber_free( ber, 1 );
 732                                 ld->ld_errno = LDAP_NO_MEMORY;
 733                                 return( -1 );
 734                         }
 735
 736                         if (( srv->lsrv_host = strdup( tmpref )) == NULL ) {
 737                                 free( (char *)srv );
 738                                 ber_free( ber, 1 );
 739                                 ld->ld_errno = LDAP_NO_MEMORY;
 740                                 return( -1 );
 741                         }
 742
 743                         if (( ports = strchr( srv->lsrv_host, ':' )) != NULL ) {
 744                                 *ports++ = '\0';
 745                                 srv->lsrv_port = atoi( ports );
 746                         } else {
 747                                 srv->lsrv_port = openldap_ldap_global_options.ldo_defport;
 748                         }
 749 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_DNS
 750                 } else {
 751                         srv = dn2servers( ld, tmpref );
 752                 }
 753 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */
 754
 755                 if ( srv != NULL && ldap_send_server_request( ld, ber, ld->ld_msgid,
 756                     lr, srv, NULL, 1 ) >= 0 ) {
 757                         ++count;
 758                 } else {
 759                         Debug( LDAP_DEBUG_ANY,
 760                             "Unable to chase referral (%s)\n",
 761                             ldap_err2string( ld->ld_errno ), 0, 0 );
 762                         rc = ldap_append_referral( ld, &unfollowed, ref );
 763                 }
 764
 765                 if ( !newdn && refdn != NULL ) {
 766                         free( refdn );
 767                 }
 768         }
 769
 770         free( *errstrp );
 771         *errstrp = unfollowed;
 772
 773         return(( rc == 0 ) ? count : rc );
 774 }
 775
 776
 777 int
 778 ldap_append_referral( LDAP *ld, char **referralsp, char *s )
 779 {
 780         int     first;
 781
 782         if ( *referralsp == NULL ) {
 783                 first = 1;
 784                 *referralsp = (char *)malloc( strlen( s ) + LDAP_REF_STR_LEN
 785                     + 1 );
 786         } else {
 787                 first = 0;
 788                 *referralsp = (char *)realloc( *referralsp,
 789                     strlen( *referralsp ) + strlen( s ) + 2 );
 790         }
 791
 792         if ( *referralsp == NULL ) {
 793                 ld->ld_errno = LDAP_NO_MEMORY;
 794                 return( -1 );
 795         }
 796
 797         if ( first ) {
 798                 strcpy( *referralsp, LDAP_REF_STR );
 799         } else {
 800                 strcat( *referralsp, "\n" );
 801         }
 802         strcat( *referralsp, s );
 803
 804         return( 0 );
 805 }
 806
 807
 808
 809 static BerElement *
 810 re_encode_request( LDAP *ld, BerElement *origber, int msgid, char **dnp )
 811 {
 812 /*
 813  * XXX this routine knows way too much about how the lber library works!
 814  */
 815         unsigned long   along, tag;
 816         long            ver;
 817         int             rc;
 818         BerElement      tmpber, *ber;
 819         char            *orig_dn;
 820
 821         Debug( LDAP_DEBUG_TRACE,
 822             "re_encode_request: new msgid %d, new dn <%s>\n",
 823             msgid, ( *dnp == NULL ) ? "NONE" : *dnp, 0 );
 824
 825         tmpber = *origber;
 826
 827         /*
 828          * all LDAP requests are sequences that start with a message id,
 829          * followed by a sequence that is tagged with the operation code
 830          */
 831         if ( ber_scanf( &tmpber, "{i", &along ) != LDAP_TAG_MSGID ||
 832             ( tag = ber_skip_tag( &tmpber, &along )) == LBER_DEFAULT ) {
 833                 ld->ld_errno = LDAP_DECODING_ERROR;
 834                 return( NULL );
 835         }
 836
 837         if (( ber = ldap_alloc_ber_with_options( ld )) == NULLBER ) {
 838                 return( NULL );
 839         }
 840
 841         /* bind requests have a version number before the DN & other stuff */
 842         if ( tag == LDAP_REQ_BIND && ber_get_int( &tmpber, (long *)&ver ) ==
 843             LBER_DEFAULT ) {
 844                 ld->ld_errno = LDAP_DECODING_ERROR;
 845                 ber_free( ber, 1 );
 846                 return( NULL );
 847         }
 848
 849         /* the rest of the request is the DN followed by other stuff */
 850         if ( ber_get_stringa( &tmpber, &orig_dn ) == LBER_DEFAULT ) {
 851                 ber_free( ber, 1 );
 852                 return( NULL );
 853         }
 854
 855         if ( *dnp == NULL ) {
 856                 *dnp = orig_dn;
 857         } else {
 858                 free( orig_dn );
 859         }
 860
 861         if ( tag == LDAP_REQ_BIND ) {
 862                 rc = ber_printf( ber, "{it{is", msgid, tag, ver, *dnp );
 863         } else {
 864                 rc = ber_printf( ber, "{it{s", msgid, tag, *dnp );
 865         }
 866
 867         if ( rc == -1 ) {
 868                 ber_free( ber, 1 );
 869                 return( NULL );
 870         }
 871
 872         if ( ber_write( ber, tmpber.ber_ptr, ( tmpber.ber_end -
 873             tmpber.ber_ptr ), 0 ) != ( tmpber.ber_end - tmpber.ber_ptr ) ||
 874             ber_printf( ber, "}}" ) == -1 ) {
 875                 ld->ld_errno = LDAP_ENCODING_ERROR;
 876                 ber_free( ber, 1 );
 877                 return( NULL );
 878         }
 879
 880 #ifdef LDAP_DEBUG
 881         if ( ldap_debug & LDAP_DEBUG_PACKETS ) {
 882                 Debug( LDAP_DEBUG_ANY, "re_encode_request new request is:\n",
 883                     0, 0, 0 );
 884                 ber_log_dump( LDAP_DEBUG_BER, ldap_debug, ber, 0 );
 885         }
 886 #endif /* LDAP_DEBUG */
 887
 888         return( ber );
 889 }
 890
 891
 892 LDAPRequest *
 893 ldap_find_request_by_msgid( LDAP *ld, int msgid )
 894 {
 895         LDAPRequest     *lr;
 896
 897         for ( lr = ld->ld_requests; lr != NULL; lr = lr->lr_next ) {
 898                 if ( msgid == lr->lr_msgid ) {
 899                         break;
 900                 }
 901         }
 902
 903         return( lr );
 904 }
 905 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS */
 906
 907
 908 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_DNS
 909 static LDAPServer *
 910 dn2servers( LDAP *ld, char *dn )        /* dn can also be a domain.... */
 911 {
 912         char            *p, *domain, *host, *server_dn, **dxs;
 913         int             i, port;
 914         LDAPServer      *srvlist, *prevsrv, *srv;
 915
 916         if (( domain = strrchr( dn, '@' )) != NULL ) {
 917                 ++domain;
 918         } else {
 919                 domain = dn;
 920         }
 921
 922         if (( dxs = ldap_getdxbyname( domain )) == NULL ) {
 923                 ld->ld_errno = LDAP_NO_MEMORY;
 924                 return( NULL );
 925         }
 926
 927         srvlist = NULL;
 928
 929         for ( i = 0; dxs[ i ] != NULL; ++i ) {
 930                 port = openldap_ldap_global_options.ldo_defport;
 931                 server_dn = NULL;
 932                 if ( strchr( dxs[ i ], ':' ) == NULL ) {
 933                         host = dxs[ i ];
 934                 } else if ( strlen( dxs[ i ] ) >= 7 &&
 935                     strncmp( dxs[ i ], "ldap://", 7 ) == 0 ) {
 936                         host = dxs[ i ] + 7;
 937                         if (( p = strchr( host, ':' )) == NULL ) {
 938                                 p = host;
 939                         } else {
 940                                 *p++ = '\0';
 941                                 port = atoi( p );
 942                         }
 943                         if (( p = strchr( p, '/' )) != NULL ) {
 944                                 server_dn = ++p;
 945                                 if ( *server_dn == '\0' ) {
 946                                         server_dn = NULL;
 947                                 }
 948                         }
 949                 } else {
 950                         host = NULL;
 951                 }
 952
 953                 if ( host != NULL ) {   /* found a server we can use */
 954                         if (( srv = (LDAPServer *)calloc( 1,
 955                             sizeof( LDAPServer ))) == NULL ) {
 956                                 free_servers( srvlist );
 957                                 srvlist = NULL;
 958                                 break;          /* exit loop & return */
 959                         }
 960
 961                         /* add to end of list of servers */
 962                         if ( srvlist == NULL ) {
 963                                 srvlist = srv;
 964                         } else {
 965                                 prevsrv->lsrv_next = srv;
 966                         }
 967                         prevsrv = srv;
 968
 969                         /* copy in info. */
 970                         if (( srv->lsrv_host = strdup( host )) == NULL ||
 971                             ( server_dn != NULL && ( srv->lsrv_dn =
 972                             strdup( server_dn )) == NULL )) {
 973                                 free_servers( srvlist );
 974                                 srvlist = NULL;
 975                                 break;          /* exit loop & return */
 976                         }
 977                         srv->lsrv_port = port;
 978                 }
 979         }
 980
 981         ldap_value_free( dxs );
 982
 983         if ( srvlist == NULL ) {
 984                 ld->ld_errno = LDAP_SERVER_DOWN;
 985         }
 986
 987         return( srvlist );
 988 }
 989 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */