1 /**
2 * Copyright (c) 2024-2025 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "libabckit/include/c/abckit.h"
17 #include "libabckit/include/c/isa/isa_static.h"
18 #include "libabckit/include/c/metadata_core.h"
19
20 #include "helpers/helpers_runtime.h"
21 #include "helpers/helpers.h"
22
23 #include <gtest/gtest.h>
24
25 // NOLINTBEGIN(readability-magic-numbers)
26 namespace libabckit::test {
27
28 namespace {
29 auto g_impl = AbckitGetApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
30 auto g_implI = AbckitGetInspectApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
31 auto g_implM = AbckitGetModifyApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
32 auto g_implG = AbckitGetGraphApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
33 auto g_statG = AbckitGetIsaApiStaticImpl(ABCKIT_VERSION_RELEASE_1_0_0);
34
CreateBBSchema()35 std::vector<helpers::BBSchema<AbckitIsaApiStaticOpcode>> CreateBBSchema()
36 {
37 helpers::BBSchema<AbckitIsaApiStaticOpcode> bb1({{}, {1}, {{}}});
38 std::vector<helpers::InstSchema<AbckitIsaApiStaticOpcode>> insts({
39 {0, ABCKIT_ISA_API_STATIC_OPCODE_INITOBJECT, {}},
40 {2, ABCKIT_ISA_API_STATIC_OPCODE_ISINSTANCE, {0}},
41 {1, ABCKIT_ISA_API_STATIC_OPCODE_RETURN, {2}},
42 });
43 helpers::BBSchema<AbckitIsaApiStaticOpcode> bb2({{0}, {2}, insts});
44 helpers::BBSchema<AbckitIsaApiStaticOpcode> bb3({{1}, {}, {}});
45 return {bb1, bb2, bb3};
46 }
47
IsetClass(AbckitGraph * graph,AbckitFile * file,AbckitCoreClass * klass,AbckitCoreClass * startKlass)48 void IsetClass(AbckitGraph *graph, AbckitFile *file, AbckitCoreClass *klass, AbckitCoreClass *startKlass)
49 {
50 AbckitInst *initObj = helpers::FindFirstInst(graph, ABCKIT_ISA_API_STATIC_OPCODE_INITOBJECT);
51 ASSERT_NE(initObj, nullptr);
52 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
53
54 AbckitInst *ret = helpers::FindFirstInst(graph, ABCKIT_ISA_API_STATIC_OPCODE_RETURN);
55 ASSERT_NE(ret, nullptr);
56 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
57
58 auto *type = g_implM->createReferenceType(file, startKlass);
59 ASSERT_NE(type, nullptr);
60 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
61
62 AbckitInst *isInstance = g_statG->iCreateIsInstance(graph, initObj, type);
63 ASSERT_NE(isInstance, nullptr);
64 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
65
66 g_statG->iSetClass(isInstance, klass);
67 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
68
69 g_implG->iInsertAfter(isInstance, initObj);
70 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
71
72 helpers::ReplaceInst(ret, g_statG->iCreateReturn(graph, isInstance));
73 }
74
IgetClass(AbckitGraph * graph)75 AbckitCoreClass *IgetClass(AbckitGraph *graph)
76 {
77 AbckitInst *inst = helpers::FindFirstInst(graph, ABCKIT_ISA_API_STATIC_OPCODE_ISINSTANCE);
78 EXPECT_TRUE(inst != nullptr);
79 EXPECT_TRUE(g_impl->getLastError() == ABCKIT_STATUS_NO_ERROR);
80
81 auto *klass = g_statG->iGetClass(inst);
82 EXPECT_TRUE(klass != nullptr);
83 EXPECT_TRUE(g_impl->getLastError() == ABCKIT_STATUS_NO_ERROR);
84 return klass;
85 }
86
TransformIrForIsetClass(AbckitFile * file,AbckitCoreFunction * method,AbckitGraph * graph)87 void TransformIrForIsetClass(AbckitFile *file, AbckitCoreFunction *method, AbckitGraph *graph)
88 {
89 auto *module = g_implI->functionGetModule(method);
90 helpers::ClassByNameContext classCtxFinder = {nullptr, "A"};
91 g_implI->moduleEnumerateClasses(module, &classCtxFinder, helpers::ClassByNameFinder);
92 ASSERT_NE(classCtxFinder.klass, nullptr);
93 helpers::ClassByNameContext startClassCtxFinder = {nullptr, "B"};
94 g_implI->moduleEnumerateClasses(module, &startClassCtxFinder, helpers::ClassByNameFinder);
95 ASSERT_NE(startClassCtxFinder.klass, nullptr);
96 IsetClass(graph, file, classCtxFinder.klass, startClassCtxFinder.klass);
97 }
98
TransformIrForIgetClass(AbckitFile * file,AbckitCoreFunction * method,AbckitGraph * graph)99 void TransformIrForIgetClass(AbckitFile *file, AbckitCoreFunction *method, AbckitGraph *graph)
100 {
101 auto *module = g_implI->functionGetModule(method);
102 helpers::ClassByNameContext classCtxFinder = {nullptr, "A"};
103 g_implI->moduleEnumerateClasses(module, &classCtxFinder, helpers::ClassByNameFinder);
104 ASSERT_NE(classCtxFinder.klass, nullptr);
105 helpers::ClassByNameContext startClassCtxFinder = {nullptr, "B"};
106 g_implI->moduleEnumerateClasses(module, &startClassCtxFinder, helpers::ClassByNameFinder);
107 ASSERT_NE(startClassCtxFinder.klass, nullptr);
108 IsetClass(graph, file, classCtxFinder.klass, startClassCtxFinder.klass);
109 auto *klass = IgetClass(graph);
110 ASSERT_EQ(klass, classCtxFinder.klass);
111 }
112
TemplateTestFunction(const std::string & inputResult,const std::string & outputResult,const std::string & funcName,void (* irTransformer)(AbckitFile * file,AbckitCoreFunction * method,AbckitGraph * graph))113 void TemplateTestFunction(const std::string &inputResult, const std::string &outputResult, const std::string &funcName,
114 void (*irTransformer)(AbckitFile *file, AbckitCoreFunction *method, AbckitGraph *graph))
115 {
116 auto output = helpers::ExecuteStaticAbc(ABCKIT_ABC_DIR "ut/isa/isa_static/classes/classes_api.abc",
117 "classes_api/ETSGLOBAL", "main");
118 EXPECT_TRUE(helpers::Match(output, inputResult));
119
120 helpers::TransformMethod(
121 ABCKIT_ABC_DIR "ut/isa/isa_static/classes/classes_api.abc",
122 ABCKIT_ABC_DIR "ut/isa/isa_static/classes/classes_api_modified.abc", funcName,
123 [&irTransformer](AbckitFile *file, AbckitCoreFunction *method, AbckitGraph *graph) {
124 irTransformer(file, method, graph);
125 },
126 [](AbckitGraph *graph) {
127 const auto &bbSchemas = CreateBBSchema();
128 helpers::VerifyGraph(graph, bbSchemas);
129 });
130
131 output = helpers::ExecuteStaticAbc(ABCKIT_ABC_DIR "ut/isa/isa_static/classes/classes_api_modified.abc",
132 "classes_api/ETSGLOBAL", "main");
133 EXPECT_TRUE(helpers::Match(output, outputResult));
134 }
135 } // namespace
136
137 class LibAbcKitClassesAPITest : public ::testing::Test {};
138
139 // Test: test-kind=api, api=IsaApiStaticImpl::iSetClass, abc-kind=ArkTS2, category=positive, extension=c
TEST_F(LibAbcKitClassesAPITest,LibAbcKitTestIsetClassIsInstanceTrue)140 TEST_F(LibAbcKitClassesAPITest, LibAbcKitTestIsetClassIsInstanceTrue)
141 {
142 TemplateTestFunction("false\ntrue\n", "true\ntrue\n", "foo", TransformIrForIsetClass);
143 }
144
145 // Test: test-kind=api, api=IsaApiStaticImpl::iSetClass, abc-kind=ArkTS2, category=positive, extension=c
TEST_F(LibAbcKitClassesAPITest,LibAbcKitTestIsetClassIsInstanceFalse)146 TEST_F(LibAbcKitClassesAPITest, LibAbcKitTestIsetClassIsInstanceFalse)
147 {
148 TemplateTestFunction("false\ntrue\n", "false\nfalse\n", "bar", TransformIrForIsetClass);
149 }
150
151 // Test: test-kind=api, api=IsaApiStaticImpl::iGetClass, abc-kind=ArkTS2, category=positive, extension=c
TEST_F(LibAbcKitClassesAPITest,LibAbcKitTestIgetClassIsInstance)152 TEST_F(LibAbcKitClassesAPITest, LibAbcKitTestIgetClassIsInstance)
153 {
154 TemplateTestFunction("false\ntrue\n", "true\ntrue\n", "foo", TransformIrForIgetClass);
155 }
156
157 } // namespace libabckit::test
158 // NOLINTEND(readability-magic-numbers)
159