1/*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 *
21 * Portions Copyright (c) 2008-2009 Stacey Son <sson@FreeBSD.org>
22 *
23 * $FreeBSD$
24 *
25 */
26
27/*
28 * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
29 * Use is subject to license terms.
30 */
31
32#include "opt_kdtrace.h"
33
34#include <sys/cdefs.h>
35#include <sys/param.h>
36#include <sys/systm.h>
37#include <sys/conf.h>
38#include <sys/kernel.h>
39#include <sys/limits.h>
40#include <sys/lock.h>
41#include <sys/linker.h>
42#include <sys/module.h>
43#include <sys/mutex.h>
44
45#include <sys/dtrace.h>
46#include <sys/lockstat.h>
47
48#if defined(__i386__) || defined(__amd64__) || \
49	defined(__mips__) || defined(__powerpc__)
50#define LOCKSTAT_AFRAMES 1
51#else
52#error "architecture not supported"
53#endif
54
55static d_open_t lockstat_open;
56static void     lockstat_provide(void *, dtrace_probedesc_t *);
57static void     lockstat_destroy(void *, dtrace_id_t, void *);
58static void     lockstat_enable(void *, dtrace_id_t, void *);
59static void     lockstat_disable(void *, dtrace_id_t, void *);
60static void     lockstat_load(void *);
61static int     	lockstat_unload(void);
62
63
64typedef struct lockstat_probe {
65	char		*lsp_func;
66	char		*lsp_name;
67	int		lsp_probe;
68	dtrace_id_t	lsp_id;
69#ifdef __FreeBSD__
70	int		lsp_frame;
71#endif
72} lockstat_probe_t;
73
74#ifdef __FreeBSD__
75lockstat_probe_t lockstat_probes[] =
76{
77  /* Spin Locks */
78  { LS_MTX_SPIN_LOCK,	LSS_ACQUIRE,	LS_MTX_SPIN_LOCK_ACQUIRE,
79	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
80  { LS_MTX_SPIN_LOCK, 	LSS_SPIN,	LS_MTX_SPIN_LOCK_SPIN,
81	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
82  { LS_MTX_SPIN_UNLOCK,	LSS_RELEASE,	LS_MTX_SPIN_UNLOCK_RELEASE,
83	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
84  /* Adaptive Locks */
85  { LS_MTX_LOCK,	LSA_ACQUIRE,	LS_MTX_LOCK_ACQUIRE,
86	  DTRACE_IDNONE, (LOCKSTAT_AFRAMES + 1) },
87  { LS_MTX_LOCK,	LSA_BLOCK,	LS_MTX_LOCK_BLOCK,
88	  DTRACE_IDNONE, (LOCKSTAT_AFRAMES + 1) },
89  { LS_MTX_LOCK,	LSA_SPIN,	LS_MTX_LOCK_SPIN,
90	  DTRACE_IDNONE, (LOCKSTAT_AFRAMES + 1) },
91  { LS_MTX_UNLOCK,	LSA_RELEASE,	LS_MTX_UNLOCK_RELEASE,
92	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
93  { LS_MTX_TRYLOCK,	LSA_ACQUIRE,	LS_MTX_TRYLOCK_ACQUIRE,
94	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
95  /* Reader/Writer Locks */
96  { LS_RW_RLOCK,	LSR_ACQUIRE,	LS_RW_RLOCK_ACQUIRE,
97	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
98  { LS_RW_RLOCK,	LSR_BLOCK,	LS_RW_RLOCK_BLOCK,
99	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
100  { LS_RW_RLOCK,	LSR_SPIN,	LS_RW_RLOCK_SPIN,
101	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
102  { LS_RW_RUNLOCK,	LSR_RELEASE,	LS_RW_RUNLOCK_RELEASE,
103	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
104  { LS_RW_WLOCK,	LSR_ACQUIRE,	LS_RW_WLOCK_ACQUIRE,
105	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
106  { LS_RW_WLOCK,	LSR_BLOCK,	LS_RW_WLOCK_BLOCK,
107	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
108  { LS_RW_WLOCK,	LSR_SPIN,	LS_RW_WLOCK_SPIN,
109	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
110  { LS_RW_WUNLOCK,	LSR_RELEASE,	LS_RW_WUNLOCK_RELEASE,
111	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
112  { LS_RW_TRYUPGRADE,	LSR_UPGRADE,   	LS_RW_TRYUPGRADE_UPGRADE,
113	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
114  { LS_RW_DOWNGRADE,	LSR_DOWNGRADE, 	LS_RW_DOWNGRADE_DOWNGRADE,
115	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
116  /* Shared/Exclusive Locks */
117  { LS_SX_SLOCK,	LSX_ACQUIRE,	LS_SX_SLOCK_ACQUIRE,
118	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
119  { LS_SX_SLOCK,	LSX_BLOCK,	LS_SX_SLOCK_BLOCK,
120	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
121  { LS_SX_SLOCK,	LSX_SPIN,	LS_SX_SLOCK_SPIN,
122	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
123  { LS_SX_SUNLOCK,	LSX_RELEASE,	LS_SX_SUNLOCK_RELEASE,
124	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
125  { LS_SX_XLOCK,	LSX_ACQUIRE,	LS_SX_XLOCK_ACQUIRE,
126	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
127  { LS_SX_XLOCK,	LSX_BLOCK,	LS_SX_XLOCK_BLOCK,
128	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
129  { LS_SX_XLOCK,	LSX_SPIN,	LS_SX_XLOCK_SPIN,
130	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
131  { LS_SX_XUNLOCK,	LSX_RELEASE,	LS_SX_XUNLOCK_RELEASE,
132	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
133  { LS_SX_TRYUPGRADE,	LSX_UPGRADE,	LS_SX_TRYUPGRADE_UPGRADE,
134	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
135  { LS_SX_DOWNGRADE,	LSX_DOWNGRADE,	LS_SX_DOWNGRADE_DOWNGRADE,
136	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
137  /* Thread Locks */
138  { LS_THREAD_LOCK,	LST_SPIN,	LS_THREAD_LOCK_SPIN,
139	  DTRACE_IDNONE, LOCKSTAT_AFRAMES },
140  { NULL }
141};
142#else
143#error "OS not supported"
144#endif
145
146
147static struct cdevsw lockstat_cdevsw = {
148	.d_version	= D_VERSION,
149	.d_open		= lockstat_open,
150	.d_name		= "lockstat",
151};
152
153static struct cdev		*lockstat_cdev;
154static dtrace_provider_id_t 	lockstat_id;
155
156/*ARGSUSED*/
157static void
158lockstat_enable(void *arg, dtrace_id_t id, void *parg)
159{
160	lockstat_probe_t *probe = parg;
161
162	ASSERT(!lockstat_probemap[probe->lsp_probe]);
163
164	lockstat_probemap[probe->lsp_probe] = id;
165#ifdef DOODAD
166	membar_producer();
167#endif
168
169	lockstat_probe_func = dtrace_probe;
170#ifdef DOODAD
171	membar_producer();
172
173	lockstat_hot_patch();
174	membar_producer();
175#endif
176}
177
178/*ARGSUSED*/
179static void
180lockstat_disable(void *arg, dtrace_id_t id, void *parg)
181{
182	lockstat_probe_t *probe = parg;
183	int i;
184
185	ASSERT(lockstat_probemap[probe->lsp_probe]);
186
187	lockstat_probemap[probe->lsp_probe] = 0;
188#ifdef DOODAD
189	lockstat_hot_patch();
190	membar_producer();
191#endif
192
193	/*
194	 * See if we have any probes left enabled.
195	 */
196	for (i = 0; i < LS_NPROBES; i++) {
197		if (lockstat_probemap[i]) {
198			/*
199			 * This probe is still enabled.  We don't need to deal
200			 * with waiting for all threads to be out of the
201			 * lockstat critical sections; just return.
202			 */
203			return;
204		}
205	}
206
207}
208
209/*ARGSUSED*/
210static int
211lockstat_open(struct cdev *dev __unused, int oflags __unused,
212	      int devtype __unused, struct thread *td __unused)
213{
214	return (0);
215}
216
217/*ARGSUSED*/
218static void
219lockstat_provide(void *arg, dtrace_probedesc_t *desc)
220{
221	int i = 0;
222
223	for (i = 0; lockstat_probes[i].lsp_func != NULL; i++) {
224		lockstat_probe_t *probe = &lockstat_probes[i];
225
226		if (dtrace_probe_lookup(lockstat_id, "kernel",
227		    probe->lsp_func, probe->lsp_name) != 0)
228			continue;
229
230		ASSERT(!probe->lsp_id);
231#ifdef __FreeBSD__
232		probe->lsp_id = dtrace_probe_create(lockstat_id,
233		    "kernel", probe->lsp_func, probe->lsp_name,
234		    probe->lsp_frame, probe);
235#else
236		probe->lsp_id = dtrace_probe_create(lockstat_id,
237		    "kernel", probe->lsp_func, probe->lsp_name,
238		    LOCKSTAT_AFRAMES, probe);
239#endif
240	}
241}
242
243/*ARGSUSED*/
244static void
245lockstat_destroy(void *arg, dtrace_id_t id, void *parg)
246{
247	lockstat_probe_t *probe = parg;
248
249	ASSERT(!lockstat_probemap[probe->lsp_probe]);
250	probe->lsp_id = 0;
251}
252
253static dtrace_pattr_t lockstat_attr = {
254{ DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_COMMON },
255{ DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
256{ DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
257{ DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_COMMON },
258{ DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_COMMON },
259};
260
261static dtrace_pops_t lockstat_pops = {
262	lockstat_provide,
263	NULL,
264	lockstat_enable,
265	lockstat_disable,
266	NULL,
267	NULL,
268	NULL,
269	NULL,
270	NULL,
271	lockstat_destroy
272};
273
274static void
275lockstat_load(void *dummy)
276{
277	/* Create the /dev/dtrace/lockstat entry. */
278	lockstat_cdev = make_dev(&lockstat_cdevsw, 0, UID_ROOT, GID_WHEEL, 0600,
279	    "dtrace/lockstat");
280
281	if (dtrace_register("lockstat", &lockstat_attr, DTRACE_PRIV_USER,
282	    NULL, &lockstat_pops, NULL, &lockstat_id) != 0)
283	        return;
284}
285
286static int
287lockstat_unload()
288{
289	int error = 0;
290
291	if ((error = dtrace_unregister(lockstat_id)) != 0)
292	    return (error);
293
294	destroy_dev(lockstat_cdev);
295
296	return (error);
297}
298
299/* ARGSUSED */
300static int
301lockstat_modevent(module_t mod __unused, int type, void *data __unused)
302{
303	int error = 0;
304
305	switch (type) {
306	case MOD_LOAD:
307		break;
308
309	case MOD_UNLOAD:
310		break;
311
312	case MOD_SHUTDOWN:
313		break;
314
315	default:
316		error = EOPNOTSUPP;
317		break;
318	}
319	return (error);
320}
321
322SYSINIT(lockstat_load, SI_SUB_DTRACE_PROVIDER, SI_ORDER_ANY, lockstat_load, NULL);
323SYSUNINIT(lockstat_unload, SI_SUB_DTRACE_PROVIDER, SI_ORDER_ANY, lockstat_unload, NULL);
324
325DEV_MODULE(lockstat, lockstat_modevent, NULL);
326MODULE_VERSION(lockstat, 1);
327MODULE_DEPEND(lockstat, dtrace, 1, 1, 1);
328MODULE_DEPEND(lockstat, opensolaris, 1, 1, 1);
329