rpikvm/hidinput.py

189 lines
4.5 KiB
Python

from flask import Response
from flask import Flask
from flask import render_template
from flask import request
import threading
import argparse
import json
from hidkeycodes import hidkeycodes
from jskeycodes import jscodehidmap
lock = threading.Lock()
# initialize a flask object
app = Flask(__name__)
hiddev = None
hidmouse = None
hidmouseabs = None
def hid_init():
global hiddev, hidmouse, hidmouseabs
hiddev = open('/dev/hidg0', 'rb+')
hidmouse = open('/dev/hidg1', 'rb+')
hidmouseabs = open('/dev/hidg2', 'rb+')
def hid_write(data):
if hiddev is None:
return False
hiddev.write(data)
hiddev.flush()
def hid_mouse_write(btn, x, y, wheel):
if hidmouse is None:
return False
data = bytearray(4)
data[0] = btn
data[1] = x
data[2] = y
data[3] = wheel
hidmouse.write(data)
hidmouse.flush()
def hid_mouse_writeabs(btn, x, y, wheel):
if hidmouseabs is None:
return False
data = bytearray(6)
data[0] = btn
data[1:3] = x.to_bytes(2, byteorder='little')
data[3:5] = y.to_bytes(2, byteorder='little')
data[5] = wheel
hidmouseabs.write(data)
hidmouseabs.flush()
def send_key(hidkey, shift, alt, ctlr, mod):
data = bytearray(8)
if shift:
data[0] += hidkeycodes['KEY_MOD_LSHIFT']
if alt:
data[0] += hidkeycodes['KEY_MOD_LALT']
if ctlr:
data[0] += hidkeycodes['KEY_MOD_LCTRL']
if mod and (hidkey == hidkeycodes['KEY_MOD_LMETA'] or hidkey == hidkeycodes['KEY_MOD_RMETA']):
data[0] += hidkey
hidkey = 0
data[2] = hidkey
hid_write(data)
hid_write(bytearray(8))
def get_hid_by_jscode(rawkeycode):
hidkeycode = None
hidkeyname = None
if rawkeycode in jscodehidmap:
hidkeyname = jscodehidmap[rawkeycode]
if hidkeyname in hidkeycodes:
hidkeycode = hidkeycodes[hidkeyname]
else:
print("Code for key {} not found in HID mapping".format(hidkeyname))
else:
print("Javascript code for key {} not found".format(rawkeycode))
print ("JS key: {} HID key: {}({})".format(rawkeycode, hidkeyname, hidkeycode))
return hidkeycode, (hidkeyname is not None and "MOD" in hidkeyname)
@app.route("/")
def index():
return render_template("hid.html")
@app.route("/mouse.html")
def mouseindex():
return render_template("mouse.html")
@app.route("/hid/mouse", methods=["POST"])
def mouse():
mouseevent = json.loads(request.data)
print(mouseevent)
btn = mouseevent['btn']
x = mouseevent['x']
y = mouseevent['y']
wheel = mouseevent['wheel']
x = x if x >= 0 else 255-abs(x)
y = y if y >= 0 else 255-abs(y)
print("X: {}, Y: {}".format(x, y))
hid_mouse_write(btn, x, y, wheel)
return Response("", mimetype="text/plain")
@app.route("/hid/mouseabs", methods=["POST"])
def mouseabs():
mouseevent = json.loads(request.data)
print(mouseevent)
btn = mouseevent['btn']
x = mouseevent['x']
y = mouseevent['y']
wheel = mouseevent['wheel']
print("X: {}, Y: {}".format(x, y))
hid_mouse_writeabs(btn, x, y, wheel)
return Response("", mimetype="text/plain")
@app.route("/hid/keyboard", methods=["POST"])
def keypress():
keyevent = json.loads(request.data)
print("Raw data: {}".format(request.data))
rawkeycode = keyevent['code']
hidkeycode, mod = get_hid_by_jscode(rawkeycode)
if hidkeycode is not None:
try:
send_key(
hidkeycode,
keyevent['shiftKey'],
keyevent['altKey'],
keyevent['ctrlKey'],
mod)
except Exception as e:
print("Error sending HID message", e)
return Response("Press {}".format(hidkeycode), mimetype="text/plain")
# check to see if this is the main thread of execution
if __name__ == '__main__':
# construct the argument parser and parse command line arguments
ap = argparse.ArgumentParser()
ap.add_argument("-i", "--ip", type=str, required=True,
help="ip address of the device")
ap.add_argument("-o", "--port", type=int, required=True,
help="ephemeral port number of the server (1024 to 65535)")
ap.add_argument("-f", "--frame-count", type=int, default=32,
help="# of frames used to construct the background model")
args = vars(ap.parse_args())
hid_init()
app.run(host=args["ip"], port=args["port"], debug=True,
threaded=True, use_reloader=False)
# Clean up
if hiddev is not None:
print("Closing hid")
hiddev.close()