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");
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
ReplaceAllPolicy(int32_t userId,const std::string & adminName,const std::string & newAdminName)72 ErrCode PolicyManager::ReplaceAllPolicy(int32_t userId, const std::string &adminName, const std::string &newAdminName)
73 {
74 auto userPolicyMgr = GetUserPolicyMgr(userId);
75 return userPolicyMgr->ReplacePolicyByAdminName(userId, adminName, newAdminName);
76 }
77
GetUserPolicyMgr(int32_t userId)78 std::shared_ptr<UserPolicyManager> PolicyManager::GetUserPolicyMgr(int32_t userId)
79 {
80 EDMLOGD("PolicyManager::GetUserPolicyMgr");
81 if (userId == EdmConstants::DEFAULT_USER_ID) {
82 if (defaultPolicyMgr_ == nullptr) {
83 defaultPolicyMgr_ = std::make_shared<UserPolicyManager>(userId);
84 defaultPolicyMgr_->Init();
85 }
86 return defaultPolicyMgr_;
87 } else {
88 auto itr = policyMgrMap_.find(userId);
89 if (itr != policyMgrMap_.end()) {
90 return itr->second;
91 } else {
92 std::shared_ptr<UserPolicyManager> userPolicyMgr = std::make_shared<UserPolicyManager>(userId);
93 userPolicyMgr->Init();
94 policyMgrMap_.insert(std::make_pair(userId, userPolicyMgr));
95 return userPolicyMgr;
96 }
97 }
98 }
99
GetPolicyUserIds(std::vector<int32_t> & userIds)100 void PolicyManager::GetPolicyUserIds(std::vector<int32_t> &userIds)
101 {
102 for (const auto& pair : policyMgrMap_) {
103 userIds.push_back(pair.first);
104 }
105 }
106
Dump()107 void PolicyManager::Dump()
108 {
109 auto defaultUserPolicyMgr = GetUserPolicyMgr(EdmConstants::DEFAULT_USER_ID);
110 defaultUserPolicyMgr->DumpAdminPolicy();
111 defaultUserPolicyMgr->DumpAdminList();
112 defaultUserPolicyMgr->DumpCombinedPolicy();
113
114 std::vector<int32_t> userIds;
115 GetPolicyUserIds(userIds);
116 EDMLOGD("PolicyManager::Dump userIds size:%{public}zu", userIds.size());
117 for (auto userId : userIds) {
118 auto userPolicyMgr = GetUserPolicyMgr(userId);
119 userPolicyMgr->DumpAdminPolicy();
120 userPolicyMgr->DumpAdminList();
121 userPolicyMgr->DumpCombinedPolicy();
122 }
123 }
124 } // namespace EDM
125 } // namespace OHOS
126