ParseXboxMetadata/CleanSS.py at main · Deterous/ParseXboxMetadata · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
import os
import sys

def get_xgd_type(ss):
    layerbreak = (ss[13] << 16) | (ss[14] << 8) | ss[15]
    if layerbreak == 0x2033AF:
        return 1
    elif layerbreak == 0x20339F:
        return 2
    elif layerbreak == 0x238E0F:
        return 3
    else:
        return None

def clean_ss(ss, ssv2, fix):
    xgd_type = get_xgd_type(ss)
    if xgd_type == 1:
        if fix:
            ss[0x200:0x2D0] = b'\x00' * (0x2D0 - 0x200)
        return True
    elif xgd_type == 2:
        ss[552] = 0x01
        ss[553] = 0x00
        ss[555] = 0x01 if ssv2 else 0x00
        ss[556] = 0x00
        ss[561] = 0x5B
        ss[562] = 0x00
        ss[564] = 0x5B if ssv2 else 0x00
        ss[565] = 0x00
        ss[570] = 0xB5
        ss[571] = 0x00
        ss[573] = 0xB5 if ssv2 else 0x00
        ss[574] = 0x00
        ss[579] = 0x0F
        ss[580] = 0x01
        ss[582] = 0x0F if ssv2 else 0x00
        ss[583] = 0x01 if ssv2 else 0x00
        return True
    elif xgd_type == 3:
        if any(x != 0 for x in ss[32:32+72]):
            ss[72] = 0x01
            ss[73] = 0x00
            ss[75] = 0x01
            ss[76] = 0x00
            ss[81] = 0x5B
            ss[82] = 0x00
            ss[84] = 0x5B
            ss[85] = 0x00
            ss[90] = 0xB5
            ss[91] = 0x00
            ss[93] = 0xB5
            ss[94] = 0x00
            ss[99] = 0x0F
            ss[100] = 0x01
            ss[102] = 0x0F
            ss[103] = 0x01
        else:
            ss[552] = 0x01
            ss[553] = 0x00
            ss[561] = 0x5B
            ss[562] = 0x00
            ss[570] = 0xB5
            ss[571] = 0x00
            ss[579] = 0x0F
            ss[580] = 0x01
        return True
    return False

def process_file(file_path, ssv2, fix):
    try:
        if os.path.isfile(file_path) and os.path.getsize(file_path) == 2048:
            with open(file_path, 'rb+') as f:
                data = bytearray(f.read())
                if clean_ss(data, ssv2, fix):
                    f.seek(0)
                    f.write(data)
                else:
                    print(f"Invalid SS: {file_path}")
        else:
            print(f"Invalid SS: {file_path}")
    except Exception as e:
        print(f"Error processing {file_path}: {e}")

def process_directory(directory, recursive, ssv2, fix):
    for root, _, files in os.walk(directory):
        for name in files:
            process_file(os.path.join(root, name), ssv2, fix)
        if not recursive:
            break

def main():
    if len(sys.argv) < 2:
        print("Usage: python CleanSS.py <file|directory> [-r|--recursive] [-s|--ssv2] [-f|--fix]")
        print("")
        print("--ssv2 cleans and upgrades the SS to SSv2 format")
        print("--fix cleans Original Xbox SS that are rarely incorrectly written to by some drives")
        return

    path = sys.argv[1]
    recursive = any(arg in ('-r', '--recursive') for arg in sys.argv[2:])
    ssv2 = any(arg in ('-s', '--ssv2') for arg in sys.argv[2:])
    fix = any(arg in ('-f', '--fix') for arg in sys.argv[2:])

    if os.path.isdir(path):
        process_directory(path, recursive, ssv2, fix)
    elif os.path.isfile(path):
        process_file(path, ssv2, fix)
    else:
        print(f"Invalid path: {path}")

if __name__ == "__main__":
    main()