hid.py

hid.py

import os
import sys
import time
import logging

import dbus
from bluetooth import *

class Hid:
   # Define the Bluetooth HID PSMs
   PSM_CTRL = 0x11
   PSM_INTR = 0x13

   # define the keyboard service record
   SERVICE_RECORD = """<?xml version="1.0" encoding="UTF-8" ?><record><attribute id="0x0001"><sequence><uuid value="0x1124" /></sequence></attribute>
<attribute id="0x0004"><sequence><sequence><uuid value="0x0100" /><uint16 value="0x0011" /></sequence><sequence><uuid value="0x0011" /></sequence></sequence></attribute>
<attribute id="0x0005"><sequence><uuid value="0x1002" /></sequence></attribute>
<attribute id="0x0006"><sequence><uint16 value="0x656e" /><uint16 value="0x006a" /><uint16 value="0x0100" /></sequence></attribute>
<attribute id="0x0009"><sequence><sequence><uuid value="0x1124" /><uint16 value="0x0100" /></sequence></sequence></attribute>
<attribute id="0x000d"><sequence><sequence><sequence><uuid value="0x0100" /><uint16 value="0x0013" /></sequence><sequence><uuid value="0x0011" /></sequence></sequence></sequence></attribute>
<attribute id="0x0100"><text value="Raspberry Pi Virtual Keyboard" /></attribute><attribute id="0x0101"><text value="USB > BT Keyboard" /></attribute>
<attribute id="0x0102"><text value="PiBoard" /></attribute><attribute id="0x0200"><uint16 value="0x0100" /></attribute>
<attribute id="0x0201"><uint16 value="0x0111" /></attribute><attribute id="0x0202"><uint8 value="0x40" /></attribute>
<attribute id="0x0203"><uint8 value="0x00" /></attribute><attribute id="0x0204"><boolean value="true" /></attribute>
<attribute id="0x0205"><boolean value="true" /></attribute><attribute id="0x0206"><sequence><sequence><uint8 value="0x22" />
<text encoding="hex" value="05010906a101850175019508050719e029e715002501810295017508810395057501050819012905910295017503910395067508150026ff000507190029ff8100c0050c0901a1018503150025017501950b0a23020a21020ab10109b809b609cd09b509e209ea09e9093081029501750d8103c0" />
</sequence></sequence></attribute><attribute id="0x0207"><sequence><sequence><uint16 value="0x0409" /><uint16 value="0x0100" /></sequence></sequence></attribute>
<attribute id="0x020b"><uint16 value="0x0100" /></attribute><attribute id="0x020c"><uint16 value="0x0c80" /></attribute>
<attribute id="0x020d"><boolean value="false" /></attribute><attribute id="0x020e"><boolean value="true" /></attribute>
<attribute id="0x020f"><uint16 value="0x0640" /></attribute><attribute id="0x0210"><uint16 value="0x0320" /></attribute></record>"""

   # define some HID key codes
   KEY_H = 11
   KEY_E = 8
   KEY_L = 15
   KEY_O = 18
   KEY_W = 26
   KEY_R = 21
   KEY_D = 7
   KEY_1 = 30
   KEY_SPACE = 44

   LOG_LEVEL = logging.DEBUG
   LOG_FILE = "/dev/stdout"
   LOG_FORMAT = "%(asctime)s %(levelname)s %(message)s"

   # Define some key modifier bitmasks
   MODIFIER_NONE = 0x00
   SHIFT_LEFT = 0x02

   def __init__(self):
       self.sock_control = None
       self.conn_control = None
       self.conn_interrupt = None
       self.sock_intetrrupt = None
       self.input_report = bytearray([
               0xA1, # Input report
               0x01, # Usage: keyboard
               0x00, # Reserved
               0x00, # Modifier bits
               0x00, 0x00, 0x00, 0x00, 0x00, 0x00 # Keys
       ])
       logging.basicConfig(filename=Hid.LOG_FILE, format=Hid.LOG_FORMAT, level=Hid.LOG_LEVEL)

   def connect(self):
       # Create the control and interrupt sockets, and bind and listen
       self.sock_control = BluetoothSocket(L2CAP)
       self.sock_control.bind(("", Hid.PSM_CTRL))
       self.sock_control.listen(1)
       self.sock_interrupt = BluetoothSocket(L2CAP)
       self.sock_interrupt.bind(("", Hid.PSM_INTR))
       self.sock_interrupt.listen(1)
       logging.info("Waiting for a connection...")

       # Accept connections from the host, first the control socket and then the interrupt
       self.conn_control, conn_info = self.sock_control.accept()
       self.conn_interrupt, conn_info = self.sock_interrupt.accept()
       logging.info("PiBoard is connected to %s" % (conn_info[0]))

   def close(self):
       self.sock_interrupt.close()
       self.sock_control.close()

   def addService(self):
       SERVICE_NAME = "org.bluez"
       OBJECT_IFACE =  "org.freedesktop.DBus.ObjectManager"
       ADAPTER_IFACE = SERVICE_NAME + ".Adapter1"
       DEVICE_IFACE = SERVICE_NAME + ".Device1"
       PROPERTIES_IFACE = "org.freedesktop.DBus.Properties"
       bus = dbus.SystemBus()
       manager = dbus.Interface(bus.get_object("org.bluez", "/"), "org.freedesktop.DBus.ObjectManager")
       objects = manager.GetManagedObjects()
      
       obj = bus.get_object(SERVICE_NAME, "/org/bluez/hci0/dev_E4_E4_AB_F1_30_0F")
       adapter_path = dbus.Interface(obj, ADAPTER_IFACE)
       service = dbus.Interface(bus.get_object("org.bluez", "/org/bluez/hci0/dev_E4_E4_AB_F1_30_0F"), "org.bluez.Service")

       logging.debug("Adding keyboard service record")
       service.AddRecord(Hid.SERVICE_RECORD)

   # Send a key press and release
   def sendKey(self, key, modifiers=None):
       # press the key
       self.pressKey(key, modifiers=modifiers)
       # Give the host some time to process; keyboards aren't expected to be fast
       time.sleep(0.01)
       # release the key
       self.releaseKey()

   def pressKey(self, key, modifiers=None):
       if (modifiers != None):
           self.input_report[2] = modifiers
       self.input_report[4] = key
       self.conn_interrupt.send(str(self.input_report))

   def releaseKey(self):
       self.input_report[4] = 0x00
       self.conn_interrupt.send(str(self.input_report))

if __name__ == "__main__":

   hid = Hid()
   hid.addService()
   hid.connect()
   time.sleep(5)
   try:
       while True:
           hid.sendKey(chr(Hid.KEY_H), modifiers=Hid.SHIFT_LEFT)
           hid.sendKey(chr(Hid.KEY_E), modifiers=Hid.MODIFIER_NONE)
           hid.sendKey(chr(Hid.KEY_L))
           hid.sendKey(chr(Hid.KEY_L))
           hid.sendKey(chr(Hid.KEY_O))
           hid.sendKey(chr(Hid.KEY_SPACE))
           hid.sendKey(chr(Hid.KEY_W), modifiers=Hid.SHIFT_LEFT)
           hid.sendKey(chr(Hid.KEY_O), modifiers=Hid.MODIFIER_NONE)
           hid.sendKey(chr(Hid.KEY_R))
           hid.sendKey(chr(Hid.KEY_L))
           hid.sendKey(chr(Hid.KEY_D))
           hid.sendKey(chr(Hid.KEY_1), modifiers=Hid.SHIFT_LEFT)
           hid.sendKey(chr(Hid.KEY_SPACE), modifiers=Hid.MODIFIER_NONE)
           time.sleep(5)
   except KeyboardInterrupt as ex:
       hid.close()
   except BluetoothError as ex:
       print("Connection was lost")

   print("Thank you for using PiBoard")