org.mockito.asm.tree.InsnList Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mockito-all Show documentation
Show all versions of mockito-all Show documentation
Mock objects library for java
The newest version!
/***
* ASM: a very small and fast Java bytecode manipulation framework
* Copyright (c) 2000-2007 INRIA, France Telecom
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of the copyright holders nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*/
package org.mockito.asm.tree;
import java.util.ListIterator;
import java.util.NoSuchElementException;
import org.mockito.asm.MethodVisitor;
/**
* A doubly linked list of {@link AbstractInsnNode} objects. This
* implementation is not thread safe.
*/
public class InsnList {
/**
* Indicates if preconditions of methods of this class must be checked.
* Checking preconditions causes the {@link #indexOf indexOf},
* {@link #set set}, {@link #insert(AbstractInsnNode, AbstractInsnNode)},
* {@link #insert(AbstractInsnNode, InsnList)}, {@link #remove remove} and
* {@link #clear} methods to execute in O(n) time instead of O(1).
*/
public static boolean check;
/**
* The number of instructions in this list.
*/
private int size;
/**
* The first instruction in this list. May be null.
*/
private AbstractInsnNode first;
/**
* The last instruction in this list. May be null.
*/
private AbstractInsnNode last;
/**
* A cache of the instructions of this list. This cache is used to improve
* the performance of the {@link #get} method.
*/
private AbstractInsnNode[] cache;
/**
* Returns the number of instructions in this list.
*
* @return the number of instructions in this list.
*/
public int size() {
return size;
}
/**
* Returns the first instruction in this list.
*
* @return the first instruction in this list, or null if the
* list is empty.
*/
public AbstractInsnNode getFirst() {
return first;
}
/**
* Returns the last instruction in this list.
*
* @return the last instruction in this list, or null if the list
* is empty.
*/
public AbstractInsnNode getLast() {
return last;
}
/**
* Returns the instruction whose index is given. This method builds a cache
* of the instructions in this list to avoid scanning the whole list each
* time it is called. Once the cache is built, this method run in constant
* time. This cache is invalidated by all the methods that modify the list.
*
* @param index the index of the instruction that must be returned.
* @return the instruction whose index is given.
* @throws IndexOutOfBoundsException if (index < 0 || index >= size()).
*/
public AbstractInsnNode get(final int index) {
if (index < 0 || index >= size) {
throw new IndexOutOfBoundsException();
}
if (cache == null) {
cache = toArray();
}
return cache[index];
}
/**
* Returns true if the given instruction belongs to this list.
* This method always scans the instructions of this list until it finds the
* given instruction or reaches the end of the list.
*
* @param insn an instruction.
* @return true if the given instruction belongs to this list.
*/
public boolean contains(final AbstractInsnNode insn) {
AbstractInsnNode i = first;
while (i != null && i != insn) {
i = i.next;
}
return i != null;
}
/**
* Returns the index of the given instruction in this list. This method
* builds a cache of the instruction indexes to avoid scanning the whole
* list each time it is called. Once the cache is built, this method run in
* constant time. The cache is invalidated by all the methods that modify
* the list.
*
* @param insn an instruction of this list.
* @return the index of the given instruction in this list. The result of
* this method is undefined if the given instruction does not belong
* to this list. Use {@link #contains contains} to test if an
* instruction belongs to an instruction list or not.
* @throws IllegalArgumentException if {@link #check} is true and
* if insn does not belong to this list.
*/
public int indexOf(final AbstractInsnNode insn) {
if (check && !contains(insn)) {
throw new IllegalArgumentException();
}
if (cache == null) {
cache = toArray();
}
return insn.index;
}
/**
* Makes the given visitor visit all of the instructions in this list.
*
* @param mv the method visitor that must visit the instructions.
*/
public void accept(final MethodVisitor mv) {
AbstractInsnNode insn = first;
while (insn != null) {
insn.accept(mv);
insn = insn.next;
}
}
/**
* Returns an iterator over the instructions in this list.
*
* @return an iterator over the instructions in this list.
*/
public ListIterator iterator() {
return iterator(0);
}
/**
* Returns an iterator over the instructions in this list.
*
* @return an iterator over the instructions in this list.
*/
public ListIterator iterator(int index) {
return new InsnListIterator(index);
}
/**
* Returns an array containing all of the instructions in this list.
*
* @return an array containing all of the instructions in this list.
*/
public AbstractInsnNode[] toArray() {
int i = 0;
AbstractInsnNode elem = first;
AbstractInsnNode[] insns = new AbstractInsnNode[size];
while (elem != null) {
insns[i] = elem;
elem.index = i++;
elem = elem.next;
}
return insns;
}
/**
* Replaces an instruction of this list with another instruction.
*
* @param location an instruction of this list.
* @param insn another instruction, which must not belong to any
* {@link InsnList}.
* @throws IllegalArgumentException if {@link #check} is true,
* and if i does not belong to this list or if insn belongs to an
* instruction list.
*/
public void set(final AbstractInsnNode location, final AbstractInsnNode insn) {
if (check && !(contains(location) && insn.index == -1)) {
throw new IllegalArgumentException();
}
AbstractInsnNode next = location.next;
insn.next = next;
if (next != null) {
next.prev = insn;
} else {
last = insn;
}
AbstractInsnNode prev = location.prev;
insn.prev = prev;
if (prev != null) {
prev.next = insn;
} else {
first = insn;
}
if (cache != null) {
int index = location.index;
cache[index] = insn;
insn.index = index;
} else {
insn.index = 0; // insn now belongs to an InsnList
}
location.index = -1; // i no longer belongs to an InsnList
location.prev = null;
location.next = null;
}
/**
* Adds the given instruction to the end of this list.
*
* @param insn an instruction, which must not belong to any
* {@link InsnList}.
* @throws IllegalArgumentException if {@link #check} is true,
* and if insn belongs to an instruction list.
*/
public void add(final AbstractInsnNode insn) {
if (check && insn.index != -1) {
throw new IllegalArgumentException();
}
++size;
if (last == null) {
first = insn;
last = insn;
} else {
last.next = insn;
insn.prev = last;
}
last = insn;
cache = null;
insn.index = 0; // insn now belongs to an InsnList
}
/**
* Adds the given instructions to the end of this list.
*
* @param insns an instruction list, which is cleared during the process.
* @throws IllegalArgumentException if {@link #check} is true,
* and if insn == this.
*/
public void add(final InsnList insns) {
if (check && insns == this) {
throw new IllegalArgumentException();
}
if (insns.size == 0) {
return;
}
size += insns.size;
if (last == null) {
first = insns.first;
last = insns.last;
} else {
AbstractInsnNode elem = insns.first;
last.next = elem;
elem.prev = last;
last = insns.last;
}
cache = null;
insns.removeAll(false);
}
/**
* Inserts the given instruction at the begining of this list.
*
* @param insn an instruction, which must not belong to any
* {@link InsnList}.
* @throws IllegalArgumentException if {@link #check} is true,
* and if insn belongs to an instruction list.
*/
public void insert(final AbstractInsnNode insn) {
if (check && insn.index != -1) {
throw new IllegalArgumentException();
}
++size;
if (first == null) {
first = insn;
last = insn;
} else {
first.prev = insn;
insn.next = first;
}
first = insn;
cache = null;
insn.index = 0; // insn now belongs to an InsnList
}
/**
* Inserts the given instructions at the begining of this list.
*
* @param insns an instruction list, which is cleared during the process.
* @throws IllegalArgumentException if {@link #check} is true,
* and if insn == this.
*/
public void insert(final InsnList insns) {
if (check && insns == this) {
throw new IllegalArgumentException();
}
if (insns.size == 0) {
return;
}
size += insns.size;
if (first == null) {
first = insns.first;
last = insns.last;
} else {
AbstractInsnNode elem = insns.last;
first.prev = elem;
elem.next = first;
first = insns.first;
}
cache = null;
insns.removeAll(false);
}
/**
* Inserts the given instruction after the specified instruction.
*
* @param location an instruction of this list after which insn must be
* inserted.
* @param insn the instruction to be inserted, which must not belong to
* any {@link InsnList}.
* @throws IllegalArgumentException if {@link #check} is true,
* and if i does not belong to this list or if insn belongs to an
* instruction list.
*/
public void insert(final AbstractInsnNode location, final AbstractInsnNode insn) {
if (check && !(contains(location) && insn.index == -1)) {
throw new IllegalArgumentException();
}
++size;
AbstractInsnNode next = location.next;
if (next == null) {
last = insn;
} else {
next.prev = insn;
}
location.next = insn;
insn.next = next;
insn.prev = location;
cache = null;
insn.index = 0; // insn now belongs to an InsnList
}
/**
* Inserts the given instructions after the specified instruction.
*
* @param location an instruction of this list after which the instructions
* must be inserted.
* @param insns the instruction list to be inserted, which is cleared during
* the process.
* @throws IllegalArgumentException if {@link #check} is true,
* and if i does not belong to this list or if insns == this.
*/
public void insert(final AbstractInsnNode location, final InsnList insns) {
if (check && !(contains(location) && insns != this)) {
throw new IllegalArgumentException();
}
if (insns.size == 0) {
return;
}
size += insns.size;
AbstractInsnNode ifirst = insns.first;
AbstractInsnNode ilast = insns.last;
AbstractInsnNode next = location.next;
if (next == null) {
last = ilast;
} else {
next.prev = ilast;
}
location.next = ifirst;
ilast.next = next;
ifirst.prev = location;
cache = null;
insns.removeAll(false);
}
/**
* Inserts the given instruction before the specified instruction.
*
* @param location an instruction of this list before which insn must be
* inserted.
* @param insn the instruction to be inserted, which must not belong to
* any {@link InsnList}.
* @throws IllegalArgumentException if {@link #check} is true,
* and if i does not belong to this list or if insn belongs to an
* instruction list.
*/
public void insertBefore(final AbstractInsnNode location, final AbstractInsnNode insn) {
if (check && !(contains(location) && insn.index == -1)) {
throw new IllegalArgumentException();
}
++size;
AbstractInsnNode prev = location.prev;
if (prev == null) {
first = insn;
} else {
prev.next = insn;
}
location.prev = insn;
insn.next = location;
insn.prev = prev;
cache = null;
insn.index = 0; // insn now belongs to an InsnList
}
/**
* Inserts the given instructions before the specified instruction.
*
* @param location an instruction of this list before which the instructions
* must be inserted.
* @param insns the instruction list to be inserted, which is cleared during
* the process.
* @throws IllegalArgumentException if {@link #check} is true,
* and if i does not belong to this list or if insns == this.
*/
public void insertBefore(final AbstractInsnNode location, final InsnList insns) {
if (check && !(contains(location ) && insns != this)) {
throw new IllegalArgumentException();
}
if (insns.size == 0) {
return;
}
size += insns.size;
AbstractInsnNode ifirst = insns.first;
AbstractInsnNode ilast = insns.last;
AbstractInsnNode prev = location .prev;
if (prev == null) {
first = ifirst;
} else {
prev.next = ifirst;
}
location .prev = ilast;
ilast.next = location ;
ifirst.prev = prev;
cache = null;
insns.removeAll(false);
}
/**
* Removes the given instruction from this list.
*
* @param insn the instruction of this list that must be removed.
* @throws IllegalArgumentException if {@link #check} is true,
* and if insn does not belong to this list.
*/
public void remove(final AbstractInsnNode insn) {
if (check && !contains(insn)) {
throw new IllegalArgumentException();
}
--size;
AbstractInsnNode next = insn.next;
AbstractInsnNode prev = insn.prev;
if (next == null) {
if (prev == null) {
first = null;
last = null;
} else {
prev.next = null;
last = prev;
}
} else {
if (prev == null) {
first = next;
next.prev = null;
} else {
prev.next = next;
next.prev = prev;
}
}
cache = null;
insn.index = -1; // insn no longer belongs to an InsnList
insn.prev = null;
insn.next = null;
}
/**
* Removes all of the instructions of this list.
*
* @param mark if the instructions must be marked as no longer belonging to
* any {@link InsnList}.
*/
private void removeAll(final boolean mark) {
if (mark) {
AbstractInsnNode insn = first;
while (insn != null) {
AbstractInsnNode next = insn.next;
insn.index = -1; // insn no longer belongs to an InsnList
insn.prev = null;
insn.next = null;
insn = next;
}
}
size = 0;
first = null;
last = null;
cache = null;
}
/**
* Removes all of the instructions of this list.
*/
public void clear() {
removeAll(check);
}
/**
* Reset all labels in the instruction list. This method should be called
* before reusing same instructions list between several
* ClassWriters.
*/
public void resetLabels() {
AbstractInsnNode insn = first;
while (insn != null) {
if (insn instanceof LabelNode) {
((LabelNode) insn).resetLabel();
}
insn = insn.next;
}
}
private final class InsnListIterator implements ListIterator {
AbstractInsnNode next;
AbstractInsnNode prev;
private InsnListIterator(int index) {
if(index==size()) {
next = null;
prev = getLast();
} else {
next = get(index);
prev = next.prev;
}
}
public boolean hasNext() {
return next != null;
}
public Object next() {
if (next == null) {
throw new NoSuchElementException();
}
AbstractInsnNode result = next;
prev = result;
next = result.next;
return result;
}
public void remove() {
InsnList.this.remove(prev);
prev = prev.prev;
}
public boolean hasPrevious() {
return prev != null;
}
public Object previous() {
AbstractInsnNode result = prev;
next = result;
prev = result.prev;
return result;
}
public int nextIndex() {
if (next == null) {
return size();
}
if (cache == null) {
cache = toArray();
}
return next.index;
}
public int previousIndex() {
if (prev == null) {
return -1;
}
if (cache == null) {
cache = toArray();
}
return prev.index;
}
public void add(Object o) {
InsnList.this.insertBefore(next, (AbstractInsnNode) o);
prev = (AbstractInsnNode) o;
}
public void set(Object o) {
InsnList.this.set(next.prev, (AbstractInsnNode) o);
prev = (AbstractInsnNode) o;
}
}
}