Merge "Add testcases for load_keys()."

class VerifierFailureTest : public VerifierTest {

};

TEST(VerifierTest, load_keys_multiple_keys) {

  std::string testkey_v4;

  ASSERT_TRUE(android::base::ReadFileToString(from_testdata_base("testkey_v4.txt"), &testkey_v4));

  std::string testkey_v3;

  ASSERT_TRUE(android::base::ReadFileToString(from_testdata_base("testkey_v3.txt"), &testkey_v3));

  std::string keys = testkey_v4 + "," + testkey_v3 + "," + testkey_v4;

  TemporaryFile key_file1;

  ASSERT_TRUE(android::base::WriteStringToFile(keys, key_file1.path));

  std::vector<Certificate> certs;

  ASSERT_TRUE(load_keys(key_file1.path, certs));

  ASSERT_EQ(3U, certs.size());

}

TEST(VerifierTest, load_keys_invalid_keys) {

  std::vector<Certificate> certs;

  ASSERT_FALSE(load_keys("/doesntexist", certs));

  // Empty file.

  TemporaryFile key_file1;

  ASSERT_FALSE(load_keys(key_file1.path, certs));

  // Invalid contents.

  ASSERT_TRUE(android::base::WriteStringToFile("invalid", key_file1.path));

  ASSERT_FALSE(load_keys(key_file1.path, certs));

  std::string testkey_v4;

  ASSERT_TRUE(android::base::ReadFileToString(from_testdata_base("testkey_v4.txt"), &testkey_v4));

  // Invalid key version: "v4 ..." => "v6 ...".

  std::string invalid_key2(testkey_v4);

  invalid_key2[1] = '6';

  TemporaryFile key_file2;

  ASSERT_TRUE(android::base::WriteStringToFile(invalid_key2, key_file2.path));

  ASSERT_FALSE(load_keys(key_file2.path, certs));

  // Invalid key content: inserted extra bytes ",2209831334".

  std::string invalid_key3(testkey_v4);

  invalid_key3.insert(invalid_key2.size() - 2, ",2209831334");

  TemporaryFile key_file3;

  ASSERT_TRUE(android::base::WriteStringToFile(invalid_key3, key_file3.path));

  ASSERT_FALSE(load_keys(key_file3.path, certs));

  // Invalid key: the last key must not end with an extra ','.

  std::string invalid_key4 = testkey_v4 + ",";

  TemporaryFile key_file4;

  ASSERT_TRUE(android::base::WriteStringToFile(invalid_key4, key_file4.path));

  ASSERT_FALSE(load_keys(key_file4.path, certs));

  // Invalid key separator.

  std::string invalid_key5 = testkey_v4 + ";" + testkey_v4;

  TemporaryFile key_file5;

  ASSERT_TRUE(android::base::WriteStringToFile(invalid_key5, key_file5.path));

  ASSERT_FALSE(load_keys(key_file5.path, certs));

}

TEST_P(VerifierSuccessTest, VerifySucceed) {

  ASSERT_EQ(verify_file(memmap.addr, memmap.length, certs, nullptr), VERIFY_SUCCESS);

}