blob: d4c61e2d13c4cea6381061244ca797947d1e0fc9 [file] [log] [blame]
Amaury Sechete8ea7d82016-02-04 23:26:19 +00001//===-- echo.cpp - tool for testing libLLVM and llvm-c API ----------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
Jeroen Ketemaad659c32016-04-08 09:19:02 +000010// This file implements the --echo command in llvm-c-test.
Amaury Sechete8ea7d82016-02-04 23:26:19 +000011//
12// This command uses the C API to read a module and output an exact copy of it
13// as output. It is used to check that the resulting module matches the input
14// to validate that the C API can read and write modules properly.
15//
16//===----------------------------------------------------------------------===//
17
18#include "llvm-c-test.h"
Robert Widmann9cba4ec2018-09-28 15:35:18 +000019#include "llvm-c/DebugInfo.h"
Amaury Sechet55909672016-02-16 05:11:24 +000020#include "llvm-c/Target.h"
Amaury Sechete8ea7d82016-02-04 23:26:19 +000021#include "llvm/ADT/DenseMap.h"
Amaury Secheta82042e2016-02-09 22:36:41 +000022#include "llvm/Support/ErrorHandling.h"
Amaury Sechete8ea7d82016-02-04 23:26:19 +000023
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +000024#include <stdio.h>
25#include <stdlib.h>
Amaury Sechete8ea7d82016-02-04 23:26:19 +000026
27using namespace llvm;
28
29// Provide DenseMapInfo for C API opaque types.
30template<typename T>
31struct CAPIDenseMap {};
32
33// The default DenseMapInfo require to know about pointer alignement.
34// Because the C API uses opaques pointer types, their alignement is unknown.
35// As a result, we need to roll out our own implementation.
36template<typename T>
37struct CAPIDenseMap<T*> {
38 struct CAPIDenseMapInfo {
39 static inline T* getEmptyKey() {
40 uintptr_t Val = static_cast<uintptr_t>(-1);
41 return reinterpret_cast<T*>(Val);
42 }
43 static inline T* getTombstoneKey() {
44 uintptr_t Val = static_cast<uintptr_t>(-2);
45 return reinterpret_cast<T*>(Val);
46 }
47 static unsigned getHashValue(const T *PtrVal) {
48 return hash_value(PtrVal);
49 }
50 static bool isEqual(const T *LHS, const T *RHS) { return LHS == RHS; }
51 };
52
53 typedef DenseMap<T*, T*, CAPIDenseMapInfo> Map;
54};
55
56typedef CAPIDenseMap<LLVMValueRef>::Map ValueMap;
Amaury Secheta82042e2016-02-09 22:36:41 +000057typedef CAPIDenseMap<LLVMBasicBlockRef>::Map BasicBlockMap;
Amaury Sechete8ea7d82016-02-04 23:26:19 +000058
Amaury Sechetaad93532016-02-10 00:38:50 +000059struct TypeCloner {
60 LLVMModuleRef M;
61 LLVMContextRef Ctx;
62
63 TypeCloner(LLVMModuleRef M): M(M), Ctx(LLVMGetModuleContext(M)) {}
64
Amaury Sechetc679dbd2016-02-14 09:30:42 +000065 LLVMTypeRef Clone(LLVMValueRef Src) {
66 return Clone(LLVMTypeOf(Src));
67 }
68
Amaury Sechetaad93532016-02-10 00:38:50 +000069 LLVMTypeRef Clone(LLVMTypeRef Src) {
70 LLVMTypeKind Kind = LLVMGetTypeKind(Src);
71 switch (Kind) {
72 case LLVMVoidTypeKind:
73 return LLVMVoidTypeInContext(Ctx);
74 case LLVMHalfTypeKind:
75 return LLVMHalfTypeInContext(Ctx);
76 case LLVMFloatTypeKind:
77 return LLVMFloatTypeInContext(Ctx);
78 case LLVMDoubleTypeKind:
79 return LLVMDoubleTypeInContext(Ctx);
80 case LLVMX86_FP80TypeKind:
81 return LLVMX86FP80TypeInContext(Ctx);
82 case LLVMFP128TypeKind:
83 return LLVMFP128TypeInContext(Ctx);
84 case LLVMPPC_FP128TypeKind:
85 return LLVMPPCFP128TypeInContext(Ctx);
86 case LLVMLabelTypeKind:
87 return LLVMLabelTypeInContext(Ctx);
88 case LLVMIntegerTypeKind:
89 return LLVMIntTypeInContext(Ctx, LLVMGetIntTypeWidth(Src));
90 case LLVMFunctionTypeKind: {
91 unsigned ParamCount = LLVMCountParamTypes(Src);
92 LLVMTypeRef* Params = nullptr;
93 if (ParamCount > 0) {
Serge Pavlov76d8cce2018-02-20 05:41:26 +000094 Params = static_cast<LLVMTypeRef*>(
95 safe_malloc(ParamCount * sizeof(LLVMTypeRef)));
Amaury Sechetaad93532016-02-10 00:38:50 +000096 LLVMGetParamTypes(Src, Params);
97 for (unsigned i = 0; i < ParamCount; i++)
98 Params[i] = Clone(Params[i]);
99 }
100
101 LLVMTypeRef FunTy = LLVMFunctionType(Clone(LLVMGetReturnType(Src)),
102 Params, ParamCount,
103 LLVMIsFunctionVarArg(Src));
104 if (ParamCount > 0)
105 free(Params);
106 return FunTy;
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000107 }
Amaury Sechetaad93532016-02-10 00:38:50 +0000108 case LLVMStructTypeKind: {
109 LLVMTypeRef S = nullptr;
110 const char *Name = LLVMGetStructName(Src);
111 if (Name) {
112 S = LLVMGetTypeByName(M, Name);
113 if (S)
114 return S;
115 S = LLVMStructCreateNamed(Ctx, Name);
116 if (LLVMIsOpaqueStruct(Src))
117 return S;
118 }
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000119
Amaury Sechetaad93532016-02-10 00:38:50 +0000120 unsigned EltCount = LLVMCountStructElementTypes(Src);
121 SmallVector<LLVMTypeRef, 8> Elts;
122 for (unsigned i = 0; i < EltCount; i++)
123 Elts.push_back(Clone(LLVMStructGetTypeAtIndex(Src, i)));
124 if (Name)
125 LLVMStructSetBody(S, Elts.data(), EltCount, LLVMIsPackedStruct(Src));
126 else
127 S = LLVMStructTypeInContext(Ctx, Elts.data(), EltCount,
128 LLVMIsPackedStruct(Src));
129 return S;
130 }
131 case LLVMArrayTypeKind:
132 return LLVMArrayType(
133 Clone(LLVMGetElementType(Src)),
134 LLVMGetArrayLength(Src)
135 );
136 case LLVMPointerTypeKind:
137 return LLVMPointerType(
138 Clone(LLVMGetElementType(Src)),
139 LLVMGetPointerAddressSpace(Src)
140 );
141 case LLVMVectorTypeKind:
142 return LLVMVectorType(
143 Clone(LLVMGetElementType(Src)),
144 LLVMGetVectorSize(Src)
145 );
146 case LLVMMetadataTypeKind:
whitequark131f98f2017-10-27 11:51:40 +0000147 return LLVMMetadataTypeInContext(Ctx);
Amaury Sechetaad93532016-02-10 00:38:50 +0000148 case LLVMX86_MMXTypeKind:
149 return LLVMX86MMXTypeInContext(Ctx);
Robert Widmann478fce92018-03-30 17:49:53 +0000150 case LLVMTokenTypeKind:
151 return LLVMTokenTypeInContext(Ctx);
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000152 }
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000153
Amaury Sechetaad93532016-02-10 00:38:50 +0000154 fprintf(stderr, "%d is not a supported typekind\n", Kind);
155 exit(-1);
156 }
157};
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000158
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000159static ValueMap clone_params(LLVMValueRef Src, LLVMValueRef Dst) {
Amaury Sechetecda4ea2016-02-14 09:14:30 +0000160 unsigned Count = LLVMCountParams(Src);
Amaury Sechetd01c8612016-02-14 10:06:34 +0000161 if (Count != LLVMCountParams(Dst))
162 report_fatal_error("Parameter count mismatch");
Amaury Sechetecda4ea2016-02-14 09:14:30 +0000163
164 ValueMap VMap;
165 if (Count == 0)
166 return VMap;
167
168 LLVMValueRef SrcFirst = LLVMGetFirstParam(Src);
169 LLVMValueRef DstFirst = LLVMGetFirstParam(Dst);
170 LLVMValueRef SrcLast = LLVMGetLastParam(Src);
171 LLVMValueRef DstLast = LLVMGetLastParam(Dst);
172
173 LLVMValueRef SrcCur = SrcFirst;
174 LLVMValueRef DstCur = DstFirst;
175 LLVMValueRef SrcNext = nullptr;
176 LLVMValueRef DstNext = nullptr;
177 while (true) {
Robert Widmann025c78f2018-05-19 15:08:36 +0000178 size_t NameLen;
179 const char *Name = LLVMGetValueName2(SrcCur, &NameLen);
180 LLVMSetValueName2(DstCur, Name, NameLen);
Amaury Sechetecda4ea2016-02-14 09:14:30 +0000181
182 VMap[SrcCur] = DstCur;
183
184 Count--;
185 SrcNext = LLVMGetNextParam(SrcCur);
186 DstNext = LLVMGetNextParam(DstCur);
187 if (SrcNext == nullptr && DstNext == nullptr) {
Amaury Sechetd01c8612016-02-14 10:06:34 +0000188 if (SrcCur != SrcLast)
189 report_fatal_error("SrcLast param does not match End");
190 if (DstCur != DstLast)
191 report_fatal_error("DstLast param does not match End");
Amaury Sechetecda4ea2016-02-14 09:14:30 +0000192 break;
193 }
194
Amaury Sechetd01c8612016-02-14 10:06:34 +0000195 if (SrcNext == nullptr)
196 report_fatal_error("SrcNext was unexpectedly null");
197 if (DstNext == nullptr)
198 report_fatal_error("DstNext was unexpectedly null");
Amaury Sechetecda4ea2016-02-14 09:14:30 +0000199
200 LLVMValueRef SrcPrev = LLVMGetPreviousParam(SrcNext);
Amaury Sechetd01c8612016-02-14 10:06:34 +0000201 if (SrcPrev != SrcCur)
202 report_fatal_error("SrcNext.Previous param is not Current");
Amaury Sechetecda4ea2016-02-14 09:14:30 +0000203
204 LLVMValueRef DstPrev = LLVMGetPreviousParam(DstNext);
Amaury Sechetd01c8612016-02-14 10:06:34 +0000205 if (DstPrev != DstCur)
206 report_fatal_error("DstNext.Previous param is not Current");
Amaury Sechetecda4ea2016-02-14 09:14:30 +0000207
208 SrcCur = SrcNext;
209 DstCur = DstNext;
210 }
211
Amaury Sechetd01c8612016-02-14 10:06:34 +0000212 if (Count != 0)
213 report_fatal_error("Parameter count does not match iteration");
Amaury Sechetecda4ea2016-02-14 09:14:30 +0000214
215 return VMap;
216}
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000217
Amaury Sechet49491312016-04-07 05:56:20 +0000218static void check_value_kind(LLVMValueRef V, LLVMValueKind K) {
219 if (LLVMGetValueKind(V) != K)
220 report_fatal_error("LLVMGetValueKind returned incorrect type");
221}
222
223static LLVMValueRef clone_constant_impl(LLVMValueRef Cst, LLVMModuleRef M);
224
225static LLVMValueRef clone_constant(LLVMValueRef Cst, LLVMModuleRef M) {
226 LLVMValueRef Ret = clone_constant_impl(Cst, M);
227 check_value_kind(Ret, LLVMGetValueKind(Cst));
228 return Ret;
229}
230
231static LLVMValueRef clone_constant_impl(LLVMValueRef Cst, LLVMModuleRef M) {
Amaury Sechetc679dbd2016-02-14 09:30:42 +0000232 if (!LLVMIsAConstant(Cst))
233 report_fatal_error("Expected a constant");
234
235 // Maybe it is a symbol
236 if (LLVMIsAGlobalValue(Cst)) {
Robert Widmann025c78f2018-05-19 15:08:36 +0000237 size_t NameLen;
238 const char *Name = LLVMGetValueName2(Cst, &NameLen);
Amaury Sechetc679dbd2016-02-14 09:30:42 +0000239
240 // Try function
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000241 if (LLVMIsAFunction(Cst)) {
Amaury Sechet49491312016-04-07 05:56:20 +0000242 check_value_kind(Cst, LLVMFunctionValueKind);
243 LLVMValueRef Dst = LLVMGetNamedFunction(M, Name);
244 if (Dst)
245 return Dst;
246 report_fatal_error("Could not find function");
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000247 }
Amaury Sechet49491312016-04-07 05:56:20 +0000248
249 // Try global variable
250 if (LLVMIsAGlobalVariable(Cst)) {
251 check_value_kind(Cst, LLVMGlobalVariableValueKind);
Robert Widmann360d6e32018-05-20 23:49:08 +0000252 LLVMValueRef Dst = LLVMGetNamedGlobal(M, Name);
Amaury Sechet49491312016-04-07 05:56:20 +0000253 if (Dst)
254 return Dst;
Robert Widmann360d6e32018-05-20 23:49:08 +0000255 report_fatal_error("Could not find variable");
256 }
257
258 // Try global alias
259 if (LLVMIsAGlobalAlias(Cst)) {
260 check_value_kind(Cst, LLVMGlobalAliasValueKind);
261 LLVMValueRef Dst = LLVMGetNamedGlobalAlias(M, Name, NameLen);
262 if (Dst)
263 return Dst;
264 report_fatal_error("Could not find alias");
Amaury Sechet49491312016-04-07 05:56:20 +0000265 }
266
267 fprintf(stderr, "Could not find @%s\n", Name);
268 exit(-1);
269 }
270
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000271 // Try integer literal
Amaury Sechet49491312016-04-07 05:56:20 +0000272 if (LLVMIsAConstantInt(Cst)) {
273 check_value_kind(Cst, LLVMConstantIntValueKind);
274 return LLVMConstInt(TypeCloner(M).Clone(Cst),
275 LLVMConstIntGetZExtValue(Cst), false);
276 }
277
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000278 // Try zeroinitializer
Amaury Sechet49491312016-04-07 05:56:20 +0000279 if (LLVMIsAConstantAggregateZero(Cst)) {
280 check_value_kind(Cst, LLVMConstantAggregateZeroValueKind);
281 return LLVMConstNull(TypeCloner(M).Clone(Cst));
282 }
283
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000284 // Try constant array
Amaury Sechet49491312016-04-07 05:56:20 +0000285 if (LLVMIsAConstantArray(Cst)) {
286 check_value_kind(Cst, LLVMConstantArrayValueKind);
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000287 LLVMTypeRef Ty = TypeCloner(M).Clone(Cst);
288 unsigned EltCount = LLVMGetArrayLength(Ty);
289 SmallVector<LLVMValueRef, 8> Elts;
290 for (unsigned i = 0; i < EltCount; i++)
291 Elts.push_back(clone_constant(LLVMGetOperand(Cst, i), M));
Amaury Sechet49491312016-04-07 05:56:20 +0000292 return LLVMConstArray(LLVMGetElementType(Ty), Elts.data(), EltCount);
293 }
294
Amaury Sechetb3256862016-03-13 00:58:25 +0000295 // Try contant data array
Amaury Sechet49491312016-04-07 05:56:20 +0000296 if (LLVMIsAConstantDataArray(Cst)) {
297 check_value_kind(Cst, LLVMConstantDataArrayValueKind);
Amaury Sechetb3256862016-03-13 00:58:25 +0000298 LLVMTypeRef Ty = TypeCloner(M).Clone(Cst);
299 unsigned EltCount = LLVMGetArrayLength(Ty);
300 SmallVector<LLVMValueRef, 8> Elts;
301 for (unsigned i = 0; i < EltCount; i++)
302 Elts.push_back(clone_constant(LLVMGetElementAsConstant(Cst, i), M));
Amaury Sechet49491312016-04-07 05:56:20 +0000303 return LLVMConstArray(LLVMGetElementType(Ty), Elts.data(), EltCount);
304 }
305
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000306 // Try constant struct
Amaury Sechet49491312016-04-07 05:56:20 +0000307 if (LLVMIsAConstantStruct(Cst)) {
308 check_value_kind(Cst, LLVMConstantStructValueKind);
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000309 LLVMTypeRef Ty = TypeCloner(M).Clone(Cst);
310 unsigned EltCount = LLVMCountStructElementTypes(Ty);
311 SmallVector<LLVMValueRef, 8> Elts;
312 for (unsigned i = 0; i < EltCount; i++)
313 Elts.push_back(clone_constant(LLVMGetOperand(Cst, i), M));
314 if (LLVMGetStructName(Ty))
Amaury Sechet49491312016-04-07 05:56:20 +0000315 return LLVMConstNamedStruct(Ty, Elts.data(), EltCount);
316 return LLVMConstStructInContext(LLVMGetModuleContext(M), Elts.data(),
317 EltCount, LLVMIsPackedStruct(Ty));
318 }
319
Amaury Sechetc679dbd2016-02-14 09:30:42 +0000320 // Try undef
Amaury Sechet49491312016-04-07 05:56:20 +0000321 if (LLVMIsUndef(Cst)) {
322 check_value_kind(Cst, LLVMUndefValueValueKind);
323 return LLVMGetUndef(TypeCloner(M).Clone(Cst));
324 }
325
Robert Widmann478fce92018-03-30 17:49:53 +0000326 // Try null
327 if (LLVMIsNull(Cst)) {
328 check_value_kind(Cst, LLVMConstantTokenNoneValueKind);
329 LLVMTypeRef Ty = TypeCloner(M).Clone(Cst);
330 return LLVMConstNull(Ty);
331 }
332
Amaury Sechet40bbe512016-02-17 23:55:59 +0000333 // Try float literal
Amaury Sechet49491312016-04-07 05:56:20 +0000334 if (LLVMIsAConstantFP(Cst)) {
335 check_value_kind(Cst, LLVMConstantFPValueKind);
Amaury Sechet40bbe512016-02-17 23:55:59 +0000336 report_fatal_error("ConstantFP is not supported");
Amaury Sechet49491312016-04-07 05:56:20 +0000337 }
338
Amaury Sechet6b16c232016-02-16 07:33:23 +0000339 // This kind of constant is not supported
Amaury Sechet49491312016-04-07 05:56:20 +0000340 if (!LLVMIsAConstantExpr(Cst))
Amaury Sechet6b16c232016-02-16 07:33:23 +0000341 report_fatal_error("Expected a constant expression");
Amaury Sechet49491312016-04-07 05:56:20 +0000342
343 // At this point, it must be a constant expression
344 check_value_kind(Cst, LLVMConstantExprValueKind);
345
346 LLVMOpcode Op = LLVMGetConstOpcode(Cst);
347 switch(Op) {
348 case LLVMBitCast:
349 return LLVMConstBitCast(clone_constant(LLVMGetOperand(Cst, 0), M),
350 TypeCloner(M).Clone(Cst));
351 default:
352 fprintf(stderr, "%d is not a supported opcode\n", Op);
353 exit(-1);
Amaury Sechet40bbe512016-02-17 23:55:59 +0000354 }
Amaury Sechetc679dbd2016-02-14 09:30:42 +0000355}
356
Amaury Secheta82042e2016-02-09 22:36:41 +0000357struct FunCloner {
358 LLVMValueRef Fun;
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000359 LLVMModuleRef M;
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000360
Amaury Secheta82042e2016-02-09 22:36:41 +0000361 ValueMap VMap;
362 BasicBlockMap BBMap;
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000363
Amaury Sechetaad93532016-02-10 00:38:50 +0000364 FunCloner(LLVMValueRef Src, LLVMValueRef Dst): Fun(Dst),
365 M(LLVMGetGlobalParent(Fun)), VMap(clone_params(Src, Dst)) {}
366
367 LLVMTypeRef CloneType(LLVMTypeRef Src) {
368 return TypeCloner(M).Clone(Src);
369 }
370
371 LLVMTypeRef CloneType(LLVMValueRef Src) {
Amaury Sechetc679dbd2016-02-14 09:30:42 +0000372 return TypeCloner(M).Clone(Src);
Amaury Sechetaad93532016-02-10 00:38:50 +0000373 }
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000374
Amaury Secheta82042e2016-02-09 22:36:41 +0000375 // Try to clone everything in the llvm::Value hierarchy.
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000376 LLVMValueRef CloneValue(LLVMValueRef Src) {
Amaury Secheta82042e2016-02-09 22:36:41 +0000377 // First, the value may be constant.
Amaury Sechetc679dbd2016-02-14 09:30:42 +0000378 if (LLVMIsAConstant(Src))
379 return clone_constant(Src, M);
Amaury Secheta82042e2016-02-09 22:36:41 +0000380
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000381 // Function argument should always be in the map already.
Amaury Sechet2f432082016-02-11 21:37:54 +0000382 auto i = VMap.find(Src);
383 if (i != VMap.end())
384 return i->second;
Amaury Secheta82042e2016-02-09 22:36:41 +0000385
Amaury Sechet2f432082016-02-11 21:37:54 +0000386 if (!LLVMIsAInstruction(Src))
387 report_fatal_error("Expected an instruction");
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000388
Amaury Sechet2f432082016-02-11 21:37:54 +0000389 auto Ctx = LLVMGetModuleContext(M);
390 auto Builder = LLVMCreateBuilderInContext(Ctx);
391 auto BB = DeclareBB(LLVMGetInstructionParent(Src));
392 LLVMPositionBuilderAtEnd(Builder, BB);
393 auto Dst = CloneInstruction(Src, Builder);
394 LLVMDisposeBuilder(Builder);
395 return Dst;
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000396 }
397
Amaury Secheta65a2372016-06-15 05:14:29 +0000398 void CloneAttrs(LLVMValueRef Src, LLVMValueRef Dst) {
399 auto Ctx = LLVMGetModuleContext(M);
400 int ArgCount = LLVMGetNumArgOperands(Src);
401 for (int i = LLVMAttributeReturnIndex; i <= ArgCount; i++) {
402 for (unsigned k = 0, e = LLVMGetLastEnumAttributeKind(); k < e; ++k) {
403 if (auto SrcA = LLVMGetCallSiteEnumAttribute(Src, i, k)) {
404 auto Val = LLVMGetEnumAttributeValue(SrcA);
405 auto A = LLVMCreateEnumAttribute(Ctx, k, Val);
406 LLVMAddCallSiteAttribute(Dst, i, A);
407 }
408 }
409 }
410 }
411
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000412 LLVMValueRef CloneInstruction(LLVMValueRef Src, LLVMBuilderRef Builder) {
Amaury Sechet49491312016-04-07 05:56:20 +0000413 check_value_kind(Src, LLVMInstructionValueKind);
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000414 if (!LLVMIsAInstruction(Src))
415 report_fatal_error("Expected an instruction");
416
Robert Widmann025c78f2018-05-19 15:08:36 +0000417 size_t NameLen;
418 const char *Name = LLVMGetValueName2(Src, &NameLen);
Peter Zotov3e4561c2016-04-06 22:21:29 +0000419
Amaury Secheta82042e2016-02-09 22:36:41 +0000420 // Check if this is something we already computed.
421 {
422 auto i = VMap.find(Src);
Amaury Sechet2f432082016-02-11 21:37:54 +0000423 if (i != VMap.end()) {
424 // If we have a hit, it means we already generated the instruction
425 // as a dependancy to somethign else. We need to make sure
426 // it is ordered properly.
427 auto I = i->second;
428 LLVMInstructionRemoveFromParent(I);
429 LLVMInsertIntoBuilderWithName(Builder, I, Name);
430 return I;
431 }
Amaury Secheta82042e2016-02-09 22:36:41 +0000432 }
433
434 // We tried everything, it must be an instruction
435 // that hasn't been generated already.
436 LLVMValueRef Dst = nullptr;
437
438 LLVMOpcode Op = LLVMGetInstructionOpcode(Src);
439 switch(Op) {
440 case LLVMRet: {
441 int OpCount = LLVMGetNumOperands(Src);
442 if (OpCount == 0)
443 Dst = LLVMBuildRetVoid(Builder);
444 else
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000445 Dst = LLVMBuildRet(Builder, CloneValue(LLVMGetOperand(Src, 0)));
Amaury Secheta82042e2016-02-09 22:36:41 +0000446 break;
447 }
448 case LLVMBr: {
Amaury Sechete7e62172016-02-09 23:15:02 +0000449 if (!LLVMIsConditional(Src)) {
450 LLVMValueRef SrcOp = LLVMGetOperand(Src, 0);
451 LLVMBasicBlockRef SrcBB = LLVMValueAsBasicBlock(SrcOp);
452 Dst = LLVMBuildBr(Builder, DeclareBB(SrcBB));
453 break;
454 }
455
456 LLVMValueRef Cond = LLVMGetCondition(Src);
457 LLVMValueRef Else = LLVMGetOperand(Src, 1);
458 LLVMBasicBlockRef ElseBB = DeclareBB(LLVMValueAsBasicBlock(Else));
459 LLVMValueRef Then = LLVMGetOperand(Src, 2);
460 LLVMBasicBlockRef ThenBB = DeclareBB(LLVMValueAsBasicBlock(Then));
Rafael Espindola92f35782018-02-21 19:55:11 +0000461 Dst = LLVMBuildCondBr(Builder, CloneValue(Cond), ThenBB, ElseBB);
Amaury Secheta82042e2016-02-09 22:36:41 +0000462 break;
463 }
464 case LLVMSwitch:
465 case LLVMIndirectBr:
Amaury Secheta82042e2016-02-09 22:36:41 +0000466 break;
Amaury Sechete39e8532016-02-18 20:38:32 +0000467 case LLVMInvoke: {
468 SmallVector<LLVMValueRef, 8> Args;
469 int ArgCount = LLVMGetNumArgOperands(Src);
470 for (int i = 0; i < ArgCount; i++)
471 Args.push_back(CloneValue(LLVMGetOperand(Src, i)));
472 LLVMValueRef Fn = CloneValue(LLVMGetCalledValue(Src));
473 LLVMBasicBlockRef Then = DeclareBB(LLVMGetNormalDest(Src));
474 LLVMBasicBlockRef Unwind = DeclareBB(LLVMGetUnwindDest(Src));
475 Dst = LLVMBuildInvoke(Builder, Fn, Args.data(), ArgCount,
476 Then, Unwind, Name);
Amaury Secheta65a2372016-06-15 05:14:29 +0000477 CloneAttrs(Src, Dst);
Amaury Sechete39e8532016-02-18 20:38:32 +0000478 break;
479 }
Amaury Secheta82042e2016-02-09 22:36:41 +0000480 case LLVMUnreachable:
481 Dst = LLVMBuildUnreachable(Builder);
482 break;
483 case LLVMAdd: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000484 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
485 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000486 Dst = LLVMBuildAdd(Builder, LHS, RHS, Name);
487 break;
488 }
489 case LLVMSub: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000490 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
491 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000492 Dst = LLVMBuildSub(Builder, LHS, RHS, Name);
493 break;
494 }
495 case LLVMMul: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000496 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
497 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000498 Dst = LLVMBuildMul(Builder, LHS, RHS, Name);
499 break;
500 }
501 case LLVMUDiv: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000502 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
503 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000504 Dst = LLVMBuildUDiv(Builder, LHS, RHS, Name);
505 break;
506 }
507 case LLVMSDiv: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000508 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
509 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000510 Dst = LLVMBuildSDiv(Builder, LHS, RHS, Name);
511 break;
512 }
513 case LLVMURem: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000514 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
515 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000516 Dst = LLVMBuildURem(Builder, LHS, RHS, Name);
517 break;
518 }
519 case LLVMSRem: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000520 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
521 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000522 Dst = LLVMBuildSRem(Builder, LHS, RHS, Name);
523 break;
524 }
525 case LLVMShl: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000526 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
527 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000528 Dst = LLVMBuildShl(Builder, LHS, RHS, Name);
529 break;
530 }
531 case LLVMLShr: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000532 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
533 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000534 Dst = LLVMBuildLShr(Builder, LHS, RHS, Name);
535 break;
536 }
537 case LLVMAShr: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000538 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
539 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000540 Dst = LLVMBuildAShr(Builder, LHS, RHS, Name);
541 break;
542 }
543 case LLVMAnd: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000544 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
545 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000546 Dst = LLVMBuildAnd(Builder, LHS, RHS, Name);
547 break;
548 }
549 case LLVMOr: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000550 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
551 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000552 Dst = LLVMBuildOr(Builder, LHS, RHS, Name);
553 break;
554 }
555 case LLVMXor: {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000556 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
557 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Secheta82042e2016-02-09 22:36:41 +0000558 Dst = LLVMBuildXor(Builder, LHS, RHS, Name);
559 break;
560 }
561 case LLVMAlloca: {
Amaury Sechetaad93532016-02-10 00:38:50 +0000562 LLVMTypeRef Ty = CloneType(LLVMGetAllocatedType(Src));
Amaury Secheta82042e2016-02-09 22:36:41 +0000563 Dst = LLVMBuildAlloca(Builder, Ty, Name);
564 break;
565 }
Amaury Sechet053ac452016-02-17 22:51:03 +0000566 case LLVMLoad: {
567 LLVMValueRef Ptr = CloneValue(LLVMGetOperand(Src, 0));
568 Dst = LLVMBuildLoad(Builder, Ptr, Name);
569 LLVMSetAlignment(Dst, LLVMGetAlignment(Src));
570 break;
571 }
572 case LLVMStore: {
573 LLVMValueRef Val = CloneValue(LLVMGetOperand(Src, 0));
574 LLVMValueRef Ptr = CloneValue(LLVMGetOperand(Src, 1));
575 Dst = LLVMBuildStore(Builder, Val, Ptr);
576 LLVMSetAlignment(Dst, LLVMGetAlignment(Src));
577 break;
578 }
579 case LLVMGetElementPtr: {
580 LLVMValueRef Ptr = CloneValue(LLVMGetOperand(Src, 0));
581 SmallVector<LLVMValueRef, 8> Idx;
582 int NumIdx = LLVMGetNumIndices(Src);
583 for (int i = 1; i <= NumIdx; i++)
584 Idx.push_back(CloneValue(LLVMGetOperand(Src, i)));
585 if (LLVMIsInBounds(Src))
586 Dst = LLVMBuildInBoundsGEP(Builder, Ptr, Idx.data(), NumIdx, Name);
587 else
588 Dst = LLVMBuildGEP(Builder, Ptr, Idx.data(), NumIdx, Name);
589 break;
590 }
Mehdi Amini43165d92016-03-19 21:28:28 +0000591 case LLVMAtomicCmpXchg: {
592 LLVMValueRef Ptr = CloneValue(LLVMGetOperand(Src, 0));
593 LLVMValueRef Cmp = CloneValue(LLVMGetOperand(Src, 1));
594 LLVMValueRef New = CloneValue(LLVMGetOperand(Src, 2));
595 LLVMAtomicOrdering Succ = LLVMGetCmpXchgSuccessOrdering(Src);
596 LLVMAtomicOrdering Fail = LLVMGetCmpXchgFailureOrdering(Src);
597 LLVMBool SingleThread = LLVMIsAtomicSingleThread(Src);
598
599 Dst = LLVMBuildAtomicCmpXchg(Builder, Ptr, Cmp, New, Succ, Fail,
600 SingleThread);
601 } break;
Amaury Sechet40bbe512016-02-17 23:55:59 +0000602 case LLVMBitCast: {
603 LLVMValueRef V = CloneValue(LLVMGetOperand(Src, 0));
604 Dst = LLVMBuildBitCast(Builder, V, CloneType(Src), Name);
605 break;
606 }
Amaury Sechete7e62172016-02-09 23:15:02 +0000607 case LLVMICmp: {
608 LLVMIntPredicate Pred = LLVMGetICmpPredicate(Src);
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000609 LLVMValueRef LHS = CloneValue(LLVMGetOperand(Src, 0));
610 LLVMValueRef RHS = CloneValue(LLVMGetOperand(Src, 1));
Amaury Sechete7e62172016-02-09 23:15:02 +0000611 Dst = LLVMBuildICmp(Builder, Pred, LHS, RHS, Name);
612 break;
613 }
Amaury Sechet2f432082016-02-11 21:37:54 +0000614 case LLVMPHI: {
Simon Pilgrim68168d12017-03-30 12:59:53 +0000615 // We need to aggressively set things here because of loops.
Amaury Sechet2f432082016-02-11 21:37:54 +0000616 VMap[Src] = Dst = LLVMBuildPhi(Builder, CloneType(Src), Name);
617
618 SmallVector<LLVMValueRef, 8> Values;
619 SmallVector<LLVMBasicBlockRef, 8> Blocks;
620
621 unsigned IncomingCount = LLVMCountIncoming(Src);
622 for (unsigned i = 0; i < IncomingCount; ++i) {
623 Blocks.push_back(DeclareBB(LLVMGetIncomingBlock(Src, i)));
624 Values.push_back(CloneValue(LLVMGetIncomingValue(Src, i)));
625 }
626
627 LLVMAddIncoming(Dst, Values.data(), Blocks.data(), IncomingCount);
628 return Dst;
629 }
Amaury Secheta82042e2016-02-09 22:36:41 +0000630 case LLVMCall: {
Amaury Secheta82042e2016-02-09 22:36:41 +0000631 SmallVector<LLVMValueRef, 8> Args;
Amaury Sechet5c7b3af2016-02-10 00:09:37 +0000632 int ArgCount = LLVMGetNumArgOperands(Src);
Amaury Secheta82042e2016-02-09 22:36:41 +0000633 for (int i = 0; i < ArgCount; i++)
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000634 Args.push_back(CloneValue(LLVMGetOperand(Src, i)));
Amaury Sechet5c7b3af2016-02-10 00:09:37 +0000635 LLVMValueRef Fn = CloneValue(LLVMGetCalledValue(Src));
Amaury Secheta82042e2016-02-09 22:36:41 +0000636 Dst = LLVMBuildCall(Builder, Fn, Args.data(), ArgCount, Name);
Amaury Sechete39e8532016-02-18 20:38:32 +0000637 LLVMSetTailCall(Dst, LLVMIsTailCall(Src));
Amaury Secheta65a2372016-06-15 05:14:29 +0000638 CloneAttrs(Src, Dst);
Amaury Sechete39e8532016-02-18 20:38:32 +0000639 break;
640 }
641 case LLVMResume: {
642 Dst = LLVMBuildResume(Builder, CloneValue(LLVMGetOperand(Src, 0)));
643 break;
644 }
645 case LLVMLandingPad: {
646 // The landing pad API is a bit screwed up for historical reasons.
647 Dst = LLVMBuildLandingPad(Builder, CloneType(Src), nullptr, 0, Name);
648 unsigned NumClauses = LLVMGetNumClauses(Src);
649 for (unsigned i = 0; i < NumClauses; ++i)
650 LLVMAddClause(Dst, CloneValue(LLVMGetClause(Src, i)));
651 LLVMSetCleanup(Dst, LLVMIsCleanup(Src));
Amaury Secheta82042e2016-02-09 22:36:41 +0000652 break;
653 }
Robert Widmann478fce92018-03-30 17:49:53 +0000654 case LLVMCleanupRet: {
655 LLVMValueRef CatchPad = CloneValue(LLVMGetOperand(Src, 0));
656 LLVMBasicBlockRef Unwind = nullptr;
657 if (LLVMBasicBlockRef UDest = LLVMGetUnwindDest(Src))
658 Unwind = DeclareBB(UDest);
659 Dst = LLVMBuildCleanupRet(Builder, CatchPad, Unwind);
660 break;
661 }
662 case LLVMCatchRet: {
663 LLVMValueRef CatchPad = CloneValue(LLVMGetOperand(Src, 0));
664 LLVMBasicBlockRef SuccBB = DeclareBB(LLVMGetSuccessor(Src, 0));
665 Dst = LLVMBuildCatchRet(Builder, CatchPad, SuccBB);
666 break;
667 }
668 case LLVMCatchPad: {
669 LLVMValueRef ParentPad = CloneValue(LLVMGetParentCatchSwitch(Src));
670 SmallVector<LLVMValueRef, 8> Args;
671 int ArgCount = LLVMGetNumArgOperands(Src);
672 for (int i = 0; i < ArgCount; i++)
673 Args.push_back(CloneValue(LLVMGetOperand(Src, i)));
674 Dst = LLVMBuildCatchPad(Builder, ParentPad,
675 Args.data(), ArgCount, Name);
676 break;
677 }
678 case LLVMCleanupPad: {
679 LLVMValueRef ParentPad = CloneValue(LLVMGetOperand(Src, 0));
680 SmallVector<LLVMValueRef, 8> Args;
681 int ArgCount = LLVMGetNumArgOperands(Src);
682 for (int i = 0; i < ArgCount; i++)
683 Args.push_back(CloneValue(LLVMGetArgOperand(Src, i)));
684 Dst = LLVMBuildCleanupPad(Builder, ParentPad,
685 Args.data(), ArgCount, Name);
686 break;
687 }
688 case LLVMCatchSwitch: {
689 LLVMValueRef ParentPad = CloneValue(LLVMGetOperand(Src, 0));
690 LLVMBasicBlockRef UnwindBB = nullptr;
691 if (LLVMBasicBlockRef UDest = LLVMGetUnwindDest(Src)) {
692 UnwindBB = DeclareBB(UDest);
693 }
694 unsigned NumHandlers = LLVMGetNumHandlers(Src);
695 Dst = LLVMBuildCatchSwitch(Builder, ParentPad, UnwindBB, NumHandlers, Name);
696 if (NumHandlers > 0) {
697 LLVMBasicBlockRef *Handlers = static_cast<LLVMBasicBlockRef*>(
698 safe_malloc(NumHandlers * sizeof(LLVMBasicBlockRef)));
699 LLVMGetHandlers(Src, Handlers);
700 for (unsigned i = 0; i < NumHandlers; i++)
701 LLVMAddHandler(Dst, DeclareBB(Handlers[i]));
702 free(Handlers);
703 }
704 break;
705 }
Amaury Sechetaad93532016-02-10 00:38:50 +0000706 case LLVMExtractValue: {
707 LLVMValueRef Agg = CloneValue(LLVMGetOperand(Src, 0));
708 if (LLVMGetNumIndices(Src) != 1)
709 report_fatal_error("Expected only one indice");
710 auto I = LLVMGetIndices(Src)[0];
711 Dst = LLVMBuildExtractValue(Builder, Agg, I, Name);
712 break;
713 }
714 case LLVMInsertValue: {
715 LLVMValueRef Agg = CloneValue(LLVMGetOperand(Src, 0));
716 LLVMValueRef V = CloneValue(LLVMGetOperand(Src, 1));
717 if (LLVMGetNumIndices(Src) != 1)
718 report_fatal_error("Expected only one indice");
719 auto I = LLVMGetIndices(Src)[0];
720 Dst = LLVMBuildInsertValue(Builder, Agg, V, I, Name);
721 break;
722 }
Amaury Secheta82042e2016-02-09 22:36:41 +0000723 default:
724 break;
725 }
726
727 if (Dst == nullptr) {
728 fprintf(stderr, "%d is not a supported opcode\n", Op);
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000729 exit(-1);
730 }
731
Robert Widmann9cba4ec2018-09-28 15:35:18 +0000732 auto Ctx = LLVMGetModuleContext(M);
733 size_t NumMetadataEntries;
734 auto *AllMetadata =
735 LLVMInstructionGetAllMetadataOtherThanDebugLoc(Src,
736 &NumMetadataEntries);
737 for (unsigned i = 0; i < NumMetadataEntries; ++i) {
738 unsigned Kind = LLVMValueMetadataEntriesGetKind(AllMetadata, i);
739 LLVMMetadataRef MD = LLVMValueMetadataEntriesGetMetadata(AllMetadata, i);
740 LLVMSetMetadata(Dst, Kind, LLVMMetadataAsValue(Ctx, MD));
741 }
742 LLVMDisposeValueMetadataEntries(AllMetadata);
743 LLVMSetInstDebugLocation(Builder, Dst);
744
Amaury Sechet49491312016-04-07 05:56:20 +0000745 check_value_kind(Dst, LLVMInstructionValueKind);
Amaury Secheta82042e2016-02-09 22:36:41 +0000746 return VMap[Src] = Dst;
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000747 }
748
Amaury Secheta82042e2016-02-09 22:36:41 +0000749 LLVMBasicBlockRef DeclareBB(LLVMBasicBlockRef Src) {
750 // Check if this is something we already computed.
751 {
752 auto i = BBMap.find(Src);
753 if (i != BBMap.end()) {
754 return i->second;
755 }
756 }
757
Amaury Secheta82042e2016-02-09 22:36:41 +0000758 LLVMValueRef V = LLVMBasicBlockAsValue(Src);
759 if (!LLVMValueIsBasicBlock(V) || LLVMValueAsBasicBlock(V) != Src)
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000760 report_fatal_error("Basic block is not a basic block");
Amaury Secheta82042e2016-02-09 22:36:41 +0000761
Amaury Sechetd01c8612016-02-14 10:06:34 +0000762 const char *Name = LLVMGetBasicBlockName(Src);
Robert Widmann025c78f2018-05-19 15:08:36 +0000763 size_t NameLen;
764 const char *VName = LLVMGetValueName2(V, &NameLen);
Amaury Secheta82042e2016-02-09 22:36:41 +0000765 if (Name != VName)
766 report_fatal_error("Basic block name mismatch");
767
768 LLVMBasicBlockRef BB = LLVMAppendBasicBlock(Fun, Name);
769 return BBMap[Src] = BB;
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000770 }
771
Amaury Secheta82042e2016-02-09 22:36:41 +0000772 LLVMBasicBlockRef CloneBB(LLVMBasicBlockRef Src) {
773 LLVMBasicBlockRef BB = DeclareBB(Src);
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000774
Amaury Secheta82042e2016-02-09 22:36:41 +0000775 // Make sure ordering is correct.
776 LLVMBasicBlockRef Prev = LLVMGetPreviousBasicBlock(Src);
777 if (Prev)
778 LLVMMoveBasicBlockAfter(BB, DeclareBB(Prev));
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000779
Amaury Secheta82042e2016-02-09 22:36:41 +0000780 LLVMValueRef First = LLVMGetFirstInstruction(Src);
781 LLVMValueRef Last = LLVMGetLastInstruction(Src);
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000782
Amaury Secheta82042e2016-02-09 22:36:41 +0000783 if (First == nullptr) {
Amaury Sechetd01c8612016-02-14 10:06:34 +0000784 if (Last != nullptr)
785 report_fatal_error("Has no first instruction, but last one");
Amaury Secheta82042e2016-02-09 22:36:41 +0000786 return BB;
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000787 }
788
Amaury Sechetaad93532016-02-10 00:38:50 +0000789 auto Ctx = LLVMGetModuleContext(M);
Amaury Secheta82042e2016-02-09 22:36:41 +0000790 LLVMBuilderRef Builder = LLVMCreateBuilderInContext(Ctx);
791 LLVMPositionBuilderAtEnd(Builder, BB);
792
793 LLVMValueRef Cur = First;
794 LLVMValueRef Next = nullptr;
795 while(true) {
Amaury Sechet1e5d7e22016-02-09 23:41:20 +0000796 CloneInstruction(Cur, Builder);
Amaury Secheta82042e2016-02-09 22:36:41 +0000797 Next = LLVMGetNextInstruction(Cur);
798 if (Next == nullptr) {
Amaury Sechetd01c8612016-02-14 10:06:34 +0000799 if (Cur != Last)
800 report_fatal_error("Final instruction does not match Last");
Amaury Secheta82042e2016-02-09 22:36:41 +0000801 break;
802 }
803
804 LLVMValueRef Prev = LLVMGetPreviousInstruction(Next);
Amaury Sechetd01c8612016-02-14 10:06:34 +0000805 if (Prev != Cur)
806 report_fatal_error("Next.Previous instruction is not Current");
Amaury Secheta82042e2016-02-09 22:36:41 +0000807
808 Cur = Next;
809 }
810
811 LLVMDisposeBuilder(Builder);
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000812 return BB;
813 }
814
Amaury Secheta82042e2016-02-09 22:36:41 +0000815 void CloneBBs(LLVMValueRef Src) {
816 unsigned Count = LLVMCountBasicBlocks(Src);
817 if (Count == 0)
818 return;
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000819
Amaury Secheta82042e2016-02-09 22:36:41 +0000820 LLVMBasicBlockRef First = LLVMGetFirstBasicBlock(Src);
821 LLVMBasicBlockRef Last = LLVMGetLastBasicBlock(Src);
822
823 LLVMBasicBlockRef Cur = First;
824 LLVMBasicBlockRef Next = nullptr;
825 while(true) {
826 CloneBB(Cur);
827 Count--;
828 Next = LLVMGetNextBasicBlock(Cur);
829 if (Next == nullptr) {
Amaury Sechetd01c8612016-02-14 10:06:34 +0000830 if (Cur != Last)
831 report_fatal_error("Final basic block does not match Last");
Amaury Secheta82042e2016-02-09 22:36:41 +0000832 break;
833 }
834
835 LLVMBasicBlockRef Prev = LLVMGetPreviousBasicBlock(Next);
Amaury Sechetd01c8612016-02-14 10:06:34 +0000836 if (Prev != Cur)
837 report_fatal_error("Next.Previous basic bloc is not Current");
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000838
Amaury Secheta82042e2016-02-09 22:36:41 +0000839 Cur = Next;
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000840 }
841
Amaury Sechetd01c8612016-02-14 10:06:34 +0000842 if (Count != 0)
843 report_fatal_error("Basic block count does not match iterration");
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000844 }
Amaury Secheta82042e2016-02-09 22:36:41 +0000845};
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000846
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000847static void declare_symbols(LLVMModuleRef Src, LLVMModuleRef M) {
Robert Widmann360d6e32018-05-20 23:49:08 +0000848 auto Ctx = LLVMGetModuleContext(M);
849
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000850 LLVMValueRef Begin = LLVMGetFirstGlobal(Src);
851 LLVMValueRef End = LLVMGetLastGlobal(Src);
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000852
853 LLVMValueRef Cur = Begin;
854 LLVMValueRef Next = nullptr;
Amaury Sechet58946a92016-02-17 22:30:05 +0000855 if (!Begin) {
856 if (End != nullptr)
Hiroshi Inouee9dea6e2017-07-13 06:48:39 +0000857 report_fatal_error("Range has an end but no beginning");
Amaury Sechet58946a92016-02-17 22:30:05 +0000858 goto FunDecl;
859 }
860
Amaury Sechete8ea7d82016-02-04 23:26:19 +0000861 while (true) {
Robert Widmann025c78f2018-05-19 15:08:36 +0000862 size_t NameLen;
863 const char *Name = LLVMGetValueName2(Cur, &NameLen);
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000864 if (LLVMGetNamedGlobal(M, Name))
865 report_fatal_error("GlobalVariable already cloned");
866 LLVMAddGlobal(M, LLVMGetElementType(TypeCloner(M).Clone(Cur)), Name);
867
868 Next = LLVMGetNextGlobal(Cur);
869 if (Next == nullptr) {
870 if (Cur != End)
871 report_fatal_error("");
872 break;
873 }
874
875 LLVMValueRef Prev = LLVMGetPreviousGlobal(Next);
876 if (Prev != Cur)
877 report_fatal_error("Next.Previous global is not Current");
878
879 Cur = Next;
880 }
881
Amaury Sechet58946a92016-02-17 22:30:05 +0000882FunDecl:
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000883 Begin = LLVMGetFirstFunction(Src);
884 End = LLVMGetLastFunction(Src);
885 if (!Begin) {
886 if (End != nullptr)
Hiroshi Inouee9dea6e2017-07-13 06:48:39 +0000887 report_fatal_error("Range has an end but no beginning");
Robert Widmann360d6e32018-05-20 23:49:08 +0000888 goto AliasDecl;
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000889 }
890
891 Cur = Begin;
892 Next = nullptr;
893 while (true) {
Robert Widmann025c78f2018-05-19 15:08:36 +0000894 size_t NameLen;
895 const char *Name = LLVMGetValueName2(Cur, &NameLen);
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000896 if (LLVMGetNamedFunction(M, Name))
897 report_fatal_error("Function already cloned");
Amaury Sechet5db224e2016-06-12 06:17:24 +0000898 auto Ty = LLVMGetElementType(TypeCloner(M).Clone(Cur));
899 auto F = LLVMAddFunction(M, Name, Ty);
900
901 // Copy attributes
902 for (int i = LLVMAttributeFunctionIndex, c = LLVMCountParams(F);
903 i <= c; ++i) {
904 for (unsigned k = 0, e = LLVMGetLastEnumAttributeKind(); k < e; ++k) {
905 if (auto SrcA = LLVMGetEnumAttributeAtIndex(Cur, i, k)) {
906 auto Val = LLVMGetEnumAttributeValue(SrcA);
907 auto DstA = LLVMCreateEnumAttribute(Ctx, k, Val);
908 LLVMAddAttributeAtIndex(F, i, DstA);
909 }
910 }
911 }
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000912
Amaury Sechetf64e8342016-02-16 07:08:49 +0000913 Next = LLVMGetNextFunction(Cur);
914 if (Next == nullptr) {
915 if (Cur != End)
916 report_fatal_error("Last function does not match End");
917 break;
918 }
919
920 LLVMValueRef Prev = LLVMGetPreviousFunction(Next);
921 if (Prev != Cur)
922 report_fatal_error("Next.Previous function is not Current");
923
924 Cur = Next;
925 }
Robert Widmann360d6e32018-05-20 23:49:08 +0000926
927AliasDecl:
928 Begin = LLVMGetFirstGlobalAlias(Src);
929 End = LLVMGetLastGlobalAlias(Src);
930 if (!Begin) {
931 if (End != nullptr)
932 report_fatal_error("Range has an end but no beginning");
Robert Widmann0a35b762018-08-30 17:09:43 +0000933 goto NamedMDDecl;
Robert Widmann360d6e32018-05-20 23:49:08 +0000934 }
935
936 Cur = Begin;
937 Next = nullptr;
938 while (true) {
939 size_t NameLen;
940 const char *Name = LLVMGetValueName2(Cur, &NameLen);
941 if (LLVMGetNamedGlobalAlias(M, Name, NameLen))
942 report_fatal_error("Global alias already cloned");
943 LLVMTypeRef CurType = TypeCloner(M).Clone(Cur);
944 // FIXME: Allow NULL aliasee.
945 LLVMAddAlias(M, CurType, LLVMGetUndef(CurType), Name);
946
947 Next = LLVMGetNextGlobalAlias(Cur);
948 if (Next == nullptr) {
949 if (Cur != End)
950 report_fatal_error("");
951 break;
952 }
953
954 LLVMValueRef Prev = LLVMGetPreviousGlobalAlias(Next);
955 if (Prev != Cur)
956 report_fatal_error("Next.Previous global is not Current");
957
958 Cur = Next;
959 }
Robert Widmann0a35b762018-08-30 17:09:43 +0000960
961NamedMDDecl:
962 LLVMNamedMDNodeRef BeginMD = LLVMGetFirstNamedMetadata(Src);
963 LLVMNamedMDNodeRef EndMD = LLVMGetLastNamedMetadata(Src);
964 if (!BeginMD) {
965 if (EndMD != nullptr)
966 report_fatal_error("Range has an end but no beginning");
967 return;
968 }
969
970 LLVMNamedMDNodeRef CurMD = BeginMD;
971 LLVMNamedMDNodeRef NextMD = nullptr;
972 while (true) {
973 size_t NameLen;
974 const char *Name = LLVMGetNamedMetadataName(CurMD, &NameLen);
975 if (LLVMGetNamedMetadata(M, Name, NameLen))
976 report_fatal_error("Named Metadata Node already cloned");
977 LLVMGetOrInsertNamedMetadata(M, Name, NameLen);
978
979 NextMD = LLVMGetNextNamedMetadata(CurMD);
980 if (NextMD == nullptr) {
981 if (CurMD != EndMD)
982 report_fatal_error("");
983 break;
984 }
985
986 LLVMNamedMDNodeRef PrevMD = LLVMGetPreviousNamedMetadata(NextMD);
987 if (PrevMD != CurMD)
988 report_fatal_error("Next.Previous global is not Current");
989
990 CurMD = NextMD;
991 }
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000992}
Amaury Sechetf64e8342016-02-16 07:08:49 +0000993
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000994static void clone_symbols(LLVMModuleRef Src, LLVMModuleRef M) {
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000995 LLVMValueRef Begin = LLVMGetFirstGlobal(Src);
996 LLVMValueRef End = LLVMGetLastGlobal(Src);
Amaury Sechete8ba2bf2016-02-17 22:13:33 +0000997
998 LLVMValueRef Cur = Begin;
999 LLVMValueRef Next = nullptr;
Amaury Sechet58946a92016-02-17 22:30:05 +00001000 if (!Begin) {
1001 if (End != nullptr)
Hiroshi Inouee9dea6e2017-07-13 06:48:39 +00001002 report_fatal_error("Range has an end but no beginning");
Amaury Sechet58946a92016-02-17 22:30:05 +00001003 goto FunClone;
1004 }
1005
Amaury Sechete8ba2bf2016-02-17 22:13:33 +00001006 while (true) {
Robert Widmann025c78f2018-05-19 15:08:36 +00001007 size_t NameLen;
1008 const char *Name = LLVMGetValueName2(Cur, &NameLen);
Amaury Sechete8ba2bf2016-02-17 22:13:33 +00001009 LLVMValueRef G = LLVMGetNamedGlobal(M, Name);
1010 if (!G)
1011 report_fatal_error("GlobalVariable must have been declared already");
1012
1013 if (auto I = LLVMGetInitializer(Cur))
1014 LLVMSetInitializer(G, clone_constant(I, M));
1015
Robert Widmann9cba4ec2018-09-28 15:35:18 +00001016 size_t NumMetadataEntries;
1017 auto *AllMetadata = LLVMGlobalCopyAllMetadata(Cur, &NumMetadataEntries);
1018 for (unsigned i = 0; i < NumMetadataEntries; ++i) {
1019 unsigned Kind = LLVMValueMetadataEntriesGetKind(AllMetadata, i);
1020 LLVMMetadataRef MD = LLVMValueMetadataEntriesGetMetadata(AllMetadata, i);
1021 LLVMGlobalSetMetadata(G, Kind, MD);
1022 }
1023 LLVMDisposeValueMetadataEntries(AllMetadata);
1024
Amaury Sechete8ba2bf2016-02-17 22:13:33 +00001025 LLVMSetGlobalConstant(G, LLVMIsGlobalConstant(Cur));
1026 LLVMSetThreadLocal(G, LLVMIsThreadLocal(Cur));
1027 LLVMSetExternallyInitialized(G, LLVMIsExternallyInitialized(Cur));
1028 LLVMSetLinkage(G, LLVMGetLinkage(Cur));
1029 LLVMSetSection(G, LLVMGetSection(Cur));
1030 LLVMSetVisibility(G, LLVMGetVisibility(Cur));
Robert Widmann4bb481b2018-03-14 06:45:51 +00001031 LLVMSetUnnamedAddress(G, LLVMGetUnnamedAddress(Cur));
Amaury Sechete8ba2bf2016-02-17 22:13:33 +00001032 LLVMSetAlignment(G, LLVMGetAlignment(Cur));
1033
1034 Next = LLVMGetNextGlobal(Cur);
1035 if (Next == nullptr) {
1036 if (Cur != End)
1037 report_fatal_error("");
1038 break;
1039 }
1040
1041 LLVMValueRef Prev = LLVMGetPreviousGlobal(Next);
1042 if (Prev != Cur)
1043 report_fatal_error("Next.Previous global is not Current");
1044
1045 Cur = Next;
1046 }
1047
Amaury Sechet58946a92016-02-17 22:30:05 +00001048FunClone:
Amaury Sechete8ba2bf2016-02-17 22:13:33 +00001049 Begin = LLVMGetFirstFunction(Src);
1050 End = LLVMGetLastFunction(Src);
1051 if (!Begin) {
1052 if (End != nullptr)
Hiroshi Inouee9dea6e2017-07-13 06:48:39 +00001053 report_fatal_error("Range has an end but no beginning");
Robert Widmann360d6e32018-05-20 23:49:08 +00001054 goto AliasClone;
Amaury Sechete8ba2bf2016-02-17 22:13:33 +00001055 }
1056
Amaury Sechetf64e8342016-02-16 07:08:49 +00001057 Cur = Begin;
1058 Next = nullptr;
1059 while (true) {
Robert Widmann025c78f2018-05-19 15:08:36 +00001060 size_t NameLen;
1061 const char *Name = LLVMGetValueName2(Cur, &NameLen);
Amaury Sechete8ba2bf2016-02-17 22:13:33 +00001062 LLVMValueRef Fun = LLVMGetNamedFunction(M, Name);
1063 if (!Fun)
1064 report_fatal_error("Function must have been declared already");
Amaury Sechete39e8532016-02-18 20:38:32 +00001065
1066 if (LLVMHasPersonalityFn(Cur)) {
Robert Widmann025c78f2018-05-19 15:08:36 +00001067 size_t FNameLen;
1068 const char *FName = LLVMGetValueName2(LLVMGetPersonalityFn(Cur),
1069 &FNameLen);
Amaury Sechete39e8532016-02-18 20:38:32 +00001070 LLVMValueRef P = LLVMGetNamedFunction(M, FName);
1071 if (!P)
1072 report_fatal_error("Could not find personality function");
1073 LLVMSetPersonalityFn(Fun, P);
1074 }
1075
Robert Widmann9cba4ec2018-09-28 15:35:18 +00001076 size_t NumMetadataEntries;
1077 auto *AllMetadata = LLVMGlobalCopyAllMetadata(Cur, &NumMetadataEntries);
1078 for (unsigned i = 0; i < NumMetadataEntries; ++i) {
1079 unsigned Kind = LLVMValueMetadataEntriesGetKind(AllMetadata, i);
1080 LLVMMetadataRef MD = LLVMValueMetadataEntriesGetMetadata(AllMetadata, i);
1081 LLVMGlobalSetMetadata(Fun, Kind, MD);
1082 }
1083 LLVMDisposeValueMetadataEntries(AllMetadata);
1084
Amaury Sechete8ba2bf2016-02-17 22:13:33 +00001085 FunCloner FC(Cur, Fun);
1086 FC.CloneBBs(Cur);
1087
Amaury Sechete8ea7d82016-02-04 23:26:19 +00001088 Next = LLVMGetNextFunction(Cur);
1089 if (Next == nullptr) {
Amaury Sechetd01c8612016-02-14 10:06:34 +00001090 if (Cur != End)
1091 report_fatal_error("Last function does not match End");
Amaury Sechete8ea7d82016-02-04 23:26:19 +00001092 break;
1093 }
1094
1095 LLVMValueRef Prev = LLVMGetPreviousFunction(Next);
Amaury Sechetd01c8612016-02-14 10:06:34 +00001096 if (Prev != Cur)
1097 report_fatal_error("Next.Previous function is not Current");
Amaury Sechete8ea7d82016-02-04 23:26:19 +00001098
1099 Cur = Next;
1100 }
Robert Widmann360d6e32018-05-20 23:49:08 +00001101
1102AliasClone:
1103 Begin = LLVMGetFirstGlobalAlias(Src);
1104 End = LLVMGetLastGlobalAlias(Src);
1105 if (!Begin) {
1106 if (End != nullptr)
1107 report_fatal_error("Range has an end but no beginning");
Robert Widmann0a35b762018-08-30 17:09:43 +00001108 goto NamedMDClone;
Robert Widmann360d6e32018-05-20 23:49:08 +00001109 }
1110
1111 Cur = Begin;
1112 Next = nullptr;
1113 while (true) {
1114 size_t NameLen;
1115 const char *Name = LLVMGetValueName2(Cur, &NameLen);
1116 LLVMValueRef Alias = LLVMGetNamedGlobalAlias(M, Name, NameLen);
1117 if (!Alias)
1118 report_fatal_error("Global alias must have been declared already");
1119
1120 if (LLVMValueRef Aliasee = LLVMAliasGetAliasee(Cur)) {
1121 LLVMAliasSetAliasee(Alias, clone_constant(Aliasee, M));
1122 }
1123
1124 LLVMSetLinkage(Alias, LLVMGetLinkage(Cur));
1125 LLVMSetUnnamedAddress(Alias, LLVMGetUnnamedAddress(Cur));
1126
1127 Next = LLVMGetNextGlobalAlias(Cur);
1128 if (Next == nullptr) {
1129 if (Cur != End)
1130 report_fatal_error("Last global alias does not match End");
1131 break;
1132 }
1133
1134 LLVMValueRef Prev = LLVMGetPreviousGlobalAlias(Next);
1135 if (Prev != Cur)
1136 report_fatal_error("Next.Previous global alias is not Current");
1137
1138 Cur = Next;
1139 }
Robert Widmann0a35b762018-08-30 17:09:43 +00001140
1141NamedMDClone:
1142 LLVMNamedMDNodeRef BeginMD = LLVMGetFirstNamedMetadata(Src);
1143 LLVMNamedMDNodeRef EndMD = LLVMGetLastNamedMetadata(Src);
1144 if (!BeginMD) {
1145 if (EndMD != nullptr)
1146 report_fatal_error("Range has an end but no beginning");
1147 return;
1148 }
1149
1150 LLVMNamedMDNodeRef CurMD = BeginMD;
1151 LLVMNamedMDNodeRef NextMD = nullptr;
1152 while (true) {
1153 size_t NameLen;
1154 const char *Name = LLVMGetNamedMetadataName(CurMD, &NameLen);
1155 LLVMNamedMDNodeRef NamedMD = LLVMGetNamedMetadata(M, Name, NameLen);
1156 if (!NamedMD)
1157 report_fatal_error("Named MD Node must have been declared already");
1158
1159 unsigned OperandCount = LLVMGetNamedMetadataNumOperands(Src, Name);
1160 LLVMValueRef *OperandBuf = static_cast<LLVMValueRef *>(
1161 safe_malloc(OperandCount * sizeof(LLVMValueRef)));
1162 LLVMGetNamedMetadataOperands(Src, Name, OperandBuf);
1163 for (unsigned i = 0, e = OperandCount; i != e; ++i) {
1164 LLVMAddNamedMetadataOperand(M, Name, OperandBuf[i]);
1165 }
1166 free(OperandBuf);
1167
1168 NextMD = LLVMGetNextNamedMetadata(CurMD);
1169 if (NextMD == nullptr) {
1170 if (CurMD != EndMD)
1171 report_fatal_error("Last Named MD Node does not match End");
1172 break;
1173 }
1174
1175 LLVMNamedMDNodeRef PrevMD = LLVMGetPreviousNamedMetadata(NextMD);
1176 if (PrevMD != CurMD)
1177 report_fatal_error("Next.Previous Named MD Node is not Current");
1178
1179 CurMD = NextMD;
1180 }
Amaury Sechete8ea7d82016-02-04 23:26:19 +00001181}
1182
Benjamin Kramer9a3bd232016-02-05 13:31:14 +00001183int llvm_echo(void) {
Amaury Sechete8ea7d82016-02-04 23:26:19 +00001184 LLVMEnablePrettyStackTrace();
1185
Benjamin Kramer9a3bd232016-02-05 13:31:14 +00001186 LLVMModuleRef Src = llvm_load_module(false, true);
Robert Widmann490a5802018-01-30 21:34:29 +00001187 size_t SourceFileLen;
1188 const char *SourceFileName = LLVMGetSourceFileName(Src, &SourceFileLen);
1189 size_t ModuleIdentLen;
1190 const char *ModuleName = LLVMGetModuleIdentifier(Src, &ModuleIdentLen);
Amaury Sechete8ea7d82016-02-04 23:26:19 +00001191 LLVMContextRef Ctx = LLVMContextCreate();
Peter Zotov0a2fa0a2016-04-05 13:56:59 +00001192 LLVMModuleRef M = LLVMModuleCreateWithNameInContext(ModuleName, Ctx);
1193
Robert Widmann490a5802018-01-30 21:34:29 +00001194 LLVMSetSourceFileName(M, SourceFileName, SourceFileLen);
1195 LLVMSetModuleIdentifier(M, ModuleName, ModuleIdentLen);
Amaury Sechete8ea7d82016-02-04 23:26:19 +00001196
Amaury Sechet55909672016-02-16 05:11:24 +00001197 LLVMSetTarget(M, LLVMGetTarget(Src));
1198 LLVMSetModuleDataLayout(M, LLVMGetModuleDataLayout(Src));
1199 if (strcmp(LLVMGetDataLayoutStr(M), LLVMGetDataLayoutStr(Src)))
1200 report_fatal_error("Inconsistent DataLayout string representation");
1201
Robert Widmannf108d572018-04-06 02:31:29 +00001202 size_t ModuleInlineAsmLen;
1203 const char *ModuleAsm = LLVMGetModuleInlineAsm(Src, &ModuleInlineAsmLen);
1204 LLVMSetModuleInlineAsm2(M, ModuleAsm, ModuleInlineAsmLen);
1205
Amaury Sechete8ba2bf2016-02-17 22:13:33 +00001206 declare_symbols(Src, M);
1207 clone_symbols(Src, M);
Amaury Sechetc679dbd2016-02-14 09:30:42 +00001208 char *Str = LLVMPrintModuleToString(M);
Amaury Sechete8ea7d82016-02-04 23:26:19 +00001209 fputs(Str, stdout);
1210
1211 LLVMDisposeMessage(Str);
Rafael Espindola92f35782018-02-21 19:55:11 +00001212 LLVMDisposeModule(Src);
Amaury Sechetc679dbd2016-02-14 09:30:42 +00001213 LLVMDisposeModule(M);
Amaury Sechete8ea7d82016-02-04 23:26:19 +00001214 LLVMContextDispose(Ctx);
1215
1216 return 0;
1217}