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