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
22/*
23 * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
24 * Use is subject to license terms.
25 */
26
27/*
28 * Copyright (c) 2012 by Delphix. All rights reserved.
29 */
30
31#ifndef	_DT_PROC_H
32#define	_DT_PROC_H
33
34#include <libproc.h>
35#include <dtrace.h>
36#include <pthread.h>
37#include <dt_list.h>
38
39#ifdef	__cplusplus
40extern "C" {
41#endif
42
43typedef struct dt_proc {
44	dt_list_t dpr_list;		/* prev/next pointers for lru chain */
45	struct dt_proc *dpr_hash;	/* next pointer for pid hash chain */
46	dtrace_hdl_t *dpr_hdl;		/* back pointer to libdtrace handle */
47	struct ps_prochandle *dpr_proc;	/* proc handle for libproc calls */
48	char dpr_errmsg[BUFSIZ];	/* error message */
49	rd_agent_t *dpr_rtld;		/* rtld handle for librtld_db calls */
50	pthread_mutex_t dpr_lock;	/* lock for manipulating dpr_hdl */
51	pthread_cond_t dpr_cv;		/* cond for dpr_stop/quit/done */
52	pid_t dpr_pid;			/* pid of process */
53	uint_t dpr_refs;		/* reference count */
54	uint8_t dpr_cacheable;		/* cache handle using lru list */
55	uint8_t dpr_stop;		/* stop mask: see flag bits below */
56	uint8_t dpr_quit;		/* quit flag: ctl thread should quit */
57	uint8_t dpr_done;		/* done flag: ctl thread has exited */
58	uint8_t dpr_usdt;		/* usdt flag: usdt initialized */
59	uint8_t dpr_stale;		/* proc flag: been deprecated */
60	uint8_t dpr_rdonly;		/* proc flag: opened read-only */
61	pthread_t dpr_tid;		/* control thread (or zero if none) */
62	dt_list_t dpr_bps;		/* list of dt_bkpt_t structures */
63} dt_proc_t;
64
65typedef struct dt_proc_notify {
66	dt_proc_t *dprn_dpr;		/* process associated with the event */
67	char dprn_errmsg[BUFSIZ];	/* error message */
68	struct dt_proc_notify *dprn_next; /* next pointer */
69} dt_proc_notify_t;
70
71#define	DT_PROC_STOP_IDLE	0x01	/* idle on owner's stop request */
72#define	DT_PROC_STOP_CREATE	0x02	/* wait on dpr_cv at process exec */
73#define	DT_PROC_STOP_GRAB	0x04	/* wait on dpr_cv at process grab */
74#define	DT_PROC_STOP_PREINIT	0x08	/* wait on dpr_cv at rtld preinit */
75#define	DT_PROC_STOP_POSTINIT	0x10	/* wait on dpr_cv at rtld postinit */
76#define	DT_PROC_STOP_MAIN	0x20	/* wait on dpr_cv at a.out`main() */
77
78typedef void dt_bkpt_f(dtrace_hdl_t *, dt_proc_t *, void *);
79
80typedef struct dt_bkpt {
81	dt_list_t dbp_list;		/* prev/next pointers for bkpt list */
82	dt_bkpt_f *dbp_func;		/* callback function to execute */
83	void *dbp_data;			/* callback function private data */
84	uintptr_t dbp_addr;		/* virtual address of breakpoint */
85	ulong_t dbp_instr;		/* saved instruction from breakpoint */
86	ulong_t dbp_hits;		/* count of breakpoint hits for debug */
87	int dbp_active;			/* flag indicating breakpoint is on */
88} dt_bkpt_t;
89
90typedef struct dt_proc_hash {
91	pthread_mutex_t dph_lock;	/* lock protecting dph_notify list */
92	pthread_cond_t dph_cv;		/* cond for waiting for dph_notify */
93	dt_proc_notify_t *dph_notify;	/* list of pending proc notifications */
94	dt_list_t dph_lrulist;		/* list of dt_proc_t's in lru order */
95	uint_t dph_lrulim;		/* limit on number of procs to hold */
96	uint_t dph_lrucnt;		/* count of cached process handles */
97	uint_t dph_hashlen;		/* size of hash chains array */
98	dt_proc_t *dph_hash[1];		/* hash chains array */
99} dt_proc_hash_t;
100
101extern struct ps_prochandle *dt_proc_create(dtrace_hdl_t *,
102    const char *, char *const *, proc_child_func *, void *);
103
104extern struct ps_prochandle *dt_proc_grab(dtrace_hdl_t *, pid_t, int, int);
105extern void dt_proc_release(dtrace_hdl_t *, struct ps_prochandle *);
106extern void dt_proc_continue(dtrace_hdl_t *, struct ps_prochandle *);
107extern void dt_proc_lock(dtrace_hdl_t *, struct ps_prochandle *);
108extern void dt_proc_unlock(dtrace_hdl_t *, struct ps_prochandle *);
109extern dt_proc_t *dt_proc_lookup(dtrace_hdl_t *, struct ps_prochandle *, int);
110
111extern void dt_proc_init(dtrace_hdl_t *);
112extern void dt_proc_fini(dtrace_hdl_t *);
113
114#ifdef	__cplusplus
115}
116#endif
117
118#endif	/* _DT_PROC_H */
119