Spaces:
Sleeping
Sleeping
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 { encodeNode } from "@webassemblyjs/wasm-gen"; | |
import { overrideBytesInBuffer } from "@webassemblyjs/helper-buffer"; | |
import constants from "@webassemblyjs/helper-wasm-bytecode"; | |
import * as t from "@webassemblyjs/ast"; | |
function findLastSection(ast, forSection) { | |
var targetSectionId = constants.sections[forSection]; // $FlowIgnore: metadata can not be empty | |
var moduleSections = ast.body[0].metadata.sections; | |
var lastSection; | |
var lastId = 0; | |
for (var i = 0, len = moduleSections.length; i < len; i++) { | |
var section = moduleSections[i]; // Ignore custom section since they can actually occur everywhere | |
if (section.section === "custom") { | |
continue; | |
} | |
var sectionId = constants.sections[section.section]; | |
if (targetSectionId > lastId && targetSectionId < sectionId) { | |
return lastSection; | |
} | |
lastId = sectionId; | |
lastSection = section; | |
} | |
return lastSection; | |
} | |
export function createEmptySection(ast, uint8Buffer, section) { | |
// previous section after which we are going to insert our section | |
var lastSection = findLastSection(ast, section); | |
var start, end; | |
/** | |
* It's the first section | |
*/ | |
if (lastSection == null || lastSection.section === "custom") { | |
start = 8 | |
/* wasm header size */ | |
; | |
end = start; | |
} else { | |
start = lastSection.startOffset + lastSection.size.value + 1; | |
end = start; | |
} // section id | |
start += 1; | |
var sizeStartLoc = { | |
line: -1, | |
column: start | |
}; | |
var sizeEndLoc = { | |
line: -1, | |
column: start + 1 | |
}; // 1 byte for the empty vector | |
var size = t.withLoc(t.numberLiteralFromRaw(1), sizeEndLoc, sizeStartLoc); | |
var vectorOfSizeStartLoc = { | |
line: -1, | |
column: sizeEndLoc.column | |
}; | |
var vectorOfSizeEndLoc = { | |
line: -1, | |
column: sizeEndLoc.column + 1 | |
}; | |
var vectorOfSize = t.withLoc(t.numberLiteralFromRaw(0), vectorOfSizeEndLoc, vectorOfSizeStartLoc); | |
var sectionMetadata = t.sectionMetadata(section, start, size, vectorOfSize); | |
var sectionBytes = encodeNode(sectionMetadata); | |
uint8Buffer = overrideBytesInBuffer(uint8Buffer, start - 1, end, sectionBytes); // Add section into the AST for later lookups | |
if (_typeof(ast.body[0].metadata) === "object") { | |
// $FlowIgnore: metadata can not be empty | |
ast.body[0].metadata.sections.push(sectionMetadata); | |
t.sortSectionMetadata(ast.body[0]); | |
} | |
/** | |
* Update AST | |
*/ | |
// Once we hit our section every that is after needs to be shifted by the delta | |
var deltaBytes = +sectionBytes.length; | |
var encounteredSection = false; | |
t.traverse(ast, { | |
SectionMetadata: function SectionMetadata(path) { | |
if (path.node.section === section) { | |
encounteredSection = true; | |
return; | |
} | |
if (encounteredSection === true) { | |
t.shiftSection(ast, path.node, deltaBytes); | |
} | |
} | |
}); | |
return { | |
uint8Buffer: uint8Buffer, | |
sectionMetadata: sectionMetadata | |
}; | |
} |