mirror of
https://git.openldap.org/openldap/openldap.git
synced 2024-12-21 03:10:25 +08:00
807 lines
18 KiB
C
807 lines
18 KiB
C
/* monitor.c - monitor mdb backend */
|
|
/* $OpenLDAP$ */
|
|
/* This work is part of OpenLDAP Software <http://www.openldap.org/>.
|
|
*
|
|
* Copyright 2000-2020 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>.
|
|
*/
|
|
|
|
#include "portable.h"
|
|
|
|
#include <stdio.h>
|
|
#include <ac/string.h>
|
|
#include <ac/unistd.h>
|
|
#include <ac/stdlib.h>
|
|
#include <ac/errno.h>
|
|
#include <sys/stat.h>
|
|
#include "lutil.h"
|
|
#include "back-mdb.h"
|
|
|
|
#include "../back-monitor/back-monitor.h"
|
|
|
|
#include "config.h"
|
|
|
|
static ObjectClass *oc_olmMDBDatabase;
|
|
|
|
static AttributeDescription *ad_olmDbDirectory;
|
|
|
|
#ifdef MDB_MONITOR_IDX
|
|
static int
|
|
mdb_monitor_idx_entry_add(
|
|
struct mdb_info *mdb,
|
|
Entry *e );
|
|
|
|
static AttributeDescription *ad_olmDbNotIndexed;
|
|
#endif /* MDB_MONITOR_IDX */
|
|
|
|
static AttributeDescription *ad_olmMDBPagesMax,
|
|
*ad_olmMDBPagesUsed, *ad_olmMDBPagesFree;
|
|
|
|
static AttributeDescription *ad_olmMDBReadersMax,
|
|
*ad_olmMDBReadersUsed;
|
|
|
|
static AttributeDescription *ad_olmMDBEntries;
|
|
|
|
/*
|
|
* NOTE: there's some confusion in monitor OID arc;
|
|
* by now, let's consider:
|
|
*
|
|
* Subsystems monitor attributes 1.3.6.1.4.1.4203.666.1.55.0
|
|
* Databases monitor attributes 1.3.6.1.4.1.4203.666.1.55.0.1
|
|
* MDB database monitor attributes 1.3.6.1.4.1.4203.666.1.55.0.1.3
|
|
*
|
|
* Subsystems monitor objectclasses 1.3.6.1.4.1.4203.666.3.16.0
|
|
* Databases monitor objectclasses 1.3.6.1.4.1.4203.666.3.16.0.1
|
|
* MDB database monitor objectclasses 1.3.6.1.4.1.4203.666.3.16.0.1.3
|
|
*/
|
|
|
|
static struct {
|
|
char *name;
|
|
char *oid;
|
|
} s_oid[] = {
|
|
{ "olmMDBAttributes", "olmDatabaseAttributes:1" },
|
|
{ "olmMDBObjectClasses", "olmDatabaseObjectClasses:1" },
|
|
|
|
{ NULL }
|
|
};
|
|
|
|
static struct {
|
|
char *desc;
|
|
AttributeDescription **ad;
|
|
} s_at[] = {
|
|
{ "( olmDatabaseAttributes:1 "
|
|
"NAME ( 'olmDbDirectory' ) "
|
|
"DESC 'Path name of the directory "
|
|
"where the database environment resides' "
|
|
"SUP monitoredInfo "
|
|
"NO-USER-MODIFICATION "
|
|
"USAGE dSAOperation )",
|
|
&ad_olmDbDirectory },
|
|
|
|
#ifdef MDB_MONITOR_IDX
|
|
{ "( olmDatabaseAttributes:2 "
|
|
"NAME ( 'olmDbNotIndexed' ) "
|
|
"DESC 'Missing indexes resulting from candidate selection' "
|
|
"SUP monitoredInfo "
|
|
"NO-USER-MODIFICATION "
|
|
"USAGE dSAOperation )",
|
|
&ad_olmDbNotIndexed },
|
|
#endif /* MDB_MONITOR_IDX */
|
|
|
|
{ "( olmMDBAttributes:1 "
|
|
"NAME ( 'olmMDBPagesMax' ) "
|
|
"DESC 'Maximum number of pages' "
|
|
"SUP monitorCounter "
|
|
"NO-USER-MODIFICATION "
|
|
"USAGE dSAOperation )",
|
|
&ad_olmMDBPagesMax },
|
|
|
|
{ "( olmMDBAttributes:2 "
|
|
"NAME ( 'olmMDBPagesUsed' ) "
|
|
"DESC 'Number of pages in use' "
|
|
"SUP monitorCounter "
|
|
"NO-USER-MODIFICATION "
|
|
"USAGE dSAOperation )",
|
|
&ad_olmMDBPagesUsed },
|
|
|
|
{ "( olmMDBAttributes:3 "
|
|
"NAME ( 'olmMDBPagesFree' ) "
|
|
"DESC 'Number of free pages' "
|
|
"SUP monitorCounter "
|
|
"NO-USER-MODIFICATION "
|
|
"USAGE dSAOperation )",
|
|
&ad_olmMDBPagesFree },
|
|
|
|
{ "( olmMDBAttributes:4 "
|
|
"NAME ( 'olmMDBReadersMax' ) "
|
|
"DESC 'Maximum number of readers' "
|
|
"SUP monitorCounter "
|
|
"NO-USER-MODIFICATION "
|
|
"USAGE dSAOperation )",
|
|
&ad_olmMDBReadersMax },
|
|
|
|
{ "( olmMDBAttributes:5 "
|
|
"NAME ( 'olmMDBReadersUsed' ) "
|
|
"DESC 'Number of readers in use' "
|
|
"SUP monitorCounter "
|
|
"NO-USER-MODIFICATION "
|
|
"USAGE dSAOperation )",
|
|
&ad_olmMDBReadersUsed },
|
|
|
|
{ "( olmMDBAttributes:6 "
|
|
"NAME ( 'olmMDBEntries' ) "
|
|
"DESC 'Number of entries in DB' "
|
|
"SUP monitorCounter "
|
|
"NO-USER-MODIFICATION "
|
|
"USAGE dSAOperation )",
|
|
&ad_olmMDBEntries },
|
|
{ NULL }
|
|
};
|
|
|
|
static struct {
|
|
char *desc;
|
|
ObjectClass **oc;
|
|
} s_oc[] = {
|
|
/* augments an existing object, so it must be AUXILIARY
|
|
* FIXME: derive from some ABSTRACT "monitoredEntity"? */
|
|
{ "( olmMDBObjectClasses:2 "
|
|
"NAME ( 'olmMDBDatabase' ) "
|
|
"SUP top AUXILIARY "
|
|
"MAY ( "
|
|
"olmDbDirectory "
|
|
#ifdef MDB_MONITOR_IDX
|
|
"$ olmDbNotIndexed "
|
|
#endif /* MDB_MONITOR_IDX */
|
|
"$ olmMDBPagesMax $ olmMDBPagesUsed $ olmMDBPagesFree "
|
|
"$ olmMDBReadersMax $ olmMDBReadersUsed $ olmMDBEntries "
|
|
") )",
|
|
&oc_olmMDBDatabase },
|
|
|
|
{ NULL }
|
|
};
|
|
|
|
static int
|
|
mdb_monitor_update(
|
|
Operation *op,
|
|
SlapReply *rs,
|
|
Entry *e,
|
|
void *priv )
|
|
{
|
|
struct mdb_info *mdb = (struct mdb_info *) priv;
|
|
Attribute *a;
|
|
char buf[ BUFSIZ ];
|
|
struct berval bv;
|
|
MDB_stat mst;
|
|
MDB_envinfo mei;
|
|
MDB_txn *txn;
|
|
int rc;
|
|
|
|
#ifdef MDB_MONITOR_IDX
|
|
|
|
mdb_monitor_idx_entry_add( mdb, e );
|
|
#endif /* MDB_MONITOR_IDX */
|
|
|
|
mdb_env_stat( mdb->mi_dbenv, &mst );
|
|
mdb_env_info( mdb->mi_dbenv, &mei );
|
|
|
|
a = attr_find( e->e_attrs, ad_olmMDBPagesMax );
|
|
assert( a != NULL );
|
|
bv.bv_val = buf;
|
|
bv.bv_len = snprintf( buf, sizeof( buf ), "%lu", mei.me_mapsize / mst.ms_psize );
|
|
ber_bvreplace( &a->a_vals[ 0 ], &bv );
|
|
|
|
a = attr_find( e->e_attrs, ad_olmMDBPagesUsed );
|
|
assert( a != NULL );
|
|
bv.bv_val = buf;
|
|
bv.bv_len = snprintf( buf, sizeof( buf ), "%lu", mei.me_last_pgno+1 );
|
|
ber_bvreplace( &a->a_vals[ 0 ], &bv );
|
|
|
|
a = attr_find( e->e_attrs, ad_olmMDBReadersMax );
|
|
assert( a != NULL );
|
|
bv.bv_val = buf;
|
|
bv.bv_len = snprintf( buf, sizeof( buf ), "%u", mei.me_maxreaders );
|
|
ber_bvreplace( &a->a_vals[ 0 ], &bv );
|
|
|
|
a = attr_find( e->e_attrs, ad_olmMDBReadersUsed );
|
|
assert( a != NULL );
|
|
bv.bv_val = buf;
|
|
bv.bv_len = snprintf( buf, sizeof( buf ), "%u", mei.me_numreaders );
|
|
ber_bvreplace( &a->a_vals[ 0 ], &bv );
|
|
|
|
rc = mdb_txn_begin( mdb->mi_dbenv, NULL, MDB_RDONLY, &txn );
|
|
if ( !rc ) {
|
|
MDB_cursor *cursor;
|
|
MDB_val key, data;
|
|
size_t pages = 0, *iptr;
|
|
|
|
rc = mdb_cursor_open( txn, 0, &cursor );
|
|
if ( !rc ) {
|
|
while (( rc = mdb_cursor_get( cursor, &key, &data, MDB_NEXT )) == 0 ) {
|
|
iptr = data.mv_data;
|
|
pages += *iptr;
|
|
}
|
|
mdb_cursor_close( cursor );
|
|
}
|
|
|
|
mdb_stat( txn, mdb->mi_id2entry, &mst );
|
|
a = attr_find( e->e_attrs, ad_olmMDBEntries );
|
|
assert( a != NULL );
|
|
bv.bv_val = buf;
|
|
bv.bv_len = snprintf( buf, sizeof( buf ), "%lu", mst.ms_entries );
|
|
ber_bvreplace( &a->a_vals[ 0 ], &bv );
|
|
|
|
mdb_txn_abort( txn );
|
|
|
|
a = attr_find( e->e_attrs, ad_olmMDBPagesFree );
|
|
assert( a != NULL );
|
|
bv.bv_val = buf;
|
|
bv.bv_len = snprintf( buf, sizeof( buf ), "%lu", pages );
|
|
ber_bvreplace( &a->a_vals[ 0 ], &bv );
|
|
}
|
|
return SLAP_CB_CONTINUE;
|
|
}
|
|
|
|
#if 0 /* uncomment if required */
|
|
static int
|
|
mdb_monitor_modify(
|
|
Operation *op,
|
|
SlapReply *rs,
|
|
Entry *e,
|
|
void *priv )
|
|
{
|
|
return SLAP_CB_CONTINUE;
|
|
}
|
|
#endif
|
|
|
|
static int
|
|
mdb_monitor_free(
|
|
Entry *e,
|
|
void **priv )
|
|
{
|
|
struct berval values[ 2 ];
|
|
Modification mod = { 0 };
|
|
|
|
const char *text;
|
|
char textbuf[ SLAP_TEXT_BUFLEN ];
|
|
|
|
int i, rc;
|
|
|
|
/* NOTE: if slap_shutdown != 0, priv might have already been freed */
|
|
*priv = NULL;
|
|
|
|
/* Remove objectClass */
|
|
mod.sm_op = LDAP_MOD_DELETE;
|
|
mod.sm_desc = slap_schema.si_ad_objectClass;
|
|
mod.sm_values = values;
|
|
mod.sm_numvals = 1;
|
|
values[ 0 ] = oc_olmMDBDatabase->soc_cname;
|
|
BER_BVZERO( &values[ 1 ] );
|
|
|
|
rc = modify_delete_values( e, &mod, 1, &text,
|
|
textbuf, sizeof( textbuf ) );
|
|
/* don't care too much about return code... */
|
|
|
|
/* remove attrs */
|
|
mod.sm_values = NULL;
|
|
mod.sm_numvals = 0;
|
|
for ( i = 0; s_at[ i ].desc != NULL; i++ ) {
|
|
mod.sm_desc = *s_at[ i ].ad;
|
|
rc = modify_delete_values( e, &mod, 1, &text,
|
|
textbuf, sizeof( textbuf ) );
|
|
/* don't care too much about return code... */
|
|
}
|
|
|
|
return SLAP_CB_CONTINUE;
|
|
}
|
|
|
|
/*
|
|
* call from within mdb_initialize()
|
|
*/
|
|
static int
|
|
mdb_monitor_initialize( void )
|
|
{
|
|
int i, code;
|
|
ConfigArgs c;
|
|
char *argv[ 3 ];
|
|
|
|
static int mdb_monitor_initialized = 0;
|
|
|
|
/* set to 0 when successfully initialized; otherwise, remember failure */
|
|
static int mdb_monitor_initialized_failure = 1;
|
|
|
|
if ( mdb_monitor_initialized++ ) {
|
|
return mdb_monitor_initialized_failure;
|
|
}
|
|
|
|
if ( backend_info( "monitor" ) == NULL ) {
|
|
return -1;
|
|
}
|
|
|
|
/* register schema here */
|
|
|
|
argv[ 0 ] = "back-mdb monitor";
|
|
c.argv = argv;
|
|
c.argc = 3;
|
|
c.fname = argv[0];
|
|
|
|
for ( i = 0; s_oid[ i ].name; i++ ) {
|
|
c.lineno = i;
|
|
argv[ 1 ] = s_oid[ i ].name;
|
|
argv[ 2 ] = s_oid[ i ].oid;
|
|
|
|
if ( parse_oidm( &c, 0, NULL ) != 0 ) {
|
|
Debug( LDAP_DEBUG_ANY, LDAP_XSTRING(mdb_monitor_initialize)
|
|
": unable to add "
|
|
"objectIdentifier \"%s=%s\"\n",
|
|
s_oid[ i ].name, s_oid[ i ].oid );
|
|
return 2;
|
|
}
|
|
}
|
|
|
|
for ( i = 0; s_at[ i ].desc != NULL; i++ ) {
|
|
code = register_at( s_at[ i ].desc, s_at[ i ].ad, 1 );
|
|
if ( code != LDAP_SUCCESS ) {
|
|
Debug( LDAP_DEBUG_ANY, LDAP_XSTRING(mdb_monitor_initialize)
|
|
": register_at failed for attributeType (%s)\n",
|
|
s_at[ i ].desc );
|
|
return 3;
|
|
|
|
} else {
|
|
(*s_at[ i ].ad)->ad_type->sat_flags |= SLAP_AT_HIDE;
|
|
}
|
|
}
|
|
|
|
for ( i = 0; s_oc[ i ].desc != NULL; i++ ) {
|
|
code = register_oc( s_oc[ i ].desc, s_oc[ i ].oc, 1 );
|
|
if ( code != LDAP_SUCCESS ) {
|
|
Debug( LDAP_DEBUG_ANY, LDAP_XSTRING(mdb_monitor_initialize)
|
|
": register_oc failed for objectClass (%s)\n",
|
|
s_oc[ i ].desc );
|
|
return 4;
|
|
|
|
} else {
|
|
(*s_oc[ i ].oc)->soc_flags |= SLAP_OC_HIDE;
|
|
}
|
|
}
|
|
|
|
return ( mdb_monitor_initialized_failure = LDAP_SUCCESS );
|
|
}
|
|
|
|
/*
|
|
* call from within mdb_db_init()
|
|
*/
|
|
int
|
|
mdb_monitor_db_init( BackendDB *be )
|
|
{
|
|
#ifdef MDB_MONITOR_IDX
|
|
struct mdb_info *mdb = (struct mdb_info *) be->be_private;
|
|
#endif /* MDB_MONITOR_IDX */
|
|
|
|
if ( mdb_monitor_initialize() == LDAP_SUCCESS ) {
|
|
/* monitoring in back-mdb is on by default */
|
|
SLAP_DBFLAGS( be ) |= SLAP_DBFLAG_MONITORING;
|
|
}
|
|
|
|
#ifdef MDB_MONITOR_IDX
|
|
mdb->mi_idx = NULL;
|
|
ldap_pvt_thread_mutex_init( &mdb->mi_idx_mutex );
|
|
#endif /* MDB_MONITOR_IDX */
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* call from within mdb_db_open()
|
|
*/
|
|
int
|
|
mdb_monitor_db_open( BackendDB *be )
|
|
{
|
|
struct mdb_info *mdb = (struct mdb_info *) be->be_private;
|
|
Attribute *a, *next;
|
|
monitor_callback_t *cb = NULL;
|
|
int rc = 0;
|
|
BackendInfo *mi;
|
|
monitor_extra_t *mbe;
|
|
|
|
if ( !SLAP_DBMONITORING( be ) ) {
|
|
return 0;
|
|
}
|
|
|
|
mi = backend_info( "monitor" );
|
|
if ( !mi || !mi->bi_extra ) {
|
|
SLAP_DBFLAGS( be ) ^= SLAP_DBFLAG_MONITORING;
|
|
return 0;
|
|
}
|
|
mbe = mi->bi_extra;
|
|
|
|
/* don't bother if monitor is not configured */
|
|
if ( !mbe->is_configured() ) {
|
|
static int warning = 0;
|
|
|
|
if ( warning++ == 0 ) {
|
|
Debug( LDAP_DEBUG_CONFIG, LDAP_XSTRING(mdb_monitor_db_open)
|
|
": monitoring disabled; "
|
|
"configure monitor database to enable\n" );
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/* alloc as many as required (plus 1 for objectClass) */
|
|
a = attrs_alloc( 1 + 7 );
|
|
if ( a == NULL ) {
|
|
rc = 1;
|
|
goto cleanup;
|
|
}
|
|
|
|
a->a_desc = slap_schema.si_ad_objectClass;
|
|
attr_valadd( a, &oc_olmMDBDatabase->soc_cname, NULL, 1 );
|
|
next = a->a_next;
|
|
|
|
{
|
|
struct berval bv = BER_BVC( "0" );
|
|
|
|
next->a_desc = ad_olmMDBPagesMax;
|
|
attr_valadd( next, &bv, NULL, 1 );
|
|
next = next->a_next;
|
|
|
|
next->a_desc = ad_olmMDBPagesUsed;
|
|
attr_valadd( next, &bv, NULL, 1 );
|
|
next = next->a_next;
|
|
|
|
next->a_desc = ad_olmMDBPagesFree;
|
|
attr_valadd( next, &bv, NULL, 1 );
|
|
next = next->a_next;
|
|
|
|
next->a_desc = ad_olmMDBReadersMax;
|
|
attr_valadd( next, &bv, NULL, 1 );
|
|
next = next->a_next;
|
|
|
|
next->a_desc = ad_olmMDBReadersUsed;
|
|
attr_valadd( next, &bv, NULL, 1 );
|
|
next = next->a_next;
|
|
|
|
next->a_desc = ad_olmMDBEntries;
|
|
attr_valadd( next, &bv, NULL, 1 );
|
|
next = next->a_next;
|
|
}
|
|
|
|
{
|
|
struct berval bv, nbv;
|
|
ber_len_t pathlen = 0, len = 0;
|
|
char path[ MAXPATHLEN ] = { '\0' };
|
|
char *fname = mdb->mi_dbenv_home,
|
|
*ptr;
|
|
|
|
len = strlen( fname );
|
|
if ( fname[ 0 ] != '/' ) {
|
|
/* get full path name */
|
|
getcwd( path, sizeof( path ) );
|
|
pathlen = strlen( path );
|
|
|
|
if ( fname[ 0 ] == '.' && fname[ 1 ] == '/' ) {
|
|
fname += 2;
|
|
len -= 2;
|
|
}
|
|
}
|
|
|
|
bv.bv_len = pathlen + STRLENOF( "/" ) + len;
|
|
ptr = bv.bv_val = ch_malloc( bv.bv_len + STRLENOF( "/" ) + 1 );
|
|
if ( pathlen ) {
|
|
ptr = lutil_strncopy( ptr, path, pathlen );
|
|
ptr[ 0 ] = '/';
|
|
ptr++;
|
|
}
|
|
ptr = lutil_strncopy( ptr, fname, len );
|
|
if ( ptr[ -1 ] != '/' ) {
|
|
ptr[ 0 ] = '/';
|
|
ptr++;
|
|
}
|
|
ptr[ 0 ] = '\0';
|
|
|
|
attr_normalize_one( ad_olmDbDirectory, &bv, &nbv, NULL );
|
|
|
|
next->a_desc = ad_olmDbDirectory;
|
|
next->a_vals = ch_calloc( sizeof( struct berval ), 2 );
|
|
next->a_vals[ 0 ] = bv;
|
|
next->a_numvals = 1;
|
|
|
|
if ( BER_BVISNULL( &nbv ) ) {
|
|
next->a_nvals = next->a_vals;
|
|
|
|
} else {
|
|
next->a_nvals = ch_calloc( sizeof( struct berval ), 2 );
|
|
next->a_nvals[ 0 ] = nbv;
|
|
}
|
|
|
|
next = next->a_next;
|
|
}
|
|
|
|
cb = ch_calloc( sizeof( monitor_callback_t ), 1 );
|
|
cb->mc_update = mdb_monitor_update;
|
|
#if 0 /* uncomment if required */
|
|
cb->mc_modify = mdb_monitor_modify;
|
|
#endif
|
|
cb->mc_free = mdb_monitor_free;
|
|
cb->mc_private = (void *)mdb;
|
|
|
|
/* make sure the database is registered; then add monitor attributes */
|
|
rc = mbe->register_database( be, &mdb->mi_monitor.mdm_ndn );
|
|
if ( rc == 0 ) {
|
|
rc = mbe->register_entry_attrs( &mdb->mi_monitor.mdm_ndn, a, cb,
|
|
NULL, -1, NULL );
|
|
}
|
|
|
|
cleanup:;
|
|
if ( rc != 0 ) {
|
|
if ( cb != NULL ) {
|
|
ch_free( cb );
|
|
cb = NULL;
|
|
}
|
|
|
|
if ( a != NULL ) {
|
|
attrs_free( a );
|
|
a = NULL;
|
|
}
|
|
}
|
|
|
|
/* store for cleanup */
|
|
mdb->mi_monitor.mdm_cb = (void *)cb;
|
|
|
|
/* we don't need to keep track of the attributes, because
|
|
* mdb_monitor_free() takes care of everything */
|
|
if ( a != NULL ) {
|
|
attrs_free( a );
|
|
}
|
|
|
|
return rc;
|
|
}
|
|
|
|
/*
|
|
* call from within mdb_db_close()
|
|
*/
|
|
int
|
|
mdb_monitor_db_close( BackendDB *be )
|
|
{
|
|
struct mdb_info *mdb = (struct mdb_info *) be->be_private;
|
|
|
|
if ( !BER_BVISNULL( &mdb->mi_monitor.mdm_ndn ) ) {
|
|
BackendInfo *mi = backend_info( "monitor" );
|
|
monitor_extra_t *mbe;
|
|
|
|
if ( mi && mi->bi_extra ) {
|
|
mbe = mi->bi_extra;
|
|
mbe->unregister_entry_callback( &mdb->mi_monitor.mdm_ndn,
|
|
(monitor_callback_t *)mdb->mi_monitor.mdm_cb,
|
|
NULL, 0, NULL );
|
|
}
|
|
|
|
memset( &mdb->mi_monitor, 0, sizeof( mdb->mi_monitor ) );
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* call from within mdb_db_destroy()
|
|
*/
|
|
int
|
|
mdb_monitor_db_destroy( BackendDB *be )
|
|
{
|
|
#ifdef MDB_MONITOR_IDX
|
|
struct mdb_info *mdb = (struct mdb_info *) be->be_private;
|
|
|
|
/* TODO: free tree */
|
|
ldap_pvt_thread_mutex_destroy( &mdb->mi_idx_mutex );
|
|
avl_free( mdb->mi_idx, ch_free );
|
|
#endif /* MDB_MONITOR_IDX */
|
|
|
|
return 0;
|
|
}
|
|
|
|
#ifdef MDB_MONITOR_IDX
|
|
|
|
#define MDB_MONITOR_IDX_TYPES (4)
|
|
|
|
typedef struct monitor_idx_t monitor_idx_t;
|
|
|
|
struct monitor_idx_t {
|
|
AttributeDescription *idx_ad;
|
|
unsigned long idx_count[MDB_MONITOR_IDX_TYPES];
|
|
};
|
|
|
|
static int
|
|
mdb_monitor_bitmask2key( slap_mask_t bitmask )
|
|
{
|
|
int key;
|
|
|
|
for ( key = 0; key < 8 * (int)sizeof(slap_mask_t) && !( bitmask & 0x1U );
|
|
key++ )
|
|
bitmask >>= 1;
|
|
|
|
return key;
|
|
}
|
|
|
|
static struct berval idxbv[] = {
|
|
BER_BVC( "present=" ),
|
|
BER_BVC( "equality=" ),
|
|
BER_BVC( "approx=" ),
|
|
BER_BVC( "substr=" ),
|
|
BER_BVNULL
|
|
};
|
|
|
|
static ber_len_t
|
|
mdb_monitor_idx2len( monitor_idx_t *idx )
|
|
{
|
|
int i;
|
|
ber_len_t len = 0;
|
|
|
|
for ( i = 0; i < MDB_MONITOR_IDX_TYPES; i++ ) {
|
|
if ( idx->idx_count[ i ] != 0 ) {
|
|
len += idxbv[i].bv_len;
|
|
}
|
|
}
|
|
|
|
return len;
|
|
}
|
|
|
|
static int
|
|
monitor_idx_cmp( const void *p1, const void *p2 )
|
|
{
|
|
const monitor_idx_t *idx1 = (const monitor_idx_t *)p1;
|
|
const monitor_idx_t *idx2 = (const monitor_idx_t *)p2;
|
|
|
|
return SLAP_PTRCMP( idx1->idx_ad, idx2->idx_ad );
|
|
}
|
|
|
|
static int
|
|
monitor_idx_dup( void *p1, void *p2 )
|
|
{
|
|
monitor_idx_t *idx1 = (monitor_idx_t *)p1;
|
|
monitor_idx_t *idx2 = (monitor_idx_t *)p2;
|
|
|
|
return SLAP_PTRCMP( idx1->idx_ad, idx2->idx_ad ) == 0 ? -1 : 0;
|
|
}
|
|
|
|
int
|
|
mdb_monitor_idx_add(
|
|
struct mdb_info *mdb,
|
|
AttributeDescription *desc,
|
|
slap_mask_t type )
|
|
{
|
|
monitor_idx_t idx_dummy = { 0 },
|
|
*idx;
|
|
int rc = 0, key;
|
|
|
|
idx_dummy.idx_ad = desc;
|
|
key = mdb_monitor_bitmask2key( type ) - 1;
|
|
if ( key >= MDB_MONITOR_IDX_TYPES ) {
|
|
/* invalid index type */
|
|
return -1;
|
|
}
|
|
|
|
ldap_pvt_thread_mutex_lock( &mdb->mi_idx_mutex );
|
|
|
|
idx = (monitor_idx_t *)avl_find( mdb->mi_idx,
|
|
(caddr_t)&idx_dummy, monitor_idx_cmp );
|
|
if ( idx == NULL ) {
|
|
idx = (monitor_idx_t *)ch_calloc( sizeof( monitor_idx_t ), 1 );
|
|
idx->idx_ad = desc;
|
|
idx->idx_count[ key ] = 1;
|
|
|
|
switch ( avl_insert( &mdb->mi_idx, (caddr_t)idx,
|
|
monitor_idx_cmp, monitor_idx_dup ) )
|
|
{
|
|
case 0:
|
|
break;
|
|
|
|
default:
|
|
ch_free( idx );
|
|
rc = -1;
|
|
}
|
|
|
|
} else {
|
|
idx->idx_count[ key ]++;
|
|
}
|
|
|
|
ldap_pvt_thread_mutex_unlock( &mdb->mi_idx_mutex );
|
|
|
|
return rc;
|
|
}
|
|
|
|
static int
|
|
mdb_monitor_idx_apply( void *v_idx, void *v_valp )
|
|
{
|
|
monitor_idx_t *idx = (monitor_idx_t *)v_idx;
|
|
BerVarray *valp = (BerVarray *)v_valp;
|
|
|
|
struct berval bv;
|
|
char *ptr;
|
|
char count_buf[ MDB_MONITOR_IDX_TYPES ][ SLAP_TEXT_BUFLEN ];
|
|
ber_len_t count_len[ MDB_MONITOR_IDX_TYPES ],
|
|
idx_len;
|
|
int i, num = 0;
|
|
|
|
idx_len = mdb_monitor_idx2len( idx );
|
|
|
|
bv.bv_len = 0;
|
|
for ( i = 0; i < MDB_MONITOR_IDX_TYPES; i++ ) {
|
|
if ( idx->idx_count[ i ] == 0 ) {
|
|
continue;
|
|
}
|
|
|
|
count_len[ i ] = snprintf( count_buf[ i ],
|
|
sizeof( count_buf[ i ] ), "%lu", idx->idx_count[ i ] );
|
|
bv.bv_len += count_len[ i ];
|
|
num++;
|
|
}
|
|
|
|
bv.bv_len += idx->idx_ad->ad_cname.bv_len
|
|
+ num
|
|
+ idx_len;
|
|
ptr = bv.bv_val = ch_malloc( bv.bv_len + 1 );
|
|
ptr = lutil_strcopy( ptr, idx->idx_ad->ad_cname.bv_val );
|
|
for ( i = 0; i < MDB_MONITOR_IDX_TYPES; i++ ) {
|
|
if ( idx->idx_count[ i ] == 0 ) {
|
|
continue;
|
|
}
|
|
|
|
ptr[ 0 ] = '#';
|
|
++ptr;
|
|
ptr = lutil_strcopy( ptr, idxbv[ i ].bv_val );
|
|
ptr = lutil_strcopy( ptr, count_buf[ i ] );
|
|
}
|
|
|
|
ber_bvarray_add( valp, &bv );
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
mdb_monitor_idx_entry_add(
|
|
struct mdb_info *mdb,
|
|
Entry *e )
|
|
{
|
|
BerVarray vals = NULL;
|
|
Attribute *a;
|
|
|
|
a = attr_find( e->e_attrs, ad_olmDbNotIndexed );
|
|
|
|
ldap_pvt_thread_mutex_lock( &mdb->mi_idx_mutex );
|
|
|
|
avl_apply( mdb->mi_idx, mdb_monitor_idx_apply,
|
|
&vals, -1, AVL_INORDER );
|
|
|
|
ldap_pvt_thread_mutex_unlock( &mdb->mi_idx_mutex );
|
|
|
|
if ( vals != NULL ) {
|
|
if ( a != NULL ) {
|
|
assert( a->a_nvals == a->a_vals );
|
|
|
|
ber_bvarray_free( a->a_vals );
|
|
|
|
} else {
|
|
Attribute **ap;
|
|
|
|
for ( ap = &e->e_attrs; *ap != NULL; ap = &(*ap)->a_next )
|
|
;
|
|
*ap = attr_alloc( ad_olmDbNotIndexed );
|
|
a = *ap;
|
|
}
|
|
a->a_vals = vals;
|
|
a->a_nvals = a->a_vals;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
#endif /* MDB_MONITOR_IDX */
|