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