dasm2dgml.js

// Support script for WinDbg disassembly test (with lines in private symbols)
// .shell -o d:\output.dgml -ci "uf ." cscript.exe /nologo dasm2dgml.js
// This is inspired from http://blogs.msdn.com/b/reiley/archive/2012/08/19/visualize-assembly-using-dgml.aspx
var EBB = [];

var hypertext=function(s){
  var r=[],L=s.length;
  for(var i=0;i<L;i++){
    var c=s.charAt(i);
    switch(c){
      case '"':r.push('&quot;');break;
      case '&':r.push('&amp;');break;
      case '<':r.push('&lt;');break;
      case '>':r.push('&gt;');break;
      default:r.push(c);}}
  return r.join('');
};

var map=function(f,v){
    var L=v.length,r=[];
    for(var i=0;i<L;i++)
        r.push(f(v[i]));
    return r;
};

(function(){
  var blk;

  var CExtendedBasicBlock = function(name, previous, next){
    this.Address = '';
    this.Code = [];
    this.Name = name;
    this.Previous = previous;
    this.Next = next;
  };

  while(true)
  {
    if(WScript.StdIn.AtEndOfStream)
      break;
    var origSourceLine = WScript.StdIn.ReadLine();

    var strSourceLine = origSourceLine.replace(/(^\s+)|(\s+$)/g, '');

    var prevEmptyLine;
    if(!strSourceLine || strSourceLine.match(/^Flow/) || strSourceLine.match(/^Debugger\./))
    {
        prevEmptyLine = true;
        continue;
    }

    if(strSourceLine.match(/.*:$/))
    {
        var blockName = strSourceLine.replace(/\[.+@\s+/, '').replace(/\]/, '');
        blk = new CExtendedBasicBlock(blockName);
        EBB.push(blk);

        prevEmptyLine = false;
    }
    else
    {
        // strip line number at first, assume line number starts with 2 spaces
        // source line # is reserved to be 5 spaces???
        if (origSourceLine.match(/^\s*\d{1,5}\s/))
        // if (!strSourceLine.match(/^\s*0/))
        {
            strSourceLine = strSourceLine.replace(/\s*\S+\s*/, '');
        }
        else
        {
            strSourceLine = strSourceLine.replace(/^\s*/, '');
        }

        if (prevEmptyLine)
        {
            var blockName = strSourceLine.match(/^\S+/)[0];
            blk = new CExtendedBasicBlock(blockName);
            EBB.push(blk);
        }
        prevEmptyLine = false;

        var headAddress = strSourceLine.match(/^\S+/)[0];
        blk.Address = blk.Address || headAddress;
        var filteredSourceLine = strSourceLine.replace(/^\S+\s+\S+\s+/, '');    // remove instr addr and encoding
        filteredSourceLine = filteredSourceLine.replace(/^call\s/, '>call'); // mark call inst
        var tokens = filteredSourceLine.match(/\S+/g);
        var instLen = tokens[0].length;
        if (instLen < 8 && tokens.length > 1) {
            var alignSpaces = new Array(9 - instLen).join(' ');
            filteredSourceLine = filteredSourceLine.replace(/^(\S+)\s+/, '\$1'+alignSpaces);
        }
        blk.Code.push(filteredSourceLine);
    }
  }
})();

EBB = EBB.sort(function(x, y){ return x.Address == y.Address ? 0 : x.Address > y.Address ? 1 : -1; });
for(var i = 1; i < EBB.length; i++)
{
  EBB[i].Previous = EBB[i - 1];
  EBB[i].Previous.Next = EBB[i];
}

WScript.Echo('<?xml version="1.0" encoding="utf-8"?>');
WScript.Echo('<DirectedGraph Background="#8ACA9A" GraphDirection="TopToBottom" xmlns="http://schemas.microsoft.com/vs/2009/dgml">');
WScript.Echo('  <Nodes>');
map(function(blk){
  var content = hypertext(blk.Name + ' (' + blk.Address + ')') + '&#xD;&#xA;';
  map(function(instruction){
    content += '&#xD;&#xA;' + hypertext(instruction);
  }, blk.Code);
  var setRetBlockColor = blk.Code[blk.Code.length - 1].match(/^ret/) ? "yellow" : ""; 
  WScript.Echo('    <Node Background="' + setRetBlockColor + '" Id="' + hypertext(blk.Name) + '" Label="' + content + '" />');
}, EBB);
WScript.Echo('  </Nodes>');
WScript.Echo('  <Links>');

map(function(blk){
  map(function(instruction){
    map(function(x){
      var idx = instruction.indexOf(x.Address);
      idx = idx >= 0 ? instruction.charAt(idx + x.Name.length) : -1;
      if(idx == '' || idx == ' ')
        WScript.Echo('    <Link Category="Jmp" Source="' + hypertext(blk.Name) + '" Target="' + hypertext(x.Name) + '" />');
    }, EBB);
  }, blk.Code);
  if(blk.Next && !(blk.Code[blk.Code.length - 1].match(/^[^\s]+/)[0] in {jmp: 0, ret: 0}))
    WScript.Echo('    <Link Category="FallThrough" Source="' + hypertext(blk.Name) + '" Target="' + hypertext(blk.Next.Name) + '" />');
}, EBB);

WScript.Echo('  </Links>');
WScript.Echo('  <Styles>');
WScript.Echo('    <Style TargetType="Node">');
WScript.Echo('      <Setter Property="FontFamily" Value="Consolas" />');
WScript.Echo('      <Setter Property="FontSize" Value="11" />');
// WScript.Echo('      <Setter Property="Background" Value="White" />');
WScript.Echo('      <Setter Property="NodeRadius" Value="2" />');
WScript.Echo('    </Style>');
WScript.Echo('    <Style TargetType="Link">');
WScript.Echo('        <Condition Expression="HasCategory(\'FallThrough\')" />');
WScript.Echo('        <Setter Property="Background" Value="Red" />');
WScript.Echo('        <Setter Property="Stroke" Value="Black" />');
WScript.Echo('    </Style>');
WScript.Echo('    <Style TargetType="Link">');
WScript.Echo('        <Condition Expression="HasCategory(\'Jmp\')" />');
WScript.Echo('        <Setter Property="Stroke" Value="Red" />');
WScript.Echo('        <Setter Property="StrokeDashArray" Value="3 3" />');
WScript.Echo('    </Style>');
WScript.Echo('  </Styles>');
WScript.Echo('</DirectedGraph>');