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)
|
||||
|
||||
for section in pe.sections:
|
||||
name = section.Name.rstrip(b"\x00").decode()
|
||||
name = section.Name.rstrip(b'\x00').decode()
|
||||
file_addr = section.PointerToRawData
|
||||
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:
|
||||
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
|
||||
|
||||
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
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
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(main())
|
||||
|
||||
Reference in New Issue
Block a user