blob: f566af453c5a79da6c822b8e09f8a9b976a0bff6 [file] [log] [blame]
Nguyen Anh Quynh6023ef72014-04-29 11:21:04 +08001/* Capstone Disassembly Engine */
2/* By Nguyen Anh Quynh <aquynh@gmail.com>, 2013-2014 */
Axel 0vercl0k Souchet779d4c72014-05-08 23:44:49 +01003#define _CRT_SECURE_NO_WARNINGS
4
5#define CAPSTONE_HAS_ARM
6#define CAPSTONE_HAS_ARM64
7#define CAPSTONE_HAS_MIPS
8#define CAPSTONE_HAS_POWERPC
9#define CAPSTONE_HAS_SPARC
10#define CAPSTONE_HAS_SYSZ
11#define CAPSTONE_HAS_X86
12#define USE_SYS_DYN_MEM
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +080013
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +080014#include <stddef.h>
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +080015#include <stdio.h>
16#include <stdlib.h>
17#include <string.h>
18#include <capstone.h>
19
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +080020#include "utils.h"
Nguyen Anh Quynhc7404072014-01-05 11:35:47 +080021#include "MCRegisterInfo.h"
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +080022
Nguyen Anh Quynh5ee2b452014-03-07 08:40:35 +080023#ifdef USE_SYS_DYN_MEM
Nguyen Anh Quynh472a4a42014-03-06 09:13:04 +080024#define INSN_CACHE_SIZE 32
Nguyen Anh Quynha836b752014-03-06 03:36:03 +080025#else
Nguyen Anh Quynh5ee2b452014-03-07 08:40:35 +080026// reduce stack variable size for kernel/firmware
Nguyen Anh Quynha836b752014-03-06 03:36:03 +080027#define INSN_CACHE_SIZE 8
28#endif
Nguyen Anh Quynhc34959b2014-01-22 09:46:42 +080029
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +080030// default SKIPDATA mnemonic
Nguyen Anh Quynhc75a9092014-04-10 10:26:49 +080031#define SKIPDATA_MNEM ".byte"
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +080032
Nguyen Anh Quynh42c6b1a2013-12-30 00:15:25 +080033cs_err (*arch_init[MAX_ARCH])(cs_struct *) = { NULL };
Nguyen Anh Quynhee143c82014-01-03 21:51:59 +080034cs_err (*arch_option[MAX_ARCH]) (cs_struct *, cs_opt_type, size_t value) = { NULL };
35void (*arch_destroy[MAX_ARCH]) (cs_struct *) = { NULL };
Nguyen Anh Quynhf1851802013-12-21 12:16:47 +080036
Nguyen Anh Quynhc272e9d2014-01-19 12:03:22 +080037extern void ARM_enable(void);
38extern void AArch64_enable(void);
39extern void Mips_enable(void);
40extern void X86_enable(void);
41extern void PPC_enable(void);
Nguyen Anh Quynh05e27132014-03-10 11:58:57 +080042extern void Sparc_enable(void);
Nguyen Anh Quynh48a14ca2014-03-23 08:35:45 +080043extern void SystemZ_enable(void);
danghvu701b8502014-01-09 11:06:44 +070044
Nguyen Anh Quynhc272e9d2014-01-19 12:03:22 +080045static void archs_enable(void)
Nguyen Anh Quynh625b5bc2014-01-09 14:33:56 +080046{
Nguyen Anh Quynhc272e9d2014-01-19 12:03:22 +080047 static bool initialized = false;
48
49 if (initialized)
50 return;
51
danghvub33bd2c2014-01-09 12:22:56 +070052#ifdef CAPSTONE_HAS_ARM
Nguyen Anh Quynhc272e9d2014-01-19 12:03:22 +080053 ARM_enable();
danghvub33bd2c2014-01-09 12:22:56 +070054#endif
55#ifdef CAPSTONE_HAS_ARM64
Nguyen Anh Quynhc272e9d2014-01-19 12:03:22 +080056 AArch64_enable();
danghvub33bd2c2014-01-09 12:22:56 +070057#endif
58#ifdef CAPSTONE_HAS_MIPS
Nguyen Anh Quynhc272e9d2014-01-19 12:03:22 +080059 Mips_enable();
danghvub33bd2c2014-01-09 12:22:56 +070060#endif
danghvub33bd2c2014-01-09 12:22:56 +070061#ifdef CAPSTONE_HAS_POWERPC
Nguyen Anh Quynhc272e9d2014-01-19 12:03:22 +080062 PPC_enable();
danghvub33bd2c2014-01-09 12:22:56 +070063#endif
Nguyen Anh Quynh05e27132014-03-10 11:58:57 +080064#ifdef CAPSTONE_HAS_SPARC
65 Sparc_enable();
66#endif
Nguyen Anh Quynh48a14ca2014-03-23 08:35:45 +080067#ifdef CAPSTONE_HAS_SYSZ
68 SystemZ_enable();
69#endif
Nguyen Anh Quynh05e27132014-03-10 11:58:57 +080070#ifdef CAPSTONE_HAS_X86
71 X86_enable();
72#endif
Nguyen Anh Quynhc272e9d2014-01-19 12:03:22 +080073
74 initialized = true;
danghvub33bd2c2014-01-09 12:22:56 +070075}
76
Nguyen Anh Quynh39a42ed2013-12-22 10:40:58 +080077unsigned int all_arch = 0;
Nguyen Anh Quynhf1851802013-12-21 12:16:47 +080078
Nguyen Anh Quynh59492c22014-01-05 23:41:31 +080079#ifdef USE_SYS_DYN_MEM
Nguyen Anh Quynha8eb7a52014-01-11 12:55:31 +080080cs_malloc_t cs_mem_malloc = malloc;
81cs_calloc_t cs_mem_calloc = calloc;
82cs_realloc_t cs_mem_realloc = realloc;
83cs_free_t cs_mem_free = free;
Nguyen Anh Quynhedeeb042014-01-15 20:44:03 +080084cs_vsnprintf_t cs_vsnprintf = vsnprintf;
Nguyen Anh Quynh59492c22014-01-05 23:41:31 +080085#else
Nguyen Anh Quynha8eb7a52014-01-11 12:55:31 +080086cs_malloc_t cs_mem_malloc = NULL;
87cs_calloc_t cs_mem_calloc = NULL;
88cs_realloc_t cs_mem_realloc = NULL;
89cs_free_t cs_mem_free = NULL;
Nguyen Anh Quynhedeeb042014-01-15 20:44:03 +080090cs_vsnprintf_t cs_vsnprintf = NULL;
Nguyen Anh Quynh59492c22014-01-05 23:41:31 +080091#endif
Nguyen Anh Quynhb8806782013-12-22 15:20:07 +080092
Nguyen Anh Quynhb90cb992013-12-28 13:59:09 +080093unsigned int cs_version(int *major, int *minor)
Nguyen Anh Quynh36df4bb2013-12-10 13:31:20 +080094{
Nguyen Anh Quynhc46c35d2014-03-05 00:21:53 +080095 archs_enable();
96
Nguyen Anh Quynh08777472013-12-22 14:16:28 +080097 if (major != NULL && minor != NULL) {
98 *major = CS_API_MAJOR;
99 *minor = CS_API_MINOR;
100 }
Nguyen Anh Quynh39a42ed2013-12-22 10:40:58 +0800101
102 return (CS_API_MAJOR << 8) + CS_API_MINOR;
103}
104
Nguyen Anh Quynhb2870e42014-02-22 23:41:16 +0800105bool cs_support(int query)
Nguyen Anh Quynh39a42ed2013-12-22 10:40:58 +0800106{
Nguyen Anh Quynhc46c35d2014-03-05 00:21:53 +0800107 archs_enable();
108
Nguyen Anh Quynhb2870e42014-02-22 23:41:16 +0800109 if (query == CS_ARCH_ALL)
Nguyen Anh Quynh39a42ed2013-12-22 10:40:58 +0800110 return all_arch == ((1 << CS_ARCH_ARM) | (1 << CS_ARCH_ARM64) |
Nguyen Anh Quynh42c6b1a2013-12-30 00:15:25 +0800111 (1 << CS_ARCH_MIPS) | (1 << CS_ARCH_X86) |
Nguyen Anh Quynh48a14ca2014-03-23 08:35:45 +0800112 (1 << CS_ARCH_PPC) | (1 << CS_ARCH_SPARC) |
113 (1 << CS_ARCH_SYSZ));
Nguyen Anh Quynh39a42ed2013-12-22 10:40:58 +0800114
Nguyen Anh Quynh48d58322014-02-27 13:54:28 +0800115 if ((unsigned int)query < CS_ARCH_MAX)
Nguyen Anh Quynhb2870e42014-02-22 23:41:16 +0800116 return all_arch & (1 << query);
117
118 if (query == CS_SUPPORT_DIET) {
119#ifdef CAPSTONE_DIET
120 return true;
121#else
122 return false;
123#endif
124 }
125
Nguyen Anh Quynh59b54892014-03-27 10:54:44 +0800126 if (query == CS_SUPPORT_X86_REDUCE) {
127#if defined(CAPSTONE_HAS_X86) && defined(CAPSTONE_X86_REDUCE)
Nguyen Anh Quynh95181482014-03-25 23:20:41 +0800128 return true;
129#else
130 return false;
131#endif
132 }
133
Nguyen Anh Quynhb2870e42014-02-22 23:41:16 +0800134 // unsupported query
135 return false;
Nguyen Anh Quynh36df4bb2013-12-10 13:31:20 +0800136}
137
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800138cs_err cs_errno(csh handle)
139{
Nguyen Anh Quynh42706a32014-05-09 07:33:35 +0800140 struct cs_struct *ud = NULL;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800141 if (!handle)
142 return CS_ERR_CSH;
143
Axel 0vercl0k Souchet779d4c72014-05-08 23:44:49 +0100144 ud = (struct cs_struct *)(uintptr_t)handle;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800145
Nguyen Anh Quynh3eb9ac92013-11-27 15:24:47 +0800146 return ud->errnum;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800147}
148
Nguyen Anh Quynh34f96382014-01-03 22:49:07 +0800149const char *cs_strerror(cs_err code)
150{
151 switch(code) {
152 default:
153 return "Unknown error code";
154 case CS_ERR_OK:
155 return "OK (CS_ERR_OK)";
156 case CS_ERR_MEM:
157 return "Out of memory (CS_ERR_MEM)";
158 case CS_ERR_ARCH:
159 return "Invalid architecture (CS_ERR_ARCH)";
160 case CS_ERR_HANDLE:
161 return "Invalid handle (CS_ERR_HANDLE)";
162 case CS_ERR_CSH:
163 return "Invalid csh (CS_ERR_CSH)";
164 case CS_ERR_MODE:
165 return "Invalid mode (CS_ERR_MODE)";
166 case CS_ERR_OPTION:
167 return "Invalid option (CS_ERR_OPTION)";
168 case CS_ERR_DETAIL:
169 return "Details are unavailable (CS_ERR_DETAIL)";
Nguyen Anh Quynhc52352d2014-01-06 09:06:30 +0800170 case CS_ERR_MEMSETUP:
171 return "Dynamic memory management uninitialized (CS_ERR_MEMSETUP)";
Nguyen Anh Quynh4f220282014-02-09 00:19:15 +0800172 case CS_ERR_VERSION:
173 return "Different API version between core & binding (CS_ERR_VERSION)";
Nguyen Anh Quynhf7cdbdf2014-02-24 16:47:36 +0800174 case CS_ERR_DIET:
175 return "Information irrelevant in diet engine (CS_ERR_DIET)";
Nguyen Anh Quynh11ec8812014-04-10 17:20:01 +0800176 case CS_ERR_SKIPDATA:
177 return "Information irrelevant for 'data' instruction in SKIPDATA mode (CS_ERR_SKIPDATA)";
Nguyen Anh Quynh34f96382014-01-03 22:49:07 +0800178 }
179}
180
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800181cs_err cs_open(cs_arch arch, cs_mode mode, csh *handle)
182{
Nguyen Anh Quynh42706a32014-05-09 07:33:35 +0800183 cs_err err;
184 struct cs_struct *ud = NULL;
Nguyen Anh Quynhedeeb042014-01-15 20:44:03 +0800185 if (!cs_mem_malloc || !cs_mem_calloc || !cs_mem_realloc || !cs_mem_free || !cs_vsnprintf)
Nguyen Anh Quynhc52352d2014-01-06 09:06:30 +0800186 // Error: before cs_open(), dynamic memory management must be initialized
187 // with cs_option(CS_OPT_MEM)
188 return CS_ERR_MEMSETUP;
189
Nguyen Anh Quynhc272e9d2014-01-19 12:03:22 +0800190 archs_enable();
191
Nguyen Anh Quynh8f7ab492014-01-06 09:52:57 +0800192 if (arch < CS_ARCH_MAX && arch_init[arch]) {
Nguyen Anh Quynha8eb7a52014-01-11 12:55:31 +0800193 ud = cs_mem_calloc(1, sizeof(*ud));
Nguyen Anh Quynh8f7ab492014-01-06 09:52:57 +0800194 if (!ud) {
195 // memory insufficient
196 return CS_ERR_MEM;
197 }
198
Nguyen Anh Quynh9a197b32013-12-22 13:41:38 +0800199 ud->errnum = CS_ERR_OK;
200 ud->arch = arch;
201 ud->mode = mode;
202 ud->big_endian = mode & CS_MODE_BIG_ENDIAN;
Nguyen Anh Quynh39b812d2014-01-07 23:36:26 +0800203 // by default, do not break instruction into details
204 ud->detail = CS_OPT_OFF;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800205
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +0800206 // default skipdata setup
207 ud->skipdata_setup.mnemonic = SKIPDATA_MNEM;
208
Axel 0vercl0k Souchet779d4c72014-05-08 23:44:49 +0100209 err = arch_init[ud->arch](ud);
Nguyen Anh Quynh06b3c052014-01-21 15:26:02 +0800210 if (err) {
211 cs_mem_free(ud);
212 *handle = 0;
213 return err;
214 }
Nguyen Anh Quynh8f7ab492014-01-06 09:52:57 +0800215
216 *handle = (uintptr_t)ud;
217
218 return CS_ERR_OK;
Nguyen Anh Quynh9a197b32013-12-22 13:41:38 +0800219 } else {
220 *handle = 0;
221 return CS_ERR_ARCH;
222 }
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800223}
224
Nguyen Anh Quynh226d7dc2014-02-27 22:20:39 +0800225cs_err cs_close(csh *handle)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800226{
Nguyen Anh Quynh42706a32014-05-09 07:33:35 +0800227 struct cs_struct *ud = NULL;
Nguyen Anh Quynhef3d04d2014-02-27 23:42:49 +0800228 if (*handle == 0)
Nguyen Anh Quynh226d7dc2014-02-27 22:20:39 +0800229 // invalid handle
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800230 return CS_ERR_CSH;
231
Axel 0vercl0k Souchet779d4c72014-05-08 23:44:49 +0100232 ud = (struct cs_struct *)(*handle);
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800233
Nguyen Anh Quynhfbe10a52014-02-27 23:59:08 +0800234 if (ud->printer_info)
235 cs_mem_free(ud->printer_info);
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800236
Nguyen Anh Quynh9fac5122014-01-07 10:56:04 +0800237 // arch_destroy[ud->arch](ud);
Nguyen Anh Quynhb2654062014-01-03 17:08:58 +0800238
Nguyen Anh Quynha8eb7a52014-01-11 12:55:31 +0800239 cs_mem_free(ud->insn_cache);
Nguyen Anh Quynhee143c82014-01-03 21:51:59 +0800240 memset(ud, 0, sizeof(*ud));
Nguyen Anh Quynha8eb7a52014-01-11 12:55:31 +0800241 cs_mem_free(ud);
Nguyen Anh Quynhee143c82014-01-03 21:51:59 +0800242
Nguyen Anh Quynhc8e07852014-02-28 09:38:11 +0800243 // invalidate this handle by ZERO out its value.
244 // this is to make sure it is unusable after cs_close()
Nguyen Anh Quynh226d7dc2014-02-27 22:20:39 +0800245 *handle = 0;
246
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800247 return CS_ERR_OK;
248}
249
250// fill insn with mnemonic & operands info
Nguyen Anh Quynha82a0892014-01-23 23:42:40 +0800251static void fill_insn(struct cs_struct *handle, cs_insn *insn, char *buffer, MCInst *mci,
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800252 PostPrinter_t postprinter, const uint8_t *code)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800253{
Nguyen Anh Quynh42706a32014-05-09 07:33:35 +0800254 char *sp = NULL;
Nguyen Anh Quynha209e672013-12-14 00:23:41 +0800255 if (handle->detail) {
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800256 // avoiding copy insn->detail
257 memcpy(insn, &mci->flat_insn, sizeof(*insn) - sizeof(insn->detail));
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800258
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800259 // NOTE: copy details in 2 chunks, since union is always put at address divisible by 8
260 // copy from @regs_read until @arm
Axel 0vercl0k Souchet779d4c72014-05-08 23:44:49 +0100261 memcpy(insn->detail, (void *)((uintptr_t)(&(mci->flat_insn)) + offsetof(cs_insn_flat, regs_read)),
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800262 offsetof(cs_detail, arm) - offsetof(cs_detail, regs_read));
263 // then copy from @arm until end
Nguyen Anh Quynha82a0892014-01-23 23:42:40 +0800264 memcpy((void *)((uintptr_t)(insn->detail) + offsetof(cs_detail, arm)),
265 (void *)((uintptr_t)(&(mci->flat_insn)) + offsetof(cs_insn_flat, arm)),
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800266 sizeof(cs_detail) - offsetof(cs_detail, arm));
Nguyen Anh Quynha209e672013-12-14 00:23:41 +0800267 } else {
268 insn->address = mci->address;
Alex Ionescu46018db2014-01-22 09:45:00 -0800269 insn->size = (uint16_t)mci->insn_size;
Nguyen Anh Quynha209e672013-12-14 00:23:41 +0800270 }
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800271
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800272 // fill the instruction bytes
273 memcpy(insn->bytes, code, MIN(sizeof(insn->bytes), insn->size));
274
Nguyen Anh Quynh4d3e8522013-12-14 10:45:09 +0800275 // map internal instruction opcode to public insn ID
276 if (handle->insn_id)
Nguyen Anh Quynh1acfd0b2014-01-06 10:56:59 +0800277 handle->insn_id(handle, insn, MCInst_getOpcode(mci));
Nguyen Anh Quynh4d3e8522013-12-14 10:45:09 +0800278
279 // alias instruction might have ID saved in OpcodePub
280 if (MCInst_getOpcodePub(mci))
281 insn->id = MCInst_getOpcodePub(mci);
282
283 // post printer handles some corner cases (hacky)
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800284 if (postprinter)
285 postprinter((csh)handle, insn, buffer);
Nguyen Anh Quynh4d3e8522013-12-14 10:45:09 +0800286
Nguyen Anh Quynhfc83a432014-02-22 23:26:27 +0800287#ifndef CAPSTONE_DIET
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800288 // fill in mnemonic & operands
Nguyen Anh Quynhdefb9bc2013-12-12 14:00:12 +0800289 // find first space or tab
Axel 0vercl0k Souchet779d4c72014-05-08 23:44:49 +0100290 sp = buffer;
Nguyen Anh Quynha5ffdc32014-05-07 08:25:24 +0800291 for (sp = buffer; *sp; sp++) {
Nguyen Anh Quynhdefb9bc2013-12-12 14:00:12 +0800292 if (*sp == ' '||*sp == '\t')
293 break;
Nguyen Anh Quynha5ffdc32014-05-07 08:25:24 +0800294 if (*sp == '|')
295 *sp = ' ';
296 }
297
Nguyen Anh Quynhdefb9bc2013-12-12 14:00:12 +0800298 if (*sp) {
299 *sp = '\0';
Nguyen Anh Quynh86dc3932013-12-12 14:43:39 +0800300 // find the next non-space char
301 sp++;
302 for (; ((*sp == ' ') || (*sp == '\t')); sp++);
303 strncpy(insn->op_str, sp, sizeof(insn->op_str) - 1);
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800304 insn->op_str[sizeof(insn->op_str) - 1] = '\0';
305 } else
306 insn->op_str[0] = '\0';
307
Nguyen Anh Quynh1dbc9592014-05-07 14:14:07 +0800308 strncpy(insn->mnemonic, buffer, sizeof(insn->mnemonic) - 1);
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800309 insn->mnemonic[sizeof(insn->mnemonic) - 1] = '\0';
Nguyen Anh Quynhfc83a432014-02-22 23:26:27 +0800310#endif
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800311}
312
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +0800313// how many bytes will we skip when encountering data (CS_OPT_SKIPDATA)?
Nguyen Anh Quynha89383e2014-04-10 11:53:46 +0800314// this very much depends on instruction alignment requirement of each arch.
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +0800315static uint8_t skipdata_size(cs_struct *handle)
316{
317 switch(handle->arch) {
318 default:
319 // should never reach
320 return -1;
321 case CS_ARCH_ARM:
Nguyen Anh Quynha89383e2014-04-10 11:53:46 +0800322 // skip 2 bytes on Thumb mode.
323 if (handle->mode & CS_MODE_THUMB)
324 return 2;
325 // otherwise, skip 4 bytes
326 return 4;
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +0800327 case CS_ARCH_ARM64:
328 case CS_ARCH_MIPS:
329 case CS_ARCH_PPC:
330 case CS_ARCH_SPARC:
Nguyen Anh Quynha89383e2014-04-10 11:53:46 +0800331 // skip 4 bytes
332 return 4;
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +0800333 case CS_ARCH_SYSZ:
Nguyen Anh Quynha89383e2014-04-10 11:53:46 +0800334 // SystemZ instruction's length can be 2, 4 or 6 bytes,
335 // so we just skip 2 bytes
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +0800336 return 2;
337 case CS_ARCH_X86:
338 // X86 has no restriction on instruction alignment
339 return 1;
340 }
341}
342
Nguyen Anh Quynhda8adad2013-12-04 09:44:07 +0800343cs_err cs_option(csh ud, cs_opt_type type, size_t value)
Nguyen Anh Quynh01aba002013-12-03 21:00:09 +0800344{
Nguyen Anh Quynh42706a32014-05-09 07:33:35 +0800345 struct cs_struct *handle = NULL;
Nguyen Anh Quynhc46c35d2014-03-05 00:21:53 +0800346 archs_enable();
347
Nguyen Anh Quynh24bf0d92014-01-05 11:19:04 +0800348 // cs_option() can be called with NULL handle just for CS_OPT_MEM
349 // This is supposed to be executed before all other APIs (even cs_open())
350 if (type == CS_OPT_MEM) {
351 cs_opt_mem *mem = (cs_opt_mem *)value;
352
Nguyen Anh Quynha8eb7a52014-01-11 12:55:31 +0800353 cs_mem_malloc = mem->malloc;
354 cs_mem_calloc = mem->calloc;
355 cs_mem_realloc = mem->realloc;
356 cs_mem_free = mem->free;
Nguyen Anh Quynhedeeb042014-01-15 20:44:03 +0800357 cs_vsnprintf = mem->vsnprintf;
Nguyen Anh Quynh24bf0d92014-01-05 11:19:04 +0800358
359 return CS_ERR_OK;
360 }
361
Axel 0vercl0k Souchet779d4c72014-05-08 23:44:49 +0100362 handle = (struct cs_struct *)(uintptr_t)ud;
danghvu2b192962013-12-19 22:40:28 -0600363 if (!handle)
364 return CS_ERR_CSH;
Nguyen Anh Quynh01aba002013-12-03 21:00:09 +0800365
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +0800366 switch(type) {
367 default:
368 break;
369 case CS_OPT_DETAIL:
370 handle->detail = value;
371 return CS_ERR_OK;
372 case CS_OPT_SKIPDATA:
373 handle->skipdata = (value == CS_OPT_ON);
374 if (handle->skipdata) {
375 if (handle->skipdata_size == 0) {
376 // set the default skipdata size
377 handle->skipdata_size = skipdata_size(handle);
378 }
379 }
380 return CS_ERR_OK;
381 case CS_OPT_SKIPDATA_SETUP:
382 if (value)
383 handle->skipdata_setup = *((cs_opt_skipdata *)value);
384 return CS_ERR_OK;
Nguyen Anh Quynh7d02c922013-12-21 09:59:31 +0800385 }
Nguyen Anh Quynha209e672013-12-14 00:23:41 +0800386
Nguyen Anh Quynhd3458392013-12-22 11:10:56 +0800387 return arch_option[handle->arch](handle, type, value);
Nguyen Anh Quynh01aba002013-12-03 21:00:09 +0800388}
389
Nguyen Anh Quynh07ffd642014-04-10 14:36:08 +0800390// generate @op_str for data instruction of SKIPDATA
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +0800391static void skipdata_opstr(char *opstr, const uint8_t *buffer, size_t size)
392{
393 char *p = opstr;
394 int len;
395 size_t i;
396
397 if (!size) {
398 opstr[0] = '\0';
399 return;
400 }
401
402 len = sprintf(p, "0x%02x", buffer[0]);
403 p+= len;
404
405 for(i = 1; i < size; i++) {
406 len = sprintf(p, ", 0x%02x", buffer[i]);
407 p+= len;
408 }
409}
410
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800411// dynamicly allocate memory to contain disasm insn
412// NOTE: caller must free() the allocated memory itself to avoid memory leaking
Nguyen Anh Quynh04c19be2013-12-25 13:26:22 +0800413size_t cs_disasm_ex(csh ud, const uint8_t *buffer, size_t size, uint64_t offset, size_t count, cs_insn **insn)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800414{
Nguyen Anh Quynha82a0892014-01-23 23:42:40 +0800415 struct cs_struct *handle = (struct cs_struct *)(uintptr_t)ud;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800416 MCInst mci;
417 uint16_t insn_size;
Nguyen Anh Quynh7772d852014-01-21 11:49:25 +0800418 size_t c = 0;
419 unsigned int f = 0;
Nguyen Anh Quynhc34959b2014-01-22 09:46:42 +0800420 cs_insn insn_cache[INSN_CACHE_SIZE];
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800421 void *total = NULL;
Nguyen Anh Quynhb42a6572013-11-29 17:40:07 +0800422 size_t total_size = 0;
Nguyen Anh Quynh034a7482014-03-06 22:40:08 +0800423 bool r;
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +0800424 void *tmp;
425 size_t skipdata_bytes;
Nguyen Anh Quynh07ffd642014-04-10 14:36:08 +0800426 uint64_t offset_org;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800427
Nguyen Anh Quynh5848aaa2014-02-27 11:10:41 +0800428 if (!handle) {
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800429 // FIXME: how to handle this case:
Nguyen Anh Quynh3eb9ac92013-11-27 15:24:47 +0800430 // handle->errnum = CS_ERR_HANDLE;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800431 return 0;
432 }
433
Nguyen Anh Quynh3eb9ac92013-11-27 15:24:47 +0800434 handle->errnum = CS_ERR_OK;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800435
Nguyen Anh Quynh4d3e8522013-12-14 10:45:09 +0800436 memset(insn_cache, 0, sizeof(insn_cache));
437
Nguyen Anh Quynh2cff6f62014-04-28 11:19:44 +0800438 // save the original offset for SKIPDATA
Nguyen Anh Quynh07ffd642014-04-10 14:36:08 +0800439 offset_org = offset;
Nguyen Anh Quynh2cff6f62014-04-28 11:19:44 +0800440
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800441 while (size > 0) {
danghvu2b192962013-12-19 22:40:28 -0600442 MCInst_Init(&mci);
Nguyen Anh Quynh42c6b1a2013-12-30 00:15:25 +0800443 mci.csh = handle;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800444
Nguyen Anh Quynha5ffdc32014-05-07 08:25:24 +0800445 r = handle->disasm(ud, buffer, size, &mci, &insn_size, offset, handle->getinsn_info);
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800446 if (r) {
447 SStream ss;
448 SStream_Init(&ss);
449
Nguyen Anh Quynha209e672013-12-14 00:23:41 +0800450 // relative branches need to know the address & size of current insn
451 mci.insn_size = insn_size;
452 mci.address = offset;
453
Nguyen Anh Quynh4d3e8522013-12-14 10:45:09 +0800454 if (handle->detail) {
Nguyen Anh Quynh4d3e8522013-12-14 10:45:09 +0800455 // save all the information for non-detailed mode
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800456 mci.flat_insn.address = offset;
457 mci.flat_insn.size = insn_size;
458 // allocate memory for @detail pointer
Nguyen Anh Quynha8eb7a52014-01-11 12:55:31 +0800459 insn_cache[f].detail = cs_mem_calloc(1, sizeof(cs_detail));
Nguyen Anh Quynha209e672013-12-14 00:23:41 +0800460 }
461
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800462 handle->printer(&mci, &ss, handle->printer_info);
463
Joxean114df0e2013-12-04 07:11:32 +0100464 fill_insn(handle, &insn_cache[f], ss.buffer, &mci, handle->post_printer, buffer);
Nguyen Anh Quynha209e672013-12-14 00:23:41 +0800465
Nguyen Anh Quynha5ffdc32014-05-07 08:25:24 +0800466 f++;
467 if (f == ARR_SIZE(insn_cache)) {
468 // resize total to contain newly disasm insns
469 total_size += (sizeof(cs_insn) * INSN_CACHE_SIZE);
470 tmp = cs_mem_realloc(total, total_size);
471 if (tmp == NULL) { // insufficient memory
472 cs_mem_free(total);
473 handle->errnum = CS_ERR_MEM;
474 return 0;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800475 }
476
Nguyen Anh Quynha5ffdc32014-05-07 08:25:24 +0800477 total = tmp;
478 memcpy((void*)((uintptr_t)total + total_size - sizeof(insn_cache)), insn_cache, sizeof(insn_cache));
479
480 // reset f back to 0
481 f = 0;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800482 }
483
Nguyen Anh Quynha5ffdc32014-05-07 08:25:24 +0800484 c++;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800485 buffer += insn_size;
Nguyen Anh Quynh2cff6f62014-04-28 11:19:44 +0800486
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800487 size -= insn_size;
488 offset += insn_size;
489
Nguyen Anh Quynha5ffdc32014-05-07 08:25:24 +0800490 if (count > 0 && c == count)
491 break;
Nguyen Anh Quynh8f13f3c2013-12-04 22:57:04 +0800492 } else {
493 // encounter a broken instruction
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +0800494 // if there is no request to skip data, or remaining data is too small,
495 // then bail out
496 if (!handle->skipdata || handle->skipdata_size > size)
497 break;
498
499 if (handle->skipdata_setup.callback) {
Nguyen Anh Quynh42288ac2014-04-14 14:53:13 +0800500 skipdata_bytes = handle->skipdata_setup.callback(buffer, offset - offset_org,
Nguyen Anh Quynhd3ffe372014-04-09 23:49:30 +0800501 handle->skipdata_setup.user_data);
502 if (skipdata_bytes > size)
503 // remaining data is not enough
504 break;
505
506 if (!skipdata_bytes)
507 // user requested not to skip data, so bail out
508 break;
509 } else
510 skipdata_bytes = handle->skipdata_size;
511
512 // we have to skip some amount of data, depending on arch & mode
513 insn_cache[f].id = 0; // invalid ID for this "data" instruction
514 insn_cache[f].address = offset;
515 insn_cache[f].size = skipdata_bytes;
516 memcpy(insn_cache[f].bytes, buffer, skipdata_bytes);
517 strncpy(insn_cache[f].mnemonic, handle->skipdata_setup.mnemonic,
518 sizeof(insn_cache[f].mnemonic) - 1);
519 skipdata_opstr(insn_cache[f].op_str, buffer, skipdata_bytes);
520 insn_cache[f].detail = NULL;
521
522 f++;
523 if (f == ARR_SIZE(insn_cache)) {
524 // resize total to contain newly disasm insns
525
526 total_size += (sizeof(cs_insn) * INSN_CACHE_SIZE);
527 tmp = cs_mem_realloc(total, total_size);
528 if (tmp == NULL) { // insufficient memory
529 cs_mem_free(total);
530 handle->errnum = CS_ERR_MEM;
531 return 0;
532 }
533
534 total = tmp;
535 memcpy((void*)((uintptr_t)total + total_size - sizeof(insn_cache)), insn_cache, sizeof(insn_cache));
536
537 // reset f back to 0
538 f = 0;
539 }
540
541 buffer += skipdata_bytes;
542 size -= skipdata_bytes;
543 offset += skipdata_bytes;
544 c++;
Nguyen Anh Quynh8f13f3c2013-12-04 22:57:04 +0800545 }
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800546 }
547
548 if (f) {
549 // resize total to contain newly disasm insns
Nguyen Anh Quynha8eb7a52014-01-11 12:55:31 +0800550 void *tmp = cs_mem_realloc(total, total_size + f * sizeof(insn_cache[0]));
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800551 if (tmp == NULL) { // insufficient memory
Nguyen Anh Quynha8eb7a52014-01-11 12:55:31 +0800552 cs_mem_free(total);
Nguyen Anh Quynh3eb9ac92013-11-27 15:24:47 +0800553 handle->errnum = CS_ERR_MEM;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800554 return 0;
555 }
556
557 total = tmp;
Alex Ionescu46018db2014-01-22 09:45:00 -0800558 memcpy((void*)((uintptr_t)total + total_size), insn_cache, f * sizeof(insn_cache[0]));
Nguyen Anh Quynha82a0892014-01-23 23:42:40 +0800559
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800560 }
561
562 *insn = total;
563
564 return c;
565}
566
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800567void cs_free(cs_insn *insn, size_t count)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800568{
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800569 size_t i;
570
571 // free all detail pointers
572 for (i = 0; i < count; i++)
Nguyen Anh Quynha8eb7a52014-01-11 12:55:31 +0800573 cs_mem_free(insn[i].detail);
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800574
575 // then free pointer to cs_insn array
Nguyen Anh Quynha8eb7a52014-01-11 12:55:31 +0800576 cs_mem_free(insn);
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800577}
578
579// return friendly name of regiser in a string
pancakef0e4eed2013-12-11 22:14:42 +0100580const char *cs_reg_name(csh ud, unsigned int reg)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800581{
Nguyen Anh Quynha82a0892014-01-23 23:42:40 +0800582 struct cs_struct *handle = (struct cs_struct *)(uintptr_t)ud;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800583
584 if (!handle || handle->reg_name == NULL) {
585 return NULL;
586 }
587
Nguyen Anh Quynha253c7a2013-12-09 10:26:18 +0800588 return handle->reg_name(ud, reg);
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800589}
590
pancakef0e4eed2013-12-11 22:14:42 +0100591const char *cs_insn_name(csh ud, unsigned int insn)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800592{
Nguyen Anh Quynha82a0892014-01-23 23:42:40 +0800593 struct cs_struct *handle = (struct cs_struct *)(uintptr_t)ud;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800594
595 if (!handle || handle->insn_name == NULL) {
596 return NULL;
597 }
598
Nguyen Anh Quynha253c7a2013-12-09 10:26:18 +0800599 return handle->insn_name(ud, insn);
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800600}
601
Nguyen Anh Quynh70083562013-12-20 22:02:20 +0800602static bool arr_exist(unsigned char *arr, unsigned char max, unsigned int id)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800603{
604 int i;
605
606 for (i = 0; i < max; i++) {
607 if (arr[i] == id)
608 return true;
609 }
610
611 return false;
612}
613
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800614bool cs_insn_group(csh ud, cs_insn *insn, unsigned int group_id)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800615{
Nguyen Anh Quynh42706a32014-05-09 07:33:35 +0800616 struct cs_struct *handle = NULL;
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800617 if (!ud)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800618 return false;
619
Axel 0vercl0k Souchet779d4c72014-05-08 23:44:49 +0100620 handle = (struct cs_struct *)(uintptr_t)ud;
Giovanni Condelloa715df12014-05-07 17:25:38 +0200621
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800622 if (!handle->detail) {
623 handle->errnum = CS_ERR_DETAIL;
624 return false;
625 }
626
Giovanni Condello95657e02014-05-07 17:31:27 +0200627 if(!insn->id) {
628 handle->errnum = CS_ERR_SKIPDATA;
629 return false;
630 }
Giovanni Condelloa715df12014-05-07 17:25:38 +0200631
Giovanni Condello95657e02014-05-07 17:31:27 +0200632 if(!insn->detail) {
633 handle->errnum = CS_ERR_DETAIL;
634 return false;
635 }
Giovanni Condelloa715df12014-05-07 17:25:38 +0200636
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800637 return arr_exist(insn->detail->groups, insn->detail->groups_count, group_id);
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800638}
639
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800640bool cs_reg_read(csh ud, cs_insn *insn, unsigned int reg_id)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800641{
Nguyen Anh Quynh42706a32014-05-09 07:33:35 +0800642 struct cs_struct *handle = NULL;
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800643 if (!ud)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800644 return false;
645
Axel 0vercl0k Souchet779d4c72014-05-08 23:44:49 +0100646 handle = (struct cs_struct *)(uintptr_t)ud;
Giovanni Condelloa715df12014-05-07 17:25:38 +0200647
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800648 if (!handle->detail) {
649 handle->errnum = CS_ERR_DETAIL;
650 return false;
651 }
652
Giovanni Condello95657e02014-05-07 17:31:27 +0200653 if(!insn->id) {
654 handle->errnum = CS_ERR_SKIPDATA;
655 return false;
656 }
Giovanni Condelloa715df12014-05-07 17:25:38 +0200657
Giovanni Condello95657e02014-05-07 17:31:27 +0200658 if(!insn->detail) {
659 handle->errnum = CS_ERR_DETAIL;
660 return false;
661 }
Giovanni Condelloa715df12014-05-07 17:25:38 +0200662
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800663 return arr_exist(insn->detail->regs_read, insn->detail->regs_read_count, reg_id);
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800664}
665
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800666bool cs_reg_write(csh ud, cs_insn *insn, unsigned int reg_id)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800667{
Nguyen Anh Quynh42706a32014-05-09 07:33:35 +0800668 struct cs_struct *handle = NULL;
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800669 if (!ud)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800670 return false;
671
Axel 0vercl0k Souchet779d4c72014-05-08 23:44:49 +0100672 handle = (struct cs_struct *)(uintptr_t)ud;
Giovanni Condello95657e02014-05-07 17:31:27 +0200673
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800674 if (!handle->detail) {
675 handle->errnum = CS_ERR_DETAIL;
676 return false;
677 }
678
Giovanni Condello95657e02014-05-07 17:31:27 +0200679 if(!insn->id) {
680 handle->errnum = CS_ERR_SKIPDATA;
681 return false;
682 }
Giovanni Condelloa715df12014-05-07 17:25:38 +0200683
Giovanni Condello95657e02014-05-07 17:31:27 +0200684 if(!insn->detail) {
685 handle->errnum = CS_ERR_DETAIL;
686 return false;
687 }
Giovanni Condelloa715df12014-05-07 17:25:38 +0200688
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800689 return arr_exist(insn->detail->regs_write, insn->detail->regs_write_count, reg_id);
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800690}
691
692int cs_op_count(csh ud, cs_insn *insn, unsigned int op_type)
693{
Nguyen Anh Quynh42706a32014-05-09 07:33:35 +0800694 struct cs_struct *handle = NULL;
695 unsigned int count = 0, i = 0;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800696 if (!ud)
697 return -1;
698
Axel 0vercl0k Souchet779d4c72014-05-08 23:44:49 +0100699 handle = (struct cs_struct *)(uintptr_t)ud;
Giovanni Condelloa715df12014-05-07 17:25:38 +0200700
Nguyen Anh Quynh24e12272014-01-15 21:27:23 +0800701 if (!handle->detail) {
702 handle->errnum = CS_ERR_DETAIL;
703 return -1;
704 }
705
Giovanni Condello95657e02014-05-07 17:31:27 +0200706 if(!insn->id) {
707 handle->errnum = CS_ERR_SKIPDATA;
708 return -1;
709 }
Giovanni Condelloa715df12014-05-07 17:25:38 +0200710
Giovanni Condello95657e02014-05-07 17:31:27 +0200711 if(!insn->detail) {
712 handle->errnum = CS_ERR_DETAIL;
713 return -1;
714 }
Giovanni Condelloa715df12014-05-07 17:25:38 +0200715
Nguyen Anh Quynh3eb9ac92013-11-27 15:24:47 +0800716 handle->errnum = CS_ERR_OK;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800717
718 switch (handle->arch) {
719 default:
Nguyen Anh Quynh3eb9ac92013-11-27 15:24:47 +0800720 handle->errnum = CS_ERR_HANDLE;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800721 return -1;
722 case CS_ARCH_ARM:
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800723 for (i = 0; i < insn->detail->arm.op_count; i++)
Alex Ionescu46018db2014-01-22 09:45:00 -0800724 if (insn->detail->arm.operands[i].type == (arm_op_type)op_type)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800725 count++;
726 break;
727 case CS_ARCH_ARM64:
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800728 for (i = 0; i < insn->detail->arm64.op_count; i++)
Alex Ionescu46018db2014-01-22 09:45:00 -0800729 if (insn->detail->arm64.operands[i].type == (arm64_op_type)op_type)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800730 count++;
731 break;
732 case CS_ARCH_X86:
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800733 for (i = 0; i < insn->detail->x86.op_count; i++)
Alex Ionescu46018db2014-01-22 09:45:00 -0800734 if (insn->detail->x86.operands[i].type == (x86_op_type)op_type)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800735 count++;
736 break;
737 case CS_ARCH_MIPS:
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800738 for (i = 0; i < insn->detail->mips.op_count; i++)
Alex Ionescu46018db2014-01-22 09:45:00 -0800739 if (insn->detail->mips.operands[i].type == (mips_op_type)op_type)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800740 count++;
741 break;
Nguyen Anh Quynh42c6b1a2013-12-30 00:15:25 +0800742 case CS_ARCH_PPC:
743 for (i = 0; i < insn->detail->ppc.op_count; i++)
Alex Ionescu46018db2014-01-22 09:45:00 -0800744 if (insn->detail->ppc.operands[i].type == (ppc_op_type)op_type)
Nguyen Anh Quynh42c6b1a2013-12-30 00:15:25 +0800745 count++;
746 break;
Nguyen Anh Quynh05e27132014-03-10 11:58:57 +0800747 case CS_ARCH_SPARC:
748 for (i = 0; i < insn->detail->sparc.op_count; i++)
749 if (insn->detail->sparc.operands[i].type == (sparc_op_type)op_type)
750 count++;
751 break;
Nguyen Anh Quynh48a14ca2014-03-23 08:35:45 +0800752 case CS_ARCH_SYSZ:
753 for (i = 0; i < insn->detail->sysz.op_count; i++)
754 if (insn->detail->sysz.operands[i].type == (sysz_op_type)op_type)
755 count++;
756 break;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800757 }
758
759 return count;
760}
761
762int cs_op_index(csh ud, cs_insn *insn, unsigned int op_type,
763 unsigned int post)
764{
Nguyen Anh Quynh42706a32014-05-09 07:33:35 +0800765 struct cs_struct *handle = NULL;
766 unsigned int count = 0, i = 0;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800767 if (!ud)
768 return -1;
769
Axel 0vercl0k Souchet779d4c72014-05-08 23:44:49 +0100770 handle = (struct cs_struct *)(uintptr_t)ud;
Giovanni Condelloa715df12014-05-07 17:25:38 +0200771
Nguyen Anh Quynh24e12272014-01-15 21:27:23 +0800772 if (!handle->detail) {
773 handle->errnum = CS_ERR_DETAIL;
774 return -1;
775 }
776
Giovanni Condello95657e02014-05-07 17:31:27 +0200777 if(!insn->id) {
778 handle->errnum = CS_ERR_SKIPDATA;
779 return -1;
780 }
Giovanni Condelloa715df12014-05-07 17:25:38 +0200781
Giovanni Condello95657e02014-05-07 17:31:27 +0200782 if(!insn->detail) {
783 handle->errnum = CS_ERR_DETAIL;
784 return -1;
785 }
Giovanni Condelloa715df12014-05-07 17:25:38 +0200786
Nguyen Anh Quynh3eb9ac92013-11-27 15:24:47 +0800787 handle->errnum = CS_ERR_OK;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800788
789 switch (handle->arch) {
790 default:
Nguyen Anh Quynh3eb9ac92013-11-27 15:24:47 +0800791 handle->errnum = CS_ERR_HANDLE;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800792 return -1;
793 case CS_ARCH_ARM:
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800794 for (i = 0; i < insn->detail->arm.op_count; i++) {
Alex Ionescu46018db2014-01-22 09:45:00 -0800795 if (insn->detail->arm.operands[i].type == (arm_op_type)op_type)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800796 count++;
797 if (count == post)
798 return i;
799 }
800 break;
801 case CS_ARCH_ARM64:
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800802 for (i = 0; i < insn->detail->arm64.op_count; i++) {
Alex Ionescu46018db2014-01-22 09:45:00 -0800803 if (insn->detail->arm64.operands[i].type == (arm64_op_type)op_type)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800804 count++;
805 if (count == post)
806 return i;
807 }
808 break;
809 case CS_ARCH_X86:
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800810 for (i = 0; i < insn->detail->x86.op_count; i++) {
Alex Ionescu46018db2014-01-22 09:45:00 -0800811 if (insn->detail->x86.operands[i].type == (x86_op_type)op_type)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800812 count++;
813 if (count == post)
814 return i;
815 }
816 break;
817 case CS_ARCH_MIPS:
Nguyen Anh Quynh4fe224b2013-12-24 16:49:36 +0800818 for (i = 0; i < insn->detail->mips.op_count; i++) {
Alex Ionescu46018db2014-01-22 09:45:00 -0800819 if (insn->detail->mips.operands[i].type == (mips_op_type)op_type)
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800820 count++;
821 if (count == post)
822 return i;
823 }
824 break;
Nguyen Anh Quynh42c6b1a2013-12-30 00:15:25 +0800825 case CS_ARCH_PPC:
826 for (i = 0; i < insn->detail->ppc.op_count; i++) {
Alex Ionescu46018db2014-01-22 09:45:00 -0800827 if (insn->detail->ppc.operands[i].type == (ppc_op_type)op_type)
Nguyen Anh Quynh42c6b1a2013-12-30 00:15:25 +0800828 count++;
829 if (count == post)
830 return i;
831 }
832 break;
Nguyen Anh Quynh05e27132014-03-10 11:58:57 +0800833 case CS_ARCH_SPARC:
834 for (i = 0; i < insn->detail->sparc.op_count; i++) {
835 if (insn->detail->sparc.operands[i].type == (sparc_op_type)op_type)
836 count++;
837 if (count == post)
838 return i;
839 }
840 break;
Nguyen Anh Quynh48a14ca2014-03-23 08:35:45 +0800841 case CS_ARCH_SYSZ:
842 for (i = 0; i < insn->detail->sysz.op_count; i++) {
843 if (insn->detail->sysz.operands[i].type == (sysz_op_type)op_type)
844 count++;
845 if (count == post)
846 return i;
847 }
848 break;
Nguyen Anh Quynh26ee41a2013-11-27 12:11:31 +0800849 }
850
851 return -1;
852}