248 lines
		
	
	
		
			6.2 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			248 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; // $FlowIgnore
 | 
						|
 | 
						|
        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;
 | 
						|
} |