Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(130)

Unified Diff: courgette/disassembler.cc

Issue 1650013002: Revert of [Courgette] Refactor: Manage AssemblyProgram and EncodedProgram with scoped_ptr. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 4 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « courgette/courgette_tool.cc ('k') | courgette/encode_decode_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: courgette/disassembler.cc
diff --git a/courgette/disassembler.cc b/courgette/disassembler.cc
index b9fce8bdc0efc7d01bc4a1864738378430ef45b3..568114c52701816720816b785a74786c66967625 100644
--- a/courgette/disassembler.cc
+++ b/courgette/disassembler.cc
@@ -4,7 +4,100 @@
#include "courgette/disassembler.h"
+#include <stddef.h>
+#include <stdint.h>
+
+#include <algorithm>
+#include <string>
+#include <vector>
+
+#include "base/logging.h"
+#include "base/memory/scoped_ptr.h"
+#include "courgette/assembly_program.h"
+#include "courgette/courgette.h"
+#include "courgette/disassembler_elf_32_arm.h"
+#include "courgette/disassembler_elf_32_x86.h"
+#include "courgette/disassembler_win32_x64.h"
+#include "courgette/disassembler_win32_x86.h"
+#include "courgette/encoded_program.h"
+
namespace courgette {
+
+////////////////////////////////////////////////////////////////////////////////
+
+Disassembler* DetectDisassembler(const void* buffer, size_t length) {
+ Disassembler* disassembler = NULL;
+
+ disassembler = new DisassemblerWin32X86(buffer, length);
+ if (disassembler->ParseHeader())
+ return disassembler;
+ else
+ delete disassembler;
+
+ disassembler = new DisassemblerWin32X64(buffer, length);
+ if (disassembler->ParseHeader())
+ return disassembler;
+ else
+ delete disassembler;
+
+ disassembler = new DisassemblerElf32X86(buffer, length);
+ if (disassembler->ParseHeader())
+ return disassembler;
+ else
+ delete disassembler;
+
+ disassembler = new DisassemblerElf32ARM(buffer, length);
+ if (disassembler->ParseHeader())
+ return disassembler;
+ else
+ delete disassembler;
+
+ return NULL;
+}
+
+Status DetectExecutableType(const void* buffer, size_t length,
+ ExecutableType* type,
+ size_t* detected_length) {
+
+ Disassembler* disassembler = DetectDisassembler(buffer, length);
+
+ if (disassembler) {
+ *type = disassembler->kind();
+ *detected_length = disassembler->length();
+ delete disassembler;
+ return C_OK;
+ }
+
+ // We failed to detect anything
+ *type = EXE_UNKNOWN;
+ *detected_length = 0;
+ return C_INPUT_NOT_RECOGNIZED;
+}
+
+Status ParseDetectedExecutable(const void* buffer, size_t length,
+ AssemblyProgram** output) {
+ *output = nullptr;
+
+ scoped_ptr<Disassembler> disassembler(DetectDisassembler(buffer, length));
+ if (!disassembler)
+ return C_INPUT_NOT_RECOGNIZED;
+
+ scoped_ptr<AssemblyProgram> program(
+ new AssemblyProgram(disassembler->kind()));
+
+ if (!disassembler->Disassemble(program.get()))
+ return C_DISASSEMBLY_FAILED;
+
+ if (!program->TrimLabels())
+ return C_TRIM_FAILED;
+
+ *output = program.release();
+ return C_OK;
+}
+
+void DeleteAssemblyProgram(AssemblyProgram* program) {
+ delete program;
+}
Disassembler::Disassembler(const void* start, size_t length)
: failure_reason_("uninitialized") {
« no previous file with comments | « courgette/courgette_tool.cc ('k') | courgette/encode_decode_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698