1 /*
2 * Copyright (c) 2023-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 "disable_usb_plugin.h"
17
18 #include "bool_serializer.h"
19 #include "edm_constants.h"
20 #include "edm_ipc_interface_code.h"
21 #include "edm_utils.h"
22 #include "usb_policy_utils.h"
23 #include "plugin_manager.h"
24
25 namespace OHOS {
26 namespace EDM {
27 const bool REGISTER_RESULT = PluginManager::GetInstance()->AddPlugin(DisableUsbPlugin::GetPlugin());
28
InitPlugin(std::shared_ptr<IPluginTemplate<DisableUsbPlugin,bool>> ptr)29 void DisableUsbPlugin::InitPlugin(std::shared_ptr<IPluginTemplate<DisableUsbPlugin, bool>> ptr)
30 {
31 EDMLOGI("DisableUsbPlugin InitPlugin...");
32 std::map<std::string, std::string> perms;
33 perms.insert(std::make_pair(EdmConstants::PERMISSION_TAG_VERSION_11,
34 "ohos.permission.ENTERPRISE_MANAGE_USB"));
35 perms.insert(std::make_pair(EdmConstants::PERMISSION_TAG_VERSION_12,
36 "ohos.permission.ENTERPRISE_MANAGE_RESTRICTIONS"));
37 IPlugin::PolicyPermissionConfig config = IPlugin::PolicyPermissionConfig(perms,
38 IPlugin::PermissionType::SUPER_DEVICE_ADMIN, IPlugin::ApiType::PUBLIC);
39 ptr->InitAttribute(EdmInterfaceCode::DISABLE_USB, "disable_usb", config, true);
40 ptr->SetSerializer(BoolSerializer::GetInstance());
41 ptr->SetOnHandlePolicyListener(&DisableUsbPlugin::OnSetPolicy, FuncOperateType::SET);
42 ptr->SetOnAdminRemoveListener(&DisableUsbPlugin::OnAdminRemove);
43 }
44
OnSetPolicy(bool & data)45 ErrCode DisableUsbPlugin::OnSetPolicy(bool &data)
46 {
47 EDMLOGI("DisableUsbPlugin OnSetPolicy...disable = %{public}d", data);
48 if (data && HasConflictPolicy()) {
49 return EdmReturnErrCode::CONFIGURATION_CONFLICT_FAILED;
50 }
51 return UsbPolicyUtils::SetUsbDisabled(data);
52 }
53
HasConflictPolicy()54 bool DisableUsbPlugin::HasConflictPolicy()
55 {
56 auto policyManager = IPolicyManager::GetInstance();
57 std::string allowUsbDevice;
58 policyManager->GetPolicy("", "allowed_usb_devices", allowUsbDevice);
59 if (!allowUsbDevice.empty()) {
60 EDMLOGE("DisableUsbPlugin POLICY CONFLICT! allowedUsbDevice: %{public}s", allowUsbDevice.c_str());
61 return true;
62 }
63 std::string disallowUsbDevice;
64 policyManager->GetPolicy("", "disallowed_usb_devices", disallowUsbDevice);
65 if (!disallowUsbDevice.empty()) {
66 EDMLOGE("DisableUsbPlugin POLICY CONFLICT! disallowUsbDevice: %{public}s", disallowUsbDevice.c_str());
67 return true;
68 }
69 std::string usbStoragePolicy;
70 policyManager->GetPolicy("", "usb_read_only", usbStoragePolicy);
71 if (usbStoragePolicy == std::to_string(EdmConstants::STORAGE_USB_POLICY_DISABLED) ||
72 usbStoragePolicy == std::to_string(EdmConstants::STORAGE_USB_POLICY_READ_ONLY)) {
73 EDMLOGE("DisableUsbPlugin POLICY CONFLICT! usbStoragePolicy: %{public}s", usbStoragePolicy.c_str());
74 return true;
75 }
76 return false;
77 }
78
OnGetPolicy(std::string & policyData,MessageParcel & data,MessageParcel & reply,int32_t userId)79 ErrCode DisableUsbPlugin::OnGetPolicy(std::string &policyData, MessageParcel &data, MessageParcel &reply,
80 int32_t userId)
81 {
82 EDMLOGI("DisableUsbPlugin OnGetPolicy %{public}s...", policyData.c_str());
83 bool isDisabled = false;
84 pluginInstance_->serializer_->Deserialize(policyData, isDisabled);
85 reply.WriteInt32(ERR_OK);
86 reply.WriteBool(isDisabled);
87 return ERR_OK;
88 }
89
OnAdminRemove(const std::string & adminName,bool & data,int32_t userId)90 ErrCode DisableUsbPlugin::OnAdminRemove(const std::string &adminName, bool &data, int32_t userId)
91 {
92 EDMLOGI("DisableUsbPlugin OnAdminRemove %{public}d...", data);
93 if (!data) {
94 return ERR_OK;
95 }
96 return UsbPolicyUtils::SetUsbDisabled(!data);
97 }
98 } // namespace EDM
99 } // namespace OHOS
100