1 /*
2 * Copyright (c) 2025 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 "allowed_wifi_list_plugin.h"
17
18 #include <algorithm>
19
20 #include "system_ability_definition.h"
21
22 #include "array_wifi_id_serializer.h"
23 #include "edm_constants.h"
24 #include "edm_ipc_interface_code.h"
25 #include "edm_sys_manager.h"
26 #include "edm_utils.h"
27 #include "iplugin_manager.h"
28 #include "wifi_policy_utils.h"
29
30 namespace OHOS {
31 namespace EDM {
32 const bool REGISTER_RESULT = IPluginManager::GetInstance()->AddPlugin(AllowWifiListPlugin::GetPlugin());
33
InitPlugin(std::shared_ptr<IPluginTemplate<AllowWifiListPlugin,std::vector<WifiId>>> ptr)34 void AllowWifiListPlugin::InitPlugin(
35 std::shared_ptr<IPluginTemplate<AllowWifiListPlugin, std::vector<WifiId>>> ptr)
36 {
37 EDMLOGI("AllowWifiListPlugin InitPlugin...");
38 ptr->InitAttribute(EdmInterfaceCode::ALLOWED_WIFI_LIST, PolicyName::POLICY_ALLOWED_WIFI_LIST,
39 EdmPermission::PERMISSION_ENTERPRISE_MANAGE_WIFI, IPlugin::PermissionType::SUPER_DEVICE_ADMIN, true);
40 ptr->SetSerializer(ArrayWifiIdSerializer::GetInstance());
41 ptr->SetOnHandlePolicyListener(&AllowWifiListPlugin::OnSetPolicy, FuncOperateType::SET);
42 ptr->SetOnHandlePolicyListener(&AllowWifiListPlugin::OnRemovePolicy, FuncOperateType::REMOVE);
43 ptr->SetOnAdminRemoveListener(&AllowWifiListPlugin::OnAdminRemove);
44 }
45
OnSetPolicy(std::vector<WifiId> & data,std::vector<WifiId> & currentData,std::vector<WifiId> & mergeData,int32_t userId)46 ErrCode AllowWifiListPlugin::OnSetPolicy(std::vector<WifiId> &data,
47 std::vector<WifiId> ¤tData, std::vector<WifiId> &mergeData, int32_t userId)
48 {
49 EDMLOGI("AllowWifiListPlugin OnSetPolicy");
50 return WifiPolicyUtils::AddWifiListPolicy(data, currentData, mergeData, userId, true);
51 }
52
OnRemovePolicy(std::vector<WifiId> & data,std::vector<WifiId> & currentData,std::vector<WifiId> & mergeData,int32_t userId)53 ErrCode AllowWifiListPlugin::OnRemovePolicy(std::vector<WifiId> &data, std::vector<WifiId> ¤tData,
54 std::vector<WifiId> &mergeData, int32_t userId)
55 {
56 EDMLOGI("AllowWifiListPlugin OnRemovePolicy");
57 return WifiPolicyUtils::RemoveWifiListPolicy(data, currentData, mergeData, userId, true);
58 }
59
OnAdminRemove(const std::string & adminName,std::vector<WifiId> & data,std::vector<WifiId> & mergeData,int32_t userId)60 ErrCode AllowWifiListPlugin::OnAdminRemove(const std::string &adminName, std::vector<WifiId> &data,
61 std::vector<WifiId> &mergeData, int32_t userId)
62 {
63 EDMLOGI("AllowWifiListPlugin OnAdminRemove");
64 if (data.empty()) {
65 EDMLOGW("AllowWifiListPlugin OnRemovePolicy data is empty:");
66 return ERR_OK;
67 }
68 return WifiPolicyUtils::SetWifiAccessList(mergeData, true);
69 }
70 } // namespace EDM
71 } // namespace OHOS