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 "libabckit/include/c/metadata_core.h"
18 #include "libabckit/include/c/ir_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
TransformCallVirtualIr(AbckitGraph * graph,AbckitCoreClass * classA)35 void TransformCallVirtualIr(AbckitGraph *graph, AbckitCoreClass *classA)
36 {
37 auto *ret = helpers::FindFirstInst(graph, ABCKIT_ISA_API_STATIC_OPCODE_RETURN_VOID);
38 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
39
40 auto *initObj = g_implG->iGetPrev(ret);
41 ASSERT_NE(initObj, nullptr);
42 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
43
44 helpers::MethodByNameContext methodCtxFinder = {nullptr, "foo"};
45 g_implI->classEnumerateMethods(classA, &methodCtxFinder, helpers::MethodByNameFinder);
46 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
47 ASSERT_NE(methodCtxFinder.method, nullptr);
48
49 auto *call = g_statG->iCreateCallVirtual(graph, initObj, methodCtxFinder.method, 0);
50 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
51
52 g_implG->iInsertBefore(call, ret);
53 ASSERT_EQ(g_impl->getLastError(), ABCKIT_STATUS_NO_ERROR);
54 }
55
56 } // namespace
57
58 class LibAbcKitCallVirtualStaticTest : public ::testing::Test {};
59
60 // Test: test-kind=api, api=IsaApiStaticImpl::iCreateCallVirtual, abc-kind=ArkTS2, category=positive, extension=c
TEST_F(LibAbcKitCallVirtualStaticTest,LibAbcKitTestCallVirtual)61 TEST_F(LibAbcKitCallVirtualStaticTest, LibAbcKitTestCallVirtual)
62 {
63 auto output = helpers::ExecuteStaticAbc(ABCKIT_ABC_DIR "ut/isa/isa_static/call/call_virtual_static.abc",
64 "call_virtual_static/ETSGLOBAL", "main");
65 EXPECT_TRUE(helpers::Match(output, ""));
66
67 helpers::TransformMethod(
68 ABCKIT_ABC_DIR "ut/isa/isa_static/call/call_virtual_static.abc",
69 ABCKIT_ABC_DIR "ut/isa/isa_static/call/call_virtual_static_modified.abc", "main",
70 [](AbckitFile * /*file*/, AbckitCoreFunction *method, AbckitGraph *graph) {
71 auto *module = g_implI->functionGetModule(method);
72 helpers::ClassByNameContext classCtxFinder = {nullptr, "A"};
73 g_implI->moduleEnumerateClasses(module, &classCtxFinder, helpers::ClassByNameFinder);
74 ASSERT_NE(classCtxFinder.klass, nullptr);
75 TransformCallVirtualIr(graph, classCtxFinder.klass);
76 },
77 [](AbckitGraph *graph) {
78 helpers::BBSchema<AbckitIsaApiStaticOpcode> bb1({{}, {1}, {{}}});
79 std::vector<helpers::InstSchema<AbckitIsaApiStaticOpcode>> insts({
80 {0, ABCKIT_ISA_API_STATIC_OPCODE_INITOBJECT, {}},
81 {2, ABCKIT_ISA_API_STATIC_OPCODE_CALL_VIRTUAL, {0}},
82 {1, ABCKIT_ISA_API_STATIC_OPCODE_RETURN_VOID, {}},
83 });
84 helpers::BBSchema<AbckitIsaApiStaticOpcode> bb2({{0}, {2}, insts});
85 helpers::BBSchema<AbckitIsaApiStaticOpcode> bb3({{1}, {}, {}});
86 std::vector<helpers::BBSchema<AbckitIsaApiStaticOpcode>> bbSchemas({bb1, bb2, bb3});
87 helpers::VerifyGraph(graph, bbSchemas);
88 });
89
90 output = helpers::ExecuteStaticAbc(ABCKIT_ABC_DIR "ut/isa/isa_static/call/call_virtual_static_modified.abc",
91 "call_virtual_static/ETSGLOBAL", "main");
92 EXPECT_TRUE(helpers::Match(output, "foo\n"));
93 }
94
95 } // namespace libabckit::test
96 // NOLINTEND(readability-magic-numbers)
97