rt/vm/src/main/java/org/apidesign/vm4brwsr/VMLazy.java
author Lubomir Nerad <lubomir.nerad@oracle.com>
Thu, 21 Mar 2013 18:48:46 +0100
branchclosure
changeset 869 151f4ccd7673
parent 829 0e4945540961
child 1020 a6bacea2518f
child 1282 8d29792a09c6
permissions -rw-r--r--
Initial attempt for advanced obfuscation
     1 /**
     2  * Back 2 Browser Bytecode Translator
     3  * Copyright (C) 2012 Jaroslav Tulach <jaroslav.tulach@apidesign.org>
     4  *
     5  * This program is free software: you can redistribute it and/or modify
     6  * it under the terms of the GNU General Public License as published by
     7  * the Free Software Foundation, version 2 of the License.
     8  *
     9  * This program is distributed in the hope that it will be useful,
    10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    12  * GNU General Public License for more details.
    13  *
    14  * You should have received a copy of the GNU General Public License
    15  * along with this program. Look for COPYING file in the top folder.
    16  * If not, see http://opensource.org/licenses/GPL-2.0.
    17  */
    18 package org.apidesign.vm4brwsr;
    19 
    20 import java.io.ByteArrayInputStream;
    21 import java.io.IOException;
    22 import java.io.InputStream;
    23 import java.lang.reflect.Array;
    24 import org.apidesign.bck2brwsr.core.JavaScriptBody;
    25 
    26 /**
    27  *
    28  * @author Jaroslav Tulach <jtulach@netbeans.org>
    29  */
    30 final class VMLazy {
    31     private final Object loader;
    32     private final Object[] args;
    33     
    34     private VMLazy(Object loader, Object[] args) {
    35         this.loader = loader;
    36         this.args = args;
    37     }
    38     
    39     static void init() {
    40     }
    41     
    42     static Object load(Object loader, String name, Object[] arguments) 
    43     throws IOException, ClassNotFoundException {
    44         return new VMLazy(loader, arguments).load(name, false);
    45     }
    46     
    47     static byte[] loadBytes(Object loader, String name, Object[] arguments) throws Exception {
    48         return Zips.loadFromCp(arguments, name);
    49     }
    50     
    51     private Object load(String name, boolean instance)
    52     throws IOException, ClassNotFoundException {
    53         String res = name.replace('.', '/') + ".class";
    54         byte[] arr = Zips.loadFromCp(args, res);
    55         if (arr == null) {
    56             throw new ClassNotFoundException(name);
    57         }
    58 //        beingDefined(loader, name);
    59         StringBuilder out = new StringBuilder(65535);
    60         out.append("var loader = arguments[0];\n");
    61         out.append("var vm = loader.vm;\n");
    62         int prelude = out.length();
    63         String initCode = new Gen(this, out).compile(new ByteArrayInputStream(arr));
    64         String code = out.toString().toString();
    65 //        dump("Loading " + name);
    66         dump(code);
    67         String under = name.replace('.', '_');
    68         Object fn = applyCode(loader, under, code, instance);
    69         
    70         if (!initCode.isEmpty()) {
    71             out.setLength(prelude);
    72             out.append(initCode);
    73             code = out.toString().toString();
    74             dump(code);
    75             applyCode(loader, null, code, false);
    76         }            
    77         
    78         return fn;
    79     }
    80 
    81 //    @JavaScriptBody(args = "s", body = "java.lang.System.out.println(s.toString());")
    82     static void dump(String s) {
    83     }
    84 
    85 /* possibly not needed:
    86     @JavaScriptBody(args = {"loader", "n" }, body =
    87         "var cls = n.replace__Ljava_lang_String_2CC(n, '.','_').toString();" +
    88         "loader.vm[cls] = true;\n"
    89     )
    90     private static native void beingDefined(Object loader, String name);
    91 */
    92     
    93 
    94     @JavaScriptBody(args = {"loader", "name", "script", "instance" }, body =
    95         "try {\n" +
    96         "  new Function(script)(loader, name);\n" +
    97         "} catch (ex) {\n" +
    98         "  throw 'Cannot compile ' + name + ' ' + ex + ' line: ' + ex.lineNumber + ' script:\\n' + script;\n" +
    99         "}\n" +
   100         "return name != null ? vm[name](instance) : null;\n"
   101     )
   102     private static native Object applyCode(Object loader, String name, String script, boolean instance);
   103     
   104     
   105     private static final class Gen extends ByteCodeToJavaScript {
   106         private final VMLazy lazy;
   107 
   108         public Gen(VMLazy vm, Appendable out) {
   109             super(out);
   110             this.lazy = vm;
   111         }
   112         
   113         @JavaScriptBody(args = {"n"},
   114         body =
   115         "var cls = n.replace__Ljava_lang_String_2CC('/','_').toString();"
   116         + "\nvar dot = n.replace__Ljava_lang_String_2CC('/','.').toString();"
   117         + "\nvar lazy = this._lazy();"
   118         + "\nvar loader = lazy._loader();"
   119         + "\nvar vm = loader.vm;"
   120         + "\nif (vm[cls]) return false;"
   121         + "\nvm[cls] = function() {"
   122         + "\n  var instance = arguments.length == 0 || arguments[0] === true;"
   123         + "\n  return lazy.load__Ljava_lang_Object_2Ljava_lang_String_2Z(dot, instance);"
   124         + "\n};"
   125         + "\nreturn true;")
   126         @Override
   127         protected boolean requireReference(String internalClassName) {
   128             throw new UnsupportedOperationException();
   129         }
   130 
   131         @Override
   132         protected void requireScript(String resourcePath) throws IOException {
   133             InputStream is = getClass().getResourceAsStream(resourcePath);
   134             StringBuilder sb = new StringBuilder();
   135             for (;;) {
   136                 int ch = is.read();
   137                 if (ch == -1) {
   138                     break;
   139                 }
   140                 sb.append((char)ch);
   141             }
   142             applyCode(lazy.loader, null, sb.toString(), false);
   143         }
   144 
   145         @Override
   146         String assignClass(String className) {
   147             return "vm[arguments[1]]=";
   148         }
   149 
   150         @Override
   151         String accessClass(String classOperation) {
   152             return "vm." + classOperation;
   153         }
   154 
   155         @Override
   156         String getVMObject() {
   157             return "vm";
   158         }
   159     }
   160 }