diff options
author | Samuel Huang <huangs@chromium.org> | 2018-03-13 18:19:34 +0000 |
---|---|---|
committer | Edward Lesmes <ehmaldonado@google.com> | 2021-07-23 21:50:59 +0000 |
commit | 06f1ae9aaca969ee95ef840f22b6b461c304542d (patch) | |
tree | f1e5c6624e70628e81fbf38d6cd14b974abe5d93 /disassembler_win32.h | |
download | zucchini-06f1ae9aaca969ee95ef840f22b6b461c304542d.tar.gz |
[Zucchini] Move Zucchini from /chrome/installer/ to /components/.
(Use "git log --follow" to see older revisions of files).
/components/ is the most logical place to put Zucchini, which only
depends on /base and /testing/gtest. This move also enables Zucchini to
be used by the Component Updater. Details:
- Move all files; run the following to change deps and guards:
sed 's/chrome\/installer/components/' *.cc *.h -i
sed 's/CHROME_INSTALLER/COMPONENTS/' *.cc *.h -i
- Sorting works out pretty well!
- Change all 'chrome/installer/zucchini' to 'components/zucchini'
throughout other parts of the repo; sort if necessary.
- Fix 6 'git cl lint' errors.
- Change 1 Bind() usage to BindRepeated().
- Update OWNER.
Bug: 729154
Change-Id: I50c5a7d411ea85f707b5994ab319dfb2a1acccf7
Reviewed-on: https://chromium-review.googlesource.com/954923
Reviewed-by: Greg Thompson <grt@chromium.org>
Reviewed-by: Jochen Eisinger <jochen@chromium.org>
Reviewed-by: Samuel Huang <huangs@chromium.org>
Commit-Queue: Samuel Huang <huangs@chromium.org>
Cr-Commit-Position: refs/heads/master@{#542857}
NOKEYCHECK=True
GitOrigin-RevId: 577ef6c435e8d43be6e3e60ccbcbd1881780f4ec
Diffstat (limited to 'disassembler_win32.h')
-rw-r--r-- | disassembler_win32.h | 129 |
1 files changed, 129 insertions, 0 deletions
diff --git a/disassembler_win32.h b/disassembler_win32.h new file mode 100644 index 0000000..8e410ee --- /dev/null +++ b/disassembler_win32.h @@ -0,0 +1,129 @@ +// Copyright 2017 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. + +#ifndef COMPONENTS_ZUCCHINI_DISASSEMBLER_WIN32_H_ +#define COMPONENTS_ZUCCHINI_DISASSEMBLER_WIN32_H_ + +#include <stddef.h> +#include <stdint.h> + +#include <memory> +#include <string> +#include <utility> +#include <vector> + +#include "base/macros.h" +#include "components/zucchini/address_translator.h" +#include "components/zucchini/buffer_view.h" +#include "components/zucchini/disassembler.h" +#include "components/zucchini/image_utils.h" +#include "components/zucchini/type_win_pe.h" + +namespace zucchini { + +class Rel32FinderX86; +class Rel32FinderX64; + +struct Win32X86Traits { + static constexpr Bitness kBitness = kBit32; + static constexpr ExecutableType kExeType = kExeTypeWin32X86; + enum : uint16_t { kMagic = 0x10B }; + enum : uint16_t { kRelocType = 3 }; + enum : offset_t { kVAWidth = 4 }; + static const char kExeTypeString[]; + + using ImageOptionalHeader = pe::ImageOptionalHeader; + using RelFinder = Rel32FinderX86; + using Address = uint32_t; +}; + +struct Win32X64Traits { + static constexpr Bitness kBitness = kBit64; + static constexpr ExecutableType kExeType = kExeTypeWin32X64; + enum : uint16_t { kMagic = 0x20B }; + enum : uint16_t { kRelocType = 10 }; + enum : offset_t { kVAWidth = 8 }; + static const char kExeTypeString[]; + + using ImageOptionalHeader = pe::ImageOptionalHeader64; + using RelFinder = Rel32FinderX64; + using Address = uint64_t; +}; + +template <class Traits> +class DisassemblerWin32 : public Disassembler { + public: + enum ReferenceType : uint8_t { kReloc, kAbs32, kRel32, kTypeCount }; + + // Applies quick checks to determine whether |image| *may* point to the start + // of an executable. Returns true iff the check passes. + static bool QuickDetect(ConstBufferView image); + + DisassemblerWin32(); + ~DisassemblerWin32() override; + + // Disassembler: + ExecutableType GetExeType() const override; + std::string GetExeTypeString() const override; + std::vector<ReferenceGroup> MakeReferenceGroups() const override; + + // Functions that return reader / writer for references. + std::unique_ptr<ReferenceReader> MakeReadRelocs(offset_t lo, offset_t hi); + std::unique_ptr<ReferenceReader> MakeReadAbs32(offset_t lo, offset_t hi); + std::unique_ptr<ReferenceReader> MakeReadRel32(offset_t lo, offset_t hi); + std::unique_ptr<ReferenceWriter> MakeWriteRelocs(MutableBufferView image); + std::unique_ptr<ReferenceWriter> MakeWriteAbs32(MutableBufferView image); + std::unique_ptr<ReferenceWriter> MakeWriteRel32(MutableBufferView image); + + private: + friend Disassembler; + + // Disassembler: + bool Parse(ConstBufferView image) override; + + // Parses the file header. Returns true iff successful. + bool ParseHeader(); + + // Parsers to extract references. These are lazily called, and return whether + // parsing was successful (failures are non-fatal). + bool ParseAndStoreRelocBlocks(); + bool ParseAndStoreAbs32(); + bool ParseAndStoreRel32(); + + // In-memory copy of sections. + std::vector<pe::ImageSectionHeader> sections_; + + // Image base address to translate between RVA and VA. + typename Traits::Address image_base_ = 0; + + // Pointer to data Directory entry of the relocation table. + const pe::ImageDataDirectory* base_relocation_table_ = nullptr; + + // Translator between offsets and RVAs. + AddressTranslator translator_; + + // Reference storage. + BufferRegion reloc_region_; + std::vector<offset_t> reloc_block_offsets_; + offset_t reloc_end_ = 0; + std::vector<offset_t> abs32_locations_; + std::vector<offset_t> rel32_locations_; + + // Initialization states of reference storage, used for lazy initialization. + // TODO(huangs): Investigate whether lazy initialization is useful for memory + // reduction. This is a carryover from Courgette. To be sure we should run + // experiment after Zucchini is able to do ensemble patching. + bool has_parsed_relocs_ = false; + bool has_parsed_abs32_ = false; + bool has_parsed_rel32_ = false; + + DISALLOW_COPY_AND_ASSIGN(DisassemblerWin32); +}; + +using DisassemblerWin32X86 = DisassemblerWin32<Win32X86Traits>; +using DisassemblerWin32X64 = DisassemblerWin32<Win32X64Traits>; + +} // namespace zucchini + +#endif // COMPONENTS_ZUCCHINI_DISASSEMBLER_WIN32_H_ |