Many upgrades to tpm_lite.

src/platform/tpm_lite/src/testsuite/readonly.c

 /* Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 /* This program mimicks the TPM usage from read-only firmware.  It exercises
  * the TPM functionality needed in the read-only firmware.  It is meant to be
  * integrated with the rest of the read-only firmware.  It is also provided as
  * a test.
  */
@@ skipping 17 matching lines @@
 
 /* This is called once at initialization time.  It may be called again from
  * recovery mode to rebuild the spaces if something incomprehensible happened
  * and the spaces are gone or messed up.  This is called after TPM_Startup and
  * before the spaces are write-locked, so there is a chance that they can be
  * recreated (but who knows---if anything can happen, there are plenty of ways
  * of making this FUBAR).
  */
 void InitializeSpaces(void) {
   uint32_t zero = 0;
+  uint32_t perm = TPM_NV_PER_WRITE_STCLEAR;
 
+  printf("Initializing spaces\n");
   TlclSetNvLocked();  /* useful only the first time */
 
-  TlclDefineSpace(INDEX0, TPM_NV_PER_WRITE_STCLEAR, 4);
+  TlclDefineSpace(INDEX0, perm, 4);
   TlclWrite(INDEX0, (uint8_t *) &zero, 4);
-  TlclDefineSpace(INDEX1, TPM_NV_PER_WRITE_STCLEAR, 4);
+  TlclDefineSpace(INDEX1, perm, 4);
   TlclWrite(INDEX1, (uint8_t *) &zero, 4);
-  TlclDefineSpace(INDEX2, TPM_NV_PER_WRITE_STCLEAR, 4);
+  TlclDefineSpace(INDEX2, perm, 4);
   TlclWrite(INDEX2, (uint8_t *) &zero, 4);
-  TlclDefineSpace(INDEX3, TPM_NV_PER_WRITE_STCLEAR, 4);
+  TlclDefineSpace(INDEX3, perm, 4);
   TlclWrite(INDEX3, (uint8_t *) &zero, 4);
 
-  TlclDefineSpace(INDEX_INITIALIZED, TPM_NV_PER_READ_STCLEAR, 1);
-  TlclReadLock(INDEX_INITIALIZED);
+  perm = TPM_NV_PER_READ_STCLEAR | TPM_NV_PER_WRITE_STCLEAR |
+    TPM_NV_PER_PPWRITE;
+  TlclDefineSpace(INDEX_INITIALIZED, perm, 1);
 }
 
 
 void EnterRecoveryMode(void) {
   printf("entering recovery mode");
   exit(0);
 }
   
 
 int main(void) {
   uint8_t c;
   uint32_t index_0, index_1, index_2, index_3;
 
   TlclLibinit();
 
 #if 0
   TlclStartup();
   TlclSelftestfull();
 #endif
 
   TlclAssertPhysicalPresence();
 
-  /* Checks if initialization has completed.
+  /* Checks if initialization has completed by trying to read-lock a space
+   * that's created at the end of initialization.
    */
-  if (TlclRead(INDEX_INITIALIZED, &c, 1) != TPM_E_DISABLED_CMD) {
+  if (TlclRead(INDEX_INITIALIZED, &c, 0) == TPM_E_BADINDEX) {
     /* The initialization did not complete.
      */
     InitializeSpaces();
   }
 
   /* Checks if spaces are OK or messed up.
    */
-  if (TlclRead(INDEX0, (uint8_t *) &index_0, sizeof(index_0)) != TPM_SUCCESS ||
-      TlclRead(INDEX1, (uint8_t *) &index_1, sizeof(index_1)) != TPM_SUCCESS ||
-      TlclRead(INDEX2, (uint8_t *) &index_2, sizeof(index_2)) != TPM_SUCCESS ||
-      TlclRead(INDEX3, (uint8_t *) &index_3, sizeof(index_3)) != TPM_SUCCESS) {
+  if (TlclRead(INDEX0, (uint8_t*) &index_0, sizeof(index_0)) != TPM_SUCCESS ||
+      TlclRead(INDEX1, (uint8_t*) &index_1, sizeof(index_1)) != TPM_SUCCESS ||
+      TlclRead(INDEX2, (uint8_t*) &index_2, sizeof(index_2)) != TPM_SUCCESS ||
+      TlclRead(INDEX3, (uint8_t*) &index_3, sizeof(index_3)) != TPM_SUCCESS) {
     EnterRecoveryMode();
   }
 
+  /* Writes space, and locks it.  Then attempts to write again.  I really wish
+   * I could use the imperative.
+   */
+  index_0 += 1;
+  if (TlclWrite(INDEX0, (uint8_t*) &index_0, sizeof(index_0) != TPM_SUCCESS)) {
+    error("could not write index 0\n");
+  }
+  TlclWriteLock(INDEX0);
+  if (TlclWrite(INDEX0, (uint8_t*) &index_0, sizeof(index_0)) == TPM_SUCCESS) {
+    error("index 0 is not locked\n");
+  }
+
   /* Done for now.
    */
+  printf("Test completed successfully\n");
   exit(0);
 }
-