fz0x1/Vbmeta
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added "if name == main"

Browse source
This commit is contained in:
WessellUrdata 2023-01-12 20:00:44 +08:00
parent aac8291219
commit 4c63f728aa
1 changed files with 25 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
patch-vbmeta.py
View file

@ -10,35 +10,35 @@ FLAGS_OFFSET = 123
FLAGS_TO_SET = b'\x03'
### main program
if __name__ == "__main__":
# if an argument is not provided
if len(sys.argv) != 2:
sys.exit("Usage: python ./" + os.path.basename(__file__) + " <vbmeta-image>")
# if an argument is not provided
if len(sys.argv) != 2:
sys.exit("Usage: python ./" + os.path.basename(__file__) + " <vbmeta-image>")
# try reading the file with read/write to make sure it exists
FILE = sys.argv[1]
# try reading the file with read/write to make sure it exists
FILE = sys.argv[1]
try:
fd = os.open(FILE, os.O_RDWR)
except OSError:
sys.exit("Error reading file: " + FILE + ". File not modified. Exiting...")
try:
fd = os.open(FILE, os.O_RDWR)
except OSError:
sys.exit("Error reading file: " + FILE + ". File not modified. Exiting...")
# making sure it's a vbmeta image by reading the magic bytes at the start of the file
magic = os.read(fd, AVB_MAGIC_LEN)
# making sure it's a vbmeta image by reading the magic bytes at the start of the file
magic = os.read(fd, AVB_MAGIC_LEN)
if (magic != AVB_MAGIC):
fd.close()
sys.exit("Error: The provided image is not a valid vbmeta image. File not modified. Exiting...")
if (magic != AVB_MAGIC):
fd.close()
sys.exit("Error: The provided image is not a valid vbmeta image. File not modified. Exiting...")
# set the disable-verity and disable-verification flags at offset 123
try:
os.lseek(fd, FLAGS_OFFSET, os.SEEK_SET)
os.write(fd, FLAGS_TO_SET)
except OSError:
fd.close()
sys.exit("Error: Failed when patching the vbmeta image. Exiting...")
# set the disable-verity and disable-verification flags at offset 123
try:
os.lseek(fd, FLAGS_OFFSET, os.SEEK_SET)
os.write(fd, FLAGS_TO_SET)
except OSError:
fd.close()
sys.exit("Error: Failed when patching the vbmeta image. Exiting...")
# end of program
os.close(fd)
print("Patching successful.")
# end of program
os.close(fd)
print("Patching successful.")