[Refinery] Introduce TypePropagatorAnalyzer - pointer types.

syzygy/refinery/types/type_repository_unittest.cc

 // Copyright 2015 Google Inc. All Rights Reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
 //
 //     http://www.apache.org/licenses/LICENSE-2.0
 //
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
 #include "syzygy/refinery/types/type_repository.h"
 
 #include "base/memory/ref_counted.h"
 #include "gtest/gtest.h"
+#include "syzygy/pe/pe_file.h"
 #include "syzygy/refinery/types/type.h"
 
 namespace refinery {
 
 TEST(TypeRepositoryTest, AddType) {
   scoped_refptr<TypeRepository> repo = new TypeRepository();
   EXPECT_EQ(0U, repo->size());
 
   // Returns a NULL type for unknown TypeId.
   EXPECT_FALSE(repo->GetType(1));
@@ skipping 48 matching lines @@
   // There is still no object with id 1.
   EXPECT_FALSE(repo->GetType(1));
 
   // This index is already taken.
   EXPECT_FALSE(repo->AddTypeWithId(t3, kId1));
 
   EXPECT_EQ(t1, repo->GetType(kId1));
   EXPECT_EQ(t2, repo->GetType(kId2));
 }
 
+TEST(TypeRepositoryTest, GetSignature) {
+  pe::PEFile::Signature retrieved_sig;
+
+  scoped_refptr<TypeRepository> repo = new TypeRepository();
+  ASSERT_FALSE(repo->GetModuleSignature(&retrieved_sig));
+
+  pe::PEFile::Signature sig(L"Path", core::AbsoluteAddress(1U), 2, 3, 4);
+  repo = new TypeRepository(sig);
+  ASSERT_TRUE(repo->GetModuleSignature(&retrieved_sig));
+  ASSERT_EQ(sig, retrieved_sig);
+}
+
 TEST(TypeRepositoryTest, Iteration) {
   scoped_refptr<TypeRepository> repo = new TypeRepository();
 
   repo->AddType(new BasicType(L"one", 4));
   repo->AddType(new BasicType(L"two", 4));
   repo->AddType(new BasicType(L"three", 4));
   EXPECT_EQ(3U, repo->size());
 
   size_t iterated = 0;
   for (auto type : *repo) {
@@ skipping 35 matching lines @@
   ASSERT_EQ(2, matching_types.size());
 
   // Match with single hit.
   matching_types.clear();
   index->GetTypes(kTypeNameTwo, &matching_types);
   ASSERT_EQ(1, matching_types.size());
   ASSERT_EQ(two.get(), matching_types[0].get());
 }
 
 }  // namespace refinery