java - Findbugs:自定义检测器

我在Findbugs中编写自定义检测器。我想知道是否可以通过某种方式跟踪ASTORE和相应的ALOAD指令？就是说如果我的字节码中出现了ASTORE 3，我首先要确定它是ASTORE指令，然后是它的索引（在本例中为3），然后查找具有相同索引的ALOAD指令（在本例中为ALOAD 3指令）。

例如，在下面显示的字节码中，我想读取ASTORE 8指令（出现在第29行），并查看是否有任何索引为8的ALOAD指令。即，ALOAD 8（可以在第73行看到） 。

  29: astore        8
  31: aload_1
  32: iconst_0
  .
  .
  .
  .
  .
  .
  60: ldc           #54                 // String number
  62: aload         11
  64: invokeinterface #56,  3           // InterfaceMethod javax/servlet/http/HttpSession.setAttribute:(Ljava/lang/String;Ljava/lang/Object;)V
  69: aload         12
  71: aload         7
  73: aload         8
  75: invokeinterface #62,  3           // InterfaceMethod com/ibm/itim/ws/services/WSSessionService.getNumber:(Ljava/lang/String;Ljava/lang/String;)Lcom/ibm/itim/ws/model/WSSession;
  80: astore        14

此外，如果我找到了相应的ALOAD指令，那么我想检查调用了哪种方法。可以使用sawOpcode（）方法检查我所知道的：

    if (seen == INVOKEINTERFACE){...}

简而言之，我想做这样的事情：

伪码

    public void sawOpcode(int seen) {
    if (seen == ASTORE){
        //code to identify its index i;  i.e, ASTORE i
        if(seen == ALOAD_i){
            //if the corresponding ALOAD instruction is found...
            if(seen == INVOKEINTERFACE){

                // Identify the method invoked

            }
        }

不知道以上方法是否正确。

最佳答案

对于简单的情况，最好扩展OpcodeStackDetector。该抽象类支持跟踪堆栈值并存储有关它们的信息。您根本不需要关心ASTORE，ALOAD等。只需检查INVOKEINTERFACE。例如，如果要查找最后一个方法参数是另一个方法的返回值的位置，则可以执行以下操作：

public void sawOpcode(int seen) {
    if(seen == INVOKEINTERFACE && getMethodDescriptorOperand().getSlashedClassName()
        .equals("com/ibm/itim/ws/services/WSSessionService") &&
        getMethodDescriptorOperand().getName().equals("getNumber'))
    Item topStackItem = getStack().getStackItem(0);
    XMethod returnOf = topStackItem.getReturnValueOf();
    if(returnOf != null && returnOf.getName().equals("getParameter"))
        // here we go
    }
}

您也可以在0调用中将getStackItem更改为其他数字，以获取其他操作数。不幸的是，通过这种方式，您可以知道该值是getParameter方法的返回值，但不知道该方法中使用了哪些参数。

如果您需要跟踪更复杂的情况，则最好使用ValueNumberAnalysis。它是一个简单但功能强大的概念：它只是为静态证明相同的值分配相同的数字。假设您要跟踪所有请求参数。让我们为方法输入做一些准备（例如，在visitCode中）：

private ValueNumberDataflow vna;
private Map<ValueNumber, String> vnToParameterName;

@Override
public void visit(Code code) {
    try {
        this.vna = getClassContext().getValueNumberDataflow(getMethod());
    } catch (DataflowAnalysisException | CFGBuilderException e) {
        bugReporter.logError("Unable to get VNA for "+getMethodDescriptor(), e);
        return;
    }
    this.vnToParameterName = new HashMap<>();
    super.visit(code);
}

Map将用于存储值和相应的参数名称。这可以在sawOpcode中完成：

@Override
public void sawOpcode(int seen) {
    if(seen == INVOKEINTERFACE) {
        if(getNameConstantOperand().equals("getParameter") &&
                getSigConstantOperand().equals("(Ljava/lang/String;)Ljava/lang/String;")
                /* && check the class if necessary */) {
            Object topValue = getStack().getStackItem(0).getConstant();
            if(topValue instanceof String) { // known parameter name like "name"
                // Iterate over locations corresponding to current PC
                // (usually only one such location exists)
                for(Location location : vna.getCFG()
                        .getLocationsContainingInstructionWithOffset(getPC())) {
                    try {
                        // This frame contains value numbers
                        // right after the INVOKEINTERFACE execution
                        ValueNumberFrame frame = vna.getFactAfterLocation(location);
                        // ValueNumber corresponding to the top stack value:
                        // the return value of getParameters() method
                        ValueNumber vn = frame.getTopValue();
                        vnToParameterName.put(vn, (String) topValue);
                    } catch (DataflowAnalysisException e) {
                        return;
                    }
                }
            }
        }
    }
}

因此，现在您可以使用此地图。向sawOpcode添加更多代码：

if(seen == INVOKEINTERFACE && getMethodDescriptorOperand().getSlashedClassName()
        .equals("com/ibm/itim/ws/services/WSSessionService") &&
        getMethodDescriptorOperand().getName().equals("getNumber"))
    for(Location location : vna.getCFG()
            .getLocationsContainingInstructionWithOffset(getPC())) {
        try {
            // This frame contains value numbers
            // right before the INVOKEINTERFACE execution
            ValueNumberFrame frame = vna.getFactAtLocation(location);
            // ValueNumber corresponding to the top stack value:
            // the last parameter for getNumber method
            ValueNumber vn = frame.getStackValue(0);
            String parameterName = vnToParameterName.get(vn);
            if(parameterName != null) {
                // hurrah: this parameter is in fact
                // the return value of getParameter(parameterName)
            }
        } catch (DataflowAnalysisException e) {
            return;
        }
    }

我没有测试此代码，因此可能会出现一些小问题。请注意，ValueNumberAnalysis是非常强大的功能。它不仅能够跟踪ASTORE / ALOAD，而且还能将这个值重新保存到另一个变量中，甚至（在某些限制下）随后续加载一起存储到字段中。当然，如果您根本不使用局部变量（例如getNumber(request.getParameter("name"))），它也将起作用。