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

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