Chromium Code Reviews| Index: syzygy/refinery/types/type_repository.h |
| diff --git a/syzygy/refinery/types/type_repository.h b/syzygy/refinery/types/type_repository.h |
| index 1c49d294f5de963774f2ff2772a3d6faaad50f2e..858921c243d2e7279636d3a5f2dab09b803d4eb9 100644 |
| --- a/syzygy/refinery/types/type_repository.h |
| +++ b/syzygy/refinery/types/type_repository.h |
| @@ -22,6 +22,7 @@ |
| #include "base/macros.h" |
| #include "base/containers/hash_tables.h" |
| #include "base/memory/ref_counted.h" |
| +#include "syzygy/pe/pe_file.h" |
| namespace refinery { |
| @@ -36,6 +37,7 @@ class TypeRepository : public base::RefCounted<TypeRepository> { |
| class Iterator; |
| TypeRepository(); |
| + explicit TypeRepository(const pe::PEFile::Signature& signature); |
|
Sigurður Ásgeirsson
2015/11/26 21:45:12
I think we want to abstract the module ID to somet
manzagop (departed)
2015/11/27 15:20:28
Yes, I was thinking we'd have a Signature interfac
|
| // Retrieve a type by @p id. |
| TypePtr GetType(TypeId id) const; |
| @@ -49,6 +51,10 @@ class TypeRepository : public base::RefCounted<TypeRepository> { |
| // @returns true on success, failure typically means id is already taken. |
| bool AddTypeWithId(TypePtr type, TypeId id); |
| + |
| + // Get the signature for the module this type represents. |
| + bool GetModuleSignature(pe::PEFile::Signature* signature); |
| + |
| // @name Accessors. |
| // @{ |
| size_t size() const; |
| @@ -60,6 +66,8 @@ class TypeRepository : public base::RefCounted<TypeRepository> { |
| friend class base::RefCounted<TypeRepository>; |
| ~TypeRepository(); |
| + bool is_signature_set_; |
| + pe::PEFile::Signature signature_; |
| base::hash_map<TypeId, TypePtr> types_; |
| DISALLOW_COPY_AND_ASSIGN(TypeRepository); |