1 /**
2 * Copyright (c) 2024 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 "logger.h"
18 #include "libabckit/include/c/metadata_core.h"
19 #include "libabckit/include/c/ir_core.h"
20 #include "ir_impl.h"
21
22 #include "helpers/helpers.h"
23 #include "helpers/helpers_runtime.h"
24
25 #include <gtest/gtest.h>
26
27 // NOLINTBEGIN(readability-magic-numbers)
28 namespace libabckit::test {
29
30 static auto g_impl = AbckitGetApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
31 static auto g_implI = AbckitGetInspectApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
32 static auto g_implM = AbckitGetModifyApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
33 static auto g_implG = AbckitGetGraphApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
34 static auto g_statG = AbckitGetIsaApiStaticImpl(ABCKIT_VERSION_RELEASE_1_0_0);
35
36 class LibAbcKitICreateObjectsStaticTest : public ::testing::Test {};
37
TransformIrCreateNewObject(AbckitGraph * graph)38 static void TransformIrCreateNewObject(AbckitGraph *graph)
39 {
40 helpers::ModuleByNameContext ctxFinder = {nullptr, "objects"};
41 g_implI->fileEnumerateModules(graph->file, &ctxFinder, helpers::ModuleByNameFinder);
42 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
43 ASSERT_NE(ctxFinder.module, nullptr);
44
45 helpers::ClassByNameContext ctxClassFinder = {nullptr, "B"};
46 g_implI->moduleEnumerateClasses(ctxFinder.module, &ctxClassFinder, helpers::ClassByNameFinder);
47 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
48 ASSERT_NE(ctxClassFinder.klass, nullptr);
49
50 auto method = helpers::FindMethodByName(graph->file, "bar");
51
52 AbckitInst *newObjectInst = g_statG->iCreateNewObject(graph, ctxClassFinder.klass);
53 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
54
55 AbckitBasicBlock *mainBB = g_implG->bbGetSuccBlock(g_implG->gGetStartBasicBlock(graph), 0);
56 AbckitInst *lastInst = g_implG->bbGetLastInst(mainBB);
57
58 g_implG->iInsertBefore(newObjectInst, lastInst);
59 AbckitInst *callInst = g_statG->iCreateCallStatic(graph, method, 1, newObjectInst);
60 g_implG->iInsertAfter(callInst, newObjectInst);
61 }
62
GetInitObjectInst(AbckitGraph * graph,AbckitInst * constantInst,size_t numArg)63 static AbckitInst *GetInitObjectInst(AbckitGraph *graph, AbckitInst *constantInst, size_t numArg)
64 {
65 switch (numArg) {
66 case 0: {
67 auto method1 = helpers::FindMethodByName(graph->file, "_ctor_:objects.C;void;");
68 EXPECT_NE(method1, nullptr);
69 return g_statG->iCreateInitObject(graph, method1, numArg);
70 }
71 case 1: {
72 auto method1 = helpers::FindMethodByName(graph->file, "_ctor_:objects.C;i32;void;");
73 EXPECT_NE(method1, nullptr);
74 return g_statG->iCreateInitObject(graph, method1, numArg, constantInst);
75 }
76 case 2U: {
77 auto method1 = helpers::FindMethodByName(graph->file, "_ctor_:objects.C;i32;i32;void;");
78 EXPECT_NE(method1, nullptr);
79 return g_statG->iCreateInitObject(graph, method1, numArg, constantInst, constantInst);
80 }
81 case 3U: {
82 auto method1 = helpers::FindMethodByName(graph->file, "_ctor_:objects.C;i32;i32;i32;void;");
83 EXPECT_NE(method1, nullptr);
84 return g_statG->iCreateInitObject(graph, method1, numArg, constantInst, constantInst, constantInst);
85 }
86 case 4U: {
87 auto method1 = helpers::FindMethodByName(graph->file, "_ctor_:objects.C;i32;i32;i32;i32;void;");
88 EXPECT_NE(method1, nullptr);
89 return g_statG->iCreateInitObject(graph, method1, numArg, constantInst, constantInst, constantInst,
90 constantInst);
91 }
92 case 5U: {
93 auto method1 = helpers::FindMethodByName(graph->file, "_ctor_:objects.C;i32;i32;i32;i32;i32;void;");
94 EXPECT_NE(method1, nullptr);
95 return g_statG->iCreateInitObject(graph, method1, numArg, constantInst, constantInst, constantInst,
96 constantInst, constantInst);
97 }
98 default:
99 LIBABCKIT_UNREACHABLE_TEST(DEBUG);
100 }
101 EXPECT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
102 }
103
TransformIrCreateInitObject(AbckitGraph * graph,size_t numArg)104 static void TransformIrCreateInitObject(AbckitGraph *graph, size_t numArg)
105 {
106 helpers::ModuleByNameContext ctxFinder = {nullptr, "objects"};
107 g_implI->fileEnumerateModules(graph->file, &ctxFinder, helpers::ModuleByNameFinder);
108 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
109 ASSERT_NE(ctxFinder.module, nullptr);
110
111 helpers::ClassByNameContext ctxClassFinder = {nullptr, "C"};
112 g_implI->moduleEnumerateClasses(ctxFinder.module, &ctxClassFinder, helpers::ClassByNameFinder);
113 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
114 ASSERT_NE(ctxClassFinder.klass, nullptr);
115
116 AbckitBasicBlock *mainBB = g_implG->bbGetSuccBlock(g_implG->gGetStartBasicBlock(graph), 0);
117 AbckitInst *lastInst = g_implG->bbGetLastInst(mainBB);
118 AbckitInst *constantInst = numArg > 0 ? g_implG->gFindOrCreateConstantI64(graph, 0x1) : nullptr;
119 if (constantInst != nullptr) {
120 g_implG->iInsertBefore(constantInst, lastInst);
121 }
122
123 g_implG->iInsertBefore(GetInitObjectInst(graph, constantInst, numArg), lastInst);
124 }
125
InitObjectTest(size_t numArg,helpers::InstSchema<AbckitIsaApiStaticOpcode> initObjectInst,const std::string & expectedOutput)126 void InitObjectTest(size_t numArg, helpers::InstSchema<AbckitIsaApiStaticOpcode> initObjectInst,
127 const std::string &expectedOutput)
128 {
129 auto output =
130 helpers::ExecuteStaticAbc(ABCKIT_ABC_DIR "ut/isa/isa_static/objects/objects.abc", "objects/ETSGLOBAL", "main");
131 EXPECT_TRUE(helpers::Match(output, "A\n"));
132
133 std::vector<helpers::InstSchema<AbckitIsaApiStaticOpcode>> startBB = {};
134 if (numArg != 0) {
135 startBB = {{0, ABCKIT_ISA_API_STATIC_OPCODE_CONSTANT, {}}};
136 }
137
138 helpers::TransformMethod(
139 ABCKIT_ABC_DIR "ut/isa/isa_static/objects/objects.abc",
140 ABCKIT_ABC_DIR "ut/isa/isa_static/objects/objects_modified.abc", "main",
141 [numArg](AbckitFile * /*file*/, AbckitCoreFunction * /*method*/, AbckitGraph *graph) {
142 TransformIrCreateInitObject(graph, numArg);
143 },
144 [&](AbckitGraph *graph) {
145 std::vector<helpers::BBSchema<AbckitIsaApiStaticOpcode>> bbSchema = {
146 {{}, {1}, startBB},
147 {
148 {0},
149 {2U},
150 {
151 {1, ABCKIT_ISA_API_STATIC_OPCODE_INITOBJECT, {}},
152 {4, ABCKIT_ISA_API_STATIC_OPCODE_CALL_VIRTUAL, {1}},
153 initObjectInst,
154 {6, ABCKIT_ISA_API_STATIC_OPCODE_RETURN_VOID, {}},
155
156 },
157 },
158 {{1}, {}, {}}};
159 helpers::VerifyGraph(graph, bbSchema);
160 });
161
162 output = helpers::ExecuteStaticAbc(ABCKIT_ABC_DIR "ut/isa/isa_static/objects/objects_modified.abc",
163 "objects/ETSGLOBAL", "main");
164 EXPECT_TRUE(helpers::Match(output, expectedOutput));
165 }
166
167 // Test: test-kind=api, api=IsaApiStaticImpl::iCreateNewObject, abc-kind=ArkTS2, category=positive, extension=c
TEST_F(LibAbcKitICreateObjectsStaticTest,LibAbcKitTestCreateIcreateNewObject)168 TEST_F(LibAbcKitICreateObjectsStaticTest, LibAbcKitTestCreateIcreateNewObject)
169 {
170 auto output =
171 helpers::ExecuteStaticAbc(ABCKIT_ABC_DIR "ut/isa/isa_static/objects/objects.abc", "objects/ETSGLOBAL", "main");
172 EXPECT_TRUE(helpers::Match(output, "A\n"));
173
174 std::vector<helpers::BBSchema<AbckitIsaApiStaticOpcode>> bbSchema = {
175 {{}, {1}, {}},
176 {
177 {0},
178 {2U},
179 {
180 {0, ABCKIT_ISA_API_STATIC_OPCODE_INITOBJECT, {}},
181 {3, ABCKIT_ISA_API_STATIC_OPCODE_CALL_VIRTUAL, {0}},
182 {4, ABCKIT_ISA_API_STATIC_OPCODE_NEWOBJECT, {}},
183 {5, ABCKIT_ISA_API_STATIC_OPCODE_CALL_STATIC, {4}},
184 {6, ABCKIT_ISA_API_STATIC_OPCODE_RETURN_VOID, {}},
185
186 },
187 },
188 {{1}, {}, {}}};
189
190 helpers::TransformMethod(
191 ABCKIT_ABC_DIR "ut/isa/isa_static/objects/objects.abc",
192 ABCKIT_ABC_DIR "ut/isa/isa_static/objects/objects_modified.abc", "main",
193 [](AbckitFile * /*file*/, AbckitCoreFunction * /*method*/, AbckitGraph *graph) {
194 TransformIrCreateNewObject(graph);
195 },
196 [&](AbckitGraph *graph) { helpers::VerifyGraph(graph, bbSchema); });
197
198 output = helpers::ExecuteStaticAbc(ABCKIT_ABC_DIR "ut/isa/isa_static/objects/objects_modified.abc",
199 "objects/ETSGLOBAL", "main");
200 EXPECT_TRUE(helpers::Match(output, "A\nB\n"));
201
202 helpers::TransformMethod(
203 ABCKIT_ABC_DIR "ut/isa/isa_static/objects/objects_modified.abc",
204 ABCKIT_ABC_DIR "ut/isa/isa_static/objects/objects_modified.abc", "main",
205 []([[maybe_unused]] AbckitFile *file, [[maybe_unused]] AbckitCoreFunction *method,
206 [[maybe_unused]] AbckitGraph *graph) {},
207 [&](AbckitGraph *graph) { helpers::VerifyGraph(graph, bbSchema); });
208
209 output = helpers::ExecuteStaticAbc(ABCKIT_ABC_DIR "ut/isa/isa_static/objects/objects_modified.abc",
210 "objects/ETSGLOBAL", "main");
211 EXPECT_TRUE(helpers::Match(output, "A\nB\n"));
212 }
213
214 // Test: test-kind=api, api=IsaApiStaticImpl::iCreateInitObject, abc-kind=ArkTS2, category=positive, extension=c
TEST_F(LibAbcKitICreateObjectsStaticTest,LibAbcKitTestCreateIcreateInitObjectArg0)215 TEST_F(LibAbcKitICreateObjectsStaticTest, LibAbcKitTestCreateIcreateInitObjectArg0)
216 {
217 InitObjectTest(0, {4U, ABCKIT_ISA_API_STATIC_OPCODE_INITOBJECT, {}}, "A\nC\n");
218 }
219
220 // Test: test-kind=api, api=IsaApiStaticImpl::iCreateInitObject, abc-kind=ArkTS2, category=positive, extension=c
TEST_F(LibAbcKitICreateObjectsStaticTest,LibAbcKitTestCreateIcreateInitObjectArg1)221 TEST_F(LibAbcKitICreateObjectsStaticTest, LibAbcKitTestCreateIcreateInitObjectArg1)
222 {
223 InitObjectTest(1, {4U, ABCKIT_ISA_API_STATIC_OPCODE_INITOBJECT, {0}}, "A\nC 1\n");
224 }
225
226 // Test: test-kind=api, api=IsaApiStaticImpl::iCreateInitObject, abc-kind=ArkTS2, category=positive, extension=c
TEST_F(LibAbcKitICreateObjectsStaticTest,LibAbcKitTestCreateIcreateInitObjectArg2)227 TEST_F(LibAbcKitICreateObjectsStaticTest, LibAbcKitTestCreateIcreateInitObjectArg2)
228 {
229 InitObjectTest(2U, {4U, ABCKIT_ISA_API_STATIC_OPCODE_INITOBJECT, {0, 0}}, "A\nC 11\n");
230 }
231
232 // Test: test-kind=api, api=IsaApiStaticImpl::iCreateInitObject, abc-kind=ArkTS2, category=positive, extension=c
TEST_F(LibAbcKitICreateObjectsStaticTest,LibAbcKitTestCreateIcreateInitObjectArg3)233 TEST_F(LibAbcKitICreateObjectsStaticTest, LibAbcKitTestCreateIcreateInitObjectArg3)
234 {
235 InitObjectTest(3U, {4U, ABCKIT_ISA_API_STATIC_OPCODE_INITOBJECT, {0, 0, 0}}, "A\nC 111\n");
236 }
237
238 // Test: test-kind=api, api=IsaApiStaticImpl::iCreateInitObject, abc-kind=ArkTS2, category=positive, extension=c
TEST_F(LibAbcKitICreateObjectsStaticTest,LibAbcKitTestCreateIcreateInitObjectArg4)239 TEST_F(LibAbcKitICreateObjectsStaticTest, LibAbcKitTestCreateIcreateInitObjectArg4)
240 {
241 InitObjectTest(4U, {4U, ABCKIT_ISA_API_STATIC_OPCODE_INITOBJECT, {0, 0, 0, 0}}, "A\nC 1111\n");
242 }
243
244 // Test: test-kind=api, api=IsaApiStaticImpl::iCreateInitObject, abc-kind=ArkTS2, category=positive, extension=c
TEST_F(LibAbcKitICreateObjectsStaticTest,LibAbcKitTestCreateIcreateInitObjectArg5)245 TEST_F(LibAbcKitICreateObjectsStaticTest, LibAbcKitTestCreateIcreateInitObjectArg5)
246 {
247 InitObjectTest(5U, {4U, ABCKIT_ISA_API_STATIC_OPCODE_INITOBJECT, {0, 0, 0, 0, 0}}, "A\nC 11111\n");
248 }
249
250 } // namespace libabckit::test
251 // NOLINTEND(readability-magic-numbers)
252