com.eg.agent.compile.InstrumentationContext Maven / Gradle / Ivy

Go to download
package com.eg.agent.compile;

/**
 * @author janakiraman.k
 *
 */

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.WeakHashMap;

public class InstrumentationContext {
    private static final String[] ANDROID_8_MISSING_CLASS_WHITE_LIST = {"android.view.View$AccessibilityDelegate", "android.view.accessibility.AccessibilityNodeProvider"};

    private static final HashMap> MISSING_CLASS_WHITE_LIST = new HashMap() {
    };

    private final WeakHashMap> cache = new WeakHashMap();
    private final ClassRemapperConfiguration config;
    private ShimInterpreter shim;
    private final EGLogger log;
    private boolean classModified;
    private String className;
    private String superClassName;
    private final ArrayList tags = new ArrayList();
    private HashMap tracedMethods;
    private HashMap skippedMethods;
    private final HashMap> tracedMethodParameters = new HashMap();

    public InstrumentationContext(ClassRemapperConfiguration config, EGLogger log) {
        this.config = config;
        this.log = log;
        this.tracedMethods = new HashMap();
        this.skippedMethods = new HashMap();
    }

    public EGLogger getLog() {
        return this.log;
    }

    public void reset() {
        this.classModified = false;
        this.className = null;
        this.superClassName = null;
        this.shim = null;
        this.tags.clear();
    }

    public void markModified() {
        this.classModified = true;
    }

    public boolean isClassModified() {
        return this.classModified;
    }

    public void addTag(String tag) {
        this.tags.add(tag);
    }

    public void addUniqueTag(String tag) {
        while (this.tags.remove(tag)) ;
        addTag(tag);
    }

    public void addTracedMethod(String name, String desc) {
        this.tracedMethods.put(this.className + "#" + name, desc);
    }

    public void addSkippedMethod(String name, String desc) {
        this.skippedMethods.put(this.className + "#" + name, desc);
    }

    public void addTracedMethodParameter(String methodName, String parameterName, String parameterClass, String parameterValue) {

        String name = this.className + "#" + methodName;
        if (!this.tracedMethodParameters.containsKey(name)) {
            this.tracedMethodParameters.put(name, new ArrayList());
        }
        ArrayList methodParameters = (ArrayList) this.tracedMethodParameters.get(name);
        methodParameters.add(parameterName);
        methodParameters.add(parameterClass);
        methodParameters.add(parameterValue);
    }

    public ArrayList getTracedMethodParameters(String methodName) {
        return (ArrayList) this.tracedMethodParameters.get(this.className + "#" + methodName);
    }

    public boolean isTracedMethod(String name, String desc) {
        return searchMethodMap(this.tracedMethods, name, desc);
    }

    public boolean isSkippedMethod(String name, String desc) {
        return searchMethodMap(this.skippedMethods, name, desc);
    }

    private boolean searchMethodMap(Map map, String name, String desc) {
        String descToMatch = (String) map.get(this.className + "#" + name);

        if (descToMatch == null) {
            return false;
        }
        if (desc.equals(desc)) {
            return true;
        }

        return false;
    }

    public List getTags() {
        return this.tags;
    }

    public boolean hasTag(String tag) {
        return this.tags.contains(tag);
    }

    public void setClassName(String className) {
        this.className = className;
    }

    public String getClassName() {
        return this.className;
    }

    public String getFriendlyClassName() {
        return this.className.replaceAll("/", ".");
    }

    public String getFriendlySuperClassName() {
        return this.superClassName.replaceAll("/", ".");
    }

    public String getSimpleClassName() {
        if (this.className.contains("/")) {
            return this.className.substring(this.className.lastIndexOf("/") + 1);
        }
        return this.className;
    }

    public void setSuperClassName(String superClassName) {
        this.superClassName = superClassName;
    }

    public String getSuperClassName() {
        return this.superClassName;
    }

    public void setShim(ShimInterpreter shim) {
        this.shim = shim;
    }

    public ShimInterpreter getShim() {
        return this.shim;
    }

    public ClassMetaData newClassData(byte[] mainClassBytes) {
        if (this.shim != null) {
            return new ClassMetaData(mainClassBytes, this.shim.getClassName(), this.shim.getBytes(), isClassModified());
        }

        return new ClassMetaData(mainClassBytes, isClassModified());
    }

    public MethodData getMethodWrapper(MethodData method) {
        return this.config.getMethodWrapper(method);
    }

    public Collection getCallSiteReplacements(String className, String methodName, String methodDesc) {
        return this.config.getCallSiteReplacements(className, methodName, methodDesc);
    }
}