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::string> perms;
36     perms.insert(std::make_pair(EdmConstants::PERMISSION_TAG_VERSION_11, "ohos.permission.ENTERPRISE_SET_WIFI"));
37     perms.insert(std::make_pair(EdmConstants::PERMISSION_TAG_VERSION_12, "ohos.permission.ENTERPRISE_MANAGE_WIFI"));
38     IPlugin::PolicyPermissionConfig config = IPlugin::PolicyPermissionConfig(perms,
39         IPlugin::PermissionType::SUPER_DEVICE_ADMIN, IPlugin::ApiType::PUBLIC);
40     ptr->InitAttribute(EdmInterfaceCode::SET_WIFI_PROFILE, "set_wifi_profile", config, false);
41     ptr->SetSerializer(WifiDeviceConfigSerializer::GetInstance());
42     ptr->SetOnHandlePolicyListener(&SetWifiProfilePlugin::OnSetPolicy, FuncOperateType::SET);
43 }
44 
OnSetPolicy(Wifi::WifiDeviceConfig & config)45 ErrCode SetWifiProfilePlugin::OnSetPolicy(Wifi::WifiDeviceConfig &config)
46 {
47     EDMLOGD("SetWifiProfilePlugin OnSetPolicy");
48     ErrCode ret = Wifi::WifiDevice::GetInstance(WIFI_DEVICE_ABILITY_ID)->ConnectToDevice(config);
49     EdmUtils::ClearString(config.preSharedKey);
50     if (sizeof(config.wepKeys) / sizeof(std::string) > 0) {
51         EdmUtils::ClearString(config.wepKeys[0]);
52     }
53     EdmUtils::ClearString(config.wifiEapConfig.password);
54     memset_s(config.wifiEapConfig.certPassword, sizeof(config.wifiEapConfig.certPassword), 0,
55         sizeof(config.wifiEapConfig.certPassword));
56     if (ret != ERR_OK) {
57         return EdmReturnErrCode::SYSTEM_ABNORMALLY;
58     }
59     return ERR_OK;
60 }
61 } // namespace EDM
62 } // namespace OHOS
63