Add googletotpkey command

hhomar · hhomar · commit 6a676fdb26a3 · 2017-10-17T20:12:56.000+01:00
This matches the web app where the value entered is only converted from
base32 to hex if Google OTP is selected.
diff --git a/onlykey/cli.py b/onlykey/cli.py
@@ -1,6 +1,8 @@
 # coding: utf-8
 from __future__ import unicode_literals, print_function
 
+import base64
+import binascii
 import time
 import logging
 import os
@@ -121,7 +123,6 @@ def prompt_key():
         key = prompt('Key: ',
                           is_password=Condition(lambda cli: hidden[0]),
                           key_bindings_registry=key_bindings_manager.registry)
-        #need base32 to hex conversion
         return key
 
     def prompt_pin():
@@ -201,6 +202,14 @@ def mprompt():
                 only_key.setslot(slot_id, MessageField.DELAY3, data[3])
             elif data[2] == 'type':
                  only_key.setslot(slot_id, MessageField.TFATYPE, data[3])
+            elif data[2] == 'googletotpkey':
+                totpkey = prompt_key()
+                totpkey = base64.b32decode(totpkey)
+                totpkey = binascii.hexlify(totpkey)
+                # pad with zeros for even digits
+                totpkey = totpkey.zfill(len(totpkey) + len(totpkey) % 2)
+                payload = [int(totpkey[i: i+2], 16) for i in range(0, len(totpkey), 2)]
+                only_key.setslot(slot_id, MessageField.TOTPKEY, payload)
             elif data[2] == 'totpkey':
                 totpkey = prompt_key()
                 only_key.setslot(slot_id, MessageField.TOTPKEY, totpkey)
