/*
* fb-contrib - Auxiliary detectors for Java programs
* Copyright (C) 2005-2017 Dave Brosius
*
* This library 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 library 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
package com.mebigfatguy.fbcontrib.detect;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import org.apache.bcel.Constants;
import org.apache.bcel.classfile.Code;
import org.apache.bcel.classfile.Field;
import org.apache.bcel.classfile.JavaClass;
import com.mebigfatguy.fbcontrib.utils.BugType;
import com.mebigfatguy.fbcontrib.utils.StopOpcodeParsingException;
import edu.umd.cs.findbugs.BugInstance;
import edu.umd.cs.findbugs.BugReporter;
import edu.umd.cs.findbugs.BytecodeScanningDetector;
import edu.umd.cs.findbugs.OpcodeStack;
import edu.umd.cs.findbugs.SourceLineAnnotation;
import edu.umd.cs.findbugs.ba.ClassContext;
import edu.umd.cs.findbugs.ba.XFactory;
/**
* looks for simple fields that only store one of several constant values. This usually is an indication that this field should really be an enum type.
*/
public class PoorMansEnum extends BytecodeScanningDetector {
private BugReporter bugReporter;
private Map<String, Set<Object>> fieldValues;
private Map<String, Field> nameToField;
private Map<String, SourceLineAnnotation> firstFieldUse;
private OpcodeStack stack;
public PoorMansEnum(BugReporter bugReporter) {
this.bugReporter = bugReporter;
}
@Override
public void visitClassContext(ClassContext classContext) {
try {
JavaClass cls = classContext.getJavaClass();
if (cls.getMajor() >= Constants.MAJOR_1_5) {
fieldValues = new HashMap<>();
nameToField = new HashMap<>();
for (Field f : cls.getFields()) {
if (f.isPrivate() && !f.isSynthetic()) {
String fieldName = f.getName();
fieldValues.put(fieldName, null);
nameToField.put(fieldName, f);
}
}
if (!fieldValues.isEmpty()) {
stack = new OpcodeStack();
firstFieldUse = new HashMap<>();
try {
super.visitClassContext(classContext);
for (Map.Entry<String, Set<Object>> fieldInfo : fieldValues.entrySet()) {
Set<Object> values = fieldInfo.getValue();
if ((values != null) && (values.size() >= 3)) {
String fieldName = fieldInfo.getKey();
bugReporter.reportBug(new BugInstance(this, BugType.PME_POOR_MANS_ENUM.name(), NORMAL_PRIORITY).addClass(this)
.addField(XFactory.createXField(cls, nameToField.get(fieldName))).addSourceLine(firstFieldUse.get(fieldName)));
}
}
} catch (StopOpcodeParsingException e) {
// no fields left
}
}
}
} finally {
fieldValues = null;
nameToField = null;
firstFieldUse = null;
stack = null;
}
}
@Override
public void visitCode(Code obj) {
if (!fieldValues.isEmpty()) {
stack.resetForMethodEntry(this);
super.visitCode(obj);
}
}
@Override
public void sawOpcode(int seen) {
try {
stack.precomputation(this);
if (seen == PUTFIELD) {
String fieldName = getNameConstantOperand();
if (fieldValues.containsKey(fieldName) && (stack.getStackDepth() > 0)) {
OpcodeStack.Item item = stack.getStackItem(0);
Object cons = item.getConstant();
if (cons == null) {
fieldValues.remove(fieldName);
nameToField.remove(fieldName);
firstFieldUse.remove(fieldName);
if (fieldValues.isEmpty()) {
throw new StopOpcodeParsingException();
}
} else {
Set<Object> values = fieldValues.get(fieldName);
if (values == null) {
values = new HashSet<>();
fieldValues.put(fieldName, values);
if (firstFieldUse.get(fieldName) == null) {
firstFieldUse.put(fieldName, SourceLineAnnotation.fromVisitedInstruction(this));
}
}
values.add(cons);
}
}
}
} finally {
stack.sawOpcode(this, seen);
}
}
}