/* * Copyright (C)2008-2017 Haxe Foundation * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associated documentation files (the "Software"), * to deal in the Software without restriction, including without limitation * the rights to use, copy, modify, merge, publish, distribute, sublicense, * and/or sell copies of the Software, and to permit persons to whom the * Software is furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in * all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER * DEALINGS IN THE SOFTWARE. */ /* * YoshiCrafter Engine fixes: * - Added Error handler * - Added Imports * - Added @:bypassAccessor */ package hscript; import haxe.iterators.StringKeyValueIteratorUnicode; import haxe.EnumTools; import haxe.display.Protocol.InitializeResult; import haxe.PosInfos; import hscript.Expr; import haxe.Constraints.IMap; using StringTools; private enum Stop { SBreak; SContinue; SReturn; } class Interp { public var scriptObject(default, set):Dynamic; public function set_scriptObject(v:Dynamic) { __instanceFields = (v == null) ? [] : Type.getInstanceFields(Type.getClass(v)); return scriptObject = v; } public var errorHandler:Error->Void; #if haxe3 public var variables:Map; public var publicVariables:Map; public var staticVariables:Map; var locals:Map; var binops:MapExpr->Dynamic>; #else public var variables:Hash; public var publicVariables:Hash; public var staticVariables:Hash; var locals:Hash<{r:Dynamic, depth:Int}>; var binops:HashExpr->Dynamic>; #end var depth:Int = 0; var inTry:Bool; var declared:Array<{n:String, old:{r:Dynamic, depth:Int}, depth:Int}>; var returnValue:Dynamic; var isBypassAccessor:Bool = false; public var importEnabled:Bool = true; public var allowStaticVariables:Bool = false; public var allowPublicVariables:Bool = false; public var importBlocklist:Array = [ // "flixel.FlxG" ]; var __instanceFields:Array = []; #if hscriptPos var curExpr:Expr; #end public function new() { #if haxe3 locals = new Map(); #else locals = new Hash(); #end declared = new Array(); resetVariables(); initOps(); } private function resetVariables() { #if haxe3 variables = new Map(); publicVariables = new Map(); staticVariables = new Map(); #else variables = new Hash(); publicVariables = new Hash(); staticVariables = new Hash(); #end variables.set("null", null); variables.set("true", true); variables.set("false", false); variables.set("trace", Reflect.makeVarArgs(function(el) { var inf = posInfos(); var v = el.shift(); if (el.length > 0) inf.customParams = el; haxe.Log.trace(Std.string(v), inf); })); } public function posInfos():PosInfos { #if hscriptPos if (curExpr != null) return cast {fileName: curExpr.origin, lineNumber: curExpr.line}; #end return cast {fileName: "hscript", lineNumber: 0}; } function initOps() { var me = this; #if haxe3 binops = new Map(); #else binops = new Hash(); #end binops.set("+", function(e1, e2) return me.expr(e1) + me.expr(e2)); binops.set("-", function(e1, e2) return me.expr(e1) - me.expr(e2)); binops.set("*", function(e1, e2) return me.expr(e1) * me.expr(e2)); binops.set("/", function(e1, e2) return me.expr(e1) / me.expr(e2)); binops.set("%", function(e1, e2) return me.expr(e1) % me.expr(e2)); binops.set("&", function(e1, e2) return me.expr(e1) & me.expr(e2)); binops.set("|", function(e1, e2) return me.expr(e1) | me.expr(e2)); binops.set("^", function(e1, e2) return me.expr(e1) ^ me.expr(e2)); binops.set("<<", function(e1, e2) return me.expr(e1) << me.expr(e2)); binops.set(">>", function(e1, e2) return me.expr(e1) >> me.expr(e2)); binops.set(">>>", function(e1, e2) return me.expr(e1) >>> me.expr(e2)); binops.set("==", function(e1, e2) return me.expr(e1) == me.expr(e2)); binops.set("!=", function(e1, e2) return me.expr(e1) != me.expr(e2)); binops.set(">=", function(e1, e2) return me.expr(e1) >= me.expr(e2)); binops.set("<=", function(e1, e2) return me.expr(e1) <= me.expr(e2)); binops.set(">", function(e1, e2) return me.expr(e1) > me.expr(e2)); binops.set("<", function(e1, e2) return me.expr(e1) < me.expr(e2)); binops.set("||", function(e1, e2) return me.expr(e1) == true || me.expr(e2) == true); binops.set("&&", function(e1, e2) return me.expr(e1) == true && me.expr(e2) == true); binops.set("=", assign); binops.set("...", function(e1, e2) return new #if (haxe_211 || haxe3) IntIterator #else IntIter #end(me.expr(e1), me.expr(e2))); assignOp("+=", function(v1:Dynamic, v2:Dynamic) return v1 + v2); assignOp("-=", function(v1:Float, v2:Float) return v1 - v2); assignOp("*=", function(v1:Float, v2:Float) return v1 * v2); assignOp("/=", function(v1:Float, v2:Float) return v1 / v2); assignOp("%=", function(v1:Float, v2:Float) return v1 % v2); assignOp("&=", function(v1, v2) return v1 & v2); assignOp("|=", function(v1, v2) return v1 | v2); assignOp("^=", function(v1, v2) return v1 ^ v2); assignOp("<<=", function(v1, v2) return v1 << v2); assignOp(">>=", function(v1, v2) return v1 >> v2); assignOp(">>>=", function(v1, v2) return v1 >>> v2); } function setVar(name:String, v:Dynamic) { if (allowStaticVariables && staticVariables.exists(name)) staticVariables.set(name, v); else if (allowPublicVariables && publicVariables.exists(name)) publicVariables.set(name, v); else variables.set(name, v); } function assign(e1:Expr, e2:Expr):Dynamic { var v = expr(e2); switch (Tools.expr(e1)) { case EIdent(id): var l = locals.get(id); if (l == null) { if (!variables.exists(id) && !staticVariables.exists(id) && !publicVariables.exists(id) && scriptObject != null) { if (Type.typeof(scriptObject) == TObject) { Reflect.setField(scriptObject, id, v); } else { if (__instanceFields.contains(id)) { Reflect.setProperty(scriptObject, id, v); } else if (__instanceFields.contains('set_$id')) { // setter Reflect.getProperty(scriptObject, 'set_$id')(v); } else { setVar(id, v); } } } else { setVar(id, v); } } else { l.r = v; if (l.depth == 0) { setVar(id, v); } } // TODO case EField(e, f): v = set(expr(e), f, v); case EArray(e, index): var arr:Dynamic = expr(e); var index:Dynamic = expr(index); if (isMap(arr)) { setMapValue(arr, index, v); } else { arr[index] = v; } default: error(EInvalidOp("=")); } return v; } function assignOp(op, fop:Dynamic->Dynamic->Dynamic) { var me = this; binops.set(op, function(e1, e2) return me.evalAssignOp(op, fop, e1, e2)); } function evalAssignOp(op, fop, e1, e2):Dynamic { var v; switch (Tools.expr(e1)) { case EIdent(id): var l = locals.get(id); v = fop(expr(e1), expr(e2)); if (l == null) { if (__instanceFields.contains(id)) { Reflect.setProperty(scriptObject, id, v); } else if (__instanceFields.contains('set_$id')) { // setter Reflect.getProperty(scriptObject, 'set_$id')(v); } else { setVar(id, v); } } else l.r = v; case EField(e, f): var obj = expr(e); v = fop(get(obj, f), expr(e2)); v = set(obj, f, v); case EArray(e, index): var arr:Dynamic = expr(e); var index:Dynamic = expr(index); if (isMap(arr)) { v = fop(getMapValue(arr, index), expr(e2)); setMapValue(arr, index, v); } else { v = fop(arr[index], expr(e2)); arr[index] = v; } default: return error(EInvalidOp(op)); } return v; } function increment(e:Expr, prefix:Bool, delta:Int):Dynamic { #if hscriptPos curExpr = e; var e = e.e; #end switch (e) { case EIdent(id): var l = locals.get(id); var v:Dynamic = (l == null) ? resolve(id) : l.r; if (prefix) { v += delta; if (l == null) setVar(id, v) else l.r = v; } else if (l == null) setVar(id, v + delta) else l.r = v + delta; return v; case EField(e, f): var obj = expr(e); var v:Dynamic = get(obj, f); if (prefix) { v += delta; set(obj, f, v); } else set(obj, f, v + delta); return v; case EArray(e, index): var arr:Dynamic = expr(e); var index:Dynamic = expr(index); if (isMap(arr)) { var v = getMapValue(arr, index); if (prefix) { v += delta; setMapValue(arr, index, v); } else { setMapValue(arr, index, v + delta); } return v; } else { var v = arr[index]; if (prefix) { v += delta; arr[index] = v; } else arr[index] = v + delta; return v; } default: return error(EInvalidOp((delta > 0) ? "++" : "--")); } } public function execute(expr:Expr):Dynamic { depth = 0; #if haxe3 locals = new Map(); #else locals = new Hash(); #end declared = new Array(); return exprReturn(expr); } function exprReturn(e):Dynamic { try { try { return expr(e); } catch (e:Stop) { switch (e) { case SBreak: throw "Invalid break"; case SContinue: throw "Invalid continue"; case SReturn: var v = returnValue; returnValue = null; return v; } } catch(e) { error(ECustom('${e.toString()}')); return null; } } catch(e:Error) { if (errorHandler != null) errorHandler(e); else throw e; return null; } catch(e) { trace(e); } return null; } function duplicate(h:#if haxe3 Map #else Hash #end) { #if haxe3 var h2 = new Map(); #else var h2 = new Hash(); #end for (k in h.keys()) h2.set(k, h.get(k)); return h2; } function restore(old:Int) { while (declared.length > old) { var d = declared.pop(); locals.set(d.n, d.old); } } inline function error(e:#if hscriptPos ErrorDef #else Error #end, rethrow = false):Dynamic { #if hscriptPos var e = new Error(e, curExpr.pmin, curExpr.pmax, curExpr.origin, curExpr.line); #end if (rethrow) { this.rethrow(e); } else { throw e; } return null; } inline function rethrow(e:Dynamic) { #if hl hl.Api.rethrow(e); #else throw e; #end } function resolve(id:String, doException:Bool = true):Dynamic { var l = locals.get(id); if (l != null) return l.r; var v = variables.get(id); // !publicVariables.exists(id) if (staticVariables.exists(id)) { return staticVariables.get(id); } else if (publicVariables.exists(id)) { return publicVariables.get(id); } else if (variables.exists(id)) { return variables.get(id); } else { if (scriptObject != null) { // search in object if (id == "this") { return scriptObject; } else if ((Type.typeof(scriptObject) == TObject) && Reflect.hasField(scriptObject, id)) { return Reflect.field(scriptObject, id); } else { if (__instanceFields.contains(id)) { return Reflect.getProperty(scriptObject, id); } else if (__instanceFields.contains('get_$id')) { // getter return Reflect.getProperty(scriptObject, 'get_$id')(); } } } if (doException) error(EUnknownVariable(id)); } return v; } public function expr(e:Expr):Dynamic { #if hscriptPos curExpr = e; var e = e.e; #end switch (e) { case EImport(c): if (!importEnabled) return null; var splitClassName = [for (e in c.split(".")) e.trim()]; var realClassName = splitClassName.join("."); if (importBlocklist.contains(realClassName)) return null; var cl = Type.resolveClass(realClassName); if (cl == null) cl = Type.resolveClass('${realClassName}_HSC'); var en = Type.resolveEnum(realClassName); if (cl == null && en == null) { // LogsOverlay.trace('Class / Enum at $realClassName does not exist.'); error(EInvalidClass(realClassName)); } else { if (en != null) { // ENUM!!!! var enumThingy = {}; for (c in en.getConstructors()) { try { Reflect.setField(enumThingy, c, en.createByName(c)); } catch(e) { try { Reflect.setField(enumThingy, c, Reflect.field(en, c)); } catch(ex) { throw e; } } } variables.set(splitClassName[splitClassName.length - 1], enumThingy); } else { variables.set(splitClassName[splitClassName.length - 1], cl); } } return null; case EConst(c): switch (c) { case CInt(v): return v; case CFloat(f): return f; case CString(s): return s; #if !haxe3 case CInt32(v): return v; #end } case EIdent(id): return resolve(id); case EVar(n, _, e, isPublic, isStatic): declared.push({n: n, old: locals.get(n), depth: depth}); locals.set(n, {r: (e == null) ? null : expr(e), depth: depth}); if (depth == 0) (isStatic == true ? staticVariables : (isPublic ? publicVariables : variables)).set(n, locals[n].r); return null; case EParent(e): return expr(e); case EBlock(exprs): var old = declared.length; var v = null; for (e in exprs) v = expr(e); restore(old); return v; case EField(e, f): return get(expr(e), f); case EBinop(op, e1, e2): var fop = binops.get(op); if (fop == null) error(EInvalidOp(op)); return fop(e1, e2); case EUnop(op, prefix, e): switch (op) { case "!": return expr(e) != true; case "-": return -expr(e); case "++": return increment(e, prefix, 1); case "--": return increment(e, prefix, -1); case "~": #if (neko && !haxe3) return haxe.Int32.complement(expr(e)); #else return ~expr(e); #end default: error(EInvalidOp(op)); } case ECall(e, params): var args = new Array(); for (p in params) args.push(expr(p)); switch (Tools.expr(e)) { case EField(e, f): var obj = expr(e); if (obj == null) error(EInvalidAccess(f)); return fcall(obj, f, args); default: return call(null, expr(e), args); } case EIf(econd, e1, e2): return if (expr(econd) == true) expr(e1) else if (e2 == null) null else expr(e2); case EWhile(econd, e): whileLoop(econd, e); return null; case EDoWhile(econd, e): doWhileLoop(econd, e); return null; case EFor(v, it, e): forLoop(v, it, e); return null; case EBreak: throw SBreak; case EContinue: throw SContinue; case EReturn(e): returnValue = e == null ? null : expr(e); throw SReturn; case EFunction(params, fexpr, name, _, isPublic, isStatic): var __capturedLocals = duplicate(locals); var capturedLocals:Map = []; for(k=>e in __capturedLocals) if (e != null && e.depth > 0) capturedLocals.set(k, e); var me = this; var hasOpt = false, minParams = 0; for (p in params) if (p.opt) hasOpt = true; else minParams++; var f = function(args:Array) { if (((args == null) ? 0 : args.length) != params.length) { if (args.length < minParams) { var str = "Invalid number of parameters. Got " + args.length + ", required " + minParams; if (name != null) str += " for function '" + name + "'"; error(ECustom(str)); } // make sure mandatory args are forced var args2 = []; var extraParams = args.length - minParams; var pos = 0; for (p in params) if (p.opt) { if (extraParams > 0) { args2.push(args[pos++]); extraParams--; } else args2.push(null); } else args2.push(args[pos++]); args = args2; } var old = me.locals, depth = me.depth; me.depth++; me.locals = me.duplicate(capturedLocals); for (i in 0...params.length) me.locals.set(params[i].name, {r: args[i], depth: depth}); var r = null; var oldDecl = declared.length; if (inTry) try { r = me.exprReturn(fexpr); } catch (e:Dynamic) { me.locals = old; me.depth = depth; #if neko neko.Lib.rethrow(e); #else throw e; #end } else r = me.exprReturn(fexpr); restore(oldDecl); me.locals = old; me.depth = depth; return r; }; var f = Reflect.makeVarArgs(f); if (name != null) { if (depth == 0) { // global function ((isStatic && allowStaticVariables) ? staticVariables : ((isPublic && allowPublicVariables) ? publicVariables : variables)).set(name, f); } else { // function-in-function is a local function declared.push({n: name, old: locals.get(name), depth: depth}); var ref = {r: f, depth: depth}; locals.set(name, ref); capturedLocals.set(name, ref); // allow self-recursion } } return f; case EArrayDecl(arr, wantedType): var isMap = false; var isTypeMap = false; if(!isMap && wantedType != null) { isMap = wantedType.match(CTPath(["Map"], [_, _])); isTypeMap = true; } else { isMap = arr.length > 0 && Tools.expr(arr[0]).match(EBinop("=>", _)); } if (isMap) { var isAllString:Bool = true; var isAllInt:Bool = true; var isAllObject:Bool = true; var isAllEnum:Bool = true; var keys:Array = []; var values:Array = []; for (e in arr) { switch (Tools.expr(e)) { case EBinop("=>", eKey, eValue): { var key:Dynamic = expr(eKey); var value:Dynamic = expr(eValue); isAllString = isAllString && (key is String); isAllInt = isAllInt && (key is Int); isAllObject = isAllObject && Reflect.isObject(key); isAllEnum = isAllEnum && Reflect.isEnumValue(key); keys.push(key); values.push(value); } default: throw("=> expected"); } } if(isTypeMap) { if(wantedType != null) { isAllString = wantedType.match(CTPath(["Map"], [CTPath(["String"], _), _])); isAllInt = wantedType.match(CTPath(["Map"], [CTPath(["Int"], _), _])); if(isAllString || isAllInt) { isAllObject = false; isAllEnum = false; } else { if(!isAllObject && !isAllEnum) { throw("Unknown Type Key"); } } } } var map:Dynamic = { if (isAllInt) new haxe.ds.IntMap(); else if (isAllString) new haxe.ds.StringMap(); else if (isAllEnum) new haxe.ds.EnumValueMap(); else if (isAllObject) new haxe.ds.ObjectMap(); else throw 'Inconsistent key types'; } for (n in 0...keys.length) { setMapValue(map, keys[n], values[n]); } return map; } else { var a = new Array(); for (e in arr) { a.push(expr(e)); } return a; } case EArray(e, index): var arr:Dynamic = expr(e); var index:Dynamic = expr(index); if (isMap(arr)) { return getMapValue(arr, index); } else { return arr[index]; } case ENew(cl, params): var a = new Array(); for (e in params) a.push(expr(e)); return cnew(cl, a); case EThrow(e): throw expr(e); case ETry(e, n, _, ecatch): var old = declared.length; var oldTry = inTry; try { inTry = true; var v:Dynamic = expr(e); restore(old); inTry = oldTry; return v; } catch (err:Stop) { inTry = oldTry; throw err; } catch (err:Dynamic) { // restore vars restore(old); inTry = oldTry; // declare 'v' declared.push({n: n, old: locals.get(n), depth: depth}); locals.set(n, {r: err, depth: depth}); var v:Dynamic = expr(ecatch); restore(old); return v; } case EObject(fl): var o = {}; for (f in fl) set(o, f.name, expr(f.e)); return o; case ETernary(econd, e1, e2): return if (expr(econd) == true) expr(e1) else expr(e2); case ESwitch(e, cases, def): var val:Dynamic = expr(e); var match = false; for (c in cases) { for (v in c.values) if (expr(v) == val) { match = true; break; } if (match) { val = expr(c.expr); break; } } if (!match) val = def == null ? null : expr(def); return val; case EMeta(a, b, e): var oldAccessor = isBypassAccessor; if(a == ":bypassAccessor") { isBypassAccessor = true; } var val = expr(e); isBypassAccessor = oldAccessor; return val; case ECheckType(e, _): return expr(e); } return null; } function doWhileLoop(econd, e) { var old = declared.length; do { try { expr(e); } catch (err:Stop) { switch (err) { case SContinue: case SBreak: break; case SReturn: throw err; } } } while (expr(econd) == true); restore(old); } function whileLoop(econd, e) { var old = declared.length; while (expr(econd) == true) { try { expr(e); } catch (err:Stop) { switch (err) { case SContinue: case SBreak: break; case SReturn: throw err; } } } restore(old); } function makeIterator(v:Dynamic):Iterator { #if ((flash && !flash9) || (php && !php7 && haxe_ver < '4.0.0')) if (v.iterator != null) v = v.iterator(); #else try v = v.iterator() catch (e:Dynamic) {}; #end if (v.hasNext == null || v.next == null) error(EInvalidIterator(v)); return v; } function forLoop(n, it, e) { var old = declared.length; declared.push({n: n, old: locals.get(n), depth: depth}); var it = makeIterator(expr(it)); while (it.hasNext()) { locals.set(n, {r: it.next(), depth: depth}); try { expr(e); } catch (err:Stop) { switch (err) { case SContinue: case SBreak: break; case SReturn: throw err; } } } restore(old); } inline function isMap(o:Dynamic):Bool { return (o is IMap); } inline function getMapValue(map:Dynamic, key:Dynamic):Dynamic { return cast(map, haxe.Constraints.IMap).get(key); } inline function setMapValue(map:Dynamic, key:Dynamic, value:Dynamic):Void { cast(map, haxe.Constraints.IMap).set(key, value); } public static var getRedirects:MapString->Dynamic> = []; public static var setRedirects:MapString->Dynamic->Dynamic> = []; function get(o:Dynamic, f:String):Dynamic { if (o == null) error(EInvalidAccess(f)); return { var redirect:Dynamic->String->Dynamic = null; var cl:String = switch (Type.typeof(o)) { case TNull: "Null"; case TInt: "Int"; case TFloat: "Float"; case TBool: "Bool"; case _: null; }; if (getRedirects.exists(cl = Type.getClassName(Type.getClass(o))) && (redirect = getRedirects[cl]) != null) { return redirect(o, f); } else if (o is IHScriptCustomBehaviour) { var obj = cast(o, IHScriptCustomBehaviour); return obj.hget(f); } else { var v = null; if(isBypassAccessor) { if ((v = Reflect.field(o, f)) == null) v = Reflect.field(Type.getClass(o), f); } if(v == null) { if ((v = Reflect.getProperty(o, f)) == null) v = Reflect.getProperty(Type.getClass(o), f); } return v; } } } function set(o:Dynamic, f:String, v:Dynamic):Dynamic { if (o == null) error(EInvalidAccess(f)); var redirect:Dynamic->String->Dynamic->Dynamic = null; var cl:String = switch (Type.typeof(o)) { case TNull: "Null"; case TInt: "Int"; case TFloat: "Float"; case TBool: "Bool"; case _: null; }; if (setRedirects.exists(cl = Type.getClassName(Type.getClass(o))) && (redirect = setRedirects[cl]) != null) return redirect(o, f, v); else if (o is IHScriptCustomBehaviour) { var obj = cast(o, IHScriptCustomBehaviour); return obj.hset(f, v); } if(isBypassAccessor) { Reflect.setField(o, f, v); } else { Reflect.setProperty(o, f, v); } return v; } function fcall(o:Dynamic, f:String, args:Array):Dynamic { return call(o, get(o, f), args); } function call(o:Dynamic, f:Dynamic, args:Array):Dynamic { return Reflect.callMethod(o, f, args); } function cnew(cl:String, args:Array):Dynamic { var c = Type.resolveClass(cl); if (c == null) c = resolve(cl); return Type.createInstance(c, args); } }