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

classic: add to .encrypt the possibilty to disable encryption

Browse Source
This commit is contained in:
Abel Lucas
2022-12-01 22:11:22 +00:00
parent 441933bd64
commit 5178c866ac
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
bumble/device.py
View File

@@ -523,8 +523,8 @@ class Connection(CompositeEventEmitter):
async def authenticate(self):
return await self.device.authenticate(self)
async def encrypt(self):
return await self.device.encrypt(self)
async def encrypt(self, enable=True):
return await self.device.encrypt(self, enable)
async def sustain(self, timeout=None):
"""Idles the current task waiting for a disconnect or timeout"""
@@ -1999,7 +1999,10 @@ class Device(CompositeEventEmitter):
'connection_authentication_failure', on_authentication_failure
)
async def encrypt(self, connection):
async def encrypt(self, connection, enable=True):
if not enable and connection.transport == BT_LE_TRANSPORT:
raise ValueError('`enable` parameter is classic only.')
# Set up event handlers
pending_encryption = asyncio.get_running_loop().create_future()
@@ -2054,7 +2057,7 @@ class Device(CompositeEventEmitter):
else:
result = await self.send_command(
HCI_Set_Connection_Encryption_Command(
connection_handle=connection.handle, encryption_enable=0x01
connection_handle=connection.handle, encryption_enable=0x01 if enable else 0x00
)
)