1/*
2 * Copyright (c) 2021 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 <HdcJdwpTest.h>
17using namespace testing::ext;
18using namespace std;
19namespace Hdc {
20void HdcJdwpTest::SetUpTestCase() {}
21
22void HdcJdwpTest::TearDownTestCase() {}
23
24void HdcJdwpTest::SetUp() {}
25
26void HdcJdwpTest::TearDown() {}
27
28std::unique_ptr<HdcJdwp> HdcJdwpTest::InstanceHdcJdwp()
29{
30    uv_loop_t loopMain;
31    uv_loop_init(&loopMain);
32    // Base::SetLogLevel(Hdc::LOG_FULL);
33    return std::make_unique<HdcJdwp>(&loopMain);
34}
35
36/*
37 * @tc.name: TestCreateFdEventPoll
38 * @tc.desc: Check the poll thread.
39 * @tc.type: FUNC
40 */
41HWTEST_F(HdcJdwpTest, TestCreateFdEventPoll, TestSize.Level1)
42{
43    std::unique_ptr<HdcJdwp> mJdwpTest = HdcJdwpTest::InstanceHdcJdwp();
44    ASSERT_NE(mJdwpTest, nullptr) << "Instanse HdcJdwp fail.";
45    ASSERT_EQ(mJdwpTest->Initial(), 0) << " Jdwp init instance fail.";
46    EXPECT_GE(mJdwpTest->pollNodeMap.size(), 0u) << " Jdwp pollNodeMap is null.";
47    mJdwpTest->Stop();
48}
49
50/*
51 * @tc.name: CreateJdwpTracker
52 * @tc.desc: Check the poll thread.
53 * @tc.type: FUNC
54 */
55HWTEST_F(HdcJdwpTest, TestCreateJdwpTracker, TestSize.Level1)
56{
57    std::unique_ptr<HdcJdwp> mJdwpTest = HdcJdwpTest::InstanceHdcJdwp();
58    ASSERT_NE(mJdwpTest, nullptr) << "Instanse HdcJdwp fail.";
59    ASSERT_EQ(mJdwpTest->Initial(), 0) << "Instanse HdcJdwp fail.";
60    ASSERT_EQ(mJdwpTest->CreateJdwpTracker(nullptr), 0) << "CreateJdwpTracker nullptr fail.";
61    HTaskInfo hTaskInfo = new TaskInformation();
62    HdcSessionBase HdcSessionTest(false);
63    hTaskInfo->ownerSessionClass = &HdcSessionTest;
64    ASSERT_EQ(mJdwpTest->CreateJdwpTracker(hTaskInfo), 1) << "CreateJdwpTracker valid param fail.";
65    delete hTaskInfo;
66    mJdwpTest->Stop();
67}
68
69/*
70 * @tc.name: TestReadStream
71 * @tc.desc: Check ReadStream interface of pipe connection.
72 * @tc.type: FUNC
73 */
74HWTEST_F(HdcJdwpTest, TestReadStream, TestSize.Level1)
75{
76    std::unique_ptr<HdcJdwp> mJdwpTest = HdcJdwpTest::InstanceHdcJdwp();
77    mJdwpTest->Initial();
78    uv_pipe_t pipe;
79    uv_stream_t *stream = (uv_stream_t *)&pipe;
80    HdcJdwp::HCtxJdwp ctxJdwp = (HdcJdwp::HCtxJdwp)mJdwpTest->MallocContext();
81    ctxJdwp->finish = true; // For skip HdcJdwp::FreeContext, because the pipe
82                            // connect hasn't actually established
83    pipe.data = ctxJdwp;
84    // invalid nread < min
85    mJdwpTest->ReadStream(stream, (HdcJdwp::JS_PKG_MIN_SIZE - 1), nullptr);
86    ASSERT_EQ(mJdwpTest->mapCtxJdwp.size(), 0u) << "Instanse HdcJdwp fail.";
87    // invalid nread > max
88    mJdwpTest->ReadStream(stream, (HdcJdwp::JS_PKG_MX_SIZE + 1), nullptr);
89    ASSERT_EQ(mJdwpTest->mapCtxJdwp.size(), 0u) << "Instanse HdcJdwp fail.";
90
91    // valid parameters
92    string pkgName = "JDWP_UTEST";
93    uint32_t pkgSize = pkgName.size() + sizeof(HdcJdwp::JsMsgHeader); // JsMsgHeader pkgName;
94    uint8_t *info = reinterpret_cast<uint8_t *>(ctxJdwp->buf);
95    ASSERT_NE(info, nullptr) << "TestReadStream info null.";
96    memset_s(info, sizeof(ctxJdwp->buf), 0, sizeof(ctxJdwp->buf));
97    HdcJdwp::JsMsgHeader *jsMsg = (HdcJdwp::JsMsgHeader *)info;
98    jsMsg->pid = 1212;
99    jsMsg->msgLen = pkgSize;
100    ASSERT_EQ(
101        memcpy_s(info + sizeof(HdcJdwp::JsMsgHeader), pkgName.size(), &pkgName[0], pkgName.size()),
102        0)
103        << "TestReadStream memcpy_s fail.";
104    mJdwpTest->ReadStream(stream, pkgSize, nullptr);
105    ASSERT_NE(mJdwpTest->AdminContext(OP_QUERY, 1212, nullptr), nullptr)
106        << "TestReadStream query fail.";
107    mJdwpTest->Stop();
108}
109} // namespace Hdc
110