spike_main/spike-dasm.cc

   1 // See LICENSE for license details.
   2
   3 // This little program finds occurrences of strings like
   4 //  DASM(ffabc013)
   5 // in its input, then replaces them with the disassembly
   6 // enclosed hexadecimal number, interpreted as a RISC-V
   7 // instruction.
   8
   9 #include "disasm.h"
  10 #include "extension.h"
  11 #include <iostream>
  12 #include <string>
  13 #include <cstdint>
  14 #include <fesvr/option_parser.h>
  15 using namespace std;
  16
  17 int main(int argc, char** argv)
  18 {
  19   string s;
  20   disassembler_t d;
  21
  22   std::function<extension_t*()> extension;
  23   option_parser_t parser;
  24   parser.option(0, "extension", 1, [&](const char* s){extension = find_extension(s);});
  25   parser.parse(argv);
  26
  27   if (extension) {
  28     for (auto disasm_insn : extension()->get_disasms())
  29       d.add_insn(disasm_insn);
  30   }
  31
  32   while (getline(cin, s))
  33   {
  34     for (size_t start = 0; (start = s.find("DASM(", start)) != string::npos; )
  35     {
  36       size_t end = s.find(')', start);
  37       if (end == string::npos)
  38         break;
  39
  40       char* endp;
  41       size_t numstart = start + strlen("DASM(");
  42       int64_t bits = strtoull(&s[numstart], &endp, 16);
  43       size_t nbits = 4 * (endp - &s[numstart]);
  44       if (nbits < 64)
  45         bits = bits << (64 - nbits) >> (64 - nbits);
  46
  47       string dis = d.disassemble(bits);
  48       s = s.substr(0, start) + dis + s.substr(end+1);
  49       start += dis.length();
  50     }
  51
  52     cout << s << '\n';
  53   }
  54
  55   return 0;
  56 }