OCUpdater/patch.py at main · OpenCortex-Research/OCUpdater · 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
#!/usr/bin/python2
import argparse, json, subprocess, sys

parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter,)

parser.add_argument("Operation", choices=["patch"])
parser.add_argument("-q", "--quiet", default=False, action="store_true", help="Reduce output text")
parser.add_argument("-s", "--silent", default=False, action="store_true", help="Stop all text output. Use for headless patching")
parser.add_argument("-d", "--settings", default="/media/p4/OpenCortex/updates/settings.json", action="store", help="Use a custom settings file")
parser.add_argument("File")

# For Dev Testing
class subprocess:
    def __init():
        pass

    def call(args):
        print(args)

args = parser.parse_args()

with open(args.settings) as jsonfile:
        settings = json.load(jsonfile)
operation = args.Operation
if args.File is None:
    parser.error("Update file required")
file = args.File

BLUE='\033[0;34m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# echo the ascii art banner
print(f"""{YELLOW}
    ____                   ______           __
   / __ \____  ___  ____  / ____/___  _____/ /____  _  __
  / / / / __ \/ _ \/ __ \/ /   / __ \/ ___/ __/ _ \| |/_/
 / /_/ / /_/ /  __/ / / / /___/ /_/ / /  / /_/  __/>  <
 \____/ .___/\___/_/ /_/\____/\____/_/   \__/\___/_/|_|
     /_/                       CorOS update patcher
{NC}""")

print(f"{BLUE}=============== OpenCortex Update Patcher ==============={NC}")

subprocess.call(["mkdir", "-r", "/media/p4/OpenCortex/updates/cache/fs/"])
subprocess.call(["mkdir", "-r", "/media/p4/OpenCortex/updates/cache/uncompressed-fs/"])

# Extract the update file

print(f"{YELLOW}[!]{NC} Extracting update file...")
subprocess.call(["gunzip", f"{file}"])
subprocess.call(["tar","-xvf", f"{file[:-3]}", "-C", "/media/p4/OpenCortex/updates/cache/uncompressed-fs/"])
print(f"{YELLOW}[!]{NC} Mounting update file...")
subprocess.call(["mount", "-t", "ext4", "/media/p4/OpenCortex/updates/cache/uncompressed-fs/rootfs.ext3", "/media/p4/OpenCortex/updates/cache/fs/"])

# Do patching here

print(f"{YELLOW}[!]{NC} Starting update file patching...")

if settings["patchSSH"]:
    subprocess.call(["curl", "https://raw.githubusercontent.com/VanIseghemThomas/OpenCortex/main/shadow", "-o", "/media/p4/OpenCortex/updates/cache/shadow"])
    subprocess.call(["mv", "/media/p4/OpenCortex/updates/cache/fs/etc/shadow", "/media/p4/OpenCortex/updates/cache/fs/etc/shadow.bckp"])
    subprocess.call(["cp", "/media/p4/OpenCortex/updates/cache/shadow", "/media/p4/OpenCortex/updates/cache/fs/etc/shadow"])
    subprocess.call(["touch", "/media/p4/OpenCortex/updates/cache/fs/opt/neuraldsp/allow_sshd"])

if settings["patchUpdater"]:
    subprocess.call(["ln", "-s", "/media/p4/OpenCortex/updates/patch.py", "/media/p4/OpenCortex/updates/cache/fs/usr/bin/oc-update"])
    subprocess.call(["chmod", "+x", "/media/p4/OpenCortex/updates/patch.py"])
    subprocess.call(["chmod", "+x", "/media/p4/OpenCortex/updates/cache/fs/usr/bin/oc-update"])

print(f"{YELLOW}[!]{NC} Finished patching...")

# Build the new update file

print(f"{YELLOW}[!]{NC} Building the new update file...")
subprocess.call(["tar", "-cvf", f"{file[:-3]}", "/media/p4/OpenCortex/updates/cache/uncompressed-fs/rootfs.ext3", "/media/p4/OpenCortex/updates/cache/uncompressed-fs/uImage", "/media/p4/OpenCortex/updates/cache/uncompressed-fs/zpu.dtb"])
subprocess.call(["gunzip", "-k", f"{file[:-3]}"])
print(f"{BLUE}[+]{NC} Update file patched: {file}")

print(f"{YELLOW}[!]{NC} Unmounting update...")
subprocess.call(["umount", "/media/p4/OpenCortex/updates/cache/fs/"])

print(f"{YELLOW}[!]{NC} Clearing cache...")
subprocess.call(["rm", "-r", "/media/p4/OpenCortex/updates/cache/"])