Skip to content

Commit

Permalink
[tcat] Add TCAT advertisement parsing in client.
Browse files Browse the repository at this point in the history
Commit adds TCAT advertisement pars and display during scan operation
perormed in python client.
  • Loading branch information
canisLupus1313 committed Nov 13, 2024
1 parent 79572ec commit 39a37aa
Show file tree
Hide file tree
Showing 5 changed files with 169 additions and 5 deletions.
53 changes: 53 additions & 0 deletions tools/tcat_ble_client/ble/ble_advertisement_data.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
"""
Copyright (c) 2024, The OpenThread Authors.
All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met:
1. Redistributions of source code must retain the above copyright
notice, this list of conditions and the following disclaimer.
2. Redistributions in binary form must reproduce the above copyright
notice, this list of conditions and the following disclaimer in the
documentation and/or other materials provided with the distribution.
3. Neither the name of the copyright holder nor the
names of its contributors may be used to endorse or promote products
derived from this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
POSSIBILITY OF SUCH DAMAGE.
"""

from tlv import advertised_tlv


class AdvertisedData:

def __init__(self):
self._op_code = None
self._version = None
self._tlvs = []

def __str__(self) -> str:
res = ''
res += f'OPCODE: {self._op_code} THREAD VERSION: {self._version}\n'
res += 'TLVs:\n'
for tlv in self._tlvs:
res += f'\t{str(tlv)}\n'
return res

@staticmethod
def from_bytes(data: bytes):
res = AdvertisedData()
res._op_code = data[0] & 0x0f
res._version = (data[0] & 0xf0) >> 4
res._tlvs.extend(advertised_tlv.parse_tlvs(data[1:]))
return res
9 changes: 7 additions & 2 deletions tools/tcat_ble_client/ble/ble_scanner.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@
from bleak.backends.device import BLEDevice
from bbtc import BBTC_SERVICE_UUID
from typing import Optional
from ble.ble_advertisement_data import AdvertisedData


async def find_first_by_name(name):
Expand All @@ -50,7 +51,11 @@ async def scan_tcat_devices(adapter: Optional[str] = None):
discovered_devices = await scanner.discover(return_adv=True,
service_uuids=[BBTC_SERVICE_UUID.lower()],
adapter=adapter)
for _, (device, _) in discovered_devices.items():
tcat_devices.append(device)
for _, (device, adv) in discovered_devices.items():
ad = None
for uuid, data in adv.service_data.items():
if BBTC_SERVICE_UUID.lower() in uuid:
ad = AdvertisedData.from_bytes(data)
tcat_devices.append((device, ad))

return tcat_devices
104 changes: 104 additions & 0 deletions tools/tcat_ble_client/tlv/advertised_tlv.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,104 @@
"""
Copyright (c) 2024, The OpenThread Authors.
All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met:
1. Redistributions of source code must retain the above copyright
notice, this list of conditions and the following disclaimer.
2. Redistributions in binary form must reproduce the above copyright
notice, this list of conditions and the following disclaimer in the
documentation and/or other materials provided with the distribution.
3. Neither the name of the copyright holder nor the
names of its contributors may be used to endorse or promote products
derived from this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
POSSIBILITY OF SUCH DAMAGE.
"""


class AdvertisedTlv:
"""
Represents a TCAT Advertisement TLV, used to advertise capabilities of a TCAT Device.
"""

DEVICE_TYPE_STATUS_TLV = 4
CAPABILITIES_TLV = 5

def __init__(self, type: int = None, size: int = None, value: bytes = None):
self._type: int = type
self._size: int = size
self._value: bytes = value

def __str__(self) -> str:
return f'TYPE 0x{self._type:02x} VALUE: {str(self._value)}'

def set_from_bytes(self, data: bytes):
self._type = (data[0] & 0xf0) >> 4
self._size = data[0] & 0xf
header_len = 1
self._value = data[header_len:header_len + self._size]


class CapabilitiesTlv(AdvertisedTlv):
G = 0x80
L = 0x40

def __str__(self) -> str:
val = int.from_bytes(self._value, 'big')
return f'TYPE 0x{self._type:02x} G: {is_set(val, self.G)} L: {is_set(val, self.L)}'


class DeviceTypeStatusTlv(AdvertisedTlv):
D = 0x80
R = 0x40
B = 0x20
T = 0x10
C = 0x08
S = 0x04
M = 0x02

def __str__(self) -> str:
val = int.from_bytes(self._value, 'big')
return f'TYPE 0x{self._type:02x} D: {is_set(val, self.D)} R: {is_set(val, self.R)} B: {is_set(val, self.B)} T: {is_set(val, self.T)} C: {is_set(val, self.C)} S: {is_set(val, self.S)} M: {is_set(val, self.M)}'


def is_set(val, flag):
return 1 if val & flag else 0


def _create_tlv(data: bytes):
res = None
header_len = 1
type = (data[0] & 0xf0) >> 4
size = data[0] & 0xf
value = data[header_len:header_len + size]

if type == AdvertisedTlv.DEVICE_TYPE_STATUS_TLV:
res = DeviceTypeStatusTlv(type, size, value)
elif type == AdvertisedTlv.CAPABILITIES_TLV:
res = CapabilitiesTlv(type, size, value)
else:
res = AdvertisedTlv(type, size, value)

return res


def parse_tlvs(data: bytes):
res = []
while data:
next_tlv = _create_tlv(data)
header_len = 1
data = data[next_tlv._size + header_len:]
res.append(next_tlv)
return res
2 changes: 1 addition & 1 deletion tools/tcat_ble_client/tlv/tlv.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ def __init__(self, type: int = None, value: bytes = None):
self.value: bytes = value

def __str__(self):
return f'TLV\n\tTYPE:\t0x{self.type:02x}\n\tVALUE:\t{self.value}'
return f'TLV\n\tTYPE:\t0x{self.type:02x}\n\tVALUE:\t{self.value.hex()}'

@staticmethod
def parse_tlvs(data: bytes) -> List[TLV]:
Expand Down
6 changes: 4 additions & 2 deletions tools/tcat_ble_client/utils/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
"""

import base64
from tlv import advertised_tlv


def get_int_in_range(min_value, max_value):
Expand All @@ -51,15 +52,16 @@ def quit_with_reason(reason):
def select_device_by_user_input(tcat_devices):
if tcat_devices:
print('Found devices:\n')
for i, device in enumerate(tcat_devices):
for i, (device, adv) in enumerate(tcat_devices):
print(f'{i + 1}: {device.name} - {device.address}')
print(str(adv))
else:
print('\nNo devices found.')
return None

print('\nSelect the target number to connect to it.')
selected = get_int_in_range(1, len(tcat_devices))
device = tcat_devices[selected - 1]
device = tcat_devices[selected - 1][0]
print('Selected ', device)

return device
Expand Down

0 comments on commit 39a37aa

Please sign in to comment.