1 /*
2 * Copyright 2012, The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include <string>
18 #include <vector>
19
20 #include <dlfcn.h>
21 #include <stdlib.h>
22
23 #include <llvm/ADT/STLExtras.h>
24 #include <llvm/ADT/SmallString.h>
25 #include <llvm/Config/config.h>
26 #include <llvm/Support/CommandLine.h>
27 #include <llvm/Support/FileSystem.h>
28 #include <llvm/Support/MemoryBuffer.h>
29 #include <llvm/Support/PluginLoader.h>
30 #include <llvm/Support/raw_ostream.h>
31
32 #include <bcc/BCCContext.h>
33 #include <bcc/Compiler.h>
34 #include <bcc/Config/Config.h>
35 #include <bcc/ExecutionEngine/CompilerRTSymbolResolver.h>
36 #include <bcc/ExecutionEngine/ObjectLoader.h>
37 #include <bcc/ExecutionEngine/SymbolResolverProxy.h>
38 #include <bcc/ExecutionEngine/SymbolResolvers.h>
39 #include <bcc/Renderscript/RSCompilerDriver.h>
40 #include <bcc/Script.h>
41 #include <bcc/Source.h>
42 #include <bcc/Support/CompilerConfig.h>
43 #include <bcc/Support/Initialization.h>
44 #include <bcc/Support/InputFile.h>
45 #include <bcc/Support/OutputFile.h>
46
47 using namespace bcc;
48
49 #define STR2(a) #a
50 #define STR(a) STR2(a)
51
52 //===----------------------------------------------------------------------===//
53 // General Options
54 //===----------------------------------------------------------------------===//
55 namespace {
56
57 llvm::cl::opt<std::string>
58 OptInputFilename(llvm::cl::Positional, llvm::cl::ValueRequired,
59 llvm::cl::desc("<input bitcode file>"));
60
61 llvm::cl::opt<std::string>
62 OptOutputFilename("o", llvm::cl::desc("Specify the output filename"),
63 llvm::cl::value_desc("filename"),
64 llvm::cl::init("bcc_output"));
65
66 llvm::cl::opt<std::string>
67 OptBCLibFilename("bclib", llvm::cl::desc("Specify the bclib filename"),
68 llvm::cl::value_desc("bclib"));
69
70 llvm::cl::opt<std::string>
71 OptOutputPath("output_path", llvm::cl::desc("Specify the output path"),
72 llvm::cl::value_desc("output path"),
73 llvm::cl::init("."));
74
75 llvm::cl::opt<bool>
76 OptEmitLLVM("emit-llvm",
77 llvm::cl::desc("Emit an LLVM-IR version of the generated program"));
78
79 llvm::cl::opt<std::string>
80 OptTargetTriple("mtriple",
81 llvm::cl::desc("Specify the target triple (default: "
82 DEFAULT_TARGET_TRIPLE_STRING ")"),
83 llvm::cl::init(DEFAULT_TARGET_TRIPLE_STRING),
84 llvm::cl::value_desc("triple"));
85
86 llvm::cl::alias OptTargetTripleC("C", llvm::cl::NotHidden,
87 llvm::cl::desc("Alias for -mtriple"),
88 llvm::cl::aliasopt(OptTargetTriple));
89
90 llvm::cl::opt<bool>
91 OptRSDebugContext("rs-debug-ctx",
92 llvm::cl::desc("Enable build to work with a RenderScript debug context"));
93
94 //===----------------------------------------------------------------------===//
95 // Compiler Options
96 //===----------------------------------------------------------------------===//
97
98 // RenderScript uses -O3 by default
99 llvm::cl::opt<char>
100 OptOptLevel("O", llvm::cl::desc("Optimization level. [-O0, -O1, -O2, or -O3] "
101 "(default: -O3)"),
102 llvm::cl::Prefix, llvm::cl::ZeroOrMore, llvm::cl::init('3'));
103
104 // Override "bcc -version" since the LLVM version information is not correct on
105 // Android build.
BCCVersionPrinter()106 void BCCVersionPrinter() {
107 llvm::raw_ostream &os = llvm::outs();
108 os << "libbcc (The Android Open Source Project, http://www.android.com/):\n"
109 << " Default target: " << DEFAULT_TARGET_TRIPLE_STRING << "\n\n"
110 << "LLVM (http://llvm.org/):\n"
111 << " Version: " << PACKAGE_VERSION << "\n";
112 return;
113 }
114
115 } // end anonymous namespace
116
117 static inline
ConfigCompiler(RSCompilerDriver & pRSCD)118 bool ConfigCompiler(RSCompilerDriver &pRSCD) {
119 RSCompiler *RSC = pRSCD.getCompiler();
120 CompilerConfig *config = NULL;
121
122 config = new (std::nothrow) CompilerConfig(OptTargetTriple);
123 if (config == NULL) {
124 llvm::errs() << "Out of memory when create the compiler configuration!\n";
125 return false;
126 }
127
128 // llvm3.5 has removed the auto-detect feature for x86 subtarget,
129 // so set features explicitly in bcc.
130 if ((config->getTriple().find("i686") != std::string::npos) ||
131 (config->getTriple().find("x86_64") != std::string::npos)) {
132 std::vector<std::string> fv;
133
134 #if defined(__SSE3__)
135 fv.push_back("+sse3");
136 #endif
137 #if defined(__SSSE3__)
138 fv.push_back("+ssse3");
139 #endif
140 #if defined(__SSE4_1__)
141 fv.push_back("+sse4.1");
142 #endif
143 #if defined(__SSE4_2__)
144 fv.push_back("+sse4.2");
145 #endif
146
147 if (fv.size()) {
148 config->setFeatureString(fv);
149 }
150 }
151
152 switch (OptOptLevel) {
153 case '0': config->setOptimizationLevel(llvm::CodeGenOpt::None); break;
154 case '1': config->setOptimizationLevel(llvm::CodeGenOpt::Less); break;
155 case '2': config->setOptimizationLevel(llvm::CodeGenOpt::Default); break;
156 case '3':
157 default: {
158 config->setOptimizationLevel(llvm::CodeGenOpt::Aggressive);
159 break;
160 }
161 }
162
163 pRSCD.setConfig(config);
164 Compiler::ErrorCode result = RSC->config(*config);
165
166 if (OptRSDebugContext) {
167 pRSCD.setDebugContext(true);
168 }
169
170 if (result != Compiler::kSuccess) {
171 llvm::errs() << "Failed to configure the compiler! (detail: "
172 << Compiler::GetErrorString(result) << ")\n";
173 return false;
174 }
175
176 return true;
177 }
178
main(int argc,char ** argv)179 int main(int argc, char **argv) {
180 llvm::cl::SetVersionPrinter(BCCVersionPrinter);
181 llvm::cl::ParseCommandLineOptions(argc, argv);
182 std::string commandLine = bcc::getCommandLine(argc, argv);
183 init::Initialize();
184
185 BCCContext context;
186 RSCompilerDriver RSCD;
187
188 if (OptBCLibFilename.empty()) {
189 ALOGE("Failed to compile bit code, -bclib was not specified");
190 return EXIT_FAILURE;
191 }
192
193 llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> mb_or_error =
194 llvm::MemoryBuffer::getFile(OptInputFilename.c_str());
195 if (mb_or_error.getError()) {
196 ALOGE("Failed to load bitcode from path %s! (%s)",
197 OptInputFilename.c_str(), mb_or_error.getError().message().c_str());
198 return EXIT_FAILURE;
199 }
200 std::unique_ptr<llvm::MemoryBuffer> input_data = std::move(mb_or_error.get());
201
202 const char *bitcode = input_data->getBufferStart();
203 size_t bitcodeSize = input_data->getBufferSize();
204
205 if (!ConfigCompiler(RSCD)) {
206 ALOGE("Failed to configure compiler");
207 return EXIT_FAILURE;
208 }
209
210 // Attempt to dynamically initialize the compiler driver if such a function
211 // is present. It is only present if passed via "-load libFOO.so".
212 RSCompilerDriverInit_t rscdi = (RSCompilerDriverInit_t)
213 dlsym(RTLD_DEFAULT, STR(RS_COMPILER_DRIVER_INIT_FN));
214 if (rscdi != NULL) {
215 rscdi(&RSCD);
216 }
217
218 bool built = RSCD.build(context, OptOutputPath.c_str(), OptOutputFilename.c_str(), bitcode,
219 bitcodeSize, commandLine.c_str(), OptBCLibFilename.c_str(), NULL,
220 OptEmitLLVM);
221
222 if (!built) {
223 return EXIT_FAILURE;
224 }
225
226 return EXIT_SUCCESS;
227 }
228