| rjw | 1f88458 | 2022-01-06 17:20:42 +0800 | [diff] [blame^] | 1 | #!/usr/bin/env python |
| 2 | |
| 3 | import sys |
| 4 | import os |
| 5 | |
| 6 | |
| 7 | if len(sys.argv) != 2: |
| 8 | print('Usage : align_4kb [file_name]') |
| 9 | sys.exit() |
| 10 | |
| 11 | if not os.path.isfile(sys.argv[1]): |
| 12 | print(sys.argv[1] + ' not exist') |
| 13 | sys.exit() |
| 14 | |
| 15 | if os.path.getsize(sys.argv[1]) % 4096 == 0: |
| 16 | print('file size is 4KB alignment') |
| 17 | sys.exit() |
| 18 | |
| 19 | file_src = sys.argv[1] |
| 20 | file_dst = os.path.splitext( |
| 21 | file_src)[0] + "_original" + os.path.splitext(file_src)[1] |
| 22 | |
| 23 | if os.path.exists(file_dst): |
| 24 | os.remove(file_dst) |
| 25 | |
| 26 | os.rename(file_src, file_dst) |
| 27 | |
| 28 | file_src = file_dst |
| 29 | file_dst = sys.argv[1] |
| 30 | |
| 31 | with open(file_dst, 'ab') as fdst: |
| 32 | with open(file_src, 'rb') as fsrc: |
| 33 | data = fsrc.read() |
| 34 | len_diff = 4096 - len(data) % 4096 |
| 35 | fdst.write(data) |
| 36 | if len_diff: |
| 37 | fdst.write(b'\xff' * len_diff) |