/* * Copyright (C) 2010-2013 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 . */ package com.jpexs.decompiler.flash; import com.jpexs.decompiler.flash.abc.avm2.AVM2Code; import com.jpexs.decompiler.flash.gui.AboutDialog; import com.jpexs.decompiler.flash.gui.LoadingDialog; import com.jpexs.decompiler.flash.gui.MainFrame; import com.jpexs.decompiler.flash.gui.ModeFrame; import com.jpexs.decompiler.flash.gui.View; import com.jpexs.decompiler.flash.gui.player.FlashPlayerPanel; import com.jpexs.decompiler.flash.gui.proxy.ProxyFrame; import java.awt.*; import java.awt.event.ActionEvent; import java.awt.event.ActionListener; import java.awt.event.MouseAdapter; import java.awt.event.MouseEvent; import java.io.*; import java.net.Socket; import java.util.Calendar; import java.util.logging.ConsoleHandler; import java.util.logging.FileHandler; import java.util.logging.Level; import java.util.logging.Logger; import java.util.logging.SimpleFormatter; import java.util.regex.Matcher; import java.util.regex.Pattern; import javax.swing.JFileChooser; import javax.swing.JFrame; import javax.swing.JOptionPane; import javax.swing.SwingWorker; import javax.swing.filechooser.FileFilter; import javax.xml.parsers.DocumentBuilder; import javax.xml.parsers.DocumentBuilderFactory; import org.w3c.dom.Document; import org.w3c.dom.Node; import org.w3c.dom.NodeList; /** * Main executable class * * @author JPEXS */ public class Main { public static ProxyFrame proxyFrame; public static String file; public static String maskURL; public static SWF swf; public static final String version = "1.4.1"; public static final String applicationName = "JPEXS Free Flash Decompiler v." + version; public static final String shortApplicationName = "FFDec"; public static final String shortApplicationVerName = shortApplicationName + " v." + version; public static final String projectPage = "http://www.free-decompiler.com/flash"; public static final String vendor = "JPEXS"; public static LoadingDialog loadingDialog; public static ModeFrame modeFrame; private static boolean working = false; private static TrayIcon trayIcon; private static MenuItem stopMenuItem; private static boolean commandLineMode = false; public static MainFrame mainFrame; public static boolean isCommandLineMode() { return commandLineMode; } public static final boolean DISPLAY_FILENAME = true; public static boolean DEBUG_COPY = false; /** * Debug mode = throwing an error when comparing original file and recompiled */ public static boolean debugMode = false; /** * Turn off reading unsafe tags (tags which can cause problems with * recompiling) */ public static boolean DISABLE_DANGEROUS = false; /** * Turn off resolving constants in ActionScript 2 */ public static final boolean RESOLVE_CONSTANTS = true; /** * Turn off decompiling if needed */ public static final boolean DO_DECOMPILE = true; /** * Find latest constant pool in the code */ public static final boolean LATEST_CONSTANTPOOL_HACK = false; /** * Dump tags to stdout */ public static boolean dump_tags = false; /** * Limit of code subs (for obfuscated code) */ public static final int SUBLIMITER = 500; //using parameter names in decompiling may cause problems because oficial programs like Flash CS 5.5 inserts wrong parameter names indices public static final boolean PARAM_NAMES_ENABLE = false; public static String getFileTitle() { if (maskURL != null) { return maskURL; } return file; } public static void setSubLimiter(boolean value) { if (value) { AVM2Code.toSourceLimit = Main.SUBLIMITER; } else { AVM2Code.toSourceLimit = -1; } } public static boolean isWorking() { return working; } public static void showProxy() { if (proxyFrame == null) { proxyFrame = new ProxyFrame(); } proxyFrame.setVisible(true); proxyFrame.setState(Frame.NORMAL); } public static void startWork(String name) { startWork(name, -1); } public static void startWork(String name, int percent) { working = true; if (mainFrame != null) { mainFrame.setWorkStatus(name); if (percent == -1) { mainFrame.hidePercent(); } else { mainFrame.setPercent(percent); } } if (loadingDialog != null) { loadingDialog.setDetail(name); if (percent == -1) { loadingDialog.hidePercent(); } else { loadingDialog.setPercent(percent); } } if (Main.isCommandLineMode()) { System.out.println(name); } } public static void stopWork() { working = false; if (mainFrame != null) { mainFrame.setWorkStatus(""); } if (loadingDialog != null) { loadingDialog.setDetail(""); } } public static SWF parseSWF(String file) throws Exception { FileInputStream fis = new FileInputStream(file); InputStream bis = new BufferedInputStream(fis); SWF locswf = new SWF(bis, new PercentListener() { @Override public void percent(int p) { startWork("Reading SWF", p); } }); locswf.addEventListener(new EventListener() { @Override public void handleEvent(String event, Object data) { if (event.equals("export")) { startWork((String) data); } } }); return locswf; } public static void saveFile(String outfile) throws IOException { file = outfile; swf.saveTo(new FileOutputStream(outfile)); } private static class OpenFileWorker extends SwingWorker { @Override protected Object doInBackground() throws Exception { try { Main.startWork("Reading SWF..."); swf = parseSWF(Main.file); FileInputStream fis = new FileInputStream(file); DEBUG_COPY = true; ByteArrayOutputStream baos = new ByteArrayOutputStream(); /*try { swf.saveTo(baos); } catch (NotSameException nse) { Logger.getLogger(Main.class.getName()).log(Level.FINE, null, nse); JOptionPane.showMessageDialog(null, "WARNING: The SWF decompiler may have problems saving this file. Recommended usage is READ ONLY."); }*/ DEBUG_COPY = false; //DEBUG_COPY=true; } catch (Exception ex) { Logger.getLogger(Main.class.getName()).log(Level.SEVERE, null, ex); JOptionPane.showMessageDialog(null, "Cannot load SWF file."); loadingDialog.setVisible(false); return false; } try { Main.startWork("Creating window..."); mainFrame = new MainFrame(swf); loadingDialog.setVisible(false); mainFrame.setVisible(true); Main.stopWork(); } catch (Exception ex) { Logger.getLogger(Main.class.getName()).log(Level.SEVERE, null, ex); } return true; } } public static boolean openFile(String swfFile) { if (mainFrame != null) { mainFrame.setVisible(false); } Main.file = swfFile; if (Main.loadingDialog == null) { Main.loadingDialog = new LoadingDialog(); } Main.loadingDialog.setVisible(true); (new OpenFileWorker()).execute(); return true; } public static boolean saveFileDialog() { JFileChooser fc = new JFileChooser(); fc.setCurrentDirectory(new File((String) Configuration.getConfig("lastSaveDir", "."))); JFrame f = new JFrame(); View.setWindowIcon(f); int returnVal = fc.showSaveDialog(f); if (returnVal == JFileChooser.APPROVE_OPTION) { File file = fc.getSelectedFile(); try { Main.saveFile(file.getAbsolutePath()); Configuration.setConfig("lastSaveDir", file.getParentFile().getAbsolutePath()); maskURL = null; return true; } catch (IOException ex) { JOptionPane.showMessageDialog(null, "Cannot write to the file"); } } return false; } public static boolean openFileDialog() { maskURL = null; JFileChooser fc = new JFileChooser(); fc.setCurrentDirectory(new File((String) Configuration.getConfig("lastOpenDir", "."))); fc.setFileFilter(new FileFilter() { @Override public boolean accept(File f) { return (f.getName().endsWith(".swf")) || (f.isDirectory()); } @Override public String getDescription() { return "SWF files (*.swf)"; } }); JFrame f = new JFrame(); View.setWindowIcon(f); int returnVal = fc.showOpenDialog(f); if (returnVal == JFileChooser.APPROVE_OPTION) { Configuration.setConfig("lastOpenDir", fc.getSelectedFile().getParentFile().getAbsolutePath()); File selfile = fc.getSelectedFile(); Main.openFile(selfile.getAbsolutePath()); return true; } else { return false; } } public static void showModeFrame() { if (modeFrame == null) { modeFrame = new ModeFrame(); } modeFrame.setVisible(true); } public static void updateLicense() { updateLicenseInDir(new File(".\\src\\")); } /** * Script for updating license header in java files :-) * * @param dir Star directory (e.g. "src/") */ public static void updateLicenseInDir(File dir) { int defaultStartYear = 2010; int defaultFinalYear = 2013; String defaultAuthor = "JPEXS"; String defaultYearStr = "" + defaultStartYear; if (defaultFinalYear != defaultStartYear) { defaultYearStr += "-" + defaultFinalYear; } String license = "/*\r\n * Copyright (C) {year} {author}\r\n * \r\n * This program is free software: you can redistribute it and/or modify\r\n * it under the terms of the GNU General Public License as published by\r\n * the Free Software Foundation, either version 3 of the License, or\r\n * (at your option) any later version.\r\n * \r\n * This program is distributed in the hope that it will be useful,\r\n * but WITHOUT ANY WARRANTY; without even the implied warranty of\r\n * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\r\n * GNU General Public License for more details.\r\n * \r\n * You should have received a copy of the GNU General Public License\r\n * along with this program. If not, see .\r\n */"; File files[] = dir.listFiles(); for (File f : files) { if (f.isDirectory()) { updateLicenseInDir(f); } else { if (f.getName().endsWith(".java")) { ByteArrayOutputStream baos = new ByteArrayOutputStream(); PrintWriter pw = null; try { pw = new PrintWriter(new OutputStreamWriter(baos, "utf8")); } catch (UnsupportedEncodingException ex) { } try { BufferedReader br = new BufferedReader(new FileReader(f)); String s; boolean packageFound = false; String author = defaultAuthor; String yearStr = defaultYearStr; while ((s = br.readLine()) != null) { if (!packageFound) { if (s.trim().startsWith("package")) { packageFound = true; pw.println(license.replace("{year}", yearStr).replace("{author}", author)); } else { Matcher mAuthor = Pattern.compile("^.*Copyright \\(C\\) ([0-9]+)(-[0-9]+)? (.*)$").matcher(s); if (mAuthor.matches()) { author = mAuthor.group(3).trim(); int startYear = Integer.parseInt(mAuthor.group(1).trim()); if (startYear == defaultFinalYear) { yearStr = "" + startYear; } else { yearStr = "" + startYear + "-" + defaultFinalYear; } if (!author.equals(defaultAuthor)) { System.out.println("Detected nodefault author:" + author + " in " + f.getAbsolutePath()); } } } } if (packageFound) { pw.println(s); } } br.close(); pw.close(); } catch (IOException ex) { } FileOutputStream fos; try { fos = new FileOutputStream(f); fos.write(baos.toByteArray()); fos.close(); } catch (IOException ex) { } } } } } public static void badArguments() { System.err.println("Error: Bad Commandline Arguments!"); printCmdLineUsage(); System.exit(1); } public static void printHeader() { System.out.println(applicationName); for (int i = 0; i < applicationName.length(); i++) { System.out.print("-"); } System.out.println(); } public static void printCmdLineUsage() { System.out.println("Commandline arguments:"); System.out.println(" 1) -help | --help | /?"); System.out.println(" ...shows commandline arguments (this help)"); System.out.println(" 2) infile"); System.out.println(" ...opens SWF file with the decompiler GUI"); System.out.println(" 3) -proxy (-PXXX)"); System.out.println(" ...auto start proxy in the tray. Optional parameter -P specifies port for proxy. Defaults to 55555. "); System.out.println(" 4) -export (as|pcode|image|shape|movie|sound) outdirectory infile"); System.out.println(" ...export infile sources to outdirectory as AsctionScript code (\"as\" argument) or as PCode (\"pcode\" argument), images, shapes or movies"); System.out.println(" 5) -dumpSWF infile"); System.out.println(" ...dumps list of SWF tags to console"); System.out.println(" 6) -compress infile outfile"); System.out.println(" ...Compress SWF infile and save it to outfile"); System.out.println(" 7) -decompress infile outfile"); System.out.println(" ...Decompress infile and save it to outfile"); System.out.println(); System.out.println("Examples:"); System.out.println("java -jar FFDec.jar myfile.swf"); System.out.println("java -jar FFDec.jar -proxy"); System.out.println("java -jar FFDec.jar -proxy -P1234"); System.out.println("java -jar FFDec.jar -export as \"C:\\decompiled\\\" myfile.swf"); System.out.println("java -jar FFDec.jar -export pcode \"C:\\decompiled\\\" myfile.swf"); System.out.println("java -jar FFDec.jar -dumpSWF myfile.swf"); System.out.println("java -jar FFDec.jar -compress myfile.swf myfiledec.swf"); System.out.println("java -jar FFDec.jar -decompress myfiledec.swf myfile.swf"); } private static void copyFile(String from, String to) throws IOException { FileInputStream fis = new FileInputStream(from); FileOutputStream fos = new FileOutputStream(to); byte buf[] = new byte[4096]; int cnt = 0; while ((cnt = fis.read(buf)) > 0) { fos.write(buf, 0, cnt); } fis.close(); fos.close(); } /** * @param args the command line arguments */ public static void main(String[] args) throws IOException { View.setLookAndFeel(); Configuration.load(); int pos = 0; if (args.length > 0) { if (args[0].equals("-debug")) { debugMode = true; pos++; } } initLogging(debugMode); if (args.length < pos + 1) { autoCheckForUpdates(); showModeFrame(); } else { if (args[pos].equals("-proxy")) { int port = 55555; for (int i = pos; i < args.length; i++) { if (args[i].startsWith("-P")) { try { port = Integer.parseInt(args[pos].substring(2)); } catch (NumberFormatException nex) { System.err.println("Bad port number"); } } } if (proxyFrame == null) { proxyFrame = new ProxyFrame(); } proxyFrame.setPort(port); addTrayIcon(); switchProxy(); } else if (args[pos].equals("-export")) { if (args.length < pos + 4) { badArguments(); } String exportFormat = args[pos + 1]; if (!exportFormat.toLowerCase().equals("as")) { if (!exportFormat.toLowerCase().equals("pcode")) { if (!exportFormat.toLowerCase().equals("image")) { if (!exportFormat.toLowerCase().equals("shape")) { if (!exportFormat.toLowerCase().equals("movie")) { if (!exportFormat.toLowerCase().equals("sound")) { System.err.println("Invalid export format:" + exportFormat); badArguments(); } } } } } } File outDir = new File(args[pos + 2]); File inFile = new File(args[pos + 3]); if (!inFile.exists()) { System.err.println("Input SWF file does not exist!"); badArguments(); } commandLineMode = true; boolean exportOK; try { printHeader(); SWF exfile = new SWF(new FileInputStream(inFile)); exfile.addEventListener(new EventListener() { @Override public void handleEvent(String event, Object data) { if (event.equals("export")) { System.out.println((String) data); } } }); if (exportFormat.equals("image")) { exfile.exportImages(outDir.getAbsolutePath()); exportOK = true; } else if (exportFormat.equals("shape")) { exfile.exportShapes(outDir.getAbsolutePath()); exportOK = true; } else if (exportFormat.equals("as") || exportFormat.equals("pcode")) { exportOK = exfile.exportActionScript(outDir.getAbsolutePath(), exportFormat.equals("pcode")); } else if (exportFormat.equals("movie")) { exfile.exportMovies(outDir.getAbsolutePath()); exportOK = true; } else if (exportFormat.equals("sound")) { exfile.exportSounds(outDir.getAbsolutePath(), true); exportOK = true; } else { exportOK = false; } } catch (Exception ex) { exportOK = false; System.err.print("FAIL: Exporting Failed on Exception - "); Logger.getLogger(Main.class.getName()).log(Level.SEVERE, null, ex); System.exit(1); } if (exportOK) { System.out.println("OK"); System.exit(0); } else { System.err.println("FAIL"); System.exit(1); } } else if (args[pos].equals("-compress")) { if (args.length < pos + 3) { badArguments(); } if (SWF.fws2cws(new FileInputStream(args[pos + 1]), new FileOutputStream(args[pos + 2]))) { System.out.println("OK"); } else { System.err.println("FAIL"); } } else if (args[pos].equals("-decompress")) { if (args.length < pos + 3) { badArguments(); } if (SWF.cws2fws(new FileInputStream(args[pos + 1]), new FileOutputStream(args[pos + 2]))) { System.out.println("OK"); System.exit(0); } else { System.err.println("FAIL"); System.exit(1); } } else if (args[pos].equals("-dumpSWF")) { if (args.length < pos + 2) { badArguments(); } try { dump_tags = true; SWF swf = parseSWF(args[pos + 1]); } catch (Exception ex) { Logger.getLogger(Main.class.getName()).log(Level.SEVERE, null, ex); System.exit(1); } System.exit(0); } else if (args[pos].equals("-help") || args[pos].equals("--help") || args[pos].equals("/?")) { printHeader(); printCmdLineUsage(); System.exit(0); } else if (args.length == pos + 1) { autoCheckForUpdates(); openFile(args[pos]); } else { badArguments(); } } } public static String tempFile(String url) { File f = new File(Configuration.getASDecHome() + "saved" + File.separator); if (!f.exists()) { f.mkdirs(); } return Configuration.getASDecHome() + "saved" + File.separator + "asdec_" + Integer.toHexString(url.hashCode()) + ".tmp"; } public static void removeTrayIcon() { if (SystemTray.isSupported()) { SystemTray tray = SystemTray.getSystemTray(); if (trayIcon != null) { tray.remove(trayIcon); trayIcon = null; } } } public static void switchProxy() { proxyFrame.switchState(); if (stopMenuItem != null) { if (proxyFrame.isRunning()) { stopMenuItem.setLabel("Stop proxy"); } else { stopMenuItem.setLabel("Start proxy"); } } } public static void addTrayIcon() { if (trayIcon != null) { return; } if (SystemTray.isSupported()) { SystemTray tray = SystemTray.getSystemTray(); trayIcon = new TrayIcon(View.loadImage("proxy16"), "JP ASDec Proxy"); trayIcon.setImageAutoSize(true); PopupMenu trayPopup = new PopupMenu(); ActionListener trayListener = new ActionListener() { /** * Invoked when an action occurs. */ @Override public void actionPerformed(ActionEvent e) { if (e.getActionCommand().equals("EXIT")) { Main.exit(); } if (e.getActionCommand().equals("SHOW")) { Main.showProxy(); } if (e.getActionCommand().equals("SWITCH")) { Main.switchProxy(); } } }; MenuItem showMenuItem = new MenuItem("Show proxy"); showMenuItem.setActionCommand("SHOW"); showMenuItem.addActionListener(trayListener); trayPopup.add(showMenuItem); stopMenuItem = new MenuItem("Start proxy"); stopMenuItem.setActionCommand("SWITCH"); stopMenuItem.addActionListener(trayListener); trayPopup.add(stopMenuItem); trayPopup.addSeparator(); MenuItem exitMenuItem = new MenuItem("Exit"); exitMenuItem.setActionCommand("EXIT"); exitMenuItem.addActionListener(trayListener); trayPopup.add(exitMenuItem); trayIcon.setPopupMenu(trayPopup); trayIcon.addMouseListener(new MouseAdapter() { /** * {@inheritDoc} */ @Override public void mouseClicked(MouseEvent e) { if (e.getButton() == MouseEvent.BUTTON1) { Main.showProxy(); } } }); try { tray.add(trayIcon); } catch (AWTException ex) { } } } public static void exit() { Configuration.save(); FlashPlayerPanel.unload(); System.exit(0); } public static void about() { (new AboutDialog()).setVisible(true); } public static void autoCheckForUpdates() { Calendar lastUpdatesCheckDate = (Calendar) Configuration.getConfig("lastUpdatesCheckDate", null); if ((lastUpdatesCheckDate == null) || (lastUpdatesCheckDate.getTime().getTime() < Calendar.getInstance().getTime().getTime() - 1000 * 60 * 60 * 24)) { checkForUpdates(); } } public static boolean checkForUpdates() { try { Socket sock = new Socket("code.google.com", 80); OutputStream os = sock.getOutputStream(); os.write("GET /feeds/p/asdec/downloads/basic HTTP/1.1\r\nHost: code.google.com\r\nConnection: close\r\n\r\n".getBytes()); BufferedReader br = new BufferedReader(new InputStreamReader(sock.getInputStream())); String s; String response = ""; boolean start = false; while ((s = br.readLine()) != null) { if (start) { response += s + "\r\n"; } if (s.equals("")) { start = true; } } DocumentBuilder builder = DocumentBuilderFactory.newInstance().newDocumentBuilder(); Document doc = builder.parse(new ByteArrayInputStream(response.getBytes())); NodeList contents = doc.getElementsByTagName("content"); for (int i = 0; i < contents.getLength(); i++) { Node nod = contents.item(i); String cont = nod.getTextContent().trim(); String parts[] = cont.split("\n"); boolean isUpdate = false; for (String part : parts) { if (part.trim().equals("Update")) { isUpdate = true; break; } } if ((parts.length > 4) && (isUpdate)) { String downloadName = parts[1]; String link = parts[parts.length - 2]; if (isUpdate) { String downVersion = "NEW"; if (downloadName.startsWith(shortApplicationName + " version ")) { downVersion = downloadName.substring((shortApplicationName + " version ").length()); if (downVersion.contains(" ")) { downVersion = downVersion.substring(0, downVersion.indexOf(" ")); } } if (link.startsWith("