Add support for parsing certificate formats other than raw, DER-encoded cert...

net/base/pem_tokenizer_unittest.cc

Index: net/base/pem_tokenizer_unittest.cc
diff --git a/net/base/pem_tokenizer_unittest.cc b/net/base/pem_tokenizer_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..af2446ca3bce6949e7a5722d76822e641676e276
--- /dev/null
+++ b/net/base/pem_tokenizer_unittest.cc
@@ -0,0 +1,169 @@
+// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "net/base/pem_tokenizer.h"
+
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace net {
+
+TEST(PEMTokenizerTest, BasicParsing) {
+  const char data[] =
+      "-----BEGIN EXPECTED-BLOCK-----\n"
+      "TWF0Y2hlc0FjY2VwdGVkQmxvY2tUeXBl\n"
+      "-----END EXPECTED-BLOCK-----\n";
+  base::StringPiece string_piece(data);
+  std::vector<std::string> accepted_types;
+  accepted_types.push_back("EXPECTED-BLOCK");
+
+  PEMTokenizer tokenizer(string_piece, accepted_types);
+  EXPECT_TRUE(tokenizer.GetNext());
+
+  EXPECT_EQ("EXPECTED-BLOCK", tokenizer.block_type());
+  EXPECT_EQ("MatchesAcceptedBlockType", tokenizer.data());
+
+  EXPECT_FALSE(tokenizer.GetNext());
+}
+
+TEST(PEMTokenizerTest, CarriageReturnLineFeeds) {
+  const char data[] =
+      "-----BEGIN EXPECTED-BLOCK-----\r\n"
+      "TWF0Y2hlc0FjY2VwdGVkQmxvY2tUeXBl\r\n"
+      "-----END EXPECTED-BLOCK-----\r\n";
+  base::StringPiece string_piece(data);
+  std::vector<std::string> accepted_types;
+  accepted_types.push_back("EXPECTED-BLOCK");
+
+  PEMTokenizer tokenizer(string_piece, accepted_types);
+  EXPECT_TRUE(tokenizer.GetNext());
+
+  EXPECT_EQ("EXPECTED-BLOCK", tokenizer.block_type());
+  EXPECT_EQ("MatchesAcceptedBlockType", tokenizer.data());
+
+  EXPECT_FALSE(tokenizer.GetNext());
+}
+
+TEST(PEMTokenizerTest, NoAcceptedBlockTypes) {
+  const char data[] =
+      "-----BEGIN UNEXPECTED-BLOCK-----\n"
+      "SWdub3Jlc1JlamVjdGVkQmxvY2tUeXBl\n"
+      "-----END UNEXPECTED-BLOCK-----\n";
+  base::StringPiece string_piece(data);
+  std::vector<std::string> accepted_types;
+  accepted_types.push_back("EXPECTED-BLOCK");
+
+  PEMTokenizer tokenizer(string_piece, accepted_types);
+  EXPECT_FALSE(tokenizer.GetNext());
+}
+
+TEST(PEMTokenizerTest, MultipleAcceptedBlockTypes) {
+  const char data[] =
+      "-----BEGIN BLOCK-ONE-----\n"
+      "RW5jb2RlZERhdGFPbmU=\n"
+      "-----END BLOCK-ONE-----\n"
+      "-----BEGIN BLOCK-TWO-----\n"
+      "RW5jb2RlZERhdGFUd28=\n"
+      "-----END BLOCK-TWO-----\n";
+  base::StringPiece string_piece(data);
+  std::vector<std::string> accepted_types;
+  accepted_types.push_back("BLOCK-ONE");
+  accepted_types.push_back("BLOCK-TWO");
+
+  PEMTokenizer tokenizer(string_piece, accepted_types);
+  EXPECT_TRUE(tokenizer.GetNext());
+
+  EXPECT_EQ("BLOCK-ONE", tokenizer.block_type());
+  EXPECT_EQ("EncodedDataOne", tokenizer.data());
+
+  EXPECT_TRUE(tokenizer.GetNext());
+
+  EXPECT_EQ("BLOCK-TWO", tokenizer.block_type());
+  EXPECT_EQ("EncodedDataTwo", tokenizer.data());
+
+  EXPECT_FALSE(tokenizer.GetNext());
+}
+
+TEST(PEMTokenizerTest, MissingFooter) {
+  const char data[] =
+      "-----BEGIN MISSING-FOOTER-----\n"
+      "RW5jb2RlZERhdGFPbmU=\n"
+      "-----END MISSING-FOOTER-----\n"
+      "-----BEGIN MISSING-FOOTER-----\n"
+      "RW5jb2RlZERhdGFUd28=\n";
+  base::StringPiece string_piece(data);
+  std::vector<std::string> accepted_types;
+  accepted_types.push_back("MISSING-FOOTER");
+
+  PEMTokenizer tokenizer(string_piece, accepted_types);
+  EXPECT_TRUE(tokenizer.GetNext());
+
+  EXPECT_EQ("MISSING-FOOTER", tokenizer.block_type());
+  EXPECT_EQ("EncodedDataOne", tokenizer.data());
+
+  EXPECT_FALSE(tokenizer.GetNext());
+}
+
+TEST(PEMTokenizerTest, NestedEncoding) {
+  const char data[] =
+      "-----BEGIN BLOCK-ONE-----\n"
+      "RW5jb2RlZERhdGFPbmU=\n"
+      "-----BEGIN BLOCK-TWO-----\n"
+      "RW5jb2RlZERhdGFUd28=\n"
+      "-----END BLOCK-TWO-----\n"
+      "-----END BLOCK-ONE-----\n"
+      "-----BEGIN BLOCK-ONE-----\n"
+      "RW5jb2RlZERhdGFUaHJlZQ==\n"
+      "-----END BLOCK-ONE-----\n";
+  base::StringPiece string_piece(data);
+  std::vector<std::string> accepted_types;
+  accepted_types.push_back("BLOCK-ONE");
+
+  PEMTokenizer tokenizer(string_piece, accepted_types);
+  EXPECT_TRUE(tokenizer.GetNext());
+
+  EXPECT_EQ("BLOCK-ONE", tokenizer.block_type());
+  EXPECT_EQ("EncodedDataThree", tokenizer.data());
+
+  EXPECT_FALSE(tokenizer.GetNext());
+}
+
+TEST(PEMTokenizerTest, EmptyAcceptedTypes) {
+  const char data[] =
+      "-----BEGIN BLOCK-ONE-----\n"
+      "RW5jb2RlZERhdGFPbmU=\n"
+      "-----END BLOCK-ONE-----\n";
+  base::StringPiece string_piece(data);
+  std::vector<std::string> accepted_types;
+
+  PEMTokenizer tokenizer(string_piece, accepted_types);
+  EXPECT_FALSE(tokenizer.GetNext());
+}
+
+TEST(PEMTokenizerTest, BlockWithHeader) {
+  const char data[] =
+      "-----BEGIN BLOCK-ONE-----\n"
+      "Header-One: Data data data\n"
+      "Header-Two: \n"
+      "  continuation\n"
+      "Header-Three: Mix-And,Match\n"
+      "\n"
+      "RW5jb2RlZERhdGFPbmU=\n"
+      "-----END BLOCK-ONE-----\n"
+      "-----BEGIN BLOCK-ONE-----\n"
+      "RW5jb2RlZERhdGFUd28=\n"
+      "-----END BLOCK-ONE-----\n";
+  base::StringPiece string_piece(data);
+  std::vector<std::string> accepted_types;
+  accepted_types.push_back("BLOCK-ONE");
+
+  PEMTokenizer tokenizer(string_piece, accepted_types);
+  EXPECT_TRUE(tokenizer.GetNext());
+
+  EXPECT_EQ("BLOCK-ONE", tokenizer.block_type());
+  EXPECT_EQ("EncodedDataTwo", tokenizer.data());
+
+  EXPECT_FALSE(tokenizer.GetNext());
+}
+
+}  // namespace net