1 /*
2  * Copyright (c) 2022 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 #ifndef FOUNDATION_BUNDLE_FRAMEWORK_SERVICE_INCLUDE_APP_CONTROL_MANAGER_HOST_IMPL_H
17 #define FOUNDATION_BUNDLE_FRAMEWORK_SERVICE_INCLUDE_APP_CONTROL_MANAGER_HOST_IMPL_H
18 
19 #include "app_control_host.h"
20 #include "app_control_manager.h"
21 #include "bundle_data_mgr.h"
22 
23 namespace OHOS {
24 namespace AppExecFwk {
25 class AppControlManagerHostImpl : public OHOS::AppExecFwk::AppControlHost {
26 public:
27     AppControlManagerHostImpl();
28     virtual ~AppControlManagerHostImpl();
29 
30     virtual ErrCode AddAppInstallControlRule(const std::vector<std::string> &appIds,
31         const AppInstallControlRuleType controlRuleType, int32_t userId) override;
32 
33     virtual ErrCode DeleteAppInstallControlRule(const AppInstallControlRuleType controlRuleType,
34         const std::vector<std::string> &appIds, int32_t userId) override;
35 
36     virtual ErrCode DeleteAppInstallControlRule(
37         const AppInstallControlRuleType controlRuleType, int32_t userId) override;
38 
39     virtual ErrCode GetAppInstallControlRule(const AppInstallControlRuleType controlRuleType, int32_t userId,
40         std::vector<std::string> &appIds) override;
41 
42     // for app running control rule
43     virtual ErrCode AddAppRunningControlRule(
44         const std::vector<AppRunningControlRule> &controlRules, int32_t userId) override;
45     virtual ErrCode DeleteAppRunningControlRule(
46         const std::vector<AppRunningControlRule> &controlRules, int32_t userId) override;
47     virtual ErrCode DeleteAppRunningControlRule(int32_t userId) override;
48     virtual ErrCode GetAppRunningControlRule(int32_t userId, std::vector<std::string> &appIds) override;
49     virtual ErrCode GetAppRunningControlRule(
50         const std::string &bundleName, int32_t userId, AppRunningControlRuleResult &controlRule) override;
51 
52     // for app jump control rule
53     virtual ErrCode ConfirmAppJumpControlRule(const std::string &callerBundleName, const std::string &targetBundleName,
54         int32_t userId) override;
55     virtual ErrCode AddAppJumpControlRule(const std::vector<AppJumpControlRule> &controlRules, int32_t userId) override;
56     virtual ErrCode DeleteAppJumpControlRule(const std::vector<AppJumpControlRule> &controlRules,
57         int32_t userId) override;
58     virtual ErrCode DeleteRuleByCallerBundleName(const std::string &callerBundleName, int32_t userId) override;
59     virtual ErrCode DeleteRuleByTargetBundleName(const std::string &targetBundleName, int32_t userId) override;
60     virtual ErrCode GetAppJumpControlRule(const std::string &callerBundleName, const std::string &targetBundleName,
61         int32_t userId, AppJumpControlRule &controlRule) override;
62 
63     virtual ErrCode SetDisposedStatus(
64         const std::string &appId, const Want &want, int32_t userId) override;
65 
66     virtual ErrCode DeleteDisposedStatus(
67         const std::string &appId, int32_t userId) override;
68 
69     virtual ErrCode GetDisposedStatus(
70         const std::string &appId, Want &want, int32_t userId) override;
71 
72     virtual ErrCode SetDisposedRule(
73         const std::string &appId, DisposedRule &DisposedRule, int32_t userId) override;
74 
75     virtual ErrCode GetDisposedRule(
76         const std::string &appId, DisposedRule &DisposedRule, int32_t userId) override;
77 
78     virtual ErrCode GetAbilityRunningControlRule(const std::string &bundleName, int32_t userId,
79         std::vector<DisposedRule>& disposedRules, int32_t appIndex = Constants::MAIN_APP_INDEX) override;
80 
81     virtual ErrCode SetDisposedRuleForCloneApp(
82         const std::string &appId, DisposedRule &DisposedRule, int32_t appIndex, int32_t userId) override;
83 
84     virtual ErrCode GetDisposedRuleForCloneApp(
85         const std::string &appId, DisposedRule &DisposedRule, int32_t appIndex, int32_t userId) override;
86 
87     virtual ErrCode DeleteDisposedRuleForCloneApp(
88         const std::string &appId, int32_t appIndex, int32_t userId) override;
89 
90 private:
91     int32_t GetCallingUserId();
92     std::string GetCallingName();
93     std::string GetControlRuleType(const AppInstallControlRuleType controlRuleType);
94     void UpdateAppControlledInfo(int32_t userId) const;
95     void GetCallerByUid(const int32_t uid, std::string &callerName);
96 
97     std::unordered_map<int32_t, std::string> callingNameMap_;
98     std::unordered_map<AppInstallControlRuleType, std::string> ruleTypeMap_;
99     std::shared_ptr<AppControlManager> appControlManager_ = nullptr;
100     std::shared_ptr<BundleDataMgr> dataMgr_ = nullptr;
101 };
102 }
103 }
104 #endif // FOUNDATION_BUNDLE_FRAMEWORK_SERVICE_INCLUDE_APP_CONTROL_MANAGER_HOST_IMPL_H
105