1/*===-- attributes.c - tool for testing libLLVM and llvm-c API ------------===*\
2|* *|
3|* Part of the LLVM Project, under the Apache License v2.0 with LLVM *|
4|* Exceptions. *|
5|* See https://llvm.org/LICENSE.txt for license information. *|
6|* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception *|
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
20int llvm_test_function_attributes(void) {
21 LLVMEnablePrettyStackTrace();
22
23 LLVMModuleRef M = llvm_load_module(C: LLVMGetGlobalContext(), false, true);
24
25 LLVMValueRef F = LLVMGetFirstFunction(M);
26 while (F) {
27 // Read attributes
28 int Idx, ParamCount;
29 for (Idx = LLVMAttributeFunctionIndex, ParamCount = LLVMCountParams(Fn: F);
30 Idx <= ParamCount; ++Idx) {
31 int AttrCount = LLVMGetAttributeCountAtIndex(F, Idx);
32 LLVMAttributeRef *Attrs = 0;
33 if (AttrCount) {
34 Attrs =
35 (LLVMAttributeRef *)malloc(size: AttrCount * sizeof(LLVMAttributeRef));
36 assert(Attrs);
37 }
38 LLVMGetAttributesAtIndex(F, Idx, Attrs);
39 free(ptr: Attrs);
40 }
41 F = LLVMGetNextFunction(Fn: F);
42 }
43
44 LLVMDisposeModule(M);
45
46 return 0;
47}
48
49int llvm_test_callsite_attributes(void) {
50 LLVMEnablePrettyStackTrace();
51
52 LLVMModuleRef M = llvm_load_module(C: LLVMGetGlobalContext(), false, true);
53
54 LLVMValueRef F = LLVMGetFirstFunction(M);
55 while (F) {
56 LLVMBasicBlockRef BB;
57 for (BB = LLVMGetFirstBasicBlock(Fn: F); BB; BB = LLVMGetNextBasicBlock(BB)) {
58 LLVMValueRef I;
59 for (I = LLVMGetFirstInstruction(BB); I; I = LLVMGetNextInstruction(Inst: I)) {
60 if (LLVMIsACallInst(Val: I)) {
61 // Read attributes
62 int Idx, ParamCount;
63 for (Idx = LLVMAttributeFunctionIndex,
64 ParamCount = LLVMCountParams(Fn: F);
65 Idx <= ParamCount; ++Idx) {
66 int AttrCount = LLVMGetCallSiteAttributeCount(C: I, Idx);
67 LLVMAttributeRef *Attrs = 0;
68 if (AttrCount) {
69 Attrs = (LLVMAttributeRef *)malloc(
70 size: AttrCount * sizeof(LLVMAttributeRef));
71 assert(Attrs);
72 }
73 LLVMGetCallSiteAttributes(C: I, Idx, Attrs);
74 free(ptr: Attrs);
75 }
76 }
77 }
78 }
79
80 F = LLVMGetNextFunction(Fn: F);
81 }
82
83 LLVMDisposeModule(M);
84
85 return 0;
86}
87