int main(int argc, char **argv) { char buffer[INSN_MAX * 2], *p, *ep, *q; char outbuf[256]; char *pname = *argv; char *filename = NULL; uint32_t nextsync, synclen, initskip = 0L; int lenread; int32_t lendis; bool autosync = false; int bits = 16, b; bool eof = false; iflag_t prefer; bool rn_error; int64_t offset; FILE *fp; nasm_ctype_init(); nasm_set_verror(ndisasm_verror); iflag_clear_all(&prefer); offset = 0; init_sync(); while (--argc) { char *v, *vv, *p = *++argv; if (*p == '-' && p[1]) { p++; while (*p) switch (nasm_tolower(*p)) { case 'a': /* auto or intelligent sync */ case 'i': autosync = true; p++; break; case 'h': fputs(help, stderr); return 0; case 'r': case 'v': fprintf(stderr, "NDISASM version %s compiled on %s\n", nasm_version, nasm_date); return 0; case 'u': /* -u for -b 32, -uu for -b 64 */ if (bits < 64) bits <<= 1; p++; break; case 'b': /* bits */ v = p[1] ? p + 1 : --argc ? *++argv : NULL; if (!v) { fprintf(stderr, "%s: `-b' requires an argument\n", pname); return 1; } b = strtoul(v, &ep, 10); if (*ep || !(bits == 16 || bits == 32 || bits == 64)) { fprintf(stderr, "%s: argument to `-b' should" " be 16, 32 or 64\n", pname); } else { bits = b; } p = ""; /* force to next argument */ break; case 'o': /* origin */ v = p[1] ? p + 1 : --argc ? *++argv : NULL; if (!v) { fprintf(stderr, "%s: `-o' requires an argument\n", pname); return 1; } offset = readnum(v, &rn_error); if (rn_error) { fprintf(stderr, "%s: `-o' requires a numeric argument\n", pname); return 1; } p = ""; /* force to next argument */ break; case 's': /* sync point */ v = p[1] ? p + 1 : --argc ? *++argv : NULL; if (!v) { fprintf(stderr, "%s: `-s' requires an argument\n", pname); return 1; } add_sync(readnum(v, &rn_error), 0L); if (rn_error) { fprintf(stderr, "%s: `-s' requires a numeric argument\n", pname); return 1; } p = ""; /* force to next argument */ break; case 'e': /* skip a header */ v = p[1] ? p + 1 : --argc ? *++argv : NULL; if (!v) { fprintf(stderr, "%s: `-e' requires an argument\n", pname); return 1; } initskip = readnum(v, &rn_error); if (rn_error) { fprintf(stderr, "%s: `-e' requires a numeric argument\n", pname); return 1; } p = ""; /* force to next argument */ break; case 'k': /* skip a region */ v = p[1] ? p + 1 : --argc ? *++argv : NULL; if (!v) { fprintf(stderr, "%s: `-k' requires an argument\n", pname); return 1; } vv = strchr(v, ','); if (!vv) { fprintf(stderr, "%s: `-k' requires two numbers separated" " by a comma\n", pname); return 1; } *vv++ = '\0'; nextsync = readnum(v, &rn_error); if (rn_error) { fprintf(stderr, "%s: `-k' requires numeric arguments\n", pname); return 1; } synclen = readnum(vv, &rn_error); if (rn_error) { fprintf(stderr, "%s: `-k' requires numeric arguments\n", pname); return 1; } add_sync(nextsync, synclen); p = ""; /* force to next argument */ break; case 'p': /* preferred vendor */ v = p[1] ? p + 1 : --argc ? *++argv : NULL; if (!v) { fprintf(stderr, "%s: `-p' requires an argument\n", pname); return 1; } if (!strcmp(v, "intel")) { iflag_clear_all(&prefer); /* default */ } else if (!strcmp(v, "amd")) { iflag_clear_all(&prefer); iflag_set(&prefer, IF_AMD); iflag_set(&prefer, IF_3DNOW); } else if (!strcmp(v, "cyrix")) { iflag_clear_all(&prefer); iflag_set(&prefer, IF_CYRIX); iflag_set(&prefer, IF_3DNOW); } else if (!strcmp(v, "idt") || !strcmp(v, "centaur") || !strcmp(v, "winchip")) { iflag_clear_all(&prefer); iflag_set(&prefer, IF_3DNOW); } else { fprintf(stderr, "%s: unknown vendor `%s' specified with `-p'\n", pname, v); return 1; } p = ""; /* force to next argument */ break; default: /*bf */ fprintf(stderr, "%s: unrecognised option `-%c'\n", pname, *p); return 1; } } else if (!filename) { filename = p; } else { fprintf(stderr, "%s: more than one filename specified\n", pname); return 1; } } if (!filename) { fprintf(stderr, help, pname); return 0; } if (strcmp(filename, "-")) { fp = fopen(filename, "rb"); if (!fp) { fprintf(stderr, "%s: unable to open `%s': %s\n", pname, filename, strerror(errno)); return 1; } } else fp = stdin; if (initskip > 0) skip(initskip, fp); /* * This main loop is really horrible, and wants rewriting with * an axe. It'll stay the way it is for a while though, until I * find the energy... */ p = q = buffer; nextsync = next_sync(offset, &synclen); do { uint32_t to_read = buffer + sizeof(buffer) - p; if ((nextsync || synclen) && to_read > nextsync - offset - (p - q)) to_read = nextsync - offset - (p - q); if (to_read) { lenread = fread(p, 1, to_read, fp); if (lenread == 0) eof = true; /* help along systems with bad feof */ } else lenread = 0; p += lenread; if ((nextsync || synclen) && (uint32_t)offset == nextsync) { if (synclen) { fprintf(stdout, "%08"PRIX64" skipping 0x%"PRIX32" bytes\n", offset, synclen); offset += synclen; skip(synclen, fp); } p = q = buffer; nextsync = next_sync(offset, &synclen); } while (p > q && (p - q >= INSN_MAX || lenread == 0)) { lendis = disasm((uint8_t *)q, INSN_MAX, outbuf, sizeof(outbuf), bits, offset, autosync, &prefer); if (!lendis || lendis > (p - q) || ((nextsync || synclen) && (uint32_t)lendis > nextsync - offset)) lendis = eatbyte((uint8_t *) q, outbuf, sizeof(outbuf), bits); output_ins(offset, (uint8_t *) q, lendis, outbuf); q += lendis; offset += lendis; } if (q >= buffer + INSN_MAX) { uint8_t *r = (uint8_t *) buffer, *s = (uint8_t *) q; int count = p - q; while (count--) *r++ = *s++; p -= (q - buffer); q = buffer; } } while (lenread > 0 || !(eof || feof(fp))); if (fp != stdin) fclose(fp); return 0; }
int main(int argc, char **argv) { unsigned char buffer[INSN_MAX * 2], *p, *q; char outbuf[256]; char *pname = *argv; char *filename = NULL; unsigned long nextsync, synclen, initskip = 0L; int lenread, lendis; int autosync = FALSE; int bits = 16; int eof = FALSE; int rn_error; long offset; FILE *fp; offset = 0; init_sync(); while (--argc) { char *v, *vv, *p = *++argv; if (*p == '-') { p++; while (*p) switch (tolower(*p)) { case 'a': /* auto or intelligent sync */ case 'i': autosync = TRUE; p++; break; case 'h': fprintf(stderr, help); return 0; case 'r': fprintf(stderr, "NDISASM version " NASM_VER "\n"); return 0; case 'u': /* USE32 */ bits = 32; p++; break; case 'b': /* bits */ v = p[1] ? p+1 : --argc ? *++argv : NULL; if (!v) { fprintf(stderr, "%s: `-b' requires an argument\n", pname); return 1; } if (!strcmp(v, "16")) bits = 16; else if (!strcmp(v, "32")) bits = 32; else { fprintf(stderr, "%s: argument to `-b' should" " be `16' or `32'\n", pname); } p = ""; /* force to next argument */ break; case 'o': /* origin */ v = p[1] ? p+1 : --argc ? *++argv : NULL; if (!v) { fprintf(stderr, "%s: `-o' requires an argument\n", pname); return 1; } offset = readnum (v, &rn_error); if (rn_error) { fprintf(stderr, "%s: `-o' requires a numeric argument\n", pname); return 1; } p = ""; /* force to next argument */ break; case 's': /* sync point */ v = p[1] ? p+1 : --argc ? *++argv : NULL; if (!v) { fprintf(stderr, "%s: `-s' requires an argument\n", pname); return 1; } add_sync (readnum (v, &rn_error), 0L); if (rn_error) { fprintf(stderr, "%s: `-s' requires a numeric argument\n", pname); return 1; } p = ""; /* force to next argument */ break; case 'e': /* skip a header */ v = p[1] ? p+1 : --argc ? *++argv : NULL; if (!v) { fprintf(stderr, "%s: `-e' requires an argument\n", pname); return 1; } initskip = readnum (v, &rn_error); if (rn_error) { fprintf(stderr, "%s: `-e' requires a numeric argument\n", pname); return 1; } p = ""; /* force to next argument */ break; case 'k': /* skip a region */ v = p[1] ? p+1 : --argc ? *++argv : NULL; if (!v) { fprintf(stderr, "%s: `-k' requires an argument\n", pname); return 1; } vv = strchr(v, ','); if (!vv) { fprintf(stderr, "%s: `-k' requires two numbers separated" " by a comma\n", pname); return 1; } *vv++ = '\0'; nextsync = readnum (v, &rn_error); if (rn_error) { fprintf(stderr, "%s: `-k' requires numeric arguments\n", pname); return 1; } synclen = readnum (vv, &rn_error); if (rn_error) { fprintf(stderr, "%s: `-k' requires numeric arguments\n", pname); return 1; } add_sync (nextsync, synclen); p = ""; /* force to next argument */ break; } } else if (!filename) { filename = p; } else { fprintf(stderr, "%s: more than one filename specified\n", pname); return 1; } } if (!filename) { fprintf(stderr, help, pname); return 0; } fp = fopen(filename, "rb"); if (!fp) { fprintf(stderr, "%s: unable to open `%s': %s\n", pname, filename, strerror(errno)); return 1; } if (initskip > 0) skip (initskip, fp); /* * This main loop is really horrible, and wants rewriting with * an axe. It'll stay the way it is for a while though, until I * find the energy... */ p = q = buffer; nextsync = next_sync (offset, &synclen); do { unsigned long to_read = buffer+sizeof(buffer)-p; if (to_read > nextsync-offset-(p-q)) to_read = nextsync-offset-(p-q); lenread = fread (p, 1, to_read, fp); if (lenread == 0) eof = TRUE; /* help along systems with bad feof */ p += lenread; if (offset == (long)nextsync) { if (synclen) { printf("%08lX skipping 0x%lX bytes\n", offset, synclen); offset += synclen; skip (synclen, fp); } p = q = buffer; nextsync = next_sync (offset, &synclen); } while (p > q && (p - q >= INSN_MAX || lenread == 0)) { lendis = disasm (q, outbuf, bits, offset, autosync); if (!lendis || lendis > (p - q) || lendis > (int)(nextsync-offset)) lendis = eatbyte (q, outbuf); output_ins (offset, q, lendis, outbuf); q += lendis; offset += lendis; } if (q >= buffer+INSN_MAX) { unsigned char *r = buffer, *s = q; int count = p - q; while (count--) *r++ = *s++; p -= (q - buffer); q = buffer; } } while (lenread > 0 || !(eof || feof(fp))); fclose (fp); return 0; }