diff options
Diffstat (limited to 'util/inject-keys.py')
-rwxr-xr-x | util/inject-keys.py | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/util/inject-keys.py b/util/inject-keys.py new file mode 100755 index 0000000000..ef87bb14b6 --- /dev/null +++ b/util/inject-keys.py @@ -0,0 +1,127 @@ +#!/usr/bin/python +# -*- coding: utf-8 -*- +# +# Copyright 2016 The Chromium OS Authors. All rights reserved. +# Use of this source code is governed by a BSD-style license that can be +# found in the LICENSE file. + +import string +import subprocess +import sys + + +KEYMATRIX = {'`': (3, 1), '1': (6, 1), '2': (6, 4), '3': (6, 2), '4': (6, 3), + '5': (3, 3), '6': (3, 6), '7': (6, 6), '8': (6, 5), '9': (6, 9), + '0': (6, 8), '-': (3, 8), '=': (0, 8), 'q': (7, 1), 'w': (7, 4), + 'e': (7, 2), 'r': (7, 3), 't': (2, 3), 'y': (2, 6), 'u': (7, 6), + 'i': (7, 5), 'o': (7, 9), 'p': (7, 8), '[': (2, 8), ']': (2, 5), + '\\': (3, 11), 'a': (4, 1), 's': (4, 4), 'd': (4, 2), 'f': (4, 3), + 'g': (1, 3), 'h': (1, 6), 'j': (4, 6), 'k': (4, 5), 'l': (4, 9), + ';': (4, 8), '\'': (1, 8), 'z': (5, 1), 'x': (5, 4), 'c': (5, 2), + 'v': (5, 3), 'b': (0, 3), 'n': (0, 6), 'm': (5, 6), ',': (5, 5), + '.': (5, 9), '/': (5, 8), ' ': (5, 11), '<right>': (6, 12), + '<alt_r>': (0, 10), '<down>': (6, 11), '<tab>': (2, 1), + '<f10>': (0, 4), '<shift_r>': (7, 7), '<ctrl_r>': (4, 0), + '<esc>': (1, 1), '<backspace>': (1, 11), '<f2>': (3, 2), + '<alt_l>': (6, 10), '<ctrl_l>': (2, 0), '<f1>': (0, 2), + '<search>': (0, 1), '<f3>': (2, 2), '<f4>': (1, 2), '<f5>': (3, 4), + '<f6>': (2, 4), '<f7>': (1, 4), '<f8>': (2, 9), '<f9>': (1, 9), + '<up>': (7, 11), '<shift_l>': (5, 7), '<enter>': (4, 11), + '<left>': (7, 12)} + + +UNSHIFT_TABLE = { '~': '`', '!': '1', '@': '2', '#': '3', '$': '4', + '%': '5', '^': '6', '&': '7', '*': '8', '(': '9', + ')': '0', '_': '-', '+': '=', '{': '[', '}': ']', + '|': '\\', + ':': ';', '"': "'", '<': ',', '>': '.', '?': '/'} + +for c in string.ascii_lowercase: + UNSHIFT_TABLE[c.upper()] = c + + +def inject_event(key, press): + if len(key) >= 2 and key[0] != '<': + key = '<' + key + '>' + if key not in KEYMATRIX: + print "%s: invalid key: %s" % (this_script, key) + sys.exit(1) + (row, col) = KEYMATRIX[key] + subprocess.call(["ectool", "kbpress", str(row), str(col), + "1" if press else "0"]) + + +def inject_key(key): + inject_event(key, True) + inject_event(key, False) + + +def inject_string(string): + for c in string: + if c in KEYMATRIX: + inject_key(c) + elif c in UNSHIFT_TABLE: + inject_event("<shift_l>", True) + inject_key(UNSHIFT_TABLE[c]) + inject_event("<shift_l>", False) + else: + print "unimplemented character:", c + sys.exit(1) + + +def usage(): + print "Usage: %s [-s <string>] [-k <key>]" % this_script, \ + "[-p <pressed-key>] [-r <released-key>] ..." + print "Examples:" + print "%s -s MyPassw0rd -k enter" % this_script + print "%s -p ctrl_l -p alt_l -k f3 -r alt_l -r ctrl_l" % this_script + + +def help(): + usage() + print "Valid keys are:" + i = 0 + for key in KEYMATRIX: + print "%12s" % key, + i += 1 + if i % 4 == 0: + print + print + print "angle brackets may be omitted" + + +def usage_check(asserted_condition, message): + if asserted_condition: + return + print "%s:" % this_script, message + usage() + sys.exit(1) + + +# -- main + +this_script = sys.argv[0] +arg_len = len(sys.argv) + +if arg_len > 1 and sys.argv[1] == "--help": + help() + sys.exit(0) + +usage_check(arg_len > 1, "not enough arguments") +usage_check(arg_len % 2 == 1, "mismatched arguments") + +for i in range(1, arg_len, 2): + usage_check(sys.argv[i] in ("-s", "-k", "-p", "-r"), + "unknown flag: %s" % sys.argv[i]) + +for i in range(1, arg_len, 2): + flag = sys.argv[i] + arg = sys.argv[i+1] + if flag == "-s": + inject_string(arg) + elif flag == "-k": + inject_key(arg) + elif flag == "-p": + inject_event(arg, True) + elif flag == "-r": + inject_event(arg, False) |