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 "multi_thread_test.h"
17
18 #include "access_token.h"
19 #include "access_token_error.h"
20 #include "accesstoken_log.h"
21 #define private public
22 #include "accesstoken_id_manager.h"
23 #undef private
24 #include "permission_validator.h"
25 #include "string_ex.h"
26 #include "token_setproc.h"
27
28
29 using namespace testing::ext;
30 using namespace testing::mt;
31 namespace OHOS {
32 namespace Security {
33 namespace AccessToken {
34 namespace {
35 bool g_register = false;
36 static std::set<AccessTokenID> g_tokenIdSet;
37 static constexpr int32_t TEST_TOKEN_ID_1 = 537800000;
38 static constexpr int32_t TEST_TOKEN_ID_2 = 537900000;
39 static constexpr int32_t MULTI_CYCLE_TIMES = 1000;
40 static constexpr OHOS::HiviewDFX::HiLogLabel LABEL = {LOG_CORE,
41 SECURITY_DOMAIN_ACCESSTOKEN, "AccessTokenMultiThreadTest"};
42 }
43
SetUpTestCase()44 void AccessTokenMultiThreadTest::SetUpTestCase()
45 {
46 }
47
TearDownTestCase()48 void AccessTokenMultiThreadTest::TearDownTestCase()
49 {
50 }
51
SetUp()52 void AccessTokenMultiThreadTest::SetUp()
53 {
54 ACCESSTOKEN_LOG_INFO(LABEL, "SetUp ok.");
55 g_tokenIdSet = AccessTokenIDManager::GetInstance().tokenIdSet_;
56 AccessTokenIDManager::GetInstance().tokenIdSet_.clear();
57 }
58
TearDown()59 void AccessTokenMultiThreadTest::TearDown()
60 {
61 AccessTokenIDManager::GetInstance().tokenIdSet_ = g_tokenIdSet; // recovery
62 }
63
TestRegisterTokenId()64 void TestRegisterTokenId()
65 {
66 AccessTokenID tokenId = TEST_TOKEN_ID_2;
67 int32_t i = MULTI_CYCLE_TIMES + 1;
68 while (i--) {
69 tokenId += i;
70 AccessTokenIDManager::GetInstance().RegisterTokenId(tokenId, TOKEN_HAP);
71 }
72 }
73
TestReleaseTokenId()74 void TestReleaseTokenId()
75 {
76 AccessTokenID releaseId = TEST_TOKEN_ID_2;
77 AccessTokenID tokenId = TEST_TOKEN_ID_1;
78 g_register = !g_register;
79 int32_t i = MULTI_CYCLE_TIMES + 1;
80 if (!g_register) {
81 while (i--) {
82 releaseId += i;
83 AccessTokenIDManager::GetInstance().ReleaseTokenId(tokenId);
84 }
85 } else {
86 while (i--) {
87 tokenId += i;
88 AccessTokenIDManager::GetInstance().RegisterTokenId(tokenId, TOKEN_HAP);
89 }
90 }
91 }
92
93 /**
94 * @tc.name: RegisterTokenId001
95 * @tc.desc: register 1000 id.
96 * @tc.type: FUNC
97 * @tc.require: Issue Number
98 */
99 HWTEST_F(AccessTokenMultiThreadTest, RegisterTokenId001, TestSize.Level1)
100 {
101 SET_THREAD_NUM(10);
102 GTEST_RUN_TASK(TestRegisterTokenId);
103 std::vector<AccessTokenID> tokenIdList;
104 AccessTokenIDManager::GetInstance().GetHapTokenIdList(tokenIdList);
105 EXPECT_EQ(MULTI_CYCLE_TIMES, tokenIdList.size());
106 }
107
108 /**
109 * @tc.name: RegisterTokenId002
110 * @tc.desc: release and register 1000 id.
111 * @tc.type: FUNC
112 * @tc.require: Issue Number
113 */
114 HWTEST_F(AccessTokenMultiThreadTest, RegisterTokenId002, TestSize.Level1)
115 {
116 SET_THREAD_NUM(10);
117 GTEST_RUN_TASK(TestReleaseTokenId);
118 std::vector<AccessTokenID> tokenIdList;
119 AccessTokenIDManager::GetInstance().GetHapTokenIdList(tokenIdList);
120 EXPECT_EQ(MULTI_CYCLE_TIMES, tokenIdList.size());
121 }
122 } // namespace AccessToken
123 } // namespace Security
124 } // namespace OHOS
125