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 "get_native_token_id_test.h"
17 #include "gtest/gtest.h"
18 #include <thread>
19 #include <unistd.h>
20
21 #include "access_token.h"
22 #include "access_token_error.h"
23 #include "accesstoken_common_log.h"
24 #include "iaccess_token_manager.h"
25 #include "permission_grant_info.h"
26 #include "permission_state_change_info_parcel.h"
27 #include "string_ex.h"
28 #include "test_common.h"
29 #include "tokenid_kit.h"
30 #include "token_setproc.h"
31
32 using namespace testing::ext;
33 namespace OHOS {
34 namespace Security {
35 namespace AccessToken {
36 namespace {
37 static uint64_t g_selfTokenId = 0;
38 };
39
SetUpTestCase()40 void GetNativeTokenIdTest::SetUpTestCase()
41 {
42 g_selfTokenId = GetSelfTokenID();
43 TestCommon::SetTestEvironment(g_selfTokenId);
44 }
45
TearDownTestCase()46 void GetNativeTokenIdTest::TearDownTestCase()
47 {
48 SetSelfTokenID(g_selfTokenId);
49 TestCommon::ResetTestEvironment();
50 }
51
SetUp()52 void GetNativeTokenIdTest::SetUp()
53 {
54 LOGI(ATM_DOMAIN, ATM_TAG, "SetUp ok.");
55 }
56
TearDown()57 void GetNativeTokenIdTest::TearDown()
58 {
59 }
60
61 /**
62 * @tc.name: GetNativeTokenIdAbnormalTest001
63 * @tc.desc: cannot get native tokenid with invalid processName.
64 * @tc.type: FUNC
65 * @tc.require: Issue Number
66 */
67 HWTEST_F(GetNativeTokenIdTest, GetNativeTokenIdAbnormalTest001, TestSize.Level0)
68 {
69 LOGI(ATM_DOMAIN, ATM_TAG, "GetNativeTokenIdAbnormalTest001");
70 MockNativeToken mock("accesstoken_service");
71 std::string processName = "";
72 EXPECT_EQ(INVALID_TOKENID, AccessTokenKit::GetNativeTokenId(processName));
73
74 processName = "invalid processName";
75 EXPECT_EQ(INVALID_TOKENID, AccessTokenKit::GetNativeTokenId(processName));
76 }
77
78 /**
79 * @tc.name: GetNativeTokenIdAbnormalTest002
80 * @tc.desc: get native tokenid with hap.
81 * @tc.type: FUNC
82 * @tc.require: Issue Number
83 */
84 HWTEST_F(GetNativeTokenIdTest, GetNativeTokenIdAbnormalTest002, TestSize.Level0)
85 {
86 LOGI(ATM_DOMAIN, ATM_TAG, "GetNativeTokenIdAbnormalTest002");
87 std::string processName = "hdcd";
88 std::vector<std::string> reqPerm;
89 MockHapToken mock("GetNativeTokenIdAbnormalTest002", reqPerm, true);
90
91 int32_t selfUid = getuid();
92 setuid(10001); // 10001： UID
93
94 ASSERT_EQ(INVALID_TOKENID, AccessTokenKit::GetNativeTokenId(processName));
95
96 // restore environment
97 setuid(selfUid);
98 }
99
100 /**
101 * @tc.name: GetNativeTokenIdAbnormalTest003
102 * @tc.desc: Verify the GetNativeTokenId abnormal branch return nullptr proxy.
103 * @tc.type: FUNC
104 * @tc.require: Issue Number
105 */
106 HWTEST_F(GetNativeTokenIdTest, GetNativeTokenIdAbnormalTest003, TestSize.Level0)
107 {
108 LOGI(ATM_DOMAIN, ATM_TAG, "GetNativeTokenIdAbnormalTest003");
109 int32_t gSelfUid = getuid();
110 setuid(1234); // 1234: UID
111
112 std::string processName = "hdcd";
113 AccessTokenID tokenID = AccessTokenKit::GetNativeTokenId(processName);
114 ASSERT_EQ(INVALID_TOKENID, tokenID);
115
116 setuid(gSelfUid);
117 ASSERT_EQ(RET_SUCCESS, SetSelfTokenID(g_selfTokenId));
118 }
119
120 /**
121 * @tc.name: GetNativeTokenIdFuncTest001
122 * @tc.desc: get native tokenid with processName.
123 * @tc.type: FUNC
124 * @tc.require: Issue Number
125 */
126 HWTEST_F(GetNativeTokenIdTest, GetNativeTokenIdFuncTest001, TestSize.Level0)
127 {
128 LOGI(ATM_DOMAIN, ATM_TAG, "GetNativeTokenIdFuncTest001");
129 MockNativeToken mock("accesstoken_service");
130
131 std::string processName = "hdcd";
132 AccessTokenID tokenID = AccessTokenKit::GetNativeTokenId(processName);
133 ASSERT_NE(INVALID_TOKENID, tokenID);
134
135 NativeTokenInfo tokenInfo;
136 ASSERT_EQ(RET_SUCCESS, AccessTokenKit::GetNativeTokenInfo(tokenID, tokenInfo));
137 ASSERT_EQ(true, tokenInfo.processName == processName);
138 }
139
140 /**
141 * @tc.name: GetNativeTokenIdFuncTest002
142 * @tc.desc: get native tokenid.
143 * @tc.type: FUNC
144 * @tc.require: Issue Number
145 */
146 HWTEST_F(GetNativeTokenIdTest, GetNativeTokenIdFuncTest002, TestSize.Level0)
147 {
148 LOGI(ATM_DOMAIN, ATM_TAG, "GetNativeTokenIdFuncTest002");
149 std::string processName = "hdcd";
150 MockNativeToken mock("accesstoken_service");
151
152 AccessTokenID tokenID = AccessTokenKit::GetNativeTokenId(processName);
153 ASSERT_NE(INVALID_TOKENID, tokenID);
154
155 NativeTokenInfo tokenInfo;
156 ASSERT_EQ(RET_SUCCESS, AccessTokenKit::GetNativeTokenInfo(tokenID, tokenInfo));
157 ASSERT_EQ(true, tokenInfo.processName == processName);
158 }
159 } // namespace AccessToken
160 } // namespace Security
161 } // namespace OHOS