| function _typeof(obj) { if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); } |
| |
| function _sliceIterator(arr, i) { var _arr = []; var _n = true; var _d = false; var _e = undefined; try { for (var _i = arr[Symbol.iterator](), _s; !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"] != null) _i["return"](); } finally { if (_d) throw _e; } } return _arr; } |
| |
| function _slicedToArray(arr, i) { if (Array.isArray(arr)) { return arr; } else if (Symbol.iterator in Object(arr)) { return _sliceIterator(arr, i); } else { throw new TypeError("Invalid attempt to destructure non-iterable instance"); } } |
| |
| import { isAnonymous, isInstruction } from "@webassemblyjs/ast"; |
| import Long from "@xtuc/long"; |
| var compact = false; |
| var space = " "; |
| |
| var quote = function quote(str) { |
| return "\"".concat(str, "\""); |
| }; |
| |
| function indent(nb) { |
| return Array(nb).fill(space + space).join(""); |
| } // TODO(sven): allow arbitrary ast nodes |
| |
| |
| export function print(n) { |
| if (n.type === "Program") { |
| return printProgram(n, 0); |
| } else { |
| throw new Error("Unsupported node in print of type: " + String(n.type)); |
| } |
| } |
| |
| function printProgram(n, depth) { |
| return n.body.reduce(function (acc, child) { |
| if (child.type === "Module") { |
| acc += printModule(child, depth + 1); |
| } |
| |
| if (child.type === "Func") { |
| acc += printFunc(child, depth + 1); |
| } |
| |
| if (child.type === "BlockComment") { |
| acc += printBlockComment(child); |
| } |
| |
| if (child.type === "LeadingComment") { |
| acc += printLeadingComment(child); |
| } |
| |
| if (compact === false) { |
| acc += "\n"; |
| } |
| |
| return acc; |
| }, ""); |
| } |
| |
| function printTypeInstruction(n) { |
| var out = ""; |
| out += "("; |
| out += "type"; |
| out += space; |
| |
| if (n.id != null) { |
| out += printIndex(n.id); |
| out += space; |
| } |
| |
| out += "("; |
| out += "func"; |
| n.functype.params.forEach(function (param) { |
| out += space; |
| out += "("; |
| out += "param"; |
| out += space; |
| out += printFuncParam(param); |
| out += ")"; |
| }); |
| n.functype.results.forEach(function (result) { |
| out += space; |
| out += "("; |
| out += "result"; |
| out += space; |
| out += result; |
| out += ")"; |
| }); |
| out += ")"; // func |
| |
| out += ")"; |
| return out; |
| } |
| |
| function printModule(n, depth) { |
| var out = "("; |
| out += "module"; |
| |
| if (typeof n.id === "string") { |
| out += space; |
| out += n.id; |
| } |
| |
| if (compact === false) { |
| out += "\n"; |
| } else { |
| out += space; |
| } |
| |
| n.fields.forEach(function (field) { |
| if (compact === false) { |
| out += indent(depth); |
| } |
| |
| switch (field.type) { |
| case "Func": |
| { |
| out += printFunc(field, depth + 1); |
| break; |
| } |
| |
| case "TypeInstruction": |
| { |
| out += printTypeInstruction(field); |
| break; |
| } |
| |
| case "Table": |
| { |
| out += printTable(field); |
| break; |
| } |
| |
| case "Global": |
| { |
| out += printGlobal(field, depth + 1); |
| break; |
| } |
| |
| case "ModuleExport": |
| { |
| out += printModuleExport(field); |
| break; |
| } |
| |
| case "ModuleImport": |
| { |
| out += printModuleImport(field); |
| break; |
| } |
| |
| case "Memory": |
| { |
| out += printMemory(field); |
| break; |
| } |
| |
| case "BlockComment": |
| { |
| out += printBlockComment(field); |
| break; |
| } |
| |
| case "LeadingComment": |
| { |
| out += printLeadingComment(field); |
| break; |
| } |
| |
| case "Start": |
| { |
| out += printStart(field); |
| break; |
| } |
| |
| case "Elem": |
| { |
| out += printElem(field, depth); |
| break; |
| } |
| |
| case "Data": |
| { |
| out += printData(field, depth); |
| break; |
| } |
| |
| default: |
| throw new Error("Unsupported node in printModule: " + String(field.type)); |
| } |
| |
| if (compact === false) { |
| out += "\n"; |
| } |
| }); |
| out += ")"; |
| return out; |
| } |
| |
| function printData(n, depth) { |
| var out = ""; |
| out += "("; |
| out += "data"; |
| out += space; |
| out += printIndex(n.memoryIndex); |
| out += space; |
| out += printInstruction(n.offset, depth); |
| out += space; |
| out += '"'; |
| n.init.values.forEach(function (byte) { |
| // Avoid non-displayable characters |
| if (byte <= 31 || byte == 34 || byte == 92 || byte >= 127) { |
| out += "\\"; |
| out += ("00" + byte.toString(16)).substr(-2); |
| } else if (byte > 255) { |
| throw new Error("Unsupported byte in data segment: " + byte); |
| } else { |
| out += String.fromCharCode(byte); |
| } |
| }); |
| out += '"'; |
| out += ")"; |
| return out; |
| } |
| |
| function printElem(n, depth) { |
| var out = ""; |
| out += "("; |
| out += "elem"; |
| out += space; |
| out += printIndex(n.table); |
| |
| var _n$offset = _slicedToArray(n.offset, 1), |
| firstOffset = _n$offset[0]; |
| |
| out += space; |
| out += "("; |
| out += "offset"; |
| out += space; |
| out += printInstruction(firstOffset, depth); |
| out += ")"; |
| n.funcs.forEach(function (func) { |
| out += space; |
| out += printIndex(func); |
| }); |
| out += ")"; |
| return out; |
| } |
| |
| function printStart(n) { |
| var out = ""; |
| out += "("; |
| out += "start"; |
| out += space; |
| out += printIndex(n.index); |
| out += ")"; |
| return out; |
| } |
| |
| function printLeadingComment(n) { |
| // Don't print leading comments in compact mode |
| if (compact === true) { |
| return ""; |
| } |
| |
| var out = ""; |
| out += ";;"; |
| out += n.value; |
| out += "\n"; |
| return out; |
| } |
| |
| function printBlockComment(n) { |
| // Don't print block comments in compact mode |
| if (compact === true) { |
| return ""; |
| } |
| |
| var out = ""; |
| out += "(;"; |
| out += n.value; |
| out += ";)"; |
| out += "\n"; |
| return out; |
| } |
| |
| function printSignature(n) { |
| var out = ""; |
| n.params.forEach(function (param) { |
| out += space; |
| out += "("; |
| out += "param"; |
| out += space; |
| out += printFuncParam(param); |
| out += ")"; |
| }); |
| n.results.forEach(function (result) { |
| out += space; |
| out += "("; |
| out += "result"; |
| out += space; |
| out += result; |
| out += ")"; |
| }); |
| return out; |
| } |
| |
| function printModuleImportDescr(n) { |
| var out = ""; |
| |
| if (n.type === "FuncImportDescr") { |
| out += "("; |
| out += "func"; |
| |
| if (isAnonymous(n.id) === false) { |
| out += space; |
| out += printIdentifier(n.id); |
| } |
| |
| out += printSignature(n.signature); |
| out += ")"; |
| } |
| |
| if (n.type === "GlobalType") { |
| out += "("; |
| out += "global"; |
| out += space; |
| out += printGlobalType(n); |
| out += ")"; |
| } |
| |
| if (n.type === "Table") { |
| out += printTable(n); |
| } |
| |
| return out; |
| } |
| |
| function printModuleImport(n) { |
| var out = ""; |
| out += "("; |
| out += "import"; |
| out += space; |
| out += quote(n.module); |
| out += space; |
| out += quote(n.name); |
| out += space; |
| out += printModuleImportDescr(n.descr); |
| out += ")"; |
| return out; |
| } |
| |
| function printGlobalType(n) { |
| var out = ""; |
| |
| if (n.mutability === "var") { |
| out += "("; |
| out += "mut"; |
| out += space; |
| out += n.valtype; |
| out += ")"; |
| } else { |
| out += n.valtype; |
| } |
| |
| return out; |
| } |
| |
| function printGlobal(n, depth) { |
| var out = ""; |
| out += "("; |
| out += "global"; |
| out += space; |
| |
| if (n.name != null && isAnonymous(n.name) === false) { |
| out += printIdentifier(n.name); |
| out += space; |
| } |
| |
| out += printGlobalType(n.globalType); |
| out += space; |
| n.init.forEach(function (i) { |
| out += printInstruction(i, depth + 1); |
| }); |
| out += ")"; |
| return out; |
| } |
| |
| function printTable(n) { |
| var out = ""; |
| out += "("; |
| out += "table"; |
| out += space; |
| |
| if (n.name != null && isAnonymous(n.name) === false) { |
| out += printIdentifier(n.name); |
| out += space; |
| } |
| |
| out += printLimit(n.limits); |
| out += space; |
| out += n.elementType; |
| out += ")"; |
| return out; |
| } |
| |
| function printFuncParam(n) { |
| var out = ""; |
| |
| if (typeof n.id === "string") { |
| out += "$" + n.id; |
| out += space; |
| } |
| |
| out += n.valtype; |
| return out; |
| } |
| |
| function printFunc(n, depth) { |
| var out = ""; |
| out += "("; |
| out += "func"; |
| |
| if (n.name != null) { |
| if (n.name.type === "Identifier" && isAnonymous(n.name) === false) { |
| out += space; |
| out += printIdentifier(n.name); |
| } |
| } |
| |
| if (n.signature.type === "Signature") { |
| out += printSignature(n.signature); |
| } else { |
| var index = n.signature; |
| out += space; |
| out += "("; |
| out += "type"; |
| out += space; |
| out += printIndex(index); |
| out += ")"; |
| } |
| |
| if (n.body.length > 0) { |
| // func is empty since we ignore the default end instruction |
| if (n.body.length === 1 && n.body[0].id === "end") { |
| out += ")"; |
| return out; |
| } |
| |
| if (compact === false) { |
| out += "\n"; |
| } |
| |
| n.body.forEach(function (i) { |
| if (i.id !== "end") { |
| out += indent(depth); |
| out += printInstruction(i, depth); |
| |
| if (compact === false) { |
| out += "\n"; |
| } |
| } |
| }); |
| out += indent(depth - 1) + ")"; |
| } else { |
| out += ")"; |
| } |
| |
| return out; |
| } |
| |
| function printInstruction(n, depth) { |
| switch (n.type) { |
| case "Instr": |
| // $FlowIgnore |
| return printGenericInstruction(n, depth + 1); |
| |
| case "BlockInstruction": |
| // $FlowIgnore |
| return printBlockInstruction(n, depth + 1); |
| |
| case "IfInstruction": |
| // $FlowIgnore |
| return printIfInstruction(n, depth + 1); |
| |
| case "CallInstruction": |
| // $FlowIgnore |
| return printCallInstruction(n, depth + 1); |
| |
| case "CallIndirectInstruction": |
| // $FlowIgnore |
| return printCallIndirectIntruction(n, depth + 1); |
| |
| case "LoopInstruction": |
| // $FlowIgnore |
| return printLoopInstruction(n, depth + 1); |
| |
| default: |
| throw new Error("Unsupported instruction: " + JSON.stringify(n.type)); |
| } |
| } |
| |
| function printCallIndirectIntruction(n, depth) { |
| var out = ""; |
| out += "("; |
| out += "call_indirect"; |
| |
| if (n.signature.type === "Signature") { |
| out += printSignature(n.signature); |
| } else if (n.signature.type === "Identifier") { |
| out += space; |
| out += "("; |
| out += "type"; |
| out += space; |
| out += printIdentifier(n.signature); |
| out += ")"; |
| } else { |
| throw new Error("CallIndirectInstruction: unsupported signature " + JSON.stringify(n.signature.type)); |
| } |
| |
| out += space; |
| |
| if (n.intrs != null) { |
| // $FlowIgnore |
| n.intrs.forEach(function (i, index) { |
| // $FlowIgnore |
| out += printInstruction(i, depth + 1); // $FlowIgnore |
| |
| if (index !== n.intrs.length - 1) { |
| out += space; |
| } |
| }); |
| } |
| |
| out += ")"; |
| return out; |
| } |
| |
| function printLoopInstruction(n, depth) { |
| var out = ""; |
| out += "("; |
| out += "loop"; |
| |
| if (n.label != null && isAnonymous(n.label) === false) { |
| out += space; |
| out += printIdentifier(n.label); |
| } |
| |
| if (typeof n.resulttype === "string") { |
| out += space; |
| out += "("; |
| out += "result"; |
| out += space; |
| out += n.resulttype; |
| out += ")"; |
| } |
| |
| if (n.instr.length > 0) { |
| n.instr.forEach(function (e) { |
| if (compact === false) { |
| out += "\n"; |
| } |
| |
| out += indent(depth); |
| out += printInstruction(e, depth + 1); |
| }); |
| |
| if (compact === false) { |
| out += "\n"; |
| out += indent(depth - 1); |
| } |
| } |
| |
| out += ")"; |
| return out; |
| } |
| |
| function printCallInstruction(n, depth) { |
| var out = ""; |
| out += "("; |
| out += "call"; |
| out += space; |
| out += printIndex(n.index); |
| |
| if (_typeof(n.instrArgs) === "object") { |
| // $FlowIgnore |
| n.instrArgs.forEach(function (arg) { |
| out += space; |
| out += printFuncInstructionArg(arg, depth + 1); |
| }); |
| } |
| |
| out += ")"; |
| return out; |
| } |
| |
| function printIfInstruction(n, depth) { |
| var out = ""; |
| out += "("; |
| out += "if"; |
| |
| if (n.testLabel != null && isAnonymous(n.testLabel) === false) { |
| out += space; |
| out += printIdentifier(n.testLabel); |
| } |
| |
| if (typeof n.result === "string") { |
| out += space; |
| out += "("; |
| out += "result"; |
| out += space; |
| out += n.result; |
| out += ")"; |
| } |
| |
| if (n.test.length > 0) { |
| out += space; |
| n.test.forEach(function (i) { |
| out += printInstruction(i, depth + 1); |
| }); |
| } |
| |
| if (n.consequent.length > 0) { |
| if (compact === false) { |
| out += "\n"; |
| } |
| |
| out += indent(depth); |
| out += "("; |
| out += "then"; |
| depth++; |
| n.consequent.forEach(function (i) { |
| if (compact === false) { |
| out += "\n"; |
| } |
| |
| out += indent(depth); |
| out += printInstruction(i, depth + 1); |
| }); |
| depth--; |
| |
| if (compact === false) { |
| out += "\n"; |
| out += indent(depth); |
| } |
| |
| out += ")"; |
| } else { |
| if (compact === false) { |
| out += "\n"; |
| out += indent(depth); |
| } |
| |
| out += "("; |
| out += "then"; |
| out += ")"; |
| } |
| |
| if (n.alternate.length > 0) { |
| if (compact === false) { |
| out += "\n"; |
| } |
| |
| out += indent(depth); |
| out += "("; |
| out += "else"; |
| depth++; |
| n.alternate.forEach(function (i) { |
| if (compact === false) { |
| out += "\n"; |
| } |
| |
| out += indent(depth); |
| out += printInstruction(i, depth + 1); |
| }); |
| depth--; |
| |
| if (compact === false) { |
| out += "\n"; |
| out += indent(depth); |
| } |
| |
| out += ")"; |
| } else { |
| if (compact === false) { |
| out += "\n"; |
| out += indent(depth); |
| } |
| |
| out += "("; |
| out += "else"; |
| out += ")"; |
| } |
| |
| if (compact === false) { |
| out += "\n"; |
| out += indent(depth - 1); |
| } |
| |
| out += ")"; |
| return out; |
| } |
| |
| function printBlockInstruction(n, depth) { |
| var out = ""; |
| out += "("; |
| out += "block"; |
| |
| if (n.label != null && isAnonymous(n.label) === false) { |
| out += space; |
| out += printIdentifier(n.label); |
| } |
| |
| if (typeof n.result === "string") { |
| out += space; |
| out += "("; |
| out += "result"; |
| out += space; |
| out += n.result; |
| out += ")"; |
| } |
| |
| if (n.instr.length > 0) { |
| n.instr.forEach(function (i) { |
| if (compact === false) { |
| out += "\n"; |
| } |
| |
| out += indent(depth); |
| out += printInstruction(i, depth + 1); |
| }); |
| |
| if (compact === false) { |
| out += "\n"; |
| } |
| |
| out += indent(depth - 1); |
| out += ")"; |
| } else { |
| out += ")"; |
| } |
| |
| return out; |
| } |
| |
| function printGenericInstruction(n, depth) { |
| var out = ""; |
| out += "("; |
| |
| if (typeof n.object === "string") { |
| out += n.object; |
| out += "."; |
| } |
| |
| out += n.id; |
| n.args.forEach(function (arg) { |
| out += space; |
| out += printFuncInstructionArg(arg, depth + 1); |
| }); |
| out += ")"; |
| return out; |
| } |
| |
| function printLongNumberLiteral(n) { |
| if (typeof n.raw === "string") { |
| return n.raw; |
| } |
| |
| var _n$value = n.value, |
| low = _n$value.low, |
| high = _n$value.high; |
| var v = new Long(low, high); |
| return v.toString(); |
| } |
| |
| function printFloatLiteral(n) { |
| if (typeof n.raw === "string") { |
| return n.raw; |
| } |
| |
| return String(n.value); |
| } |
| |
| function printFuncInstructionArg(n, depth) { |
| var out = ""; |
| |
| if (n.type === "NumberLiteral") { |
| out += printNumberLiteral(n); |
| } |
| |
| if (n.type === "LongNumberLiteral") { |
| out += printLongNumberLiteral(n); |
| } |
| |
| if (n.type === "Identifier" && isAnonymous(n) === false) { |
| out += printIdentifier(n); |
| } |
| |
| if (n.type === "ValtypeLiteral") { |
| out += n.name; |
| } |
| |
| if (n.type === "FloatLiteral") { |
| out += printFloatLiteral(n); |
| } |
| |
| if (isInstruction(n)) { |
| out += printInstruction(n, depth + 1); |
| } |
| |
| return out; |
| } |
| |
| function printNumberLiteral(n) { |
| if (typeof n.raw === "string") { |
| return n.raw; |
| } |
| |
| return String(n.value); |
| } |
| |
| function printModuleExport(n) { |
| var out = ""; |
| out += "("; |
| out += "export"; |
| out += space; |
| out += quote(n.name); |
| |
| if (n.descr.exportType === "Func") { |
| out += space; |
| out += "("; |
| out += "func"; |
| out += space; |
| out += printIndex(n.descr.id); |
| out += ")"; |
| } else if (n.descr.exportType === "Global") { |
| out += space; |
| out += "("; |
| out += "global"; |
| out += space; |
| out += printIndex(n.descr.id); |
| out += ")"; |
| } else if (n.descr.exportType === "Memory" || n.descr.exportType === "Mem") { |
| out += space; |
| out += "("; |
| out += "memory"; |
| out += space; |
| out += printIndex(n.descr.id); |
| out += ")"; |
| } else if (n.descr.exportType === "Table") { |
| out += space; |
| out += "("; |
| out += "table"; |
| out += space; |
| out += printIndex(n.descr.id); |
| out += ")"; |
| } else { |
| throw new Error("printModuleExport: unknown type: " + n.descr.exportType); |
| } |
| |
| out += ")"; |
| return out; |
| } |
| |
| function printIdentifier(n) { |
| return "$" + n.value; |
| } |
| |
| function printIndex(n) { |
| if (n.type === "Identifier") { |
| return printIdentifier(n); |
| } else if (n.type === "NumberLiteral") { |
| return printNumberLiteral(n); |
| } else { |
| throw new Error("Unsupported index: " + n.type); |
| } |
| } |
| |
| function printMemory(n) { |
| var out = ""; |
| out += "("; |
| out += "memory"; |
| |
| if (n.id != null) { |
| out += space; |
| out += printIndex(n.id); |
| out += space; |
| } |
| |
| out += printLimit(n.limits); |
| out += ")"; |
| return out; |
| } |
| |
| function printLimit(n) { |
| var out = ""; |
| out += n.min + ""; |
| |
| if (n.max != null) { |
| out += space; |
| out += String(n.max); |
| } |
| |
| return out; |
| } |