blob: 435107902b434e36bf2234d67b745a97d411a403 [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
Andrew Hsieh998ec832011-11-21 02:36:11 -080094
95#if !defined(__HOST__)
96 #define TARGET_TRIPLE_STRING DEFAULT_TARGET_TRIPLE_STRING
97#else
98// In host TARGET_TRIPLE_STRING is a variable to allow cross-compilation.
99 #if defined(__cplusplus)
100 extern "C" {
101 #endif
102 char *TARGET_TRIPLE_STRING = (char*)DEFAULT_TARGET_TRIPLE_STRING;
103 #if defined(__cplusplus)
104 };
105 #endif
106#endif
107
Logan1f028c02010-11-27 01:02:48 +0800108// Code generation optimization level for the compiler
109llvm::CodeGenOpt::Level Compiler::CodeGenOptLevel;
110
111std::string Compiler::Triple;
112
113std::string Compiler::CPU;
114
115std::vector<std::string> Compiler::Features;
116
Stephen Hines071288a2011-01-27 14:38:26 -0800117// Name of metadata node where pragma info resides (should be synced with
Logan1f028c02010-11-27 01:02:48 +0800118// slang.cpp)
119const llvm::StringRef Compiler::PragmaMetadataName = "#pragma";
120
Stephen Hines071288a2011-01-27 14:38:26 -0800121// Name of metadata node where exported variable names reside (should be
Logan1f028c02010-11-27 01:02:48 +0800122// synced with slang_rs_metadata.h)
123const llvm::StringRef Compiler::ExportVarMetadataName = "#rs_export_var";
124
Stephen Hines071288a2011-01-27 14:38:26 -0800125// Name of metadata node where exported function names reside (should be
Logan1f028c02010-11-27 01:02:48 +0800126// synced with slang_rs_metadata.h)
127const llvm::StringRef Compiler::ExportFuncMetadataName = "#rs_export_func";
128
Stephen Hines071288a2011-01-27 14:38:26 -0800129// Name of metadata node where RS object slot info resides (should be
130// synced with slang_rs_metadata.h)
131const llvm::StringRef Compiler::ObjectSlotMetadataName = "#rs_object_slots";
Logan1f028c02010-11-27 01:02:48 +0800132
133//////////////////////////////////////////////////////////////////////////////
134// Compiler
135//////////////////////////////////////////////////////////////////////////////
136
137void Compiler::GlobalInitialization() {
Shih-wei Liao40bcd662011-10-22 17:51:01 -0700138 if (GlobalInitialized) {
Logan1f028c02010-11-27 01:02:48 +0800139 return;
Shih-wei Liao40bcd662011-10-22 17:51:01 -0700140 }
141
Logan1f028c02010-11-27 01:02:48 +0800142 // if (!llvm::llvm_is_multithreaded())
143 // llvm::llvm_start_multithreaded();
144
145 // Set Triple, CPU and Features here
146 Triple = TARGET_TRIPLE_STRING;
147
Logan Chien3bb77072011-09-17 16:53:53 +0800148#if defined(DEFAULT_ARM_CODEGEN)
149
Stephen Hinesa12d2f32011-09-07 15:30:06 -0700150#if defined(ARCH_ARM_HAVE_VFP)
151 Features.push_back("+vfp3");
152#if !defined(ARCH_ARM_HAVE_VFP_D32)
153 Features.push_back("+d16");
154#endif
155#endif
156
Shih-wei Liao40bcd662011-10-22 17:51:01 -0700157#if defined(ARCH_ARM_HAVE_NEON)
Joseph Wen51001b82011-06-23 18:56:45 -0700158 Features.push_back("+neon");
159 Features.push_back("+neonfp");
Stephen Hines7f079eb2011-11-09 15:19:03 -0800160#else
161 Features.push_back("-neon");
162 Features.push_back("-neonfp");
Shih-wei Liao40bcd662011-10-22 17:51:01 -0700163#endif
164
165#if defined(DISABLE_ARCH_ARM_HAVE_NEON)
Joseph Wen51001b82011-06-23 18:56:45 -0700166 Features.push_back("-neon");
Logan4fe966f2011-02-27 08:26:40 +0800167 Features.push_back("-neonfp");
Shih-wei Liao40bcd662011-10-22 17:51:01 -0700168#endif
169
Logan Chien3bb77072011-09-17 16:53:53 +0800170#endif // DEFAULT_ARM_CODEGEN
Joseph Wen51001b82011-06-23 18:56:45 -0700171
Logan Chien3bb77072011-09-17 16:53:53 +0800172#if defined(PROVIDE_ARM_CODEGEN)
Logan Chienb9ef9ab2011-07-20 11:11:06 +0800173 LLVMInitializeARMAsmPrinter();
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800174 LLVMInitializeARMTargetMC();
Logan1f028c02010-11-27 01:02:48 +0800175 LLVMInitializeARMTargetInfo();
176 LLVMInitializeARMTarget();
Logan1f028c02010-11-27 01:02:48 +0800177#endif
178
Logan Chien3bb77072011-09-17 16:53:53 +0800179#if defined(PROVIDE_X86_CODEGEN)
Logan Chienb9ef9ab2011-07-20 11:11:06 +0800180 LLVMInitializeX86AsmPrinter();
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800181 LLVMInitializeX86TargetMC();
Logan1f028c02010-11-27 01:02:48 +0800182 LLVMInitializeX86TargetInfo();
183 LLVMInitializeX86Target();
Logan1f028c02010-11-27 01:02:48 +0800184#endif
Logan Chien9347e0b2011-07-07 19:51:47 +0800185
186#if USE_DISASSEMBLER
187 InitializeDisassembler();
Logan1f028c02010-11-27 01:02:48 +0800188#endif
189
190 // -O0: llvm::CodeGenOpt::None
191 // -O1: llvm::CodeGenOpt::Less
192 // -O2: llvm::CodeGenOpt::Default
193 // -O3: llvm::CodeGenOpt::Aggressive
Shih-wei Liao72f67a62010-12-14 13:36:15 -0800194 CodeGenOptLevel = llvm::CodeGenOpt::Aggressive;
Logan1f028c02010-11-27 01:02:48 +0800195
196 // Below are the global settings to LLVM
197
198 // Disable frame pointer elimination optimization
199 llvm::NoFramePointerElim = false;
200
201 // Use hardfloat ABI
202 //
203 // TODO(all): Need to detect the CPU capability and decide whether to use
204 // softfp. To use softfp, change following 2 lines to
205 //
206 // llvm::FloatABIType = llvm::FloatABI::Soft;
207 // llvm::UseSoftFloat = true;
208 //
Shih-wei Liaoe728cb82010-12-15 15:20:47 -0800209 llvm::FloatABIType = llvm::FloatABI::Soft;
Logan1f028c02010-11-27 01:02:48 +0800210 llvm::UseSoftFloat = false;
211
Logan1f028c02010-11-27 01:02:48 +0800212 // Register the scheduler
213 llvm::RegisterScheduler::setDefault(llvm::createDefaultScheduler);
214
215 // Register allocation policy:
216 // createFastRegisterAllocator: fast but bad quality
217 // createLinearScanRegisterAllocator: not so fast but good quality
218 llvm::RegisterRegAlloc::setDefault
219 ((CodeGenOptLevel == llvm::CodeGenOpt::None) ?
220 llvm::createFastRegisterAllocator :
Logan Chiene1bff142011-11-15 15:55:41 +0800221 llvm::createGreedyRegisterAllocator);
Logan1f028c02010-11-27 01:02:48 +0800222
Logan35849002011-01-15 07:30:43 +0800223#if USE_CACHE
Ying Wang26fea102011-07-05 15:12:25 -0700224 // Read in SHA1 checksum of libbcc and libRS.
225 readSHA1(sha1LibBCC_SHA1, sizeof(sha1LibBCC_SHA1), pathLibBCC_SHA1);
Joseph Wen76919072011-07-07 23:06:15 -0700226
227 calcFileSHA1(sha1LibRS, pathLibRS);
Logan35849002011-01-15 07:30:43 +0800228#endif
Logan75cc8a52011-01-07 06:06:52 +0800229
Logan1f028c02010-11-27 01:02:48 +0800230 GlobalInitialized = true;
231}
232
233
234void Compiler::LLVMErrorHandler(void *UserData, const std::string &Message) {
235 std::string *Error = static_cast<std::string*>(UserData);
236 Error->assign(Message);
237 LOGE("%s", Message.c_str());
238 exit(1);
239}
240
241
Logan Chienda5e0c32011-06-13 03:47:21 +0800242#if USE_OLD_JIT
Logan1f028c02010-11-27 01:02:48 +0800243CodeMemoryManager *Compiler::createCodeMemoryManager() {
244 mCodeMemMgr.reset(new CodeMemoryManager());
245 return mCodeMemMgr.get();
246}
Logan Chienda5e0c32011-06-13 03:47:21 +0800247#endif
Logan1f028c02010-11-27 01:02:48 +0800248
249
Logan Chienda5e0c32011-06-13 03:47:21 +0800250#if USE_OLD_JIT
Logan1f028c02010-11-27 01:02:48 +0800251CodeEmitter *Compiler::createCodeEmitter() {
Logan7dcaac92011-01-06 04:26:23 +0800252 mCodeEmitter.reset(new CodeEmitter(mpResult, mCodeMemMgr.get()));
Logan1f028c02010-11-27 01:02:48 +0800253 return mCodeEmitter.get();
254}
Logan Chienda5e0c32011-06-13 03:47:21 +0800255#endif
Logan1f028c02010-11-27 01:02:48 +0800256
257
Logan2a6dc822011-01-06 04:05:20 +0800258Compiler::Compiler(ScriptCompiled *result)
259 : mpResult(result),
Logan Chienda5e0c32011-06-13 03:47:21 +0800260#if USE_MCJIT
261 mRSExecutable(NULL),
262#endif
Logan1f028c02010-11-27 01:02:48 +0800263 mpSymbolLookupFn(NULL),
264 mpSymbolLookupContext(NULL),
265 mContext(NULL),
266 mModule(NULL),
267 mHasLinked(false) /* Turn off linker */ {
268 llvm::remove_fatal_error_handler();
269 llvm::install_fatal_error_handler(LLVMErrorHandler, &mError);
270 mContext = new llvm::LLVMContext();
271 return;
272}
273
Logan1f028c02010-11-27 01:02:48 +0800274
Logan474cbd22011-01-31 01:47:44 +0800275llvm::Module *Compiler::parseBitcodeFile(llvm::MemoryBuffer *MEM) {
276 llvm::Module *result = llvm::ParseBitcodeFile(MEM, *mContext, &mError);
Logan1f028c02010-11-27 01:02:48 +0800277
Logan474cbd22011-01-31 01:47:44 +0800278 if (!result) {
279 LOGE("Unable to ParseBitcodeFile: %s\n", mError.c_str());
280 return NULL;
Logan1f028c02010-11-27 01:02:48 +0800281 }
282
Logan474cbd22011-01-31 01:47:44 +0800283 return result;
Logan1f028c02010-11-27 01:02:48 +0800284}
285
286
Logan474cbd22011-01-31 01:47:44 +0800287int Compiler::linkModule(llvm::Module *moduleWith) {
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800288 if (llvm::Linker::LinkModules(mModule, moduleWith,
289 llvm::Linker::DestroySource,
290 &mError) != 0) {
Logan1f028c02010-11-27 01:02:48 +0800291 return hasError();
292 }
293
Logan1f028c02010-11-27 01:02:48 +0800294 // Everything for linking should be settled down here with no error occurs
295 mHasLinked = true;
296 return hasError();
297}
298
299
Joseph Wen34c600a2011-07-25 17:59:17 -0700300int Compiler::compile(bool compileOnly) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800301 llvm::Target const *Target = NULL;
Logan1f028c02010-11-27 01:02:48 +0800302 llvm::TargetData *TD = NULL;
Logan1f028c02010-11-27 01:02:48 +0800303 llvm::TargetMachine *TM = NULL;
Logan Chienda5e0c32011-06-13 03:47:21 +0800304
Logan1f028c02010-11-27 01:02:48 +0800305 std::string FeaturesStr;
306
Logan Chienda5e0c32011-06-13 03:47:21 +0800307 llvm::NamedMDNode const *PragmaMetadata;
308 llvm::NamedMDNode const *ExportVarMetadata;
309 llvm::NamedMDNode const *ExportFuncMetadata;
310 llvm::NamedMDNode const *ObjectSlotMetadata;
Logan1f028c02010-11-27 01:02:48 +0800311
Logan1f028c02010-11-27 01:02:48 +0800312 if (mModule == NULL) // No module was loaded
313 return 0;
314
315 // Create TargetMachine
316 Target = llvm::TargetRegistry::lookupTarget(Triple, mError);
317 if (hasError())
318 goto on_bcc_compile_error;
319
320 if (!CPU.empty() || !Features.empty()) {
321 llvm::SubtargetFeatures F;
Logana4994f52010-11-27 14:06:02 +0800322
323 for (std::vector<std::string>::const_iterator
324 I = Features.begin(), E = Features.end(); I != E; I++) {
Logan1f028c02010-11-27 01:02:48 +0800325 F.AddFeature(*I);
Logana4994f52010-11-27 14:06:02 +0800326 }
327
Logan1f028c02010-11-27 01:02:48 +0800328 FeaturesStr = F.getString();
329 }
330
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800331#if defined(DEFAULT_X86_64_CODEGEN)
332 // Data address in X86_64 architecture may reside in a far-away place
Logan Chiend3fe15c2011-07-20 22:08:39 +0800333 TM = Target->createTargetMachine(Triple, CPU, FeaturesStr,
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800334 llvm::Reloc::Static,
335 llvm::CodeModel::Medium);
336#else
337 // This is set for the linker (specify how large of the virtual addresses
338 // we can access for all unknown symbols.)
339 TM = Target->createTargetMachine(Triple, CPU, FeaturesStr,
340 llvm::Reloc::Static,
341 llvm::CodeModel::Small);
342#endif
Logan1f028c02010-11-27 01:02:48 +0800343 if (TM == NULL) {
344 setError("Failed to create target machine implementation for the"
345 " specified triple '" + Triple + "'");
346 goto on_bcc_compile_error;
347 }
348
Logan Chienda5e0c32011-06-13 03:47:21 +0800349 // Get target data from Module
350 TD = new llvm::TargetData(mModule);
351
352 // Load named metadata
353 ExportVarMetadata = mModule->getNamedMetadata(ExportVarMetadataName);
354 ExportFuncMetadata = mModule->getNamedMetadata(ExportFuncMetadataName);
355 PragmaMetadata = mModule->getNamedMetadata(PragmaMetadataName);
356 ObjectSlotMetadata = mModule->getNamedMetadata(ObjectSlotMetadataName);
357
358 // Perform link-time optimization if we have multiple modules
359 if (mHasLinked) {
360 runLTO(new llvm::TargetData(*TD), ExportVarMetadata, ExportFuncMetadata);
361 }
362
363 // Perform code generation
364#if USE_OLD_JIT
365 if (runCodeGen(new llvm::TargetData(*TD), TM,
366 ExportVarMetadata, ExportFuncMetadata) != 0) {
367 goto on_bcc_compile_error;
368 }
369#endif
370
371#if USE_MCJIT
Joseph Wen34c600a2011-07-25 17:59:17 -0700372 if (runMCCodeGen(new llvm::TargetData(*TD), TM) != 0) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800373 goto on_bcc_compile_error;
374 }
Logan Chien9347e0b2011-07-07 19:51:47 +0800375
Joseph Wen34c600a2011-07-25 17:59:17 -0700376 if (compileOnly)
377 return 0;
378
379 // Load the ELF Object
380 mRSExecutable =
381 rsloaderCreateExec((unsigned char *)&*mEmittedELFExecutable.begin(),
382 mEmittedELFExecutable.size(),
383 &resolveSymbolAdapter, this);
384
385 if (!mRSExecutable) {
386 setError("Fail to load emitted ELF relocatable file");
387 goto on_bcc_compile_error;
388 }
389
390 if (ExportVarMetadata) {
391 ScriptCompiled::ExportVarList &varList = mpResult->mExportVars;
392 std::vector<std::string> &varNameList = mpResult->mExportVarsName;
393
394 for (int i = 0, e = ExportVarMetadata->getNumOperands(); i != e; i++) {
395 llvm::MDNode *ExportVar = ExportVarMetadata->getOperand(i);
396 if (ExportVar != NULL && ExportVar->getNumOperands() > 1) {
397 llvm::Value *ExportVarNameMDS = ExportVar->getOperand(0);
398 if (ExportVarNameMDS->getValueID() == llvm::Value::MDStringVal) {
399 llvm::StringRef ExportVarName =
400 static_cast<llvm::MDString*>(ExportVarNameMDS)->getString();
401
402 varList.push_back(
403 rsloaderGetSymbolAddress(mRSExecutable,
404 ExportVarName.str().c_str()));
405 varNameList.push_back(ExportVarName.str());
406#if DEBUG_MCJIT_REFLECT
407 LOGD("runMCCodeGen(): Exported Var: %s @ %p\n", ExportVarName.str().c_str(),
408 varList.back());
409#endif
410 continue;
411 }
412 }
413
414 varList.push_back(NULL);
415 }
416 }
417
418 if (ExportFuncMetadata) {
419 ScriptCompiled::ExportFuncList &funcList = mpResult->mExportFuncs;
420 std::vector<std::string> &funcNameList = mpResult->mExportFuncsName;
421
422 for (int i = 0, e = ExportFuncMetadata->getNumOperands(); i != e; i++) {
423 llvm::MDNode *ExportFunc = ExportFuncMetadata->getOperand(i);
424 if (ExportFunc != NULL && ExportFunc->getNumOperands() > 0) {
425 llvm::Value *ExportFuncNameMDS = ExportFunc->getOperand(0);
426 if (ExportFuncNameMDS->getValueID() == llvm::Value::MDStringVal) {
427 llvm::StringRef ExportFuncName =
428 static_cast<llvm::MDString*>(ExportFuncNameMDS)->getString();
429
430 funcList.push_back(
431 rsloaderGetSymbolAddress(mRSExecutable,
432 ExportFuncName.str().c_str()));
433 funcNameList.push_back(ExportFuncName.str());
434#if DEBUG_MCJIT_RELECT
435 LOGD("runMCCodeGen(): Exported Func: %s @ %p\n", ExportFuncName.str().c_str(),
436 funcList.back());
437#endif
438 }
439 }
440 }
441 }
442
Logan Chien4885cf82011-07-20 10:18:05 +0800443#if DEBUG_MCJIT_DISASSEMBLER
Shih-wei Liao90cd3d12011-06-20 15:43:34 -0700444 {
Shih-wei Liaod3c551f2011-07-01 04:28:27 -0700445 // Get MC codegen emitted function name list
446 size_t func_list_size = rsloaderGetFuncCount(mRSExecutable);
447 std::vector<char const *> func_list(func_list_size, NULL);
448 rsloaderGetFuncNameList(mRSExecutable, func_list_size, &*func_list.begin());
Shih-wei Liao320b5492011-06-20 22:53:33 -0700449
Shih-wei Liaod3c551f2011-07-01 04:28:27 -0700450 // Disassemble each function
Shih-wei Liao90cd3d12011-06-20 15:43:34 -0700451 for (size_t i = 0; i < func_list_size; ++i) {
452 void *func = rsloaderGetSymbolAddress(mRSExecutable, func_list[i]);
453 if (func) {
454 size_t size = rsloaderGetSymbolSize(mRSExecutable, func_list[i]);
Logan Chien9347e0b2011-07-07 19:51:47 +0800455 Disassemble(DEBUG_MCJIT_DISASSEMBLER_FILE,
456 Target, TM, func_list[i], (unsigned char const *)func, size);
Shih-wei Liao90cd3d12011-06-20 15:43:34 -0700457 }
458 }
459 }
460#endif
Logan Chienda5e0c32011-06-13 03:47:21 +0800461#endif
462
463 // Read pragma information from the metadata node of the module.
464 if (PragmaMetadata) {
465 ScriptCompiled::PragmaList &pragmaList = mpResult->mPragmas;
466
467 for (int i = 0, e = PragmaMetadata->getNumOperands(); i != e; i++) {
468 llvm::MDNode *Pragma = PragmaMetadata->getOperand(i);
469 if (Pragma != NULL &&
470 Pragma->getNumOperands() == 2 /* should have exactly 2 operands */) {
471 llvm::Value *PragmaNameMDS = Pragma->getOperand(0);
472 llvm::Value *PragmaValueMDS = Pragma->getOperand(1);
473
474 if ((PragmaNameMDS->getValueID() == llvm::Value::MDStringVal) &&
475 (PragmaValueMDS->getValueID() == llvm::Value::MDStringVal)) {
476 llvm::StringRef PragmaName =
477 static_cast<llvm::MDString*>(PragmaNameMDS)->getString();
478 llvm::StringRef PragmaValue =
479 static_cast<llvm::MDString*>(PragmaValueMDS)->getString();
480
481 pragmaList.push_back(
482 std::make_pair(std::string(PragmaName.data(),
483 PragmaName.size()),
484 std::string(PragmaValue.data(),
485 PragmaValue.size())));
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700486#if DEBUG_BCC_REFLECT
487 LOGD("compile(): Pragma: %s -> %s\n",
488 pragmaList.back().first.c_str(),
489 pragmaList.back().second.c_str());
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700490#endif
Logan Chienda5e0c32011-06-13 03:47:21 +0800491 }
492 }
493 }
Logan Chienda5e0c32011-06-13 03:47:21 +0800494 }
495
496 if (ObjectSlotMetadata) {
497 ScriptCompiled::ObjectSlotList &objectSlotList = mpResult->mObjectSlots;
498
499 for (int i = 0, e = ObjectSlotMetadata->getNumOperands(); i != e; i++) {
500 llvm::MDNode *ObjectSlot = ObjectSlotMetadata->getOperand(i);
501 if (ObjectSlot != NULL &&
502 ObjectSlot->getNumOperands() == 1) {
503 llvm::Value *SlotMDS = ObjectSlot->getOperand(0);
504 if (SlotMDS->getValueID() == llvm::Value::MDStringVal) {
505 llvm::StringRef Slot =
506 static_cast<llvm::MDString*>(SlotMDS)->getString();
507 uint32_t USlot = 0;
508 if (Slot.getAsInteger(10, USlot)) {
509 setError("Non-integer object slot value '" + Slot.str() + "'");
510 goto on_bcc_compile_error;
511 }
512 objectSlotList.push_back(USlot);
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700513#if DEBUG_BCC_REFLECT
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700514 LOGD("compile(): RefCount Slot: %s @ %u\n", Slot.str().c_str(), USlot);
515#endif
Logan Chienda5e0c32011-06-13 03:47:21 +0800516 }
517 }
518 }
Logan Chienda5e0c32011-06-13 03:47:21 +0800519 }
520
521on_bcc_compile_error:
522 // LOGE("on_bcc_compiler_error");
523 if (TD) {
524 delete TD;
525 }
526
527 if (TM) {
528 delete TM;
529 }
530
531 if (mError.empty()) {
532 return 0;
533 }
534
535 // LOGE(getErrorMessage());
536 return 1;
537}
538
539
540#if USE_OLD_JIT
541int Compiler::runCodeGen(llvm::TargetData *TD, llvm::TargetMachine *TM,
542 llvm::NamedMDNode const *ExportVarMetadata,
543 llvm::NamedMDNode const *ExportFuncMetadata) {
Logan1f028c02010-11-27 01:02:48 +0800544 // Create memory manager for creation of code emitter later.
545 if (!mCodeMemMgr.get() && !createCodeMemoryManager()) {
546 setError("Failed to startup memory management for further compilation");
Logan Chienda5e0c32011-06-13 03:47:21 +0800547 return 1;
Logan1f028c02010-11-27 01:02:48 +0800548 }
Logan02286cb2011-01-07 00:30:47 +0800549
550 mpResult->mContext = (char *) (mCodeMemMgr.get()->getCodeMemBase());
Logan1f028c02010-11-27 01:02:48 +0800551
552 // Create code emitter
553 if (!mCodeEmitter.get()) {
554 if (!createCodeEmitter()) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800555 setError("Failed to create machine code emitter for compilation");
556 return 1;
Logan1f028c02010-11-27 01:02:48 +0800557 }
558 } else {
559 // Reuse the code emitter
560 mCodeEmitter->reset();
561 }
562
563 mCodeEmitter->setTargetMachine(*TM);
564 mCodeEmitter->registerSymbolCallback(mpSymbolLookupFn,
565 mpSymbolLookupContext);
566
Logan1f028c02010-11-27 01:02:48 +0800567 // Create code-gen pass to run the code emitter
Logan Chienda5e0c32011-06-13 03:47:21 +0800568 llvm::OwningPtr<llvm::FunctionPassManager> CodeGenPasses(
569 new llvm::FunctionPassManager(mModule));
Logan1f028c02010-11-27 01:02:48 +0800570
Logan Chienda5e0c32011-06-13 03:47:21 +0800571 // Add TargetData to code generation pass manager
572 CodeGenPasses->add(TD);
573
574 // Add code emit passes
Logan1f028c02010-11-27 01:02:48 +0800575 if (TM->addPassesToEmitMachineCode(*CodeGenPasses,
576 *mCodeEmitter,
577 CodeGenOptLevel)) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800578 setError("The machine code emission is not supported on '" + Triple + "'");
579 return 1;
Logan1f028c02010-11-27 01:02:48 +0800580 }
581
Logan Chienda5e0c32011-06-13 03:47:21 +0800582 // Run the code emitter on every non-declaration function in the module
Logan1f028c02010-11-27 01:02:48 +0800583 CodeGenPasses->doInitialization();
Logan Chienda5e0c32011-06-13 03:47:21 +0800584 for (llvm::Module::iterator
585 I = mModule->begin(), E = mModule->end(); I != E; I++) {
Logan1f028c02010-11-27 01:02:48 +0800586 if (!I->isDeclaration()) {
587 CodeGenPasses->run(*I);
588 }
589 }
590
591 CodeGenPasses->doFinalization();
Logan Chienb0ceca22011-06-12 13:34:49 +0800592
Logan1f028c02010-11-27 01:02:48 +0800593 // Copy the global address mapping from code emitter and remapping
594 if (ExportVarMetadata) {
Logan2a6dc822011-01-06 04:05:20 +0800595 ScriptCompiled::ExportVarList &varList = mpResult->mExportVars;
596
Logan1f028c02010-11-27 01:02:48 +0800597 for (int i = 0, e = ExportVarMetadata->getNumOperands(); i != e; i++) {
598 llvm::MDNode *ExportVar = ExportVarMetadata->getOperand(i);
599 if (ExportVar != NULL && ExportVar->getNumOperands() > 1) {
600 llvm::Value *ExportVarNameMDS = ExportVar->getOperand(0);
601 if (ExportVarNameMDS->getValueID() == llvm::Value::MDStringVal) {
602 llvm::StringRef ExportVarName =
603 static_cast<llvm::MDString*>(ExportVarNameMDS)->getString();
604
605 CodeEmitter::global_addresses_const_iterator I, E;
606 for (I = mCodeEmitter->global_address_begin(),
607 E = mCodeEmitter->global_address_end();
608 I != E; I++) {
609 if (I->first->getValueID() != llvm::Value::GlobalVariableVal)
610 continue;
611 if (ExportVarName == I->first->getName()) {
Logan2a6dc822011-01-06 04:05:20 +0800612 varList.push_back(I->second);
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700613#if DEBUG_BCC_REFLECT
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700614 LOGD("runCodeGen(): Exported VAR: %s @ %p\n", ExportVarName.str().c_str(), I->second);
615#endif
Logan1f028c02010-11-27 01:02:48 +0800616 break;
617 }
618 }
619 if (I != mCodeEmitter->global_address_end())
620 continue; // found
Logan Chien7d1bf582011-06-13 23:22:40 +0800621
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700622#if DEBUG_BCC_REFLECT
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700623 LOGD("runCodeGen(): Exported VAR: %s @ %p\n",
Logan Chien7d1bf582011-06-13 23:22:40 +0800624 ExportVarName.str().c_str(), (void *)0);
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700625#endif
Logan1f028c02010-11-27 01:02:48 +0800626 }
627 }
628 // if reaching here, we know the global variable record in metadata is
629 // not found. So we make an empty slot
Logan2a6dc822011-01-06 04:05:20 +0800630 varList.push_back(NULL);
Logan1f028c02010-11-27 01:02:48 +0800631 }
Logan2a6dc822011-01-06 04:05:20 +0800632
Stephen Hinesbbcef8a2011-05-04 19:40:10 -0700633 bccAssert((varList.size() == ExportVarMetadata->getNumOperands()) &&
634 "Number of slots doesn't match the number of export variables!");
Logan1f028c02010-11-27 01:02:48 +0800635 }
636
637 if (ExportFuncMetadata) {
Logan2a6dc822011-01-06 04:05:20 +0800638 ScriptCompiled::ExportFuncList &funcList = mpResult->mExportFuncs;
639
Logan1f028c02010-11-27 01:02:48 +0800640 for (int i = 0, e = ExportFuncMetadata->getNumOperands(); i != e; i++) {
641 llvm::MDNode *ExportFunc = ExportFuncMetadata->getOperand(i);
642 if (ExportFunc != NULL && ExportFunc->getNumOperands() > 0) {
643 llvm::Value *ExportFuncNameMDS = ExportFunc->getOperand(0);
644 if (ExportFuncNameMDS->getValueID() == llvm::Value::MDStringVal) {
645 llvm::StringRef ExportFuncName =
646 static_cast<llvm::MDString*>(ExportFuncNameMDS)->getString();
Logan7dcaac92011-01-06 04:26:23 +0800647 funcList.push_back(mpResult->lookup(ExportFuncName.str().c_str()));
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700648#if DEBUG_BCC_REFLECT
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700649 LOGD("runCodeGen(): Exported Func: %s @ %p\n", ExportFuncName.str().c_str(),
Logan Chien7d1bf582011-06-13 23:22:40 +0800650 funcList.back());
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700651#endif
Logan1f028c02010-11-27 01:02:48 +0800652 }
653 }
654 }
655 }
656
657 // Tell code emitter now can release the memory using during the JIT since
658 // we have done the code emission
659 mCodeEmitter->releaseUnnecessary();
660
Logan Chienda5e0c32011-06-13 03:47:21 +0800661 return 0;
Logan1f028c02010-11-27 01:02:48 +0800662}
Logan Chienda5e0c32011-06-13 03:47:21 +0800663#endif // USE_OLD_JIT
Logan1f028c02010-11-27 01:02:48 +0800664
665
Logan Chienda5e0c32011-06-13 03:47:21 +0800666#if USE_MCJIT
Joseph Wen34c600a2011-07-25 17:59:17 -0700667int Compiler::runMCCodeGen(llvm::TargetData *TD, llvm::TargetMachine *TM) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800668 // Decorate mEmittedELFExecutable with formatted ostream
669 llvm::raw_svector_ostream OutSVOS(mEmittedELFExecutable);
670
671 // Relax all machine instructions
672 TM->setMCRelaxAll(/* RelaxAll= */ true);
673
674 // Create MC code generation pass manager
675 llvm::PassManager MCCodeGenPasses;
676
677 // Add TargetData to MC code generation pass manager
678 MCCodeGenPasses.add(TD);
679
680 // Add MC code generation passes to pass manager
681 llvm::MCContext *Ctx;
682 if (TM->addPassesToEmitMC(MCCodeGenPasses, Ctx, OutSVOS,
683 CodeGenOptLevel, false)) {
684 setError("Fail to add passes to emit file");
685 return 1;
686 }
687
688 MCCodeGenPasses.run(*mModule);
689 OutSVOS.flush();
Logan Chienda5e0c32011-06-13 03:47:21 +0800690 return 0;
691}
692#endif // USE_MCJIT
693
694
695int Compiler::runLTO(llvm::TargetData *TD,
696 llvm::NamedMDNode const *ExportVarMetadata,
697 llvm::NamedMDNode const *ExportFuncMetadata) {
Logan Chien4cc00332011-06-12 14:00:46 +0800698 llvm::PassManager LTOPasses;
699
700 // Add TargetData to LTO passes
701 LTOPasses.add(TD);
702
703 // Collect All Exported Symbols
704 std::vector<const char*> ExportSymbols;
705
706 // Note: This is a workaround for getting export variable and function name.
707 // We should refine it soon.
708 if (ExportVarMetadata) {
709 for (int i = 0, e = ExportVarMetadata->getNumOperands(); i != e; i++) {
710 llvm::MDNode *ExportVar = ExportVarMetadata->getOperand(i);
711 if (ExportVar != NULL && ExportVar->getNumOperands() > 1) {
712 llvm::Value *ExportVarNameMDS = ExportVar->getOperand(0);
713 if (ExportVarNameMDS->getValueID() == llvm::Value::MDStringVal) {
714 llvm::StringRef ExportVarName =
715 static_cast<llvm::MDString*>(ExportVarNameMDS)->getString();
716 ExportSymbols.push_back(ExportVarName.data());
717 }
718 }
719 }
720 }
721
722 if (ExportFuncMetadata) {
723 for (int i = 0, e = ExportFuncMetadata->getNumOperands(); i != e; i++) {
724 llvm::MDNode *ExportFunc = ExportFuncMetadata->getOperand(i);
725 if (ExportFunc != NULL && ExportFunc->getNumOperands() > 0) {
726 llvm::Value *ExportFuncNameMDS = ExportFunc->getOperand(0);
727 if (ExportFuncNameMDS->getValueID() == llvm::Value::MDStringVal) {
728 llvm::StringRef ExportFuncName =
729 static_cast<llvm::MDString*>(ExportFuncNameMDS)->getString();
730 ExportSymbols.push_back(ExportFuncName.data());
731 }
732 }
733 }
734 }
735
Logan Chien7890d432011-08-03 14:55:17 +0800736 // TODO(logan): Remove this after we have finished the
737 // bccMarkExternalSymbol API.
738
Stephen Hines64160102011-09-01 17:30:26 -0700739 // root(), init(), and .rs.dtor() are born to be exported
Logan Chien4cc00332011-06-12 14:00:46 +0800740 ExportSymbols.push_back("root");
741 ExportSymbols.push_back("init");
Stephen Hines64160102011-09-01 17:30:26 -0700742 ExportSymbols.push_back(".rs.dtor");
Logan Chien4cc00332011-06-12 14:00:46 +0800743
Logan Chien7890d432011-08-03 14:55:17 +0800744 // User-defined exporting symbols
745 std::vector<char const *> const &UserDefinedExternalSymbols =
746 mpResult->getUserDefinedExternalSymbols();
747
748 std::copy(UserDefinedExternalSymbols.begin(),
749 UserDefinedExternalSymbols.end(),
750 std::back_inserter(ExportSymbols));
751
Logan Chien4cc00332011-06-12 14:00:46 +0800752 // We now create passes list performing LTO. These are copied from
753 // (including comments) llvm::createStandardLTOPasses().
754
755 // Internalize all other symbols not listed in ExportSymbols
756 LTOPasses.add(llvm::createInternalizePass(ExportSymbols));
757
758 // Propagate constants at call sites into the functions they call. This
759 // opens opportunities for globalopt (and inlining) by substituting
760 // function pointers passed as arguments to direct uses of functions.
761 LTOPasses.add(llvm::createIPSCCPPass());
762
763 // Now that we internalized some globals, see if we can hack on them!
764 LTOPasses.add(llvm::createGlobalOptimizerPass());
765
766 // Linking modules together can lead to duplicated global constants, only
767 // keep one copy of each constant...
768 LTOPasses.add(llvm::createConstantMergePass());
769
770 // Remove unused arguments from functions...
771 LTOPasses.add(llvm::createDeadArgEliminationPass());
772
773 // Reduce the code after globalopt and ipsccp. Both can open up
774 // significant simplification opportunities, and both can propagate
775 // functions through function pointers. When this happens, we often have
776 // to resolve varargs calls, etc, so let instcombine do this.
777 LTOPasses.add(llvm::createInstructionCombiningPass());
778
779 // Inline small functions
780 LTOPasses.add(llvm::createFunctionInliningPass());
781
782 // Remove dead EH info.
783 LTOPasses.add(llvm::createPruneEHPass());
784
785 // Internalize the globals again after inlining
786 LTOPasses.add(llvm::createGlobalOptimizerPass());
787
788 // Remove dead functions.
789 LTOPasses.add(llvm::createGlobalDCEPass());
790
791 // If we didn't decide to inline a function, check to see if we can
792 // transform it to pass arguments by value instead of by reference.
793 LTOPasses.add(llvm::createArgumentPromotionPass());
794
795 // The IPO passes may leave cruft around. Clean up after them.
796 LTOPasses.add(llvm::createInstructionCombiningPass());
797 LTOPasses.add(llvm::createJumpThreadingPass());
798
799 // Break up allocas
800 LTOPasses.add(llvm::createScalarReplAggregatesPass());
801
802 // Run a few AA driven optimizations here and now, to cleanup the code.
803 LTOPasses.add(llvm::createFunctionAttrsPass()); // Add nocapture.
804 LTOPasses.add(llvm::createGlobalsModRefPass()); // IP alias analysis.
805
806 // Hoist loop invariants.
807 LTOPasses.add(llvm::createLICMPass());
808
809 // Remove redundancies.
810 LTOPasses.add(llvm::createGVNPass());
811
812 // Remove dead memcpys.
813 LTOPasses.add(llvm::createMemCpyOptPass());
814
815 // Nuke dead stores.
816 LTOPasses.add(llvm::createDeadStoreEliminationPass());
817
818 // Cleanup and simplify the code after the scalar optimizations.
819 LTOPasses.add(llvm::createInstructionCombiningPass());
820
821 LTOPasses.add(llvm::createJumpThreadingPass());
822
823 // Delete basic blocks, which optimization passes may have killed.
824 LTOPasses.add(llvm::createCFGSimplificationPass());
825
826 // Now that we have optimized the program, discard unreachable functions.
827 LTOPasses.add(llvm::createGlobalDCEPass());
828
829 LTOPasses.run(*mModule);
Logan Chienda5e0c32011-06-13 03:47:21 +0800830
831 return 0;
Logan Chien4cc00332011-06-12 14:00:46 +0800832}
833
834
Logan Chienda5e0c32011-06-13 03:47:21 +0800835#if USE_MCJIT
Logan Chienda5e0c32011-06-13 03:47:21 +0800836void *Compiler::getSymbolAddress(char const *name) {
837 return rsloaderGetSymbolAddress(mRSExecutable, name);
838}
839#endif
840
841
842#if USE_MCJIT
843void *Compiler::resolveSymbolAdapter(void *context, char const *name) {
844 Compiler *self = reinterpret_cast<Compiler *>(context);
845
846 if (void *Addr = FindRuntimeFunction(name)) {
847 return Addr;
848 }
849
850 if (self->mpSymbolLookupFn) {
851 if (void *Addr = self->mpSymbolLookupFn(self->mpSymbolLookupContext, name)) {
852 return Addr;
853 }
854 }
855
856 LOGE("Unable to resolve symbol: %s\n", name);
857 return NULL;
858}
859#endif
860
861
Logan1f028c02010-11-27 01:02:48 +0800862Compiler::~Compiler() {
Logan1f028c02010-11-27 01:02:48 +0800863 delete mModule;
Logan1f028c02010-11-27 01:02:48 +0800864 delete mContext;
Logana4994f52010-11-27 14:06:02 +0800865
Logan Chienda5e0c32011-06-13 03:47:21 +0800866#if USE_MCJIT
867 rsloaderDisposeExec(mRSExecutable);
868#endif
869
Logana4994f52010-11-27 14:06:02 +0800870 // llvm::llvm_shutdown();
Logan1f028c02010-11-27 01:02:48 +0800871}
872
Shih-wei Liao90cd3d12011-06-20 15:43:34 -0700873
Logan1f028c02010-11-27 01:02:48 +0800874} // namespace bcc