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
|
#include "ast.h"
#include "utils.h"
static void dump_usage() {
std::string Usage = R"usage(
usage: command <filename> <libname> [guest_triple] [host_triple] -- <clang_flags>
<filename> : set the header file to be parsed
<libname> : set libname required for wrapping func
[guest_triple]: set guest triple: can be arm32/arm64/x86/x64, default is x64
[host_triple] : set host triple: can be arm32/arm64/x86/x64, default is arm64
-- : mandatory
<clang_flags> : extra compiler flags
)usage";
std::cerr << Usage << std::endl;
}
std::string parse_triple(const char* arg) {
if (strcmp(arg, "x86") == 0) {
return "i386-pc-linux-gnu";
} else if (strcmp(arg, "x64") == 0) {
return "x86_64-pc-linux-gnu";
} else if (strcmp(arg, "arm32") == 0) {
return "armv7-unknown-linux-gnueabihf";
} else if (strcmp(arg, "arm64") == 0) {
return "aarch64-unknown-linux-gnu";
} else {
std::cerr << "Invalid triple: '" << arg << "'\n";
dump_usage();
return "";
}
}
int main(int argc, const char* argv[]) {
if (argc < 4) {
dump_usage();
return 0;
}
std::string libname = argv[2];
std::string guest_triple = parse_triple("x64");
std::string host_triple = parse_triple("arm64");
if (argc >= 5) {
guest_triple = parse_triple(argv[3]);
}
if (argc >= 6) {
host_triple = parse_triple(argv[4]);
}
bool has_necessary_tag = false;
for (int i = 0; i < argc; i++) {
if (strcmp(argv[i], "--") == 0) {
has_necessary_tag = true;
break;
}
}
if (!has_necessary_tag) {
std::cerr << "Please add '--' after the triples" << std::endl;
dump_usage();
return 0;
}
std::string err;
auto compile_db = clang::tooling::FixedCompilationDatabase::loadFromCommandLine(argc, argv, err);
clang::tooling::ClangTool Tool(*compile_db, {argv[1]});
Tool.appendArgumentsAdjuster([&guest_triple](const clang::tooling::CommandLineArguments &args, clang::StringRef) {
clang::tooling::CommandLineArguments adjusted_args = args;
adjusted_args.push_back(std::string{"-target"});
adjusted_args.push_back(guest_triple);
return adjusted_args;
});
return Tool.run(std::make_unique<MyFrontendActionFactory>(libname, host_triple, guest_triple).get());
}
|