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
40 extern "C" {
41 #endif
42 
43 typedef 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 
65 typedef 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 
78 typedef void dt_bkpt_f(dtrace_hdl_t *, dt_proc_t *, void *);
79 
80 typedef 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 
90 typedef 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 
101 extern struct ps_prochandle *dt_proc_create(dtrace_hdl_t *,
102     const char *, char *const *);
103 
104 extern struct ps_prochandle *dt_proc_grab(dtrace_hdl_t *, pid_t, int, int);
105 extern void dt_proc_release(dtrace_hdl_t *, struct ps_prochandle *);
106 extern void dt_proc_continue(dtrace_hdl_t *, struct ps_prochandle *);
107 extern void dt_proc_lock(dtrace_hdl_t *, struct ps_prochandle *);
108 extern void dt_proc_unlock(dtrace_hdl_t *, struct ps_prochandle *);
109 extern dt_proc_t *dt_proc_lookup(dtrace_hdl_t *, struct ps_prochandle *, int);
110 
111 extern void dt_proc_init(dtrace_hdl_t *);
112 extern void dt_proc_fini(dtrace_hdl_t *);
113 
114 #ifdef	__cplusplus
115 }
116 #endif
117 
118 #endif	/* _DT_PROC_H */
119