Merge pull request #221 from crocs-muni/feat/track-doc-hashes

Add document hashes.
crocs-muni · May 30, 2022 · 191997b · 191997b
2 parents de9e12c + ff6deb9
commit 191997b
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/sec_certs/sample/common_criteria.py b/sec_certs/sample/common_criteria.py
@@ -98,8 +98,12 @@ class InternalState(ComplexSerializableType):
 
         st_pdf_path: Path
         report_pdf_path: Path
+        st_pdf_hash: Optional[str]
+        report_pdf_hash: Optional[str]
         st_txt_path: Path
         report_txt_path: Path
+        st_txt_hash: Optional[str]
+        report_txt_hash: Optional[str]
 
         def __init__(
             self,
@@ -124,8 +128,12 @@ def serialized_attributes(self) -> List[str]:
             return [
                 "st_download_ok",
                 "report_download_ok",
+                "st_pdf_hash",
+                "report_pdf_hash",
                 "st_convert_ok",
                 "report_convert_ok",
+                "st_txt_hash",
+                "report_txt_hash",
                 "st_extract_ok",
                 "report_extract_ok",
                 "errors",
@@ -673,6 +681,8 @@ def download_pdf_report(cert: CommonCriteriaCert) -> CommonCriteriaCert:
             logger.error(f"Cert dgst: {cert.dgst} " + error_msg)
             cert.state.report_download_ok = False
             cert.state.errors.append(error_msg)
+        else:
+            cert.state.report_pdf_hash = helpers.get_sha256_filepath(cert.state.report_pdf_path)
         return cert
 
     @staticmethod
@@ -693,6 +703,8 @@ def download_pdf_st(cert: CommonCriteriaCert) -> CommonCriteriaCert:
             logger.error(f"Cert dgst: {cert.dgst}" + error_msg)
             cert.state.st_download_ok = False
             cert.state.errors.append(error_msg)
+        else:
+            cert.state.st_pdf_hash = helpers.get_sha256_filepath(cert.state.st_pdf_path)
         return cert
 
     @staticmethod
@@ -709,6 +721,8 @@ def convert_report_pdf(cert: CommonCriteriaCert) -> CommonCriteriaCert:
             logger.error(f"Cert dgst: {cert.dgst}" + error_msg)
             cert.state.report_convert_ok = False
             cert.state.errors.append(error_msg)
+        else:
+            cert.state.report_txt_hash = helpers.get_sha256_filepath(cert.state.report_txt_path)
         return cert
 
     @staticmethod
@@ -725,6 +739,8 @@ def convert_st_pdf(cert: CommonCriteriaCert) -> CommonCriteriaCert:
             logger.error(f"Cert dgst: {cert.dgst}" + error_msg)
             cert.state.st_convert_ok = False
             cert.state.errors.append(error_msg)
+        else:
+            cert.state.st_txt_hash = helpers.get_sha256_filepath(cert.state.st_txt_path)
         return cert
 
     @staticmethod