now with dynamic challenge

1 files changed, 7 insertions, 2 deletions

diff --git a/verify.py b/verify.py
index d410f67..25e3c42 100755
--- a/verify.py
+++ b/verify.py
@@ -4,15 +4,20 @@ import binascii
 import codecs
 import hashlib
 import hmac
+import random
+import string
 import subprocess
 import sys
 
 ssh = b'840fedcacdb9e86f1ae1e6a0060de471d4f5fb35'
 ss = codecs.decode(ssh, 'hex')
 
-actual = subprocess.check_output(['ykchalresp', '-2', 'deadbeef']).strip().decode()
+chh = ''.join(random.choice('0123456789abcdef') for i in range(126))
+ch = codecs.decode(chh, 'hex')
 
-expected = hmac.new(ss, b'deadbeef', hashlib.sha1).hexdigest()
+actual = subprocess.check_output(['ykchalresp', '-2', '-x', chh]).strip().decode()
+
+expected = hmac.new(ss, ch, hashlib.sha1).hexdigest()
 
 if actual == expected:
     sys.exit(0)