1 | /*===-- disassemble.c - tool for testing libLLVM and llvm-c API -----------===*\ |
2 | |* *| |
3 | |* Part of the LLVM Project, under the Apache License v2.0 with LLVM *| |
4 | |* Exceptions. *| |
5 | |* See https://llvm.org/LICENSE.txt for license information. *| |
6 | |* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception *| |
7 | |* *| |
8 | |*===----------------------------------------------------------------------===*| |
9 | |* *| |
10 | |* This file implements the --disassemble command in llvm-c-test. *| |
11 | |* --disassemble reads lines from stdin, parses them as a triple and hex *| |
12 | |* machine code, and prints disassembly of the machine code. *| |
13 | |* *| |
14 | \*===----------------------------------------------------------------------===*/ |
15 | |
16 | #include "llvm-c-test.h" |
17 | #include "llvm-c/Disassembler.h" |
18 | #include "llvm-c/Target.h" |
19 | #include <stdio.h> |
20 | #include <stdlib.h> |
21 | #include <string.h> |
22 | |
23 | static void pprint(int pos, unsigned char *buf, int len, const char *disasm) { |
24 | int i; |
25 | printf(format: "%04x: " , pos); |
26 | for (i = 0; i < 8; i++) { |
27 | if (i < len) { |
28 | printf(format: "%02x " , buf[i]); |
29 | } else { |
30 | printf(format: " " ); |
31 | } |
32 | } |
33 | |
34 | printf(format: " %s\n" , disasm); |
35 | } |
36 | |
37 | static void do_disassemble(const char *triple, const char *features, |
38 | unsigned char *buf, int siz) { |
39 | LLVMDisasmContextRef D = LLVMCreateDisasmCPUFeatures(Triple: triple, CPU: "" , Features: features, |
40 | NULL, TagType: 0, NULL, NULL); |
41 | char outline[1024]; |
42 | int pos; |
43 | |
44 | if (!D) { |
45 | printf(format: "ERROR: Couldn't create disassembler for triple %s\n" , triple); |
46 | return; |
47 | } |
48 | |
49 | pos = 0; |
50 | while (pos < siz) { |
51 | size_t l = LLVMDisasmInstruction(DC: D, Bytes: buf + pos, BytesSize: siz - pos, PC: 0, OutString: outline, |
52 | OutStringSize: sizeof(outline)); |
53 | if (!l) { |
54 | pprint(pos, buf: buf + pos, len: 1, disasm: "\t???" ); |
55 | pos++; |
56 | } else { |
57 | pprint(pos, buf: buf + pos, len: l, disasm: outline); |
58 | pos += l; |
59 | } |
60 | } |
61 | |
62 | LLVMDisasmDispose(DC: D); |
63 | } |
64 | |
65 | static void handle_line(char **tokens, int ntokens) { |
66 | unsigned char disbuf[128]; |
67 | size_t disbuflen = 0; |
68 | const char *triple = tokens[0]; |
69 | const char *features = tokens[1]; |
70 | int i; |
71 | |
72 | printf(format: "triple: %s, features: %s\n" , triple, features); |
73 | if (!strcmp(s1: features, s2: "NULL" )) |
74 | features = "" ; |
75 | |
76 | for (i = 2; i < ntokens; i++) { |
77 | disbuf[disbuflen++] = strtol(nptr: tokens[i], NULL, base: 16); |
78 | if (disbuflen >= sizeof(disbuf)) { |
79 | fprintf(stderr, format: "Warning: Too long line, truncating\n" ); |
80 | break; |
81 | } |
82 | } |
83 | do_disassemble(triple, features, buf: disbuf, siz: disbuflen); |
84 | } |
85 | |
86 | int llvm_disassemble(void) { |
87 | LLVMInitializeAllTargetInfos(); |
88 | LLVMInitializeAllTargetMCs(); |
89 | LLVMInitializeAllDisassemblers(); |
90 | |
91 | llvm_tokenize_stdin(cb: handle_line); |
92 | |
93 | return 0; |
94 | } |
95 | |