soot.jimple.spark.pag.VarNode Maven / Gradle / Ivy

package soot.jimple.spark.pag;

/*-
 * #%L
 * Soot - a J*va Optimization Framework
 * %%
 * Copyright (C) 2002 Ondrej Lhotak
 * %%
 * 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 2.1 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 General Lesser Public License for more details.
 * 
 * You should have received a copy of the GNU General Lesser Public
 * License along with this program.  If not, see
 * .
 * #L%
 */

import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import soot.AnySubType;
import soot.Context;
import soot.RefLikeType;
import soot.Type;
import soot.toolkits.scalar.Pair;

/**
 * Represents a simple variable node (Green) in the pointer assignment graph.
 *
 * @author Ondrej Lhotak
 */
public abstract class VarNode extends ValNode implements Comparable {
  private static final Logger logger = LoggerFactory.getLogger(VarNode.class);

  public Context context() {
    return null;
  }

  /** Returns all field ref nodes having this node as their base. */
  public Collection getAllFieldRefs() {
    if (fields == null) {
      return Collections.emptyList();
    }
    return fields.values();
  }

  /**
   * Returns the field ref node having this node as its base, and field as its field; null if nonexistent.
   */
  public FieldRefNode dot(SparkField field) {
    return fields == null ? null : fields.get(field);
  }

  public int compareTo(Object o) {
    VarNode other = (VarNode) o;
    if (other.finishingNumber == finishingNumber && other != this) {
      logger.debug("" + "This is: " + this + " with id " + getNumber() + " and number " + finishingNumber);
      logger.debug("" + "Other is: " + other + " with id " + other.getNumber() + " and number " + other.finishingNumber);
      throw new RuntimeException("Comparison error");
    }
    return other.finishingNumber - finishingNumber;
  }

  public void setFinishingNumber(int i) {
    finishingNumber = i;
    if (i > pag.maxFinishNumber) {
      pag.maxFinishNumber = i;
    }
  }

  /** Returns the underlying variable that this node represents. */
  public Object getVariable() {
    return variable;
  }

  /**
   * Designates this node as the potential target of a interprocedural assignment edge which may be added during on-the-fly
   * call graph updating.
   */
  public void setInterProcTarget() {
    interProcTarget = true;
  }

  /**
   * Returns true if this node is the potential target of a interprocedural assignment edge which may be added during
   * on-the-fly call graph updating.
   */
  public boolean isInterProcTarget() {
    return interProcTarget;
  }

  /**
   * Designates this node as the potential source of a interprocedural assignment edge which may be added during on-the-fly
   * call graph updating.
   */
  public void setInterProcSource() {
    interProcSource = true;
  }

  /**
   * Returns true if this node is the potential source of a interprocedural assignment edge which may be added during
   * on-the-fly call graph updating.
   */
  public boolean isInterProcSource() {
    return interProcSource;
  }

  /** Returns true if this VarNode represents the THIS pointer */
  public boolean isThisPtr() {
    if (variable instanceof Pair) {
      Pair o = (Pair) variable;
      return o.isThisParameter();
    }

    return false;
  }

  /* End of public methods. */

  VarNode(PAG pag, Object variable, Type t) {
    super(pag, t);
    if (!(t instanceof RefLikeType) || t instanceof AnySubType) {
      throw new RuntimeException("Attempt to create VarNode of type " + t);
    }
    this.variable = variable;
    pag.getVarNodeNumberer().add(this);
    setFinishingNumber(++pag.maxFinishNumber);
  }

  /** Registers a frn as having this node as its base. */
  void addField(FieldRefNode frn, SparkField field) {
    if (fields == null) {
      fields = new HashMap();
    }
    fields.put(field, frn);
  }

  /* End of package methods. */

  protected Object variable;
  protected Map fields;
  protected int finishingNumber = 0;
  protected boolean interProcTarget = false;
  protected boolean interProcSource = false;
  protected int numDerefs = 0;
}