rizin/librz/core/cfile.c
Khairul Azhar Kasmiran c1fc1419e4
Rename some rz_list funcs to rz_list_XXX_val (part 1) (#5671)
* Rename rz_list_find_ptr() to rz_list_find_val()
* Rename rz_list_delete_data() to rz_list_delete_val()
2025-12-25 22:01:49 +08:00

1743 lines
47 KiB
C

// SPDX-FileCopyrightText: 2009-2021 pancake <pancake@nopcode.org>
// SPDX-License-Identifier: LGPL-3.0-only
#include <rz_core.h>
#include <stdlib.h>
#include <string.h>
#include "core_private.h"
#include "core_private_base.h"
static bool core_file_do_load_for_debug(RzCore *r, ut64 baseaddr, const char *filenameuri);
static bool core_file_do_load_for_io_plugin(RzCore *r, ut64 baseaddr, ut64 loadaddr);
static RzCoreFile *core_file_new(RzCore *core, int fd) {
RzCoreFile *r = RZ_NEW0(RzCoreFile);
if (!r) {
return NULL;
}
r->core = core;
r->fd = fd;
rz_pvector_init(&r->binfiles, NULL);
rz_pvector_init(&r->extra_files, NULL);
rz_pvector_init(&r->maps, NULL);
return r;
}
RZ_IPI void rz_core_file_free(RzCoreFile *cf) {
if (!cf) {
return;
}
rz_pvector_fini(&cf->binfiles);
rz_pvector_fini(&cf->extra_files);
rz_pvector_fini(&cf->maps);
free(cf);
}
static bool __isMips(RzAsm *a) {
return a && a->cur && a->cur->arch && strstr(a->cur->arch, "mips");
}
static void loadGP(RzCore *core) {
if (__isMips(core->rasm)) {
ut64 gp = rz_num_math(core->num, "loc._gp");
if (!gp || gp == UT64_MAX) {
rz_config_set(core->config, "analysis.roregs", "zero");
ut64 addr = rz_num_math(core->num, "entry");
rz_core_seek_opt(core, addr, true, false);
rz_core_debug_step_one(core, 10);
rz_config_set(core->config, "analysis.roregs", "zero,gp");
gp = rz_reg_getv(core->analysis->reg, "gp");
}
// RZ_LOG_DEBUG("[mips] gp: 0x%08"PFMT64x"\n", gp);
rz_config_set_i(core->config, "analysis.gp", gp);
}
}
static RZ_OWN RzPVector /*<RzBinSection *>*/ *__save_old_sections(RzCore *core) {
RzBinObject *obj = rz_bin_cur_object(core->bin);
const RzPVector *sections = obj ? rz_bin_object_get_sections_all(obj) : NULL;
void **it;
RzBinSection *sec;
RzPVector *old_sections = rz_pvector_new(NULL);
// Return an empty list
if (!sections) {
RZ_LOG_WARN("No sections found, functions and flags won't be rebased\n");
return old_sections;
}
old_sections->v.free_user = sections->v.free_user;
rz_pvector_foreach (sections, it) {
sec = *it;
RzBinSection *old_sec = RZ_NEW0(RzBinSection);
if (!old_sec) {
break;
}
*old_sec = *sec;
old_sec->name = rz_str_dup(sec->name);
rz_pvector_push(old_sections, old_sec);
}
return old_sections;
}
struct __rebase_struct {
RzCore *core;
RzPVector /*<RzBinSection *>*/ *old_sections;
ut64 old_base;
ut64 diff;
int type;
};
#define __is_inside_section(item_addr, section) \
(item_addr >= old_base + section->vaddr && item_addr <= old_base + section->vaddr + section->vsize)
static bool __rebase_flags(RzFlagItem *flag, void *user) {
struct __rebase_struct *reb = user;
ut64 old_base = reb->old_base;
void **it;
RzBinSection *sec;
// Only rebase flags that were in the rebased sections, otherwise it will take too long
rz_pvector_foreach (reb->old_sections, it) {
sec = *it;
if (__is_inside_section(flag->offset, sec)) {
rz_flag_set(reb->core->flags, flag->name, flag->offset + reb->diff, flag->size);
break;
}
}
return true;
}
static bool __rebase_xrefs_i(void *user, const ut64 k, const void *v) {
struct __rebase_struct *reb = (void *)user;
RzAnalysisXRef *xref = (RzAnalysisXRef *)v;
xref->from += reb->diff;
xref->to += reb->diff;
rz_analysis_xrefs_set(reb->core->analysis, xref->from, xref->to, xref->type);
return true;
}
static bool __rebase_xrefs(void *user, const ut64 k, const void *v) {
HtUP *ht = (HtUP *)v;
ht_up_foreach(ht, __rebase_xrefs_i, user);
return true;
}
static void __rebase_everything(RzCore *core, RzPVector /*<RzBinSection *>*/ *old_sections, ut64 old_base) {
RzListIter *it;
RzAnalysisFunction *fcn;
ut64 new_base = core->bin->cur->o->baddr_shift;
RzBinSection *old_section;
ut64 diff = new_base - old_base;
if (!diff) {
return;
}
// FUNCTIONS
rz_list_foreach (core->analysis->fcns, it, fcn) {
void **iter;
rz_pvector_foreach (old_sections, iter) {
old_section = *iter;
if (!__is_inside_section(fcn->addr, old_section)) {
continue;
}
rz_analysis_function_relocate(fcn, fcn->addr + diff);
ut64 new_sec_addr = new_base + old_section->vaddr;
void **vit;
RzAnalysisBlock *bb;
rz_pvector_foreach (fcn->bbs, vit) {
bb = (RzAnalysisBlock *)*vit;
if (bb->addr >= new_sec_addr && bb->addr <= new_sec_addr + old_section->vsize) {
// Todo: Find better way to check if bb was already rebased
continue;
}
rz_analysis_block_relocate(bb, bb->addr + diff, bb->size);
if (bb->jump != UT64_MAX) {
bb->jump += diff;
}
if (bb->fail != UT64_MAX) {
bb->fail += diff;
}
}
break;
}
}
// FLAGS
struct __rebase_struct reb = {
core,
old_sections,
old_base,
diff
};
rz_flag_foreach(core->flags, __rebase_flags, &reb);
// META
rz_meta_rebase(core->analysis, diff);
// XREFS
HtUP *xrefs_from = core->analysis->ht_xrefs_from;
HtUP *xrefs_to = core->analysis->ht_xrefs_to;
core->analysis->ht_xrefs_from = NULL;
core->analysis->ht_xrefs_to = NULL;
rz_analysis_xrefs_init(core->analysis);
ht_up_foreach(xrefs_from, __rebase_xrefs, &reb);
ht_up_free(xrefs_from);
ht_up_free(xrefs_to);
// BREAKPOINTS
rz_debug_bp_rebase(core->dbg, old_base, new_base);
}
/**
* \brief Tries to open the file, load binary info and make RzIOMap
* \return Success?
*/
RZ_API bool rz_core_file_open_load(RZ_NONNULL RzCore *core, RZ_NONNULL const char *filepath, ut64 addr, int perms, bool write_mode) {
rz_return_val_if_fail(core && filepath, false);
RzCoreFile *cfile = rz_core_file_open(core, filepath, perms, addr);
if (!cfile) {
RZ_LOG_ERROR("Cannot open file '%s'\n", filepath);
return false;
}
core->num->value = cfile->fd;
if (addr == 0) { // if no baddr defined, use the one provided by the file
addr = UT64_MAX;
}
if (!rz_core_bin_load(core, filepath, addr)) {
RZ_LOG_ERROR("Cannot load binary info of '%s'.\n", filepath);
return false;
}
if (write_mode) {
RzIODesc *desc = rz_io_desc_get(core->io, cfile->fd);
if (!desc || !(desc->perm & RZ_PERM_W)) {
RZ_LOG_WARN("Cannot make maps for %s writable.\n", filepath);
return false;
}
void **it;
rz_pvector_foreach (&cfile->maps, it) {
RzIOMap *map = *it;
map->perm |= RZ_PERM_WX;
}
}
rz_core_block_read(core);
return true;
}
RZ_API void rz_core_file_reopen_remote_debug(RzCore *core, const char *uri, ut64 addr) {
RzCoreFile *ofile = core->file;
RzIODesc *desc;
RzCoreFile *file;
int fd;
if (!ofile || !(desc = rz_io_desc_get(core->io, ofile->fd)) || !desc->uri) {
RZ_LOG_ERROR("No file open?\n");
return;
}
core->dbg->main_arena_resolved = false;
RzPVector *old_sections = __save_old_sections(core);
ut64 old_base = core->bin->cur->o->baddr_shift;
int bits = core->rasm->bits;
rz_config_set_i(core->config, "asm.bits", bits);
rz_config_set_b(core->config, "cfg.debug", true);
// Set referer as the original uri so we could return to it with `oo`
desc->referer = desc->uri;
desc->uri = rz_str_dup(uri);
if ((file = rz_core_file_open(core, uri, RZ_PERM_R | RZ_PERM_W, addr))) {
fd = file->fd;
core->num->value = fd;
// if no baddr is defined, use the one provided by the file
if (addr == 0) {
desc = rz_io_desc_get(core->io, file->fd);
if (desc->plugin->isdbg) {
addr = rz_debug_get_baddr(core->dbg, desc->name);
} else {
addr = rz_bin_get_baddr(core->bin);
}
}
rz_core_bin_load(core, uri, addr);
} else {
RZ_LOG_ERROR("Cannot open file '%s'\n", uri);
rz_pvector_free(old_sections);
return;
}
rz_core_block_read(core);
if (rz_config_get_i(core->config, "dbg.rebase")) {
__rebase_everything(core, old_sections, old_base);
}
rz_pvector_free(old_sections);
rz_core_seek_to_register(core, "PC", false);
}
RZ_API void rz_core_file_reopen_debug(RzCore *core, const char *args) {
RzCoreFile *ofile = core->file;
RzIODesc *desc;
if (!ofile || !(desc = rz_io_desc_get(core->io, ofile->fd)) || !desc->uri) {
RZ_LOG_ERROR("No file open?\n");
return;
}
// Reopen the original file as read only since we can't open native debug while the
// file is open with write permissions
if (!(desc->plugin && desc->plugin->isdbg) && (desc->perm & RZ_PERM_W)) {
RZ_LOG_ERROR("Cannot debug file (%s) with permissions set to 0x%x.\n"
"Reopening the original file in read-only mode.\n",
desc->name, desc->perm);
rz_io_reopen(core->io, ofile->fd, RZ_PERM_R, 644);
desc = rz_io_desc_get(core->io, ofile->fd);
}
RzBinFile *bf = rz_bin_file_find_by_fd(core->bin, ofile->fd);
char *binpath = (bf && bf->file) ? rz_str_dup(bf->file) : NULL;
if (!binpath) {
if (rz_file_exists(desc->name)) {
binpath = rz_str_dup(desc->name);
}
}
if (!binpath) {
/* fallback to oo */
rz_core_io_file_open(core, core->io->desc->fd);
return;
}
core->dbg->main_arena_resolved = false;
RzPVector *old_sections = __save_old_sections(core);
ut64 old_base = core->bin->cur->o->baddr_shift;
int bits = core->rasm->bits;
char *bin_abspath = rz_file_abspath(binpath);
char *escaped_path = rz_str_arg_escape(bin_abspath);
char *newfile = RZ_STR_ISEMPTY(args) ? rz_str_newf("dbg://%s", escaped_path)
: rz_str_newf("dbg://%s %s", escaped_path, args);
desc->uri = newfile;
desc->referer = NULL;
rz_config_set_i(core->config, "asm.bits", bits);
rz_config_set_b(core->config, "cfg.debug", true);
rz_core_file_reopen(core, newfile, 0, 2);
if (rz_config_get_i(core->config, "dbg.rebase")) {
__rebase_everything(core, old_sections, old_base);
}
rz_pvector_free(old_sections);
rz_core_seek_to_register(core, "PC", false);
free(bin_abspath);
free(escaped_path);
free(binpath);
}
RZ_API bool rz_core_file_reopen(RzCore *core, const char *args, int perm, int loadbin) {
int isdebug = rz_config_get_b(core->config, "cfg.debug");
char *path;
ut64 laddr = rz_config_get_i(core->config, "bin.laddr");
RzCoreFile *file = NULL;
RzCoreFile *ofile = core->file;
RzBinFile *bf = ofile ? rz_bin_file_find_by_fd(core->bin, ofile->fd)
: NULL;
RzIODesc *odesc = (core->io && ofile) ? rz_io_desc_get(core->io, ofile->fd) : NULL;
char *ofilepath = NULL, *obinfilepath = (bf && bf->file) ? rz_str_dup(bf->file) : NULL;
int ret = false;
ut64 origoff = core->offset;
if (odesc) {
if (odesc->referer) {
ofilepath = odesc->referer;
} else if (odesc->uri) {
ofilepath = odesc->uri;
}
}
ut64 new_baddr = UT64_MAX;
if (args) {
new_baddr = rz_num_math(core->num, args);
if (new_baddr && new_baddr != UT64_MAX) {
rz_config_set_i(core->config, "bin.baddr", new_baddr);
} else {
new_baddr = UT64_MAX;
}
}
if (new_baddr == UT64_MAX) {
new_baddr = rz_config_get_i(core->config, "bin.baddr");
}
if (!core->file) {
RZ_LOG_ERROR("No file opened to reopen\n");
free(ofilepath);
free(obinfilepath);
return false;
}
int newpid = odesc ? odesc->fd : -1;
if (isdebug) {
rz_debug_kill(core->dbg, core->dbg->pid, core->dbg->tid, 9); // SIGKILL
do {
rz_debug_continue(core->dbg);
} while (!rz_debug_is_dead(core->dbg));
rz_debug_detach(core->dbg, core->dbg->pid);
perm = 7;
} else {
if (!perm) {
perm = 4; // RZ_PERM_R;
}
}
if (!ofilepath) {
RZ_LOG_ERROR("Unknown file path\n");
free(obinfilepath);
return false;
}
// HACK: move last mapped address to higher place
// XXX - why does this hack work?
// when the new memory maps are created.
path = rz_str_dup(ofilepath);
free(obinfilepath);
obinfilepath = rz_str_dup(ofilepath);
// rz_str_trim (path);
file = rz_core_file_open(core, path, perm, laddr);
if (isdebug) {
int newtid = newpid;
// XXX - select the right backend
if (core->file) {
newpid = rz_io_fd_get_pid(core->io, core->file->fd);
#if __linux__
core->dbg->main_pid = newpid;
newtid = newpid;
#else
newtid = rz_io_fd_get_tid(core->io, core->file->fd);
#endif
}
// Reset previous pid and tid
core->dbg->pid = -1;
core->dbg->tid = -1;
core->dbg->recoil_mode = RZ_DBG_RECOIL_NONE;
memset(&core->dbg->reason, 0, sizeof(core->dbg->reason));
// Reopen and attach
rz_core_setup_debugger(core, "native", true);
rz_debug_select(core->dbg, newpid, newtid);
}
if (file) {
bool had_rbin_info = false;
if (ofile && bf) {
if (rz_bin_file_delete(core->bin, bf)) {
had_rbin_info = true;
}
}
rz_core_file_close(ofile);
rz_core_file_set_by_file(core, file);
ofile = NULL;
odesc = NULL;
RZ_LOG_WARN("File %s reopened in %s mode\n", path,
(perm & RZ_PERM_W) ? "read-write" : "read-only");
if (loadbin && (loadbin == 2 || had_rbin_info)) {
ut64 baddr;
if (isdebug) {
baddr = rz_debug_get_baddr(core->dbg, path);
} else if (new_baddr != UT64_MAX) {
baddr = new_baddr;
} else {
baddr = rz_config_get_i(core->config, "bin.baddr");
}
ret = rz_core_bin_load(core, obinfilepath, baddr);
rz_core_bin_update_arch_bits(core);
if (!ret) {
RZ_LOG_ERROR("Error: Failed to reload rbin for: '%s'\n", path);
}
origoff = rz_num_math(core->num, "entry0");
}
if (core->bin->cur && core->io && rz_io_desc_get(core->io, file->fd) && !loadbin) {
// force here NULL because is causing uaf look this better in future XXX @alvarofe
core->bin->cur = NULL;
}
// close old file
} else if (ofile) {
RZ_LOG_ERROR("rz_core_file_reopen: Cannot reopen file: '%s' with perms 0x%x,"
" attempting to open read-only.\n",
path, perm);
// lower it down back
// ofile = rz_core_file_open (core, path, RZ_PERM_R, addr);
rz_core_file_set_by_file(core, ofile);
} else {
RZ_LOG_ERROR("Cannot reopen '%s'\n", path);
}
if (core->file) {
rz_io_use_fd(core->io, core->file->fd);
rz_core_block_read(core);
}
rz_core_seek(core, origoff, true);
if (isdebug) {
rz_core_debug_map_update_flags(core);
rz_core_reg_update_flags(core);
rz_core_seek_to_register(core, "PC", false);
} else {
loadGP(core);
}
// update analysis io bind
rz_io_bind(core->io, &(core->analysis->iob));
if (core->file && core->file->fd >= 0) {
rz_core_file_close_all_but(core);
}
rz_core_file_close_all_but(core);
// This is done to ensure that the file is correctly
// loaded into the view
free(obinfilepath);
// free (ofilepath);
free(path);
return ret;
}
static bool file_resize(RzCore *core, ut64 newsize, st64 delta) {
int ret;
ut64 oldsize = (core->file) ? rz_io_fd_size(core->io, core->file->fd) : 0;
if (delta) {
newsize = oldsize + delta;
}
bool grow = (newsize > oldsize);
if (grow) {
ret = rz_io_resize(core->io, newsize);
if (ret < 1) {
RZ_LOG_ERROR("rz_io_resize: cannot resize\n");
return false;
}
}
if (delta && core->offset < newsize) {
rz_io_shift(core->io, core->offset, grow ? newsize : oldsize, delta);
}
if (!grow) {
ret = rz_io_resize(core->io, newsize);
if (ret < 1) {
RZ_LOG_ERROR("rz_io_resize: cannot resize\n");
return false;
}
}
if (newsize < core->offset + core->blocksize || oldsize < core->offset + core->blocksize) {
rz_core_block_read(core);
}
return true;
}
RZ_API bool rz_core_file_resize(RzCore *core, ut64 newsize) {
return file_resize(core, newsize, 0);
}
RZ_API bool rz_core_file_resize_delta(RzCore *core, st64 delta) {
return file_resize(core, 0, delta);
}
RZ_API void rz_core_sysenv_end(RzCore *core) {
// This will be deprecated when moving the . commands to newshell
rz_sys_setenv("RZ_FILE", NULL);
rz_sys_setenv("RZ_OFFSET", NULL);
// remove temporary RZ_CONFIG file
char *rz_config = rz_sys_getenv("RZ_CONFIG");
if (rz_config) {
rz_file_rm(rz_config);
rz_sys_setenv("RZ_CONFIG", NULL);
free(rz_config);
}
}
RZ_API void rz_core_sysenv_begin(RzCore *core) {
char tmpbuf[64];
// This will be deprecated when moving the . commands to newshell
RzIODesc *desc = core->file ? rz_io_desc_get(core->io, core->file->fd) : NULL;
rz_sys_setenv("RZ_BIN_PDBSERVER", rz_config_get(core->config, "pdb.server"));
if (desc && desc->name) {
rz_sys_setenv("RZ_FILE", desc->name);
rz_sys_setenv("RZ_SIZE", rz_strf(tmpbuf, "%" PFMT64d, rz_io_desc_size(desc)));
}
rz_sys_setenv("RZ_OFFSET", rz_strf(tmpbuf, "%" PFMT64d, core->offset));
rz_sys_setenv("RZ_XOFFSET", rz_strf(tmpbuf, "0x%08" PFMT64x, core->offset));
rz_sys_setenv("RZ_ENDIAN", core->rasm->big_endian ? "big" : "little");
rz_sys_setenv("RZ_BSIZE", rz_strf(tmpbuf, "%d", core->blocksize));
// dump current config file so other r2 tools can use the same options
char *config_sdb_path = NULL;
int config_sdb_fd = rz_file_mkstemp(NULL, &config_sdb_path);
if (config_sdb_fd >= 0) {
close(config_sdb_fd);
}
Sdb *config_sdb = sdb_new(NULL, config_sdb_path, 0);
rz_config_serialize(core->config, config_sdb);
sdb_sync(config_sdb);
sdb_free(config_sdb);
rz_sys_setenv("RZ_CONFIG", config_sdb_path);
rz_sys_setenv("RZ_BIN_LANG", rz_config_get(core->config, "bin.lang"));
rz_sys_setenv("RZ_BIN_DEMANGLE", rz_config_get(core->config, "bin.demangle"));
rz_sys_setenv("RZ_ARCH", rz_config_get(core->config, "asm.arch"));
rz_sys_setenv("RZ_BITS", rz_config_get(core->config, "asm.bits"));
rz_sys_setenv("RZ_COLOR", rz_config_get_i(core->config, "scr.color") ? "1" : "0");
rz_sys_setenv("RZ_DEBUG", rz_config_get_b(core->config, "cfg.debug") ? "1" : "0");
rz_sys_setenv("RZ_IOVA", rz_config_get_i(core->config, "io.va") ? "1" : "0");
free(config_sdb_path);
}
#if !__linux__ && !__WINDOWS__
static ut64 get_base_from_maps(RzCore *core, const char *file) {
RzDebugMap *map;
RzListIter *iter;
ut64 b = 0LL;
rz_debug_map_sync(core->dbg); // update process memory maps
// Find the base address of the process,usually the first map entry with at least read permission, this is set to bin.baddr.
rz_list_foreach (core->dbg->maps, iter, map) {
if (map->perm & RZ_PERM_R) {
// TODO: make this more flexible
// XXX - why "copy/" here?
if (map->name && strstr(map->name, "copy/")) {
return map->addr;
}
if (map->file && !strcmp(map->file, file)) {
return map->addr;
}
if (map->name && !strcmp(map->name, file)) {
return map->addr;
}
// XXX - Commented out, as this could unexpected results
// b = map->addr;
}
}
// fallback resolution copied from cmd_debug.c:rz_debug_get_baddr
// Find the base address of the process,usually the first map entry with at least read permission, this is set to bin.baddr.
rz_list_foreach (core->dbg->maps, iter, map) {
if (map->perm & RZ_PERM_R) { // r--
return map->addr;
}
}
return b;
}
#endif
#if __linux__ || __APPLE__
static bool setbpint(RzCore *r, const char *mode, const char *sym) {
RzBreakpointItem *bp;
RzFlagItem *fi = rz_flag_get(r->flags, sym);
if (!fi) {
return false;
}
bp = rz_bp_add_sw(r->dbg->bp, fi->offset, 1, RZ_PERM_X);
if (bp) {
bp->internal = true;
#if __linux__
bp->data = rz_str_newf("echo %s: %s", mode, sym);
#else
bp->data = rz_str_newf("echo %s: %s;ps@rdi", mode, sym);
#endif
return true;
}
RZ_LOG_ERROR("Cannot set breakpoint at %s\n", sym);
return false;
}
#endif
// XXX - need to handle index selection during debugging
static bool core_file_do_load_for_debug(RzCore *r, ut64 baseaddr, const char *filenameuri) {
RzCoreFile *cf = rz_core_file_cur(r);
RzIODesc *desc = cf ? rz_io_desc_get(r->io, cf->fd) : NULL;
RzBinPlugin *plugin;
int xtr_idx = 0; // if 0, load all if xtr is used
// TODO : Honor file.path eval var too?
if (!strncmp("dbg://", filenameuri, 6)) {
filenameuri += 6;
}
if (!desc) {
return false;
}
if (cf) {
rz_debug_select(r->dbg, rz_io_fd_get_pid(r->io, cf->fd),
rz_io_fd_get_tid(r->io, cf->fd));
}
#if !__linux__
#if !__WINDOWS__
baseaddr = get_base_from_maps(r, filenameuri);
#endif
if (baseaddr != UT64_MAX) {
rz_config_set_i(r->config, "bin.baddr", baseaddr);
} else if (desc) {
ut64 base;
if (rz_io_desc_get_base(desc, &base) && base != UT64_MAX) {
baseaddr = base;
rz_config_set_i(r->config, "bin.baddr", baseaddr);
}
}
#endif
int fd = cf ? cf->fd : -1;
RzBinOptions opt;
rz_bin_options_init(&opt, fd, baseaddr, UT64_MAX, false);
opt.obj_opts.elf_load_sections = rz_config_get_b(r->config, "elf.load.sections");
opt.obj_opts.elf_checks_sections = rz_config_get_b(r->config, "elf.checks.sections");
opt.obj_opts.elf_checks_segments = rz_config_get_b(r->config, "elf.checks.segments");
opt.obj_opts.big_endian = rz_config_get_b(r->config, "cfg.bigendian");
opt.xtr_idx = xtr_idx;
RzBinFile *binfile = rz_bin_open(r->bin, filenameuri, &opt);
if (!binfile) {
RZ_LOG_ERROR("bin: debug: Cannot open '%s'\n", filenameuri);
return false;
}
if (binfile && cf) {
rz_pvector_push(&cf->binfiles, binfile);
}
if (*rz_config_get(r->config, "dbg.libs")) {
rz_core_cmd0(r, ".dmm*");
#if __linux__
setbpint(r, "dbg.libs", "sym.imp.dlopen");
setbpint(r, "dbg.libs", "sym.imp.dlmopen");
setbpint(r, "dbg.unlibs", "sym.imp.dlclose");
#elif __APPLE__
setbpint(r, "dbg.libs", "sym._dlopen");
setbpint(r, "dbg.libs", "sym._dlclose");
#endif
}
rz_core_bin_apply_all_info(r, binfile);
rz_debug_reg_profile_sync(r->dbg);
plugin = rz_bin_file_cur_plugin(binfile);
if (!(plugin && !strcmp(plugin->name, "any")) && binfile) {
RzBinObject *obj = rz_bin_cur_object(r->bin);
RzBinInfo *info = obj ? obj->info : NULL;
if (plugin && info) {
rz_core_bin_set_arch_bits(r, binfile->file, info->arch, info->bits);
}
}
return true;
}
static bool core_file_do_load_for_io_plugin(RzCore *r, ut64 baseaddr, ut64 loadaddr) {
RzCoreFile *cf = rz_core_file_cur(r);
int fd = cf ? cf->fd : -1;
int xtr_idx = 0; // if 0, load all if xtr is used
RzBinPlugin *plugin;
if (fd < 0) {
return false;
}
rz_io_use_fd(r->io, fd);
RzBinOptions opt;
rz_core_bin_options_init(r, &opt, fd, baseaddr, loadaddr);
opt.xtr_idx = xtr_idx;
RzBinFile *binfile = rz_bin_open_io(r->bin, &opt);
if (!binfile) {
// RZ_LOG_ERROR("Failed to load the bin with an IO Plugin.\n");
return false;
}
if (cf) {
rz_pvector_push(&cf->binfiles, binfile);
}
rz_core_bin_apply_all_info(r, binfile);
plugin = rz_bin_file_cur_plugin(binfile);
if (plugin && !strcmp(plugin->name, "any")) {
RzBinObject *obj = rz_bin_cur_object(r->bin);
RzBinInfo *info = obj ? obj->info : NULL;
if (!info) {
return false;
}
info->bits = r->rasm->bits;
rz_core_bin_set_arch_bits(r, binfile->file, info->arch, info->bits);
} else if (binfile) {
RzBinObject *obj = rz_bin_cur_object(r->bin);
RzBinInfo *info = obj ? obj->info : NULL;
if (!info) {
return false;
}
if (plugin && info) {
rz_core_bin_set_arch_bits(r, binfile->file,
info->arch, info->bits);
}
}
return true;
}
static bool try_loadlib(RzCore *core, const char *lib, ut64 addr) {
if (rz_core_file_open(core, lib, 0, addr) != NULL) {
rz_core_bin_load(core, lib, addr);
return true;
}
return false;
}
RZ_API bool rz_core_file_loadlib(RzCore *core, const char *lib, ut64 libaddr) {
const char *dirlibs = rz_config_get(core->config, "dir.libs");
char *libdir = rz_path_libdir(core->sys_path);
if (!dirlibs || !*dirlibs) {
dirlibs = "." RZ_SYS_DIR;
}
const char *ldlibrarypath[] = {
dirlibs,
libdir,
#ifndef __WINDOWS__
"/usr/local/lib",
"/usr/lib",
"/lib",
#endif
"." RZ_SYS_DIR,
NULL
};
const char **libpath = (const char **)&ldlibrarypath;
bool ret = false;
#ifdef __WINDOWS__
if (strlen(lib) >= 3 && lib[1] == ':' && lib[2] == '\\') {
#else
if (*lib == '/') {
#endif
if (try_loadlib(core, lib, libaddr)) {
ret = true;
}
} else {
while (*libpath) {
char *s = rz_str_newf("%s" RZ_SYS_DIR "%s", *libpath, lib);
if (try_loadlib(core, s, libaddr)) {
ret = true;
}
free(s);
if (ret) {
break;
}
libpath++;
}
}
free(libdir);
return ret;
}
static void load_scripts_for(RzCore *core, const char *name) {
char *file;
RzListIter *iter;
char *binrc = rz_path_home_prefix(RZ_BINRC);
char tmp[50];
char *hdir = rz_file_path_join(binrc, rz_strf(tmp, "bin-%s", name));
free(binrc);
RzList *files = rz_sys_dir(hdir);
if (!rz_list_empty(files)) {
RZ_LOG_INFO("[binrc] path: '%s'\n", hdir);
}
rz_list_foreach (files, iter, file) {
if (*file && *file != '.') {
RZ_LOG_INFO("[binrc] loading '%s'\n", file);
char *fullpath = rz_file_path_join(hdir, file);
rz_core_run_script(core, fullpath);
free(fullpath);
}
}
rz_list_free(files);
free(hdir);
}
typedef struct {
const char *name;
bool found;
} RzCoreFileData;
static bool filecb(void *user, void *data, ut32 id) {
RzCoreFileData *filedata = user;
RzIODesc *desc = (RzIODesc *)data;
if (!strcmp(desc->name, filedata->name)) {
filedata->found = true;
}
return true;
}
static bool file_is_loaded(RzCore *core, const char *lib) {
RzCoreFileData filedata = { lib, false };
rz_id_storage_foreach(core->io->files, filecb, &filedata);
return filedata.found;
}
typedef struct {
const char *name;
ut64 addr;
RzBin *bin;
} RzCoreLinkData;
static bool resolve_import_cb(RzCoreLinkData *ld, RzIODesc *desc, ut32 id) {
rz_return_val_if_fail(ld && desc, false);
RzBinFile *bf = rz_bin_file_find_by_fd(ld->bin, desc->fd);
if (!bf) {
return true;
}
void **iter;
RzBinSymbol *sym;
RzPVector *symbols = rz_bin_file_get_symbols(bf);
rz_pvector_foreach (symbols, iter) {
sym = *iter;
if (!strcmp(sym->name, ld->name)) {
ld->addr = sym->vaddr;
return false;
}
}
return true;
}
static bool map_multi_dex(RzCore *core, RzIODesc *desc, ut32 id) {
rz_return_val_if_fail(core && desc, false);
if (!rz_str_endswith(desc->name, ".dex")) {
return true;
}
RzCoreFile *cf = rz_core_file_cur(core);
// adds the current size and aligns next address
ut64 base_address = RZ_CORE_BASE_ADDRESS_DEX;
RzBinFile *cur = rz_pvector_empty(&cf->binfiles) ? NULL : rz_pvector_tail(&cf->binfiles);
if (cur) {
RzIODesc *iod = rz_io_desc_get(core->io, cur->fd);
if (iod) {
base_address = cur->loadaddr;
base_address += rz_io_desc_size(iod);
rz_core_align_base_address(base_address);
}
}
ut64 size = rz_io_desc_size(desc);
RZ_LOG_INFO("Mapping at 0x%08" PFMT64x " with size 0x08%" PFMT64x " %s\n", base_address, size, desc->name);
rz_io_use_fd(core->io, desc->fd);
RzBinOptions opt;
rz_core_bin_options_init(core, &opt, desc->fd, base_address, 0);
opt.xtr_idx = 0;
RzBinFile *binfile = rz_bin_open_io(core->bin, &opt);
if (!binfile) {
RZ_LOG_ERROR("Cannot load bin file %s.\n", desc->name);
return true;
}
binfile->loadaddr = base_address;
rz_pvector_push(&cf->binfiles, binfile);
rz_core_bin_apply_all_info(core, binfile);
return true;
}
/**
* \brief Load the binary specified by the \p filenameuri URI using \p baddr base address
*
* \param core A pointer to the `RzCore` object used for opening file into
* \param filenameuri A file name/path or any other URI supported by Rizin
* \param baddr The base address to use for loading. UT64_MAX value enables autodetection
*
* \return true if loading successfully completed, false otherwise.
*/
RZ_API bool rz_core_bin_load(RZ_NONNULL RzCore *r, RZ_NULLABLE const char *filenameuri, ut64 baddr) {
rz_return_val_if_fail(r, false);
RzCoreFile *cf = rz_core_file_cur(r);
RzIODesc *desc = cf ? rz_io_desc_get(r->io, cf->fd) : NULL;
ut64 laddr = rz_config_get_i(r->config, "bin.laddr");
RzBinFile *binfile = NULL;
RzBinPlugin *plugin = NULL;
bool is_io_load = false;
const char *cmd_load;
if (!cf) {
return false;
}
is_io_load = desc && desc->plugin;
if (desc && RZ_STR_ISEMPTY(filenameuri)) {
filenameuri = desc->name;
}
if (RZ_STR_ISEMPTY(filenameuri)) {
RZ_LOG_ERROR("rz_core_bin_load: no file specified\n");
return false;
}
if (is_io_load) {
// TODO? necessary to restore the desc back?
// Fix to select pid before trying to load the binary
if ((desc->plugin && desc->plugin->isdbg) || rz_config_get_b(r->config, "cfg.debug")) {
core_file_do_load_for_debug(r, baddr, filenameuri);
} else {
core_file_do_load_for_io_plugin(r, baddr, 0LL);
if (!strncmp(filenameuri, "apk://", 6) && r->io->files->size > 1) {
RZ_LOG_INFO("Found multidex APK, mapping extra files\n");
rz_id_storage_foreach(r->io->files, (RzIDStorageForeachCb)map_multi_dex, r);
// forbid loading libraries even when set.
rz_config_set_b(r->config, "bin.libs", false);
}
}
// Restore original desc
rz_io_use_fd(r->io, desc->fd);
}
binfile = rz_bin_cur(r->bin);
if (cf && binfile && desc) {
binfile->fd = desc->fd;
}
// rz_core_bin_apply_all_info (r, binfile);
plugin = rz_bin_file_cur_plugin(binfile);
if (plugin) {
if (plugin->strfilter) {
char msg[2];
msg[0] = plugin->strfilter;
msg[1] = 0;
rz_config_set(r->config, "bin.str.filter", msg);
}
if (plugin->name) {
load_scripts_for(r, plugin->name);
}
}
cmd_load = rz_config_get(r->config, "cmd.load");
if (cmd_load && *cmd_load) {
rz_core_cmd(r, cmd_load, 0);
}
if (plugin && plugin->name) {
if (!strcmp(plugin->name, "any")) {
if (rz_str_startswith(desc->name, "rap") && strstr(desc->name, "://")) {
rz_io_map_new(r->io, desc->fd, desc->perm, 0, laddr, UT64_MAX);
} else {
rz_io_map_new(r->io, desc->fd, desc->perm, 0, laddr, rz_io_desc_size(desc));
}
} else if (binfile) {
RzBinObject *obj = rz_bin_cur_object(r->bin);
if (obj) {
bool va = obj->info ? obj->info->has_va : 0;
if (!va) {
rz_config_set_i(r->config, "io.va", 0);
}
// workaround to map correctly malloc:// and raw binaries
if (rz_io_desc_is_dbg(desc) || (rz_pvector_empty(obj->maps) || !va)) {
rz_io_map_new(r->io, desc->fd, desc->perm, 0, laddr, rz_io_desc_size(desc));
}
RzBinInfo *info = obj->info;
if (info) {
rz_core_bin_set_arch_bits(r, binfile->file, info->arch, info->bits);
} else {
rz_core_bin_set_arch_bits(r, binfile->file,
rz_config_get(r->config, "asm.arch"),
rz_config_get_i(r->config, "asm.bits"));
}
}
}
} else {
if (desc) {
rz_io_map_new(r->io, desc->fd, desc->perm, 0, laddr, rz_io_desc_size(desc));
}
if (binfile) {
ut16 bits = rz_config_get_i(r->config, "asm.bits");
const char *arch = rz_config_get(r->config, "asm.arch");
rz_core_bin_set_arch_bits(r, binfile->file, arch, bits);
}
}
if (desc && rz_config_get_i(r->config, "io.exec")) {
desc->perm |= RZ_PERM_X;
}
if (!rz_config_get_b(r->config, "cfg.debug")) {
loadGP(r);
}
if (rz_config_get_b(r->config, "bin.libs")) {
const char *lib;
void **iter;
RzBinObject *o = rz_bin_cur_object(r->bin);
const RzPVector *libs = rz_bin_object_get_libs(o);
if (libs) {
rz_pvector_foreach (libs, iter) {
lib = *iter;
if (file_is_loaded(r, lib)) {
continue;
}
RZ_LOG_INFO("Opening library %s\n", lib);
ut64 baddr = rz_io_map_location(r->io, 0x200000);
if (baddr != UT64_MAX) {
rz_core_file_loadlib(r, lib, baddr);
}
}
}
rz_core_bin_raise(r, 0);
ut64 ocurr = r->offset;
ut64 entry0addr = rz_num_math(r->num, "entry0");
rz_core_seek(r, entry0addr, true);
rz_config_set_b(r->config, "bin.at", true);
RZ_LOG_INFO("Linking imports...\n");
RzBinImport *imp;
RzBinObject *bin_obj = rz_bin_cur_object(r->bin);
const RzPVector *imports = rz_bin_object_get_imports(bin_obj);
void **vec_iter;
rz_pvector_foreach (imports, vec_iter) {
imp = *vec_iter;
char *name = rz_str_newf("sym.imp.%s", imp->name);
rz_name_filter(name + 8, strlen(name + 8) + 1, true);
RzFlagItem *flag = rz_flag_get(r->flags, name);
if (!flag) {
// RZ_LOG_DEBUG("Cannot find flag %s\n", name);
free(name);
continue;
}
ut64 imp_addr = flag->offset;
RzCoreLinkData linkdata = { imp->name, UT64_MAX, r->bin };
rz_id_storage_foreach(r->io->files, (RzIDStorageForeachCb)resolve_import_cb, &linkdata);
if (linkdata.addr != UT64_MAX) {
RZ_LOG_INFO("Resolved %s with address 0x%08" PFMT64x "\n", name, linkdata.addr);
ut64 a = linkdata.addr;
ut64 b = imp_addr;
rz_analysis_xrefs_set(r->analysis, b, a, RZ_ANALYSIS_XREF_TYPE_NULL);
} else {
RZ_LOG_ERROR("Cannot resolve %s\n", name);
}
free(name);
}
rz_core_seek(r, ocurr, true);
}
// If type == RZ_BIN_TYPE_CORE, we need to create all the maps
if (plugin && binfile && plugin->file_type && plugin->file_type(binfile) == RZ_BIN_TYPE_CORE) {
// Setting the right arch and bits, so regstate will be shown correctly
if (plugin->info) {
RzBinInfo *inf = plugin->info(binfile);
RZ_LOG_INFO("Setting up coredump arch-bits to: %s-%d\n", inf->arch, inf->bits);
rz_config_set(r->config, "asm.arch", inf->arch);
rz_config_set_i(r->config, "asm.bits", inf->bits);
rz_bin_info_free(inf);
}
if (binfile->o->regstate) {
if (rz_reg_arena_set_bytes(r->analysis->reg, binfile->o->regstate)) {
RZ_LOG_WARN("Setting up coredump: Problem while setting the registers\n");
} else {
RZ_LOG_INFO("Setting up coredump: Registers have been set\n");
}
}
}
return true;
}
/**
* \brief Open the file as a compilation of files
*
* Calls rz_io_open_many and maps all the file descriptors to an RzCoreFile
*/
RZ_API RZ_BORROW RzCoreFile *rz_core_file_open_many(RZ_NONNULL RzCore *r, RZ_NULLABLE const char *file, int perm, ut64 base_address) {
RzList *list_fds = rz_io_open_many(r->io, file, perm, 0644);
if (rz_list_empty(list_fds)) {
rz_list_free(list_fds);
return NULL;
}
if (!base_address) {
base_address = RZ_CORE_BASE_ADDRESS_DEFAULT;
}
RzListIter *it = NULL;
RzIODesc *desc = NULL;
RzIODesc *first = NULL;
rz_list_foreach (list_fds, it, desc) {
if (!rz_io_desc_add(r->io, desc)) {
rz_io_desc_free(desc);
continue;
} else if (!first) {
first = desc;
}
RzCoreFile *fh = core_file_new(r, desc->fd);
if (!fh) {
continue;
}
r->file = fh;
rz_list_append(r->files, fh);
ut64 size = rz_io_desc_size(desc);
RZ_LOG_INFO("Mapping at 0x%08" PFMT64x " with size 0x08%" PFMT64x " %s\n", base_address, size, desc->name);
if (!rz_core_bin_load(r, desc->name, base_address)) {
RZ_LOG_ERROR("failed to load %s\n", desc->name);
}
// adds the current size and aligns next address
base_address += size;
rz_core_align_base_address(base_address);
}
rz_list_free(list_fds);
return rz_list_first_val(r->files);
}
/**
* \brief Tries to open the file as is, otherwise tries as is a compilation of files
*
* Calls rz_io_open_nomap but if it fails, then tries with rz_core_file_open_many;
* Also, loadaddr is rizin -m (mapaddr)
*/
RZ_API RZ_BORROW RzCoreFile *rz_core_file_open(RZ_NONNULL RzCore *r, RZ_NONNULL const char *file, int flags, ut64 loadaddr) {
rz_return_val_if_fail(r && file, NULL);
ut64 prev = rz_time_now_mono();
const bool openmany = rz_config_get_i(r->config, "file.openmany");
RzCoreFile *fh = NULL;
if (!strcmp(file, "=")) {
file = "malloc://512";
}
// if not flags was passed open it with -r--
if (!flags) {
flags = RZ_PERM_R;
}
r->io->bits = r->rasm->bits; // TODO: we need an api for this
RzIODesc *fd = rz_io_open_nomap(r->io, file, flags, 0644);
if (rz_cons_is_breaked()) {
goto beach;
}
if (!fd && openmany) {
if (!(fh = rz_core_file_open_many(r, file, flags, loadaddr))) {
goto beach;
}
fd = rz_io_desc_get(r->io, fh->fd);
}
if (!fd) {
if (flags & RZ_PERM_W) {
// flags |= RZ_IO_CREAT;
if (!(fd = rz_io_open_nomap(r->io, file, flags, 0644))) {
goto beach;
}
} else {
goto beach;
}
}
if (rz_io_is_listener(r->io)) {
rz_io_desc_free(fd);
goto beach;
}
if (!fh) {
fh = core_file_new(r, fd->fd);
if (!fh) {
RZ_LOG_ERROR("rz_core_file_open: failed to allocate RzCoreFile.\n");
goto beach;
}
}
{
const char *cp = rz_config_get(r->config, "cmd.open");
if (cp && *cp) {
rz_core_cmd(r, cp, 0);
}
char *absfile = rz_file_abspath(file);
rz_config_set(r->config, "file.path", absfile);
free(absfile);
}
r->file = fh;
rz_io_use_fd(r->io, fd->fd);
if (!rz_list_find_val(r->files, fh)) {
rz_list_append(r->files, fh);
}
if (rz_config_get_b(r->config, "cfg.debug")) {
bool swstep = true;
if (r->dbg->cur && r->dbg->cur->canstep) {
swstep = false;
}
rz_config_set_i(r->config, "dbg.swstep", swstep);
// Set the correct debug handle
if (fd->plugin && fd->plugin->isdbg) {
char *dh = rz_str_ndup(file, (strstr(file, "://") - file));
if (dh) {
rz_debug_use(r->dbg, dh);
free(dh);
}
}
}
// used by rz_core_bin_load otherwise won't load correctly
// this should be argument of rz_core_bin_load <shrug>
if (loadaddr != UT64_MAX) {
rz_config_set_i(r->config, "bin.laddr", loadaddr);
}
beach:
r->times->file_open_time = rz_time_now_mono() - prev;
return fh;
}
/**
* Allocates the memory chunk of \p len size, copies the contents at the current offset, and opens it as a file.
* \param core RzCore instanced
* \param len Size of the chunk to be created and copied
* \param offset Offset from where to copy bytes
* \return true on success, false otherwise
*/
RZ_API bool rz_core_file_malloc_copy_chunk(RzCore *core, size_t len, ut64 offset) {
rz_return_val_if_fail(core && len, false);
bool res = false;
ut8 *data = RZ_NEWS(ut8, len);
if (!data) {
return false;
}
if (!rz_io_read_at(core->io, offset, data, len)) {
RZ_LOG_ERROR("Cannot read %zu bytes from offset 0x%" PFMT64x ".\n", len, offset);
goto err;
}
char uri[100];
rz_strf(uri, "malloc://%zu", len);
RzCoreFile *cfile = rz_core_file_open(core, uri, RZ_PERM_RWX, 0);
if (!cfile) {
RZ_LOG_ERROR("Cannot open '%s'.\n", uri);
goto err;
}
if (!rz_core_bin_load(core, uri, 0)) {
RZ_LOG_ERROR("Cannot load binary info of '%s'.\n", uri);
goto err;
}
RzIODesc *desc = rz_io_desc_get(core->io, cfile->fd);
rz_warn_if_fail(desc);
rz_io_desc_write_at(desc, 0, data, len);
res = true;
err:
free(data);
return res;
}
RZ_IPI void rz_core_file_io_desc_closed(RzCore *core, RzIODesc *desc) {
// remove all references to the closed desc
RzListIter *it;
RzCoreFile *cf;
rz_list_foreach (core->files, it, cf) {
rz_pvector_remove_data(&cf->extra_files, desc);
}
}
RZ_IPI void rz_core_file_io_map_deleted(RzCore *core, RzIOMap *map) {
// remove all references to the deleted map
RzListIter *it;
RzCoreFile *cf;
rz_list_foreach (core->files, it, cf) {
rz_pvector_remove_data(&cf->maps, map);
}
rz_core_io_map_info_free(map->user);
}
RZ_IPI void rz_core_file_bin_file_deleted(RzCore *core, RzBinFile *bf) {
// remove all references to the deleted binfile
RzListIter *it;
RzCoreFile *cf;
rz_list_foreach (core->files, it, cf) {
rz_pvector_remove_data(&cf->binfiles, bf);
}
}
RZ_API void rz_core_file_close(RzCoreFile *fh) {
rz_return_if_fail(fh && fh->core);
RzCore *r = fh->core;
RzListIter *fh_it = rz_list_find_val(r->files, fh);
rz_return_if_fail(fh_it);
RzIODesc *desc = rz_io_desc_get(r->io, fh->fd);
if (desc) {
rz_io_desc_close(desc);
}
while (!rz_pvector_empty(&fh->maps)) {
// The element will automatically be removed from the vector through events
// always delete the last to avoid unnecessary copies
RzIOMap *map = rz_pvector_at(&fh->maps, rz_pvector_len(&fh->maps) - 1);
rz_io_map_del(r->io, map->id);
}
while (!rz_pvector_empty(&fh->extra_files)) {
// same as for maps above
rz_io_desc_close(rz_pvector_at(&fh->extra_files, rz_pvector_len(&fh->extra_files) - 1));
}
while (!rz_pvector_empty(&fh->binfiles)) {
// same as for maps above
rz_bin_file_delete(r->bin, rz_pvector_at(&fh->binfiles, rz_pvector_len(&fh->binfiles) - 1));
}
if (r->file == fh) {
r->file = NULL;
}
rz_list_delete(r->files, fh_it);
}
RZ_API RzCoreFile *rz_core_file_get_by_fd(RzCore *core, int fd) {
RzCoreFile *file;
RzListIter *iter;
rz_list_foreach (core->files, iter, file) {
if (file->fd == fd) {
return file;
}
}
return NULL;
}
RZ_API bool rz_core_raw_file_print(RzCore *core) {
RzCoreFile *f;
RzIODesc *desc;
RzBinFile *bf;
RzListIter *it1, *it2, *it3;
rz_list_foreach (core->files, it1, f) {
desc = rz_io_desc_get(core->io, f->fd);
if (!desc) {
continue;
}
bool header_loaded = false;
rz_list_foreach (core->bin->binfiles, it2, bf) {
if (bf->fd == f->fd) {
header_loaded = true;
break;
}
}
if (!header_loaded) {
RzList *maps = rz_io_map_get_for_fd(core->io, f->fd);
RzIOMap *current_map;
char *absfile = rz_file_abspath(desc->uri);
rz_list_foreach (maps, it3, current_map) {
if (current_map) {
rz_cons_printf("on %s 0x%" PFMT64x "\n", absfile, current_map->itv.addr);
}
}
rz_list_free(maps);
free(absfile);
}
}
return true;
}
RZ_API bool rz_core_file_print(RzCore *core, RzOutputMode mode) {
int count = 0;
RzCoreFile *f;
RzIODesc *desc;
ut64 from;
RzListIter *iter;
PJ *pj = NULL;
if (mode == RZ_OUTPUT_MODE_JSON) {
pj = pj_new();
if (!pj) {
return false;
}
pj_a(pj);
}
rz_list_foreach (core->files, iter, f) {
desc = rz_io_desc_get(core->io, f->fd);
if (!desc) {
// cannot find desc for this fd, RzCoreFile inconsistency!!!1
continue;
}
from = 0LL;
switch (mode) {
case RZ_OUTPUT_MODE_JSON: { // "oij"
pj_o(pj);
pj_kb(pj, "raised", core->io->desc->fd == f->fd);
pj_ki(pj, "fd", f->fd);
pj_ks(pj, "uri", desc->uri);
pj_kn(pj, "from", (ut64)from);
pj_kb(pj, "writable", desc->perm & RZ_PERM_W);
pj_ki(pj, "size", (int)rz_io_desc_size(desc));
pj_end(pj);
break;
}
default: {
ut64 sz = rz_io_desc_size(desc);
const char *fmt;
if (sz == UT64_MAX) {
fmt = "%c %d %d %s @ 0x%" PFMT64x " ; %s size=%" PFMT64d "\n";
} else {
fmt = "%c %d %d %s @ 0x%" PFMT64x " ; %s size=%" PFMT64u "\n";
}
rz_cons_printf(fmt,
core->io->desc->fd == f->fd ? '*' : '-',
count,
(int)f->fd, desc->uri, (ut64)from,
desc->perm & RZ_PERM_W ? "rw" : "r",
rz_io_desc_size(desc));
} break;
}
count++;
}
if (mode == RZ_OUTPUT_MODE_JSON) {
pj_end(pj);
rz_cons_println(pj_string(pj));
pj_free(pj);
}
return true;
}
// XXX - needs to account for binfile index and bin object index
RZ_API bool rz_core_file_bin_raise(RzCore *core, ut32 bfid) {
RzBin *bin = core->bin;
RzBinFile *bf = rz_list_get_n(bin->binfiles, bfid);
bool res = false;
if (bf) {
res = rz_bin_file_set_obj(bf, bf->o);
res &= rz_bin_set_cur_binfile(bin, bf);
if (res) {
rz_io_use_fd(core->io, bf->fd);
}
res = res ? rz_core_file_set_by_fd(core, bf->fd) : res;
}
return res;
}
RZ_API int rz_core_file_binlist(RzCore *core) {
int count = 0;
RzListIter *iter;
RzCoreFile *cur_cf = core->file, *cf = NULL;
RzBinFile *binfile = NULL;
RzIODesc *desc;
RzBin *bin = core->bin;
const RzList *binfiles = bin ? bin->binfiles : NULL;
if (!binfiles) {
return false;
}
rz_list_foreach (binfiles, iter, binfile) {
int fd = binfile->fd;
cf = rz_core_file_get_by_fd(core, fd);
desc = rz_io_desc_get(core->io, fd);
if (cf) {
rz_cons_printf("%c %d %s ; %s\n",
core->io->desc == desc ? '*' : '-',
fd, desc->uri, desc->perm & RZ_PERM_W ? "rw" : "r");
}
}
rz_core_file_set_by_file(core, cur_cf);
// rz_core_bin_bind (core, cur_bf);
return count;
}
static bool close_but_cb(void *user, void *data, ut32 id) {
RzCore *core = (RzCore *)user;
RzIODesc *desc = (RzIODesc *)data;
if (core && desc && core->file) {
if (desc->fd != core->file->fd) {
if (!rz_core_file_close_fd(core, desc->fd)) {
return false;
}
}
}
return true;
}
RZ_API bool rz_core_file_close_all_but(RzCore *core) {
rz_id_storage_foreach(core->io->files, close_but_cb, core);
return true;
}
RZ_API bool rz_core_file_close_fd(RzCore *core, int fd) {
RzCoreFile *file;
RzListIter *iter;
if (fd == -1) {
while (!rz_list_empty(core->files)) {
rz_core_file_close(rz_list_first_val(core->files));
}
return true;
}
rz_list_foreach (core->files, iter, file) {
if (file->fd == fd) {
rz_core_file_close(file);
return true;
}
}
return rz_io_fd_close(core->io, fd);
}
RZ_API RzCoreFile *rz_core_file_find_by_fd(RzCore *core, ut64 fd) {
RzListIter *iter;
RzCoreFile *cf = NULL;
rz_list_foreach (core->files, iter, cf) {
if (cf && cf->fd == fd) {
break;
}
cf = NULL;
}
return cf;
}
RZ_API RzCoreFile *rz_core_file_find_by_name(RzCore *core, const char *name) {
RzListIter *iter;
RzCoreFile *cf = NULL;
RzIODesc *desc;
if (!core) {
return NULL;
}
rz_list_foreach (core->files, iter, cf) {
desc = rz_io_desc_get(core->io, cf->fd);
if (desc && !strcmp(desc->name, name)) {
break;
}
cf = NULL;
}
return cf;
}
RZ_API int rz_core_file_set_by_fd(RzCore *core, ut64 fd) {
if (core) {
rz_io_use_fd(core->io, fd);
rz_core_bin_set_by_fd(core, fd);
return true;
}
return false;
}
RZ_API int rz_core_file_set_by_name(RzCore *core, const char *name) {
RzCoreFile *cf = rz_core_file_find_by_name(core, name);
return rz_core_file_set_by_file(core, cf);
}
RZ_API int rz_core_file_set_by_file(RzCore *core, RzCoreFile *cf) {
if (core && cf) {
if (!rz_core_file_set_by_fd(core, cf->fd)) {
return false;
}
core->file = cf;
return true;
}
return false;
}
RZ_API ut32 rz_core_file_cur_fd(RzCore *core) {
if (core && core->file) {
return core->file->fd;
}
return UT32_MAX;
}
RZ_API RzCoreFile *rz_core_file_cur(RzCore *r) {
// Add any locks here
return r->file;
}
/**
* \brief Open file use read-only Permission
* \param core RzCore instance
* \param fd File descriptor
*/
RZ_API void rz_core_io_file_open(RZ_NONNULL RzCore *core, int fd) {
rz_return_if_fail(core && fd >= 0);
if (!rz_config_get_b(core->config, "cfg.debug")) {
rz_io_reopen(core->io, fd, RZ_PERM_R, 644);
return;
}
RzBinFile *bf = rz_bin_cur(core->bin);
if (!(bf && rz_file_exists(bf->file))) {
RZ_LOG_WARN("Cannot open current RzBinFile.\n");
return;
}
// Escape spaces so that o's argv parse will detect the path properly
char *file = rz_str_path_escape(bf->file);
// Backup the baddr and sections that were already rebased to
// revert the rebase after the debug session is closed
ut64 orig_baddr = core->bin->cur->o->baddr_shift;
RzPVector *orig_sections = __save_old_sections(core);
rz_bin_file_delete_all(core->bin);
rz_io_close_all(core->io);
rz_config_set_b(core->config, "cfg.debug", false);
RzCoreFile *cfile = rz_core_file_open(core, file, RZ_PERM_R, 0);
if (!cfile) {
rz_pvector_free(orig_sections);
RZ_LOG_ERROR("Cannot open file '%s'\n", file);
return;
}
core->num->value = cfile->fd;
// If no baddr defined, use the one provided by the file
if (!rz_core_bin_load(core, file, UT64_MAX)) {
rz_pvector_free(orig_sections);
RZ_LOG_ERROR("Cannot load binary info of '%s'.\n", file);
return;
}
rz_core_block_read(core);
__rebase_everything(core, orig_sections, orig_baddr);
rz_pvector_free(orig_sections);
free(file);
}
/**
* \brief Reopen file
* \param core RzCore instance
* \param fd File descriptor
* \param perms Permission \s RZ_PERM_R
*/
RZ_API void rz_core_io_file_reopen(RZ_NONNULL RzCore *core, int fd, int perms) {
rz_return_if_fail(core && fd >= 0);
if (rz_io_reopen(core->io, fd, perms, 644)) {
void **it;
RzPVector *maps = rz_io_maps(core->io);
if (!maps) {
return;
}
rz_pvector_foreach_prev(maps, it) {
RzIOMap *map = *it;
if (map->fd == fd) {
map->perm |= RZ_PERM_WX;
}
}
}
}
static bool reopen_in_malloc_cb(void *user, void *data, ut32 id) {
RzIO *io = (RzIO *)user;
RzIODesc *desc = (RzIODesc *)data;
if (rz_io_desc_is_blockdevice(desc) || rz_io_desc_is_dbg(desc)) {
return true;
}
if (strstr(desc->uri, "://")) {
return true;
}
ut64 size = rz_io_desc_size(desc);
char *uri = rz_str_newf("malloc://%" PFMT64u, size);
if (!uri) {
return false;
}
ut8 *buf = malloc(size);
// if malloc fails, we can just abort the loop by returning false
if (!buf) {
free(uri);
return false;
}
RzIODesc *ndesc = rz_io_open_nomap(io, uri, RZ_PERM_RW, 0);
free(uri);
if (!ndesc) {
free(buf);
return false;
}
rz_io_desc_read_at(desc, 0LL, buf, (int)size); // that cast o_O
rz_io_desc_write_at(ndesc, 0LL, buf, (int)size);
free(buf);
rz_io_desc_exchange(io, desc->fd, ndesc->fd);
rz_io_desc_close(desc);
return true;
}
RZ_API void rz_core_file_reopen_in_malloc(RzCore *core) {
if (core && core->io && core->io->files) {
rz_id_storage_foreach(core->io->files, reopen_in_malloc_cb, core->io);
}
}
/* --------------------------------------------------------------------------------- */
RZ_IPI RzCoreIOMapInfo *rz_core_io_map_info_new(RzCoreFile *cf, int perm_orig) {
RzCoreIOMapInfo *info = RZ_NEW(RzCoreIOMapInfo);
if (!info) {
return NULL;
}
info->cf = cf;
info->perm_orig = perm_orig;
return info;
}
RZ_IPI void rz_core_io_map_info_free(RzCoreIOMapInfo *info) {
free(info);
}