You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							306 lines
						
					
					
						
							8.8 KiB
						
					
					
				
			
		
		
	
	
							306 lines
						
					
					
						
							8.8 KiB
						
					
					
				"use strict"; | 
						|
 | 
						|
Object.defineProperty(exports, "__esModule", { | 
						|
  value: true | 
						|
}); | 
						|
exports.isAnonymous = isAnonymous; | 
						|
exports.getSectionMetadata = getSectionMetadata; | 
						|
exports.getSectionMetadatas = getSectionMetadatas; | 
						|
exports.sortSectionMetadata = sortSectionMetadata; | 
						|
exports.orderedInsertNode = orderedInsertNode; | 
						|
exports.assertHasLoc = assertHasLoc; | 
						|
exports.getEndOfSection = getEndOfSection; | 
						|
exports.shiftLoc = shiftLoc; | 
						|
exports.shiftSection = shiftSection; | 
						|
exports.signatureForOpcode = signatureForOpcode; | 
						|
exports.getUniqueNameGenerator = getUniqueNameGenerator; | 
						|
exports.getStartByteOffset = getStartByteOffset; | 
						|
exports.getEndByteOffset = getEndByteOffset; | 
						|
exports.getFunctionBeginingByteOffset = getFunctionBeginingByteOffset; | 
						|
exports.getEndBlockByteOffset = getEndBlockByteOffset; | 
						|
exports.getStartBlockByteOffset = getStartBlockByteOffset; | 
						|
 | 
						|
var _signatures = require("./signatures"); | 
						|
 | 
						|
var _traverse = require("./traverse"); | 
						|
 | 
						|
var _helperWasmBytecode = _interopRequireWildcard(require("@webassemblyjs/helper-wasm-bytecode")); | 
						|
 | 
						|
function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = Object.defineProperty && Object.getOwnPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : {}; if (desc.get || desc.set) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; return newObj; } } | 
						|
 | 
						|
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"); } } | 
						|
 | 
						|
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 isAnonymous(ident) { | 
						|
  return ident.raw === ""; | 
						|
} | 
						|
 | 
						|
function getSectionMetadata(ast, name) { | 
						|
  var section; | 
						|
  (0, _traverse.traverse)(ast, { | 
						|
    SectionMetadata: function (_SectionMetadata) { | 
						|
      function SectionMetadata(_x) { | 
						|
        return _SectionMetadata.apply(this, arguments); | 
						|
      } | 
						|
 | 
						|
      SectionMetadata.toString = function () { | 
						|
        return _SectionMetadata.toString(); | 
						|
      }; | 
						|
 | 
						|
      return SectionMetadata; | 
						|
    }(function (_ref) { | 
						|
      var node = _ref.node; | 
						|
 | 
						|
      if (node.section === name) { | 
						|
        section = node; | 
						|
      } | 
						|
    }) | 
						|
  }); | 
						|
  return section; | 
						|
} | 
						|
 | 
						|
function getSectionMetadatas(ast, name) { | 
						|
  var sections = []; | 
						|
  (0, _traverse.traverse)(ast, { | 
						|
    SectionMetadata: function (_SectionMetadata2) { | 
						|
      function SectionMetadata(_x2) { | 
						|
        return _SectionMetadata2.apply(this, arguments); | 
						|
      } | 
						|
 | 
						|
      SectionMetadata.toString = function () { | 
						|
        return _SectionMetadata2.toString(); | 
						|
      }; | 
						|
 | 
						|
      return SectionMetadata; | 
						|
    }(function (_ref2) { | 
						|
      var node = _ref2.node; | 
						|
 | 
						|
      if (node.section === name) { | 
						|
        sections.push(node); | 
						|
      } | 
						|
    }) | 
						|
  }); | 
						|
  return sections; | 
						|
} | 
						|
 | 
						|
function sortSectionMetadata(m) { | 
						|
  if (m.metadata == null) { | 
						|
    console.warn("sortSectionMetadata: no metadata to sort"); | 
						|
    return; | 
						|
  } // $FlowIgnore | 
						|
 | 
						|
 | 
						|
  m.metadata.sections.sort(function (a, b) { | 
						|
    var aId = _helperWasmBytecode.default.sections[a.section]; | 
						|
    var bId = _helperWasmBytecode.default.sections[b.section]; | 
						|
 | 
						|
    if (typeof aId !== "number" || typeof bId !== "number") { | 
						|
      throw new Error("Section id not found"); | 
						|
    } | 
						|
 | 
						|
    return aId - bId; | 
						|
  }); | 
						|
} | 
						|
 | 
						|
function orderedInsertNode(m, n) { | 
						|
  assertHasLoc(n); | 
						|
  var didInsert = false; | 
						|
 | 
						|
  if (n.type === "ModuleExport") { | 
						|
    m.fields.push(n); | 
						|
    return; | 
						|
  } | 
						|
 | 
						|
  m.fields = m.fields.reduce(function (acc, field) { | 
						|
    var fieldEndCol = Infinity; | 
						|
 | 
						|
    if (field.loc != null) { | 
						|
      // $FlowIgnore | 
						|
      fieldEndCol = field.loc.end.column; | 
						|
    } // $FlowIgnore: assertHasLoc ensures that | 
						|
 | 
						|
 | 
						|
    if (didInsert === false && n.loc.start.column < fieldEndCol) { | 
						|
      didInsert = true; | 
						|
      acc.push(n); | 
						|
    } | 
						|
 | 
						|
    acc.push(field); | 
						|
    return acc; | 
						|
  }, []); // Handles empty modules or n is the last element | 
						|
 | 
						|
  if (didInsert === false) { | 
						|
    m.fields.push(n); | 
						|
  } | 
						|
} | 
						|
 | 
						|
function assertHasLoc(n) { | 
						|
  if (n.loc == null || n.loc.start == null || n.loc.end == null) { | 
						|
    throw new Error("Internal failure: node (".concat(JSON.stringify(n.type), ") has no location information")); | 
						|
  } | 
						|
} | 
						|
 | 
						|
function getEndOfSection(s) { | 
						|
  assertHasLoc(s.size); | 
						|
  return s.startOffset + s.size.value + ( // $FlowIgnore | 
						|
  s.size.loc.end.column - s.size.loc.start.column); | 
						|
} | 
						|
 | 
						|
function shiftLoc(node, delta) { | 
						|
  // $FlowIgnore | 
						|
  node.loc.start.column += delta; // $FlowIgnore | 
						|
 | 
						|
  node.loc.end.column += delta; | 
						|
} | 
						|
 | 
						|
function shiftSection(ast, node, delta) { | 
						|
  if (node.type !== "SectionMetadata") { | 
						|
    throw new Error("Can not shift node " + JSON.stringify(node.type)); | 
						|
  } | 
						|
 | 
						|
  node.startOffset += delta; | 
						|
 | 
						|
  if (_typeof(node.size.loc) === "object") { | 
						|
    shiftLoc(node.size, delta); | 
						|
  } // Custom sections doesn't have vectorOfSize | 
						|
 | 
						|
 | 
						|
  if (_typeof(node.vectorOfSize) === "object" && _typeof(node.vectorOfSize.loc) === "object") { | 
						|
    shiftLoc(node.vectorOfSize, delta); | 
						|
  } | 
						|
 | 
						|
  var sectionName = node.section; // shift node locations within that section | 
						|
 | 
						|
  (0, _traverse.traverse)(ast, { | 
						|
    Node: function Node(_ref3) { | 
						|
      var node = _ref3.node; | 
						|
      var section = (0, _helperWasmBytecode.getSectionForNode)(node); | 
						|
 | 
						|
      if (section === sectionName && _typeof(node.loc) === "object") { | 
						|
        shiftLoc(node, delta); | 
						|
      } | 
						|
    } | 
						|
  }); | 
						|
} | 
						|
 | 
						|
function signatureForOpcode(object, name) { | 
						|
  var opcodeName = name; | 
						|
 | 
						|
  if (object !== undefined && object !== "") { | 
						|
    opcodeName = object + "." + name; | 
						|
  } | 
						|
 | 
						|
  var sign = _signatures.signatures[opcodeName]; | 
						|
 | 
						|
  if (sign == undefined) { | 
						|
    // TODO: Uncomment this when br_table and others has been done | 
						|
    //throw new Error("Invalid opcode: "+opcodeName); | 
						|
    return [object, object]; | 
						|
  } | 
						|
 | 
						|
  return sign[0]; | 
						|
} | 
						|
 | 
						|
function getUniqueNameGenerator() { | 
						|
  var inc = {}; | 
						|
  return function () { | 
						|
    var prefix = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : "temp"; | 
						|
 | 
						|
    if (!(prefix in inc)) { | 
						|
      inc[prefix] = 0; | 
						|
    } else { | 
						|
      inc[prefix] = inc[prefix] + 1; | 
						|
    } | 
						|
 | 
						|
    return prefix + "_" + inc[prefix]; | 
						|
  }; | 
						|
} | 
						|
 | 
						|
function getStartByteOffset(n) { | 
						|
  // $FlowIgnore | 
						|
  if (typeof n.loc === "undefined" || typeof n.loc.start === "undefined") { | 
						|
    throw new Error( // $FlowIgnore | 
						|
    "Can not get byte offset without loc informations, node: " + String(n.id)); | 
						|
  } | 
						|
 | 
						|
  return n.loc.start.column; | 
						|
} | 
						|
 | 
						|
function getEndByteOffset(n) { | 
						|
  // $FlowIgnore | 
						|
  if (typeof n.loc === "undefined" || typeof n.loc.end === "undefined") { | 
						|
    throw new Error("Can not get byte offset without loc informations, node: " + n.type); | 
						|
  } | 
						|
 | 
						|
  return n.loc.end.column; | 
						|
} | 
						|
 | 
						|
function getFunctionBeginingByteOffset(n) { | 
						|
  if (!(n.body.length > 0)) { | 
						|
    throw new Error('n.body.length > 0' + " error: " + (undefined || "unknown")); | 
						|
  } | 
						|
 | 
						|
  var _n$body = _slicedToArray(n.body, 1), | 
						|
      firstInstruction = _n$body[0]; | 
						|
 | 
						|
  return getStartByteOffset(firstInstruction); | 
						|
} | 
						|
 | 
						|
function getEndBlockByteOffset(n) { | 
						|
  // $FlowIgnore | 
						|
  if (!(n.instr.length > 0 || n.body.length > 0)) { | 
						|
    throw new Error('n.instr.length > 0 || n.body.length > 0' + " error: " + (undefined || "unknown")); | 
						|
  } | 
						|
 | 
						|
  var lastInstruction; | 
						|
 | 
						|
  if (n.instr) { | 
						|
    // $FlowIgnore | 
						|
    lastInstruction = n.instr[n.instr.length - 1]; | 
						|
  } | 
						|
 | 
						|
  if (n.body) { | 
						|
    // $FlowIgnore | 
						|
    lastInstruction = n.body[n.body.length - 1]; | 
						|
  } | 
						|
 | 
						|
  if (!(_typeof(lastInstruction) === "object")) { | 
						|
    throw new Error('typeof lastInstruction === "object"' + " error: " + (undefined || "unknown")); | 
						|
  } | 
						|
 | 
						|
  // $FlowIgnore | 
						|
  return getStartByteOffset(lastInstruction); | 
						|
} | 
						|
 | 
						|
function getStartBlockByteOffset(n) { | 
						|
  // $FlowIgnore | 
						|
  if (!(n.instr.length > 0 || n.body.length > 0)) { | 
						|
    throw new Error('n.instr.length > 0 || n.body.length > 0' + " error: " + (undefined || "unknown")); | 
						|
  } | 
						|
 | 
						|
  var fistInstruction; | 
						|
 | 
						|
  if (n.instr) { | 
						|
    // $FlowIgnore | 
						|
    var _n$instr = _slicedToArray(n.instr, 1); | 
						|
 | 
						|
    fistInstruction = _n$instr[0]; | 
						|
  } | 
						|
 | 
						|
  if (n.body) { | 
						|
    // $FlowIgnore | 
						|
    var _n$body2 = _slicedToArray(n.body, 1); | 
						|
 | 
						|
    fistInstruction = _n$body2[0]; | 
						|
  } | 
						|
 | 
						|
  if (!(_typeof(fistInstruction) === "object")) { | 
						|
    throw new Error('typeof fistInstruction === "object"' + " error: " + (undefined || "unknown")); | 
						|
  } | 
						|
 | 
						|
  // $FlowIgnore | 
						|
  return getStartByteOffset(fistInstruction); | 
						|
} |