mirror of
https://github.com/CloverHackyColor/CloverBootloader.git
synced 2024-12-10 14:23:31 +01:00
221 lines
9.7 KiB
Python
221 lines
9.7 KiB
Python
|
## @file
|
||
|
|
# Convert a binary file to a VOID* PCD value or DSC file VOID* PCD statement.
|
||
|
|
#
|
||
|
|
# Copyright (c) 2016 - 2018, Intel Corporation. All rights reserved.<BR>
|
||
|
|
# SPDX-License-Identifier: BSD-2-Clause-Patent
|
||
|
|
#
|
||
|
|
|
||
|
|
'''
|
||
|
|
BinToPcd
|
||
|
|
'''
|
||
|
|
from __future__ import print_function
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import argparse
|
||
|
|
import re
|
||
|
|
import xdrlib
|
||
|
|
|
||
|
|
#
|
||
|
|
# Globals for help information
|
||
|
|
#
|
||
|
|
__prog__ = 'BinToPcd'
|
||
|
|
__copyright__ = 'Copyright (c) 2016 - 2018, Intel Corporation. All rights reserved.'
|
||
|
|
__description__ = 'Convert one or more binary files to a VOID* PCD value or DSC file VOID* PCD statement.\n'
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
def ValidateUnsignedInteger (Argument):
|
||
|
|
try:
|
||
|
|
Value = int (Argument, 0)
|
||
|
|
except:
|
||
|
|
Message = '{Argument} is not a valid integer value.'.format (Argument = Argument)
|
||
|
|
raise argparse.ArgumentTypeError (Message)
|
||
|
|
if Value < 0:
|
||
|
|
Message = '{Argument} is a negative value.'.format (Argument = Argument)
|
||
|
|
raise argparse.ArgumentTypeError (Message)
|
||
|
|
return Value
|
||
|
|
|
||
|
|
def ValidatePcdName (Argument):
|
||
|
|
if re.split ('[a-zA-Z\_][a-zA-Z0-9\_]*\.[a-zA-Z\_][a-zA-Z0-9\_]*', Argument) != ['', '']:
|
||
|
|
Message = '{Argument} is not in the form <PcdTokenSpaceGuidCName>.<PcdCName>'.format (Argument = Argument)
|
||
|
|
raise argparse.ArgumentTypeError (Message)
|
||
|
|
return Argument
|
||
|
|
|
||
|
|
def ValidateGuidName (Argument):
|
||
|
|
if re.split ('[a-zA-Z\_][a-zA-Z0-9\_]*', Argument) != ['', '']:
|
||
|
|
Message = '{Argument} is not a valid GUID C name'.format (Argument = Argument)
|
||
|
|
raise argparse.ArgumentTypeError (Message)
|
||
|
|
return Argument
|
||
|
|
|
||
|
|
def ByteArray (Buffer, Xdr = False):
|
||
|
|
if Xdr:
|
||
|
|
#
|
||
|
|
# If Xdr flag is set then encode data using the Variable-Length Opaque
|
||
|
|
# Data format of RFC 4506 External Data Representation Standard (XDR).
|
||
|
|
#
|
||
|
|
XdrEncoder = xdrlib.Packer ()
|
||
|
|
for Item in Buffer:
|
||
|
|
XdrEncoder.pack_bytes (Item)
|
||
|
|
Buffer = bytearray (XdrEncoder.get_buffer ())
|
||
|
|
else:
|
||
|
|
#
|
||
|
|
# If Xdr flag is not set, then concatenate all the data
|
||
|
|
#
|
||
|
|
Buffer = bytearray (b''.join (Buffer))
|
||
|
|
#
|
||
|
|
# Return a PCD value of the form '{0x01, 0x02, ...}' along with the PCD length in bytes
|
||
|
|
#
|
||
|
|
return '{' + (', '.join (['0x{Byte:02X}'.format (Byte = Item) for Item in Buffer])) + '}', len (Buffer)
|
||
|
|
|
||
|
|
#
|
||
|
|
# Create command line argument parser object
|
||
|
|
#
|
||
|
|
parser = argparse.ArgumentParser (prog = __prog__,
|
||
|
|
description = __description__ + __copyright__,
|
||
|
|
conflict_handler = 'resolve')
|
||
|
|
parser.add_argument ("-i", "--input", dest = 'InputFile', type = argparse.FileType ('rb'), action='append', required = True,
|
||
|
|
help = "Input binary filename. Multiple input files are combined into a single PCD.")
|
||
|
|
parser.add_argument ("-o", "--output", dest = 'OutputFile', type = argparse.FileType ('w'),
|
||
|
|
help = "Output filename for PCD value or PCD statement")
|
||
|
|
parser.add_argument ("-p", "--pcd", dest = 'PcdName', type = ValidatePcdName,
|
||
|
|
help = "Name of the PCD in the form <PcdTokenSpaceGuidCName>.<PcdCName>")
|
||
|
|
parser.add_argument ("-t", "--type", dest = 'PcdType', default = None, choices = ['VPD', 'HII'],
|
||
|
|
help = "PCD statement type (HII or VPD). Default is standard.")
|
||
|
|
parser.add_argument ("-m", "--max-size", dest = 'MaxSize', type = ValidateUnsignedInteger,
|
||
|
|
help = "Maximum size of the PCD. Ignored with --type HII.")
|
||
|
|
parser.add_argument ("-f", "--offset", dest = 'Offset', type = ValidateUnsignedInteger,
|
||
|
|
help = "VPD offset if --type is VPD. UEFI Variable offset if --type is HII. Must be 8-byte aligned.")
|
||
|
|
parser.add_argument ("-n", "--variable-name", dest = 'VariableName',
|
||
|
|
help = "UEFI variable name. Only used with --type HII.")
|
||
|
|
parser.add_argument ("-g", "--variable-guid", type = ValidateGuidName, dest = 'VariableGuid',
|
||
|
|
help = "UEFI variable GUID C name. Only used with --type HII.")
|
||
|
|
parser.add_argument ("-x", "--xdr", dest = 'Xdr', action = "store_true",
|
||
|
|
help = "Encode PCD using the Variable-Length Opaque Data format of RFC 4506 External Data Representation Standard (XDR)")
|
||
|
|
parser.add_argument ("-v", "--verbose", dest = 'Verbose', action = "store_true",
|
||
|
|
help = "Increase output messages")
|
||
|
|
parser.add_argument ("-q", "--quiet", dest = 'Quiet', action = "store_true",
|
||
|
|
help = "Reduce output messages")
|
||
|
|
parser.add_argument ("--debug", dest = 'Debug', type = int, metavar = '[0-9]', choices = range (0, 10), default = 0,
|
||
|
|
help = "Set debug level")
|
||
|
|
|
||
|
|
#
|
||
|
|
# Parse command line arguments
|
||
|
|
#
|
||
|
|
args = parser.parse_args ()
|
||
|
|
|
||
|
|
#
|
||
|
|
# Read all binary input files
|
||
|
|
#
|
||
|
|
Buffer = []
|
||
|
|
for File in args.InputFile:
|
||
|
|
try:
|
||
|
|
Buffer.append (File.read ())
|
||
|
|
File.close ()
|
||
|
|
except:
|
||
|
|
print ('BinToPcd: error: can not read binary input file {File}'.format (File = File))
|
||
|
|
sys.exit (1)
|
||
|
|
|
||
|
|
#
|
||
|
|
# Convert PCD to an encoded string of hex values and determine the size of
|
||
|
|
# the encoded PCD in bytes.
|
||
|
|
#
|
||
|
|
PcdValue, PcdSize = ByteArray (Buffer, args.Xdr)
|
||
|
|
|
||
|
|
#
|
||
|
|
# Convert binary buffer to a DSC file PCD statement
|
||
|
|
#
|
||
|
|
if args.PcdName is None:
|
||
|
|
#
|
||
|
|
# If PcdName is None, then only a PCD value is being requested.
|
||
|
|
#
|
||
|
|
Pcd = PcdValue
|
||
|
|
if args.Verbose:
|
||
|
|
print ('BinToPcd: Convert binary file to PCD Value')
|
||
|
|
elif args.PcdType is None:
|
||
|
|
#
|
||
|
|
# If --type is neither VPD nor HII, then use PCD statement syntax that is
|
||
|
|
# compatible with [PcdsFixedAtBuild], [PcdsPatchableInModule],
|
||
|
|
# [PcdsDynamicDefault], and [PcdsDynamicExDefault].
|
||
|
|
#
|
||
|
|
if args.MaxSize is None:
|
||
|
|
#
|
||
|
|
# If --max-size is not provided, then do not generate the syntax that
|
||
|
|
# includes the maximum size.
|
||
|
|
#
|
||
|
|
Pcd = ' {Name}|{Value}'.format (Name = args.PcdName, Value = PcdValue)
|
||
|
|
elif args.MaxSize < PcdSize:
|
||
|
|
print ('BinToPcd: error: argument --max-size is smaller than input file.')
|
||
|
|
sys.exit (1)
|
||
|
|
else:
|
||
|
|
Pcd = ' {Name}|{Value}|VOID*|{Size}'.format (Name = args.PcdName, Value = PcdValue, Size = args.MaxSize)
|
||
|
|
|
||
|
|
if args.Verbose:
|
||
|
|
print ('BinToPcd: Convert binary file to PCD statement compatible with PCD sections:')
|
||
|
|
print (' [PcdsFixedAtBuild]')
|
||
|
|
print (' [PcdsPatchableInModule]')
|
||
|
|
print (' [PcdsDynamicDefault]')
|
||
|
|
print (' [PcdsDynamicExDefault]')
|
||
|
|
elif args.PcdType == 'VPD':
|
||
|
|
if args.MaxSize is None:
|
||
|
|
#
|
||
|
|
# If --max-size is not provided, then set maximum size to the size of the
|
||
|
|
# binary input file
|
||
|
|
#
|
||
|
|
args.MaxSize = PcdSize
|
||
|
|
if args.MaxSize < PcdSize:
|
||
|
|
print ('BinToPcd: error: argument --max-size is smaller than input file.')
|
||
|
|
sys.exit (1)
|
||
|
|
if args.Offset is None:
|
||
|
|
#
|
||
|
|
# if --offset is not provided, then set offset field to '*' so build
|
||
|
|
# tools will compute offset of PCD in VPD region.
|
||
|
|
#
|
||
|
|
Pcd = ' {Name}|*|{Size}|{Value}'.format (Name = args.PcdName, Size = args.MaxSize, Value = PcdValue)
|
||
|
|
else:
|
||
|
|
#
|
||
|
|
# --offset value must be 8-byte aligned
|
||
|
|
#
|
||
|
|
if (args.Offset % 8) != 0:
|
||
|
|
print ('BinToPcd: error: argument --offset must be 8-byte aligned.')
|
||
|
|
sys.exit (1)
|
||
|
|
#
|
||
|
|
# Use the --offset value provided.
|
||
|
|
#
|
||
|
|
Pcd = ' {Name}|{Offset}|{Size}|{Value}'.format (Name = args.PcdName, Offset = args.Offset, Size = args.MaxSize, Value = PcdValue)
|
||
|
|
if args.Verbose:
|
||
|
|
print ('BinToPcd: Convert binary file to PCD statement compatible with PCD sections')
|
||
|
|
print (' [PcdsDynamicVpd]')
|
||
|
|
print (' [PcdsDynamicExVpd]')
|
||
|
|
elif args.PcdType == 'HII':
|
||
|
|
if args.VariableGuid is None or args.VariableName is None:
|
||
|
|
print ('BinToPcd: error: arguments --variable-guid and --variable-name are required for --type HII.')
|
||
|
|
sys.exit (1)
|
||
|
|
if args.Offset is None:
|
||
|
|
#
|
||
|
|
# Use UEFI Variable offset of 0 if --offset is not provided
|
||
|
|
#
|
||
|
|
args.Offset = 0
|
||
|
|
#
|
||
|
|
# --offset value must be 8-byte aligned
|
||
|
|
#
|
||
|
|
if (args.Offset % 8) != 0:
|
||
|
|
print ('BinToPcd: error: argument --offset must be 8-byte aligned.')
|
||
|
|
sys.exit (1)
|
||
|
|
Pcd = ' {Name}|L"{VarName}"|{VarGuid}|{Offset}|{Value}'.format (Name = args.PcdName, VarName = args.VariableName, VarGuid = args.VariableGuid, Offset = args.Offset, Value = PcdValue)
|
||
|
|
if args.Verbose:
|
||
|
|
print ('BinToPcd: Convert binary file to PCD statement compatible with PCD sections')
|
||
|
|
print (' [PcdsDynamicHii]')
|
||
|
|
print (' [PcdsDynamicExHii]')
|
||
|
|
|
||
|
|
#
|
||
|
|
# Write PCD value or PCD statement to the output file
|
||
|
|
#
|
||
|
|
try:
|
||
|
|
args.OutputFile.write (Pcd)
|
||
|
|
args.OutputFile.close ()
|
||
|
|
except:
|
||
|
|
#
|
||
|
|
# If output file is not specified or it can not be written, then write the
|
||
|
|
# PCD value or PCD statement to the console
|
||
|
|
#
|
||
|
|
print (Pcd)
|