377 lines
		
	
	
		
			9.4 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			377 lines
		
	
	
		
			9.4 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
 | |
| 
 | |
| function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } }
 | |
| 
 | |
| function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); return Constructor; }
 | |
| 
 | |
| import { isSignature, isNumberLiteral } from "../../nodes.js";
 | |
| export function moduleContextFromModuleAST(m) {
 | |
|   var moduleContext = new ModuleContext();
 | |
| 
 | |
|   if (!(m.type === "Module")) {
 | |
|     throw new Error('m.type === "Module"' + " error: " + (undefined || "unknown"));
 | |
|   }
 | |
| 
 | |
|   m.fields.forEach(function (field) {
 | |
|     switch (field.type) {
 | |
|       case "Start":
 | |
|         {
 | |
|           moduleContext.setStart(field.index);
 | |
|           break;
 | |
|         }
 | |
| 
 | |
|       case "TypeInstruction":
 | |
|         {
 | |
|           moduleContext.addType(field);
 | |
|           break;
 | |
|         }
 | |
| 
 | |
|       case "Func":
 | |
|         {
 | |
|           moduleContext.addFunction(field);
 | |
|           break;
 | |
|         }
 | |
| 
 | |
|       case "Global":
 | |
|         {
 | |
|           moduleContext.defineGlobal(field);
 | |
|           break;
 | |
|         }
 | |
| 
 | |
|       case "ModuleImport":
 | |
|         {
 | |
|           switch (field.descr.type) {
 | |
|             case "GlobalType":
 | |
|               {
 | |
|                 moduleContext.importGlobal(field.descr.valtype, field.descr.mutability);
 | |
|                 break;
 | |
|               }
 | |
| 
 | |
|             case "Memory":
 | |
|               {
 | |
|                 moduleContext.addMemory(field.descr.limits.min, field.descr.limits.max);
 | |
|                 break;
 | |
|               }
 | |
| 
 | |
|             case "FuncImportDescr":
 | |
|               {
 | |
|                 moduleContext.importFunction(field.descr);
 | |
|                 break;
 | |
|               }
 | |
| 
 | |
|             case "Table":
 | |
|               {
 | |
|                 // FIXME(sven): not implemented yet
 | |
|                 break;
 | |
|               }
 | |
| 
 | |
|             default:
 | |
|               throw new Error("Unsupported ModuleImport of type " + JSON.stringify(field.descr.type));
 | |
|           }
 | |
| 
 | |
|           break;
 | |
|         }
 | |
| 
 | |
|       case "Memory":
 | |
|         {
 | |
|           moduleContext.addMemory(field.limits.min, field.limits.max);
 | |
|           break;
 | |
|         }
 | |
|     }
 | |
|   });
 | |
|   return moduleContext;
 | |
| }
 | |
| /**
 | |
|  * Module context for type checking
 | |
|  */
 | |
| 
 | |
| export var ModuleContext = /*#__PURE__*/function () {
 | |
|   function ModuleContext() {
 | |
|     _classCallCheck(this, ModuleContext);
 | |
| 
 | |
|     this.funcs = [];
 | |
|     this.funcsOffsetByIdentifier = [];
 | |
|     this.types = [];
 | |
|     this.globals = [];
 | |
|     this.globalsOffsetByIdentifier = [];
 | |
|     this.mems = []; // Current stack frame
 | |
| 
 | |
|     this.locals = [];
 | |
|     this.labels = [];
 | |
|     this["return"] = [];
 | |
|     this.debugName = "unknown";
 | |
|     this.start = null;
 | |
|   }
 | |
|   /**
 | |
|    * Set start segment
 | |
|    */
 | |
| 
 | |
| 
 | |
|   _createClass(ModuleContext, [{
 | |
|     key: "setStart",
 | |
|     value: function setStart(index) {
 | |
|       this.start = index.value;
 | |
|     }
 | |
|     /**
 | |
|      * Get start function
 | |
|      */
 | |
| 
 | |
|   }, {
 | |
|     key: "getStart",
 | |
|     value: function getStart() {
 | |
|       return this.start;
 | |
|     }
 | |
|     /**
 | |
|      * Reset the active stack frame
 | |
|      */
 | |
| 
 | |
|   }, {
 | |
|     key: "newContext",
 | |
|     value: function newContext(debugName, expectedResult) {
 | |
|       this.locals = [];
 | |
|       this.labels = [expectedResult];
 | |
|       this["return"] = expectedResult;
 | |
|       this.debugName = debugName;
 | |
|     }
 | |
|     /**
 | |
|      * Functions
 | |
|      */
 | |
| 
 | |
|   }, {
 | |
|     key: "addFunction",
 | |
|     value: function addFunction(func) {
 | |
|       /* eslint-disable */
 | |
|       // $FlowIgnore
 | |
|       var _ref = func.signature || {},
 | |
|           _ref$params = _ref.params,
 | |
|           args = _ref$params === void 0 ? [] : _ref$params,
 | |
|           _ref$results = _ref.results,
 | |
|           result = _ref$results === void 0 ? [] : _ref$results;
 | |
|       /* eslint-enable */
 | |
| 
 | |
| 
 | |
|       args = args.map(function (arg) {
 | |
|         return arg.valtype;
 | |
|       });
 | |
|       this.funcs.push({
 | |
|         args: args,
 | |
|         result: result
 | |
|       });
 | |
| 
 | |
|       if (typeof func.name !== "undefined") {
 | |
|         // $FlowIgnore
 | |
|         this.funcsOffsetByIdentifier[func.name.value] = this.funcs.length - 1;
 | |
|       }
 | |
|     }
 | |
|   }, {
 | |
|     key: "importFunction",
 | |
|     value: function importFunction(funcimport) {
 | |
|       if (isSignature(funcimport.signature)) {
 | |
|         // eslint-disable-next-line prefer-const
 | |
|         var _funcimport$signature = funcimport.signature,
 | |
|             args = _funcimport$signature.params,
 | |
|             result = _funcimport$signature.results;
 | |
|         args = args.map(function (arg) {
 | |
|           return arg.valtype;
 | |
|         });
 | |
|         this.funcs.push({
 | |
|           args: args,
 | |
|           result: result
 | |
|         });
 | |
|       } else {
 | |
|         if (!isNumberLiteral(funcimport.signature)) {
 | |
|           throw new Error('isNumberLiteral(funcimport.signature)' + " error: " + (undefined || "unknown"));
 | |
|         }
 | |
| 
 | |
|         var typeId = funcimport.signature.value;
 | |
| 
 | |
|         if (!this.hasType(typeId)) {
 | |
|           throw new Error('this.hasType(typeId)' + " error: " + (undefined || "unknown"));
 | |
|         }
 | |
| 
 | |
|         var signature = this.getType(typeId);
 | |
|         this.funcs.push({
 | |
|           args: signature.params.map(function (arg) {
 | |
|             return arg.valtype;
 | |
|           }),
 | |
|           result: signature.results
 | |
|         });
 | |
|       }
 | |
| 
 | |
|       if (typeof funcimport.id !== "undefined") {
 | |
|         // imports are first, we can assume their index in the array
 | |
|         this.funcsOffsetByIdentifier[funcimport.id.value] = this.funcs.length - 1;
 | |
|       }
 | |
|     }
 | |
|   }, {
 | |
|     key: "hasFunction",
 | |
|     value: function hasFunction(index) {
 | |
|       return typeof this.getFunction(index) !== "undefined";
 | |
|     }
 | |
|   }, {
 | |
|     key: "getFunction",
 | |
|     value: function getFunction(index) {
 | |
|       if (typeof index !== "number") {
 | |
|         throw new Error("getFunction only supported for number index");
 | |
|       }
 | |
| 
 | |
|       return this.funcs[index];
 | |
|     }
 | |
|   }, {
 | |
|     key: "getFunctionOffsetByIdentifier",
 | |
|     value: function getFunctionOffsetByIdentifier(name) {
 | |
|       if (!(typeof name === "string")) {
 | |
|         throw new Error('typeof name === "string"' + " error: " + (undefined || "unknown"));
 | |
|       }
 | |
| 
 | |
|       return this.funcsOffsetByIdentifier[name];
 | |
|     }
 | |
|     /**
 | |
|      * Labels
 | |
|      */
 | |
| 
 | |
|   }, {
 | |
|     key: "addLabel",
 | |
|     value: function addLabel(result) {
 | |
|       this.labels.unshift(result);
 | |
|     }
 | |
|   }, {
 | |
|     key: "hasLabel",
 | |
|     value: function hasLabel(index) {
 | |
|       return this.labels.length > index && index >= 0;
 | |
|     }
 | |
|   }, {
 | |
|     key: "getLabel",
 | |
|     value: function getLabel(index) {
 | |
|       return this.labels[index];
 | |
|     }
 | |
|   }, {
 | |
|     key: "popLabel",
 | |
|     value: function popLabel() {
 | |
|       this.labels.shift();
 | |
|     }
 | |
|     /**
 | |
|      * Locals
 | |
|      */
 | |
| 
 | |
|   }, {
 | |
|     key: "hasLocal",
 | |
|     value: function hasLocal(index) {
 | |
|       return typeof this.getLocal(index) !== "undefined";
 | |
|     }
 | |
|   }, {
 | |
|     key: "getLocal",
 | |
|     value: function getLocal(index) {
 | |
|       return this.locals[index];
 | |
|     }
 | |
|   }, {
 | |
|     key: "addLocal",
 | |
|     value: function addLocal(type) {
 | |
|       this.locals.push(type);
 | |
|     }
 | |
|     /**
 | |
|      * Types
 | |
|      */
 | |
| 
 | |
|   }, {
 | |
|     key: "addType",
 | |
|     value: function addType(type) {
 | |
|       if (!(type.functype.type === "Signature")) {
 | |
|         throw new Error('type.functype.type === "Signature"' + " error: " + (undefined || "unknown"));
 | |
|       }
 | |
| 
 | |
|       this.types.push(type.functype);
 | |
|     }
 | |
|   }, {
 | |
|     key: "hasType",
 | |
|     value: function hasType(index) {
 | |
|       return this.types[index] !== undefined;
 | |
|     }
 | |
|   }, {
 | |
|     key: "getType",
 | |
|     value: function getType(index) {
 | |
|       return this.types[index];
 | |
|     }
 | |
|     /**
 | |
|      * Globals
 | |
|      */
 | |
| 
 | |
|   }, {
 | |
|     key: "hasGlobal",
 | |
|     value: function hasGlobal(index) {
 | |
|       return this.globals.length > index && index >= 0;
 | |
|     }
 | |
|   }, {
 | |
|     key: "getGlobal",
 | |
|     value: function getGlobal(index) {
 | |
|       return this.globals[index].type;
 | |
|     }
 | |
|   }, {
 | |
|     key: "getGlobalOffsetByIdentifier",
 | |
|     value: function getGlobalOffsetByIdentifier(name) {
 | |
|       if (!(typeof name === "string")) {
 | |
|         throw new Error('typeof name === "string"' + " error: " + (undefined || "unknown"));
 | |
|       }
 | |
| 
 | |
|       // $FlowIgnore
 | |
|       return this.globalsOffsetByIdentifier[name];
 | |
|     }
 | |
|   }, {
 | |
|     key: "defineGlobal",
 | |
|     value: function defineGlobal(global) {
 | |
|       var type = global.globalType.valtype;
 | |
|       var mutability = global.globalType.mutability;
 | |
|       this.globals.push({
 | |
|         type: type,
 | |
|         mutability: mutability
 | |
|       });
 | |
| 
 | |
|       if (typeof global.name !== "undefined") {
 | |
|         // $FlowIgnore
 | |
|         this.globalsOffsetByIdentifier[global.name.value] = this.globals.length - 1;
 | |
|       }
 | |
|     }
 | |
|   }, {
 | |
|     key: "importGlobal",
 | |
|     value: function importGlobal(type, mutability) {
 | |
|       this.globals.push({
 | |
|         type: type,
 | |
|         mutability: mutability
 | |
|       });
 | |
|     }
 | |
|   }, {
 | |
|     key: "isMutableGlobal",
 | |
|     value: function isMutableGlobal(index) {
 | |
|       return this.globals[index].mutability === "var";
 | |
|     }
 | |
|   }, {
 | |
|     key: "isImmutableGlobal",
 | |
|     value: function isImmutableGlobal(index) {
 | |
|       return this.globals[index].mutability === "const";
 | |
|     }
 | |
|     /**
 | |
|      * Memories
 | |
|      */
 | |
| 
 | |
|   }, {
 | |
|     key: "hasMemory",
 | |
|     value: function hasMemory(index) {
 | |
|       return this.mems.length > index && index >= 0;
 | |
|     }
 | |
|   }, {
 | |
|     key: "addMemory",
 | |
|     value: function addMemory(min, max) {
 | |
|       this.mems.push({
 | |
|         min: min,
 | |
|         max: max
 | |
|       });
 | |
|     }
 | |
|   }, {
 | |
|     key: "getMemory",
 | |
|     value: function getMemory(index) {
 | |
|       return this.mems[index];
 | |
|     }
 | |
|   }]);
 | |
| 
 | |
|   return ModuleContext;
 | |
| }(); |