git.sur5r.net Git - openldap/blob - servers/slapd/back-monitor/rww.c

   1 /* readw.c - deal with read waiters subsystem */
   2 /* $OpenLDAP$ */
   3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
   4  *
   5  * Copyright 2001-2006 The OpenLDAP Foundation.
   6  * Portions Copyright 2001-2003 Pierangelo Masarati.
   7  * All rights reserved.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted only as authorized by the OpenLDAP
  11  * Public License.
  12  *
  13  * A copy of this license is available in file LICENSE in the
  14  * top-level directory of the distribution or, alternatively, at
  15  * <http://www.OpenLDAP.org/license.html>.
  16  */
  17 /* ACKNOWLEDGEMENTS:
  18  * This work was initially developed by Pierangelo Masarati for inclusion
  19  * in OpenLDAP Software.
  20  */
  21
  22 #include "portable.h"
  23
  24 #include <stdio.h>
  25 #include <ac/string.h>
  26
  27 #include "slap.h"
  28 #include "lutil.h"
  29 #include "back-monitor.h"
  30
  31 static int
  32 monitor_subsys_rww_destroy(
  33         BackendDB               *be,
  34         monitor_subsys_t        *ms );
  35
  36 static int
  37 monitor_subsys_rww_update(
  38         Operation               *op,
  39         SlapReply               *rs,
  40         Entry                   *e );
  41
  42 enum {
  43         MONITOR_RWW_READ = 0,
  44         MONITOR_RWW_WRITE,
  45
  46         MONITOR_RWW_LAST
  47 };
  48
  49 static struct monitor_rww_t {
  50         struct berval   rdn;
  51         struct berval   nrdn;
  52 } monitor_rww[] = {
  53         { BER_BVC("cn=Read"),           BER_BVNULL },
  54         { BER_BVC("cn=Write"),          BER_BVNULL },
  55         { BER_BVNULL,                   BER_BVNULL }
  56 };
  57
  58 int
  59 monitor_subsys_rww_init(
  60         BackendDB               *be,
  61         monitor_subsys_t        *ms )
  62 {
  63         monitor_info_t  *mi;
  64
  65         Entry           **ep, *e_conn;
  66         monitor_entry_t *mp;
  67         int                     i;
  68
  69         assert( be != NULL );
  70
  71         ms->mss_destroy = monitor_subsys_rww_destroy;
  72         ms->mss_update = monitor_subsys_rww_update;
  73
  74         mi = ( monitor_info_t * )be->be_private;
  75
  76         if ( monitor_cache_get( mi, &ms->mss_ndn, &e_conn ) ) {
  77                 Debug( LDAP_DEBUG_ANY,
  78                         "monitor_subsys_rww_init: "
  79                         "unable to get entry \"%s\"\n",
  80                         ms->mss_ndn.bv_val, 0, 0 );
  81                 return( -1 );
  82         }
  83
  84         mp = ( monitor_entry_t * )e_conn->e_private;
  85         mp->mp_children = NULL;
  86         ep = &mp->mp_children;
  87
  88         for ( i = 0; i < MONITOR_RWW_LAST; i++ ) {
  89                 char                    buf[ BACKMONITOR_BUFSIZE ];
  90                 struct berval           nrdn, bv;
  91                 Entry                   *e;
  92
  93                 e = monitor_entry_stub( &ms->mss_dn, &ms->mss_ndn, &monitor_rww[i].rdn,
  94                         mi->mi_oc_monitorCounterObject, mi, NULL, NULL );
  95                 if ( e == NULL ) {
  96                         Debug( LDAP_DEBUG_ANY,
  97                                 "monitor_subsys_rww_init: "
  98                                 "unable to create entry \"cn=Read,%s\"\n",
  99                                 ms->mss_ndn.bv_val, 0, 0 );
 100                         return( -1 );
 101                 }
 102
 103                 /* steal normalized RDN */
 104                 dnRdn( &e->e_nname, &nrdn );
 105                 ber_dupbv( &monitor_rww[ i ].nrdn, &nrdn );
 106
 107                 BER_BVSTR( &bv, "0" );
 108                 attr_merge_normalize_one( e, mi->mi_ad_monitorCounter, &bv, NULL );
 109
 110                 mp = monitor_entrypriv_create();
 111                 if ( mp == NULL ) {
 112                         return -1;
 113                 }
 114                 e->e_private = ( void * )mp;
 115                 mp->mp_info = ms;
 116                 mp->mp_flags = ms->mss_flags \
 117                         | MONITOR_F_SUB | MONITOR_F_PERSISTENT;
 118
 119                 if ( monitor_cache_add( mi, e ) ) {
 120                         Debug( LDAP_DEBUG_ANY,
 121                                 "monitor_subsys_rww_init: "
 122                                 "unable to add entry \"%s,%s\"\n",
 123                                 monitor_rww[ i ].rdn.bv_val,
 124                                 ms->mss_ndn.bv_val, 0 );
 125                         return( -1 );
 126                 }
 127
 128                 *ep = e;
 129                 ep = &mp->mp_next;
 130         }
 131
 132         monitor_cache_release( mi, e_conn );
 133
 134         return( 0 );
 135 }
 136
 137 static int
 138 monitor_subsys_rww_destroy(
 139         BackendDB               *be,
 140         monitor_subsys_t        *ms )
 141 {
 142         int             i;
 143
 144         for ( i = 0; i < MONITOR_RWW_LAST; i++ ) {
 145                 ber_memfree_x( monitor_rww[ i ].nrdn.bv_val, NULL );
 146         }
 147
 148         return 0;
 149 }
 150
 151 static int
 152 monitor_subsys_rww_update(
 153         Operation               *op,
 154         SlapReply               *rs,
 155         Entry                   *e )
 156 {
 157         monitor_info_t *mi = (monitor_info_t *)op->o_bd->be_private;
 158         Connection      *c;
 159         int             connindex;
 160         long            nconns, nwritewaiters, nreadwaiters;
 161
 162         int             i;
 163         struct berval   nrdn;
 164
 165         Attribute       *a;
 166         char            buf[] = "+9223372036854775807L";
 167         long            num = 0;
 168         ber_len_t       len;
 169
 170         assert( mi != NULL );
 171         assert( e != NULL );
 172
 173         dnRdn( &e->e_nname, &nrdn );
 174
 175         for ( i = 0; !BER_BVISNULL( &monitor_rww[ i ].nrdn ); i++ ) {
 176                 if ( dn_match( &nrdn, &monitor_rww[ i ].nrdn ) ) {
 177                         break;
 178                 }
 179         }
 180
 181         if ( i == MONITOR_RWW_LAST ) {
 182                 return SLAP_CB_CONTINUE;
 183         }
 184
 185         nconns = nwritewaiters = nreadwaiters = 0;
 186         for ( c = connection_first( &connindex );
 187                         c != NULL;
 188                         c = connection_next( c, &connindex ), nconns++ )
 189         {
 190                 if ( c->c_writewaiter ) {
 191                         nwritewaiters++;
 192                 }
 193
 194                 /* FIXME: ?!? */
 195                 if ( c->c_currentber != NULL ) {
 196                         nreadwaiters++;
 197                 }
 198         }
 199         connection_done(c);
 200
 201         switch ( i ) {
 202         case MONITOR_RWW_READ:
 203                 num = nreadwaiters;
 204                 break;
 205
 206         case MONITOR_RWW_WRITE:
 207                 num = nwritewaiters;
 208                 break;
 209
 210         default:
 211                 assert( 0 );
 212         }
 213
 214         snprintf( buf, sizeof( buf ), "%ld", num );
 215
 216         a = attr_find( e->e_attrs, mi->mi_ad_monitorCounter );
 217         assert( a != NULL );
 218         len = strlen( buf );
 219         if ( len > a->a_vals[ 0 ].bv_len ) {
 220                 a->a_vals[ 0 ].bv_val = ber_memrealloc( a->a_vals[ 0 ].bv_val, len + 1 );
 221                 if ( BER_BVISNULL( &a->a_vals[ 0 ] ) ) {
 222                         BER_BVZERO( &a->a_vals[ 0 ] );
 223                         return SLAP_CB_CONTINUE;
 224                 }
 225         }
 226         AC_MEMCPY( a->a_vals[ 0 ].bv_val, buf, len + 1 );
 227         a->a_vals[ 0 ].bv_len = len;
 228
 229         /* FIXME: touch modifyTimestamp? */
 230
 231         return SLAP_CB_CONTINUE;
 232 }
 233