| rjw | 1f88458 | 2022-01-06 17:20:42 +0800 | [diff] [blame^] | 1 | #!/usr/bin/env python |
| 2 | |
| 3 | import argparse |
| 4 | import sys |
| 5 | import os |
| 6 | |
| 7 | def padding(data, size): |
| 8 | return data + '\xff' * (size - len(data)) |
| 9 | |
| 10 | def main(argv): |
| 11 | parser = argparse.ArgumentParser() |
| 12 | parser.add_argument('in_image', help = 'raw input image which has GPT') |
| 13 | parser.add_argument('part_size', help = 'partition size') |
| 14 | args = parser.parse_args() |
| 15 | part_size = int(args.part_size, 16) |
| 16 | file_len = os.path.getsize(args.in_image) |
| 17 | |
| 18 | with open(args.in_image, "rb+") as f: |
| 19 | f.read() |
| 20 | f.write('\xff' * (part_size - file_len)) |
| 21 | |
| 22 | if __name__ == "__main__": |
| 23 | sys.exit(main(sys.argv)) |