#include "portable.h"
#include <stdio.h>
#include "slap.h"
#include "back-relay.h"
static const struct relay_fail_modes_s {
slap_mask_t rf_bd, rf_op;
#define RB_ERR_MASK 0x0000FFFFU
#define RB_BDERR 0x80000000U
#define RB_OPERR 0x40000000U
#define RB_REF 0x20000000U
#define RB_SEND 0x10000000U
#define RB_SENDREF 0
#define RB_NO_BIND (RB_OPERR | LDAP_INVALID_CREDENTIALS)
#define RB_NOT_SUPP (RB_OPERR | LDAP_UNWILLING_TO_PERFORM)
#define RB_NO_OBJ (RB_REF | LDAP_NO_SUCH_OBJECT)
#define RB_CHK_REF (RB_REF | RB_SENDREF | LDAP_SUCCESS)
} relay_fail_modes[relay_op_last] = {
# define RB_OP(b, o) { (b) | RB_BD2ERR(b), (o) | RB_BD2ERR(b) }
# define RB_BD2ERR(b) ((b) ? RB_BDERR : 0)
RB_OP(RB_NO_BIND|RB_SEND, RB_NO_BIND |RB_SEND),
RB_OP(0, LDAP_SUCCESS),
RB_OP(RB_NO_OBJ |RB_SEND, RB_NOT_SUPP |RB_SEND),
RB_OP(RB_NO_OBJ |RB_SEND, SLAP_CB_CONTINUE),
RB_OP(RB_NO_OBJ |RB_SEND, RB_NOT_SUPP |RB_SEND),
RB_OP(RB_NO_OBJ |RB_SEND, RB_NOT_SUPP |RB_SEND),
RB_OP(RB_NO_OBJ |RB_SEND, RB_NOT_SUPP |RB_SEND),
RB_OP(RB_NO_OBJ |RB_SEND, RB_NOT_SUPP |RB_SEND),
RB_OP(0, LDAP_SUCCESS),
RB_OP(RB_NO_OBJ, RB_NOT_SUPP),
RB_OP(0, SLAP_CB_CONTINUE),
RB_OP(0, LDAP_SUCCESS),
RB_OP(RB_CHK_REF, LDAP_SUCCESS),
RB_OP(0, SLAP_CB_CONTINUE),
RB_OP(0, 0),
RB_OP(0, 0),
RB_OP(0, 0),
};
typedef struct relay_callback {
slap_callback rcb_sc;
BackendDB *rcb_bd;
} relay_callback;
static int
relay_back_cleanup_cb( Operation *op, SlapReply *rs )
{
op->o_bd = ((relay_callback *) op->o_callback)->rcb_bd;
return SLAP_CB_CONTINUE;
}
static int
relay_back_response_cb( Operation *op, SlapReply *rs )
{
relay_callback *rcb = (relay_callback *) op->o_callback;
rcb->rcb_sc.sc_cleanup = relay_back_cleanup_cb;
rcb->rcb_bd = op->o_bd;
op->o_bd = op->o_callback->sc_private;
return SLAP_CB_CONTINUE;
}
#define relay_back_add_cb( rcb, op ) { \
(rcb)->rcb_sc.sc_next = (op)->o_callback; \
(rcb)->rcb_sc.sc_response = relay_back_response_cb; \
(rcb)->rcb_sc.sc_cleanup = 0; \
(rcb)->rcb_sc.sc_private = (op)->o_bd; \
(op)->o_callback = (slap_callback *) (rcb); \
}
#define relay_back_remove_cb( rcb, op ) { \
slap_callback **sc = &(op)->o_callback; \
for ( ;; sc = &(*sc)->sc_next ) \
if ( *sc == (slap_callback *) (rcb) ) { \
*sc = (*sc)->sc_next; break; \
} else if ( *sc == NULL ) break; \
}
static BackendDB *
relay_back_select_backend( Operation *op, SlapReply *rs, int which )
{
OpExtra *oex;
char *key = (char *) op->o_bd->be_private;
BackendDB *bd = ((relay_back_info *) key)->ri_bd;
slap_mask_t fail_mode = relay_fail_modes[which].rf_bd;
int useDN = 0, rc = ( fail_mode & RB_ERR_MASK );
if ( bd == NULL && !BER_BVISNULL( &op->o_req_ndn ) ) {
useDN = 1;
bd = select_backend( &op->o_req_ndn, 1 );
}
if ( bd != NULL ) {
key += which;
LDAP_SLIST_FOREACH( oex, &op->o_extra, oe_next ) {
if ( oex->oe_key == key )
break;
}
if ( oex == NULL ) {
return bd;
}
Debug( LDAP_DEBUG_ANY,
"%s: back-relay for DN=\"%s\" would call self.\n",
op->o_log_prefix, op->o_req_dn.bv_val, 0 );
} else if ( useDN && ( fail_mode & RB_REF ) && default_referral ) {
rc = LDAP_REFERRAL;
rs->sr_ref = referral_rewrite(
default_referral, NULL, &op->o_req_dn,
op->o_tag == LDAP_REQ_SEARCH ?
op->ors_scope : LDAP_SCOPE_DEFAULT );
if ( rs->sr_ref != NULL ) {
rs->sr_flags |= REP_REF_MUSTBEFREED;
} else {
rs->sr_ref = default_referral;
}
if ( fail_mode & RB_SENDREF )
fail_mode = (RB_BDERR | RB_SEND);
}
if ( fail_mode & RB_BDERR ) {
rs->sr_err = rc;
if ( fail_mode & RB_SEND ) {
send_ldap_result( op, rs );
}
}
return NULL;
}
#define RELAY_WRAP_OP( op, bd, which, act ) { \
OpExtraDB wrap_oex; \
BackendDB *const wrap_bd = (op)->o_bd; \
wrap_oex.oe_db = wrap_bd; \
wrap_oex.oe.oe_key = (char *) wrap_bd->be_private + (which); \
LDAP_SLIST_INSERT_HEAD( &(op)->o_extra, &wrap_oex.oe, oe_next ); \
(op)->o_bd = (bd); \
act; \
(op)->o_bd = wrap_bd; \
LDAP_SLIST_REMOVE( &(op)->o_extra, &wrap_oex.oe, OpExtra, oe_next ); \
}
static int
relay_back_op( Operation *op, SlapReply *rs, int which )
{
BackendDB *bd;
BI_op_bind *func;
slap_mask_t fail_mode = relay_fail_modes[which].rf_op;
int rc = ( fail_mode & RB_ERR_MASK );
bd = relay_back_select_backend( op, rs, which );
if ( bd == NULL ) {
if ( fail_mode & RB_BDERR )
return rs->sr_err;
} else if ( (func = (&bd->be_bind)[which]) != 0 ) {
relay_callback rcb;
relay_back_add_cb( &rcb, op );
RELAY_WRAP_OP( op, bd, which, {
rc = func( op, rs );
});
relay_back_remove_cb( &rcb, op );
} else if ( fail_mode & RB_OPERR ) {
rs->sr_err = rc;
if ( rc == LDAP_UNWILLING_TO_PERFORM ) {
rs->sr_text = "operation not supported within naming context";
}
if ( fail_mode & RB_SEND ) {
send_ldap_result( op, rs );
}
}
return rc;
}
int
relay_back_op_bind( Operation *op, SlapReply *rs )
{
switch ( be_rootdn_bind( op, rs ) ) {
case SLAP_CB_CONTINUE:
break;
default:
return rs->sr_err;
}
return relay_back_op( op, rs, op_bind );
}
#define RELAY_DEFOP(func, which) \
int func( Operation *op, SlapReply *rs ) \
{ return relay_back_op( op, rs, which ); }
RELAY_DEFOP( relay_back_op_search, op_search )
RELAY_DEFOP( relay_back_op_compare, op_compare )
RELAY_DEFOP( relay_back_op_modify, op_modify )
RELAY_DEFOP( relay_back_op_modrdn, op_modrdn )
RELAY_DEFOP( relay_back_op_add, op_add )
RELAY_DEFOP( relay_back_op_delete, op_delete )
RELAY_DEFOP( relay_back_op_extended, op_extended )
RELAY_DEFOP( relay_back_operational, op_aux_operational )
int
relay_back_entry_release_rw( Operation *op, Entry *e, int rw )
{
BackendDB *bd;
int rc = LDAP_UNWILLING_TO_PERFORM;
bd = relay_back_select_backend( op, NULL, relay_op_entry_release );
if ( bd && bd->be_release ) {
RELAY_WRAP_OP( op, bd, relay_op_entry_release, {
rc = bd->be_release( op, e, rw );
});
} else if ( e->e_private == NULL ) {
entry_free( e );
rc = LDAP_SUCCESS;
}
return rc;
}
int
relay_back_entry_get_rw( Operation *op, struct berval *ndn,
ObjectClass *oc, AttributeDescription *at, int rw, Entry **e )
{
BackendDB *bd;
int rc = LDAP_NO_SUCH_OBJECT;
bd = relay_back_select_backend( op, NULL, relay_op_entry_get );
if ( bd && bd->be_fetch ) {
RELAY_WRAP_OP( op, bd, relay_op_entry_get, {
rc = bd->be_fetch( op, ndn, oc, at, rw, e );
});
}
return rc;
}
#if 0
RELAY_DEFOP( relay_back_chk_referrals, op_aux_chk_referrals )
#endif
int
relay_back_has_subordinates( Operation *op, Entry *e, int *hasSubs )
{
BackendDB *bd;
int rc = LDAP_OTHER;
bd = relay_back_select_backend( op, NULL, relay_op_has_subordinates );
if ( bd && bd->be_has_subordinates ) {
RELAY_WRAP_OP( op, bd, relay_op_has_subordinates, {
rc = bd->be_has_subordinates( op, e, hasSubs );
});
}
return rc;
}