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