tools/llvm-c-test/disassemble.c

   1 /*===-- disassemble.c - tool for testing libLLVM and llvm-c API -----------===*\
   2 |*                                                                            *|
   3 |* Part of the LLVM Project, under the Apache License v2.0 with LLVM          *|
   4 |* Exceptions.                                                                *|
   5 |* See https://llvm.org/LICENSE.txt for license information.                  *|
   6 |* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception                    *|
   7 |*                                                                            *|
   8 |*===----------------------------------------------------------------------===*|
   9 |*                                                                            *|
  10 |* This file implements the --disassemble command in llvm-c-test.             *|
  11 |* --disassemble reads lines from stdin, parses them as a triple and hex      *|
  12 |*  machine code, and prints disassembly of the machine code.                 *|
  13 |*                                                                            *|
  14 \*===----------------------------------------------------------------------===*/
  15
  16 #include "llvm-c-test.h"
  17 #include "llvm-c/Disassembler.h"
  18 #include "llvm-c/Target.h"
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <string.h>
  22
  23 static void pprint(int pos, unsigned char *buf, int len, const char *disasm) {
  24   int i;
  25   printf("%04x:  ", pos);
  26   for (i = 0; i < 8; i++) {
  27     if (i < len) {
  28       printf("%02x ", buf[i]);
  29     } else {
  30       printf("   ");
  31     }
  32   }
  33
  34   printf("   %s\n", disasm);
  35 }
  36
  37 static void do_disassemble(const char *triple, const char *features,
  38                            unsigned char *buf, int siz) {
  39   LLVMDisasmContextRef D = LLVMCreateDisasmCPUFeatures(triple, "", features,
  40                                                        NULL, 0, NULL, NULL);
  41   char outline[1024];
  42   int pos;
  43
  44   if (!D) {
  45     printf("ERROR: Couldn't create disassembler for triple %s\n", triple);
  46     return;
  47   }
  48
  49   pos = 0;
  50   while (pos < siz) {
  51     size_t l = LLVMDisasmInstruction(D, buf + pos, siz - pos, 0, outline,
  52                                      sizeof(outline));
  53     if (!l) {
  54       pprint(pos, buf + pos, 1, "\t???");
  55       pos++;
  56     } else {
  57       pprint(pos, buf + pos, l, outline);
  58       pos += l;
  59     }
  60   }
  61
  62   LLVMDisasmDispose(D);
  63 }
  64
  65 static void handle_line(char **tokens, int ntokens) {
  66   unsigned char disbuf[128];
  67   size_t disbuflen = 0;
  68   const char *triple = tokens[0];
  69   const char *features = tokens[1];
  70   int i;
  71
  72   printf("triple: %s, features: %s\n", triple, features);
  73   if (!strcmp(features, "NULL"))
  74     features = "";
  75
  76   for (i = 2; i < ntokens; i++) {
  77     disbuf[disbuflen++] = strtol(tokens[i], NULL, 16);
  78     if (disbuflen >= sizeof(disbuf)) {
  79       fprintf(stderr, "Warning: Too long line, truncating\n");
  80       break;
  81     }
  82   }
  83   do_disassemble(triple, features, disbuf, disbuflen);
  84 }
  85
  86 int llvm_disassemble(void) {
  87   LLVMInitializeAllTargetInfos();
  88   LLVMInitializeAllTargetMCs();
  89   LLVMInitializeAllDisassemblers();
  90
  91   llvm_tokenize_stdin(handle_line);
  92
  93   return 0;
  94 }