ndisasm.c

   1 /* ----------------------------------------------------------------------- *
   2  *
   3  *   Copyright 1996-2009 The NASM Authors - All Rights Reserved
   4  *   See the file AUTHORS included with the NASM distribution for
   5  *   the specific copyright holders.
   6  *
   7  *   Redistribution and use in source and binary forms, with or without
   8  *   modification, are permitted provided that the following
   9  *   conditions are met:
  10  *
  11  *   * Redistributions of source code must retain the above copyright
  12  *     notice, this list of conditions and the following disclaimer.
  13  *   * Redistributions in binary form must reproduce the above
  14  *     copyright notice, this list of conditions and the following
  15  *     disclaimer in the documentation and/or other materials provided
  16  *     with the distribution.
  17  *
  18  *     THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
  19  *     CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
  20  *     INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  21  *     MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  22  *     DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  23  *     CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24  *     SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  25  *     NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  26  *     LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  *     HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  28  *     CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  29  *     OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
  30  *     EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31  *
  32  * ----------------------------------------------------------------------- */
  33
  34 /*
  35  * ndisasm.c   the Netwide Disassembler main module
  36  */
  37
  38 #include "compiler.h"
  39
  40 #include <stdio.h>
  41 #include <stdarg.h>
  42 #include <stdlib.h>
  43 #include <string.h>
  44 #include <ctype.h>
  45 #include <errno.h>
  46 #include <inttypes.h>
  47
  48 #include "insns.h"
  49 #include "nasm.h"
  50 #include "nasmlib.h"
  51 #include "sync.h"
  52 #include "disasm.h"
  53
  54 #define BPL 8                   /* bytes per line of hex dump */
  55
  56 static const char *help =
  57     "usage: ndisasm [-a] [-i] [-h] [-r] [-u] [-b bits] [-o origin] [-s sync...]\n"
  58     "               [-e bytes] [-k start,bytes] [-p vendor] file\n"
  59     "   -a or -i activates auto (intelligent) sync\n"
  60     "   -u same as -b 32\n"
  61     "   -b 16, -b 32 or -b 64 sets the processor mode\n"
  62     "   -h displays this text\n"
  63     "   -r or -v displays the version number\n"
  64     "   -e skips <bytes> bytes of header\n"
  65     "   -k avoids disassembling <bytes> bytes from position <start>\n"
  66     "   -p selects the preferred vendor instruction set (intel, amd, cyrix, idt)\n";
  67
  68 static void output_ins(uint32_t, uint8_t *, int, char *);
  69 static void skip(uint32_t dist, FILE * fp);
  70
  71 static void ndisasm_verror(int severity, const char *fmt, va_list va)
  72 {
  73     vfprintf(stderr, fmt, va);
  74
  75     if (severity & ERR_FATAL)
  76         exit(1);
  77 }
  78
  79 int main(int argc, char **argv)
  80 {
  81     char buffer[INSN_MAX * 2], *p, *ep, *q;
  82     char outbuf[256];
  83     char *pname = *argv;
  84     char *filename = NULL;
  85     uint32_t nextsync, synclen, initskip = 0L;
  86     int lenread;
  87     int32_t lendis;
  88     bool autosync = false;
  89     int bits = 16, b;
  90     bool eof = false;
  91     iflags_t prefer = 0;
  92     bool rn_error;
  93     int32_t offset;
  94     FILE *fp;
  95
  96     tolower_init();
  97     nasm_set_verror(ndisasm_verror);
  98     nasm_init_malloc_error();
  99
 100     offset = 0;
 101     init_sync();
 102
 103     while (--argc) {
 104         char *v, *vv, *p = *++argv;
 105         if (*p == '-' && p[1]) {
 106             p++;
 107             while (*p)
 108                 switch (nasm_tolower(*p)) {
 109                 case 'a':      /* auto or intelligent sync */
 110                 case 'i':
 111                     autosync = true;
 112                     p++;
 113                     break;
 114                 case 'h':
 115                     fputs(help, stderr);
 116                     return 0;
 117                 case 'r':
 118                 case 'v':
 119                     fprintf(stderr,
 120                             "NDISASM version %s compiled on %s\n",
 121                             nasm_version, nasm_date);
 122                     return 0;
 123                 case 'u':       /* -u for -b 32, -uu for -b 64 */
 124                     if (bits < 64)
 125                         bits <<= 1;
 126                     p++;
 127                     break;
 128                 case 'b':      /* bits */
 129                     v = p[1] ? p + 1 : --argc ? *++argv : NULL;
 130                     if (!v) {
 131                         fprintf(stderr, "%s: `-b' requires an argument\n",
 132                                 pname);
 133                         return 1;
 134                     }
 135                     b = strtoul(v, &ep, 10);
 136                     if (*ep || !(bits == 16 || bits == 32 || bits == 64)) {
 137                         fprintf(stderr, "%s: argument to `-b' should"
 138                                 " be 16, 32 or 64\n", pname);
 139                     } else {
 140                         bits = b;
 141                     }
 142                     p = "";     /* force to next argument */
 143                     break;
 144                 case 'o':      /* origin */
 145                     v = p[1] ? p + 1 : --argc ? *++argv : NULL;
 146                     if (!v) {
 147                         fprintf(stderr, "%s: `-o' requires an argument\n",
 148                                 pname);
 149                         return 1;
 150                     }
 151                     offset = readnum(v, &rn_error);
 152                     if (rn_error) {
 153                         fprintf(stderr,
 154                                 "%s: `-o' requires a numeric argument\n",
 155                                 pname);
 156                         return 1;
 157                     }
 158                     p = "";     /* force to next argument */
 159                     break;
 160                 case 's':      /* sync point */
 161                     v = p[1] ? p + 1 : --argc ? *++argv : NULL;
 162                     if (!v) {
 163                         fprintf(stderr, "%s: `-s' requires an argument\n",
 164                                 pname);
 165                         return 1;
 166                     }
 167                     add_sync(readnum(v, &rn_error), 0L);
 168                     if (rn_error) {
 169                         fprintf(stderr,
 170                                 "%s: `-s' requires a numeric argument\n",
 171                                 pname);
 172                         return 1;
 173                     }
 174                     p = "";     /* force to next argument */
 175                     break;
 176                 case 'e':      /* skip a header */
 177                     v = p[1] ? p + 1 : --argc ? *++argv : NULL;
 178                     if (!v) {
 179                         fprintf(stderr, "%s: `-e' requires an argument\n",
 180                                 pname);
 181                         return 1;
 182                     }
 183                     initskip = readnum(v, &rn_error);
 184                     if (rn_error) {
 185                         fprintf(stderr,
 186                                 "%s: `-e' requires a numeric argument\n",
 187                                 pname);
 188                         return 1;
 189                     }
 190                     p = "";     /* force to next argument */
 191                     break;
 192                 case 'k':      /* skip a region */
 193                     v = p[1] ? p + 1 : --argc ? *++argv : NULL;
 194                     if (!v) {
 195                         fprintf(stderr, "%s: `-k' requires an argument\n",
 196                                 pname);
 197                         return 1;
 198                     }
 199                     vv = strchr(v, ',');
 200                     if (!vv) {
 201                         fprintf(stderr,
 202                                 "%s: `-k' requires two numbers separated"
 203                                 " by a comma\n", pname);
 204                         return 1;
 205                     }
 206                     *vv++ = '\0';
 207                     nextsync = readnum(v, &rn_error);
 208                     if (rn_error) {
 209                         fprintf(stderr,
 210                                 "%s: `-k' requires numeric arguments\n",
 211                                 pname);
 212                         return 1;
 213                     }
 214                     synclen = readnum(vv, &rn_error);
 215                     if (rn_error) {
 216                         fprintf(stderr,
 217                                 "%s: `-k' requires numeric arguments\n",
 218                                 pname);
 219                         return 1;
 220                     }
 221                     add_sync(nextsync, synclen);
 222                     p = "";     /* force to next argument */
 223                     break;
 224                 case 'p':      /* preferred vendor */
 225                     v = p[1] ? p + 1 : --argc ? *++argv : NULL;
 226                     if (!v) {
 227                         fprintf(stderr, "%s: `-p' requires an argument\n",
 228                                 pname);
 229                         return 1;
 230                     }
 231                     if (!strcmp(v, "intel")) {
 232                         prefer = 0;     /* Default */
 233                     } else if (!strcmp(v, "amd")) {
 234                         prefer = IF_AMD | IF_3DNOW;
 235                     } else if (!strcmp(v, "cyrix")) {
 236                         prefer = IF_CYRIX | IF_3DNOW;
 237                     } else if (!strcmp(v, "idt") || !strcmp(v, "centaur")
 238                                || !strcmp(v, "winchip")) {
 239                         prefer = IF_3DNOW;
 240                     } else {
 241                         fprintf(stderr,
 242                                 "%s: unknown vendor `%s' specified with `-p'\n",
 243                                 pname, v);
 244                         return 1;
 245                     }
 246                     p = "";     /* force to next argument */
 247                     break;
 248                 default:       /*bf */
 249                     fprintf(stderr, "%s: unrecognised option `-%c'\n",
 250                             pname, *p);
 251                     return 1;
 252                 }
 253         } else if (!filename) {
 254             filename = p;
 255         } else {
 256             fprintf(stderr, "%s: more than one filename specified\n",
 257                     pname);
 258             return 1;
 259         }
 260     }
 261
 262     if (!filename) {
 263         fprintf(stderr, help, pname);
 264         return 0;
 265     }
 266
 267     if (strcmp(filename, "-")) {
 268         fp = fopen(filename, "rb");
 269         if (!fp) {
 270             fprintf(stderr, "%s: unable to open `%s': %s\n",
 271                     pname, filename, strerror(errno));
 272             return 1;
 273         }
 274     } else
 275         fp = stdin;
 276
 277     if (initskip > 0)
 278         skip(initskip, fp);
 279
 280     /*
 281      * This main loop is really horrible, and wants rewriting with
 282      * an axe. It'll stay the way it is for a while though, until I
 283      * find the energy...
 284      */
 285
 286     p = q = buffer;
 287     nextsync = next_sync(offset, &synclen);
 288     do {
 289         uint32_t to_read = buffer + sizeof(buffer) - p;
 290         if ((nextsync || synclen) &&
 291             to_read > nextsync - offset - (p - q))
 292             to_read = nextsync - offset - (p - q);
 293         if (to_read) {
 294             lenread = fread(p, 1, to_read, fp);
 295             if (lenread == 0)
 296                 eof = true;     /* help along systems with bad feof */
 297         } else
 298             lenread = 0;
 299         p += lenread;
 300         if ((nextsync || synclen) &&
 301             (uint32_t)offset == nextsync) {
 302             if (synclen) {
 303                 fprintf(stdout, "%08"PRIX32"  skipping 0x%"PRIX32" bytes\n",
 304                         offset, synclen);
 305                 offset += synclen;
 306                 skip(synclen, fp);
 307             }
 308             p = q = buffer;
 309             nextsync = next_sync(offset, &synclen);
 310         }
 311         while (p > q && (p - q >= INSN_MAX || lenread == 0)) {
 312             lendis =
 313                 disasm((uint8_t *) q, outbuf, sizeof(outbuf), bits,
 314                        offset, autosync, prefer);
 315             if (!lendis || lendis > (p - q)
 316                 || ((nextsync || synclen) &&
 317                     (uint32_t)lendis > nextsync - offset))
 318                 lendis = eatbyte((uint8_t *) q, outbuf, sizeof(outbuf), bits);
 319             output_ins(offset, (uint8_t *) q, lendis, outbuf);
 320             q += lendis;
 321             offset += lendis;
 322         }
 323         if (q >= buffer + INSN_MAX) {
 324             uint8_t *r = (uint8_t *) buffer, *s = (uint8_t *) q;
 325             int count = p - q;
 326             while (count--)
 327                 *r++ = *s++;
 328             p -= (q - buffer);
 329             q = buffer;
 330         }
 331     } while (lenread > 0 || !(eof || feof(fp)));
 332
 333     if (fp != stdin)
 334         fclose(fp);
 335
 336     return 0;
 337 }
 338
 339 static void output_ins(uint32_t offset, uint8_t *data,
 340                        int datalen, char *insn)
 341 {
 342     int bytes;
 343     fprintf(stdout, "%08"PRIX32"  ", offset);
 344
 345     bytes = 0;
 346     while (datalen > 0 && bytes < BPL) {
 347         fprintf(stdout, "%02X", *data++);
 348         bytes++;
 349         datalen--;
 350     }
 351
 352     fprintf(stdout, "%*s%s\n", (BPL + 1 - bytes) * 2, "", insn);
 353
 354     while (datalen > 0) {
 355         fprintf(stdout, "         -");
 356         bytes = 0;
 357         while (datalen > 0 && bytes < BPL) {
 358             fprintf(stdout, "%02X", *data++);
 359             bytes++;
 360             datalen--;
 361         }
 362         fprintf(stdout, "\n");
 363     }
 364 }
 365
 366 /*
 367  * Skip a certain amount of data in a file, either by seeking if
 368  * possible, or if that fails then by reading and discarding.
 369  */
 370 static void skip(uint32_t dist, FILE * fp)
 371 {
 372     char buffer[256];           /* should fit on most stacks :-) */
 373
 374     /*
 375      * Got to be careful with fseek: at least one fseek I've tried
 376      * doesn't approve of SEEK_CUR. So I'll use SEEK_SET and
 377      * ftell... horrible but apparently necessary.
 378      */
 379     if (fseek(fp, dist + ftell(fp), SEEK_SET)) {
 380         while (dist > 0) {
 381             uint32_t len = (dist < sizeof(buffer) ?
 382                                  dist : sizeof(buffer));
 383             if (fread(buffer, 1, len, fp) < len) {
 384                 perror("fread");
 385                 exit(1);
 386             }
 387             dist -= len;
 388         }
 389     }
 390 }