Merge pull request #43 from Ruoyu-y/enhance_go

cc-api · Apr 22, 2024 · 287d71c · 287d71c
2 parents bcc11dc + f75cdd0
commit 287d71c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/golang/cctrusted_vm/cvm.go b/src/golang/cctrusted_vm/cvm.go
@@ -78,6 +78,10 @@ func (d *GenericDevice) Report(nonce, userData string, extraArgs map[string]any)
 			return cctrusted_base.CcReport{}, errors.New("Failed to get generation info.")
 		}
 		generation, _ = strconv.Atoi(string(rawGeneration))
+		// Check if the outblob has been corrupted during file open
+		if generation > 1 {
+			return cctrusted_base.CcReport{}, errors.New("Found corrupted generation.")
+		}
 	}
 
 	if _, err = os.Stat(filepath.Join(tempdir, "provider")); !os.IsNotExist(err) {