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 "plugin_manager.h"
26
27 namespace OHOS {
28 namespace EDM {
29 const bool REGISTER_RESULT = PluginManager::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 "ohos.permission.ENTERPRISE_SET_WIFI");
40 typePermissionsForTag12.emplace(IPlugin::PermissionType::SUPER_DEVICE_ADMIN,
41 "ohos.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, "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 ErrCode ret = Wifi::WifiDevice::GetInstance(WIFI_DEVICE_ABILITY_ID)->ConnectToDevice(config);
55 EdmUtils::ClearString(config.preSharedKey);
56 if (sizeof(config.wepKeys) / sizeof(std::string) > 0) {
57 EdmUtils::ClearString(config.wepKeys[0]);
58 }
59 EdmUtils::ClearString(config.wifiEapConfig.password);
60 memset_s(config.wifiEapConfig.certPassword, sizeof(config.wifiEapConfig.certPassword), 0,
61 sizeof(config.wifiEapConfig.certPassword));
62 if (ret != ERR_OK) {
63 return EdmReturnErrCode::SYSTEM_ABNORMALLY;
64 }
65 return ERR_OK;
66 }
67 } // namespace EDM
68 } // namespace OHOS
69