qemu/roms/seabios/scripts/checkrom.py

   1 #!/usr/bin/env python
   2 # Script to check a bios image and report info on it.
   3 #
   4 # Copyright (C) 2008  Kevin O'Connor <kevin@koconnor.net>
   5 #
   6 # This file may be distributed under the terms of the GNU GPLv3 license.
   7
   8 import sys, struct
   9 import layoutrom, buildrom
  10
  11 from python23compat import as_bytes
  12
  13 def subst(data, offset, new):
  14     return data[:offset] + new + data[offset + len(new):]
  15
  16 def checksum(data, start, size, csum):
  17     sumbyte = buildrom.checksum(data[start:start+size])
  18     return subst(data, start+csum, sumbyte)
  19
  20 def main():
  21     # Get args
  22     objinfo, finalsize, rawfile, outfile = sys.argv[1:]
  23
  24     # Read in symbols
  25     objinfofile = open(objinfo, 'r')
  26     symbols = layoutrom.parseObjDump(objinfofile, 'in')[1]
  27
  28     # Read in raw file
  29     f = open(rawfile, 'rb')
  30     rawdata = f.read()
  31     f.close()
  32     datasize = len(rawdata)
  33     finalsize = int(finalsize) * 1024
  34     if finalsize == 0:
  35         finalsize = 64*1024
  36         if datasize > 64*1024:
  37             finalsize = 128*1024
  38             if datasize > 128*1024:
  39                 finalsize = 256*1024
  40     if datasize > finalsize:
  41         print("Error!  ROM doesn't fit (%d > %d)" % (datasize, finalsize))
  42         print("   You have to either increate the size (CONFIG_ROM_SIZE)")
  43         print("   or turn off some features (such as hardware support not")
  44         print("   needed) to make it fit.  Trying a more recent gcc version")
  45         print("   might work too.")
  46         sys.exit(1)
  47
  48     # Sanity checks
  49     start = symbols['code32flat_start'].offset
  50     end = symbols['code32flat_end'].offset
  51     expend = layoutrom.BUILD_BIOS_ADDR + layoutrom.BUILD_BIOS_SIZE
  52     if end != expend:
  53         print("Error!  Code does not end at 0x%x (got 0x%x)" % (
  54             expend, end))
  55         sys.exit(1)
  56     if datasize > finalsize:
  57         print("Error!  Code is too big (0x%x vs 0x%x)" % (
  58             datasize, finalsize))
  59         sys.exit(1)
  60     expdatasize = end - start
  61     if datasize != expdatasize:
  62         print("Error!  Unknown extra data (0x%x vs 0x%x)" % (
  63             datasize, expdatasize))
  64         sys.exit(1)
  65
  66     # Fix up CSM Compatibility16 table
  67     if 'csm_compat_table' in symbols and 'entry_csm' in symbols:
  68         # Field offsets within EFI_COMPATIBILITY16_TABLE
  69         ENTRY_FIELD_OFS = 14 # Compatibility16CallOffset (UINT16)
  70         SIZE_FIELD_OFS = 5   # TableLength (UINT8)
  71         CSUM_FIELD_OFS = 4   # TableChecksum (UINT8)
  72
  73         tableofs = symbols['csm_compat_table'].offset - symbols['code32flat_start'].offset
  74         entry_addr = symbols['entry_csm'].offset - layoutrom.BUILD_BIOS_ADDR
  75         entry_addr = struct.pack('<H', entry_addr)
  76         rawdata = subst(rawdata, tableofs+ENTRY_FIELD_OFS, entry_addr)
  77
  78         tsfield = tableofs+SIZE_FIELD_OFS
  79         tablesize = ord(rawdata[tsfield:tsfield+1])
  80         rawdata = checksum(rawdata, tableofs, tablesize, CSUM_FIELD_OFS)
  81
  82     # Print statistics
  83     runtimesize = end - symbols['code32init_end'].offset
  84     print("Total size: %d  Fixed: %d  Free: %d (used %.1f%% of %dKiB rom)" % (
  85         datasize, runtimesize, finalsize - datasize
  86         , (datasize / float(finalsize)) * 100.0
  87         , int(finalsize / 1024)))
  88
  89     # Write final file
  90     f = open(outfile, 'wb')
  91     f.write((as_bytes("\0") * (finalsize - datasize)) + rawdata)
  92     f.close()
  93
  94 if __name__ == '__main__':
  95     main()