| # |
| # Copyright 2020 - The Android Open Source Project |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| import time |
| |
| from acts.controllers.openwrt_ap import MOBLY_CONTROLLER_CONFIG_NAME as OPENWRT |
| from acts.test_decorators import test_tracker_info |
| from acts_contrib.test_utils.net import connectivity_const |
| from acts_contrib.test_utils.net import net_test_utils as nutils |
| from acts_contrib.test_utils.wifi import wifi_test_utils as wutils |
| from acts_contrib.test_utils.wifi.WifiBaseTest import WifiBaseTest |
| |
| |
| VPN_CONST = connectivity_const.VpnProfile |
| VPN_TYPE = connectivity_const.VpnProfileType |
| VPN_PARAMS = connectivity_const.VpnReqParams |
| |
| |
| class IKEv2VpnOverWifiTest(WifiBaseTest): |
| """IKEv2 VPN tests.""" |
| |
| def setup_class(self): |
| """Setup wi-fi connection and unpack params.""" |
| self.dut = self.android_devices[0] |
| req_params = dir(VPN_PARAMS) |
| req_params = [ |
| x for x in req_params if not x.startswith("__") |
| ] |
| opt_params = ["wifi_network", "vpn_cert_country", |
| "vpn_cert_org", "configure_OpenWrt"] |
| self.unpack_userparams(req_param_names=req_params, |
| opt_param_names=opt_params) |
| |
| wutils.wifi_test_device_init(self.dut) |
| wutils.wifi_toggle_state(self.dut, True) |
| if OPENWRT in self.user_params: |
| self.openwrt = self.access_points[0] |
| if hasattr(self, "configure_OpenWrt") and self.configure_OpenWrt == "skip": |
| self.dut.log.info("Skip configure Wifi interface due to config setup.") |
| else: |
| self.configure_openwrt_ap_and_start(wpa_network=True) |
| self.wifi_network = self.openwrt.get_wifi_network() |
| # Wait for OpenWrt statement update |
| time.sleep(10) |
| self.openwrt.network_setting.setup_vpn_l2tp_server( |
| self.vpn_server_hostname, |
| self.vpn_verify_addresses["IKEV2_IPSEC_RSA"][0], |
| self.vpn_username, |
| self.vpn_password, |
| self.vpn_identity, |
| "ikev2-server", |
| self.vpn_cert_country, |
| self.vpn_cert_org |
| ) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wifi_network["SSID"]) |
| wutils.wifi_connect(self.dut, self.wifi_network) |
| time.sleep(3) |
| |
| self.vpn_params = {"vpn_username": self.vpn_username, |
| "vpn_password": self.vpn_password, |
| "psk_secret": self.psk_secret, |
| "client_pkcs_file_name": self.client_pkcs_file_name, |
| "cert_path_vpnserver": self.cert_path_vpnserver, |
| "cert_password": self.cert_password, |
| "vpn_identity": self.vpn_identity} |
| |
| def teardown_class(self): |
| wutils.reset_wifi(self.dut) |
| |
| def on_fail(self, test_name, begin_time): |
| self.dut.take_bug_report(test_name, begin_time) |
| |
| ### Helper methods ### |
| |
| def _test_ikev2_vpn(self, vpn, hostname=None): |
| """Verify IKEv2 VPN connection. |
| |
| Args: |
| vpn: type of VPN. |
| hostname: hostname or IP address of the server. |
| """ |
| server_addr = self.vpn_server_addresses[vpn.name][0] |
| self.vpn_params["server_addr"] = server_addr |
| if not hostname: |
| hostname = server_addr |
| vpn_addr = self.vpn_verify_addresses[vpn.name][0] |
| vpn_profile = nutils.generate_ikev2_vpn_profile( |
| self.dut, self.vpn_params, vpn, hostname, self.log_path) |
| nutils.legacy_vpn_connection_test_logic(self.dut, vpn_profile, vpn_addr) |
| |
| ### Test cases ### |
| |
| @test_tracker_info(uuid="4991755c-321d-4e9a-ada9-fc821a35bb5b") |
| def test_ikev2_psk_vpn_wifi(self): |
| self._test_ikev2_vpn(VPN_TYPE.IKEV2_IPSEC_PSK) |
| |
| @test_tracker_info(uuid="04d88575-7b96-4746-bff8-a1d6841e202e") |
| def test_ikev2_mschapv2_vpn_wifi(self): |
| self._test_ikev2_vpn(VPN_TYPE.IKEV2_IPSEC_USER_PASS) |
| |
| @test_tracker_info(uuid="e65f8a3e-f807-4493-822e-377dd6fa89cd") |
| def test_ikev2_rsa_vpn_wifi(self): |
| self._test_ikev2_vpn(VPN_TYPE.IKEV2_IPSEC_RSA) |
| |
| @test_tracker_info(uuid="bdd8a967-8dac-4e48-87b7-2ce9f7d32158") |
| def test_ikev2_psk_vpn_wifi_with_hostname(self): |
| self._test_ikev2_vpn(VPN_TYPE.IKEV2_IPSEC_PSK, |
| self.vpn_server_hostname) |
| |
| @test_tracker_info(uuid="19692520-c123-4b42-8549-08dda9c4873e") |
| def test_ikev2_mschapv2_vpn_wifi_with_hostname(self): |
| self._test_ikev2_vpn(VPN_TYPE.IKEV2_IPSEC_USER_PASS, |
| self.vpn_server_hostname) |
| |
| @test_tracker_info(uuid="bdaaf6e3-6671-4533-baba-2951009c7d69") |
| def test_ikev2_rsa_vpn_wifi_with_hostname(self): |
| self._test_ikev2_vpn(VPN_TYPE.IKEV2_IPSEC_RSA, |
| self.vpn_server_hostname) |