com.github.wolf480pl.mias4j.core.InvocationType Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of mias4j Show documentation
Method Interception and Sandboxing for Java - an alternative Java sandbox based on bytecode rewrite
There is a newer version: 0.1.1
/*
 * Copyright (c) 2014 Wolf480pl 
 * This program is licensed under the GNU Lesser General Public License.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see .
 */
package com.github.wolf480pl.mias4j.core;

import java.util.HashMap;
import java.util.Map;

import org.objectweb.asm.Opcodes;

public enum InvocationType {

    INVOKEVIRTUAL(Opcodes.H_INVOKEVIRTUAL, Opcodes.INVOKEVIRTUAL),
    INVOKESTATIC(Opcodes.H_INVOKESTATIC, Opcodes.INVOKESTATIC),
    INVOKESPECIAL(Opcodes.H_INVOKESPECIAL, Opcodes.INVOKESPECIAL),
    INVOKESUPERINITSPECIAL(Opcodes.H_INVOKESPECIAL, Opcodes.INVOKESPECIAL, false, false), // FIXME: This one's id() isn't unique, shouldn't be a problem for now, but we should fix it anyway
    INVOKENEWSPECIAL(Opcodes.H_NEWINVOKESPECIAL, Opcodes.INVOKESPECIAL, false),
    INVOKEINTERFACE(Opcodes.H_INVOKEINTERFACE, Opcodes.INVOKEINTERFACE);


    public final int handleOpcode;
    public final int insnOpcode;
    private final boolean registerHOpcode;
    private final boolean registerIOpcode;

    private InvocationType(int hopcode, int insn) {
        this(hopcode, insn, true);
    }

    private InvocationType(int hopcode, int insn, boolean registerIOpcode) {
        this(hopcode, insn, registerIOpcode, true);
    }

    private InvocationType(int hopcode, int insn, boolean registerIOpcode, boolean registerHOpcode) {
        this.handleOpcode = hopcode;
        this.insnOpcode = insn;
        this.registerHOpcode = registerHOpcode;
        this.registerIOpcode = registerIOpcode;
    }

    public int id() {
        return handleOpcode; // Subject to change
    }

    private static InvocationType[] byHOpcode = new InvocationType[10];
    private static Map byIOpcode = new HashMap<>();

    static {
        for (InvocationType t : InvocationType.values()) {
            register(t);
        }
    }

    private static void register(InvocationType t) {
        if (t.registerHOpcode) {
            byHOpcode[t.handleOpcode] = t;
        }
        if (t.registerIOpcode) {
            byIOpcode.put(t.insnOpcode, t);
        }
    }

    public static InvocationType fromHandleOpcode(int handleOpcode) {
        if (handleOpcode < 0 || handleOpcode >= byHOpcode.length) {
            return null;
        }
        return byHOpcode[handleOpcode];
    }

    public static InvocationType fromInstruction(int insnOpcode) {
        return byIOpcode.get(insnOpcode);
    }

    public static InvocationType fromID(int id) {
        return fromHandleOpcode(id); // Subject to change
    }
}