#!/usr/bin/env python3

# Copyright (c) 2021 Arista Networks, Inc.  All rights reserved.
# Arista Networks, Inc. Confidential and Proprietary.

import argparse
import sys

import TpmGeneric.Defs
import TpmGeneric.Tpm

def main():
   parser = argparse.ArgumentParser( description='Read TPM PCR register' )
   parser.add_argument( 'pcrIndex', metavar='PCR', type=int,
                        help='Index of the pcr to extend' )
   args = parser.parse_args()

   try:
      tpm = TpmGeneric.Tpm.TpmGeneric()
      stdout = sys.stdout.buffer
      stdout.write( tpm.pcrRead( args.pcrIndex ) + b'\n' )
      return 0
   except TpmGeneric.Defs.Error as e:
      print( e )
      return 1

if __name__ == '__main__':
   sys.exit( main() )
