check_elf_page_size.py

check_elf_page_size.py

#!/usr/bin/env python3

# Released into the public domain. No attribution required.

__doc__ = """
This script reads an ELF file and prints the page size of the first LOAD segment.
"""

import os
import sys

# third_party

# to install, run `pip install pyelftools`
from elftools.elf.elffile import ELFFile

def get_page_size(elf_file_path):
  with open(elf_file_path, 'rb') as file:
    try:
      elf = ELFFile(file)
      for segment in elf.iter_segments():
        if segment.header.p_type == 'PT_LOAD':
          print(f"Page Size of {elf_file_path}: {segment.header.p_align}")
          break
    except ELFError as e:
      print(f"Failed to parse file: {elf_file_path}: {e}")


def main(argv):
  if len(argv) < 2:
    print(__doc__)
    print(f"Usage: {argv[0]} <path_to_elf_file> [<path_to_elf_file> ...]")
    return 1
  else:
    for elf_file in argv[1:]:
      if os.path.exists(elf_file):
        get_page_size(elf_file)
      else:
        print(f"File not found: {elf_file}")
        sys.exit(1)


if __name__ == "__main__":
  sys.exit(main(sys.argv))