added password reader

slip-0016/pwdreader.py

+#!/usr/bin/env python2
+
+from trezorlib.client import TrezorClient
+from trezorlib.transport_hid import HidTransport
+from binascii import hexlify, unhexlify
+from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
+from cryptography.hazmat.backends import default_backend
+import hmac, hashlib, base58, json, sys
+
+
+#return path by BIP-32
+def getPath():
+    return client.expand_path("10016'/0");
+
+#Deriving master key
+def getMasterKey():
+    bip32_path = getPath()
+    ENC_KEY = "Activate TREZOR Password Manager?"
+    ENC_VALUE = unhexlify("2d650551248d792eabf628f451200d7f51cb63e46aadcbb1038aacb05e8c8aee2d650551248d792eabf628f451200d7f51cb63e46aadcbb1038aacb05e8c8aee")
+    key = hexlify(client.encrypt_keyvalue(
+        bip32_path,
+        ENC_KEY,
+        ENC_VALUE,
+        True,
+        True
+    ))
+    return key;
+
+#Deriving file name and encryption key
+def getFileEncKey(key):
+    filekey, enckey = key[:len(key)/2], key[len(key)/2:]
+    FILENAME_MESS = "5f91add3fa1c3c76e90c90a3bd0999e2bd7833d06a483fe884ee60397aca277a"
+    digest = hmac.new(filekey, FILENAME_MESS, hashlib.sha256).hexdigest()
+    filename = ''.join((digest, '.pswd'))
+    return [filename, filekey, enckey];
+
+#Path to locally stored file
+def getFilePath():
+    return '/home/chren/';
+
+#File level decryption and file reading
+def decryptStorage(path, key):
+    cipherkey = unhexlify(key)
+    with open(path, "rb") as f:
+        iv = f.read(12)
+        tag = f.read(16)
+        cipher = Cipher(algorithms.AES(cipherkey), modes.GCM(iv, tag), backend=default_backend())
+        decryptor = cipher.decryptor()
+        data = "";
+        while True:
+            block = f.read(16)
+            # data are not authenticated yet
+            if block:
+                data = data + decryptor.update(block)
+            else:
+                break
+        # throws exception when the tag is wrong
+        data = data + decryptor.finalize()
+    return json.loads(data);
+
+def decryptEntryValue(nonce, val):
+    cipherkey = unhexlify(nonce)
+    iv = val[:12]
+    tag = val[12:28]
+    cipher = Cipher(algorithms.AES(cipherkey), modes.GCM(iv, tag), backend=default_backend())
+    decryptor = cipher.decryptor()
+    data = "";
+    inputData = val[28:]
+    while True:
+        block = inputData[:16]
+        inputData = inputData[16:]
+        if block:
+            data = data + decryptor.update(block)
+        else:
+            break
+        # throws exception when the tag is wrong
+    data = data + decryptor.finalize()
+    return json.loads(data);
+
+#decrypt give entry nonce
+def getDecryptedNonce(entry):
+    print '\nWaiting for TREZOR input!\n'
+    ENC_KEY = ''.join(('Unlock ', entry['title'], ' for user ', entry['username'], '?'))
+    ENC_VALUE = entry['nonce']
+    decrypted_nonce =  hexlify(client.decrypt_keyvalue(
+        getPath(),
+        ENC_KEY,
+        unhexlify(ENC_VALUE),
+        False,
+        True
+    ))
+    return decrypted_nonce;
+
+#list whatever
+def printList(obj, val):
+    objList = obj[val]
+    print '\n'
+    print 'Entry list:'
+    for x in xrange (len(objList)):
+        keys = objList[str(x)].keys()
+        print 'Entry id ', x
+        for y in xrange (len(keys)):
+            print keys[y],': ',objList[str(x)][keys[y]]
+        print '\n'
+    return;
+
+
+def main():
+    masterKey = getMasterKey()
+    #print 'master key: ', masterKey
+
+    fileName = getFileEncKey(masterKey)[0]
+    #print 'file name: ', fileName
+
+    path = getFilePath()
+    #print 'path to file: ', path
+
+    encKey = getFileEncKey(masterKey)[2]
+    #print 'enckey: ', encKey
+
+    full_path = ''.join((path, fileName))
+    parsed_json = decryptStorage(full_path, encKey)
+
+    #list entries
+    printList(parsed_json, 'entries')
+    entries = parsed_json['entries']
+
+    entry_id = raw_input('Select entry number to decrypt: ')
+
+    plain_nonce = getDecryptedNonce(entries[str(entry_id)])
+
+
+    pwdArr = entries[str(entry_id)]['password']['data']
+    pwdHex = ''.join([ hex(x)[2:].zfill(2) for x in pwdArr ])
+    print 'password : ', decryptEntryValue(plain_nonce, unhexlify(pwdHex))
+
+    safeNoteArr = entries[str(entry_id)]['safe_note']['data']
+    safeNoteHex = ''.join([ hex(x)[2:].zfill(2) for x in safeNoteArr ])
+    print 'safe_note : ', decryptEntryValue(plain_nonce, unhexlify(safeNoteHex))
+    return;
+
+if __name__ == "__main__":
+    try:
+        # init Trezor transport
+        client = TrezorClient(HidTransport(HidTransport.enumerate()[0]))
+    except:
+        print 'TREZOR is not plugged in. Please, connect TREZOR and retry.'
+    else:
+        main()
\ No newline at end of file