blob: f4560667a7a8d87fcb3735e9467d19b736336a98 [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 Chien21392f02011-11-26 20:32:01 +0800179#if defined(PROVIDE_MIPS_CODEGEN)
180 LLVMInitializeMipsAsmPrinter();
181 LLVMInitializeMipsTargetMC();
182 LLVMInitializeMipsTargetInfo();
183 LLVMInitializeMipsTarget();
184#endif
185
Logan Chien3bb77072011-09-17 16:53:53 +0800186#if defined(PROVIDE_X86_CODEGEN)
Logan Chienb9ef9ab2011-07-20 11:11:06 +0800187 LLVMInitializeX86AsmPrinter();
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800188 LLVMInitializeX86TargetMC();
Logan1f028c02010-11-27 01:02:48 +0800189 LLVMInitializeX86TargetInfo();
190 LLVMInitializeX86Target();
Logan1f028c02010-11-27 01:02:48 +0800191#endif
Logan Chien9347e0b2011-07-07 19:51:47 +0800192
193#if USE_DISASSEMBLER
194 InitializeDisassembler();
Logan1f028c02010-11-27 01:02:48 +0800195#endif
196
197 // -O0: llvm::CodeGenOpt::None
198 // -O1: llvm::CodeGenOpt::Less
199 // -O2: llvm::CodeGenOpt::Default
200 // -O3: llvm::CodeGenOpt::Aggressive
Shih-wei Liao72f67a62010-12-14 13:36:15 -0800201 CodeGenOptLevel = llvm::CodeGenOpt::Aggressive;
Logan1f028c02010-11-27 01:02:48 +0800202
203 // Below are the global settings to LLVM
204
205 // Disable frame pointer elimination optimization
206 llvm::NoFramePointerElim = false;
207
208 // Use hardfloat ABI
209 //
210 // TODO(all): Need to detect the CPU capability and decide whether to use
211 // softfp. To use softfp, change following 2 lines to
212 //
213 // llvm::FloatABIType = llvm::FloatABI::Soft;
214 // llvm::UseSoftFloat = true;
215 //
Shih-wei Liaoe728cb82010-12-15 15:20:47 -0800216 llvm::FloatABIType = llvm::FloatABI::Soft;
Logan1f028c02010-11-27 01:02:48 +0800217 llvm::UseSoftFloat = false;
218
Logan1f028c02010-11-27 01:02:48 +0800219 // Register the scheduler
220 llvm::RegisterScheduler::setDefault(llvm::createDefaultScheduler);
221
222 // Register allocation policy:
223 // createFastRegisterAllocator: fast but bad quality
224 // createLinearScanRegisterAllocator: not so fast but good quality
225 llvm::RegisterRegAlloc::setDefault
226 ((CodeGenOptLevel == llvm::CodeGenOpt::None) ?
227 llvm::createFastRegisterAllocator :
Logan Chiene1bff142011-11-15 15:55:41 +0800228 llvm::createGreedyRegisterAllocator);
Logan1f028c02010-11-27 01:02:48 +0800229
Logan35849002011-01-15 07:30:43 +0800230#if USE_CACHE
Ying Wang26fea102011-07-05 15:12:25 -0700231 // Read in SHA1 checksum of libbcc and libRS.
232 readSHA1(sha1LibBCC_SHA1, sizeof(sha1LibBCC_SHA1), pathLibBCC_SHA1);
Joseph Wen76919072011-07-07 23:06:15 -0700233
234 calcFileSHA1(sha1LibRS, pathLibRS);
Logan35849002011-01-15 07:30:43 +0800235#endif
Logan75cc8a52011-01-07 06:06:52 +0800236
Logan1f028c02010-11-27 01:02:48 +0800237 GlobalInitialized = true;
238}
239
240
241void Compiler::LLVMErrorHandler(void *UserData, const std::string &Message) {
242 std::string *Error = static_cast<std::string*>(UserData);
243 Error->assign(Message);
244 LOGE("%s", Message.c_str());
245 exit(1);
246}
247
248
Logan Chienda5e0c32011-06-13 03:47:21 +0800249#if USE_OLD_JIT
Logan1f028c02010-11-27 01:02:48 +0800250CodeMemoryManager *Compiler::createCodeMemoryManager() {
251 mCodeMemMgr.reset(new CodeMemoryManager());
252 return mCodeMemMgr.get();
253}
Logan Chienda5e0c32011-06-13 03:47:21 +0800254#endif
Logan1f028c02010-11-27 01:02:48 +0800255
256
Logan Chienda5e0c32011-06-13 03:47:21 +0800257#if USE_OLD_JIT
Logan1f028c02010-11-27 01:02:48 +0800258CodeEmitter *Compiler::createCodeEmitter() {
Logan7dcaac92011-01-06 04:26:23 +0800259 mCodeEmitter.reset(new CodeEmitter(mpResult, mCodeMemMgr.get()));
Logan1f028c02010-11-27 01:02:48 +0800260 return mCodeEmitter.get();
261}
Logan Chienda5e0c32011-06-13 03:47:21 +0800262#endif
Logan1f028c02010-11-27 01:02:48 +0800263
264
Logan2a6dc822011-01-06 04:05:20 +0800265Compiler::Compiler(ScriptCompiled *result)
266 : mpResult(result),
Logan Chienda5e0c32011-06-13 03:47:21 +0800267#if USE_MCJIT
268 mRSExecutable(NULL),
269#endif
Logan1f028c02010-11-27 01:02:48 +0800270 mpSymbolLookupFn(NULL),
271 mpSymbolLookupContext(NULL),
272 mContext(NULL),
273 mModule(NULL),
274 mHasLinked(false) /* Turn off linker */ {
275 llvm::remove_fatal_error_handler();
276 llvm::install_fatal_error_handler(LLVMErrorHandler, &mError);
277 mContext = new llvm::LLVMContext();
278 return;
279}
280
Logan1f028c02010-11-27 01:02:48 +0800281
Logan474cbd22011-01-31 01:47:44 +0800282llvm::Module *Compiler::parseBitcodeFile(llvm::MemoryBuffer *MEM) {
283 llvm::Module *result = llvm::ParseBitcodeFile(MEM, *mContext, &mError);
Logan1f028c02010-11-27 01:02:48 +0800284
Logan474cbd22011-01-31 01:47:44 +0800285 if (!result) {
286 LOGE("Unable to ParseBitcodeFile: %s\n", mError.c_str());
287 return NULL;
Logan1f028c02010-11-27 01:02:48 +0800288 }
289
Logan474cbd22011-01-31 01:47:44 +0800290 return result;
Logan1f028c02010-11-27 01:02:48 +0800291}
292
293
Logan474cbd22011-01-31 01:47:44 +0800294int Compiler::linkModule(llvm::Module *moduleWith) {
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800295 if (llvm::Linker::LinkModules(mModule, moduleWith,
296 llvm::Linker::DestroySource,
297 &mError) != 0) {
Logan1f028c02010-11-27 01:02:48 +0800298 return hasError();
299 }
300
Logan1f028c02010-11-27 01:02:48 +0800301 // Everything for linking should be settled down here with no error occurs
302 mHasLinked = true;
303 return hasError();
304}
305
306
Joseph Wen34c600a2011-07-25 17:59:17 -0700307int Compiler::compile(bool compileOnly) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800308 llvm::Target const *Target = NULL;
Logan1f028c02010-11-27 01:02:48 +0800309 llvm::TargetData *TD = NULL;
Logan1f028c02010-11-27 01:02:48 +0800310 llvm::TargetMachine *TM = NULL;
Logan Chienda5e0c32011-06-13 03:47:21 +0800311
Logan1f028c02010-11-27 01:02:48 +0800312 std::string FeaturesStr;
313
Logan Chienda5e0c32011-06-13 03:47:21 +0800314 llvm::NamedMDNode const *PragmaMetadata;
315 llvm::NamedMDNode const *ExportVarMetadata;
316 llvm::NamedMDNode const *ExportFuncMetadata;
317 llvm::NamedMDNode const *ObjectSlotMetadata;
Logan1f028c02010-11-27 01:02:48 +0800318
Logan1f028c02010-11-27 01:02:48 +0800319 if (mModule == NULL) // No module was loaded
320 return 0;
321
322 // Create TargetMachine
323 Target = llvm::TargetRegistry::lookupTarget(Triple, mError);
324 if (hasError())
325 goto on_bcc_compile_error;
326
327 if (!CPU.empty() || !Features.empty()) {
328 llvm::SubtargetFeatures F;
Logana4994f52010-11-27 14:06:02 +0800329
330 for (std::vector<std::string>::const_iterator
331 I = Features.begin(), E = Features.end(); I != E; I++) {
Logan1f028c02010-11-27 01:02:48 +0800332 F.AddFeature(*I);
Logana4994f52010-11-27 14:06:02 +0800333 }
334
Logan1f028c02010-11-27 01:02:48 +0800335 FeaturesStr = F.getString();
336 }
337
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800338#if defined(DEFAULT_X86_64_CODEGEN)
339 // Data address in X86_64 architecture may reside in a far-away place
Logan Chiend3fe15c2011-07-20 22:08:39 +0800340 TM = Target->createTargetMachine(Triple, CPU, FeaturesStr,
Logan Chienbc9eb8f2011-10-21 15:17:45 +0800341 llvm::Reloc::Static,
342 llvm::CodeModel::Medium);
343#else
344 // This is set for the linker (specify how large of the virtual addresses
345 // we can access for all unknown symbols.)
346 TM = Target->createTargetMachine(Triple, CPU, FeaturesStr,
347 llvm::Reloc::Static,
348 llvm::CodeModel::Small);
349#endif
Logan1f028c02010-11-27 01:02:48 +0800350 if (TM == NULL) {
351 setError("Failed to create target machine implementation for the"
352 " specified triple '" + Triple + "'");
353 goto on_bcc_compile_error;
354 }
355
Logan Chienda5e0c32011-06-13 03:47:21 +0800356 // Get target data from Module
357 TD = new llvm::TargetData(mModule);
358
359 // Load named metadata
360 ExportVarMetadata = mModule->getNamedMetadata(ExportVarMetadataName);
361 ExportFuncMetadata = mModule->getNamedMetadata(ExportFuncMetadataName);
362 PragmaMetadata = mModule->getNamedMetadata(PragmaMetadataName);
363 ObjectSlotMetadata = mModule->getNamedMetadata(ObjectSlotMetadataName);
364
365 // Perform link-time optimization if we have multiple modules
366 if (mHasLinked) {
367 runLTO(new llvm::TargetData(*TD), ExportVarMetadata, ExportFuncMetadata);
368 }
369
370 // Perform code generation
371#if USE_OLD_JIT
372 if (runCodeGen(new llvm::TargetData(*TD), TM,
373 ExportVarMetadata, ExportFuncMetadata) != 0) {
374 goto on_bcc_compile_error;
375 }
376#endif
377
378#if USE_MCJIT
Joseph Wen34c600a2011-07-25 17:59:17 -0700379 if (runMCCodeGen(new llvm::TargetData(*TD), TM) != 0) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800380 goto on_bcc_compile_error;
381 }
Logan Chien9347e0b2011-07-07 19:51:47 +0800382
Joseph Wen34c600a2011-07-25 17:59:17 -0700383 if (compileOnly)
384 return 0;
385
386 // Load the ELF Object
387 mRSExecutable =
388 rsloaderCreateExec((unsigned char *)&*mEmittedELFExecutable.begin(),
389 mEmittedELFExecutable.size(),
390 &resolveSymbolAdapter, this);
391
392 if (!mRSExecutable) {
393 setError("Fail to load emitted ELF relocatable file");
394 goto on_bcc_compile_error;
395 }
396
397 if (ExportVarMetadata) {
398 ScriptCompiled::ExportVarList &varList = mpResult->mExportVars;
399 std::vector<std::string> &varNameList = mpResult->mExportVarsName;
400
401 for (int i = 0, e = ExportVarMetadata->getNumOperands(); i != e; i++) {
402 llvm::MDNode *ExportVar = ExportVarMetadata->getOperand(i);
403 if (ExportVar != NULL && ExportVar->getNumOperands() > 1) {
404 llvm::Value *ExportVarNameMDS = ExportVar->getOperand(0);
405 if (ExportVarNameMDS->getValueID() == llvm::Value::MDStringVal) {
406 llvm::StringRef ExportVarName =
407 static_cast<llvm::MDString*>(ExportVarNameMDS)->getString();
408
409 varList.push_back(
410 rsloaderGetSymbolAddress(mRSExecutable,
411 ExportVarName.str().c_str()));
412 varNameList.push_back(ExportVarName.str());
413#if DEBUG_MCJIT_REFLECT
414 LOGD("runMCCodeGen(): Exported Var: %s @ %p\n", ExportVarName.str().c_str(),
415 varList.back());
416#endif
417 continue;
418 }
419 }
420
421 varList.push_back(NULL);
422 }
423 }
424
425 if (ExportFuncMetadata) {
426 ScriptCompiled::ExportFuncList &funcList = mpResult->mExportFuncs;
427 std::vector<std::string> &funcNameList = mpResult->mExportFuncsName;
428
429 for (int i = 0, e = ExportFuncMetadata->getNumOperands(); i != e; i++) {
430 llvm::MDNode *ExportFunc = ExportFuncMetadata->getOperand(i);
431 if (ExportFunc != NULL && ExportFunc->getNumOperands() > 0) {
432 llvm::Value *ExportFuncNameMDS = ExportFunc->getOperand(0);
433 if (ExportFuncNameMDS->getValueID() == llvm::Value::MDStringVal) {
434 llvm::StringRef ExportFuncName =
435 static_cast<llvm::MDString*>(ExportFuncNameMDS)->getString();
436
437 funcList.push_back(
438 rsloaderGetSymbolAddress(mRSExecutable,
439 ExportFuncName.str().c_str()));
440 funcNameList.push_back(ExportFuncName.str());
441#if DEBUG_MCJIT_RELECT
442 LOGD("runMCCodeGen(): Exported Func: %s @ %p\n", ExportFuncName.str().c_str(),
443 funcList.back());
444#endif
445 }
446 }
447 }
448 }
449
Logan Chien4885cf82011-07-20 10:18:05 +0800450#if DEBUG_MCJIT_DISASSEMBLER
Shih-wei Liao90cd3d12011-06-20 15:43:34 -0700451 {
Shih-wei Liaod3c551f2011-07-01 04:28:27 -0700452 // Get MC codegen emitted function name list
453 size_t func_list_size = rsloaderGetFuncCount(mRSExecutable);
454 std::vector<char const *> func_list(func_list_size, NULL);
455 rsloaderGetFuncNameList(mRSExecutable, func_list_size, &*func_list.begin());
Shih-wei Liao320b5492011-06-20 22:53:33 -0700456
Shih-wei Liaod3c551f2011-07-01 04:28:27 -0700457 // Disassemble each function
Shih-wei Liao90cd3d12011-06-20 15:43:34 -0700458 for (size_t i = 0; i < func_list_size; ++i) {
459 void *func = rsloaderGetSymbolAddress(mRSExecutable, func_list[i]);
460 if (func) {
461 size_t size = rsloaderGetSymbolSize(mRSExecutable, func_list[i]);
Logan Chien9347e0b2011-07-07 19:51:47 +0800462 Disassemble(DEBUG_MCJIT_DISASSEMBLER_FILE,
463 Target, TM, func_list[i], (unsigned char const *)func, size);
Shih-wei Liao90cd3d12011-06-20 15:43:34 -0700464 }
465 }
466 }
467#endif
Logan Chienda5e0c32011-06-13 03:47:21 +0800468#endif
469
470 // Read pragma information from the metadata node of the module.
471 if (PragmaMetadata) {
472 ScriptCompiled::PragmaList &pragmaList = mpResult->mPragmas;
473
474 for (int i = 0, e = PragmaMetadata->getNumOperands(); i != e; i++) {
475 llvm::MDNode *Pragma = PragmaMetadata->getOperand(i);
476 if (Pragma != NULL &&
477 Pragma->getNumOperands() == 2 /* should have exactly 2 operands */) {
478 llvm::Value *PragmaNameMDS = Pragma->getOperand(0);
479 llvm::Value *PragmaValueMDS = Pragma->getOperand(1);
480
481 if ((PragmaNameMDS->getValueID() == llvm::Value::MDStringVal) &&
482 (PragmaValueMDS->getValueID() == llvm::Value::MDStringVal)) {
483 llvm::StringRef PragmaName =
484 static_cast<llvm::MDString*>(PragmaNameMDS)->getString();
485 llvm::StringRef PragmaValue =
486 static_cast<llvm::MDString*>(PragmaValueMDS)->getString();
487
488 pragmaList.push_back(
489 std::make_pair(std::string(PragmaName.data(),
490 PragmaName.size()),
491 std::string(PragmaValue.data(),
492 PragmaValue.size())));
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700493#if DEBUG_BCC_REFLECT
494 LOGD("compile(): Pragma: %s -> %s\n",
495 pragmaList.back().first.c_str(),
496 pragmaList.back().second.c_str());
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700497#endif
Logan Chienda5e0c32011-06-13 03:47:21 +0800498 }
499 }
500 }
Logan Chienda5e0c32011-06-13 03:47:21 +0800501 }
502
503 if (ObjectSlotMetadata) {
504 ScriptCompiled::ObjectSlotList &objectSlotList = mpResult->mObjectSlots;
505
506 for (int i = 0, e = ObjectSlotMetadata->getNumOperands(); i != e; i++) {
507 llvm::MDNode *ObjectSlot = ObjectSlotMetadata->getOperand(i);
508 if (ObjectSlot != NULL &&
509 ObjectSlot->getNumOperands() == 1) {
510 llvm::Value *SlotMDS = ObjectSlot->getOperand(0);
511 if (SlotMDS->getValueID() == llvm::Value::MDStringVal) {
512 llvm::StringRef Slot =
513 static_cast<llvm::MDString*>(SlotMDS)->getString();
514 uint32_t USlot = 0;
515 if (Slot.getAsInteger(10, USlot)) {
516 setError("Non-integer object slot value '" + Slot.str() + "'");
517 goto on_bcc_compile_error;
518 }
519 objectSlotList.push_back(USlot);
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700520#if DEBUG_BCC_REFLECT
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700521 LOGD("compile(): RefCount Slot: %s @ %u\n", Slot.str().c_str(), USlot);
522#endif
Logan Chienda5e0c32011-06-13 03:47:21 +0800523 }
524 }
525 }
Logan Chienda5e0c32011-06-13 03:47:21 +0800526 }
527
528on_bcc_compile_error:
529 // LOGE("on_bcc_compiler_error");
530 if (TD) {
531 delete TD;
532 }
533
534 if (TM) {
535 delete TM;
536 }
537
538 if (mError.empty()) {
539 return 0;
540 }
541
542 // LOGE(getErrorMessage());
543 return 1;
544}
545
546
547#if USE_OLD_JIT
548int Compiler::runCodeGen(llvm::TargetData *TD, llvm::TargetMachine *TM,
549 llvm::NamedMDNode const *ExportVarMetadata,
550 llvm::NamedMDNode const *ExportFuncMetadata) {
Logan1f028c02010-11-27 01:02:48 +0800551 // Create memory manager for creation of code emitter later.
552 if (!mCodeMemMgr.get() && !createCodeMemoryManager()) {
553 setError("Failed to startup memory management for further compilation");
Logan Chienda5e0c32011-06-13 03:47:21 +0800554 return 1;
Logan1f028c02010-11-27 01:02:48 +0800555 }
Logan02286cb2011-01-07 00:30:47 +0800556
557 mpResult->mContext = (char *) (mCodeMemMgr.get()->getCodeMemBase());
Logan1f028c02010-11-27 01:02:48 +0800558
559 // Create code emitter
560 if (!mCodeEmitter.get()) {
561 if (!createCodeEmitter()) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800562 setError("Failed to create machine code emitter for compilation");
563 return 1;
Logan1f028c02010-11-27 01:02:48 +0800564 }
565 } else {
566 // Reuse the code emitter
567 mCodeEmitter->reset();
568 }
569
570 mCodeEmitter->setTargetMachine(*TM);
571 mCodeEmitter->registerSymbolCallback(mpSymbolLookupFn,
572 mpSymbolLookupContext);
573
Logan1f028c02010-11-27 01:02:48 +0800574 // Create code-gen pass to run the code emitter
Logan Chienda5e0c32011-06-13 03:47:21 +0800575 llvm::OwningPtr<llvm::FunctionPassManager> CodeGenPasses(
576 new llvm::FunctionPassManager(mModule));
Logan1f028c02010-11-27 01:02:48 +0800577
Logan Chienda5e0c32011-06-13 03:47:21 +0800578 // Add TargetData to code generation pass manager
579 CodeGenPasses->add(TD);
580
581 // Add code emit passes
Logan1f028c02010-11-27 01:02:48 +0800582 if (TM->addPassesToEmitMachineCode(*CodeGenPasses,
583 *mCodeEmitter,
584 CodeGenOptLevel)) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800585 setError("The machine code emission is not supported on '" + Triple + "'");
586 return 1;
Logan1f028c02010-11-27 01:02:48 +0800587 }
588
Logan Chienda5e0c32011-06-13 03:47:21 +0800589 // Run the code emitter on every non-declaration function in the module
Logan1f028c02010-11-27 01:02:48 +0800590 CodeGenPasses->doInitialization();
Logan Chienda5e0c32011-06-13 03:47:21 +0800591 for (llvm::Module::iterator
592 I = mModule->begin(), E = mModule->end(); I != E; I++) {
Logan1f028c02010-11-27 01:02:48 +0800593 if (!I->isDeclaration()) {
594 CodeGenPasses->run(*I);
595 }
596 }
597
598 CodeGenPasses->doFinalization();
Logan Chienb0ceca22011-06-12 13:34:49 +0800599
Logan1f028c02010-11-27 01:02:48 +0800600 // Copy the global address mapping from code emitter and remapping
601 if (ExportVarMetadata) {
Logan2a6dc822011-01-06 04:05:20 +0800602 ScriptCompiled::ExportVarList &varList = mpResult->mExportVars;
603
Logan1f028c02010-11-27 01:02:48 +0800604 for (int i = 0, e = ExportVarMetadata->getNumOperands(); i != e; i++) {
605 llvm::MDNode *ExportVar = ExportVarMetadata->getOperand(i);
606 if (ExportVar != NULL && ExportVar->getNumOperands() > 1) {
607 llvm::Value *ExportVarNameMDS = ExportVar->getOperand(0);
608 if (ExportVarNameMDS->getValueID() == llvm::Value::MDStringVal) {
609 llvm::StringRef ExportVarName =
610 static_cast<llvm::MDString*>(ExportVarNameMDS)->getString();
611
612 CodeEmitter::global_addresses_const_iterator I, E;
613 for (I = mCodeEmitter->global_address_begin(),
614 E = mCodeEmitter->global_address_end();
615 I != E; I++) {
616 if (I->first->getValueID() != llvm::Value::GlobalVariableVal)
617 continue;
618 if (ExportVarName == I->first->getName()) {
Logan2a6dc822011-01-06 04:05:20 +0800619 varList.push_back(I->second);
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700620#if DEBUG_BCC_REFLECT
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700621 LOGD("runCodeGen(): Exported VAR: %s @ %p\n", ExportVarName.str().c_str(), I->second);
622#endif
Logan1f028c02010-11-27 01:02:48 +0800623 break;
624 }
625 }
626 if (I != mCodeEmitter->global_address_end())
627 continue; // found
Logan Chien7d1bf582011-06-13 23:22:40 +0800628
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700629#if DEBUG_BCC_REFLECT
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700630 LOGD("runCodeGen(): Exported VAR: %s @ %p\n",
Logan Chien7d1bf582011-06-13 23:22:40 +0800631 ExportVarName.str().c_str(), (void *)0);
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700632#endif
Logan1f028c02010-11-27 01:02:48 +0800633 }
634 }
635 // if reaching here, we know the global variable record in metadata is
636 // not found. So we make an empty slot
Logan2a6dc822011-01-06 04:05:20 +0800637 varList.push_back(NULL);
Logan1f028c02010-11-27 01:02:48 +0800638 }
Logan2a6dc822011-01-06 04:05:20 +0800639
Stephen Hinesbbcef8a2011-05-04 19:40:10 -0700640 bccAssert((varList.size() == ExportVarMetadata->getNumOperands()) &&
641 "Number of slots doesn't match the number of export variables!");
Logan1f028c02010-11-27 01:02:48 +0800642 }
643
644 if (ExportFuncMetadata) {
Logan2a6dc822011-01-06 04:05:20 +0800645 ScriptCompiled::ExportFuncList &funcList = mpResult->mExportFuncs;
646
Logan1f028c02010-11-27 01:02:48 +0800647 for (int i = 0, e = ExportFuncMetadata->getNumOperands(); i != e; i++) {
648 llvm::MDNode *ExportFunc = ExportFuncMetadata->getOperand(i);
649 if (ExportFunc != NULL && ExportFunc->getNumOperands() > 0) {
650 llvm::Value *ExportFuncNameMDS = ExportFunc->getOperand(0);
651 if (ExportFuncNameMDS->getValueID() == llvm::Value::MDStringVal) {
652 llvm::StringRef ExportFuncName =
653 static_cast<llvm::MDString*>(ExportFuncNameMDS)->getString();
Logan7dcaac92011-01-06 04:26:23 +0800654 funcList.push_back(mpResult->lookup(ExportFuncName.str().c_str()));
Shih-wei Liao9f73de02011-07-01 04:40:24 -0700655#if DEBUG_BCC_REFLECT
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700656 LOGD("runCodeGen(): Exported Func: %s @ %p\n", ExportFuncName.str().c_str(),
Logan Chien7d1bf582011-06-13 23:22:40 +0800657 funcList.back());
Shih-wei Liao749a51c2011-06-17 16:02:18 -0700658#endif
Logan1f028c02010-11-27 01:02:48 +0800659 }
660 }
661 }
662 }
663
664 // Tell code emitter now can release the memory using during the JIT since
665 // we have done the code emission
666 mCodeEmitter->releaseUnnecessary();
667
Logan Chienda5e0c32011-06-13 03:47:21 +0800668 return 0;
Logan1f028c02010-11-27 01:02:48 +0800669}
Logan Chienda5e0c32011-06-13 03:47:21 +0800670#endif // USE_OLD_JIT
Logan1f028c02010-11-27 01:02:48 +0800671
672
Logan Chienda5e0c32011-06-13 03:47:21 +0800673#if USE_MCJIT
Joseph Wen34c600a2011-07-25 17:59:17 -0700674int Compiler::runMCCodeGen(llvm::TargetData *TD, llvm::TargetMachine *TM) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800675 // Decorate mEmittedELFExecutable with formatted ostream
676 llvm::raw_svector_ostream OutSVOS(mEmittedELFExecutable);
677
678 // Relax all machine instructions
679 TM->setMCRelaxAll(/* RelaxAll= */ true);
680
681 // Create MC code generation pass manager
682 llvm::PassManager MCCodeGenPasses;
683
684 // Add TargetData to MC code generation pass manager
685 MCCodeGenPasses.add(TD);
686
687 // Add MC code generation passes to pass manager
688 llvm::MCContext *Ctx;
Logan Chien4e4485d2011-11-25 18:12:33 +0800689 if (TM->addPassesToEmitMC(MCCodeGenPasses, Ctx, OutSVOS, false)) {
Logan Chienda5e0c32011-06-13 03:47:21 +0800690 setError("Fail to add passes to emit file");
691 return 1;
692 }
693
694 MCCodeGenPasses.run(*mModule);
695 OutSVOS.flush();
Logan Chienda5e0c32011-06-13 03:47:21 +0800696 return 0;
697}
698#endif // USE_MCJIT
699
700
701int Compiler::runLTO(llvm::TargetData *TD,
702 llvm::NamedMDNode const *ExportVarMetadata,
703 llvm::NamedMDNode const *ExportFuncMetadata) {
Logan Chien4cc00332011-06-12 14:00:46 +0800704 llvm::PassManager LTOPasses;
705
706 // Add TargetData to LTO passes
707 LTOPasses.add(TD);
708
709 // Collect All Exported Symbols
710 std::vector<const char*> ExportSymbols;
711
712 // Note: This is a workaround for getting export variable and function name.
713 // We should refine it soon.
714 if (ExportVarMetadata) {
715 for (int i = 0, e = ExportVarMetadata->getNumOperands(); i != e; i++) {
716 llvm::MDNode *ExportVar = ExportVarMetadata->getOperand(i);
717 if (ExportVar != NULL && ExportVar->getNumOperands() > 1) {
718 llvm::Value *ExportVarNameMDS = ExportVar->getOperand(0);
719 if (ExportVarNameMDS->getValueID() == llvm::Value::MDStringVal) {
720 llvm::StringRef ExportVarName =
721 static_cast<llvm::MDString*>(ExportVarNameMDS)->getString();
722 ExportSymbols.push_back(ExportVarName.data());
723 }
724 }
725 }
726 }
727
728 if (ExportFuncMetadata) {
729 for (int i = 0, e = ExportFuncMetadata->getNumOperands(); i != e; i++) {
730 llvm::MDNode *ExportFunc = ExportFuncMetadata->getOperand(i);
731 if (ExportFunc != NULL && ExportFunc->getNumOperands() > 0) {
732 llvm::Value *ExportFuncNameMDS = ExportFunc->getOperand(0);
733 if (ExportFuncNameMDS->getValueID() == llvm::Value::MDStringVal) {
734 llvm::StringRef ExportFuncName =
735 static_cast<llvm::MDString*>(ExportFuncNameMDS)->getString();
736 ExportSymbols.push_back(ExportFuncName.data());
737 }
738 }
739 }
740 }
741
Logan Chien7890d432011-08-03 14:55:17 +0800742 // TODO(logan): Remove this after we have finished the
743 // bccMarkExternalSymbol API.
744
Stephen Hines64160102011-09-01 17:30:26 -0700745 // root(), init(), and .rs.dtor() are born to be exported
Logan Chien4cc00332011-06-12 14:00:46 +0800746 ExportSymbols.push_back("root");
747 ExportSymbols.push_back("init");
Stephen Hines64160102011-09-01 17:30:26 -0700748 ExportSymbols.push_back(".rs.dtor");
Logan Chien4cc00332011-06-12 14:00:46 +0800749
Logan Chien7890d432011-08-03 14:55:17 +0800750 // User-defined exporting symbols
751 std::vector<char const *> const &UserDefinedExternalSymbols =
752 mpResult->getUserDefinedExternalSymbols();
753
754 std::copy(UserDefinedExternalSymbols.begin(),
755 UserDefinedExternalSymbols.end(),
756 std::back_inserter(ExportSymbols));
757
Logan Chien4cc00332011-06-12 14:00:46 +0800758 // We now create passes list performing LTO. These are copied from
759 // (including comments) llvm::createStandardLTOPasses().
760
761 // Internalize all other symbols not listed in ExportSymbols
762 LTOPasses.add(llvm::createInternalizePass(ExportSymbols));
763
764 // Propagate constants at call sites into the functions they call. This
765 // opens opportunities for globalopt (and inlining) by substituting
766 // function pointers passed as arguments to direct uses of functions.
767 LTOPasses.add(llvm::createIPSCCPPass());
768
769 // Now that we internalized some globals, see if we can hack on them!
770 LTOPasses.add(llvm::createGlobalOptimizerPass());
771
772 // Linking modules together can lead to duplicated global constants, only
773 // keep one copy of each constant...
774 LTOPasses.add(llvm::createConstantMergePass());
775
776 // Remove unused arguments from functions...
777 LTOPasses.add(llvm::createDeadArgEliminationPass());
778
779 // Reduce the code after globalopt and ipsccp. Both can open up
780 // significant simplification opportunities, and both can propagate
781 // functions through function pointers. When this happens, we often have
782 // to resolve varargs calls, etc, so let instcombine do this.
783 LTOPasses.add(llvm::createInstructionCombiningPass());
784
785 // Inline small functions
786 LTOPasses.add(llvm::createFunctionInliningPass());
787
788 // Remove dead EH info.
789 LTOPasses.add(llvm::createPruneEHPass());
790
791 // Internalize the globals again after inlining
792 LTOPasses.add(llvm::createGlobalOptimizerPass());
793
794 // Remove dead functions.
795 LTOPasses.add(llvm::createGlobalDCEPass());
796
797 // If we didn't decide to inline a function, check to see if we can
798 // transform it to pass arguments by value instead of by reference.
799 LTOPasses.add(llvm::createArgumentPromotionPass());
800
801 // The IPO passes may leave cruft around. Clean up after them.
802 LTOPasses.add(llvm::createInstructionCombiningPass());
803 LTOPasses.add(llvm::createJumpThreadingPass());
804
805 // Break up allocas
806 LTOPasses.add(llvm::createScalarReplAggregatesPass());
807
808 // Run a few AA driven optimizations here and now, to cleanup the code.
809 LTOPasses.add(llvm::createFunctionAttrsPass()); // Add nocapture.
810 LTOPasses.add(llvm::createGlobalsModRefPass()); // IP alias analysis.
811
812 // Hoist loop invariants.
813 LTOPasses.add(llvm::createLICMPass());
814
815 // Remove redundancies.
816 LTOPasses.add(llvm::createGVNPass());
817
818 // Remove dead memcpys.
819 LTOPasses.add(llvm::createMemCpyOptPass());
820
821 // Nuke dead stores.
822 LTOPasses.add(llvm::createDeadStoreEliminationPass());
823
824 // Cleanup and simplify the code after the scalar optimizations.
825 LTOPasses.add(llvm::createInstructionCombiningPass());
826
827 LTOPasses.add(llvm::createJumpThreadingPass());
828
829 // Delete basic blocks, which optimization passes may have killed.
830 LTOPasses.add(llvm::createCFGSimplificationPass());
831
832 // Now that we have optimized the program, discard unreachable functions.
833 LTOPasses.add(llvm::createGlobalDCEPass());
834
835 LTOPasses.run(*mModule);
Logan Chienda5e0c32011-06-13 03:47:21 +0800836
837 return 0;
Logan Chien4cc00332011-06-12 14:00:46 +0800838}
839
840
Logan Chienda5e0c32011-06-13 03:47:21 +0800841#if USE_MCJIT
Logan Chienda5e0c32011-06-13 03:47:21 +0800842void *Compiler::getSymbolAddress(char const *name) {
843 return rsloaderGetSymbolAddress(mRSExecutable, name);
844}
845#endif
846
847
848#if USE_MCJIT
849void *Compiler::resolveSymbolAdapter(void *context, char const *name) {
850 Compiler *self = reinterpret_cast<Compiler *>(context);
851
852 if (void *Addr = FindRuntimeFunction(name)) {
853 return Addr;
854 }
855
856 if (self->mpSymbolLookupFn) {
857 if (void *Addr = self->mpSymbolLookupFn(self->mpSymbolLookupContext, name)) {
858 return Addr;
859 }
860 }
861
862 LOGE("Unable to resolve symbol: %s\n", name);
863 return NULL;
864}
865#endif
866
867
Logan1f028c02010-11-27 01:02:48 +0800868Compiler::~Compiler() {
Logan1f028c02010-11-27 01:02:48 +0800869 delete mModule;
Logan1f028c02010-11-27 01:02:48 +0800870 delete mContext;
Logana4994f52010-11-27 14:06:02 +0800871
Logan Chienda5e0c32011-06-13 03:47:21 +0800872#if USE_MCJIT
873 rsloaderDisposeExec(mRSExecutable);
874#endif
875
Logana4994f52010-11-27 14:06:02 +0800876 // llvm::llvm_shutdown();
Logan1f028c02010-11-27 01:02:48 +0800877}
878
Shih-wei Liao90cd3d12011-06-20 15:43:34 -0700879
Logan1f028c02010-11-27 01:02:48 +0800880} // namespace bcc