Skip to content

Commit

Permalink
Add mechanism to make script work for refactored NFC module (flipperd…
Browse files Browse the repository at this point in the history 
…evices/flipperzero-firmware#3188 (comment))
  • Loading branch information
@fanfannnmn
fanfannnmn committed Nov 29, 2023
1 parent befbcd0 commit f738481
Showing 1 changed file with 43 additions and 18 deletions.
61 changes: 43 additions & 18 deletions AmiiboConverter.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -20,13 +20,15 @@
https://hax0kartik.github.io/amiibo-generator/
https://github.com/socram8888/amiitool
https://github.com/tobywf/pyamiibo
https://github.com/turbospok/Flipper-NTAG215-password-converter
"""
import argparse
import logging
import os
import pathlib
import re
import random
import re

from amiibo import AmiiboDump, AmiiboMasterKey, crypto


Expand Down Expand Up @@ -61,7 +63,7 @@ def _check_for_keys(self) -> bool or str:
if os.path.exists("unfixed-info.bin"):
if os.path.exists("locked-secret.bin"):
with open("unfixed-info.bin", "rb") as fp_d, open(
"locked-secret.bin", "rb"
"locked-secret.bin", "rb"
) as fp_t:
logging.info("Loaded decryption keys.")
return AmiiboMasterKey.from_separate_bin(fp_d.read(), fp_t.read())
Expand Down Expand Up @@ -111,7 +113,7 @@ def read_file(self, input_path: str) -> bool:
data = self._build_from_amiiboid(input_path.strip())
if not data:
pass
#return False
# return False
while len(data) < 540:
data += bytes(1)
self.byte_data = data[:540]
Expand All @@ -126,6 +128,7 @@ def _assemble_nfc(self) -> str:
:return: Flipper Compatible string
"""
pages, page_count = self._make_pages()

nfc_content = (
f"Filetype: Flipper NFC device\n"
f"Version: 2\n"
Expand All @@ -136,7 +139,7 @@ def _assemble_nfc(self) -> str:
f"ATQA: 44 00\n"
f"SAK: 00\n"
f"# Mifare Ultralight specific data\n"
f"Signature: {('00 '*32).strip()}\n"
f"Signature: {('00 ' * 32).strip()}\n"
f"Mifare version: 00 04 04 02 01 00 11 03\n"
f"Counter 0: 0\n"
f"Tearing 0: 00\n"
Expand All @@ -150,6 +153,25 @@ def _assemble_nfc(self) -> str:

return nfc_content

def _calculate_password(self) -> str:
"""
Calculate the password for Page 133
:return: Password string
"""
uid = bytes.fromhex(self._get_uid())

pwd = []
pwd_str = ""
if len(uid) == 7:
pwd.append(uid[1] ^ uid[3] ^ 0xAA)
pwd.append(uid[2] ^ uid[4] ^ 0x55)
pwd.append(uid[3] ^ uid[5] ^ 0xAA)
pwd.append(uid[4] ^ uid[6] ^ 0x55)
pwd_str = ' '.join('{:02X}'.format(byte) for byte in pwd)

return pwd_str

def _get_uid(self) -> str:
"""
Get the UID of the byte data, .nfc files needs this
Expand All @@ -158,10 +180,10 @@ def _get_uid(self) -> str:
"""
uid = []
for i in range(3):
byte = self.byte_data[i : i + 1].hex()
byte = self.byte_data[i: i + 1].hex()
uid.append(byte)
for i in range(4, 8):
byte = self.byte_data[i : i + 1].hex()
byte = self.byte_data[i: i + 1].hex()
uid.append(byte)

return " ".join(uid).upper()
Expand All @@ -177,12 +199,15 @@ def _make_pages(self) -> [str, int]:
page_count = 0
page = []
for i in range(len(self.byte_data)):
byte = self.byte_data[i : i + 1].hex()
byte = self.byte_data[i: i + 1].hex()
page.append(byte)
if len(page) == 4:
pages.append(f"Page {page_count}: {' '.join(page).upper()}")
page = []
page_count += 1

pages[133] = f"Page 133: {self._calculate_password()}"
pages[134] = "Page 134: 80 80 00 00"
return "\n".join(pages), page_count

def _randomize_uid(self) -> bool:
Expand Down Expand Up @@ -315,8 +340,8 @@ def write_files(self, multi_input: list or None = None, multi: int or None = 0)
logging.debug(f"Parsing {file[0]}")
if multi > 1:
multiply = []
for i in range(0,multi):
multiply.append([file[0], file[1]+"_"+str(i+1).zfill(2)])
for i in range(0, multi):
multiply.append([file[0], file[1] + "_" + str(i + 1).zfill(2)])
result = self.write_files(multi_input=multiply)
counter += result[0]
for fail in result[1]:
Expand Down Expand Up @@ -419,10 +444,10 @@ def set_output(self, output_path: str):
:param output_folder: Root folder to store the files
"""
#if os.path.splitext(output_path)[1] in [".nfc", ".bin"]:
# if os.path.splitext(output_path)[1] in [".nfc", ".bin"]:
if os.path.splitext(output_path)[1] not in [".nfc", ".bin"]:
# self.name = os.path.splitext(output_path)[0]
#else:
# self.name = os.path.splitext(output_path)[0]
# else:
self.output_folder = output_path

def _get_save_path(self, parsed_data: list) -> str:
Expand Down Expand Up @@ -451,7 +476,7 @@ def _get_save_path(self, parsed_data: list) -> str:
filename = os.path.join(path, parsed_data[1])
else:
save_path = path
return os.path.join(save_path, parsed_data[1])#, filename, extension
return os.path.join(save_path, parsed_data[1]) # , filename, extension


def confirm_prompt(question: str) -> bool:
Expand Down Expand Up @@ -484,7 +509,7 @@ def get_args():
required=False,
type=pathlib.Path,
help="Directory or file to save to. Will be created if it doesn't exist. If not specified,"
" the output will be stored in the same location as the original file.",
" the output will be stored in the same location as the original file.",
)
parser.add_argument(
"-r",
Expand Down Expand Up @@ -519,10 +544,10 @@ def validate_arguments(args) -> bool:
:return: True if everything checks out
"""
if ((args.mode in ["bin2bin", "nfc2nfc"]) and not args.output) or (
(args.mode in ["bin2bin", "nfc2nfc"]) and (str(args.output) in args.input)
(args.mode in ["bin2bin", "nfc2nfc"]) and (str(args.output) in args.input)
):
if not confirm_prompt(
f"This will overwrite existing files in {args.output}, do you wish to continue?"
f"This will overwrite existing files in {args.output}, do you wish to continue?"
):
return False
if args.output:
Expand Down Expand Up @@ -577,6 +602,6 @@ def main():

if __name__ == "__main__":
if main():
print(f"{'#'*12} Done! {'#'*12}")
print(f"{'#' * 12} Done! {'#' * 12}")
else:
print(f"{'#'*12} Aborted! {'#'*12}")
print(f"{'#' * 12} Aborted! {'#' * 12}")

0 comments on commit f738481

Please sign in to comment.