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