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 #ifndef SERVICES_EDM_PLUGIN_INCLUDE_DISALLOWED_USB_DEVICES_PLUGIN_H
17 #define SERVICES_EDM_PLUGIN_INCLUDE_DISALLOWED_USB_DEVICES_PLUGIN_H
18 
19 #include "plugin_singleton.h"
20 #include "usb_interface_type.h"
21 
22 namespace OHOS {
23 namespace EDM {
24 class DisallowedUsbDevicesPlugin : public IPlugin {
25 public:
26     DisallowedUsbDevicesPlugin();
27     ErrCode OnHandlePolicy(std::uint32_t funcCode, MessageParcel &data, MessageParcel &reply,
28         HandlePolicyData &policyData, int32_t userId) override;
OnHandlePolicyDone(std::uint32_t funcCode,const std::string & adminName,bool isGlobalChanged,int32_t userId)29     void OnHandlePolicyDone(std::uint32_t funcCode, const std::string &adminName, bool isGlobalChanged,
30         int32_t userId) override{};
31     ErrCode OnAdminRemove(const std::string &adminName, const std::string &policyData, int32_t userId) override;
OnAdminRemoveDone(const std::string & adminName,const std::string & currentJsonData,int32_t userId)32     void OnAdminRemoveDone(const std::string &adminName, const std::string &currentJsonData, int32_t userId) override{};
33     ErrCode OnGetPolicy(std::string &policyData, MessageParcel &data, MessageParcel &reply, int32_t userId) override;
34 
35 private:
36     bool HasConflictPolicy(const FuncOperateType type);
37     bool CombinePolicyDataAndBeforeData(const FuncOperateType type, std::vector<USB::UsbDeviceType> policyDevices,
38         std::vector<USB::UsbDeviceType> beforeDevices, std::vector<USB::UsbDeviceType> &mergeDevices);
39     void CombineDataWithStorageAccessPolicy(std::vector<USB::UsbDeviceType> policyData,
40         std::vector<USB::UsbDeviceType> &combineData);
41 };
42 } // namespace EDM
43 } // namespace OHOS
44 
45 #endif // SERVICES_EDM_PLUGIN_INCLUDE_DISALLOWED_USB_DEVICES_PLUGIN_H
46