1/*	$NetBSD: sent.c,v 1.3 2021/08/14 16:15:00 christos Exp $	*/
2
3/* sent.c - deal with data sent subsystem */
4/* $OpenLDAP$ */
5/* This work is part of OpenLDAP Software <http://www.openldap.org/>.
6 *
7 * Copyright 2001-2021 The OpenLDAP Foundation.
8 * Portions Copyright 2001-2003 Pierangelo Masarati.
9 * All rights reserved.
10 *
11 * Redistribution and use in source and binary forms, with or without
12 * modification, are permitted only as authorized by the OpenLDAP
13 * Public License.
14 *
15 * A copy of this license is available in file LICENSE in the
16 * top-level directory of the distribution or, alternatively, at
17 * <http://www.OpenLDAP.org/license.html>.
18 */
19/* ACKNOWLEDGEMENTS:
20 * This work was initially developed by Pierangelo Masarati for inclusion
21 * in OpenLDAP Software.
22 */
23
24#include <sys/cdefs.h>
25__RCSID("$NetBSD: sent.c,v 1.3 2021/08/14 16:15:00 christos Exp $");
26
27#include "portable.h"
28
29#include <stdio.h>
30#include <ac/string.h>
31
32#include "slap.h"
33#include "back-monitor.h"
34
35static int
36monitor_subsys_sent_destroy(
37	BackendDB		*be,
38	monitor_subsys_t	*ms );
39
40static int
41monitor_subsys_sent_update(
42	Operation		*op,
43	SlapReply		*rs,
44	Entry                   *e );
45
46enum {
47	MONITOR_SENT_BYTES = 0,
48	MONITOR_SENT_PDU,
49	MONITOR_SENT_ENTRIES,
50	MONITOR_SENT_REFERRALS,
51
52	MONITOR_SENT_LAST
53};
54
55struct monitor_sent_t {
56	struct berval	rdn;
57	struct berval	nrdn;
58} monitor_sent[] = {
59	{ BER_BVC("cn=Bytes"),		BER_BVNULL },
60	{ BER_BVC("cn=PDU"),		BER_BVNULL },
61	{ BER_BVC("cn=Entries"),	BER_BVNULL },
62	{ BER_BVC("cn=Referrals"),	BER_BVNULL },
63	{ BER_BVNULL,			BER_BVNULL }
64};
65
66int
67monitor_subsys_sent_init(
68	BackendDB		*be,
69	monitor_subsys_t	*ms )
70{
71	monitor_info_t	*mi;
72
73	Entry		**ep, *e_sent;
74	monitor_entry_t	*mp;
75	int			i;
76
77	assert( be != NULL );
78
79	ms->mss_destroy = monitor_subsys_sent_destroy;
80	ms->mss_update = monitor_subsys_sent_update;
81
82	mi = ( monitor_info_t * )be->be_private;
83
84	if ( monitor_cache_get( mi, &ms->mss_ndn, &e_sent ) ) {
85		Debug( LDAP_DEBUG_ANY,
86			"monitor_subsys_sent_init: "
87			"unable to get entry \"%s\"\n",
88			ms->mss_ndn.bv_val );
89		return( -1 );
90	}
91
92	mp = ( monitor_entry_t * )e_sent->e_private;
93	mp->mp_children = NULL;
94	ep = &mp->mp_children;
95
96	for ( i = 0; i < MONITOR_SENT_LAST; i++ ) {
97		struct berval		nrdn, bv;
98		Entry			*e;
99
100		e = monitor_entry_stub( &ms->mss_dn, &ms->mss_ndn,
101			&monitor_sent[i].rdn, mi->mi_oc_monitorCounterObject,
102			NULL, NULL );
103
104		if ( e == NULL ) {
105			Debug( LDAP_DEBUG_ANY,
106				"monitor_subsys_sent_init: "
107				"unable to create entry \"%s,%s\"\n",
108				monitor_sent[ i ].rdn.bv_val,
109				ms->mss_ndn.bv_val );
110			return( -1 );
111		}
112
113		/* steal normalized RDN */
114		dnRdn( &e->e_nname, &nrdn );
115		ber_dupbv( &monitor_sent[ i ].nrdn, &nrdn );
116
117		BER_BVSTR( &bv, "0" );
118		attr_merge_one( e, mi->mi_ad_monitorCounter, &bv, NULL );
119
120		mp = monitor_entrypriv_create();
121		if ( mp == NULL ) {
122			return -1;
123		}
124		e->e_private = ( void * )mp;
125		mp->mp_info = ms;
126		mp->mp_flags = ms->mss_flags \
127			| MONITOR_F_SUB | MONITOR_F_PERSISTENT;
128
129		if ( monitor_cache_add( mi, e ) ) {
130			Debug( LDAP_DEBUG_ANY,
131				"monitor_subsys_sent_init: "
132				"unable to add entry \"%s,%s\"\n",
133				monitor_sent[ i ].rdn.bv_val,
134				ms->mss_ndn.bv_val );
135			return( -1 );
136		}
137
138		*ep = e;
139		ep = &mp->mp_next;
140	}
141
142	monitor_cache_release( mi, e_sent );
143
144	return( 0 );
145}
146
147static int
148monitor_subsys_sent_destroy(
149	BackendDB		*be,
150	monitor_subsys_t	*ms )
151{
152	int		i;
153
154	for ( i = 0; i < MONITOR_SENT_LAST; i++ ) {
155		if ( !BER_BVISNULL( &monitor_sent[ i ].nrdn ) ) {
156			ch_free( monitor_sent[ i ].nrdn.bv_val );
157		}
158	}
159
160	return 0;
161}
162
163static int
164monitor_subsys_sent_update(
165	Operation		*op,
166	SlapReply		*rs,
167	Entry                   *e )
168{
169	monitor_info_t	*mi = ( monitor_info_t *)op->o_bd->be_private;
170
171	struct berval		nrdn;
172	ldap_pvt_mp_t		n;
173	Attribute		*a;
174	slap_counters_t *sc;
175	int			i;
176
177	assert( mi != NULL );
178	assert( e != NULL );
179
180	dnRdn( &e->e_nname, &nrdn );
181
182	for ( i = 0; i < MONITOR_SENT_LAST; i++ ) {
183		if ( dn_match( &nrdn, &monitor_sent[ i ].nrdn ) ) {
184			break;
185		}
186	}
187
188	if ( i == MONITOR_SENT_LAST ) {
189		return SLAP_CB_CONTINUE;
190	}
191
192	ldap_pvt_thread_mutex_lock(&slap_counters.sc_mutex);
193	switch ( i ) {
194	case MONITOR_SENT_ENTRIES:
195		ldap_pvt_mp_init_set( n, slap_counters.sc_entries );
196		for ( sc = slap_counters.sc_next; sc; sc = sc->sc_next ) {
197			ldap_pvt_thread_mutex_lock( &sc->sc_mutex );
198			ldap_pvt_mp_add( n, sc->sc_entries );
199			ldap_pvt_thread_mutex_unlock( &sc->sc_mutex );
200		}
201		break;
202
203	case MONITOR_SENT_REFERRALS:
204		ldap_pvt_mp_init_set( n, slap_counters.sc_refs );
205		for ( sc = slap_counters.sc_next; sc; sc = sc->sc_next ) {
206			ldap_pvt_thread_mutex_lock( &sc->sc_mutex );
207			ldap_pvt_mp_add( n, sc->sc_refs );
208			ldap_pvt_thread_mutex_unlock( &sc->sc_mutex );
209		}
210		break;
211
212	case MONITOR_SENT_PDU:
213		ldap_pvt_mp_init_set( n, slap_counters.sc_pdu );
214		for ( sc = slap_counters.sc_next; sc; sc = sc->sc_next ) {
215			ldap_pvt_thread_mutex_lock( &sc->sc_mutex );
216			ldap_pvt_mp_add( n, sc->sc_pdu );
217			ldap_pvt_thread_mutex_unlock( &sc->sc_mutex );
218		}
219		break;
220
221	case MONITOR_SENT_BYTES:
222		ldap_pvt_mp_init_set( n, slap_counters.sc_bytes );
223		for ( sc = slap_counters.sc_next; sc; sc = sc->sc_next ) {
224			ldap_pvt_thread_mutex_lock( &sc->sc_mutex );
225			ldap_pvt_mp_add( n, sc->sc_bytes );
226			ldap_pvt_thread_mutex_unlock( &sc->sc_mutex );
227		}
228		break;
229
230	default:
231		assert(0);
232	}
233	ldap_pvt_thread_mutex_unlock(&slap_counters.sc_mutex);
234
235	a = attr_find( e->e_attrs, mi->mi_ad_monitorCounter );
236	assert( a != NULL );
237
238	/* NOTE: no minus sign is allowed in the counters... */
239	UI2BV( &a->a_vals[ 0 ], n );
240	ldap_pvt_mp_clear( n );
241
242	/* FIXME: touch modifyTimestamp? */
243
244	return SLAP_CB_CONTINUE;
245}
246
247