org.pitest.mutationtest.mocksupport.BendJavassistToMyWillTransformer Maven / Gradle / Ivy

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

Mutation testing system for Java.

There is a newer version: 1.17.0

package org.pitest.mutationtest.mocksupport;

import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.IllegalClassFormatException;
import java.security.ProtectionDomain;
import java.util.function.Function;
import java.util.function.Predicate;

import org.objectweb.asm.ClassReader;
import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.ClassWriter;
import org.pitest.bytecode.FrameOptions;

public class BendJavassistToMyWillTransformer implements ClassFileTransformer {

  private final Predicate filter;
  private final Function transformation;

  public BendJavassistToMyWillTransformer(final Predicate filter, Function transformation) {
    this.filter = filter;
    this.transformation = transformation;
  }

  @Override
  public byte[] transform(final ClassLoader loader, final String className,
      final Class classBeingRedefined,
      final ProtectionDomain protectionDomain, final byte[] classfileBuffer)
          throws IllegalClassFormatException {

    if (shouldInclude(className)) {

      final ClassReader reader = new ClassReader(classfileBuffer);
      final ClassWriter writer = new ClassWriter(
          FrameOptions.pickFlags(classfileBuffer));

      reader.accept(this.transformation.apply(writer),
          ClassReader.EXPAND_FRAMES);
      return writer.toByteArray();
    } else {
      return null;
    }
  }

  private boolean shouldInclude(final String className) {
    return this.filter.test(className);
  }

}