1 files changed, 7 insertions, 33 deletions
diff --git a/tests/tpm_lite/earlynvram2.c b/tests/tpm_lite/earlynvram2.c
index 91b8d2ec..2bd2d712 100644
--- a/tests/tpm_lite/earlynvram2.c
+++ b/tests/tpm_lite/earlynvram2.c
@@ -11,43 +11,17 @@
 #include <stdlib.h>
 
 #include "tlcl.h"
+#include "tlcl_tests.h"
 #include "utility.h"
 
-#define INDEX0 0xcafe
-
 int main(int argc, char** argv) {
-  uint32_t perm;
-  uint32_t result;
   uint32_t x;
 
   TlclLibInit();
-  TlclStartup();
-  TlclContinueSelfTest();
-
-  do {
-    result = TlclAssertPhysicalPresence();
-    printf("result of AssertPP = %d\n", result);
-  } while (result == TPM_E_DOING_SELFTEST ||
-           result == TPM_E_NEEDS_SELFTEST);
-
-  if (result != TPM_SUCCESS) {
-    error("AssertPP failed with error %d\n", result);
-  }
-
-  do {
-    result = TlclWrite(INDEX0, (uint8_t*) &x, sizeof(x));
-    printf("result of WriteValue = %d\n", result);
-  } while (result == TPM_E_DOING_SELFTEST ||
-           result == TPM_E_NEEDS_SELFTEST);
-
-  if (result == TPM_E_BADINDEX) {
-    VBDEBUG(("creating INDEX0\n"));
-    perm = TPM_NV_PER_PPWRITE;
-    TlclDefineSpace(INDEX0, perm, sizeof(uint32_t));
-  } else if (result != TPM_SUCCESS) {
-    error("Write failed with result %d\n", result);
-  }
-
-  printf("Test completed successfully\n");
-  exit(0);
+  TPM_CHECK(TlclStartup());
+  TPM_CHECK(TlclContinueSelfTest());
+  TPM_CHECK(TlclAssertPhysicalPresence());
+  TPM_CHECK(TlclWrite(INDEX0, (uint8_t*) &x, sizeof(x)));
+  printf("TEST SUCCEEDED\n");
+  return 0;
 }