Implemented pipelined hash calculation in LoadFirmware()

tests/vboot_common2_tests.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.
  *
  * Tests for firmware image library.
  */
 
 #include <stdio.h>
 #include <stdlib.h>
 
@@ skipping 27 matching lines @@
     TEST_EQ(rsa->algorithm, key->algorithm, "PublicKeyToRSA() algorithm");
     RSAPublicKeyFree(rsa);
   }
 }
 
 
 static void VerifyDataTest(const VbPublicKey* public_key,
                            const VbPrivateKey* private_key) {
 
   const uint8_t test_data[] = "This is some test data to sign.";
-  VbSignature *sig;
-  RSAPublicKey *rsa;
+  VbSignature* sig;
+  RSAPublicKey* rsa;
 
   sig = CalculateSignature(test_data, sizeof(test_data), private_key);
   rsa = PublicKeyToRSA(public_key);
   TEST_NEQ(sig && rsa, 0, "VerifyData() prerequisites");
   if (!sig || !rsa)
     return;
 
   TEST_EQ(VerifyData(test_data, sig, rsa), 0, "VerifyData() ok");
 
   sig->sig_size -= 16;
   TEST_EQ(VerifyData(test_data, sig, rsa), 1, "VerifyData() wrong sig size");
   sig->sig_size += 16;
 
   GetSignatureData(sig)[0] ^= 0x5A;
   TEST_EQ(VerifyData(test_data, sig, rsa), 1, "VerifyData() wrong sig");
 
   RSAPublicKeyFree(rsa);
   Free(sig);
 }
 
 
+static void VerifyDigestTest(const VbPublicKey* public_key,
+                           const VbPrivateKey* private_key) {
+
+  const uint8_t test_data[] = "This is some other test data to sign.";
+  VbSignature* sig;
+  RSAPublicKey* rsa;
+  uint8_t* digest;
+
+  sig = CalculateSignature(test_data, sizeof(test_data), private_key);
+  rsa = PublicKeyToRSA(public_key);
+  digest = DigestBuf(test_data, sizeof(test_data), public_key->algorithm);
+  TEST_NEQ(sig && rsa && digest, 0, "VerifyData() prerequisites");
+  if (!sig || !rsa || !digest)
+    return;
+
+  TEST_EQ(VerifyDigest(digest, sig, rsa), 0, "VerifyDigest() ok");
+
+  GetSignatureData(sig)[0] ^= 0x5A;
+  TEST_EQ(VerifyDigest(digest, sig, rsa), 1, "VerifyDigest() wrong sig");
+
+  RSAPublicKeyFree(rsa);
+  Free(sig);
+  Free(digest);
+}
+
+
 static void ReSignKernelPreamble(VbKernelPreambleHeader *h,
                                  const VbPrivateKey *key) {
   VbSignature *sig = CalculateSignature((const uint8_t*)h,
                                         h->preamble_signature.data_size, key);
 
   SignatureCopy(&h->preamble_signature, sig);
   Free(sig);
 }
 
 
@@ skipping 113 matching lines @@
   }
 
   public_key = PublicKeyReadKeyb(argv[3], key_algorithm, 1);
   if (!public_key) {
     fprintf(stderr, "Error reading public_key");
     return 1;
   }
 
   VerifyPublicKeyToRSA(public_key);
   VerifyDataTest(public_key, private_key);
+  VerifyDigestTest(public_key, private_key);
   VerifyKernelPreambleTest(public_key, private_key);
 
   if (public_key)
     Free(public_key);
   if (private_key)
     Free(private_key);
 
   return error_code;
 }