mirror of
https://git.huckle.dev/Huckles-Minecraft-Archive/jpexs-decompiler.git
synced 2026-06-16 08:41:54 +00:00
285 lines
12 KiB
Java
285 lines
12 KiB
Java
/*
|
|
* Copyright (C) 2010-2015 JPEXS
|
|
*
|
|
* 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/>.
|
|
*/
|
|
package com.jpexs.decompiler.flash.gui.dumpview;
|
|
|
|
import com.jpexs.decompiler.flash.SWF;
|
|
import com.jpexs.decompiler.flash.SWFInputStream;
|
|
import com.jpexs.decompiler.flash.abc.ABC;
|
|
import com.jpexs.decompiler.flash.abc.ABCInputStream;
|
|
import com.jpexs.decompiler.flash.abc.avm2.AVM2Code;
|
|
import com.jpexs.decompiler.flash.action.Action;
|
|
import com.jpexs.decompiler.flash.action.ActionListReader;
|
|
import com.jpexs.decompiler.flash.configuration.Configuration;
|
|
import com.jpexs.decompiler.flash.dumpview.DumpInfo;
|
|
import com.jpexs.decompiler.flash.dumpview.DumpInfoSwfNode;
|
|
import com.jpexs.decompiler.flash.gui.Main;
|
|
import com.jpexs.decompiler.flash.gui.MainPanel;
|
|
import com.jpexs.decompiler.flash.gui.View;
|
|
import com.jpexs.helpers.Helper;
|
|
import com.jpexs.helpers.MemoryInputStream;
|
|
import java.awt.Color;
|
|
import java.awt.event.ActionEvent;
|
|
import java.awt.event.MouseAdapter;
|
|
import java.awt.event.MouseEvent;
|
|
import java.io.BufferedOutputStream;
|
|
import java.io.File;
|
|
import java.io.FileOutputStream;
|
|
import java.io.IOException;
|
|
import java.io.OutputStream;
|
|
import java.util.List;
|
|
import java.util.logging.Level;
|
|
import java.util.logging.Logger;
|
|
import javax.swing.JFileChooser;
|
|
import javax.swing.JFrame;
|
|
import javax.swing.JMenuItem;
|
|
import javax.swing.JPopupMenu;
|
|
import javax.swing.JTree;
|
|
import javax.swing.SwingUtilities;
|
|
import javax.swing.plaf.basic.BasicLabelUI;
|
|
import javax.swing.plaf.basic.BasicTreeUI;
|
|
import javax.swing.tree.DefaultTreeCellRenderer;
|
|
import javax.swing.tree.TreeModel;
|
|
import javax.swing.tree.TreePath;
|
|
|
|
/**
|
|
*
|
|
* @author JPEXS
|
|
*/
|
|
public class DumpTree extends JTree {
|
|
|
|
private static final Logger logger = Logger.getLogger(DumpTree.class.getName());
|
|
|
|
private final MainPanel mainPanel;
|
|
|
|
public class DumpTreeCellRenderer extends DefaultTreeCellRenderer {
|
|
|
|
public DumpTreeCellRenderer() {
|
|
setUI(new BasicLabelUI());
|
|
setOpaque(false);
|
|
setBackgroundNonSelectionColor(Color.white);
|
|
}
|
|
}
|
|
|
|
public DumpTree(DumpTreeModel treeModel, MainPanel mainPanel) {
|
|
super(treeModel);
|
|
this.mainPanel = mainPanel;
|
|
setCellRenderer(new DumpTreeCellRenderer());
|
|
setRootVisible(false);
|
|
setBackground(Color.white);
|
|
setUI(new BasicTreeUI() {
|
|
{
|
|
setHashColor(Color.gray);
|
|
}
|
|
});
|
|
}
|
|
|
|
public void createContextMenu() {
|
|
final JPopupMenu contextPopupMenu = new JPopupMenu();
|
|
|
|
final JMenuItem expandRecursiveMenuItem = new JMenuItem(mainPanel.translate("contextmenu.expandAll"));
|
|
expandRecursiveMenuItem.addActionListener(this::expandRecursiveButtonActionPerformed);
|
|
contextPopupMenu.add(expandRecursiveMenuItem);
|
|
|
|
final JMenuItem saveToFileMenuItem = new JMenuItem(mainPanel.translate("contextmenu.saveToFile"));
|
|
saveToFileMenuItem.addActionListener(this::saveToFileButtonActionPerformed);
|
|
contextPopupMenu.add(saveToFileMenuItem);
|
|
|
|
final JMenuItem closeSelectionMenuItem = new JMenuItem(mainPanel.translate("contextmenu.closeSwf"));
|
|
closeSelectionMenuItem.addActionListener(this::closeSwfButtonActionPerformed);
|
|
contextPopupMenu.add(closeSelectionMenuItem);
|
|
|
|
final JMenuItem parseActionsMenuItem = new JMenuItem(mainPanel.translate("contextmenu.parseActions"));
|
|
parseActionsMenuItem.addActionListener(this::parseActionsButtonActionPerformed);
|
|
contextPopupMenu.add(parseActionsMenuItem);
|
|
|
|
final JMenuItem parseAbcMenuItem = new JMenuItem(mainPanel.translate("contextmenu.parseABC"));
|
|
parseAbcMenuItem.addActionListener(this::parseAbcButtonActionPerformed);
|
|
contextPopupMenu.add(parseAbcMenuItem);
|
|
|
|
final JMenuItem parseInstructionsMenuItem = new JMenuItem(mainPanel.translate("contextmenu.parseInstructions"));
|
|
parseInstructionsMenuItem.addActionListener(this::parseInstructionsButtonActionPerformed);
|
|
contextPopupMenu.add(parseInstructionsMenuItem);
|
|
|
|
addMouseListener(new MouseAdapter() {
|
|
@Override
|
|
public void mouseClicked(MouseEvent e) {
|
|
if (SwingUtilities.isRightMouseButton(e)) {
|
|
|
|
int row = getClosestRowForLocation(e.getX(), e.getY());
|
|
int[] selectionRows = getSelectionRows();
|
|
if (!Helper.contains(selectionRows, row)) {
|
|
setSelectionRow(row);
|
|
}
|
|
|
|
TreePath[] paths = getSelectionPaths();
|
|
if (paths == null || paths.length == 0) {
|
|
return;
|
|
}
|
|
closeSelectionMenuItem.setVisible(false);
|
|
expandRecursiveMenuItem.setVisible(false);
|
|
saveToFileMenuItem.setVisible(false);
|
|
parseActionsMenuItem.setVisible(false);
|
|
parseAbcMenuItem.setVisible(false);
|
|
parseInstructionsMenuItem.setVisible(false);
|
|
|
|
if (paths.length == 1) {
|
|
DumpInfo treeNode = (DumpInfo) paths[0].getLastPathComponent();
|
|
|
|
if (treeNode instanceof DumpInfoSwfNode) {
|
|
closeSelectionMenuItem.setVisible(true);
|
|
}
|
|
|
|
if (treeNode.getEndByte() - treeNode.startByte > 3) {
|
|
saveToFileMenuItem.setVisible(true);
|
|
}
|
|
|
|
// todo honfika: do not use string names, because it has conflicts e.g with DefineFont.code
|
|
if (treeNode.name.equals("actionBytes") && treeNode.getChildCount() == 0) {
|
|
parseActionsMenuItem.setVisible(true);
|
|
}
|
|
|
|
if (treeNode.name.equals("abcBytes") && treeNode.getChildCount() == 0) {
|
|
parseAbcMenuItem.setVisible(true);
|
|
}
|
|
|
|
if (treeNode.name.equals("code") && treeNode.parent.name.equals("method_body") && treeNode.getChildCount() == 0) {
|
|
parseInstructionsMenuItem.setVisible(true);
|
|
}
|
|
|
|
TreeModel model = getModel();
|
|
expandRecursiveMenuItem.setVisible(model.getChildCount(treeNode) > 0);
|
|
}
|
|
|
|
contextPopupMenu.show(e.getComponent(), e.getX(), e.getY());
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
private void expandRecursiveButtonActionPerformed(ActionEvent evt) {
|
|
TreePath path = getSelectionPath();
|
|
if (path == null) {
|
|
return;
|
|
}
|
|
View.expandTreeNodes(this, path, true);
|
|
}
|
|
|
|
private void saveToFileButtonActionPerformed(ActionEvent evt) {
|
|
TreePath[] paths = getSelectionPaths();
|
|
DumpInfo dumpInfo = (DumpInfo) paths[0].getLastPathComponent();
|
|
JFileChooser fc = new JFileChooser();
|
|
String selDir = Configuration.lastOpenDir.get();
|
|
fc.setCurrentDirectory(new File(selDir));
|
|
JFrame f = new JFrame();
|
|
View.setWindowIcon(f);
|
|
if (fc.showSaveDialog(f) == JFileChooser.APPROVE_OPTION) {
|
|
File sf = Helper.fixDialogFile(fc.getSelectedFile());
|
|
try (OutputStream fos = new BufferedOutputStream(new FileOutputStream(sf))) {
|
|
byte[] data = DumpInfoSwfNode.getSwfNode(dumpInfo).getSwf().originalUncompressedData;
|
|
fos.write(data, (int) dumpInfo.startByte, (int) (dumpInfo.getEndByte() - dumpInfo.startByte + 1));
|
|
} catch (IOException ex) {
|
|
logger.log(Level.SEVERE, null, ex);
|
|
}
|
|
}
|
|
}
|
|
|
|
private void parseActionsButtonActionPerformed(ActionEvent evt) {
|
|
TreePath[] paths = getSelectionPaths();
|
|
DumpInfo dumpInfo = (DumpInfo) paths[0].getLastPathComponent();
|
|
SWF swf = DumpInfoSwfNode.getSwfNode(dumpInfo).getSwf();
|
|
byte[] data = swf.originalUncompressedData;
|
|
int prevLength = (int) dumpInfo.startByte;
|
|
try {
|
|
SWFInputStream rri = new SWFInputStream(swf, data);
|
|
if (prevLength != 0) {
|
|
rri.seek(prevLength);
|
|
}
|
|
List<Action> actions = ActionListReader.getOriginalActions(rri, prevLength, (int) dumpInfo.getEndByte());
|
|
for (Action action : actions) {
|
|
DumpInfo di = new DumpInfo(action.toString(), "Action", null, action.getAddress(), action.getTotalActionLength());
|
|
di.parent = dumpInfo;
|
|
rri.dumpInfo = di;
|
|
rri.seek(action.getAddress());
|
|
rri.readAction();
|
|
dumpInfo.getChildInfos().add(di);
|
|
}
|
|
repaint();
|
|
} catch (IOException | InterruptedException ex) {
|
|
logger.log(Level.SEVERE, null, ex);
|
|
}
|
|
}
|
|
|
|
private void parseAbcButtonActionPerformed(ActionEvent evt) {
|
|
TreePath[] paths = getSelectionPaths();
|
|
DumpInfo dumpInfo = (DumpInfo) paths[0].getLastPathComponent();
|
|
SWF swf = DumpInfoSwfNode.getSwfNode(dumpInfo).getSwf();
|
|
byte[] data = swf.originalUncompressedData;
|
|
int prevLength = (int) dumpInfo.startByte;
|
|
try {
|
|
ABCInputStream ais = new ABCInputStream(new MemoryInputStream(data, 0, prevLength + (int) dumpInfo.lengthBytes));
|
|
ais.seek(prevLength);
|
|
ais.dumpInfo = dumpInfo;
|
|
new ABC(ais, swf, null);
|
|
} catch (IOException ex) {
|
|
logger.log(Level.SEVERE, null, ex);
|
|
}
|
|
repaint();
|
|
}
|
|
|
|
private void parseInstructionsButtonActionPerformed(ActionEvent evt) {
|
|
TreePath[] paths = getSelectionPaths();
|
|
DumpInfo dumpInfo = (DumpInfo) paths[0].getLastPathComponent();
|
|
SWF swf = DumpInfoSwfNode.getSwfNode(dumpInfo).getSwf();
|
|
byte[] data = swf.originalUncompressedData;
|
|
int prevLength = (int) dumpInfo.startByte;
|
|
try {
|
|
ABCInputStream ais = new ABCInputStream(new MemoryInputStream(data, 0, prevLength + (int) dumpInfo.lengthBytes));
|
|
ais.seek(prevLength);
|
|
ais.dumpInfo = dumpInfo;
|
|
new AVM2Code(ais, null /*FIXME! Pass correct body!*/);
|
|
} catch (IOException ex) {
|
|
logger.log(Level.SEVERE, null, ex);
|
|
}
|
|
repaint();
|
|
}
|
|
|
|
private void closeSwfButtonActionPerformed(ActionEvent evt) {
|
|
Main.closeFile(mainPanel.getCurrentSwfList());
|
|
}
|
|
|
|
@Override
|
|
public DumpTreeModel getModel() {
|
|
return (DumpTreeModel) super.getModel();
|
|
}
|
|
|
|
public void expandRoot() {
|
|
DumpTreeModel dtm = getModel();
|
|
DumpInfo root = dtm.getRoot();
|
|
expandPath(new TreePath(new Object[]{root}));
|
|
}
|
|
|
|
public void expandFirstLevelNodes() {
|
|
DumpTreeModel dtm = getModel();
|
|
DumpInfo root = dtm.getRoot();
|
|
int childCount = dtm.getChildCount(root);
|
|
expandPath(new TreePath(new Object[]{root}));
|
|
for (int i = 0; i < childCount; i++) {
|
|
expandPath(new TreePath(new Object[]{root, dtm.getChild(root, i)}));
|
|
}
|
|
}
|
|
}
|