public static string Disassemble(this UndertaleCode code, IList <UndertaleVariable> vars, UndertaleCodeLocals locals)
{
StringBuilder sb = new StringBuilder();
if (locals == null && !code.WeirdLocalFlag)
{
sb.Append("; WARNING: Missing code locals, possibly due to unsupported bytecode version or a brand new code entry.\n");
}
else
{
sb.Append(code.GenerateLocalVarDefinitions(vars, locals));
}
Dictionary <uint, string> fragments = new Dictionary <uint, string>();
foreach (var dup in code.Duplicates)
{
fragments.Add(dup.Offset / 4, (dup.Name?.Content ?? "<null>") + $" (locals={dup.LocalsCount}, argc={dup.ArgumentsCount})");
}
List <uint> blocks = FindBlockAddresses(code);
foreach (var inst in code.Instructions)
{
bool doNewline = true;
if (fragments.TryGetValue(inst.Address, out string entry))
{
sb.AppendLine();
sb.AppendLine($"> {entry}");
doNewline = false;
}
int ind = blocks.IndexOf(inst.Address);
if (ind != -1)
{
if (doNewline)
{
sb.AppendLine();
}
sb.AppendLine($":[{ind}]");
}
sb.AppendLine(inst.ToString(code, blocks));
}
sb.AppendLine();
sb.Append(":[end]");
return(sb.ToString());
}
private void DisassembleCode(UndertaleCode code)
{
code.UpdateAddresses();
FlowDocument document = new FlowDocument();
document.PagePadding = new Thickness(0);
document.PageWidth = 2048; // Speed-up.
document.FontFamily = new FontFamily("Lucida Console");
Paragraph par = new Paragraph();
par.Margin = new Thickness(0);
if (code.Instructions.Count > 5000)
{
// Disable syntax highlighting. Loading it can take a few MINUTES on large scripts.
var data = (Application.Current.MainWindow as MainWindow).Data;
string[] split = code.Disassemble(data.Variables, data.CodeLocals.For(code)).Split('\n');
for (var i = 0; i < split.Length; i++)
{ // Makes it possible to select text.
if (i > 0 && (i % 100) == 0)
{
document.Blocks.Add(par);
par = new Paragraph();
par.Margin = new Thickness(0);
}
par.Inlines.Add(split[i] + (split.Length > i + 1 && ((i + 1) % 100) != 0 ? "\n" : ""));
}
}
else
{
Brush addressBrush = new SolidColorBrush(Color.FromRgb(50, 50, 50));
Brush opcodeBrush = new SolidColorBrush(Color.FromRgb(0, 100, 0));
Brush argBrush = new SolidColorBrush(Color.FromRgb(0, 0, 150));
Brush typeBrush = new SolidColorBrush(Color.FromRgb(0, 0, 50));
var data = (Application.Current.MainWindow as MainWindow).Data;
par.Inlines.Add(new Run(code.GenerateLocalVarDefinitions(data.Variables, data.CodeLocals.For(code)))
{
Foreground = addressBrush
});
foreach (var instr in code.Instructions)
{
par.Inlines.Add(new Run(instr.Address.ToString("D5") + ": ")
{
Foreground = addressBrush
});
par.Inlines.Add(new Run(instr.Kind.ToString().ToLower())
{
Foreground = opcodeBrush, FontWeight = FontWeights.Bold
});
switch (UndertaleInstruction.GetInstructionType(instr.Kind))
{
case UndertaleInstruction.InstructionType.SingleTypeInstruction:
par.Inlines.Add(new Run("." + instr.Type1.ToOpcodeParam())
{
Foreground = typeBrush
});
if (instr.Kind == UndertaleInstruction.Opcode.Dup || instr.Kind == UndertaleInstruction.Opcode.CallV)
{
par.Inlines.Add(new Run(" "));
par.Inlines.Add(new Run(instr.Extra.ToString())
{
Foreground = argBrush
});
}
break;
case UndertaleInstruction.InstructionType.DoubleTypeInstruction:
par.Inlines.Add(new Run("." + instr.Type1.ToOpcodeParam())
{
Foreground = typeBrush
});
par.Inlines.Add(new Run("." + instr.Type2.ToOpcodeParam())
{
Foreground = typeBrush
});
break;
case UndertaleInstruction.InstructionType.ComparisonInstruction:
par.Inlines.Add(new Run("." + instr.Type1.ToOpcodeParam())
{
Foreground = typeBrush
});
par.Inlines.Add(new Run("." + instr.Type2.ToOpcodeParam())
{
Foreground = typeBrush
});
par.Inlines.Add(new Run(" "));
par.Inlines.Add(new Run(instr.ComparisonKind.ToString())
{
Foreground = opcodeBrush
});
break;
case UndertaleInstruction.InstructionType.GotoInstruction:
par.Inlines.Add(new Run(" "));
string tgt = (instr.Address + instr.JumpOffset).ToString("D5");
if (instr.Address + instr.JumpOffset == code.Length / 4)
{
tgt = "func_end";
}
if (instr.JumpOffsetPopenvExitMagic)
{
tgt = "[drop]";
}
par.Inlines.Add(new Run(tgt)
{
Foreground = argBrush, ToolTip = "$" + instr.JumpOffset.ToString("+#;-#;0")
});
break;
case UndertaleInstruction.InstructionType.PopInstruction:
par.Inlines.Add(new Run("." + instr.Type1.ToOpcodeParam())
{
Foreground = typeBrush
});
par.Inlines.Add(new Run("." + instr.Type2.ToOpcodeParam())
{
Foreground = typeBrush
});
par.Inlines.Add(new Run(" "));
if (instr.Type1 == UndertaleInstruction.DataType.Int16)
{
// Special scenario - the swap instruction
// TODO: Figure out the proper syntax, see #129
Run runType = new Run(instr.SwapExtra.ToString().ToLower())
{
Foreground = argBrush
};
par.Inlines.Add(runType);
}
else
{
if (instr.Type1 == UndertaleInstruction.DataType.Variable && instr.TypeInst != UndertaleInstruction.InstanceType.Undefined)
{
par.Inlines.Add(new Run(instr.TypeInst.ToString().ToLower())
{
Foreground = typeBrush
});
par.Inlines.Add(new Run("."));
}
Run runDest = new Run(instr.Destination.ToString())
{
Foreground = argBrush, Cursor = Cursors.Hand
};
runDest.MouseDown += (sender, e) =>
{
(Application.Current.MainWindow as MainWindow).ChangeSelection(instr.Destination);
};
par.Inlines.Add(runDest);
}
break;
case UndertaleInstruction.InstructionType.PushInstruction:
par.Inlines.Add(new Run("." + instr.Type1.ToOpcodeParam())
{
Foreground = typeBrush
});
par.Inlines.Add(new Run(" "));
if (instr.Type1 == UndertaleInstruction.DataType.Variable && instr.TypeInst != UndertaleInstruction.InstanceType.Undefined)
{
par.Inlines.Add(new Run(instr.TypeInst.ToString().ToLower())
{
Foreground = typeBrush
});
par.Inlines.Add(new Run("."));
}
Run valueRun = new Run((instr.Value as IFormattable)?.ToString(null, CultureInfo.InvariantCulture) ?? instr.Value.ToString())
{
Foreground = argBrush, Cursor = (instr.Value is UndertaleObject || instr.Value is UndertaleResourceRef) ? Cursors.Hand : Cursors.Arrow
};
if (instr.Value is UndertaleResourceRef)
{
valueRun.MouseDown += (sender, e) =>
{
(Application.Current.MainWindow as MainWindow).ChangeSelection((instr.Value as UndertaleResourceRef).Resource);
};
}
else if (instr.Value is UndertaleObject)
{
valueRun.MouseDown += (sender, e) =>
{
(Application.Current.MainWindow as MainWindow).ChangeSelection(instr.Value);
};
}
par.Inlines.Add(valueRun);
break;
case UndertaleInstruction.InstructionType.CallInstruction:
par.Inlines.Add(new Run("." + instr.Type1.ToOpcodeParam())
{
Foreground = typeBrush
});
par.Inlines.Add(new Run(" "));
par.Inlines.Add(new Run(instr.Function.ToString())
{
Foreground = argBrush
});
par.Inlines.Add(new Run("(argc="));
par.Inlines.Add(new Run(instr.ArgumentsCount.ToString())
{
Foreground = argBrush
});
par.Inlines.Add(new Run(")"));
break;
case UndertaleInstruction.InstructionType.BreakInstruction:
par.Inlines.Add(new Run("." + instr.Type1.ToOpcodeParam())
{
Foreground = typeBrush
});
par.Inlines.Add(new Run(" "));
par.Inlines.Add(new Run(instr.Value.ToString())
{
Foreground = argBrush
});
break;
}
if (par.Inlines.Count >= 250)
{ // Makes selecting text possible.
document.Blocks.Add(par);
par = new Paragraph();
par.Margin = new Thickness(0);
}
else
{
par.Inlines.Add(new Run("\n"));
}
}
}
document.Blocks.Add(par);
DisassemblyView.Document = document;
CurrentDisassembled = code;
}
