Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 58199a43 authored by Martin Brabham's avatar Martin Brabham Committed by Automerger Merge Worker
Browse files

SecurityTest: Add double pairing test on same ACL am: e389270c 

Original change: https://android-review.googlesource.com/c/platform/system/bt/+/1324299

Change-Id: I46bf1adaf31ab93d46ea95caefd8f8ba3a37310b
parents 69a5d592 e389270c

system/gd/security/cert/security_test.py

+47 −3
Original line number Diff line number Diff line
@@ -20,8 +20,9 @@ from cert.event_stream import EventStream 
from cert.gd_base_test import GdBaseTestClass

from cert.py_security import PySecurity

from facade import common_pb2 as common

from hci.facade import controller_facade_pb2 as controller_facade

from google.protobuf import empty_pb2 as empty_proto

from hci.facade import controller_facade_pb2 as controller_facade

from l2cap.classic.facade_pb2 import ClassicSecurityPolicy

from neighbor.facade import facade_pb2 as neighbor_facade

from security.cert.cert_security import CertSecurity

from security.facade_pb2 import AuthenticationRequirements
@@ -63,13 +64,21 @@ class SecurityTest(GdBaseTestClass): 
        self.cert_security.close()

        super().teardown_test()



    # SSP Numeric Comparison test cases

    # Initiates the numeric comparison test

    def _run_ssp_numeric_comparison(self, initiator, responder, init_ui_response, resp_ui_response,

                                    expected_init_ui_event, expected_resp_ui_event, expected_init_bond_event,

                                    expected_resp_bond_event):

        initiator.enable_secure_simple_pairing()

        responder.enable_secure_simple_pairing()

        initiator.create_bond(responder.get_address(), common.BluetoothAddressTypeEnum.PUBLIC_DEVICE_ADDRESS)

        self._verify_ssp_numeric_comparison(initiator, responder, init_ui_response, resp_ui_response,

                                            expected_init_ui_event, expected_resp_ui_event, expected_init_bond_event,

                                            expected_resp_bond_event)



    # Verifies the events for the numeric comparion test

    def _verify_ssp_numeric_comparison(self, initiator, responder, init_ui_response, resp_ui_response,

                                       expected_init_ui_event, expected_resp_ui_event, expected_init_bond_event,

                                       expected_resp_bond_event):

        responder.accept_pairing(initiator.get_address(), resp_ui_response)

        initiator.on_user_input(responder.get_address(), init_ui_response, expected_init_ui_event)

        initiator.wait_for_bond_event(expected_init_bond_event)
@@ -296,7 +305,42 @@ class SecurityTest(GdBaseTestClass): 
            expected_resp_bond_event=None)



    # no_input_no_output + no_input_no_output is JustWorks no confirmation

    def test_dut_initiated_no_input_no_output_no_input_no_output_twice(self):

    def test_dut_initiated_no_input_no_output_no_input_no_output_twice_same_acl(self):

        # Arrange

        self.dut_security.set_io_capabilities(IoCapabilities.NO_INPUT_NO_OUTPUT)

        self.dut_security.set_authentication_requirements(AuthenticationRequirements.DEDICATED_BONDING_MITM_PROTECTION)

        self.dut_security.set_oob_data(OobDataPresent.NOT_PRESENT)

        self.cert_security.set_io_capabilities(IoCapabilities.NO_INPUT_NO_OUTPUT)

        self.cert_security.set_authentication_requirements(AuthenticationRequirements.DEDICATED_BONDING_MITM_PROTECTION)

        self.cert_security.set_oob_data(OobDataPresent.NOT_PRESENT)



        # Act and Assert

        self._run_ssp_numeric_comparison(

            initiator=self.dut_security,

            responder=self.cert_security,

            init_ui_response=True,

            resp_ui_response=True,

            expected_init_ui_event=None,

            expected_resp_ui_event=None,

            expected_init_bond_event=BondMsgType.DEVICE_BONDED,

            expected_resp_bond_event=None)



        self.dut_security.enforce_security_policy(self.cert.address,

                                                  common.BluetoothAddressTypeEnum.PUBLIC_DEVICE_ADDRESS,

                                                  ClassicSecurityPolicy.AUTHENTICATED_ENCRYPTED_TRANSPORT)



        self._verify_ssp_numeric_comparison(

            initiator=self.dut_security,

            responder=self.cert_security,

            init_ui_response=True,

            resp_ui_response=True,

            expected_init_ui_event=None,

            expected_resp_ui_event=None,

            expected_init_bond_event=BondMsgType.DEVICE_BONDED,

            expected_resp_bond_event=None)



    # no_input_no_output + no_input_no_output is JustWorks no confirmation

    def test_dut_initiated_no_input_no_output_no_input_no_output_twice_with_remove_bond(self):

        # Arrange

        self.dut_security.set_io_capabilities(IoCapabilities.NO_INPUT_NO_OUTPUT)

        self.dut_security.set_authentication_requirements(AuthenticationRequirements.DEDICATED_BONDING_MITM_PROTECTION)