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  * Copyright 2012 Joshua M. Clulow <josh@sysmgr.org>
26  * Copyright 2015 Nexenta Systems, Inc.  All rights reserved.
27  */
28 
29 #include <libdisasm.h>
30 
31 #include "dis_tables.h"
32 #include "libdisasm_impl.h"
33 
34 typedef struct dis_handle_i386 {
35 	int		dhx_mode;
36 	dis86_t		dhx_dis;
37 	uint64_t	dhx_end;
38 } dis_handle_i386_t;
39 
40 /*
41  * Returns true if we are near the end of a function.  This is a cheap hack at
42  * detecting NULL padding between functions.  If we're within a few bytes of the
43  * next function, or past the start, then return true.
44  */
45 static int
check_func(void * data)46 check_func(void *data)
47 {
48 	dis_handle_t *dhp = data;
49 	uint64_t start;
50 	size_t len;
51 
52 	if (dhp->dh_lookup(dhp->dh_data, dhp->dh_addr, NULL, 0, &start, &len)
53 	    != 0)
54 		return (0);
55 
56 	if (start < dhp->dh_addr)
57 		return (dhp->dh_addr > start + len - 0x10);
58 
59 	return (1);
60 }
61 
62 static int
get_byte(void * data)63 get_byte(void *data)
64 {
65 	uchar_t byte;
66 	dis_handle_t *dhp = data;
67 
68 	if (dhp->dh_read(dhp->dh_data, dhp->dh_addr, &byte, sizeof (byte)) !=
69 	    sizeof (byte))
70 		return (-1);
71 
72 	dhp->dh_addr++;
73 
74 	return ((int)byte);
75 }
76 
77 static int
do_lookup(void * data,uint64_t addr,char * buf,size_t buflen)78 do_lookup(void *data, uint64_t addr, char *buf, size_t buflen)
79 {
80 	dis_handle_t *dhp = data;
81 
82 	return (dhp->dh_lookup(dhp->dh_data, addr, buf, buflen, NULL, NULL));
83 }
84 
85 static void
dis_i386_handle_detach(dis_handle_t * dhp)86 dis_i386_handle_detach(dis_handle_t *dhp)
87 {
88 	dis_free(dhp->dh_arch_private, sizeof (dis_handle_i386_t));
89 	dhp->dh_arch_private = NULL;
90 }
91 
92 static int
dis_i386_handle_attach(dis_handle_t * dhp)93 dis_i386_handle_attach(dis_handle_t *dhp)
94 {
95 	dis_handle_i386_t *dhx;
96 
97 	/*
98 	 * Validate architecture flags
99 	 */
100 	if (dhp->dh_flags & ~(DIS_X86_SIZE16 | DIS_X86_SIZE32 | DIS_X86_SIZE64 |
101 	    DIS_OCTAL | DIS_NOIMMSYM)) {
102 		(void) dis_seterrno(E_DIS_INVALFLAG);
103 		return (-1);
104 	}
105 
106 	/*
107 	 * Create and initialize the internal structure
108 	 */
109 	if ((dhx = dis_zalloc(sizeof (dis_handle_i386_t))) == NULL) {
110 		(void) dis_seterrno(E_DIS_NOMEM);
111 		return (-1);
112 	}
113 	dhp->dh_arch_private = dhx;
114 
115 	/*
116 	 * Initialize x86-specific architecture structure
117 	 */
118 	if (dhp->dh_flags & DIS_X86_SIZE16)
119 		dhx->dhx_mode = SIZE16;
120 	else if (dhp->dh_flags & DIS_X86_SIZE64)
121 		dhx->dhx_mode = SIZE64;
122 	else
123 		dhx->dhx_mode = SIZE32;
124 
125 	if (dhp->dh_flags & DIS_OCTAL)
126 		dhx->dhx_dis.d86_flags = DIS_F_OCTAL;
127 
128 	dhx->dhx_dis.d86_sprintf_func = dis_snprintf;
129 	dhx->dhx_dis.d86_get_byte = get_byte;
130 	dhx->dhx_dis.d86_sym_lookup = do_lookup;
131 	dhx->dhx_dis.d86_check_func = check_func;
132 
133 	dhx->dhx_dis.d86_data = dhp;
134 
135 	return (0);
136 }
137 
138 static int
dis_i386_disassemble(dis_handle_t * dhp,uint64_t addr,char * buf,size_t buflen)139 dis_i386_disassemble(dis_handle_t *dhp, uint64_t addr, char *buf,
140     size_t buflen)
141 {
142 	dis_handle_i386_t *dhx = dhp->dh_arch_private;
143 	dhp->dh_addr = addr;
144 
145 	/* DIS_NOIMMSYM might not be set until now, so update */
146 	if (dhp->dh_flags & DIS_NOIMMSYM)
147 		dhx->dhx_dis.d86_flags |= DIS_F_NOIMMSYM;
148 	else
149 		dhx->dhx_dis.d86_flags &= ~DIS_F_NOIMMSYM;
150 
151 	if (dtrace_disx86(&dhx->dhx_dis, dhx->dhx_mode) != 0)
152 		return (-1);
153 
154 	if (buf != NULL)
155 		dtrace_disx86_str(&dhx->dhx_dis, dhx->dhx_mode, addr, buf,
156 		    buflen);
157 
158 	return (0);
159 }
160 
161 /* ARGSUSED */
162 static int
dis_i386_max_instrlen(dis_handle_t * dhp)163 dis_i386_max_instrlen(dis_handle_t *dhp)
164 {
165 	return (15);
166 }
167 
168 /* ARGSUSED */
169 static int
dis_i386_min_instrlen(dis_handle_t * dhp)170 dis_i386_min_instrlen(dis_handle_t *dhp)
171 {
172 	return (1);
173 }
174 
175 static int
dis_i386_supports_flags(int flags)176 dis_i386_supports_flags(int flags)
177 {
178 	int archflags = flags & DIS_ARCH_MASK;
179 
180 	if (archflags == DIS_X86_SIZE16 || archflags == DIS_X86_SIZE32 ||
181 	    archflags == DIS_X86_SIZE64)
182 		return (1);
183 
184 	return (0);
185 }
186 
187 dis_arch_t dis_arch_i386 = {
188 	.da_supports_flags	= dis_i386_supports_flags,
189 	.da_handle_attach	= dis_i386_handle_attach,
190 	.da_handle_detach	= dis_i386_handle_detach,
191 	.da_disassemble		= dis_i386_disassemble,
192 	.da_min_instrlen	= dis_i386_min_instrlen,
193 	.da_max_instrlen	= dis_i386_max_instrlen,
194 };
195