]> git.sur5r.net Git - openldap/blobdiff - contrib/slapd-modules/authzid/authzid.c
Merge branch 'mdb.master' of ssh://git-master.openldap.org/~git/git/openldap
[openldap] / contrib / slapd-modules / authzid / authzid.c
index 2a028ad0d6ad1e6efa4e8135a7cac79d2743854b..972deadb0e913163956d00b5ec6ba6665205697d 100644 (file)
@@ -1,8 +1,8 @@
-/* vc.c - LDAP Verify Credentials extop (no spec yet) */
+/* authzid.c - RFC 3829 Authzid Control */
 /* $OpenLDAP$ */
 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
  *
- * Copyright 2010 The OpenLDAP Foundation.
+ * Copyright 2010-2011 The OpenLDAP Foundation.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
 
 /*
  * RFC 3829 Authzid
+ *
+ * must be instantiated as a global overlay
  */
 
 #include "portable.h"
 
 #include "slap.h"
+#include "config.h"
 #include "lutil.h"
 #include "ac/string.h"
 
-typedef struct ConnExtraAuthzID {
-       ConnExtra ce;
+typedef struct authzid_conn_t {
+       Connection *conn;
+       int refcnt;
        char authzid_flag;
-} ConnExtraAuthzID;
+} authzid_conn_t;
+
+static ldap_pvt_thread_mutex_t authzid_mutex;
+static Avlnode *authzid_tree;
+
+static int
+authzid_conn_cmp( const void *c1, const void *c2 )
+{
+       const authzid_conn_t *ac1 = (const authzid_conn_t *)c1;
+       const authzid_conn_t *ac2 = (const authzid_conn_t *)c2;
+
+       return SLAP_PTRCMP( ac1->conn, ac2->conn );
+}
+
+static int
+authzid_conn_dup( void *c1, void *c2 )
+{
+       authzid_conn_t *ac1 = (authzid_conn_t *)c1;
+       authzid_conn_t *ac2 = (authzid_conn_t *)c2;
+
+       if ( ac1->conn == ac2->conn ) {
+               return -1;
+       }
+
+       return 0;
+}
 
 static int authzid_cid;
 static slap_overinst authzid;
 
-static ConnExtraAuthzID *
-authzid_extra_find( Connection *c )
+static authzid_conn_t *
+authzid_conn_find( Connection *c )
 {
-       ConnExtra *cex;
+       authzid_conn_t *ac = NULL, tmp = { 0 };
 
-       LDAP_SLIST_FOREACH( cex, &c->c_extra, ce_next ) {
-               if ( cex->ce_key == (void *)&authzid_cid )
-                       break;
+       tmp.conn = c;
+       ac = (authzid_conn_t *)avl_find( authzid_tree, (caddr_t)&tmp, authzid_conn_cmp );
+       if ( ac == NULL || ( ac != NULL && ac->refcnt != 0 ) ) {
+               ac = NULL;
+       }
+       if ( ac ) {
+               ac->refcnt++;
        }
 
-       return (ConnExtraAuthzID *)cex;
+       return ac;
 }
 
-static ConnExtraAuthzID *
-authzid_extra_insert( Connection *c )
+static authzid_conn_t *
+authzid_conn_get( Connection *c )
 {
-       ConnExtraAuthzID *cex;
+       authzid_conn_t *ac = NULL;
 
-       cex = SLAP_CALLOC( 1, sizeof( ConnExtraAuthzID ) );
-       cex->ce.ce_key = (void *)&authzid_cid;
+       ldap_pvt_thread_mutex_lock( &authzid_mutex );
+       ac = authzid_conn_find( c );
+       if ( ac && ac->refcnt ) ac = NULL;
+       if ( ac ) ac->refcnt++;
+       ldap_pvt_thread_mutex_unlock( &authzid_mutex );
 
-       LDAP_SLIST_INSERT_HEAD( &c->c_extra, &cex->ce, ce_next );
+       return ac;
+}
 
-       return cex;
+static void
+authzid_conn_release( authzid_conn_t *ac )
+{
+       ldap_pvt_thread_mutex_lock( &authzid_mutex );
+       ac->refcnt--;
+       ldap_pvt_thread_mutex_unlock( &authzid_mutex );
 }
 
 static int
-authzid_extra_remove( Connection *c )
+authzid_conn_insert( Connection *c, char flag )
 {
-       ConnExtra *cex;
-       int found = 0;
-
-       cex = (ConnExtra *)authzid_extra_find( c );
-       if ( cex ) {
-               found = 1;
-               LDAP_SLIST_REMOVE( &c->c_extra, cex, ConnExtra, ce_next );
-               SLAP_FREE( cex );
+       authzid_conn_t *ac;
+       int rc;
+
+       ldap_pvt_thread_mutex_lock( &authzid_mutex );
+       ac = authzid_conn_find( c );
+       if ( ac ) {
+               ldap_pvt_thread_mutex_unlock( &authzid_mutex );
+               return -1;
        }
 
-       return found;
+       ac = SLAP_MALLOC( sizeof( authzid_conn_t ) );
+       ac->conn = c;
+       ac->refcnt = 0;
+       ac->authzid_flag = flag;
+       rc = avl_insert( &authzid_tree, (caddr_t)ac,
+               authzid_conn_cmp, authzid_conn_dup );
+       ldap_pvt_thread_mutex_unlock( &authzid_mutex );
+
+       return rc;
+}
+
+static int
+authzid_conn_remove( Connection *c )
+{
+       authzid_conn_t *ac, *tmp;
+
+       ldap_pvt_thread_mutex_lock( &authzid_mutex );
+       ac = authzid_conn_find( c );
+       if ( !ac ) {
+               ldap_pvt_thread_mutex_unlock( &authzid_mutex );
+               return -1;
+       }
+       tmp = avl_delete( &authzid_tree, (caddr_t)ac, authzid_conn_cmp );
+       ldap_pvt_thread_mutex_unlock( &authzid_mutex );
+
+       assert( tmp == ac );
+       SLAP_FREE( ac );
+
+       return 0;
 }
 
 static int
@@ -83,7 +153,6 @@ authzid_response(
        Operation *op,
        SlapReply *rs )
 {
-       ConnExtraAuthzID *cex;
        LDAPControl **ctrls;
        struct berval edn = BER_BVNULL;
        ber_len_t len = 0;
@@ -91,20 +160,17 @@ authzid_response(
 
        assert( rs->sr_tag = LDAP_RES_BIND );
 
-       cex = authzid_extra_find( op->o_conn );
        if ( rs->sr_err == LDAP_SASL_BIND_IN_PROGRESS ) {
-               if ( !cex ) {
-                       cex = authzid_extra_insert( op->o_conn );
-                       cex->authzid_flag = op->o_ctrlflag[ authzid_cid ];
+               authzid_conn_t *ac = op->o_controls[ authzid_cid ];
+               if ( ac ) {
+                       authzid_conn_release( ac );
+               } else {
+                       (void)authzid_conn_insert( op->o_conn, op->o_ctrlflag[ authzid_cid ] );
                }
-
                return SLAP_CB_CONTINUE;
-
        }
 
-       if ( cex ) {
-               authzid_extra_remove( op->o_conn );
-       }
+       (void)authzid_conn_remove( op->o_conn );
 
        if ( rs->sr_err != LDAP_SUCCESS ) {
                return SLAP_CB_CONTINUE;
@@ -112,11 +178,32 @@ authzid_response(
 
        if ( !BER_BVISEMPTY( &op->orb_edn ) ) {
                edn = op->orb_edn;
+
        } else if ( !BER_BVISEMPTY( &op->o_conn->c_dn ) ) {
                edn = op->o_conn->c_dn;
        }
 
        if ( !BER_BVISEMPTY( &edn ) ) {
+               ber_tag_t save_tag = op->o_tag;
+               struct berval save_dn = op->o_dn;
+               struct berval save_ndn = op->o_ndn;
+               int rc;
+
+               /* pretend it's an extop without data,
+                * so it is treated as a generic write
+                */
+               op->o_tag = LDAP_REQ_EXTENDED;
+               op->o_dn = edn;
+               op->o_ndn = edn;
+               rc = backend_check_restrictions( op, rs, NULL );
+               op->o_tag = save_tag;
+               op->o_dn = save_dn;
+               op->o_ndn = save_ndn;
+               if ( rc != LDAP_SUCCESS ) {
+                       rs->sr_err = LDAP_CONFIDENTIALITY_REQUIRED;
+                       return SLAP_CB_CONTINUE;
+               }
+
                len = STRLENOF("dn:") + edn.bv_len;
        }
 
@@ -168,8 +255,8 @@ authzid_cleanup(
                /* if ours, cleanup */
                ctrl = ldap_control_find( LDAP_CONTROL_AUTHZID_RESPONSE, rs->sr_ctrls, NULL );
                if ( ctrl ) {
-                       op->o_tmpfree( ctrl, op->o_tmpmemctx );
                        op->o_tmpfree( rs->sr_ctrls, op->o_tmpmemctx );
+                       rs->sr_ctrls = NULL;
                }
 
                if ( op->o_callback->sc_private != NULL ) {
@@ -192,9 +279,10 @@ authzid_op_bind(
        slap_callback *sc;
 
        if ( op->o_ctrlflag[ authzid_cid ] <= SLAP_CONTROL_IGNORED ) {
-               ConnExtraAuthzID *cex = authzid_extra_find( op->o_conn );
-               if ( cex ) {
-                       op->o_ctrlflag[ authzid_cid ] = cex->authzid_flag;
+               authzid_conn_t *ac = authzid_conn_get( op->o_conn );
+               if ( ac ) {
+                       op->o_ctrlflag[ authzid_cid ] = ac->authzid_flag;
+                       op->o_controls[ authzid_cid] = ac;
                }
        }
 
@@ -227,47 +315,53 @@ parse_authzid_ctrl(
        }
 
        /* drop ongoing requests */
-       (void)authzid_extra_remove( op->o_conn );
+       (void)authzid_conn_remove( op->o_conn );
 
        op->o_ctrlflag[ authzid_cid ] = ctrl->ldctl_iscritical ?  SLAP_CONTROL_CRITICAL : SLAP_CONTROL_NONCRITICAL;
 
        return LDAP_SUCCESS;
 }
 
-static int authzid_cnt;
-
 static int
-authzid_db_init( BackendDB *be, ConfigReply *cr)
+authzid_db_init( BackendDB *be, ConfigReply *cr )
 {
-       if ( authzid_cnt++ == 0 ) {
-               int rc;
-
-               rc = register_supported_control( LDAP_CONTROL_AUTHZID_REQUEST,
-                       SLAP_CTRL_GLOBAL|SLAP_CTRL_BIND|SLAP_CTRL_HIDE, NULL,
-                       parse_authzid_ctrl, &authzid_cid );
-               if ( rc != LDAP_SUCCESS ) {
-                       Debug( LDAP_DEBUG_ANY,
-                               "authzid_initialize: Failed to register control '%s' (%d)\n",
-                               LDAP_CONTROL_AUTHZID_REQUEST, rc, 0 );
-                       return rc;
+       if ( !SLAP_ISGLOBALOVERLAY( be ) ) {
+               /* do not allow slapo-ppolicy to be global by now (ITS#5858) */
+               if ( cr ) {
+                       snprintf( cr->msg, sizeof(cr->msg), 
+                               "slapo-authzid must be global" );
+                       Debug( LDAP_DEBUG_ANY, "%s\n", cr->msg, 0, 0 );
                }
+               return 1;
+       }
+               
+       int rc;
+
+       rc = register_supported_control( LDAP_CONTROL_AUTHZID_REQUEST,
+               SLAP_CTRL_GLOBAL|SLAP_CTRL_BIND|SLAP_CTRL_HIDE, NULL,
+               parse_authzid_ctrl, &authzid_cid );
+       if ( rc != LDAP_SUCCESS ) {
+               Debug( LDAP_DEBUG_ANY,
+                       "authzid_initialize: Failed to register control '%s' (%d)\n",
+                       LDAP_CONTROL_AUTHZID_REQUEST, rc, 0 );
+               return rc;
        }
 
        return LDAP_SUCCESS;
 }
 
+/*
+ * Almost pointless, by now, since this overlay needs to be global,
+ * and global overlays deletion is not supported yet.
+ */
 static int
 authzid_db_destroy( BackendDB *be, ConfigReply *cr )
 {
-       assert( authzid_cnt > 0 );
-
 #ifdef SLAP_CONFIG_DELETE
        overlay_unregister_control( be, LDAP_CONTROL_AUTHZID_REQUEST );
 #endif /* SLAP_CONFIG_DELETE */
 
-       if ( --authzid_cnt == 0 ) {
-               unregister_supported_control( LDAP_CONTROL_AUTHZID_REQUEST );
-       }
+       unregister_supported_control( LDAP_CONTROL_AUTHZID_REQUEST );
 
        return 0;
 }
@@ -275,6 +369,8 @@ authzid_db_destroy( BackendDB *be, ConfigReply *cr )
 static int
 authzid_initialize( void )
 {
+       ldap_pvt_thread_mutex_init( &authzid_mutex );
+
        authzid.on_bi.bi_type = "authzid";
 
        authzid.on_bi.bi_db_init = authzid_db_init;