openldap/servers/slapd/slapi/slapi_overlay.c

/* slapi_overlay.c - SLAPI overlay */
/* $OpenLDAP$ */
/* This work is part of OpenLDAP Software <http://www.openldap.org/>.
 *
 * Copyright 2001-2005 The OpenLDAP Foundation.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted only as authorized by the OpenLDAP
 * Public License.
 *
 * A copy of this license is available in the file LICENSE in the
 * top-level directory of the distribution or, alternatively, at
 * <http://www.OpenLDAP.org/license.html>.
 */
/* ACKNOWLEDGEMENTS:
 * This work was initially developed by Luke Howard for inclusion
 * in OpenLDAP Software.
 */

#include "portable.h"

#include <stdio.h>

#include <ac/string.h>
#include <ac/socket.h>

#include "slap.h"
#include "slapi.h"

#ifdef LDAP_SLAPI

static slap_overinst slapi;

static int slapi_over_response( Operation *op, SlapReply *rs );
static int slapi_over_cleanup( Operation *op, SlapReply *rs );

static Slapi_PBlock *
slapi_over_pblock_new ( Operation *op )
{
	Slapi_PBlock		*pb;

	pb = slapi_pblock_new();
	if ( pb == NULL ) {
		return NULL;
	}

	slapi_int_pblock_set_operation( pb, op );

	return pb;
}

static int
slapi_op_internal_p( Operation *op, slap_callback *cb )
{
	int			internal_op = 0;
	Slapi_PBlock		*pb = NULL;
	slap_callback		*pcb;

	/* Abstraction violating check for SLAPI internal operations */
	/* allows pblock to remain consistent when invoking internal op plugins */
	for ( pcb = op->o_callback; pcb != NULL; pcb = pcb->sc_next ) {
		if ( pcb->sc_response == slapi_int_response ) {
			pb = (Slapi_PBlock *)pcb->sc_private;
			internal_op = 1;
			break;
		}
	}

	if ( cb != NULL ) {
		if ( !internal_op ) {
			cb->sc_response = slapi_over_response;
			cb->sc_cleanup = slapi_over_cleanup;
		} else {
			cb->sc_response = NULL;
			cb->sc_cleanup = NULL;
		}

		if ( pb == NULL ) {
			pb = slapi_over_pblock_new( op );
		}

		cb->sc_private = pb;
		cb->sc_next = op->o_callback;
		op->o_callback = cb;
	}

	return internal_op;
}

static int
slapi_over_compute_output(
	computed_attr_context *c,
	Slapi_Attr *attribute,
	Slapi_Entry *entry
)
{
	Attribute		**a;
	AttributeDescription	*desc;
	SlapReply		*rs = (SlapReply *)c->cac_private;

	if ( c == NULL || attribute == NULL || entry == NULL ) {
		return 0;
	}

	assert( rs->sr_entry == entry );

	desc = attribute->a_desc;

	if ( rs->sr_attrs == NULL ) {
		/* All attrs request, skip operational attributes */
		if ( is_at_operational( desc->ad_type ) ) {
			return 0;
		}
	} else {
		/* Specific attributes requested */
		if ( is_at_operational( desc->ad_type ) ) {
			if ( !SLAP_OPATTRS( rs->sr_attr_flags ) &&
			     !ad_inlist( desc, rs->sr_attrs ) ) {
				return 0;
			}
		} else {
			if ( !SLAP_USERATTRS( rs->sr_attr_flags ) &&
			     !ad_inlist( desc, rs->sr_attrs ) ) {
				return 0;
			}
		}
	}

	/* XXX perhaps we should check for existing attributes and merge */
	for ( a = &rs->sr_operational_attrs; *a != NULL; a = &(*a)->a_next )
		;

	*a = attr_dup( attribute );

	return 0;
}

static int
slapi_over_aux_operational( Operation *op, SlapReply *rs )
{
	/* Support for computed attribute plugins */
	computed_attr_context    ctx;
	AttributeName		*anp;

	if ( slapi_op_internal_p( op, NULL ) ) {
		return SLAP_CB_CONTINUE;
	}

	ctx.cac_pb = slapi_over_pblock_new( op );
	ctx.cac_op = op;
	ctx.cac_private = rs;

	if ( rs->sr_entry != NULL ) {
		/*
		 * For each client requested attribute, call the plugins.
		 */
		if ( rs->sr_attrs != NULL ) {
			for ( anp = rs->sr_attrs; anp->an_name.bv_val != NULL; anp++ ) {
				if ( compute_evaluator( &ctx, anp->an_name.bv_val,
					rs->sr_entry, slapi_over_compute_output ) == 1 ) {
					break;
				}
			}
		} else {
			/*
			 * Technically we shouldn't be returning operational attributes
			 * when the user requested only user attributes. We'll let the
			 * plugin decide whether to be naughty or not.
			 */
			compute_evaluator( &ctx, "*", rs->sr_entry, slapi_over_compute_output );
		}
	}

	slapi_pblock_destroy( ctx.cac_pb );

	return SLAP_CB_CONTINUE;
}

static int
slapi_over_search( Operation *op, SlapReply *rs, int type )
{
	int			rc;
	Slapi_PBlock		*pb;

	assert( rs->sr_type == REP_SEARCH || rs->sr_type == REP_SEARCHREF );

	/* create a new pblock to not trample on result controls */
	pb = slapi_over_pblock_new( op );

	slapi_pblock_set( pb, SLAPI_RESCONTROLS,         (void *)rs->sr_ctrls );
	slapi_pblock_set( pb, SLAPI_SEARCH_RESULT_ENTRY, (void *)rs->sr_entry );

	rc = slapi_int_call_plugins( op->o_bd, type, pb );
	if ( rc >= 0 ) /* 1 means no plugins called */
		rc = SLAP_CB_CONTINUE;
	else
		rc = LDAP_SUCCESS; /* confusing: don't abort, but don't send */

	slapi_pblock_set( pb, SLAPI_RESCONTROLS, NULL ); /* don't free */
	slapi_pblock_destroy(pb);

	return rc;
}

static int
slapi_over_count_controls( LDAPControl **controls )
{
	int			i;

	if ( controls == NULL )
		return 0;

	for ( i = 0; controls[i] != NULL; i++ )
		;

	return i;
}

static int
slapi_over_merge_controls( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );
	LDAPControl		**slapiControls = NULL, **mergedControls;
	int			nSlapiControls = 0;
	int			nResControls = 0;
	int			i;

	slapi_pblock_get( pb, SLAPI_RESCONTROLS, (void **)&slapiControls );
	nSlapiControls = slapi_over_count_controls( slapiControls );

	if ( nSlapiControls == 0 ) {
		/* short-circuit */
		return LDAP_SUCCESS;
	}

	nResControls = slapi_over_count_controls( rs->sr_ctrls );

	/* XXX this is a bit tricky, rs->sr_ctrls may have been allocated on stack */
	mergedControls = (LDAPControl **)op->o_tmpalloc( ( nResControls + nSlapiControls + 1 ) *
							 sizeof( LDAPControl *), op->o_tmpmemctx );
	if ( mergedControls == NULL ) {
		return LDAP_NO_MEMORY;
	}

	if ( rs->sr_ctrls != NULL ) {
		for ( i = 0; i < nResControls; i++ )
			mergedControls[i] = rs->sr_ctrls[i];
	}
	if ( slapiControls != NULL ) {
		for ( i = 0; i < nSlapiControls; i++ )
			mergedControls[nResControls + i] = slapiControls[i];
	}
	mergedControls[nResControls + nSlapiControls] = NULL;

	if ( slapiControls != NULL ) {
		slapi_ch_free( (void **)&slapiControls );
		slapi_pblock_set( pb, SLAPI_RESCONTROLS, NULL ); /* don't free */
	}

	rs->sr_ctrls = mergedControls;

	return LDAP_SUCCESS;
}

/*
 * Call pre- and post-result plugins
 */
static int
slapi_over_result( Operation *op, SlapReply *rs, int type )
{
	int			rc;
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );

	assert( rs->sr_type == REP_RESULT );

	slapi_pblock_set( pb, SLAPI_RESULT_CODE,    (void *)rs->sr_err );
	slapi_pblock_set( pb, SLAPI_RESULT_TEXT,    (void *)rs->sr_text );
	slapi_pblock_set( pb, SLAPI_RESULT_MATCHED, (void *)rs->sr_matched );

	rc = slapi_int_call_plugins( op->o_bd, type, pb );
	
	slapi_pblock_get( pb, SLAPI_RESULT_CODE,    (void **)&rs->sr_err );
	slapi_pblock_get( pb, SLAPI_RESULT_TEXT,    (void **)&rs->sr_text );
	slapi_pblock_get( pb, SLAPI_RESULT_MATCHED, (void **)&rs->sr_matched );

	if ( type == SLAPI_PLUGIN_PRE_RESULT_FN ) {
		rc = slapi_over_merge_controls( op, rs );
	}

	return SLAP_CB_CONTINUE;
}

static int
slapi_op_add_init( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );

	slapi_pblock_set( pb, SLAPI_ADD_ENTRY, (void *)op->ora_e );

	return LDAP_SUCCESS;
}

static int
slapi_op_bind_preop_init( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );

	slapi_pblock_set( pb, SLAPI_BIND_TARGET,      (void *)op->o_req_dn.bv_val );
	slapi_pblock_set( pb, SLAPI_BIND_METHOD,      (void *)op->orb_method );
	slapi_pblock_set( pb, SLAPI_BIND_CREDENTIALS, (void *)&op->orb_cred );
	slapi_pblock_set( pb, SLAPI_CONN_DN,          NULL );

	return LDAP_SUCCESS;
}

static int
slapi_op_bind_postop_init( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );

	if ( rs->sr_err == LDAP_SUCCESS ) {
		/* fix for ITS#2971 */
		slapi_pblock_set( pb, SLAPI_CONN_DN, op->o_conn->c_authz.sai_dn.bv_val );
	}

	return LDAP_SUCCESS;
}

static int
slapi_op_bind_callback( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );
	int			rc = rs->sr_err;

	switch ( rc ) {
	case SLAPI_BIND_SUCCESS:
		/* Continue with backend processing */
		break;
	case SLAPI_BIND_FAIL:
		/* Failure, frontend (that's us) sends result */
		rs->sr_err = LDAP_INVALID_CREDENTIALS;
		send_ldap_result( op, rs );
		return rs->sr_err;
		break;
	case SLAPI_BIND_ANONYMOUS: /* undocumented */
	default: /* plugin sent result or no plugins called */
		if ( slapi_pblock_get( pb, SLAPI_RESULT_CODE, (void **)&rs->sr_err ) != 0 ) {
			rs->sr_err = LDAP_OTHER;
		}

		BER_BVZERO( &op->orb_edn );

		if ( rs->sr_err == LDAP_SUCCESS ) {
			slapi_pblock_get( pb, SLAPI_CONN_DN, (void *)&op->orb_edn.bv_val );
			if ( BER_BVISNULL( &op->orb_edn ) ) {
				if ( rc == 1 ) {
					/* No plugins were called; continue processing */
					return LDAP_SUCCESS;
				}
			} else {
				op->orb_edn.bv_len = strlen( op->orb_edn.bv_val );
			}
			rs->sr_err = dnPrettyNormal( NULL, &op->orb_edn,
				&op->o_req_dn, &op->o_req_ndn, op->o_tmpmemctx );

			ldap_pvt_thread_mutex_lock( &op->o_conn->c_mutex );
			ber_dupbv(&op->o_conn->c_dn, &op->o_req_dn);
			ber_dupbv(&op->o_conn->c_ndn, &op->o_req_ndn);
			op->o_tmpfree( op->o_req_dn.bv_val, op->o_tmpmemctx );
			BER_BVZERO( &op->o_req_dn );
			op->o_tmpfree( op->o_req_ndn.bv_val, op->o_tmpmemctx );
			BER_BVZERO( &op->o_req_ndn );
			if ( !BER_BVISEMPTY( &op->o_conn->c_dn ) ) {
				ber_len_t max = sockbuf_max_incoming_auth;
				ber_sockbuf_ctrl( op->o_conn->c_sb,
					LBER_SB_OPT_SET_MAX_INCOMING, &max );
			}

			/* log authorization identity */
			Statslog( LDAP_DEBUG_STATS,
				"%s BIND dn=\"%s\" mech=%s (SLAPI) ssf=0\n",
				op->o_log_prefix,
				BER_BVISNULL( &op->o_conn->c_dn )
					? "<empty>" : op->o_conn->c_dn.bv_val,
				op->orb_tmp_mech.bv_val, 0, 0 );

			ldap_pvt_thread_mutex_unlock( &op->o_conn->c_mutex );
			return -1;
		}
		break;
	}

	return rc;
}

static int
slapi_op_compare_init( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );

	slapi_pblock_set( pb, SLAPI_COMPARE_TYPE,  (void *)op->orc_ava->aa_desc->ad_cname.bv_val );
	slapi_pblock_set( pb, SLAPI_COMPARE_VALUE, (void *)&op->orc_ava->aa_value );

	return LDAP_SUCCESS;
}

static int
slapi_op_modify_init( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );
	LDAPMod			**modv = NULL;

	modv = slapi_int_modifications2ldapmods( &op->orm_modlist );
	slapi_pblock_set( pb, SLAPI_MODIFY_MODS, (void *)modv );

	return LDAP_SUCCESS;
}

static int
slapi_op_modify_callback( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );
	LDAPMod			**modv = NULL;

	/* check preoperation result code */
	if ( rs->sr_err < 0 ) {
		slapi_pblock_get( pb, SLAPI_RESULT_CODE, (void **)&rs->sr_err );
		return rs->sr_err;
	}

	/*
	 * NB: it is valid for the plugin to return no modifications
	 * (for example, a plugin might store some attributes elsewhere
	 * and remove them from the modification list; if only those
	 * attribute types were included in the modification request,
	 * then slapi_int_ldapmods2modifications() above will return
	 * NULL).
	 *
	 * However, the post-operation plugin should still be
	 * called.
	 */

	slapi_pblock_get( pb, SLAPI_MODIFY_MODS, (void **)&modv );
	op->orm_modlist = slapi_int_ldapmods2modifications( modv );

	return LDAP_SUCCESS;
}

static int
slapi_op_modify_cleanup( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );
	LDAPMod			**modv = NULL;

	slapi_pblock_get( pb, SLAPI_MODIFY_MODS, (void **)&modv );

	if ( modv != NULL )
		slapi_int_free_ldapmods( modv );

	return LDAP_SUCCESS;
}

static int
slapi_op_modrdn_init( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );

	slapi_pblock_set( pb, SLAPI_MODRDN_NEWRDN,      (void *)op->orr_newrdn.bv_val );
	slapi_pblock_set( pb, SLAPI_MODRDN_NEWSUPERIOR, (void *)op->orr_newSup->bv_val );
	slapi_pblock_set( pb, SLAPI_MODRDN_DELOLDRDN,   (void *)op->orr_deleteoldrdn );

	return LDAP_SUCCESS;
}

static int
slapi_op_search_init( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );
	char			**attrs;

	attrs = anlist2charray_x( op->ors_attrs, 0, op->o_tmpmemctx );

	slapi_pblock_set( pb, SLAPI_SEARCH_SCOPE,     (void *)op->ors_scope );
	slapi_pblock_set( pb, SLAPI_SEARCH_DEREF,     (void *)op->ors_deref );
	slapi_pblock_set( pb, SLAPI_SEARCH_SIZELIMIT, (void *)op->ors_slimit );
	slapi_pblock_set( pb, SLAPI_SEARCH_TIMELIMIT, (void *)op->ors_tlimit );
	slapi_pblock_set( pb, SLAPI_SEARCH_FILTER,    (void *)op->ors_filter );
	slapi_pblock_set( pb, SLAPI_SEARCH_STRFILTER, (void *)op->ors_filterstr.bv_val );
	slapi_pblock_set( pb, SLAPI_SEARCH_ATTRS,     (void *)attrs );
	slapi_pblock_set( pb, SLAPI_SEARCH_ATTRSONLY, (void *)op->ors_attrsonly );

	return LDAP_SUCCESS;
}

static int
slapi_op_search_callback( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );

	/* check preoperation result code */
	if ( rs->sr_err < 0 ) {
		slapi_pblock_get( pb, SLAPI_RESULT_CODE, (void **)&rs->sr_err );
		return rs->sr_err;
	}

	if ( slapi_int_call_plugins( op->o_bd, SLAPI_PLUGIN_COMPUTE_SEARCH_REWRITER_FN, pb ) != 0 ) {
		return LDAP_SUCCESS;
	}

	/*
	 * The plugin can set the SLAPI_SEARCH_FILTER.
	 * SLAPI_SEARCH_STRFILER is not normative.
	 */
	slapi_pblock_get( pb, SLAPI_SEARCH_FILTER, (void *)&op->ors_filter );
	op->o_tmpfree( op->ors_filterstr.bv_val, op->o_tmpmemctx );
	filter2bv_x( op, op->ors_filter, &op->ors_filterstr );

	slapi_pblock_get( pb, SLAPI_SEARCH_TARGET, (void **)&op->o_req_dn.bv_val );
	op->o_req_dn.bv_len = strlen( op->o_req_dn.bv_val );

	if( !BER_BVISNULL( &op->o_req_ndn ) ) {
		slap_sl_free( op->o_req_ndn.bv_val, op->o_tmpmemctx );
	}
	rs->sr_err = dnNormalize( 0, NULL, NULL, &op->o_req_dn, &op->o_req_ndn,
				  op->o_tmpmemctx );
	if ( rs->sr_err != LDAP_SUCCESS ) {
		send_ldap_result( op, rs );
		return rs->sr_err;
	}

	slapi_pblock_get( pb, SLAPI_SEARCH_SCOPE, (void **)&op->ors_scope );
	slapi_pblock_get( pb, SLAPI_SEARCH_DEREF, (void **)&op->ors_deref );

	return LDAP_SUCCESS;
}

static int
slapi_op_search_cleanup( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb = SLAPI_OPERATION_PBLOCK( op );
	char			**attrs = NULL;

	slapi_pblock_get( pb, SLAPI_SEARCH_ATTRS,     (void *)&attrs );

	if ( attrs != NULL )
		op->o_tmpfree( attrs, op->o_tmpmemctx );

	return LDAP_SUCCESS;
}

struct slapi_op_info {
	int soi_preop;			/* preoperation plugin parameter */
	int soi_postop;			/* postoperation plugin parameter */
	int soi_internal_preop;		/* internal preoperation plugin parameter */
	int soi_internal_postop;	/* internal postoperation plugin parameter */
	slap_response *soi_preop_init;	/* preoperation pblock init function */
	slap_response *soi_callback; 	/* preoperation result handler */
	slap_response *soi_postop_init;	/* postoperation pblock init function */
	slap_response *soi_cleanup;	/* cleanup function (not for internal ops) */
} slapi_op_dispatch_table[] = {
	{
		SLAPI_PLUGIN_PRE_BIND_FN,
		SLAPI_PLUGIN_POST_BIND_FN,
		0,
		0,
		slapi_op_bind_preop_init,
		slapi_op_bind_callback,
		slapi_op_bind_postop_init,
		NULL
	},
	{
		SLAPI_PLUGIN_PRE_UNBIND_FN,
		SLAPI_PLUGIN_POST_UNBIND_FN,
		0,
		0,
		NULL,
		NULL,
		NULL,
		NULL
	},
	{
		SLAPI_PLUGIN_PRE_SEARCH_FN,
		SLAPI_PLUGIN_POST_SEARCH_FN,
		0,
		0,
		slapi_op_search_init,
		slapi_op_search_callback,
		NULL,
		slapi_op_search_cleanup
	},
	{
		SLAPI_PLUGIN_PRE_COMPARE_FN,
		SLAPI_PLUGIN_POST_COMPARE_FN,
		0,
		0,
		slapi_op_compare_init,
		NULL,
		NULL,
		NULL
	},
	{
		SLAPI_PLUGIN_PRE_MODIFY_FN,
		SLAPI_PLUGIN_POST_MODIFY_FN,
		SLAPI_PLUGIN_INTERNAL_PRE_MODIFY_FN,
		SLAPI_PLUGIN_INTERNAL_POST_MODIFY_FN,
		slapi_op_modify_init,
		slapi_op_modify_callback,
		NULL,
		slapi_op_modify_cleanup
	},
	{
		SLAPI_PLUGIN_PRE_MODRDN_FN,
		SLAPI_PLUGIN_POST_MODRDN_FN,
		SLAPI_PLUGIN_INTERNAL_PRE_MODRDN_FN,
		SLAPI_PLUGIN_INTERNAL_POST_MODRDN_FN,
		slapi_op_modrdn_init,
		NULL,
		NULL,
		NULL
	},
	{
		SLAPI_PLUGIN_PRE_ADD_FN,
		SLAPI_PLUGIN_POST_ADD_FN,
		SLAPI_PLUGIN_INTERNAL_PRE_ADD_FN,
		SLAPI_PLUGIN_INTERNAL_POST_ADD_FN,
		slapi_op_add_init,
		NULL,
		NULL,
		NULL
	},
	{
		SLAPI_PLUGIN_PRE_DELETE_FN,
		SLAPI_PLUGIN_POST_DELETE_FN,
		SLAPI_PLUGIN_INTERNAL_PRE_DELETE_FN,
		SLAPI_PLUGIN_INTERNAL_POST_DELETE_FN,
		NULL,
		NULL,
		NULL,
		NULL
	},
	{
		SLAPI_PLUGIN_PRE_ABANDON_FN,
		SLAPI_PLUGIN_POST_ABANDON_FN,
		0,
		0,
		NULL,
		NULL,
		NULL,
		NULL
	},
	{
		0,
		0,
		0,
		0,
		NULL,
		NULL,
		NULL,
		NULL
	}
};

slap_operation_t
slapi_tag2op( ber_tag_t tag )
{
	slap_operation_t op;

	switch ( tag ) {
	case LDAP_REQ_BIND:
		op = op_bind;
		break;
	case LDAP_REQ_ADD:
		op = op_add;
		break;
	case LDAP_REQ_DELETE:
		op = op_compare;
		break;
	case LDAP_REQ_MODRDN:
		op = op_modrdn;
		break;
	case LDAP_REQ_MODIFY:
		op = op_modify;
		break;
	case LDAP_REQ_COMPARE:
		op = op_compare;
		break;
	case LDAP_REQ_SEARCH:
		op = op_search;
		break;
	case LDAP_REQ_UNBIND:
		op = op_unbind;
		break;
	default:
		op = op_last;
		break;
	}

	return op;
}

static int
slapi_over_response( Operation *op, SlapReply *rs )
{
	int			rc;

	switch ( rs->sr_type ) {
	case REP_RESULT:
		rc = slapi_over_result( op, rs, SLAPI_PLUGIN_PRE_RESULT_FN );
		break;
	case REP_SEARCH:
		rc = slapi_over_search( op, rs, SLAPI_PLUGIN_PRE_ENTRY_FN );
		break;
	case REP_SEARCHREF:
		rc = slapi_over_search( op, rs, SLAPI_PLUGIN_PRE_REFERRAL_FN );
		break;
	default:
		rc = SLAP_CB_CONTINUE;
		break;
	}

	return rc;
}

static int
slapi_over_cleanup( Operation *op, SlapReply *rs )
{
	int			rc;

	switch ( rs->sr_type ) {
	case REP_RESULT:
		rc = slapi_over_result( op, rs, SLAPI_PLUGIN_POST_RESULT_FN );
		break;
	case REP_SEARCH:
		rc = slapi_over_search( op, rs, SLAPI_PLUGIN_POST_ENTRY_FN );
		break;
	case REP_SEARCHREF:
		rc = slapi_over_search( op, rs, SLAPI_PLUGIN_POST_REFERRAL_FN );
		break;
	default:
		rc = SLAP_CB_CONTINUE;
		break;
	}

	return rc;
}

static int
slapi_op_func( Operation *op, SlapReply *rs )
{
	Slapi_PBlock		*pb;
	slap_operation_t	which;
	struct slapi_op_info	*opinfo;
	int			rc;
	slap_overinfo		*oi;
	slap_overinst		*on;
	slap_callback		cb;
	int			internal_op;
	int			preop_type, postop_type;

	if ( !slapi_plugins_used )
		return SLAP_CB_CONTINUE;

	/*
	 * Find the SLAPI operation information for this LDAP
	 * operation; this will contain the preop and postop
	 * plugin types, as well as optional callbacks for
	 * setting up the SLAPI environment.
	 */
	which = slapi_tag2op( op->o_tag );
	if ( which >= op_last ) {
		/* invalid operation, but let someone else deal with it */
		return SLAP_CB_CONTINUE;
	}

	opinfo = &slapi_op_dispatch_table[which];
	if ( opinfo == NULL ) {
		/* no SLAPI plugin types for this operation */
		return SLAP_CB_CONTINUE;
	}

	internal_op = slapi_op_internal_p( op, &cb );

	if ( internal_op ) {
		preop_type = opinfo->soi_internal_preop;
		postop_type = opinfo->soi_internal_postop;
	} else {
		preop_type = opinfo->soi_preop;
		postop_type = opinfo->soi_postop;
	}

	if ( preop_type == 0 ) {
		/* no SLAPI plugin types for this operation */
		rs->sr_err = SLAP_CB_CONTINUE;
		goto cleanup;
	}

	pb = SLAPI_OPERATION_PBLOCK( op );

	slapi_pblock_set( pb, SLAPI_TARGET_ADDRESS, NULL );
	slapi_pblock_set( pb, SLAPI_TARGET_UNIQUEID, NULL );
	slapi_pblock_set( pb, SLAPI_TARGET_DN, (void *)op->o_req_dn.bv_val );

	/*
	 * Call preoperation plugins 
	 */
	if ( opinfo->soi_preop_init != NULL ) {
		rs->sr_err = (opinfo->soi_preop_init)( op, rs );
		if ( rs->sr_err != LDAP_SUCCESS )
			goto cleanup;
	}

	rs->sr_err = slapi_int_call_plugins( op->o_bd, preop_type, pb );

	/*
	 * soi_callback is responsible for examining the result code
	 * of the preoperation plugin and determining whether to
	 * abort. This is needed because of special SLAPI behaviour
	 * with bind preoperation plugins.
	 *
	 * The soi_callback function is also used to reset any values
	 * returned from the preoperation plugin before calling the
	 * backend (for the success case).
	 */
	if ( opinfo->soi_callback == NULL ) {
		/* default behaviour is preop plugin can abort operation */
		if ( rs->sr_err < 0 ) {
			slapi_pblock_get( pb, SLAPI_RESULT_CODE, (void **)&rs->sr_err );
			goto cleanup;
		} else {
			rs->sr_err = LDAP_SUCCESS;
		}
	} else {
		rc = (opinfo->soi_callback)( op, rs );
		if ( rc )
			goto cleanup;
	}

	/*
	 * Call actual backend (or next overlay in stack). We need to
	 * do this rather than returning SLAP_CB_CONTINUE and calling
	 * postoperation plugins in a response handler to match the
	 * behaviour of SLAPI in OpenLDAP 2.2, where postoperation
	 * plugins are called after the backend has completely
	 * finished processing the operation.
	 */
	on = (slap_overinst *)op->o_bd->bd_info;
	oi = on->on_info;

	rs->sr_err = overlay_op_walk( op, rs, which, oi, on->on_next );

	/*
	 * Call postoperation plugins
	 */
	slapi_pblock_set( pb, SLAPI_RESULT_CODE, (void *)rs->sr_err );

	if ( opinfo->soi_postop_init != NULL ) {
		(opinfo->soi_postop_init)( op, rs );
	}

	slapi_int_call_plugins( op->o_bd, postop_type, pb );

cleanup:
	if ( !internal_op ) {
		if ( opinfo->soi_cleanup != NULL ) {
			(opinfo->soi_cleanup)( op, rs );
		}
		slapi_pblock_destroy(pb);
		cb.sc_private = NULL;
	}

	op->o_callback = cb.sc_next;

	return rs->sr_err;
}

static int
slapi_over_extended( Operation *op, SlapReply *rs )
{
	Slapi_PBlock	*pb;
	SLAPI_FUNC	callback;
	int		rc;
	struct berval	reqdata = BER_BVNULL;
	int		internal_op;
	slap_callback	cb;

	slapi_int_get_extop_plugin( &op->ore_reqoid, &callback );
	if ( callback == NULL ) {
		return SLAP_CB_CONTINUE;
	}

	internal_op = slapi_op_internal_p( op, &cb );
	if ( internal_op ) {
		return SLAP_CB_CONTINUE;
	}

	pb = SLAPI_OPERATION_PBLOCK( op );

	if ( op->ore_reqdata != NULL ) {
		reqdata = *op->ore_reqdata;
	}

	slapi_pblock_set( pb, SLAPI_EXT_OP_REQ_OID,   (void *)op->ore_reqoid.bv_val);
	slapi_pblock_set( pb, SLAPI_EXT_OP_REQ_VALUE, (void *)&reqdata);

	rc = (*callback)( pb );
	if ( rc == SLAPI_PLUGIN_EXTENDED_SENT_RESULT ) {
		slapi_pblock_destroy( pb );
		return rc;
	} else if ( rc == SLAPI_PLUGIN_EXTENDED_NOT_HANDLED ) {
		slapi_pblock_destroy( pb );
		return SLAP_CB_CONTINUE;
	}

	slapi_pblock_get( pb, SLAPI_EXT_OP_RET_OID,   (void **)&rs->sr_rspoid );
	slapi_pblock_get( pb, SLAPI_EXT_OP_RET_VALUE, (void **)&rs->sr_rspdata );

	rs->sr_err = rc;
	send_ldap_extended( op, rs );

	slapi_ch_free_string( (char **)&rs->sr_rspoid );

	if ( rs->sr_rspdata != NULL )
		ber_bvfree( rs->sr_rspdata );

	slapi_pblock_destroy( pb );

	return rs->sr_err;
}

static int
slapi_over_access_allowed(
	Operation		*op,
	Entry			*e,
	AttributeDescription	*desc,
	struct berval		*val,
	slap_access_t		access,
	AccessControlState	*state,
	slap_mask_t		*maskp )
{
	int			rc;
	Slapi_PBlock		*pb;
	slap_callback		cb;
	int			internal_op;

	internal_op = slapi_op_internal_p( op, &cb );

	cb.sc_response = NULL;
	cb.sc_cleanup = NULL;

	pb = SLAPI_OPERATION_PBLOCK( op );

	rc = slapi_int_access_allowed( op, e, desc, val, access, state );
	if ( rc ) {
		rc = SLAP_CB_CONTINUE;
	}

	op->o_callback = cb.sc_next;

	if ( !internal_op )
		slapi_pblock_destroy( pb );

	return rc;
}

static int
slapi_over_acl_group(
	Operation		*op,
	Entry			*target,
	struct berval		*gr_ndn,
	struct berval		*op_ndn,
	ObjectClass		*group_oc,
	AttributeDescription	*group_at )
{
	Slapi_Entry		*e;
	int			rc;
	Slapi_PBlock		*pb;
	BackendDB		*be = op->o_bd;
	GroupAssertion		*g;

	op->o_bd = select_backend( gr_ndn, 0, 0 );

	for ( g = op->o_groups; g; g = g->ga_next ) {
		if ( g->ga_be != op->o_bd || g->ga_oc != group_oc ||
			g->ga_at != group_at || g->ga_len != gr_ndn->bv_len )
		{
			continue;
		}
		if ( strcmp( g->ga_ndn, gr_ndn->bv_val ) == 0 ) {
			break;
		}
	}
	if ( g != NULL ) {
		rc = g->ga_res;
		goto done;
	}

	if ( target != NULL && dn_match( &target->e_nname, gr_ndn ) ) {
		e = target;
		rc = 0;
	} else {
		rc = be_entry_get_rw( op, gr_ndn, group_oc, group_at, 0, &e );
	}
	if ( e != NULL ) {
		int			internal_op;
		slap_callback		cb;

		internal_op = slapi_op_internal_p( op, &cb );

		cb.sc_response = NULL;
		cb.sc_cleanup = NULL;

		pb = SLAPI_OPERATION_PBLOCK( op );

		slapi_pblock_set( pb, SLAPI_X_GROUP_ENTRY,        (void *)e );
		slapi_pblock_set( pb, SLAPI_X_GROUP_OPERATION_DN, (void *)op_ndn->bv_val );
		slapi_pblock_set( pb, SLAPI_X_GROUP_ATTRIBUTE,    (void *)group_at->ad_cname.bv_val );
		slapi_pblock_set( pb, SLAPI_X_GROUP_TARGET_ENTRY, (void *)target );

		rc = slapi_int_call_plugins( op->o_bd, SLAPI_X_PLUGIN_PRE_GROUP_FN, pb );
		if ( rc >= 0 ) /* 1 means no plugins called */
			rc = SLAP_CB_CONTINUE;
		else
			slapi_pblock_get( pb, SLAPI_RESULT_CODE, (void **)&rc );

		if ( !internal_op )
			slapi_pblock_destroy( pb );

		if ( e != target ) {
			be_entry_release_r( op, e );
		}
	} else {
		rc = LDAP_NO_SUCH_OBJECT; /* return SLAP_CB_CONTINUE for correctness? */
	}

	if ( op->o_tag != LDAP_REQ_BIND && !op->o_do_not_cache &&
	     rc != SLAP_CB_CONTINUE ) {
		g = op->o_tmpalloc( sizeof( GroupAssertion ) + gr_ndn->bv_len,
			op->o_tmpmemctx );
		g->ga_be = op->o_bd;
		g->ga_oc = group_oc;
		g->ga_at = group_at;
		g->ga_res = rc;
		g->ga_len = gr_ndn->bv_len;
		strcpy( g->ga_ndn, gr_ndn->bv_val );
		g->ga_next = op->o_groups;
		op->o_groups = g;
	}
	/*
	 * XXX don't call POST_GROUP_FN, I have no idea what the point of
	 * that plugin function was anyway
	 */
done:
	op->o_bd = be;
	return rc;
}

int
slapi_int_overlay_init()
{
	memset( &slapi, 0, sizeof(slapi) );

	slapi.on_bi.bi_type = SLAPI_OVERLAY_NAME;

	slapi.on_bi.bi_op_bind 		= slapi_op_func;
	slapi.on_bi.bi_op_unbind	= slapi_op_func;
	slapi.on_bi.bi_op_search	= slapi_op_func;
	slapi.on_bi.bi_op_compare	= slapi_op_func;
	slapi.on_bi.bi_op_modify	= slapi_op_func;
	slapi.on_bi.bi_op_modrdn	= slapi_op_func;
	slapi.on_bi.bi_op_add		= slapi_op_func;
	slapi.on_bi.bi_op_delete	= slapi_op_func;
	slapi.on_bi.bi_op_abandon	= slapi_op_func;
	slapi.on_bi.bi_op_cancel	= slapi_op_func;

	slapi.on_bi.bi_extended		= slapi_over_extended;
	slapi.on_bi.bi_access_allowed	= slapi_over_access_allowed;
	slapi.on_bi.bi_operational	= slapi_over_aux_operational;
	slapi.on_bi.bi_acl_group	= slapi_over_acl_group;

	return overlay_register( &slapi );
}

#endif /* LDAP_SLAPI */