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 #include "switch_wifi_plugin.h"
16
17 #include "parameters.h"
18 #include "wifi_device.h"
19
20 #include "bool_serializer.h"
21 #include "edm_ipc_interface_code.h"
22 #include "iplugin_manager.h"
23
24 namespace OHOS {
25 namespace EDM {
26
27 const bool REGISTER_RESULT = IPluginManager::GetInstance()->AddPlugin(SwitchWifiPlugin::GetPlugin());
28 const std::string MDM_WIFI_PROP = "persist.edm.wifi_enable";
29 const std::string PARAM_FORCE_OPEN_WIFI = "persist.edm.force_open_wifi";
30
InitPlugin(std::shared_ptr<IPluginTemplate<SwitchWifiPlugin,bool>> ptr)31 void SwitchWifiPlugin::InitPlugin(std::shared_ptr<IPluginTemplate<SwitchWifiPlugin, bool>> ptr)
32 {
33 EDMLOGI("SwitchWifiPlugin InitPlugin...");
34 ptr->InitAttribute(EdmInterfaceCode::SWITCH_WIFI, "switch_wifi",
35 EdmPermission::PERMISSION_ENTERPRISE_MANAGE_WIFI, IPlugin::PermissionType::SUPER_DEVICE_ADMIN, true);
36 ptr->SetSerializer(BoolSerializer::GetInstance());
37 ptr->SetOnHandlePolicyListener(&SwitchWifiPlugin::OnTurnOnPolicy, FuncOperateType::SET);
38 ptr->SetOnHandlePolicyListener(&SwitchWifiPlugin::OnTurnOffPolicy, FuncOperateType::REMOVE);
39 ptr->SetOnAdminRemoveListener(&SwitchWifiPlugin::OnAdminRemove);
40 }
41
OnTurnOnPolicy(bool & isForce)42 ErrCode SwitchWifiPlugin::OnTurnOnPolicy(bool &isForce)
43 {
44 EDMLOGI("SwitchWifiPlugin OnSetPolicy isForce %{public}d", isForce);
45 if (system::GetBoolParameter(MDM_WIFI_PROP.c_str(), false)) {
46 EDMLOGE("SwitchWifiPlugin OnTurnOnPolicy failed, because wifi disabled");
47 return EdmReturnErrCode::ENTERPRISE_POLICES_DENIED;
48 }
49
50 std::shared_ptr<Wifi::WifiDevice> devicePtr = Wifi::WifiDevice::GetInstance(WIFI_DEVICE_ABILITY_ID);
51 if (devicePtr == nullptr) {
52 EDMLOGE("SwitchWifiPlugin:OnTurnOnPolicy WifiDevice null");
53 return EdmReturnErrCode::SYSTEM_ABNORMALLY;
54 }
55 int ret = devicePtr->EnableWifi();
56 if (ret != Wifi::WIFI_OPT_SUCCESS || !system::SetParameter(PARAM_FORCE_OPEN_WIFI, isForce ? "true" : "false")) {
57 EDMLOGE("SwitchWifiPlugin:OnTurnOnPolicy SetParameter fail or send request fail. %{public}d", ret);
58 return EdmReturnErrCode::SYSTEM_ABNORMALLY;
59 }
60 return ERR_OK;
61 }
62
OnTurnOffPolicy()63 ErrCode SwitchWifiPlugin::OnTurnOffPolicy()
64 {
65 if (system::GetBoolParameter(MDM_WIFI_PROP.c_str(), false)) {
66 EDMLOGE("SwitchWifiPlugin OnTurnOffPolicy failed, because wifi disabled");
67 return EdmReturnErrCode::ENTERPRISE_POLICES_DENIED;
68 }
69
70 if (!system::SetParameter(PARAM_FORCE_OPEN_WIFI, "false")) {
71 EDMLOGE("SwitchWifiPlugin:OnTurnOffPolicy SetParameter fail. ");
72 return EdmReturnErrCode::SYSTEM_ABNORMALLY;
73 }
74
75 std::shared_ptr<Wifi::WifiDevice> devicePtr = Wifi::WifiDevice::GetInstance(WIFI_DEVICE_ABILITY_ID);
76 if (devicePtr == nullptr) {
77 EDMLOGE("SwitchWifiPlugin:OnTurnOffPolicy WifiDevice null");
78 return EdmReturnErrCode::SYSTEM_ABNORMALLY;
79 }
80 int ret = devicePtr->DisableWifi();
81 if (ret != Wifi::WIFI_OPT_SUCCESS) {
82 EDMLOGE("SwitchWifiPlugin:OnTurnOffPolicy send request fail. %{public}d", ret);
83 return EdmReturnErrCode::SYSTEM_ABNORMALLY;
84 }
85 return ERR_OK;
86 }
87
OnAdminRemove()88 ErrCode SwitchWifiPlugin::OnAdminRemove()
89 {
90 EDMLOGI("SwitchWifiPlugin OnAdminRemove");
91 system::SetParameter(PARAM_FORCE_OPEN_WIFI, "false");
92 return ERR_OK;
93 }
94
95 } // namespace EDM
96 } // namespace OHOS