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

   1 /*
   2  *  Copyright (c) 1995 Regents of the University of Michigan.
   3  *  All rights reserved.
   4  *
   5  *  request.c - sending of ldap requests; handling of referrals
   6  */
   7
   8 #ifndef lint
   9 static char copyright[] = "@(#) Copyright (c) 1995 Regents of the University of Michigan.\nAll rights reserved.\n";
  10 #endif
  11
  12 #include <stdio.h>
  13 #include <string.h>
  14 #ifdef MACOS
  15 #include <stdlib.h>
  16 #include <time.h>
  17 #include "macos.h"
  18 #else /* MACOS */
  19 #if defined( DOS ) || defined( _WIN32 )
  20 #include "msdos.h"
  21 #include <time.h>
  22 #include <stdlib.h>
  23 #ifdef PCNFS
  24 #include <tklib.h>
  25 #include <tk_errno.h>
  26 #include <bios.h>
  27 #endif /* PCNFS */
  28 #ifdef NCSA
  29 #include "externs.h"
  30 #endif /* NCSA */
  31 #else /* DOS */
  32 #include <sys/time.h>
  33 #include <sys/types.h>
  34 #include <sys/socket.h>
  35 #include <errno.h>
  36 #ifdef _AIX
  37 #include <sys/select.h>
  38 #endif /* _AIX */
  39 #include "portable.h"
  40 #endif /* DOS */
  41 #endif /* MACOS */
  42 #ifdef VMS
  43 #include "ucx_select.h"
  44 #endif
  45 #include "lber.h"
  46 #include "ldap.h"
  47 #include "ldap-int.h"
  48
  49 #ifdef USE_SYSCONF
  50 #include <unistd.h>
  51 #endif /* USE_SYSCONF */
  52
  53
  54 #if defined( LDAP_REFERRALS ) || defined( LDAP_DNS )
  55 #ifdef NEEDPROTOS
  56 static LDAPConn *find_connection( LDAP *ld, LDAPServer *srv, int any );
  57 static void use_connection( LDAP *ld, LDAPConn *lc );
  58 static void free_servers( LDAPServer *srvlist );
  59 #else /* NEEDPROTOS */
  60 static LDAPConn *find_connection();
  61 static void use_connection();
  62 static void free_servers();
  63 #endif /* NEEDPROTOS */
  64 #endif /* LDAP_REFERRALS || LDAP_DNS */
  65
  66
  67 #ifdef LDAP_DNS
  68 #ifdef NEEDPROTOS
  69 static LDAPServer *dn2servers( LDAP *ld, char *dn );
  70 #else /* NEEDPROTOS */
  71 static LDAPServer *dn2servers();
  72 #endif /* NEEDPROTOS */
  73 #endif /* LDAP_DNS */
  74
  75 #ifdef LDAP_REFERRALS
  76 #ifdef NEEDPROTOS
  77 static BerElement *re_encode_request( LDAP *ld, BerElement *origber,
  78     int msgid, char **dnp );
  79 #else /* NEEDPROTOS */
  80 static BerElement *re_encode_request();
  81 #endif /* NEEDPROTOS */
  82 #endif /* LDAP_REFERRALS */
  83
  84
  85 BerElement *
  86 alloc_ber_with_options( LDAP *ld )
  87 {
  88         BerElement      *ber;
  89
  90         if (( ber = ber_alloc_t( ld->ld_lberoptions )) == NULLBER ) {
  91                 ld->ld_errno = LDAP_NO_MEMORY;
  92 #ifdef STR_TRANSLATION
  93         } else {
  94                 set_ber_options( ld, ber );
  95 #endif /* STR_TRANSLATION */
  96         }
  97
  98         return( ber );
  99 }
 100
 101
 102 void
 103 set_ber_options( LDAP *ld, BerElement *ber )
 104 {
 105         ber->ber_options = ld->ld_lberoptions;
 106 #ifdef STR_TRANSLATION
 107         if (( ld->ld_lberoptions & LBER_TRANSLATE_STRINGS ) != 0 ) {
 108                 ber_set_string_translators( ber,
 109                     ld->ld_lber_encode_translate_proc,
 110                     ld->ld_lber_decode_translate_proc );
 111         }
 112 #endif /* STR_TRANSLATION */
 113 }
 114
 115
 116 int
 117 send_initial_request( LDAP *ld, unsigned long msgtype, char *dn,
 118         BerElement *ber )
 119 {
 120 #if defined( LDAP_REFERRALS ) || defined( LDAP_DNS )
 121         LDAPServer      *servers;
 122 #endif /* LDAP_REFERRALS || LDAP_DNS */
 123
 124         Debug( LDAP_DEBUG_TRACE, "send_initial_request\n", 0, 0, 0 );
 125
 126 #if !defined( LDAP_REFERRALS ) && !defined( LDAP_DNS )
 127         if ( ber_flush( &ld->ld_sb, ber, 1 ) != 0 ) {
 128                 ld->ld_errno = LDAP_SERVER_DOWN;
 129                 return( -1 );
 130         }
 131
 132         ld->ld_errno = LDAP_SUCCESS;
 133         return( ld->ld_msgid );
 134 #else /* !LDAP_REFERRALS && !LDAP_DNS */
 135
 136 #ifdef LDAP_DNS
 137         if (( ld->ld_options & LDAP_OPT_DNS ) != 0 && ldap_is_dns_dn( dn )) {
 138                 if (( servers = dn2servers( ld, dn )) == NULL ) {
 139                         ber_free( ber, 1 );
 140                         return( -1 );
 141                 }
 142
 143 #ifdef LDAP_DEBUG
 144                 if ( ldap_debug & LDAP_DEBUG_TRACE ) {
 145                         LDAPServer      *srv;
 146
 147                         for ( srv = servers; srv != NULL;
 148                             srv = srv->lsrv_next ) {
 149                                 fprintf( stderr,
 150                                     "LDAP server %s:  dn %s, port %d\n",
 151                                     srv->lsrv_host, ( srv->lsrv_dn == NULL ) ?
 152                                     "(default)" : srv->lsrv_dn,
 153                                     srv->lsrv_port );
 154                         }
 155                 }
 156 #endif /* LDAP_DEBUG */
 157         } else {
 158 #endif /* LDAP_DNS */
 159                 /*
 160                  * use of DNS is turned off or this is an X.500 DN...
 161                  * use our default connection
 162                  */
 163                 servers = NULL;
 164 #ifdef LDAP_DNS
 165         }
 166 #endif /* LDAP_DNS */
 167
 168         return( send_server_request( ld, ber, ld->ld_msgid, NULL, servers,
 169             NULL, 0 ));
 170 #endif /* !LDAP_REFERRALS && !LDAP_DNS */
 171 }
 172
 173
 174
 175 #if defined( LDAP_REFERRALS ) || defined( LDAP_DNS )
 176 int
 177 send_server_request( LDAP *ld, BerElement *ber, int msgid, LDAPRequest
 178         *parentreq, LDAPServer *srvlist, LDAPConn *lc, int bind )
 179 {
 180         LDAPRequest     *lr;
 181
 182         Debug( LDAP_DEBUG_TRACE, "send_server_request\n", 0, 0, 0 );
 183
 184         ld->ld_errno = LDAP_SUCCESS;    /* optimistic */
 185
 186         if ( lc == NULL ) {
 187                 if ( srvlist == NULL ) {
 188                         lc = ld->ld_defconn;
 189                 } else {
 190                         if (( lc = find_connection( ld, srvlist, 1 )) ==
 191                             NULL ) {
 192                                 lc = new_connection( ld, &srvlist, 0, 1, bind );
 193                         }
 194                         free_servers( srvlist );
 195                 }
 196         }
 197
 198         if ( lc == NULL || lc->lconn_status != LDAP_CONNST_CONNECTED ) {
 199                 ber_free( ber, 1 );
 200                 if ( ld->ld_errno == LDAP_SUCCESS ) {
 201                         ld->ld_errno = LDAP_SERVER_DOWN;
 202                 }
 203                 return( -1 );
 204         }
 205
 206         use_connection( ld, lc );
 207         if (( lr = (LDAPRequest *)calloc( 1, sizeof( LDAPRequest ))) ==
 208             NULL ) {
 209                 ld->ld_errno = LDAP_NO_MEMORY;
 210                 free_connection( ld, lc, 0, 0 );
 211                 ber_free( ber, 1 );
 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                 ++parentreq->lr_outrefcnt;
 221                 lr->lr_origid = parentreq->lr_origid;
 222                 lr->lr_parentcnt = parentreq->lr_parentcnt + 1;
 223                 lr->lr_parent = parentreq;
 224                 lr->lr_refnext = parentreq->lr_refnext;
 225                 parentreq->lr_refnext = lr;
 226         } else {                        /* original request */
 227                 lr->lr_origid = lr->lr_msgid;
 228         }
 229
 230         if (( lr->lr_next = ld->ld_requests ) != NULL ) {
 231                 lr->lr_next->lr_prev = lr;
 232         }
 233         ld->ld_requests = lr;
 234         lr->lr_prev = NULL;
 235
 236         if ( ber_flush( lc->lconn_sb, ber, 0 ) != 0 ) {
 237 #ifdef notyet
 238                 extern int      errno;
 239
 240                 if ( errno == EWOULDBLOCK ) {
 241                         /* need to continue write later */
 242                         lr->lr_status = LDAP_REQST_WRITING;
 243                         mark_select_write( ld, lc->lconn_sb );
 244                 } else {
 245 #else /* notyet */
 246                         ld->ld_errno = LDAP_SERVER_DOWN;
 247                         free_request( ld, lr );
 248                         free_connection( ld, lc, 0, 0 );
 249                         return( -1 );
 250 #endif /* notyet */
 251 #ifdef notyet
 252                 }
 253 #endif /* notyet */
 254         } else {
 255                 if ( parentreq == NULL ) {
 256                         ber->ber_end = ber->ber_ptr;
 257                         ber->ber_ptr = ber->ber_buf;
 258                 }
 259
 260                 /* sent -- waiting for a response */
 261                 mark_select_read( ld, lc->lconn_sb );
 262         }
 263
 264         ld->ld_errno = LDAP_SUCCESS;
 265         return( msgid );
 266 }
 267
 268
 269 LDAPConn *
 270 new_connection( LDAP *ld, LDAPServer **srvlistp, int use_ldsb,
 271         int connect, int bind )
 272 {
 273         LDAPConn        *lc;
 274         LDAPServer      *prevsrv, *srv;
 275         Sockbuf         *sb;
 276
 277         /*
 278          * make a new LDAP server connection
 279          * XXX open connection synchronously for now
 280          */
 281         if (( lc = (LDAPConn *)calloc( 1, sizeof( LDAPConn ))) == NULL ||
 282             ( !use_ldsb && ( sb = (Sockbuf *)calloc( 1, sizeof( Sockbuf )))
 283             == NULL )) {
 284                 if ( lc != NULL ) {
 285                         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                         free( (char *)lc->lconn_sb );
 308                     }
 309                     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 == NULL ) {
 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                         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( 0 );
 413 }
 414
 415
 416 void
 417 free_connection( LDAP *ld, LDAPConn *lc, int force, int unbind )
 418 {
 419         LDAPConn        *tmplc, *prevlc;
 420
 421         Debug( LDAP_DEBUG_TRACE, "free_connection\n", 0, 0, 0 );
 422
 423         if ( force || --lc->lconn_refcnt <= 0 ) {
 424                 if ( lc->lconn_status == LDAP_CONNST_CONNECTED ) {
 425                         mark_select_clear( ld, lc->lconn_sb );
 426                         if ( unbind ) {
 427                                 send_unbind( ld, lc->lconn_sb );
 428                         }
 429                         close_connection( lc->lconn_sb );
 430                         if ( lc->lconn_sb->sb_ber.ber_buf != NULL ) {
 431                                 free( lc->lconn_sb->sb_ber.ber_buf );
 432                         }
 433                 }
 434                 prevlc = NULL;
 435                 for ( tmplc = ld->ld_conns; tmplc != NULL;
 436                     tmplc = tmplc->lconn_next ) {
 437                         if ( tmplc == lc ) {
 438                                 if ( prevlc == NULL ) {
 439                                     ld->ld_conns = tmplc->lconn_next;
 440                                 } else {
 441                                     prevlc->lconn_next = tmplc->lconn_next;
 442                                 }
 443                                 break;
 444                         }
 445                 }
 446                 free_servers( lc->lconn_server );
 447                 if ( lc->lconn_krbinstance != NULL ) {
 448                         free( lc->lconn_krbinstance );
 449                 }
 450                 if ( lc->lconn_sb != &ld->ld_sb ) {
 451                         free( (char *)lc->lconn_sb );
 452                 }
 453                 free( lc );
 454                 Debug( LDAP_DEBUG_TRACE, "free_connection: actually freed\n",
 455                     0, 0, 0 );
 456         } else {
 457                 lc->lconn_lastused = time( 0 );
 458                 Debug( LDAP_DEBUG_TRACE, "free_connection: refcnt %d\n",
 459                     lc->lconn_refcnt, 0, 0 );
 460         }
 461 }
 462
 463
 464 #ifdef LDAP_DEBUG
 465 void
 466 dump_connection( LDAP *ld, LDAPConn *lconns, int all )
 467 {
 468         LDAPConn        *lc;
 469
 470         fprintf( stderr, "** Connection%s:\n", all ? "s" : "" );
 471         for ( lc = lconns; lc != NULL; lc = lc->lconn_next ) {
 472                 if ( lc->lconn_server != NULL ) {
 473                         fprintf( stderr, "* host: %s  port: %d%s\n",
 474                             ( lc->lconn_server->lsrv_host == NULL ) ? "(null)"
 475                             : lc->lconn_server->lsrv_host,
 476                             lc->lconn_server->lsrv_port, ( lc->lconn_sb ==
 477                             &ld->ld_sb ) ? "  (default)" : "" );
 478                 }
 479                 fprintf( stderr, "  refcnt: %d  status: %s\n", lc->lconn_refcnt,
 480                     ( lc->lconn_status == LDAP_CONNST_NEEDSOCKET ) ?
 481                     "NeedSocket" : ( lc->lconn_status ==
 482                     LDAP_CONNST_CONNECTING ) ? "Connecting" : "Connected" );
 483                 fprintf( stderr, "  last used: %s\n",
 484                     ctime( &lc->lconn_lastused ));
 485                 if ( !all ) {
 486                         break;
 487                 }
 488         }
 489 }
 490
 491
 492 void
 493 dump_requests_and_responses( LDAP *ld )
 494 {
 495         LDAPRequest     *lr;
 496         LDAPMessage     *lm, *l;
 497
 498         fprintf( stderr, "** Outstanding Requests:\n" );
 499         if (( lr = ld->ld_requests ) == NULL ) {
 500                 fprintf( stderr, "   Empty\n" );
 501         }
 502         for ( ; lr != NULL; lr = lr->lr_next ) {
 503             fprintf( stderr, " * msgid %d,  origid %d, status %s\n",
 504                 lr->lr_msgid, lr->lr_origid, ( lr->lr_status ==
 505                 LDAP_REQST_INPROGRESS ) ? "InProgress" :
 506                 ( lr->lr_status == LDAP_REQST_CHASINGREFS ) ? "ChasingRefs" :
 507                 ( lr->lr_status == LDAP_REQST_NOTCONNECTED ) ? "NotConnected" :
 508                 "Writing" );
 509             fprintf( stderr, "   outstanding referrals %d, parent count %d\n",
 510                     lr->lr_outrefcnt, lr->lr_parentcnt );
 511         }
 512
 513         fprintf( stderr, "** Response Queue:\n" );
 514         if (( lm = ld->ld_responses ) == NULLMSG ) {
 515                 fprintf( stderr, "   Empty\n" );
 516         }
 517         for ( ; lm != NULLMSG; lm = lm->lm_next ) {
 518                 fprintf( stderr, " * msgid %d,  type %d\n",
 519                     lm->lm_msgid, lm->lm_msgtype );
 520                 if (( l = lm->lm_chain ) != NULL ) {
 521                         fprintf( stderr, "   chained responses:\n" );
 522                         for ( ; l != NULLMSG; l = l->lm_chain ) {
 523                                 fprintf( stderr,
 524                                     "  * msgid %d,  type %d\n",
 525                                     l->lm_msgid, l->lm_msgtype );
 526                         }
 527                 }
 528         }
 529 }
 530 #endif /* LDAP_DEBUG */
 531
 532
 533 void
 534 free_request( LDAP *ld, LDAPRequest *lr )
 535 {
 536         LDAPRequest     *tmplr, *nextlr;
 537
 538         Debug( LDAP_DEBUG_TRACE, "free_request (origid %d, msgid %d)\n",
 539                 lr->lr_origid, lr->lr_msgid, 0 );
 540
 541         if ( lr->lr_parent != NULL ) {
 542                 --lr->lr_parent->lr_outrefcnt;
 543         } else {
 544                 /* free all referrals (child requests) */
 545                 for ( tmplr = lr->lr_refnext; tmplr != NULL; tmplr = nextlr ) {
 546                         nextlr = tmplr->lr_refnext;
 547                         free_request( ld, tmplr );
 548                 }
 549         }
 550
 551         if ( lr->lr_prev == NULL ) {
 552                 ld->ld_requests = lr->lr_next;
 553         } else {
 554                 lr->lr_prev->lr_next = lr->lr_next;
 555         }
 556
 557         if ( lr->lr_next != NULL ) {
 558                 lr->lr_next->lr_prev = lr->lr_prev;
 559         }
 560
 561         if ( lr->lr_ber != NULL ) {
 562                 ber_free( lr->lr_ber, 1 );
 563         }
 564
 565         if ( lr->lr_res_error != NULL ) {
 566                 free( lr->lr_res_error );
 567         }
 568
 569         if ( lr->lr_res_matched != NULL ) {
 570                 free( lr->lr_res_matched );
 571         }
 572
 573         free( lr );
 574 }
 575
 576
 577 static void
 578 free_servers( LDAPServer *srvlist )
 579 {
 580     LDAPServer  *nextsrv;
 581
 582     while ( srvlist != NULL ) {
 583         nextsrv = srvlist->lsrv_next;
 584         if ( srvlist->lsrv_dn != NULL ) {
 585                 free( srvlist->lsrv_dn );
 586         }
 587         if ( srvlist->lsrv_host != NULL ) {
 588                 free( srvlist->lsrv_host );
 589         }
 590         free( srvlist );
 591         srvlist = nextsrv;
 592     }
 593 }
 594 #endif /* LDAP_REFERRALS || LDAP_DNS */
 595
 596
 597 #ifdef LDAP_REFERRALS
 598 /*
 599  * XXX merging of errors in this routine needs to be improved
 600  */
 601 int
 602 chase_referrals( LDAP *ld, LDAPRequest *lr, char **errstrp, int *hadrefp )
 603 {
 604         int             rc, count, len, newdn;
 605 #ifdef LDAP_DNS
 606         int             ldapref;
 607 #endif /* LDAP_DNS */
 608         char            *p, *ports, *ref, *tmpref, *refdn, *unfollowed;
 609         LDAPRequest     *origreq;
 610         LDAPServer      *srv;
 611         BerElement      *ber;
 612
 613         Debug( LDAP_DEBUG_TRACE, "chase_referrals\n", 0, 0, 0 );
 614
 615         ld->ld_errno = LDAP_SUCCESS;    /* optimistic */
 616         *hadrefp = 0;
 617
 618         if ( *errstrp == NULL ) {
 619                 return( 0 );
 620         }
 621
 622         len = strlen( *errstrp );
 623         for ( p = *errstrp; len >= LDAP_REF_STR_LEN; ++p, --len ) {
 624                 if (( *p == 'R' || *p == 'r' ) && strncasecmp( p,
 625                     LDAP_REF_STR, LDAP_REF_STR_LEN ) == 0 ) {
 626                         *p = '\0';
 627                         p += LDAP_REF_STR_LEN;
 628                         break;
 629                 }
 630         }
 631
 632         if ( len < LDAP_REF_STR_LEN ) {
 633                 return( 0 );
 634         }
 635
 636         if ( lr->lr_parentcnt >= ld->ld_refhoplimit ) {
 637                 Debug( LDAP_DEBUG_ANY,
 638                     "more than %d referral hops (dropping)\n",
 639                     ld->ld_refhoplimit, 0, 0 );
 640                     /* XXX report as error in ld->ld_errno? */
 641                     return( 0 );
 642         }
 643
 644         /* find original request */
 645         for ( origreq = lr; origreq->lr_parent != NULL;
 646              origreq = origreq->lr_parent ) {
 647                 ;
 648         }
 649
 650         unfollowed = NULL;
 651         rc = count = 0;
 652
 653         /* parse out & follow referrals */
 654         for ( ref = p; rc == 0 && ref != NULL; ref = p ) {
 655 #ifdef LDAP_DNS
 656                 ldapref = 0;
 657 #endif /* LDAP_DNS */
 658
 659                 if (( p = strchr( ref, '\n' )) != NULL ) {
 660                         *p++ = '\0';
 661                 } else {
 662                         p = NULL;
 663                 }
 664
 665                 len = strlen( ref );
 666                 if ( len > LDAP_LDAP_REF_STR_LEN && strncasecmp( ref,
 667                     LDAP_LDAP_REF_STR, LDAP_LDAP_REF_STR_LEN ) == 0 ) {
 668                         Debug( LDAP_DEBUG_TRACE,
 669                             "chasing LDAP referral: <%s>\n", ref, 0, 0 );
 670 #ifdef LDAP_DNS
 671                         ldapref = 1;
 672 #endif /* LDAP_DNS */
 673                         tmpref = ref + LDAP_LDAP_REF_STR_LEN;
 674 #ifdef LDAP_DNS
 675                 } else if ( len > LDAP_DX_REF_STR_LEN && strncasecmp( ref,
 676                     LDAP_DX_REF_STR, LDAP_DX_REF_STR_LEN ) == 0 ) {
 677                         Debug( LDAP_DEBUG_TRACE,
 678                             "chasing DX referral: <%s>\n", ref, 0, 0 );
 679                         tmpref = ref + LDAP_DX_REF_STR_LEN;
 680 #endif /* LDAP_DNS */
 681                 } else {
 682                         Debug( LDAP_DEBUG_TRACE,
 683                             "ignoring unknown referral <%s>\n", ref, 0, 0 );
 684                         rc = append_referral( ld, &unfollowed, ref );
 685                         *hadrefp = 1;
 686                         continue;
 687                 }
 688
 689                 *hadrefp = 1;
 690                 if (( refdn = strchr( tmpref, '/' )) != NULL ) {
 691                         *refdn++ = '\0';
 692                         newdn = 1;
 693                 } else {
 694                         newdn = 0;
 695                 }
 696
 697                 if (( ber = re_encode_request( ld, origreq->lr_ber,
 698                     ++ld->ld_msgid, &refdn )) == NULL ) {
 699                         return( -1 );
 700                 }
 701
 702 #ifdef LDAP_DNS
 703                 if ( ldapref ) {
 704 #endif /* LDAP_DNS */
 705                         if (( srv = (LDAPServer *)calloc( 1,
 706                             sizeof( LDAPServer ))) == NULL ) {
 707                                 ber_free( ber, 1 );
 708                                 ld->ld_errno = LDAP_NO_MEMORY;
 709                                 return( -1 );
 710                         }
 711
 712                         if (( srv->lsrv_host = strdup( tmpref )) == NULL ) {
 713                                 free( (char *)srv );
 714                                 ber_free( ber, 1 );
 715                                 ld->ld_errno = LDAP_NO_MEMORY;
 716                                 return( -1 );
 717                         }
 718
 719                         if (( ports = strchr( srv->lsrv_host, ':' )) != NULL ) {
 720                                 *ports++ = '\0';
 721                                 srv->lsrv_port = atoi( ports );
 722                         } else {
 723                                 srv->lsrv_port = LDAP_PORT;
 724                         }
 725 #ifdef LDAP_DNS
 726                 } else {
 727                         srv = dn2servers( ld, tmpref );
 728                 }
 729 #endif /* LDAP_DNS */
 730
 731                 if ( srv != NULL && send_server_request( ld, ber, ld->ld_msgid,
 732                     lr, srv, NULL, 1 ) >= 0 ) {
 733                         ++count;
 734                 } else {
 735                         Debug( LDAP_DEBUG_ANY,
 736                             "Unable to chase referral (%s)\n",
 737                             ldap_err2string( ld->ld_errno ), 0, 0 );
 738                         rc = append_referral( ld, &unfollowed, ref );
 739                 }
 740
 741                 if ( !newdn && refdn != NULL ) {
 742                         free( refdn );
 743                 }
 744         }
 745
 746         free( *errstrp );
 747         *errstrp = unfollowed;
 748
 749         return(( rc == 0 ) ? count : rc );
 750 }
 751
 752
 753 int
 754 append_referral( LDAP *ld, char **referralsp, char *s )
 755 {
 756         int     first;
 757
 758         if ( *referralsp == NULL ) {
 759                 first = 1;
 760                 *referralsp = (char *)malloc( strlen( s ) + LDAP_REF_STR_LEN
 761                     + 1 );
 762         } else {
 763                 first = 0;
 764                 *referralsp = (char *)realloc( *referralsp,
 765                     strlen( *referralsp ) + strlen( s ) + 2 );
 766         }
 767
 768         if ( *referralsp == NULL ) {
 769                 ld->ld_errno = LDAP_NO_MEMORY;
 770                 return( -1 );
 771         }
 772
 773         if ( first ) {
 774                 strcpy( *referralsp, LDAP_REF_STR );
 775         } else {
 776                 strcat( *referralsp, "\n" );
 777         }
 778         strcat( *referralsp, s );
 779
 780         return( 0 );
 781 }
 782
 783
 784
 785 static BerElement *
 786 re_encode_request( LDAP *ld, BerElement *origber, int msgid, char **dnp )
 787 {
 788 /*
 789  * XXX this routine knows way too much about how the lber library works!
 790  */
 791         unsigned long   along, tag;
 792         long            ver;
 793         int             rc;
 794         BerElement      tmpber, *ber;
 795         char            *orig_dn;
 796
 797         Debug( LDAP_DEBUG_TRACE,
 798             "re_encode_request: new msgid %d, new dn <%s>\n",
 799             msgid, ( *dnp == NULL ) ? "NONE" : *dnp, 0 );
 800
 801         tmpber = *origber;
 802
 803         /*
 804          * all LDAP requests are sequences that start with a message id,
 805          * followed by a sequence that is tagged with the operation code
 806          */
 807         if ( ber_scanf( &tmpber, "{i", &along ) != LDAP_TAG_MSGID ||
 808             ( tag = ber_skip_tag( &tmpber, &along )) == LBER_DEFAULT ) {
 809                 ld->ld_errno = LDAP_DECODING_ERROR;
 810                 return( NULL );
 811         }
 812
 813         if (( ber = alloc_ber_with_options( ld )) == NULLBER ) {
 814                 return( NULL );
 815         }
 816
 817         /* bind requests have a version number before the DN & other stuff */
 818         if ( tag == LDAP_REQ_BIND && ber_get_int( &tmpber, (long *)&ver ) ==
 819             LBER_DEFAULT ) {
 820                 ld->ld_errno = LDAP_DECODING_ERROR;
 821                 ber_free( ber, 1 );
 822                 return( NULL );
 823         }
 824
 825         /* the rest of the request is the DN followed by other stuff */
 826         if ( ber_get_stringa( &tmpber, &orig_dn ) == LBER_DEFAULT ) {
 827                 ber_free( ber, 1 );
 828                 return( NULL );
 829         }
 830
 831         if ( *dnp == NULL ) {
 832                 *dnp = orig_dn;
 833         } else {
 834                 free( orig_dn );
 835         }
 836
 837         if ( tag == LDAP_REQ_BIND ) {
 838                 rc = ber_printf( ber, "{it{is", msgid, tag, ver, *dnp );
 839         } else {
 840                 rc = ber_printf( ber, "{it{s", msgid, tag, *dnp );
 841         }
 842
 843         if ( rc == -1 ) {
 844                 ber_free( ber, 1 );
 845                 return( NULL );
 846         }
 847
 848         if ( ber_write( ber, tmpber.ber_ptr, ( tmpber.ber_end -
 849             tmpber.ber_ptr ), 0 ) != ( tmpber.ber_end - tmpber.ber_ptr ) ||
 850             ber_printf( ber, "}}" ) == -1 ) {
 851                 ld->ld_errno = LDAP_ENCODING_ERROR;
 852                 ber_free( ber, 1 );
 853                 return( NULL );
 854         }
 855
 856 #ifdef LDAP_DEBUG
 857         if ( ldap_debug & LDAP_DEBUG_PACKETS ) {
 858                 Debug( LDAP_DEBUG_ANY, "re_encode_request new request is:\n",
 859                     0, 0, 0 );
 860                 ber_dump( ber, 0 );
 861         }
 862 #endif /* LDAP_DEBUG */
 863
 864         return( ber );
 865 }
 866
 867
 868 LDAPRequest *
 869 find_request_by_msgid( LDAP *ld, int msgid )
 870 {
 871         LDAPRequest     *lr;
 872
 873         for ( lr = ld->ld_requests; lr != NULL; lr = lr->lr_next ) {
 874                 if ( msgid == lr->lr_msgid ) {
 875                         break;
 876                 }
 877         }
 878
 879         return( lr );
 880 }
 881 #endif /* LDAP_REFERRALS */
 882
 883
 884 #ifdef LDAP_DNS
 885 static LDAPServer *
 886 dn2servers( LDAP *ld, char *dn )        /* dn can also be a domain.... */
 887 {
 888         char            *p, *domain, *host, *server_dn, **dxs;
 889         int             i, port;
 890         LDAPServer      *srvlist, *prevsrv, *srv;
 891
 892         if (( domain = strrchr( dn, '@' )) != NULL ) {
 893                 ++domain;
 894         } else {
 895                 domain = dn;
 896         }
 897
 898         if (( dxs = getdxbyname( domain )) == NULL ) {
 899                 ld->ld_errno = LDAP_NO_MEMORY;
 900                 return( NULL );
 901         }
 902
 903         srvlist = NULL;
 904
 905         for ( i = 0; dxs[ i ] != NULL; ++i ) {
 906                 port = LDAP_PORT;
 907                 server_dn = NULL;
 908                 if ( strchr( dxs[ i ], ':' ) == NULL ) {
 909                         host = dxs[ i ];
 910                 } else if ( strlen( dxs[ i ] ) >= 7 &&
 911                     strncmp( dxs[ i ], "ldap://", 7 ) == 0 ) {
 912                         host = dxs[ i ] + 7;
 913                         if (( p = strchr( host, ':' )) == NULL ) {
 914                                 p = host;
 915                         } else {
 916                                 *p++ = '\0';
 917                                 port = atoi( p );
 918                         }
 919                         if (( p = strchr( p, '/' )) != NULL ) {
 920                                 server_dn = ++p;
 921                                 if ( *server_dn == '\0' ) {
 922                                         server_dn = NULL;
 923                                 }
 924                         }
 925                 } else {
 926                         host = NULL;
 927                 }
 928
 929                 if ( host != NULL ) {   /* found a server we can use */
 930                         if (( srv = (LDAPServer *)calloc( 1,
 931                             sizeof( LDAPServer ))) == NULL ) {
 932                                 free_servers( srvlist );
 933                                 srvlist = NULL;
 934                                 break;          /* exit loop & return */
 935                         }
 936
 937                         /* add to end of list of servers */
 938                         if ( srvlist == NULL ) {
 939                                 srvlist = srv;
 940                         } else {
 941                                 prevsrv->lsrv_next = srv;
 942                         }
 943                         prevsrv = srv;
 944
 945                         /* copy in info. */
 946                         if (( srv->lsrv_host = strdup( host )) == NULL ||
 947                             ( server_dn != NULL && ( srv->lsrv_dn =
 948                             strdup( server_dn )) == NULL )) {
 949                                 free_servers( srvlist );
 950                                 srvlist = NULL;
 951                                 break;          /* exit loop & return */
 952                         }
 953                         srv->lsrv_port = port;
 954                 }
 955         }
 956
 957         ldap_value_free( dxs );
 958
 959         if ( srvlist == NULL ) {
 960                 ld->ld_errno = LDAP_SERVER_DOWN;
 961         }
 962
 963         return( srvlist );
 964 }
 965 #endif /* LDAP_DNS */