auracaster/bumble_mirror
3
0
Fork 1
mirror of https://github.com/google/bumble.git synced 2026-05-08 03:58:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

smp: add simple Session proxy

Browse Source 
This allow modifying the SMP behavior at runtime for testing purpose.
This commit is contained in:
uael
2023-05-02 05:31:52 +00:00
parent 3c81b248a3
commit 5d4f811a65
2 changed files with 17 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
bumble/device.py
View File

@@ -23,7 +23,7 @@ import asyncio
import logging import logging
from contextlib import asynccontextmanager, AsyncExitStack from contextlib import asynccontextmanager, AsyncExitStack
from dataclasses import dataclass from dataclasses import dataclass
from typing import Any, Callable, ClassVar, Dict, List, Optional, Tuple, Union from typing import Any, Callable, ClassVar, Dict, List, Optional, Tuple, Type, Union
from .colors import color from .colors import color
from .att import ATT_CID, ATT_DEFAULT_MTU, ATT_PDU from .att import ATT_CID, ATT_DEFAULT_MTU, ATT_PDU
@@ -2206,6 +2206,14 @@ class Device(CompositeEventEmitter):
) -> None: ) -> None:
self.smp_manager.pairing_config_factory = pairing_config_factory self.smp_manager.pairing_config_factory = pairing_config_factory
@property
def smp_session_proxy(self) -> Type[smp.Session]:
return self.smp_manager.session_proxy
@smp_session_proxy.setter
def smp_session_proxy(self, session_proxy: Type[smp.Session]) -> None:
self.smp_manager.session_proxy = session_proxy
async def pair(self, connection): async def pair(self, connection):
return await self.smp_manager.pair(connection) return await self.smp_manager.pair(connection)

10
bumble/smp.py
View File

@@ -1711,6 +1711,7 @@ class Manager(EventEmitter):
device: Device device: Device
sessions: Dict[int, Session] sessions: Dict[int, Session]
pairing_config_factory: Callable[[Connection], PairingConfig] pairing_config_factory: Callable[[Connection], PairingConfig]
session_proxy: Type[Session]
def __init__( def __init__(
self, self,
@@ -1722,6 +1723,7 @@ class Manager(EventEmitter):
self.sessions = {} self.sessions = {}
self._ecc_key = None self._ecc_key = None
self.pairing_config_factory = pairing_config_factory self.pairing_config_factory = pairing_config_factory
self.session_proxy = Session
def send_command(self, connection: Connection, command: SMP_Command) -> None: def send_command(self, connection: Connection, command: SMP_Command) -> None:
logger.debug( logger.debug(
@@ -1737,7 +1739,9 @@ class Manager(EventEmitter):
if connection.role == BT_CENTRAL_ROLE: if connection.role == BT_CENTRAL_ROLE:
logger.warning('Remote starts pairing as Peripheral!') logger.warning('Remote starts pairing as Peripheral!')
pairing_config = self.pairing_config_factory(connection) pairing_config = self.pairing_config_factory(connection)
session = Session(self, connection, pairing_config, is_initiator=False) session = self.session_proxy(
self, connection, pairing_config, is_initiator=False
)
self.sessions[connection.handle] = session self.sessions[connection.handle] = session
# Parse the L2CAP payload into an SMP Command object # Parse the L2CAP payload into an SMP Command object
@@ -1762,7 +1766,9 @@ class Manager(EventEmitter):
if connection.role != BT_CENTRAL_ROLE: if connection.role != BT_CENTRAL_ROLE:
logger.warning('Start pairing as Peripheral!') logger.warning('Start pairing as Peripheral!')
pairing_config = self.pairing_config_factory(connection) pairing_config = self.pairing_config_factory(connection)
session = Session(self, connection, pairing_config, is_initiator=True) session = self.session_proxy(
self, connection, pairing_config, is_initiator=True
)
self.sessions[connection.handle] = session self.sessions[connection.handle] = session
return await session.pair() return await session.pair()