package org.jf.dexlib2.util;
import com.google.common.base.Predicate;
import org.jf.dexlib2.AccessFlags;
import org.jf.dexlib2.iface.Method;
import org.jf.dexlib2.iface.reference.MethodReference;
import org.jf.util.CharSequenceUtils;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.Collection;
public final class MethodUtil {
private static int directMask = AccessFlags.STATIC.getValue() | AccessFlags.PRIVATE.getValue() |
AccessFlags.CONSTRUCTOR.getValue();
public static Predicate<Method> METHOD_IS_DIRECT = new Predicate<Method>() {
@Override public boolean apply(@Nullable Method input) {
return input != null && isDirect(input);
}
};
public static Predicate<Method> METHOD_IS_VIRTUAL = new Predicate<Method>() {
@Override public boolean apply(@Nullable Method input) {
return input != null && !isDirect(input);
}
};
public static boolean isDirect(@Nonnull Method method) {
return (method.getAccessFlags() & directMask) != 0;
}
public static boolean isStatic(@Nonnull Method method) {
return AccessFlags.STATIC.isSet(method.getAccessFlags());
}
public static boolean isConstructor(@Nonnull MethodReference methodReference) {
return methodReference.getName().equals("<init>");
}
public static boolean isPackagePrivate(@Nonnull Method method) {
return (method.getAccessFlags() & (AccessFlags.PRIVATE.getValue() |
AccessFlags.PROTECTED.getValue() |
AccessFlags.PUBLIC.getValue())) == 0;
}
public static int getParameterRegisterCount(@Nonnull Method method) {
return getParameterRegisterCount(method, MethodUtil.isStatic(method));
}
public static int getParameterRegisterCount(@Nonnull MethodReference methodRef, boolean isStatic) {
return getParameterRegisterCount(methodRef.getParameterTypes(), isStatic);
}
public static int getParameterRegisterCount(@Nonnull Collection<? extends CharSequence> parameterTypes,
boolean isStatic) {
int regCount = 0;
for (CharSequence paramType: parameterTypes) {
int firstChar = paramType.charAt(0);
if (firstChar == 'J' || firstChar == 'D') {
regCount += 2;
} else {
regCount++;
}
}
if (!isStatic) {
regCount++;
}
return regCount;
}
private static char getShortyType(CharSequence type) {
if (type.length() > 1) {
return 'L';
}
return type.charAt(0);
}
public static String getShorty(Collection<? extends CharSequence> params, String returnType) {
StringBuilder sb = new StringBuilder(params.size() + 1);
sb.append(getShortyType(returnType));
for (CharSequence typeRef: params) {
sb.append(getShortyType(typeRef));
}
return sb.toString();
}
public static boolean methodSignaturesMatch(@Nonnull MethodReference a, @Nonnull MethodReference b) {
return (a.getName().equals(b.getName()) &&
a.getReturnType().equals(b.getReturnType()) &&
CharSequenceUtils.listEquals(a.getParameterTypes(), b.getParameterTypes()));
}
private MethodUtil() {}
}