1 //===-- TargetMachine.cpp -------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file implements the LLVM-C part of TargetMachine.h
10 //
11 //===----------------------------------------------------------------------===//
12
13 #include "llvm-c/Core.h"
14 #include "llvm-c/Target.h"
15 #include "llvm-c/TargetMachine.h"
16 #include "llvm/Analysis/TargetTransformInfo.h"
17 #include "llvm/IR/DataLayout.h"
18 #include "llvm/IR/LegacyPassManager.h"
19 #include "llvm/IR/Module.h"
20 #include "llvm/MC/SubtargetFeature.h"
21 #include "llvm/Support/FileSystem.h"
22 #include "llvm/Support/FormattedStream.h"
23 #include "llvm/Support/Host.h"
24 #include "llvm/Support/TargetRegistry.h"
25 #include "llvm/Support/raw_ostream.h"
26 #include "llvm/Target/CodeGenCWrappers.h"
27 #include "llvm/Target/TargetMachine.h"
28 #include <cassert>
29 #include <cstdlib>
30 #include <cstring>
31
32 using namespace llvm;
33
unwrap(LLVMTargetMachineRef P)34 static TargetMachine *unwrap(LLVMTargetMachineRef P) {
35 return reinterpret_cast<TargetMachine *>(P);
36 }
unwrap(LLVMTargetRef P)37 static Target *unwrap(LLVMTargetRef P) {
38 return reinterpret_cast<Target*>(P);
39 }
wrap(const TargetMachine * P)40 static LLVMTargetMachineRef wrap(const TargetMachine *P) {
41 return reinterpret_cast<LLVMTargetMachineRef>(const_cast<TargetMachine *>(P));
42 }
wrap(const Target * P)43 static LLVMTargetRef wrap(const Target * P) {
44 return reinterpret_cast<LLVMTargetRef>(const_cast<Target*>(P));
45 }
46
LLVMGetFirstTarget()47 LLVMTargetRef LLVMGetFirstTarget() {
48 if (TargetRegistry::targets().begin() == TargetRegistry::targets().end()) {
49 return nullptr;
50 }
51
52 const Target *target = &*TargetRegistry::targets().begin();
53 return wrap(target);
54 }
LLVMGetNextTarget(LLVMTargetRef T)55 LLVMTargetRef LLVMGetNextTarget(LLVMTargetRef T) {
56 return wrap(unwrap(T)->getNext());
57 }
58
LLVMGetTargetFromName(const char * Name)59 LLVMTargetRef LLVMGetTargetFromName(const char *Name) {
60 StringRef NameRef = Name;
61 auto I = find_if(TargetRegistry::targets(),
62 [&](const Target &T) { return T.getName() == NameRef; });
63 return I != TargetRegistry::targets().end() ? wrap(&*I) : nullptr;
64 }
65
LLVMGetTargetFromTriple(const char * TripleStr,LLVMTargetRef * T,char ** ErrorMessage)66 LLVMBool LLVMGetTargetFromTriple(const char* TripleStr, LLVMTargetRef *T,
67 char **ErrorMessage) {
68 std::string Error;
69
70 *T = wrap(TargetRegistry::lookupTarget(TripleStr, Error));
71
72 if (!*T) {
73 if (ErrorMessage)
74 *ErrorMessage = strdup(Error.c_str());
75
76 return 1;
77 }
78
79 return 0;
80 }
81
LLVMGetTargetName(LLVMTargetRef T)82 const char * LLVMGetTargetName(LLVMTargetRef T) {
83 return unwrap(T)->getName();
84 }
85
LLVMGetTargetDescription(LLVMTargetRef T)86 const char * LLVMGetTargetDescription(LLVMTargetRef T) {
87 return unwrap(T)->getShortDescription();
88 }
89
LLVMTargetHasJIT(LLVMTargetRef T)90 LLVMBool LLVMTargetHasJIT(LLVMTargetRef T) {
91 return unwrap(T)->hasJIT();
92 }
93
LLVMTargetHasTargetMachine(LLVMTargetRef T)94 LLVMBool LLVMTargetHasTargetMachine(LLVMTargetRef T) {
95 return unwrap(T)->hasTargetMachine();
96 }
97
LLVMTargetHasAsmBackend(LLVMTargetRef T)98 LLVMBool LLVMTargetHasAsmBackend(LLVMTargetRef T) {
99 return unwrap(T)->hasMCAsmBackend();
100 }
101
LLVMCreateTargetMachine(LLVMTargetRef T,const char * Triple,const char * CPU,const char * Features,LLVMCodeGenOptLevel Level,LLVMRelocMode Reloc,LLVMCodeModel CodeModel)102 LLVMTargetMachineRef LLVMCreateTargetMachine(LLVMTargetRef T,
103 const char *Triple, const char *CPU, const char *Features,
104 LLVMCodeGenOptLevel Level, LLVMRelocMode Reloc,
105 LLVMCodeModel CodeModel) {
106 Optional<Reloc::Model> RM;
107 switch (Reloc){
108 case LLVMRelocStatic:
109 RM = Reloc::Static;
110 break;
111 case LLVMRelocPIC:
112 RM = Reloc::PIC_;
113 break;
114 case LLVMRelocDynamicNoPic:
115 RM = Reloc::DynamicNoPIC;
116 break;
117 case LLVMRelocROPI:
118 RM = Reloc::ROPI;
119 break;
120 case LLVMRelocRWPI:
121 RM = Reloc::RWPI;
122 break;
123 case LLVMRelocROPI_RWPI:
124 RM = Reloc::ROPI_RWPI;
125 break;
126 default:
127 break;
128 }
129
130 bool JIT;
131 Optional<CodeModel::Model> CM = unwrap(CodeModel, JIT);
132
133 CodeGenOpt::Level OL;
134 switch (Level) {
135 case LLVMCodeGenLevelNone:
136 OL = CodeGenOpt::None;
137 break;
138 case LLVMCodeGenLevelLess:
139 OL = CodeGenOpt::Less;
140 break;
141 case LLVMCodeGenLevelAggressive:
142 OL = CodeGenOpt::Aggressive;
143 break;
144 default:
145 OL = CodeGenOpt::Default;
146 break;
147 }
148
149 TargetOptions opt;
150 return wrap(unwrap(T)->createTargetMachine(Triple, CPU, Features, opt, RM, CM,
151 OL, JIT));
152 }
153
LLVMDisposeTargetMachine(LLVMTargetMachineRef T)154 void LLVMDisposeTargetMachine(LLVMTargetMachineRef T) { delete unwrap(T); }
155
LLVMGetTargetMachineTarget(LLVMTargetMachineRef T)156 LLVMTargetRef LLVMGetTargetMachineTarget(LLVMTargetMachineRef T) {
157 const Target* target = &(unwrap(T)->getTarget());
158 return wrap(target);
159 }
160
LLVMGetTargetMachineTriple(LLVMTargetMachineRef T)161 char* LLVMGetTargetMachineTriple(LLVMTargetMachineRef T) {
162 std::string StringRep = unwrap(T)->getTargetTriple().str();
163 return strdup(StringRep.c_str());
164 }
165
LLVMGetTargetMachineCPU(LLVMTargetMachineRef T)166 char* LLVMGetTargetMachineCPU(LLVMTargetMachineRef T) {
167 std::string StringRep = unwrap(T)->getTargetCPU();
168 return strdup(StringRep.c_str());
169 }
170
LLVMGetTargetMachineFeatureString(LLVMTargetMachineRef T)171 char* LLVMGetTargetMachineFeatureString(LLVMTargetMachineRef T) {
172 std::string StringRep = unwrap(T)->getTargetFeatureString();
173 return strdup(StringRep.c_str());
174 }
175
LLVMSetTargetMachineAsmVerbosity(LLVMTargetMachineRef T,LLVMBool VerboseAsm)176 void LLVMSetTargetMachineAsmVerbosity(LLVMTargetMachineRef T,
177 LLVMBool VerboseAsm) {
178 unwrap(T)->Options.MCOptions.AsmVerbose = VerboseAsm;
179 }
180
LLVMCreateTargetDataLayout(LLVMTargetMachineRef T)181 LLVMTargetDataRef LLVMCreateTargetDataLayout(LLVMTargetMachineRef T) {
182 return wrap(new DataLayout(unwrap(T)->createDataLayout()));
183 }
184
LLVMTargetMachineEmit(LLVMTargetMachineRef T,LLVMModuleRef M,raw_pwrite_stream & OS,LLVMCodeGenFileType codegen,char ** ErrorMessage)185 static LLVMBool LLVMTargetMachineEmit(LLVMTargetMachineRef T, LLVMModuleRef M,
186 raw_pwrite_stream &OS,
187 LLVMCodeGenFileType codegen,
188 char **ErrorMessage) {
189 TargetMachine* TM = unwrap(T);
190 Module* Mod = unwrap(M);
191
192 legacy::PassManager pass;
193
194 std::string error;
195
196 Mod->setDataLayout(TM->createDataLayout());
197
198 CodeGenFileType ft;
199 switch (codegen) {
200 case LLVMAssemblyFile:
201 ft = CGFT_AssemblyFile;
202 break;
203 default:
204 ft = CGFT_ObjectFile;
205 break;
206 }
207 if (TM->addPassesToEmitFile(pass, OS, nullptr, ft)) {
208 error = "TargetMachine can't emit a file of this type";
209 *ErrorMessage = strdup(error.c_str());
210 return true;
211 }
212
213 pass.run(*Mod);
214
215 OS.flush();
216 return false;
217 }
218
LLVMTargetMachineEmitToFile(LLVMTargetMachineRef T,LLVMModuleRef M,char * Filename,LLVMCodeGenFileType codegen,char ** ErrorMessage)219 LLVMBool LLVMTargetMachineEmitToFile(LLVMTargetMachineRef T, LLVMModuleRef M,
220 char* Filename, LLVMCodeGenFileType codegen, char** ErrorMessage) {
221 std::error_code EC;
222 raw_fd_ostream dest(Filename, EC, sys::fs::OF_None);
223 if (EC) {
224 *ErrorMessage = strdup(EC.message().c_str());
225 return true;
226 }
227 bool Result = LLVMTargetMachineEmit(T, M, dest, codegen, ErrorMessage);
228 dest.flush();
229 return Result;
230 }
231
LLVMTargetMachineEmitToMemoryBuffer(LLVMTargetMachineRef T,LLVMModuleRef M,LLVMCodeGenFileType codegen,char ** ErrorMessage,LLVMMemoryBufferRef * OutMemBuf)232 LLVMBool LLVMTargetMachineEmitToMemoryBuffer(LLVMTargetMachineRef T,
233 LLVMModuleRef M, LLVMCodeGenFileType codegen, char** ErrorMessage,
234 LLVMMemoryBufferRef *OutMemBuf) {
235 SmallString<0> CodeString;
236 raw_svector_ostream OStream(CodeString);
237 bool Result = LLVMTargetMachineEmit(T, M, OStream, codegen, ErrorMessage);
238
239 StringRef Data = OStream.str();
240 *OutMemBuf =
241 LLVMCreateMemoryBufferWithMemoryRangeCopy(Data.data(), Data.size(), "");
242 return Result;
243 }
244
LLVMGetDefaultTargetTriple(void)245 char *LLVMGetDefaultTargetTriple(void) {
246 return strdup(sys::getDefaultTargetTriple().c_str());
247 }
248
LLVMNormalizeTargetTriple(const char * triple)249 char *LLVMNormalizeTargetTriple(const char* triple) {
250 return strdup(Triple::normalize(StringRef(triple)).c_str());
251 }
252
LLVMGetHostCPUName(void)253 char *LLVMGetHostCPUName(void) {
254 return strdup(sys::getHostCPUName().data());
255 }
256
LLVMGetHostCPUFeatures(void)257 char *LLVMGetHostCPUFeatures(void) {
258 SubtargetFeatures Features;
259 StringMap<bool> HostFeatures;
260
261 if (sys::getHostCPUFeatures(HostFeatures))
262 for (auto &F : HostFeatures)
263 Features.AddFeature(F.first(), F.second);
264
265 return strdup(Features.getString().c_str());
266 }
267
LLVMAddAnalysisPasses(LLVMTargetMachineRef T,LLVMPassManagerRef PM)268 void LLVMAddAnalysisPasses(LLVMTargetMachineRef T, LLVMPassManagerRef PM) {
269 unwrap(PM)->add(
270 createTargetTransformInfoWrapperPass(unwrap(T)->getTargetIRAnalysis()));
271 }
272