blob: 3150c9eccdbda11e1d5a0c7a7d742cd4d9f5ae51 [file] [log] [blame]
import { traverse, shiftSection } from "@webassemblyjs/ast";
import { encodeU32 } from "@webassemblyjs/wasm-gen/lib/encoder";
import { overrideBytesInBuffer } from "@webassemblyjs/helper-buffer";
function shiftFollowingSections(ast, _ref, deltaInSizeEncoding) {
var section = _ref.section;
// Once we hit our section every that is after needs to be shifted by the delta
var encounteredSection = false;
traverse(ast, {
SectionMetadata: function SectionMetadata(path) {
if (path.node.section === section) {
encounteredSection = true;
return;
}
if (encounteredSection === true) {
shiftSection(ast, path.node, deltaInSizeEncoding);
}
}
});
}
export function shrinkPaddedLEB128(ast, uint8Buffer) {
traverse(ast, {
SectionMetadata: function SectionMetadata(_ref2) {
var node = _ref2.node;
/**
* Section size
*/
{
var newu32Encoded = encodeU32(node.size.value);
var newu32EncodedLen = newu32Encoded.length;
var start = node.size.loc.start.column;
var end = node.size.loc.end.column;
var oldu32EncodedLen = end - start;
if (newu32EncodedLen !== oldu32EncodedLen) {
var deltaInSizeEncoding = oldu32EncodedLen - newu32EncodedLen;
uint8Buffer = overrideBytesInBuffer(uint8Buffer, start, end, newu32Encoded);
shiftFollowingSections(ast, node, -deltaInSizeEncoding);
}
}
}
});
return uint8Buffer;
}