radare2/libr/io/undo.c

298 lines
6.4 KiB
C
Raw Normal View History

2018-01-10 06:43:23 +00:00
/* radare - LGPL - Copyright 2007-2018 - pancake */
#include <r_io.h>
#if 0
* TODO:
* - make path of indirections shortr (io->undo.foo is slow) */
* - Plugin changes in write and seeks
* - Per-fd history log
#endif
R_API int r_io_undo_init(RIO *io) {
/* seek undo */
r_io_sundo_reset (io);
/* write undo */
io->undo.w_init = 0;
io->undo.w_enable = 0;
io->undo.w_enable = 0;
io->undo.w_list = r_list_new ();
2015-09-14 00:08:31 +00:00
return true;
}
R_API void r_io_undo_enable(RIO *io, int s, int w) {
io->undo.s_enable = s;
io->undo.w_enable = w;
}
/* undo seekz */
R_API RIOUndos *r_io_sundo(RIO *io, ut64 offset) {
RIOUndos *undo;
RIOSection *sec;
2018-01-10 06:43:23 +00:00
if (!io->undo.s_enable || !io->undo.undos) {
return NULL;
2018-01-10 06:43:23 +00:00
}
/* No redos yet, store the current seek so we can redo to it. */
if (!io->undo.redos) {
undo = &io->undo.seek[io->undo.idx];
undo->off = offset;
undo->cursor = 0;
}
io->undo.idx = (io->undo.idx - 1 + R_IO_UNDOS) % R_IO_UNDOS;
io->undo.undos--;
io->undo.redos++;
undo = &io->undo.seek[io->undo.idx];
sec = r_io_section_vget (io, undo->off);
if (!sec || (sec->paddr == sec->vaddr)) {
io->off = undo->off;
} else {
io->off = undo->off - sec->vaddr + sec->paddr;
}
return undo;
}
R_API RIOUndos *r_io_sundo_redo(RIO *io) {
RIOUndos *undo;
RIOSection *sec;
2018-01-10 06:43:23 +00:00
if (!io->undo.s_enable || !io->undo.redos) {
return NULL;
2018-01-10 06:43:23 +00:00
}
io->undo.idx = (io->undo.idx + 1) % R_IO_UNDOS;
io->undo.undos++;
io->undo.redos--;
undo = &io->undo.seek[io->undo.idx];
sec = r_io_section_vget (io, undo->off);
if (!sec || (sec->paddr == sec->vaddr)) {
io->off = undo->off;
} else {
io->off = undo->off - sec->vaddr + sec->paddr;
}
return undo;
}
R_API void r_io_sundo_push(RIO *io, ut64 off, int cursor) {
2018-01-10 06:43:23 +00:00
if (!io->undo.s_enable) {
return;
}
// the first insert
if (io->undo.idx > 0) {
2018-01-10 06:43:23 +00:00
RIOUndos *undo = &io->undo.seek[io->undo.idx - 1];
if (undo->off == off && undo->cursor == cursor) {
return;
}
}
undo = &io->undo.seek[io->undo.idx];
undo->off = off;
undo->cursor = cursor;
io->undo.idx = (io->undo.idx + 1) % R_IO_UNDOS;
/* Only R_IO_UNDOS - 1 undos can be used because r_io_sundo_undo () must
* push the current position for redo as well, which takes one entry in
* the table. */
if (io->undo.undos < R_IO_UNDOS - 1)
io->undo.undos++;
/* We only have linear undo/redo, no tree. So after this new possible
* undo, all redos are lost. */
io->undo.redos = 0;
}
R_API void r_io_sundo_reset(RIO *io) {
io->undo.idx = 0;
io->undo.undos = 0;
io->undo.redos = 0;
}
R_API RList *r_io_sundo_list(RIO *io, int mode) {
int idx, undos, redos, i, j, start, end;
2017-03-27 11:42:52 +00:00
RList* list = NULL;
if (mode == '!') {
mode = 0;
}
2016-11-24 15:11:43 +00:00
if (!io->undo.s_enable) {
return NULL;
2016-11-24 15:11:43 +00:00
}
undos = io->undo.undos;
redos = io->undo.redos;
idx = io->undo.idx;
start = (idx - undos + R_IO_UNDOS) % R_IO_UNDOS;
end = (idx + redos + 1) % R_IO_UNDOS;
j = 0;
2016-05-09 08:39:19 +00:00
switch (mode) {
case 'j':
io->cb_printf ("[");
break;
case 0:
list = r_list_newf (free);
break;
}
2016-11-24 15:11:43 +00:00
const char *comma = "";
2016-05-09 08:39:19 +00:00
for (i = start; i < end || j == 0; i = (i + 1) % R_IO_UNDOS) {
2018-01-10 06:43:23 +00:00
int idx = (j < undos)? undos - j - 1: j - undos - 1;
RIOUndos *undo = &io->undo.seek[i];
ut64 addr = undo->off;
ut64 notLast = (j + 1 < undos) && (i != end - 1);
2016-05-09 08:39:19 +00:00
switch (mode) {
case '=':
if (j < undos) {
io->cb_printf ("0x%"PFMT64x"%s", addr, notLast? " > ": "");
}
break;
case 'j':
if (j < undos) {
io->cb_printf ("%"PFMT64d"%s", addr, notLast? ",": "");
2016-11-24 15:11:43 +00:00
comma = ",";
2016-05-09 08:39:19 +00:00
}
break;
case '*':
if (j < undos) {
io->cb_printf ("f undo_%d @ 0x%"PFMT64x"\n", idx, addr);
} else if (j == undos && j != 0 && redos != 0) {
io->cb_printf ("# Current undo/redo position.\n");
} else if (j != undos) {
io->cb_printf ("f redo_%d @ 0x%"PFMT64x"\n", idx, addr);
}
2017-03-27 11:42:52 +00:00
break;
case 0:
if (list) {
RIOUndos *u = R_NEW0 (RIOUndos);
if (u) {
memcpy (u, undo, sizeof (RIOUndos));
r_list_append (list, u);
}
}
break;
}
j++;
}
2016-05-09 08:39:19 +00:00
switch (mode) {
case '=':
io->cb_printf ("\n");
break;
case 'j':
2016-11-24 15:11:43 +00:00
io->cb_printf ("%s%"PFMT64d"]\n", comma, io->off);
2016-05-09 08:39:19 +00:00
break;
}
return list;
}
/* undo writez */
R_API void r_io_wundo_new(RIO *io, ut64 off, const ut8 *data, int len) {
RIOUndoWrite *uw;
if (!io->undo.w_enable)
return;
/* undo write changes */
2016-05-24 20:22:15 +00:00
uw = R_NEW0 (RIOUndoWrite);
2018-01-10 06:43:23 +00:00
if (!uw) {
return;
}
2015-09-14 00:08:31 +00:00
uw->set = true;
uw->off = off;
uw->len = len;
uw->n = (ut8*) malloc (len);
2016-05-24 20:22:15 +00:00
if (!uw->n) {
2018-01-10 06:43:23 +00:00
free (uw);
2016-05-24 20:22:15 +00:00
return;
}
2018-01-10 06:43:23 +00:00
memcpy (uw->n, data, len);
uw->o = (ut8*) malloc (len);
2016-05-24 20:22:15 +00:00
if (!uw->o) {
R_FREE (uw);
return;
}
2018-01-10 06:43:23 +00:00
memset (uw->o, 0xff, len);
r_io_read_at (io, off, uw->o, len);
r_list_append (io->undo.w_list, uw);
}
R_API void r_io_wundo_clear(RIO *io) {
// XXX memory leak
io->undo.w_list = r_list_new ();
}
// rename to r_io_undo_length ?
R_API int r_io_wundo_size(RIO *io) {
2017-08-19 17:17:09 +00:00
return r_list_length (io->undo.w_list);
}
// TODO: Deprecate or so? iterators must be language-wide, but helpers are useful
R_API void r_io_wundo_list(RIO *io) {
#define BW 8 /* byte wrap */
RListIter *iter;
RIOUndoWrite *u;
int i = 0, j, len;
if (io->undo.w_init)
r_list_foreach (io->undo.w_list, iter, u) {
io->cb_printf ("%02d %c %d %08"PFMT64x": ", i, u->set?'+':'-', u->len, u->off);
len = (u->len>BW)?BW:u->len;
for (j=0;j<len;j++) io->cb_printf ("%02x ", u->o[j]);
if (len == BW) io->cb_printf (".. ");
io->cb_printf ("=> ");
for (j=0;j<len;j++) io->cb_printf ("%02x ", u->n[j]);
if (len == BW) io->cb_printf (".. ");
io->cb_printf ("\n");
i++;
}
}
2018-01-10 06:43:23 +00:00
R_API int r_io_wundo_apply(RIO *io, RIOUndoWrite *u, int set) {
int orig = io->undo.w_enable;
io->undo.w_enable = 0;
if (set) {
r_io_write_at (io, u->off, u->n, u->len);
2015-09-14 00:08:31 +00:00
u->set = true;
} else {
r_io_write_at (io, u->off, u->o, u->len);
2015-09-14 00:08:31 +00:00
u->set = false;
}
io->undo.w_enable = orig;
return 0;
}
R_API void r_io_wundo_apply_all(RIO *io, int set) {
RListIter *iter;
RIOUndoWrite *u;
r_list_foreach_prev (io->undo.w_list, iter, u) {
r_io_wundo_apply (io, u, set); //UNDO_WRITE_UNSET);
eprintf ("%s 0x%08"PFMT64x"\n", set?"redo":"undo", u->off);
}
}
/* sets or unsets the writes done */
/* if ( set == 0 ) unset(n) */
R_API int r_io_wundo_set(RIO *io, int n, int set) {
RListIter *iter;
RIOUndoWrite *u = NULL;
int i = 0;
if (io->undo.w_init) {
2018-01-10 06:43:23 +00:00
r_list_foreach_prev (io->undo.w_list, iter, u) {
if (i++ == n) {
break;
2018-01-10 06:43:23 +00:00
}
}
if (u) { // wtf?
r_io_wundo_apply (io, u, set);
2015-09-14 00:08:31 +00:00
return true;
}
eprintf ("invalid undo-write index\n");
2018-01-10 06:43:23 +00:00
} else {
eprintf ("no writes done\n");
}
2015-09-14 00:08:31 +00:00
return false;
}