1 /*===-- attributes.c - 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 |* *|
10 |* This file implements the --test-attributes and --test-callsite-attributes *|
11 |* commands in llvm-c-test. *|
12 |* *|
13 \*===----------------------------------------------------------------------===*/
14
15 #include "llvm-c-test.h"
16
17 #include <assert.h>
18 #include <stdlib.h>
19
llvm_test_function_attributes(void)20 int llvm_test_function_attributes(void) {
21 LLVMEnablePrettyStackTrace();
22
23 LLVMModuleRef M = llvm_load_module(false, true);
24
25 LLVMValueRef F = LLVMGetFirstFunction(M);
26 while (F) {
27 // Read attributes
28 int Idx, ParamCount;
29 for (Idx = LLVMAttributeFunctionIndex, ParamCount = LLVMCountParams(F);
30 Idx <= ParamCount; ++Idx) {
31 int AttrCount = LLVMGetAttributeCountAtIndex(F, Idx);
32 LLVMAttributeRef *Attrs =
33 (LLVMAttributeRef *)malloc(AttrCount * sizeof(LLVMAttributeRef));
34 assert(Attrs);
35 LLVMGetAttributesAtIndex(F, Idx, Attrs);
36 free(Attrs);
37 }
38 F = LLVMGetNextFunction(F);
39 }
40
41 LLVMDisposeModule(M);
42
43 return 0;
44 }
45
llvm_test_callsite_attributes(void)46 int llvm_test_callsite_attributes(void) {
47 LLVMEnablePrettyStackTrace();
48
49 LLVMModuleRef M = llvm_load_module(false, true);
50
51 LLVMValueRef F = LLVMGetFirstFunction(M);
52 while (F) {
53 LLVMBasicBlockRef BB;
54 for (BB = LLVMGetFirstBasicBlock(F); BB; BB = LLVMGetNextBasicBlock(BB)) {
55 LLVMValueRef I;
56 for (I = LLVMGetFirstInstruction(BB); I; I = LLVMGetNextInstruction(I)) {
57 if (LLVMIsACallInst(I)) {
58 // Read attributes
59 int Idx, ParamCount;
60 for (Idx = LLVMAttributeFunctionIndex,
61 ParamCount = LLVMCountParams(F);
62 Idx <= ParamCount; ++Idx) {
63 int AttrCount = LLVMGetCallSiteAttributeCount(I, Idx);
64 LLVMAttributeRef *Attrs = (LLVMAttributeRef *)malloc(
65 AttrCount * sizeof(LLVMAttributeRef));
66 assert(Attrs);
67 LLVMGetCallSiteAttributes(I, Idx, Attrs);
68 free(Attrs);
69 }
70 }
71 }
72 }
73
74 F = LLVMGetNextFunction(F);
75 }
76
77 LLVMDisposeModule(M);
78
79 return 0;
80 }
81