From 37bf786c017c75061f5bb3d1e7819c9655620ead Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marco=20Trevisan=20=28Trevi=C3=B1o=29?= <mail@3v1n0.net>
Date: Wed, 10 Jun 2020 01:42:22 +0200
Subject: [PATCH] led-dance: Add example to perform led tests

---
 led-dance.py | 62 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 led-dance.py

diff --git a/led-dance.py b/led-dance.py
new file mode 100644
index 0000000..cf0e21f
--- /dev/null
+++ b/led-dance.py
@@ -0,0 +1,62 @@
+#!/usr/bin/env python3
+
+import os
+
+from binascii import unhexlify
+from enum import Enum
+from proto9x.flash import read_flash
+from proto9x.sensor import *
+from proto9x.tls import tls as vfs_tls
+from proto9x.usb import usb as vfs_usb
+from time import sleep
+from usb import core as usb_core
+
+class VFS(Enum):
+    DEV_90 = 0x0090
+    DEV_97 = 0x0097
+
+
+if __name__ == "__main__":
+    if os.geteuid() != 0:
+        raise Exception('This script needs to be executed as root')
+
+    usb_dev = None
+    for d in VFS:
+        dev = usb_core.find(idVendor=0x138a, idProduct=d.value)
+        if dev:
+            usb_dev = dev
+
+    if not usb_dev:
+        raise Exception('No supported validity device found')
+
+    vfs_usb.open(product=usb_dev.idProduct)
+    vfs_usb.send_init()
+
+    try:
+        with open('/sys/class/dmi/id/product_name', 'r') as node:
+            product_name = node.read().strip()
+        with open('/sys/class/dmi/id/product_serial', 'r') as node:
+            product_serial = node.read().strip()
+    except:
+        product_name = 'VirtualBox'
+        product_serial = '0'
+
+    vfs_tls.set_hwkey(product_name=product_name, serial_number=product_serial)
+
+    # try to init TLS session from the flash
+    vfs_tls.parseTlsFlash(read_flash(1, 0, 0x1000))
+    vfs_tls.open()
+
+    for i in range(0, 10):
+        glow_start_scan()
+        sleep(0.05)
+        glow_end_enroll()
+        sleep(0.05)
+
+    led_script = unhexlify(
+        '39ff100000ff03000001ff002000000000ffff0000ffff0000ff03000001ff00' \
+        '200000000000000000ffff0000ff03000001ff002000000000ffff0000000000' \
+        '0000000000000000000000000000000000000000000000000000000000000000' \
+        '0000000000000000000000000000000000000000000000000000000000')
+
+    assert_status(tls.app(led_script))