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.
36 lines
1.3 KiB
36 lines
1.3 KiB
import { traverse, getSectionMetadatas, shiftSection } from "@webassemblyjs/ast"; |
|
import { overrideBytesInBuffer } from "@webassemblyjs/helper-buffer"; |
|
export function removeSections(ast, uint8Buffer, section) { |
|
var sectionMetadatas = getSectionMetadatas(ast, section); |
|
|
|
if (sectionMetadatas.length === 0) { |
|
throw new Error("Section metadata not found"); |
|
} |
|
|
|
return sectionMetadatas.reverse().reduce(function (uint8Buffer, sectionMetadata) { |
|
var startsIncludingId = sectionMetadata.startOffset - 1; |
|
var ends = section === "start" ? sectionMetadata.size.loc.end.column + 1 : sectionMetadata.startOffset + sectionMetadata.size.value + 1; |
|
var delta = -(ends - startsIncludingId); |
|
/** |
|
* update AST |
|
*/ |
|
// 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 path.remove(); |
|
} |
|
|
|
if (encounteredSection === true) { |
|
shiftSection(ast, path.node, delta); |
|
} |
|
} |
|
}); // replacement is nothing |
|
|
|
var replacement = []; |
|
return overrideBytesInBuffer(uint8Buffer, startsIncludingId, ends, replacement); |
|
}, uint8Buffer); |
|
} |