io.vproxy.pni.exec.internal.ASTReader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pni-exec Show documentation
panama native interface pni-exec
The newest version!
package io.vproxy.pni.exec.internal;

import io.vproxy.pni.exec.CompilerOptions;
import io.vproxy.pni.exec.ast.AstClass;
import io.vproxy.pni.exec.type.TypePool;
import org.objectweb.asm.ClassReader;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.tree.ClassNode;

import java.util.ArrayList;
import java.util.List;

public class ASTReader {
    private final List classReaders;
    private final CompilerOptions opts;
    private final TypePool pool = new TypePool();

    private final List classes = new ArrayList<>();

    public ASTReader(List classReaders, CompilerOptions opts) {
        this.classReaders = classReaders;
        this.opts = opts;
    }

    public List read() {
        // load all classes which requires handling
        for (var r : classReaders) {
            var classNode = new ClassNode();
            r.accept(classNode, ClassReader.SKIP_FRAMES | ClassReader.SKIP_CODE);
            if ((classNode.access & Opcodes.ACC_SYNTHETIC) == Opcodes.ACC_SYNTHETIC) {
                continue;
            }

            var astClass = new AstClass(classNode, opts);
            if (!requiresHandling(astClass)) {
                PNILogger.debug(opts, "skipping " + astClass.name);
                continue;
            }
            classes.add(astClass);
            if (!astClass.isInterface) {
                pool.record(astClass);
            }
        }
        // make all classes reference to each other
        for (var cls : classes) {
            cls.ref(pool);
        }

        // verbose
        if (opts.isVerbose()) {
            for (var cls : classes) {
                PNILogger.debug(opts, "-----BEGIN CLASS-----");
                PNILogger.debug(opts, "" + cls);
                PNILogger.debug(opts, "-----END CLASS-----");
            }
        }

        // validate
        var errors = new ArrayList();
        for (var cls : classes) {
            cls.validate(opts, errors);
        }
        if (!errors.isEmpty()) {
            throw new RuntimeException("Error!\n" + String.join("\n", errors));
        }
        for (var cls : classes) {
            cls.validateDependency(errors);
        }
        if (!errors.isEmpty()) {
            throw new RuntimeException("Error!\n" + String.join("\n", errors));
        }
        for (var cls : classes) {
            cls.getNativeMemorySize(); // ensure padding calculated
        }
        for (var cls : classes) {
            cls.validateAlignment(errors);
        }
        if (!errors.isEmpty()) {
            throw new RuntimeException("Error!\n" + String.join("\n", errors));
        }
        return classes;
    }

    private boolean requiresHandling(AstClass astClass) {
        for (var a : astClass.annos) {
            if (a.type.desc.startsWith("Lio/vproxy/pni/annotation/")) {
                var n = a.type.desc.substring("Lio/vproxy/pni/annotation/".length(), a.type.desc.length() - 1);
                if (n.equals("Downcall") || n.equals("Struct") || n.equals("Union") || n.equals("Upcall")) {
                    return true;
                }
            }
        }
        return false;
    }
}