]> git.sur5r.net Git - openldap/blobdiff - servers/slapd/back-monitor/thread.c
don't leak callbacks if stuff cannot be registered; provide a means to dispose of...
[openldap] / servers / slapd / back-monitor / thread.c
index c76acaa86468941669261ff0e7bd336aa0b81ab4..3a476cc0d8d2b4d5780fb2971f1c1daa6a0a335c 100644 (file)
@@ -2,7 +2,7 @@
 /* $OpenLDAP$ */
 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
  *
- * Copyright 2001-2005 The OpenLDAP Foundation.
+ * Copyright 2001-2006 The OpenLDAP Foundation.
  * Portions Copyright 2001-2003 Pierangelo Masarati.
  * All rights reserved.
  *
 
 #include <ldap_rq.h>
 
+#ifndef NO_THREADS
+typedef enum {
+       MT_UNKNOWN,
+       MT_RUNQUEUE,
+       MT_TASKLIST,
+
+       MT_LAST
+} monitor_thread_t;
+
+static struct {
+       struct berval                   rdn;
+       struct berval                   nrdn;
+       ldap_pvt_thread_pool_param_t    param;
+       monitor_thread_t                mt;
+}              mt[] = {
+       { BER_BVC( "cn=Max" ),          BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_MAX,         MT_UNKNOWN },
+       { BER_BVC( "cn=Max Pending" ),  BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_MAX_PENDING, MT_UNKNOWN },
+       { BER_BVC( "cn=Open" ),         BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_OPEN,        MT_UNKNOWN },
+       { BER_BVC( "cn=Starting" ),     BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_STARTING,    MT_UNKNOWN },
+       { BER_BVC( "cn=Active" ),       BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_ACTIVE,      MT_UNKNOWN },
+       { BER_BVC( "cn=Pending" ),      BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_PENDING,     MT_UNKNOWN },
+       { BER_BVC( "cn=Backload" ),     BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_BACKLOAD,    MT_UNKNOWN },
+#if 0  /* not meaningful right now */
+       { BER_BVC( "cn=Active Max" ),   BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_ACTIVE_MAX,  MT_UNKNOWN },
+       { BER_BVC( "cn=Pending Max" ),  BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_PENDING_MAX, MT_UNKNOWN },
+       { BER_BVC( "cn=Backload Max" ), BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_BACKLOAD_MAX,MT_UNKNOWN },
+#endif
+       { BER_BVC( "cn=State" ),        BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_STATE,       MT_UNKNOWN },
+
+       { BER_BVC( "cn=Runqueue" ),     BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_UNKNOWN,     MT_RUNQUEUE },
+       { BER_BVC( "cn=Tasklist" ),     BER_BVNULL,     LDAP_PVT_THREAD_POOL_PARAM_UNKNOWN,     MT_TASKLIST },
+
+       { BER_BVNULL }
+};
+
 static int 
 monitor_subsys_thread_update( 
        Operation               *op,
        SlapReply               *rs,
        Entry                   *e );
+#endif /* ! NO_THREADS */
 
 /*
  * initializes log subentry
@@ -41,13 +77,13 @@ monitor_subsys_thread_update(
 int
 monitor_subsys_thread_init(
        BackendDB               *be,
-       monitor_subsys_t        *ms
-)
+       monitor_subsys_t        *ms )
 {
+#ifndef NO_THREADS
        monitor_info_t  *mi;
        monitor_entry_t *mp;
        Entry           *e, **ep, *e_thread;
-       static char     buf[ BACKMONITOR_BUFSIZE ];
+       int             i;
 
        ms->mss_update = monitor_subsys_thread_update;
 
@@ -65,165 +101,88 @@ monitor_subsys_thread_init(
        mp->mp_children = NULL;
        ep = &mp->mp_children;
 
-       /*
-        * Max
-        */
-       snprintf( buf, sizeof( buf ),
-                       "dn: cn=Max,%s\n"
-                       "objectClass: %s\n"
-                       "structuralObjectClass: %s\n"
-                       "cn: Max\n"
-                       "%s: %d\n"
-                       "creatorsName: %s\n"
-                       "modifiersName: %s\n"
-                       "createTimestamp: %s\n"
-                       "modifyTimestamp: %s\n", 
-                       ms->mss_dn.bv_val,
-                       mi->mi_oc_monitoredObject->soc_cname.bv_val,
-                       mi->mi_oc_monitoredObject->soc_cname.bv_val,
-                       mi->mi_ad_monitoredInfo->ad_cname.bv_val,
-                       connection_pool_max,
-                       mi->mi_creatorsName.bv_val,
-                       mi->mi_creatorsName.bv_val,
-                       mi->mi_startTime.bv_val,
-                       mi->mi_startTime.bv_val );
-
-       e = str2entry( buf );
-       if ( e == NULL ) {
-               Debug( LDAP_DEBUG_ANY,
-                       "monitor_subsys_thread_init: "
-                       "unable to create entry \"cn=Max,%s\"\n",
-                       ms->mss_ndn.bv_val, 0, 0 );
-               return( -1 );
-       }
-       
-       mp = monitor_entrypriv_create();
-       if ( mp == NULL ) {
-               return -1;
-       }
-       e->e_private = ( void * )mp;
-       mp->mp_info = ms;
-       mp->mp_flags = ms->mss_flags \
-               | MONITOR_F_SUB | MONITOR_F_PERSISTENT;
-
-       if ( monitor_cache_add( mi, e ) ) {
-               Debug( LDAP_DEBUG_ANY,
-                       "monitor_subsys_thread_init: "
-                       "unable to add entry \"cn=Max,%s\"\n",
-                       ms->mss_ndn.bv_val, 0, 0 );
-               return( -1 );
-       }
-       
-       *ep = e;
-       ep = &mp->mp_next;
-
-       /*
-        * Backload
-        */
-       snprintf( buf, sizeof( buf ),
-                       "dn: cn=Backload,%s\n"
-                       "objectClass: %s\n"
-                       "structuralObjectClass: %s\n"
-                       "cn: Backload\n"
-                       "%s: 0\n"
-                       "creatorsName: %s\n"
-                       "modifiersName: %s\n"
-                       "createTimestamp: %s\n"
-                       "modifyTimestamp: %s\n",
-                       ms->mss_dn.bv_val,
-                       mi->mi_oc_monitoredObject->soc_cname.bv_val,
-                       mi->mi_oc_monitoredObject->soc_cname.bv_val,
-                       mi->mi_ad_monitoredInfo->ad_cname.bv_val,
-                       mi->mi_creatorsName.bv_val,
-                       mi->mi_creatorsName.bv_val,
-                       mi->mi_startTime.bv_val,
-                       mi->mi_startTime.bv_val );
-
-       e = str2entry( buf );
-       if ( e == NULL ) {
-               Debug( LDAP_DEBUG_ANY,
-                       "monitor_subsys_thread_init: "
-                       "unable to create entry \"cn=Backload,%s\"\n",
-                       ms->mss_ndn.bv_val, 0, 0 );
-               return( -1 );
-       }
+       for ( i = 0; !BER_BVISNULL( &mt[ i ].rdn ); i++ ) {
+               static char     buf[ BACKMONITOR_BUFSIZE ];
+               int             count = -1;
+               char            *state = NULL;
+               struct berval   bv = BER_BVNULL;
+
+               /*
+                * Max
+                */
+               e = monitor_entry_stub( &ms->mss_dn, &ms->mss_ndn,
+                       &mt[ i ].rdn,
+                       mi->mi_oc_monitoredObject, mi, NULL, NULL );
+               if ( e == NULL ) {
+                       Debug( LDAP_DEBUG_ANY,
+                               "monitor_subsys_thread_init: "
+                               "unable to create entry \"%s,%s\"\n",
+                               mt[ i ].rdn.bv_val,
+                               ms->mss_ndn.bv_val, 0 );
+                       return( -1 );
+               }
 
-       mp = monitor_entrypriv_create();
-       if ( mp == NULL ) {
-               return -1;
-       }
-       e->e_private = ( void * )mp;
-       mp->mp_info = ms;
-       mp->mp_flags = ms->mss_flags \
-               | MONITOR_F_SUB | MONITOR_F_PERSISTENT;
+               /* NOTE: reference to the normalized DN of the entry,
+                * under the assumption it's not modified */
+               dnRdn( &e->e_nname, &mt[ i ].nrdn );
+
+               switch ( mt[ i ].param ) {
+               case LDAP_PVT_THREAD_POOL_PARAM_UNKNOWN:
+                       break;
+
+               case LDAP_PVT_THREAD_POOL_PARAM_STATE:
+                       if ( ldap_pvt_thread_pool_query( &connection_pool,
+                               mt[ i ].param, (void *)&state ) == 0 )
+                       {
+                               ber_str2bv( state, 0, 0, &bv );
+
+                       } else {
+                               BER_BVSTR( &bv, "unknown" );
+                       }
+                       break;
+
+               default:
+                       /* NOTE: in case of error, it'll be set to -1 */
+                       (void)ldap_pvt_thread_pool_query( &connection_pool,
+                               mt[ i ].param, (void *)&count );
+                       bv.bv_val = buf;
+                       bv.bv_len = snprintf( buf, sizeof( buf ), "%d", count );
+                       break;
+               }
 
-       if ( monitor_cache_add( mi, e ) ) {
-               Debug( LDAP_DEBUG_ANY,
-                       "monitor_subsys_thread_init: "
-                       "unable to add entry \"cn=Backload,%s\"\n",
-                       ms->mss_ndn.bv_val, 0, 0 );
-               return( -1 );
-       }
+               if ( !BER_BVISNULL( &bv ) ) {
+                       attr_merge_normalize_one( e, mi->mi_ad_monitoredInfo, &bv, NULL );
+               }
        
-       *ep = e;
-       ep = &mp->mp_next;
-
-       /*
-        * Runqueue runners
-        */
-       snprintf( buf, sizeof( buf ),
-                       "dn: cn=Runqueue,%s\n"
-                       "objectClass: %s\n"
-                       "structuralObjectClass: %s\n"
-                       "cn: Runqueue\n"
-                       "%s: 0\n"
-                       "creatorsName: %s\n"
-                       "modifiersName: %s\n"
-                       "createTimestamp: %s\n"
-                       "modifyTimestamp: %s\n",
-                       ms->mss_dn.bv_val,
-                       mi->mi_oc_monitoredObject->soc_cname.bv_val,
-                       mi->mi_oc_monitoredObject->soc_cname.bv_val,
-                       mi->mi_ad_monitoredInfo->ad_cname.bv_val,
-                       mi->mi_creatorsName.bv_val,
-                       mi->mi_creatorsName.bv_val,
-                       mi->mi_startTime.bv_val,
-                       mi->mi_startTime.bv_val );
-
-       e = str2entry( buf );
-       if ( e == NULL ) {
-               Debug( LDAP_DEBUG_ANY,
-                       "monitor_subsys_thread_init: "
-                       "unable to create entry \"cn=Runqueue,%s\"\n",
-                       ms->mss_ndn.bv_val, 0, 0 );
-               return( -1 );
-       }
-
-       mp = monitor_entrypriv_create();
-       if ( mp == NULL ) {
-               return -1;
-       }
-       e->e_private = ( void * )mp;
-       mp->mp_info = ms;
-       mp->mp_flags = ms->mss_flags \
-               | MONITOR_F_SUB | MONITOR_F_PERSISTENT;
-
-       if ( monitor_cache_add( mi, e ) ) {
-               Debug( LDAP_DEBUG_ANY,
-                       "monitor_subsys_thread_init: "
-                       "unable to add entry \"cn=Runqueue,%s\"\n",
-                       ms->mss_ndn.bv_val, 0, 0 );
-               return( -1 );
-       }
+               mp = monitor_entrypriv_create();
+               if ( mp == NULL ) {
+                       return -1;
+               }
+               e->e_private = ( void * )mp;
+               mp->mp_info = ms;
+               mp->mp_flags = ms->mss_flags \
+                       | MONITOR_F_SUB | MONITOR_F_PERSISTENT;
+
+               if ( monitor_cache_add( mi, e ) ) {
+                       Debug( LDAP_DEBUG_ANY,
+                               "monitor_subsys_thread_init: "
+                               "unable to add entry \"%s,%s\"\n",
+                               mt[ i ].rdn.bv_val,
+                               ms->mss_ndn.bv_val, 0 );
+                       return( -1 );
+               }
        
-       *ep = e;
-       ep = &mp->mp_next;
+               *ep = e;
+               ep = &mp->mp_next;
+       }
 
        monitor_cache_release( mi, e_thread );
 
+#endif /* ! NO_THREADS */
        return( 0 );
 }
 
+#ifndef NO_THREADS
 static int 
 monitor_subsys_thread_update( 
        Operation               *op,
@@ -232,57 +191,122 @@ monitor_subsys_thread_update(
 {
        monitor_info_t  *mi = ( monitor_info_t * )op->o_bd->be_private;
        Attribute               *a;
+       BerVarray               vals = NULL;
        char                    buf[ BACKMONITOR_BUFSIZE ];
-       static struct berval    backload_bv = BER_BVC( "cn=backload" );
-       static struct berval    runqueue_bv = BER_BVC( "cn=runqueue" );
        struct berval           rdn, bv;
-       ber_len_t               len;
-       int which = 0, i;
-       struct re_s *re;
+       int                     which, i;
+       struct re_s             *re;
+       int                     count = -1;
+       char                    *state = NULL;
 
        assert( mi != NULL );
 
        dnRdn( &e->e_nname, &rdn );
-       if ( dn_match( &rdn, &backload_bv ) ) {
-               which = 1;
 
-       } else if ( dn_match( &rdn, &runqueue_bv ) ) {
-               which = 2;
+       for ( i = 0; !BER_BVISNULL( &mt[ i ].nrdn ); i++ ) {
+               if ( dn_match( &mt[ i ].nrdn, &rdn ) ) {
+                       break;
+               }
+       }
 
-       } else {
+       which = i;
+       if ( BER_BVISNULL( &mt[ which ].nrdn ) ) {
                return SLAP_CB_CONTINUE;
        }
 
        a = attr_find( e->e_attrs, mi->mi_ad_monitoredInfo );
-       if ( a == NULL ) {
-               return rs->sr_err = LDAP_OTHER;
-       }
 
-       switch ( which ) {
-       case 1:
-               snprintf( buf, sizeof( buf ), "%d", 
-                       ldap_pvt_thread_pool_backload( &connection_pool ) );
-               len = strlen( buf );
-               if ( len > a->a_vals[ 0 ].bv_len ) {
-                       a->a_vals[ 0 ].bv_val = ber_memrealloc( a->a_vals[ 0 ].bv_val, len + 1 );
+       switch ( mt[ which ].param ) {
+       case LDAP_PVT_THREAD_POOL_PARAM_UNKNOWN:
+               switch ( mt[ which ].mt ) {
+               case MT_RUNQUEUE:
+                       if ( a != NULL ) {
+                               if ( a->a_nvals != a->a_vals ) {
+                                       ber_bvarray_free( a->a_nvals );
+                               }
+                               ber_bvarray_free( a->a_vals );
+                               a->a_vals = NULL;
+                               a->a_nvals = NULL;
+                       }
+
+                       i = 0;
+                       bv.bv_val = buf;
+                       ldap_pvt_thread_mutex_lock( &slapd_rq.rq_mutex );
+                       LDAP_STAILQ_FOREACH( re, &slapd_rq.run_list, rnext ) {
+                               bv.bv_len = snprintf( buf, sizeof( buf ), "{%d}%s(%s)",
+                                       i, re->tname, re->tspec );
+                               if ( bv.bv_len < sizeof( buf ) ) {
+                                       value_add_one( &vals, &bv );
+                               }
+                               i++;
+                       }
+                       ldap_pvt_thread_mutex_unlock( &slapd_rq.rq_mutex );
+       
+                       if ( vals ) {
+                               attr_merge_normalize( e, mi->mi_ad_monitoredInfo, vals, NULL );
+                               ber_bvarray_free( vals );
+                       }
+                       break;
+
+               case MT_TASKLIST:
+                       if ( a != NULL ) {
+                               if ( a->a_nvals != a->a_vals ) {
+                                       ber_bvarray_free( a->a_nvals );
+                               }
+                               ber_bvarray_free( a->a_vals );
+                               a->a_vals = NULL;
+                               a->a_nvals = NULL;
+                       }
+       
+                       i = 0;
+                       bv.bv_val = buf;
+                       ldap_pvt_thread_mutex_lock( &slapd_rq.rq_mutex );
+                       LDAP_STAILQ_FOREACH( re, &slapd_rq.task_list, tnext ) {
+                               bv.bv_len = snprintf( buf, sizeof( buf ), "{%d}%s(%s)",
+                                       i, re->tname, re->tspec );
+                               if ( bv.bv_len < sizeof( buf ) ) {
+                                       value_add_one( &vals, &bv );
+                               }
+                               i++;
+                       }
+                       ldap_pvt_thread_mutex_unlock( &slapd_rq.rq_mutex );
+       
+                       if ( vals ) {
+                               attr_merge_normalize( e, mi->mi_ad_monitoredInfo, vals, NULL );
+                               ber_bvarray_free( vals );
+                       }
+                       break;
+
+               default:
+                       assert( 0 );
                }
-               a->a_vals[ 0 ].bv_len = len;
-               AC_MEMCPY( a->a_vals[ 0 ].bv_val, buf, len + 1 );
                break;
 
-       case 2:
-               for ( i = 0; !BER_BVISNULL( a->a_vals + i ); i++) {
-                       ch_free( a->a_vals[i].bv_val );
-                       BER_BVZERO( a->a_vals + i );
+       case LDAP_PVT_THREAD_POOL_PARAM_STATE:
+               if ( a == NULL ) {
+                       return rs->sr_err = LDAP_OTHER;
                }
-               bv.bv_val = buf;
-               ldap_pvt_thread_mutex_lock( &slapd_rq.rq_mutex );
-               LDAP_STAILQ_FOREACH( re, &slapd_rq.run_list, rnext ) {
-                       bv.bv_len = snprintf( buf, sizeof( buf ), "%s(%s)",
-                               re->tname, re->tspec );
-                       value_add_one( &a->a_vals, &bv );
+               if ( ldap_pvt_thread_pool_query( &connection_pool,
+                       mt[ i ].param, (void *)&state ) == 0 )
+               {
+                       ber_str2bv( state, 0, 0, &bv );
+                       ber_bvreplace( &a->a_vals[ 0 ], &bv );
+               }
+               break;
+
+       default:
+               if ( a == NULL ) {
+                       return rs->sr_err = LDAP_OTHER;
+               }
+               if ( ldap_pvt_thread_pool_query( &connection_pool,
+                       mt[ i ].param, (void *)&count ) == 0 )
+               {
+                       bv.bv_val = buf;
+                       bv.bv_len = snprintf( buf, sizeof( buf ), "%d", count );
+                       if ( bv.bv_len < sizeof( buf ) ) {
+                               ber_bvreplace( &a->a_vals[ 0 ], &bv );
+                       }
                }
-               ldap_pvt_thread_mutex_unlock( &slapd_rq.rq_mutex );
                break;
        }
 
@@ -290,4 +314,4 @@ monitor_subsys_thread_update(
 
        return SLAP_CB_CONTINUE;
 }
-
+#endif /* ! NO_THREADS */