123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900 |
- 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;
- }
|