def __init__(self, device, acl_stream, remote_addr, handle):
"""
An abstract representation for an LE ACL connection in GD certification test
:param device: The GD device
:param acl_stream: The ACL stream for this connection
:param remote_addr: Remote device address
:param handle: Connection handle
"""
self.device = device
self.handle = handle
# todo enable filtering after sorting out handles
#self.our_acl_stream = FilteringEventStream(acl_stream, None)
self.our_acl_stream = acl_stream
if remote_addr:
remote_addr_bytes = bytes(
remote_addr.address.address,
'utf8') if type(remote_addr.address.address) is str else bytes(remote_addr.address.address)
self.connection_event_stream = EventStream(
self.device.hci_le_acl_manager.CreateConnection(
le_acl_manager_facade.LeConnectionMsg(
address_type=int(remote_addr.type),
address=remote_addr_bytes)))
else:
self.connection_event_stream = None
def test_dut_connects(self):
self.register_for_le_event(
hci_packets.SubeventCode.CONNECTION_COMPLETE)
with EventCallbackStream(self.cert_device.hci.FetchLeSubevents(empty_proto.Empty())) as cert_hci_le_event_stream, \
EventCallbackStream(self.cert_device.hci.FetchAclPackets(empty_proto.Empty())) as cert_acl_data_stream, \
EventCallbackStream(self.device_under_test.hci_le_acl_manager.FetchAclData(empty_proto.Empty())) as acl_data_stream:
cert_hci_le_event_asserts = EventAsserts(cert_hci_le_event_stream)
acl_data_asserts = EventAsserts(acl_data_stream)
cert_acl_data_asserts = EventAsserts(cert_acl_data_stream)
# Cert Advertises
advertising_handle = 0
self.enqueue_hci_command(
hci_packets.LeSetExtendedAdvertisingLegacyParametersBuilder(
advertising_handle,
hci_packets.LegacyAdvertisingProperties.ADV_IND,
400,
450,
7,
hci_packets.OwnAddressType.RANDOM_DEVICE_ADDRESS,
hci_packets.PeerAddressType.
PUBLIC_DEVICE_OR_IDENTITY_ADDRESS,
'00:00:00:00:00:00',
hci_packets.AdvertisingFilterPolicy.ALL_DEVICES,
0xF8,
1, #SID
hci_packets.Enable.DISABLED # Scan request notification
),
True)
self.enqueue_hci_command(
hci_packets.LeSetExtendedAdvertisingRandomAddressBuilder(
advertising_handle, '0C:05:04:03:02:01'), True)
gap_name = hci_packets.GapData()
gap_name.data_type = hci_packets.GapDataType.COMPLETE_LOCAL_NAME
gap_name.data = list(bytes(b'Im_A_Cert'))
self.enqueue_hci_command(
hci_packets.LeSetExtendedAdvertisingDataBuilder(
advertising_handle,
hci_packets.Operation.COMPLETE_ADVERTISEMENT,
hci_packets.FragmentPreference.CONTROLLER_SHOULD_NOT,
[gap_name]), True)
gap_short_name = hci_packets.GapData()
gap_short_name.data_type = hci_packets.GapDataType.SHORTENED_LOCAL_NAME
gap_short_name.data = list(bytes(b'Im_A_C'))
self.enqueue_hci_command(
hci_packets.LeSetExtendedAdvertisingScanResponseBuilder(
advertising_handle,
hci_packets.Operation.COMPLETE_ADVERTISEMENT,
hci_packets.FragmentPreference.CONTROLLER_SHOULD_NOT,
[gap_short_name]), True)
enabled_set = hci_packets.EnabledSet()
enabled_set.advertising_handle = advertising_handle
enabled_set.duration = 0
enabled_set.max_extended_advertising_events = 0
self.enqueue_hci_command(
hci_packets.LeSetExtendedAdvertisingEnableBuilder(
hci_packets.Enable.ENABLED, [enabled_set]), True)
with EventCallbackStream(
self.device_under_test.hci_le_acl_manager.CreateConnection(
le_acl_manager_facade.LeConnectionMsg(
address_type=int(
hci_packets.AddressType.RANDOM_DEVICE_ADDRESS),
address=bytes(
'0C:05:04:03:02:01',
'utf8')))) as connection_event_stream:
connection_event_asserts = EventAsserts(
connection_event_stream)
# Cert gets ConnectionComplete with a handle and sends ACL data
handle = 0xfff
def get_handle(packet):
packet_bytes = packet.event
nonlocal handle
if b'\x3e\x13\x01\x00' in packet_bytes:
cc_view = hci_packets.LeConnectionCompleteView(
hci_packets.LeMetaEventView(
hci_packets.EventPacketView(
bt_packets.PacketViewLittleEndian(
list(packet_bytes)))))
handle = cc_view.GetConnectionHandle()
return True
if b'\x3e\x13\x0A\x00' in packet_bytes:
cc_view = hci_packets.LeEnhancedConnectionCompleteView(
hci_packets.LeMetaEventView(
hci_packets.EventPacketView(
bt_packets.PacketViewLittleEndian(
list(packet_bytes)))))
handle = cc_view.GetConnectionHandle()
return True
return False
cert_hci_le_event_asserts.assert_event_occurs(get_handle)
cert_handle = handle
self.enqueue_acl_data(
cert_handle, hci_packets.PacketBoundaryFlag.
FIRST_AUTOMATICALLY_FLUSHABLE,
hci_packets.BroadcastFlag.POINT_TO_POINT,
bytes(b'\x19\x00\x07\x00SomeAclData from the Cert'))
# DUT gets a connection complete event and sends and receives
handle = 0xfff
connection_event_asserts.assert_event_occurs(get_handle)
self.device_under_test.hci_le_acl_manager.SendAclData(
le_acl_manager_facade.LeAclData(
handle=handle,
payload=bytes(
b'\x1C\x00\x07\x00SomeMoreAclData from the DUT')))
cert_acl_data_asserts.assert_event_occurs(
lambda packet: b'SomeMoreAclData' in packet.data)
acl_data_asserts.assert_event_occurs(
lambda packet: b'SomeAclData' in packet.payload)