radare2/libr/core/project.c
BlueC0re 40aa48681d Fix vmarks and save them in the project
removing deletion of marks after dump and using UT64_MAX as an unset flag throughout the code.
2016-08-26 21:35:10 +02:00

533 lines
12 KiB
C

/* radare - LGPL - Copyright 2010-2016 - pancake, maijin */
#include <r_types.h>
#include <r_list.h>
#include <r_flags.h>
#include <r_core.h>
static int is_valid_project_name (const char *name) {
int i;
for (i=0; name[i]; i++) {
switch (name[i]) {
case '\\': // for w32
case '.':
case '_':
case ':':
continue;
}
if (name[i] >= 'a' && name[i] <= 'z')
continue;
if (name[i] >= 'A' && name[i] <= 'Z')
continue;
if (name[i] >= '0' && name[i] <= '9')
continue;
return 0;
}
return 1;
}
static char *r_core_project_file(RCore *core, const char *file) {
const char *magic = "# r2 rdb project file";
char *data, *prjfile;
if (r_file_is_abspath (file)) {
prjfile = strdup (file);
} else {
if (!is_valid_project_name (file)) {
return NULL;
}
prjfile = r_file_abspath (r_config_get (
core->config, "dir.projects"));
prjfile = r_str_concat (prjfile, R_SYS_DIR);
prjfile = r_str_concat (prjfile, file);
}
data = r_file_slurp (prjfile, NULL);
if (data) {
if (strncmp (data, magic, strlen (magic))) {
R_FREE (prjfile);
}
}
free (data);
return prjfile;
}
static int r_core_project_init(RCore *core) {
char *prjdir = r_file_abspath (r_config_get (
core->config, "dir.projects"));
int ret = r_sys_mkdirp (prjdir);
if (!ret) eprintf ("Cannot mkdir dir.projects\n");
free (prjdir);
return ret;
}
static int r_core_is_project(RCore *core, const char *name) {
int ret = 0;
if (name && *name && *name!='.') {
char *path = r_core_project_file (core, name);
if (!path)
return 0;
path = r_str_concat (path, ".d");
if (r_file_is_directory (path))
ret = 1;
free (path);
}
return ret;
}
R_API int r_core_project_cat(RCore *core, const char *name) {
char *path = r_core_project_file (core, name);
if (path) {
char *data = r_file_slurp (path, NULL);
if (data) {
r_cons_println (data);
free (data);
}
}
free (path);
return 0;
}
R_API int r_core_project_list(RCore *core, int mode) {
RListIter *iter;
RList *list;
int isfirst = 1;
char *foo, *path = r_file_abspath (r_config_get (core->config, "dir.projects"));
if (!path)
return 0;
list = r_sys_dir (path);
switch (mode) {
case 'j':
r_cons_printf ("[");
r_list_foreach (list, iter, foo) {
// todo. escape string
if (r_core_is_project (core, foo)) {
r_cons_printf ("%s\"%s\"",
isfirst?"":",", foo);
isfirst = 0;
}
}
r_cons_printf ("]\n");
break;
default:
r_list_foreach (list, iter, foo) {
if (r_core_is_project (core, foo))
r_cons_println (foo);
}
break;
}
r_list_free (list);
free (path);
return 0;
}
R_API int r_core_project_delete(RCore *core, const char *prjfile) {
char *path;
if (r_sandbox_enable (0)) {
eprintf ("Cannot delete project in sandbox mode\n");
return 0;
}
path = r_core_project_file (core, prjfile);
if (!path) {
eprintf ("Invalid project name '%s'\n", prjfile);
return false;
}
if (r_core_is_project (core, prjfile)) {
// rm project file
r_file_rm (path);
eprintf ("rm %s\n", path);
path = r_str_concat (path, ".d");
if (r_file_is_directory (path)) {
char *f;
RListIter *iter;
RList *files = r_sys_dir (path);
r_list_foreach (files, iter, f) {
char *filepath = r_str_concat (strdup (path), R_SYS_DIR);
filepath =r_str_concat (filepath, f);
if (!r_file_is_directory (filepath)) {
eprintf ("rm %s\n", filepath);
r_file_rm (filepath);
}
free (filepath);
}
r_file_rm (path);
eprintf ("rm %s\n", path);
r_list_free (files);
}
// TODO: remove .d directory (BEWARE OF ROOT RIMRAFS!)
// TODO: r_file_rmrf (path);
}
free (path);
return 0;
}
static bool r_core_rop_load(RCore *core, const char *prjfile) {
char *path, *db;
bool found = 0;
SdbListIter *it;
SdbNs *ns;
const char *prjdir = r_config_get (core->config, "dir.projects");
Sdb *rop_db = sdb_ns (core->sdb, "rop", false);
if (!prjfile || !*prjfile) {
return false;
}
if (*prjfile == '/') {
db = r_str_newf ("%s.d", prjfile);
if (!db) return false;
path = strdup (db);
} else {
db = r_str_newf ("%s/%s.d", prjdir, prjfile);
if (!db) return false;
path = r_file_abspath (db);
}
if (!path) {
free (db);
return false;
}
if (rop_db) {
ls_foreach (core->sdb->ns, it, ns){
if (ns->sdb == rop_db) {
ls_delete (core->sdb->ns, it);
found = true;
break;
}
}
}
if (!found) {
sdb_free (rop_db);
}
rop_db = sdb_new (path, "rop", 0);
if (!rop_db) {
free (db);
free (path);
return false;
}
sdb_ns_set (core->sdb, "rop", rop_db);
free (path);
free (db);
return true;
}
R_API void r_core_project_load(RCore *core, const char *prjfile) {
(void)r_core_rop_load (core, prjfile);
(void)r_core_project_load_xrefs (core, prjfile);
}
R_API int r_core_project_open(RCore *core, const char *prjfile) {
int askuser = 1;
int ret, close_current_session = 1;
char *prj, *filepath;
if (!prjfile || !*prjfile) {
return false;
}
const bool cfg_fortunes = r_config_get_i (core->config, "cfg.fortunes");
const bool scr_interactive = r_config_get_i (core->config, "scr.interactive");
const bool scr_prompt = r_config_get_i (core->config, "scr.prompt");
prj = r_core_project_file (core, prjfile);
if (!prj) {
eprintf ("Invalid project name '%s'\n", prjfile);
return false;
}
filepath = r_core_project_info (core, prj);
//eprintf ("OPENING (%s) from %s\n", prj, r_config_get (core->config, "file.path"));
/* if it is not an URI */
if (!filepath) {
eprintf ("Cannot retrieve information for project '%s'\n", prj);
free (prj);
return false;
}
if (!strstr (filepath, "://")) {
/* check if path exists */
if (!r_file_exists (filepath)) {
eprintf ("Cannot find file '%s'\n", filepath);
free (prj);
free (filepath);
return false;
}
}
if (!strcmp (prjfile, r_config_get (core->config, "file.project"))) {
//eprintf ("Reloading project\n");
askuser = 0;
#if 0
free (prj);
free (filepath);
return false;
#endif
}
if (askuser) {
if (r_config_get_i (core->config, "scr.interactive")) {
close_current_session = r_cons_yesno ('y', "Close current session? (Y/n)");
}
}
if (close_current_session) {
RCoreFile *fh;
// delete
r_core_file_close_fd (core, -1);
r_io_close_all (core->io);
r_anal_purge (core->anal);
r_flag_unset_all (core->flags);
r_bin_file_delete_all (core->bin);
// open new file
// TODO: handle read/read-write mode
// TODO: handle mapaddr (io.maps are not saved in projects yet)
fh = r_core_file_open (core, filepath, 0, 0);
if (!fh) {
eprintf ("Cannot open file '%s'\n", filepath);
free (filepath);
free (prj);
return false;
}
// TODO: handle load bin info or not
// TODO: handle base address
r_core_bin_load (core, filepath, UT64_MAX);
}
/* load sdb stuff in here */
r_core_project_load (core, prjfile);
ret = r_core_cmd_file (core, prj);
r_config_set_i (core->config, "cfg.fortunes", cfg_fortunes);
r_config_set_i (core->config, "scr.interactive", scr_interactive);
r_config_set_i (core->config, "scr.prompt", scr_prompt);
r_config_bump (core->config, "asm.arch");
free (filepath);
free (prj);
return ret;
}
R_API char *r_core_project_info(RCore *core, const char *prjfile) {
FILE *fd;
char buf[256], *file = NULL;
char *prj = r_core_project_file (core, prjfile);
if (!prj) {
eprintf ("Invalid project name '%s'\n", prjfile);
return NULL;
}
fd = r_sandbox_fopen (prj, "r");
if (fd) {
for (;;) {
fgets (buf, sizeof (buf), fd);
if (feof (fd))
break;
if (!strncmp (buf, "\"e file.path = ", 15)) {
buf[strlen (buf) - 2] = 0;
file = r_str_new (buf+15);
break;
}
// TODO: deprecate before 1.0
if (!strncmp (buf, "e file.path = ", 14)) {
buf[strlen (buf)-1] = 0;
file = r_str_new (buf+14);
break;
}
}
fclose (fd);
} else {
eprintf ("Cannot open project info (%s)\n", prj);
}
#if 0
if (file) {
r_cons_printf ("Project: %s\n", prj);
r_cons_printf ("FilePath: %s\n", file);
}
#endif
free (prj);
return file;
}
R_API bool r_core_project_save_rdb(RCore *core, const char *file, int opts) {
char *filename, *hl, *ohl = NULL;
int fd, fdold, tmp;
if (file == NULL || *file == '\0')
return false;
filename = r_str_word_get_first (file);
fd = r_sandbox_open (file, O_BINARY|O_RDWR|O_CREAT|O_TRUNC, 0644);
if (fd == -1) {
free (filename);
return false;
}
hl = r_cons_singleton ()->highlight;
if (hl) {
ohl = strdup (hl);
r_cons_highlight (NULL);
}
fdold = r_cons_singleton ()->fdout;
r_cons_singleton ()->fdout = fd;
r_cons_singleton ()->is_interactive = false;
r_str_write (fd, "# r2 rdb project file\n");
if (opts & R_CORE_PRJ_FLAGS) {
r_str_write (fd, "# flags\n");
tmp = core->flags->space_idx;
core->flags->space_idx = -1;
r_flag_list (core->flags, true, NULL);
core->flags->space_idx = tmp;
r_cons_flush ();
}
if (opts & R_CORE_PRJ_EVAL) {
r_str_write (fd, "# eval\n");
r_config_list (core->config, NULL, true);
r_cons_flush ();
}
if (opts & R_CORE_PRJ_IO_MAPS) {
r_core_cmd (core, "om*", 0);
r_cons_flush ();
}
if (opts & R_CORE_PRJ_SECTIONS) {
r_str_write (fd, "# sections\n");
r_io_section_list (core->io, core->offset, 1);
r_cons_flush ();
}
if (opts & R_CORE_PRJ_META) {
r_str_write (fd, "# meta\n");
r_meta_list (core->anal, R_META_TYPE_ANY, 1);
r_cons_flush ();
r_core_cmd (core, "fV*", 0);
r_cons_flush ();
}
if (opts & R_CORE_PRJ_XREFS) {
r_core_cmd (core, "ax*", 0);
r_cons_flush ();
}
if (opts & R_CORE_PRJ_FCNS) {
r_core_cmd (core, "afl*", 0);
r_cons_flush ();
}
if (opts & R_CORE_PRJ_FLAGS) {
r_core_cmd (core, "f.**", 0);
r_cons_flush ();
}
if (opts & R_CORE_PRJ_DBG_BREAK) {
r_core_cmd (core, "db*", 0);
r_cons_flush ();
}
if (opts & R_CORE_PRJ_ANAL_HINTS) {
r_core_cmd (core, "ah*", 0);
r_cons_flush ();
}
if (opts & R_CORE_PRJ_ANAL_TYPES) {
r_str_write (fd, "# types\n");
r_core_cmd (core, "t*", 0);
r_cons_flush ();
}
if (opts & R_CORE_PRJ_ANAL_MACROS) {
r_str_write (fd, "# macros\n");
r_core_cmd (core, "(*", 0);
r_cons_flush ();
}
if (opts & R_CORE_PRJ_ANAL_SEEK) {
r_cons_printf ("# seek\n"
"s 0x%08"PFMT64x"\n", core->offset);
r_cons_flush ();
}
r_cons_singleton ()->fdout = fdold;
r_cons_singleton ()->is_interactive = true;
if (ohl) {
r_cons_highlight (ohl);
free (ohl);
}
close (fd);
free (filename);
return true;
}
R_API bool r_core_project_save(RCore *core, const char *file) {
bool ret = true;
char *prj, buf[1024];
if (!file || !*file) {
return false;
}
prj = r_core_project_file (core, file);
if (!prj) {
eprintf ("Invalid project name '%s'\n", file);
return false;
}
if (r_file_is_directory (prj)) {
eprintf ("Error: Target is a directory\n");
free (prj);
return false;
}
r_core_project_init (core);
snprintf (buf, sizeof (buf), "%s.d" R_SYS_DIR "xrefs", prj);
r_anal_project_save (core->anal, buf);
Sdb *rop_db = sdb_ns (core->sdb, "rop", false);
if (rop_db) {
snprintf (buf, sizeof (buf), "%s.d" R_SYS_DIR "rop", prj);
sdb_file (rop_db, buf);
sdb_sync (rop_db);
}
if (!r_core_project_save_rdb (core, prj, R_CORE_PRJ_ALL^R_CORE_PRJ_XREFS)) {
eprintf ("Cannot open '%s' for writing\n", prj);
ret = false;
}
free (prj);
return ret;
}
R_API char *r_core_project_notes_file (RCore *core, const char *file) {
char *notes_txt;
const char *prjdir = r_config_get (core->config, "dir.projects");
char *prjpath = r_file_abspath (prjdir);
notes_txt = r_str_newf ("%s"R_SYS_DIR"%s.d"R_SYS_DIR"notes.txt", prjpath, file);
free (prjpath);
return notes_txt;
}
#define DB core->anal->sdb_xrefs
R_API bool r_core_project_load_xrefs(RCore *core, const char *prjfile) {
char *path, *db;
const char *prjdir = r_config_get (core->config, "dir.projects");
if (!prjfile || !*prjfile) {
return false;
}
if (prjfile[0] == '/') {
db = r_str_newf ("%s.d", prjfile);
if (!db) return false;
path = strdup (db);
} else {
db = r_str_newf ("%s/%s.d", prjdir, prjfile);
if (!db) return false;
path = r_file_abspath (db);
}
if (!path) {
free (db);
return false;
}
if (!sdb_ns_unset (core->anal->sdb, NULL, DB)) {
sdb_free (DB);
}
DB = sdb_new (path, "xrefs", 0);
if (!DB) {
free (db);
free (path);
return false;
}
sdb_ns_set (core->anal->sdb, "xrefs", DB);
free (path);
free (db);
return true;
}