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 <gtest/gtest.h>
17 #include <cstring>
18
19 #include "libabckit/include/c/abckit.h"
20 #include "helpers/helpers.h"
21 #include "helpers/helpers_runtime.h"
22 #include "libabckit/include/c/metadata_core.h"
23 #include "libabckit/include/c/extensions/js/metadata_js.h"
24 #include "ir_impl.h"
25 #include "libabckit/include/c/isa/isa_dynamic.h"
26 #include "libabckit/include/c/statuses.h"
27
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_implJsI = AbckitGetJsInspectApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
33 static auto g_implM = AbckitGetModifyApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
34 static auto g_implJsM = AbckitGetJsModifyApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
35 static auto g_implG = AbckitGetGraphApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
36 static auto g_dynG = AbckitGetIsaApiDynamicImpl(ABCKIT_VERSION_RELEASE_1_0_0);
37
38 class LibAbcKitJSModifyApiModulesTest : public ::testing::Test {};
39
40 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateNewobjrange, abc-kind=JS, category=positive
TEST_F(LibAbcKitJSModifyApiModulesTest,DynamicIcreateNewobjrange)41 TEST_F(LibAbcKitJSModifyApiModulesTest, DynamicIcreateNewobjrange)
42 {
43 auto output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/newobj.abc", "newobj");
44 auto expected = "1 2\n";
45 EXPECT_TRUE(helpers::Match(output, expected));
46
47 helpers::TransformMethod(
48 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/newobj.abc",
49 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/newobj_modified.abc", "newobj.func_main_0",
50 [](AbckitFile *, AbckitCoreFunction *, AbckitGraph *graph) {
51 AbckitInst *defineClass =
52 helpers::FindFirstInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_DEFINECLASSWITHBUFFER);
53
54 AbckitInst *a = g_implG->gFindOrCreateConstantI32(graph, 3);
55 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
56 ASSERT_NE(a, nullptr);
57 AbckitInst *storeThis = g_dynG->iCreateNewobjrange(graph, 3, defineClass, a, a);
58 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
59 ASSERT_NE(storeThis, nullptr);
60
61 AbckitInst *ret = helpers::FindFirstInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_RETURNUNDEFINED);
62 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
63 ASSERT_NE(ret, nullptr);
64
65 g_implG->iInsertBefore(storeThis, ret);
66 },
67 [&]([[maybe_unused]] AbckitGraph *graph) {});
68
69 output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/newobj_modified.abc", "newobj");
70 expected = "1 2\n3 3\n";
71 EXPECT_TRUE(helpers::Match(output, expected));
72 }
73
74 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateWideNewobjrange, abc-kind=JS, category=positive
TEST_F(LibAbcKitJSModifyApiModulesTest,DynamicIcreateWideNewobjrange)75 TEST_F(LibAbcKitJSModifyApiModulesTest, DynamicIcreateWideNewobjrange)
76 {
77 auto output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/newobj.abc", "newobj");
78 auto expected = "1 2\n";
79 EXPECT_TRUE(helpers::Match(output, expected));
80
81 helpers::TransformMethod(
82 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/newobj.abc",
83 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/newobj_modified.abc", "newobj.func_main_0",
84 [](AbckitFile *, AbckitCoreFunction *, AbckitGraph *graph) {
85 AbckitInst *defineClass =
86 helpers::FindFirstInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_DEFINECLASSWITHBUFFER);
87
88 AbckitInst *a = g_implG->gFindOrCreateConstantI32(graph, 4);
89 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
90 ASSERT_NE(a, nullptr);
91 AbckitInst *storeThis = g_dynG->iCreateWideNewobjrange(graph, 3, defineClass, a, a);
92 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
93 ASSERT_NE(storeThis, nullptr);
94
95 AbckitInst *ret = helpers::FindFirstInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_RETURNUNDEFINED);
96 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
97 ASSERT_NE(ret, nullptr);
98
99 g_implG->iInsertBefore(storeThis, ret);
100 },
101 [&]([[maybe_unused]] AbckitGraph *graph) {});
102
103 output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/newobj_modified.abc", "newobj");
104 expected = "1 2\n4 4\n";
105 EXPECT_TRUE(helpers::Match(output, expected));
106 }
107
108 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateDelobjprop, abc-kind=JS, category=positive
TEST_F(LibAbcKitJSModifyApiModulesTest,DynamicIcreateDelobjprop)109 TEST_F(LibAbcKitJSModifyApiModulesTest, DynamicIcreateDelobjprop)
110 {
111 auto output =
112 helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/delobjprop.abc", "delobjprop");
113 auto expected = "undefined 2\n";
114 EXPECT_TRUE(helpers::Match(output, expected));
115
116 auto cb = [](AbckitFile *file, AbckitCoreFunction *, AbckitGraph *graph) {
117 AbckitInst *param = helpers::FindLastInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER);
118 // CC-OFFNXT(G.FMT.02)
119 AbckitInst *firstDelobj = helpers::FindFirstInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_DELOBJPROP);
120 // CC-OFFNXT(G.FMT.02)
121 AbckitString *str = g_implM->createString(file, "y", strlen("y"));
122 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
123 ASSERT_NE(str, nullptr);
124
125 AbckitInst *loadString = g_dynG->iCreateLoadString(graph, str);
126 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
127 ASSERT_NE(loadString, nullptr);
128
129 g_implG->iInsertAfter(loadString, firstDelobj);
130 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
131
132 AbckitInst *secondDelobj = g_dynG->iCreateDelobjprop(graph, loadString, param);
133 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
134 ASSERT_NE(secondDelobj, nullptr);
135
136 g_implG->iInsertAfter(secondDelobj, loadString);
137 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
138 };
139
140 helpers::TransformMethod(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/delobjprop.abc",
141 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/delobjprop_modified.abc", "foo", cb);
142
143 output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/delobjprop_modified.abc",
144 "delobjprop");
145 expected = "undefined undefined\n";
146 EXPECT_TRUE(helpers::Match(output, expected));
147 }
148
149 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateLdprivateproperty, abc-kind=JS, category=positive
TEST_F(LibAbcKitJSModifyApiModulesTest,DynamicIcreateLdprivateproperty)150 TEST_F(LibAbcKitJSModifyApiModulesTest, DynamicIcreateLdprivateproperty)
151 {
152 auto output =
153 helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/privateprop.abc", "privateprop");
154 auto expected = "1\n2\n";
155 EXPECT_TRUE(helpers::Match(output, expected));
156
157 auto cb = [](AbckitFile *, AbckitCoreFunction *, AbckitGraph *graph) {
158 AbckitInst *param2 = helpers::FindLastInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER);
159 // CC-OFFNXT(G.FMT.02)
160 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
161 ASSERT_NE(param2, nullptr);
162
163 AbckitInst *load = g_dynG->iCreateLdprivateproperty(graph, param2, 0, 0);
164 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
165 ASSERT_NE(load, nullptr);
166
167 AbckitInst *ret = helpers::FindFirstInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_RETURN);
168 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
169 ASSERT_NE(ret, nullptr);
170 g_implG->iInsertBefore(load, ret);
171 g_implG->iRemove(ret);
172
173 ret = g_dynG->iCreateReturn(graph, load);
174
175 g_implG->iInsertAfter(ret, load);
176
177 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
178 };
179
180 helpers::TransformMethod(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/privateprop.abc",
181 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/privateprop_modified.abc", "getT", cb);
182
183 output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/privateprop_modified.abc",
184 "privateprop");
185 expected = "2\n2\n";
186 EXPECT_TRUE(helpers::Match(output, expected));
187 }
188
189 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateStprivateproperty, abc-kind=JS, category=positive
TEST_F(LibAbcKitJSModifyApiModulesTest,DynamicIcreateStprivateproperty)190 TEST_F(LibAbcKitJSModifyApiModulesTest, DynamicIcreateStprivateproperty)
191 {
192 auto output =
193 helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/privateprop.abc", "privateprop");
194 auto expected = "1\n2\n";
195 EXPECT_TRUE(helpers::Match(output, expected));
196
197 auto cb = [](AbckitFile *, AbckitCoreFunction *, AbckitGraph *graph) {
198 AbckitInst *param3 = helpers::FindLastInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER);
199 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
200 ASSERT_NE(param3, nullptr);
201
202 AbckitInst *param2 = g_implG->iGetPrev(param3);
203 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
204 ASSERT_NE(param2, nullptr);
205
206 AbckitInst *store = g_dynG->iCreateStprivateproperty(graph, param3, 0, 0, param2);
207 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
208 ASSERT_NE(store, nullptr);
209
210 AbckitInst *load = helpers::FindFirstInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_LDPRIVATEPROPERTY);
211 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
212 ASSERT_NE(load, nullptr);
213
214 g_implG->iInsertBefore(store, load);
215 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
216 };
217
218 helpers::TransformMethod(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/privateprop.abc",
219 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/privateprop_modified.abc", "changeT", cb);
220
221 output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/privateprop_modified.abc",
222 "privateprop");
223 expected = "1\n3\n";
224 EXPECT_TRUE(helpers::Match(output, expected));
225 }
226
227 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateSetobjectwithproto, abc-kind=JS, category=positive
TEST_F(LibAbcKitJSModifyApiModulesTest,DynamicIcreateSetobjectwithproto)228 TEST_F(LibAbcKitJSModifyApiModulesTest, DynamicIcreateSetobjectwithproto)
229 {
230 auto output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/proto.abc", "proto");
231 auto expected = "false\nfalse\n";
232 EXPECT_TRUE(helpers::Match(output, expected));
233
234 auto cb = [](AbckitFile *, AbckitCoreFunction *, AbckitGraph *graph) {
235 auto *ldlexvar = g_dynG->iCreateLdlexvar(graph, 0, 0);
236 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
237 ASSERT_NE(ldlexvar, nullptr);
238
239 auto *emptyObj = helpers::FindLastInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_CREATEEMPTYOBJECT);
240 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
241 ASSERT_NE(emptyObj, nullptr);
242
243 auto *set = g_dynG->iCreateSetobjectwithproto(graph, emptyObj, ldlexvar);
244 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
245 ASSERT_NE(set, nullptr);
246
247 g_implG->iInsertAfter(ldlexvar, emptyObj);
248 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
249
250 g_implG->iInsertAfter(set, ldlexvar);
251 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
252 };
253
254 helpers::TransformMethod(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/proto.abc",
255 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/proto_modified.abc", "setProto", cb);
256
257 output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/proto_modified.abc", "proto");
258 expected = "false\ntrue\n";
259 EXPECT_TRUE(helpers::Match(output, expected));
260 }
261
262 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateCallruntimeCreateprivateproperty, abc-kind=JS, category=positive
TEST_F(LibAbcKitJSModifyApiModulesTest,DynamicIcreateCallruntimeCreateprivateproperty)263 TEST_F(LibAbcKitJSModifyApiModulesTest, DynamicIcreateCallruntimeCreateprivateproperty)
264 {
265 auto output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/callruntime_private.abc",
266 "callruntime_private");
267 auto expected = "30\n";
268 EXPECT_TRUE(helpers::Match(output, expected));
269
270 helpers::TransformMethod(
271 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/callruntime_private.abc",
272 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/callruntime_private_modified.abc",
273 "callruntime_private.func_main_0", [](AbckitFile *, AbckitCoreFunction *, AbckitGraph *graph) {
274 auto *create =
275 helpers::FindLastInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_CALLRUNTIME_CREATEPRIVATEPROPERTY);
276 uint64_t imm0 = g_implG->iGetImmediate(create, 0);
277 auto *litArr = g_implG->iGetLiteralArray(create);
278 auto *newCreate = g_dynG->iCreateCallruntimeCreateprivateproperty(graph, imm0, litArr);
279 helpers::ReplaceInst(create, newCreate);
280 });
281
282 output = helpers::ExecuteDynamicAbc(
283 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/callruntime_private_modified.abc", "callruntime_private");
284 expected = "30\n";
285 EXPECT_TRUE(helpers::Match(output, expected));
286 }
287
288 // Test: test-kind=api, api=IsaApiDynamicImpl::iCreateCallruntimeDefineprivateproperty, abc-kind=JS, category=positive
TEST_F(LibAbcKitJSModifyApiModulesTest,DynamicIcreateCallruntimeDefineprivateproperty)289 TEST_F(LibAbcKitJSModifyApiModulesTest, DynamicIcreateCallruntimeDefineprivateproperty)
290 {
291 auto output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/callruntime_private.abc",
292 "callruntime_private");
293 auto expected = "30\n";
294 EXPECT_TRUE(helpers::Match(output, expected));
295
296 auto cb = [](AbckitFile *, AbckitCoreFunction *, AbckitGraph *graph) {
297 // CC-OFFNXT(G.FMT.02)
298 auto *oldDefine = helpers::FindLastInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_CALLRUNTIME_DEFINEPRIVATEPROPERTY);
299 uint64_t imm1 = g_implG->iGetImmediate(oldDefine, 1);
300 auto *acc = g_implG->iGetInput(oldDefine, 0);
301 auto *input = g_implG->iGetInput(oldDefine, 1);
302 // CC-OFFNXT(G.FMT.02)
303 auto *newDefine = g_dynG->iCreateCallruntimeDefineprivateproperty(graph, acc, imm1, 2, input);
304 helpers::ReplaceInst(oldDefine, newDefine);
305 };
306 helpers::TransformMethod(ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/callruntime_private.abc",
307 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/callruntime_private_modified.abc",
308 "instance_initializer", cb);
309
310 output = helpers::ExecuteDynamicAbc(
311 ABCKIT_ABC_DIR "ut/extensions/js/modify_api/obj/callruntime_private_modified.abc", "callruntime_private");
312 expected = "30\n";
313 EXPECT_TRUE(helpers::Match(output, expected));
314 }
315
316 } // namespace libabckit::test
317