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::string> perms;
34 perms.insert(std::make_pair(EdmConstants::PERMISSION_TAG_VERSION_11,
35 "ohos.permission.ENTERPRISE_MANAGE_WIFI"));
36 perms.insert(std::make_pair(EdmConstants::PERMISSION_TAG_VERSION_12,
37 "ohos.permission.ENTERPRISE_MANAGE_RESTRICTIONS"));
38 IPlugin::PolicyPermissionConfig config = IPlugin::PolicyPermissionConfig(perms,
39 IPlugin::PermissionType::SUPER_DEVICE_ADMIN, IPlugin::ApiType::PUBLIC);
40 ptr->InitAttribute(EdmInterfaceCode::DISABLE_WIFI, "disable_wifi", config, false);
41 ptr->SetSerializer(BoolSerializer::GetInstance());
42 ptr->SetOnHandlePolicyListener(&SetWifiDisabledPlugin::OnSetPolicy, FuncOperateType::SET);
43 }
44
OnSetPolicy(bool & isDisable)45 ErrCode SetWifiDisabledPlugin::OnSetPolicy(bool &isDisable)
46 {
47 EDMLOGI("SetWifiDisabledPlugin OnSetPolicy %{public}d", isDisable);
48 std::string value = isDisable ? "true" : "false";
49 if (isDisable) {
50 ErrCode ret = Wifi::WifiDevice::GetInstance(WIFI_DEVICE_ABILITY_ID)->DisableWifi();
51 if (ret != ERR_OK) {
52 return EdmReturnErrCode::SYSTEM_ABNORMALLY;
53 }
54 }
55 return system::SetParameter(KEY_DISABLE_WIFI, value) ? ERR_OK : EdmReturnErrCode::SYSTEM_ABNORMALLY;
56 }
57
OnGetPolicy(std::string & policyData,MessageParcel & data,MessageParcel & reply,int32_t userId)58 ErrCode SetWifiDisabledPlugin::OnGetPolicy(std::string &policyData, MessageParcel &data, MessageParcel &reply,
59 int32_t userId)
60 {
61 EDMLOGI("SetWifiDisabledPlugin OnGetPolicy.");
62 bool ret = system::GetBoolParameter(KEY_DISABLE_WIFI, false);
63 reply.WriteInt32(ERR_OK);
64 reply.WriteBool(ret);
65 return ERR_OK;
66 }
67 } // namespace EDM
68 } // namespace OHOS
69