blob: be67bdfa7888bca2d7e257825e68aee69fb0fc8c [file] [log] [blame]
Logan1f028c02010-11-27 01:02:48 +08001/*
2 * Copyright 2010, 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
Loganc4395232010-11-27 18:54:17 +080017#include "Compiler.h"
Logan1f028c02010-11-27 01:02:48 +080018
Logan35849002011-01-15 07:30:43 +080019#include "Config.h"
20
Logan Chiend2a5f302011-07-19 20:32:25 +080021#if USE_OLD_JIT
22#include "OldJIT/ContextManager.h"
23#endif
24
Logan Chien4885cf82011-07-20 10:18:05 +080025#if USE_DISASSEMBLER
26#include "Disassembler/Disassembler.h"
27#endif
28
Logan4dcd6792011-02-28 05:12:00 +080029#include "DebugHelper.h"
Shih-wei Liao6c0c7b02011-05-21 21:47:14 -070030#include "FileHandle.h"
Logan Chienda5e0c32011-06-13 03:47:21 +080031#include "Runtime.h"
Logan2a6dc822011-01-06 04:05:20 +080032#include "ScriptCompiled.h"
Logan75cc8a52011-01-07 06:06:52 +080033#include "Sha1Helper.h"
Loganeb3d12b2010-12-16 06:20:18 +080034
Shih-wei Liao320b5492011-06-20 22:53:33 -070035#if USE_MCJIT
Logan Chienda5e0c32011-06-13 03:47:21 +080036#include "librsloader.h"
Shih-wei Liao320b5492011-06-20 22:53:33 -070037#endif
Logan Chienda5e0c32011-06-13 03:47:21 +080038
Logandf23afa2010-11-27 11:04:54 +080039#include "llvm/ADT/StringRef.h"
Logan1f028c02010-11-27 01:02:48 +080040
Logandf23afa2010-11-27 11:04:54 +080041#include "llvm/Analysis/Passes.h"
Logan1f028c02010-11-27 01:02:48 +080042
Logan1f028c02010-11-27 01:02:48 +080043#include "llvm/Bitcode/ReaderWriter.h"
44
Logan1f028c02010-11-27 01:02:48 +080045#include "llvm/CodeGen/Passes.h"
Logan1f028c02010-11-27 01:02:48 +080046#include "llvm/CodeGen/RegAllocRegistry.h"
47#include "llvm/CodeGen/SchedulerRegistry.h"
Logan1f028c02010-11-27 01:02:48 +080048
Logan Chien9347e0b2011-07-07 19:51:47 +080049#include "llvm/MC/SubtargetFeature.h"
50
Logandf23afa2010-11-27 11:04:54 +080051#include "llvm/Transforms/IPO.h"
52#include "llvm/Transforms/Scalar.h"
53
Logandf23afa2010-11-27 11:04:54 +080054#include "llvm/Target/TargetData.h"
55#include "llvm/Target/TargetMachine.h"
56#include "llvm/Target/TargetOptions.h"
Logandf23afa2010-11-27 11:04:54 +080057
58#include "llvm/Support/ErrorHandling.h"
Shih-wei Liao898c5a92011-05-18 07:02:39 -070059#include "llvm/Support/FormattedStream.h"
Logandf23afa2010-11-27 11:04:54 +080060#include "llvm/Support/MemoryBuffer.h"
Logan Chienbc9eb8f2011-10-21 15:17:45 +080061#include "llvm/Support/TargetRegistry.h"
62#include "llvm/Support/TargetSelect.h"
Logandf23afa2010-11-27 11:04:54 +080063
Shih-wei Liao90cd3d12011-06-20 15:43:34 -070064#include "llvm/Type.h"
Logandf23afa2010-11-27 11:04:54 +080065#include "llvm/GlobalValue.h"
66#include "llvm/Linker.h"
67#include "llvm/LLVMContext.h"
68#include "llvm/Metadata.h"
69#include "llvm/Module.h"
70#include "llvm/PassManager.h"
71#include "llvm/Value.h"
72
73#include <errno.h>
74#include <sys/file.h>
Logandf23afa2010-11-27 11:04:54 +080075#include <sys/stat.h>
76#include <sys/types.h>
77#include <unistd.h>
78
Logan75cc8a52011-01-07 06:06:52 +080079#include <string.h>
Logan8b77a772010-12-21 09:11:01 +080080
Logan Chien7890d432011-08-03 14:55:17 +080081#include <algorithm>
82#include <iterator>
Logandf23afa2010-11-27 11:04:54 +080083#include <string>
84#include <vector>
Logan1f028c02010-11-27 01:02:48 +080085
Logan1f028c02010-11-27 01:02:48 +080086namespace bcc {
87
88//////////////////////////////////////////////////////////////////////////////
89// BCC Compiler Static Variables
90//////////////////////////////////////////////////////////////////////////////
91
92bool Compiler::GlobalInitialized = false;
93
Logan1f028c02010-11-27 01:02:48 +080094// Code generation optimization level for the compiler
95llvm::CodeGenOpt::Level Compiler::CodeGenOptLevel;
96
97std::string Compiler::Triple;
98
99std::string Compiler::CPU;
100
101std::vector<std::string> Compiler::Features;
102
Stephen Hines071288a2011-01-27 14:38:26 -0800103// Name of metadata node where pragma info resides (should be synced with
Logan1f028c02010-11-27 01:02:48 +0800104// slang.cpp)
105const llvm::StringRef Compiler::PragmaMetadataName = "#pragma";
106
Stephen Hines071288a2011-01-27 14:38:26 -0800107// Name of metadata node where exported variable names reside (should be
Logan1f028c02010-11-27 01:02:48 +0800108// synced with slang_rs_metadata.h)
109const llvm::StringRef Compiler::ExportVarMetadataName = "#rs_export_var";
110
Stephen Hines071288a2011-01-27 14:38:26 -0800111// Name of metadata node where exported function names reside (should be
Logan1f028c02010-11-27 01:02:48 +0800112// synced with slang_rs_metadata.h)
113const llvm::StringRef Compiler::ExportFuncMetadataName = "#rs_export_func";
114
Stephen Hines071288a2011-01-27 14:38:26 -0800115// Name of metadata node where RS object slot info resides (should be
116// synced with slang_rs_metadata.h)
117const llvm::StringRef Compiler::ObjectSlotMetadataName = "#rs_object_slots";
Logan1f028c02010-11-27 01:02:48 +0800118
119//////////////////////////////////////////////////////////////////////////////
120// Compiler
121//////////////////////////////////////////////////////////////////////////////
122
123void Compiler::GlobalInitialization() {
124 if (GlobalInitialized)
125 return;
Logan1f028c02010-11-27 01:02:48 +0800126 // if (!llvm::llvm_is_multithreaded())
127 // llvm::llvm_start_multithreaded();
128
129 // Set Triple, CPU and Features here
130 Triple = TARGET_TRIPLE_STRING;
131
Logan Chien3bb77072011-09-17 16:53:53 +0800132#if defined(DEFAULT_ARM_CODEGEN)
133
Stephen Hinesa12d2f32011-09-07 15:30:06 -0700134#if defined(ARCH_ARM_HAVE_VFP)
135 Features.push_back("+vfp3");
136#if !defined(ARCH_ARM_HAVE_VFP_D32)
137 Features.push_back("+d16");
138#endif
139#endif
140
Logan Chien3bb77072011-09-17 16:53:53 +0800141 // NOTE: Currently, we have to turn off the support for NEON explicitly.
142 // Since the ARMCodeEmitter.cpp is not ready for JITing NEON
143 // instructions.
144
145 // FIXME: Re-enable NEON when ARMCodeEmitter supports NEON.
146#define USE_ARM_NEON 0
147#if USE_ARM_NEON
Joseph Wen51001b82011-06-23 18:56:45 -0700148 Features.push_back("+neon");
149 Features.push_back("+neonfp");
150#else
Joseph Wen51001b82011-06-23 18:56:45 -0700151 Features.push_back("-neon");
Logan4fe966f2011-02-27 08:26:40 +0800152 Features.push_back("-neonfp");
Logan Chien3bb77072011-09-17 16:53:53 +0800153#endif // USE_ARM_NEON
154#endif // DEFAULT_ARM_CODEGEN
Joseph Wen51001b82011-06-23 18:56:45 -0700155
Logan Chien3bb77072011-09-17 16:53:53 +0800156#if defined(PROVIDE_ARM_CODEGEN)
Logan Chienb9ef9ab2011-07-20 11:11:06 +0800157 LLVMInitializeARMAsmPrinter();
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800158 LLVMInitializeARMTargetMC();
Logan1f028c02010-11-27 01:02:48 +0800159 LLVMInitializeARMTargetInfo();
160 LLVMInitializeARMTarget();
Logan1f028c02010-11-27 01:02:48 +0800161#endif
162
Logan Chien3bb77072011-09-17 16:53:53 +0800163#if defined(PROVIDE_X86_CODEGEN)
Logan Chienb9ef9ab2011-07-20 11:11:06 +0800164 LLVMInitializeX86AsmPrinter();
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800165 LLVMInitializeX86TargetMC();
Logan1f028c02010-11-27 01:02:48 +0800166 LLVMInitializeX86TargetInfo();
167 LLVMInitializeX86Target();
Logan1f028c02010-11-27 01:02:48 +0800168#endif
Logan Chien9347e0b2011-07-07 19:51:47 +0800169
170#if USE_DISASSEMBLER
171 InitializeDisassembler();
Logan1f028c02010-11-27 01:02:48 +0800172#endif
173
174 // -O0: llvm::CodeGenOpt::None
175 // -O1: llvm::CodeGenOpt::Less
176 // -O2: llvm::CodeGenOpt::Default
177 // -O3: llvm::CodeGenOpt::Aggressive
Shih-wei Liao72f67a62010-12-14 13:36:15 -0800178 CodeGenOptLevel = llvm::CodeGenOpt::Aggressive;
Logan1f028c02010-11-27 01:02:48 +0800179
180 // Below are the global settings to LLVM
181
182 // Disable frame pointer elimination optimization
183 llvm::NoFramePointerElim = false;
184
185 // Use hardfloat ABI
186 //
187 // TODO(all): Need to detect the CPU capability and decide whether to use
188 // softfp. To use softfp, change following 2 lines to
189 //
190 // llvm::FloatABIType = llvm::FloatABI::Soft;
191 // llvm::UseSoftFloat = true;
192 //
Shih-wei Liaoe728cb82010-12-15 15:20:47 -0800193 llvm::FloatABIType = llvm::FloatABI::Soft;
Logan1f028c02010-11-27 01:02:48 +0800194 llvm::UseSoftFloat = false;
195
Logan1f028c02010-11-27 01:02:48 +0800196 // Register the scheduler
197 llvm::RegisterScheduler::setDefault(llvm::createDefaultScheduler);
198
199 // Register allocation policy:
200 // createFastRegisterAllocator: fast but bad quality
201 // createLinearScanRegisterAllocator: not so fast but good quality
202 llvm::RegisterRegAlloc::setDefault
203 ((CodeGenOptLevel == llvm::CodeGenOpt::None) ?
204 llvm::createFastRegisterAllocator :
205 llvm::createLinearScanRegisterAllocator);
206
Logan35849002011-01-15 07:30:43 +0800207#if USE_CACHE
Ying Wang26fea102011-07-05 15:12:25 -0700208 // Read in SHA1 checksum of libbcc and libRS.
209 readSHA1(sha1LibBCC_SHA1, sizeof(sha1LibBCC_SHA1), pathLibBCC_SHA1);
Joseph Wen76919072011-07-07 23:06:15 -0700210
211 calcFileSHA1(sha1LibRS, pathLibRS);
Logan35849002011-01-15 07:30:43 +0800212#endif
Logan75cc8a52011-01-07 06:06:52 +0800213
Logan1f028c02010-11-27 01:02:48 +0800214 GlobalInitialized = true;
215}
216
217
218void Compiler::LLVMErrorHandler(void *UserData, const std::string &Message) {
219 std::string *Error = static_cast<std::string*>(UserData);
220 Error->assign(Message);
221 LOGE("%s", Message.c_str());
222 exit(1);
223}
224
225
Logan Chienda5e0c32011-06-13 03:47:21 +0800226#if USE_OLD_JIT
Logan1f028c02010-11-27 01:02:48 +0800227CodeMemoryManager *Compiler::createCodeMemoryManager() {
228 mCodeMemMgr.reset(new CodeMemoryManager());
229 return mCodeMemMgr.get();
230}
Logan Chienda5e0c32011-06-13 03:47:21 +0800231#endif
Logan1f028c02010-11-27 01:02:48 +0800232
233
Logan Chienda5e0c32011-06-13 03:47:21 +0800234#if USE_OLD_JIT
Logan1f028c02010-11-27 01:02:48 +0800235CodeEmitter *Compiler::createCodeEmitter() {
Logan7dcaac92011-01-06 04:26:23 +0800236 mCodeEmitter.reset(new CodeEmitter(mpResult, mCodeMemMgr.get()));
Logan1f028c02010-11-27 01:02:48 +0800237 return mCodeEmitter.get();
238}
Logan Chienda5e0c32011-06-13 03:47:21 +0800239#endif
Logan1f028c02010-11-27 01:02:48 +0800240
241
Logan2a6dc822011-01-06 04:05:20 +0800242Compiler::Compiler(ScriptCompiled *result)
243 : mpResult(result),
Logan Chienda5e0c32011-06-13 03:47:21 +0800244#if USE_MCJIT
245 mRSExecutable(NULL),
246#endif
Logan1f028c02010-11-27 01:02:48 +0800247 mpSymbolLookupFn(NULL),
248 mpSymbolLookupContext(NULL),
249 mContext(NULL),
250 mModule(NULL),
251 mHasLinked(false) /* Turn off linker */ {
252 llvm::remove_fatal_error_handler();
253 llvm::install_fatal_error_handler(LLVMErrorHandler, &mError);
254 mContext = new llvm::LLVMContext();
255 return;
256}
257
Logan1f028c02010-11-27 01:02:48 +0800258
Logan474cbd22011-01-31 01:47:44 +0800259llvm::Module *Compiler::parseBitcodeFile(llvm::MemoryBuffer *MEM) {
260 llvm::Module *result = llvm::ParseBitcodeFile(MEM, *mContext, &mError);
Logan1f028c02010-11-27 01:02:48 +0800261
Logan474cbd22011-01-31 01:47:44 +0800262 if (!result) {
263 LOGE("Unable to ParseBitcodeFile: %s\n", mError.c_str());
264 return NULL;
Logan1f028c02010-11-27 01:02:48 +0800265 }
266
Logan474cbd22011-01-31 01:47:44 +0800267 return result;
Logan1f028c02010-11-27 01:02:48 +0800268}
269
270
Logan474cbd22011-01-31 01:47:44 +0800271int Compiler::linkModule(llvm::Module *moduleWith) {
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800272 if (llvm::Linker::LinkModules(mModule, moduleWith,
273 llvm::Linker::DestroySource,
274 &mError) != 0) {
Logan1f028c02010-11-27 01:02:48 +0800275 return hasError();
276 }
277
Logan1f028c02010-11-27 01:02:48 +0800278 // Everything for linking should be settled down here with no error occurs
279 mHasLinked = true;
280 return hasError();
281}
282
283
Joseph Wen34c600a2011-07-25 17:59:17 -0700284int Compiler::compile(bool compileOnly) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800285 llvm::Target const *Target = NULL;
Logan1f028c02010-11-27 01:02:48 +0800286 llvm::TargetData *TD = NULL;
Logan1f028c02010-11-27 01:02:48 +0800287 llvm::TargetMachine *TM = NULL;
Logan Chienda5e0c32011-06-13 03:47:21 +0800288
Logan1f028c02010-11-27 01:02:48 +0800289 std::string FeaturesStr;
290
Logan Chienda5e0c32011-06-13 03:47:21 +0800291 llvm::NamedMDNode const *PragmaMetadata;
292 llvm::NamedMDNode const *ExportVarMetadata;
293 llvm::NamedMDNode const *ExportFuncMetadata;
294 llvm::NamedMDNode const *ObjectSlotMetadata;
Logan1f028c02010-11-27 01:02:48 +0800295
Logan1f028c02010-11-27 01:02:48 +0800296 if (mModule == NULL) // No module was loaded
297 return 0;
298
299 // Create TargetMachine
300 Target = llvm::TargetRegistry::lookupTarget(Triple, mError);
301 if (hasError())
302 goto on_bcc_compile_error;
303
304 if (!CPU.empty() || !Features.empty()) {
305 llvm::SubtargetFeatures F;
Logana4994f52010-11-27 14:06:02 +0800306
307 for (std::vector<std::string>::const_iterator
308 I = Features.begin(), E = Features.end(); I != E; I++) {
Logan1f028c02010-11-27 01:02:48 +0800309 F.AddFeature(*I);
Logana4994f52010-11-27 14:06:02 +0800310 }
311
Logan1f028c02010-11-27 01:02:48 +0800312 FeaturesStr = F.getString();
313 }
314
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800315#if defined(DEFAULT_X86_64_CODEGEN)
316 // Data address in X86_64 architecture may reside in a far-away place
Logan Chiend3fe15c2011-07-20 22:08:39 +0800317 TM = Target->createTargetMachine(Triple, CPU, FeaturesStr,
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800318 llvm::Reloc::Static,
319 llvm::CodeModel::Medium);
320#else
321 // This is set for the linker (specify how large of the virtual addresses
322 // we can access for all unknown symbols.)
323 TM = Target->createTargetMachine(Triple, CPU, FeaturesStr,
324 llvm::Reloc::Static,
325 llvm::CodeModel::Small);
326#endif
Logan1f028c02010-11-27 01:02:48 +0800327 if (TM == NULL) {
328 setError("Failed to create target machine implementation for the"
329 " specified triple '" + Triple + "'");
330 goto on_bcc_compile_error;
331 }
332
Logan Chienda5e0c32011-06-13 03:47:21 +0800333 // Get target data from Module
334 TD = new llvm::TargetData(mModule);
335
336 // Load named metadata
337 ExportVarMetadata = mModule->getNamedMetadata(ExportVarMetadataName);
338 ExportFuncMetadata = mModule->getNamedMetadata(ExportFuncMetadataName);
339 PragmaMetadata = mModule->getNamedMetadata(PragmaMetadataName);
340 ObjectSlotMetadata = mModule->getNamedMetadata(ObjectSlotMetadataName);
341
342 // Perform link-time optimization if we have multiple modules
343 if (mHasLinked) {
344 runLTO(new llvm::TargetData(*TD), ExportVarMetadata, ExportFuncMetadata);
345 }
346
347 // Perform code generation
348#if USE_OLD_JIT
349 if (runCodeGen(new llvm::TargetData(*TD), TM,
350 ExportVarMetadata, ExportFuncMetadata) != 0) {
351 goto on_bcc_compile_error;
352 }
353#endif
354
355#if USE_MCJIT
Joseph Wen34c600a2011-07-25 17:59:17 -0700356 if (runMCCodeGen(new llvm::TargetData(*TD), TM) != 0) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800357 goto on_bcc_compile_error;
358 }
Logan Chien9347e0b2011-07-07 19:51:47 +0800359
Joseph Wen34c600a2011-07-25 17:59:17 -0700360 if (compileOnly)
361 return 0;
362
363 // Load the ELF Object
364 mRSExecutable =
365 rsloaderCreateExec((unsigned char *)&*mEmittedELFExecutable.begin(),
366 mEmittedELFExecutable.size(),
367 &resolveSymbolAdapter, this);
368
369 if (!mRSExecutable) {
370 setError("Fail to load emitted ELF relocatable file");
371 goto on_bcc_compile_error;
372 }
373
374 if (ExportVarMetadata) {
375 ScriptCompiled::ExportVarList &varList = mpResult->mExportVars;
376 std::vector<std::string> &varNameList = mpResult->mExportVarsName;
377
378 for (int i = 0, e = ExportVarMetadata->getNumOperands(); i != e; i++) {
379 llvm::MDNode *ExportVar = ExportVarMetadata->getOperand(i);
380 if (ExportVar != NULL && ExportVar->getNumOperands() > 1) {
381 llvm::Value *ExportVarNameMDS = ExportVar->getOperand(0);
382 if (ExportVarNameMDS->getValueID() == llvm::Value::MDStringVal) {
383 llvm::StringRef ExportVarName =
384 static_cast<llvm::MDString*>(ExportVarNameMDS)->getString();
385
386 varList.push_back(
387 rsloaderGetSymbolAddress(mRSExecutable,
388 ExportVarName.str().c_str()));
389 varNameList.push_back(ExportVarName.str());
390#if DEBUG_MCJIT_REFLECT
391 LOGD("runMCCodeGen(): Exported Var: %s @ %p\n", ExportVarName.str().c_str(),
392 varList.back());
393#endif
394 continue;
395 }
396 }
397
398 varList.push_back(NULL);
399 }
400 }
401
402 if (ExportFuncMetadata) {
403 ScriptCompiled::ExportFuncList &funcList = mpResult->mExportFuncs;
404 std::vector<std::string> &funcNameList = mpResult->mExportFuncsName;
405
406 for (int i = 0, e = ExportFuncMetadata->getNumOperands(); i != e; i++) {
407 llvm::MDNode *ExportFunc = ExportFuncMetadata->getOperand(i);
408 if (ExportFunc != NULL && ExportFunc->getNumOperands() > 0) {
409 llvm::Value *ExportFuncNameMDS = ExportFunc->getOperand(0);
410 if (ExportFuncNameMDS->getValueID() == llvm::Value::MDStringVal) {
411 llvm::StringRef ExportFuncName =
412 static_cast<llvm::MDString*>(ExportFuncNameMDS)->getString();
413
414 funcList.push_back(
415 rsloaderGetSymbolAddress(mRSExecutable,
416 ExportFuncName.str().c_str()));
417 funcNameList.push_back(ExportFuncName.str());
418#if DEBUG_MCJIT_RELECT
419 LOGD("runMCCodeGen(): Exported Func: %s @ %p\n", ExportFuncName.str().c_str(),
420 funcList.back());
421#endif
422 }
423 }
424 }
425 }
426
Logan Chien4885cf82011-07-20 10:18:05 +0800427#if DEBUG_MCJIT_DISASSEMBLER
Shih-wei Liao90cd3d12011-06-20 15:43:34 -0700428 {
Shih-wei Liaod3c551f2011-07-01 04:28:27 -0700429 // Get MC codegen emitted function name list
430 size_t func_list_size = rsloaderGetFuncCount(mRSExecutable);
431 std::vector<char const *> func_list(func_list_size, NULL);
432 rsloaderGetFuncNameList(mRSExecutable, func_list_size, &*func_list.begin());
Shih-wei Liao320b5492011-06-20 22:53:33 -0700433
Shih-wei Liaod3c551f2011-07-01 04:28:27 -0700434 // Disassemble each function
Shih-wei Liao90cd3d12011-06-20 15:43:34 -0700435 for (size_t i = 0; i < func_list_size; ++i) {
436 void *func = rsloaderGetSymbolAddress(mRSExecutable, func_list[i]);
437 if (func) {
438 size_t size = rsloaderGetSymbolSize(mRSExecutable, func_list[i]);
Logan Chien9347e0b2011-07-07 19:51:47 +0800439 Disassemble(DEBUG_MCJIT_DISASSEMBLER_FILE,
440 Target, TM, func_list[i], (unsigned char const *)func, size);
Shih-wei Liao90cd3d12011-06-20 15:43:34 -0700441 }
442 }
443 }
444#endif
Logan Chienda5e0c32011-06-13 03:47:21 +0800445#endif
446
447 // Read pragma information from the metadata node of the module.
448 if (PragmaMetadata) {
449 ScriptCompiled::PragmaList &pragmaList = mpResult->mPragmas;
450
451 for (int i = 0, e = PragmaMetadata->getNumOperands(); i != e; i++) {
452 llvm::MDNode *Pragma = PragmaMetadata->getOperand(i);
453 if (Pragma != NULL &&
454 Pragma->getNumOperands() == 2 /* should have exactly 2 operands */) {
455 llvm::Value *PragmaNameMDS = Pragma->getOperand(0);
456 llvm::Value *PragmaValueMDS = Pragma->getOperand(1);
457
458 if ((PragmaNameMDS->getValueID() == llvm::Value::MDStringVal) &&
459 (PragmaValueMDS->getValueID() == llvm::Value::MDStringVal)) {
460 llvm::StringRef PragmaName =
461 static_cast<llvm::MDString*>(PragmaNameMDS)->getString();
462 llvm::StringRef PragmaValue =
463 static_cast<llvm::MDString*>(PragmaValueMDS)->getString();
464
465 pragmaList.push_back(
466 std::make_pair(std::string(PragmaName.data(),
467 PragmaName.size()),
468 std::string(PragmaValue.data(),
469 PragmaValue.size())));
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700470#if DEBUG_BCC_REFLECT
471 LOGD("compile(): Pragma: %s -> %s\n",
472 pragmaList.back().first.c_str(),
473 pragmaList.back().second.c_str());
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700474#endif
Logan Chienda5e0c32011-06-13 03:47:21 +0800475 }
476 }
477 }
Logan Chienda5e0c32011-06-13 03:47:21 +0800478 }
479
480 if (ObjectSlotMetadata) {
481 ScriptCompiled::ObjectSlotList &objectSlotList = mpResult->mObjectSlots;
482
483 for (int i = 0, e = ObjectSlotMetadata->getNumOperands(); i != e; i++) {
484 llvm::MDNode *ObjectSlot = ObjectSlotMetadata->getOperand(i);
485 if (ObjectSlot != NULL &&
486 ObjectSlot->getNumOperands() == 1) {
487 llvm::Value *SlotMDS = ObjectSlot->getOperand(0);
488 if (SlotMDS->getValueID() == llvm::Value::MDStringVal) {
489 llvm::StringRef Slot =
490 static_cast<llvm::MDString*>(SlotMDS)->getString();
491 uint32_t USlot = 0;
492 if (Slot.getAsInteger(10, USlot)) {
493 setError("Non-integer object slot value '" + Slot.str() + "'");
494 goto on_bcc_compile_error;
495 }
496 objectSlotList.push_back(USlot);
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700497#if DEBUG_BCC_REFLECT
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700498 LOGD("compile(): RefCount Slot: %s @ %u\n", Slot.str().c_str(), USlot);
499#endif
Logan Chienda5e0c32011-06-13 03:47:21 +0800500 }
501 }
502 }
Logan Chienda5e0c32011-06-13 03:47:21 +0800503 }
504
505on_bcc_compile_error:
506 // LOGE("on_bcc_compiler_error");
507 if (TD) {
508 delete TD;
509 }
510
511 if (TM) {
512 delete TM;
513 }
514
515 if (mError.empty()) {
516 return 0;
517 }
518
519 // LOGE(getErrorMessage());
520 return 1;
521}
522
523
524#if USE_OLD_JIT
525int Compiler::runCodeGen(llvm::TargetData *TD, llvm::TargetMachine *TM,
526 llvm::NamedMDNode const *ExportVarMetadata,
527 llvm::NamedMDNode const *ExportFuncMetadata) {
Logan1f028c02010-11-27 01:02:48 +0800528 // Create memory manager for creation of code emitter later.
529 if (!mCodeMemMgr.get() && !createCodeMemoryManager()) {
530 setError("Failed to startup memory management for further compilation");
Logan Chienda5e0c32011-06-13 03:47:21 +0800531 return 1;
Logan1f028c02010-11-27 01:02:48 +0800532 }
Logan02286cb2011-01-07 00:30:47 +0800533
534 mpResult->mContext = (char *) (mCodeMemMgr.get()->getCodeMemBase());
Logan1f028c02010-11-27 01:02:48 +0800535
536 // Create code emitter
537 if (!mCodeEmitter.get()) {
538 if (!createCodeEmitter()) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800539 setError("Failed to create machine code emitter for compilation");
540 return 1;
Logan1f028c02010-11-27 01:02:48 +0800541 }
542 } else {
543 // Reuse the code emitter
544 mCodeEmitter->reset();
545 }
546
547 mCodeEmitter->setTargetMachine(*TM);
548 mCodeEmitter->registerSymbolCallback(mpSymbolLookupFn,
549 mpSymbolLookupContext);
550
Logan1f028c02010-11-27 01:02:48 +0800551 // Create code-gen pass to run the code emitter
Logan Chienda5e0c32011-06-13 03:47:21 +0800552 llvm::OwningPtr<llvm::FunctionPassManager> CodeGenPasses(
553 new llvm::FunctionPassManager(mModule));
Logan1f028c02010-11-27 01:02:48 +0800554
Logan Chienda5e0c32011-06-13 03:47:21 +0800555 // Add TargetData to code generation pass manager
556 CodeGenPasses->add(TD);
557
558 // Add code emit passes
Logan1f028c02010-11-27 01:02:48 +0800559 if (TM->addPassesToEmitMachineCode(*CodeGenPasses,
560 *mCodeEmitter,
561 CodeGenOptLevel)) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800562 setError("The machine code emission is not supported on '" + Triple + "'");
563 return 1;
Logan1f028c02010-11-27 01:02:48 +0800564 }
565
Logan Chienda5e0c32011-06-13 03:47:21 +0800566 // Run the code emitter on every non-declaration function in the module
Logan1f028c02010-11-27 01:02:48 +0800567 CodeGenPasses->doInitialization();
Logan Chienda5e0c32011-06-13 03:47:21 +0800568 for (llvm::Module::iterator
569 I = mModule->begin(), E = mModule->end(); I != E; I++) {
Logan1f028c02010-11-27 01:02:48 +0800570 if (!I->isDeclaration()) {
571 CodeGenPasses->run(*I);
572 }
573 }
574
575 CodeGenPasses->doFinalization();
Logan Chienb0ceca22011-06-12 13:34:49 +0800576
Logan1f028c02010-11-27 01:02:48 +0800577 // Copy the global address mapping from code emitter and remapping
578 if (ExportVarMetadata) {
Logan2a6dc822011-01-06 04:05:20 +0800579 ScriptCompiled::ExportVarList &varList = mpResult->mExportVars;
580
Logan1f028c02010-11-27 01:02:48 +0800581 for (int i = 0, e = ExportVarMetadata->getNumOperands(); i != e; i++) {
582 llvm::MDNode *ExportVar = ExportVarMetadata->getOperand(i);
583 if (ExportVar != NULL && ExportVar->getNumOperands() > 1) {
584 llvm::Value *ExportVarNameMDS = ExportVar->getOperand(0);
585 if (ExportVarNameMDS->getValueID() == llvm::Value::MDStringVal) {
586 llvm::StringRef ExportVarName =
587 static_cast<llvm::MDString*>(ExportVarNameMDS)->getString();
588
589 CodeEmitter::global_addresses_const_iterator I, E;
590 for (I = mCodeEmitter->global_address_begin(),
591 E = mCodeEmitter->global_address_end();
592 I != E; I++) {
593 if (I->first->getValueID() != llvm::Value::GlobalVariableVal)
594 continue;
595 if (ExportVarName == I->first->getName()) {
Logan2a6dc822011-01-06 04:05:20 +0800596 varList.push_back(I->second);
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700597#if DEBUG_BCC_REFLECT
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700598 LOGD("runCodeGen(): Exported VAR: %s @ %p\n", ExportVarName.str().c_str(), I->second);
599#endif
Logan1f028c02010-11-27 01:02:48 +0800600 break;
601 }
602 }
603 if (I != mCodeEmitter->global_address_end())
604 continue; // found
Logan Chien7d1bf582011-06-13 23:22:40 +0800605
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700606#if DEBUG_BCC_REFLECT
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700607 LOGD("runCodeGen(): Exported VAR: %s @ %p\n",
Logan Chien7d1bf582011-06-13 23:22:40 +0800608 ExportVarName.str().c_str(), (void *)0);
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700609#endif
Logan1f028c02010-11-27 01:02:48 +0800610 }
611 }
612 // if reaching here, we know the global variable record in metadata is
613 // not found. So we make an empty slot
Logan2a6dc822011-01-06 04:05:20 +0800614 varList.push_back(NULL);
Logan1f028c02010-11-27 01:02:48 +0800615 }
Logan2a6dc822011-01-06 04:05:20 +0800616
Stephen Hinesbbcef8a2011-05-04 19:40:10 -0700617 bccAssert((varList.size() == ExportVarMetadata->getNumOperands()) &&
618 "Number of slots doesn't match the number of export variables!");
Logan1f028c02010-11-27 01:02:48 +0800619 }
620
621 if (ExportFuncMetadata) {
Logan2a6dc822011-01-06 04:05:20 +0800622 ScriptCompiled::ExportFuncList &funcList = mpResult->mExportFuncs;
623
Logan1f028c02010-11-27 01:02:48 +0800624 for (int i = 0, e = ExportFuncMetadata->getNumOperands(); i != e; i++) {
625 llvm::MDNode *ExportFunc = ExportFuncMetadata->getOperand(i);
626 if (ExportFunc != NULL && ExportFunc->getNumOperands() > 0) {
627 llvm::Value *ExportFuncNameMDS = ExportFunc->getOperand(0);
628 if (ExportFuncNameMDS->getValueID() == llvm::Value::MDStringVal) {
629 llvm::StringRef ExportFuncName =
630 static_cast<llvm::MDString*>(ExportFuncNameMDS)->getString();
Logan7dcaac92011-01-06 04:26:23 +0800631 funcList.push_back(mpResult->lookup(ExportFuncName.str().c_str()));
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700632#if DEBUG_BCC_REFLECT
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700633 LOGD("runCodeGen(): Exported Func: %s @ %p\n", ExportFuncName.str().c_str(),
Logan Chien7d1bf582011-06-13 23:22:40 +0800634 funcList.back());
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700635#endif
Logan1f028c02010-11-27 01:02:48 +0800636 }
637 }
638 }
639 }
640
641 // Tell code emitter now can release the memory using during the JIT since
642 // we have done the code emission
643 mCodeEmitter->releaseUnnecessary();
644
Logan Chienda5e0c32011-06-13 03:47:21 +0800645 return 0;
Logan1f028c02010-11-27 01:02:48 +0800646}
Logan Chienda5e0c32011-06-13 03:47:21 +0800647#endif // USE_OLD_JIT
Logan1f028c02010-11-27 01:02:48 +0800648
649
Logan Chienda5e0c32011-06-13 03:47:21 +0800650#if USE_MCJIT
Joseph Wen34c600a2011-07-25 17:59:17 -0700651int Compiler::runMCCodeGen(llvm::TargetData *TD, llvm::TargetMachine *TM) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800652 // Decorate mEmittedELFExecutable with formatted ostream
653 llvm::raw_svector_ostream OutSVOS(mEmittedELFExecutable);
654
655 // Relax all machine instructions
656 TM->setMCRelaxAll(/* RelaxAll= */ true);
657
658 // Create MC code generation pass manager
659 llvm::PassManager MCCodeGenPasses;
660
661 // Add TargetData to MC code generation pass manager
662 MCCodeGenPasses.add(TD);
663
664 // Add MC code generation passes to pass manager
665 llvm::MCContext *Ctx;
666 if (TM->addPassesToEmitMC(MCCodeGenPasses, Ctx, OutSVOS,
667 CodeGenOptLevel, false)) {
668 setError("Fail to add passes to emit file");
669 return 1;
670 }
671
672 MCCodeGenPasses.run(*mModule);
673 OutSVOS.flush();
Logan Chienda5e0c32011-06-13 03:47:21 +0800674 return 0;
675}
676#endif // USE_MCJIT
677
678
679int Compiler::runLTO(llvm::TargetData *TD,
680 llvm::NamedMDNode const *ExportVarMetadata,
681 llvm::NamedMDNode const *ExportFuncMetadata) {
Logan Chien4cc00332011-06-12 14:00:46 +0800682 llvm::PassManager LTOPasses;
683
684 // Add TargetData to LTO passes
685 LTOPasses.add(TD);
686
687 // Collect All Exported Symbols
688 std::vector<const char*> ExportSymbols;
689
690 // Note: This is a workaround for getting export variable and function name.
691 // We should refine it soon.
692 if (ExportVarMetadata) {
693 for (int i = 0, e = ExportVarMetadata->getNumOperands(); i != e; i++) {
694 llvm::MDNode *ExportVar = ExportVarMetadata->getOperand(i);
695 if (ExportVar != NULL && ExportVar->getNumOperands() > 1) {
696 llvm::Value *ExportVarNameMDS = ExportVar->getOperand(0);
697 if (ExportVarNameMDS->getValueID() == llvm::Value::MDStringVal) {
698 llvm::StringRef ExportVarName =
699 static_cast<llvm::MDString*>(ExportVarNameMDS)->getString();
700 ExportSymbols.push_back(ExportVarName.data());
701 }
702 }
703 }
704 }
705
706 if (ExportFuncMetadata) {
707 for (int i = 0, e = ExportFuncMetadata->getNumOperands(); i != e; i++) {
708 llvm::MDNode *ExportFunc = ExportFuncMetadata->getOperand(i);
709 if (ExportFunc != NULL && ExportFunc->getNumOperands() > 0) {
710 llvm::Value *ExportFuncNameMDS = ExportFunc->getOperand(0);
711 if (ExportFuncNameMDS->getValueID() == llvm::Value::MDStringVal) {
712 llvm::StringRef ExportFuncName =
713 static_cast<llvm::MDString*>(ExportFuncNameMDS)->getString();
714 ExportSymbols.push_back(ExportFuncName.data());
715 }
716 }
717 }
718 }
719
Logan Chien7890d432011-08-03 14:55:17 +0800720 // TODO(logan): Remove this after we have finished the
721 // bccMarkExternalSymbol API.
722
Stephen Hines64160102011-09-01 17:30:26 -0700723 // root(), init(), and .rs.dtor() are born to be exported
Logan Chien4cc00332011-06-12 14:00:46 +0800724 ExportSymbols.push_back("root");
725 ExportSymbols.push_back("init");
Stephen Hines64160102011-09-01 17:30:26 -0700726 ExportSymbols.push_back(".rs.dtor");
Logan Chien4cc00332011-06-12 14:00:46 +0800727
Logan Chien7890d432011-08-03 14:55:17 +0800728 // User-defined exporting symbols
729 std::vector<char const *> const &UserDefinedExternalSymbols =
730 mpResult->getUserDefinedExternalSymbols();
731
732 std::copy(UserDefinedExternalSymbols.begin(),
733 UserDefinedExternalSymbols.end(),
734 std::back_inserter(ExportSymbols));
735
Logan Chien4cc00332011-06-12 14:00:46 +0800736 // We now create passes list performing LTO. These are copied from
737 // (including comments) llvm::createStandardLTOPasses().
738
739 // Internalize all other symbols not listed in ExportSymbols
740 LTOPasses.add(llvm::createInternalizePass(ExportSymbols));
741
742 // Propagate constants at call sites into the functions they call. This
743 // opens opportunities for globalopt (and inlining) by substituting
744 // function pointers passed as arguments to direct uses of functions.
745 LTOPasses.add(llvm::createIPSCCPPass());
746
747 // Now that we internalized some globals, see if we can hack on them!
748 LTOPasses.add(llvm::createGlobalOptimizerPass());
749
750 // Linking modules together can lead to duplicated global constants, only
751 // keep one copy of each constant...
752 LTOPasses.add(llvm::createConstantMergePass());
753
754 // Remove unused arguments from functions...
755 LTOPasses.add(llvm::createDeadArgEliminationPass());
756
757 // Reduce the code after globalopt and ipsccp. Both can open up
758 // significant simplification opportunities, and both can propagate
759 // functions through function pointers. When this happens, we often have
760 // to resolve varargs calls, etc, so let instcombine do this.
761 LTOPasses.add(llvm::createInstructionCombiningPass());
762
763 // Inline small functions
764 LTOPasses.add(llvm::createFunctionInliningPass());
765
766 // Remove dead EH info.
767 LTOPasses.add(llvm::createPruneEHPass());
768
769 // Internalize the globals again after inlining
770 LTOPasses.add(llvm::createGlobalOptimizerPass());
771
772 // Remove dead functions.
773 LTOPasses.add(llvm::createGlobalDCEPass());
774
775 // If we didn't decide to inline a function, check to see if we can
776 // transform it to pass arguments by value instead of by reference.
777 LTOPasses.add(llvm::createArgumentPromotionPass());
778
779 // The IPO passes may leave cruft around. Clean up after them.
780 LTOPasses.add(llvm::createInstructionCombiningPass());
781 LTOPasses.add(llvm::createJumpThreadingPass());
782
783 // Break up allocas
784 LTOPasses.add(llvm::createScalarReplAggregatesPass());
785
786 // Run a few AA driven optimizations here and now, to cleanup the code.
787 LTOPasses.add(llvm::createFunctionAttrsPass()); // Add nocapture.
788 LTOPasses.add(llvm::createGlobalsModRefPass()); // IP alias analysis.
789
790 // Hoist loop invariants.
791 LTOPasses.add(llvm::createLICMPass());
792
793 // Remove redundancies.
794 LTOPasses.add(llvm::createGVNPass());
795
796 // Remove dead memcpys.
797 LTOPasses.add(llvm::createMemCpyOptPass());
798
799 // Nuke dead stores.
800 LTOPasses.add(llvm::createDeadStoreEliminationPass());
801
802 // Cleanup and simplify the code after the scalar optimizations.
803 LTOPasses.add(llvm::createInstructionCombiningPass());
804
805 LTOPasses.add(llvm::createJumpThreadingPass());
806
807 // Delete basic blocks, which optimization passes may have killed.
808 LTOPasses.add(llvm::createCFGSimplificationPass());
809
810 // Now that we have optimized the program, discard unreachable functions.
811 LTOPasses.add(llvm::createGlobalDCEPass());
812
813 LTOPasses.run(*mModule);
Logan Chienda5e0c32011-06-13 03:47:21 +0800814
815 return 0;
Logan Chien4cc00332011-06-12 14:00:46 +0800816}
817
818
Logan Chienda5e0c32011-06-13 03:47:21 +0800819#if USE_MCJIT
Logan Chienda5e0c32011-06-13 03:47:21 +0800820void *Compiler::getSymbolAddress(char const *name) {
821 return rsloaderGetSymbolAddress(mRSExecutable, name);
822}
823#endif
824
825
826#if USE_MCJIT
827void *Compiler::resolveSymbolAdapter(void *context, char const *name) {
828 Compiler *self = reinterpret_cast<Compiler *>(context);
829
830 if (void *Addr = FindRuntimeFunction(name)) {
831 return Addr;
832 }
833
834 if (self->mpSymbolLookupFn) {
835 if (void *Addr = self->mpSymbolLookupFn(self->mpSymbolLookupContext, name)) {
836 return Addr;
837 }
838 }
839
840 LOGE("Unable to resolve symbol: %s\n", name);
841 return NULL;
842}
843#endif
844
845
Logan1f028c02010-11-27 01:02:48 +0800846Compiler::~Compiler() {
Logan1f028c02010-11-27 01:02:48 +0800847 delete mModule;
Logan1f028c02010-11-27 01:02:48 +0800848 delete mContext;
Logana4994f52010-11-27 14:06:02 +0800849
Logan Chienda5e0c32011-06-13 03:47:21 +0800850#if USE_MCJIT
851 rsloaderDisposeExec(mRSExecutable);
852#endif
853
Logana4994f52010-11-27 14:06:02 +0800854 // llvm::llvm_shutdown();
Logan1f028c02010-11-27 01:02:48 +0800855}
856
Shih-wei Liao90cd3d12011-06-20 15:43:34 -0700857
Logan1f028c02010-11-27 01:02:48 +0800858} // namespace bcc