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