From d05ae0e167690172533a4fdea603080cb4582f86 Mon Sep 17 00:00:00 2001
From: Paul Selkirk <paul@psgd.org>
Date: Tue, 12 Jul 2016 14:53:24 -0400
Subject: Allow wheel to upload firmware/bitstreams

---
 projects/hsm/cryptech_upload | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'projects')

diff --git a/projects/hsm/cryptech_upload b/projects/hsm/cryptech_upload
index 6d6d4f7..64b9440 100755
--- a/projects/hsm/cryptech_upload
+++ b/projects/hsm/cryptech_upload
@@ -128,11 +128,15 @@ def _execute(dst, cmd):
     _write(dst, "\r")
     prompt = _read(dst)
     if prompt.endswith("Username: "):
-        _write(dst, "so\r")
+        if sys.version_info[0] > 2:
+            user = input("Username (so, wheel): ")
+        else:
+            user = raw_input("Username (so, wheel): ")
+        _write(dst, user + "\r")
         prompt = _read(dst)
         if prompt.endswith("Password: "):
             if not pin:
-                pin = getpass.getpass("SO PIN: ")
+                pin = getpass.getpass("PIN: ")
             _write(dst, pin + "\r")
             prompt = _read(dst)
     if not prompt.endswith(("> ", "# ")):
-- 
cgit v1.2.3