lib/System/Disassembler.cpp

   1 //===- lib/System/Disassembler.cpp ------------------------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file implements the necessary glue to call external disassembler
  11 // libraries.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/Config/config.h"
  16 #include "llvm/System/Disassembler.h"
  17
  18 #include <cassert>
  19 #include <iomanip>
  20 #include <string>
  21 #include <sstream>
  22
  23 #if USE_UDIS86
  24 #include <udis86.h>
  25 #endif
  26
  27 using namespace llvm;
  28
  29 bool llvm::sys::hasDisassembler(void)
  30 {
  31 #if defined (__i386__) || defined (__amd64__) || defined (__x86_64__)
  32   // We have option to enable udis86 library.
  33   return true;
  34 #else
  35   return false;
  36 #endif
  37 }
  38
  39 std::string llvm::sys::disassembleBuffer(uint8_t* start, size_t length,
  40                                          uint64_t pc) {
  41   std::stringstream res;
  42
  43 #if defined (__i386__) || defined (__amd64__) || defined (__x86_64__)
  44   unsigned bits;
  45 # if defined(__i386__)
  46   bits = 32;
  47 # else
  48   bits = 64;
  49 # endif
  50
  51 # if USE_UDIS86
  52   ud_t ud_obj;
  53
  54   ud_init(&ud_obj);
  55   ud_set_input_buffer(&ud_obj, start, length);
  56   ud_set_mode(&ud_obj, bits);
  57   ud_set_pc(&ud_obj, pc);
  58   ud_set_syntax(&ud_obj, UD_SYN_ATT);
  59
  60   res << std::setbase(16)
  61       << std::setw(bits/4);
  62
  63   while (ud_disassemble(&ud_obj)) {
  64     res << ud_insn_off(&ud_obj) << ":\t" << ud_insn_asm(&ud_obj) << "\n";
  65   }
  66 # else
  67   res << "No disassembler available. See configure help for options.\n";
  68 # endif
  69
  70 #else
  71   res << "No disassembler available. See configure help for options.\n";
  72 #endif
  73
  74   return res.str();
  75 }