radare2/shlr/www/enyo/js/r2ui.js

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(x) { });
  } 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);
}