236 lines
11 KiB
Python
236 lines
11 KiB
Python
"""
|
|
Bluetooth-HID
|
|
|
|
Micropython library for the Bluetooth-HID(Compliant with equipment)
|
|
=======================================================
|
|
@dahanzimin From the Mixly Team
|
|
"""
|
|
import bluetooth
|
|
import struct, time
|
|
from micropython import const
|
|
from ubinascii import hexlify
|
|
from ble_advertising import advertising_payload
|
|
from bluetooth import UUID, FLAG_READ, FLAG_WRITE ,FLAG_NOTIFY, FLAG_WRITE_NO_RESPONSE
|
|
|
|
_IRQ_CENTRAL_CONNECT = const(1)
|
|
_IRQ_CENTRAL_DISCONNECT = const(2)
|
|
_IRQ_GATTS_WRITE = const(3)
|
|
_IRQ_MTU_EXCHANGED = const(21)
|
|
_IRQ_CONNECTION_UPDATE = const(27)
|
|
_IRQ_GET_SECRET = const(29)
|
|
_IRQ_SET_SECRET = const(30)
|
|
_IRQ_PASSKEY_ACTION = const(31)
|
|
_PASSKEY_ACTION_INPUT = const(2)
|
|
_PASSKEY_ACTION_DISP = const(3)
|
|
_PASSKEY_ACTION_NUMCMP = const(4)
|
|
|
|
#HID 鼠标、键盘、手柄设备报告描述符
|
|
_HID_INPUT_REPORT = const(b'\x05\x01\t\x02\xa1\x01\x85\x01\t\x01\xa1\x00\x05\t\x19\x01)\x03\x15\x00%\x01\x95\x03u\x01\x81\x02\x95\x01u\x05\x81\x03\x05\x01\t0\t1\t8\x15\x81%\x7fu\x08\x95\x03\x81\x06\xc0\xc0\x05\x01\t\x06\xa1\x01\x85\x02u\x01\x95\x08\x05\x07\x19\xe0)\xe7\x15\x00%\x01\x81\x02\x95\x01u\x08\x81\x01\x95\x05u\x01\x05\x08\x19\x01)\x05\x91\x02\x95\x01u\x03\x91\x01\x95\x06u\x08\x15\x00%e\x05\x07\x19\x00)e\x81\x00\xc0\x05\x01\t\x04\xa1\x01\x85\x03\xa1\x00\t0\t1\x15\x81%\x7fu\x08\x95\x02\x81\x02\x05\t)\x08\x19\x01\x95\x08u\x01%\x01\x15\x00\x81\x02\xc0\xc0')
|
|
_KEYCODE = const(b'\x00\x00\x00\x00\x00\x00\x00\x00*+(\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00)\x00\x00\x00\x00,\x9e\xb4\xa0\xa1\xa2\xa44\xa6\xa7\xa5\xae6-78\'\x1e\x1f !"#$%&\xb33\xb6.\xb7\xb8\x9f\x84\x85\x86\x87\x88\x89\x8a\x8b\x8c\x8d\x8e\x8f\x90\x91\x92\x93\x94\x95\x96\x97\x98\x99\x9a\x9b\x9c\x9d/10\xa3\xad5\x04\x05\x06\x07\x08\t\n\x0b\x0c\r\x0e\x0f\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1a\x1b\x1c\x1d\xaf\xb1\xb0\xb5L')
|
|
|
|
_DIS = (UUID(0x180A), ( (UUID(0x2A24), FLAG_READ),
|
|
(UUID(0x2A25), FLAG_READ),
|
|
(UUID(0x2A26), FLAG_READ),
|
|
(UUID(0x2A27), FLAG_READ),
|
|
(UUID(0x2A28), FLAG_READ),
|
|
(UUID(0x2A29), FLAG_READ),
|
|
(UUID(0x2A50), FLAG_READ), ), )
|
|
|
|
_BAS = (UUID(0x180F), ( (UUID(0x2A19), FLAG_READ | FLAG_NOTIFY, ((UUID(0x2902), 0x03), (UUID(0x2904), 0x01),)), ), )
|
|
|
|
_HIDS = (UUID(0x1812), ((UUID(0x2A4A), FLAG_READ),
|
|
(UUID(0x2A4B), FLAG_READ),
|
|
(UUID(0x2A4C), FLAG_READ | FLAG_WRITE | FLAG_WRITE_NO_RESPONSE),
|
|
(UUID(0x2A4D), FLAG_READ | FLAG_NOTIFY, ((UUID(0x2902), 0x03), (UUID(0x2908), 0x03),)), #鼠标
|
|
(UUID(0x2A4D), FLAG_READ | FLAG_NOTIFY, ((UUID(0x2902), 0x03), (UUID(0x2908), 0x03),)), #键盘发送
|
|
(UUID(0x2A4D), FLAG_READ | FLAG_WRITE, ((UUID(0x2902), 0x03), (UUID(0x2908), 0x03),)), #键盘状态
|
|
(UUID(0x2A4D), FLAG_READ | FLAG_NOTIFY, ((UUID(0x2902), 0x03), (UUID(0x2908), 0x03),)), #游戏手柄
|
|
(UUID(0x2A4E), FLAG_READ | FLAG_WRITE | FLAG_WRITE_NO_RESPONSE), ), )
|
|
|
|
class HID:
|
|
def __init__(self, name=None, passkey=1234, battery_level=100):
|
|
if (name is '') or (name is None):
|
|
name = "Mixgo_" + self.mac[-6:].upper()
|
|
print("HID Name:", name)
|
|
|
|
self._ble = bluetooth.BLE()
|
|
self._ble.active(True)
|
|
self._ble.irq(self._irq)
|
|
self._ble.config(gap_name=name)
|
|
self._ble.config(mtu=23)
|
|
self.device_state = False
|
|
self.conn_handle = None
|
|
self.passkey = passkey
|
|
self.battery_level = battery_level
|
|
self.report = b'\x00'
|
|
try:
|
|
import ble_hid_key
|
|
self.keys = ble_hid_key.keys
|
|
except:
|
|
self.keys = {}
|
|
|
|
handles = self._ble.gatts_register_services((_DIS, _BAS, _HIDS))
|
|
self._service_characteristics(handles)
|
|
self._payload = advertising_payload(name=name, services=[UUID(0x1812)], appearance=const(960))
|
|
self.advertise()
|
|
|
|
def _irq(self, event, data):
|
|
# Interrupt request callback function
|
|
if event == _IRQ_CENTRAL_CONNECT:
|
|
self.conn_handle, _, _ = data
|
|
print("HID connected:", self.conn_handle)
|
|
self.device_state = True
|
|
elif event == _IRQ_CENTRAL_DISCONNECT:
|
|
self.conn_handle = None
|
|
conn_handle, addr_type, addr = data
|
|
print("HID disconnected:", conn_handle)
|
|
self.advertise()
|
|
self.device_state = False
|
|
elif event == _IRQ_MTU_EXCHANGED:
|
|
conn_handle, mtu = data
|
|
self._ble.config(mtu=mtu)
|
|
print("MTU exchanged:", mtu)
|
|
elif event == _IRQ_CONNECTION_UPDATE:
|
|
self.conn_handle, _, _, _, _ = data
|
|
print("Connection update")
|
|
elif event == _IRQ_PASSKEY_ACTION:
|
|
conn_handle, action, passkey = data
|
|
print("Passkey action", conn_handle, action, passkey)
|
|
if action == _PASSKEY_ACTION_NUMCMP:
|
|
self._ble.gap_passkey(conn_handle, action, False)
|
|
elif action == _PASSKEY_ACTION_DISP:
|
|
print("Displaying passkey")
|
|
self._ble.gap_passkey(conn_handle, action, self.passkey)
|
|
elif action == _PASSKEY_ACTION_INPUT:
|
|
print("Prompting for passkey")
|
|
self._ble.gap_passkey(conn_handle, action, None)
|
|
else:
|
|
print("Unknown action")
|
|
elif event == _IRQ_GATTS_WRITE:
|
|
conn_handle, attr_handle = data
|
|
self.report = self._ble.gatts_read(attr_handle)
|
|
elif event == _IRQ_SET_SECRET:
|
|
sec_type, key, value = data
|
|
key = sec_type, bytes(key)
|
|
value = bytes(value) if value else None
|
|
#print("Set secret: ", key, value)
|
|
if value is None:
|
|
if key in self.keys:
|
|
del self.keys[key]
|
|
self._key_secrets(self.keys)
|
|
return True
|
|
else:
|
|
return False
|
|
else:
|
|
self.keys[key] = value
|
|
self._key_secrets(self.keys)
|
|
return True
|
|
elif event == _IRQ_GET_SECRET:
|
|
sec_type, index, key = data
|
|
#print("Get secret: ", sec_type, index, bytes(key) if key else None)
|
|
if key is None:
|
|
i = 0
|
|
for (t, _key), value in self.keys.items():
|
|
if t == sec_type:
|
|
if i == index:
|
|
return value
|
|
i += 1
|
|
return None
|
|
else:
|
|
key = sec_type, bytes(key)
|
|
return self.keys.get(key, None)
|
|
# else:
|
|
# print("Unhandled IRQ event: ", event, data)
|
|
|
|
def _key_secrets(self, keys={}):
|
|
"""Save pairing key"""
|
|
with open("ble_hid_key.py", "w+") as s_f:
|
|
s_f.write("keys=" + str(keys) + "\n")
|
|
|
|
def _service_characteristics(self, handles):
|
|
"""Write BAS&service characteristics."""
|
|
(h_mod, h_ser, h_fwr, h_hwr, h_swr, h_man, h_pnp) = handles[0]
|
|
(self.h_bat, h_ccc, h_bfmt) = handles[1]
|
|
(h_info, h_hid, _, self.m_rep, _, h_d1, self.k_rep, _, h_d2, _, _, h_d3, self.j_rep, _, h_d4, h_proto) = handles[2]
|
|
# Write DIS characteristics.
|
|
self._ble.gatts_write(h_mod, b'1')
|
|
self._ble.gatts_write(h_ser, b'1')
|
|
self._ble.gatts_write(h_fwr, b'1')
|
|
self._ble.gatts_write(h_hwr, b'1')
|
|
self._ble.gatts_write(h_swr, b'1')
|
|
self._ble.gatts_write(h_man, b'Homebrew')
|
|
self._ble.gatts_write(h_pnp, b'\x01a\xfe\x01\x00#\x01')
|
|
# Write BAS characteristics.
|
|
self._ble.gatts_write(self.h_bat, struct.pack("<B", self.battery_level))
|
|
self._ble.gatts_write(h_bfmt, b'\x04\x00\xad\x27\x01\x00\x00')
|
|
self._ble.gatts_write(h_ccc, b'\x00\x00')
|
|
# Write service characteristics
|
|
self._ble.gatts_write(h_info, b"\x01\x01\x00\x02")
|
|
self._ble.gatts_write(h_hid, _HID_INPUT_REPORT)
|
|
self._ble.gatts_write(h_d1, b'\x01\x01')
|
|
self._ble.gatts_write(h_d2, b'\x02\x01')
|
|
self._ble.gatts_write(h_d3, b'\x02\x02')
|
|
self._ble.gatts_write(h_d4, b'\x03\x01')
|
|
self._ble.gatts_write(h_proto, b"\x01")
|
|
|
|
def is_connected(self):
|
|
'''蓝牙是否连接成功'''
|
|
return self.device_state
|
|
|
|
def advertise(self, interval_us=100000):
|
|
'''蓝牙广播'''
|
|
print("Starting advertising")
|
|
self._ble.gap_advertise(interval_us, adv_data=self._payload)
|
|
|
|
def battery_notify(self, level):
|
|
'''电池电量%'''
|
|
if self.is_connected():
|
|
self.battery_level = max(min(level, 100), 0)
|
|
self._ble.gatts_notify(self.conn_handle, self.h_bat, struct.pack("<B", self.battery_level))
|
|
|
|
def keyboard_state(self):
|
|
'''获取键盘 指示灯状态'''
|
|
return int.from_bytes(self.report, 'big')
|
|
|
|
def keyboard_notify(self, special=0, general=0, release=True):
|
|
'''键盘 特殊按键 + 常规组合按键*6'''
|
|
if self.is_connected():
|
|
_keys = bytearray(6)
|
|
if type(general) in (tuple, list):
|
|
for i in range(len(general)):
|
|
if i > 5: break
|
|
_keys[i] = general[i]
|
|
else:
|
|
_keys[0] = general
|
|
self._ble.gatts_notify(self.conn_handle, self.k_rep, bytes([special & 0xFF, 0]) + _keys)
|
|
if release:
|
|
time.sleep_ms(10)
|
|
self._ble.gatts_notify(self.conn_handle, self.k_rep, b'\x00\x00\x00\x00\x00\x00\x00\x00')
|
|
|
|
def keyboard_str(self, string, delay=0):
|
|
'''键盘发送ASCLL码'''
|
|
for char in str(string):
|
|
char = max(min(ord(char), 127), 0)
|
|
self.keyboard_notify( 0x02 if _KEYCODE[char] >> 7 else 0x00, _KEYCODE[char] & 0x7F)
|
|
time.sleep_ms(20 + delay)
|
|
|
|
def mouse_notify(self, keys=0, move=(0, 0), wheel=0, release=True):
|
|
'''鼠标 按键*3 + 位移 + 滚轮'''
|
|
if self.is_connected():
|
|
self._ble.gatts_notify(self.conn_handle, self.m_rep, bytes([keys & 0x0F, move[0] & 0xFF, move[1] & 0xFF, wheel & 0xFF]))
|
|
if release:
|
|
time.sleep_ms(10)
|
|
self._ble.gatts_notify(self.conn_handle, self.m_rep, b'\x00\x00\x00\x00')
|
|
|
|
def Joystick_notify(self, keys=0, axes=(0, 0), release=False):
|
|
'''手柄 按键*8 + 摇杆'''
|
|
if self.is_connected():
|
|
self._ble.gatts_notify(self.conn_handle, self.j_rep, bytes([axes[0] & 0xFF, axes[1] & 0xFF, keys & 0xFF]))
|
|
if release:
|
|
time.sleep_ms(10)
|
|
self._ble.gatts_notify(self.conn_handle, self.j_rep, b'\x00\x00\x00')
|
|
@property
|
|
def mac(self):
|
|
'''Get mac address'''
|
|
return hexlify(self._ble.config('mac')[1]).decode()
|