pstruebi/bumble_fork
1
0
Fork 0
forked from auracaster/bumble_mirror
Code Pull Requests Activity

Renamed the status message class

Browse Source
This commit is contained in:
dhavan
2023-11-22 17:14:24 +00:00
parent dc410b14c4
commit d6cefdff8e
2 changed files with 25 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
bumble/hid.py
View File

@@ -298,7 +298,6 @@ class HID(EventEmitter):
def on_ctrl_pdu(self, pdu: bytes) -> None:
logger.debug(f'<<< HID CONTROL PDU: {pdu.hex()}')
# Here we will receive all kinds of packets, parse and then call respective callbacks
param = pdu[0] & 0x0F
message_type = pdu[0] >> 4
@@ -369,7 +368,7 @@ class HID(EventEmitter):
class Device(HID):
class ReportStatus(enum.IntEnum):
class GetSetReturn(enum.IntEnum):
FAILURE = 0x00
REPORT_ID_NOT_FOUND = 0x01
ERR_UNSUPPORTED_REQUEST = 0x02
@@ -377,7 +376,7 @@ class Device(HID):
SUCCESS = 0xff
class GetReportStatus():
class GetSetStatus():
def __init__(self) -> None:
self.status = 0
self.data=None
@@ -398,7 +397,7 @@ class Device(HID):
self.send_pdu_on_ctrl(hid_message)
def handle_get_report(self, pdu: bytes):
ret = self.GetReportStatus()
ret = self.GetSetStatus()
report_type=pdu[0] & 0x03
buffer_flag = (pdu[0] & 0x08) >> 3
report_id = pdu[1]
@@ -411,18 +410,18 @@ class Device(HID):
if(self.get_report_cb != None):
ret = self.get_report_cb(report_id, report_type, buffer_size)
if(ret.status == self.ReportStatus.FAILURE):
if(ret.status == self.GetSetReturn.FAILURE):
self.send_handshake_message(Message.Handshake.ERR_UNKNOWN)
elif(ret.status == self.ReportStatus.SUCCESS):
elif(ret.status == self.GetSetReturn.SUCCESS):
data = bytearray()
data.append(report_id)
data.extend(ret.data)
#TODO Check the data size and MTU size here and only then send out
#the message
self.send_control_data(report_type=report_type, data = data)
elif(ret.status == self.ReportStatus.REPORT_ID_NOT_FOUND):
elif(ret.status == self.GetSetReturn.REPORT_ID_NOT_FOUND):
self.send_handshake_message(Message.Handshake.ERR_INVALID_REPORT_ID)
elif(ret.status == self.ReportStatus.ERR_UNSUPPORTED_REQUEST):
elif(ret.status == self.GetSetReturn.ERR_UNSUPPORTED_REQUEST):
self.send_handshake_message(Message.Handshake.ERR_UNSUPPORTED_REQUEST)
else:
logger.debug("GetReport callback not registered !!")
@@ -438,7 +437,7 @@ class Device(HID):
report_id = pdu[1]
report_data = pdu[2:]
ret = self.set_report_cb(report_id, report_type, report_data)
if(ret.status == self.ReportStatus.SUCCESS):
if(ret.status == self.GetSetReturn.SUCCESS):
self.send_handshake_message(Message.Handshake.SUCCESSFUL)
return
else:
@@ -451,10 +450,10 @@ class Device(HID):
logger.debug("SetReport callback registered successfully")
def handle_get_protocol(self, pdu: bytes):
ret = self.GetReportStatus()
ret = self.GetSetStatus()
if(self.get_protocol_cb != None):
ret=self.get_protocol_cb()
if(ret.status == self.ReportStatus.SUCCESS):
if(ret.status == self.GetSetReturn.SUCCESS):
self.send_control_data(Message.ReportType.OTHER_REPORT, ret.data)
return
else:
@@ -467,10 +466,10 @@ class Device(HID):
logger.debug("GetProtocol callback registered successfully")
def handle_set_protocol(self, pdu: bytes):
ret = self.GetReportStatus()
ret = self.GetSetStatus()
if(self.set_protocol_cb != None):
ret=self.set_protocol_cb(pdu[0] & 0x01)
if(ret.status == self.ReportStatus.SUCCESS):
if(ret.status == self.GetSetReturn.SUCCESS):
return
else:
logger.debug("SetProtocol callback not registered !!")