Diff for /libraries/libldap/result.c between versions 1.124.2.21 and 1.175

version 1.124.2.21, 2010/04/13 20:22:59 version 1.175, 2010/11/24 13:47:46
Line 1 Line 1
 /* result.c - wait for an ldap result */  /* result.c - wait for an ldap result */
 /* $OpenLDAP: pkg/ldap/libraries/libldap/result.c,v 1.124.2.20 2009/11/18 17:04:31 quanah Exp $ */  /* $OpenLDAP: pkg/ldap/libraries/libldap/result.c,v 1.174 2010/10/22 19:45:49 hyc Exp $ */
 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.  /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
  *   *
  * Copyright 1998-2010 The OpenLDAP Foundation.   * Copyright 1998-2010 The OpenLDAP Foundation.
Line 66 Line 66
 #include "ldap_log.h"  #include "ldap_log.h"
 #include "lutil.h"  #include "lutil.h"
   
 static int ldap_abandoned LDAP_P(( LDAP *ld, ber_int_t msgid, int *idx ));  static int ldap_abandoned LDAP_P(( LDAP *ld, ber_int_t msgid ));
 static int ldap_mark_abandoned LDAP_P(( LDAP *ld, ber_int_t msgid, int idx ));  static int ldap_mark_abandoned LDAP_P(( LDAP *ld, ber_int_t msgid ));
 static int wait4msg LDAP_P(( LDAP *ld, ber_int_t msgid, int all, struct timeval *timeout,  static int wait4msg LDAP_P(( LDAP *ld, ber_int_t msgid, int all, struct timeval *timeout,
         LDAPMessage **result ));          LDAPMessage **result ));
 static ber_tag_t try_read1msg LDAP_P(( LDAP *ld, ber_int_t msgid,  static ber_tag_t try_read1msg LDAP_P(( LDAP *ld, ber_int_t msgid,
Line 113  ldap_result( Line 113  ldap_result(
   
         Debug( LDAP_DEBUG_TRACE, "ldap_result ld %p msgid %d\n", (void *)ld, msgid, 0 );          Debug( LDAP_DEBUG_TRACE, "ldap_result ld %p msgid %d\n", (void *)ld, msgid, 0 );
   
 #ifdef LDAP_R_COMPILE          LDAP_MUTEX_LOCK( &ld->ld_res_mutex );
         ldap_pvt_thread_mutex_lock( &ld->ld_res_mutex );  
 #endif  
   
         rc = wait4msg( ld, msgid, all, timeout, result );          rc = wait4msg( ld, msgid, all, timeout, result );
           LDAP_MUTEX_UNLOCK( &ld->ld_res_mutex );
 #ifdef LDAP_R_COMPILE  
         ldap_pvt_thread_mutex_unlock( &ld->ld_res_mutex );  
 #endif  
   
         return rc;          return rc;
 }  }
   
   /* protected by res_mutex */
 static LDAPMessage *  static LDAPMessage *
 chkResponseList(  chkResponseList(
         LDAP *ld,          LDAP *ld,
Line 142  chkResponseList( Line 137  chkResponseList(
          * wait until it arrives or timeout occurs.           * wait until it arrives or timeout occurs.
          */           */
   
 #ifdef LDAP_R_COMPILE          LDAP_ASSERT_MUTEX_OWNER( &ld->ld_res_mutex );
         LDAP_PVT_THREAD_ASSERT_MUTEX_OWNER( &ld->ld_res_mutex );  
 #endif  
   
         Debug( LDAP_DEBUG_TRACE,          Debug( LDAP_DEBUG_TRACE,
                 "ldap_chkResponseList ld %p msgid %d all %d\n",                  "ldap_chkResponseList ld %p msgid %d all %d\n",
Line 152  chkResponseList( Line 145  chkResponseList(
   
         lastlm = &ld->ld_responses;          lastlm = &ld->ld_responses;
         for ( lm = ld->ld_responses; lm != NULL; lm = nextlm ) {          for ( lm = ld->ld_responses; lm != NULL; lm = nextlm ) {
                 int     idx;  
   
                 nextlm = lm->lm_next;                  nextlm = lm->lm_next;
                 ++cnt;                  ++cnt;
   
                 if ( ldap_abandoned( ld, lm->lm_msgid, &idx ) ) {                  if ( ldap_abandoned( ld, lm->lm_msgid ) ) {
                         Debug( LDAP_DEBUG_ANY,                          Debug( LDAP_DEBUG_ANY,
                                 "response list msg abandoned, "                                  "response list msg abandoned, "
                                 "msgid %d message type %s\n",                                  "msgid %d message type %s\n",
Line 172  chkResponseList( Line 163  chkResponseList(
                         default:                          default:
                                 /* there's no need to keep the id                                  /* there's no need to keep the id
                                  * in the abandoned list any longer */                                   * in the abandoned list any longer */
                                 ldap_mark_abandoned( ld, lm->lm_msgid, idx );                                  ldap_mark_abandoned( ld, lm->lm_msgid );
                                 break;                                  break;
                         }                          }
   
Line 239  chkResponseList( Line 230  chkResponseList(
         return lm;          return lm;
 }  }
   
   /* protected by res_mutex */
 static int  static int
 wait4msg(  wait4msg(
         LDAP *ld,          LDAP *ld,
Line 257  wait4msg( Line 249  wait4msg(
         assert( ld != NULL );          assert( ld != NULL );
         assert( result != NULL );          assert( result != NULL );
   
 #ifdef LDAP_R_COMPILE          LDAP_ASSERT_MUTEX_OWNER( &ld->ld_res_mutex );
         LDAP_PVT_THREAD_ASSERT_MUTEX_OWNER( &ld->ld_res_mutex );  
 #endif  
   
         if ( timeout == NULL && ld->ld_options.ldo_tm_api.tv_sec >= 0 ) {          if ( timeout == NULL && ld->ld_options.ldo_tm_api.tv_sec >= 0 ) {
                 tv = ld->ld_options.ldo_tm_api;                  tv = ld->ld_options.ldo_tm_api;
Line 294  wait4msg( Line 284  wait4msg(
                 if ( ldap_debug & LDAP_DEBUG_TRACE ) {                  if ( ldap_debug & LDAP_DEBUG_TRACE ) {
                         Debug( LDAP_DEBUG_TRACE, "wait4msg continue ld %p msgid %d all %d\n",                          Debug( LDAP_DEBUG_TRACE, "wait4msg continue ld %p msgid %d all %d\n",
                                 (void *)ld, msgid, all );                                  (void *)ld, msgid, all );
 #ifdef LDAP_R_COMPILE  
                         ldap_pvt_thread_mutex_lock( &ld->ld_conn_mutex );  
 #endif  
                         ldap_dump_connection( ld, ld->ld_conns, 1 );                          ldap_dump_connection( ld, ld->ld_conns, 1 );
 #ifdef LDAP_R_COMPILE                          LDAP_MUTEX_LOCK( &ld->ld_req_mutex );
                         ldap_pvt_thread_mutex_unlock( &ld->ld_conn_mutex );  
                         ldap_pvt_thread_mutex_lock( &ld->ld_req_mutex );  
 #endif  
                         ldap_dump_requests_and_responses( ld );                          ldap_dump_requests_and_responses( ld );
 #ifdef LDAP_R_COMPILE                          LDAP_MUTEX_UNLOCK( &ld->ld_req_mutex );
                         ldap_pvt_thread_mutex_unlock( &ld->ld_req_mutex );  
 #endif  
                 }                  }
 #endif /* LDAP_DEBUG */  #endif /* LDAP_DEBUG */
   
Line 315  wait4msg( Line 297  wait4msg(
                 } else {                  } else {
                         int lc_ready = 0;                          int lc_ready = 0;
   
 #ifdef LDAP_R_COMPILE                          LDAP_MUTEX_LOCK( &ld->ld_conn_mutex );
                         ldap_pvt_thread_mutex_lock( &ld->ld_conn_mutex );  
 #endif  
                         for ( lc = ld->ld_conns; lc != NULL; lc = lc->lconn_next ) {                          for ( lc = ld->ld_conns; lc != NULL; lc = lc->lconn_next ) {
                                 if ( ber_sockbuf_ctrl( lc->lconn_sb,                                  if ( ber_sockbuf_ctrl( lc->lconn_sb,
                                         LBER_SB_OPT_DATA_READY, NULL ) )                                          LBER_SB_OPT_DATA_READY, NULL ) )
Line 326  wait4msg( Line 306  wait4msg(
                                         break;                                          break;
                                 }                                  }
                         }                          }
 #ifdef LDAP_R_COMPILE  
                         ldap_pvt_thread_mutex_unlock( &ld->ld_conn_mutex );  
 #endif  
   
                         if ( !lc_ready ) {                          if ( !lc_ready ) {
                                 int err;                                  int err;
Line 348  wait4msg( Line 325  wait4msg(
                                 {                                  {
                                         ld->ld_errno = (rc == -1 ? LDAP_SERVER_DOWN :                                          ld->ld_errno = (rc == -1 ? LDAP_SERVER_DOWN :
                                                 LDAP_TIMEOUT);                                                  LDAP_TIMEOUT);
                                           LDAP_MUTEX_UNLOCK( &ld->ld_conn_mutex );
                                         return( rc );                                          return( rc );
                                 }                                  }
   
Line 361  wait4msg( Line 339  wait4msg(
                         if ( lc_ready ) {                          if ( lc_ready ) {
                                 LDAPConn *lnext;                                  LDAPConn *lnext;
                                 rc = LDAP_MSG_X_KEEP_LOOKING;                                  rc = LDAP_MSG_X_KEEP_LOOKING;
 #ifdef LDAP_R_COMPILE                                  LDAP_MUTEX_LOCK( &ld->ld_req_mutex );
                                 ldap_pvt_thread_mutex_lock( &ld->ld_req_mutex );  
 #endif  
                                 if ( ld->ld_requests &&                                  if ( ld->ld_requests &&
                                         ld->ld_requests->lr_status == LDAP_REQST_WRITING &&                                          ld->ld_requests->lr_status == LDAP_REQST_WRITING &&
                                         ldap_is_write_ready( ld,                                          ldap_is_write_ready( ld,
Line 371  wait4msg( Line 347  wait4msg(
                                 {                                  {
                                         ldap_int_flush_request( ld, ld->ld_requests );                                          ldap_int_flush_request( ld, ld->ld_requests );
                                 }                                  }
 #ifdef LDAP_R_COMPILE  
                                 ldap_pvt_thread_mutex_unlock( &ld->ld_req_mutex );  
                                 ldap_pvt_thread_mutex_lock( &ld->ld_conn_mutex );  
 #endif  
                                 for ( lc = ld->ld_conns;                                  for ( lc = ld->ld_conns;
                                         rc == LDAP_MSG_X_KEEP_LOOKING && lc != NULL;                                          rc == LDAP_MSG_X_KEEP_LOOKING && lc != NULL;
                                         lc = lnext )                                          lc = lnext )
Line 384  wait4msg( Line 356  wait4msg(
                                         {                                          {
                                                 /* Don't let it get freed out from under us */                                                  /* Don't let it get freed out from under us */
                                                 ++lc->lconn_refcnt;                                                  ++lc->lconn_refcnt;
 #ifdef LDAP_R_COMPILE  
                                                 ldap_pvt_thread_mutex_unlock( &ld->ld_conn_mutex );  
 #endif  
                                                 rc = try_read1msg( ld, msgid, all, lc, result );                                                  rc = try_read1msg( ld, msgid, all, lc, result );
                                                 lnext = lc->lconn_next;                                                  lnext = lc->lconn_next;
   
                                                 /* Only take locks if we're really freeing */                                                  /* Only take locks if we're really freeing */
                                                 if ( lc->lconn_refcnt <= 1 ) {                                                  if ( lc->lconn_refcnt <= 1 ) {
 #ifdef LDAP_R_COMPILE  
                                                         ldap_pvt_thread_mutex_lock( &ld->ld_req_mutex );  
 #endif  
                                                         ldap_free_connection( ld, lc, 0, 1 );                                                          ldap_free_connection( ld, lc, 0, 1 );
 #ifdef LDAP_R_COMPILE  
                                                         ldap_pvt_thread_mutex_unlock( &ld->ld_req_mutex );  
 #endif  
                                                 } else {                                                  } else {
                                                         --lc->lconn_refcnt;                                                          --lc->lconn_refcnt;
                                                 }                                                  }
 #ifdef LDAP_R_COMPILE  
                                                 ldap_pvt_thread_mutex_lock( &ld->ld_conn_mutex );  
 #endif  
                                         } else {                                          } else {
                                                 lnext = lc->lconn_next;                                                  lnext = lc->lconn_next;
                                         }                                          }
                                 }                                  }
 #ifdef LDAP_R_COMPILE                                  LDAP_MUTEX_UNLOCK( &ld->ld_req_mutex );
                                 ldap_pvt_thread_mutex_unlock( &ld->ld_conn_mutex );  
 #endif  
                         }                          }
                           LDAP_MUTEX_UNLOCK( &ld->ld_conn_mutex );
                 }                  }
   
                 if ( rc == LDAP_MSG_X_KEEP_LOOKING && tvp != NULL ) {                  if ( rc == LDAP_MSG_X_KEEP_LOOKING && tvp != NULL ) {
Line 466  wait4msg( Line 425  wait4msg(
 }  }
   
   
   /* protected by res_mutex, conn_mutex and req_mutex */
 static ber_tag_t  static ber_tag_t
 try_read1msg(  try_read1msg(
         LDAP *ld,          LDAP *ld,
Line 477  try_read1msg( Line 437  try_read1msg(
         BerElement      *ber;          BerElement      *ber;
         LDAPMessage     *newmsg, *l, *prev;          LDAPMessage     *newmsg, *l, *prev;
         ber_int_t       id;          ber_int_t       id;
         int             idx;  
         ber_tag_t       tag;          ber_tag_t       tag;
         ber_len_t       len;          ber_len_t       len;
         int             foundit = 0;          int             foundit = 0;
Line 494  try_read1msg( Line 453  try_read1msg(
         assert( ld != NULL );          assert( ld != NULL );
         assert( lc != NULL );          assert( lc != NULL );
                   
 #ifdef LDAP_R_COMPILE          LDAP_ASSERT_MUTEX_OWNER( &ld->ld_res_mutex );
         LDAP_PVT_THREAD_ASSERT_MUTEX_OWNER( &ld->ld_res_mutex );          LDAP_ASSERT_MUTEX_OWNER( &ld->ld_conn_mutex );
 #endif          LDAP_ASSERT_MUTEX_OWNER( &ld->ld_req_mutex );
   
         Debug( LDAP_DEBUG_TRACE, "read1msg: ld %p msgid %d all %d\n",          Debug( LDAP_DEBUG_TRACE, "read1msg: ld %p msgid %d all %d\n",
                 (void *)ld, msgid, all );                  (void *)ld, msgid, all );
Line 539  nextresp3: Line 498  nextresp3:
                 Debug( LDAP_DEBUG_CONNS,                  Debug( LDAP_DEBUG_CONNS,
                         "ber_get_next failed.\n", 0, 0, 0 );                          "ber_get_next failed.\n", 0, 0, 0 );
 #endif               #endif             
 #ifdef EWOULDBLOCK                        
                 if ( err == EWOULDBLOCK ) return LDAP_MSG_X_KEEP_LOOKING;                  if ( err == EWOULDBLOCK ) return LDAP_MSG_X_KEEP_LOOKING;
 #endif  
 #ifdef EAGAIN  
                 if ( err == EAGAIN ) return LDAP_MSG_X_KEEP_LOOKING;                  if ( err == EAGAIN ) return LDAP_MSG_X_KEEP_LOOKING;
 #endif  
                 ld->ld_errno = LDAP_SERVER_DOWN;                  ld->ld_errno = LDAP_SERVER_DOWN;
                 --lc->lconn_refcnt;                  --lc->lconn_refcnt;
                 lc->lconn_status = 0;                  lc->lconn_status = 0;
Line 571  nextresp3: Line 526  nextresp3:
                   
         /* if it's been abandoned, toss it */          /* if it's been abandoned, toss it */
         if ( id > 0 ) {          if ( id > 0 ) {
                 if ( ldap_abandoned( ld, id, &idx ) ) {                  if ( ldap_abandoned( ld, id ) ) {
                         /* the message type */                          /* the message type */
                         tag = ber_peek_tag( ber, &len );                          tag = ber_peek_tag( ber, &len );
                         switch ( tag ) {                          switch ( tag ) {
Line 584  nextresp3: Line 539  nextresp3:
                         default:                          default:
                                 /* there's no need to keep the id                                  /* there's no need to keep the id
                                  * in the abandoned list any longer */                                   * in the abandoned list any longer */
                                 ldap_mark_abandoned( ld, id, idx );                                  ldap_mark_abandoned( ld, id );
                                 break;                                  break;
                         }                          }
   
Line 747  nextresp2: Line 702  nextresp2:
                         }                          }
   
                         /* Do we need to check for referrals? */                          /* Do we need to check for referrals? */
                         if ( LDAP_BOOL_GET(&ld->ld_options, LDAP_BOOL_REFERRALS) ||                          if ( tag != LDAP_RES_BIND &&
                                         lr->lr_parent != NULL )                                  ( LDAP_BOOL_GET(&ld->ld_options, LDAP_BOOL_REFERRALS) ||
                                           lr->lr_parent != NULL ))
                         {                          {
                                 char            **refs = NULL;                                  char            **refs = NULL;
                                 ber_len_t       len;                                  ber_len_t       len;
Line 993  nextresp2: Line 949  nextresp2:
   
                         /* need to return -1, because otherwise                          /* need to return -1, because otherwise
                          * a valid result is expected */                           * a valid result is expected */
                           ld->ld_errno = lderr;
                         return -1;                          return -1;
                 }                  }
         }          }
Line 1329  ldap_msgdelete( LDAP *ld, int msgid ) Line 1286  ldap_msgdelete( LDAP *ld, int msgid )
         Debug( LDAP_DEBUG_TRACE, "ldap_msgdelete ld=%p msgid=%d\n",          Debug( LDAP_DEBUG_TRACE, "ldap_msgdelete ld=%p msgid=%d\n",
                 (void *)ld, msgid, 0 );                  (void *)ld, msgid, 0 );
   
 #ifdef LDAP_R_COMPILE          LDAP_MUTEX_LOCK( &ld->ld_res_mutex );
         ldap_pvt_thread_mutex_lock( &ld->ld_res_mutex );  
 #endif  
         prev = NULL;          prev = NULL;
         for ( lm = ld->ld_responses; lm != NULL; lm = lm->lm_next ) {          for ( lm = ld->ld_responses; lm != NULL; lm = lm->lm_next ) {
                 if ( lm->lm_msgid == msgid ) {                  if ( lm->lm_msgid == msgid ) {
Line 1350  ldap_msgdelete( LDAP *ld, int msgid ) Line 1305  ldap_msgdelete( LDAP *ld, int msgid )
                         prev->lm_next = lm->lm_next;                          prev->lm_next = lm->lm_next;
                 }                  }
         }          }
 #ifdef LDAP_R_COMPILE          LDAP_MUTEX_UNLOCK( &ld->ld_res_mutex );
         ldap_pvt_thread_mutex_unlock( &ld->ld_res_mutex );  
 #endif  
         if ( lm ) {          if ( lm ) {
                 switch ( ldap_msgfree( lm ) ) {                  switch ( ldap_msgfree( lm ) ) {
                 case LDAP_RES_SEARCH_ENTRY:                  case LDAP_RES_SEARCH_ENTRY:
Line 1375  ldap_msgdelete( LDAP *ld, int msgid ) Line 1328  ldap_msgdelete( LDAP *ld, int msgid )
  *   *
  * return the location of the message id in the array of abandoned   * return the location of the message id in the array of abandoned
  * message ids, or -1   * message ids, or -1
  *  
  * expects ld_res_mutex to be locked  
  */   */
 static int  static int
 ldap_abandoned( LDAP *ld, ber_int_t msgid, int *idxp )  ldap_abandoned( LDAP *ld, ber_int_t msgid )
 {  {
 #ifdef LDAP_R_COMPILE          int     ret, idx;
         LDAP_PVT_THREAD_ASSERT_MUTEX_OWNER( &ld->ld_res_mutex );  
 #endif  
   
         assert( idxp != NULL );  
         assert( msgid >= 0 );          assert( msgid >= 0 );
   
         return ldap_int_bisect_find( ld->ld_abandoned, ld->ld_nabandoned, msgid, idxp );          LDAP_MUTEX_LOCK( &ld->ld_abandon_mutex );
           ret = ldap_int_bisect_find( ld->ld_abandoned, ld->ld_nabandoned, msgid, &idx );
           LDAP_MUTEX_UNLOCK( &ld->ld_abandon_mutex );
           return ret;
 }  }
   
 /*  /*
  * ldap_mark_abandoned   * ldap_mark_abandoned
  *  
  * expects ld_res_mutex to be locked  
  */   */
 static int  static int
 ldap_mark_abandoned( LDAP *ld, ber_int_t msgid, int idx )  ldap_mark_abandoned( LDAP *ld, ber_int_t msgid )
 {  {
 #ifdef LDAP_R_COMPILE          int     ret, idx;
         LDAP_PVT_THREAD_ASSERT_MUTEX_OWNER( &ld->ld_res_mutex );  
 #endif  
   
         /* NOTE: those assertions are repeated in ldap_int_bisect_delete() */  
         assert( idx >= 0 );  
         assert( (unsigned) idx < ld->ld_nabandoned );  
         assert( ld->ld_abandoned[ idx ] == msgid );  
   
         return ldap_int_bisect_delete( &ld->ld_abandoned, &ld->ld_nabandoned,          assert( msgid >= 0 );
           LDAP_MUTEX_LOCK( &ld->ld_abandon_mutex );
           ret = ldap_int_bisect_find( ld->ld_abandoned, ld->ld_nabandoned, msgid, &idx );
           if (ret <= 0) {         /* error or already deleted by another thread */
                   LDAP_MUTEX_UNLOCK( &ld->ld_abandon_mutex );
                   return ret;
           }
           /* still in abandoned array, so delete */
           ret = ldap_int_bisect_delete( &ld->ld_abandoned, &ld->ld_nabandoned,
                 msgid, idx );                  msgid, idx );
           LDAP_MUTEX_UNLOCK( &ld->ld_abandon_mutex );
           return ret;
 }  }

Removed from v.1.124.2.21  
changed lines
  Added in v.1.175


______________
© Copyright 1998-2020, OpenLDAP Foundation, info@OpenLDAP.org