1 /*
2 * Copyright (c) 2023-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 #include "set_wifi_profile_plugin.h"
17
18 #include <system_ability_definition.h>
19
20 #include "edm_constants.h"
21 #include "edm_ipc_interface_code.h"
22 #include "edm_utils.h"
23 #include "wifi_device.h"
24 #include "wifi_device_config_serializer.h"
25 #include "iplugin_manager.h"
26
27 namespace OHOS {
28 namespace EDM {
29 const bool REGISTER_RESULT = IPluginManager::GetInstance()->AddPlugin(SetWifiProfilePlugin::GetPlugin());
30
InitPlugin(std::shared_ptr<IPluginTemplate<SetWifiProfilePlugin,Wifi::WifiDeviceConfig>> ptr)31 void SetWifiProfilePlugin::InitPlugin(
32 std::shared_ptr<IPluginTemplate<SetWifiProfilePlugin, Wifi::WifiDeviceConfig>> ptr)
33 {
34 EDMLOGI("SetWifiProfilePlugin InitPlugin...");
35 std::map<std::string, std::map<IPlugin::PermissionType, std::string>> tagPermissions;
36 std::map<IPlugin::PermissionType, std::string> typePermissionsForTag11;
37 std::map<IPlugin::PermissionType, std::string> typePermissionsForTag12;
38 typePermissionsForTag11.emplace(IPlugin::PermissionType::SUPER_DEVICE_ADMIN,
39 EdmPermission::PERMISSION_ENTERPRISE_SET_WIFI);
40 typePermissionsForTag12.emplace(IPlugin::PermissionType::SUPER_DEVICE_ADMIN,
41 EdmPermission::PERMISSION_ENTERPRISE_MANAGE_WIFI);
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::SET_WIFI_PROFILE, PolicyName::POLICY_SET_WIFI_PROFILE, config, false);
47 ptr->SetSerializer(WifiDeviceConfigSerializer::GetInstance());
48 ptr->SetOnHandlePolicyListener(&SetWifiProfilePlugin::OnSetPolicy, FuncOperateType::SET);
49 }
50
OnSetPolicy(Wifi::WifiDeviceConfig & config)51 ErrCode SetWifiProfilePlugin::OnSetPolicy(Wifi::WifiDeviceConfig &config)
52 {
53 EDMLOGD("SetWifiProfilePlugin OnSetPolicy");
54 auto wifiDevice = Wifi::WifiDevice::GetInstance(WIFI_DEVICE_ABILITY_ID);
55 if (wifiDevice == nullptr) {
56 EDMLOGE("wifiDevice GetInstance null");
57 return EdmReturnErrCode::SYSTEM_ABNORMALLY;
58 }
59 ErrCode ret = wifiDevice->ConnectToDevice(config);
60 EdmUtils::ClearString(config.preSharedKey);
61 if (sizeof(config.wepKeys) / sizeof(std::string) > 0) {
62 EdmUtils::ClearString(config.wepKeys[0]);
63 }
64 EdmUtils::ClearString(config.wifiEapConfig.password);
65 memset_s(config.wifiEapConfig.certPassword, sizeof(config.wifiEapConfig.certPassword), 0,
66 sizeof(config.wifiEapConfig.certPassword));
67 if (ret != ERR_OK) {
68 EDMLOGE("SetWifiProfilePlugin ConnectToDevice ret %{public}d", ret);
69 return EdmReturnErrCode::SYSTEM_ABNORMALLY;
70 }
71 return ERR_OK;
72 }
73 } // namespace EDM
74 } // namespace OHOS
75