boot/src/main/java/org/apidesign/html/boot/impl/JsClassLoader.java
author Jaroslav Tulach <jaroslav.tulach@apidesign.org>
Tue, 25 Jun 2013 09:27:40 +0200
branchclassloader
changeset 153 16a9e0101224
parent 143 b3b56bed079e
child 160 b5a06e2b3b92
permissions -rw-r--r--
Can return boolean value
jaroslav@123
     1
/**
jaroslav@123
     2
 * HTML via Java(tm) Language Bindings
jaroslav@123
     3
 * Copyright (C) 2013 Jaroslav Tulach <jaroslav.tulach@apidesign.org>
jaroslav@123
     4
 *
jaroslav@123
     5
 * This program is free software: you can redistribute it and/or modify
jaroslav@123
     6
 * it under the terms of the GNU General Public License as published by
jaroslav@123
     7
 * the Free Software Foundation, version 2 of the License.
jaroslav@123
     8
 *
jaroslav@123
     9
 * This program is distributed in the hope that it will be useful,
jaroslav@123
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
jaroslav@123
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
jaroslav@123
    12
 * GNU General Public License for more details. apidesign.org
jaroslav@123
    13
 * designates this particular file as subject to the
jaroslav@123
    14
 * "Classpath" exception as provided by apidesign.org
jaroslav@123
    15
 * in the License file that accompanied this code.
jaroslav@123
    16
 *
jaroslav@123
    17
 * You should have received a copy of the GNU General Public License
jaroslav@123
    18
 * along with this program. Look for COPYING file in the top folder.
jaroslav@123
    19
 * If not, see http://wiki.apidesign.org/wiki/GPLwithClassPathException
jaroslav@123
    20
 */
jaroslav@123
    21
package org.apidesign.html.boot.impl;
jaroslav@123
    22
jaroslav@123
    23
import org.apidesign.html.boot.spi.Fn;
jaroslav@123
    24
import java.io.IOException;
jaroslav@123
    25
import java.io.InputStream;
jaroslav@123
    26
import java.net.URL;
jaroslav@123
    27
import java.util.ArrayList;
jaroslav@123
    28
import java.util.Enumeration;
jaroslav@123
    29
import java.util.List;
jaroslav@123
    30
import org.objectweb.asm.AnnotationVisitor;
jaroslav@123
    31
import org.objectweb.asm.ClassReader;
jaroslav@123
    32
import org.objectweb.asm.ClassVisitor;
jaroslav@123
    33
import org.objectweb.asm.ClassWriter;
jaroslav@123
    34
import org.objectweb.asm.Label;
jaroslav@123
    35
import org.objectweb.asm.MethodVisitor;
jaroslav@123
    36
import org.objectweb.asm.Opcodes;
jaroslav@123
    37
import org.objectweb.asm.Type;
jaroslav@123
    38
import org.objectweb.asm.signature.SignatureReader;
jaroslav@123
    39
import org.objectweb.asm.signature.SignatureVisitor;
jaroslav@123
    40
jaroslav@123
    41
/** 
jaroslav@123
    42
 *
jaroslav@123
    43
 * @author Jaroslav Tulach <jaroslav.tulach@apidesign.org>
jaroslav@123
    44
 */
jaroslav@123
    45
abstract class JsClassLoader extends ClassLoader {
jaroslav@123
    46
    JsClassLoader(ClassLoader parent) {
jaroslav@123
    47
        super(parent);
jaroslav@143
    48
        setDefaultAssertionStatus(JsClassLoader.class.desiredAssertionStatus());
jaroslav@123
    49
    }
jaroslav@123
    50
    
jaroslav@123
    51
    @Override
jaroslav@123
    52
    protected abstract URL findResource(String name);
jaroslav@123
    53
    
jaroslav@123
    54
    @Override
jaroslav@123
    55
    protected abstract Enumeration<URL> findResources(String name);
jaroslav@123
    56
jaroslav@123
    57
    @Override
jaroslav@123
    58
    protected Class<?> findClass(String name) throws ClassNotFoundException {
jaroslav@123
    59
        if (name.startsWith("javafx")) {
jaroslav@123
    60
            return Class.forName(name);
jaroslav@123
    61
        }
jaroslav@123
    62
        if (name.startsWith("netscape")) {
jaroslav@123
    63
            return Class.forName(name);
jaroslav@123
    64
        }
jaroslav@123
    65
        if (name.startsWith("com.sun")) {
jaroslav@123
    66
            return Class.forName(name);
jaroslav@123
    67
        }
jaroslav@123
    68
        if (name.equals(JsClassLoader.class.getName())) {
jaroslav@123
    69
            return JsClassLoader.class;
jaroslav@123
    70
        }
jaroslav@123
    71
        if (name.equals(Fn.class.getName())) {
jaroslav@123
    72
            return Fn.class;
jaroslav@123
    73
        }
jaroslav@124
    74
        if (name.equals(FnUtils.class.getName())) {
jaroslav@124
    75
            return FnUtils.class;
jaroslav@124
    76
        }
jaroslav@123
    77
        URL u = findResource(name.replace('.', '/') + ".class");
jaroslav@123
    78
        if (u != null) {
jaroslav@123
    79
            InputStream is = null;
jaroslav@123
    80
            try {
jaroslav@123
    81
                is = u.openStream();
jaroslav@123
    82
                byte[] arr = new byte[is.available()];
jaroslav@123
    83
                int len = 0;
jaroslav@123
    84
                while (len < arr.length) {
jaroslav@123
    85
                    int read = is.read(arr, len, arr.length - len);
jaroslav@123
    86
                    if (read == -1) {
jaroslav@123
    87
                        throw new IOException("Can't read " + u);
jaroslav@123
    88
                    }
jaroslav@123
    89
                    len += read;
jaroslav@123
    90
                }
jaroslav@123
    91
                is.close();
jaroslav@123
    92
                is = null;
jaroslav@123
    93
                ClassReader cr = new ClassReader(arr);
jaroslav@123
    94
                FindInClass tst = new FindInClass(null);
jaroslav@123
    95
                cr.accept(tst, 0);
jaroslav@123
    96
                if (tst.found > 0) {
jaroslav@123
    97
                    ClassWriter w = new ClassWriterEx(cr, ClassWriter.COMPUTE_MAXS | ClassWriter.COMPUTE_FRAMES);
jaroslav@123
    98
                    FindInClass fic = new FindInClass(w);
jaroslav@123
    99
                    cr.accept(fic, 0);
jaroslav@123
   100
                    arr = w.toByteArray();
jaroslav@123
   101
                }
jaroslav@123
   102
                if (arr != null) {
jaroslav@123
   103
                    return defineClass(name, arr, 0, arr.length);
jaroslav@123
   104
                }
jaroslav@123
   105
            } catch (IOException ex) {
jaroslav@123
   106
                throw new ClassNotFoundException("Can't load " + name, ex);
jaroslav@123
   107
            } finally {
jaroslav@123
   108
                try {
jaroslav@123
   109
                    if (is != null) is.close();
jaroslav@123
   110
                } catch (IOException ex) {
jaroslav@123
   111
                    throw new ClassNotFoundException(null, ex);
jaroslav@123
   112
                }
jaroslav@123
   113
            }
jaroslav@123
   114
        }
jaroslav@124
   115
        if (
jaroslav@124
   116
            name.equals("org.apidesign.html.boot.spi.Fn") ||
jaroslav@124
   117
            name.equals("org.apidesign.html.boot.impl.FnUtils")
jaroslav@124
   118
        ) {
jaroslav@123
   119
            return Class.forName(name);
jaroslav@123
   120
        }
jaroslav@123
   121
        
jaroslav@123
   122
        return super.findClass(name);
jaroslav@123
   123
    }
jaroslav@123
   124
    
jaroslav@123
   125
    protected abstract Fn defineFn(String code, String... names);
jaroslav@123
   126
    
jaroslav@123
   127
    
jaroslav@123
   128
    private static final class FindInClass extends ClassVisitor {
jaroslav@123
   129
        private String name;
jaroslav@123
   130
        private int found;
jaroslav@123
   131
        
jaroslav@123
   132
        public FindInClass(ClassVisitor cv) {
jaroslav@123
   133
            super(Opcodes.ASM4, cv);
jaroslav@123
   134
        }
jaroslav@123
   135
jaroslav@123
   136
        @Override
jaroslav@123
   137
        public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {
jaroslav@123
   138
            this.name = name;
jaroslav@123
   139
            super.visit(version, access, name, signature, superName, interfaces);
jaroslav@123
   140
        }
jaroslav@123
   141
        
jaroslav@123
   142
        
jaroslav@123
   143
jaroslav@123
   144
        @Override
jaroslav@123
   145
        public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
jaroslav@123
   146
            return new FindInMethod(access, name, desc,
jaroslav@123
   147
                super.visitMethod(access & (~Opcodes.ACC_NATIVE), name, desc, signature, exceptions)
jaroslav@123
   148
            );
jaroslav@123
   149
        }
jaroslav@123
   150
        
jaroslav@123
   151
        private final class FindInMethod extends MethodVisitor {
jaroslav@123
   152
            private final String name;
jaroslav@123
   153
            private final String desc;
jaroslav@123
   154
            private final int access;
jaroslav@123
   155
            private List<String> args;
jaroslav@123
   156
            private String body;
jaroslav@123
   157
            private boolean bodyGenerated;
jaroslav@123
   158
            
jaroslav@123
   159
            public FindInMethod(int access, String name, String desc, MethodVisitor mv) {
jaroslav@123
   160
                super(Opcodes.ASM4, mv);
jaroslav@123
   161
                this.access = access;
jaroslav@123
   162
                this.name = name;
jaroslav@123
   163
                this.desc = desc;
jaroslav@123
   164
            }
jaroslav@123
   165
jaroslav@123
   166
            @Override
jaroslav@123
   167
            public AnnotationVisitor visitAnnotation(String desc, boolean visible) {
jaroslav@132
   168
                if (
jaroslav@132
   169
                    "Lnet/java/html/js/JavaScriptBody;".equals(desc) // NOI18N
jaroslav@132
   170
                    || "Lorg/apidesign/bck2brwsr/core/JavaScriptBody;".equals(desc) // NOI18N
jaroslav@132
   171
                ) {
jaroslav@123
   172
                    found++;
jaroslav@123
   173
                    return new FindInAnno();
jaroslav@123
   174
                }
jaroslav@123
   175
                return super.visitAnnotation(desc, visible);
jaroslav@123
   176
            }
jaroslav@123
   177
jaroslav@123
   178
            private void generateJSBody(List<String> args, String body) {
jaroslav@123
   179
                this.args = args;
jaroslav@123
   180
                this.body = body;
jaroslav@123
   181
            }
jaroslav@123
   182
            
jaroslav@123
   183
            @Override
jaroslav@123
   184
            public void visitCode() {
jaroslav@123
   185
                if (body == null) {
jaroslav@123
   186
                    return;
jaroslav@123
   187
                } 
jaroslav@123
   188
                generateBody();
jaroslav@123
   189
            }
jaroslav@123
   190
            
jaroslav@123
   191
            private boolean generateBody() {
jaroslav@123
   192
                if (bodyGenerated) {
jaroslav@123
   193
                    return false;
jaroslav@123
   194
                }
jaroslav@123
   195
                bodyGenerated = true;
jaroslav@123
   196
                
jaroslav@123
   197
                super.visitFieldInsn(
jaroslav@123
   198
                    Opcodes.GETSTATIC, FindInClass.this.name, 
jaroslav@123
   199
                    "$$fn$$" + name + "_" + found, 
jaroslav@123
   200
                    "Lorg/apidesign/html/boot/spi/Fn;"
jaroslav@123
   201
                );
jaroslav@123
   202
                super.visitInsn(Opcodes.DUP);
jaroslav@123
   203
                Label ifNotNull = new Label();
jaroslav@123
   204
                super.visitJumpInsn(Opcodes.IFNONNULL, ifNotNull);
jaroslav@123
   205
                
jaroslav@123
   206
                // init Fn
jaroslav@123
   207
                super.visitInsn(Opcodes.POP);
jaroslav@123
   208
                super.visitLdcInsn(Type.getObjectType(FindInClass.this.name));
jaroslav@123
   209
                super.visitLdcInsn(body);
jaroslav@123
   210
                super.visitIntInsn(Opcodes.SIPUSH, args.size());
jaroslav@123
   211
                super.visitTypeInsn(Opcodes.ANEWARRAY, "java/lang/String");
jaroslav@123
   212
                for (int i = 0; i < args.size(); i++) {
jaroslav@123
   213
                    String name = args.get(i);
jaroslav@123
   214
                    super.visitInsn(Opcodes.DUP);
jaroslav@123
   215
                    super.visitIntInsn(Opcodes.BIPUSH, i);
jaroslav@123
   216
                    super.visitLdcInsn(name);
jaroslav@123
   217
                    super.visitInsn(Opcodes.AASTORE);
jaroslav@123
   218
                }
jaroslav@123
   219
                super.visitMethodInsn(Opcodes.INVOKESTATIC, 
jaroslav@124
   220
                    "org/apidesign/html/boot/impl/FnUtils", "define", 
jaroslav@123
   221
                    "(Ljava/lang/Class;Ljava/lang/String;[Ljava/lang/String;)Lorg/apidesign/html/boot/spi/Fn;"
jaroslav@123
   222
                );
jaroslav@123
   223
                // end of Fn init
jaroslav@123
   224
                
jaroslav@123
   225
                super.visitLabel(ifNotNull);
jaroslav@123
   226
                
jaroslav@123
   227
                final int offset;
jaroslav@123
   228
                if ((access & Opcodes.ACC_STATIC) == 0) {
jaroslav@123
   229
                    offset = 1;
jaroslav@123
   230
                    super.visitIntInsn(Opcodes.ALOAD, 0);
jaroslav@123
   231
                } else {
jaroslav@123
   232
                    offset = 0;
jaroslav@123
   233
                    super.visitInsn(Opcodes.ACONST_NULL);
jaroslav@123
   234
                }
jaroslav@123
   235
                
jaroslav@123
   236
                super.visitIntInsn(Opcodes.SIPUSH, args.size());
jaroslav@123
   237
                super.visitTypeInsn(Opcodes.ANEWARRAY, "java/lang/Object");
jaroslav@123
   238
                
jaroslav@123
   239
                class SV extends SignatureVisitor {
jaroslav@123
   240
                    private boolean nowReturn;
jaroslav@123
   241
                    private Type returnType;
jaroslav@123
   242
                    private int index;
jaroslav@123
   243
                    
jaroslav@123
   244
                    public SV() {
jaroslav@123
   245
                        super(Opcodes.ASM4);
jaroslav@123
   246
                    }
jaroslav@123
   247
                    
jaroslav@123
   248
                    @Override
jaroslav@123
   249
                    public void visitBaseType(char descriptor) {
jaroslav@123
   250
                        final Type t = Type.getType("" + descriptor);
jaroslav@123
   251
                        if (nowReturn) {
jaroslav@123
   252
                            returnType = t;
jaroslav@123
   253
                            return;
jaroslav@123
   254
                        }
jaroslav@123
   255
                        FindInMethod.super.visitInsn(Opcodes.DUP);
jaroslav@123
   256
                        FindInMethod.super.visitIntInsn(Opcodes.SIPUSH, index);
jaroslav@123
   257
                        FindInMethod.super.visitVarInsn(t.getOpcode(Opcodes.ILOAD), index + offset);
jaroslav@123
   258
                        String factory;
jaroslav@123
   259
                        switch (descriptor) {
jaroslav@123
   260
                        case 'I': factory = "java/lang/Integer"; break;
jaroslav@123
   261
                        case 'J': factory = "java/lang/Long"; break;
jaroslav@123
   262
                        case 'S': factory = "java/lang/Short"; break;
jaroslav@123
   263
                        case 'F': factory = "java/lang/Float"; break;
jaroslav@123
   264
                        case 'D': factory = "java/lang/Double"; break;
jaroslav@123
   265
                        case 'Z': factory = "java/lang/Boolean"; break;
jaroslav@123
   266
                        case 'C': factory = "java/lang/Character"; break;
jaroslav@123
   267
                        case 'B': factory = "java/lang/Byte"; break;
jaroslav@123
   268
                        default: throw new IllegalStateException(t.toString());
jaroslav@123
   269
                        }
jaroslav@123
   270
                        FindInMethod.super.visitMethodInsn(Opcodes.INVOKESTATIC,
jaroslav@123
   271
                            factory, "valueOf", "(" + descriptor + ")L" + factory + ";"
jaroslav@123
   272
                        );
jaroslav@123
   273
                        FindInMethod.super.visitInsn(Opcodes.AASTORE);
jaroslav@123
   274
                        index++;
jaroslav@123
   275
                    }
jaroslav@123
   276
jaroslav@123
   277
                    @Override
jaroslav@123
   278
                    public void visitClassType(String name) {
jaroslav@123
   279
                        if (nowReturn) {
jaroslav@123
   280
                            returnType = Type.getObjectType(name);
jaroslav@123
   281
                            return;
jaroslav@123
   282
                        }
jaroslav@123
   283
                        FindInMethod.super.visitInsn(Opcodes.DUP);
jaroslav@123
   284
                        FindInMethod.super.visitIntInsn(Opcodes.SIPUSH, index);
jaroslav@123
   285
                        FindInMethod.super.visitVarInsn(Opcodes.ALOAD, index + offset);
jaroslav@123
   286
                        FindInMethod.super.visitInsn(Opcodes.AASTORE);
jaroslav@123
   287
                        index++;
jaroslav@123
   288
                    }
jaroslav@123
   289
jaroslav@123
   290
                    @Override
jaroslav@123
   291
                    public SignatureVisitor visitReturnType() {
jaroslav@123
   292
                        nowReturn = true;
jaroslav@123
   293
                        return this;
jaroslav@123
   294
                    }
jaroslav@123
   295
                    
jaroslav@123
   296
                    
jaroslav@123
   297
                }
jaroslav@123
   298
                SV sv = new SV();
jaroslav@123
   299
                SignatureReader sr = new SignatureReader(desc);
jaroslav@123
   300
                sr.accept(sv);
jaroslav@123
   301
                
jaroslav@123
   302
                super.visitMethodInsn(Opcodes.INVOKEVIRTUAL, 
jaroslav@123
   303
                    "org/apidesign/html/boot/spi/Fn", "invoke", "(Ljava/lang/Object;[Ljava/lang/Object;)Ljava/lang/Object;"
jaroslav@123
   304
                );
jaroslav@123
   305
                switch (sv.returnType.getSort()) {
jaroslav@123
   306
                case Type.VOID: 
jaroslav@123
   307
                    super.visitInsn(Opcodes.RETURN);
jaroslav@123
   308
                    break;
jaroslav@123
   309
                case Type.ARRAY:
jaroslav@123
   310
                case Type.OBJECT:
jaroslav@123
   311
                    super.visitTypeInsn(Opcodes.CHECKCAST, sv.returnType.getInternalName());
jaroslav@123
   312
                    super.visitInsn(Opcodes.ARETURN);
jaroslav@123
   313
                    break;
jaroslav@153
   314
                case Type.BOOLEAN:
jaroslav@153
   315
                    super.visitTypeInsn(Opcodes.CHECKCAST, "java/lang/Boolean");
jaroslav@153
   316
                    super.visitMethodInsn(Opcodes.INVOKEVIRTUAL, 
jaroslav@153
   317
                        "java/lang/Boolean", "booleanValue", "()Z"
jaroslav@153
   318
                    );
jaroslav@153
   319
                    super.visitInsn(Opcodes.IRETURN);
jaroslav@153
   320
                    break;
jaroslav@123
   321
                default:
jaroslav@123
   322
                    super.visitTypeInsn(Opcodes.CHECKCAST, "java/lang/Number");
jaroslav@123
   323
                    super.visitMethodInsn(Opcodes.INVOKEVIRTUAL, 
jaroslav@123
   324
                        "java/lang/Number", sv.returnType.getClassName() + "Value", "()" + sv.returnType.getDescriptor()
jaroslav@123
   325
                    );
jaroslav@123
   326
                    super.visitInsn(sv.returnType.getOpcode(Opcodes.IRETURN));
jaroslav@123
   327
                }
jaroslav@123
   328
                return true;
jaroslav@123
   329
            }
jaroslav@123
   330
jaroslav@123
   331
            @Override
jaroslav@123
   332
            public void visitEnd() {
jaroslav@123
   333
                super.visitEnd();
jaroslav@123
   334
                if (body != null) {
jaroslav@123
   335
                    if (generateBody()) {
jaroslav@123
   336
                        // native method
jaroslav@123
   337
                        super.visitMaxs(1, 0);
jaroslav@123
   338
                    }
jaroslav@123
   339
                    FindInClass.this.visitField(
jaroslav@123
   340
                        Opcodes.ACC_PRIVATE | Opcodes.ACC_STATIC, 
jaroslav@123
   341
                        "$$fn$$" + name + "_" + found, 
jaroslav@123
   342
                        "Lorg/apidesign/html/boot/spi/Fn;", 
jaroslav@123
   343
                        null, null
jaroslav@123
   344
                    );
jaroslav@123
   345
                }
jaroslav@123
   346
            }
jaroslav@123
   347
            
jaroslav@123
   348
            
jaroslav@123
   349
            
jaroslav@123
   350
            
jaroslav@123
   351
        
jaroslav@123
   352
            private final class FindInAnno extends AnnotationVisitor {
jaroslav@123
   353
                private List<String> args = new ArrayList<String>();
jaroslav@123
   354
                private String body;
jaroslav@123
   355
jaroslav@123
   356
                public FindInAnno() {
jaroslav@123
   357
                    super(Opcodes.ASM4);
jaroslav@123
   358
                }
jaroslav@123
   359
jaroslav@123
   360
                @Override
jaroslav@123
   361
                public void visit(String name, Object value) {
jaroslav@123
   362
                    if (name == null) {
jaroslav@123
   363
                        args.add((String) value);
jaroslav@123
   364
                        return;
jaroslav@123
   365
                    }
jaroslav@123
   366
                    assert name.equals("body");
jaroslav@123
   367
                    body = (String) value;
jaroslav@123
   368
                }
jaroslav@123
   369
jaroslav@123
   370
                @Override
jaroslav@123
   371
                public AnnotationVisitor visitArray(String name) {
jaroslav@123
   372
                    return this;
jaroslav@123
   373
                }
jaroslav@123
   374
jaroslav@123
   375
                @Override
jaroslav@123
   376
                public void visitEnd() {
jaroslav@123
   377
                    if (body != null) {
jaroslav@123
   378
                        generateJSBody(args, body);
jaroslav@123
   379
                    }
jaroslav@123
   380
                }
jaroslav@123
   381
            }
jaroslav@123
   382
        }
jaroslav@123
   383
    }
jaroslav@123
   384
    
jaroslav@123
   385
    private class ClassWriterEx extends ClassWriter {
jaroslav@123
   386
jaroslav@123
   387
        public ClassWriterEx(ClassReader classReader, int flags) {
jaroslav@123
   388
            super(classReader, flags);
jaroslav@123
   389
        }
jaroslav@123
   390
        
jaroslav@123
   391
        @Override
jaroslav@123
   392
        protected String getCommonSuperClass(final String type1, final String type2) {
jaroslav@123
   393
            Class<?> c, d;
jaroslav@123
   394
            ClassLoader classLoader = JsClassLoader.this;
jaroslav@123
   395
            try {
jaroslav@123
   396
                c = Class.forName(type1.replace('/', '.'), false, classLoader);
jaroslav@123
   397
                d = Class.forName(type2.replace('/', '.'), false, classLoader);
jaroslav@123
   398
            } catch (Exception e) {
jaroslav@123
   399
                throw new RuntimeException(e.toString());
jaroslav@123
   400
            }
jaroslav@123
   401
            if (c.isAssignableFrom(d)) {
jaroslav@123
   402
                return type1;
jaroslav@123
   403
            }
jaroslav@123
   404
            if (d.isAssignableFrom(c)) {
jaroslav@123
   405
                return type2;
jaroslav@123
   406
            }
jaroslav@123
   407
            if (c.isInterface() || d.isInterface()) {
jaroslav@123
   408
                return "java/lang/Object";
jaroslav@123
   409
            } else {
jaroslav@123
   410
                do {
jaroslav@123
   411
                    c = c.getSuperclass();
jaroslav@123
   412
                } while (!c.isAssignableFrom(d));
jaroslav@123
   413
                return c.getName().replace('.', '/');
jaroslav@123
   414
            }
jaroslav@123
   415
        }        
jaroslav@123
   416
    }
jaroslav@123
   417
}