diff --git a/horizonzdr/hzdr.py b/horizonzdr/hzdr.py index be312ed..ae3de4d 100644 --- a/horizonzdr/hzdr.py +++ b/horizonzdr/hzdr.py @@ -65,7 +65,7 @@ def run_benchmark() -> tuple[float]: time.sleep(10) - process_registry_file(input_file, config_file) + process_registry_file(hive, subkey, input_file, config_file) sys.exit(1) # Make sure the game started correctly result = kerasService.look_for_word("remastered", 10, 5) diff --git a/horizonzdr/hzdr_utils.py b/horizonzdr/hzdr_utils.py index eca9620..6241041 100644 --- a/horizonzdr/hzdr_utils.py +++ b/horizonzdr/hzdr_utils.py @@ -36,7 +36,7 @@ def export_registry_key(hive, subkey, input_file): def convert_dword_to_decimal(dword_hex): return int(dword_hex, 16) -def process_registry_file(input_file, config_file): +def process_registry_file(hive, subkey, input_file, config_file): export_registry_key(hive, subkey, input_file) with open(input_file, 'r') as file: lines = file.readlines()