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

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