pcr.py: add initial support for Google vTPM (#128)

Signed-off-by: Linda <bluewishken@163.com>
cc-api · May 29, 2024 · adbbe84 · adbbe84
1 parent d58732d
commit adbbe84
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/common/python/cctrusted_base/tcg.py b/common/python/cctrusted_base/tcg.py
@@ -52,8 +52,8 @@ def get_algorithm_string(alg_id: int) -> str:
  return "UNKNOWN"
 
  def __init__(self, alg_id: int) -> None:
- assert alg_id in TcgAlgorithmRegistry.TPM_ALG_TABLE, \
- "invalid parameter alg_id"
+ #assert alg_id in TcgAlgorithmRegistry.TPM_ALG_TABLE, \
+ # "invalid parameter alg_id"
  self._alg_id = alg_id
 
  @property

diff --git a/common/python/cctrusted_base/tpm/pcr.py b/common/python/cctrusted_base/tpm/pcr.py
@@ -3,10 +3,15 @@
 """
 
 from cctrusted_base.imr import TcgIMR
+from cctrusted_base.tcg import TcgAlgorithmRegistry
 
 class TpmPCR(TcgIMR):
  """PCR class defined for TPM"""
 
  @property
  def max_index(self):
  return 23
+
+ def __init__(self, index, digest_hash):
+ super().__init__(index, TcgAlgorithmRegistry.TPM_ALG_SHA256,
+ digest_hash)