mirror of
https://github.com/google/bumble.git
synced 2026-06-08 08:42:26 +00:00
Compare commits
39 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 32bb7cdaf3 | |||
| b4261548e8 | |||
| 9161cea577 | |||
| 3f643de4c1 | |||
| 7c7b792cf9 | |||
| 8e28f4e159 | |||
| 8823cf108f | |||
| 4fb501a0ef | |||
| ad0753b959 | |||
| f12cccf6cd | |||
| 5bbbe5e40f | |||
| 793fcd750c | |||
| ae2c638256 | |||
| 9ad0eafe37 | |||
| 618e977f20 | |||
| 7fdc4f624e | |||
| 255ca60d95 | |||
| 716f57de46 | |||
| 95a987d3a4 | |||
| 6858c591aa | |||
| e03b9cb441 | |||
| ade36f8d04 | |||
| 48744ee9db | |||
| 302e496890 | |||
| 6649464cd6 | |||
| c46df21385 | |||
| 7a35f5d095 | |||
| 73f2853c5e | |||
| de3009e296 | |||
| e47cb5512c | |||
| 3171b5a19e | |||
| 456cb59b48 | |||
| 33ca324e41 | |||
| a84f0279b1 | |||
| b93ba007ed | |||
| d2a4c2a8e4 | |||
| 57e05781ad | |||
| bae6c1df97 | |||
| 7292c2785e |
@@ -23,13 +23,13 @@ jobs:
|
||||
|
||||
steps:
|
||||
- name: Check out from Git
|
||||
uses: actions/checkout@v3
|
||||
uses: actions/checkout@v6
|
||||
- name: Get history and tags for SCM versioning to work
|
||||
run: |
|
||||
git fetch --prune --unshallow
|
||||
git fetch --depth=1 origin +refs/tags/*:refs/tags/*
|
||||
- name: Set up Python
|
||||
uses: actions/setup-python@v3
|
||||
uses: actions/setup-python@v6
|
||||
with:
|
||||
python-version: ${{ matrix.python-version }}
|
||||
- name: Install dependencies
|
||||
|
||||
@@ -40,7 +40,7 @@ jobs:
|
||||
|
||||
steps:
|
||||
- name: Checkout repository
|
||||
uses: actions/checkout@v3
|
||||
uses: actions/checkout@v6
|
||||
|
||||
# Initializes the CodeQL tools for scanning.
|
||||
- name: Initialize CodeQL
|
||||
|
||||
@@ -22,10 +22,10 @@ jobs:
|
||||
|
||||
steps:
|
||||
- name: Check out from Git
|
||||
uses: actions/checkout@v3
|
||||
uses: actions/checkout@v6
|
||||
|
||||
- name: Set up JDK
|
||||
uses: actions/setup-java@v4
|
||||
uses: actions/setup-java@v5
|
||||
with:
|
||||
distribution: 'zulu'
|
||||
java-version: 17
|
||||
|
||||
@@ -26,9 +26,9 @@ jobs:
|
||||
21/24, 22/24, 23/24, 24/24,
|
||||
]
|
||||
steps:
|
||||
- uses: actions/checkout@v3
|
||||
- uses: actions/checkout@v6
|
||||
- name: Set Up Python 3.11
|
||||
uses: actions/setup-python@v4
|
||||
uses: actions/setup-python@v6
|
||||
with:
|
||||
python-version: 3.11
|
||||
- name: Install
|
||||
@@ -46,7 +46,7 @@ jobs:
|
||||
run: cat rootcanal.log
|
||||
- name: Upload Mobly logs
|
||||
if: always()
|
||||
uses: actions/upload-artifact@v4
|
||||
uses: actions/upload-artifact@v6
|
||||
with:
|
||||
name: mobly-logs-${{ strategy.job-index }}
|
||||
path: /tmp/logs/mobly/bumble.bumbles/
|
||||
|
||||
@@ -23,13 +23,13 @@ jobs:
|
||||
|
||||
steps:
|
||||
- name: Check out from Git
|
||||
uses: actions/checkout@v3
|
||||
uses: actions/checkout@v6
|
||||
- name: Get history and tags for SCM versioning to work
|
||||
run: |
|
||||
git fetch --prune --unshallow
|
||||
git fetch --depth=1 origin +refs/tags/*:refs/tags/*
|
||||
- name: Set up Python ${{ matrix.python-version }}
|
||||
uses: actions/setup-python@v4
|
||||
uses: actions/setup-python@v6
|
||||
with:
|
||||
python-version: ${{ matrix.python-version }}
|
||||
- name: Install dependencies
|
||||
@@ -50,13 +50,13 @@ jobs:
|
||||
matrix:
|
||||
# Rust runtime doesn't support 3.14 yet.
|
||||
python-version: ["3.10", "3.11", "3.12", "3.13"]
|
||||
rust-version: [ "1.80.0", "stable" ]
|
||||
rust-version: [ "1.80.0", "1.91.0" ]
|
||||
fail-fast: false
|
||||
steps:
|
||||
- name: Check out from Git
|
||||
uses: actions/checkout@v3
|
||||
uses: actions/checkout@v6
|
||||
- name: Set up Python ${{ matrix.python-version }}
|
||||
uses: actions/setup-python@v4
|
||||
uses: actions/setup-python@v6
|
||||
with:
|
||||
python-version: ${{ matrix.python-version }}
|
||||
- name: Install Python dependencies
|
||||
@@ -69,7 +69,7 @@ jobs:
|
||||
components: clippy,rustfmt
|
||||
toolchain: ${{ matrix.rust-version }}
|
||||
- name: Install Rust dependencies
|
||||
run: cargo install cargo-all-features # allows building/testing combinations of features
|
||||
run: cargo install cargo-all-features --version 1.11.0 # allows building/testing combinations of features
|
||||
- name: Check License Headers
|
||||
run: cd rust && cargo run --features dev-tools --bin file-header check-all
|
||||
- name: Rust Build
|
||||
|
||||
@@ -14,13 +14,13 @@ jobs:
|
||||
|
||||
steps:
|
||||
- name: Check out from Git
|
||||
uses: actions/checkout@v3
|
||||
uses: actions/checkout@v6
|
||||
- name: Get history and tags for SCM versioning to work
|
||||
run: |
|
||||
git fetch --prune --unshallow
|
||||
git fetch --depth=1 origin +refs/tags/*:refs/tags/*
|
||||
- name: Set up Python
|
||||
uses: actions/setup-python@v3
|
||||
uses: actions/setup-python@v6
|
||||
with:
|
||||
python-version: '3.10'
|
||||
- name: Install dependencies
|
||||
@@ -31,7 +31,7 @@ jobs:
|
||||
run: python -m build
|
||||
- name: Publish package to PyPI
|
||||
if: github.event_name == 'release' && startsWith(github.ref, 'refs/tags')
|
||||
uses: pypa/gh-action-pypi-publish@release/v1
|
||||
uses: pypa/gh-action-pypi-publish@ed0c53931b1dc9bd32cbe73a98c7f6766f8a527e # release/v1.13
|
||||
with:
|
||||
user: __token__
|
||||
password: ${{ secrets.PYPI_API_TOKEN }}
|
||||
|
||||
+762
-489
File diff suppressed because it is too large
Load Diff
+4
-5
@@ -22,7 +22,6 @@ import logging
|
||||
import statistics
|
||||
import struct
|
||||
import time
|
||||
from typing import Optional
|
||||
|
||||
import click
|
||||
|
||||
@@ -257,8 +256,8 @@ async def pre_power_on(device: Device, classic: bool) -> None:
|
||||
|
||||
async def post_power_on(
|
||||
device: Device,
|
||||
le_scan: Optional[tuple[int, int]],
|
||||
le_advertise: Optional[int],
|
||||
le_scan: tuple[int, int] | None,
|
||||
le_advertise: int | None,
|
||||
classic_page_scan: bool,
|
||||
classic_inquiry_scan: bool,
|
||||
) -> None:
|
||||
@@ -1300,7 +1299,7 @@ class IsoClient(StreamedPacketIO):
|
||||
super().__init__()
|
||||
self.device = device
|
||||
self.ready = asyncio.Event()
|
||||
self.cis_link: Optional[CisLink] = None
|
||||
self.cis_link: CisLink | None = None
|
||||
|
||||
async def on_connection(
|
||||
self, connection: Connection, cis_link: CisLink, sender: bool
|
||||
@@ -1341,7 +1340,7 @@ class IsoServer(StreamedPacketIO):
|
||||
):
|
||||
super().__init__()
|
||||
self.device = device
|
||||
self.cis_link: Optional[CisLink] = None
|
||||
self.cis_link: CisLink | None = None
|
||||
self.ready = asyncio.Event()
|
||||
|
||||
logging.info(
|
||||
|
||||
+5
-10
@@ -24,7 +24,6 @@ import logging
|
||||
import os
|
||||
import re
|
||||
from collections import OrderedDict
|
||||
from typing import Optional, Union
|
||||
|
||||
import click
|
||||
import humanize
|
||||
@@ -126,8 +125,8 @@ def parse_phys(phys):
|
||||
# Console App
|
||||
# -----------------------------------------------------------------------------
|
||||
class ConsoleApp:
|
||||
connected_peer: Optional[Peer]
|
||||
connection_phy: Optional[ConnectionPHY]
|
||||
connected_peer: Peer | None
|
||||
connection_phy: ConnectionPHY | None
|
||||
|
||||
def __init__(self):
|
||||
self.known_addresses = set()
|
||||
@@ -520,7 +519,7 @@ class ConsoleApp:
|
||||
|
||||
self.show_attributes(attributes)
|
||||
|
||||
def find_remote_characteristic(self, param) -> Optional[CharacteristicProxy]:
|
||||
def find_remote_characteristic(self, param) -> CharacteristicProxy | None:
|
||||
if not self.connected_peer:
|
||||
return None
|
||||
parts = param.split('.')
|
||||
@@ -542,9 +541,7 @@ class ConsoleApp:
|
||||
|
||||
return None
|
||||
|
||||
def find_local_attribute(
|
||||
self, param
|
||||
) -> Optional[Union[Characteristic, Descriptor]]:
|
||||
def find_local_attribute(self, param) -> Characteristic | Descriptor | None:
|
||||
parts = param.split('.')
|
||||
if len(parts) == 3:
|
||||
service_uuid = UUID(parts[0])
|
||||
@@ -1096,9 +1093,7 @@ class DeviceListener(Device.Listener, Connection.Listener):
|
||||
if self.app.connected_peer.connection.is_encrypted
|
||||
else 'not encrypted'
|
||||
)
|
||||
self.app.append_to_output(
|
||||
'connection encryption change: ' f'{encryption_state}'
|
||||
)
|
||||
self.app.append_to_output(f'connection encryption change: {encryption_state}')
|
||||
|
||||
def on_connection_data_length_change(self):
|
||||
self.app.append_to_output(
|
||||
|
||||
+12
-7
@@ -35,8 +35,6 @@ from bumble.hci import (
|
||||
HCI_READ_BUFFER_SIZE_COMMAND,
|
||||
HCI_READ_LOCAL_NAME_COMMAND,
|
||||
HCI_SUCCESS,
|
||||
HCI_VERSION_NAMES,
|
||||
LMP_VERSION_NAMES,
|
||||
CodecID,
|
||||
HCI_Command,
|
||||
HCI_Command_Complete_Event,
|
||||
@@ -54,6 +52,7 @@ from bumble.hci import (
|
||||
HCI_Read_Local_Supported_Codecs_V2_Command,
|
||||
HCI_Read_Local_Version_Information_Command,
|
||||
LeFeature,
|
||||
SpecificationVersion,
|
||||
map_null_terminated_utf8_string,
|
||||
)
|
||||
from bumble.host import Host
|
||||
@@ -275,7 +274,7 @@ async def async_main(
|
||||
(
|
||||
f'min={min(latencies):.2f}, '
|
||||
f'max={max(latencies):.2f}, '
|
||||
f'average={sum(latencies)/len(latencies):.2f},'
|
||||
f'average={sum(latencies) / len(latencies):.2f},'
|
||||
),
|
||||
[f'{latency:.4}' for latency in latencies],
|
||||
'\n',
|
||||
@@ -289,14 +288,20 @@ async def async_main(
|
||||
)
|
||||
print(
|
||||
color(' HCI Version: ', 'green'),
|
||||
name_or_number(HCI_VERSION_NAMES, host.local_version.hci_version),
|
||||
SpecificationVersion(host.local_version.hci_version).name,
|
||||
)
|
||||
print(
|
||||
color(' HCI Subversion:', 'green'),
|
||||
f'0x{host.local_version.hci_subversion:04x}',
|
||||
)
|
||||
print(color(' HCI Subversion:', 'green'), host.local_version.hci_subversion)
|
||||
print(
|
||||
color(' LMP Version: ', 'green'),
|
||||
name_or_number(LMP_VERSION_NAMES, host.local_version.lmp_version),
|
||||
SpecificationVersion(host.local_version.lmp_version).name,
|
||||
)
|
||||
print(
|
||||
color(' LMP Subversion:', 'green'),
|
||||
f'0x{host.local_version.lmp_subversion:04x}',
|
||||
)
|
||||
print(color(' LMP Subversion:', 'green'), host.local_version.lmp_subversion)
|
||||
|
||||
# Get the Classic info
|
||||
await get_classic_info(host)
|
||||
|
||||
@@ -17,7 +17,6 @@
|
||||
# -----------------------------------------------------------------------------
|
||||
import asyncio
|
||||
import time
|
||||
from typing import Optional
|
||||
|
||||
import click
|
||||
|
||||
@@ -41,7 +40,7 @@ class Loopback:
|
||||
self.transport = transport
|
||||
self.packet_size = packet_size
|
||||
self.packet_count = packet_count
|
||||
self.connection_handle: Optional[int] = None
|
||||
self.connection_handle: int | None = None
|
||||
self.connection_event = asyncio.Event()
|
||||
self.done = asyncio.Event()
|
||||
self.expected_cid = 0
|
||||
|
||||
+2
-3
@@ -16,7 +16,7 @@
|
||||
# Imports
|
||||
# -----------------------------------------------------------------------------
|
||||
import asyncio
|
||||
from typing import Callable, Iterable, Optional
|
||||
from collections.abc import Callable, Iterable
|
||||
|
||||
import click
|
||||
|
||||
@@ -174,7 +174,7 @@ async def show_vcs(vcs: VolumeControlServiceProxy) -> None:
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
async def show_device_info(peer, done: Optional[asyncio.Future]) -> None:
|
||||
async def show_device_info(peer, done: asyncio.Future | None) -> None:
|
||||
try:
|
||||
# Discover all services
|
||||
print(color('### Discovering Services and Characteristics', 'magenta'))
|
||||
@@ -215,7 +215,6 @@ async def show_device_info(peer, done: Optional[asyncio.Future]) -> None:
|
||||
# -----------------------------------------------------------------------------
|
||||
async def async_main(device_config, encrypt, transport, address_or_name):
|
||||
async with await open_transport(transport) as (hci_source, hci_sink):
|
||||
|
||||
# Create a device
|
||||
if device_config:
|
||||
device = Device.from_config_file_with_hci(
|
||||
|
||||
@@ -61,7 +61,6 @@ async def dump_gatt_db(peer, done):
|
||||
# -----------------------------------------------------------------------------
|
||||
async def async_main(device_config, encrypt, transport, address_or_name):
|
||||
async with await open_transport(transport) as (hci_source, hci_sink):
|
||||
|
||||
# Create a device
|
||||
if device_config:
|
||||
device = Device.from_config_file_with_hci(
|
||||
|
||||
@@ -268,7 +268,6 @@ class UiServer:
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
class Speaker:
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
device_config_path: str | None,
|
||||
|
||||
+7
-7
@@ -18,7 +18,6 @@
|
||||
import asyncio
|
||||
import logging
|
||||
import os
|
||||
import struct
|
||||
|
||||
import click
|
||||
from prompt_toolkit.shortcuts import PromptSession
|
||||
@@ -528,7 +527,9 @@ async def pair(
|
||||
if advertise_appearance:
|
||||
advertise_appearance = advertise_appearance.upper()
|
||||
try:
|
||||
advertise_appearance_int = int(advertise_appearance)
|
||||
appearance = data_types.Appearance.from_int(
|
||||
int(advertise_appearance)
|
||||
)
|
||||
except ValueError:
|
||||
category, subcategory = advertise_appearance.split('/')
|
||||
try:
|
||||
@@ -546,12 +547,11 @@ async def pair(
|
||||
except ValueError:
|
||||
print(color(f'Invalid subcategory {subcategory}', 'red'))
|
||||
return
|
||||
advertise_appearance_int = int(
|
||||
Appearance(category_enum, subcategory_enum)
|
||||
appearance = data_types.Appearance(
|
||||
category_enum, subcategory_enum
|
||||
)
|
||||
advertising_data_types.append(
|
||||
data_types.Appearance(category_enum, subcategory_enum)
|
||||
)
|
||||
|
||||
advertising_data_types.append(appearance)
|
||||
device.advertising_data = bytes(AdvertisingData(advertising_data_types))
|
||||
await device.start_advertising(
|
||||
auto_restart=True,
|
||||
|
||||
@@ -19,7 +19,7 @@ ROOTCANAL_PORT_CUTTLEFISH = 7300
|
||||
@click.option(
|
||||
'--transport',
|
||||
help='HCI transport',
|
||||
default=f'tcp-client:127.0.0.1:<rootcanal-port>',
|
||||
default='tcp-client:127.0.0.1:<rootcanal-port>',
|
||||
)
|
||||
@click.option(
|
||||
'--config',
|
||||
@@ -44,7 +44,7 @@ def retrieve_config(config: str) -> dict[str, Any]:
|
||||
if not config:
|
||||
return {}
|
||||
|
||||
with open(config, 'r') as f:
|
||||
with open(config) as f:
|
||||
return json.load(f)
|
||||
|
||||
|
||||
|
||||
+10
-12
@@ -19,7 +19,6 @@ from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
import logging
|
||||
from typing import Optional, Union
|
||||
|
||||
import click
|
||||
|
||||
@@ -47,14 +46,13 @@ from bumble.avdtp import (
|
||||
AVDTP_DELAY_REPORTING_SERVICE_CATEGORY,
|
||||
MediaCodecCapabilities,
|
||||
MediaPacketPump,
|
||||
find_avdtp_service_with_connection,
|
||||
)
|
||||
from bumble.avdtp import Protocol as AvdtpProtocol
|
||||
from bumble.avdtp import find_avdtp_service_with_connection
|
||||
from bumble.avrcp import Protocol as AvrcpProtocol
|
||||
from bumble.colors import color
|
||||
from bumble.core import AdvertisingData
|
||||
from bumble.core import AdvertisingData, DeviceClass, PhysicalTransport
|
||||
from bumble.core import ConnectionError as BumbleConnectionError
|
||||
from bumble.core import DeviceClass, PhysicalTransport
|
||||
from bumble.device import Connection, Device, DeviceConfiguration
|
||||
from bumble.hci import HCI_CONNECTION_ALREADY_EXISTS_ERROR, Address, HCI_Constant
|
||||
from bumble.pairing import PairingConfig
|
||||
@@ -191,7 +189,7 @@ class Player:
|
||||
def __init__(
|
||||
self,
|
||||
transport: str,
|
||||
device_config: Optional[str],
|
||||
device_config: str | None,
|
||||
authenticate: bool,
|
||||
encrypt: bool,
|
||||
) -> None:
|
||||
@@ -199,8 +197,8 @@ class Player:
|
||||
self.device_config = device_config
|
||||
self.authenticate = authenticate
|
||||
self.encrypt = encrypt
|
||||
self.avrcp_protocol: Optional[AvrcpProtocol] = None
|
||||
self.done: Optional[asyncio.Event]
|
||||
self.avrcp_protocol: AvrcpProtocol | None = None
|
||||
self.done: asyncio.Event | None
|
||||
|
||||
async def run(self, workload) -> None:
|
||||
self.done = asyncio.Event()
|
||||
@@ -315,7 +313,7 @@ class Player:
|
||||
codec_type: int,
|
||||
vendor_id: int,
|
||||
codec_id: int,
|
||||
packet_source: Union[SbcPacketSource, AacPacketSource, OpusPacketSource],
|
||||
packet_source: SbcPacketSource | AacPacketSource | OpusPacketSource,
|
||||
codec_capabilities: MediaCodecCapabilities,
|
||||
):
|
||||
# Discover all endpoints on the remote device
|
||||
@@ -381,11 +379,11 @@ class Player:
|
||||
print(f">>> {color(address.to_string(False), 'yellow')}:")
|
||||
print(f" Device Class (raw): {class_of_device:06X}")
|
||||
major_class_name = DeviceClass.major_device_class_name(major_device_class)
|
||||
print(" Device Major Class: " f"{major_class_name}")
|
||||
print(f" Device Major Class: {major_class_name}")
|
||||
minor_class_name = DeviceClass.minor_device_class_name(
|
||||
major_device_class, minor_device_class
|
||||
)
|
||||
print(" Device Minor Class: " f"{minor_class_name}")
|
||||
print(f" Device Minor Class: {minor_class_name}")
|
||||
print(
|
||||
" Device Services: "
|
||||
f"{', '.join(DeviceClass.service_class_labels(service_classes))}"
|
||||
@@ -420,7 +418,7 @@ class Player:
|
||||
async def play(
|
||||
self,
|
||||
device: Device,
|
||||
address: Optional[str],
|
||||
address: str | None,
|
||||
audio_format: str,
|
||||
audio_file: str,
|
||||
) -> None:
|
||||
@@ -449,7 +447,7 @@ class Player:
|
||||
return input_file.read(byte_count)
|
||||
|
||||
# Obtain the codec capabilities from the stream
|
||||
packet_source: Union[SbcPacketSource, AacPacketSource, OpusPacketSource]
|
||||
packet_source: SbcPacketSource | AacPacketSource | OpusPacketSource
|
||||
vendor_id = 0
|
||||
codec_id = 0
|
||||
if audio_format == "sbc":
|
||||
|
||||
+10
-11
@@ -17,7 +17,6 @@
|
||||
# -----------------------------------------------------------------------------
|
||||
import asyncio
|
||||
import time
|
||||
from typing import Optional
|
||||
|
||||
import click
|
||||
|
||||
@@ -82,14 +81,14 @@ class ServerBridge:
|
||||
def __init__(
|
||||
self, channel: int, uuid: str, trace: bool, tcp_host: str, tcp_port: int
|
||||
) -> None:
|
||||
self.device: Optional[Device] = None
|
||||
self.device: Device | None = None
|
||||
self.channel = channel
|
||||
self.uuid = uuid
|
||||
self.tcp_host = tcp_host
|
||||
self.tcp_port = tcp_port
|
||||
self.rfcomm_channel: Optional[rfcomm.DLC] = None
|
||||
self.tcp_tracer: Optional[Tracer]
|
||||
self.rfcomm_tracer: Optional[Tracer]
|
||||
self.rfcomm_channel: rfcomm.DLC | None = None
|
||||
self.tcp_tracer: Tracer | None
|
||||
self.rfcomm_tracer: Tracer | None
|
||||
|
||||
if trace:
|
||||
self.tcp_tracer = Tracer(color("RFCOMM->TCP", "cyan"))
|
||||
@@ -242,14 +241,14 @@ class ClientBridge:
|
||||
self.tcp_port = tcp_port
|
||||
self.authenticate = authenticate
|
||||
self.encrypt = encrypt
|
||||
self.device: Optional[Device] = None
|
||||
self.connection: Optional[Connection] = None
|
||||
self.rfcomm_client: Optional[rfcomm.Client]
|
||||
self.rfcomm_mux: Optional[rfcomm.Multiplexer]
|
||||
self.device: Device | None = None
|
||||
self.connection: Connection | None = None
|
||||
self.rfcomm_client: rfcomm.Client | None
|
||||
self.rfcomm_mux: rfcomm.Multiplexer | None
|
||||
self.tcp_connected: bool = False
|
||||
|
||||
self.tcp_tracer: Optional[Tracer]
|
||||
self.rfcomm_tracer: Optional[Tracer]
|
||||
self.tcp_tracer: Tracer | None
|
||||
self.rfcomm_tracer: Tracer | None
|
||||
|
||||
if trace:
|
||||
self.tcp_tracer = Tracer(color("RFCOMM->TCP", "cyan"))
|
||||
|
||||
+1
-3
@@ -217,9 +217,7 @@ async def scan(
|
||||
@click.option(
|
||||
'--irk',
|
||||
metavar='<IRK_HEX>:<ADDRESS>',
|
||||
help=(
|
||||
'Use this IRK for resolving private addresses ' '(may be used more than once)'
|
||||
),
|
||||
help=('Use this IRK for resolving private addresses (may be used more than once)'),
|
||||
multiple=True,
|
||||
)
|
||||
@click.option(
|
||||
|
||||
@@ -26,7 +26,6 @@ import pathlib
|
||||
import subprocess
|
||||
import weakref
|
||||
from importlib import resources
|
||||
from typing import Optional
|
||||
|
||||
import aiohttp
|
||||
import click
|
||||
@@ -156,7 +155,7 @@ class QueuedOutput(Output):
|
||||
|
||||
packets: asyncio.Queue
|
||||
extractor: AudioExtractor
|
||||
packet_pump_task: Optional[asyncio.Task]
|
||||
packet_pump_task: asyncio.Task | None
|
||||
started: bool
|
||||
|
||||
def __init__(self, extractor):
|
||||
@@ -230,8 +229,8 @@ class WebSocketOutput(QueuedOutput):
|
||||
class FfplayOutput(QueuedOutput):
|
||||
MAX_QUEUE_SIZE = 32768
|
||||
|
||||
subprocess: Optional[asyncio.subprocess.Process]
|
||||
ffplay_task: Optional[asyncio.Task]
|
||||
subprocess: asyncio.subprocess.Process | None
|
||||
ffplay_task: asyncio.Task | None
|
||||
|
||||
def __init__(self, codec: str) -> None:
|
||||
super().__init__(AudioExtractor.create(codec))
|
||||
|
||||
+5
-5
@@ -22,9 +22,9 @@ import enum
|
||||
import logging
|
||||
import struct
|
||||
from collections.abc import AsyncGenerator, Awaitable, Callable
|
||||
from typing import Union
|
||||
from typing import ClassVar
|
||||
|
||||
from typing_extensions import ClassVar, Self
|
||||
from typing_extensions import Self
|
||||
|
||||
from bumble import utils
|
||||
from bumble.codecs import AacAudioRtpPacket
|
||||
@@ -266,7 +266,7 @@ class MediaCodecInformation:
|
||||
@classmethod
|
||||
def create(
|
||||
cls, media_codec_type: int, data: bytes
|
||||
) -> Union[MediaCodecInformation, bytes]:
|
||||
) -> MediaCodecInformation | bytes:
|
||||
if media_codec_type == CodecType.SBC:
|
||||
return SbcMediaCodecInformation.from_bytes(data)
|
||||
elif media_codec_type == CodecType.MPEG_2_4_AAC:
|
||||
@@ -490,7 +490,7 @@ class VendorSpecificMediaCodecInformation(MediaCodecInformation):
|
||||
'VendorSpecificMediaCodecInformation(',
|
||||
f' vendor_id: {self.vendor_id:08X} ({name_or_number(COMPANY_IDENTIFIERS, self.vendor_id & 0xFFFF)})',
|
||||
f' codec_id: {self.codec_id:04X}',
|
||||
f' value: {self.value.hex()}' ')',
|
||||
f' value: {self.value.hex()})',
|
||||
]
|
||||
)
|
||||
|
||||
@@ -686,7 +686,7 @@ class SbcPacketSource:
|
||||
# Prepare for next packets
|
||||
sequence_number += 1
|
||||
sequence_number &= 0xFFFF
|
||||
sample_count += sum((frame.sample_count for frame in frames))
|
||||
sample_count += sum(frame.sample_count for frame in frames)
|
||||
frames = [frame]
|
||||
frames_size = len(frame.payload)
|
||||
else:
|
||||
|
||||
+4
-5
@@ -12,7 +12,6 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from typing import Union
|
||||
|
||||
from bumble import core
|
||||
|
||||
@@ -36,7 +35,7 @@ def tokenize_parameters(buffer: bytes) -> list[bytes]:
|
||||
|
||||
if in_quotes:
|
||||
token.extend(char)
|
||||
if char == b'\"':
|
||||
if char == b'"':
|
||||
in_quotes = False
|
||||
tokens.append(token[1:-1])
|
||||
token = bytearray()
|
||||
@@ -63,18 +62,18 @@ def tokenize_parameters(buffer: bytes) -> list[bytes]:
|
||||
return [bytes(token) for token in tokens if len(token) > 0]
|
||||
|
||||
|
||||
def parse_parameters(buffer: bytes) -> list[Union[bytes, list]]:
|
||||
def parse_parameters(buffer: bytes) -> list[bytes | list]:
|
||||
"""Parse the parameters using the comma and parenthesis separators.
|
||||
Raises AtParsingError in case of invalid input string."""
|
||||
|
||||
tokens = tokenize_parameters(buffer)
|
||||
accumulator: list[list] = [[]]
|
||||
current: Union[bytes, list] = bytes()
|
||||
current: bytes | list = b''
|
||||
|
||||
for token in tokens:
|
||||
if token == b',':
|
||||
accumulator[-1].append(current)
|
||||
current = bytes()
|
||||
current = b''
|
||||
elif token == b'(':
|
||||
accumulator.append([])
|
||||
elif token == b')':
|
||||
|
||||
+17
-22
@@ -29,15 +29,12 @@ import enum
|
||||
import functools
|
||||
import inspect
|
||||
import struct
|
||||
from collections.abc import Awaitable, Callable
|
||||
from typing import (
|
||||
TYPE_CHECKING,
|
||||
Awaitable,
|
||||
Callable,
|
||||
ClassVar,
|
||||
Generic,
|
||||
Optional,
|
||||
TypeVar,
|
||||
Union,
|
||||
)
|
||||
|
||||
from bumble import hci, utils
|
||||
@@ -220,7 +217,7 @@ class ATT_PDU:
|
||||
fields: ClassVar[hci.Fields] = ()
|
||||
op_code: int = dataclasses.field(init=False)
|
||||
name: str = dataclasses.field(init=False)
|
||||
_payload: Optional[bytes] = dataclasses.field(default=None, init=False)
|
||||
_payload: bytes | None = dataclasses.field(default=None, init=False)
|
||||
|
||||
@classmethod
|
||||
def from_bytes(cls, pdu: bytes) -> ATT_PDU:
|
||||
@@ -760,26 +757,24 @@ class AttributeValue(Generic[_T]):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
read: Union[
|
||||
Callable[[Connection], _T],
|
||||
Callable[[Connection], Awaitable[_T]],
|
||||
None,
|
||||
] = None,
|
||||
write: Union[
|
||||
Callable[[Connection, _T], None],
|
||||
Callable[[Connection, _T], Awaitable[None]],
|
||||
None,
|
||||
] = None,
|
||||
read: (
|
||||
Callable[[Connection], _T] | Callable[[Connection], Awaitable[_T]] | None
|
||||
) = None,
|
||||
write: (
|
||||
Callable[[Connection, _T], None]
|
||||
| Callable[[Connection, _T], Awaitable[None]]
|
||||
| None
|
||||
) = None,
|
||||
):
|
||||
self._read = read
|
||||
self._write = write
|
||||
|
||||
def read(self, connection: Connection) -> Union[_T, Awaitable[_T]]:
|
||||
def read(self, connection: Connection) -> _T | Awaitable[_T]:
|
||||
if self._read is None:
|
||||
raise InvalidOperationError('AttributeValue has no read function')
|
||||
return self._read(connection)
|
||||
|
||||
def write(self, connection: Connection, value: _T) -> Union[Awaitable[None], None]:
|
||||
def write(self, connection: Connection, value: _T) -> Awaitable[None] | None:
|
||||
if self._write is None:
|
||||
raise InvalidOperationError('AttributeValue has no write function')
|
||||
return self._write(connection, value)
|
||||
@@ -828,13 +823,13 @@ class Attribute(utils.EventEmitter, Generic[_T]):
|
||||
EVENT_READ = "read"
|
||||
EVENT_WRITE = "write"
|
||||
|
||||
value: Union[AttributeValue[_T], _T, None]
|
||||
value: AttributeValue[_T] | _T | None
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
attribute_type: Union[str, bytes, UUID],
|
||||
permissions: Union[str, Attribute.Permissions],
|
||||
value: Union[AttributeValue[_T], _T, None] = None,
|
||||
attribute_type: str | bytes | UUID,
|
||||
permissions: str | Attribute.Permissions,
|
||||
value: AttributeValue[_T] | _T | None = None,
|
||||
) -> None:
|
||||
utils.EventEmitter.__init__(self)
|
||||
self.handle = 0
|
||||
@@ -883,7 +878,7 @@ class Attribute(utils.EventEmitter, Generic[_T]):
|
||||
error_code=ATT_INSUFFICIENT_AUTHORIZATION_ERROR, att_handle=self.handle
|
||||
)
|
||||
|
||||
value: Union[_T, None]
|
||||
value: _T | None
|
||||
if isinstance(self.value, AttributeValue):
|
||||
try:
|
||||
read_value = self.value.read(connection)
|
||||
|
||||
+5
-3
@@ -26,7 +26,8 @@ import logging
|
||||
import pathlib
|
||||
import sys
|
||||
import wave
|
||||
from typing import TYPE_CHECKING, AsyncGenerator, BinaryIO
|
||||
from collections.abc import AsyncGenerator
|
||||
from typing import TYPE_CHECKING, BinaryIO
|
||||
|
||||
from bumble.colors import color
|
||||
|
||||
@@ -545,5 +546,6 @@ class SoundDeviceAudioInput(ThreadedAudioInput):
|
||||
return bytes(pcm_buffer)
|
||||
|
||||
def _close(self):
|
||||
self._stream.stop()
|
||||
self._stream = None
|
||||
if self._stream:
|
||||
self._stream.stop()
|
||||
self._stream = None
|
||||
|
||||
+1
-2
@@ -19,7 +19,6 @@ from __future__ import annotations
|
||||
|
||||
import enum
|
||||
import struct
|
||||
from typing import Union
|
||||
|
||||
from bumble import core, utils
|
||||
|
||||
@@ -166,7 +165,7 @@ class Frame:
|
||||
|
||||
def to_bytes(
|
||||
self,
|
||||
ctype_or_response: Union[CommandFrame.CommandType, ResponseFrame.ResponseCode],
|
||||
ctype_or_response: CommandFrame.CommandType | ResponseFrame.ResponseCode,
|
||||
) -> bytes:
|
||||
# TODO: support extended subunit types and ids.
|
||||
return (
|
||||
|
||||
+2
-3
@@ -21,7 +21,6 @@ import logging
|
||||
import struct
|
||||
from collections.abc import Callable
|
||||
from enum import IntEnum
|
||||
from typing import Optional
|
||||
|
||||
from bumble import core, l2cap
|
||||
from bumble.colors import color
|
||||
@@ -147,7 +146,7 @@ class MessageAssembler:
|
||||
class Protocol:
|
||||
CommandHandler = Callable[[int, bytes], None]
|
||||
command_handlers: dict[int, CommandHandler] # Command handlers, by PID
|
||||
ResponseHandler = Callable[[int, Optional[bytes]], None]
|
||||
ResponseHandler = Callable[[int, bytes | None], None]
|
||||
response_handlers: dict[int, ResponseHandler] # Response handlers, by PID
|
||||
next_transaction_label: int
|
||||
message_assembler: MessageAssembler
|
||||
@@ -258,7 +257,7 @@ class Protocol:
|
||||
|
||||
def send_ipid(self, transaction_label: int, pid: int) -> None:
|
||||
logger.debug(
|
||||
">>> AVCTP ipid: " f"transaction_label={transaction_label}, " f"pid={pid}"
|
||||
f">>> AVCTP ipid: transaction_label={transaction_label}, pid={pid}"
|
||||
)
|
||||
self.send_message(transaction_label, False, True, pid, b'')
|
||||
|
||||
|
||||
+53
-57
@@ -22,16 +22,13 @@ import enum
|
||||
import logging
|
||||
import time
|
||||
import warnings
|
||||
from collections.abc import AsyncGenerator, Awaitable, Iterable
|
||||
from collections.abc import AsyncGenerator, Awaitable, Callable, Iterable
|
||||
from dataclasses import dataclass, field
|
||||
from typing import (
|
||||
Any,
|
||||
Callable,
|
||||
ClassVar,
|
||||
Optional,
|
||||
SupportsBytes,
|
||||
TypeVar,
|
||||
Union,
|
||||
cast,
|
||||
)
|
||||
|
||||
@@ -184,7 +181,7 @@ class State(utils.OpenIntEnum):
|
||||
# -----------------------------------------------------------------------------
|
||||
async def find_avdtp_service_with_sdp_client(
|
||||
sdp_client: sdp.Client,
|
||||
) -> Optional[tuple[int, int]]:
|
||||
) -> tuple[int, int] | None:
|
||||
'''
|
||||
Find an AVDTP service, using a connected SDP client, and return its version,
|
||||
or None if none is found
|
||||
@@ -214,7 +211,7 @@ async def find_avdtp_service_with_sdp_client(
|
||||
# -----------------------------------------------------------------------------
|
||||
async def find_avdtp_service_with_connection(
|
||||
connection: device.Connection,
|
||||
) -> Optional[tuple[int, int]]:
|
||||
) -> tuple[int, int] | None:
|
||||
'''
|
||||
Find an AVDTP service, for a connection, and return its version,
|
||||
or None if none is found
|
||||
@@ -239,7 +236,7 @@ class RealtimeClock:
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
class MediaPacketPump:
|
||||
pump_task: Optional[asyncio.Task]
|
||||
pump_task: asyncio.Task | None
|
||||
|
||||
def __init__(
|
||||
self, packets: AsyncGenerator, clock: RealtimeClock = RealtimeClock()
|
||||
@@ -296,7 +293,7 @@ class MediaPacketPump:
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
class MessageAssembler:
|
||||
message: Optional[bytes]
|
||||
message: bytes | None
|
||||
signal_identifier: SignalIdentifier
|
||||
|
||||
def __init__(self, callback: Callable[[int, Message], Any]) -> None:
|
||||
@@ -470,15 +467,15 @@ class MediaCodecCapabilities(ServiceCapabilities):
|
||||
|
||||
media_type: MediaType
|
||||
media_codec_type: a2dp.CodecType
|
||||
media_codec_information: Union[bytes, SupportsBytes]
|
||||
media_codec_information: bytes | SupportsBytes
|
||||
|
||||
# Override init to allow passing service_capabilities_bytes.
|
||||
def __init__(
|
||||
self,
|
||||
media_type: MediaType,
|
||||
media_codec_type: a2dp.CodecType,
|
||||
media_codec_information: Union[bytes, SupportsBytes],
|
||||
service_capabilities_bytes: Optional[bytes] = None,
|
||||
media_codec_information: bytes | SupportsBytes,
|
||||
service_capabilities_bytes: bytes | None = None,
|
||||
) -> None:
|
||||
self.media_type = media_type
|
||||
self.media_codec_type = media_codec_type
|
||||
@@ -553,7 +550,7 @@ class Message:
|
||||
|
||||
message_type: MessageType
|
||||
signal_identifier: SignalIdentifier
|
||||
_payload: Optional[bytes] = None
|
||||
_payload: bytes | None = None
|
||||
fields: ClassVar[hci.Fields] = ()
|
||||
|
||||
@property
|
||||
@@ -607,7 +604,7 @@ class Message:
|
||||
instance.signal_identifier = signal_identifier
|
||||
return instance
|
||||
|
||||
def to_string(self, details: Union[str, Iterable[str]]) -> str:
|
||||
def to_string(self, details: str | Iterable[str]) -> str:
|
||||
base = color(
|
||||
f'{self.signal_identifier.name}_{self.message_type.name}',
|
||||
'yellow',
|
||||
@@ -1266,9 +1263,9 @@ class Protocol(utils.EventEmitter):
|
||||
local_endpoints: list[LocalStreamEndPoint]
|
||||
remote_endpoints: dict[int, DiscoveredStreamEndPoint]
|
||||
streams: dict[int, Stream]
|
||||
transaction_results: list[Optional[asyncio.Future[Message]]]
|
||||
transaction_results: list[asyncio.Future[Message] | None]
|
||||
channel_connector: Callable[[], Awaitable[l2cap.ClassicChannel]]
|
||||
channel_acceptor: Optional[Stream]
|
||||
channel_acceptor: Stream | None
|
||||
|
||||
EVENT_OPEN = "open"
|
||||
EVENT_CLOSE = "close"
|
||||
@@ -1311,7 +1308,7 @@ class Protocol(utils.EventEmitter):
|
||||
l2cap_channel.on(l2cap_channel.EVENT_OPEN, self.on_l2cap_channel_open)
|
||||
l2cap_channel.on(l2cap_channel.EVENT_CLOSE, self.on_l2cap_channel_close)
|
||||
|
||||
def get_local_endpoint_by_seid(self, seid: int) -> Optional[LocalStreamEndPoint]:
|
||||
def get_local_endpoint_by_seid(self, seid: int) -> LocalStreamEndPoint | None:
|
||||
if 0 < seid <= len(self.local_endpoints):
|
||||
return self.local_endpoints[seid - 1]
|
||||
|
||||
@@ -1385,7 +1382,7 @@ class Protocol(utils.EventEmitter):
|
||||
|
||||
def find_remote_sink_by_codec(
|
||||
self, media_type: int, codec_type: int, vendor_id: int = 0, codec_id: int = 0
|
||||
) -> Optional[DiscoveredStreamEndPoint]:
|
||||
) -> DiscoveredStreamEndPoint | None:
|
||||
for endpoint in self.remote_endpoints.values():
|
||||
if (
|
||||
not endpoint.in_use
|
||||
@@ -1569,10 +1566,9 @@ class Protocol(utils.EventEmitter):
|
||||
|
||||
assert False # Should never reach this
|
||||
|
||||
async def get_capabilities(self, seid: int) -> Union[
|
||||
Get_Capabilities_Response,
|
||||
Get_All_Capabilities_Response,
|
||||
]:
|
||||
async def get_capabilities(
|
||||
self, seid: int
|
||||
) -> Get_Capabilities_Response | Get_All_Capabilities_Response:
|
||||
if self.version > (1, 2):
|
||||
return await self.send_command(Get_All_Capabilities_Command(seid))
|
||||
|
||||
@@ -1604,7 +1600,7 @@ class Protocol(utils.EventEmitter):
|
||||
async def abort(self, seid: int) -> Abort_Response:
|
||||
return await self.send_command(Abort_Command(seid))
|
||||
|
||||
def on_discover_command(self, command: Discover_Command) -> Optional[Message]:
|
||||
def on_discover_command(self, command: Discover_Command) -> Message | None:
|
||||
endpoint_infos = [
|
||||
EndPointInfo(endpoint.seid, 0, endpoint.media_type, endpoint.tsep)
|
||||
for endpoint in self.local_endpoints
|
||||
@@ -1613,7 +1609,7 @@ class Protocol(utils.EventEmitter):
|
||||
|
||||
def on_get_capabilities_command(
|
||||
self, command: Get_Capabilities_Command
|
||||
) -> Optional[Message]:
|
||||
) -> Message | None:
|
||||
endpoint = self.get_local_endpoint_by_seid(command.acp_seid)
|
||||
if endpoint is None:
|
||||
return Get_Capabilities_Reject(AVDTP_BAD_ACP_SEID_ERROR)
|
||||
@@ -1622,7 +1618,7 @@ class Protocol(utils.EventEmitter):
|
||||
|
||||
def on_get_all_capabilities_command(
|
||||
self, command: Get_All_Capabilities_Command
|
||||
) -> Optional[Message]:
|
||||
) -> Message | None:
|
||||
endpoint = self.get_local_endpoint_by_seid(command.acp_seid)
|
||||
if endpoint is None:
|
||||
return Get_All_Capabilities_Reject(AVDTP_BAD_ACP_SEID_ERROR)
|
||||
@@ -1631,7 +1627,7 @@ class Protocol(utils.EventEmitter):
|
||||
|
||||
def on_set_configuration_command(
|
||||
self, command: Set_Configuration_Command
|
||||
) -> Optional[Message]:
|
||||
) -> Message | None:
|
||||
endpoint = self.get_local_endpoint_by_seid(command.acp_seid)
|
||||
if endpoint is None:
|
||||
return Set_Configuration_Reject(error_code=AVDTP_BAD_ACP_SEID_ERROR)
|
||||
@@ -1649,7 +1645,7 @@ class Protocol(utils.EventEmitter):
|
||||
|
||||
def on_get_configuration_command(
|
||||
self, command: Get_Configuration_Command
|
||||
) -> Optional[Message]:
|
||||
) -> Message | None:
|
||||
endpoint = self.get_local_endpoint_by_seid(command.acp_seid)
|
||||
if endpoint is None:
|
||||
return Get_Configuration_Reject(AVDTP_BAD_ACP_SEID_ERROR)
|
||||
@@ -1658,7 +1654,7 @@ class Protocol(utils.EventEmitter):
|
||||
|
||||
return endpoint.stream.on_get_configuration_command()
|
||||
|
||||
def on_reconfigure_command(self, command: Reconfigure_Command) -> Optional[Message]:
|
||||
def on_reconfigure_command(self, command: Reconfigure_Command) -> Message | None:
|
||||
endpoint = self.get_local_endpoint_by_seid(command.acp_seid)
|
||||
if endpoint is None:
|
||||
return Reconfigure_Reject(error_code=AVDTP_BAD_ACP_SEID_ERROR)
|
||||
@@ -1668,7 +1664,7 @@ class Protocol(utils.EventEmitter):
|
||||
result = endpoint.stream.on_reconfigure_command(command.capabilities)
|
||||
return result or Reconfigure_Response()
|
||||
|
||||
def on_open_command(self, command: Open_Command) -> Optional[Message]:
|
||||
def on_open_command(self, command: Open_Command) -> Message | None:
|
||||
endpoint = self.get_local_endpoint_by_seid(command.acp_seid)
|
||||
if endpoint is None:
|
||||
return Open_Reject(AVDTP_BAD_ACP_SEID_ERROR)
|
||||
@@ -1678,7 +1674,7 @@ class Protocol(utils.EventEmitter):
|
||||
result = endpoint.stream.on_open_command()
|
||||
return result or Open_Response()
|
||||
|
||||
def on_start_command(self, command: Start_Command) -> Optional[Message]:
|
||||
def on_start_command(self, command: Start_Command) -> Message | None:
|
||||
for seid in command.acp_seids:
|
||||
endpoint = self.get_local_endpoint_by_seid(seid)
|
||||
if endpoint is None:
|
||||
@@ -1697,7 +1693,7 @@ class Protocol(utils.EventEmitter):
|
||||
|
||||
return Start_Response()
|
||||
|
||||
def on_suspend_command(self, command: Suspend_Command) -> Optional[Message]:
|
||||
def on_suspend_command(self, command: Suspend_Command) -> Message | None:
|
||||
for seid in command.acp_seids:
|
||||
endpoint = self.get_local_endpoint_by_seid(seid)
|
||||
if endpoint is None:
|
||||
@@ -1716,7 +1712,7 @@ class Protocol(utils.EventEmitter):
|
||||
|
||||
return Suspend_Response()
|
||||
|
||||
def on_close_command(self, command: Close_Command) -> Optional[Message]:
|
||||
def on_close_command(self, command: Close_Command) -> Message | None:
|
||||
endpoint = self.get_local_endpoint_by_seid(command.acp_seid)
|
||||
if endpoint is None:
|
||||
return Close_Reject(AVDTP_BAD_ACP_SEID_ERROR)
|
||||
@@ -1726,7 +1722,7 @@ class Protocol(utils.EventEmitter):
|
||||
result = endpoint.stream.on_close_command()
|
||||
return result or Close_Response()
|
||||
|
||||
def on_abort_command(self, command: Abort_Command) -> Optional[Message]:
|
||||
def on_abort_command(self, command: Abort_Command) -> Message | None:
|
||||
endpoint = self.get_local_endpoint_by_seid(command.acp_seid)
|
||||
if endpoint is None or endpoint.stream is None:
|
||||
return Abort_Response()
|
||||
@@ -1736,7 +1732,7 @@ class Protocol(utils.EventEmitter):
|
||||
|
||||
def on_security_control_command(
|
||||
self, command: Security_Control_Command
|
||||
) -> Optional[Message]:
|
||||
) -> Message | None:
|
||||
endpoint = self.get_local_endpoint_by_seid(command.acp_seid)
|
||||
if endpoint is None:
|
||||
return Security_Control_Reject(AVDTP_BAD_ACP_SEID_ERROR)
|
||||
@@ -1744,7 +1740,7 @@ class Protocol(utils.EventEmitter):
|
||||
result = endpoint.on_security_control_command(command.data)
|
||||
return result or Security_Control_Response()
|
||||
|
||||
def on_delayreport_command(self, command: DelayReport_Command) -> Optional[Message]:
|
||||
def on_delayreport_command(self, command: DelayReport_Command) -> Message | None:
|
||||
endpoint = self.get_local_endpoint_by_seid(command.acp_seid)
|
||||
if endpoint is None:
|
||||
return DelayReport_Reject(AVDTP_BAD_ACP_SEID_ERROR)
|
||||
@@ -1825,7 +1821,7 @@ class Stream:
|
||||
Pair of a local and a remote stream endpoint that can stream from one to the other
|
||||
'''
|
||||
|
||||
rtp_channel: Optional[l2cap.ClassicChannel]
|
||||
rtp_channel: l2cap.ClassicChannel | None
|
||||
|
||||
def change_state(self, state: State) -> None:
|
||||
logger.debug(f'{self} state change -> {color(state.name, "cyan")}')
|
||||
@@ -1914,7 +1910,7 @@ class Stream:
|
||||
|
||||
def on_set_configuration_command(
|
||||
self, configuration: Iterable[ServiceCapabilities]
|
||||
) -> Optional[Message]:
|
||||
) -> Message | None:
|
||||
if self.state != State.IDLE:
|
||||
return Set_Configuration_Reject(error_code=AVDTP_BAD_STATE_ERROR)
|
||||
|
||||
@@ -1925,7 +1921,7 @@ class Stream:
|
||||
self.change_state(State.CONFIGURED)
|
||||
return None
|
||||
|
||||
def on_get_configuration_command(self) -> Optional[Message]:
|
||||
def on_get_configuration_command(self) -> Message | None:
|
||||
if self.state not in (
|
||||
State.CONFIGURED,
|
||||
State.OPEN,
|
||||
@@ -1937,7 +1933,7 @@ class Stream:
|
||||
|
||||
def on_reconfigure_command(
|
||||
self, configuration: Iterable[ServiceCapabilities]
|
||||
) -> Optional[Message]:
|
||||
) -> Message | None:
|
||||
if self.state != State.OPEN:
|
||||
return Reconfigure_Reject(error_code=AVDTP_BAD_STATE_ERROR)
|
||||
|
||||
@@ -1947,7 +1943,7 @@ class Stream:
|
||||
|
||||
return None
|
||||
|
||||
def on_open_command(self) -> Optional[Message]:
|
||||
def on_open_command(self) -> Message | None:
|
||||
if self.state != State.CONFIGURED:
|
||||
return Open_Reject(AVDTP_BAD_STATE_ERROR)
|
||||
|
||||
@@ -1961,7 +1957,7 @@ class Stream:
|
||||
self.change_state(State.OPEN)
|
||||
return None
|
||||
|
||||
def on_start_command(self) -> Optional[Message]:
|
||||
def on_start_command(self) -> Message | None:
|
||||
if self.state != State.OPEN:
|
||||
return Open_Reject(AVDTP_BAD_STATE_ERROR)
|
||||
|
||||
@@ -1977,7 +1973,7 @@ class Stream:
|
||||
self.change_state(State.STREAMING)
|
||||
return None
|
||||
|
||||
def on_suspend_command(self) -> Optional[Message]:
|
||||
def on_suspend_command(self) -> Message | None:
|
||||
if self.state != State.STREAMING:
|
||||
return Open_Reject(AVDTP_BAD_STATE_ERROR)
|
||||
|
||||
@@ -1988,7 +1984,7 @@ class Stream:
|
||||
self.change_state(State.OPEN)
|
||||
return None
|
||||
|
||||
def on_close_command(self) -> Optional[Message]:
|
||||
def on_close_command(self) -> Message | None:
|
||||
if self.state not in (State.OPEN, State.STREAMING):
|
||||
return Open_Reject(AVDTP_BAD_STATE_ERROR)
|
||||
|
||||
@@ -2007,7 +2003,7 @@ class Stream:
|
||||
|
||||
return None
|
||||
|
||||
def on_abort_command(self) -> Optional[Message]:
|
||||
def on_abort_command(self) -> Message | None:
|
||||
if self.rtp_channel is None:
|
||||
# No need to wait
|
||||
self.change_state(State.IDLE)
|
||||
@@ -2120,7 +2116,7 @@ class DiscoveredStreamEndPoint(StreamEndPoint, StreamEndPointProxy):
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
class LocalStreamEndPoint(StreamEndPoint, utils.EventEmitter):
|
||||
stream: Optional[Stream]
|
||||
stream: Stream | None
|
||||
|
||||
EVENT_CONFIGURATION = "configuration"
|
||||
EVENT_OPEN = "open"
|
||||
@@ -2142,7 +2138,7 @@ class LocalStreamEndPoint(StreamEndPoint, utils.EventEmitter):
|
||||
media_type: MediaType,
|
||||
tsep: StreamEndPointType,
|
||||
capabilities: Iterable[ServiceCapabilities],
|
||||
configuration: Optional[Iterable[ServiceCapabilities]] = None,
|
||||
configuration: Iterable[ServiceCapabilities] | None = None,
|
||||
):
|
||||
StreamEndPoint.__init__(self, seid, media_type, tsep, 0, capabilities)
|
||||
utils.EventEmitter.__init__(self)
|
||||
@@ -2161,13 +2157,13 @@ class LocalStreamEndPoint(StreamEndPoint, utils.EventEmitter):
|
||||
|
||||
def on_reconfigure_command(
|
||||
self, command: Iterable[ServiceCapabilities]
|
||||
) -> Optional[Message]:
|
||||
) -> Message | None:
|
||||
del command # unused.
|
||||
return None
|
||||
|
||||
def on_set_configuration_command(
|
||||
self, configuration: Iterable[ServiceCapabilities]
|
||||
) -> Optional[Message]:
|
||||
) -> Message | None:
|
||||
logger.debug(
|
||||
'<<< received configuration: '
|
||||
f'{",".join([str(capability) for capability in configuration])}'
|
||||
@@ -2176,34 +2172,34 @@ class LocalStreamEndPoint(StreamEndPoint, utils.EventEmitter):
|
||||
self.emit(self.EVENT_CONFIGURATION)
|
||||
return None
|
||||
|
||||
def on_get_configuration_command(self) -> Optional[Message]:
|
||||
def on_get_configuration_command(self) -> Message | None:
|
||||
return Get_Configuration_Response(self.configuration)
|
||||
|
||||
def on_open_command(self) -> Optional[Message]:
|
||||
def on_open_command(self) -> Message | None:
|
||||
self.emit(self.EVENT_OPEN)
|
||||
return None
|
||||
|
||||
def on_start_command(self) -> Optional[Message]:
|
||||
def on_start_command(self) -> Message | None:
|
||||
self.emit(self.EVENT_START)
|
||||
return None
|
||||
|
||||
def on_suspend_command(self) -> Optional[Message]:
|
||||
def on_suspend_command(self) -> Message | None:
|
||||
self.emit(self.EVENT_SUSPEND)
|
||||
return None
|
||||
|
||||
def on_close_command(self) -> Optional[Message]:
|
||||
def on_close_command(self) -> Message | None:
|
||||
self.emit(self.EVENT_CLOSE)
|
||||
return None
|
||||
|
||||
def on_abort_command(self) -> Optional[Message]:
|
||||
def on_abort_command(self) -> Message | None:
|
||||
self.emit(self.EVENT_ABORT)
|
||||
return None
|
||||
|
||||
def on_delayreport_command(self, delay: int) -> Optional[Message]:
|
||||
def on_delayreport_command(self, delay: int) -> Message | None:
|
||||
self.emit(self.EVENT_DELAY_REPORT, delay)
|
||||
return None
|
||||
|
||||
def on_security_control_command(self, data: bytes) -> Optional[Message]:
|
||||
def on_security_control_command(self, data: bytes) -> Message | None:
|
||||
self.emit(self.EVENT_SECURITY_CONTROL, data)
|
||||
return None
|
||||
|
||||
@@ -2255,12 +2251,12 @@ class LocalSource(LocalStreamEndPoint):
|
||||
self.emit(self.EVENT_STOP)
|
||||
|
||||
@override
|
||||
def on_start_command(self) -> Optional[Message]:
|
||||
def on_start_command(self) -> Message | None:
|
||||
asyncio.create_task(self.start())
|
||||
return None
|
||||
|
||||
@override
|
||||
def on_suspend_command(self) -> Optional[Message]:
|
||||
def on_suspend_command(self) -> Message | None:
|
||||
asyncio.create_task(self.stop())
|
||||
return None
|
||||
|
||||
|
||||
+25
-27
@@ -24,7 +24,7 @@ import logging
|
||||
import struct
|
||||
from collections.abc import AsyncIterator, Awaitable, Callable, Iterable, Sequence
|
||||
from dataclasses import dataclass, field
|
||||
from typing import ClassVar, Optional, SupportsBytes, TypeVar, Union
|
||||
from typing import ClassVar, SupportsBytes, TypeVar
|
||||
|
||||
from bumble import avc, avctp, core, hci, l2cap, utils
|
||||
from bumble.colors import color
|
||||
@@ -196,7 +196,7 @@ def make_controller_service_sdp_records(
|
||||
service_record_handle: int,
|
||||
avctp_version: tuple[int, int] = (1, 4),
|
||||
avrcp_version: tuple[int, int] = (1, 6),
|
||||
supported_features: Union[int, ControllerFeatures] = 1,
|
||||
supported_features: int | ControllerFeatures = 1,
|
||||
) -> list[ServiceAttribute]:
|
||||
avctp_version_int = avctp_version[0] << 8 | avctp_version[1]
|
||||
avrcp_version_int = avrcp_version[0] << 8 | avrcp_version[1]
|
||||
@@ -288,7 +288,7 @@ def make_target_service_sdp_records(
|
||||
service_record_handle: int,
|
||||
avctp_version: tuple[int, int] = (1, 4),
|
||||
avrcp_version: tuple[int, int] = (1, 6),
|
||||
supported_features: Union[int, TargetFeatures] = 0x23,
|
||||
supported_features: int | TargetFeatures = 0x23,
|
||||
) -> list[ServiceAttribute]:
|
||||
# TODO: support a way to compute the supported features from a feature list
|
||||
avctp_version_int = avctp_version[0] << 8 | avctp_version[1]
|
||||
@@ -478,7 +478,7 @@ class BrowseableItem:
|
||||
MEDIA_ELEMENT = 0x03
|
||||
|
||||
item_type: ClassVar[Type]
|
||||
_payload: Optional[bytes] = None
|
||||
_payload: bytes | None = None
|
||||
|
||||
subclasses: ClassVar[dict[Type, type[BrowseableItem]]] = {}
|
||||
fields: ClassVar[hci.Fields] = ()
|
||||
@@ -672,7 +672,7 @@ class PduAssembler:
|
||||
6.3.1 AVRCP specific AV//C commands
|
||||
"""
|
||||
|
||||
pdu_id: Optional[PduId]
|
||||
pdu_id: PduId | None
|
||||
payload: bytes
|
||||
|
||||
def __init__(self, callback: Callable[[PduId, bytes], None]) -> None:
|
||||
@@ -725,7 +725,7 @@ class PduAssembler:
|
||||
# -----------------------------------------------------------------------------
|
||||
class Command:
|
||||
pdu_id: ClassVar[PduId]
|
||||
_payload: Optional[bytes] = None
|
||||
_payload: bytes | None = None
|
||||
|
||||
_Command = TypeVar('_Command', bound='Command')
|
||||
subclasses: ClassVar[dict[int, type[Command]]] = {}
|
||||
@@ -1017,7 +1017,7 @@ class AddToNowPlayingCommand(Command):
|
||||
# -----------------------------------------------------------------------------
|
||||
class Response:
|
||||
pdu_id: PduId
|
||||
_payload: Optional[bytes] = None
|
||||
_payload: bytes | None = None
|
||||
|
||||
fields: ClassVar[hci.Fields] = ()
|
||||
subclasses: ClassVar[dict[PduId, type[Response]]] = {}
|
||||
@@ -1079,7 +1079,7 @@ class NotImplementedResponse(Response):
|
||||
class GetCapabilitiesResponse(Response):
|
||||
pdu_id = PduId.GET_CAPABILITIES
|
||||
capability_id: GetCapabilitiesCommand.CapabilityId
|
||||
capabilities: Sequence[Union[SupportsBytes, bytes]]
|
||||
capabilities: Sequence[SupportsBytes | bytes]
|
||||
|
||||
@classmethod
|
||||
def from_parameters(cls, parameters: bytes) -> Response:
|
||||
@@ -1092,7 +1092,7 @@ class GetCapabilitiesResponse(Response):
|
||||
capability_id = GetCapabilitiesCommand.CapabilityId(parameters[0])
|
||||
capability_count = parameters[1]
|
||||
|
||||
capabilities: list[Union[SupportsBytes, bytes]]
|
||||
capabilities: list[SupportsBytes | bytes]
|
||||
if capability_id == GetCapabilitiesCommand.CapabilityId.EVENTS_SUPPORTED:
|
||||
capabilities = [EventId(parameters[2 + x]) for x in range(capability_count)]
|
||||
else:
|
||||
@@ -1363,7 +1363,7 @@ class AddToNowPlayingResponse(Response):
|
||||
# -----------------------------------------------------------------------------
|
||||
class Event:
|
||||
event_id: EventId
|
||||
_pdu: Optional[bytes] = None
|
||||
_pdu: bytes | None = None
|
||||
|
||||
_Event = TypeVar('_Event', bound='Event')
|
||||
subclasses: ClassVar[dict[int, type[Event]]] = {}
|
||||
@@ -1436,13 +1436,13 @@ class PlayerApplicationSettingChangedEvent(Event):
|
||||
attribute_id: ApplicationSetting.AttributeId = field(
|
||||
metadata=ApplicationSetting.AttributeId.type_metadata(1)
|
||||
)
|
||||
value_id: Union[
|
||||
ApplicationSetting.EqualizerOnOffStatus,
|
||||
ApplicationSetting.RepeatModeStatus,
|
||||
ApplicationSetting.ShuffleOnOffStatus,
|
||||
ApplicationSetting.ScanOnOffStatus,
|
||||
ApplicationSetting.GenericValue,
|
||||
] = field(metadata=hci.metadata(1))
|
||||
value_id: (
|
||||
ApplicationSetting.EqualizerOnOffStatus
|
||||
| ApplicationSetting.RepeatModeStatus
|
||||
| ApplicationSetting.ShuffleOnOffStatus
|
||||
| ApplicationSetting.ScanOnOffStatus
|
||||
| ApplicationSetting.GenericValue
|
||||
) = field(metadata=hci.metadata(1))
|
||||
|
||||
def __post_init__(self) -> None:
|
||||
super().__post_init__()
|
||||
@@ -1628,17 +1628,17 @@ class Protocol(utils.EventEmitter):
|
||||
delegate: Delegate
|
||||
send_transaction_label: int
|
||||
command_pdu_assembler: PduAssembler
|
||||
receive_command_state: Optional[ReceiveCommandState]
|
||||
receive_command_state: ReceiveCommandState | None
|
||||
response_pdu_assembler: PduAssembler
|
||||
receive_response_state: Optional[ReceiveResponseState]
|
||||
avctp_protocol: Optional[avctp.Protocol]
|
||||
receive_response_state: ReceiveResponseState | None
|
||||
avctp_protocol: avctp.Protocol | None
|
||||
free_commands: asyncio.Queue
|
||||
pending_commands: dict[int, PendingCommand] # Pending commands, by label
|
||||
notification_listeners: dict[EventId, NotificationListener]
|
||||
|
||||
@staticmethod
|
||||
def _check_vendor_dependent_frame(
|
||||
frame: Union[avc.VendorDependentCommandFrame, avc.VendorDependentResponseFrame],
|
||||
frame: avc.VendorDependentCommandFrame | avc.VendorDependentResponseFrame,
|
||||
) -> bool:
|
||||
if frame.company_id != AVRCP_BLUETOOTH_SIG_COMPANY_ID:
|
||||
logger.debug("unsupported company id, ignoring")
|
||||
@@ -1650,7 +1650,7 @@ class Protocol(utils.EventEmitter):
|
||||
|
||||
return True
|
||||
|
||||
def __init__(self, delegate: Optional[Delegate] = None) -> None:
|
||||
def __init__(self, delegate: Delegate | None = None) -> None:
|
||||
super().__init__()
|
||||
self.delegate = delegate if delegate else Delegate()
|
||||
self.command_pdu_assembler = PduAssembler(self._on_command_pdu)
|
||||
@@ -2011,7 +2011,7 @@ class Protocol(utils.EventEmitter):
|
||||
f"{command} is not a valid AV/C Command Frame"
|
||||
)
|
||||
logger.debug(
|
||||
f"<<< AVCTP Command, transaction_label={transaction_label}: " f"{command}"
|
||||
f"<<< AVCTP Command, transaction_label={transaction_label}: {command}"
|
||||
)
|
||||
|
||||
# Only addressing the unit, or the PANEL subunit with subunit ID 0 is supported
|
||||
@@ -2067,9 +2067,7 @@ class Protocol(utils.EventEmitter):
|
||||
# TODO handle other types
|
||||
self.send_not_implemented_response(transaction_label, command)
|
||||
|
||||
def _on_avctp_response(
|
||||
self, transaction_label: int, payload: Optional[bytes]
|
||||
) -> None:
|
||||
def _on_avctp_response(self, transaction_label: int, payload: bytes | None) -> None:
|
||||
response = avc.ResponseFrame.from_bytes(payload) if payload else None
|
||||
if not isinstance(response, avc.ResponseFrame):
|
||||
raise core.InvalidPacketError(
|
||||
@@ -2176,7 +2174,7 @@ class Protocol(utils.EventEmitter):
|
||||
# NOTE: with a small number of supported responses, a manual switch like this
|
||||
# is Ok, but if/when more responses are supported, a lookup mechanism would be
|
||||
# more appropriate.
|
||||
response: Optional[Response] = None
|
||||
response: Response | None = None
|
||||
if response_code == avc.ResponseFrame.ResponseCode.REJECTED:
|
||||
response = RejectedResponse(pdu_id=pdu_id, status_code=StatusCode(pdu[0]))
|
||||
elif response_code == avc.ResponseFrame.ResponseCode.NOT_IMPLEMENTED:
|
||||
|
||||
+15
-15
@@ -163,23 +163,23 @@ class AacAudioRtpPacket:
|
||||
cls, reader: BitReader, channel_configuration: int, audio_object_type: int
|
||||
) -> Self:
|
||||
# GASpecificConfig - ISO/EIC 14496-3 Table 4.1
|
||||
frame_length_flag = reader.read(1)
|
||||
reader.read(1) # frame_length_flag
|
||||
depends_on_core_coder = reader.read(1)
|
||||
if depends_on_core_coder:
|
||||
core_coder_delay = reader.read(14)
|
||||
reader.read(14) # core_coder_delay
|
||||
extension_flag = reader.read(1)
|
||||
if not channel_configuration:
|
||||
raise core.InvalidPacketError('program_config_element not supported')
|
||||
if audio_object_type in (6, 20):
|
||||
layer_nr = reader.read(3)
|
||||
reader.read(3) # layer_nr
|
||||
if extension_flag:
|
||||
if audio_object_type == 22:
|
||||
num_of_sub_frame = reader.read(5)
|
||||
layer_length = reader.read(11)
|
||||
reader.read(5) # num_of_sub_frame
|
||||
reader.read(11) # layer_length
|
||||
if audio_object_type in (17, 19, 20, 23):
|
||||
aac_section_data_resilience_flags = reader.read(1)
|
||||
aac_scale_factor_data_resilience_flags = reader.read(1)
|
||||
aac_spectral_data_resilience_flags = reader.read(1)
|
||||
reader.read(1) # aac_section_data_resilience_flags
|
||||
reader.read(1) # aac_scale_factor_data_resilience_flags
|
||||
reader.read(1) # aac_spectral_data_resilience_flags
|
||||
extension_flag_3 = reader.read(1)
|
||||
if extension_flag_3 == 1:
|
||||
raise core.InvalidPacketError('extensionFlag3 == 1 not supported')
|
||||
@@ -364,10 +364,10 @@ class AacAudioRtpPacket:
|
||||
if audio_mux_version_a != 0:
|
||||
raise core.InvalidPacketError('audioMuxVersionA != 0 not supported')
|
||||
if audio_mux_version == 1:
|
||||
tara_buffer_fullness = AacAudioRtpPacket.read_latm_value(reader)
|
||||
stream_cnt = 0
|
||||
all_streams_same_time_framing = reader.read(1)
|
||||
num_sub_frames = reader.read(6)
|
||||
AacAudioRtpPacket.read_latm_value(reader) # tara_buffer_fullness
|
||||
# stream_cnt = 0
|
||||
reader.read(1) # all_streams_same_time_framing
|
||||
reader.read(6) # num_sub_frames
|
||||
num_program = reader.read(4)
|
||||
if num_program != 0:
|
||||
raise core.InvalidPacketError('num_program != 0 not supported')
|
||||
@@ -391,9 +391,9 @@ class AacAudioRtpPacket:
|
||||
reader.skip(asc_len)
|
||||
frame_length_type = reader.read(3)
|
||||
if frame_length_type == 0:
|
||||
latm_buffer_fullness = reader.read(8)
|
||||
reader.read(8) # latm_buffer_fullness
|
||||
elif frame_length_type == 1:
|
||||
frame_length = reader.read(9)
|
||||
reader.read(9) # frame_length
|
||||
else:
|
||||
raise core.InvalidPacketError(
|
||||
f'frame_length_type {frame_length_type} not supported'
|
||||
@@ -413,7 +413,7 @@ class AacAudioRtpPacket:
|
||||
break
|
||||
crc_check_present = reader.read(1)
|
||||
if crc_check_present:
|
||||
crc_checksum = reader.read(8)
|
||||
reader.read(8) # crc_checksum
|
||||
|
||||
return cls(other_data_present, other_data_len_bits, audio_specific_config)
|
||||
|
||||
|
||||
+7
-8
@@ -13,7 +13,6 @@
|
||||
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
||||
|
||||
from functools import partial
|
||||
from typing import Optional, Union
|
||||
|
||||
|
||||
class ColorError(ValueError):
|
||||
@@ -38,7 +37,7 @@ STYLES = (
|
||||
)
|
||||
|
||||
|
||||
ColorSpec = Union[str, int]
|
||||
ColorSpec = str | int
|
||||
|
||||
|
||||
def _join(*values: ColorSpec) -> str:
|
||||
@@ -56,14 +55,14 @@ def _color_code(spec: ColorSpec, base: int) -> str:
|
||||
elif isinstance(spec, int) and 0 <= spec <= 255:
|
||||
return _join(base + 8, 5, spec)
|
||||
else:
|
||||
raise ColorError('Invalid color spec "%s"' % spec)
|
||||
raise ColorError(f'Invalid color spec "{spec}"')
|
||||
|
||||
|
||||
def color(
|
||||
s: str,
|
||||
fg: Optional[ColorSpec] = None,
|
||||
bg: Optional[ColorSpec] = None,
|
||||
style: Optional[str] = None,
|
||||
fg: ColorSpec | None = None,
|
||||
bg: ColorSpec | None = None,
|
||||
style: str | None = None,
|
||||
) -> str:
|
||||
codes: list[ColorSpec] = []
|
||||
|
||||
@@ -76,10 +75,10 @@ def color(
|
||||
if style_part in STYLES:
|
||||
codes.append(STYLES.index(style_part))
|
||||
else:
|
||||
raise ColorError('Invalid style "%s"' % style_part)
|
||||
raise ColorError(f'Invalid style "{style_part}"')
|
||||
|
||||
if codes:
|
||||
return '\x1b[{0}m{1}\x1b[0m'.format(_join(*codes), s)
|
||||
return f'\x1b[{_join(*codes)}m{s}\x1b[0m'
|
||||
else:
|
||||
return s
|
||||
|
||||
|
||||
+670
-398
File diff suppressed because it is too large
Load Diff
+41
-49
@@ -20,15 +20,11 @@ from __future__ import annotations
|
||||
import dataclasses
|
||||
import enum
|
||||
import struct
|
||||
from collections.abc import Iterable
|
||||
from typing import (
|
||||
TYPE_CHECKING,
|
||||
Any,
|
||||
ClassVar,
|
||||
Iterable,
|
||||
Literal,
|
||||
Optional,
|
||||
Type,
|
||||
Union,
|
||||
cast,
|
||||
overload,
|
||||
)
|
||||
@@ -103,7 +99,7 @@ class BaseError(BaseBumbleError):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
error_code: Optional[int],
|
||||
error_code: int | None,
|
||||
error_namespace: str = '',
|
||||
error_name: str = '',
|
||||
details: str = '',
|
||||
@@ -216,11 +212,9 @@ class UUID:
|
||||
UUIDS: list[UUID] = [] # Registry of all instances created
|
||||
|
||||
uuid_bytes: bytes
|
||||
name: Optional[str]
|
||||
name: str | None
|
||||
|
||||
def __init__(
|
||||
self, uuid_str_or_int: Union[str, int], name: Optional[str] = None
|
||||
) -> None:
|
||||
def __init__(self, uuid_str_or_int: str | int, name: str | None = None) -> None:
|
||||
if isinstance(uuid_str_or_int, int):
|
||||
self.uuid_bytes = struct.pack('<H', uuid_str_or_int)
|
||||
else:
|
||||
@@ -253,7 +247,7 @@ class UUID:
|
||||
return self
|
||||
|
||||
@classmethod
|
||||
def from_bytes(cls, uuid_bytes: bytes, name: Optional[str] = None) -> UUID:
|
||||
def from_bytes(cls, uuid_bytes: bytes, name: str | None = None) -> UUID:
|
||||
if len(uuid_bytes) in (2, 4, 16):
|
||||
self = cls.__new__(cls)
|
||||
self.uuid_bytes = uuid_bytes
|
||||
@@ -264,11 +258,11 @@ class UUID:
|
||||
raise InvalidArgumentError('only 2, 4 and 16 bytes are allowed')
|
||||
|
||||
@classmethod
|
||||
def from_16_bits(cls, uuid_16: int, name: Optional[str] = None) -> UUID:
|
||||
def from_16_bits(cls, uuid_16: int, name: str | None = None) -> UUID:
|
||||
return cls.from_bytes(struct.pack('<H', uuid_16), name)
|
||||
|
||||
@classmethod
|
||||
def from_32_bits(cls, uuid_32: int, name: Optional[str] = None) -> UUID:
|
||||
def from_32_bits(cls, uuid_32: int, name: str | None = None) -> UUID:
|
||||
return cls.from_bytes(struct.pack('<I', uuid_32), name)
|
||||
|
||||
@classmethod
|
||||
@@ -734,7 +728,7 @@ class ClassOfDevice:
|
||||
MajorDeviceClass.HEALTH: HEALTH_MINOR_DEVICE_CLASS_LABELS,
|
||||
}
|
||||
|
||||
_MINOR_DEVICE_CLASSES: ClassVar[dict[MajorDeviceClass, Type]] = {
|
||||
_MINOR_DEVICE_CLASSES: ClassVar[dict[MajorDeviceClass, type]] = {
|
||||
MajorDeviceClass.COMPUTER: ComputerMinorDeviceClass,
|
||||
MajorDeviceClass.PHONE: PhoneMinorDeviceClass,
|
||||
MajorDeviceClass.LAN_NETWORK_ACCESS_POINT: LanNetworkMinorDeviceClass,
|
||||
@@ -749,17 +743,17 @@ class ClassOfDevice:
|
||||
|
||||
major_service_classes: MajorServiceClasses
|
||||
major_device_class: MajorDeviceClass
|
||||
minor_device_class: Union[
|
||||
ComputerMinorDeviceClass,
|
||||
PhoneMinorDeviceClass,
|
||||
LanNetworkMinorDeviceClass,
|
||||
AudioVideoMinorDeviceClass,
|
||||
PeripheralMinorDeviceClass,
|
||||
WearableMinorDeviceClass,
|
||||
ToyMinorDeviceClass,
|
||||
HealthMinorDeviceClass,
|
||||
int,
|
||||
]
|
||||
minor_device_class: (
|
||||
ComputerMinorDeviceClass
|
||||
| PhoneMinorDeviceClass
|
||||
| LanNetworkMinorDeviceClass
|
||||
| AudioVideoMinorDeviceClass
|
||||
| PeripheralMinorDeviceClass
|
||||
| WearableMinorDeviceClass
|
||||
| ToyMinorDeviceClass
|
||||
| HealthMinorDeviceClass
|
||||
| int
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def from_int(cls, class_of_device: int) -> Self:
|
||||
@@ -1548,7 +1542,7 @@ class DataType:
|
||||
return f"{self.__class__.__name__}({self.value_string()})"
|
||||
|
||||
@classmethod
|
||||
def from_advertising_data(cls, advertising_data: AdvertisingData) -> Optional[Self]:
|
||||
def from_advertising_data(cls, advertising_data: AdvertisingData) -> Self | None:
|
||||
if (data := advertising_data.get(cls.ad_type, raw=True)) is None:
|
||||
return None
|
||||
|
||||
@@ -1576,16 +1570,16 @@ class DataType:
|
||||
# -----------------------------------------------------------------------------
|
||||
# Advertising Data
|
||||
# -----------------------------------------------------------------------------
|
||||
AdvertisingDataObject = Union[
|
||||
list[UUID],
|
||||
tuple[UUID, bytes],
|
||||
bytes,
|
||||
str,
|
||||
int,
|
||||
tuple[int, int],
|
||||
tuple[int, bytes],
|
||||
Appearance,
|
||||
]
|
||||
AdvertisingDataObject = (
|
||||
list[UUID]
|
||||
| tuple[UUID, bytes]
|
||||
| bytes
|
||||
| str
|
||||
| int
|
||||
| tuple[int, int]
|
||||
| tuple[int, bytes]
|
||||
| Appearance
|
||||
)
|
||||
|
||||
|
||||
class AdvertisingData:
|
||||
@@ -1722,7 +1716,7 @@ class AdvertisingData:
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
ad_structures: Optional[Iterable[Union[tuple[int, bytes], DataType]]] = None,
|
||||
ad_structures: Iterable[tuple[int, bytes] | DataType] | None = None,
|
||||
) -> None:
|
||||
if ad_structures is None:
|
||||
ad_structures = []
|
||||
@@ -2020,7 +2014,7 @@ class AdvertisingData:
|
||||
AdvertisingData.Type.LIST_OF_128_BIT_SERVICE_SOLICITATION_UUIDS,
|
||||
],
|
||||
raw: Literal[False] = False,
|
||||
) -> Optional[list[UUID]]: ...
|
||||
) -> list[UUID] | None: ...
|
||||
|
||||
@overload
|
||||
def get(
|
||||
@@ -2031,7 +2025,7 @@ class AdvertisingData:
|
||||
AdvertisingData.Type.SERVICE_DATA_128_BIT_UUID,
|
||||
],
|
||||
raw: Literal[False] = False,
|
||||
) -> Optional[tuple[UUID, bytes]]: ...
|
||||
) -> tuple[UUID, bytes] | None: ...
|
||||
|
||||
@overload
|
||||
def get(
|
||||
@@ -2043,7 +2037,7 @@ class AdvertisingData:
|
||||
AdvertisingData.Type.BROADCAST_NAME,
|
||||
],
|
||||
raw: Literal[False] = False,
|
||||
) -> Optional[Optional[str]]: ...
|
||||
) -> str | None: ...
|
||||
|
||||
@overload
|
||||
def get(
|
||||
@@ -2055,38 +2049,36 @@ class AdvertisingData:
|
||||
AdvertisingData.Type.CLASS_OF_DEVICE,
|
||||
],
|
||||
raw: Literal[False] = False,
|
||||
) -> Optional[int]: ...
|
||||
) -> int | None: ...
|
||||
|
||||
@overload
|
||||
def get(
|
||||
self,
|
||||
type_id: Literal[AdvertisingData.Type.PERIPHERAL_CONNECTION_INTERVAL_RANGE,],
|
||||
raw: Literal[False] = False,
|
||||
) -> Optional[tuple[int, int]]: ...
|
||||
) -> tuple[int, int] | None: ...
|
||||
|
||||
@overload
|
||||
def get(
|
||||
self,
|
||||
type_id: Literal[AdvertisingData.Type.MANUFACTURER_SPECIFIC_DATA,],
|
||||
raw: Literal[False] = False,
|
||||
) -> Optional[tuple[int, bytes]]: ...
|
||||
) -> tuple[int, bytes] | None: ...
|
||||
|
||||
@overload
|
||||
def get(
|
||||
self,
|
||||
type_id: Literal[AdvertisingData.Type.APPEARANCE,],
|
||||
raw: Literal[False] = False,
|
||||
) -> Optional[Appearance]: ...
|
||||
) -> Appearance | None: ...
|
||||
|
||||
@overload
|
||||
def get(self, type_id: int, raw: Literal[True]) -> Optional[bytes]: ...
|
||||
def get(self, type_id: int, raw: Literal[True]) -> bytes | None: ...
|
||||
|
||||
@overload
|
||||
def get(
|
||||
self, type_id: int, raw: bool = False
|
||||
) -> Optional[AdvertisingDataObject]: ...
|
||||
def get(self, type_id: int, raw: bool = False) -> AdvertisingDataObject | None: ...
|
||||
|
||||
def get(self, type_id: int, raw: bool = False) -> Optional[AdvertisingDataObject]:
|
||||
def get(self, type_id: int, raw: bool = False) -> AdvertisingDataObject | None:
|
||||
'''
|
||||
Get advertising data as a simple AdvertisingDataObject object.
|
||||
|
||||
|
||||
@@ -25,10 +25,11 @@ try:
|
||||
from bumble.crypto.cryptography import EccKey, aes_cmac, e
|
||||
except ImportError:
|
||||
logging.getLogger(__name__).debug(
|
||||
"Unable to import cryptography, use built-in primitives."
|
||||
"Unable to import cryptography, using built-in primitives."
|
||||
)
|
||||
from bumble.crypto.builtin import EccKey, aes_cmac, e # type: ignore[assignment]
|
||||
|
||||
_EccKey = EccKey # For the linter only
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Logging
|
||||
|
||||
@@ -29,7 +29,6 @@ import dataclasses
|
||||
import functools
|
||||
import secrets
|
||||
import struct
|
||||
from typing import Optional
|
||||
|
||||
from bumble import core
|
||||
|
||||
@@ -85,7 +84,6 @@ class _AES:
|
||||
# fmt: on
|
||||
|
||||
def __init__(self, key: bytes) -> None:
|
||||
|
||||
if len(key) not in (16, 24, 32):
|
||||
raise core.InvalidArgumentError(f'Invalid key size {len(key)}')
|
||||
|
||||
@@ -112,7 +110,6 @@ class _AES:
|
||||
r_con_pointer = 0
|
||||
t = kc
|
||||
while t < round_key_count:
|
||||
|
||||
tt = tk[kc - 1]
|
||||
tk[0] ^= (
|
||||
(self._S[(tt >> 16) & 0xFF] << 24)
|
||||
@@ -269,7 +266,6 @@ class _ECB:
|
||||
|
||||
|
||||
class _CBC:
|
||||
|
||||
def __init__(self, key: bytes, iv: bytes = bytes(16)) -> None:
|
||||
if len(iv) != 16:
|
||||
raise core.InvalidArgumentError(
|
||||
@@ -302,7 +298,6 @@ class _CBC:
|
||||
|
||||
|
||||
class _CMAC:
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
key: bytes,
|
||||
@@ -313,7 +308,7 @@ class _CMAC:
|
||||
self.digest_size = mac_len
|
||||
self._key = key
|
||||
self._block_size = bs = 16
|
||||
self._mac_tag: Optional[bytes] = None
|
||||
self._mac_tag: bytes | None = None
|
||||
self._update_after_digest = update_after_digest
|
||||
|
||||
# Section 5.3 of NIST SP 800 38B and Appendix B
|
||||
@@ -352,7 +347,7 @@ class _CMAC:
|
||||
self._last_ct = zero_block
|
||||
|
||||
# Last block that was encrypted with AES
|
||||
self._last_pt: Optional[bytes] = None
|
||||
self._last_pt: bytes | None = None
|
||||
|
||||
# Counter for total message size
|
||||
self._data_size = 0
|
||||
@@ -414,7 +409,6 @@ class _CMAC:
|
||||
self._last_pt = _xor(second_last, data_block[-bs:])
|
||||
|
||||
def digest(self) -> bytes:
|
||||
|
||||
bs = self._block_size
|
||||
|
||||
if self._mac_tag is not None and not self._update_after_digest:
|
||||
|
||||
@@ -25,7 +25,8 @@ from __future__ import annotations
|
||||
import dataclasses
|
||||
import math
|
||||
import struct
|
||||
from typing import Any, ClassVar, Sequence
|
||||
from collections.abc import Sequence
|
||||
from typing import Any, ClassVar
|
||||
|
||||
from typing_extensions import Self
|
||||
|
||||
|
||||
+2
-3
@@ -12,7 +12,6 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from typing import Union
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Constants
|
||||
@@ -167,12 +166,12 @@ class G722Decoder:
|
||||
# The initial value in BLOCK 3H
|
||||
self._band[1].det = 8
|
||||
|
||||
def decode_frame(self, encoded_data: Union[bytes, bytearray]) -> bytearray:
|
||||
def decode_frame(self, encoded_data: bytes | bytearray) -> bytearray:
|
||||
result_array = bytearray(len(encoded_data) * 4)
|
||||
self.g722_decode(result_array, encoded_data)
|
||||
return result_array
|
||||
|
||||
def g722_decode(self, result_array, encoded_data: Union[bytes, bytearray]) -> int:
|
||||
def g722_decode(self, result_array, encoded_data: bytes | bytearray) -> int:
|
||||
"""Decode the data frame using g722 decoder."""
|
||||
result_length = 0
|
||||
|
||||
|
||||
+139
-122
@@ -25,20 +25,15 @@ import itertools
|
||||
import json
|
||||
import logging
|
||||
import secrets
|
||||
import sys
|
||||
from collections.abc import Iterable, Sequence
|
||||
from collections.abc import Awaitable, Callable, Iterable, Sequence
|
||||
from contextlib import AsyncExitStack, asynccontextmanager, closing
|
||||
from dataclasses import dataclass, field
|
||||
from enum import Enum, IntEnum
|
||||
from typing import (
|
||||
TYPE_CHECKING,
|
||||
Any,
|
||||
Awaitable,
|
||||
Callable,
|
||||
ClassVar,
|
||||
Optional,
|
||||
TypeVar,
|
||||
Union,
|
||||
cast,
|
||||
overload,
|
||||
)
|
||||
@@ -189,7 +184,7 @@ class Advertisement:
|
||||
self.data = AdvertisingData.from_bytes(self.data_bytes)
|
||||
|
||||
@classmethod
|
||||
def from_advertising_report(cls, report) -> Optional[Advertisement]:
|
||||
def from_advertising_report(cls, report) -> Advertisement | None:
|
||||
if isinstance(report, hci.HCI_LE_Advertising_Report_Event.Report):
|
||||
return LegacyAdvertisement.from_advertising_report(report)
|
||||
|
||||
@@ -265,12 +260,22 @@ class ExtendedAdvertisement(Advertisement):
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
class AdvertisementDataAccumulator:
|
||||
last_advertisement: Advertisement | None
|
||||
last_data: bytes
|
||||
passive: bool
|
||||
|
||||
def __init__(self, passive: bool = False):
|
||||
self.passive = passive
|
||||
self.last_advertisement = None
|
||||
self.last_data = b''
|
||||
|
||||
def update(self, report):
|
||||
def update(
|
||||
self,
|
||||
report: (
|
||||
hci.HCI_LE_Advertising_Report_Event.Report
|
||||
| hci.HCI_LE_Extended_Advertising_Report_Event.Report
|
||||
),
|
||||
) -> Advertisement | None:
|
||||
advertisement = Advertisement.from_advertising_report(report)
|
||||
if advertisement is None:
|
||||
return None
|
||||
@@ -283,10 +288,12 @@ class AdvertisementDataAccumulator:
|
||||
and not self.last_advertisement.is_scan_response
|
||||
):
|
||||
# This is the response to a scannable advertisement
|
||||
result = Advertisement.from_advertising_report(report)
|
||||
result.is_connectable = self.last_advertisement.is_connectable
|
||||
result.is_scannable = True
|
||||
result.data = AdvertisingData.from_bytes(self.last_data + report.data)
|
||||
if result := Advertisement.from_advertising_report(report):
|
||||
result.is_connectable = self.last_advertisement.is_connectable
|
||||
result.is_scannable = True
|
||||
result.data = AdvertisingData.from_bytes(
|
||||
self.last_data + report.data
|
||||
)
|
||||
self.last_data = b''
|
||||
else:
|
||||
if (
|
||||
@@ -593,11 +600,11 @@ class AdvertisingSet(utils.EventEmitter):
|
||||
device: Device
|
||||
advertising_handle: int
|
||||
auto_restart: bool
|
||||
random_address: Optional[hci.Address]
|
||||
random_address: hci.Address | None
|
||||
advertising_parameters: AdvertisingParameters
|
||||
advertising_data: bytes
|
||||
scan_response_data: bytes
|
||||
periodic_advertising_parameters: Optional[PeriodicAdvertisingParameters]
|
||||
periodic_advertising_parameters: PeriodicAdvertisingParameters | None
|
||||
periodic_advertising_data: bytes
|
||||
selected_tx_power: int = 0
|
||||
enabled: bool = False
|
||||
@@ -844,7 +851,7 @@ class PeriodicAdvertisingSync(utils.EventEmitter):
|
||||
TERMINATED = 6
|
||||
|
||||
_state: State
|
||||
sync_handle: Optional[int]
|
||||
sync_handle: int | None
|
||||
advertiser_address: hci.Address
|
||||
sid: int
|
||||
skip: int
|
||||
@@ -857,8 +864,8 @@ class PeriodicAdvertisingSync(utils.EventEmitter):
|
||||
|
||||
EVENT_STATE_CHANGE = "state_change"
|
||||
EVENT_ESTABLISHMENT = "establishment"
|
||||
EVENT_ESTABLISHMENT_ERROR = "establishment_error"
|
||||
EVENT_CANCELLATION = "cancellation"
|
||||
EVENT_ERROR = "error"
|
||||
EVENT_LOSS = "loss"
|
||||
EVENT_PERIODIC_ADVERTISEMENT = "periodic_advertisement"
|
||||
EVENT_BIGINFO_ADVERTISEMENT = "biginfo_advertisement"
|
||||
@@ -991,7 +998,7 @@ class PeriodicAdvertisingSync(utils.EventEmitter):
|
||||
return
|
||||
|
||||
self.state = self.State.ERROR
|
||||
self.emit(self.EVENT_ERROR)
|
||||
self.emit(self.EVENT_ESTABLISHMENT_ERROR)
|
||||
|
||||
def on_loss(self):
|
||||
self.state = self.State.LOST
|
||||
@@ -1271,7 +1278,7 @@ class Peer:
|
||||
return mtu
|
||||
|
||||
async def discover_service(
|
||||
self, uuid: Union[core.UUID, str]
|
||||
self, uuid: core.UUID | str
|
||||
) -> list[gatt_client.ServiceProxy]:
|
||||
return await self.gatt_client.discover_service(uuid)
|
||||
|
||||
@@ -1287,8 +1294,8 @@ class Peer:
|
||||
|
||||
async def discover_characteristics(
|
||||
self,
|
||||
uuids: Iterable[Union[core.UUID, str]] = (),
|
||||
service: Optional[gatt_client.ServiceProxy] = None,
|
||||
uuids: Iterable[core.UUID | str] = (),
|
||||
service: gatt_client.ServiceProxy | None = None,
|
||||
) -> list[gatt_client.CharacteristicProxy[bytes]]:
|
||||
return await self.gatt_client.discover_characteristics(
|
||||
uuids=uuids, service=service
|
||||
@@ -1296,9 +1303,9 @@ class Peer:
|
||||
|
||||
async def discover_descriptors(
|
||||
self,
|
||||
characteristic: Optional[gatt_client.CharacteristicProxy] = None,
|
||||
start_handle: Optional[int] = None,
|
||||
end_handle: Optional[int] = None,
|
||||
characteristic: gatt_client.CharacteristicProxy | None = None,
|
||||
start_handle: int | None = None,
|
||||
end_handle: int | None = None,
|
||||
):
|
||||
return await self.gatt_client.discover_descriptors(
|
||||
characteristic, start_handle, end_handle
|
||||
@@ -1319,7 +1326,7 @@ class Peer:
|
||||
async def subscribe(
|
||||
self,
|
||||
characteristic: gatt_client.CharacteristicProxy,
|
||||
subscriber: Optional[Callable[[bytes], Any]] = None,
|
||||
subscriber: Callable[[bytes], Any] | None = None,
|
||||
prefer_notify: bool = True,
|
||||
) -> None:
|
||||
return await self.gatt_client.subscribe(
|
||||
@@ -1329,25 +1336,23 @@ class Peer:
|
||||
async def unsubscribe(
|
||||
self,
|
||||
characteristic: gatt_client.CharacteristicProxy,
|
||||
subscriber: Optional[Callable[[bytes], Any]] = None,
|
||||
subscriber: Callable[[bytes], Any] | None = None,
|
||||
) -> None:
|
||||
return await self.gatt_client.unsubscribe(characteristic, subscriber)
|
||||
|
||||
async def read_value(
|
||||
self, attribute: Union[int, gatt_client.AttributeProxy]
|
||||
) -> bytes:
|
||||
async def read_value(self, attribute: int | gatt_client.AttributeProxy) -> bytes:
|
||||
return await self.gatt_client.read_value(attribute)
|
||||
|
||||
async def write_value(
|
||||
self,
|
||||
attribute: Union[int, gatt_client.AttributeProxy],
|
||||
attribute: int | gatt_client.AttributeProxy,
|
||||
value: bytes,
|
||||
with_response: bool = False,
|
||||
) -> None:
|
||||
return await self.gatt_client.write_value(attribute, value, with_response)
|
||||
|
||||
async def read_characteristics_by_uuid(
|
||||
self, uuid: core.UUID, service: Optional[gatt_client.ServiceProxy] = None
|
||||
self, uuid: core.UUID, service: gatt_client.ServiceProxy | None = None
|
||||
) -> list[bytes]:
|
||||
return await self.gatt_client.read_characteristics_by_uuid(uuid, service)
|
||||
|
||||
@@ -1357,7 +1362,7 @@ class Peer:
|
||||
def get_characteristics_by_uuid(
|
||||
self,
|
||||
uuid: core.UUID,
|
||||
service: Optional[Union[gatt_client.ServiceProxy, core.UUID]] = None,
|
||||
service: gatt_client.ServiceProxy | core.UUID | None = None,
|
||||
) -> list[gatt_client.CharacteristicProxy[bytes]]:
|
||||
if isinstance(service, core.UUID):
|
||||
return list(
|
||||
@@ -1373,7 +1378,7 @@ class Peer:
|
||||
|
||||
def create_service_proxy(
|
||||
self, proxy_class: type[_PROXY_CLASS]
|
||||
) -> Optional[_PROXY_CLASS]:
|
||||
) -> _PROXY_CLASS | None:
|
||||
if proxy := proxy_class.from_client(self.gatt_client):
|
||||
return cast(_PROXY_CLASS, proxy)
|
||||
|
||||
@@ -1381,7 +1386,7 @@ class Peer:
|
||||
|
||||
async def discover_service_and_create_proxy(
|
||||
self, proxy_class: type[_PROXY_CLASS]
|
||||
) -> Optional[_PROXY_CLASS]:
|
||||
) -> _PROXY_CLASS | None:
|
||||
# Discover the first matching service and its characteristics
|
||||
services = await self.discover_service(proxy_class.SERVICE_CLASS.UUID)
|
||||
if services:
|
||||
@@ -1390,7 +1395,7 @@ class Peer:
|
||||
return self.create_service_proxy(proxy_class)
|
||||
return None
|
||||
|
||||
async def sustain(self, timeout: Optional[float] = None) -> None:
|
||||
async def sustain(self, timeout: float | None = None) -> None:
|
||||
await self.connection.sustain(timeout)
|
||||
|
||||
# [Classic only]
|
||||
@@ -1433,7 +1438,7 @@ class ScoLink(utils.CompositeEventEmitter):
|
||||
acl_connection: Connection
|
||||
handle: int
|
||||
link_type: int
|
||||
sink: Optional[Callable[[hci.HCI_SynchronousDataPacket], Any]] = None
|
||||
sink: Callable[[hci.HCI_SynchronousDataPacket], Any] | None = None
|
||||
|
||||
EVENT_DISCONNECTION: ClassVar[str] = "disconnection"
|
||||
EVENT_DISCONNECTION_FAILURE: ClassVar[str] = "disconnection_failure"
|
||||
@@ -1616,8 +1621,8 @@ class CisLink(utils.EventEmitter, _IsoLink):
|
||||
cis_sync_delay: int = 0 # CIS sync delay, in microseconds
|
||||
transport_latency_c_to_p: int = 0 # C->P transport latency, in microseconds
|
||||
transport_latency_p_to_c: int = 0 # P->C transport latency, in microseconds
|
||||
phy_c_to_p: Optional[hci.Phy] = None
|
||||
phy_p_to_c: Optional[hci.Phy] = None
|
||||
phy_c_to_p: hci.Phy | None = None
|
||||
phy_p_to_c: hci.Phy | None = None
|
||||
nse: int = 0
|
||||
bn_c_to_p: int = 0
|
||||
bn_p_to_c: int = 0
|
||||
@@ -1705,11 +1710,11 @@ class Connection(utils.CompositeEventEmitter):
|
||||
handle: int
|
||||
transport: core.PhysicalTransport
|
||||
self_address: hci.Address
|
||||
self_resolvable_address: Optional[hci.Address]
|
||||
self_resolvable_address: hci.Address | None
|
||||
peer_address: hci.Address
|
||||
peer_name: Optional[str]
|
||||
peer_resolvable_address: Optional[hci.Address]
|
||||
peer_le_features: Optional[hci.LeFeatureMask]
|
||||
peer_name: str | None
|
||||
peer_resolvable_address: hci.Address | None
|
||||
peer_le_features: hci.LeFeatureMask | None
|
||||
role: hci.Role
|
||||
parameters: Parameters
|
||||
encryption: int
|
||||
@@ -1717,8 +1722,8 @@ class Connection(utils.CompositeEventEmitter):
|
||||
authenticated: bool
|
||||
sc: bool
|
||||
gatt_client: gatt_client.Client
|
||||
pairing_peer_io_capability: Optional[int]
|
||||
pairing_peer_authentication_requirements: Optional[int]
|
||||
pairing_peer_io_capability: int | None
|
||||
pairing_peer_authentication_requirements: int | None
|
||||
cs_configs: dict[int, ChannelSoundingConfig] # Config ID to Configuration
|
||||
cs_procedures: dict[int, ChannelSoundingProcedure] # Config ID to Procedures
|
||||
classic_mode: int = hci.HCI_Mode_Change_Event.Mode.ACTIVE
|
||||
@@ -1820,9 +1825,9 @@ class Connection(utils.CompositeEventEmitter):
|
||||
handle: int,
|
||||
transport: core.PhysicalTransport,
|
||||
self_address: hci.Address,
|
||||
self_resolvable_address: Optional[hci.Address],
|
||||
self_resolvable_address: hci.Address | None,
|
||||
peer_address: hci.Address,
|
||||
peer_resolvable_address: Optional[hci.Address],
|
||||
peer_resolvable_address: hci.Address | None,
|
||||
role: hci.Role,
|
||||
parameters: Parameters,
|
||||
):
|
||||
@@ -1885,8 +1890,8 @@ class Connection(utils.CompositeEventEmitter):
|
||||
) -> l2cap.LeCreditBasedChannel: ...
|
||||
|
||||
async def create_l2cap_channel(
|
||||
self, spec: Union[l2cap.ClassicChannelSpec, l2cap.LeCreditBasedChannelSpec]
|
||||
) -> Union[l2cap.ClassicChannel, l2cap.LeCreditBasedChannel]:
|
||||
self, spec: l2cap.ClassicChannelSpec | l2cap.LeCreditBasedChannelSpec
|
||||
) -> l2cap.ClassicChannel | l2cap.LeCreditBasedChannel:
|
||||
return await self.device.create_l2cap_channel(connection=self, spec=spec)
|
||||
|
||||
async def disconnect(
|
||||
@@ -1910,7 +1915,7 @@ class Connection(utils.CompositeEventEmitter):
|
||||
async def switch_role(self, role: hci.Role) -> None:
|
||||
return await self.device.switch_role(self, role)
|
||||
|
||||
async def sustain(self, timeout: Optional[float] = None) -> None:
|
||||
async def sustain(self, timeout: float | None = None) -> None:
|
||||
"""Idles the current task waiting for a disconnect or timeout"""
|
||||
|
||||
abort = asyncio.get_running_loop().create_future()
|
||||
@@ -1954,8 +1959,8 @@ class Connection(utils.CompositeEventEmitter):
|
||||
|
||||
async def set_phy(
|
||||
self,
|
||||
tx_phys: Optional[Iterable[hci.Phy]] = None,
|
||||
rx_phys: Optional[Iterable[hci.Phy]] = None,
|
||||
tx_phys: Iterable[hci.Phy] | None = None,
|
||||
rx_phys: Iterable[hci.Phy] | None = None,
|
||||
phy_options: int = 0,
|
||||
):
|
||||
return await self.device.set_connection_phy(self, tx_phys, rx_phys, phy_options)
|
||||
@@ -2059,15 +2064,21 @@ class DeviceConfiguration:
|
||||
AdvertisingData([data_types.CompleteLocalName(DEVICE_DEFAULT_NAME)])
|
||||
)
|
||||
irk: bytes = bytes(16) # This really must be changed for any level of security
|
||||
keystore: Optional[str] = None
|
||||
keystore: str | None = None
|
||||
address_resolution_offload: bool = False
|
||||
address_generation_offload: bool = False
|
||||
cis_enabled: bool = False
|
||||
channel_sounding_enabled: bool = False
|
||||
identity_address_type: Optional[int] = None
|
||||
identity_address_type: int | None = None
|
||||
io_capability: int = pairing.PairingDelegate.IoCapability.NO_OUTPUT_NO_INPUT
|
||||
gap_service_enabled: bool = True
|
||||
gatt_service_enabled: bool = True
|
||||
enhanced_retransmission_supported: bool = False
|
||||
l2cap_extended_features: Sequence[int] = (
|
||||
l2cap.L2CAP_Information_Request.ExtendedFeatures.FIXED_CHANNELS,
|
||||
l2cap.L2CAP_Information_Request.ExtendedFeatures.FCS_OPTION,
|
||||
l2cap.L2CAP_Information_Request.ExtendedFeatures.ENHANCED_RETRANSMISSION_MODE,
|
||||
)
|
||||
|
||||
def __post_init__(self) -> None:
|
||||
self.gatt_services: list[dict[str, Any]] = []
|
||||
@@ -2126,7 +2137,7 @@ class DeviceConfiguration:
|
||||
setattr(self, key, value)
|
||||
|
||||
def load_from_file(self, filename: str) -> None:
|
||||
with open(filename, 'r', encoding='utf-8') as file:
|
||||
with open(filename, encoding='utf-8') as file:
|
||||
self.load_from_dict(json.load(file))
|
||||
|
||||
@classmethod
|
||||
@@ -2237,12 +2248,12 @@ class Device(utils.CompositeEventEmitter):
|
||||
pending_connections: dict[hci.Address, Connection]
|
||||
classic_pending_accepts: dict[
|
||||
hci.Address,
|
||||
list[asyncio.Future[Union[Connection, tuple[hci.Address, int, int]]]],
|
||||
list[asyncio.Future[Connection | tuple[hci.Address, int, int]]],
|
||||
]
|
||||
advertisement_accumulators: dict[hci.Address, AdvertisementDataAccumulator]
|
||||
periodic_advertising_syncs: list[PeriodicAdvertisingSync]
|
||||
config: DeviceConfiguration
|
||||
legacy_advertiser: Optional[LegacyAdvertiser]
|
||||
legacy_advertiser: LegacyAdvertiser | None
|
||||
sco_links: dict[int, ScoLink]
|
||||
cis_links: dict[int, CisLink]
|
||||
bigs: dict[int, Big]
|
||||
@@ -2330,13 +2341,17 @@ class Device(utils.CompositeEventEmitter):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
name: Optional[str] = None,
|
||||
address: Optional[hci.Address] = None,
|
||||
config: Optional[DeviceConfiguration] = None,
|
||||
host: Optional[Host] = None,
|
||||
name: str | None = None,
|
||||
address: hci.Address | None = None,
|
||||
config: DeviceConfiguration | None = None,
|
||||
host: Host | None = None,
|
||||
) -> None:
|
||||
super().__init__()
|
||||
|
||||
# Use the initial config or a default
|
||||
config = config or DeviceConfiguration()
|
||||
self.config = config
|
||||
|
||||
self._host = None
|
||||
self.powered_on = False
|
||||
self.auto_restart_inquiry = True
|
||||
@@ -2344,7 +2359,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
self.gatt_server = gatt_server.Server(self)
|
||||
self.sdp_server = sdp.Server(self)
|
||||
self.l2cap_channel_manager = l2cap.ChannelManager(
|
||||
[l2cap.L2CAP_Information_Request.EXTENDED_FEATURE_FIXED_CHANNELS]
|
||||
config.l2cap_extended_features
|
||||
)
|
||||
self.advertisement_accumulators = {} # Accumulators, by address
|
||||
self.periodic_advertising_syncs = []
|
||||
@@ -2375,10 +2390,6 @@ class Device(utils.CompositeEventEmitter):
|
||||
# Own address type cache
|
||||
self.connect_own_address_type = None
|
||||
|
||||
# Use the initial config or a default
|
||||
config = config or DeviceConfiguration()
|
||||
self.config = config
|
||||
|
||||
self.name = config.name
|
||||
self.public_address = hci.Address.ANY
|
||||
self.random_address = config.address
|
||||
@@ -2390,7 +2401,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
self.le_simultaneous_enabled = config.le_simultaneous_enabled
|
||||
self.le_privacy_enabled = config.le_privacy_enabled
|
||||
self.le_rpa_timeout = config.le_rpa_timeout
|
||||
self.le_rpa_periodic_update_task: Optional[asyncio.Task] = None
|
||||
self.le_rpa_periodic_update_task: asyncio.Task | None = None
|
||||
self.le_subrate_enabled = config.le_subrate_enabled
|
||||
self.classic_enabled = config.classic_enabled
|
||||
self.cis_enabled = config.cis_enabled
|
||||
@@ -2414,8 +2425,8 @@ class Device(utils.CompositeEventEmitter):
|
||||
# can be initialized from a config object, and for backward compatibility for
|
||||
# client code that may set those values directly before calling
|
||||
# start_advertising().
|
||||
self.legacy_advertising_set: Optional[AdvertisingSet] = None
|
||||
self.legacy_advertiser: Optional[LegacyAdvertiser] = None
|
||||
self.legacy_advertising_set: AdvertisingSet | None = None
|
||||
self.legacy_advertiser: LegacyAdvertiser | None = None
|
||||
self.advertising_data = config.advertising_data
|
||||
self.scan_response_data = config.scan_response_data
|
||||
self.advertising_interval_min = config.advertising_interval_min
|
||||
@@ -2533,7 +2544,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
def sdp_service_records(self, service_records):
|
||||
self.sdp_server.service_records = service_records
|
||||
|
||||
def lookup_connection(self, connection_handle: int) -> Optional[Connection]:
|
||||
def lookup_connection(self, connection_handle: int) -> Connection | None:
|
||||
if connection := self.connections.get(connection_handle):
|
||||
return connection
|
||||
|
||||
@@ -2542,9 +2553,9 @@ class Device(utils.CompositeEventEmitter):
|
||||
def find_connection_by_bd_addr(
|
||||
self,
|
||||
bd_addr: hci.Address,
|
||||
transport: Optional[int] = None,
|
||||
transport: int | None = None,
|
||||
check_address_type: bool = False,
|
||||
) -> Optional[Connection]:
|
||||
) -> Connection | None:
|
||||
for connection in self.connections.values():
|
||||
if bytes(connection.peer_address) == bytes(bd_addr):
|
||||
if (
|
||||
@@ -2559,7 +2570,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
|
||||
def lookup_periodic_advertising_sync(
|
||||
self, sync_handle: int
|
||||
) -> Optional[PeriodicAdvertisingSync]:
|
||||
) -> PeriodicAdvertisingSync | None:
|
||||
return next(
|
||||
(
|
||||
sync
|
||||
@@ -2597,8 +2608,8 @@ class Device(utils.CompositeEventEmitter):
|
||||
async def create_l2cap_channel(
|
||||
self,
|
||||
connection: Connection,
|
||||
spec: Union[l2cap.ClassicChannelSpec, l2cap.LeCreditBasedChannelSpec],
|
||||
) -> Union[l2cap.ClassicChannel, l2cap.LeCreditBasedChannel]:
|
||||
spec: l2cap.ClassicChannelSpec | l2cap.LeCreditBasedChannelSpec,
|
||||
) -> l2cap.ClassicChannel | l2cap.LeCreditBasedChannel:
|
||||
if isinstance(spec, l2cap.ClassicChannelSpec):
|
||||
return await self.l2cap_channel_manager.create_classic_channel(
|
||||
connection=connection, spec=spec
|
||||
@@ -2612,25 +2623,25 @@ class Device(utils.CompositeEventEmitter):
|
||||
def create_l2cap_server(
|
||||
self,
|
||||
spec: l2cap.ClassicChannelSpec,
|
||||
handler: Optional[Callable[[l2cap.ClassicChannel], Any]] = None,
|
||||
handler: Callable[[l2cap.ClassicChannel], Any] | None = None,
|
||||
) -> l2cap.ClassicChannelServer: ...
|
||||
|
||||
@overload
|
||||
def create_l2cap_server(
|
||||
self,
|
||||
spec: l2cap.LeCreditBasedChannelSpec,
|
||||
handler: Optional[Callable[[l2cap.LeCreditBasedChannel], Any]] = None,
|
||||
handler: Callable[[l2cap.LeCreditBasedChannel], Any] | None = None,
|
||||
) -> l2cap.LeCreditBasedChannelServer: ...
|
||||
|
||||
def create_l2cap_server(
|
||||
self,
|
||||
spec: Union[l2cap.ClassicChannelSpec, l2cap.LeCreditBasedChannelSpec],
|
||||
handler: Union[
|
||||
Callable[[l2cap.ClassicChannel], Any],
|
||||
Callable[[l2cap.LeCreditBasedChannel], Any],
|
||||
None,
|
||||
] = None,
|
||||
) -> Union[l2cap.ClassicChannelServer, l2cap.LeCreditBasedChannelServer]:
|
||||
spec: l2cap.ClassicChannelSpec | l2cap.LeCreditBasedChannelSpec,
|
||||
handler: (
|
||||
Callable[[l2cap.ClassicChannel], Any]
|
||||
| Callable[[l2cap.LeCreditBasedChannel], Any]
|
||||
| None
|
||||
) = None,
|
||||
) -> l2cap.ClassicChannelServer | l2cap.LeCreditBasedChannelServer:
|
||||
if isinstance(spec, l2cap.ClassicChannelSpec):
|
||||
return self.l2cap_channel_manager.create_classic_server(
|
||||
spec=spec,
|
||||
@@ -2932,13 +2943,13 @@ class Device(utils.CompositeEventEmitter):
|
||||
async def start_advertising(
|
||||
self,
|
||||
advertising_type: AdvertisingType = AdvertisingType.UNDIRECTED_CONNECTABLE_SCANNABLE,
|
||||
target: Optional[hci.Address] = None,
|
||||
target: hci.Address | None = None,
|
||||
own_address_type: hci.OwnAddressType = hci.OwnAddressType.RANDOM,
|
||||
auto_restart: bool = False,
|
||||
advertising_data: Optional[bytes] = None,
|
||||
scan_response_data: Optional[bytes] = None,
|
||||
advertising_interval_min: Optional[float] = None,
|
||||
advertising_interval_max: Optional[float] = None,
|
||||
advertising_data: bytes | None = None,
|
||||
scan_response_data: bytes | None = None,
|
||||
advertising_interval_min: float | None = None,
|
||||
advertising_interval_max: float | None = None,
|
||||
) -> None:
|
||||
"""Start legacy advertising.
|
||||
|
||||
@@ -3042,11 +3053,11 @@ class Device(utils.CompositeEventEmitter):
|
||||
|
||||
async def create_advertising_set(
|
||||
self,
|
||||
advertising_parameters: Optional[AdvertisingParameters] = None,
|
||||
random_address: Optional[hci.Address] = None,
|
||||
advertising_parameters: AdvertisingParameters | None = None,
|
||||
random_address: hci.Address | None = None,
|
||||
advertising_data: bytes = b'',
|
||||
scan_response_data: bytes = b'',
|
||||
periodic_advertising_parameters: Optional[PeriodicAdvertisingParameters] = None,
|
||||
periodic_advertising_parameters: PeriodicAdvertisingParameters | None = None,
|
||||
periodic_advertising_data: bytes = b'',
|
||||
auto_start: bool = True,
|
||||
auto_restart: bool = False,
|
||||
@@ -3333,7 +3344,13 @@ class Device(utils.CompositeEventEmitter):
|
||||
return self.scanning
|
||||
|
||||
@host_event_handler
|
||||
def on_advertising_report(self, report):
|
||||
def on_advertising_report(
|
||||
self,
|
||||
report: (
|
||||
hci.HCI_LE_Advertising_Report_Event.Report
|
||||
| hci.HCI_LE_Extended_Advertising_Report_Event.Report
|
||||
),
|
||||
) -> None:
|
||||
if not (accumulator := self.advertisement_accumulators.get(report.address)):
|
||||
accumulator = AdvertisementDataAccumulator(passive=self.scanning_is_passive)
|
||||
self.advertisement_accumulators[report.address] = accumulator
|
||||
@@ -3576,13 +3593,13 @@ class Device(utils.CompositeEventEmitter):
|
||||
|
||||
async def connect(
|
||||
self,
|
||||
peer_address: Union[hci.Address, str],
|
||||
peer_address: hci.Address | str,
|
||||
transport: core.PhysicalTransport = PhysicalTransport.LE,
|
||||
connection_parameters_preferences: Optional[
|
||||
dict[hci.Phy, ConnectionParametersPreferences]
|
||||
] = None,
|
||||
connection_parameters_preferences: (
|
||||
dict[hci.Phy, ConnectionParametersPreferences] | None
|
||||
) = None,
|
||||
own_address_type: hci.OwnAddressType = hci.OwnAddressType.RANDOM,
|
||||
timeout: Optional[float] = DEVICE_DEFAULT_CONNECT_TIMEOUT,
|
||||
timeout: float | None = DEVICE_DEFAULT_CONNECT_TIMEOUT,
|
||||
always_resolve: bool = False,
|
||||
) -> Connection:
|
||||
'''
|
||||
@@ -3892,9 +3909,9 @@ class Device(utils.CompositeEventEmitter):
|
||||
|
||||
async def accept(
|
||||
self,
|
||||
peer_address: Union[hci.Address, str] = hci.Address.ANY,
|
||||
peer_address: hci.Address | str = hci.Address.ANY,
|
||||
role: hci.Role = hci.Role.PERIPHERAL,
|
||||
timeout: Optional[float] = DEVICE_DEFAULT_CONNECT_TIMEOUT,
|
||||
timeout: float | None = DEVICE_DEFAULT_CONNECT_TIMEOUT,
|
||||
) -> Connection:
|
||||
'''
|
||||
Wait and accept any incoming connection or a connection from `peer_address` when
|
||||
@@ -4018,7 +4035,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
self.pending_connections.pop(peer_address, None)
|
||||
|
||||
@asynccontextmanager
|
||||
async def connect_as_gatt(self, peer_address: Union[hci.Address, str]):
|
||||
async def connect_as_gatt(self, peer_address: hci.Address | str):
|
||||
async with AsyncExitStack() as stack:
|
||||
connection = await stack.enter_async_context(
|
||||
await self.connect(peer_address)
|
||||
@@ -4065,7 +4082,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
)
|
||||
|
||||
async def disconnect(
|
||||
self, connection: Union[Connection, ScoLink, CisLink], reason: int
|
||||
self, connection: Connection | ScoLink | CisLink, reason: int
|
||||
) -> None:
|
||||
# Create a future so that we can wait for the disconnection's result
|
||||
pending_disconnection = asyncio.get_running_loop().create_future()
|
||||
@@ -4204,8 +4221,8 @@ class Device(utils.CompositeEventEmitter):
|
||||
async def set_connection_phy(
|
||||
self,
|
||||
connection: Connection,
|
||||
tx_phys: Optional[Iterable[hci.Phy]] = None,
|
||||
rx_phys: Optional[Iterable[hci.Phy]] = None,
|
||||
tx_phys: Iterable[hci.Phy] | None = None,
|
||||
rx_phys: Iterable[hci.Phy] | None = None,
|
||||
phy_options: int = 0,
|
||||
):
|
||||
if not self.host.supports_command(hci.HCI_LE_SET_PHY_COMMAND):
|
||||
@@ -4229,8 +4246,8 @@ class Device(utils.CompositeEventEmitter):
|
||||
|
||||
async def set_default_phy(
|
||||
self,
|
||||
tx_phys: Optional[Iterable[hci.Phy]] = None,
|
||||
rx_phys: Optional[Iterable[hci.Phy]] = None,
|
||||
tx_phys: Iterable[hci.Phy] | None = None,
|
||||
rx_phys: Iterable[hci.Phy] | None = None,
|
||||
):
|
||||
all_phys_bits = (1 if tx_phys is None else 0) | (
|
||||
(1 if rx_phys is None else 0) << 1
|
||||
@@ -4284,7 +4301,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
if local_name == name:
|
||||
peer_address.set_result(address)
|
||||
|
||||
listener: Optional[Callable[..., None]] = None
|
||||
listener: Callable[..., None] | None = None
|
||||
was_scanning = self.scanning
|
||||
was_discovering = self.discovering
|
||||
try:
|
||||
@@ -4398,7 +4415,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
|
||||
async def get_long_term_key(
|
||||
self, connection_handle: int, rand: bytes, ediv: int
|
||||
) -> Optional[bytes]:
|
||||
) -> bytes | None:
|
||||
if (connection := self.lookup_connection(connection_handle)) is None:
|
||||
return None
|
||||
|
||||
@@ -4422,7 +4439,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
return keys.ltk_peripheral.value
|
||||
return None
|
||||
|
||||
async def get_link_key(self, address: hci.Address) -> Optional[bytes]:
|
||||
async def get_link_key(self, address: hci.Address) -> bytes | None:
|
||||
if self.keystore is None:
|
||||
return None
|
||||
|
||||
@@ -4560,7 +4577,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
await connection.cancel_on_disconnection(pending_role_change)
|
||||
|
||||
# [Classic only]
|
||||
async def request_remote_name(self, remote: Union[hci.Address, Connection]) -> str:
|
||||
async def request_remote_name(self, remote: hci.Address | Connection) -> str:
|
||||
# Set up event handlers
|
||||
pending_name: asyncio.Future[str] = asyncio.get_running_loop().create_future()
|
||||
|
||||
@@ -5130,7 +5147,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
self,
|
||||
connection: Connection,
|
||||
attribute: Attribute,
|
||||
value: Optional[Any] = None,
|
||||
value: Any | None = None,
|
||||
force: bool = False,
|
||||
) -> None:
|
||||
"""
|
||||
@@ -5149,7 +5166,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
await self.gatt_server.notify_subscriber(connection, attribute, value, force)
|
||||
|
||||
async def notify_subscribers(
|
||||
self, attribute: Attribute, value=None, force=False
|
||||
self, attribute: Attribute, value: Any | None = None, force: bool = False
|
||||
) -> None:
|
||||
"""
|
||||
Send a notification to all the subscribers of an attribute.
|
||||
@@ -5169,7 +5186,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
self,
|
||||
connection: Connection,
|
||||
attribute: Attribute,
|
||||
value: Optional[Any] = None,
|
||||
value: Any | None = None,
|
||||
force: bool = False,
|
||||
):
|
||||
"""
|
||||
@@ -5190,7 +5207,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
await self.gatt_server.indicate_subscriber(connection, attribute, value, force)
|
||||
|
||||
async def indicate_subscribers(
|
||||
self, attribute: Attribute, value: Optional[Any] = None, force: bool = False
|
||||
self, attribute: Attribute, value: Any | None = None, force: bool = False
|
||||
):
|
||||
"""
|
||||
Send an indication to all the subscribers of an attribute.
|
||||
@@ -5225,8 +5242,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
|
||||
if status != hci.HCI_SUCCESS:
|
||||
logger.debug(
|
||||
f'advertising set {advertising_handle} '
|
||||
f'terminated with status {status}'
|
||||
f'advertising set {advertising_handle} terminated with status {status}'
|
||||
)
|
||||
return
|
||||
|
||||
@@ -5415,8 +5431,8 @@ class Device(utils.CompositeEventEmitter):
|
||||
self,
|
||||
connection_handle: int,
|
||||
peer_address: hci.Address,
|
||||
self_resolvable_address: Optional[hci.Address],
|
||||
peer_resolvable_address: Optional[hci.Address],
|
||||
self_resolvable_address: hci.Address | None,
|
||||
peer_resolvable_address: hci.Address | None,
|
||||
role: hci.Role,
|
||||
connection_interval: int,
|
||||
peripheral_latency: int,
|
||||
@@ -5451,7 +5467,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
peer_address = resolved_address
|
||||
|
||||
self_address = None
|
||||
own_address_type: Optional[hci.OwnAddressType] = None
|
||||
own_address_type: hci.OwnAddressType | None = None
|
||||
if role == hci.Role.CENTRAL:
|
||||
own_address_type = self.connect_own_address_type
|
||||
assert own_address_type is not None
|
||||
@@ -5605,7 +5621,8 @@ class Device(utils.CompositeEventEmitter):
|
||||
|
||||
self.host.send_command_sync(
|
||||
hci.HCI_Accept_Connection_Request_Command(
|
||||
bd_addr=bd_addr, role=0x01 # Remain the peripheral
|
||||
bd_addr=bd_addr,
|
||||
role=0x01, # Remain the peripheral
|
||||
)
|
||||
)
|
||||
|
||||
@@ -5915,7 +5932,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
@host_event_handler
|
||||
@try_with_connection_from_address
|
||||
def on_remote_name(
|
||||
self, connection: Optional[Connection], address: hci.Address, remote_name: bytes
|
||||
self, connection: Connection | None, address: hci.Address, remote_name: bytes
|
||||
):
|
||||
# Try to decode the name
|
||||
try:
|
||||
@@ -5934,7 +5951,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
@host_event_handler
|
||||
@try_with_connection_from_address
|
||||
def on_remote_name_failure(
|
||||
self, connection: Optional[Connection], address: hci.Address, error: int
|
||||
self, connection: Connection | None, address: hci.Address, error: int
|
||||
):
|
||||
if connection:
|
||||
connection.emit(connection.EVENT_REMOTE_NAME_FAILURE, error)
|
||||
@@ -6379,7 +6396,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
@host_event_handler
|
||||
@try_with_connection_from_address
|
||||
def on_role_change_failure(
|
||||
self, connection: Optional[Connection], address: hci.Address, error: int
|
||||
self, connection: Connection | None, address: hci.Address, error: int
|
||||
):
|
||||
if connection:
|
||||
connection.emit(connection.EVENT_ROLE_CHANGE_FAILURE, error)
|
||||
@@ -6403,7 +6420,7 @@ class Device(utils.CompositeEventEmitter):
|
||||
def on_pairing(
|
||||
self,
|
||||
connection: Connection,
|
||||
identity_address: Optional[hci.Address],
|
||||
identity_address: hci.Address | None,
|
||||
keys: PairingKeys,
|
||||
sc: bool,
|
||||
) -> None:
|
||||
|
||||
@@ -24,7 +24,8 @@ from __future__ import annotations
|
||||
import logging
|
||||
import pathlib
|
||||
import platform
|
||||
from typing import TYPE_CHECKING, Iterable, Optional
|
||||
from collections.abc import Iterable
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from bumble.drivers import intel, rtk
|
||||
from bumble.drivers.common import Driver
|
||||
@@ -41,7 +42,7 @@ logger = logging.getLogger(__name__)
|
||||
# -----------------------------------------------------------------------------
|
||||
# Functions
|
||||
# -----------------------------------------------------------------------------
|
||||
async def get_driver_for_host(host: Host) -> Optional[Driver]:
|
||||
async def get_driver_for_host(host: Host) -> Driver | None:
|
||||
"""Probe diver classes until one returns a valid instance for a host, or none is
|
||||
found.
|
||||
If a "driver" HCI metadata entry is present, only that driver class will be probed.
|
||||
|
||||
@@ -29,7 +29,7 @@ import os
|
||||
import pathlib
|
||||
import platform
|
||||
import struct
|
||||
from typing import TYPE_CHECKING, Any, Optional
|
||||
from typing import TYPE_CHECKING, Any
|
||||
|
||||
from bumble import core, hci, utils
|
||||
from bumble.drivers import common
|
||||
@@ -353,8 +353,8 @@ class Driver(common.Driver):
|
||||
self.reset_complete = asyncio.Event()
|
||||
|
||||
# Parse configuration options from the driver name.
|
||||
self.ddc_addon: Optional[bytes] = None
|
||||
self.ddc_override: Optional[bytes] = None
|
||||
self.ddc_addon: bytes | None = None
|
||||
self.ddc_override: bytes | None = None
|
||||
driver = host.hci_metadata.get("driver")
|
||||
if driver is not None and driver.startswith("intel/"):
|
||||
for key, value in [
|
||||
@@ -380,7 +380,7 @@ class Driver(common.Driver):
|
||||
|
||||
if (vendor_id, product_id) not in INTEL_USB_PRODUCTS:
|
||||
logger.debug(
|
||||
f"USB device ({vendor_id:04X}, {product_id:04X}) " "not in known list"
|
||||
f"USB device ({vendor_id:04X}, {product_id:04X}) not in known list"
|
||||
)
|
||||
return False
|
||||
|
||||
@@ -483,9 +483,7 @@ class Driver(common.Driver):
|
||||
raise DriverError("insufficient device info, missing CNVI or CNVR")
|
||||
|
||||
firmware_base_name = (
|
||||
"ibt-"
|
||||
f"{device_info[ValueType.CNVI]:04X}-"
|
||||
f"{device_info[ValueType.CNVR]:04X}"
|
||||
f"ibt-{device_info[ValueType.CNVI]:04X}-{device_info[ValueType.CNVR]:04X}"
|
||||
)
|
||||
logger.debug(f"FW base name: {firmware_base_name}")
|
||||
|
||||
@@ -604,7 +602,7 @@ class Driver(common.Driver):
|
||||
|
||||
await self.load_ddc_if_any(firmware_base_name)
|
||||
|
||||
async def load_ddc_if_any(self, firmware_base_name: Optional[str] = None) -> None:
|
||||
async def load_ddc_if_any(self, firmware_base_name: str | None = None) -> None:
|
||||
"""
|
||||
Check for and load any Device Data Configuration (DDC) blobs.
|
||||
|
||||
|
||||
@@ -484,7 +484,7 @@ class Driver(common.Driver):
|
||||
|
||||
if (vendor_id, product_id) not in RTK_USB_PRODUCTS:
|
||||
logger.debug(
|
||||
f"USB device ({vendor_id:04X}, {product_id:04X}) " "not in known list"
|
||||
f"USB device ({vendor_id:04X}, {product_id:04X}) not in known list"
|
||||
)
|
||||
return False
|
||||
|
||||
|
||||
+7
-7
@@ -28,7 +28,8 @@ import enum
|
||||
import functools
|
||||
import logging
|
||||
import struct
|
||||
from typing import Iterable, Optional, Sequence, TypeVar, Union
|
||||
from collections.abc import Iterable, Sequence
|
||||
from typing import TypeVar
|
||||
|
||||
from bumble.att import Attribute, AttributeValue
|
||||
from bumble.colors import color
|
||||
@@ -227,7 +228,6 @@ GATT_MEDIA_CONTROL_POINT_CHARACTERISTIC = UUID.from_16_bits(0x
|
||||
GATT_MEDIA_CONTROL_POINT_OPCODES_SUPPORTED_CHARACTERISTIC = UUID.from_16_bits(0x2BA5, 'Media Control Point Opcodes Supported')
|
||||
GATT_SEARCH_RESULTS_OBJECT_ID_CHARACTERISTIC = UUID.from_16_bits(0x2BA6, 'Search Results Object ID')
|
||||
GATT_SEARCH_CONTROL_POINT_CHARACTERISTIC = UUID.from_16_bits(0x2BA7, 'Search Control Point')
|
||||
GATT_CONTENT_CONTROL_ID_CHARACTERISTIC = UUID.from_16_bits(0x2BBA, 'Content Control Id')
|
||||
|
||||
# Telephone Bearer Service (TBS)
|
||||
GATT_BEARER_PROVIDER_NAME_CHARACTERISTIC = UUID.from_16_bits(0x2BB3, 'Bearer Provider Name')
|
||||
@@ -356,7 +356,7 @@ class Service(Attribute):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
uuid: Union[str, UUID],
|
||||
uuid: str | UUID,
|
||||
characteristics: Iterable[Characteristic],
|
||||
primary=True,
|
||||
included_services: Iterable[Service] = (),
|
||||
@@ -379,7 +379,7 @@ class Service(Attribute):
|
||||
self.characteristics = list(characteristics)
|
||||
self.primary = primary
|
||||
|
||||
def get_advertising_data(self) -> Optional[bytes]:
|
||||
def get_advertising_data(self) -> bytes | None:
|
||||
"""
|
||||
Get Service specific advertising data
|
||||
Defined by each Service, default value is empty
|
||||
@@ -503,10 +503,10 @@ class Characteristic(Attribute[_T]):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
uuid: Union[str, bytes, UUID],
|
||||
uuid: str | bytes | UUID,
|
||||
properties: Characteristic.Properties,
|
||||
permissions: Union[str, Attribute.Permissions],
|
||||
value: Union[AttributeValue[_T], _T, None] = None,
|
||||
permissions: str | Attribute.Permissions,
|
||||
value: AttributeValue[_T] | _T | None = None,
|
||||
descriptors: Sequence[Descriptor] = (),
|
||||
):
|
||||
super().__init__(uuid, permissions, value)
|
||||
|
||||
@@ -22,7 +22,8 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import struct
|
||||
from typing import Any, Callable, Generic, Iterable, Literal, Optional, TypeVar
|
||||
from collections.abc import Callable, Iterable
|
||||
from typing import Any, Generic, Literal, TypeVar
|
||||
|
||||
from bumble import utils
|
||||
from bumble.core import InvalidOperationError
|
||||
@@ -74,8 +75,8 @@ class DelegatedCharacteristicAdapter(CharacteristicAdapter[_T]):
|
||||
def __init__(
|
||||
self,
|
||||
characteristic: Characteristic,
|
||||
encode: Optional[Callable[[_T], bytes]] = None,
|
||||
decode: Optional[Callable[[bytes], _T]] = None,
|
||||
encode: Callable[[_T], bytes] | None = None,
|
||||
decode: Callable[[bytes], _T] | None = None,
|
||||
):
|
||||
super().__init__(characteristic)
|
||||
self.encode = encode
|
||||
@@ -101,8 +102,8 @@ class DelegatedCharacteristicProxyAdapter(CharacteristicProxyAdapter[_T]):
|
||||
def __init__(
|
||||
self,
|
||||
characteristic_proxy: CharacteristicProxy,
|
||||
encode: Optional[Callable[[_T], bytes]] = None,
|
||||
decode: Optional[Callable[[bytes], _T]] = None,
|
||||
encode: Callable[[_T], bytes] | None = None,
|
||||
decode: Callable[[bytes], _T] | None = None,
|
||||
):
|
||||
super().__init__(characteristic_proxy)
|
||||
self.encode = encode
|
||||
@@ -361,5 +362,4 @@ class EnumCharacteristicProxyAdapter(CharacteristicProxyAdapter[_T3]):
|
||||
|
||||
def decode_value(self, value: bytes) -> _T3:
|
||||
int_value = int.from_bytes(value, self.byteorder)
|
||||
a = self.cls(int_value)
|
||||
return self.cls(int_value)
|
||||
|
||||
+27
-31
@@ -28,16 +28,13 @@ from __future__ import annotations
|
||||
import asyncio
|
||||
import logging
|
||||
import struct
|
||||
from collections.abc import Callable, Iterable
|
||||
from datetime import datetime
|
||||
from typing import (
|
||||
TYPE_CHECKING,
|
||||
Any,
|
||||
Callable,
|
||||
Generic,
|
||||
Iterable,
|
||||
Optional,
|
||||
TypeVar,
|
||||
Union,
|
||||
)
|
||||
|
||||
from bumble import att, core, utils
|
||||
@@ -192,7 +189,7 @@ class CharacteristicProxy(AttributeProxy[_T]):
|
||||
self.descriptors_discovered = False
|
||||
self.subscribers = {} # Map from subscriber to proxy subscriber
|
||||
|
||||
def get_descriptor(self, descriptor_type: UUID) -> Optional[DescriptorProxy]:
|
||||
def get_descriptor(self, descriptor_type: UUID) -> DescriptorProxy | None:
|
||||
for descriptor in self.descriptors:
|
||||
if descriptor.type == descriptor_type:
|
||||
return descriptor
|
||||
@@ -204,7 +201,7 @@ class CharacteristicProxy(AttributeProxy[_T]):
|
||||
|
||||
async def subscribe(
|
||||
self,
|
||||
subscriber: Optional[Callable[[_T], Any]] = None,
|
||||
subscriber: Callable[[_T], Any] | None = None,
|
||||
prefer_notify: bool = True,
|
||||
) -> None:
|
||||
if subscriber is not None:
|
||||
@@ -253,7 +250,7 @@ class ProfileServiceProxy:
|
||||
SERVICE_CLASS: type[TemplateService]
|
||||
|
||||
@classmethod
|
||||
def from_client(cls, client: Client) -> Optional[ProfileServiceProxy]:
|
||||
def from_client(cls, client: Client) -> ProfileServiceProxy | None:
|
||||
return ServiceProxy.from_client(cls, client, cls.SERVICE_CLASS.UUID)
|
||||
|
||||
|
||||
@@ -264,13 +261,11 @@ class Client:
|
||||
services: list[ServiceProxy]
|
||||
cached_values: dict[int, tuple[datetime, bytes]]
|
||||
notification_subscribers: dict[
|
||||
int, set[Union[CharacteristicProxy, Callable[[bytes], Any]]]
|
||||
int, set[CharacteristicProxy | Callable[[bytes], Any]]
|
||||
]
|
||||
indication_subscribers: dict[
|
||||
int, set[Union[CharacteristicProxy, Callable[[bytes], Any]]]
|
||||
]
|
||||
pending_response: Optional[asyncio.futures.Future[att.ATT_PDU]]
|
||||
pending_request: Optional[att.ATT_PDU]
|
||||
indication_subscribers: dict[int, set[CharacteristicProxy | Callable[[bytes], Any]]]
|
||||
pending_response: asyncio.futures.Future[att.ATT_PDU] | None
|
||||
pending_request: att.ATT_PDU | None
|
||||
|
||||
def __init__(self, connection: Connection) -> None:
|
||||
self.connection = connection
|
||||
@@ -360,7 +355,7 @@ class Client:
|
||||
return [service for service in self.services if service.uuid == uuid]
|
||||
|
||||
def get_characteristics_by_uuid(
|
||||
self, uuid: UUID, service: Optional[ServiceProxy] = None
|
||||
self, uuid: UUID, service: ServiceProxy | None = None
|
||||
) -> list[CharacteristicProxy[bytes]]:
|
||||
services = [service] if service else self.services
|
||||
return [
|
||||
@@ -369,13 +364,14 @@ class Client:
|
||||
if c.uuid == uuid
|
||||
]
|
||||
|
||||
def get_attribute_grouping(self, attribute_handle: int) -> Optional[
|
||||
Union[
|
||||
ServiceProxy,
|
||||
tuple[ServiceProxy, CharacteristicProxy],
|
||||
tuple[ServiceProxy, CharacteristicProxy, DescriptorProxy],
|
||||
]
|
||||
]:
|
||||
def get_attribute_grouping(
|
||||
self, attribute_handle: int
|
||||
) -> (
|
||||
ServiceProxy
|
||||
| tuple[ServiceProxy, CharacteristicProxy]
|
||||
| tuple[ServiceProxy, CharacteristicProxy, DescriptorProxy]
|
||||
| None
|
||||
):
|
||||
"""
|
||||
Get the attribute(s) associated with an attribute handle
|
||||
"""
|
||||
@@ -478,7 +474,7 @@ class Client:
|
||||
|
||||
return services
|
||||
|
||||
async def discover_service(self, uuid: Union[str, UUID]) -> list[ServiceProxy]:
|
||||
async def discover_service(self, uuid: str | UUID) -> list[ServiceProxy]:
|
||||
'''
|
||||
See Vol 3, Part G - 4.4.2 Discover Primary Service by Service UUID
|
||||
'''
|
||||
@@ -612,7 +608,7 @@ class Client:
|
||||
return included_services
|
||||
|
||||
async def discover_characteristics(
|
||||
self, uuids, service: Optional[ServiceProxy]
|
||||
self, uuids, service: ServiceProxy | None
|
||||
) -> list[CharacteristicProxy[bytes]]:
|
||||
'''
|
||||
See Vol 3, Part G - 4.6.1 Discover All Characteristics of a Service and 4.6.2
|
||||
@@ -699,9 +695,9 @@ class Client:
|
||||
|
||||
async def discover_descriptors(
|
||||
self,
|
||||
characteristic: Optional[CharacteristicProxy] = None,
|
||||
start_handle: Optional[int] = None,
|
||||
end_handle: Optional[int] = None,
|
||||
characteristic: CharacteristicProxy | None = None,
|
||||
start_handle: int | None = None,
|
||||
end_handle: int | None = None,
|
||||
) -> list[DescriptorProxy]:
|
||||
'''
|
||||
See Vol 3, Part G - 4.7.1 Discover All Characteristic Descriptors
|
||||
@@ -810,7 +806,7 @@ class Client:
|
||||
async def subscribe(
|
||||
self,
|
||||
characteristic: CharacteristicProxy,
|
||||
subscriber: Optional[Callable[[Any], Any]] = None,
|
||||
subscriber: Callable[[Any], Any] | None = None,
|
||||
prefer_notify: bool = True,
|
||||
) -> None:
|
||||
# If we haven't already discovered the descriptors for this characteristic,
|
||||
@@ -860,7 +856,7 @@ class Client:
|
||||
async def unsubscribe(
|
||||
self,
|
||||
characteristic: CharacteristicProxy,
|
||||
subscriber: Optional[Callable[[Any], Any]] = None,
|
||||
subscriber: Callable[[Any], Any] | None = None,
|
||||
force: bool = False,
|
||||
) -> None:
|
||||
'''
|
||||
@@ -925,7 +921,7 @@ class Client:
|
||||
await self.write_value(cccd, b'\x00\x00', with_response=True)
|
||||
|
||||
async def read_value(
|
||||
self, attribute: Union[int, AttributeProxy], no_long_read: bool = False
|
||||
self, attribute: int | AttributeProxy, no_long_read: bool = False
|
||||
) -> bytes:
|
||||
'''
|
||||
See Vol 3, Part G - 4.8.1 Read Characteristic Value
|
||||
@@ -980,7 +976,7 @@ class Client:
|
||||
return attribute_value
|
||||
|
||||
async def read_characteristics_by_uuid(
|
||||
self, uuid: UUID, service: Optional[ServiceProxy]
|
||||
self, uuid: UUID, service: ServiceProxy | None
|
||||
) -> list[bytes]:
|
||||
'''
|
||||
See Vol 3, Part G - 4.8.2 Read Using Characteristic UUID
|
||||
@@ -1038,7 +1034,7 @@ class Client:
|
||||
|
||||
async def write_value(
|
||||
self,
|
||||
attribute: Union[int, AttributeProxy],
|
||||
attribute: int | AttributeProxy,
|
||||
value: bytes,
|
||||
with_response: bool = False,
|
||||
) -> None:
|
||||
|
||||
+13
-12
@@ -29,7 +29,8 @@ import asyncio
|
||||
import logging
|
||||
import struct
|
||||
from collections import defaultdict
|
||||
from typing import TYPE_CHECKING, Iterable, Optional, TypeVar
|
||||
from collections.abc import Iterable
|
||||
from typing import TYPE_CHECKING, TypeVar
|
||||
|
||||
from bumble import att, utils
|
||||
from bumble.colors import color
|
||||
@@ -73,7 +74,7 @@ class Server(utils.EventEmitter):
|
||||
attributes_by_handle: dict[int, att.Attribute]
|
||||
subscribers: dict[int, dict[int, bytes]]
|
||||
indication_semaphores: defaultdict[int, asyncio.Semaphore]
|
||||
pending_confirmations: defaultdict[int, Optional[asyncio.futures.Future]]
|
||||
pending_confirmations: defaultdict[int, asyncio.futures.Future | None]
|
||||
|
||||
EVENT_CHARACTERISTIC_SUBSCRIPTION = "characteristic_subscription"
|
||||
|
||||
@@ -109,7 +110,7 @@ class Server(utils.EventEmitter):
|
||||
and (data := attribute.get_advertising_data())
|
||||
}
|
||||
|
||||
def get_attribute(self, handle: int) -> Optional[att.Attribute]:
|
||||
def get_attribute(self, handle: int) -> att.Attribute | None:
|
||||
attribute = self.attributes_by_handle.get(handle)
|
||||
if attribute:
|
||||
return attribute
|
||||
@@ -126,7 +127,7 @@ class Server(utils.EventEmitter):
|
||||
|
||||
def get_attribute_group(
|
||||
self, handle: int, group_type: type[AttributeGroupType]
|
||||
) -> Optional[AttributeGroupType]:
|
||||
) -> AttributeGroupType | None:
|
||||
return next(
|
||||
(
|
||||
attribute
|
||||
@@ -137,7 +138,7 @@ class Server(utils.EventEmitter):
|
||||
None,
|
||||
)
|
||||
|
||||
def get_service_attribute(self, service_uuid: UUID) -> Optional[Service]:
|
||||
def get_service_attribute(self, service_uuid: UUID) -> Service | None:
|
||||
return next(
|
||||
(
|
||||
attribute
|
||||
@@ -151,7 +152,7 @@ class Server(utils.EventEmitter):
|
||||
|
||||
def get_characteristic_attributes(
|
||||
self, service_uuid: UUID, characteristic_uuid: UUID
|
||||
) -> Optional[tuple[CharacteristicDeclaration, Characteristic]]:
|
||||
) -> tuple[CharacteristicDeclaration, Characteristic] | None:
|
||||
service_handle = self.get_service_attribute(service_uuid)
|
||||
if not service_handle:
|
||||
return None
|
||||
@@ -176,7 +177,7 @@ class Server(utils.EventEmitter):
|
||||
|
||||
def get_descriptor_attribute(
|
||||
self, service_uuid: UUID, characteristic_uuid: UUID, descriptor_uuid: UUID
|
||||
) -> Optional[Descriptor]:
|
||||
) -> Descriptor | None:
|
||||
characteristics = self.get_characteristic_attributes(
|
||||
service_uuid, characteristic_uuid
|
||||
)
|
||||
@@ -334,7 +335,7 @@ class Server(utils.EventEmitter):
|
||||
self,
|
||||
connection: Connection,
|
||||
attribute: att.Attribute,
|
||||
value: Optional[bytes] = None,
|
||||
value: bytes | None = None,
|
||||
force: bool = False,
|
||||
) -> None:
|
||||
# Check if there's a subscriber
|
||||
@@ -377,7 +378,7 @@ class Server(utils.EventEmitter):
|
||||
self,
|
||||
connection: Connection,
|
||||
attribute: att.Attribute,
|
||||
value: Optional[bytes] = None,
|
||||
value: bytes | None = None,
|
||||
force: bool = False,
|
||||
) -> None:
|
||||
# Check if there's a subscriber
|
||||
@@ -437,7 +438,7 @@ class Server(utils.EventEmitter):
|
||||
self,
|
||||
indicate: bool,
|
||||
attribute: att.Attribute,
|
||||
value: Optional[bytes] = None,
|
||||
value: bytes | None = None,
|
||||
force: bool = False,
|
||||
) -> None:
|
||||
# Get all the connections for which there's at least one subscription
|
||||
@@ -464,7 +465,7 @@ class Server(utils.EventEmitter):
|
||||
async def notify_subscribers(
|
||||
self,
|
||||
attribute: att.Attribute,
|
||||
value: Optional[bytes] = None,
|
||||
value: bytes | None = None,
|
||||
force: bool = False,
|
||||
):
|
||||
return await self._notify_or_indicate_subscribers(
|
||||
@@ -474,7 +475,7 @@ class Server(utils.EventEmitter):
|
||||
async def indicate_subscribers(
|
||||
self,
|
||||
attribute: att.Attribute,
|
||||
value: Optional[bytes] = None,
|
||||
value: bytes | None = None,
|
||||
force: bool = False,
|
||||
):
|
||||
return await self._notify_or_indicate_subscribers(True, attribute, value, force)
|
||||
|
||||
+87
-74
@@ -24,17 +24,13 @@ import functools
|
||||
import logging
|
||||
import secrets
|
||||
import struct
|
||||
from collections.abc import Sequence
|
||||
from collections.abc import Callable, Iterable, Sequence
|
||||
from dataclasses import field
|
||||
from typing import (
|
||||
Any,
|
||||
Callable,
|
||||
ClassVar,
|
||||
Iterable,
|
||||
Literal,
|
||||
Optional,
|
||||
TypeVar,
|
||||
Union,
|
||||
cast,
|
||||
)
|
||||
|
||||
@@ -106,7 +102,7 @@ def map_class_of_device(class_of_device):
|
||||
)
|
||||
|
||||
|
||||
def phy_list_to_bits(phys: Optional[Iterable[Phy]]) -> int:
|
||||
def phy_list_to_bits(phys: Iterable[Phy] | None) -> int:
|
||||
if phys is None:
|
||||
return 0
|
||||
|
||||
@@ -119,7 +115,6 @@ def phy_list_to_bits(phys: Optional[Iterable[Phy]]) -> int:
|
||||
|
||||
|
||||
class SpecableEnum(utils.OpenIntEnum):
|
||||
|
||||
@classmethod
|
||||
def type_spec(cls, size: int, byteorder: Literal['little', 'big'] = 'little'):
|
||||
return {
|
||||
@@ -147,7 +142,6 @@ class SpecableEnum(utils.OpenIntEnum):
|
||||
|
||||
|
||||
class SpecableFlag(enum.IntFlag):
|
||||
|
||||
@classmethod
|
||||
def type_spec(cls, size: int, byteorder: Literal['little', 'big'] = 'little'):
|
||||
return {
|
||||
@@ -186,8 +180,8 @@ class SpecableFlag(enum.IntFlag):
|
||||
# - "v" for variable length bytes with a leading length byte
|
||||
# - an integer [1, 4] for 1-byte, 2-byte or 4-byte unsigned little-endian integers
|
||||
# - an integer [-2, -1] for 1-byte, 2-byte signed little-endian integers
|
||||
FieldSpec = Union[dict[str, Any], Callable[[bytes, int], tuple[int, Any]], str, int]
|
||||
Fields = Sequence[Union[tuple[str, FieldSpec], 'Fields']]
|
||||
FieldSpec = dict[str, Any] | Callable[[bytes, int], tuple[int, Any]] | str | int
|
||||
Fields = Sequence['tuple[str, FieldSpec] | Fields']
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
@@ -213,22 +207,44 @@ def metadata(
|
||||
|
||||
HCI_VENDOR_OGF = 0x3F
|
||||
|
||||
# HCI Version
|
||||
HCI_VERSION_BLUETOOTH_CORE_1_0B = 0
|
||||
HCI_VERSION_BLUETOOTH_CORE_1_1 = 1
|
||||
HCI_VERSION_BLUETOOTH_CORE_1_2 = 2
|
||||
HCI_VERSION_BLUETOOTH_CORE_2_0_EDR = 3
|
||||
HCI_VERSION_BLUETOOTH_CORE_2_1_EDR = 4
|
||||
HCI_VERSION_BLUETOOTH_CORE_3_0_HS = 5
|
||||
HCI_VERSION_BLUETOOTH_CORE_4_0 = 6
|
||||
HCI_VERSION_BLUETOOTH_CORE_4_1 = 7
|
||||
HCI_VERSION_BLUETOOTH_CORE_4_2 = 8
|
||||
HCI_VERSION_BLUETOOTH_CORE_5_0 = 9
|
||||
HCI_VERSION_BLUETOOTH_CORE_5_1 = 10
|
||||
HCI_VERSION_BLUETOOTH_CORE_5_2 = 11
|
||||
HCI_VERSION_BLUETOOTH_CORE_5_3 = 12
|
||||
HCI_VERSION_BLUETOOTH_CORE_5_4 = 13
|
||||
HCI_VERSION_BLUETOOTH_CORE_6_0 = 14
|
||||
# Specification Version
|
||||
class SpecificationVersion(utils.OpenIntEnum):
|
||||
BLUETOOTH_CORE_1_0B = 0
|
||||
BLUETOOTH_CORE_1_1 = 1
|
||||
BLUETOOTH_CORE_1_2 = 2
|
||||
BLUETOOTH_CORE_2_0_EDR = 3
|
||||
BLUETOOTH_CORE_2_1_EDR = 4
|
||||
BLUETOOTH_CORE_3_0_HS = 5
|
||||
BLUETOOTH_CORE_4_0 = 6
|
||||
BLUETOOTH_CORE_4_1 = 7
|
||||
BLUETOOTH_CORE_4_2 = 8
|
||||
BLUETOOTH_CORE_5_0 = 9
|
||||
BLUETOOTH_CORE_5_1 = 10
|
||||
BLUETOOTH_CORE_5_2 = 11
|
||||
BLUETOOTH_CORE_5_3 = 12
|
||||
BLUETOOTH_CORE_5_4 = 13
|
||||
BLUETOOTH_CORE_6_0 = 14
|
||||
BLUETOOTH_CORE_6_1 = 15
|
||||
BLUETOOTH_CORE_6_2 = 16
|
||||
|
||||
# For backwards compatibility only
|
||||
HCI_VERSION_BLUETOOTH_CORE_1_0B = SpecificationVersion.BLUETOOTH_CORE_1_0B
|
||||
HCI_VERSION_BLUETOOTH_CORE_1_1 = SpecificationVersion.BLUETOOTH_CORE_1_1
|
||||
HCI_VERSION_BLUETOOTH_CORE_1_2 = SpecificationVersion.BLUETOOTH_CORE_1_2
|
||||
HCI_VERSION_BLUETOOTH_CORE_2_0_EDR = SpecificationVersion.BLUETOOTH_CORE_2_0_EDR
|
||||
HCI_VERSION_BLUETOOTH_CORE_2_1_EDR = SpecificationVersion.BLUETOOTH_CORE_2_1_EDR
|
||||
HCI_VERSION_BLUETOOTH_CORE_3_0_HS = SpecificationVersion.BLUETOOTH_CORE_3_0_HS
|
||||
HCI_VERSION_BLUETOOTH_CORE_4_0 = SpecificationVersion.BLUETOOTH_CORE_4_0
|
||||
HCI_VERSION_BLUETOOTH_CORE_4_1 = SpecificationVersion.BLUETOOTH_CORE_4_1
|
||||
HCI_VERSION_BLUETOOTH_CORE_4_2 = SpecificationVersion.BLUETOOTH_CORE_4_2
|
||||
HCI_VERSION_BLUETOOTH_CORE_5_0 = SpecificationVersion.BLUETOOTH_CORE_5_0
|
||||
HCI_VERSION_BLUETOOTH_CORE_5_1 = SpecificationVersion.BLUETOOTH_CORE_5_1
|
||||
HCI_VERSION_BLUETOOTH_CORE_5_2 = SpecificationVersion.BLUETOOTH_CORE_5_2
|
||||
HCI_VERSION_BLUETOOTH_CORE_5_3 = SpecificationVersion.BLUETOOTH_CORE_5_3
|
||||
HCI_VERSION_BLUETOOTH_CORE_5_4 = SpecificationVersion.BLUETOOTH_CORE_5_4
|
||||
HCI_VERSION_BLUETOOTH_CORE_6_0 = SpecificationVersion.BLUETOOTH_CORE_6_0
|
||||
HCI_VERSION_BLUETOOTH_CORE_6_1 = SpecificationVersion.BLUETOOTH_CORE_6_1
|
||||
HCI_VERSION_BLUETOOTH_CORE_6_2 = SpecificationVersion.BLUETOOTH_CORE_6_2
|
||||
|
||||
HCI_VERSION_NAMES = {
|
||||
HCI_VERSION_BLUETOOTH_CORE_1_0B: 'HCI_VERSION_BLUETOOTH_CORE_1_0B',
|
||||
@@ -246,9 +262,10 @@ HCI_VERSION_NAMES = {
|
||||
HCI_VERSION_BLUETOOTH_CORE_5_3: 'HCI_VERSION_BLUETOOTH_CORE_5_3',
|
||||
HCI_VERSION_BLUETOOTH_CORE_5_4: 'HCI_VERSION_BLUETOOTH_CORE_5_4',
|
||||
HCI_VERSION_BLUETOOTH_CORE_6_0: 'HCI_VERSION_BLUETOOTH_CORE_6_0',
|
||||
HCI_VERSION_BLUETOOTH_CORE_6_1: 'HCI_VERSION_BLUETOOTH_CORE_6_1',
|
||||
HCI_VERSION_BLUETOOTH_CORE_6_2: 'HCI_VERSION_BLUETOOTH_CORE_6_2',
|
||||
}
|
||||
|
||||
# LMP Version
|
||||
LMP_VERSION_NAMES = HCI_VERSION_NAMES
|
||||
|
||||
# HCI Packet types
|
||||
@@ -1786,20 +1803,20 @@ class HCI_Object:
|
||||
|
||||
@classmethod
|
||||
def dict_and_offset_from_bytes(
|
||||
cls, data: bytes, offset: int, fields: Fields
|
||||
cls, data: bytes, offset: int, object_fields: Fields
|
||||
) -> tuple[int, collections.OrderedDict[str, Any]]:
|
||||
result = collections.OrderedDict[str, Any]()
|
||||
for field in fields:
|
||||
if isinstance(field, list):
|
||||
for object_field in object_fields:
|
||||
if isinstance(object_field, list):
|
||||
# This is an array field, starting with a 1-byte item count.
|
||||
item_count = data[offset]
|
||||
offset += 1
|
||||
# Set fields first, because item_count might be 0.
|
||||
for sub_field_name, _ in field:
|
||||
for sub_field_name, _ in object_field:
|
||||
result[sub_field_name] = []
|
||||
|
||||
for _ in range(item_count):
|
||||
for sub_field_name, sub_field_type in field:
|
||||
for sub_field_name, sub_field_type in object_field:
|
||||
value, size = HCI_Object.parse_field(
|
||||
data, offset, sub_field_type
|
||||
)
|
||||
@@ -1807,7 +1824,7 @@ class HCI_Object:
|
||||
offset += size
|
||||
continue
|
||||
|
||||
field_name, field_type = field
|
||||
field_name, field_type = object_field
|
||||
assert isinstance(field_name, str)
|
||||
field_value, field_size = HCI_Object.parse_field(
|
||||
data, offset, cast(FieldSpec, field_type)
|
||||
@@ -1890,26 +1907,26 @@ class HCI_Object:
|
||||
return field_bytes
|
||||
|
||||
@staticmethod
|
||||
def dict_to_bytes(hci_object, fields):
|
||||
def dict_to_bytes(hci_object, object_fields):
|
||||
result = bytearray()
|
||||
for field in fields:
|
||||
if isinstance(field, list):
|
||||
for object_field in object_fields:
|
||||
if isinstance(object_field, list):
|
||||
# The field is an array. The serialized form starts with a 1-byte
|
||||
# item count. We use the length of the first array field as the
|
||||
# array count, since all array fields have the same number of items.
|
||||
item_count = len(hci_object[field[0][0]])
|
||||
item_count = len(hci_object[object_field[0][0]])
|
||||
result += bytes([item_count]) + b''.join(
|
||||
b''.join(
|
||||
HCI_Object.serialize_field(
|
||||
hci_object[sub_field_name][i], sub_field_type
|
||||
)
|
||||
for sub_field_name, sub_field_type in field
|
||||
for sub_field_name, sub_field_type in object_field
|
||||
)
|
||||
for i in range(item_count)
|
||||
)
|
||||
continue
|
||||
|
||||
(field_name, field_type) = field
|
||||
(field_name, field_type) = object_field
|
||||
result += HCI_Object.serialize_field(hci_object[field_name], field_type)
|
||||
|
||||
return bytes(result)
|
||||
@@ -1967,15 +1984,15 @@ class HCI_Object:
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
def format_fields(hci_object, fields, indentation='', value_mappers=None):
|
||||
if not fields:
|
||||
def format_fields(hci_object, object_fields, indentation='', value_mappers=None):
|
||||
if not object_fields:
|
||||
return ''
|
||||
|
||||
# Build array of formatted key:value pairs
|
||||
field_strings = []
|
||||
for field in fields:
|
||||
if isinstance(field, list):
|
||||
for sub_field in field:
|
||||
for object_field in object_fields:
|
||||
if isinstance(object_field, list):
|
||||
for sub_field in object_field:
|
||||
sub_field_name, sub_field_type = sub_field
|
||||
item_count = len(hci_object[sub_field_name])
|
||||
for i in range(item_count):
|
||||
@@ -1993,7 +2010,7 @@ class HCI_Object:
|
||||
)
|
||||
continue
|
||||
|
||||
field_name, field_type = field
|
||||
field_name, field_type = object_field
|
||||
field_value = hci_object[field_name]
|
||||
field_strings.append(
|
||||
(
|
||||
@@ -2016,16 +2033,16 @@ class HCI_Object:
|
||||
@classmethod
|
||||
def fields_from_dataclass(cls, obj: Any) -> list[Any]:
|
||||
stack: list[list[Any]] = [[]]
|
||||
for field in dataclasses.fields(obj):
|
||||
for object_field in dataclasses.fields(obj):
|
||||
# Fields without metadata should be ignored.
|
||||
if not isinstance(
|
||||
(metadata := field.metadata.get("bumble.hci")), FieldMetadata
|
||||
(metadata := object_field.metadata.get("bumble.hci")), FieldMetadata
|
||||
):
|
||||
continue
|
||||
if metadata.list_begin:
|
||||
stack.append([])
|
||||
if metadata.spec:
|
||||
stack[-1].append((field.name, metadata.spec))
|
||||
stack[-1].append((object_field.name, metadata.spec))
|
||||
if metadata.list_end:
|
||||
top = stack.pop()
|
||||
stack[-1].append(top)
|
||||
@@ -2158,7 +2175,7 @@ class Address:
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
address: Union[bytes, str],
|
||||
address: bytes | str,
|
||||
address_type: AddressType = RANDOM_DEVICE_ADDRESS,
|
||||
) -> None:
|
||||
'''
|
||||
@@ -2423,9 +2440,9 @@ class HCI_Command(HCI_Packet):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
parameters: Optional[bytes] = None,
|
||||
parameters: bytes | None = None,
|
||||
*,
|
||||
op_code: Optional[int] = None,
|
||||
op_code: int | None = None,
|
||||
**kwargs,
|
||||
) -> None:
|
||||
# op_code should be set in cls.
|
||||
@@ -5716,7 +5733,7 @@ class HCI_Event(HCI_Packet):
|
||||
hci_packet_type = HCI_EVENT_PACKET
|
||||
event_names: dict[int, str] = {}
|
||||
event_classes: dict[int, type[HCI_Event]] = {}
|
||||
vendor_factories: list[Callable[[bytes], Optional[HCI_Event]]] = []
|
||||
vendor_factories: list[Callable[[bytes], HCI_Event | None]] = []
|
||||
event_code: int
|
||||
fields: Fields = ()
|
||||
_parameters: bytes = b''
|
||||
@@ -5777,14 +5794,12 @@ class HCI_Event(HCI_Packet):
|
||||
return event_class
|
||||
|
||||
@classmethod
|
||||
def add_vendor_factory(
|
||||
cls, factory: Callable[[bytes], Optional[HCI_Event]]
|
||||
) -> None:
|
||||
def add_vendor_factory(cls, factory: Callable[[bytes], HCI_Event | None]) -> None:
|
||||
cls.vendor_factories.append(factory)
|
||||
|
||||
@classmethod
|
||||
def remove_vendor_factory(
|
||||
cls, factory: Callable[[bytes], Optional[HCI_Event]]
|
||||
cls, factory: Callable[[bytes], HCI_Event | None]
|
||||
) -> None:
|
||||
if factory in cls.vendor_factories:
|
||||
cls.vendor_factories.remove(factory)
|
||||
@@ -5797,7 +5812,7 @@ class HCI_Event(HCI_Packet):
|
||||
if len(parameters) != length:
|
||||
raise InvalidPacketError('invalid packet length')
|
||||
|
||||
subclass: Optional[type[HCI_Event]]
|
||||
subclass: type[HCI_Event] | None
|
||||
if event_code == HCI_LE_META_EVENT:
|
||||
# We do this dispatch here and not in the subclass in order to avoid call
|
||||
# loops
|
||||
@@ -5835,9 +5850,9 @@ class HCI_Event(HCI_Packet):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
parameters: Optional[bytes] = None,
|
||||
parameters: bytes | None = None,
|
||||
*,
|
||||
event_code: Optional[int] = None,
|
||||
event_code: int | None = None,
|
||||
**kwargs,
|
||||
):
|
||||
if event_code is not None:
|
||||
@@ -5946,9 +5961,7 @@ class HCI_Extended_Event(HCI_Event):
|
||||
cls.subevent_names.update(cls.subevent_map(symbols))
|
||||
|
||||
@classmethod
|
||||
def subclass_from_parameters(
|
||||
cls, parameters: bytes
|
||||
) -> Optional[HCI_Extended_Event]:
|
||||
def subclass_from_parameters(cls, parameters: bytes) -> HCI_Extended_Event | None:
|
||||
"""
|
||||
Factory method that parses the subevent code, finds a registered subclass,
|
||||
and creates an instance if found.
|
||||
@@ -5968,9 +5981,9 @@ class HCI_Extended_Event(HCI_Event):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
parameters: Optional[bytes] = None,
|
||||
parameters: bytes | None = None,
|
||||
*,
|
||||
subevent_code: Optional[int] = None,
|
||||
subevent_code: int | None = None,
|
||||
**kwargs,
|
||||
) -> None:
|
||||
if subevent_code is not None:
|
||||
@@ -6966,7 +6979,7 @@ class HCI_Command_Complete_Event(HCI_Event):
|
||||
command_opcode: int = field(
|
||||
metadata=metadata({'size': 2, 'mapper': HCI_Command.command_name})
|
||||
)
|
||||
return_parameters: Union[bytes, HCI_Object, int] = field(metadata=metadata("*"))
|
||||
return_parameters: bytes | HCI_Object | int = field(metadata=metadata("*"))
|
||||
|
||||
def map_return_parameters(self, return_parameters):
|
||||
'''Map simple 'status' return parameters to their named constant form'''
|
||||
@@ -7550,20 +7563,20 @@ class HCI_IsoDataPacket(HCI_Packet):
|
||||
iso_sdu_fragment: bytes
|
||||
pb_flag: int
|
||||
ts_flag: int = 0
|
||||
time_stamp: Optional[int] = None
|
||||
packet_sequence_number: Optional[int] = None
|
||||
iso_sdu_length: Optional[int] = None
|
||||
packet_status_flag: Optional[int] = None
|
||||
time_stamp: int | None = None
|
||||
packet_sequence_number: int | None = None
|
||||
iso_sdu_length: int | None = None
|
||||
packet_status_flag: int | None = None
|
||||
|
||||
def __post_init__(self) -> None:
|
||||
self.ts_flag = self.time_stamp is not None
|
||||
|
||||
@staticmethod
|
||||
def from_bytes(packet: bytes) -> HCI_IsoDataPacket:
|
||||
time_stamp: Optional[int] = None
|
||||
packet_sequence_number: Optional[int] = None
|
||||
iso_sdu_length: Optional[int] = None
|
||||
packet_status_flag: Optional[int] = None
|
||||
time_stamp: int | None = None
|
||||
packet_sequence_number: int | None = None
|
||||
iso_sdu_length: int | None = None
|
||||
packet_status_flag: int | None = None
|
||||
|
||||
pos = 1
|
||||
pdu_info, data_total_length = struct.unpack_from('<HH', packet, pos)
|
||||
@@ -7646,7 +7659,7 @@ class HCI_IsoDataPacket(HCI_Packet):
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
class HCI_AclDataPacketAssembler:
|
||||
current_data: Optional[bytes]
|
||||
current_data: bytes | None
|
||||
|
||||
def __init__(self, callback: Callable[[bytes], Any]) -> None:
|
||||
self.callback = callback
|
||||
|
||||
+5
-5
@@ -20,7 +20,7 @@ from __future__ import annotations
|
||||
import datetime
|
||||
import logging
|
||||
from collections.abc import Callable, MutableMapping
|
||||
from typing import Any, Optional, cast
|
||||
from typing import Any, cast
|
||||
|
||||
from bumble import avc, avctp, avdtp, avrcp, crypto, rfcomm, sdp
|
||||
from bumble.att import ATT_CID, ATT_PDU
|
||||
@@ -70,7 +70,7 @@ AVCTP_PID_NAMES = {avrcp.AVRCP_PID: 'AVRCP'}
|
||||
class PacketTracer:
|
||||
class AclStream:
|
||||
psms: MutableMapping[int, int]
|
||||
peer: Optional[PacketTracer.AclStream]
|
||||
peer: PacketTracer.AclStream | None
|
||||
avdtp_assemblers: MutableMapping[int, avdtp.MessageAssembler]
|
||||
avctp_assemblers: MutableMapping[int, avctp.MessageAssembler]
|
||||
|
||||
@@ -201,7 +201,7 @@ class PacketTracer:
|
||||
self.label = label
|
||||
self.emit_message = emit_message
|
||||
self.acl_streams = {} # ACL streams, by connection handle
|
||||
self.packet_timestamp: Optional[datetime.datetime] = None
|
||||
self.packet_timestamp: datetime.datetime | None = None
|
||||
|
||||
def start_acl_stream(self, connection_handle: int) -> PacketTracer.AclStream:
|
||||
logger.info(
|
||||
@@ -230,7 +230,7 @@ class PacketTracer:
|
||||
self.peer.end_acl_stream(connection_handle)
|
||||
|
||||
def on_packet(
|
||||
self, timestamp: Optional[datetime.datetime], packet: HCI_Packet
|
||||
self, timestamp: datetime.datetime | None, packet: HCI_Packet
|
||||
) -> None:
|
||||
self.packet_timestamp = timestamp
|
||||
self.emit(packet)
|
||||
@@ -262,7 +262,7 @@ class PacketTracer:
|
||||
self,
|
||||
packet: HCI_Packet,
|
||||
direction: int = 0,
|
||||
timestamp: Optional[datetime.datetime] = None,
|
||||
timestamp: datetime.datetime | None = None,
|
||||
) -> None:
|
||||
if direction == 0:
|
||||
self.host_to_controller_analyzer.on_packet(timestamp, packet)
|
||||
|
||||
+27
-26
@@ -25,7 +25,8 @@ import enum
|
||||
import logging
|
||||
import re
|
||||
import traceback
|
||||
from typing import TYPE_CHECKING, Any, ClassVar, Iterable, Optional, Union
|
||||
from collections.abc import Iterable
|
||||
from typing import TYPE_CHECKING, Any, ClassVar
|
||||
|
||||
from typing_extensions import Self
|
||||
|
||||
@@ -80,7 +81,7 @@ class HfpProtocol:
|
||||
|
||||
dlc.sink = self.feed
|
||||
|
||||
def feed(self, data: Union[bytes, str]) -> None:
|
||||
def feed(self, data: bytes | str) -> None:
|
||||
# Convert the data to a string if needed
|
||||
if isinstance(data, bytes):
|
||||
data = data.decode('utf-8')
|
||||
@@ -324,8 +325,8 @@ class CallInfo:
|
||||
status: CallInfoStatus
|
||||
mode: CallInfoMode
|
||||
multi_party: CallInfoMultiParty
|
||||
number: Optional[str] = None
|
||||
type: Optional[int] = None
|
||||
number: str | None = None
|
||||
type: int | None = None
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
@@ -353,10 +354,10 @@ class CallLineIdentification:
|
||||
|
||||
number: str
|
||||
type: int
|
||||
subaddr: Optional[str] = None
|
||||
satype: Optional[int] = None
|
||||
alpha: Optional[str] = None
|
||||
cli_validity: Optional[int] = None
|
||||
subaddr: str | None = None
|
||||
satype: int | None = None
|
||||
alpha: str | None = None
|
||||
cli_validity: int | None = None
|
||||
|
||||
@classmethod
|
||||
def parse_from(cls, parameters: list[bytes]) -> Self:
|
||||
@@ -584,7 +585,7 @@ class AgIndicatorState:
|
||||
indicator: AgIndicator
|
||||
supported_values: set[int]
|
||||
current_status: int
|
||||
index: Optional[int] = None
|
||||
index: int | None = None
|
||||
enabled: bool = True
|
||||
|
||||
@property
|
||||
@@ -597,7 +598,7 @@ class AgIndicatorState:
|
||||
supported_values_text = (
|
||||
f'({",".join(str(v) for v in self.supported_values)})'
|
||||
)
|
||||
return f'(\"{self.indicator.value}\",{supported_values_text})'
|
||||
return f'("{self.indicator.value}",{supported_values_text})'
|
||||
|
||||
@classmethod
|
||||
def call(cls: type[Self]) -> Self:
|
||||
@@ -728,7 +729,7 @@ class HfProtocol(utils.EventEmitter):
|
||||
command_lock: asyncio.Lock
|
||||
if TYPE_CHECKING:
|
||||
response_queue: asyncio.Queue[AtResponse]
|
||||
unsolicited_queue: asyncio.Queue[Optional[AtResponse]]
|
||||
unsolicited_queue: asyncio.Queue[AtResponse | None]
|
||||
else:
|
||||
response_queue: asyncio.Queue
|
||||
unsolicited_queue: asyncio.Queue
|
||||
@@ -820,7 +821,7 @@ class HfProtocol(utils.EventEmitter):
|
||||
cmd: str,
|
||||
timeout: float = 1.0,
|
||||
response_type: AtResponseType = AtResponseType.NONE,
|
||||
) -> Union[None, AtResponse, list[AtResponse]]:
|
||||
) -> None | AtResponse | list[AtResponse]:
|
||||
"""
|
||||
Sends an AT command and wait for the peer response.
|
||||
Wait for the AT responses sent by the peer, to the status code.
|
||||
@@ -1351,7 +1352,7 @@ class AgProtocol(utils.EventEmitter):
|
||||
logger.warning(f'AG indicator {indicator} is disabled')
|
||||
|
||||
indicator_state.current_status = value
|
||||
self.send_response(f'+CIEV: {index+1},{value}')
|
||||
self.send_response(f'+CIEV: {index + 1},{value}')
|
||||
|
||||
async def negotiate_codec(self, codec: AudioCodec) -> None:
|
||||
"""Starts codec negotiation."""
|
||||
@@ -1411,13 +1412,13 @@ class AgProtocol(utils.EventEmitter):
|
||||
self.emit(self.EVENT_VOICE_RECOGNITION, VoiceRecognitionState(int(vrec)))
|
||||
|
||||
def _on_chld(self, operation_code: bytes) -> None:
|
||||
call_index: Optional[int] = None
|
||||
call_index: int | None = None
|
||||
if len(operation_code) > 1:
|
||||
call_index = int(operation_code[1:])
|
||||
operation_code = operation_code[:1] + b'x'
|
||||
try:
|
||||
operation = CallHoldOperation(operation_code.decode())
|
||||
except:
|
||||
except Exception:
|
||||
logger.error(f'Invalid operation: {operation_code.decode()}')
|
||||
self.send_cme_error(CmeError.OPERATION_NOT_SUPPORTED)
|
||||
return
|
||||
@@ -1481,8 +1482,8 @@ class AgProtocol(utils.EventEmitter):
|
||||
def _on_cmer(
|
||||
self,
|
||||
mode: bytes,
|
||||
keypad: Optional[bytes] = None,
|
||||
display: Optional[bytes] = None,
|
||||
keypad: bytes | None = None,
|
||||
display: bytes | None = None,
|
||||
indicator: bytes = b'',
|
||||
) -> None:
|
||||
if (
|
||||
@@ -1589,7 +1590,7 @@ class AgProtocol(utils.EventEmitter):
|
||||
|
||||
def _on_clcc(self) -> None:
|
||||
for call in self.calls:
|
||||
number_text = f',\"{call.number}\"' if call.number is not None else ''
|
||||
number_text = f',"{call.number}"' if call.number is not None else ''
|
||||
type_text = f',{call.type}' if call.type is not None else ''
|
||||
response = (
|
||||
f'+CLCC: {call.index}'
|
||||
@@ -1844,7 +1845,7 @@ def make_ag_sdp_records(
|
||||
|
||||
async def find_hf_sdp_record(
|
||||
connection: device.Connection,
|
||||
) -> Optional[tuple[int, ProfileVersion, HfSdpFeature]]:
|
||||
) -> tuple[int, ProfileVersion, HfSdpFeature] | None:
|
||||
"""Searches a Hands-Free SDP record from remote device.
|
||||
|
||||
Args:
|
||||
@@ -1864,9 +1865,9 @@ async def find_hf_sdp_record(
|
||||
],
|
||||
)
|
||||
for attribute_lists in search_result:
|
||||
channel: Optional[int] = None
|
||||
version: Optional[ProfileVersion] = None
|
||||
features: Optional[HfSdpFeature] = None
|
||||
channel: int | None = None
|
||||
version: ProfileVersion | None = None
|
||||
features: HfSdpFeature | None = None
|
||||
for attribute in attribute_lists:
|
||||
# The layout is [[L2CAP_PROTOCOL], [RFCOMM_PROTOCOL, RFCOMM_CHANNEL]].
|
||||
if attribute.id == sdp.SDP_PROTOCOL_DESCRIPTOR_LIST_ATTRIBUTE_ID:
|
||||
@@ -1896,7 +1897,7 @@ async def find_hf_sdp_record(
|
||||
|
||||
async def find_ag_sdp_record(
|
||||
connection: device.Connection,
|
||||
) -> Optional[tuple[int, ProfileVersion, AgSdpFeature]]:
|
||||
) -> tuple[int, ProfileVersion, AgSdpFeature] | None:
|
||||
"""Searches an Audio-Gateway SDP record from remote device.
|
||||
|
||||
Args:
|
||||
@@ -1915,9 +1916,9 @@ async def find_ag_sdp_record(
|
||||
],
|
||||
)
|
||||
for attribute_lists in search_result:
|
||||
channel: Optional[int] = None
|
||||
version: Optional[ProfileVersion] = None
|
||||
features: Optional[AgSdpFeature] = None
|
||||
channel: int | None = None
|
||||
version: ProfileVersion | None = None
|
||||
features: AgSdpFeature | None = None
|
||||
for attribute in attribute_lists:
|
||||
# The layout is [[L2CAP_PROTOCOL], [RFCOMM_PROTOCOL, RFCOMM_CHANNEL]].
|
||||
if attribute.id == sdp.SDP_PROTOCOL_DESCRIPTOR_LIST_ATTRIBUTE_ID:
|
||||
|
||||
+9
-9
@@ -21,8 +21,8 @@ import enum
|
||||
import logging
|
||||
import struct
|
||||
from abc import ABC, abstractmethod
|
||||
from collections.abc import Callable
|
||||
from dataclasses import dataclass
|
||||
from typing import Callable, Optional
|
||||
|
||||
from typing_extensions import override
|
||||
|
||||
@@ -195,9 +195,9 @@ class SendHandshakeMessage(Message):
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
class HID(ABC, utils.EventEmitter):
|
||||
l2cap_ctrl_channel: Optional[l2cap.ClassicChannel] = None
|
||||
l2cap_intr_channel: Optional[l2cap.ClassicChannel] = None
|
||||
connection: Optional[device.Connection] = None
|
||||
l2cap_ctrl_channel: l2cap.ClassicChannel | None = None
|
||||
l2cap_intr_channel: l2cap.ClassicChannel | None = None
|
||||
connection: device.Connection | None = None
|
||||
|
||||
EVENT_INTERRUPT_DATA = "interrupt_data"
|
||||
EVENT_CONTROL_DATA = "control_data"
|
||||
@@ -212,7 +212,7 @@ class HID(ABC, utils.EventEmitter):
|
||||
|
||||
def __init__(self, device: device.Device, role: Role) -> None:
|
||||
super().__init__()
|
||||
self.remote_device_bd_address: Optional[Address] = None
|
||||
self.remote_device_bd_address: Address | None = None
|
||||
self.device = device
|
||||
self.role = role
|
||||
|
||||
@@ -353,10 +353,10 @@ class Device(HID):
|
||||
data: bytes = b''
|
||||
status: int = 0
|
||||
|
||||
get_report_cb: Optional[Callable[[int, int, int], GetSetStatus]] = None
|
||||
set_report_cb: Optional[Callable[[int, int, int, bytes], GetSetStatus]] = None
|
||||
get_protocol_cb: Optional[Callable[[], GetSetStatus]] = None
|
||||
set_protocol_cb: Optional[Callable[[int], GetSetStatus]] = None
|
||||
get_report_cb: Callable[[int, int, int], GetSetStatus] | None = None
|
||||
set_report_cb: Callable[[int, int, int, bytes], GetSetStatus] | None = None
|
||||
get_protocol_cb: Callable[[], GetSetStatus] | None = None
|
||||
set_protocol_cb: Callable[[int], GetSetStatus] | None = None
|
||||
|
||||
def __init__(self, device: device.Device) -> None:
|
||||
super().__init__(device, HID.Role.DEVICE)
|
||||
|
||||
+44
-50
@@ -22,7 +22,8 @@ import collections
|
||||
import dataclasses
|
||||
import logging
|
||||
import struct
|
||||
from typing import TYPE_CHECKING, Any, Awaitable, Callable, Optional, Union, cast
|
||||
from collections.abc import Awaitable, Callable
|
||||
from typing import TYPE_CHECKING, Any, cast
|
||||
|
||||
from bumble import drivers, hci, utils
|
||||
from bumble.colors import color
|
||||
@@ -108,8 +109,7 @@ class DataPacketQueue(utils.EventEmitter):
|
||||
|
||||
if self._packets:
|
||||
logger.debug(
|
||||
f'{self._in_flight} packets in flight, '
|
||||
f'{len(self._packets)} in queue'
|
||||
f'{self._in_flight} packets in flight, {len(self._packets)} in queue'
|
||||
)
|
||||
|
||||
def flush(self, connection_handle: int) -> None:
|
||||
@@ -199,7 +199,7 @@ class Connection:
|
||||
self.peer_address = peer_address
|
||||
self.assembler = hci.HCI_AclDataPacketAssembler(self.on_acl_pdu)
|
||||
self.transport = transport
|
||||
acl_packet_queue: Optional[DataPacketQueue] = (
|
||||
acl_packet_queue: DataPacketQueue | None = (
|
||||
host.le_acl_packet_queue
|
||||
if transport == PhysicalTransport.LE
|
||||
else host.acl_packet_queue
|
||||
@@ -242,20 +242,18 @@ class Host(utils.EventEmitter):
|
||||
bis_links: dict[int, IsoLink]
|
||||
sco_links: dict[int, ScoLink]
|
||||
bigs: dict[int, set[int]]
|
||||
acl_packet_queue: Optional[DataPacketQueue] = None
|
||||
le_acl_packet_queue: Optional[DataPacketQueue] = None
|
||||
iso_packet_queue: Optional[DataPacketQueue] = None
|
||||
hci_sink: Optional[TransportSink] = None
|
||||
acl_packet_queue: DataPacketQueue | None = None
|
||||
le_acl_packet_queue: DataPacketQueue | None = None
|
||||
iso_packet_queue: DataPacketQueue | None = None
|
||||
hci_sink: TransportSink | None = None
|
||||
hci_metadata: dict[str, Any]
|
||||
long_term_key_provider: Optional[
|
||||
Callable[[int, bytes, int], Awaitable[Optional[bytes]]]
|
||||
]
|
||||
link_key_provider: Optional[Callable[[hci.Address], Awaitable[Optional[bytes]]]]
|
||||
long_term_key_provider: Callable[[int, bytes, int], Awaitable[bytes | None]] | None
|
||||
link_key_provider: Callable[[hci.Address], Awaitable[bytes | None]] | None
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
controller_source: Optional[TransportSource] = None,
|
||||
controller_sink: Optional[TransportSink] = None,
|
||||
controller_source: TransportSource | None = None,
|
||||
controller_sink: TransportSink | None = None,
|
||||
) -> None:
|
||||
super().__init__()
|
||||
|
||||
@@ -267,7 +265,7 @@ class Host(utils.EventEmitter):
|
||||
self.sco_links = {} # SCO links, by connection handle
|
||||
self.bigs = {} # BIG Handle to BIS Handles
|
||||
self.pending_command = None
|
||||
self.pending_response: Optional[asyncio.Future[Any]] = None
|
||||
self.pending_response: asyncio.Future[Any] | None = None
|
||||
self.number_of_supported_advertising_sets = 0
|
||||
self.maximum_advertising_data_length = 31
|
||||
self.local_version = None
|
||||
@@ -280,7 +278,7 @@ class Host(utils.EventEmitter):
|
||||
self.long_term_key_provider = None
|
||||
self.link_key_provider = None
|
||||
self.pairing_io_capability_provider = None # Classic only
|
||||
self.snooper: Optional[Snooper] = None
|
||||
self.snooper: Snooper | None = None
|
||||
|
||||
# Connect to the source and sink if specified
|
||||
if controller_source:
|
||||
@@ -291,9 +289,9 @@ class Host(utils.EventEmitter):
|
||||
def find_connection_by_bd_addr(
|
||||
self,
|
||||
bd_addr: hci.Address,
|
||||
transport: Optional[int] = None,
|
||||
transport: int | None = None,
|
||||
check_address_type: bool = False,
|
||||
) -> Optional[Connection]:
|
||||
) -> Connection | None:
|
||||
for connection in self.connections.values():
|
||||
if bytes(connection.peer_address) == bytes(bd_addr):
|
||||
if (
|
||||
@@ -633,7 +631,7 @@ class Host(utils.EventEmitter):
|
||||
)
|
||||
|
||||
@property
|
||||
def controller(self) -> Optional[TransportSink]:
|
||||
def controller(self) -> TransportSink | None:
|
||||
return self.hci_sink
|
||||
|
||||
@controller.setter
|
||||
@@ -642,7 +640,7 @@ class Host(utils.EventEmitter):
|
||||
if controller:
|
||||
self.set_packet_source(controller)
|
||||
|
||||
def set_packet_sink(self, sink: Optional[TransportSink]) -> None:
|
||||
def set_packet_sink(self, sink: TransportSink | None) -> None:
|
||||
self.hci_sink = sink
|
||||
|
||||
def set_packet_source(self, source: TransportSource) -> None:
|
||||
@@ -657,7 +655,7 @@ class Host(utils.EventEmitter):
|
||||
self.hci_sink.on_packet(bytes(packet))
|
||||
|
||||
async def send_command(
|
||||
self, command, check_result=False, response_timeout: Optional[int] = None
|
||||
self, command, check_result=False, response_timeout: int | None = None
|
||||
):
|
||||
# Wait until we can send (only one pending command at a time)
|
||||
async with self.command_semaphore:
|
||||
@@ -707,7 +705,7 @@ class Host(utils.EventEmitter):
|
||||
|
||||
asyncio.create_task(send_command(command))
|
||||
|
||||
def send_l2cap_pdu(self, connection_handle: int, cid: int, pdu: bytes) -> None:
|
||||
def send_acl_sdu(self, connection_handle: int, sdu: bytes) -> None:
|
||||
if not (connection := self.connections.get(connection_handle)):
|
||||
logger.warning(f'connection 0x{connection_handle:04X} not found')
|
||||
return
|
||||
@@ -718,27 +716,24 @@ class Host(utils.EventEmitter):
|
||||
)
|
||||
return
|
||||
|
||||
# Create a PDU
|
||||
l2cap_pdu = bytes(L2CAP_PDU(cid, pdu))
|
||||
|
||||
# Send the data to the controller via ACL packets
|
||||
bytes_remaining = len(l2cap_pdu)
|
||||
offset = 0
|
||||
pb_flag = 0
|
||||
while bytes_remaining:
|
||||
data_total_length = min(bytes_remaining, packet_queue.max_packet_size)
|
||||
max_packet_size = packet_queue.max_packet_size
|
||||
for offset in range(0, len(sdu), max_packet_size):
|
||||
pdu = sdu[offset : offset + max_packet_size]
|
||||
acl_packet = hci.HCI_AclDataPacket(
|
||||
connection_handle=connection_handle,
|
||||
pb_flag=pb_flag,
|
||||
pb_flag=1 if offset > 0 else 0,
|
||||
bc_flag=0,
|
||||
data_total_length=data_total_length,
|
||||
data=l2cap_pdu[offset : offset + data_total_length],
|
||||
data_total_length=len(pdu),
|
||||
data=pdu,
|
||||
)
|
||||
logger.debug(
|
||||
'>>> ACL packet enqueue: (Handle=0x%04X) %s', connection_handle, pdu
|
||||
)
|
||||
logger.debug(f'>>> ACL packet enqueue: (CID={cid}) {acl_packet}')
|
||||
packet_queue.enqueue(acl_packet, connection_handle)
|
||||
pb_flag = 1
|
||||
offset += data_total_length
|
||||
bytes_remaining -= data_total_length
|
||||
|
||||
def send_l2cap_pdu(self, connection_handle: int, cid: int, pdu: bytes) -> None:
|
||||
self.send_acl_sdu(connection_handle, bytes(L2CAP_PDU(cid, pdu)))
|
||||
|
||||
def get_data_packet_queue(self, connection_handle: int) -> DataPacketQueue | None:
|
||||
if connection := self.connections.get(connection_handle):
|
||||
@@ -903,7 +898,7 @@ class Host(utils.EventEmitter):
|
||||
self.emit('l2cap_pdu', connection.handle, cid, pdu)
|
||||
|
||||
def on_command_processed(
|
||||
self, event: Union[hci.HCI_Command_Complete_Event, hci.HCI_Command_Status_Event]
|
||||
self, event: hci.HCI_Command_Complete_Event | hci.HCI_Command_Status_Event
|
||||
):
|
||||
if self.pending_response:
|
||||
# Check that it is what we were expecting
|
||||
@@ -966,11 +961,11 @@ class Host(utils.EventEmitter):
|
||||
|
||||
def on_hci_le_connection_complete_event(
|
||||
self,
|
||||
event: Union[
|
||||
hci.HCI_LE_Connection_Complete_Event,
|
||||
hci.HCI_LE_Enhanced_Connection_Complete_Event,
|
||||
hci.HCI_LE_Enhanced_Connection_Complete_V2_Event,
|
||||
],
|
||||
event: (
|
||||
hci.HCI_LE_Connection_Complete_Event
|
||||
| hci.HCI_LE_Enhanced_Connection_Complete_Event
|
||||
| hci.HCI_LE_Enhanced_Connection_Complete_V2_Event
|
||||
),
|
||||
):
|
||||
# Check if this is a cancellation
|
||||
if event.status == hci.HCI_SUCCESS:
|
||||
@@ -1015,10 +1010,10 @@ class Host(utils.EventEmitter):
|
||||
|
||||
def on_hci_le_enhanced_connection_complete_event(
|
||||
self,
|
||||
event: Union[
|
||||
hci.HCI_LE_Enhanced_Connection_Complete_Event,
|
||||
hci.HCI_LE_Enhanced_Connection_Complete_V2_Event,
|
||||
],
|
||||
event: (
|
||||
hci.HCI_LE_Enhanced_Connection_Complete_Event
|
||||
| hci.HCI_LE_Enhanced_Connection_Complete_V2_Event
|
||||
),
|
||||
):
|
||||
# Just use the same implementation as for the non-enhanced event for now
|
||||
self.on_hci_le_connection_complete_event(event)
|
||||
@@ -1397,8 +1392,7 @@ class Host(utils.EventEmitter):
|
||||
if event.status == hci.HCI_SUCCESS:
|
||||
# Create/update the connection
|
||||
logger.debug(
|
||||
f'### SCO CONNECTION: [0x{event.connection_handle:04X}] '
|
||||
f'{event.bd_addr}'
|
||||
f'### SCO CONNECTION: [0x{event.connection_handle:04X}] {event.bd_addr}'
|
||||
)
|
||||
|
||||
self.sco_links[event.connection_handle] = ScoLink(
|
||||
@@ -1450,7 +1444,7 @@ class Host(utils.EventEmitter):
|
||||
def on_hci_le_data_length_change_event(
|
||||
self, event: hci.HCI_LE_Data_Length_Change_Event
|
||||
):
|
||||
if (connection := self.connections.get(event.connection_handle)) is None:
|
||||
if event.connection_handle not in self.connections:
|
||||
logger.warning('!!! DATA LENGTH CHANGE: unknown handle')
|
||||
return
|
||||
|
||||
|
||||
+17
-17
@@ -27,7 +27,7 @@ import dataclasses
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
from typing import TYPE_CHECKING, Any, Optional
|
||||
from typing import TYPE_CHECKING, Any
|
||||
|
||||
from typing_extensions import Self
|
||||
|
||||
@@ -51,8 +51,8 @@ class PairingKeys:
|
||||
class Key:
|
||||
value: bytes
|
||||
authenticated: bool = False
|
||||
ediv: Optional[int] = None
|
||||
rand: Optional[bytes] = None
|
||||
ediv: int | None = None
|
||||
rand: bytes | None = None
|
||||
|
||||
@classmethod
|
||||
def from_dict(cls, key_dict: dict[str, Any]) -> PairingKeys.Key:
|
||||
@@ -74,17 +74,17 @@ class PairingKeys:
|
||||
|
||||
return key_dict
|
||||
|
||||
address_type: Optional[hci.AddressType] = None
|
||||
ltk: Optional[Key] = None
|
||||
ltk_central: Optional[Key] = None
|
||||
ltk_peripheral: Optional[Key] = None
|
||||
irk: Optional[Key] = None
|
||||
csrk: Optional[Key] = None
|
||||
link_key: Optional[Key] = None # Classic
|
||||
link_key_type: Optional[int] = None # Classic
|
||||
address_type: hci.AddressType | None = None
|
||||
ltk: Key | None = None
|
||||
ltk_central: Key | None = None
|
||||
ltk_peripheral: Key | None = None
|
||||
irk: Key | None = None
|
||||
csrk: Key | None = None
|
||||
link_key: Key | None = None # Classic
|
||||
link_key_type: int | None = None # Classic
|
||||
|
||||
@classmethod
|
||||
def key_from_dict(cls, keys_dict: dict[str, Any], key_name: str) -> Optional[Key]:
|
||||
def key_from_dict(cls, keys_dict: dict[str, Any], key_name: str) -> Key | None:
|
||||
key_dict = keys_dict.get(key_name)
|
||||
if key_dict is None:
|
||||
return None
|
||||
@@ -156,7 +156,7 @@ class KeyStore:
|
||||
async def update(self, name: str, keys: PairingKeys) -> None:
|
||||
pass
|
||||
|
||||
async def get(self, _name: str) -> Optional[PairingKeys]:
|
||||
async def get(self, _name: str) -> PairingKeys | None:
|
||||
return None
|
||||
|
||||
async def get_all(self) -> list[tuple[str, PairingKeys]]:
|
||||
@@ -274,7 +274,7 @@ class JsonKeyStore(KeyStore):
|
||||
|
||||
@classmethod
|
||||
def from_device(
|
||||
cls: type[Self], device: Device, filename: Optional[str] = None
|
||||
cls: type[Self], device: Device, filename: str | None = None
|
||||
) -> Self:
|
||||
if not filename:
|
||||
# Extract the filename from the config if there is one
|
||||
@@ -297,7 +297,7 @@ class JsonKeyStore(KeyStore):
|
||||
# Try to open the file, without failing. If the file does not exist, it
|
||||
# will be created upon saving.
|
||||
try:
|
||||
with open(self.filename, 'r', encoding='utf-8') as json_file:
|
||||
with open(self.filename, encoding='utf-8') as json_file:
|
||||
db = json.load(json_file)
|
||||
except FileNotFoundError:
|
||||
db = {}
|
||||
@@ -348,7 +348,7 @@ class JsonKeyStore(KeyStore):
|
||||
key_map.clear()
|
||||
await self.save(db)
|
||||
|
||||
async def get(self, name: str) -> Optional[PairingKeys]:
|
||||
async def get(self, name: str) -> PairingKeys | None:
|
||||
_, key_map = await self.load()
|
||||
if name not in key_map:
|
||||
return None
|
||||
@@ -370,7 +370,7 @@ class MemoryKeyStore(KeyStore):
|
||||
async def update(self, name: str, keys: PairingKeys) -> None:
|
||||
self.all_keys[name] = keys
|
||||
|
||||
async def get(self, name: str) -> Optional[PairingKeys]:
|
||||
async def get(self, name: str) -> PairingKeys | None:
|
||||
return self.all_keys.get(name)
|
||||
|
||||
async def get_all(self) -> list[tuple[str, PairingKeys]]:
|
||||
|
||||
+1016
-225
File diff suppressed because it is too large
Load Diff
+30
-163
@@ -19,9 +19,12 @@ import asyncio
|
||||
# Imports
|
||||
# -----------------------------------------------------------------------------
|
||||
import logging
|
||||
from typing import Optional
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from bumble import controller, core, hci, lmp
|
||||
from bumble import core, hci, ll, lmp
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from bumble import controller
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Logging
|
||||
@@ -29,11 +32,6 @@ from bumble import controller, core, hci, lmp
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Utils
|
||||
# -----------------------------------------------------------------------------
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# TODO: add more support for various LL exchanges
|
||||
# (see Vol 6, Part B - 2.4 DATA CHANNEL PDU)
|
||||
@@ -47,7 +45,6 @@ class LocalLink:
|
||||
|
||||
def __init__(self):
|
||||
self.controllers = set()
|
||||
self.pending_connection = None
|
||||
self.pending_classic_connection = None
|
||||
|
||||
############################################################
|
||||
@@ -61,23 +58,21 @@ class LocalLink:
|
||||
def remove_controller(self, controller: controller.Controller):
|
||||
self.controllers.remove(controller)
|
||||
|
||||
def find_controller(self, address: hci.Address) -> controller.Controller | None:
|
||||
def find_le_controller(self, address: hci.Address) -> controller.Controller | None:
|
||||
for controller in self.controllers:
|
||||
if controller.random_address == address:
|
||||
return controller
|
||||
for connection in controller.le_connections.values():
|
||||
if connection.self_address == address:
|
||||
return controller
|
||||
return None
|
||||
|
||||
def find_classic_controller(
|
||||
self, address: hci.Address
|
||||
) -> Optional[controller.Controller]:
|
||||
) -> controller.Controller | None:
|
||||
for controller in self.controllers:
|
||||
if controller.public_address == address:
|
||||
return controller
|
||||
return None
|
||||
|
||||
def get_pending_connection(self):
|
||||
return self.pending_connection
|
||||
|
||||
############################################################
|
||||
# LE handlers
|
||||
############################################################
|
||||
@@ -85,12 +80,6 @@ class LocalLink:
|
||||
def on_address_changed(self, controller):
|
||||
pass
|
||||
|
||||
def send_advertising_data(self, sender_address: hci.Address, data: bytes):
|
||||
# Send the advertising data to all controllers, except the sender
|
||||
for controller in self.controllers:
|
||||
if controller.random_address != sender_address:
|
||||
controller.on_link_advertising_data(sender_address, data)
|
||||
|
||||
def send_acl_data(
|
||||
self,
|
||||
sender_controller: controller.Controller,
|
||||
@@ -100,7 +89,7 @@ class LocalLink:
|
||||
):
|
||||
# Send the data to the first controller with a matching address
|
||||
if transport == core.PhysicalTransport.LE:
|
||||
destination_controller = self.find_controller(destination_address)
|
||||
destination_controller = self.find_le_controller(destination_address)
|
||||
source_address = sender_controller.random_address
|
||||
elif transport == core.PhysicalTransport.BR_EDR:
|
||||
destination_controller = self.find_classic_controller(destination_address)
|
||||
@@ -115,152 +104,30 @@ class LocalLink:
|
||||
)
|
||||
)
|
||||
|
||||
def on_connection_complete(self) -> None:
|
||||
# Check that we expect this call
|
||||
if not self.pending_connection:
|
||||
logger.warning('on_connection_complete with no pending connection')
|
||||
return
|
||||
def send_advertising_pdu(
|
||||
self,
|
||||
sender_controller: controller.Controller,
|
||||
packet: ll.AdvertisingPdu,
|
||||
):
|
||||
loop = asyncio.get_running_loop()
|
||||
for c in self.controllers:
|
||||
if c != sender_controller:
|
||||
loop.call_soon(c.on_ll_advertising_pdu, packet)
|
||||
|
||||
central_address, le_create_connection_command = self.pending_connection
|
||||
self.pending_connection = None
|
||||
|
||||
# Find the controller that initiated the connection
|
||||
if not (central_controller := self.find_controller(central_address)):
|
||||
logger.warning('!!! Initiating controller not found')
|
||||
return
|
||||
|
||||
# Connect to the first controller with a matching address
|
||||
if peripheral_controller := self.find_controller(
|
||||
le_create_connection_command.peer_address
|
||||
):
|
||||
central_controller.on_link_peripheral_connection_complete(
|
||||
le_create_connection_command, hci.HCI_SUCCESS
|
||||
def send_ll_control_pdu(
|
||||
self,
|
||||
sender_address: hci.Address,
|
||||
receiver_address: hci.Address,
|
||||
packet: ll.ControlPdu,
|
||||
):
|
||||
if not (receiver_controller := self.find_le_controller(receiver_address)):
|
||||
raise core.InvalidArgumentError(
|
||||
f"Unable to find controller for address {receiver_address}"
|
||||
)
|
||||
peripheral_controller.on_link_central_connected(central_address)
|
||||
return
|
||||
|
||||
# No peripheral found
|
||||
central_controller.on_link_peripheral_connection_complete(
|
||||
le_create_connection_command, hci.HCI_CONNECTION_ACCEPT_TIMEOUT_ERROR
|
||||
)
|
||||
|
||||
def connect(
|
||||
self,
|
||||
central_address: hci.Address,
|
||||
le_create_connection_command: hci.HCI_LE_Create_Connection_Command,
|
||||
):
|
||||
logger.debug(
|
||||
f'$$$ CONNECTION {central_address} -> '
|
||||
f'{le_create_connection_command.peer_address}'
|
||||
)
|
||||
self.pending_connection = (central_address, le_create_connection_command)
|
||||
asyncio.get_running_loop().call_soon(self.on_connection_complete)
|
||||
|
||||
def on_disconnection_complete(
|
||||
self,
|
||||
initiating_address: hci.Address,
|
||||
target_address: hci.Address,
|
||||
disconnect_command: hci.HCI_Disconnect_Command,
|
||||
):
|
||||
# Find the controller that initiated the disconnection
|
||||
if not (initiating_controller := self.find_controller(initiating_address)):
|
||||
logger.warning('!!! Initiating controller not found')
|
||||
return
|
||||
|
||||
# Disconnect from the first controller with a matching address
|
||||
if target_controller := self.find_controller(target_address):
|
||||
target_controller.on_link_disconnected(
|
||||
initiating_address, disconnect_command.reason
|
||||
)
|
||||
|
||||
initiating_controller.on_link_disconnection_complete(
|
||||
disconnect_command, hci.HCI_SUCCESS
|
||||
)
|
||||
|
||||
def disconnect(
|
||||
self,
|
||||
initiating_address: hci.Address,
|
||||
target_address: hci.Address,
|
||||
disconnect_command: hci.HCI_Disconnect_Command,
|
||||
):
|
||||
logger.debug(
|
||||
f'$$$ DISCONNECTION {initiating_address} -> '
|
||||
f'{target_address}: reason = {disconnect_command.reason}'
|
||||
)
|
||||
asyncio.get_running_loop().call_soon(
|
||||
lambda: self.on_disconnection_complete(
|
||||
initiating_address, target_address, disconnect_command
|
||||
)
|
||||
lambda: receiver_controller.on_ll_control_pdu(sender_address, packet)
|
||||
)
|
||||
|
||||
def on_connection_encrypted(
|
||||
self,
|
||||
central_address: hci.Address,
|
||||
peripheral_address: hci.Address,
|
||||
rand: bytes,
|
||||
ediv: int,
|
||||
ltk: bytes,
|
||||
):
|
||||
logger.debug(f'*** ENCRYPTION {central_address} -> {peripheral_address}')
|
||||
|
||||
if central_controller := self.find_controller(central_address):
|
||||
central_controller.on_link_encrypted(peripheral_address, rand, ediv, ltk)
|
||||
|
||||
if peripheral_controller := self.find_controller(peripheral_address):
|
||||
peripheral_controller.on_link_encrypted(central_address, rand, ediv, ltk)
|
||||
|
||||
def create_cis(
|
||||
self,
|
||||
central_controller: controller.Controller,
|
||||
peripheral_address: hci.Address,
|
||||
cig_id: int,
|
||||
cis_id: int,
|
||||
) -> None:
|
||||
logger.debug(
|
||||
f'$$$ CIS Request {central_controller.random_address} -> {peripheral_address}'
|
||||
)
|
||||
if peripheral_controller := self.find_controller(peripheral_address):
|
||||
asyncio.get_running_loop().call_soon(
|
||||
peripheral_controller.on_link_cis_request,
|
||||
central_controller.random_address,
|
||||
cig_id,
|
||||
cis_id,
|
||||
)
|
||||
|
||||
def accept_cis(
|
||||
self,
|
||||
peripheral_controller: controller.Controller,
|
||||
central_address: hci.Address,
|
||||
cig_id: int,
|
||||
cis_id: int,
|
||||
) -> None:
|
||||
logger.debug(
|
||||
f'$$$ CIS Accept {peripheral_controller.random_address} -> {central_address}'
|
||||
)
|
||||
if central_controller := self.find_controller(central_address):
|
||||
loop = asyncio.get_running_loop()
|
||||
loop.call_soon(central_controller.on_link_cis_established, cig_id, cis_id)
|
||||
loop.call_soon(
|
||||
peripheral_controller.on_link_cis_established, cig_id, cis_id
|
||||
)
|
||||
|
||||
def disconnect_cis(
|
||||
self,
|
||||
initiator_controller: controller.Controller,
|
||||
peer_address: hci.Address,
|
||||
cig_id: int,
|
||||
cis_id: int,
|
||||
) -> None:
|
||||
logger.debug(
|
||||
f'$$$ CIS Disconnect {initiator_controller.random_address} -> {peer_address}'
|
||||
)
|
||||
if peer_controller := self.find_controller(peer_address):
|
||||
loop = asyncio.get_running_loop()
|
||||
loop.call_soon(
|
||||
initiator_controller.on_link_cis_disconnected, cig_id, cis_id
|
||||
)
|
||||
loop.call_soon(peer_controller.on_link_cis_disconnected, cig_id, cis_id)
|
||||
|
||||
############################################################
|
||||
# Classic handlers
|
||||
############################################################
|
||||
|
||||
+200
@@ -0,0 +1,200 @@
|
||||
# Copyright 2021-2025 Google LLC
|
||||
#
|
||||
# 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
|
||||
#
|
||||
# https://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.
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Imports
|
||||
# -----------------------------------------------------------------------------
|
||||
from __future__ import annotations
|
||||
|
||||
import dataclasses
|
||||
from typing import ClassVar
|
||||
|
||||
from bumble import hci
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Advertising PDU
|
||||
# -----------------------------------------------------------------------------
|
||||
class AdvertisingPdu:
|
||||
"""Base Advertising Physical Channel PDU class.
|
||||
|
||||
See Core Spec 6.0, Volume 6, Part B, 2.3. Advertising physical channel PDU.
|
||||
|
||||
Currently these messages don't really follow the LL spec, because LL protocol is
|
||||
context-aware and we don't have real physical transport.
|
||||
"""
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class ConnectInd(AdvertisingPdu):
|
||||
initiator_address: hci.Address
|
||||
advertiser_address: hci.Address
|
||||
interval: int
|
||||
latency: int
|
||||
timeout: int
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class AdvInd(AdvertisingPdu):
|
||||
advertiser_address: hci.Address
|
||||
data: bytes
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class AdvDirectInd(AdvertisingPdu):
|
||||
advertiser_address: hci.Address
|
||||
target_address: hci.Address
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class AdvNonConnInd(AdvertisingPdu):
|
||||
advertiser_address: hci.Address
|
||||
data: bytes
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class AdvExtInd(AdvertisingPdu):
|
||||
advertiser_address: hci.Address
|
||||
data: bytes
|
||||
|
||||
target_address: hci.Address | None = None
|
||||
adi: int | None = None
|
||||
tx_power: int | None = None
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# LL Control PDU
|
||||
# -----------------------------------------------------------------------------
|
||||
class ControlPdu:
|
||||
"""Base LL Control PDU Class.
|
||||
|
||||
See Core Spec 6.0, Volume 6, Part B, 2.4.2. LL Control PDU.
|
||||
|
||||
Currently these messages don't really follow the LL spec, because LL protocol is
|
||||
context-aware and we don't have real physical transport.
|
||||
"""
|
||||
|
||||
class Opcode(hci.SpecableEnum):
|
||||
LL_CONNECTION_UPDATE_IND = 0x00
|
||||
LL_CHANNEL_MAP_IND = 0x01
|
||||
LL_TERMINATE_IND = 0x02
|
||||
LL_ENC_REQ = 0x03
|
||||
LL_ENC_RSP = 0x04
|
||||
LL_START_ENC_REQ = 0x05
|
||||
LL_START_ENC_RSP = 0x06
|
||||
LL_UNKNOWN_RSP = 0x07
|
||||
LL_FEATURE_REQ = 0x08
|
||||
LL_FEATURE_RSP = 0x09
|
||||
LL_PAUSE_ENC_REQ = 0x0A
|
||||
LL_PAUSE_ENC_RSP = 0x0B
|
||||
LL_VERSION_IND = 0x0C
|
||||
LL_REJECT_IND = 0x0D
|
||||
LL_PERIPHERAL_FEATURE_REQ = 0x0E
|
||||
LL_CONNECTION_PARAM_REQ = 0x0F
|
||||
LL_CONNECTION_PARAM_RSP = 0x10
|
||||
LL_REJECT_EXT_IND = 0x11
|
||||
LL_PING_REQ = 0x12
|
||||
LL_PING_RSP = 0x13
|
||||
LL_LENGTH_REQ = 0x14
|
||||
LL_LENGTH_RSP = 0x15
|
||||
LL_PHY_REQ = 0x16
|
||||
LL_PHY_RSP = 0x17
|
||||
LL_PHY_UPDATE_IND = 0x18
|
||||
LL_MIN_USED_CHANNELS_IND = 0x19
|
||||
LL_CTE_REQ = 0x1A
|
||||
LL_CTE_RSP = 0x1B
|
||||
LL_PERIODIC_SYNC_IND = 0x1C
|
||||
LL_CLOCK_ACCURACY_REQ = 0x1D
|
||||
LL_CLOCK_ACCURACY_RSP = 0x1E
|
||||
LL_CIS_REQ = 0x1F
|
||||
LL_CIS_RSP = 0x20
|
||||
LL_CIS_IND = 0x21
|
||||
LL_CIS_TERMINATE_IND = 0x22
|
||||
LL_POWER_CONTROL_REQ = 0x23
|
||||
LL_POWER_CONTROL_RSP = 0x24
|
||||
LL_POWER_CHANGE_IND = 0x25
|
||||
LL_SUBRATE_REQ = 0x26
|
||||
LL_SUBRATE_IND = 0x27
|
||||
LL_CHANNEL_REPORTING_IND = 0x28
|
||||
LL_CHANNEL_STATUS_IND = 0x29
|
||||
LL_PERIODIC_SYNC_WR_IND = 0x2A
|
||||
LL_FEATURE_EXT_REQ = 0x2B
|
||||
LL_FEATURE_EXT_RSP = 0x2C
|
||||
LL_CS_SEC_RSP = 0x2D
|
||||
LL_CS_CAPABILITIES_REQ = 0x2E
|
||||
LL_CS_CAPABILITIES_RSP = 0x2F
|
||||
LL_CS_CONFIG_REQ = 0x30
|
||||
LL_CS_CONFIG_RSP = 0x31
|
||||
LL_CS_REQ = 0x32
|
||||
LL_CS_RSP = 0x33
|
||||
LL_CS_IND = 0x34
|
||||
LL_CS_TERMINATE_REQ = 0x35
|
||||
LL_CS_FAE_REQ = 0x36
|
||||
LL_CS_FAE_RSP = 0x37
|
||||
LL_CS_CHANNEL_MAP_IND = 0x38
|
||||
LL_CS_SEC_REQ = 0x39
|
||||
LL_CS_TERMINATE_RSP = 0x3A
|
||||
LL_FRAME_SPACE_REQ = 0x3B
|
||||
LL_FRAME_SPACE_RSP = 0x3C
|
||||
|
||||
opcode: ClassVar[Opcode]
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class TerminateInd(ControlPdu):
|
||||
opcode = ControlPdu.Opcode.LL_TERMINATE_IND
|
||||
|
||||
error_code: int
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class EncReq(ControlPdu):
|
||||
opcode = ControlPdu.Opcode.LL_ENC_REQ
|
||||
|
||||
rand: bytes
|
||||
ediv: int
|
||||
ltk: bytes
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class CisReq(ControlPdu):
|
||||
opcode = ControlPdu.Opcode.LL_CIS_REQ
|
||||
|
||||
cig_id: int
|
||||
cis_id: int
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class CisRsp(ControlPdu):
|
||||
opcode = ControlPdu.Opcode.LL_CIS_REQ
|
||||
|
||||
cig_id: int
|
||||
cis_id: int
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class CisInd(ControlPdu):
|
||||
opcode = ControlPdu.Opcode.LL_CIS_REQ
|
||||
|
||||
cig_id: int
|
||||
cis_id: int
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class CisTerminateInd(ControlPdu):
|
||||
opcode = ControlPdu.Opcode.LL_CIS_TERMINATE_IND
|
||||
|
||||
cig_id: int
|
||||
cis_id: int
|
||||
error_code: int
|
||||
+14
-15
@@ -20,7 +20,6 @@ from __future__ import annotations
|
||||
import enum
|
||||
import secrets
|
||||
from dataclasses import dataclass
|
||||
from typing import Optional
|
||||
|
||||
from bumble import hci
|
||||
from bumble.core import AdvertisingData, LeRole
|
||||
@@ -45,16 +44,16 @@ from bumble.smp import (
|
||||
class OobData:
|
||||
"""OOB data that can be sent from one device to another."""
|
||||
|
||||
address: Optional[hci.Address] = None
|
||||
role: Optional[LeRole] = None
|
||||
shared_data: Optional[OobSharedData] = None
|
||||
legacy_context: Optional[OobLegacyContext] = None
|
||||
address: hci.Address | None = None
|
||||
role: LeRole | None = None
|
||||
shared_data: OobSharedData | None = None
|
||||
legacy_context: OobLegacyContext | None = None
|
||||
|
||||
@classmethod
|
||||
def from_ad(cls, ad: AdvertisingData) -> OobData:
|
||||
instance = cls()
|
||||
shared_data_c: Optional[bytes] = None
|
||||
shared_data_r: Optional[bytes] = None
|
||||
shared_data_c: bytes | None = None
|
||||
shared_data_r: bytes | None = None
|
||||
for ad_type, ad_data in ad.ad_structures:
|
||||
if ad_type == AdvertisingData.LE_BLUETOOTH_DEVICE_ADDRESS:
|
||||
instance.address = hci.Address(ad_data)
|
||||
@@ -181,14 +180,14 @@ class PairingDelegate:
|
||||
"""Compare two numbers."""
|
||||
return True
|
||||
|
||||
async def get_number(self) -> Optional[int]:
|
||||
async def get_number(self) -> int | None:
|
||||
"""
|
||||
Return an optional number as an answer to a passkey request.
|
||||
Returning `None` will result in a negative reply.
|
||||
"""
|
||||
return 0
|
||||
|
||||
async def get_string(self, max_length: int) -> Optional[str]:
|
||||
async def get_string(self, max_length: int) -> str | None:
|
||||
"""
|
||||
Return a string whose utf-8 encoding is up to max_length bytes.
|
||||
"""
|
||||
@@ -239,18 +238,18 @@ class PairingConfig:
|
||||
class OobConfig:
|
||||
"""Config for OOB pairing."""
|
||||
|
||||
our_context: Optional[OobContext]
|
||||
peer_data: Optional[OobSharedData]
|
||||
legacy_context: Optional[OobLegacyContext]
|
||||
our_context: OobContext | None
|
||||
peer_data: OobSharedData | None
|
||||
legacy_context: OobLegacyContext | None
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
sc: bool = True,
|
||||
mitm: bool = True,
|
||||
bonding: bool = True,
|
||||
delegate: Optional[PairingDelegate] = None,
|
||||
identity_address_type: Optional[AddressType] = None,
|
||||
oob: Optional[OobConfig] = None,
|
||||
delegate: PairingDelegate | None = None,
|
||||
identity_address_type: AddressType | None = None,
|
||||
oob: OobConfig | None = None,
|
||||
) -> None:
|
||||
self.sc = sc
|
||||
self.mitm = mitm
|
||||
|
||||
@@ -19,7 +19,7 @@ This module implement the Pandora Bluetooth test APIs for the Bumble stack.
|
||||
|
||||
__version__ = "0.0.1"
|
||||
|
||||
from typing import Callable, List, Optional
|
||||
from collections.abc import Callable
|
||||
|
||||
import grpc
|
||||
import grpc.aio
|
||||
@@ -58,7 +58,7 @@ def register_servicer_hook(
|
||||
async def serve(
|
||||
bumble: PandoraDevice,
|
||||
config: Config = Config(),
|
||||
grpc_server: Optional[grpc.aio.Server] = None,
|
||||
grpc_server: grpc.aio.Server | None = None,
|
||||
port: int = 0,
|
||||
) -> None:
|
||||
# initialize a gRPC server if not provided.
|
||||
|
||||
@@ -16,7 +16,7 @@
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any, Optional
|
||||
from typing import Any
|
||||
|
||||
from bumble import transport
|
||||
from bumble.core import (
|
||||
@@ -54,7 +54,7 @@ class PandoraDevice:
|
||||
|
||||
# HCI transport name & instance.
|
||||
_hci_name: str
|
||||
_hci: Optional[transport.Transport] # type: ignore[name-defined]
|
||||
_hci: transport.Transport | None # type: ignore[name-defined]
|
||||
|
||||
def __init__(self, config: dict[str, Any]) -> None:
|
||||
self.config = config
|
||||
@@ -74,7 +74,9 @@ class PandoraDevice:
|
||||
|
||||
# open HCI transport & set device host.
|
||||
self._hci = await transport.open_transport(self._hci_name)
|
||||
self.device.host = Host(controller_source=self._hci.source, controller_sink=self._hci.sink) # type: ignore[no-untyped-call]
|
||||
self.device.host = Host(
|
||||
controller_source=self._hci.source, controller_sink=self._hci.sink
|
||||
) # type: ignore[no-untyped-call]
|
||||
|
||||
# power-on.
|
||||
await self.device.power_on()
|
||||
@@ -96,7 +98,7 @@ class PandoraDevice:
|
||||
await self.close()
|
||||
await self.open()
|
||||
|
||||
def info(self) -> Optional[dict[str, str]]:
|
||||
def info(self) -> dict[str, str] | None:
|
||||
return {
|
||||
'public_bd_address': str(self.device.public_address),
|
||||
'random_address': str(self.device.random_address),
|
||||
|
||||
+19
-10
@@ -17,12 +17,15 @@ from __future__ import annotations
|
||||
import asyncio
|
||||
import logging
|
||||
import struct
|
||||
from typing import AsyncGenerator, Optional, cast
|
||||
from collections.abc import AsyncGenerator
|
||||
from typing import cast
|
||||
|
||||
import grpc
|
||||
import grpc.aio
|
||||
from google.protobuf import any_pb2 # pytype: disable=pyi-error
|
||||
from google.protobuf import empty_pb2 # pytype: disable=pyi-error
|
||||
from google.protobuf import (
|
||||
any_pb2, # pytype: disable=pyi-error
|
||||
empty_pb2, # pytype: disable=pyi-error
|
||||
)
|
||||
from pandora import host_pb2
|
||||
from pandora.host_grpc_aio import HostServicer
|
||||
from pandora.host_pb2 import (
|
||||
@@ -302,7 +305,9 @@ class HostService(HostServicer):
|
||||
await disconnection_future
|
||||
self.log.debug("Disconnected")
|
||||
finally:
|
||||
connection.remove_listener(connection.EVENT_DISCONNECTION, on_disconnection) # type: ignore
|
||||
connection.remove_listener(
|
||||
connection.EVENT_DISCONNECTION, on_disconnection
|
||||
) # type: ignore
|
||||
|
||||
return empty_pb2.Empty()
|
||||
|
||||
@@ -539,7 +544,7 @@ class HostService(HostServicer):
|
||||
await bumble.utils.cancel_on_event(
|
||||
self.device, 'flush', self.device.stop_advertising()
|
||||
)
|
||||
except:
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
@utils.rpc
|
||||
@@ -609,7 +614,7 @@ class HostService(HostServicer):
|
||||
await bumble.utils.cancel_on_event(
|
||||
self.device, 'flush', self.device.stop_scanning()
|
||||
)
|
||||
except:
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
@utils.rpc
|
||||
@@ -619,7 +624,7 @@ class HostService(HostServicer):
|
||||
self.log.debug('Inquiry')
|
||||
|
||||
inquiry_queue: asyncio.Queue[
|
||||
Optional[tuple[Address, int, AdvertisingData, int]]
|
||||
tuple[Address, int, AdvertisingData, int] | None
|
||||
] = asyncio.Queue()
|
||||
complete_handler = self.device.on(
|
||||
self.device.EVENT_INQUIRY_COMPLETE, lambda: inquiry_queue.put_nowait(None)
|
||||
@@ -644,14 +649,18 @@ class HostService(HostServicer):
|
||||
)
|
||||
|
||||
finally:
|
||||
self.device.remove_listener(self.device.EVENT_INQUIRY_COMPLETE, complete_handler) # type: ignore
|
||||
self.device.remove_listener(self.device.EVENT_INQUIRY_RESULT, result_handler) # type: ignore
|
||||
self.device.remove_listener(
|
||||
self.device.EVENT_INQUIRY_COMPLETE, complete_handler
|
||||
) # type: ignore
|
||||
self.device.remove_listener(
|
||||
self.device.EVENT_INQUIRY_RESULT, result_handler
|
||||
) # type: ignore
|
||||
try:
|
||||
self.log.debug('Stop inquiry')
|
||||
await bumble.utils.cancel_on_event(
|
||||
self.device, 'flush', self.device.stop_discovery()
|
||||
)
|
||||
except:
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
@utils.rpc
|
||||
|
||||
@@ -18,15 +18,15 @@ import json
|
||||
import logging
|
||||
from asyncio import Future
|
||||
from asyncio import Queue as AsyncQueue
|
||||
from collections.abc import AsyncGenerator
|
||||
from dataclasses import dataclass
|
||||
from typing import AsyncGenerator, Optional, Union
|
||||
|
||||
import grpc
|
||||
from google.protobuf import any_pb2, empty_pb2 # pytype: disable=pyi-error
|
||||
from pandora.l2cap_grpc_aio import L2CAPServicer # pytype: disable=pyi-error
|
||||
from pandora.l2cap_pb2 import COMMAND_NOT_UNDERSTOOD, INVALID_CID_IN_REQUEST
|
||||
from pandora.l2cap_pb2 import Channel as PandoraChannel # pytype: disable=pyi-error
|
||||
from pandora.l2cap_pb2 import (
|
||||
COMMAND_NOT_UNDERSTOOD,
|
||||
INVALID_CID_IN_REQUEST,
|
||||
ConnectRequest,
|
||||
ConnectResponse,
|
||||
CreditBasedChannelRequest,
|
||||
@@ -41,6 +41,7 @@ from pandora.l2cap_pb2 import (
|
||||
WaitDisconnectionRequest,
|
||||
WaitDisconnectionResponse,
|
||||
)
|
||||
from pandora.l2cap_pb2 import Channel as PandoraChannel # pytype: disable=pyi-error
|
||||
|
||||
from bumble.core import InvalidArgumentError, OutOfResourcesError
|
||||
from bumble.device import Device
|
||||
@@ -55,7 +56,7 @@ from bumble.l2cap import (
|
||||
from bumble.pandora import utils
|
||||
from bumble.pandora.config import Config
|
||||
|
||||
L2capChannel = Union[ClassicChannel, LeCreditBasedChannel]
|
||||
L2capChannel = ClassicChannel | LeCreditBasedChannel
|
||||
|
||||
|
||||
@dataclass
|
||||
@@ -106,10 +107,8 @@ class L2CAPService(L2CAPServicer):
|
||||
oneof = request.WhichOneof('type')
|
||||
self.log.debug(f'WaitConnection channel request type: {oneof}.')
|
||||
channel_type = getattr(request, oneof)
|
||||
spec: Optional[Union[ClassicChannelSpec, LeCreditBasedChannelSpec]] = None
|
||||
l2cap_server: Optional[
|
||||
Union[ClassicChannelServer, LeCreditBasedChannelServer]
|
||||
] = None
|
||||
spec: ClassicChannelSpec | LeCreditBasedChannelSpec | None = None
|
||||
l2cap_server: ClassicChannelServer | LeCreditBasedChannelServer | None = None
|
||||
if isinstance(channel_type, CreditBasedChannelRequest):
|
||||
spec = LeCreditBasedChannelSpec(
|
||||
psm=channel_type.spsm,
|
||||
@@ -216,7 +215,7 @@ class L2CAPService(L2CAPServicer):
|
||||
oneof = request.WhichOneof('type')
|
||||
self.log.debug(f'Channel request type: {oneof}.')
|
||||
channel_type = getattr(request, oneof)
|
||||
spec: Optional[Union[ClassicChannelSpec, LeCreditBasedChannelSpec]] = None
|
||||
spec: ClassicChannelSpec | LeCreditBasedChannelSpec | None = None
|
||||
if isinstance(channel_type, CreditBasedChannelRequest):
|
||||
spec = LeCreditBasedChannelSpec(
|
||||
psm=channel_type.spsm,
|
||||
|
||||
+18
-16
@@ -17,13 +17,15 @@ from __future__ import annotations
|
||||
import asyncio
|
||||
import contextlib
|
||||
import logging
|
||||
from collections.abc import Awaitable
|
||||
from typing import Any, AsyncGenerator, AsyncIterator, Callable, Optional, Union
|
||||
from collections.abc import AsyncGenerator, AsyncIterator, Awaitable, Callable
|
||||
from typing import Any
|
||||
|
||||
import grpc
|
||||
from google.protobuf import any_pb2 # pytype: disable=pyi-error
|
||||
from google.protobuf import empty_pb2 # pytype: disable=pyi-error
|
||||
from google.protobuf import wrappers_pb2 # pytype: disable=pyi-error
|
||||
from google.protobuf import (
|
||||
any_pb2, # pytype: disable=pyi-error
|
||||
empty_pb2, # pytype: disable=pyi-error
|
||||
wrappers_pb2, # pytype: disable=pyi-error
|
||||
)
|
||||
from pandora.host_pb2 import Connection
|
||||
from pandora.security_grpc_aio import SecurityServicer, SecurityStorageServicer
|
||||
from pandora.security_pb2 import (
|
||||
@@ -64,7 +66,7 @@ class PairingDelegate(BasePairingDelegate):
|
||||
def __init__(
|
||||
self,
|
||||
connection: BumbleConnection,
|
||||
service: "SecurityService",
|
||||
service: SecurityService,
|
||||
io_capability: BasePairingDelegate.IoCapability = BasePairingDelegate.NO_OUTPUT_NO_INPUT,
|
||||
local_initiator_key_distribution: BasePairingDelegate.KeyDistribution = BasePairingDelegate.DEFAULT_KEY_DISTRIBUTION,
|
||||
local_responder_key_distribution: BasePairingDelegate.KeyDistribution = BasePairingDelegate.DEFAULT_KEY_DISTRIBUTION,
|
||||
@@ -130,7 +132,7 @@ class PairingDelegate(BasePairingDelegate):
|
||||
assert answer.answer_variant() == 'confirm' and answer.confirm is not None
|
||||
return answer.confirm
|
||||
|
||||
async def get_number(self) -> Optional[int]:
|
||||
async def get_number(self) -> int | None:
|
||||
self.log.debug(
|
||||
f"Pairing event: `passkey_entry_request` (io_capability: {self.io_capability})"
|
||||
)
|
||||
@@ -147,7 +149,7 @@ class PairingDelegate(BasePairingDelegate):
|
||||
assert answer.answer_variant() == 'passkey'
|
||||
return answer.passkey
|
||||
|
||||
async def get_string(self, max_length: int) -> Optional[str]:
|
||||
async def get_string(self, max_length: int) -> str | None:
|
||||
self.log.debug(
|
||||
f"Pairing event: `pin_code_request` (io_capability: {self.io_capability})"
|
||||
)
|
||||
@@ -195,8 +197,8 @@ class SecurityService(SecurityServicer):
|
||||
self.log = utils.BumbleServerLoggerAdapter(
|
||||
logging.getLogger(), {'service_name': 'Security', 'device': device}
|
||||
)
|
||||
self.event_queue: Optional[asyncio.Queue[PairingEvent]] = None
|
||||
self.event_answer: Optional[AsyncIterator[PairingEventAnswer]] = None
|
||||
self.event_queue: asyncio.Queue[PairingEvent] | None = None
|
||||
self.event_answer: AsyncIterator[PairingEventAnswer] | None = None
|
||||
self.device = device
|
||||
self.config = config
|
||||
|
||||
@@ -231,7 +233,7 @@ class SecurityService(SecurityServicer):
|
||||
if level == LEVEL2:
|
||||
return connection.encryption != 0 and connection.authenticated
|
||||
|
||||
link_key_type: Optional[int] = None
|
||||
link_key_type: int | None = None
|
||||
if (keystore := connection.device.keystore) and (
|
||||
keys := await keystore.get(str(connection.peer_address))
|
||||
):
|
||||
@@ -410,8 +412,8 @@ class SecurityService(SecurityServicer):
|
||||
wait_for_security: asyncio.Future[str] = (
|
||||
asyncio.get_running_loop().create_future()
|
||||
)
|
||||
authenticate_task: Optional[asyncio.Future[None]] = None
|
||||
pair_task: Optional[asyncio.Future[None]] = None
|
||||
authenticate_task: asyncio.Future[None] | None = None
|
||||
pair_task: asyncio.Future[None] | None = None
|
||||
|
||||
async def authenticate() -> None:
|
||||
if (encryption := connection.encryption) != 0:
|
||||
@@ -455,9 +457,9 @@ class SecurityService(SecurityServicer):
|
||||
|
||||
def pair(*_: Any) -> None:
|
||||
if self.need_pairing(connection, level):
|
||||
pair_task = asyncio.create_task(connection.pair())
|
||||
bumble.utils.AsyncRunner.spawn(connection.pair())
|
||||
|
||||
listeners: dict[str, Callable[..., Union[None, Awaitable[None]]]] = {
|
||||
listeners: dict[str, Callable[..., None | Awaitable[None]]] = {
|
||||
'disconnection': set_failure('connection_died'),
|
||||
'pairing_failure': set_failure('pairing_failure'),
|
||||
'connection_authentication_failure': set_failure('authentication_failure'),
|
||||
@@ -500,7 +502,7 @@ class SecurityService(SecurityServicer):
|
||||
return WaitSecurityResponse(**kwargs)
|
||||
|
||||
async def reached_security_level(
|
||||
self, connection: BumbleConnection, level: Union[SecurityLevel, LESecurityLevel]
|
||||
self, connection: BumbleConnection, level: SecurityLevel | LESecurityLevel
|
||||
) -> bool:
|
||||
self.log.debug(
|
||||
str(
|
||||
|
||||
@@ -18,7 +18,8 @@ import contextlib
|
||||
import functools
|
||||
import inspect
|
||||
import logging
|
||||
from typing import Any, Generator, MutableMapping, Optional
|
||||
from collections.abc import Generator, MutableMapping
|
||||
from typing import Any
|
||||
|
||||
import grpc
|
||||
from google.protobuf.message import Message # pytype: disable=pyi-error
|
||||
@@ -34,7 +35,7 @@ ADDRESS_TYPES: dict[str, AddressType] = {
|
||||
}
|
||||
|
||||
|
||||
def address_from_request(request: Message, field: Optional[str]) -> Address:
|
||||
def address_from_request(request: Message, field: str | None) -> Address:
|
||||
if field is None:
|
||||
return Address.ANY
|
||||
return Address(bytes(reversed(getattr(request, field))), ADDRESS_TYPES[field])
|
||||
@@ -95,8 +96,7 @@ def rpc(func: Any) -> Any:
|
||||
@functools.wraps(func)
|
||||
def gen_wrapper(self: Any, request: Any, context: grpc.ServicerContext) -> Any:
|
||||
with exception_to_rpc_error(context):
|
||||
for v in func(self, request, context):
|
||||
yield v
|
||||
yield from func(self, request, context)
|
||||
|
||||
@functools.wraps(func)
|
||||
def wrapper(self: Any, request: Any, context: grpc.ServicerContext) -> Any:
|
||||
|
||||
@@ -22,7 +22,6 @@ from __future__ import annotations
|
||||
import logging
|
||||
import struct
|
||||
from dataclasses import dataclass
|
||||
from typing import Optional
|
||||
|
||||
from bumble import utils
|
||||
from bumble.att import ATT_Error
|
||||
@@ -129,7 +128,7 @@ class AudioInputState:
|
||||
mute: Mute = Mute.NOT_MUTED
|
||||
gain_mode: GainMode = GainMode.MANUAL
|
||||
change_counter: int = 0
|
||||
attribute: Optional[Attribute] = None
|
||||
attribute: Attribute | None = None
|
||||
|
||||
def __bytes__(self) -> bytes:
|
||||
return bytes(
|
||||
@@ -199,7 +198,6 @@ class AudioInputControlPoint:
|
||||
gain_settings_properties: GainSettingsProperties
|
||||
|
||||
async def on_write(self, connection: Connection, value: bytes) -> None:
|
||||
|
||||
opcode = AudioInputControlPointOpCode(value[0])
|
||||
|
||||
if opcode == AudioInputControlPointOpCode.SET_GAIN_SETTING:
|
||||
@@ -317,7 +315,7 @@ class AudioInputDescription:
|
||||
'''
|
||||
|
||||
audio_input_description: str = "Bluetooth"
|
||||
attribute: Optional[Attribute] = None
|
||||
attribute: Attribute | None = None
|
||||
|
||||
def on_read(self, _connection: Connection) -> str:
|
||||
return self.audio_input_description
|
||||
@@ -340,11 +338,11 @@ class AICSService(TemplateService):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
audio_input_state: Optional[AudioInputState] = None,
|
||||
gain_settings_properties: Optional[GainSettingsProperties] = None,
|
||||
audio_input_state: AudioInputState | None = None,
|
||||
gain_settings_properties: GainSettingsProperties | None = None,
|
||||
audio_input_type: str = "local",
|
||||
audio_input_status: Optional[AudioInputStatus] = None,
|
||||
audio_input_description: Optional[AudioInputDescription] = None,
|
||||
audio_input_status: AudioInputStatus | None = None,
|
||||
audio_input_description: AudioInputDescription | None = None,
|
||||
):
|
||||
self.audio_input_state = (
|
||||
AudioInputState() if audio_input_state is None else audio_input_state
|
||||
|
||||
@@ -25,7 +25,7 @@ import asyncio
|
||||
import dataclasses
|
||||
import enum
|
||||
import logging
|
||||
from typing import Iterable, Optional, Union
|
||||
from collections.abc import Iterable
|
||||
|
||||
from bumble import utils
|
||||
from bumble.device import Peer
|
||||
@@ -230,7 +230,7 @@ class AmsClient(utils.EventEmitter):
|
||||
self.supported_commands = set()
|
||||
|
||||
@classmethod
|
||||
async def for_peer(cls, peer: Peer) -> Optional[AmsClient]:
|
||||
async def for_peer(cls, peer: Peer) -> AmsClient | None:
|
||||
ams_proxy = await peer.discover_service_and_create_proxy(AmsProxy)
|
||||
if ams_proxy is None:
|
||||
return None
|
||||
@@ -263,9 +263,7 @@ class AmsClient(utils.EventEmitter):
|
||||
async def observe(
|
||||
self,
|
||||
entity: EntityId,
|
||||
attributes: Iterable[
|
||||
Union[PlayerAttributeId, QueueAttributeId, TrackAttributeId]
|
||||
],
|
||||
attributes: Iterable[PlayerAttributeId | QueueAttributeId | TrackAttributeId],
|
||||
) -> None:
|
||||
await self._ams_proxy.entity_update.write_value(
|
||||
bytes([entity] + list(attributes)), with_response=True
|
||||
|
||||
+11
-11
@@ -27,7 +27,7 @@ import datetime
|
||||
import enum
|
||||
import logging
|
||||
import struct
|
||||
from typing import Optional, Sequence, Union
|
||||
from collections.abc import Sequence
|
||||
|
||||
from bumble import utils
|
||||
from bumble.att import ATT_Error
|
||||
@@ -116,7 +116,7 @@ class NotificationAttributeId(utils.OpenIntEnum):
|
||||
@dataclasses.dataclass
|
||||
class NotificationAttribute:
|
||||
attribute_id: NotificationAttributeId
|
||||
value: Union[str, int, datetime.datetime]
|
||||
value: str | int | datetime.datetime
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
@@ -242,10 +242,10 @@ class AncsProxy(ProfileServiceProxy):
|
||||
|
||||
|
||||
class AncsClient(utils.EventEmitter):
|
||||
_expected_response_command_id: Optional[CommandId]
|
||||
_expected_response_notification_uid: Optional[int]
|
||||
_expected_response_app_identifier: Optional[str]
|
||||
_expected_app_identifier: Optional[str]
|
||||
_expected_response_command_id: CommandId | None
|
||||
_expected_response_notification_uid: int | None
|
||||
_expected_response_app_identifier: str | None
|
||||
_expected_app_identifier: str | None
|
||||
_expected_response_tuples: int
|
||||
_response_accumulator: bytes
|
||||
|
||||
@@ -255,12 +255,12 @@ class AncsClient(utils.EventEmitter):
|
||||
super().__init__()
|
||||
self._ancs_proxy = ancs_proxy
|
||||
self._command_semaphore = asyncio.Semaphore()
|
||||
self._response: Optional[asyncio.Future] = None
|
||||
self._response: asyncio.Future | None = None
|
||||
self._reset_response()
|
||||
self._started = False
|
||||
|
||||
@classmethod
|
||||
async def for_peer(cls, peer: Peer) -> Optional[AncsClient]:
|
||||
async def for_peer(cls, peer: Peer) -> AncsClient | None:
|
||||
ancs_proxy = await peer.discover_service_and_create_proxy(AncsProxy)
|
||||
if ancs_proxy is None:
|
||||
return None
|
||||
@@ -316,7 +316,7 @@ class AncsClient(utils.EventEmitter):
|
||||
# Not enough data yet.
|
||||
return
|
||||
|
||||
attributes: list[Union[NotificationAttribute, AppAttribute]] = []
|
||||
attributes: list[NotificationAttribute | AppAttribute] = []
|
||||
|
||||
if command_id == CommandId.GET_NOTIFICATION_ATTRIBUTES:
|
||||
(notification_uid,) = struct.unpack_from(
|
||||
@@ -342,7 +342,7 @@ class AncsClient(utils.EventEmitter):
|
||||
str_value = attribute_data[3 : 3 + attribute_data_length].decode(
|
||||
"utf-8"
|
||||
)
|
||||
value: Union[str, int, datetime.datetime]
|
||||
value: str | int | datetime.datetime
|
||||
if attribute_id == NotificationAttributeId.MESSAGE_SIZE:
|
||||
value = int(str_value)
|
||||
elif attribute_id == NotificationAttributeId.DATE:
|
||||
@@ -415,7 +415,7 @@ class AncsClient(utils.EventEmitter):
|
||||
self,
|
||||
notification_uid: int,
|
||||
attributes: Sequence[
|
||||
Union[NotificationAttributeId, tuple[NotificationAttributeId, int]]
|
||||
NotificationAttributeId | tuple[NotificationAttributeId, int]
|
||||
],
|
||||
) -> list[NotificationAttribute]:
|
||||
if not self._started:
|
||||
|
||||
@@ -24,7 +24,7 @@ import logging
|
||||
import struct
|
||||
from collections.abc import Sequence
|
||||
from dataclasses import dataclass, field
|
||||
from typing import Any, Optional, TypeVar, Union
|
||||
from typing import Any, TypeVar
|
||||
|
||||
from bumble import colors, device, gatt, gatt_client, hci, utils
|
||||
from bumble.profiles import le_audio
|
||||
@@ -49,7 +49,7 @@ class ASE_Operation:
|
||||
classes: dict[int, type[ASE_Operation]] = {}
|
||||
op_code: Opcode
|
||||
name: str
|
||||
fields: Optional[Sequence[Any]] = None
|
||||
fields: Sequence[Any] | None = None
|
||||
ase_id: Sequence[int]
|
||||
|
||||
class Opcode(enum.IntEnum):
|
||||
@@ -278,7 +278,7 @@ class AseStateMachine(gatt.Characteristic):
|
||||
|
||||
EVENT_STATE_CHANGE = "state_change"
|
||||
|
||||
cis_link: Optional[device.CisLink] = None
|
||||
cis_link: device.CisLink | None = None
|
||||
|
||||
# Additional parameters in CODEC_CONFIGURED State
|
||||
preferred_framing = 0 # Unframed PDU supported
|
||||
@@ -290,7 +290,7 @@ class AseStateMachine(gatt.Characteristic):
|
||||
preferred_presentation_delay_min = 0
|
||||
preferred_presentation_delay_max = 0
|
||||
codec_id = hci.CodingFormat(hci.CodecID.LC3)
|
||||
codec_specific_configuration: Union[CodecSpecificConfiguration, bytes] = b''
|
||||
codec_specific_configuration: CodecSpecificConfiguration | bytes = b''
|
||||
|
||||
# Additional parameters in QOS_CONFIGURED State
|
||||
cig_id = 0
|
||||
@@ -610,7 +610,7 @@ class AudioStreamControlService(gatt.TemplateService):
|
||||
|
||||
ase_state_machines: dict[int, AseStateMachine]
|
||||
ase_control_point: gatt.Characteristic[bytes]
|
||||
_active_client: Optional[device.Connection] = None
|
||||
_active_client: device.Connection | None = None
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
|
||||
+10
-9
@@ -19,7 +19,8 @@
|
||||
import enum
|
||||
import logging
|
||||
import struct
|
||||
from typing import Any, Callable, Optional, Union
|
||||
from collections.abc import Callable
|
||||
from typing import Any
|
||||
|
||||
from bumble import data_types, gatt, gatt_client, l2cap, utils
|
||||
from bumble.core import AdvertisingData
|
||||
@@ -90,20 +91,20 @@ class AshaService(gatt.TemplateService):
|
||||
EVENT_DISCONNECTED = "disconnected"
|
||||
EVENT_VOLUME_CHANGED = "volume_changed"
|
||||
|
||||
audio_sink: Optional[Callable[[bytes], Any]]
|
||||
active_codec: Optional[Codec] = None
|
||||
audio_type: Optional[AudioType] = None
|
||||
volume: Optional[int] = None
|
||||
other_state: Optional[int] = None
|
||||
connection: Optional[Connection] = None
|
||||
audio_sink: Callable[[bytes], Any] | None
|
||||
active_codec: Codec | None = None
|
||||
audio_type: AudioType | None = None
|
||||
volume: int | None = None
|
||||
other_state: int | None = None
|
||||
connection: Connection | None = None
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
capability: int,
|
||||
hisyncid: Union[list[int], bytes],
|
||||
hisyncid: list[int] | bytes,
|
||||
device: Device,
|
||||
psm: int = 0,
|
||||
audio_sink: Optional[Callable[[bytes], Any]] = None,
|
||||
audio_sink: Callable[[bytes], Any] | None = None,
|
||||
feature_map: int = FeatureMap.LE_COC_AUDIO_OUTPUT_STREAMING_SUPPORTED,
|
||||
protocol_version: int = 0x01,
|
||||
render_delay_milliseconds: int = 0,
|
||||
|
||||
@@ -21,7 +21,8 @@ from __future__ import annotations
|
||||
import dataclasses
|
||||
import logging
|
||||
import struct
|
||||
from typing import ClassVar, Optional, Sequence
|
||||
from collections.abc import Sequence
|
||||
from typing import ClassVar
|
||||
|
||||
from bumble import core, device, gatt, gatt_adapters, gatt_client, hci, utils
|
||||
|
||||
@@ -337,7 +338,12 @@ class BroadcastAudioScanService(gatt.TemplateService):
|
||||
b"12", # TEST
|
||||
)
|
||||
|
||||
super().__init__([self.battery_level_characteristic])
|
||||
super().__init__(
|
||||
[
|
||||
self.broadcast_audio_scan_control_point_characteristic,
|
||||
self.broadcast_receive_state_characteristic,
|
||||
]
|
||||
)
|
||||
|
||||
def on_broadcast_audio_scan_control_point_write(
|
||||
self, connection: device.Connection, value: bytes
|
||||
@@ -351,7 +357,7 @@ class BroadcastAudioScanServiceProxy(gatt_client.ProfileServiceProxy):
|
||||
|
||||
broadcast_audio_scan_control_point: gatt_client.CharacteristicProxy[bytes]
|
||||
broadcast_receive_states: list[
|
||||
gatt_client.CharacteristicProxy[Optional[BroadcastReceiveState]]
|
||||
gatt_client.CharacteristicProxy[BroadcastReceiveState | None]
|
||||
]
|
||||
|
||||
def __init__(self, service_proxy: gatt_client.ServiceProxy):
|
||||
|
||||
@@ -16,7 +16,6 @@
|
||||
# -----------------------------------------------------------------------------
|
||||
# Imports
|
||||
# -----------------------------------------------------------------------------
|
||||
from typing import Optional
|
||||
|
||||
from bumble.gatt import (
|
||||
GATT_BATTERY_LEVEL_CHARACTERISTIC,
|
||||
@@ -56,7 +55,7 @@ class BatteryService(TemplateService):
|
||||
class BatteryServiceProxy(ProfileServiceProxy):
|
||||
SERVICE_CLASS = BatteryService
|
||||
|
||||
battery_level: Optional[CharacteristicProxy[int]]
|
||||
battery_level: CharacteristicProxy[int] | None
|
||||
|
||||
def __init__(self, service_proxy):
|
||||
self.service_proxy = service_proxy
|
||||
|
||||
+9
-10
@@ -20,7 +20,6 @@ from __future__ import annotations
|
||||
|
||||
import enum
|
||||
import struct
|
||||
from typing import Optional
|
||||
|
||||
from bumble import core, crypto, device, gatt, gatt_client
|
||||
|
||||
@@ -96,17 +95,17 @@ class CoordinatedSetIdentificationService(gatt.TemplateService):
|
||||
|
||||
set_identity_resolving_key: bytes
|
||||
set_identity_resolving_key_characteristic: gatt.Characteristic[bytes]
|
||||
coordinated_set_size_characteristic: Optional[gatt.Characteristic[bytes]] = None
|
||||
set_member_lock_characteristic: Optional[gatt.Characteristic[bytes]] = None
|
||||
set_member_rank_characteristic: Optional[gatt.Characteristic[bytes]] = None
|
||||
coordinated_set_size_characteristic: gatt.Characteristic[bytes] | None = None
|
||||
set_member_lock_characteristic: gatt.Characteristic[bytes] | None = None
|
||||
set_member_rank_characteristic: gatt.Characteristic[bytes] | None = None
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
set_identity_resolving_key: bytes,
|
||||
set_identity_resolving_key_type: SirkType,
|
||||
coordinated_set_size: Optional[int] = None,
|
||||
set_member_lock: Optional[MemberLock] = None,
|
||||
set_member_rank: Optional[int] = None,
|
||||
coordinated_set_size: int | None = None,
|
||||
set_member_lock: MemberLock | None = None,
|
||||
set_member_rank: int | None = None,
|
||||
) -> None:
|
||||
if len(set_identity_resolving_key) != SET_IDENTITY_RESOLVING_KEY_LENGTH:
|
||||
raise core.InvalidArgumentError(
|
||||
@@ -198,9 +197,9 @@ class CoordinatedSetIdentificationProxy(gatt_client.ProfileServiceProxy):
|
||||
SERVICE_CLASS = CoordinatedSetIdentificationService
|
||||
|
||||
set_identity_resolving_key: gatt_client.CharacteristicProxy[bytes]
|
||||
coordinated_set_size: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
set_member_lock: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
set_member_rank: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
coordinated_set_size: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
set_member_lock: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
set_member_rank: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
|
||||
def __init__(self, service_proxy: gatt_client.ServiceProxy) -> None:
|
||||
self.service_proxy = service_proxy
|
||||
|
||||
@@ -17,7 +17,6 @@
|
||||
# Imports
|
||||
# -----------------------------------------------------------------------------
|
||||
import struct
|
||||
from typing import Optional
|
||||
|
||||
from bumble.gatt import (
|
||||
GATT_DEVICE_INFORMATION_SERVICE,
|
||||
@@ -54,14 +53,14 @@ class DeviceInformationService(TemplateService):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
manufacturer_name: Optional[str] = None,
|
||||
model_number: Optional[str] = None,
|
||||
serial_number: Optional[str] = None,
|
||||
hardware_revision: Optional[str] = None,
|
||||
firmware_revision: Optional[str] = None,
|
||||
software_revision: Optional[str] = None,
|
||||
system_id: Optional[tuple[int, int]] = None, # (OUI, Manufacturer ID)
|
||||
ieee_regulatory_certification_data_list: Optional[bytes] = None,
|
||||
manufacturer_name: str | None = None,
|
||||
model_number: str | None = None,
|
||||
serial_number: str | None = None,
|
||||
hardware_revision: str | None = None,
|
||||
firmware_revision: str | None = None,
|
||||
software_revision: str | None = None,
|
||||
system_id: tuple[int, int] | None = None, # (OUI, Manufacturer ID)
|
||||
ieee_regulatory_certification_data_list: bytes | None = None,
|
||||
# TODO: pnp_id
|
||||
):
|
||||
characteristics: list[Characteristic[bytes]] = [
|
||||
@@ -109,14 +108,14 @@ class DeviceInformationService(TemplateService):
|
||||
class DeviceInformationServiceProxy(ProfileServiceProxy):
|
||||
SERVICE_CLASS = DeviceInformationService
|
||||
|
||||
manufacturer_name: Optional[CharacteristicProxy[str]]
|
||||
model_number: Optional[CharacteristicProxy[str]]
|
||||
serial_number: Optional[CharacteristicProxy[str]]
|
||||
hardware_revision: Optional[CharacteristicProxy[str]]
|
||||
firmware_revision: Optional[CharacteristicProxy[str]]
|
||||
software_revision: Optional[CharacteristicProxy[str]]
|
||||
system_id: Optional[CharacteristicProxy[tuple[int, int]]]
|
||||
ieee_regulatory_certification_data_list: Optional[CharacteristicProxy[bytes]]
|
||||
manufacturer_name: CharacteristicProxy[str] | None
|
||||
model_number: CharacteristicProxy[str] | None
|
||||
serial_number: CharacteristicProxy[str] | None
|
||||
hardware_revision: CharacteristicProxy[str] | None
|
||||
firmware_revision: CharacteristicProxy[str] | None
|
||||
software_revision: CharacteristicProxy[str] | None
|
||||
system_id: CharacteristicProxy[tuple[int, int]] | None
|
||||
ieee_regulatory_certification_data_list: CharacteristicProxy[bytes] | None
|
||||
|
||||
def __init__(self, service_proxy: ServiceProxy):
|
||||
self.service_proxy = service_proxy
|
||||
|
||||
@@ -19,7 +19,6 @@
|
||||
# -----------------------------------------------------------------------------
|
||||
import logging
|
||||
import struct
|
||||
from typing import Optional, Union
|
||||
|
||||
from bumble.core import Appearance
|
||||
from bumble.gatt import (
|
||||
@@ -54,7 +53,7 @@ class GenericAccessService(TemplateService):
|
||||
appearance_characteristic: Characteristic[bytes]
|
||||
|
||||
def __init__(
|
||||
self, device_name: str, appearance: Union[Appearance, tuple[int, int], int] = 0
|
||||
self, device_name: str, appearance: Appearance | tuple[int, int] | int = 0
|
||||
):
|
||||
if isinstance(appearance, int):
|
||||
appearance_int = appearance
|
||||
@@ -88,8 +87,8 @@ class GenericAccessService(TemplateService):
|
||||
class GenericAccessServiceProxy(ProfileServiceProxy):
|
||||
SERVICE_CLASS = GenericAccessService
|
||||
|
||||
device_name: Optional[CharacteristicProxy[str]]
|
||||
appearance: Optional[CharacteristicProxy[Appearance]]
|
||||
device_name: CharacteristicProxy[str] | None
|
||||
appearance: CharacteristicProxy[Appearance] | None
|
||||
|
||||
def __init__(self, service_proxy: ServiceProxy):
|
||||
self.service_proxy = service_proxy
|
||||
|
||||
@@ -40,7 +40,6 @@ class GenericAttributeProfileService(gatt.TemplateService):
|
||||
database_hash_enabled: bool = True,
|
||||
service_change_enabled: bool = True,
|
||||
) -> None:
|
||||
|
||||
if server_supported_features is not None:
|
||||
self.server_supported_features_characteristic = gatt.Characteristic(
|
||||
uuid=gatt.GATT_SERVER_SUPPORTED_FEATURES_CHARACTERISTIC,
|
||||
|
||||
+12
-13
@@ -19,7 +19,6 @@
|
||||
# -----------------------------------------------------------------------------
|
||||
import struct
|
||||
from enum import IntFlag
|
||||
from typing import Optional
|
||||
|
||||
from bumble.gatt import (
|
||||
GATT_BGR_FEATURES_CHARACTERISTIC,
|
||||
@@ -77,18 +76,18 @@ class GamingAudioService(TemplateService):
|
||||
UUID = GATT_GAMING_AUDIO_SERVICE
|
||||
|
||||
gmap_role: Characteristic
|
||||
ugg_features: Optional[Characteristic] = None
|
||||
ugt_features: Optional[Characteristic] = None
|
||||
bgs_features: Optional[Characteristic] = None
|
||||
bgr_features: Optional[Characteristic] = None
|
||||
ugg_features: Characteristic | None = None
|
||||
ugt_features: Characteristic | None = None
|
||||
bgs_features: Characteristic | None = None
|
||||
bgr_features: Characteristic | None = None
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
gmap_role: GmapRole,
|
||||
ugg_features: Optional[UggFeatures] = None,
|
||||
ugt_features: Optional[UgtFeatures] = None,
|
||||
bgs_features: Optional[BgsFeatures] = None,
|
||||
bgr_features: Optional[BgrFeatures] = None,
|
||||
ugg_features: UggFeatures | None = None,
|
||||
ugt_features: UgtFeatures | None = None,
|
||||
bgs_features: BgsFeatures | None = None,
|
||||
bgr_features: BgrFeatures | None = None,
|
||||
) -> None:
|
||||
characteristics = []
|
||||
|
||||
@@ -150,10 +149,10 @@ class GamingAudioService(TemplateService):
|
||||
class GamingAudioServiceProxy(ProfileServiceProxy):
|
||||
SERVICE_CLASS = GamingAudioService
|
||||
|
||||
ugg_features: Optional[CharacteristicProxy[UggFeatures]] = None
|
||||
ugt_features: Optional[CharacteristicProxy[UgtFeatures]] = None
|
||||
bgs_features: Optional[CharacteristicProxy[BgsFeatures]] = None
|
||||
bgr_features: Optional[CharacteristicProxy[BgrFeatures]] = None
|
||||
ugg_features: CharacteristicProxy[UggFeatures] | None = None
|
||||
ugt_features: CharacteristicProxy[UgtFeatures] | None = None
|
||||
bgs_features: CharacteristicProxy[BgsFeatures] | None = None
|
||||
bgr_features: CharacteristicProxy[BgrFeatures] | None = None
|
||||
|
||||
def __init__(self, service_proxy: ServiceProxy) -> None:
|
||||
self.service_proxy = service_proxy
|
||||
|
||||
@@ -20,7 +20,7 @@ from __future__ import annotations
|
||||
import asyncio
|
||||
import logging
|
||||
from dataclasses import dataclass, field
|
||||
from typing import Any, Optional, Union
|
||||
from typing import Any
|
||||
|
||||
from bumble import att, gatt, gatt_adapters, gatt_client, utils
|
||||
from bumble.core import InvalidArgumentError, InvalidStateError
|
||||
@@ -145,7 +145,7 @@ class PresetChangedOperation:
|
||||
return bytes([self.prev_index]) + bytes(self.preset_record)
|
||||
|
||||
change_id: ChangeId
|
||||
additional_parameters: Union[Generic, int]
|
||||
additional_parameters: Generic | int
|
||||
|
||||
def to_bytes(self, is_last: bool) -> bytes:
|
||||
if isinstance(self.additional_parameters, PresetChangedOperation.Generic):
|
||||
@@ -235,7 +235,7 @@ class HearingAccessService(gatt.TemplateService):
|
||||
preset_records: dict[int, PresetRecord] # key is the preset index
|
||||
read_presets_request_in_progress: bool
|
||||
|
||||
other_server_in_binaural_set: Optional[HearingAccessService] = None
|
||||
other_server_in_binaural_set: HearingAccessService | None = None
|
||||
|
||||
preset_changed_operations_history_per_device: dict[
|
||||
Address, list[PresetChangedOperation]
|
||||
|
||||
@@ -20,7 +20,6 @@ from __future__ import annotations
|
||||
|
||||
import struct
|
||||
from enum import IntEnum
|
||||
from typing import Optional
|
||||
|
||||
from bumble import core
|
||||
from bumble.att import ATT_Error
|
||||
@@ -207,13 +206,13 @@ class HeartRateService(TemplateService):
|
||||
class HeartRateServiceProxy(ProfileServiceProxy):
|
||||
SERVICE_CLASS = HeartRateService
|
||||
|
||||
heart_rate_measurement: Optional[
|
||||
CharacteristicProxy[HeartRateService.HeartRateMeasurement]
|
||||
]
|
||||
body_sensor_location: Optional[
|
||||
CharacteristicProxy[HeartRateService.BodySensorLocation]
|
||||
]
|
||||
heart_rate_control_point: Optional[CharacteristicProxy[int]]
|
||||
heart_rate_measurement: (
|
||||
CharacteristicProxy[HeartRateService.HeartRateMeasurement] | None
|
||||
)
|
||||
body_sensor_location: (
|
||||
CharacteristicProxy[HeartRateService.BodySensorLocation] | None
|
||||
)
|
||||
heart_rate_control_point: CharacteristicProxy[int] | None
|
||||
|
||||
def __init__(self, service_proxy):
|
||||
self.service_proxy = service_proxy
|
||||
|
||||
@@ -137,7 +137,7 @@ class Metadata:
|
||||
values.append(str(decoded))
|
||||
|
||||
return '\n'.join(
|
||||
f'{indent}{key}: {" " * (max_key_length-len(key))}{value}'
|
||||
f'{indent}{key}: {" " * (max_key_length - len(key))}{value}'
|
||||
for key, value in zip(keys, values)
|
||||
)
|
||||
|
||||
|
||||
+28
-28
@@ -22,7 +22,7 @@ import asyncio
|
||||
import dataclasses
|
||||
import enum
|
||||
import struct
|
||||
from typing import TYPE_CHECKING, ClassVar, Optional
|
||||
from typing import TYPE_CHECKING, ClassVar
|
||||
|
||||
from typing_extensions import Self
|
||||
|
||||
@@ -196,7 +196,7 @@ class MediaControlService(gatt.TemplateService):
|
||||
|
||||
UUID = gatt.GATT_MEDIA_CONTROL_SERVICE
|
||||
|
||||
def __init__(self, media_player_name: Optional[str] = None) -> None:
|
||||
def __init__(self, media_player_name: str | None = None) -> None:
|
||||
self.track_position = 0
|
||||
|
||||
self.media_player_name_characteristic = gatt.Characteristic(
|
||||
@@ -337,32 +337,32 @@ class MediaControlServiceProxy(
|
||||
EVENT_TRACK_DURATION = "track_duration"
|
||||
EVENT_TRACK_POSITION = "track_position"
|
||||
|
||||
media_player_name: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
media_player_icon_object_id: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
media_player_icon_url: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
track_changed: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
track_title: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
track_duration: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
track_position: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
playback_speed: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
seeking_speed: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
current_track_segments_object_id: Optional[
|
||||
gatt_client.CharacteristicProxy[bytes]
|
||||
] = None
|
||||
current_track_object_id: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
next_track_object_id: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
parent_group_object_id: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
current_group_object_id: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
playing_order: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
playing_orders_supported: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
media_state: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
media_control_point: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
media_control_point_opcodes_supported: Optional[
|
||||
gatt_client.CharacteristicProxy[bytes]
|
||||
] = None
|
||||
search_control_point: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
search_results_object_id: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
content_control_id: Optional[gatt_client.CharacteristicProxy[bytes]] = None
|
||||
media_player_name: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
media_player_icon_object_id: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
media_player_icon_url: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
track_changed: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
track_title: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
track_duration: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
track_position: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
playback_speed: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
seeking_speed: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
current_track_segments_object_id: gatt_client.CharacteristicProxy[bytes] | None = (
|
||||
None
|
||||
)
|
||||
current_track_object_id: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
next_track_object_id: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
parent_group_object_id: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
current_group_object_id: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
playing_order: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
playing_orders_supported: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
media_state: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
media_control_point: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
media_control_point_opcodes_supported: (
|
||||
gatt_client.CharacteristicProxy[bytes] | None
|
||||
) = None
|
||||
search_control_point: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
search_results_object_id: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
content_control_id: gatt_client.CharacteristicProxy[bytes] | None = None
|
||||
|
||||
if TYPE_CHECKING:
|
||||
media_control_point_notifications: asyncio.Queue[bytes]
|
||||
|
||||
+13
-17
@@ -21,7 +21,7 @@ from __future__ import annotations
|
||||
import dataclasses
|
||||
import logging
|
||||
import struct
|
||||
from typing import Optional, Sequence, Union
|
||||
from collections.abc import Sequence
|
||||
|
||||
from bumble import gatt, gatt_adapters, gatt_client, hci
|
||||
from bumble.profiles import le_audio
|
||||
@@ -39,7 +39,7 @@ class PacRecord:
|
||||
'''Published Audio Capabilities Service, Table 3.2/3.4.'''
|
||||
|
||||
coding_format: hci.CodingFormat
|
||||
codec_specific_capabilities: Union[CodecSpecificCapabilities, bytes]
|
||||
codec_specific_capabilities: CodecSpecificCapabilities | bytes
|
||||
metadata: le_audio.Metadata = dataclasses.field(default_factory=le_audio.Metadata)
|
||||
|
||||
@classmethod
|
||||
@@ -56,7 +56,7 @@ class PacRecord:
|
||||
offset += 1
|
||||
metadata = le_audio.Metadata.from_bytes(data[offset : offset + metadata_size])
|
||||
|
||||
codec_specific_capabilities: Union[CodecSpecificCapabilities, bytes]
|
||||
codec_specific_capabilities: CodecSpecificCapabilities | bytes
|
||||
if coding_format.codec_id == hci.CodecID.VENDOR_SPECIFIC:
|
||||
codec_specific_capabilities = codec_specific_capabilities_bytes
|
||||
else:
|
||||
@@ -101,10 +101,10 @@ class PacRecord:
|
||||
class PublishedAudioCapabilitiesService(gatt.TemplateService):
|
||||
UUID = gatt.GATT_PUBLISHED_AUDIO_CAPABILITIES_SERVICE
|
||||
|
||||
sink_pac: Optional[gatt.Characteristic[bytes]]
|
||||
sink_audio_locations: Optional[gatt.Characteristic[bytes]]
|
||||
source_pac: Optional[gatt.Characteristic[bytes]]
|
||||
source_audio_locations: Optional[gatt.Characteristic[bytes]]
|
||||
sink_pac: gatt.Characteristic[bytes] | None
|
||||
sink_audio_locations: gatt.Characteristic[bytes] | None
|
||||
source_pac: gatt.Characteristic[bytes] | None
|
||||
source_audio_locations: gatt.Characteristic[bytes] | None
|
||||
available_audio_contexts: gatt.Characteristic[bytes]
|
||||
supported_audio_contexts: gatt.Characteristic[bytes]
|
||||
|
||||
@@ -115,9 +115,9 @@ class PublishedAudioCapabilitiesService(gatt.TemplateService):
|
||||
available_source_context: ContextType,
|
||||
available_sink_context: ContextType,
|
||||
sink_pac: Sequence[PacRecord] = (),
|
||||
sink_audio_locations: Optional[AudioLocation] = None,
|
||||
sink_audio_locations: AudioLocation | None = None,
|
||||
source_pac: Sequence[PacRecord] = (),
|
||||
source_audio_locations: Optional[AudioLocation] = None,
|
||||
source_audio_locations: AudioLocation | None = None,
|
||||
) -> None:
|
||||
characteristics = []
|
||||
|
||||
@@ -183,14 +183,10 @@ class PublishedAudioCapabilitiesService(gatt.TemplateService):
|
||||
class PublishedAudioCapabilitiesServiceProxy(gatt_client.ProfileServiceProxy):
|
||||
SERVICE_CLASS = PublishedAudioCapabilitiesService
|
||||
|
||||
sink_pac: Optional[gatt_client.CharacteristicProxy[list[PacRecord]]] = None
|
||||
sink_audio_locations: Optional[gatt_client.CharacteristicProxy[AudioLocation]] = (
|
||||
None
|
||||
)
|
||||
source_pac: Optional[gatt_client.CharacteristicProxy[list[PacRecord]]] = None
|
||||
source_audio_locations: Optional[gatt_client.CharacteristicProxy[AudioLocation]] = (
|
||||
None
|
||||
)
|
||||
sink_pac: gatt_client.CharacteristicProxy[list[PacRecord]] | None = None
|
||||
sink_audio_locations: gatt_client.CharacteristicProxy[AudioLocation] | None = None
|
||||
source_pac: gatt_client.CharacteristicProxy[list[PacRecord]] | None = None
|
||||
source_audio_locations: gatt_client.CharacteristicProxy[AudioLocation] | None = None
|
||||
available_audio_contexts: gatt_client.CharacteristicProxy[tuple[ContextType, ...]]
|
||||
supported_audio_contexts: gatt_client.CharacteristicProxy[tuple[ContextType, ...]]
|
||||
|
||||
|
||||
@@ -22,6 +22,7 @@ import enum
|
||||
|
||||
from typing_extensions import Self
|
||||
|
||||
from bumble import core, data_types, gatt
|
||||
from bumble.profiles import le_audio
|
||||
|
||||
|
||||
@@ -46,3 +47,18 @@ class PublicBroadcastAnnouncement:
|
||||
return cls(
|
||||
features=features, metadata=le_audio.Metadata.from_bytes(metadata_ltv)
|
||||
)
|
||||
|
||||
def get_advertising_data(self) -> bytes:
|
||||
return bytes(
|
||||
core.AdvertisingData(
|
||||
[
|
||||
data_types.ServiceData16BitUUID(
|
||||
gatt.GATT_PUBLIC_BROADCAST_ANNOUNCEMENT_SERVICE, bytes(self)
|
||||
)
|
||||
]
|
||||
)
|
||||
)
|
||||
|
||||
def __bytes__(self) -> bytes:
|
||||
metadata_bytes = bytes(self.metadata)
|
||||
return bytes([self.features, len(metadata_bytes)]) + metadata_bytes
|
||||
|
||||
@@ -20,9 +20,9 @@ from __future__ import annotations
|
||||
|
||||
import dataclasses
|
||||
import enum
|
||||
from typing import Sequence
|
||||
from collections.abc import Sequence
|
||||
|
||||
from bumble import att, device, gatt, gatt_adapters, gatt_client, utils
|
||||
from bumble import att, device, gatt, gatt_adapters, gatt_client
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Constants
|
||||
|
||||
@@ -18,7 +18,6 @@
|
||||
# -----------------------------------------------------------------------------
|
||||
import struct
|
||||
from dataclasses import dataclass
|
||||
from typing import Optional
|
||||
|
||||
from bumble import utils
|
||||
from bumble.att import ATT_Error
|
||||
@@ -69,7 +68,7 @@ class ErrorCode(utils.OpenIntEnum):
|
||||
class VolumeOffsetState:
|
||||
volume_offset: int = 0
|
||||
change_counter: int = 0
|
||||
attribute: Optional[Characteristic] = None
|
||||
attribute: Characteristic | None = None
|
||||
|
||||
def __bytes__(self) -> bytes:
|
||||
return struct.pack('<hB', self.volume_offset, self.change_counter)
|
||||
@@ -93,7 +92,7 @@ class VolumeOffsetState:
|
||||
@dataclass
|
||||
class VocsAudioLocation:
|
||||
audio_location: AudioLocation = AudioLocation.NOT_ALLOWED
|
||||
attribute: Optional[Characteristic] = None
|
||||
attribute: Characteristic | None = None
|
||||
|
||||
def __bytes__(self) -> bytes:
|
||||
return struct.pack('<I', self.audio_location)
|
||||
@@ -118,7 +117,6 @@ class VolumeOffsetControlPoint:
|
||||
volume_offset_state: VolumeOffsetState
|
||||
|
||||
async def on_write(self, connection: Connection, value: bytes) -> None:
|
||||
|
||||
opcode = value[0]
|
||||
if opcode != SetVolumeOffsetOpCode.SET_VOLUME_OFFSET:
|
||||
raise ATT_Error(ErrorCode.OPCODE_NOT_SUPPORTED)
|
||||
@@ -148,7 +146,7 @@ class VolumeOffsetControlPoint:
|
||||
@dataclass
|
||||
class AudioOutputDescription:
|
||||
audio_output_description: str = ''
|
||||
attribute: Optional[Characteristic] = None
|
||||
attribute: Characteristic | None = None
|
||||
|
||||
@classmethod
|
||||
def from_bytes(cls, data: bytes):
|
||||
@@ -173,11 +171,10 @@ class VolumeOffsetControlService(TemplateService):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
volume_offset_state: Optional[VolumeOffsetState] = None,
|
||||
audio_location: Optional[VocsAudioLocation] = None,
|
||||
audio_output_description: Optional[AudioOutputDescription] = None,
|
||||
volume_offset_state: VolumeOffsetState | None = None,
|
||||
audio_location: VocsAudioLocation | None = None,
|
||||
audio_output_description: AudioOutputDescription | None = None,
|
||||
) -> None:
|
||||
|
||||
self.volume_offset_state = (
|
||||
VolumeOffsetState() if volume_offset_state is None else volume_offset_state
|
||||
)
|
||||
|
||||
+19
-18
@@ -22,7 +22,8 @@ import collections
|
||||
import dataclasses
|
||||
import enum
|
||||
import logging
|
||||
from typing import TYPE_CHECKING, Callable, Optional, Union
|
||||
from collections.abc import Callable
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from typing_extensions import Self
|
||||
|
||||
@@ -119,7 +120,7 @@ RFCOMM_DYNAMIC_CHANNEL_NUMBER_END = 30
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
def make_service_sdp_records(
|
||||
service_record_handle: int, channel: int, uuid: Optional[UUID] = None
|
||||
service_record_handle: int, channel: int, uuid: UUID | None = None
|
||||
) -> list[sdp.ServiceAttribute]:
|
||||
"""
|
||||
Create SDP records for an RFComm service given a channel number and an
|
||||
@@ -186,7 +187,7 @@ async def find_rfcomm_channels(connection: Connection) -> dict[int, list[UUID]]:
|
||||
)
|
||||
for attribute_lists in search_result:
|
||||
service_classes: list[UUID] = []
|
||||
channel: Optional[int] = None
|
||||
channel: int | None = None
|
||||
for attribute in attribute_lists:
|
||||
# The layout is [[L2CAP_PROTOCOL], [RFCOMM_PROTOCOL, RFCOMM_CHANNEL]].
|
||||
if attribute.id == sdp.SDP_PROTOCOL_DESCRIPTOR_LIST_ATTRIBUTE_ID:
|
||||
@@ -207,7 +208,7 @@ async def find_rfcomm_channels(connection: Connection) -> dict[int, list[UUID]]:
|
||||
# -----------------------------------------------------------------------------
|
||||
async def find_rfcomm_channel_with_uuid(
|
||||
connection: Connection, uuid: str | UUID
|
||||
) -> Optional[int]:
|
||||
) -> int | None:
|
||||
"""Searches an RFCOMM channel associated with given UUID from service records.
|
||||
|
||||
Args:
|
||||
@@ -473,15 +474,15 @@ class DLC(utils.EventEmitter):
|
||||
self.state = DLC.State.INIT
|
||||
self.role = multiplexer.role
|
||||
self.c_r = 1 if self.role == Multiplexer.Role.INITIATOR else 0
|
||||
self.connection_result: Optional[asyncio.Future] = None
|
||||
self.disconnection_result: Optional[asyncio.Future] = None
|
||||
self.connection_result: asyncio.Future | None = None
|
||||
self.disconnection_result: asyncio.Future | None = None
|
||||
self.drained = asyncio.Event()
|
||||
self.drained.set()
|
||||
# Queued packets when sink is not set.
|
||||
self._enqueued_rx_packets: collections.deque[bytes] = collections.deque(
|
||||
maxlen=DEFAULT_RX_QUEUE_SIZE
|
||||
)
|
||||
self._sink: Optional[Callable[[bytes], None]] = None
|
||||
self._sink: Callable[[bytes], None] | None = None
|
||||
|
||||
# Compute the MTU
|
||||
max_overhead = 4 + 1 # header with 2-byte length + fcs
|
||||
@@ -490,11 +491,11 @@ class DLC(utils.EventEmitter):
|
||||
)
|
||||
|
||||
@property
|
||||
def sink(self) -> Optional[Callable[[bytes], None]]:
|
||||
def sink(self) -> Callable[[bytes], None] | None:
|
||||
return self._sink
|
||||
|
||||
@sink.setter
|
||||
def sink(self, sink: Optional[Callable[[bytes], None]]) -> None:
|
||||
def sink(self, sink: Callable[[bytes], None] | None) -> None:
|
||||
self._sink = sink
|
||||
# Dump queued packets to sink
|
||||
if sink:
|
||||
@@ -712,7 +713,7 @@ class DLC(utils.EventEmitter):
|
||||
self.drained.set()
|
||||
|
||||
# Stream protocol
|
||||
def write(self, data: Union[bytes, str]) -> None:
|
||||
def write(self, data: bytes | str) -> None:
|
||||
# We can only send bytes
|
||||
if not isinstance(data, bytes):
|
||||
if isinstance(data, str):
|
||||
@@ -769,10 +770,10 @@ class Multiplexer(utils.EventEmitter):
|
||||
|
||||
EVENT_DLC = "dlc"
|
||||
|
||||
connection_result: Optional[asyncio.Future]
|
||||
disconnection_result: Optional[asyncio.Future]
|
||||
open_result: Optional[asyncio.Future]
|
||||
acceptor: Optional[Callable[[int], Optional[tuple[int, int]]]]
|
||||
connection_result: asyncio.Future | None
|
||||
disconnection_result: asyncio.Future | None
|
||||
open_result: asyncio.Future | None
|
||||
acceptor: Callable[[int], tuple[int, int] | None] | None
|
||||
dlcs: dict[int, DLC]
|
||||
|
||||
def __init__(self, l2cap_channel: l2cap.ClassicChannel, role: Role) -> None:
|
||||
@@ -784,7 +785,7 @@ class Multiplexer(utils.EventEmitter):
|
||||
self.connection_result = None
|
||||
self.disconnection_result = None
|
||||
self.open_result = None
|
||||
self.open_pn: Optional[RFCOMM_MCC_PN] = None
|
||||
self.open_pn: RFCOMM_MCC_PN | None = None
|
||||
self.open_rx_max_credits = 0
|
||||
self.acceptor = None
|
||||
|
||||
@@ -1031,8 +1032,8 @@ class Multiplexer(utils.EventEmitter):
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
class Client:
|
||||
multiplexer: Optional[Multiplexer]
|
||||
l2cap_channel: Optional[l2cap.ClassicChannel]
|
||||
multiplexer: Multiplexer | None
|
||||
l2cap_channel: l2cap.ClassicChannel | None
|
||||
|
||||
def __init__(
|
||||
self, connection: Connection, l2cap_mtu: int = RFCOMM_DEFAULT_L2CAP_MTU
|
||||
@@ -1145,7 +1146,7 @@ class Server(utils.EventEmitter):
|
||||
# Notify
|
||||
self.emit(self.EVENT_START, multiplexer)
|
||||
|
||||
def accept_dlc(self, channel_number: int) -> Optional[tuple[int, int]]:
|
||||
def accept_dlc(self, channel_number: int) -> tuple[int, int] | None:
|
||||
return self.dlc_configs.get(channel_number)
|
||||
|
||||
def on_dlc(self, dlc: DLC) -> None:
|
||||
|
||||
+12
-11
@@ -20,7 +20,8 @@ from __future__ import annotations
|
||||
import asyncio
|
||||
import logging
|
||||
import struct
|
||||
from typing import TYPE_CHECKING, Iterable, NewType, Optional, Sequence, Union
|
||||
from collections.abc import Iterable, Sequence
|
||||
from typing import TYPE_CHECKING, NewType
|
||||
|
||||
from typing_extensions import Self
|
||||
|
||||
@@ -497,7 +498,7 @@ class ServiceAttribute:
|
||||
@staticmethod
|
||||
def find_attribute_in_list(
|
||||
attribute_list: Iterable[ServiceAttribute], attribute_id: int
|
||||
) -> Optional[DataElement]:
|
||||
) -> DataElement | None:
|
||||
return next(
|
||||
(
|
||||
attribute.value
|
||||
@@ -528,7 +529,7 @@ class ServiceAttribute:
|
||||
def to_string(self, with_colors=False):
|
||||
if with_colors:
|
||||
return (
|
||||
f'Attribute(id={color(self.id_name(self.id),"magenta")},'
|
||||
f'Attribute(id={color(self.id_name(self.id), "magenta")},'
|
||||
f'value={self.value})'
|
||||
)
|
||||
|
||||
@@ -778,11 +779,11 @@ class SDP_ServiceSearchAttributeResponse(SDP_PDU):
|
||||
class Client:
|
||||
def __init__(self, connection: Connection, mtu: int = 0) -> None:
|
||||
self.connection = connection
|
||||
self.channel: Optional[l2cap.ClassicChannel] = None
|
||||
self.channel: l2cap.ClassicChannel | None = None
|
||||
self.mtu = mtu
|
||||
self.request_semaphore = asyncio.Semaphore(1)
|
||||
self.pending_request: Optional[SDP_PDU] = None
|
||||
self.pending_response: Optional[asyncio.futures.Future[SDP_PDU]] = None
|
||||
self.pending_request: SDP_PDU | None = None
|
||||
self.pending_response: asyncio.futures.Future[SDP_PDU] | None = None
|
||||
self.next_transaction_id = 0
|
||||
|
||||
async def connect(self) -> None:
|
||||
@@ -898,7 +899,7 @@ class Client:
|
||||
async def search_attributes(
|
||||
self,
|
||||
uuids: Iterable[core.UUID],
|
||||
attribute_ids: Iterable[Union[int, tuple[int, int]]],
|
||||
attribute_ids: Iterable[int | tuple[int, int]],
|
||||
) -> list[list[ServiceAttribute]]:
|
||||
"""
|
||||
Search for attributes by UUID and attribute IDs.
|
||||
@@ -970,7 +971,7 @@ class Client:
|
||||
async def get_attributes(
|
||||
self,
|
||||
service_record_handle: int,
|
||||
attribute_ids: Iterable[Union[int, tuple[int, int]]],
|
||||
attribute_ids: Iterable[int | tuple[int, int]],
|
||||
) -> list[ServiceAttribute]:
|
||||
"""
|
||||
Get attributes for a service.
|
||||
@@ -1042,10 +1043,10 @@ class Client:
|
||||
# -----------------------------------------------------------------------------
|
||||
class Server:
|
||||
CONTINUATION_STATE = bytes([0x01, 0x00])
|
||||
channel: Optional[l2cap.ClassicChannel]
|
||||
channel: l2cap.ClassicChannel | None
|
||||
Service = NewType('Service', list[ServiceAttribute])
|
||||
service_records: dict[int, Service]
|
||||
current_response: Union[None, bytes, tuple[int, list[int]]]
|
||||
current_response: None | bytes | tuple[int, list[int]]
|
||||
|
||||
def __init__(self, device: Device) -> None:
|
||||
self.device = device
|
||||
@@ -1123,7 +1124,7 @@ class Server:
|
||||
self,
|
||||
continuation_state: bytes,
|
||||
transaction_id: int,
|
||||
) -> Optional[bool]:
|
||||
) -> bool | None:
|
||||
# Check if this is a valid continuation
|
||||
if len(continuation_state) > 1:
|
||||
if (
|
||||
|
||||
+20
-30
@@ -27,17 +27,9 @@ from __future__ import annotations
|
||||
import asyncio
|
||||
import enum
|
||||
import logging
|
||||
from collections.abc import Awaitable, Callable
|
||||
from dataclasses import dataclass, field
|
||||
from typing import (
|
||||
TYPE_CHECKING,
|
||||
Any,
|
||||
Awaitable,
|
||||
Callable,
|
||||
ClassVar,
|
||||
Optional,
|
||||
TypeVar,
|
||||
cast,
|
||||
)
|
||||
from typing import TYPE_CHECKING, ClassVar, TypeVar, cast
|
||||
|
||||
from bumble import crypto, utils
|
||||
from bumble.colors import color
|
||||
@@ -213,10 +205,10 @@ class SMP_Command:
|
||||
fields: ClassVar[Fields]
|
||||
code: int = field(default=0, init=False)
|
||||
name: str = field(default='', init=False)
|
||||
_payload: Optional[bytes] = field(default=None, init=False)
|
||||
_payload: bytes | None = field(default=None, init=False)
|
||||
|
||||
@classmethod
|
||||
def from_bytes(cls, pdu: bytes) -> "SMP_Command":
|
||||
def from_bytes(cls, pdu: bytes) -> SMP_Command:
|
||||
code = pdu[0]
|
||||
|
||||
subclass = SMP_Command.smp_classes.get(code)
|
||||
@@ -554,7 +546,7 @@ class OobContext:
|
||||
r: bytes
|
||||
|
||||
def __init__(
|
||||
self, ecc_key: Optional[crypto.EccKey] = None, r: Optional[bytes] = None
|
||||
self, ecc_key: crypto.EccKey | None = None, r: bytes | None = None
|
||||
) -> None:
|
||||
self.ecc_key = crypto.EccKey.generate() if ecc_key is None else ecc_key
|
||||
self.r = crypto.r() if r is None else r
|
||||
@@ -570,7 +562,7 @@ class OobLegacyContext:
|
||||
|
||||
tk: bytes
|
||||
|
||||
def __init__(self, tk: Optional[bytes] = None) -> None:
|
||||
def __init__(self, tk: bytes | None = None) -> None:
|
||||
self.tk = crypto.r() if tk is None else tk
|
||||
|
||||
|
||||
@@ -677,31 +669,31 @@ class Session:
|
||||
self.stk = None
|
||||
self.ltk_ediv = 0
|
||||
self.ltk_rand = bytes(8)
|
||||
self.link_key: Optional[bytes] = None
|
||||
self.link_key: bytes | None = None
|
||||
self.maximum_encryption_key_size: int = 0
|
||||
self.initiator_key_distribution: int = 0
|
||||
self.responder_key_distribution: int = 0
|
||||
self.peer_random_value: Optional[bytes] = None
|
||||
self.peer_random_value: bytes | None = None
|
||||
self.peer_public_key_x: bytes = bytes(32)
|
||||
self.peer_public_key_y = bytes(32)
|
||||
self.peer_ltk = None
|
||||
self.peer_ediv = None
|
||||
self.peer_rand: Optional[bytes] = None
|
||||
self.peer_rand: bytes | None = None
|
||||
self.peer_identity_resolving_key = None
|
||||
self.peer_bd_addr: Optional[Address] = None
|
||||
self.peer_bd_addr: Address | None = None
|
||||
self.peer_signature_key = None
|
||||
self.peer_expected_distributions: list[type[SMP_Command]] = []
|
||||
self.dh_key = b''
|
||||
self.confirm_value = None
|
||||
self.passkey: Optional[int] = None
|
||||
self.passkey: int | None = None
|
||||
self.passkey_ready = asyncio.Event()
|
||||
self.passkey_step = 0
|
||||
self.passkey_display = False
|
||||
self.pairing_method: PairingMethod = PairingMethod.JUST_WORKS
|
||||
self.pairing_config = pairing_config
|
||||
self.wait_before_continuing: Optional[asyncio.Future[None]] = None
|
||||
self.wait_before_continuing: asyncio.Future[None] | None = None
|
||||
self.completed = False
|
||||
self.ctkd_task: Optional[Awaitable[None]] = None
|
||||
self.ctkd_task: Awaitable[None] | None = None
|
||||
|
||||
# Decide if we're the initiator or the responder
|
||||
self.is_initiator = is_initiator
|
||||
@@ -720,7 +712,7 @@ class Session:
|
||||
|
||||
# Create a future that can be used to wait for the session to complete
|
||||
if self.is_initiator:
|
||||
self.pairing_result: Optional[asyncio.Future[None]] = (
|
||||
self.pairing_result: asyncio.Future[None] | None = (
|
||||
asyncio.get_running_loop().create_future()
|
||||
)
|
||||
else:
|
||||
@@ -828,7 +820,7 @@ class Session:
|
||||
def auth_req(self) -> int:
|
||||
return smp_auth_req(self.bonding, self.mitm, self.sc, self.keypress, self.ct2)
|
||||
|
||||
def get_long_term_key(self, rand: bytes, ediv: int) -> Optional[bytes]:
|
||||
def get_long_term_key(self, rand: bytes, ediv: int) -> bytes | None:
|
||||
if not self.sc and not self.completed:
|
||||
if rand == self.ltk_rand and ediv == self.ltk_ediv:
|
||||
return self.stk
|
||||
@@ -939,7 +931,7 @@ class Session:
|
||||
self.pairing_config.delegate.display_number(self.passkey, digits=6)
|
||||
)
|
||||
|
||||
def input_passkey(self, next_steps: Optional[Callable[[], None]] = None) -> None:
|
||||
def input_passkey(self, next_steps: Callable[[], None] | None = None) -> None:
|
||||
# Prompt the user for the passkey displayed on the peer
|
||||
def after_input(passkey: int) -> None:
|
||||
self.passkey = passkey
|
||||
@@ -956,7 +948,7 @@ class Session:
|
||||
self.prompt_user_for_number(after_input)
|
||||
|
||||
def display_or_input_passkey(
|
||||
self, next_steps: Optional[Callable[[], None]] = None
|
||||
self, next_steps: Callable[[], None] | None = None
|
||||
) -> None:
|
||||
if self.passkey_display:
|
||||
|
||||
@@ -1006,7 +998,6 @@ class Session:
|
||||
self.send_command(response)
|
||||
|
||||
def send_pairing_confirm_command(self) -> None:
|
||||
|
||||
if self.pairing_method != PairingMethod.OOB:
|
||||
self.r = crypto.r()
|
||||
logger.debug(f'generated random: {self.r.hex()}')
|
||||
@@ -1842,7 +1833,6 @@ class Session:
|
||||
self.send_public_key_command()
|
||||
|
||||
def next_steps() -> None:
|
||||
|
||||
if self.pairing_method in (
|
||||
PairingMethod.JUST_WORKS,
|
||||
PairingMethod.NUMERIC_COMPARISON,
|
||||
@@ -1929,7 +1919,7 @@ class Manager(utils.EventEmitter):
|
||||
sessions: dict[int, Session]
|
||||
pairing_config_factory: Callable[[Connection], PairingConfig]
|
||||
session_proxy: type[Session]
|
||||
_ecc_key: Optional[crypto.EccKey]
|
||||
_ecc_key: crypto.EccKey | None
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
@@ -2022,7 +2012,7 @@ class Manager(utils.EventEmitter):
|
||||
self.device.on_pairing_start(session.connection)
|
||||
|
||||
async def on_pairing(
|
||||
self, session: Session, identity_address: Optional[Address], keys: PairingKeys
|
||||
self, session: Session, identity_address: Address | None, keys: PairingKeys
|
||||
) -> None:
|
||||
# Store the keys in the key store
|
||||
if self.device.keystore and identity_address is not None:
|
||||
@@ -2041,7 +2031,7 @@ class Manager(utils.EventEmitter):
|
||||
|
||||
def get_long_term_key(
|
||||
self, connection: Connection, rand: bytes, ediv: int
|
||||
) -> Optional[bytes]:
|
||||
) -> bytes | None:
|
||||
if session := self.sessions.get(connection.handle):
|
||||
return session.get_long_term_key(rand, ediv)
|
||||
|
||||
|
||||
+2
-1
@@ -16,13 +16,14 @@ import datetime
|
||||
import logging
|
||||
import os
|
||||
import struct
|
||||
from collections.abc import Generator
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Imports
|
||||
# -----------------------------------------------------------------------------
|
||||
from contextlib import contextmanager
|
||||
from enum import IntEnum
|
||||
from typing import BinaryIO, Generator
|
||||
from typing import BinaryIO
|
||||
|
||||
from bumble import core
|
||||
from bumble.hci import HCI_COMMAND_PACKET, HCI_EVENT_PACKET
|
||||
|
||||
@@ -18,7 +18,6 @@
|
||||
import logging
|
||||
import os
|
||||
import re
|
||||
from typing import Optional
|
||||
|
||||
from bumble import utils
|
||||
from bumble.snoop import create_snooper
|
||||
@@ -111,7 +110,7 @@ async def open_transport(name: str) -> Transport:
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
async def _open_transport(scheme: str, spec: Optional[str]) -> Transport:
|
||||
async def _open_transport(scheme: str, spec: str | None) -> Transport:
|
||||
# pylint: disable=import-outside-toplevel
|
||||
# pylint: disable=too-many-return-statements
|
||||
|
||||
|
||||
@@ -16,7 +16,6 @@
|
||||
# Imports
|
||||
# -----------------------------------------------------------------------------
|
||||
import logging
|
||||
from typing import Optional, Union
|
||||
|
||||
import grpc.aio
|
||||
|
||||
@@ -44,7 +43,7 @@ logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
async def open_android_emulator_transport(spec: Optional[str]) -> Transport:
|
||||
async def open_android_emulator_transport(spec: str | None) -> Transport:
|
||||
'''
|
||||
Open a transport connection to an Android emulator via its gRPC interface.
|
||||
The parameter string has this syntax:
|
||||
@@ -89,7 +88,7 @@ async def open_android_emulator_transport(spec: Optional[str]) -> Transport:
|
||||
logger.debug('connecting to gRPC server at %s', server_address)
|
||||
channel = grpc.aio.insecure_channel(server_address)
|
||||
|
||||
service: Union[EmulatedBluetoothServiceStub, VhciForwardingServiceStub]
|
||||
service: EmulatedBluetoothServiceStub | VhciForwardingServiceStub
|
||||
if mode == 'host':
|
||||
# Connect as a host
|
||||
service = EmulatedBluetoothServiceStub(channel)
|
||||
|
||||
@@ -22,7 +22,6 @@ import os
|
||||
import pathlib
|
||||
import platform
|
||||
import sys
|
||||
from typing import Optional
|
||||
|
||||
import grpc.aio
|
||||
|
||||
@@ -66,7 +65,7 @@ DEFAULT_VARIANT = ''
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
def get_ini_dir() -> Optional[pathlib.Path]:
|
||||
def get_ini_dir() -> pathlib.Path | None:
|
||||
if sys.platform == 'darwin':
|
||||
if tmpdir := os.getenv('TMPDIR', None):
|
||||
return pathlib.Path(tmpdir)
|
||||
@@ -100,7 +99,7 @@ def find_grpc_port(instance_number: int) -> int:
|
||||
ini_file = ini_dir / ini_file_name(instance_number)
|
||||
logger.debug(f'Looking for .ini file at {ini_file}')
|
||||
if ini_file.is_file():
|
||||
with open(ini_file, 'r') as ini_file_data:
|
||||
with open(ini_file) as ini_file_data:
|
||||
for line in ini_file_data.readlines():
|
||||
if '=' in line:
|
||||
key, value = line.split('=')
|
||||
@@ -146,7 +145,7 @@ def publish_grpc_port(grpc_port: int, instance_number: int) -> bool:
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
async def open_android_netsim_controller_transport(
|
||||
server_host: Optional[str], server_port: int, options: dict[str, str]
|
||||
server_host: str | None, server_port: int, options: dict[str, str]
|
||||
) -> Transport:
|
||||
if server_host == '_' or not server_host:
|
||||
server_host = 'localhost'
|
||||
@@ -156,21 +155,26 @@ async def open_android_netsim_controller_transport(
|
||||
logger.warning("unable to publish gRPC port")
|
||||
|
||||
class HciDevice:
|
||||
def __init__(self, context, on_data_received):
|
||||
def __init__(self, context, server):
|
||||
self.context = context
|
||||
self.on_data_received = on_data_received
|
||||
self.server = server
|
||||
self.name = None
|
||||
self.sink = None
|
||||
self.loop = asyncio.get_running_loop()
|
||||
self.done = self.loop.create_future()
|
||||
self.task = self.loop.create_task(self.pump())
|
||||
|
||||
async def pump(self):
|
||||
try:
|
||||
await self.pump_loop()
|
||||
except asyncio.CancelledError:
|
||||
logger.debug('Pump task canceled')
|
||||
if not self.done.done():
|
||||
self.done.set_result(None)
|
||||
finally:
|
||||
if self.sink:
|
||||
logger.debug('Releasing sink')
|
||||
self.server.release_sink()
|
||||
self.sink = None
|
||||
|
||||
logger.debug('Pump task terminated')
|
||||
|
||||
async def pump_loop(self):
|
||||
while True:
|
||||
@@ -186,15 +190,26 @@ async def open_android_netsim_controller_transport(
|
||||
if request.WhichOneof('request_type') == 'initial_info':
|
||||
logger.debug(f'Received initial info: {request}')
|
||||
|
||||
self.name = request.initial_info.name
|
||||
|
||||
# We only accept BLUETOOTH
|
||||
if request.initial_info.chip.kind != ChipKind.BLUETOOTH:
|
||||
logger.warning('Unsupported chip type')
|
||||
error = PacketResponse(error='Unsupported chip type')
|
||||
await self.context.write(error)
|
||||
return
|
||||
# return
|
||||
continue
|
||||
|
||||
self.name = request.initial_info.name
|
||||
continue
|
||||
# Lease the sink so that no other device can send
|
||||
self.sink = self.server.lease_sink(self)
|
||||
if self.sink is None:
|
||||
logger.warning('Another device is already connected')
|
||||
error = PacketResponse(error='Device busy')
|
||||
await self.context.write(error)
|
||||
# return
|
||||
continue
|
||||
|
||||
continue
|
||||
|
||||
# Expect a data packet
|
||||
request_type = request.WhichOneof('request_type')
|
||||
@@ -205,10 +220,10 @@ async def open_android_netsim_controller_transport(
|
||||
continue
|
||||
|
||||
# Process the packet
|
||||
data = (
|
||||
assert self.sink is not None
|
||||
self.sink(
|
||||
bytes([request.hci_packet.packet_type]) + request.hci_packet.packet
|
||||
)
|
||||
self.on_data_received(data)
|
||||
|
||||
async def send_packet(self, data):
|
||||
return await self.context.write(
|
||||
@@ -217,12 +232,6 @@ async def open_android_netsim_controller_transport(
|
||||
)
|
||||
)
|
||||
|
||||
def terminate(self):
|
||||
self.task.cancel()
|
||||
|
||||
async def wait_for_termination(self):
|
||||
await self.done
|
||||
|
||||
server_address = f'{server_host}:{server_port}'
|
||||
|
||||
class Server(PacketStreamerServicer, ParserSource):
|
||||
@@ -258,27 +267,27 @@ async def open_android_netsim_controller_transport(
|
||||
|
||||
return await self.device.send_packet(packet)
|
||||
|
||||
async def StreamPackets(self, _request_iterator, context):
|
||||
def lease_sink(self, device):
|
||||
if self.device:
|
||||
return None
|
||||
self.device = device
|
||||
return self.parser.feed_data
|
||||
|
||||
def release_sink(self):
|
||||
self.device = None
|
||||
|
||||
async def StreamPackets(self, request_iterator, context):
|
||||
logger.debug('StreamPackets request')
|
||||
|
||||
# Check that we don't already have a device
|
||||
if self.device:
|
||||
logger.debug('Busy, already serving a device')
|
||||
return PacketResponse(error='Busy')
|
||||
|
||||
# Instantiate a new device
|
||||
self.device = HciDevice(context, self.parser.feed_data)
|
||||
device = HciDevice(context, self)
|
||||
|
||||
# Wait for the device to terminate
|
||||
logger.debug('Waiting for device to terminate')
|
||||
# Pump packets to/from the device
|
||||
logger.debug('Pumping device packets')
|
||||
try:
|
||||
await self.device.wait_for_termination()
|
||||
except asyncio.CancelledError:
|
||||
logger.debug('Request canceled')
|
||||
self.device.terminate()
|
||||
|
||||
logger.debug('Device terminated')
|
||||
self.device = None
|
||||
await device.pump()
|
||||
finally:
|
||||
logger.debug('Pump terminated')
|
||||
|
||||
server = Server()
|
||||
await server.start()
|
||||
@@ -291,9 +300,9 @@ async def open_android_netsim_controller_transport(
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
async def open_android_netsim_host_transport_with_address(
|
||||
server_host: Optional[str],
|
||||
server_host: str | None,
|
||||
server_port: int,
|
||||
options: Optional[dict[str, str]] = None,
|
||||
options: dict[str, str] | None = None,
|
||||
):
|
||||
if server_host == '_' or not server_host:
|
||||
server_host = 'localhost'
|
||||
@@ -318,7 +327,7 @@ async def open_android_netsim_host_transport_with_address(
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
async def open_android_netsim_host_transport_with_channel(
|
||||
channel, options: Optional[dict[str, str]] = None
|
||||
channel, options: dict[str, str] | None = None
|
||||
):
|
||||
# Wrapper for I/O operations
|
||||
class HciDevice:
|
||||
@@ -398,7 +407,7 @@ async def open_android_netsim_host_transport_with_channel(
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
async def open_android_netsim_transport(spec: Optional[str]) -> Transport:
|
||||
async def open_android_netsim_transport(spec: str | None) -> Transport:
|
||||
'''
|
||||
Open a transport connection as a client or server, implementing Android's `netsim`
|
||||
simulator protocol over gRPC.
|
||||
|
||||
@@ -23,7 +23,7 @@ import io
|
||||
import logging
|
||||
import struct
|
||||
from collections.abc import Awaitable, Callable
|
||||
from typing import Any, ContextManager, Optional, Protocol
|
||||
from typing import Any, Protocol
|
||||
|
||||
from bumble import core, hci
|
||||
from bumble.colors import color
|
||||
@@ -107,11 +107,11 @@ class PacketParser:
|
||||
NEED_LENGTH = 1
|
||||
NEED_BODY = 2
|
||||
|
||||
sink: Optional[TransportSink]
|
||||
sink: TransportSink | None
|
||||
extended_packet_info: dict[int, tuple[int, int, str]]
|
||||
packet_info: Optional[tuple[int, int, str]] = None
|
||||
packet_info: tuple[int, int, str] | None = None
|
||||
|
||||
def __init__(self, sink: Optional[TransportSink] = None) -> None:
|
||||
def __init__(self, sink: TransportSink | None = None) -> None:
|
||||
self.sink = sink
|
||||
self.extended_packet_info = {}
|
||||
self.reset()
|
||||
@@ -176,7 +176,7 @@ class PacketReader:
|
||||
self.source = source
|
||||
self.at_end = False
|
||||
|
||||
def next_packet(self) -> Optional[bytes]:
|
||||
def next_packet(self) -> bytes | None:
|
||||
# Get the packet type
|
||||
packet_type = self.source.read(1)
|
||||
if len(packet_type) != 1:
|
||||
@@ -253,7 +253,7 @@ class BaseSource:
|
||||
"""
|
||||
|
||||
terminated: asyncio.Future[None]
|
||||
sink: Optional[TransportSink]
|
||||
sink: TransportSink | None
|
||||
|
||||
def __init__(self) -> None:
|
||||
self.terminated = asyncio.get_running_loop().create_future()
|
||||
@@ -357,7 +357,7 @@ class Transport:
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
class PumpedPacketSource(ParserSource):
|
||||
pump_task: Optional[asyncio.Task[None]]
|
||||
pump_task: asyncio.Task[None] | None
|
||||
|
||||
def __init__(self, receive) -> None:
|
||||
super().__init__()
|
||||
@@ -390,7 +390,7 @@ class PumpedPacketSource(ParserSource):
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
class PumpedPacketSink:
|
||||
pump_task: Optional[asyncio.Task[None]]
|
||||
pump_task: asyncio.Task[None] | None
|
||||
|
||||
def __init__(self, send: Callable[[bytes], Awaitable[Any]]):
|
||||
self.send_function = send
|
||||
@@ -443,7 +443,7 @@ class SnoopingTransport(Transport):
|
||||
|
||||
@staticmethod
|
||||
def create_with(
|
||||
transport: Transport, snooper: ContextManager[Snooper]
|
||||
transport: Transport, snooper: contextlib.AbstractContextManager[Snooper]
|
||||
) -> SnoopingTransport:
|
||||
"""
|
||||
Create an instance given a snooper that works as as context manager.
|
||||
|
||||
@@ -16,7 +16,6 @@
|
||||
# Imports
|
||||
# -----------------------------------------------------------------------------
|
||||
import asyncio
|
||||
import io
|
||||
import logging
|
||||
|
||||
from bumble.transport.common import StreamPacketSink, StreamPacketSource, Transport
|
||||
@@ -36,7 +35,7 @@ async def open_file_transport(spec: str) -> Transport:
|
||||
'''
|
||||
|
||||
# Open the file
|
||||
file = io.open(spec, 'r+b', buffering=0)
|
||||
file = open(spec, 'r+b', buffering=0)
|
||||
|
||||
# Setup reading
|
||||
read_transport, packet_source = await asyncio.get_running_loop().connect_read_pipe(
|
||||
|
||||
@@ -22,7 +22,6 @@ import logging
|
||||
import os
|
||||
import socket
|
||||
import struct
|
||||
from typing import Optional
|
||||
|
||||
from bumble.transport.common import ParserSource, Transport
|
||||
|
||||
@@ -33,7 +32,7 @@ logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
async def open_hci_socket_transport(spec: Optional[str]) -> Transport:
|
||||
async def open_hci_socket_transport(spec: str | None) -> Transport:
|
||||
'''
|
||||
Open an HCI Socket (only available on some platforms).
|
||||
The parameter string is either empty (to use the first/default Bluetooth adapter)
|
||||
@@ -87,7 +86,7 @@ async def open_hci_socket_transport(spec: Optional[str]) -> Transport:
|
||||
)
|
||||
!= 0
|
||||
):
|
||||
raise IOError(ctypes.get_errno(), os.strerror(ctypes.get_errno()))
|
||||
raise OSError(ctypes.get_errno(), os.strerror(ctypes.get_errno()))
|
||||
|
||||
class HciSocketSource(ParserSource):
|
||||
def __init__(self, hci_socket):
|
||||
|
||||
@@ -17,12 +17,10 @@
|
||||
# -----------------------------------------------------------------------------
|
||||
import asyncio
|
||||
import atexit
|
||||
import io
|
||||
import logging
|
||||
import os
|
||||
import pty
|
||||
import tty
|
||||
from typing import Optional
|
||||
|
||||
from bumble.transport.common import StreamPacketSink, StreamPacketSource, Transport
|
||||
|
||||
@@ -33,7 +31,7 @@ logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
async def open_pty_transport(spec: Optional[str]) -> Transport:
|
||||
async def open_pty_transport(spec: str | None) -> Transport:
|
||||
'''
|
||||
Open a PTY transport.
|
||||
The parameter string may be empty, or a path name where a symbolic link
|
||||
@@ -48,11 +46,11 @@ async def open_pty_transport(spec: Optional[str]) -> Transport:
|
||||
tty.setraw(replica)
|
||||
|
||||
read_transport, packet_source = await asyncio.get_running_loop().connect_read_pipe(
|
||||
StreamPacketSource, io.open(primary, 'rb', closefd=False)
|
||||
StreamPacketSource, open(primary, 'rb', closefd=False)
|
||||
)
|
||||
|
||||
write_transport, _ = await asyncio.get_running_loop().connect_write_pipe(
|
||||
asyncio.BaseProtocol, io.open(primary, 'wb', closefd=False)
|
||||
asyncio.BaseProtocol, open(primary, 'wb', closefd=False)
|
||||
)
|
||||
packet_sink = StreamPacketSink(write_transport)
|
||||
|
||||
|
||||
@@ -19,7 +19,6 @@ import asyncio
|
||||
import logging
|
||||
import threading
|
||||
import time
|
||||
from typing import Optional
|
||||
|
||||
import usb.core
|
||||
import usb.util
|
||||
@@ -284,7 +283,9 @@ async def open_pyusb_transport(spec: str) -> Transport:
|
||||
device = await _power_cycle(device) # type: ignore
|
||||
except Exception as e:
|
||||
logging.debug(e, stack_info=True)
|
||||
logging.info(f"Unable to power cycle {hex(device.idVendor)} {hex(device.idProduct)}") # type: ignore
|
||||
logging.info(
|
||||
f"Unable to power cycle {hex(device.idVendor)} {hex(device.idProduct)}"
|
||||
) # type: ignore
|
||||
|
||||
# Collect the metadata
|
||||
device_metadata = {'vendor_id': device.idVendor, 'product_id': device.idProduct}
|
||||
@@ -370,7 +371,9 @@ async def _power_cycle(device: UsbDevice) -> UsbDevice:
|
||||
# Device needs to be find again otherwise it will appear as disconnected
|
||||
return usb.core.find(idVendor=device.idVendor, idProduct=device.idProduct) # type: ignore
|
||||
except USBError:
|
||||
logger.exception(f"Adjustment needed: Please revise the udev rule for device {hex(device.idVendor)}:{hex(device.idProduct)} for proper recognition.") # type: ignore
|
||||
logger.exception(
|
||||
f"Adjustment needed: Please revise the udev rule for device {hex(device.idVendor)}:{hex(device.idProduct)} for proper recognition."
|
||||
) # type: ignore
|
||||
|
||||
return device
|
||||
|
||||
@@ -385,7 +388,7 @@ def _set_port_status(device: UsbDevice, port: int, on: bool):
|
||||
)
|
||||
|
||||
|
||||
def _find_device_by_path(sys_path: str) -> Optional[UsbDevice]:
|
||||
def _find_device_by_path(sys_path: str) -> UsbDevice | None:
|
||||
"""Finds a USB device based on its system path."""
|
||||
bus_num, *port_parts = sys_path.split('-')
|
||||
ports = [int(port) for port in port_parts[0].split('.')]
|
||||
@@ -398,7 +401,7 @@ def _find_device_by_path(sys_path: str) -> Optional[UsbDevice]:
|
||||
return None
|
||||
|
||||
|
||||
def _find_hub_by_device_path(sys_path: str) -> Optional[UsbDevice]:
|
||||
def _find_hub_by_device_path(sys_path: str) -> UsbDevice | None:
|
||||
"""Finds the USB hub associated with a specific device path."""
|
||||
hub_sys_path = sys_path.rsplit('.', 1)[0]
|
||||
hub_device = _find_device_by_path(hub_sys_path)
|
||||
|
||||
@@ -17,7 +17,6 @@
|
||||
# -----------------------------------------------------------------------------
|
||||
import asyncio
|
||||
import logging
|
||||
from typing import Optional
|
||||
|
||||
import serial_asyncio
|
||||
|
||||
@@ -52,7 +51,7 @@ class SerialPacketSource(StreamPacketSource):
|
||||
logger.debug('connection made')
|
||||
self._ready.set()
|
||||
|
||||
def connection_lost(self, exc: Optional[Exception]) -> None:
|
||||
def connection_lost(self, exc: Exception | None) -> None:
|
||||
logger.debug('connection lost')
|
||||
self.on_transport_lost()
|
||||
|
||||
|
||||
@@ -16,7 +16,6 @@
|
||||
# Imports
|
||||
# -----------------------------------------------------------------------------
|
||||
import logging
|
||||
from typing import Optional
|
||||
|
||||
from bumble.transport.common import Transport
|
||||
from bumble.transport.file import open_file_transport
|
||||
@@ -28,7 +27,7 @@ logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
async def open_vhci_transport(spec: Optional[str]) -> Transport:
|
||||
async def open_vhci_transport(spec: str | None) -> Transport:
|
||||
'''
|
||||
Open a VHCI transport (only available on some platforms).
|
||||
The parameter string is either empty (to use the default VHCI device
|
||||
|
||||
@@ -16,7 +16,6 @@
|
||||
# Imports
|
||||
# -----------------------------------------------------------------------------
|
||||
import logging
|
||||
from typing import Optional
|
||||
|
||||
import websockets.asyncio.server
|
||||
|
||||
@@ -43,8 +42,8 @@ async def open_ws_server_transport(spec: str) -> Transport:
|
||||
class WsServerTransport(Transport):
|
||||
sink: PumpedPacketSink
|
||||
source: ParserSource
|
||||
connection: Optional[websockets.asyncio.server.ServerConnection]
|
||||
server: Optional[websockets.asyncio.server.Server]
|
||||
connection: websockets.asyncio.server.ServerConnection | None
|
||||
server: websockets.asyncio.server.Server | None
|
||||
|
||||
def __init__(self) -> None:
|
||||
source = ParserSource()
|
||||
|
||||
+22
-9
@@ -22,16 +22,12 @@ import collections
|
||||
import enum
|
||||
import functools
|
||||
import logging
|
||||
import sys
|
||||
import warnings
|
||||
from collections.abc import Awaitable, Callable
|
||||
from typing import (
|
||||
Any,
|
||||
Awaitable,
|
||||
Callable,
|
||||
Optional,
|
||||
Protocol,
|
||||
TypeVar,
|
||||
Union,
|
||||
overload,
|
||||
)
|
||||
|
||||
@@ -170,8 +166,8 @@ class EventWatcher:
|
||||
) -> _Handler: ...
|
||||
|
||||
def on(
|
||||
self, emitter: pyee.EventEmitter, event: str, handler: Optional[_Handler] = None
|
||||
) -> Union[_Handler, Callable[[_Handler], _Handler]]:
|
||||
self, emitter: pyee.EventEmitter, event: str, handler: _Handler | None = None
|
||||
) -> _Handler | Callable[[_Handler], _Handler]:
|
||||
'''Watch an event until the context is closed.
|
||||
|
||||
Args:
|
||||
@@ -199,8 +195,8 @@ class EventWatcher:
|
||||
) -> _Handler: ...
|
||||
|
||||
def once(
|
||||
self, emitter: pyee.EventEmitter, event: str, handler: Optional[_Handler] = None
|
||||
) -> Union[_Handler, Callable[[_Handler], _Handler]]:
|
||||
self, emitter: pyee.EventEmitter, event: str, handler: _Handler | None = None
|
||||
) -> _Handler | Callable[[_Handler], _Handler]:
|
||||
'''Watch an event for once.
|
||||
|
||||
Args:
|
||||
@@ -533,3 +529,20 @@ class IntConvertible(Protocol):
|
||||
|
||||
def __init__(self, value: int) -> None: ...
|
||||
def __int__(self) -> int: ...
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
def crc_16(data: bytes) -> int:
|
||||
"""Calculate CRC-16-IBM of given data.
|
||||
|
||||
Polynomial = x^16 + x^15 + x^2 + 1 = 0x8005 or 0xA001(Reversed)
|
||||
"""
|
||||
crc = 0x0000
|
||||
for byte in data:
|
||||
crc ^= byte
|
||||
for _ in range(8):
|
||||
if (crc & 0x0001) > 0:
|
||||
crc = (crc >> 1) ^ 0xA001
|
||||
else:
|
||||
crc = crc >> 1
|
||||
return crc
|
||||
|
||||
Vendored
+4
-2
@@ -18,7 +18,6 @@
|
||||
import dataclasses
|
||||
import struct
|
||||
from dataclasses import field
|
||||
from typing import Optional
|
||||
|
||||
from bumble import hci
|
||||
|
||||
@@ -51,6 +50,7 @@ class HCI_LE_Get_Vendor_Capabilities_Command(hci.HCI_Command):
|
||||
'''
|
||||
See https://source.android.com/docs/core/connect/bluetooth/hci_requirements#vendor-specific-capabilities
|
||||
'''
|
||||
|
||||
return_parameters_fields = [
|
||||
('status', hci.STATUS_SPEC),
|
||||
('max_advt_instances', 1),
|
||||
@@ -137,6 +137,7 @@ class HCI_Get_Controller_Activity_Energy_Info_Command(hci.HCI_Command):
|
||||
'''
|
||||
See https://source.android.com/docs/core/connect/bluetooth/hci_requirements#le_get_controller_activity_energy_info
|
||||
'''
|
||||
|
||||
return_parameters_fields = [
|
||||
('status', hci.STATUS_SPEC),
|
||||
('total_tx_time_ms', 4),
|
||||
@@ -207,7 +208,7 @@ class HCI_Android_Vendor_Event(hci.HCI_Extended_Event):
|
||||
@classmethod
|
||||
def subclass_from_parameters(
|
||||
cls, parameters: bytes
|
||||
) -> Optional[hci.HCI_Extended_Event]:
|
||||
) -> hci.HCI_Extended_Event | None:
|
||||
subevent_code = parameters[0]
|
||||
if subevent_code == HCI_BLUETOOTH_QUALITY_REPORT_EVENT:
|
||||
quality_report_id = parameters[1]
|
||||
@@ -229,6 +230,7 @@ class HCI_Bluetooth_Quality_Report_Event(HCI_Android_Vendor_Event):
|
||||
'''
|
||||
See https://source.android.com/docs/core/connect/bluetooth/hci_requirements#bluetooth-quality-report-sub-event
|
||||
'''
|
||||
|
||||
quality_report_id: int = field(metadata=hci.metadata(1))
|
||||
packet_types: int = field(metadata=hci.metadata(1))
|
||||
connection_handle: int = field(metadata=hci.metadata(2))
|
||||
|
||||
@@ -2,5 +2,5 @@
|
||||
mkdocs == 1.4.0
|
||||
mkdocs-material == 8.5.6
|
||||
mkdocs-material-extensions == 1.0.3
|
||||
pymdown-extensions == 10.0
|
||||
pymdown-extensions == 10.16.1
|
||||
mkdocstrings-python == 0.7.1
|
||||
|
||||
@@ -104,7 +104,8 @@ The `--output` option specifies where to send the decoded audio samples.
|
||||
The following outputs are supported:
|
||||
|
||||
### Sound Device
|
||||
The `--output` argument is either `device`, to send the audio to the hosts's default sound device, or `device:<DEVICE_ID>` where `<DEVICE_ID>`
|
||||
The `--output` argument is either `device`, to send the audio to the hosts's
|
||||
default sound device, or `device:<DEVICE_ID>` where `<DEVICE_ID>`
|
||||
is the integer ID of one of the available sound devices.
|
||||
When invoked with `--output "device:?"`, a list of available devices and
|
||||
their IDs is printed out.
|
||||
@@ -115,17 +116,24 @@ standard output (currently always as float32 PCM samples)
|
||||
|
||||
### FFPlay
|
||||
With `--output ffplay`, the decoded audio samples are piped to `ffplay`
|
||||
in a child process. This option is only available if `ffplay` is a command that is available on the host.
|
||||
in a child process. This option is only available if `ffplay` is a command
|
||||
that is available on the host.
|
||||
|
||||
### File
|
||||
With `--output <filename>` or `--output file:<filename>`, the decoded audio
|
||||
samples are written to a file (currently always as float32 PCM)
|
||||
|
||||
## `transmit`
|
||||
Broadcast an audio source as a transmitter.
|
||||
Broadcast one or more audio sources as a transmitter.
|
||||
|
||||
The `--input` and `--input-format` options specify what audio input
|
||||
source to transmit.
|
||||
|
||||
Optionally, you can use the `--broadcast-list` option,
|
||||
specifying a TOML configuration file, as a convenient way to specify
|
||||
audio source configurations for one or more audio sources.
|
||||
See `examples/auracast_broadcasts.toml` for an example.
|
||||
|
||||
The following inputs are supported:
|
||||
|
||||
### Sound Device
|
||||
@@ -146,7 +154,8 @@ are read from a .wav or raw PCM file.
|
||||
Use the `--input-format <FORMAT>` option to specify the format of the audio
|
||||
samples in raw PCM files. `<FORMAT>` is expressed as:
|
||||
`<sample-type>,<sample-rate>,<channels>`
|
||||
(the only supported <sample-type> currently is 'int16le' for 16 bit signed integers with little-endian byte order)
|
||||
(the only supported <sample-type> currently is 'int16le' for 16 bit signed
|
||||
integers with little-endian byte order)
|
||||
|
||||
## `scan`
|
||||
Scan for public broadcasts.
|
||||
@@ -164,6 +173,7 @@ be shared to allow better compatibiity with certain products.
|
||||
The `receive` command has been tested to successfully receive broadcasts from
|
||||
the following transmitters:
|
||||
|
||||
* Android's "Audio Sharing"
|
||||
* JBL GO 4
|
||||
* Flairmesh FlooGoo FMA120
|
||||
* Eppfun AK3040Pro Max
|
||||
@@ -193,10 +203,12 @@ Use the `--manufacturer-data` option of the `transmit` command in order to inclu
|
||||
that will let the speaker recognize the broadcast as a compatible source.
|
||||
|
||||
The manufacturer ID for JBL is 87.
|
||||
Using an option like `--manufacturer-data 87:00000000000000000000000000000000dffd` should work (tested on the
|
||||
JBL GO 4. The `dffd` value at the end of the payload may be different on other models?).
|
||||
Using an option like `--manufacturer-data 87:00000000000000000000000000000000dffd` should work
|
||||
(tested on the JBL GO 4.
|
||||
The `dffd` value at the end of the payload may be different on other models?).
|
||||
|
||||
|
||||
### Others
|
||||
|
||||
* Android
|
||||
* Nexum Audio VOCE and USB dongle
|
||||
@@ -13,7 +13,11 @@ It is also possible to use an embedded development board, running a specialized
|
||||
|
||||
Some specific USB dongles and embedded boards that are known to work include:
|
||||
|
||||
* [Nordic nRF52840 DK board](https://www.nordicsemi.com/Products/Development-hardware/nrf52840-dk) with the Zephyr `HCI UART` application
|
||||
* [Nordic nRF52840 DK board](https://www.nordicsemi.com/Products/Development-hardware/nrf52840-dk) with the mynewt `blehci` application
|
||||
* [Nordic nrf52840 dongle](https://www.nordicsemi.com/Products/Development-hardware/nRF52840-Dongle) with the Zephyr `HCI USB` application
|
||||
* [BT820 USB Dongle](https://www.lairdconnect.com/wireless-modules/bluetooth-modules/bluetooth-42-and-40-modules/bt800-series-bluetooth-module)
|
||||
* [Nordic nRF52840 DK board](https://www.nordicsemi.com/Products/Development-hardware/nrf52840-dk) with the Zephyr `HCI UART` application
|
||||
* [Nordic nRF52840 DK board](https://www.nordicsemi.com/Products/Development-hardware/nrf52840-dk) with the mynewt `blehci` application
|
||||
* [Nordic nrf52840 dongle](https://www.nordicsemi.com/Products/Development-hardware/nRF52840-Dongle) with the Zephyr `HCI USB` application
|
||||
* [BT820 USB Dongle](https://www.lairdconnect.com/wireless-modules/bluetooth-modules/bluetooth-42-and-40-modules/bt800-series-bluetooth-module)
|
||||
|
||||
However, many Bluetooth controllers do not fully implement the complete set of Bluetooth specifications. To help us improve compatibility and expand support, we are collecting data on user hardware.
|
||||
|
||||
We would greatly appreciate it if you could share the capabilities and purchase details for any controllers you have available for testing. We will later build a capability table here.
|
||||
|
||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user