kolibrios-fun/contrib/other/udcli/udcli.c
maxcodehack 903215408b Move C header for console loading to newlib/libc/include
git-svn-id: svn://kolibrios.org@8518 a494cfbc-eb01-0410-851d-a64ba20cac60
2021-01-05 10:04:54 +00:00

268 lines
6.3 KiB
C
Executable File

/* -----------------------------------------------------------------------------
* udcli.c - front end to udis86.
*
* Copyright (c) 2004,2005,2006,2007 Vivek Mohan <vivek@sig9.com>
* All rights reserved.
* See (LICENSE)
* -----------------------------------------------------------------------------
*/
#include <sys/kos_LoadConsole.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include "udis86.h"
#include "config.h"
#if defined(__amd64__) || defined(__x86_64__)
# define FMT "l"
#else
# define FMT "ll"
#endif
#if defined(__DJGPP__) || defined(_WIN32)
# include <io.h>
# include <fcntl.h>
#endif
#ifdef __DJGPP__
# include <unistd.h> /* for isatty() */
//# define _setmode setmode
# define _fileno fileno
# define _O_BINARY O_BINARY
#endif
/* help string */
static char help[] =
{
"Usage: %s [-option[s]] file\n"
"Options:\n"
" -16 : Set the disassembly mode to 16 bits. \n"
" -32 : Set the disassembly mode to 32 bits. (default)\n"
" -64 : Set the disassembly mode to 64 bits.\n"
" -intel : Set the output to INTEL (NASM like) syntax. (default)\n"
" -att : Set the output to AT&T (GAS like) syntax.\n"
" -v <v> : Set vendor. <v> = {intel, amd}.\n"
" -o <pc> : Set the value of program counter to <pc>. (default = 0)\n"
" -s <n> : Set the number of bytes to skip before disassembly to <n>.\n"
" -c <n> : Set the number of bytes to disassemble to <n>.\n"
" -x : Set the input mode to whitespace seperated 8-bit numbers in\n"
" hexadecimal representation. Example: 0f 01 ae 00\n"
" -noff : Do not display the offset of instructions.\n"
" -nohex : Do not display the hexadecimal code of instructions.\n"
" -h : Display this help message.\n"
" --version: Show version.\n"
"\n"
"Udcli is a front-end to the Udis86 Disassembler Library.\n"
"http://udis86.sourceforge.net/\n"
};
FILE* fptr = NULL;
uint64_t o_skip = 0;
uint64_t o_count = 0;
unsigned char o_do_count= 0;
unsigned char o_do_off = 1;
unsigned char o_do_hex = 1;
unsigned char o_do_x = 0;
unsigned o_vendor = UD_VENDOR_AMD;
int input_hook_x(ud_t* u);
int input_hook_file(ud_t* u);
int main(int argc, char **argv)
{
load_console();
con_set_title("udi disassembler");
FILE *stream;
stream = fopen( "disasm.out", "w" );
char *prog_path = *argv;
char *s;
ud_t ud_obj;
int mode = 0;
/* initialize */
ud_init(&ud_obj);
ud_set_mode(&ud_obj, 32);
ud_set_syntax(&ud_obj, UD_SYN_INTEL);
#ifdef __DJGPP__
if ( !isatty( fileno( stdin ) ) )
#endif
#if defined(__DJGPP) || defined(_WIN32)
// _setmode(_fileno(stdin), _O_BINARY);
#endif
fptr = stdin;
argv++;
/* loop through the args */
while(--argc > 0) {
if (strcmp(*argv,"-16") == 0) {
ud_set_mode(&ud_obj, 16);
mode = 16;
} else if (strcmp(*argv,"-32") == 0) {
ud_set_mode(&ud_obj, 32);
mode = 32;
} else if (strcmp(*argv,"-64") == 0) {
ud_set_mode(&ud_obj, 64);
mode = 64;
} else if (strcmp(*argv,"-intel") == 0)
ud_set_syntax(&ud_obj, UD_SYN_INTEL);
else if (strcmp(*argv,"-att") == 0)
ud_set_syntax(&ud_obj, UD_SYN_ATT);
else if (strcmp(*argv,"-noff") == 0)
o_do_off = 0;
else if (strcmp(*argv,"-nohex") == 0)
o_do_hex = 0;
else if (strcmp(*argv,"-x") == 0)
o_do_x = 1;
else if (strcmp(*argv,"-s") == 0)
if (--argc) {
s = *(++argv);
if (sscanf(s, "%" FMT "d", &o_skip) == 0)
printf( "Invalid value given for -s.\n");
} else {
printf( "No value given for -s.\n");
printf(help, prog_path);
exit(EXIT_FAILURE);
}
else if (strcmp(*argv,"-c") == 0)
if (--argc) {
o_do_count= 1;
s = *(++argv);
if (sscanf(s, "%" FMT "d", &o_count) == 0)
printf( "Invalid value given for -c.\n");
} else {
printf( "No value given for -c.\n");
printf(help, prog_path);
exit(EXIT_FAILURE);
}
else if (strcmp(*argv,"-v") == 0)
if (--argc) {
s = *(++argv);
if (*s == 'i')
ud_set_vendor(&ud_obj, UD_VENDOR_INTEL);
} else {
printf( "No value given for -v.\n");
printf(help, prog_path);
exit(EXIT_FAILURE);
}
else if (strcmp(*argv,"-o") == 0) {
if (--argc) {
uint64_t pc = 0;
s = *(++argv);
if (sscanf(s, "%" FMT "x", &pc) == 0)
printf( "Invalid value given for -o.\n");
ud_set_pc(&ud_obj, pc);
} else {
printf( "No value given for -o.\n");
printf(help, prog_path);
exit(EXIT_FAILURE);
}
} else if ( strcmp( *argv, "--version" ) == 0 ) {
printf( "%s\n", PACKAGE_STRING );
exit(0);
} else if((*argv)[0] == '-') {
printf( "Invalid option %s.\n", *argv);
printf(help, prog_path);
exit(EXIT_FAILURE);
} else {
static int i = 0;
s = *argv;
if (i) {
printf( "Multiple files specified.\n");
exit(EXIT_FAILURE);
} else i = 1;
if ((fptr = fopen(s, "rb")) == NULL) {
printf( "Failed to open file: %s.\n", s);
exit(EXIT_FAILURE);
}
}
argv++;
}
if (o_do_x)
ud_set_input_hook(&ud_obj, input_hook_x);
else ud_set_input_hook(&ud_obj, input_hook_file);
if (o_skip) {
o_count += o_skip;
ud_input_skip(&ud_obj, o_skip);
}
/* disassembly loop */
while (ud_disassemble(&ud_obj)) {
if (o_do_off)
{printf("%016" FMT "x ", ud_insn_off(&ud_obj));
fprintf(stream,"%016" FMT "x ", ud_insn_off(&ud_obj));}
if (o_do_hex) {
char* hex1, *hex2;
char c;
hex1 = ud_insn_hex(&ud_obj);
hex2 = hex1 + 16;
c = hex1[16];
hex1[16] = 0;
printf("%-16s %-24s", hex1, ud_insn_asm(&ud_obj));
fprintf(stream,"%-16s %-24s", hex1, ud_insn_asm(&ud_obj));
hex1[16] = c;
if (strlen(hex1) > 16) {
printf("\n");
if (o_do_off)
printf("%15s -", "");
printf("%-16s", hex2);
}
}
else {printf(" %-24s", ud_insn_asm(&ud_obj));
fprintf(stream," %-24s", ud_insn_asm(&ud_obj));
}
printf("\n");
fprintf(stream,"\n");
}
exit(EXIT_SUCCESS);
fclose( stream );
return 0;
}
int input_hook_x(ud_t* u)
{
unsigned int c, i;
if (o_do_count) {
if (! o_count)
return UD_EOI;
else --o_count;
}
i = fscanf(fptr, "%x", &c);
if (i == EOF)
return UD_EOI;
if (i == 0) {
printf( "Error: Invalid input, should be in hexadecimal form (8-bit).\n");
return UD_EOI;
}
if (c > 0xFF)
printf( "Warning: Casting non-8-bit input (%x), to %x.\n", c, c & 0xFF);
return (int) (c & 0xFF);
}
int input_hook_file(ud_t* u)
{
int c;
if (o_do_count) {
if (! o_count) {
return -1;
} else o_count -- ;
}
if ((c = fgetc(fptr)) == EOF)
return UD_EOI;
return c;
}