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/metadata_core.h"
17 #include "libabckit/include/c/ir_core.h"
18 #include "libabckit/include/c/isa/isa_dynamic.h"
19 #include "libabckit/include/c/abckit.h"
20
21 #include "helpers/helpers.h"
22 #include "helpers/helpers_runtime.h"
23
24 #include <gtest/gtest.h>
25
26 // NOLINTBEGIN(readability-magic-numbers)
27 namespace libabckit::test {
28
29 static auto g_impl = AbckitGetApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
30 static auto g_implI = AbckitGetInspectApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
31 static auto g_implM = AbckitGetModifyApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
32 static auto g_implG = AbckitGetGraphApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
33 static auto g_dynG = AbckitGetIsaApiDynamicImpl(ABCKIT_VERSION_RELEASE_1_0_0);
34
TransformIrDynLdInstValid(AbckitGraph * graph,AbckitInst * (* ldInstToCheck)(AbckitGraph * graph))35 static void TransformIrDynLdInstValid(AbckitGraph *graph, AbckitInst *(*ldInstToCheck)(AbckitGraph *graph))
36 {
37 AbckitInst *retOp = helpers::FindFirstInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_RETURN);
38 auto *mainInst = ldInstToCheck(graph);
39 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
40 g_implG->iInsertBefore(mainInst, retOp);
41 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
42 g_implG->iSetInput(retOp, mainInst, 0);
43 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
44 }
45
CreateBBSchemaForDynLd(AbckitIsaApiDynamicOpcode opcode)46 static std::vector<helpers::BBSchema<AbckitIsaApiDynamicOpcode>> CreateBBSchemaForDynLd(
47 AbckitIsaApiDynamicOpcode opcode)
48 {
49 return {{{},
50 {1},
51 {{0, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER, {}},
52 {1, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER, {}},
53 {2, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER, {}},
54 {3, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER, {}}}},
55 {{0},
56 {2},
57 {
58 {4, opcode, {}},
59 {5, ABCKIT_ISA_API_DYNAMIC_OPCODE_RETURN, {4}},
60 }},
61 {{1}, {}, {}}};
62 }
63
ValidTest(AbckitInst * (* ldInstToCheck)(AbckitGraph * graph),AbckitIsaApiDynamicOpcode expectedOpcode,const std::string & expectedOutput,bool skipExecute=false)64 static void ValidTest(AbckitInst *(*ldInstToCheck)(AbckitGraph *graph), AbckitIsaApiDynamicOpcode expectedOpcode,
65 const std::string &expectedOutput, bool skipExecute = false)
66 {
67 auto output =
68 helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/isa/isa_dynamic/loadstore/ld_dynamic.abc", "ld_dynamic");
69 EXPECT_TRUE(helpers::Match(output, "10\n"));
70
71 helpers::TransformMethod(
72 ABCKIT_ABC_DIR "ut/isa/isa_dynamic/loadstore/ld_dynamic.abc",
73 ABCKIT_ABC_DIR "ut/isa/isa_dynamic/loadstore/ld_dynamic_modified.abc", "foo",
74 [&](AbckitFile * /*file*/, AbckitCoreFunction * /*method*/, AbckitGraph *graph) {
75 TransformIrDynLdInstValid(graph, ldInstToCheck);
76 },
77 [&](AbckitGraph *graph) {
78 std::vector<helpers::BBSchema<AbckitIsaApiDynamicOpcode>> bbSchemas(CreateBBSchemaForDynLd(expectedOpcode));
79 helpers::VerifyGraph(graph, bbSchemas);
80 });
81
82 if (!skipExecute) {
83 output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/isa/isa_dynamic/loadstore/ld_dynamic_modified.abc",
84 "ld_dynamic");
85 EXPECT_TRUE(helpers::Match(output, expectedOutput));
86 }
87 }
88
89 class LibAbcKitDynLdInstTest : public ::testing::Test {};
90
91 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdnan, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynLdnanValid)92 TEST_F(LibAbcKitDynLdInstTest, CreateDynLdnanValid)
93 {
94 ValidTest(g_dynG->iCreateLdnan, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDNAN, "NaN\n");
95 }
96
97 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdinfinity, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynLdinfinityValid)98 TEST_F(LibAbcKitDynLdInstTest, CreateDynLdinfinityValid)
99 {
100 ValidTest(g_dynG->iCreateLdinfinity, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDINFINITY, "Infinity\n");
101 }
102
103 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdundefined, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynLdundefinedValid)104 TEST_F(LibAbcKitDynLdInstTest, CreateDynLdundefinedValid)
105 {
106 ValidTest(g_dynG->iCreateLdundefined, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDUNDEFINED, "undefined\n");
107 }
108
109 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdnull, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynLdnullValid)110 TEST_F(LibAbcKitDynLdInstTest, CreateDynLdnullValid)
111 {
112 ValidTest(g_dynG->iCreateLdnull, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDNULL, "null\n");
113 }
114
115 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdsymbol, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynLdsymbolValid)116 TEST_F(LibAbcKitDynLdInstTest, CreateDynLdsymbolValid)
117 {
118 ValidTest(g_dynG->iCreateLdsymbol, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDSYMBOL,
119 "function Symbol\\(\\) \\{ \\[native code\\] \\}\n");
120 }
121
122 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdglobal, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynLdglobalValid)123 TEST_F(LibAbcKitDynLdInstTest, CreateDynLdglobalValid)
124 {
125 ValidTest(g_dynG->iCreateLdglobal, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDGLOBAL, "\\[object Object\\]\n");
126 }
127
128 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdtrue, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynLdtrueValid)129 TEST_F(LibAbcKitDynLdInstTest, CreateDynLdtrueValid)
130 {
131 ValidTest(g_dynG->iCreateLdtrue, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDTRUE, "true\n");
132 }
133
134 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdfalse, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynLdfalseValid)135 TEST_F(LibAbcKitDynLdInstTest, CreateDynLdfalseValid)
136 {
137 ValidTest(g_dynG->iCreateLdfalse, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDFALSE, "false\n");
138 }
139
140 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdhole, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynLdholeValid)141 TEST_F(LibAbcKitDynLdInstTest, CreateDynLdholeValid)
142 {
143 ValidTest(g_dynG->iCreateLdhole, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDHOLE, "\n");
144 }
145
146 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdfunction, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynLdfunctionValid)147 TEST_F(LibAbcKitDynLdInstTest, CreateDynLdfunctionValid)
148 {
149 ValidTest(g_dynG->iCreateLdfunction, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDFUNCTION, "", true);
150 }
151
152 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdnewtarget, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynLdnewtargetValid)153 TEST_F(LibAbcKitDynLdInstTest, CreateDynLdnewtargetValid)
154 {
155 ValidTest(g_dynG->iCreateLdnewtarget, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDNEWTARGET, "\n");
156 }
157
158 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdthis, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynLdthisValid)159 TEST_F(LibAbcKitDynLdInstTest, CreateDynLdthisValid)
160 {
161 ValidTest(g_dynG->iCreateLdthis, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDTHIS, "\\[object Object\\]\n");
162 }
163
164 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateTryldglobalbyname, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynTryldglobalbynameValid)165 TEST_F(LibAbcKitDynLdInstTest, CreateDynTryldglobalbynameValid)
166 {
167 auto output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/isa/isa_dynamic/loadstore/tryldglobalbyname.abc",
168 "tryldglobalbyname");
169 EXPECT_TRUE(helpers::Match(output, ""));
170
171 auto cb = [](AbckitFile *file, AbckitCoreFunction * /*method*/, AbckitGraph *graph) {
172 auto *stringPrint = g_implM->createString(file, "print", strlen("print"));
173 auto *tryldglobalbyname = g_dynG->iCreateTryldglobalbyname(graph, stringPrint);
174 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
175 auto *newobjrange = helpers::FindFirstInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_NEWOBJRANGE);
176 // CC-OFFNXT(G.FMT.02)
177 auto *stringf = g_implM->createString(file, "f", strlen("f"));
178 auto *ldobjbyname = g_dynG->iCreateLdobjbyname(graph, newobjrange, stringf);
179 auto *callarg1 = g_dynG->iCreateCallarg1(graph, tryldglobalbyname, ldobjbyname);
180 auto *returnundefined = helpers::FindFirstInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_RETURNUNDEFINED);
181 // CC-OFFNXT(G.FMT.02)
182 g_implG->iInsertBefore(tryldglobalbyname, returnundefined);
183 g_implG->iInsertBefore(ldobjbyname, returnundefined);
184 g_implG->iInsertBefore(callarg1, returnundefined);
185 };
186
187 helpers::TransformMethod(ABCKIT_ABC_DIR "ut/isa/isa_dynamic/loadstore/tryldglobalbyname.abc",
188 ABCKIT_ABC_DIR "ut/isa/isa_dynamic/loadstore/tryldglobalbyname_modified.abc",
189 "func_main_0", cb);
190
191 output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/isa/isa_dynamic/loadstore/tryldglobalbyname_modified.abc",
192 "tryldglobalbyname");
193 EXPECT_TRUE(helpers::Match(output, "123\n"));
194 }
195
196 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateTrystglobalbyname, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitDynLdInstTest,CreateDynTrystglobalbynameValid)197 TEST_F(LibAbcKitDynLdInstTest, CreateDynTrystglobalbynameValid)
198 {
199 helpers::TransformMethod(
200 ABCKIT_ABC_DIR "ut/isa/isa_dynamic/loadstore/trystglobalbyname.abc",
201 ABCKIT_ABC_DIR "ut/isa/isa_dynamic/loadstore/trystglobalbyname_modified.abc", "func_main_0",
202 [&](AbckitFile *file, AbckitCoreFunction * /*method*/, AbckitGraph *graph) {
203 auto arr = std::vector<AbckitLiteral *>();
204 arr.emplace_back(g_implM->createLiteralU32(file, 0));
205 arr.emplace_back(g_implM->createLiteralU32(file, 111));
206 arr.emplace_back(g_implM->createLiteralString(file, "hello", strlen("hello")));
207 auto *litArr = g_implM->createLiteralArray(file, arr.data(), arr.size());
208
209 auto *createarraywithbuffer = g_dynG->iCreateCreatearraywithbuffer(graph, litArr);
210 auto *stringO = g_implM->createString(file, "o", strlen("o"));
211 auto *trystglobalbyname = g_dynG->iCreateTrystglobalbyname(graph, createarraywithbuffer, stringO);
212 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
213 auto *callarg0 = helpers::FindFirstInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_CALLARG0);
214
215 g_implG->iInsertBefore(createarraywithbuffer, callarg0);
216 g_implG->iInsertBefore(trystglobalbyname, callarg0);
217 },
218 [&](AbckitGraph *graph) {
219 std::vector<helpers::BBSchema<AbckitIsaApiDynamicOpcode>> bbSchemas = {
220 {{},
221 {1},
222 {{0, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER, {}},
223 {1, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER, {}},
224 {2, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER, {}}}},
225 {{0},
226 {2},
227 {
228 {3, ABCKIT_ISA_API_DYNAMIC_OPCODE_DEFINEFUNC, {}},
229 {4, ABCKIT_ISA_API_DYNAMIC_OPCODE_CREATEARRAYWITHBUFFER, {}},
230 {5, ABCKIT_ISA_API_DYNAMIC_OPCODE_TRYSTGLOBALBYNAME, {4}},
231 {6, ABCKIT_ISA_API_DYNAMIC_OPCODE_CALLARG0, {3}},
232 {7, ABCKIT_ISA_API_DYNAMIC_OPCODE_RETURNUNDEFINED, {}},
233 }},
234 {{1}, {}, {}}};
235 helpers::VerifyGraph(graph, bbSchemas);
236 });
237 }
238
239 } // namespace libabckit::test
240 // NOLINTEND(readability-magic-numbers)
241