Add C API bindings for DIBuilder 'Type' APIs
[lldb.git] / llvm / tools / llvm-c-test / attributes.c
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
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
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 }