1 /*
2 * Copyright (c) 2022-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 "policy_manager.h"
17 #include <algorithm>
18 #include <ctime>
19 #include <fstream>
20 #include <unistd.h>
21 #include "directory_ex.h"
22 #include "edm_log.h"
23
24 namespace OHOS {
25 namespace EDM {
26
GetAdminByPolicyName(const std::string & policyName,AdminValueItemsMap & adminValueItems,int32_t userId)27 ErrCode PolicyManager::GetAdminByPolicyName(const std::string &policyName, AdminValueItemsMap &adminValueItems,
28 int32_t userId)
29 {
30 auto userPolicyMgr = GetUserPolicyMgr(userId);
31 return userPolicyMgr->GetAdminByPolicyName(policyName, adminValueItems);
32 }
33
GetPolicy(const std::string & adminName,const std::string & policyName,std::string & policyValue,int32_t userId)34 ErrCode PolicyManager::GetPolicy(const std::string &adminName, const std::string &policyName, std::string &policyValue,
35 int32_t userId)
36 {
37 auto userPolicyMgr = GetUserPolicyMgr(userId);
38 return userPolicyMgr->GetPolicy(adminName, policyName, policyValue);
39 }
40
Init(std::vector<int32_t> userIds)41 void PolicyManager::Init(std::vector<int32_t> userIds)
42 {
43 for (auto userId : userIds) {
44 EDMLOGI("PolicyManager::Init userId %{public}d", userId);
45 if (userId == EdmConstants::DEFAULT_USER_ID) {
46 defaultPolicyMgr_ = std::make_shared<UserPolicyManager>(userId);
47 defaultPolicyMgr_->Init();
48 } else {
49 std::shared_ptr<UserPolicyManager> userPolicyMgr = std::make_shared<UserPolicyManager>(userId);
50 userPolicyMgr->Init();
51 policyMgrMap_.insert(std::make_pair(userId, userPolicyMgr));
52 }
53 }
54 return;
55 }
56
SetPolicy(const std::string & adminName,const std::string & policyName,const std::string & adminPolicyValue,const std::string & mergedPolicyValue,int32_t userId)57 ErrCode PolicyManager::SetPolicy(const std::string &adminName, const std::string &policyName,
58 const std::string &adminPolicyValue, const std::string &mergedPolicyValue,
59 int32_t userId)
60 {
61 auto userPolicyMgr = GetUserPolicyMgr(userId);
62 return userPolicyMgr->SetPolicy(adminName, policyName, adminPolicyValue, mergedPolicyValue);
63 }
64
GetAllPolicyByAdmin(const std::string & adminName,PolicyItemsMap & allAdminPolicy,int32_t userId)65 ErrCode PolicyManager::GetAllPolicyByAdmin(const std::string &adminName, PolicyItemsMap &allAdminPolicy,
66 int32_t userId)
67 {
68 auto userPolicyMgr = GetUserPolicyMgr(userId);
69 return userPolicyMgr->GetAllPolicyByAdmin(adminName, allAdminPolicy);
70 }
71
GetUserPolicyMgr(int32_t userId)72 std::shared_ptr<UserPolicyManager> PolicyManager::GetUserPolicyMgr(int32_t userId)
73 {
74 EDMLOGD("PolicyManager::GetUserPolicyMgr by userId:%{public}d", userId);
75 if (userId == EdmConstants::DEFAULT_USER_ID) {
76 if (defaultPolicyMgr_ == nullptr) {
77 defaultPolicyMgr_ = std::make_shared<UserPolicyManager>(userId);
78 defaultPolicyMgr_->Init();
79 }
80 return defaultPolicyMgr_;
81 } else {
82 auto itr = policyMgrMap_.find(userId);
83 if (itr != policyMgrMap_.end()) {
84 return itr->second;
85 } else {
86 std::shared_ptr<UserPolicyManager> userPolicyMgr = std::make_shared<UserPolicyManager>(userId);
87 userPolicyMgr->Init();
88 policyMgrMap_.insert(std::make_pair(userId, userPolicyMgr));
89 return userPolicyMgr;
90 }
91 }
92 }
93
GetPolicyUserIds(std::vector<int32_t> & userIds)94 void PolicyManager::GetPolicyUserIds(std::vector<int32_t> &userIds)
95 {
96 for (const auto& pair : policyMgrMap_) {
97 userIds.push_back(pair.first);
98 }
99 }
100
Dump()101 void PolicyManager::Dump()
102 {
103 auto defaultUserPolicyMgr = GetUserPolicyMgr(EdmConstants::DEFAULT_USER_ID);
104 defaultUserPolicyMgr->DumpAdminPolicy();
105 defaultUserPolicyMgr->DumpAdminList();
106 defaultUserPolicyMgr->DumpCombinedPolicy();
107
108 std::vector<int32_t> userIds;
109 GetPolicyUserIds(userIds);
110 EDMLOGD("PolicyManager::Dump userIds size:%{public}zu", userIds.size());
111 for (auto userId : userIds) {
112 auto userPolicyMgr = GetUserPolicyMgr(userId);
113 userPolicyMgr->DumpAdminPolicy();
114 userPolicyMgr->DumpAdminList();
115 userPolicyMgr->DumpCombinedPolicy();
116 }
117 }
118 } // namespace EDM
119 } // namespace OHOS
120