You cannot select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
	
	
		
			123 lines
		
	
	
		
			4.7 KiB
		
	
	
	
		
			JavaScript
		
	
		
		
			
		
	
	
			123 lines
		
	
	
		
			4.7 KiB
		
	
	
	
		
			JavaScript
		
	
| 
											9 months ago
										 | "use strict"; | ||
|  | 
 | ||
|  | Object.defineProperty(exports, "__esModule", { | ||
|  |   value: true | ||
|  | }); | ||
|  | exports.createEmptySection = createEmptySection; | ||
|  | 
 | ||
|  | var _wasmGen = require("@webassemblyjs/wasm-gen"); | ||
|  | 
 | ||
|  | var _helperBuffer = require("@webassemblyjs/helper-buffer"); | ||
|  | 
 | ||
|  | var _helperWasmBytecode = _interopRequireDefault(require("@webassemblyjs/helper-wasm-bytecode")); | ||
|  | 
 | ||
|  | var t = _interopRequireWildcard(require("@webassemblyjs/ast")); | ||
|  | 
 | ||
|  | function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); } | ||
|  | 
 | ||
|  | function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || _typeof(obj) !== "object" && typeof obj !== "function") { return { "default": obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj["default"] = obj; if (cache) { cache.set(obj, newObj); } return newObj; } | ||
|  | 
 | ||
|  | function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; } | ||
|  | 
 | ||
|  | function _typeof(obj) { "@babel/helpers - typeof"; 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 findLastSection(ast, forSection) { | ||
|  |   var targetSectionId = _helperWasmBytecode["default"].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 = _helperWasmBytecode["default"].sections[section.section]; | ||
|  | 
 | ||
|  |     if (targetSectionId > lastId && targetSectionId < sectionId) { | ||
|  |       return lastSection; | ||
|  |     } | ||
|  | 
 | ||
|  |     lastId = sectionId; | ||
|  |     lastSection = section; | ||
|  |   } | ||
|  | 
 | ||
|  |   return lastSection; | ||
|  | } | ||
|  | 
 | ||
|  | 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 = (0, _wasmGen.encodeNode)(sectionMetadata); | ||
|  |   uint8Buffer = (0, _helperBuffer.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 | ||
|  |   }; | ||
|  | } |