cmd_egg.c File Reference

#include <rz_core.h>
#include <rz_egg.h>

Go to the source code of this file.

Functions
static void	egg_option (RzEgg *egg, const char *key, const char *input)
static void	showBuffer (RzBuffer *b)
static bool	egg_compile (RzEgg *egg)
static bool	rz_core_egg_compile (RzEgg *egg)
static RzEgg *	rz_core_egg_setup (RzCore *core)
static RzCmdStatus	rz_core_egg_compile_file (RzCore *core, const char *file)
RZ_IPI RzCmdStatus	rz_egg_compile_handler (RzCore *core, int argc, const char **argv)
RZ_IPI RzCmdStatus	rz_egg_config_handler (RzCore *core, int argc, const char **argv)
RZ_IPI RzCmdStatus	rz_egg_list_plugins_handler (RzCore *core, int argc, const char **argv)
RZ_IPI RzCmdStatus	rz_egg_syscall_handler (RzCore *core, int argc, const char **argv)
RZ_IPI RzCmdStatus	rz_egg_type_handler (RzCore *core, int argc, const char **argv)
RZ_IPI RzCmdStatus	rz_egg_padding_handler (RzCore *core, int argc, const char **argv)
RZ_IPI RzCmdStatus	rz_egg_encoder_handler (RzCore *core, int argc, const char **argv)
RZ_IPI RzCmdStatus	rz_egg_reset_handler (RzCore *core, int argc, const char **argv)
RZ_IPI RzCmdStatus	rz_egg_show_config_handler (RzCore *core, int argc, const char **argv)

Variables
static const char *	RzEggConfigOptions []

Function Documentation

egg_compile()

static bool egg_compile ( RzEgg *  egg )

static

Definition at line 48 of file cmd_egg.c.

                                    {
    rz_egg_compile(egg);
    if (!rz_egg_assemble(egg)) {
        eprintf("rz_egg_assemble: invalid assembly\n");
        return false;
    }
    char *p = rz_egg_option_get(egg, "egg.padding");
    if (p && *p) {
        rz_egg_padding(egg, p);
        free(p);
    }
    p = rz_egg_option_get(egg, "egg.encoder");
    if (p && *p) {
        rz_egg_encode(egg, p);
        free(p);
    }
    RzBuffer *b;
    if ((b = rz_egg_get_bin(egg))) {
        showBuffer(b);
        return true;
    }
    return false;
}

References b, eprintf, free(), p, rz_egg_assemble(), rz_egg_compile(), rz_egg_encode(), rz_egg_get_bin(), rz_egg_option_get(), rz_egg_padding(), and showBuffer().

Referenced by rz_core_egg_compile(), and rz_core_egg_compile_file().

egg_option()

static void egg_option ( RzEgg *  egg, const char *  key, const char *  input  )

static

Definition at line 18 of file cmd_egg.c.

                                                                       {
    if (!*input) {
        return;
    }
    if (input[1] != ' ') {
        char *a = rz_egg_option_get(egg, key);
        if (a) {
            rz_cons_println(a);
            free(a);
        }
    } else {
        rz_egg_option_set(egg, key, input + 2);
    }
}

References a, free(), input(), key, rz_cons_println(), rz_egg_option_get(), and rz_egg_option_set().

Referenced by rz_egg_reset_handler().

rz_core_egg_compile()

static bool rz_core_egg_compile ( RzEgg *  egg )

static

Definition at line 72 of file cmd_egg.c.

                                            {
    int ret = false;
    char *p = rz_egg_option_get(egg, "egg.shellcode");
    if (p && *p) {
        if (!rz_egg_shellcode(egg, p)) {
            eprintf("Unknown shellcode '%s'\n", p);
            free(p);
            return false;
        }
        free(p);
    } else {
        eprintf("Setup a shellcode before (gi command)\n");
        free(p);
        return false;
    }
    ret = egg_compile(egg);
    rz_egg_option_set(egg, "egg.shellcode", "");
    rz_egg_option_set(egg, "egg.padding", "");
    rz_egg_option_set(egg, "egg.encoder", "");
    rz_egg_option_set(egg, "key", "");
 
    rz_egg_reset(egg);
    return ret;
}

References egg_compile(), eprintf, free(), p, rz_egg_option_get(), rz_egg_option_set(), rz_egg_reset(), and rz_egg_shellcode().

Referenced by rz_egg_compile_handler().

rz_core_egg_compile_file()

static RzCmdStatus rz_core_egg_compile_file ( RzCore *  core, const char *  file  )

static

Definition at line 110 of file cmd_egg.c.

                                                                            {
    rz_return_val_if_fail(file, false);
    RzEgg *egg = rz_core_egg_setup(core);
    if (!egg) {
        return RZ_CMD_STATUS_ERROR;
    }
    if (!rz_egg_load_file(egg, file)) {
        RZ_LOG_ERROR("Cannot load file \"%s\"\n", file);
        return RZ_CMD_STATUS_ERROR;
    }
    if (!egg_compile(egg)) {
        RZ_LOG_ERROR("Cannot compile file \"%s\"\n", file);
        return RZ_CMD_STATUS_ERROR;
    }
    return RZ_CMD_STATUS_OK;
}

References egg_compile(), RZ_CMD_STATUS_ERROR, RZ_CMD_STATUS_OK, rz_core_egg_setup(), rz_egg_load_file(), RZ_LOG_ERROR, and rz_return_val_if_fail.

Referenced by rz_egg_compile_handler().

rz_core_egg_setup()

static RzEgg* rz_core_egg_setup ( RzCore *  core )

static

Definition at line 97 of file cmd_egg.c.

                                              {
    RzEgg *egg = core->egg;
    const char *arch = rz_config_get(core->config, "asm.arch");
    const char *os = rz_config_get(core->config, "asm.os");
    int bits = rz_config_get_i(core->config, "asm.bits");
 
    if (!rz_egg_setup(egg, arch, bits, 0, os)) {
        RZ_LOG_ERROR("Cannot setup shellcode compiler for chosen configuration\n");
        return NULL;
    }
    return egg;
}

References arch, bits(), rz_core_t::config, rz_core_t::egg, NULL, rz_config_get(), rz_config_get_i(), rz_egg_setup(), and RZ_LOG_ERROR.

Referenced by rz_core_egg_compile_file(), rz_egg_compile_handler(), rz_egg_config_handler(), rz_egg_encoder_handler(), rz_egg_list_plugins_handler(), rz_egg_padding_handler(), rz_egg_reset_handler(), rz_egg_show_config_handler(), rz_egg_syscall_handler(), and rz_egg_type_handler().

rz_egg_compile_handler()

RZ_IPI RzCmdStatus rz_egg_compile_handler	(	RzCore *	core,
		int	argc,
		const char **	argv
	)

Definition at line 127 of file cmd_egg.c.

                                                                                     {
    if (argc > 1) {
        return rz_core_egg_compile_file(core, argv[1]);
    } else {
        RzEgg *egg = rz_core_egg_setup(core);
        if (!egg) {
            return RZ_CMD_STATUS_ERROR;
        }
        if (!rz_core_egg_compile(egg)) {
            RZ_LOG_ERROR("Cannot compile the shellcode\n");
            return RZ_CMD_STATUS_ERROR;
        }
    }
    return RZ_CMD_STATUS_OK;
}

References argv, RZ_CMD_STATUS_ERROR, RZ_CMD_STATUS_OK, rz_core_egg_compile(), rz_core_egg_compile_file(), rz_core_egg_setup(), and RZ_LOG_ERROR.

Referenced by rzshell_cmddescs_init().

rz_egg_config_handler()

RZ_IPI RzCmdStatus rz_egg_config_handler	(	RzCore *	core,
		int	argc,
		const char **	argv
	)

Definition at line 143 of file cmd_egg.c.

                                                                                    {
    RzEgg *egg = rz_core_egg_setup(core);
    if (!egg) {
        return RZ_CMD_STATUS_ERROR;
    }
    if (argc < 2) {
        // List all available config options and their values
        size_t i;
        for (i = 0; RzEggConfigOptions[i]; i++) {
            const char *p = RzEggConfigOptions[i];
            if (rz_egg_option_get(egg, p)) {
                rz_cons_printf("%s : %s\n", p, rz_egg_option_get(egg, p));
            } else {
                rz_cons_printf("%s : %s\n", p, "");
            }
        }
        return RZ_CMD_STATUS_OK;
    }
    int i;
    for (i = 1; i < argc; i++) {
        // Set the config option
        RzList *l = rz_str_split_duplist_n(argv[i], "=", 1, false);
        if (!l) {
            return RZ_CMD_STATUS_ERROR;
        }
        size_t llen = rz_list_length(l);
        if (!llen) {
            return RZ_CMD_STATUS_ERROR;
        }
        char *key = rz_list_get_n(l, 0);
        if (RZ_STR_ISEMPTY(key)) {
            RZ_LOG_ERROR("No config option name specified\n");
            rz_list_free(l);
            return RZ_CMD_STATUS_ERROR;
        }
        // If there is value specified - set it, if not - just show the value
        if (llen == 1) {
            // No value
            char *o = rz_egg_option_get(egg, key);
            if (!o) {
                RZ_LOG_ERROR("No such config option exists\n");
                rz_list_free(l);
                return RZ_CMD_STATUS_ERROR;
            }
            rz_cons_print(o);
            free(o);
        } else if (llen == 2) {
            char *value = rz_list_get_n(l, 1);
            if (RZ_STR_ISEMPTY(value)) {
                RZ_LOG_ERROR("No config option value  specified\n");
                rz_list_free(l);
                return RZ_CMD_STATUS_ERROR;
            }
            rz_egg_option_set(egg, key, value);
        }
        rz_list_free(l);
    }
    return RZ_CMD_STATUS_OK;
}

References argv, free(), i, key, p, RZ_CMD_STATUS_ERROR, RZ_CMD_STATUS_OK, rz_cons_printf(), rz_core_egg_setup(), rz_egg_option_get(), rz_egg_option_set(), rz_list_free(), rz_list_get_n(), rz_list_length(), RZ_LOG_ERROR, RZ_STR_ISEMPTY, rz_str_split_duplist_n(), RzEggConfigOptions, and value.

Referenced by rzshell_cmddescs_init().

rz_egg_encoder_handler()

RZ_IPI RzCmdStatus rz_egg_encoder_handler	(	RzCore *	core,
		int	argc,
		const char **	argv
	)

Definition at line 255 of file cmd_egg.c.

                                                                                     {
    RzEgg *egg = rz_core_egg_setup(core);
    if (!egg) {
        return RZ_CMD_STATUS_ERROR;
    }
    rz_egg_option_set(egg, "key", argv[2]);
    rz_egg_option_set(egg, "egg.encoder", argv[1]);
    return RZ_CMD_STATUS_OK;
}

References argv, RZ_CMD_STATUS_ERROR, RZ_CMD_STATUS_OK, rz_core_egg_setup(), and rz_egg_option_set().

Referenced by rzshell_cmddescs_init().

rz_egg_list_plugins_handler()

RZ_IPI RzCmdStatus rz_egg_list_plugins_handler	(	RzCore *	core,
		int	argc,
		const char **	argv
	)

Definition at line 203 of file cmd_egg.c.

                                                                                          {
    RzEgg *egg = rz_core_egg_setup(core);
    if (!egg) {
        return RZ_CMD_STATUS_ERROR;
    }
    RzListIter *iter;
    RzEggPlugin *p;
    rz_list_foreach (egg->plugins, iter, p) {
        rz_cons_printf("%s  %6s : %s\n",
            (p->type == RZ_EGG_PLUGIN_SHELLCODE) ? "shc" : "enc", p->name, p->desc);
    }
    return RZ_CMD_STATUS_OK;
}

References p, rz_egg_t::plugins, RZ_CMD_STATUS_ERROR, RZ_CMD_STATUS_OK, rz_cons_printf(), rz_core_egg_setup(), and RZ_EGG_PLUGIN_SHELLCODE.

Referenced by rzshell_cmddescs_init().

rz_egg_padding_handler()

RZ_IPI RzCmdStatus rz_egg_padding_handler	(	RzCore *	core,
		int	argc,
		const char **	argv
	)

Definition at line 246 of file cmd_egg.c.

                                                                                     {
    RzEgg *egg = rz_core_egg_setup(core);
    if (!egg) {
        return RZ_CMD_STATUS_ERROR;
    }
    rz_egg_option_set(egg, "egg.padding", argv[1]);
    return RZ_CMD_STATUS_OK;
}

References argv, RZ_CMD_STATUS_ERROR, RZ_CMD_STATUS_OK, rz_core_egg_setup(), and rz_egg_option_set().

Referenced by rzshell_cmddescs_init().

rz_egg_reset_handler()

RZ_IPI RzCmdStatus rz_egg_reset_handler	(	RzCore *	core,
		int	argc,
		const char **	argv
	)

Definition at line 265 of file cmd_egg.c.

                                                                                   {
    RzEgg *egg = rz_core_egg_setup(core);
    if (!egg) {
        return RZ_CMD_STATUS_ERROR;
    }
    egg_option(egg, "egg.padding", "");
    egg_option(egg, "egg.shellcode", "");
    egg_option(egg, "egg.encoder", "");
    return RZ_CMD_STATUS_OK;
}

References egg_option(), RZ_CMD_STATUS_ERROR, RZ_CMD_STATUS_OK, and rz_core_egg_setup().

Referenced by rzshell_cmddescs_init().

rz_egg_show_config_handler()

RZ_IPI RzCmdStatus rz_egg_show_config_handler	(	RzCore *	core,
		int	argc,
		const char **	argv
	)

Definition at line 276 of file cmd_egg.c.

                                                                                         {
    RzEgg *egg = rz_core_egg_setup(core);
    if (!egg) {
        return RZ_CMD_STATUS_ERROR;
    }
    rz_cons_printf("Configuration options\n");
    int i;
    for (i = 0; RzEggConfigOptions[i]; i++) {
        const char *p = RzEggConfigOptions[i];
        if (rz_egg_option_get(egg, p)) {
            rz_cons_printf("%s : %s\n", p, rz_egg_option_get(egg, p));
        } else {
            rz_cons_printf("%s : %s\n", p, "");
        }
    }
    rz_cons_printf("\nTarget options\n");
    rz_cons_printf("arch : %s\n", core->analysis->cpu);
    rz_cons_printf("os   : %s\n", core->analysis->os);
    rz_cons_printf("bits : %d\n", core->analysis->bits);
    return RZ_CMD_STATUS_OK;
}

References rz_core_t::analysis, rz_analysis_t::bits, rz_analysis_t::cpu, i, rz_analysis_t::os, p, RZ_CMD_STATUS_ERROR, RZ_CMD_STATUS_OK, rz_cons_printf(), rz_core_egg_setup(), rz_egg_option_get(), and RzEggConfigOptions.

Referenced by rzshell_cmddescs_init().

rz_egg_syscall_handler()

RZ_IPI RzCmdStatus rz_egg_syscall_handler	(	RzCore *	core,
		int	argc,
		const char **	argv
	)

Definition at line 217 of file cmd_egg.c.

                                                                                     {
    RzEgg *egg = rz_core_egg_setup(core);
    if (!egg) {
        return RZ_CMD_STATUS_ERROR;
    }
    RzBuffer *buf = NULL;
    if (argc > 2) {
        // With syscall parameters specified
        buf = rz_core_syscall(core, argv[1], argv[2]);
    } else {
        // Without any parameters
        buf = rz_core_syscall(core, argv[1], "");
    }
    if (buf) {
        showBuffer(buf);
    }
    egg->lang.nsyscalls = 0;
    return RZ_CMD_STATUS_OK;
}

References argv, rz_egg_t::lang, rz_egg_lang_t::nsyscalls, NULL, RZ_CMD_STATUS_ERROR, RZ_CMD_STATUS_OK, rz_core_egg_setup(), rz_core_syscall(), and showBuffer().

Referenced by rzshell_cmddescs_init().

rz_egg_type_handler()

RZ_IPI RzCmdStatus rz_egg_type_handler	(	RzCore *	core,
		int	argc,
		const char **	argv
	)

Definition at line 237 of file cmd_egg.c.

                                                                                  {
    RzEgg *egg = rz_core_egg_setup(core);
    if (!egg) {
        return RZ_CMD_STATUS_ERROR;
    }
    rz_egg_option_set(egg, "egg.shellcode", argv[1]);
    return RZ_CMD_STATUS_OK;
}

References argv, RZ_CMD_STATUS_ERROR, RZ_CMD_STATUS_OK, rz_core_egg_setup(), and rz_egg_option_set().

Referenced by rzshell_cmddescs_init().

showBuffer()

static void showBuffer ( RzBuffer *  b )

static

Definition at line 33 of file cmd_egg.c.

                                    {
    int i;
    if (b && rz_buf_size(b) > 0) {
        rz_buf_seek(b, 0, RZ_BUF_SET);
        for (i = 0; i < rz_buf_size(b); i++) {
            ut8 tmp;
            if (!rz_buf_read8(b, &tmp)) {
                return;
            }
            rz_cons_printf("%02x", tmp);
        }
        rz_cons_newline();
    }
}

References b, i, rz_buf_read8(), rz_buf_seek(), RZ_BUF_SET, rz_buf_size(), rz_cons_newline(), rz_cons_printf(), and autogen_x86imm::tmp.

Referenced by egg_compile(), and rz_egg_syscall_handler().

Variable Documentation

RzEggConfigOptions

const char* RzEggConfigOptions[]

static

Initial value:

= {
    "egg.shellcode",
    "egg.encoder",
    "egg.padding",
    "key",
    "cmd",
    "suid",
    NULL
}

Definition at line 8 of file cmd_egg.c.

Referenced by rz_egg_config_handler(), and rz_egg_show_config_handler().