aboutsummaryrefslogtreecommitdiff
path: root/element_detection.cc
diff options
context:
space:
mode:
authorSamuel Huang <huangs@chromium.org>2018-03-13 18:19:34 +0000
committerEdward Lesmes <ehmaldonado@google.com>2021-07-23 21:50:59 +0000
commit06f1ae9aaca969ee95ef840f22b6b461c304542d (patch)
treef1e5c6624e70628e81fbf38d6cd14b974abe5d93 /element_detection.cc
downloadzucchini-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 'element_detection.cc')
-rw-r--r--element_detection.cc84
1 files changed, 84 insertions, 0 deletions
diff --git a/element_detection.cc b/element_detection.cc
new file mode 100644
index 0000000..d6bba5f
--- /dev/null
+++ b/element_detection.cc
@@ -0,0 +1,84 @@
+// 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.
+
+#include "components/zucchini/element_detection.h"
+
+#include <utility>
+
+#include "base/logging.h"
+#include "components/zucchini/disassembler.h"
+#include "components/zucchini/disassembler_no_op.h"
+#include "components/zucchini/disassembler_win32.h"
+
+namespace zucchini {
+
+namespace {
+
+// Impose a minimal program size to eliminate pathological cases.
+constexpr size_t kMinProgramSize = 16;
+
+} // namespace
+
+/******** Utility Functions ********/
+
+std::unique_ptr<Disassembler> MakeDisassemblerWithoutFallback(
+ ConstBufferView image) {
+ if (DisassemblerWin32X86::QuickDetect(image)) {
+ auto disasm = Disassembler::Make<DisassemblerWin32X86>(image);
+ if (disasm && disasm->size() >= kMinProgramSize)
+ return disasm;
+ }
+
+ if (DisassemblerWin32X64::QuickDetect(image)) {
+ auto disasm = Disassembler::Make<DisassemblerWin32X64>(image);
+ if (disasm && disasm->size() >= kMinProgramSize)
+ return disasm;
+ }
+
+ return nullptr;
+}
+
+std::unique_ptr<Disassembler> MakeDisassemblerOfType(ConstBufferView image,
+ ExecutableType exe_type) {
+ switch (exe_type) {
+ case kExeTypeWin32X86:
+ return Disassembler::Make<DisassemblerWin32X86>(image);
+ case kExeTypeWin32X64:
+ return Disassembler::Make<DisassemblerWin32X64>(image);
+ case kExeTypeNoOp:
+ return Disassembler::Make<DisassemblerNoOp>(image);
+ default:
+ return nullptr;
+ }
+}
+
+base::Optional<Element> DetectElementFromDisassembler(ConstBufferView image) {
+ std::unique_ptr<Disassembler> disasm = MakeDisassemblerWithoutFallback(image);
+ if (disasm)
+ return Element({0, disasm->size()}, disasm->GetExeType());
+ return base::nullopt;
+}
+
+/******** ProgramScanner ********/
+
+ElementFinder::ElementFinder(ConstBufferView image, ElementDetector&& detector)
+ : image_(image), detector_(std::move(detector)) {}
+
+ElementFinder::~ElementFinder() = default;
+
+base::Optional<Element> ElementFinder::GetNext() {
+ for (; pos_ < image_.size(); ++pos_) {
+ ConstBufferView test_image =
+ ConstBufferView::FromRange(image_.begin() + pos_, image_.end());
+ base::Optional<Element> element = detector_.Run(test_image);
+ if (element) {
+ element->offset += pos_;
+ pos_ = element->EndOffset();
+ return element;
+ }
+ }
+ return base::nullopt;
+}
+
+} // namespace zucchini