mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-30 03:18:20 +00:00
139 lines
3.6 KiB
Python
Executable File
139 lines
3.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Use TREZOR as a hardware key for opening EncFS filesystem!
|
|
|
|
Usage:
|
|
|
|
encfs --standard --extpass=./encfs_aes_getpass.py ~/.crypt ~/crypt
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
import hashlib
|
|
|
|
import trezorlib
|
|
|
|
version_tuple = tuple(map(int, trezorlib.__version__.split(".")))
|
|
if not (0, 11) <= version_tuple < (0, 12):
|
|
raise RuntimeError("trezorlib version mismatch (0.11.x is required)")
|
|
|
|
from trezorlib.client import TrezorClient
|
|
from trezorlib.transport import enumerate_devices
|
|
from trezorlib.ui import ClickUI
|
|
|
|
import trezorlib.misc
|
|
|
|
|
|
def wait_for_devices():
|
|
devices = enumerate_devices()
|
|
while not len(devices):
|
|
sys.stderr.write("Please connect TREZOR to computer and press Enter...")
|
|
input()
|
|
devices = enumerate_devices()
|
|
|
|
return devices
|
|
|
|
|
|
def choose_device(devices):
|
|
if not len(devices):
|
|
raise RuntimeError("No TREZOR connected!")
|
|
|
|
if len(devices) == 1:
|
|
try:
|
|
return devices[0]
|
|
except IOError:
|
|
raise RuntimeError("Device is currently in use")
|
|
|
|
i = 0
|
|
sys.stderr.write("----------------------------\n")
|
|
sys.stderr.write("Available devices:\n")
|
|
for d in devices:
|
|
try:
|
|
client = TrezorClient(d, ui=ClickUI())
|
|
except IOError:
|
|
sys.stderr.write("[-] <device is currently in use>\n")
|
|
continue
|
|
|
|
if client.features.label:
|
|
sys.stderr.write("[%d] %s\n" % (i, client.features.label))
|
|
else:
|
|
sys.stderr.write("[%d] <no label>\n" % i)
|
|
client.close()
|
|
i += 1
|
|
|
|
sys.stderr.write("----------------------------\n")
|
|
sys.stderr.write("Please choose device to use:")
|
|
|
|
try:
|
|
device_id = int(input())
|
|
return devices[device_id]
|
|
except Exception:
|
|
raise ValueError("Invalid choice, exiting...")
|
|
|
|
|
|
def main():
|
|
|
|
if "encfs_root" not in os.environ:
|
|
sys.stderr.write(
|
|
"\nThis is not a standalone script and is not meant to be run independently.\n"
|
|
)
|
|
sys.stderr.write(
|
|
"\nUsage: encfs --standard --extpass=./encfs_aes_getpass.py ~/.crypt ~/crypt\n"
|
|
)
|
|
sys.exit(1)
|
|
|
|
devices = wait_for_devices()
|
|
transport = choose_device(devices)
|
|
client = TrezorClient(transport, ui=ClickUI())
|
|
|
|
rootdir = os.environ["encfs_root"] # Read "man encfs" for more
|
|
passw_file = os.path.join(rootdir, "password.dat")
|
|
|
|
if not os.path.exists(passw_file):
|
|
# New encfs drive, let's generate password
|
|
|
|
sys.stderr.write("Please provide label for new drive: ")
|
|
label = input()
|
|
|
|
sys.stderr.write("Computer asked TREZOR for new strong password.\n")
|
|
|
|
# 32 bytes, good for AES
|
|
trezor_entropy = trezorlib.misc.get_entropy(client, 32)
|
|
urandom_entropy = os.urandom(32)
|
|
passw = hashlib.sha256(trezor_entropy + urandom_entropy).digest()
|
|
|
|
if len(passw) != 32:
|
|
raise ValueError("32 bytes password expected")
|
|
|
|
bip32_path = [10, 0]
|
|
passw_encrypted = trezorlib.misc.encrypt_keyvalue(
|
|
client, bip32_path, label, passw, False, True
|
|
)
|
|
|
|
data = {
|
|
"label": label,
|
|
"bip32_path": bip32_path,
|
|
"password_encrypted_hex": passw_encrypted.hex(),
|
|
}
|
|
|
|
json.dump(data, open(passw_file, "w"))
|
|
|
|
# Let's load password
|
|
data = json.load(open(passw_file, "r"))
|
|
|
|
passw = trezorlib.misc.decrypt_keyvalue(
|
|
client,
|
|
data["bip32_path"],
|
|
data["label"],
|
|
bytes.fromhex(data["password_encrypted_hex"]),
|
|
False,
|
|
True,
|
|
)
|
|
|
|
print(passw)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|