boomerang.scene.jimple.AccessPathParser Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of boomerangPDS Show documentation

There is a newer version: 3.2.2

package boomerang.scene.jimple;

import boomerang.scene.Field;
import boomerang.util.AccessPath;
import com.google.common.collect.Lists;
import com.google.common.collect.Sets;
import java.util.Collection;
import java.util.List;
import java.util.Set;
import soot.Local;
import soot.RefType;
import soot.SootField;
import soot.util.Chain;

public class AccessPathParser {

  public static Collection parseAllFromString(String value, JimpleMethod m) {
    Set results = Sets.newHashSet();
    for (String v : value.split(";")) {
      results.add(parseAccessPathFromString(v, m));
    }
    return results;
  }

  private static AccessPath parseAccessPathFromString(String value, JimpleMethod m) {
    List fieldNames = Lists.newArrayList();
    String baseName;
    boolean overApproximated = value.endsWith("*");
    if (!value.contains("[")) {
      baseName = value;
    } else {
      int i = value.indexOf("[");
      baseName = value.substring(0, i);
      fieldNames =
          Lists.newArrayList(
              value.substring(i + 1, value.length() - (!overApproximated ? 1 : 2)).split(","));
    }
    List fields = Lists.newArrayList();
    Local base = getLocal(m, baseName);
    soot.Type type = base.getType();
    for (String fieldName : fieldNames) {
      if (type instanceof RefType) {
        RefType refType = (RefType) type;
        SootField fieldByName = refType.getSootClass().getFieldByName(fieldName);
        fields.add(new JimpleField(fieldByName));
        type = fieldByName.getType();
      }
    }
    return new AccessPath(
        new JimpleVal(base, m), (!overApproximated ? fields : Sets.newHashSet(fields)));
  }

  private static Local getLocal(JimpleMethod m, String baseName) {
    Chain locals = m.getDelegate().getActiveBody().getLocals();
    for (Local l : locals) {
      if (l.getName().equals(baseName)) return l;
    }
    throw new RuntimeException(
        "Could not find local with name " + baseName + " in method " + m.getDelegate());
  }
}