aboutsummaryrefslogtreecommitdiff
path: root/scripts/checkrom.py
diff options
context:
space:
mode:
authorKevin O'Connor <kevin@koconnor.net>2013-08-17 10:13:14 -0400
committerKevin O'Connor <kevin@koconnor.net>2013-08-17 10:23:46 -0400
commitb942ce0a03a8b83c4b52e8af5d649b3661034a2d (patch)
treed0625ea547ceba4b857d8ddc1502c5f33c60c3fb /scripts/checkrom.py
parent915f64aee37159d21760d44cdc5778220468fe0b (diff)
downloadseabios-b942ce0a03a8b83c4b52e8af5d649b3661034a2d.zip
seabios-b942ce0a03a8b83c4b52e8af5d649b3661034a2d.tar.gz
seabios-b942ce0a03a8b83c4b52e8af5d649b3661034a2d.tar.bz2
Rename tools/ directory to scripts/ directory.
It's common for other projects (eg, QEMU, Linux) to put build scripts into a "scripts/" directory. There's no reason for SeaBIOS to be different, so rename the "tools/" directory to "scripts/". Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
Diffstat (limited to 'scripts/checkrom.py')
-rwxr-xr-xscripts/checkrom.py88
1 files changed, 88 insertions, 0 deletions
diff --git a/scripts/checkrom.py b/scripts/checkrom.py
new file mode 100755
index 0000000..6f07ac8
--- /dev/null
+++ b/scripts/checkrom.py
@@ -0,0 +1,88 @@
+#!/usr/bin/env python
+# Script to check a bios image and report info on it.
+#
+# Copyright (C) 2008 Kevin O'Connor <kevin@koconnor.net>
+#
+# This file may be distributed under the terms of the GNU GPLv3 license.
+
+import sys
+import layoutrom
+
+def subst(data, offset, new):
+ return data[:offset] + new + data[offset + len(new):]
+
+def checksum(data, start, size, csum):
+ sumbyte = 0
+ while size:
+ sumbyte = sumbyte + ord(data[start + size - 1])
+ size = size - 1
+ sumbyte = (0x100 - sumbyte) & 0xff
+ return subst(data, start+csum, chr(sumbyte))
+
+def main():
+ # Get args
+ objinfo, rawfile, outfile = sys.argv[1:]
+
+ # Read in symbols
+ objinfofile = open(objinfo, 'rb')
+ symbols = layoutrom.parseObjDump(objinfofile, 'in')[1]
+
+ # Read in raw file
+ f = open(rawfile, 'rb')
+ rawdata = f.read()
+ f.close()
+ datasize = len(rawdata)
+ finalsize = 64*1024
+ if datasize > 64*1024:
+ finalsize = 128*1024
+ if datasize > 128*1024:
+ finalsize = 256*1024
+
+ # Sanity checks
+ start = symbols['code32flat_start'].offset
+ end = symbols['code32flat_end'].offset
+ expend = layoutrom.BUILD_BIOS_ADDR + layoutrom.BUILD_BIOS_SIZE
+ if end != expend:
+ print "Error! Code does not end at 0x%x (got 0x%x)" % (
+ expend, end)
+ sys.exit(1)
+ if datasize > finalsize:
+ print "Error! Code is too big (0x%x vs 0x%x)" % (
+ datasize, finalsize)
+ sys.exit(1)
+ expdatasize = end - start
+ if datasize != expdatasize:
+ print "Error! Unknown extra data (0x%x vs 0x%x)" % (
+ datasize, expdatasize)
+ sys.exit(1)
+
+ # Fix up CSM Compatibility16 table
+ if 'csm_compat_table' in symbols and 'entry_csm' in symbols:
+ # Field offsets within EFI_COMPATIBILITY16_TABLE
+ ENTRY_FIELD_OFS = 14 # Compatibility16CallOffset (UINT16)
+ SIZE_FIELD_OFS = 5 # TableLength (UINT8)
+ CSUM_FIELD_OFS = 4 # TableChecksum (UINT8)
+
+ tableofs = symbols['csm_compat_table'].offset - symbols['code32flat_start'].offset
+ entry_addr = symbols['entry_csm'].offset - layoutrom.BUILD_BIOS_ADDR
+ byte1 = chr(entry_addr & 0xff)
+ byte2 = chr(entry_addr >> 8)
+ rawdata = subst(rawdata, tableofs+ENTRY_FIELD_OFS, byte1+byte2)
+
+ tablesize = ord(rawdata[tableofs+SIZE_FIELD_OFS])
+ rawdata = checksum(rawdata, tableofs, tablesize, CSUM_FIELD_OFS)
+
+ # Print statistics
+ runtimesize = end - symbols['code32init_end'].offset
+ print "Total size: %d Fixed: %d Free: %d (used %.1f%% of %dKiB rom)" % (
+ datasize, runtimesize, finalsize - datasize
+ , (datasize / float(finalsize)) * 100.0
+ , finalsize / 1024)
+
+ # Write final file
+ f = open(outfile, 'wb')
+ f.write(("\0" * (finalsize - datasize)) + rawdata)
+ f.close()
+
+if __name__ == '__main__':
+ main()