def setup_test(self):
super().setup_test()
self.cert_hci = PyHci(self.cert, acl_streaming=True)
self.cert_hci.send_command(hci_packets.WriteScanEnableBuilder(hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN))
self.cert_name = b'Im_A_Cert'
self.cert_address = self.cert_hci.read_own_address()
self.cert_name += b'@' + self.cert_address.encode('utf8')
self.dut_neighbor = PyNeighbor(self.dut)
def test_inquiry_from_dut(self):
inquiry_msg = neighbor_facade.InquiryMsg(
inquiry_mode=neighbor_facade.DiscoverabilityMode.GENERAL,
result_mode=neighbor_facade.ResultMode.STANDARD,
length_1_28s=3,
max_results=0)
session = self.dut_neighbor.set_inquiry_mode(inquiry_msg)
self.cert_hci.send_command(hci_packets.WriteScanEnableBuilder(hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN))
assertThat(session).emits(NeighborMatchers.InquiryResult(self.cert_address), timeout=timedelta(seconds=10))
def test_connection_dut_connects(self):
self.dut_hci.send_command_with_complete(
hci_packets.WritePageTimeoutBuilder(0x4000))
# CERT Enables scans and gets its address
self.send_hal_hci_command(hci_packets.ReadBdAddrBuilder())
cert_read_bd_addr_capture = HalCaptures.ReadBdAddrCompleteCapture()
assertThat(self.cert_hal.get_hci_event_stream()).emits(
cert_read_bd_addr_capture)
address = cert_read_bd_addr_capture.get().GetBdAddr()
self.send_hal_hci_command(
hci_packets.WriteScanEnableBuilder(
hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN))
# DUT Connects
self.dut_hci.send_command_with_status(
hci_packets.CreateConnectionBuilder(
address,
0xcc18, # Packet Type
hci_packets.PageScanRepetitionMode.R0,
0,
hci_packets.ClockOffsetValid.INVALID,
hci_packets.CreateConnectionRoleSwitch.ALLOW_ROLE_SWITCH))
# Cert Accepts
connect_request_capture = HalCaptures.ConnectionRequestCapture()
assertThat(self.cert_hal.get_hci_event_stream()).emits(
connect_request_capture, timeout=timedelta(seconds=20))
connection_request = connect_request_capture.get()
self.send_hal_hci_command(
hci_packets.AcceptConnectionRequestBuilder(
connection_request.GetBdAddr(),
hci_packets.AcceptConnectionRequestRole.REMAIN_SLAVE))
(dut_handle, cert_handle) = self._verify_connection_complete()
# Send ACL Data
self.enqueue_acl_data(
dut_handle,
hci_packets.PacketBoundaryFlag.FIRST_NON_AUTOMATICALLY_FLUSHABLE,
hci_packets.BroadcastFlag.POINT_TO_POINT,
bytes(b'Just SomeAclData'))
self.send_hal_acl_data(
cert_handle,
hci_packets.PacketBoundaryFlag.FIRST_NON_AUTOMATICALLY_FLUSHABLE,
hci_packets.BroadcastFlag.POINT_TO_POINT,
bytes(b'Just SomeMoreAclData'))
assertThat(self.cert_hal.get_acl_stream()).emits(
lambda packet: b'SomeAclData' in packet.payload)
assertThat(self.dut_hci.get_raw_acl_stream()).emits(
lambda packet: b'SomeMoreAclData' in packet.data)
def test_inquiry_from_dut(self):
self.dut_hci.register_for_events(hci_packets.EventCode.INQUIRY_RESULT)
self.send_hal_hci_command(
hci_packets.WriteScanEnableBuilder(
hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN))
lap = hci_packets.Lap()
lap.lap = 0x33
self.dut_hci.send_command_with_status(
hci_packets.InquiryBuilder(lap, 0x30, 0xff))
assertThat(self.dut_hci.get_event_stream()).emits(
HciMatchers.EventWithCode(hci_packets.EventCode.INQUIRY_RESULT))
def test_remote_name(self):
self.register_for_dut_event(
hci_packets.EventCode.REMOTE_HOST_SUPPORTED_FEATURES_NOTIFICATION)
with EventCallbackStream(self.cert_device.hci.FetchEvents(empty_proto.Empty())) as hci_event_stream, \
EventCallbackStream(self.device_under_test.neighbor.GetRemoteNameEvents(empty_proto.Empty())) as name_event_stream:
name_event_asserts = EventAsserts(name_event_stream)
hci_event_asserts = EventAsserts(hci_event_stream)
cert_name = b'Im_A_Cert'
padded_name = cert_name
while len(padded_name) < 248:
padded_name = padded_name + b'\0'
self.enqueue_hci_command(
hci_packets.WriteLocalNameBuilder(padded_name), True)
hci_event_asserts.assert_event_occurs(
lambda msg: b'\x0e\x04\x01\x13\x0c' in msg.event)
address = hci_packets.Address()
def get_address_from_complete(packet):
packet_bytes = packet.event
if b'\x0e\x0a\x01\x09\x10' in packet_bytes:
nonlocal address
addr_view = hci_packets.ReadBdAddrCompleteView(
hci_packets.CommandCompleteView(
hci_packets.EventPacketView(
bt_packets.PacketViewLittleEndian(
list(packet_bytes)))))
address = addr_view.GetBdAddr()
return True
return False
# DUT Enables scans and gets its address
self.enqueue_hci_command(
hci_packets.WriteScanEnableBuilder(
hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN), True)
self.enqueue_hci_command(hci_packets.ReadBdAddrBuilder(), True)
hci_event_asserts.assert_event_occurs(get_address_from_complete)
cert_address = address.encode('utf8')
self.device_under_test.neighbor.ReadRemoteName(
neighbor_facade.RemoteNameRequestMsg(
address=cert_address,
page_scan_repetition_mode=1,
clock_offset=0x6855))
name_event_asserts.assert_event_occurs(
lambda msg: cert_name in msg.name)
def test_inquiry_from_dut(self):
self.cert_hal.send_hci_command(
hci_packets.WriteScanEnableBuilder(
hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN))
lap = hci_packets.Lap()
lap.lap = 0x33
self.dut_hal.send_hci_command(
hci_packets.InquiryBuilder(lap, 0x30, 0xff))
assertThat(self.dut_hal.get_hci_event_stream()).emits(
lambda packet: b'\x02\x0f' in packet.payload
# Expecting an HCI Event (code 0x02, length 0x0f)
)
def test_inquiry_rssi_from_dut(self):
inquiry_msg = neighbor_facade.InquiryMsg(
inquiry_mode=neighbor_facade.DiscoverabilityMode.GENERAL,
result_mode=neighbor_facade.ResultMode.RSSI,
length_1_28s=3,
max_results=0)
with EventStream(self.dut.neighbor.SetInquiryMode(
inquiry_msg)) as inquiry_event_stream:
self.enqueue_hci_command(
hci_packets.WriteScanEnableBuilder(
hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN), True)
inquiry_event_stream.assert_event_occurs(
lambda msg: b'\x22\x0f' in msg.packet
# Expecting an HCI Event (code 0x22, length 0x0f)
)
def test_inquiry_from_dut(self):
with EventCallbackStream(
self.device_under_test.hal.FetchHciEvent(
empty_pb2.Empty())) as hci_event_stream:
hci_event_asserts = EventAsserts(hci_event_stream)
self.send_cert_hci_command(
hci_packets.WriteScanEnableBuilder(
hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN))
lap = hci_packets.Lap()
lap.lap = 0x33
self.send_dut_hci_command(
hci_packets.InquiryBuilder(lap, 0x30, 0xff))
hci_event_asserts.assert_event_occurs(
lambda packet: b'\x02\x0f' in packet.payload
# Expecting an HCI Event (code 0x02, length 0x0f)
)
def test_inquiry_from_dut(self):
self.register_for_event(hci_packets.EventCode.INQUIRY_RESULT)
with EventStream(self.dut.hci.FetchEvents(
empty_proto.Empty())) as hci_event_stream:
self.send_hal_hci_command(
hci_packets.WriteScanEnableBuilder(
hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN))
lap = hci_packets.Lap()
lap.lap = 0x33
self.enqueue_hci_command(
hci_packets.InquiryBuilder(lap, 0x30, 0xff), False)
hci_event_stream.assert_event_occurs(
lambda packet: b'\x02\x0f' in packet.event
# Expecting an HCI Event (code 0x02, length 0x0f)
)
def test_inquiry_from_dut(self):
inquiry_msg = neighbor_facade.InquiryMsg(
inquiry_mode=neighbor_facade.DiscoverabilityMode.GENERAL,
result_mode=neighbor_facade.ResultMode.STANDARD,
length_1_28s=3,
max_results=0)
with EventCallbackStream(
self.device_under_test.neighbor.SetInquiryMode(
inquiry_msg)) as inquiry_event_stream:
hci_event_asserts = EventAsserts(inquiry_event_stream)
self.enqueue_hci_command(
hci_packets.WriteScanEnableBuilder(
hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN), True)
hci_event_asserts.assert_event_occurs(
lambda msg: b'\x02\x0f' in msg.packet
# Expecting an HCI Event (code 0x02, length 0x0f)
)
def test_inquiry_extended_from_dut(self):
self._set_name()
gap_name = hci_packets.GapData()
gap_name.data_type = hci_packets.GapDataType.COMPLETE_LOCAL_NAME
gap_name.data = list(bytes(self.cert_name))
gap_data = list([gap_name])
self.cert_hci.send_command(
hci_packets.WriteExtendedInquiryResponseBuilder(hci_packets.FecRequired.NOT_REQUIRED, gap_data))
inquiry_msg = neighbor_facade.InquiryMsg(
inquiry_mode=neighbor_facade.DiscoverabilityMode.GENERAL,
result_mode=neighbor_facade.ResultMode.EXTENDED,
length_1_28s=8,
max_results=0)
session = self.dut_neighbor.set_inquiry_mode(inquiry_msg)
self.cert_hci.send_command(hci_packets.WriteScanEnableBuilder(hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN))
assertThat(session).emits(
NeighborMatchers.ExtendedInquiryResult(self.cert_address), timeout=timedelta(seconds=10))
def test_inquiry_extended_from_dut(self):
name_string = b'Im_A_Cert'
gap_name = hci_packets.GapData()
gap_name.data_type = hci_packets.GapDataType.COMPLETE_LOCAL_NAME
gap_name.data = list(bytes(name_string))
gap_data = list([gap_name])
self.enqueue_hci_command(
hci_packets.WriteExtendedInquiryResponseBuilder(
hci_packets.FecRequired.NOT_REQUIRED, gap_data), True)
inquiry_msg = neighbor_facade.InquiryMsg(
inquiry_mode=neighbor_facade.DiscoverabilityMode.GENERAL,
result_mode=neighbor_facade.ResultMode.EXTENDED,
length_1_28s=3,
max_results=0)
with EventStream(self.dut.neighbor.SetInquiryMode(
inquiry_msg)) as inquiry_event_stream:
self.enqueue_hci_command(
hci_packets.WriteScanEnableBuilder(
hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN), True)
inquiry_event_stream.assert_event_occurs(
lambda msg: name_string in msg.packet)
def enable_inquiry_and_page_scan(self):
self.send_command_with_complete(
hci_packets.WriteScanEnableBuilder(
hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN))
def test_connection_cert_connects(self):
self.register_for_event(hci_packets.EventCode.CONNECTION_COMPLETE)
self.register_for_event(
hci_packets.EventCode.CONNECTION_PACKET_TYPE_CHANGED)
self.register_for_event(hci_packets.EventCode.CONNECTION_REQUEST)
self.send_hal_hci_command(hci_packets.WritePageTimeoutBuilder(0x4000))
with EventStream(self.dut.hci.FetchEvents(empty_proto.Empty())) as hci_event_stream, \
EventStream(self.dut.hci.FetchAclPackets(empty_proto.Empty())) as acl_data_stream, \
EventStream(self.cert.hal.FetchHciEvent(empty_proto.Empty())) as cert_hci_event_stream, \
EventStream(self.cert.hal.FetchHciAcl(empty_proto.Empty())) as cert_acl_data_stream:
address = hci_packets.Address()
def get_address_from_complete(packet):
packet_bytes = packet.event
if b'\x0e\x0a\x01\x09\x10' in packet_bytes:
nonlocal address
addr_view = hci_packets.ReadBdAddrCompleteView(
hci_packets.CommandCompleteView(
hci_packets.EventPacketView(
bt_packets.PacketViewLittleEndian(
list(packet_bytes)))))
address = addr_view.GetBdAddr()
return True
return False
# DUT Enables scans and gets its address
self.enqueue_hci_command(
hci_packets.WriteScanEnableBuilder(
hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN), True)
self.enqueue_hci_command(hci_packets.ReadBdAddrBuilder(), True)
hci_event_stream.assert_event_occurs(get_address_from_complete)
# Cert Connects
self.send_hal_hci_command(
hci_packets.CreateConnectionBuilder(
address,
0xcc18, # Packet Type
hci_packets.PageScanRepetitionMode.R0,
0,
hci_packets.ClockOffsetValid.INVALID,
hci_packets.CreateConnectionRoleSwitch.ALLOW_ROLE_SWITCH))
# DUT Accepts
connection_request = None
def get_connect_request(packet):
if b'\x04\x0a' in packet.event:
nonlocal connection_request
connection_request = hci_packets.ConnectionRequestView(
hci_packets.EventPacketView(
bt_packets.PacketViewLittleEndian(
list(packet.event))))
return True
return False
hci_event_stream.assert_event_occurs(
get_connect_request, timeout=timedelta(seconds=20))
self.enqueue_hci_command(
hci_packets.AcceptConnectionRequestBuilder(
connection_request.GetBdAddr(),
hci_packets.AcceptConnectionRequestRole.REMAIN_SLAVE),
False)
conn_handle = 0xfff
def get_handle(packet_bytes):
if b'\x03\x0b\x00' in packet_bytes:
nonlocal conn_handle
cc_view = hci_packets.ConnectionCompleteView(
hci_packets.EventPacketView(
bt_packets.PacketViewLittleEndian(
list(packet_bytes))))
conn_handle = cc_view.GetConnectionHandle()
return True
return False
def event_handle(packet):
packet_bytes = packet.event
return get_handle(packet_bytes)
def payload_handle(packet):
packet_bytes = packet.payload
return get_handle(packet_bytes)
cert_hci_event_stream.assert_event_occurs(payload_handle)
cert_handle = conn_handle
conn_handle = 0xfff
hci_event_stream.assert_event_occurs(event_handle)
dut_handle = conn_handle
if dut_handle == 0xfff:
logging.warning("Failed to get the DUT handle")
return False
if cert_handle == 0xfff:
logging.warning("Failed to get the CERT handle")
return False
# Send ACL Data
self.enqueue_acl_data(
dut_handle, hci_packets.PacketBoundaryFlag.
FIRST_NON_AUTOMATICALLY_FLUSHABLE,
hci_packets.BroadcastFlag.POINT_TO_POINT,
bytes(b'This is just SomeAclData'))
self.send_hal_acl_data(
cert_handle, hci_packets.PacketBoundaryFlag.
FIRST_NON_AUTOMATICALLY_FLUSHABLE,
hci_packets.BroadcastFlag.POINT_TO_POINT,
bytes(b'This is just SomeMoreAclData'))
cert_acl_data_stream.assert_event_occurs(
lambda packet: b'SomeAclData' in packet.payload)
acl_data_stream.assert_event_occurs(
lambda packet: b'SomeMoreAclData' in packet.data)
def test_dut_connects(self):
self.register_for_event(hci_packets.EventCode.CONNECTION_REQUEST)
self.register_for_event(hci_packets.EventCode.CONNECTION_COMPLETE)
self.register_for_event(
hci_packets.EventCode.CONNECTION_PACKET_TYPE_CHANGED)
with EventCallbackStream(self.cert_device.hci.FetchEvents(empty_proto.Empty())) as cert_hci_event_stream, \
EventCallbackStream(self.cert_device.hci.FetchAclPackets(empty_proto.Empty())) as cert_acl_data_stream, \
EventCallbackStream(self.device_under_test.hci_acl_manager.FetchAclData(empty_proto.Empty())) as acl_data_stream:
cert_hci_event_asserts = EventAsserts(cert_hci_event_stream)
acl_data_asserts = EventAsserts(acl_data_stream)
cert_acl_data_asserts = EventAsserts(cert_acl_data_stream)
# CERT Enables scans and gets its address
self.enqueue_hci_command(
hci_packets.WriteScanEnableBuilder(
hci_packets.ScanEnable.INQUIRY_AND_PAGE_SCAN), True)
cert_address = None
def get_address_from_complete(packet):
packet_bytes = packet.event
if b'\x0e\x0a\x01\x09\x10' in packet_bytes:
nonlocal cert_address
addr_view = hci_packets.ReadBdAddrCompleteView(
hci_packets.CommandCompleteView(
hci_packets.EventPacketView(
bt_packets.PacketViewLittleEndian(
list(packet_bytes)))))
cert_address = addr_view.GetBdAddr()
return True
return False
self.enqueue_hci_command(hci_packets.ReadBdAddrBuilder(), True)
cert_hci_event_asserts.assert_event_occurs(
get_address_from_complete)
with EventCallbackStream(
self.device_under_test.hci_acl_manager.CreateConnection(
acl_manager_facade.ConnectionMsg(
address_type=int(
hci_packets.AddressType.PUBLIC_DEVICE_ADDRESS),
address=bytes(
cert_address,
'utf8')))) as connection_event_stream:
connection_event_asserts = EventAsserts(
connection_event_stream)
connection_request = None
def get_connect_request(packet):
if b'\x04\x0a' in packet.event:
nonlocal connection_request
connection_request = hci_packets.ConnectionRequestView(
hci_packets.EventPacketView(
bt_packets.PacketViewLittleEndian(
list(packet.event))))
return True
return False
# Cert Accepts
cert_hci_event_asserts.assert_event_occurs(get_connect_request)
self.enqueue_hci_command(
hci_packets.AcceptConnectionRequestBuilder(
connection_request.GetBdAddr(),
hci_packets.AcceptConnectionRequestRole.REMAIN_SLAVE),
False)
# Cert gets ConnectionComplete with a handle and sends ACL data
handle = 0xfff
def get_handle(packet):
packet_bytes = packet.event
if b'\x03\x0b\x00' in packet_bytes:
nonlocal handle
cc_view = hci_packets.ConnectionCompleteView(
hci_packets.EventPacketView(
bt_packets.PacketViewLittleEndian(
list(packet_bytes))))
handle = cc_view.GetConnectionHandle()
return True
return False
cert_hci_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'\x26\x00\x07\x00This is just SomeAclData 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_acl_manager.SendAclData(
acl_manager_facade.AclData(
handle=handle,
payload=bytes(
b'\x29\x00\x07\x00This is just SomeMoreAclData 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)
