def payload_to_block3(passphrase, uid):
passphrase_bytes_list = [base58decode(chunk).to_bytes(3, byteorder='little') for chunk in passphrase.split('-')]
uid_bytes = base58decode(uid).to_bytes(4, byteorder='little')
binary = bytearray(32)
binary[7:10] = passphrase_bytes_list[0]
binary[10:12] = passphrase_bytes_list[1][0:2]
binary[20] = passphrase_bytes_list[1][2]
binary[21:23] = passphrase_bytes_list[2][0:2]
binary[24] = passphrase_bytes_list[2][2]
binary[25:28] = passphrase_bytes_list[3]
binary[28:32] = uid_bytes
return binary
def handle_block3_fuses(set_block_3, uid, passphrase):
if not set_block_3:
print("Block 3 eFuses already set. UID: {}, Passphrase valid".format(
uid))
return uid, passphrase
print("Reading staging password")
try:
file_directory = os.path.dirname(os.path.realpath(__file__))
with open('staging_password.txt', 'rb') as f:
staging_password = f.read().decode('utf-8').split('\n')[0].strip()
except:
print('staging_password.txt missing or malformed')
sys.exit(0)
print("Installing auth_handler")
if sys.version_info < (3, 5, 3):
context = ssl.SSLContext(protocol=ssl.PROTOCOL_SSLv23)
else:
context = ssl.SSLContext()
#context.verify_mode = ssl.CERT_REQUIRED
#context.load_verify_locations(certifi.where())
https_handler = urllib.request.HTTPSHandler(context=context)
auth_handler = urllib.request.HTTPBasicAuthHandler()
auth_handler.add_password(realm='Staging',
uri='https://stagingwww.tinkerforge.com',
user='******',
passwd=staging_password)
opener = urllib.request.build_opener(https_handler, auth_handler)
urllib.request.install_opener(opener)
print("Generating passphrase")
# smallest 4-char-base58 string is "2111" = 195112 ("ZZZ"(= 195111) + 1)
# largest 4-char-base58 string is "ZZZZ" = 11316495
# Directly selecting chars out of the BASE58 alphabet can result in numbers with leading 1s
# (those map to 0, so de- and reencoding will produce the same number without the leading 1)
wifi_passphrase = [
base58encode(rnd.randint(base58decode("2111"), base58decode("ZZZZ")))
for i in range(4)
]
print("Generating UID")
uid = base58encode(get_new_uid())
print("UID: " + uid)
print("Generating efuse binary")
uid_bytes = base58decode(uid).to_bytes(4, byteorder='little')
passphrase_bytes_list = [
base58decode(chunk).to_bytes(3, byteorder='little')
for chunk in wifi_passphrase
]
#56-95: 5 byte
#160-183: 3 byte
#192-255: 8 byte
# = 16 byte
# 4 byte (uid) + 3 byte * 4 (wifi_passphrase) = 16 byte
binary = bytearray(32)
binary[7:10] = passphrase_bytes_list[0]
binary[10:12] = passphrase_bytes_list[1][0:2]
binary[20] = passphrase_bytes_list[1][2]
binary[21:23] = passphrase_bytes_list[2][0:2]
binary[24] = passphrase_bytes_list[2][2]
binary[25:28] = passphrase_bytes_list[3]
binary[28:32] = uid_bytes
with temp_file() as (fd, name):
with os.fdopen(fd, 'wb') as f:
f.write(binary)
print("Burning UID and Wifi passphrase eFuses")
espefuse([
"--port", PORT, "burn_block_data", "BLOCK3", name,
"--do-not-confirm"
])
return uid, '-'.join(wifi_passphrase)