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