'''\r
Rsa2048Sha256Sign\r
'''\r
+from __future__ import print_function\r
\r
import os\r
import sys\r
# UINT8 Signature[256];\r
# } EFI_CERT_BLOCK_RSA_2048_SHA256;\r
#\r
-EFI_CERT_BLOCK_RSA_2048_SHA256 = collections.namedtuple('EFI_CERT_BLOCK_RSA_2048_SHA256', ['HashType','PublicKey','Signature'])\r
+EFI_CERT_BLOCK_RSA_2048_SHA256 = collections.namedtuple('EFI_CERT_BLOCK_RSA_2048_SHA256', ['HashType', 'PublicKey', 'Signature'])\r
EFI_CERT_BLOCK_RSA_2048_SHA256_STRUCT = struct.Struct('16s256s256s')\r
\r
#\r
parser.add_argument("--private-key", dest='PrivateKeyFile', type=argparse.FileType('rb'), help="specify the private key filename. If not specified, a test signing key is used.")\r
parser.add_argument("-v", "--verbose", dest='Verbose', action="store_true", help="increase output messages")\r
parser.add_argument("-q", "--quiet", dest='Quiet', action="store_true", help="reduce output messages")\r
- parser.add_argument("--debug", dest='Debug', type=int, metavar='[0-9]', choices=range(0,10), default=0, help="set debug level")\r
+ parser.add_argument("--debug", dest='Debug', type=int, metavar='[0-9]', choices=range(0, 10), default=0, help="set debug level")\r
parser.add_argument(metavar="input_file", dest='InputFile', type=argparse.FileType('rb'), help="specify the input filename")\r
\r
#\r
try:\r
Process = subprocess.Popen('%s version' % (OpenSslCommand), stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)\r
except: \r
- print 'ERROR: Open SSL command not available. Please verify PATH or set OPENSSL_PATH'\r
+ print('ERROR: Open SSL command not available. Please verify PATH or set OPENSSL_PATH')\r
sys.exit(1)\r
\r
Version = Process.communicate()\r
- if Process.returncode <> 0:\r
- print 'ERROR: Open SSL command not available. Please verify PATH or set OPENSSL_PATH'\r
+ if Process.returncode != 0:\r
+ print('ERROR: Open SSL command not available. Please verify PATH or set OPENSSL_PATH')\r
sys.exit(Process.returncode)\r
- print Version[0]\r
+ print(Version[0])\r
\r
#\r
# Read input file into a buffer and save input filename\r
#\r
OutputDir = os.path.dirname(args.OutputFile)\r
if not os.path.exists(OutputDir):\r
- print 'ERROR: The output path does not exist: %s' % OutputDir\r
+ print('ERROR: The output path does not exist: %s' % OutputDir)\r
sys.exit(1)\r
args.OutputFileName = args.OutputFile\r
\r
args.PrivateKeyFile = open(args.PrivateKeyFileName, 'rb')\r
args.PrivateKeyFile.close()\r
except:\r
- print 'ERROR: test signing private key file %s missing' % (args.PrivateKeyFileName)\r
+ print('ERROR: test signing private key file %s missing' % (args.PrivateKeyFileName))\r
sys.exit(1)\r
\r
#\r
PublicKeyHexString = Process.communicate()[0].split('=')[1].strip()\r
PublicKey = ''\r
while len(PublicKeyHexString) > 0:\r
- PublicKey = PublicKey + chr(int(PublicKeyHexString[0:2],16))\r
+ PublicKey = PublicKey + chr(int(PublicKeyHexString[0:2], 16))\r
PublicKeyHexString=PublicKeyHexString[2:]\r
- if Process.returncode <> 0:\r
+ if Process.returncode != 0:\r
sys.exit(Process.returncode)\r
\r
if args.MonotonicCountStr:\r
#\r
Process = subprocess.Popen('%s dgst -sha256 -sign "%s"' % (OpenSslCommand, args.PrivateKeyFileName), stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)\r
Signature = Process.communicate(input=FullInputFileBuffer)[0]\r
- if Process.returncode <> 0:\r
+ if Process.returncode != 0:\r
sys.exit(Process.returncode)\r
\r
#\r
#\r
# Verify that the Hash Type matches the expected SHA256 type\r
#\r
- if uuid.UUID(bytes_le = Header.HashType) <> EFI_HASH_ALGORITHM_SHA256_GUID:\r
- print 'ERROR: unsupport hash GUID'\r
+ if uuid.UUID(bytes_le = Header.HashType) != EFI_HASH_ALGORITHM_SHA256_GUID:\r
+ print('ERROR: unsupport hash GUID')\r
sys.exit(1)\r
\r
#\r
# Verify the public key\r
#\r
- if Header.PublicKey <> PublicKey:\r
- print 'ERROR: Public key in input file does not match public key from private key file'\r
+ if Header.PublicKey != PublicKey:\r
+ print('ERROR: Public key in input file does not match public key from private key file')\r
sys.exit(1)\r
\r
FullInputFileBuffer = args.InputFileBuffer\r
# \r
Process = subprocess.Popen('%s dgst -sha256 -prverify "%s" -signature %s' % (OpenSslCommand, args.PrivateKeyFileName, args.OutputFileName), stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)\r
Process.communicate(input=FullInputFileBuffer)\r
- if Process.returncode <> 0:\r
- print 'ERROR: Verification failed'\r
+ if Process.returncode != 0:\r
+ print('ERROR: Verification failed')\r
os.remove (args.OutputFileName)\r
sys.exit(Process.returncode)\r
\r