1 /*
2 * Copyright (c) 2023 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 "set_wifi_disabled_plugin.h"
17
18 #include "bool_serializer.h"
19 #include "edm_constants.h"
20 #include "edm_ipc_interface_code.h"
21 #include "parameters.h"
22 #include "wifi_device.h"
23 #include "plugin_manager.h"
24
25 namespace OHOS {
26 namespace EDM {
27 const bool REGISTER_RESULT = PluginManager::GetInstance()->AddPlugin(SetWifiDisabledPlugin::GetPlugin());
28 const std::string KEY_DISABLE_WIFI = "persist.edm.wifi_enable";
29
InitPlugin(std::shared_ptr<IPluginTemplate<SetWifiDisabledPlugin,bool>> ptr)30 void SetWifiDisabledPlugin::InitPlugin(std::shared_ptr<IPluginTemplate<SetWifiDisabledPlugin, bool>> ptr)
31 {
32 EDMLOGI("SetWifiDisabledPlugin InitPlugin...");
33 std::map<std::string, std::map<IPlugin::PermissionType, std::string>> tagPermissions;
34 std::map<IPlugin::PermissionType, std::string> typePermissionsForTag11;
35 std::map<IPlugin::PermissionType, std::string> typePermissionsForTag12;
36 typePermissionsForTag11.emplace(IPlugin::PermissionType::SUPER_DEVICE_ADMIN,
37 "ohos.permission.ENTERPRISE_MANAGE_WIFI");
38 typePermissionsForTag12.emplace(IPlugin::PermissionType::SUPER_DEVICE_ADMIN,
39 "ohos.permission.ENTERPRISE_MANAGE_RESTRICTIONS");
40 typePermissionsForTag12.emplace(IPlugin::PermissionType::BYOD_DEVICE_ADMIN,
41 "ohos.permission.PERSONAL_MANAGE_RESTRICTIONS");
42 tagPermissions.emplace(EdmConstants::PERMISSION_TAG_VERSION_11, typePermissionsForTag11);
43 tagPermissions.emplace(EdmConstants::PERMISSION_TAG_VERSION_12, typePermissionsForTag12);
44
45 IPlugin::PolicyPermissionConfig config = IPlugin::PolicyPermissionConfig(tagPermissions, IPlugin::ApiType::PUBLIC);
46 ptr->InitAttribute(EdmInterfaceCode::DISABLE_WIFI, "disable_wifi", config, true);
47 ptr->SetSerializer(BoolSerializer::GetInstance());
48 ptr->SetOnHandlePolicyListener(&SetWifiDisabledPlugin::OnSetPolicy, FuncOperateType::SET);
49 ptr->SetOnAdminRemoveListener(&SetWifiDisabledPlugin::OnAdminRemove);
50 }
51
OnSetPolicy(bool & isDisable)52 ErrCode SetWifiDisabledPlugin::OnSetPolicy(bool &isDisable)
53 {
54 EDMLOGI("SetWifiDisabledPlugin OnSetPolicy %{public}d", isDisable);
55 std::string value = isDisable ? "true" : "false";
56 if (isDisable) {
57 ErrCode ret = Wifi::WifiDevice::GetInstance(WIFI_DEVICE_ABILITY_ID)->DisableWifi();
58 if (ret != ERR_OK) {
59 return EdmReturnErrCode::SYSTEM_ABNORMALLY;
60 }
61 }
62 return system::SetParameter(KEY_DISABLE_WIFI, value) ? ERR_OK : EdmReturnErrCode::SYSTEM_ABNORMALLY;
63 }
64
OnAdminRemove(const std::string & adminName,bool & data,int32_t userId)65 ErrCode SetWifiDisabledPlugin::OnAdminRemove(const std::string &adminName, bool &data, int32_t userId)
66 {
67 EDMLOGI("SetWifiDisabledPlugin OnAdminRemove %{public}d...", data);
68 if (!data) {
69 return ERR_OK;
70 }
71 bool reset = false;
72 return OnSetPolicy(reset);
73 }
74
OnGetPolicy(std::string & policyData,MessageParcel & data,MessageParcel & reply,int32_t userId)75 ErrCode SetWifiDisabledPlugin::OnGetPolicy(std::string &policyData, MessageParcel &data, MessageParcel &reply,
76 int32_t userId)
77 {
78 EDMLOGI("SetWifiDisabledPlugin OnGetPolicy.");
79 bool ret = system::GetBoolParameter(KEY_DISABLE_WIFI, false);
80 reply.WriteInt32(ERR_OK);
81 reply.WriteBool(ret);
82 return ERR_OK;
83 }
84 } // namespace EDM
85 } // namespace OHOS
86