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