ssc/sbdump/main.cs
2022-01-26 06:47:34 -06:00

666 lines
24 KiB
C#

using arookas.IO.Binary;
using System;
using System.IO;
using System.Linq;
using System.Text;
using System.Collections.Generic;
namespace arookas {
class sbdump {
static CommandLineSettings sSettings;
static aBinaryReader sReader;
static TextWriter sWriter;
static uint sTextOffset, sDataOffset, sSymOffset;
static int sDataCount, sSymCount, sVarCount;
const string cTitle = "sbdump arookas";
static readonly string[] sSymbolTypes = { "builtin", "function", "var", };
static readonly string[] sCommandNames = {
"int", "flt", "str", "adr", "var", "nop", "inc", "dec",
"add", "sub", "mul", "div", "mod", "ass", "eq", "ne",
"gt", "lt", "ge", "le", "neg", "not", "and", "or",
"band", "bor", "shl", "shr", "call", "func", "mkfr", "mkds",
"ret", "ret0", "jne", "jmp", "pop", "int0", "int1", "end",
};
static int Main(string[] args) {
#if !DEBUG
try {
#endif
Console.WriteLine(cTitle);
ReadCommandLine(args);
Console.WriteLine("Opening input file...");
using (var sb = File.OpenRead(sSettings.Input))
{
CreateReader(sb);
Console.WriteLine("Creating output file...");
using (sWriter = File.CreateText(sSettings.Output))
{
ReadHeader();
if (sSettings.OutputSun)
{
WriteSun();
}
else
{
WritePreamble();
if (sSettings.OutputHeader)
{
WriteHeader();
}
if (sSettings.OutputText)
{
WriteText();
}
if (sSettings.OutputData)
{
WriteData();
}
if (sSettings.OutputSym)
{
WriteSym();
}
if (sSettings.OutputVars)
{
WriteVars();
}
}
Console.WriteLine("Closing output file...");
}
Console.WriteLine("Closing input file...");
}
Console.WriteLine("Done.");
#if !DEBUG
}
catch (Exception e) {
Console.WriteLine(e.Message);
return 1;
}
#endif
return 0;
}
static void ReadCommandLine(string[] args)
{
Console.WriteLine("Reading command line...");
sSettings = new CommandLineSettings(new aCommandLine(args));
}
static void CreateReader(Stream stream) {
Console.WriteLine("Creating binary reader...");
sReader = new aBinaryReader(stream, Endianness.Big, Encoding.GetEncoding(932));
}
static void WritePreamble() {
Console.WriteLine("Writing preamble...");
sWriter.WriteLine("# {0}", cTitle);
sWriter.WriteLine("# {0}", DateTime.Now);
sWriter.WriteLine("# {0}", Path.GetFileName(sSettings.Input));
sWriter.WriteLine();
}
static void ReadHeader() {
Console.WriteLine("Reading header...");
if (sReader.Read32() != 0x53504342u) { // 'SPCB'
throw new Exception("Invalid magic.");
}
sTextOffset = sReader.Read32();
sDataOffset = sReader.Read32();
sDataCount = sReader.ReadS32();
sSymOffset = sReader.Read32();
sSymCount = sReader.ReadS32();
sVarCount = sReader.ReadS32();
}
static void WriteHeader()
{
Console.WriteLine("Outputting header...");
sWriter.WriteLine("# Header information");
sWriter.WriteLine("# .text offset : {0:X8}", sTextOffset);
sWriter.WriteLine("# .data offset : {0:X8}", sDataOffset);
sWriter.WriteLine("# .data count : {0}", sDataCount);
sWriter.WriteLine("# .sym offset : {0:X8}", sSymOffset);
sWriter.WriteLine("# .sym count : {0}", sSymCount);
sWriter.WriteLine("# var count : {0}", sVarCount);
sWriter.WriteLine();
}
static void WriteText()
{
Console.WriteLine("Outputting .text...");
sWriter.WriteLine(".text");
WriteText(0u);
var symbols = new Symbol[sSymCount];
for (var i = 0; i < sSymCount; ++i)
{
symbols[i] = FetchSymbol(i);
}
foreach (var symbol in symbols.Where(i => i.Type == SymbolType.Function).OrderBy(i => i.Data))
{
sWriter.WriteLine("{0}: ", FetchSymbolName(symbol));
WriteText(symbol.Data);
}
}
static void WriteText(uint ofs) {
byte command;
sReader.Keep();
sReader.Goto(sTextOffset + ofs);
var maxofs = 0u;
do {
var pos = sReader.Position - sTextOffset;
command = sReader.Read8();
sWriter.Write(" {0:X8} {1}", pos, sCommandNames[command]);
var nextofs = 0u;
switch (command) {
case 0x00: sWriter.Write(" {0}", sReader.ReadS32()); break;
case 0x01: sWriter.Write(" {0}", sReader.ReadF32()); break;
case 0x02: {
var data = sReader.ReadS32();
var value = FetchDataValue(data);
sWriter.Write(" {0} # \"{1}\"", data, value);
break;
}
case 0x03: sWriter.Write(" ${0:X8}", sReader.Read32()); break;
case 0x04: WriteVar(); break;
case 0x06: break;
case 0x07: break;
case 0x0D: {
sReader.Read8(); // TSpcInterp skips this byte
WriteVar();
break;
}
case 0x1C: {
var dest = sReader.Read32();
var args = sReader.ReadS32();
var symbol = FetchSymbol(i => i.Data == dest);
if (symbol != null) {
sWriter.Write(" {0}, {1}", FetchSymbolName(symbol), args);
}
else {
sWriter.Write(" ${0:X8}, {1}", dest, args);
}
break;
}
case 0x1D: sWriter.Write(" {0}, {1}", FetchSymbolName(FetchSymbol(sReader.ReadS32())), sReader.ReadS32()); break;
case 0x1E: sWriter.Write(" {0}", sReader.ReadS32()); break;
case 0x1F: sWriter.Write(" {0}", sReader.ReadS32()); break;
case 0x22: nextofs = WriteJmp(ofs); break;
case 0x23: nextofs = WriteJmp(ofs); break;
}
sWriter.WriteLine();
if (nextofs > maxofs) {
maxofs = nextofs;
}
} while (!IsReturnCommand(command) || sReader.Position <= sTextOffset + maxofs);
sWriter.WriteLine();
sReader.Back();
}
static void WriteVar() {
var display = sReader.ReadS32();
var data = sReader.ReadS32();
sWriter.Write(" {0} {1}", display, data);
switch (display) {
case 0: sWriter.Write(" # {0}", FetchSymbolName(FetchSymbol(i => i.Type == SymbolType.Variable && i.Data == data))); break;
case 1: sWriter.Write(" # local{0}", data); break;
}
}
static uint WriteJmp(uint ofs) {
var dest = sReader.Read32();
var symbol = FetchSymbol(i => i.Data == ofs);
if (ofs > 0 && symbol != null) {
var name = FetchSymbolName(symbol);
sWriter.Write(" {0} + ${1:X4} # ${2:X8}", name, dest - ofs, dest);
}
else {
sWriter.Write(" ${0:X8}", dest);
}
return dest;
}
static void WriteData() {
Console.WriteLine("Outputting .data...");
sWriter.WriteLine(".data");
sReader.Goto(sDataOffset);
for (var i = 0; i < sDataCount; ++i) {
var ofs = sReader.Read32();
var data = FetchDataValue(ofs);
sWriter.WriteLine(" .string \"{0}\"", data);
}
sWriter.WriteLine();
}
static void WriteSym() {
Console.WriteLine("Outputting .sym...");
sWriter.WriteLine(".sym");
sReader.Goto(sSymOffset);
for (var i = 0; i < sSymCount; ++i) {
var symbol = new Symbol(sReader);
var name = FetchSymbolName(symbol);
sWriter.WriteLine(" .{0} {1}", sSymbolTypes[(int)symbol.Type], name);
}
sWriter.WriteLine();
}
static void WriteVars() {
Console.WriteLine("Outputting variables...");
sWriter.WriteLine("# variables:");
for (var i = 0; i < sVarCount; ++i) {
var symbol = FetchSymbol(j => j.Type == SymbolType.Variable && j.Data == i);
if (symbol != null) {
sWriter.WriteLine("# {0}", FetchSymbolName(symbol));
}
else {
sWriter.WriteLine("# (NULL)");
}
}
sWriter.WriteLine();
}
static void WriteSun()
{
Console.WriteLine("Decompiling sb file...");
var symbols = new Symbol[sSymCount];
for (var i = 0; i < sSymCount; ++i)
{
symbols[i] = FetchSymbol(i);
}
foreach (var symbol in symbols.Where(i => i.Type == SymbolType.Function).OrderBy(i => i.Data))
{
sWriter.WriteLine();
sWriter.WriteLine("function "+ FetchSymbolName(symbol)+ "(...)" + Environment.NewLine+ "{");
DecompFunction(symbol.Data, 1);
sWriter.WriteLine("}");
sWriter.WriteLine();
}
DecompFunction(0u, 0);//Decompile main part.
}
static void DecompFunction(uint ofs, int IndentL)
{
byte command;
sReader.Keep();
sReader.Goto(sTextOffset + ofs);
var maxofs = 0u;
Stack<string> Stack = new Stack<string>();
CodeGraph FuncCodeGraph = new CodeGraph();
do
{
var pos = sReader.Position - sTextOffset;
command = sReader.Read8();
var nextofs = 0u;
switch (command)
{
case 0x00: //int
{
Stack.Push(sReader.ReadS32().ToString());
break;
}
case 0x01: //flt
{
Stack.Push(sReader.ReadF32().ToString("0.0###############"));
break;
}
case 0x02: //str
{
var data = sReader.ReadS32();
var value = FetchDataValue(data);
Stack.Push("\"" + value.Replace("\n", "\\n") +"\"");
break;
}
case 0x03: //adr
{
Stack.Push("$" + sReader.ReadS32().ToString("X8"));
break;
}
case 0x04: //var
{
var display = sReader.ReadS32();
var data = sReader.ReadS32();
switch (display)
{
case 0: Stack.Push(FetchSymbolName(FetchSymbol(i => i.Type == SymbolType.Variable && i.Data == data))); break;
case 1: Stack.Push("local " + data.ToString()); break;
}
break;
}
case 0x06: //inc
{
sReader.ReadS8(); // Ignore inline var
var display = sReader.ReadS32();
var data = sReader.ReadS32();
switch (display)
{
case 0: Stack.Push("++" + FetchSymbolName(FetchSymbol(i => i.Type == SymbolType.Variable && i.Data == data))); break;
case 1: Stack.Push("++" + data.ToString()); break;
}
break;
}
case 0x07: //dec
{
sReader.ReadS8(); // Ignore inline var
var display = sReader.ReadS32();
var data = sReader.ReadS32();
switch (display)
{
case 0: Stack.Push("--" + FetchSymbolName(FetchSymbol(i => i.Type == SymbolType.Variable && i.Data == data))); break;
case 1: Stack.Push("--" + data.ToString()); break;
}
break;
}
case 0x08: //add
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " + " + Op2 + ")");
break;
}
case 0x09: //sub
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " - " + Op2 + ")");
break;
}
case 0x0A: //mul
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " * " + Op2 + ")");
break;
}
case 0x0B: //div
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " / " + Op2 + ")");
break;
}
case 0x0C: //mod
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " % " + Op2 + ")");
break;
}
case 0x0D: //ass
{
sReader.Read8(); //Ignore this byte
var display = sReader.ReadS32();
var data = sReader.ReadS32();
string VariableName = "";
switch (display)
{
case 0: VariableName = "var " + FetchSymbolName(FetchSymbol(i => i.Type == SymbolType.Variable && i.Data == data)); break;
case 1: VariableName = "var local " + data.ToString(); break;
}
CodeVertex NewLine = new CodeVertex(VertexType.CodeBlock , -1, VariableName + " = " + Stack.Pop() + ";", pos);
FuncCodeGraph.AddVertex(NewLine);
break;
}
case 0x0E: //eq
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " == " + Op2 + ")");
break;
}
case 0x0F: //ne
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " != " + Op2 + ")");
break;
}
case 0x10: //gt
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " > " + Op2 + ")");
break;
}
case 0x11: //lt
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " < " + Op2 + ")");
break;
}
case 0x12: //ge
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " >= " + Op2 + ")");
break;
}
case 0x13: //le
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " <= " + Op2 + ")");
break;
}
case 0x14: //neg
{
string Op1 = Stack.Pop();
Stack.Push("-(" + Op1 + ")");
break;
}
case 0x15: //not
{
string Op1 = Stack.Pop();
Stack.Push("!(" + Op1 + ")");
break;
}
case 0x16: //and
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " && " + Op2 + ")");
break;
}
case 0x17: //or
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " || " + Op2 + ")");
break;
}
case 0x18: //band
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " & " + Op2 + ")");
break;
}
case 0x19: //bor
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " ^ " + Op2 + ")");
break;
}
case 0x1A: //shl
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " << " + Op2 + ")");
break;
}
case 0x1B: //shr
{
string Op2 = Stack.Pop();
string Op1 = Stack.Pop();
Stack.Push("(" + Op1 + " >> " + Op2 + ")");
break;
}
case 0x1C: //call
{
var dest = sReader.Read32();
var args = sReader.ReadS32();
var symbol = FetchSymbol(i => i.Data == dest);
string FuncName = "";
if (symbol != null)
{
FuncName = FetchSymbolName(symbol);
}
else
{
FuncName = dest.ToString("X8");
}
string FuncInput = "";
for(int i = 0; i < args; i++)
{
string Op = Stack.Pop();
if (i != 0)
FuncInput = Op + "," + FuncInput;
else
FuncInput = Op;
}
Stack.Push(FuncName + "(" + FuncInput + ")");
break;
}
case 0x1D: //func
{
string FuncName = FetchSymbolName(FetchSymbol(sReader.ReadS32()));
var args = sReader.ReadS32();
string FuncInput = "";
for (int i = 0; i < args; i++)
{
string Op = Stack.Pop();
if (i != 0)
FuncInput = Op + "," + FuncInput;
else
FuncInput = Op;
}
Stack.Push(FuncName + "(" + FuncInput + ")");
break;
}
case 0x1E: sReader.ReadS32(); break;
case 0x1F: sReader.ReadS32(); break;
case 0x20: //ret
{
string Op = Stack.Pop();
CodeVertex NewLine = new CodeVertex(VertexType.CodeBlock , -1, "return " + Op + ";", pos);
FuncCodeGraph.AddVertex(NewLine);
break;
}
case 0x21: //ret0
{
CodeVertex NewLine = new CodeVertex(VertexType.CodeBlock, -1, "return 0;", pos);
FuncCodeGraph.AddVertex(NewLine);
break;
}
case 0x22: //jne
{
var dest = sReader.Read32();
nextofs = dest;
string Op = Stack.Pop();
CodeVertex NewLine = new CodeVertex(VertexType.ConditionalBranch, dest, Op, pos);
FuncCodeGraph.AddVertex(NewLine);
break;
}
case 0x23: //jmp
{
var dest = sReader.Read32();
nextofs = dest;
CodeVertex NewLine = new CodeVertex(VertexType.Branch, dest, "", pos);
FuncCodeGraph.AddVertex(NewLine);
break;
}
case 0x24: //pop
{
string Op = Stack.Pop();
CodeVertex NewLine = new CodeVertex(VertexType.CodeBlock, -1, Op + ";", pos);
FuncCodeGraph.AddVertex(NewLine);
break;
}
case 0x25: //int0
{
Stack.Push("0");
break;
}
case 0x26: //int1
{
Stack.Push("1");
break;
}
}
if (nextofs > maxofs)
{
maxofs = nextofs;
}
} while (!IsReturnCommand(command) || sReader.Position <= sTextOffset + maxofs);
WriteCode(FuncCodeGraph, IndentL);
sWriter.WriteLine();
sReader.Back();
}
static void WriteCode(CodeGraph FuncCodeGraph, int Indent)
{
sWriter.Write(FuncCodeGraph.OutputCode(Indent));
}
static uint FetchData(int i) {
sReader.Keep();
sReader.Goto(sDataOffset + (4 * i));
var data = sReader.Read32();
sReader.Back();
return data;
}
static string FetchDataValue(int i) {
if (i < 0 || i >= sDataCount) {
return "(NULL)";
}
return FetchDataValue(FetchData(i));
}
static string FetchDataValue(uint ofs) {
sReader.Keep();
sReader.Goto(sDataOffset + (4 * sDataCount) + ofs);
var data = sReader.ReadString<aZSTR>();
sReader.Back();
return data;
}
static Symbol FetchSymbol(int i) {
sReader.Keep();
sReader.Goto(sSymOffset + (20 * i));
var symbol = new Symbol(sReader);
sReader.Back();
return symbol;
}
static Symbol FetchSymbol(Predicate<Symbol> predicate) {
if (predicate == null) {
throw new ArgumentNullException("predicate");
}
Symbol found = null;
sReader.Keep();
sReader.Goto(sSymOffset);
for (var i = 0; i < sSymCount; ++i) {
var symbol = new Symbol(sReader);
if (predicate(symbol)) {
found = symbol;
break;
}
}
sReader.Back();
return found;
}
static string FetchSymbolName(Symbol symbol) {
sReader.Keep();
sReader.Goto(sSymOffset + (20 * sSymCount) + symbol.StringOffset);
var name = sReader.ReadString<aZSTR>();
sReader.Back();
return name;
}
static bool IsReturnCommand(byte cmd) {
return (
cmd == 0x20 || // ret
cmd == 0x21 || // ret0
cmd == 0x27 // end
);
}
}
}