radare/radare2-webui

var r2ui = {};

r2ui.selected_panel = 'Disassembler';

r2ui.history = [];
r2ui.history_idx = 0;

r2ui.colors = {};

r2ui.load_colors = function() {
    // change css on the fly!
    r2.cmdj('ecj', function(x) {
        for (var i in x) {
            r2ui.colors['.ec_' + i.replace('gui.', 'gui_')] = 'rgb(' + String(x[i]) + ')';
        }
    });
    for (var k in document.styleSheets) {
        var mysheet = document.styleSheets[k];
        var myrules = mysheet.cssRules ? mysheet.cssRules : mysheet.rules;
        var targetrule;
        for (var j in myrules) {
            if (myrules[j].selectorText !== undefined && myrules[j].selectorText !== null) {
                if (myrules[j].selectorText.toLowerCase().indexOf('.ec_') === 0) {
                    var sel = myrules[j].selectorText; //.toLowerCase().replace("gui.","gui_");
                    var color = r2ui.colors[sel];
                    if (color !== undefined && color !== null) {
                        if (sel == '.ec_gui_background' || sel == '.ec_gui_alt_background') {
                            myrules[j].style.backgroundColor = color;
                        } else if (sel == '.ec_border') myrules[j].style.borderColor = color;
                        else myrules[j].style.color = color;
                    } else {
                        if (sel == '.ec_gui_background' || sel == '.ec_gui_alt_background') r2ui.colors[sel] = myrules[j].style.backgroundColor;
                        else if (sel == '.ec_gui_border') r2ui.colors[sel] = myrules[j].style.borderColor;
                        else r2ui.colors[sel] = myrules[j].style.color;
                    }
                }
            }
        }
    }
};

r2ui.basic_blocks = {};
r2ui.use_sdb = false;
r2ui.get_fcn_BB = function(fcn_offset, bb_offset) {
    if (r2ui.use_sdb) {
        var path = 'webui/graph/' + fcn_offset + '/' + bb_offset;
        var bb = undefined;
        r2.cmd('k ' + path, function(x) {
            var lines = decodeURIComponent(x).split('\n');
            for (var l in lines) {
                var line = lines[l];
                if (line !== '') {
                    bb = {};
                    var props = line.split(',');
                    bb.x = props[0];
                    bb.y = props[1];
                    if (props[2] == '' || props[2] === undefined) props[2] = 'transparent';
                    bb.color = props[2].replace(/\*\*/g, ',');
                }
            }
        });
        return bb;
    } else {
        return r2ui.basic_blocks[bb_offset];
    }
};
r2ui.get_fcn_BBs = function(fcn_offset) {
    if (r2ui.use_sdb) {
        var path = 'webui/graph/' + fcn_offset + '/*';
        var BBs = {};
        r2.cmd('k ' + path, function(x) {
            var lines = decodeURIComponent(x).split('\n');
            for (var l in lines) {
                var line = lines[l];
                if (line !== '') {
                    offset = line.split('=')[0];
                    line = line.split('=')[1];
                    var bb = {};
                    var props = line.split(',');
                    bb.x = props[0];
                    bb.y = props[1];
                    if (props[2] == '' || props[2] === undefined) props[2] = 'transparent';
                    bb.color = props[2].replace(/\*\*/g, ',');
                    BBs[offset] = bb;
                }
            }
        });
        return BBs;
    } else {
        return r2ui.basic_blocks;
    }
};
r2ui.update_fcn_BB = function(fcn_offset, bb_offset, bbinfo) {
    if (r2ui.use_sdb) {
        var path = 'webui/graph/' + fcn_offset + '/' + bb_offset;
        if (bbinfo.color === undefined) bbinfo.color = 'transparent';
        var value = bbinfo.x + ',' + bbinfo.y + ',' + bbinfo.color.replace(/,/g, '**');
        r2.cmd('k ' + path + '=' + encodeURIComponent(value), function() { });
    } else {
        r2ui.basic_blocks[bb_offset] = bbinfo;
    }
};

r2ui.current_fcn_offset = null;
r2ui.graph = null;
r2ui.console_lang = 'r2';
r2ui.toggle_console_lang = function() {
    if (r2ui.console_lang == 'r2') r2ui.console_lang = 'js';
    else if (r2ui.console_lang == 'js') r2ui.console_lang = 'r2';
    $('#cmd_input > label').html(r2ui.console_lang);
};

r2ui.history_push = function(x) {
    if (x != r2ui.history_last()) {
        if (r2ui.history_idx != r2ui.history.length)
            r2ui.history = r2ui.history.splice(0, r2ui.history_idx);
        r2ui.history_idx++;
        r2ui.history.push(x);
    }
};

r2ui.history_pop = function() {
    // console.log("history pop");
    if (r2ui.history_idx == r2ui.history.length) r2ui.history_idx--;
    // console.log(r2ui.history_idx + "/" + r2ui.history.length);
    return r2ui.history.pop();
};

r2ui.history_last = function() {
    if (r2ui.history.length > 0) {
        return r2ui.history[r2ui.history_idx - 1];
    }
};

r2ui.history_prev = function() {
    // console.log("history prev");
    if (r2ui.history_idx > 1) r2ui.history_idx--;
    // console.log(r2ui.history_idx + "/" + r2ui.history.length);
    return r2ui.history[r2ui.history_idx - 1];
};

r2ui.history_next = function() {
    // console.log("history next");
    var ret = r2ui.history[r2ui.history_idx];
    if (r2ui.history_idx < r2ui.history.length) r2ui.history_idx++;
    // console.log(r2ui.history_idx + "/" + r2ui.history.length);
    return ret;
};

r2ui.next_instruction = function() {
    var offset = parseInt(r2ui.history_last(), 16);
    r2.cmd('pdl 1', function(x) {
        offset += parseInt(x.trim());
    });
    return '0x' + offset.toString(16);
};

r2ui.prev_instruction = function() {
    var offset = parseInt(r2ui.history_last(), 16);
    r2.cmdj('pdfj', function(x) {
        if (x !== undefined && x !== null) {
            for (var i in x.ops) {
                if (i === 0) continue;
                var opcode = x.ops[i];
                if (opcode.offset == offset) {
                    offset =  x.ops[i - 1].offset;
                    break;
                }
            }
        }
    });
    return '0x' + offset.toString(16);;
};

r2ui.seek = function(addr, push, scroll) {
    if (addr === undefined) return;
    // Resolve flag in case we dont have an address
    if (addr.indexOf('0x') === 0) {
        addr = address_canonicalize(addr);
    } else {
        var a = r2.get_flag_address(addr);
        if (a !== null) {
            addr = address_canonicalize(a);
        } else {
            r2.cmd('s ' + addr + ';s|', function(x) {
                addr = address_canonicalize(x.replace('\n', ''));
            });
        }
    }

    if (push) r2ui.history_push(addr);

    // if (r2ui.ra.getIndex ()==2) r2ui.ra.setIndex (1);

    r2.cmd('s ' + addr, function() {
        r2ui._dis.seek(addr, scroll);
        r2ui._hex.seek(addr, scroll);
    });
    // if (r2ui._dis.tmp_address !== undefined) r2ui._dis.tmp_address = addr;
};

r2ui.seek_in_graph = function(addr, push) {
    if (push) r2ui.history_push(addr);

    // if (r2ui.ra.getIndex ()==2) r2ui.ra.setIndex (1);

    r2.cmd('s ' + addr, function() {
        rehighlight_iaddress(addr);
        r2ui._hex.seek(addr);
        r2ui._hex.scrollTo(0, 0);
    });
};

r2ui.seek_prev = function() {
    // XXX. this is only for disasm
    var addr = r2ui.history.pop();
    r2.cmd('s ' + addr, function() {
        r2ui._dis.seek(addr);
        r2ui._dis.scrollTo(0, 0);
        r2ui._hex.seek(addr);
        r2ui._hex.scrollTo(0, 0);
    });
};

/* used from mainpanel */
r2ui.openpage = function(addr, idx) {
    if (idx === undefined) {
        idx = addr;
        addr = undefined;
    }
    if (addr !== undefined) {
        r2ui.seek(addr, true);
    }

    // What is this for?
    if (r2ui.ra.getIndex() == 2) r2ui.ra.setIndex(1);

    r2ui.mp.openPage(idx);

};

r2ui.opendis = function(addr) {
    r2ui.openpage(addr, 0);
};

r2ui.openhex = function(addr) {
    r2ui.openpage(addr, 2);
};