tools/extract-map.py

   1 #!/usr/bin/python3
   2
   3 import argparse
   4
   5
   6 def line_index(lines, start):
   7     for i, line in enumerate(lines):
   8         if line.startswith(start):
   9             return i
  10
  11
  12 def extract_vars(lines):
  13     result = []
  14     lines = lines[line_index(lines, ".data"):line_index(lines, ".memory")]
  15     i = 0
  16     while i < len(lines):
  17         line = lines[i]
  18         i += 1
  19         if line.startswith("*"):
  20             continue
  21         if line.startswith(" .data.") or line.startswith(" .bss."):
  22             fields = (line + lines[i]).split()
  23             # print(fields)
  24             i += 1
  25             var = fields[0].split(".")[-1]
  26             offset = int(fields[1], 16)
  27             size = int(fields[2], 16)
  28             result.append((var, offset, size))
  29     return result
  30
  31
  32 def main():
  33     parser = argparse.ArgumentParser(description="Extract firmware.map")
  34     parser.add_argument("file", type=argparse.FileType("r"))
  35
  36     args = parser.parse_args()
  37
  38     f = args.file
  39     lines = f.readlines()
  40
  41     vars = extract_vars(lines)
  42     vars.sort(key=lambda var: "%08d %s" % (var[2], var[0]))
  43     for var, offset, size in vars:
  44         print("%s\t %d" % (var, size))
  45
  46
  47 if __name__ == "__main__":
  48     main()