the.bytecode.club.bytecodeviewer.searching.FieldCallSearch.java Source code

Introduction

Here is the source code for the.bytecode.club.bytecodeviewer.searching.FieldCallSearch.java

Source

package the.bytecode.club.bytecodeviewer.searching;

import eu.bibl.banalysis.asm.desc.OpcodeInfo;
import org.objectweb.asm.Type;
import org.objectweb.asm.tree.*;

import javax.swing.*;
import java.awt.*;
import java.util.Iterator;
import java.util.ListIterator;

/***************************************************************************
 * Bytecode Viewer (BCV) - Java & Android Reverse Engineering Suite        *
 * Copyright (C) 2014 Kalen 'Konloch' Kinloch - http://bytecodeviewer.com  *
 *                                                                         *
 * This program is free software: you can redistribute it and/or modify    *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation, either version 3 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 Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>. *
 ***************************************************************************/

/**
 * Field call searching
 * 
 * @author Konloch
 * @author Water Wolf
 * 
 */

public class FieldCallSearch implements SearchTypeDetails {

    JTextField mOwner = new JTextField(""), mName = new JTextField(""), mDesc = new JTextField("");
    JPanel myPanel = null;

    @Override
    public JPanel getPanel() {
        if (myPanel == null) {
            myPanel = new JPanel(new GridLayout(3, 2));
            myPanel.add(new JLabel("Owner: "));
            myPanel.add(mOwner);
            myPanel.add(new JLabel("Name: "));
            myPanel.add(mName);
            myPanel.add(new JLabel("Desc: "));
            myPanel.add(mDesc);
        }

        return myPanel;
    }

    @Override
    public void search(final ClassNode node, final SearchResultNotifier srn, boolean exact) {
        final Iterator<MethodNode> methods = node.methods.iterator();
        String owner = mOwner.getText();
        if (owner.isEmpty()) {
            owner = null;
        }
        String name = mName.getText();
        if (name.isEmpty()) {
            name = null;
        }
        String desc = mDesc.getText();
        if (desc.isEmpty()) {
            desc = null;
        }
        while (methods.hasNext()) {
            final MethodNode method = methods.next();

            final InsnList insnlist = method.instructions;
            final ListIterator<AbstractInsnNode> instructions = insnlist.iterator();
            while (instructions.hasNext()) {
                final AbstractInsnNode insnNode = instructions.next();
                if (insnNode instanceof FieldInsnNode) {
                    final FieldInsnNode min = (FieldInsnNode) insnNode;
                    if (name == null && owner == null && desc == null)
                        continue;
                    if (exact) {
                        if (name != null && !name.equals(min.name)) {
                            continue;
                        }
                        if (owner != null && !owner.equals(min.owner)) {
                            continue;
                        }
                        if (desc != null && !desc.equals(min.desc)) {
                            continue;
                        }
                        String desc2 = method.desc;
                        try {
                            desc2 = Type.getType(method.desc).toString();
                            if (desc2 == null || desc2.equals("null"))
                                desc2 = method.desc;
                        } catch (java.lang.ArrayIndexOutOfBoundsException e) {

                        }
                        srn.notifyOfResult(node.name + "." + method.name + desc2 + " > "
                                + OpcodeInfo.OPCODES.get(insnNode.opcode()).toLowerCase());
                    } else {

                        if (name != null && !min.name.contains(name)) {
                            continue;
                        }
                        if (owner != null && !min.owner.contains(owner)) {
                            continue;
                        }
                        if (desc != null && !min.desc.contains(desc)) {
                            continue;
                        }
                        String desc2 = method.desc;
                        try {
                            desc2 = Type.getType(method.desc).toString();
                            if (desc2 == null || desc2.equals("null"))
                                desc2 = method.desc;
                        } catch (java.lang.ArrayIndexOutOfBoundsException e) {

                        }
                        srn.notifyOfResult(node.name + "." + method.name + desc2 + " > "
                                + OpcodeInfo.OPCODES.get(insnNode.opcode()).toLowerCase());
                    }
                }
            }

        }
    }
}