1 files changed, 105 insertions, 0 deletions

diff --git a/scripts/tests/sendkeys.py b/scripts/tests/sendkeys.py
new file mode 100755
index 0000000..9300242
--- /dev/null
+++ b/scripts/tests/sendkeys.py
@@ -0,0 +1,105 @@
+#!/usr/bin/env python3
+"""Type lines into a QEMU guest via the monitor's `sendkey` command.
+
+Each positional argument is one line: it is typed character by character
+(US layout) followed by Enter.  Use it to drive the rs-dos shell from a
+test script without a display.
+
+Used by scripts/tests/test-*.sh, and manually against an interactive
+session started with `./scripts/run.sh --monitor`:
+
+    ./scripts/tests/sendkeys.py "ls /disk" "date" "poweroff"
+    ./scripts/tests/sendkeys.py --sock /tmp/rsdos-monitor.sock "date"
+
+Environment: QMON_SOCK (default: /tmp/rsdos-monitor.sock)
+"""
+import os
+import socket
+import sys
+import time
+
+SOCK = os.environ.get("QMON_SOCK", "/tmp/rsdos-monitor.sock")
+
+# Unshifted punctuation needs QEMU's *named* keys — bare '-'/'.'/'/' are
+# not valid sendkey names (and '-' is the modifier separator).
+BASE = {
+    "-": "minus", "=": "equal", "[": "bracketleft", "]": "bracketright",
+    "\\": "backslash", ";": "semicolon", "'": "apostrophe",
+    ",": "comma", ".": "dot", "/": "slash", "`": "grave",
+}
+
+# Characters that need a shift key on US layout, and the key QEMU's
+# `sendkey` understands for them.
+SHIFT = {
+    "~": "grave", "!": "1", "@": "2", "#": "3", "$": "4", "%": "5",
+    "^": "6", "&": "7", "*": "8", "(": "9", ")": "0",
+    "_": "minus", "+": "equal", "{": "bracketleft", "}": "bracketright",
+    "|": "backslash", ":": "semicolon", '"': "apostrophe",
+    "<": "comma", ">": "dot", "?": "slash",
+}
+
+ALNUM = set("abcdefghijklmnopqrstuvwxyz0123456789")
+
+
+def connect(path, timeout=60):
+    """Connect to the monitor socket, retrying until QEMU is up."""
+    deadline = time.time() + timeout
+    while True:
+        try:
+            s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+            s.connect(path)
+            return s
+        except OSError:
+            if time.time() > deadline:
+                raise
+            time.sleep(0.2)
+
+
+def sendkey(sock, keys):
+    sock.sendall(("sendkey %s\n" % keys).encode())
+
+
+def type_line(sock, line):
+    for ch in line:
+        if ch == " ":
+            sendkey(sock, "spc")
+        elif ch in BASE:
+            sendkey(sock, BASE[ch])
+        elif ch in SHIFT:
+            sendkey(sock, "shift-" + SHIFT[ch])
+        elif ch.isupper():
+            sendkey(sock, "shift-" + ch.lower())
+        elif ch in ALNUM:
+            sendkey(sock, ch)
+        else:
+            sys.stderr.write("warning: no sendkey mapping for %r\n" % ch)
+            continue
+        time.sleep(0.25)
+    sendkey(sock, "ret")
+    time.sleep(1.5)
+
+
+def main():
+    args = [a for a in sys.argv[1:] if a != "--sock"]
+    sock_path = SOCK
+    if "--sock" in sys.argv:
+        i = sys.argv.index("--sock")
+        sock_path = sys.argv[i + 1]
+        args = sys.argv[i + 2:]
+    if not args:
+        sys.stderr.write(__doc__ or "")
+        return 1
+    sock = connect(sock_path)
+    # Let the monitor drain any startup banner before typing.
+    time.sleep(0.5)
+    for line in args:
+        try:
+            type_line(sock, line)
+        except OSError:
+            # Guest powered off (last line was usually "poweroff").
+            break
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())