X-Git-Url: https://git.sur5r.net/?a=blobdiff_plain;ds=sidebyside;f=servers%2Fslapd%2Fresult.c;h=0803f7fb80ce42add0af8176db03d900fa712232;hb=b508c611a6582fc9f23e0094b26a89b7c075131a;hp=b5c6b90f5f3d5086c0ca1f93e39692d49acf4de1;hpb=1a171b07d3776ee1643ea93adfd0c4f9230f8012;p=openldap diff --git a/servers/slapd/result.c b/servers/slapd/result.c index b5c6b90f5f..0803f7fb80 100644 --- a/servers/slapd/result.c +++ b/servers/slapd/result.c @@ -2,7 +2,7 @@ /* $OpenLDAP$ */ /* This work is part of OpenLDAP Software . * - * Copyright 1998-2008 The OpenLDAP Foundation. + * Copyright 1998-2011 The OpenLDAP Foundation. * All rights reserved. * * Redistribution and use in source and binary forms, with or without @@ -132,41 +132,208 @@ slap_req2res( ber_tag_t tag ) return tag; } +/* SlapReply debugging, prodo-slap.h overrides it in OpenLDAP releases */ +#if defined(LDAP_TEST) || (defined(USE_RS_ASSERT) && (USE_RS_ASSERT)) + +int rs_suppress_assert = 0; + +/* RS_ASSERT() helper function */ +void rs_assert_(const char*file, unsigned line, const char*fn, const char*cond) +{ + int no_assert = rs_suppress_assert, save_errno = errno; + const char *s; + + if ( no_assert >= 0 ) { + if ( no_assert == 0 && (s = getenv( "NO_RS_ASSERT" )) && *s ) { + no_assert = rs_suppress_assert = atoi( s ); + } + if ( no_assert > 0 ) { + errno = save_errno; + return; + } + } + +#ifdef rs_assert_ /* proto-slap.h #defined away the fn parameter */ + fprintf( stderr,"%s:%u: " "RS_ASSERT(%s) failed.\n", file,line,cond ); +#else + fprintf( stderr,"%s:%u: %s: RS_ASSERT(%s) failed.\n", file,line,fn,cond ); +#endif + fflush( stderr ); + + errno = save_errno; + /* $NO_RS_ASSERT > 0: ignore rs_asserts, 0: abort, < 0: just warn */ + if ( !no_assert /* from $NO_RS_ASSERT */ ) abort(); +} + +/* SlapReply is consistent */ +void +(rs_assert_ok)( const SlapReply *rs ) +{ + const slap_mask_t flags = rs->sr_flags; + + if ( flags & REP_ENTRY_MASK ) { + RS_ASSERT( !(flags & REP_ENTRY_MUSTRELEASE) + || !(flags & (REP_ENTRY_MASK ^ REP_ENTRY_MUSTRELEASE)) ); + RS_ASSERT( rs->sr_entry != NULL ); + RS_ASSERT( (1 << rs->sr_type) & + ((1 << REP_SEARCH) | (1 << REP_SEARCHREF) | + (1 << REP_RESULT) | (1 << REP_GLUE_RESULT)) ); + } +#if defined(USE_RS_ASSERT) && (USE_RS_ASSERT) > 1 /* TODO: Enable when safe */ + if ( (flags & (REP_MATCHED_MASK | REP_REF_MASK | REP_CTRLS_MASK)) ) { + RS_ASSERT( !(flags & REP_MATCHED_MASK) || rs->sr_matched ); + RS_ASSERT( !(flags & REP_CTRLS_MASK ) || rs->sr_ctrls ); + /* Note: LDAP_REFERRAL + !sr_ref is OK, becomes LDAP_NO_SUCH_OBJECT */ + } +#if (USE_RS_ASSERT) > 2 + if ( rs->sr_err == LDAP_SUCCESS ) { + RS_ASSERT( rs->sr_text == NULL ); + RS_ASSERT( rs->sr_matched == NULL ); + } +#endif +#endif +} + +/* Ready for calling a new backend operation */ +void +(rs_assert_ready)( const SlapReply *rs ) +{ + RS_ASSERT( !rs->sr_entry ); +#if defined(USE_RS_ASSERT) && (USE_RS_ASSERT) > 1 /* TODO: Enable when safe */ + RS_ASSERT( !rs->sr_text ); + RS_ASSERT( !rs->sr_ref ); + RS_ASSERT( !rs->sr_matched ); + RS_ASSERT( !rs->sr_ctrls ); + RS_ASSERT( !rs->sr_flags ); +#if (USE_RS_ASSERT) > 2 + RS_ASSERT( rs->sr_err == LDAP_SUCCESS ); +#endif +#else + RS_ASSERT( !(rs->sr_flags & REP_ENTRY_MASK) ); +#endif +} + +/* Backend operation done */ +void +(rs_assert_done)( const SlapReply *rs ) +{ +#if defined(USE_RS_ASSERT) && (USE_RS_ASSERT) > 1 /* TODO: Enable when safe */ + RS_ASSERT( !(rs->sr_flags & ~(REP_ENTRY_MODIFIABLE|REP_NO_OPERATIONALS)) ); + rs_assert_ok( rs ); +#else + RS_ASSERT( !(rs->sr_flags & REP_ENTRY_MUSTFLUSH) ); +#endif +} + +#endif /* LDAP_TEST || USE_RS_ASSERT */ + +/* Reset a used SlapReply whose contents has been flushed (freed/released) */ +void +(rs_reinit)( SlapReply *rs, slap_reply_t type ) +{ + rs_reinit( rs, type ); /* proto-slap.h macro */ +} + +/* Obey and clear rs->sr_flags & REP_ENTRY_MASK. Clear sr_entry if freed. */ +void +rs_flush_entry( Operation *op, SlapReply *rs, slap_overinst *on ) +{ + rs_assert_ok( rs ); + + if ( (rs->sr_flags & REP_ENTRY_MUSTFLUSH) && rs->sr_entry != NULL ) { + if ( !(rs->sr_flags & REP_ENTRY_MUSTRELEASE) ) { + entry_free( rs->sr_entry ); + } else if ( on != NULL ) { + overlay_entry_release_ov( op, rs->sr_entry, 0, on ); + } else { + be_entry_release_rw( op, rs->sr_entry, 0 ); + } + rs->sr_entry = NULL; + } + + rs->sr_flags &= ~REP_ENTRY_MASK; +} + +/* Set rs->sr_entry after obeying and clearing sr_flags & REP_ENTRY_MASK. */ +void +rs_replace_entry( Operation *op, SlapReply *rs, slap_overinst *on, Entry *e ) +{ + rs_flush_entry( op, rs, on ); + rs->sr_entry = e; +} + +/* + * Ensure rs->sr_entry is modifiable, by duplicating it if necessary. + * Obey sr_flags. Set REP_ENTRY_. + * Return nonzero if rs->sr_entry was replaced. + */ +int +rs_entry2modifiable( Operation *op, SlapReply *rs, slap_overinst *on ) +{ + if ( rs->sr_flags & REP_ENTRY_MODIFIABLE ) { + rs_assert_ok( rs ); + return 0; + } + rs_replace_entry( op, rs, on, entry_dup( rs->sr_entry )); + rs->sr_flags |= REP_ENTRY_MODIFIABLE | REP_ENTRY_MUSTBEFREED; + return 1; +} + static long send_ldap_ber( - Connection *conn, + Operation *op, BerElement *ber ) { + Connection *conn = op->o_conn; ber_len_t bytes; long ret = 0; - int closing = 0; ber_get_option( ber, LBER_OPT_BER_BYTES_TO_WRITE, &bytes ); /* write only one pdu at a time - wait til it's our turn */ ldap_pvt_thread_mutex_lock( &conn->c_write1_mutex ); - while ( conn->c_writers > 0 ) { + if (( op->o_abandon && !op->o_cancel ) || !connection_valid( conn ) || + conn->c_writers < 0 ) { + ldap_pvt_thread_mutex_unlock( &conn->c_write1_mutex ); + return 0; + } + + conn->c_writers++; + + while ( conn->c_writers > 0 && conn->c_writing ) { ldap_pvt_thread_cond_wait( &conn->c_write1_cv, &conn->c_write1_mutex ); } + /* connection was closed under us */ if ( conn->c_writers < 0 ) { - closing = 1; /* we're the last waiter, let the closer continue */ if ( conn->c_writers == -1 ) ldap_pvt_thread_cond_signal( &conn->c_write1_cv ); + conn->c_writers++; + ldap_pvt_thread_mutex_unlock( &conn->c_write1_mutex ); + return 0; } - conn->c_writers++; - ldap_pvt_thread_mutex_unlock( &conn->c_write1_mutex ); - - if ( closing ) - return 0; + /* Our turn */ + conn->c_writing = 1; /* write the pdu */ while( 1 ) { int err; /* lock the connection */ - ldap_pvt_thread_mutex_lock( &conn->c_mutex ); + if ( ldap_pvt_thread_mutex_trylock( &conn->c_mutex )) { + if ( !connection_valid(conn)) { + ret = 0; + break; + } + ldap_pvt_thread_mutex_unlock( &conn->c_write1_mutex ); + ldap_pvt_thread_mutex_lock( &conn->c_write1_mutex ); + if ( conn->c_writers < 0 ) { + ret = 0; + break; + } + continue; + } if ( ber_flush2( conn->c_sb, ber, LBER_FLUSH_FREE_NEVER ) == 0 ) { ldap_pvt_thread_mutex_unlock( &conn->c_mutex ); @@ -186,33 +353,33 @@ static long send_ldap_ber( err, sock_errstr(err), 0 ); if ( err != EWOULDBLOCK && err != EAGAIN ) { + conn->c_writers--; + conn->c_writing = 0; + ldap_pvt_thread_mutex_unlock( &conn->c_write1_mutex ); connection_closing( conn, "connection lost on write" ); ldap_pvt_thread_mutex_unlock( &conn->c_mutex ); - - ret = -1; - break; + return -1; } /* wait for socket to be write-ready */ ldap_pvt_thread_mutex_lock( &conn->c_write2_mutex ); conn->c_writewaiter = 1; - slapd_set_write( conn->c_sd, 1 ); + slapd_set_write( conn->c_sd, 2 ); + ldap_pvt_thread_mutex_unlock( &conn->c_write1_mutex ); ldap_pvt_thread_mutex_unlock( &conn->c_mutex ); ldap_pvt_thread_cond_wait( &conn->c_write2_cv, &conn->c_write2_mutex ); conn->c_writewaiter = 0; ldap_pvt_thread_mutex_unlock( &conn->c_write2_mutex ); ldap_pvt_thread_mutex_lock( &conn->c_write1_mutex ); - closing = ( conn->c_writers < 0 ); - ldap_pvt_thread_mutex_unlock( &conn->c_write1_mutex ); - if ( closing ) { + if ( conn->c_writers < 0 ) { ret = 0; break; } } - ldap_pvt_thread_mutex_lock( &conn->c_write1_mutex ); + conn->c_writing = 0; if ( conn->c_writers < 0 ) { conn->c_writers++; if ( !conn->c_writers ) @@ -282,11 +449,11 @@ send_ldap_controls( Operation *o, BerElement *ber, LDAPControl **c ) ber_printf( sber, "{e}", LDAP_UNWILLING_TO_PERFORM ); - if( ber_flatten2( ber, &sorted.ldctl_value, 0 ) == -1 ) { + if( ber_flatten2( sber, &sorted.ldctl_value, 0 ) == -1 ) { return -1; } - (void) ber_free_buf( ber ); + (void) ber_free_buf( sber ); rc = send_ldap_control( ber, &sorted ); if( rc == -1 ) return rc; @@ -403,7 +570,7 @@ send_ldap_response( int rc = LDAP_SUCCESS; long bytes; - if ( rs->sr_err == SLAPD_ABANDON || op->o_abandon ) { + if (( rs->sr_err == SLAPD_ABANDON || op->o_abandon ) && !op->o_cancel ) { rc = SLAPD_ABANDON; goto clean2; } @@ -425,9 +592,13 @@ send_ldap_response( ber_set_option( ber, LBER_OPT_BER_MEMCTX, &op->o_tmpmemctx ); } + rc = rs->sr_err; + if ( rc == SLAPD_ABANDON && op->o_cancel ) + rc = LDAP_CANCELLED; + Debug( LDAP_DEBUG_TRACE, "send_ldap_response: msgid=%d tag=%lu err=%d\n", - rs->sr_msgid, rs->sr_tag, rs->sr_err ); + rs->sr_msgid, rs->sr_tag, rc ); if( rs->sr_ref ) { Debug( LDAP_DEBUG_ARGS, "send_ldap_response: ref=\"%s\"\n", @@ -440,7 +611,7 @@ send_ldap_response( op->o_protocol == LDAP_VERSION2 ) { rc = ber_printf( ber, "t{ess" /*"}"*/, - rs->sr_tag, rs->sr_err, + rs->sr_tag, rc, rs->sr_matched == NULL ? "" : rs->sr_matched, rs->sr_text == NULL ? "" : rs->sr_text ); } else @@ -451,7 +622,7 @@ send_ldap_response( } else { rc = ber_printf( ber, "{it{ess" /*"}}"*/, - rs->sr_msgid, rs->sr_tag, rs->sr_err, + rs->sr_msgid, rs->sr_tag, rc, rs->sr_matched == NULL ? "" : rs->sr_matched, rs->sr_text == NULL ? "" : rs->sr_text ); } @@ -521,7 +692,7 @@ send_ldap_response( } /* send BER */ - bytes = send_ldap_ber( op->o_conn, ber ); + bytes = send_ldap_ber( op, ber ); #ifdef LDAP_CONNECTIONLESS if (!op->o_conn || op->o_conn->c_is_udp == 0) #endif @@ -589,13 +760,17 @@ send_ldap_disconnect( Operation *op, SlapReply *rs ) || (e) == LDAP_STRONG_AUTH_REQUIRED \ || (e) == LDAP_UNAVAILABLE ) - assert( LDAP_UNSOLICITED_ERROR( rs->sr_err ) ); - - rs->sr_type = REP_EXTENDED; - Debug( LDAP_DEBUG_TRACE, "send_ldap_disconnect %d:%s\n", rs->sr_err, rs->sr_text ? rs->sr_text : "", NULL ); + assert( LDAP_UNSOLICITED_ERROR( rs->sr_err ) ); + + /* TODO: Flush the entry if sr_type == REP_SEARCH/REP_SEARCHREF? */ + RS_ASSERT( !(rs->sr_flags & REP_ENTRY_MASK) ); + rs->sr_flags &= ~REP_ENTRY_MASK; /* paranoia */ + + rs->sr_type = REP_EXTENDED; + rs->sr_rspdata = NULL; if ( op->o_protocol < LDAP_VERSION3 ) { rs->sr_rspoid = NULL; @@ -629,25 +804,20 @@ slap_send_ldap_result( Operation *op, SlapReply *rs ) if ( rs->sr_err == SLAPD_ABANDON || op->o_abandon ) goto abandon; - assert( !LDAP_API_ERROR( rs->sr_err ) ); - Debug( LDAP_DEBUG_TRACE, "send_ldap_result: %s p=%d\n", op->o_log_prefix, op->o_protocol, 0 ); - Debug( LDAP_DEBUG_ARGS, "send_ldap_result: err=%d matched=\"%s\" text=\"%s\"\n", rs->sr_err, rs->sr_matched ? rs->sr_matched : "", rs->sr_text ? rs->sr_text : "" ); - - if( rs->sr_ref ) { Debug( LDAP_DEBUG_ARGS, "send_ldap_result: referral=\"%s\"\n", rs->sr_ref[0].bv_val ? rs->sr_ref[0].bv_val : "NULL", NULL, NULL ); } - + assert( !LDAP_API_ERROR( rs->sr_err ) ); assert( rs->sr_err != LDAP_PARTIAL_RESULTS ); if ( rs->sr_err == LDAP_REFERRAL ) { @@ -700,11 +870,14 @@ abandon: void send_ldap_sasl( Operation *op, SlapReply *rs ) { - rs->sr_type = REP_SASL; Debug( LDAP_DEBUG_TRACE, "send_ldap_sasl: err=%d len=%ld\n", rs->sr_err, rs->sr_sasldata ? (long) rs->sr_sasldata->bv_len : -1, NULL ); + RS_ASSERT( !(rs->sr_flags & REP_ENTRY_MASK) ); + rs->sr_flags &= ~REP_ENTRY_MASK; /* paranoia */ + + rs->sr_type = REP_SASL; rs->sr_tag = slap_req2res( op->o_tag ); rs->sr_msgid = (rs->sr_tag != LBER_SEQUENCE) ? op->o_msgid : 0; @@ -719,14 +892,16 @@ send_ldap_sasl( Operation *op, SlapReply *rs ) void slap_send_ldap_extended( Operation *op, SlapReply *rs ) { - rs->sr_type = REP_EXTENDED; - Debug( LDAP_DEBUG_TRACE, "send_ldap_extended: err=%d oid=%s len=%ld\n", rs->sr_err, rs->sr_rspoid ? rs->sr_rspoid : "", rs->sr_rspdata != NULL ? rs->sr_rspdata->bv_len : 0 ); + RS_ASSERT( !(rs->sr_flags & REP_ENTRY_MASK) ); + rs->sr_flags &= ~REP_ENTRY_MASK; /* paranoia */ + + rs->sr_type = REP_EXTENDED; rs->sr_tag = slap_req2res( op->o_tag ); rs->sr_msgid = (rs->sr_tag != LBER_SEQUENCE) ? op->o_msgid : 0; @@ -741,12 +916,16 @@ slap_send_ldap_extended( Operation *op, SlapReply *rs ) void slap_send_ldap_intermediate( Operation *op, SlapReply *rs ) { - rs->sr_type = REP_INTERMEDIATE; Debug( LDAP_DEBUG_TRACE, "send_ldap_intermediate: err=%d oid=%s len=%ld\n", rs->sr_err, rs->sr_rspoid ? rs->sr_rspoid : "", rs->sr_rspdata != NULL ? rs->sr_rspdata->bv_len : 0 ); + + RS_ASSERT( !(rs->sr_flags & REP_ENTRY_MASK) ); + rs->sr_flags &= ~REP_ENTRY_MASK; /* paranoia */ + + rs->sr_type = REP_INTERMEDIATE; rs->sr_tag = LDAP_RES_INTERMEDIATE; rs->sr_msgid = op->o_msgid; if ( send_ldap_response( op, rs ) == SLAP_CB_CONTINUE ) { @@ -902,7 +1081,7 @@ slap_send_search_entry( Operation *op, SlapReply *rs ) if( e_flags == NULL ) { Debug( LDAP_DEBUG_ANY, "send_search_entry: conn %lu slap_sl_calloc failed\n", - op->o_connid ? op->o_connid : 0, 0, 0 ); + op->o_connid, 0, 0 ); ber_free( ber, 1 ); send_ldap_error( op, rs, LDAP_OTHER, "out of memory" ); @@ -920,7 +1099,7 @@ slap_send_search_entry( Operation *op, SlapReply *rs ) if ( rc == -1 ) { Debug( LDAP_DEBUG_ANY, "send_search_entry: " "conn %lu matched values filtering failed\n", - op->o_connid ? op->o_connid : 0, 0, 0 ); + op->o_connid, 0, 0 ); if ( op->o_res_ber == NULL ) ber_free_buf( ber ); send_ldap_error( op, rs, LDAP_OTHER, "matched values filtering error" ); @@ -1087,7 +1266,7 @@ slap_send_search_entry( Operation *op, SlapReply *rs ) Debug( LDAP_DEBUG_ANY, "send_search_entry: conn %lu " "matched values filtering failed\n", - op->o_connid ? op->o_connid : 0, 0, 0); + op->o_connid, 0, 0); if ( op->o_res_ber == NULL ) ber_free_buf( ber ); send_ldap_error( op, rs, LDAP_OTHER, "matched values filtering error" ); @@ -1225,14 +1404,10 @@ slap_send_search_entry( Operation *op, SlapReply *rs ) Statslog( LDAP_DEBUG_STATS2, "%s ENTRY dn=\"%s\"\n", op->o_log_prefix, rs->sr_entry->e_nname.bv_val, 0, 0, 0 ); - if ( rs->sr_flags & REP_ENTRY_MUSTRELEASE ) { - be_entry_release_rw( op, rs->sr_entry, 0 ); - rs->sr_flags ^= REP_ENTRY_MUSTRELEASE; - rs->sr_entry = NULL; - } + rs_flush_entry( op, rs, NULL ); if ( op->o_res_ber == NULL ) { - bytes = send_ldap_ber( op->o_conn, ber ); + bytes = send_ldap_ber( op, ber ); ber_free_buf( ber ); if ( bytes < 0 ) { @@ -1266,25 +1441,25 @@ error_return:; slap_sl_free( e_flags, op->o_tmpmemctx ); } + /* FIXME: Can break if rs now contains an extended response */ if ( rs->sr_operational_attrs ) { attrs_free( rs->sr_operational_attrs ); rs->sr_operational_attrs = NULL; } rs->sr_attr_flags = SLAP_ATTRS_UNDEFINED; - /* FIXME: I think rs->sr_type should be explicitly set to - * REP_SEARCH here. That's what it was when we entered this - * function. send_ldap_error may have changed it, but we - * should set it back so that the cleanup functions know - * what they're doing. - */ - if ( op->o_tag == LDAP_REQ_SEARCH && rs->sr_type == REP_SEARCH - && rs->sr_entry - && ( rs->sr_flags & REP_ENTRY_MUSTBEFREED ) ) - { - entry_free( rs->sr_entry ); - rs->sr_entry = NULL; - rs->sr_flags &= ~REP_ENTRY_MUSTBEFREED; + if ( op->o_tag == LDAP_REQ_SEARCH && rs->sr_type == REP_SEARCH ) { + rs_flush_entry( op, rs, NULL ); + } else { + RS_ASSERT( (rs->sr_flags & REP_ENTRY_MASK) == 0 ); + } + + if ( rs->sr_flags & REP_CTRLS_MUSTBEFREED ) { + rs->sr_flags ^= REP_CTRLS_MUSTBEFREED; /* paranoia */ + if ( rs->sr_ctrls ) { + slap_free_ctrls( op, rs->sr_ctrls ); + rs->sr_ctrls = NULL; + } } return( rc ); @@ -1395,17 +1570,12 @@ slap_send_search_reference( Operation *op, SlapReply *rs ) } rc = 0; - if ( rs->sr_flags & REP_ENTRY_MUSTRELEASE ) { - assert( rs->sr_entry != NULL ); - be_entry_release_rw( op, rs->sr_entry, 0 ); - rs->sr_flags ^= REP_ENTRY_MUSTRELEASE; - rs->sr_entry = NULL; - } + rs_flush_entry( op, rs, NULL ); #ifdef LDAP_CONNECTIONLESS if (!op->o_conn || op->o_conn->c_is_udp == 0) { #endif - bytes = send_ldap_ber( op->o_conn, ber ); + bytes = send_ldap_ber( op, ber ); ber_free_buf( ber ); if ( bytes < 0 ) { @@ -1441,6 +1611,14 @@ rel: (void)slap_cleanup_play( op, rs ); } + if ( rs->sr_flags & REP_CTRLS_MUSTBEFREED ) { + rs->sr_flags ^= REP_CTRLS_MUSTBEFREED; /* paranoia */ + if ( rs->sr_ctrls ) { + slap_free_ctrls( op, rs->sr_ctrls ); + rs->sr_ctrls = NULL; + } + } + return rc; } @@ -1645,12 +1823,11 @@ slap_attr_flags( AttributeName *an ) flags |= ( SLAP_OPATTRS_NO | SLAP_USERATTRS_YES ); } else { - flags |= an_find( an, &AllOper ) + flags |= an_find( an, slap_bv_all_operational_attrs ) ? SLAP_OPATTRS_YES : SLAP_OPATTRS_NO; - flags |= an_find( an, &AllUser ) + flags |= an_find( an, slap_bv_all_user_attrs ) ? SLAP_USERATTRS_YES : SLAP_USERATTRS_NO; } return flags; } -