Added all code for integrity check transform

syzygy/pe/pe_utils.h

 // Copyright 2012 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.
 
 #ifndef SYZYGY_PE_PE_UTILS_H_
 #define SYZYGY_PE_PE_UTILS_H_
 
 #include <windows.h>
 #include <winnt.h>
 
 #include "syzygy/block_graph/block_graph.h"
+#include "syzygy/core/address.h"
+#include "syzygy/pe/pe_file_writer.h"
 
 namespace pe {
 
 // @name Operations on PE/COFF headers.
 // @{
 // Known section types.
 enum SectionType {
   kSectionCode,
   kSectionData,
   kSectionUnknown
 };
 
+// The file ranges of each section. This is populated by
+// CalculateSectionRanges and is a map from section index (as ordered in
+// the image layout) to section ranges on disk.
+typedef core::AddressRange<core::FileOffsetAddress, size_t> FileRange;
+typedef std::map<size_t, FileRange> SectionIndexFileRangeMap;
+
+// This stores an address-space from RVAs to section indices and is populated
+// by CalculateSectionRanges. This can be used to map from a block's
+// address to the index of its section. This is needed for finalizing
+// references.
+typedef core::AddressSpace<core::RelativeAddress, size_t, size_t>
+SectionIndexSpace;
+
 // Typical section names.
 extern const char kCodeSectionName[];
 extern const char kReadOnlyDataSectionName[];
 extern const char kReadWriteDataSectionName[];
 extern const char kRelocSectionName[];
 extern const char kResourceSectionName[];
 extern const char kTlsSectionName[];
 
 // Typical section characteristics.
 extern const DWORD kCodeCharacteristics;
@@ skipping 140 matching lines @@
 
 // Redirect references in a block-graph, except for references originating from
 // PE structures. Any non-PE-structure block in src_blocks will have its
 // references examined. Any reference found as a key in @p redirects will be
 // remapped to its corresponding value.
 // @param src The original referred destination that is to be redirected.
 // @param dst The redirected destination to be referred to.
 // @param redirects A map of original to redirected destinations.
 void RedirectReferences(const ReferenceMap& redirects);
 
+template <class Type>
+bool UpdateReference(size_t start, Type new_value, std::vector<uint8_t>* data);
+
+bool ResolveReferences(core::AbsoluteAddress image_base,
+                       const block_graph::BlockGraph::Block* block,
+                       core::RelativeAddress addr,
+                       const core::FileOffsetAddress file_offs,
+                       SectionIndexFileRangeMap section_file_range_map,
+                       SectionIndexSpace section_index_space,
+                       bool write_references_in_place,
+                       std::vector<uint8_t>* buffer,
+                       const ImageLayout& image_layout);
+
 }  // namespace pe
 
 #include "syzygy/pe/pe_utils_impl.h"
 
 #endif  // SYZYGY_PE_PE_UTILS_H_