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