openldap/servers/slapd/back-monitor/sent.c
Quanah Gibson-Mount b45a6a7dc7 Happy New Year!
2019-01-14 18:46:16 +00:00

242 lines
5.4 KiB
C

/* sent.c - deal with data sent subsystem */
/* $OpenLDAP$ */
/* This work is part of OpenLDAP Software <http://www.openldap.org/>.
*
* Copyright 2001-2019 The OpenLDAP Foundation.
* Portions Copyright 2001-2003 Pierangelo Masarati.
* 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 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 Pierangelo Masarati for inclusion
* in OpenLDAP Software.
*/
#include "portable.h"
#include <stdio.h>
#include <ac/string.h>
#include "slap.h"
#include "back-monitor.h"
static int
monitor_subsys_sent_destroy(
BackendDB *be,
monitor_subsys_t *ms );
static int
monitor_subsys_sent_update(
Operation *op,
SlapReply *rs,
Entry *e );
enum {
MONITOR_SENT_BYTES = 0,
MONITOR_SENT_PDU,
MONITOR_SENT_ENTRIES,
MONITOR_SENT_REFERRALS,
MONITOR_SENT_LAST
};
struct monitor_sent_t {
struct berval rdn;
struct berval nrdn;
} monitor_sent[] = {
{ BER_BVC("cn=Bytes"), BER_BVNULL },
{ BER_BVC("cn=PDU"), BER_BVNULL },
{ BER_BVC("cn=Entries"), BER_BVNULL },
{ BER_BVC("cn=Referrals"), BER_BVNULL },
{ BER_BVNULL, BER_BVNULL }
};
int
monitor_subsys_sent_init(
BackendDB *be,
monitor_subsys_t *ms )
{
monitor_info_t *mi;
Entry **ep, *e_sent;
monitor_entry_t *mp;
int i;
assert( be != NULL );
ms->mss_destroy = monitor_subsys_sent_destroy;
ms->mss_update = monitor_subsys_sent_update;
mi = ( monitor_info_t * )be->be_private;
if ( monitor_cache_get( mi, &ms->mss_ndn, &e_sent ) ) {
Debug( LDAP_DEBUG_ANY,
"monitor_subsys_sent_init: "
"unable to get entry \"%s\"\n",
ms->mss_ndn.bv_val, 0, 0 );
return( -1 );
}
mp = ( monitor_entry_t * )e_sent->e_private;
mp->mp_children = NULL;
ep = &mp->mp_children;
for ( i = 0; i < MONITOR_SENT_LAST; i++ ) {
struct berval nrdn, bv;
Entry *e;
e = monitor_entry_stub( &ms->mss_dn, &ms->mss_ndn,
&monitor_sent[i].rdn, mi->mi_oc_monitorCounterObject,
NULL, NULL );
if ( e == NULL ) {
Debug( LDAP_DEBUG_ANY,
"monitor_subsys_sent_init: "
"unable to create entry \"%s,%s\"\n",
monitor_sent[ i ].rdn.bv_val,
ms->mss_ndn.bv_val, 0 );
return( -1 );
}
/* steal normalized RDN */
dnRdn( &e->e_nname, &nrdn );
ber_dupbv( &monitor_sent[ i ].nrdn, &nrdn );
BER_BVSTR( &bv, "0" );
attr_merge_one( e, mi->mi_ad_monitorCounter, &bv, NULL );
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_sent_init: "
"unable to add entry \"%s,%s\"\n",
monitor_sent[ i ].rdn.bv_val,
ms->mss_ndn.bv_val, 0 );
return( -1 );
}
*ep = e;
ep = &mp->mp_next;
}
monitor_cache_release( mi, e_sent );
return( 0 );
}
static int
monitor_subsys_sent_destroy(
BackendDB *be,
monitor_subsys_t *ms )
{
int i;
for ( i = 0; i < MONITOR_SENT_LAST; i++ ) {
if ( !BER_BVISNULL( &monitor_sent[ i ].nrdn ) ) {
ch_free( monitor_sent[ i ].nrdn.bv_val );
}
}
return 0;
}
static int
monitor_subsys_sent_update(
Operation *op,
SlapReply *rs,
Entry *e )
{
monitor_info_t *mi = ( monitor_info_t *)op->o_bd->be_private;
struct berval nrdn;
ldap_pvt_mp_t n;
Attribute *a;
slap_counters_t *sc;
int i;
assert( mi != NULL );
assert( e != NULL );
dnRdn( &e->e_nname, &nrdn );
for ( i = 0; i < MONITOR_SENT_LAST; i++ ) {
if ( dn_match( &nrdn, &monitor_sent[ i ].nrdn ) ) {
break;
}
}
if ( i == MONITOR_SENT_LAST ) {
return SLAP_CB_CONTINUE;
}
ldap_pvt_thread_mutex_lock(&slap_counters.sc_mutex);
switch ( i ) {
case MONITOR_SENT_ENTRIES:
ldap_pvt_mp_init_set( n, slap_counters.sc_entries );
for ( sc = slap_counters.sc_next; sc; sc = sc->sc_next ) {
ldap_pvt_thread_mutex_lock( &sc->sc_mutex );
ldap_pvt_mp_add( n, sc->sc_entries );
ldap_pvt_thread_mutex_unlock( &sc->sc_mutex );
}
break;
case MONITOR_SENT_REFERRALS:
ldap_pvt_mp_init_set( n, slap_counters.sc_refs );
for ( sc = slap_counters.sc_next; sc; sc = sc->sc_next ) {
ldap_pvt_thread_mutex_lock( &sc->sc_mutex );
ldap_pvt_mp_add( n, sc->sc_refs );
ldap_pvt_thread_mutex_unlock( &sc->sc_mutex );
}
break;
case MONITOR_SENT_PDU:
ldap_pvt_mp_init_set( n, slap_counters.sc_pdu );
for ( sc = slap_counters.sc_next; sc; sc = sc->sc_next ) {
ldap_pvt_thread_mutex_lock( &sc->sc_mutex );
ldap_pvt_mp_add( n, sc->sc_pdu );
ldap_pvt_thread_mutex_unlock( &sc->sc_mutex );
}
break;
case MONITOR_SENT_BYTES:
ldap_pvt_mp_init_set( n, slap_counters.sc_bytes );
for ( sc = slap_counters.sc_next; sc; sc = sc->sc_next ) {
ldap_pvt_thread_mutex_lock( &sc->sc_mutex );
ldap_pvt_mp_add( n, sc->sc_bytes );
ldap_pvt_thread_mutex_unlock( &sc->sc_mutex );
}
break;
default:
assert(0);
}
ldap_pvt_thread_mutex_unlock(&slap_counters.sc_mutex);
a = attr_find( e->e_attrs, mi->mi_ad_monitorCounter );
assert( a != NULL );
/* NOTE: no minus sign is allowed in the counters... */
UI2BV( &a->a_vals[ 0 ], n );
ldap_pvt_mp_clear( n );
/* FIXME: touch modifyTimestamp? */
return SLAP_CB_CONTINUE;
}