soot.jimple.internal.AbstractInvokeExpr Maven / Gradle / Ivy

package soot.jimple.internal;

/*-
 * #%L
 * Soot - a J*va Optimization Framework
 * %%
 * Copyright (C) 1999 Patrick Lam
 * Copyright (C) 2004 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.ArrayList;
import java.util.Collections;
import java.util.List;

import soot.SootMethod;
import soot.SootMethodRef;
import soot.Type;
import soot.Value;
import soot.ValueBox;
import soot.jimple.InvokeExpr;

@SuppressWarnings("serial")
abstract public class AbstractInvokeExpr implements InvokeExpr {
  protected SootMethodRef methodRef;
  final protected ValueBox[] argBoxes;

  protected AbstractInvokeExpr(SootMethodRef methodRef, ValueBox[] argBoxes) {
    this.methodRef = methodRef;
    this.argBoxes = argBoxes.length == 0 ? null : argBoxes;
  }

  public void setMethodRef(SootMethodRef methodRef) {
    this.methodRef = methodRef;
  }

  public SootMethodRef getMethodRef() {
    return methodRef;
  }

  public SootMethod getMethod() {
    return methodRef.resolve();
  }

  public abstract Object clone();

  public Value getArg(int index) {
    return argBoxes[index].getValue();
  }

  public List getArgs() {
    List l = new ArrayList<>();
    if (argBoxes != null) {
      for (ValueBox element : argBoxes) {
        l.add(element.getValue());
      }
    }
    return l;
  }

  public int getArgCount() {
    return argBoxes == null ? 0 : argBoxes.length;
  }

  public void setArg(int index, Value arg) {
    argBoxes[index].setValue(arg);
  }

  public ValueBox getArgBox(int index) {
    return argBoxes[index];
  }

  public Type getType() {
    return methodRef.returnType();
  }

  @Override
  public List getUseBoxes() {
    if (argBoxes == null) {
      return Collections.emptyList();
    }

    List list = new ArrayList();
    Collections.addAll(list, argBoxes);

    for (ValueBox element : argBoxes) {
      list.addAll(element.getValue().getUseBoxes());
    }

    return list;
  }

}