radare2/libr/core/cmd_hash.c
2015-09-14 12:35:38 +02:00

260 lines
7.1 KiB
C

/* radare - LGPL - Copyright 2009-2015 - pancake, nibble */
typedef void (*HashHandler)(const ut8 *block, int len);
static void handle_md4 (const ut8 *block, int len);
static void handle_md5 (const ut8 *block, int len);
static void handle_sha1 (const ut8 *block, int len);
static void handle_sha256 (const ut8 *block, int len);
static void handle_sha512 (const ut8 *block, int len);
static void handle_adler32 (const ut8 *block, int len);
static void handle_crc32 (const ut8 *block, int len);
static void handle_crc16 (const ut8 *block, int len);
static void handle_xor (const ut8 *block, int len);
static void handle_entropy (const ut8 *block, int len);
static void handle_hamdist (const ut8 *block, int len);
static void handle_parity (const ut8 *block, int len);
static void handle_pcprint (const ut8 *input, int len);
typedef struct {
const char *name;
HashHandler handler;
} RHashHashHandlers;
static RHashHashHandlers hash_handlers[] = {
{"md4", handle_md4},
{"md5", handle_md5},
{"sha1", handle_sha1},
{"sha256", handle_sha256},
{"sha512", handle_sha512},
{"adler32", handle_adler32},
{"crc32", handle_crc32},
{"crc16", handle_crc16},
{"xor", handle_xor},
{"entropy", handle_entropy},
{"parity", handle_parity},
{"hamdist", handle_hamdist},
{"pcprint", handle_pcprint},
{NULL, NULL},
};
static void handle_md4 (const ut8 *block, int len) {
int i = 0;
RHash *ctx = r_hash_new (true, R_HASH_MD4);
const ut8 *c = r_hash_do_md4 (ctx, block, len);
for (i=0; i<R_HASH_SIZE_MD4; i++) r_cons_printf ("%02x", c[i]);
r_cons_newline ();
r_hash_free (ctx);
}
static void handle_md5 (const ut8 *block, int len) {
int i = 0;
RHash *ctx = r_hash_new (true, R_HASH_MD5);
const ut8 *c = r_hash_do_md5 (ctx, block, len);
for (i=0; i<R_HASH_SIZE_MD5; i++) r_cons_printf ("%02x", c[i]);
r_cons_newline ();
r_hash_free (ctx);
}
static void handle_sha1 (const ut8 *block, int len) {
int i = 0;
RHash *ctx = r_hash_new (true, R_HASH_SHA1);
const ut8 *c = r_hash_do_sha1 (ctx, block, len);
for (i=0; i<R_HASH_SIZE_SHA1; i++) r_cons_printf ("%02x", c[i]);
r_cons_newline ();
r_hash_free (ctx);
}
static void handle_sha256 (const ut8 *block, int len) {
int i = 0;
RHash *ctx = r_hash_new (true, R_HASH_SHA256);
const ut8 *c = r_hash_do_sha256 (ctx, block, len);
for (i=0; i<R_HASH_SIZE_SHA256; i++) r_cons_printf ("%02x", c[i]);
r_cons_newline ();
r_hash_free (ctx);
}
static void handle_sha512 (const ut8 *block, int len) {
int i = 0;
RHash *ctx = r_hash_new (true, R_HASH_SHA512);
const ut8 *c = r_hash_do_sha512 (ctx, block, len);
for (i=0; i<R_HASH_SIZE_SHA512; i++) r_cons_printf ("%02x", c[i]);
r_cons_newline ();
r_hash_free (ctx);
}
static void handle_adler32 (const ut8 *block, int len) {
ut32 hn = r_hash_adler32 (block, len);
ut8 *b = (ut8*)&hn;
r_cons_printf ("%02x%02x%02x%02x\n", b[0], b[1], b[2], b[3]);
}
static void handle_crc32 (const ut8 *block, int len) {
r_cons_printf ("%04x\n", r_hash_crc32 (block, len));
}
static void handle_crc16 (const ut8 *block, int len) {
r_cons_printf ("%04x\n", r_hash_crc16 (0, block, len));
}
static void handle_xor (const ut8 *block, int len) {
r_cons_printf ("%02x\n", r_hash_xor (block, len));
}
static void handle_entropy (const ut8 *block, int len) {
r_cons_printf ("%f\n", r_hash_entropy (block, len));
}
static void handle_parity (const ut8 *block, int len) {
r_cons_printf ("%d\n", r_hash_parity (block, len)?1:0);
}
static void handle_hamdist (const ut8 *block, int len) {
r_cons_printf ("%02x\n", r_hash_hamdist (block, len));
}
static void handle_pcprint (const ut8 *block, int len) {
r_cons_printf ("%d\n", r_hash_pcprint (block, len));
//r_cons_printf ("%02x\n", r_hash_pcprint (block, len));
}
static void algolist(int mode) {
const char *name;
ut64 bits;
int i;
r_cons_printf ("| ");
for (i=0; ; i++) {
bits = ((ut64)1)<<i;
name = r_hash_name (bits);
if (!name||!*name) break;
if (mode) {
r_cons_printf ("%s\n| ", name);
} else {
r_cons_printf (" #%s", name);
if (!((i+1)%6)) r_cons_printf ("\n| ");
}
}
if (!mode) r_cons_newline ();
}
static int cmd_hash_bang (RCore *core, const char *input) {
char *p;
const char *lang = input+1;
if (r_sandbox_enable (0)) {
eprintf ("hashbang disabled in sandbox mode\n");
return false;
}
if (*lang=='/') {
const char *ptr = lang+1;
while (*lang) {
if (*lang=='/')
ptr = lang+1;
lang++;
}
RLangPlugin *p = r_lang_get_by_extension (core->lang, ptr);
if (p && p->name) lang = p->name;
}
if (*lang==' ') {
RLangPlugin *p = r_lang_get_by_extension (core->lang, input+2);
if (p && p->name) lang = p->name;
} else if (input[1]=='?' || input[1]=='*' || input[1]=='\0') {
r_lang_list (core->lang);
return true;
}
p = strchr (input, ' ');
if (p) *p=0;
// TODO: set argv here
if (r_lang_use (core->lang, lang)) {
r_lang_setup (core->lang);
if (p) {
r_lang_run_file (core->lang, p+1);
} else {
if (r_config_get_i (core->config, "scr.interactive")) {
r_lang_prompt (core->lang);
} else eprintf ("Error: scr.interactive required to run the rlang prompt\n");
}
} else {
if (!p || *p==' ')
eprintf ("Invalid hashbang. See '#!' for help.\n");
}
return true;
}
static int cmd_hash(void *data, const char *input) {
char algo[32];
RCore *core = (RCore *)data;
ut32 osize = 0, len = core->blocksize;
const char *ptr;
int pos = 0, handled_cmd = false;
switch (*input) {
case '\t':
case ' ':
return 0;
case '#':
if (!input[1]) {
algolist (1);
return true;
}
case '!':
return cmd_hash_bang (core, input);
}
ptr = strchr (input, ' ');
sscanf (input, "%31s", algo);
if (ptr && *(ptr+1) && r_num_is_valid_input (core->num, ptr+1)) {
int nlen = r_num_math (core->num, ptr+1);
if (nlen>0) len = nlen;
osize = core->blocksize;
if (nlen>core->blocksize) {
r_core_block_size (core, nlen);
if (nlen != core->blocksize) {
eprintf ("Invalid block size\n");
r_core_block_size (core, osize);
return true;
}
}
} else if (!ptr || !*(ptr+1)) osize = len;
/* TODO: Simplify this spaguetti monster */
while (osize > 0 && hash_handlers[pos].name != NULL) {
if (!r_str_ccmp (input, hash_handlers[pos].name, ' ')) {
hash_handlers[pos].handler (core->block, len);
handled_cmd = true;
break;
}
pos++;
}
if (!osize) {
eprintf ("Error: provided size must be size > 0\n");
}
if (*input == '?') {
const char *helpmsg[] = {
"Usage: #algo <size> @ addr", "", "",
" #"," comment","note the space after the sharp sign",
" ##","","List hash/checksum algorithms.",
" #sha256", " 10K @ 33","calculate sha256 of 10K at 33",
NULL
};
const char *helpmsg2[] = {
"Hashes:","","", NULL };
const char *helpmsg3[] = {
"Usage #!interpreter [<args>] [<file] [<<eof]","","",
" #!","","list all available interpreters",
" #!python","","run python commandline",
" #!python"," foo.py","run foo.py python script (same as '. foo.py')",
//" #!python <<EOF get python code until 'EOF' mark\n"
" #!python"," arg0 a1 <<q","set arg0 and arg1 and read until 'q'",
NULL};
r_core_cmd_help (core, helpmsg);
r_core_cmd_help (core, helpmsg2);
algolist (0);
r_core_cmd_help (core, helpmsg3);
}
if (osize)
r_core_block_size (core, osize);
return handled_cmd;
}