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