Nguyen Anh Quynh | 6023ef7 | 2014-04-29 11:21:04 +0800 | [diff] [blame] | 1 | /* Capstone Disassembly Engine */ |
| 2 | /* By Nguyen Anh Quynh <aquynh@gmail.com>, 2013-2014 */ |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 3 | |
reverser | bcf09f4 | 2015-04-09 18:28:19 +0100 | [diff] [blame] | 4 | #if defined(CAPSTONE_HAS_OSXKERNEL) |
| 5 | #include <libkern/libkern.h> |
| 6 | #else |
Nguyen Anh Quynh | b265406 | 2014-01-03 17:08:58 +0800 | [diff] [blame] | 7 | #include <stdlib.h> |
reverser | bcf09f4 | 2015-04-09 18:28:19 +0100 | [diff] [blame] | 8 | #endif |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 9 | #include <string.h> |
| 10 | |
| 11 | #include "utils.h" |
| 12 | |
Nguyen Anh Quynh | b265406 | 2014-01-03 17:08:58 +0800 | [diff] [blame] | 13 | // create a cache for fast id lookup |
| 14 | static unsigned short *make_id2insn(insn_map *insns, unsigned int size) |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 15 | { |
Nguyen Anh Quynh | b265406 | 2014-01-03 17:08:58 +0800 | [diff] [blame] | 16 | // NOTE: assume that the max id is always put at the end of insns array |
| 17 | unsigned short max_id = insns[size - 1].id; |
Alex Ionescu | 46018db | 2014-01-22 09:45:00 -0800 | [diff] [blame] | 18 | unsigned short i; |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 19 | |
Nguyen Anh Quynh | aaddb25 | 2014-06-17 13:32:37 +0800 | [diff] [blame] | 20 | unsigned short *cache = (unsigned short *)cs_mem_malloc(sizeof(*cache) * (max_id + 1)); |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 21 | |
Nguyen Anh Quynh | b265406 | 2014-01-03 17:08:58 +0800 | [diff] [blame] | 22 | for (i = 1; i < size; i++) |
| 23 | cache[insns[i].id] = i; |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 24 | |
Nguyen Anh Quynh | b265406 | 2014-01-03 17:08:58 +0800 | [diff] [blame] | 25 | return cache; |
| 26 | } |
| 27 | |
| 28 | // look for @id in @insns, given its size in @max. first time call will update @cache. |
| 29 | // return 0 if not found |
| 30 | unsigned short insn_find(insn_map *insns, unsigned int max, unsigned int id, unsigned short **cache) |
| 31 | { |
| 32 | if (id > insns[max - 1].id) |
| 33 | return 0; |
| 34 | |
| 35 | if (*cache == NULL) |
| 36 | *cache = make_id2insn(insns, max); |
| 37 | |
| 38 | return (*cache)[id]; |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 39 | } |
| 40 | |
pancake | f0e4eed | 2013-12-11 22:14:42 +0100 | [diff] [blame] | 41 | int name2id(name_map* map, int max, const char *name) |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 42 | { |
| 43 | int i; |
| 44 | |
| 45 | for (i = 0; i < max; i++) { |
Nguyen Anh Quynh | 19146e9 | 2014-05-28 12:41:31 +0800 | [diff] [blame] | 46 | if (!strcmp(map[i].name, name)) { |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 47 | return map[i].id; |
| 48 | } |
| 49 | } |
| 50 | |
| 51 | // nothing match |
| 52 | return -1; |
| 53 | } |
| 54 | |
Nguyen Anh Quynh | f35e2ad | 2013-12-03 11:10:26 +0800 | [diff] [blame] | 55 | // count number of positive members in a list. |
| 56 | // NOTE: list must be guaranteed to end in 0 |
Nguyen Anh Quynh | 18103e4 | 2013-12-20 17:35:15 +0800 | [diff] [blame] | 57 | unsigned int count_positive(unsigned char *list) |
Nguyen Anh Quynh | f35e2ad | 2013-12-03 11:10:26 +0800 | [diff] [blame] | 58 | { |
| 59 | unsigned int c; |
| 60 | |
| 61 | for (c = 0; list[c] > 0; c++); |
| 62 | |
| 63 | return c; |
| 64 | } |
Nguyen Anh Quynh | a9ffb44 | 2014-01-15 18:27:01 +0800 | [diff] [blame] | 65 | |
| 66 | char *cs_strdup(const char *str) |
| 67 | { |
| 68 | size_t len = strlen(str)+ 1; |
| 69 | void *new = cs_mem_malloc(len); |
| 70 | |
| 71 | if (new == NULL) |
| 72 | return NULL; |
| 73 | |
| 74 | return (char *)memmove(new, str, len); |
| 75 | } |
Nguyen Anh Quynh | 4b6b15f | 2014-08-26 15:57:04 +0800 | [diff] [blame] | 76 | |
| 77 | // we need this since Windows doesnt have snprintf() |
| 78 | int cs_snprintf(char *buffer, size_t size, const char *fmt, ...) |
| 79 | { |
| 80 | int ret; |
| 81 | |
| 82 | va_list ap; |
| 83 | va_start(ap, fmt); |
| 84 | ret = cs_vsnprintf(buffer, size, fmt, ap); |
| 85 | va_end(ap); |
| 86 | |
| 87 | return ret; |
| 88 | } |