summaryrefslogtreecommitdiff
path: root/util/apcb/apcb_edit.py
diff options
context:
space:
mode:
Diffstat (limited to 'util/apcb/apcb_edit.py')
-rwxr-xr-xutil/apcb/apcb_edit.py185
1 files changed, 185 insertions, 0 deletions
diff --git a/util/apcb/apcb_edit.py b/util/apcb/apcb_edit.py
new file mode 100755
index 0000000000..4a7e683ebe
--- /dev/null
+++ b/util/apcb/apcb_edit.py
@@ -0,0 +1,185 @@
+#! /usr/bin/env python
+
+# Script for editing APCB binaries, such as injecting SPDs and GPIO
+# configurations.
+
+import sys
+import re
+import argparse
+from collections import namedtuple
+from struct import *
+
+GPIO_MAGIC = bytes.fromhex('fadeddad' * 3)
+SPD_MAGIC = bytes.fromhex('f005ba110000')
+EMPTY_SPD = b'\x00' * 512
+
+spd_ssp_struct_fmt = '??B?IIBBBxIIBBBx'
+spd_ssp_struct = namedtuple(
+ 'spd_ssp_struct', 'SpdValid, DimmPresent, \
+ PageAddress, NvDimmPresent, \
+ DramManufacturersIDCode, Address, \
+ SpdMuxPresent, MuxI2CAddress, MuxChannel, \
+ Technology, Package, SocketNumber, \
+ ChannelNumber, DimmNumber')
+
+
+def parseargs():
+ parser = argparse.ArgumentParser(description='Inject SPDs and SPD GPIO \
+ selection pins into APCB binaries')
+ parser.add_argument(
+ 'apcb_in',
+ nargs='?',
+ type=argparse.FileType('rb'),
+ default=sys.stdin,
+ help='APCB input file')
+ parser.add_argument(
+ 'apcb_out',
+ nargs='?',
+ type=argparse.FileType('wb'),
+ default=sys.stdout,
+ help='APCB output file')
+ parser.add_argument(
+ '--spd_0_0',
+ type=argparse.FileType('r'),
+ help='SPD input file for channel 0, dimm 0')
+ parser.add_argument(
+ '--spd_0_1',
+ type=argparse.FileType('r'),
+ help='SPD input file for channel 0, dimm 1')
+ parser.add_argument(
+ '--spd_1_0',
+ type=argparse.FileType('r'),
+ help='SPD input file for channel 1, dimm 0')
+ parser.add_argument(
+ '--spd_1_1',
+ type=argparse.FileType('r'),
+ help='SPD input file for channel 1, dimm 1')
+ parser.add_argument(
+ '--hex',
+ action='store_true',
+ help='SPD input file is hex encoded, binary otherwise')
+ parser.add_argument(
+ '--board_id_gpio0',
+ type=int,
+ required=True,
+ nargs=3,
+ help='Board ID GPIO 0: NUMBER IO_MUX BANK_CTRL')
+ parser.add_argument(
+ '--board_id_gpio1',
+ type=int,
+ required=True,
+ nargs=3,
+ help='Board ID GPIO 1: NUMBER IO_MUX BANK_CTRL')
+ parser.add_argument(
+ '--board_id_gpio2',
+ type=int,
+ required=True,
+ nargs=3,
+ help='Board ID GPIO 2: NUMBER IO_MUX BANK_CTRL')
+ parser.add_argument(
+ '--board_id_gpio3',
+ type=int,
+ required=True,
+ nargs=3,
+ help='Board ID GPIO 3: NUMBER IO_MUX BANK_CTRL')
+ return parser.parse_args()
+
+
+def chksum(data):
+ sum = 0
+ for b in data[:16] + data[17:]:
+ sum = (sum + b) & 0xff
+ return (0x100 - sum) & 0xff
+
+
+def inject(orig, insert, offset):
+ return b''.join([orig[:offset], insert, orig[offset + len(insert):]])
+
+
+def main():
+ args = parseargs()
+
+ print("Reading input APCB from %s\n" % (args.apcb_in.name))
+
+ apcb = args.apcb_in.read()
+
+ orig_apcb_len = len(apcb)
+
+ gpio_offset = apcb.find(GPIO_MAGIC)
+ assert gpio_offset > 0, "GPIO magic number not found"
+ print('GPIO magic number found at offset 0x%x\n' % gpio_offset)
+ gpio_array = (args.board_id_gpio0 + args.board_id_gpio1 +
+ args.board_id_gpio2 + args.board_id_gpio3)
+ print('Writing SPD GPIO array %s\n' % gpio_array)
+ apcb = inject(apcb, pack('BBBBBBBBBBBB', *gpio_array), gpio_offset)
+
+ spd_offset = 0
+ while True:
+ spd_offset = apcb.find(SPD_MAGIC, spd_offset)
+ if spd_offset < 0:
+ break
+
+ spd_ssp_offset = spd_offset - calcsize(spd_ssp_struct_fmt)
+ spd_ssp_bytes = apcb[spd_ssp_offset:spd_offset]
+ spd_ssp = spd_ssp_struct._make(
+ unpack(spd_ssp_struct_fmt, spd_ssp_bytes))
+
+ assert spd_ssp.DimmNumber >= 0 and spd_ssp.DimmNumber <= 1, \
+ "Unexpected dimm number found in APCB"
+ assert spd_ssp.ChannelNumber >= 0 and spd_ssp.ChannelNumber <= 1, \
+ "Unexpected channel number found in APCB"
+
+ print("Found SPD magic number with channel %d and dimm %d "
+ "at offset 0x%x\n" % (spd_ssp.ChannelNumber, spd_ssp.DimmNumber,
+ spd_offset))
+
+ dimm_channel = (spd_ssp.ChannelNumber, spd_ssp.DimmNumber)
+ spd = None
+ if dimm_channel == (0, 0) and args.spd_0_0:
+ spd = args.spd_0_0.read()
+ elif dimm_channel == (0, 1) and args.spd_0_1:
+ spd = args.spd_0_1.read()
+ elif dimm_channel == (1, 0) and args.spd_1_0:
+ spd = args.spd_1_0.read()
+ elif dimm_channel == (1, 1) and args.spd_1_1:
+ spd = args.spd_1_0.read()
+
+ if spd:
+ if args.hex:
+ spd = re.sub(r'#.*', '', spd)
+ spd = re.sub(r'\s+', '', spd)
+ spd = bytes.fromhex(spd)
+ else:
+ spd = spd.encode()
+
+ assert len(spd) == 512, \
+ "Expected SPD to be 512 bytes, got %d" % len(spd)
+
+ print("Enabling channel %d, dimm %d and injecting SPD\n" %
+ (spd_ssp.ChannelNumber, spd_ssp.DimmNumber))
+ spd_ssp = spd_ssp._replace(SpdValid=True, DimmPresent=True)
+
+ else:
+ print("Disabling channel %d, dimm %d and clearing SPD\n" %
+ (spd_ssp.ChannelNumber, spd_ssp.DimmNumber))
+ spd_ssp = spd_ssp._replace(SpdValid=False, DimmPresent=False)
+ spd = EMPTY_SPD
+
+ apcb = inject(apcb, pack(spd_ssp_struct_fmt, *spd_ssp), spd_ssp_offset)
+ apcb = inject(apcb, spd, spd_offset)
+
+ spd_offset += 512
+
+ print("Fixing checksum and writing to %s\n" % (args.apcb_out.name))
+
+ apcb = inject(apcb, bytes([chksum(apcb)]), 16)
+
+ assert chksum(apcb) == apcb[16], "Checksum is invalid"
+ assert orig_apcb_len == len(apcb), \
+ "The size of the APCB binary changed, this should not happen."
+
+ args.apcb_out.write(apcb)
+
+
+if __name__ == "__main__":
+ main()