247 lines
6.2 KiB
JavaScript
247 lines
6.2 KiB
JavaScript
|
import * as decoder from "./decoder";
|
||
|
import * as t from "@webassemblyjs/ast";
|
||
|
/**
|
||
|
* TODO(sven): I added initial props, but we should rather fix
|
||
|
* https://github.com/xtuc/webassemblyjs/issues/405
|
||
|
*/
|
||
|
|
||
|
var defaultDecoderOpts = {
|
||
|
dump: false,
|
||
|
ignoreCodeSection: false,
|
||
|
ignoreDataSection: false,
|
||
|
ignoreCustomNameSection: false
|
||
|
}; // traverses the AST, locating function name metadata, which is then
|
||
|
// used to update index-based identifiers with function names
|
||
|
|
||
|
function restoreFunctionNames(ast) {
|
||
|
var functionNames = [];
|
||
|
t.traverse(ast, {
|
||
|
FunctionNameMetadata: function FunctionNameMetadata(_ref) {
|
||
|
var node = _ref.node;
|
||
|
functionNames.push({
|
||
|
name: node.value,
|
||
|
index: node.index
|
||
|
});
|
||
|
}
|
||
|
});
|
||
|
|
||
|
if (functionNames.length === 0) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
t.traverse(ast, {
|
||
|
Func: function (_Func) {
|
||
|
function Func(_x) {
|
||
|
return _Func.apply(this, arguments);
|
||
|
}
|
||
|
|
||
|
Func.toString = function () {
|
||
|
return _Func.toString();
|
||
|
};
|
||
|
|
||
|
return Func;
|
||
|
}(function (_ref2) {
|
||
|
var node = _ref2.node;
|
||
|
// $FlowIgnore
|
||
|
var nodeName = node.name;
|
||
|
var indexBasedFunctionName = nodeName.value;
|
||
|
var index = Number(indexBasedFunctionName.replace("func_", ""));
|
||
|
var functionName = functionNames.find(function (f) {
|
||
|
return f.index === index;
|
||
|
});
|
||
|
|
||
|
if (functionName) {
|
||
|
var oldValue = nodeName.value;
|
||
|
nodeName.value = functionName.name;
|
||
|
nodeName.numeric = oldValue; // $FlowIgnore
|
||
|
|
||
|
delete nodeName.raw;
|
||
|
}
|
||
|
}),
|
||
|
// Also update the reference in the export
|
||
|
ModuleExport: function (_ModuleExport) {
|
||
|
function ModuleExport(_x2) {
|
||
|
return _ModuleExport.apply(this, arguments);
|
||
|
}
|
||
|
|
||
|
ModuleExport.toString = function () {
|
||
|
return _ModuleExport.toString();
|
||
|
};
|
||
|
|
||
|
return ModuleExport;
|
||
|
}(function (_ref3) {
|
||
|
var node = _ref3.node;
|
||
|
|
||
|
if (node.descr.exportType === "Func") {
|
||
|
// $FlowIgnore
|
||
|
var nodeName = node.descr.id;
|
||
|
var index = nodeName.value;
|
||
|
var functionName = functionNames.find(function (f) {
|
||
|
return f.index === index;
|
||
|
});
|
||
|
|
||
|
if (functionName) {
|
||
|
node.descr.id = t.identifier(functionName.name);
|
||
|
}
|
||
|
}
|
||
|
}),
|
||
|
ModuleImport: function (_ModuleImport) {
|
||
|
function ModuleImport(_x3) {
|
||
|
return _ModuleImport.apply(this, arguments);
|
||
|
}
|
||
|
|
||
|
ModuleImport.toString = function () {
|
||
|
return _ModuleImport.toString();
|
||
|
};
|
||
|
|
||
|
return ModuleImport;
|
||
|
}(function (_ref4) {
|
||
|
var node = _ref4.node;
|
||
|
|
||
|
if (node.descr.type === "FuncImportDescr") {
|
||
|
// $FlowIgnore
|
||
|
var indexBasedFunctionName = node.descr.id;
|
||
|
var index = Number(indexBasedFunctionName.replace("func_", ""));
|
||
|
var functionName = functionNames.find(function (f) {
|
||
|
return f.index === index;
|
||
|
});
|
||
|
|
||
|
if (functionName) {
|
||
|
// $FlowIgnore
|
||
|
node.descr.id = t.identifier(functionName.name);
|
||
|
}
|
||
|
}
|
||
|
}),
|
||
|
CallInstruction: function (_CallInstruction) {
|
||
|
function CallInstruction(_x4) {
|
||
|
return _CallInstruction.apply(this, arguments);
|
||
|
}
|
||
|
|
||
|
CallInstruction.toString = function () {
|
||
|
return _CallInstruction.toString();
|
||
|
};
|
||
|
|
||
|
return CallInstruction;
|
||
|
}(function (nodePath) {
|
||
|
var node = nodePath.node;
|
||
|
var index = node.index.value;
|
||
|
var functionName = functionNames.find(function (f) {
|
||
|
return f.index === index;
|
||
|
});
|
||
|
|
||
|
if (functionName) {
|
||
|
var oldValue = node.index;
|
||
|
node.index = t.identifier(functionName.name);
|
||
|
node.numeric = oldValue; // $FlowIgnore
|
||
|
|
||
|
delete node.raw;
|
||
|
}
|
||
|
})
|
||
|
});
|
||
|
}
|
||
|
|
||
|
function restoreLocalNames(ast) {
|
||
|
var localNames = [];
|
||
|
t.traverse(ast, {
|
||
|
LocalNameMetadata: function LocalNameMetadata(_ref5) {
|
||
|
var node = _ref5.node;
|
||
|
localNames.push({
|
||
|
name: node.value,
|
||
|
localIndex: node.localIndex,
|
||
|
functionIndex: node.functionIndex
|
||
|
});
|
||
|
}
|
||
|
});
|
||
|
|
||
|
if (localNames.length === 0) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
t.traverse(ast, {
|
||
|
Func: function (_Func2) {
|
||
|
function Func(_x5) {
|
||
|
return _Func2.apply(this, arguments);
|
||
|
}
|
||
|
|
||
|
Func.toString = function () {
|
||
|
return _Func2.toString();
|
||
|
};
|
||
|
|
||
|
return Func;
|
||
|
}(function (_ref6) {
|
||
|
var node = _ref6.node;
|
||
|
var signature = node.signature;
|
||
|
|
||
|
if (signature.type !== "Signature") {
|
||
|
return;
|
||
|
} // $FlowIgnore
|
||
|
|
||
|
|
||
|
var nodeName = node.name;
|
||
|
var indexBasedFunctionName = nodeName.value;
|
||
|
var functionIndex = Number(indexBasedFunctionName.replace("func_", ""));
|
||
|
signature.params.forEach(function (param, paramIndex) {
|
||
|
var paramName = localNames.find(function (f) {
|
||
|
return f.localIndex === paramIndex && f.functionIndex === functionIndex;
|
||
|
});
|
||
|
|
||
|
if (paramName && paramName.name !== "") {
|
||
|
param.id = paramName.name;
|
||
|
}
|
||
|
});
|
||
|
})
|
||
|
});
|
||
|
}
|
||
|
|
||
|
function restoreModuleName(ast) {
|
||
|
t.traverse(ast, {
|
||
|
ModuleNameMetadata: function (_ModuleNameMetadata) {
|
||
|
function ModuleNameMetadata(_x6) {
|
||
|
return _ModuleNameMetadata.apply(this, arguments);
|
||
|
}
|
||
|
|
||
|
ModuleNameMetadata.toString = function () {
|
||
|
return _ModuleNameMetadata.toString();
|
||
|
};
|
||
|
|
||
|
return ModuleNameMetadata;
|
||
|
}(function (moduleNameMetadataPath) {
|
||
|
// update module
|
||
|
t.traverse(ast, {
|
||
|
Module: function (_Module) {
|
||
|
function Module(_x7) {
|
||
|
return _Module.apply(this, arguments);
|
||
|
}
|
||
|
|
||
|
Module.toString = function () {
|
||
|
return _Module.toString();
|
||
|
};
|
||
|
|
||
|
return Module;
|
||
|
}(function (_ref7) {
|
||
|
var node = _ref7.node;
|
||
|
var name = moduleNameMetadataPath.node.value; // compatiblity with wast-parser
|
||
|
|
||
|
if (name === "") {
|
||
|
name = null;
|
||
|
}
|
||
|
|
||
|
node.id = name;
|
||
|
})
|
||
|
});
|
||
|
})
|
||
|
});
|
||
|
}
|
||
|
|
||
|
export function decode(buf, customOpts) {
|
||
|
var opts = Object.assign({}, defaultDecoderOpts, customOpts);
|
||
|
var ast = decoder.decode(buf, opts);
|
||
|
|
||
|
if (opts.ignoreCustomNameSection === false) {
|
||
|
restoreFunctionNames(ast);
|
||
|
restoreLocalNames(ast);
|
||
|
restoreModuleName(ast);
|
||
|
}
|
||
|
|
||
|
return ast;
|
||
|
}
|