check-efi-alignment: apply "ruff format"
This commit is contained in:
@@ -15,22 +15,23 @@ def main():
|
|||||||
pe = pefile.PE(sys.argv[1], fast_load=True)
|
pe = pefile.PE(sys.argv[1], fast_load=True)
|
||||||
|
|
||||||
for section in pe.sections:
|
for section in pe.sections:
|
||||||
name = section.Name.rstrip(b"\x00").decode()
|
name = section.Name.rstrip(b'\x00').decode()
|
||||||
file_addr = section.PointerToRawData
|
file_addr = section.PointerToRawData
|
||||||
virt_addr = section.VirtualAddress
|
virt_addr = section.VirtualAddress
|
||||||
print(f"{name:10s} file=0x{file_addr:08x} virt=0x{virt_addr:08x}")
|
print(f'{name:10s} file=0x{file_addr:08x} virt=0x{virt_addr:08x}')
|
||||||
|
|
||||||
if file_addr % 512 != 0:
|
if file_addr % 512 != 0:
|
||||||
print(f"File address of {name} section is not aligned to 512 bytes", file=sys.stderr)
|
print(f'File address of {name} section is not aligned to 512 bytes', file=sys.stderr)
|
||||||
return 1
|
return 1
|
||||||
|
|
||||||
if virt_addr % 512 != 0:
|
if virt_addr % 512 != 0:
|
||||||
print(f"Virt address of {name} section is not aligned to 512 bytes", file=sys.stderr)
|
print(f'Virt address of {name} section is not aligned to 512 bytes', file=sys.stderr)
|
||||||
return 1
|
return 1
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
if len(sys.argv) != 2:
|
if len(sys.argv) != 2:
|
||||||
print(f"Usage: {sys.argv[0]} pe-image")
|
print(f'Usage: {sys.argv[0]} pe-image')
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
sys.exit(main())
|
sys.exit(main())
|
||||||
|
|||||||
Reference in New Issue
Block a user