forked from zhurui/management
180 lines
4.7 KiB
JavaScript
180 lines
4.7 KiB
JavaScript
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); }
|
|
|
|
import { signatures } from "./signatures";
|
|
import { traverse } from "./traverse";
|
|
import constants from "@webassemblyjs/helper-wasm-bytecode";
|
|
import { getSectionForNode } from "@webassemblyjs/helper-wasm-bytecode";
|
|
export function isAnonymous(ident) {
|
|
return ident.raw === "";
|
|
}
|
|
export function getSectionMetadata(ast, name) {
|
|
var section;
|
|
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;
|
|
}
|
|
export function getSectionMetadatas(ast, name) {
|
|
var sections = [];
|
|
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;
|
|
}
|
|
export 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 = constants.sections[a.section];
|
|
var bId = constants.sections[b.section];
|
|
|
|
if (typeof aId !== "number" || typeof bId !== "number") {
|
|
throw new Error("Section id not found");
|
|
}
|
|
|
|
return aId - bId;
|
|
});
|
|
}
|
|
export 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);
|
|
}
|
|
}
|
|
export 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"));
|
|
}
|
|
}
|
|
export function getEndOfSection(s) {
|
|
assertHasLoc(s.size);
|
|
return s.startOffset + s.size.value + ( // $FlowIgnore
|
|
s.size.loc.end.column - s.size.loc.start.column);
|
|
}
|
|
export function shiftLoc(node, delta) {
|
|
// $FlowIgnore
|
|
node.loc.start.column += delta; // $FlowIgnore
|
|
|
|
node.loc.end.column += delta;
|
|
}
|
|
export 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
|
|
|
|
traverse(ast, {
|
|
Node: function Node(_ref3) {
|
|
var node = _ref3.node;
|
|
var section = getSectionForNode(node);
|
|
|
|
if (section === sectionName && _typeof(node.loc) === "object") {
|
|
shiftLoc(node, delta);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
export function signatureForOpcode(object, name) {
|
|
var opcodeName = name;
|
|
|
|
if (object !== undefined && object !== "") {
|
|
opcodeName = object + "." + name;
|
|
}
|
|
|
|
var sign = 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];
|
|
}
|
|
export 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];
|
|
};
|
|
} |