mirror of
https://git.huckle.dev/Huckles-Minecraft-Archive/jpexs-decompiler.git
synced 2026-06-05 13:14:53 +00:00
2557 lines
107 KiB
Java
2557 lines
107 KiB
Java
/*
|
|
* 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 <http://www.gnu.org/licenses/>.
|
|
*/
|
|
package com.jpexs.decompiler.flash;
|
|
|
|
import SevenZip.Compression.LZMA.Encoder;
|
|
import com.jpexs.decompiler.flash.abc.ABC;
|
|
import com.jpexs.decompiler.flash.abc.ClassPath;
|
|
import com.jpexs.decompiler.flash.abc.RenameType;
|
|
import com.jpexs.decompiler.flash.abc.ScriptPack;
|
|
import com.jpexs.decompiler.flash.action.Action;
|
|
import com.jpexs.decompiler.flash.action.ActionGraphSource;
|
|
import com.jpexs.decompiler.flash.action.model.ConstantPool;
|
|
import com.jpexs.decompiler.flash.action.model.DirectValueActionItem;
|
|
import com.jpexs.decompiler.flash.action.model.FunctionActionItem;
|
|
import com.jpexs.decompiler.flash.action.model.GetMemberActionItem;
|
|
import com.jpexs.decompiler.flash.action.model.GetVariableActionItem;
|
|
import com.jpexs.decompiler.flash.action.model.clauses.ClassActionItem;
|
|
import com.jpexs.decompiler.flash.action.model.clauses.InterfaceActionItem;
|
|
import com.jpexs.decompiler.flash.action.swf4.ActionEquals;
|
|
import com.jpexs.decompiler.flash.action.swf4.ActionGetVariable;
|
|
import com.jpexs.decompiler.flash.action.swf4.ActionIf;
|
|
import com.jpexs.decompiler.flash.action.swf4.ActionPush;
|
|
import com.jpexs.decompiler.flash.action.swf4.ActionSetVariable;
|
|
import com.jpexs.decompiler.flash.action.swf4.ConstantIndex;
|
|
import com.jpexs.decompiler.flash.action.swf5.ActionCallFunction;
|
|
import com.jpexs.decompiler.flash.action.swf5.ActionCallMethod;
|
|
import com.jpexs.decompiler.flash.action.swf5.ActionConstantPool;
|
|
import com.jpexs.decompiler.flash.action.swf5.ActionDefineFunction;
|
|
import com.jpexs.decompiler.flash.action.swf5.ActionDefineLocal;
|
|
import com.jpexs.decompiler.flash.action.swf5.ActionDefineLocal2;
|
|
import com.jpexs.decompiler.flash.action.swf5.ActionEquals2;
|
|
import com.jpexs.decompiler.flash.action.swf5.ActionGetMember;
|
|
import com.jpexs.decompiler.flash.action.swf5.ActionNewMethod;
|
|
import com.jpexs.decompiler.flash.action.swf5.ActionNewObject;
|
|
import com.jpexs.decompiler.flash.action.swf5.ActionSetMember;
|
|
import com.jpexs.decompiler.flash.action.swf7.ActionDefineFunction2;
|
|
import com.jpexs.decompiler.flash.ecma.Null;
|
|
import com.jpexs.decompiler.flash.flv.AUDIODATA;
|
|
import com.jpexs.decompiler.flash.flv.FLVOutputStream;
|
|
import com.jpexs.decompiler.flash.flv.FLVTAG;
|
|
import com.jpexs.decompiler.flash.flv.VIDEODATA;
|
|
import com.jpexs.decompiler.flash.helpers.collections.MyEntry;
|
|
import com.jpexs.decompiler.flash.tags.ABCContainerTag;
|
|
import com.jpexs.decompiler.flash.tags.DefineBinaryDataTag;
|
|
import com.jpexs.decompiler.flash.tags.DefineButton2Tag;
|
|
import com.jpexs.decompiler.flash.tags.DefineButtonTag;
|
|
import com.jpexs.decompiler.flash.tags.DefineSoundTag;
|
|
import com.jpexs.decompiler.flash.tags.DefineSpriteTag;
|
|
import com.jpexs.decompiler.flash.tags.DefineVideoStreamTag;
|
|
import com.jpexs.decompiler.flash.tags.DoInitActionTag;
|
|
import com.jpexs.decompiler.flash.tags.ExportAssetsTag;
|
|
import com.jpexs.decompiler.flash.tags.FileAttributesTag;
|
|
import com.jpexs.decompiler.flash.tags.SetBackgroundColorTag;
|
|
import com.jpexs.decompiler.flash.tags.ShowFrameTag;
|
|
import com.jpexs.decompiler.flash.tags.SoundStreamBlockTag;
|
|
import com.jpexs.decompiler.flash.tags.SymbolClassTag;
|
|
import com.jpexs.decompiler.flash.tags.Tag;
|
|
import com.jpexs.decompiler.flash.tags.VideoFrameTag;
|
|
import com.jpexs.decompiler.flash.tags.base.ASMSource;
|
|
import com.jpexs.decompiler.flash.tags.base.BoundedTag;
|
|
import com.jpexs.decompiler.flash.tags.base.CharacterIdTag;
|
|
import com.jpexs.decompiler.flash.tags.base.CharacterTag;
|
|
import com.jpexs.decompiler.flash.tags.base.Container;
|
|
import com.jpexs.decompiler.flash.tags.base.DrawableTag;
|
|
import com.jpexs.decompiler.flash.tags.base.ImageTag;
|
|
import com.jpexs.decompiler.flash.tags.base.PlaceObjectTypeTag;
|
|
import com.jpexs.decompiler.flash.tags.base.RemoveTag;
|
|
import com.jpexs.decompiler.flash.tags.base.ShapeTag;
|
|
import com.jpexs.decompiler.flash.tags.base.SoundStreamHeadTypeTag;
|
|
import com.jpexs.decompiler.flash.tags.base.TextTag;
|
|
import com.jpexs.decompiler.flash.types.CXFORM;
|
|
import com.jpexs.decompiler.flash.types.CXFORMWITHALPHA;
|
|
import com.jpexs.decompiler.flash.types.MATRIX;
|
|
import com.jpexs.decompiler.flash.types.RECT;
|
|
import com.jpexs.decompiler.flash.types.filters.BlendComposite;
|
|
import com.jpexs.decompiler.flash.types.filters.FILTER;
|
|
import com.jpexs.decompiler.flash.types.filters.Filtering;
|
|
import com.jpexs.decompiler.flash.types.shaperecords.SHAPERECORD;
|
|
import com.jpexs.decompiler.flash.types.sound.AdpcmDecoder;
|
|
import com.jpexs.decompiler.flash.xfl.XFLConverter;
|
|
import com.jpexs.decompiler.graph.Graph;
|
|
import com.jpexs.decompiler.graph.GraphSourceItem;
|
|
import com.jpexs.decompiler.graph.GraphSourceItemContainer;
|
|
import com.jpexs.decompiler.graph.GraphTargetItem;
|
|
import com.jpexs.helpers.Cache;
|
|
import com.jpexs.helpers.Helper;
|
|
import com.jpexs.helpers.ProgressListener;
|
|
import java.awt.AlphaComposite;
|
|
import java.awt.Color;
|
|
import java.awt.Graphics2D;
|
|
import java.awt.Point;
|
|
import java.awt.Rectangle;
|
|
import java.awt.RenderingHints;
|
|
import java.awt.geom.AffineTransform;
|
|
import java.awt.image.BufferedImage;
|
|
import java.io.*;
|
|
import java.util.ArrayList;
|
|
import java.util.Arrays;
|
|
import java.util.HashMap;
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.Locale;
|
|
import java.util.Map;
|
|
import java.util.Random;
|
|
import java.util.Set;
|
|
import java.util.Stack;
|
|
import java.util.concurrent.Callable;
|
|
import java.util.concurrent.ExecutionException;
|
|
import java.util.concurrent.ExecutorService;
|
|
import java.util.concurrent.Executors;
|
|
import java.util.concurrent.Future;
|
|
import java.util.concurrent.TimeUnit;
|
|
import java.util.concurrent.atomic.AtomicInteger;
|
|
import java.util.logging.Level;
|
|
import java.util.logging.Logger;
|
|
import java.util.regex.Pattern;
|
|
import java.util.zip.DeflaterOutputStream;
|
|
import java.util.zip.InflaterInputStream;
|
|
import javax.imageio.ImageIO;
|
|
|
|
/**
|
|
* Class representing SWF file
|
|
*
|
|
* @author JPEXS
|
|
*/
|
|
public class SWF {
|
|
|
|
/**
|
|
* Default version of SWF file format
|
|
*/
|
|
public static final int DEFAULT_VERSION = 10;
|
|
/**
|
|
* Tags inside of file
|
|
*/
|
|
public List<Tag> tags = new ArrayList<>();
|
|
/**
|
|
* Rectangle for the display
|
|
*/
|
|
public RECT displayRect;
|
|
/**
|
|
* Movie frame rate
|
|
*/
|
|
public int frameRate;
|
|
/**
|
|
* Number of frames in movie
|
|
*/
|
|
public int frameCount;
|
|
/**
|
|
* Version of SWF
|
|
*/
|
|
public int version;
|
|
/**
|
|
* Size of the file
|
|
*/
|
|
public long fileSize;
|
|
/**
|
|
* Use compression
|
|
*/
|
|
public boolean compressed = false;
|
|
/**
|
|
* Use LZMA compression
|
|
*/
|
|
public boolean lzma = false;
|
|
/**
|
|
* Compressed size of the file (LZMA)
|
|
*/
|
|
public long compressedSize;
|
|
/**
|
|
* LZMA Properties
|
|
*/
|
|
public byte lzmaProperties[];
|
|
public FileAttributesTag fileAttributes;
|
|
|
|
/**
|
|
* Gets all tags with specified id
|
|
*
|
|
* @param tagId Identificator of tag type
|
|
* @return List of tags
|
|
*/
|
|
public List<Tag> getTagData(int tagId) {
|
|
List<Tag> ret = new ArrayList<>();
|
|
for (Tag tag : tags) {
|
|
if (tag.getId() == tagId) {
|
|
ret.add(tag);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
* Saves this SWF into new file
|
|
*
|
|
* @param os OutputStream to save SWF in
|
|
* @throws IOException
|
|
*/
|
|
public void saveTo(OutputStream os) throws IOException {
|
|
try {
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
SWFOutputStream sos = new SWFOutputStream(baos, version);
|
|
sos.writeRECT(displayRect);
|
|
sos.writeUI8(0);
|
|
sos.writeUI8(frameRate);
|
|
sos.writeUI16(frameCount);
|
|
|
|
sos.writeTags(tags);
|
|
sos.writeUI16(0);
|
|
sos.close();
|
|
if (compressed && lzma) {
|
|
os.write('Z');
|
|
} else if (compressed) {
|
|
os.write('C');
|
|
} else {
|
|
os.write('F');
|
|
}
|
|
os.write('W');
|
|
os.write('S');
|
|
os.write(version);
|
|
byte data[] = baos.toByteArray();
|
|
sos = new SWFOutputStream(os, version);
|
|
sos.writeUI32(data.length + 8);
|
|
|
|
if (compressed) {
|
|
if (lzma) {
|
|
Encoder enc = new Encoder();
|
|
int val = lzmaProperties[0] & 0xFF;
|
|
int lc = val % 9;
|
|
int remainder = val / 9;
|
|
int lp = remainder % 5;
|
|
int pb = remainder / 5;
|
|
int dictionarySize = 0;
|
|
for (int i = 0; i < 4; i++) {
|
|
dictionarySize += ((int) (lzmaProperties[1 + i]) & 0xFF) << (i * 8);
|
|
}
|
|
enc.SetDictionarySize(dictionarySize);
|
|
enc.SetLcLpPb(lc, lp, pb);
|
|
baos = new ByteArrayOutputStream();
|
|
enc.SetEndMarkerMode(true);
|
|
enc.Code(new ByteArrayInputStream(data), baos, -1, -1, null);
|
|
data = baos.toByteArray();
|
|
byte udata[] = new byte[4];
|
|
udata[0] = (byte) (data.length & 0xFF);
|
|
udata[1] = (byte) ((data.length >> 8) & 0xFF);
|
|
udata[2] = (byte) ((data.length >> 16) & 0xFF);
|
|
udata[3] = (byte) ((data.length >> 24) & 0xFF);
|
|
os.write(udata);
|
|
os.write(lzmaProperties);
|
|
} else {
|
|
os = new DeflaterOutputStream(os);
|
|
}
|
|
}
|
|
os.write(data);
|
|
} finally {
|
|
if (os != null) {
|
|
os.close();
|
|
}
|
|
}
|
|
}
|
|
|
|
public SWF(InputStream is, boolean parallelRead) throws IOException {
|
|
this(is, null, parallelRead);
|
|
}
|
|
|
|
/**
|
|
* Construct SWF from stream
|
|
*
|
|
* @param is Stream to read SWF from
|
|
* @param listener
|
|
* @param parallelRead Use parallel threads?
|
|
* @throws IOException
|
|
*/
|
|
public SWF(InputStream is, ProgressListener listener, boolean parallelRead) throws IOException {
|
|
this(is, listener, parallelRead, false);
|
|
}
|
|
|
|
/**
|
|
* Construct SWF from stream
|
|
*
|
|
* @param is Stream to read SWF from
|
|
* @param listener
|
|
* @param parallelRead Use parallel threads?
|
|
* @param checkOnly Check only file validity
|
|
* @throws IOException
|
|
*/
|
|
public SWF(InputStream is, ProgressListener listener, boolean parallelRead, boolean checkOnly) throws IOException {
|
|
byte hdr[] = new byte[3];
|
|
is.read(hdr);
|
|
String shdr = new String(hdr, "utf-8");
|
|
if ((!shdr.equals("FWS")) && (!shdr.equals("CWS")) && (!shdr.equals("ZWS"))) {
|
|
throw new IOException("Invalid SWF file");
|
|
}
|
|
version = is.read();
|
|
SWFInputStream sis = new SWFInputStream(is, version, 4);
|
|
fileSize = sis.readUI32();
|
|
|
|
if (hdr[0] == 'C') {
|
|
sis = new SWFInputStream(new InflaterInputStream(is), version, 8);
|
|
compressed = true;
|
|
}
|
|
|
|
if (hdr[0] == 'Z') {
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
sis.readUI32(); //outSize
|
|
int propertiesSize = 5;
|
|
lzmaProperties = new byte[propertiesSize];
|
|
if (sis.read(lzmaProperties, 0, propertiesSize) != propertiesSize) {
|
|
throw new IOException("LZMA:input .lzma file is too short");
|
|
}
|
|
long dictionarySize = 0;
|
|
for (int i = 0; i < 4; i++) {
|
|
dictionarySize += ((int) (lzmaProperties[1 + i]) & 0xFF) << (i * 8);
|
|
if (dictionarySize > Runtime.getRuntime().freeMemory()) {
|
|
throw new IOException("LZMA: Too large dictionary size");
|
|
}
|
|
}
|
|
|
|
SevenZip.Compression.LZMA.Decoder decoder = new SevenZip.Compression.LZMA.Decoder();
|
|
if (!decoder.SetDecoderProperties(lzmaProperties)) {
|
|
throw new IOException("LZMA:Incorrect stream properties");
|
|
}
|
|
|
|
if (!decoder.Code(sis, baos, fileSize - 8)) {
|
|
throw new IOException("LZMA:Error in data stream");
|
|
}
|
|
sis = new SWFInputStream(new ByteArrayInputStream(baos.toByteArray()), version, 8);
|
|
compressed = true;
|
|
lzma = true;
|
|
}
|
|
|
|
|
|
if (listener != null) {
|
|
sis.addPercentListener(listener);
|
|
}
|
|
sis.setPercentMax(fileSize);
|
|
displayRect = sis.readRECT();
|
|
// FIXED8 (16 bit fixed point) frameRate
|
|
sis.readUI8(); //tmpFirstByetOfFrameRate
|
|
frameRate = sis.readUI8();
|
|
frameCount = sis.readUI16();
|
|
if (checkOnly) {
|
|
return;
|
|
}
|
|
tags = sis.readTagList(this, 0, parallelRead, true, !checkOnly);
|
|
if (!checkOnly) {
|
|
assignExportNamesToSymbols();
|
|
assignClassesToSymbols();
|
|
findFileAttributes();
|
|
}
|
|
}
|
|
|
|
private void findFileAttributes() {
|
|
for (Tag t : tags) {
|
|
if (t instanceof FileAttributesTag) {
|
|
fileAttributes = (FileAttributesTag) t;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
private void assignExportNamesToSymbols() {
|
|
HashMap<Integer, String> exportNames = new HashMap<>();
|
|
for (Tag t : tags) {
|
|
if (t instanceof ExportAssetsTag) {
|
|
ExportAssetsTag eat = (ExportAssetsTag) t;
|
|
for (int i = 0; i < eat.tags.size(); i++) {
|
|
if ((!exportNames.containsKey(eat.tags.get(i))) && (!exportNames.containsValue(eat.names.get(i)))) {
|
|
exportNames.put(eat.tags.get(i), eat.names.get(i));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
for (Tag t : tags) {
|
|
if (t instanceof CharacterIdTag) {
|
|
CharacterIdTag ct = (CharacterIdTag) t;
|
|
if (exportNames.containsKey(ct.getCharacterId())) {
|
|
ct.setExportName(exportNames.get(ct.getCharacterId()));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public void assignClassesToSymbols() {
|
|
HashMap<Integer, String> classes = new HashMap<>();
|
|
for (Tag t : tags) {
|
|
if (t instanceof SymbolClassTag) {
|
|
SymbolClassTag sct = (SymbolClassTag) t;
|
|
for (int i = 0; i < sct.tagIDs.length; i++) {
|
|
if ((!classes.containsKey(sct.tagIDs[i])) && (!classes.containsValue(sct.classNames[i]))) {
|
|
classes.put(sct.tagIDs[i], sct.classNames[i]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
for (Tag t : tags) {
|
|
if (t instanceof CharacterIdTag) {
|
|
CharacterIdTag ct = (CharacterIdTag) t;
|
|
if (classes.containsKey(ct.getCharacterId())) {
|
|
ct.setClassName(classes.get(ct.getCharacterId()));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Compress SWF file
|
|
*
|
|
* @param fis Input stream
|
|
* @param fos Output stream
|
|
* @return True on success
|
|
*/
|
|
public static boolean fws2cws(InputStream fis, OutputStream fos) {
|
|
try {
|
|
byte swfHead[] = new byte[8];
|
|
fis.read(swfHead);
|
|
|
|
if (swfHead[0] != 'F') {
|
|
fis.close();
|
|
return false;
|
|
}
|
|
swfHead[0] = 'C';
|
|
fos.write(swfHead);
|
|
fos = new DeflaterOutputStream(fos);
|
|
int i;
|
|
while ((i = fis.read()) != -1) {
|
|
fos.write(i);
|
|
}
|
|
|
|
fis.close();
|
|
fos.close();
|
|
} catch (FileNotFoundException ex) {
|
|
return false;
|
|
} catch (IOException ex) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
public static boolean decompress(InputStream fis, OutputStream fos) {
|
|
try {
|
|
byte hdr[] = new byte[3];
|
|
fis.read(hdr);
|
|
String shdr = new String(hdr, "utf-8");
|
|
if (shdr.equals("CWS")) {
|
|
int version = fis.read();
|
|
SWFInputStream sis = new SWFInputStream(fis, version, 4);
|
|
long fileSize = sis.readUI32();
|
|
SWFOutputStream sos = new SWFOutputStream(fos, version);
|
|
sos.write("FWS".getBytes("utf-8"));
|
|
sos.writeUI8(version);
|
|
sos.writeUI32(fileSize);
|
|
InflaterInputStream iis = new InflaterInputStream(fis);
|
|
int i;
|
|
while ((i = iis.read()) != -1) {
|
|
fos.write(i);
|
|
}
|
|
|
|
fis.close();
|
|
fos.close();
|
|
} else if (shdr.equals("ZWS")) {
|
|
int version = fis.read();
|
|
SWFInputStream sis = new SWFInputStream(fis, version, 4);
|
|
long fileSize = sis.readUI32();
|
|
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
sis.readUI32(); //outSize
|
|
int propertiesSize = 5;
|
|
byte lzmaProperties[] = new byte[propertiesSize];
|
|
if (sis.read(lzmaProperties, 0, propertiesSize) != propertiesSize) {
|
|
throw new IOException("LZMA:input .lzma file is too short");
|
|
}
|
|
SevenZip.Compression.LZMA.Decoder decoder = new SevenZip.Compression.LZMA.Decoder();
|
|
if (!decoder.SetDecoderProperties(lzmaProperties)) {
|
|
throw new IOException("LZMA:Incorrect stream properties");
|
|
}
|
|
|
|
if (!decoder.Code(sis, baos, fileSize - 8)) {
|
|
throw new IOException("LZMA:Error in data stream");
|
|
}
|
|
try (SWFOutputStream sos = new SWFOutputStream(fos, version)) {
|
|
sos.write("FWS".getBytes("utf-8"));
|
|
sos.write(version);
|
|
sos.writeUI32(fileSize);
|
|
sos.write(baos.toByteArray());
|
|
}
|
|
fis.close();
|
|
fos.close();
|
|
} else {
|
|
return false;
|
|
}
|
|
} catch (FileNotFoundException ex) {
|
|
return false;
|
|
} catch (IOException ex) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
public boolean exportAS3Class(String className, String outdir, boolean isPcode, boolean parallel) throws Exception {
|
|
List<ABCContainerTag> abcTags = new ArrayList<>();
|
|
|
|
for (Tag t : tags) {
|
|
if (t instanceof ABCContainerTag) {
|
|
ABCContainerTag cnt = (ABCContainerTag) t;
|
|
abcTags.add(cnt);
|
|
}
|
|
}
|
|
for (int i = 0; i < abcTags.size(); i++) {
|
|
ABC abc = abcTags.get(i).getABC();
|
|
ScriptPack scr = abc.findScriptPackByPath(className);
|
|
if (scr != null) {
|
|
String cnt = "";
|
|
if (abc.script_info.length > 1) {
|
|
cnt = "script " + (i + 1) + "/" + abc.script_info.length + " ";
|
|
}
|
|
String exStr = "Exporting " + "tag " + (i + 1) + "/" + abcTags.size() + " " + cnt + scr.getPath() + " ...";
|
|
informListeners("export", exStr);
|
|
scr.export(outdir, abcTags, isPcode, parallel);
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private List<MyEntry<ClassPath, ScriptPack>> uniqueAS3Packs(List<MyEntry<ClassPath, ScriptPack>> packs) {
|
|
List<MyEntry<ClassPath, ScriptPack>> ret = new ArrayList<>();
|
|
for (MyEntry<ClassPath, ScriptPack> item : packs) {
|
|
for (MyEntry<ClassPath, ScriptPack> itemOld : ret) {
|
|
if (item.key.equals(itemOld.key)) {
|
|
Logger.getLogger(SWF.class.getName()).log(Level.SEVERE, "Duplicate pack path found!");
|
|
break;
|
|
}
|
|
}
|
|
ret.add(item);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public List<MyEntry<ClassPath, ScriptPack>> getAS3Packs() {
|
|
List<ABCContainerTag> abcTags = new ArrayList<>();
|
|
for (Tag t : tags) {
|
|
if (t instanceof ABCContainerTag) {
|
|
abcTags.add((ABCContainerTag) t);
|
|
}
|
|
}
|
|
List<MyEntry<ClassPath, ScriptPack>> packs = new ArrayList<>();
|
|
for (int i = 0; i < abcTags.size(); i++) {
|
|
ABCContainerTag t = abcTags.get(i);
|
|
packs.addAll(t.getABC().getScriptPacks());
|
|
}
|
|
return uniqueAS3Packs(packs);
|
|
}
|
|
|
|
private class ExportPackTask implements Callable<File> {
|
|
|
|
ScriptPack pack;
|
|
String directory;
|
|
List<ABCContainerTag> abcList;
|
|
boolean pcode;
|
|
ClassPath path;
|
|
AtomicInteger index;
|
|
int count;
|
|
boolean parallel;
|
|
AbortRetryIgnoreHandler handler;
|
|
long startTime;
|
|
long stopTime;
|
|
|
|
public ExportPackTask(AbortRetryIgnoreHandler handler, AtomicInteger index, int count, ClassPath path, ScriptPack pack, String directory, List<ABCContainerTag> abcList, boolean pcode, boolean parallel) {
|
|
this.pack = pack;
|
|
this.directory = directory;
|
|
this.abcList = abcList;
|
|
this.pcode = pcode;
|
|
this.path = path;
|
|
this.index = index;
|
|
this.count = count;
|
|
this.parallel = parallel;
|
|
this.handler = handler;
|
|
}
|
|
|
|
@Override
|
|
public File call() throws Exception {
|
|
RunnableIOExResult<File> rio = new RunnableIOExResult<File>() {
|
|
@Override
|
|
public void run() throws IOException {
|
|
startTime = System.currentTimeMillis();
|
|
this.result = pack.export(directory, abcList, pcode, parallel);
|
|
stopTime = System.currentTimeMillis();
|
|
}
|
|
};
|
|
new RetryTask(rio, handler).run();
|
|
synchronized (ABC.class) {
|
|
long time = stopTime - startTime;
|
|
informListeners("export", "Exported script " + index.getAndIncrement() + "/" + count + " " + path + ", " + Helper.formatTimeSec(time));
|
|
}
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public List<File> exportActionScript2(AbortRetryIgnoreHandler handler, String outdir, boolean isPcode, boolean parallel, EventListener evl) throws IOException {
|
|
List<File> ret = new ArrayList<>();
|
|
List<Object> list2 = new ArrayList<>();
|
|
list2.addAll(tags);
|
|
List<TagNode> list = createASTagList(list2, null);
|
|
|
|
TagNode.setExport(list, true);
|
|
if (!outdir.endsWith(File.separator)) {
|
|
outdir += File.separator;
|
|
}
|
|
outdir += "scripts" + File.separator;
|
|
AtomicInteger cnt = new AtomicInteger(1);
|
|
int totalCount = TagNode.getTagCountRecursive(list);
|
|
ret.addAll(TagNode.exportNodeAS(tags, handler, list, outdir, isPcode, cnt, totalCount, evl));
|
|
return ret;
|
|
}
|
|
|
|
public List<File> exportActionScript3(AbortRetryIgnoreHandler handler, String outdir, boolean isPcode, boolean parallel) {
|
|
ExecutorService executor = Executors.newFixedThreadPool(parallel ? 20 : 1);
|
|
List<Future<File>> futureResults = new ArrayList<>();
|
|
AtomicInteger cnt = new AtomicInteger(1);
|
|
List<ABCContainerTag> abcTags = new ArrayList<>();
|
|
for (Tag t : tags) {
|
|
if (t instanceof ABCContainerTag) {
|
|
abcTags.add((ABCContainerTag) t);
|
|
}
|
|
}
|
|
List<MyEntry<ClassPath, ScriptPack>> packs = getAS3Packs();
|
|
for (MyEntry<ClassPath, ScriptPack> item : packs) {
|
|
Future<File> future = executor.submit(new ExportPackTask(handler, cnt, packs.size(), item.key, item.value, outdir, abcTags, isPcode, parallel));
|
|
futureResults.add(future);
|
|
}
|
|
|
|
List<File> ret = new ArrayList<>();
|
|
for (int f = 0; f < futureResults.size(); f++) {
|
|
try {
|
|
ret.add(futureResults.get(f).get());
|
|
} catch (InterruptedException | ExecutionException ex) {
|
|
Logger.getLogger(SWF.class.getName()).log(Level.SEVERE, "Error during ABC export", ex);
|
|
}
|
|
}
|
|
|
|
try {
|
|
executor.shutdown();
|
|
executor.awaitTermination(Configuration.DECOMPILATION_TIMEOUT, TimeUnit.SECONDS);
|
|
} catch (InterruptedException ex) {
|
|
Logger.getLogger(ABC.class.getName()).log(Level.SEVERE, Helper.formatTimeToText(Configuration.DECOMPILATION_TIMEOUT) + " ActionScript export limit reached", ex);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public List<File> exportActionScript(AbortRetryIgnoreHandler handler, String outdir, boolean isPcode, boolean parallel) throws Exception {
|
|
boolean asV3Found = false;
|
|
List<File> ret = new ArrayList<>();
|
|
final EventListener evl = new EventListener() {
|
|
@Override
|
|
public void handleEvent(String event, Object data) {
|
|
if (event.equals("export")) {
|
|
informListeners(event, data);
|
|
}
|
|
}
|
|
};
|
|
for (Tag t : tags) {
|
|
if (t instanceof ABCContainerTag) {
|
|
asV3Found = true;
|
|
}
|
|
}
|
|
|
|
if (asV3Found) {
|
|
ret.addAll(exportActionScript3(handler, outdir, isPcode, parallel));
|
|
} else {
|
|
ret.addAll(exportActionScript2(handler, outdir, isPcode, parallel, evl));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public static List<TagNode> createASTagList(List<Object> list, Object parent) {
|
|
List<TagNode> ret = new ArrayList<>();
|
|
int frame = 1;
|
|
List<TagNode> frames = new ArrayList<>();
|
|
|
|
List<ExportAssetsTag> exportAssetsTags = new ArrayList<>();
|
|
for (Object t : list) {
|
|
if (t instanceof ExportAssetsTag) {
|
|
exportAssetsTags.add((ExportAssetsTag) t);
|
|
}
|
|
TagNode addNode = null;
|
|
if (t instanceof ShowFrameTag) {
|
|
TagNode tti = new TagNode(new FrameNode(frame, parent, false));
|
|
|
|
for (int r = ret.size() - 1; r >= 0; r--) {
|
|
if (!(ret.get(r).tag instanceof DefineSpriteTag)) {
|
|
if (!(ret.get(r).tag instanceof DefineButtonTag)) {
|
|
if (!(ret.get(r).tag instanceof DefineButton2Tag)) {
|
|
if (!(ret.get(r).tag instanceof DoInitActionTag)) {
|
|
if (!(ret.get(r).tag instanceof PackageNode)) {
|
|
tti.subItems.add(ret.get(r));
|
|
ret.remove(r);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
frame++;
|
|
frames.add(tti);
|
|
} else if (t instanceof ASMSource) {
|
|
TagNode tti = new TagNode(t);
|
|
//ret.add(tti);
|
|
addNode = tti;
|
|
} else if (t instanceof Container) {
|
|
if (((Container) t).getItemCount() > 0) {
|
|
|
|
TagNode tti = new TagNode(t);
|
|
List<Object> subItems = ((Container) t).getSubItems();
|
|
|
|
tti.subItems = createASTagList(subItems, t);
|
|
addNode = tti;
|
|
//ret.add(tti);
|
|
}
|
|
}
|
|
if (addNode != null) {
|
|
if (addNode.tag instanceof CharacterIdTag) {
|
|
CharacterIdTag cit = (CharacterIdTag) addNode.tag;
|
|
String path = cit.getExportName();
|
|
if (path == null) {
|
|
path = "";
|
|
}
|
|
String pathParts[];
|
|
if (path.contains(".")) {
|
|
pathParts = path.split("\\.");
|
|
} else {
|
|
pathParts = new String[]{path};
|
|
}
|
|
List<TagNode> items = ret;
|
|
int pos = 0;
|
|
TagNode selNode = null;
|
|
do {
|
|
if (pos == pathParts.length - 1) {
|
|
break;
|
|
}
|
|
selNode = null;
|
|
for (TagNode node : items) {
|
|
if (node.tag instanceof PackageNode) {
|
|
PackageNode pkg = (PackageNode) node.tag;
|
|
if (pkg.packageName.equals(pathParts[pos])) {
|
|
selNode = node;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
int pkgCount = 0;
|
|
for (; pkgCount < items.size(); pkgCount++) {
|
|
if (items.get(pkgCount).tag instanceof PackageNode) {
|
|
PackageNode pkg = (PackageNode) items.get(pkgCount).tag;
|
|
if (pkg.packageName.compareTo(pathParts[pos]) > 0) {
|
|
break;
|
|
}
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
if (selNode == null) {
|
|
items.add(pkgCount, selNode = new TagNode(new PackageNode(pathParts[pos])));
|
|
}
|
|
pos++;
|
|
if (selNode != null) {
|
|
items = selNode.subItems;
|
|
}
|
|
|
|
} while (selNode != null);
|
|
|
|
int clsCount = 0;
|
|
for (; clsCount < items.size(); clsCount++) {
|
|
if (items.get(clsCount).tag instanceof CharacterIdTag) {
|
|
CharacterIdTag ct = (CharacterIdTag) items.get(clsCount).tag;
|
|
String expName = ct.getExportName();
|
|
if (expName == null) {
|
|
expName = "";
|
|
}
|
|
if (expName.contains(".")) {
|
|
expName = expName.substring(expName.lastIndexOf(".") + 1);
|
|
}
|
|
if ((ct.getClass().getName() + "_" + expName).compareTo(addNode.tag.getClass().getName() + "_" + pathParts[pos]) > 0) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
items.add(clsCount, addNode);
|
|
} else {
|
|
ret.add(addNode);
|
|
}
|
|
}
|
|
|
|
}
|
|
ret.addAll(frames);
|
|
for (int i = ret.size() - 1; i >= 0; i--) {
|
|
if (ret.get(i).tag instanceof DefineSpriteTag) {
|
|
((DefineSpriteTag) ret.get(i).tag).exportAssetsTags = exportAssetsTags;
|
|
}
|
|
if (ret.get(i).tag instanceof DefineButtonTag) {
|
|
((DefineButtonTag) ret.get(i).tag).exportAssetsTags = exportAssetsTags;
|
|
}
|
|
if (ret.get(i).tag instanceof DefineButton2Tag) {
|
|
((DefineButton2Tag) ret.get(i).tag).exportAssetsTags = exportAssetsTags;
|
|
}
|
|
/*if (ret.get(i).tag instanceof DoInitActionTag) {
|
|
//((DoInitActionTag) ret.get(i).tag).exportAssetsTags = exportAssetsTags;
|
|
}*/
|
|
if (ret.get(i).tag instanceof ASMSource) {
|
|
ASMSource ass = (ASMSource) ret.get(i).tag;
|
|
if (ass.containsSource()) {
|
|
continue;
|
|
}
|
|
}
|
|
if (ret.get(i).subItems.isEmpty()) {
|
|
ret.remove(i);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
private HashSet<EventListener> listeners = new HashSet<>();
|
|
|
|
public final void addEventListener(EventListener listener) {
|
|
listeners.add(listener);
|
|
for (Tag t : tags) {
|
|
if (t instanceof ABCContainerTag) {
|
|
(((ABCContainerTag) t).getABC()).addEventListener(listener);
|
|
}
|
|
}
|
|
}
|
|
|
|
public final void removeEventListener(EventListener listener) {
|
|
listeners.remove(listener);
|
|
for (Tag t : tags) {
|
|
if (t instanceof ABCContainerTag) {
|
|
(((ABCContainerTag) t).getABC()).removeEventListener(listener);
|
|
}
|
|
}
|
|
}
|
|
|
|
protected void informListeners(String event, Object data) {
|
|
for (EventListener listener : listeners) {
|
|
listener.handleEvent(event, data);
|
|
}
|
|
}
|
|
|
|
public static boolean hasErrorHeader(byte data[]) {
|
|
if (data.length > 4) {
|
|
if ((data[0] & 0xff) == 0xff) {
|
|
if ((data[1] & 0xff) == 0xd9) {
|
|
if ((data[2] & 0xff) == 0xff) {
|
|
if ((data[3] & 0xff) == 0xd8) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public static void populateSoundStreamBlocks(List<Object> tags, Tag head, List<SoundStreamBlockTag> output) {
|
|
boolean found = false;
|
|
for (Object t : tags) {
|
|
if (t == head) {
|
|
found = true;
|
|
continue;
|
|
}
|
|
if (!found) {
|
|
continue;
|
|
}
|
|
if (t instanceof SoundStreamBlockTag) {
|
|
output.add((SoundStreamBlockTag) t);
|
|
}
|
|
if (t instanceof SoundStreamHeadTypeTag) {
|
|
break;
|
|
}
|
|
if (t instanceof Container) {
|
|
populateSoundStreamBlocks(((Container) t).getSubItems(), head, output);
|
|
}
|
|
}
|
|
}
|
|
|
|
public void populateVideoFrames(int streamId, List<Object> tags, HashMap<Integer, VideoFrameTag> output) {
|
|
for (Object t : tags) {
|
|
if (t instanceof VideoFrameTag) {
|
|
output.put(((VideoFrameTag) t).frameNum, (VideoFrameTag) t);
|
|
}
|
|
if (t instanceof Container) {
|
|
populateVideoFrames(streamId, ((Container) t).getSubItems(), output);
|
|
}
|
|
}
|
|
}
|
|
|
|
public void exportMovies(AbortRetryIgnoreHandler handler, String outdir) throws IOException {
|
|
exportMovies(handler, outdir, tags);
|
|
}
|
|
|
|
public void exportSounds(AbortRetryIgnoreHandler handler, String outdir, boolean mp3, boolean wave) throws IOException {
|
|
exportSounds(handler, outdir, tags, mp3, wave);
|
|
}
|
|
|
|
public byte[] exportSound(Tag t) throws IOException {
|
|
boolean mp3 = true;
|
|
boolean wave = true;
|
|
ByteArrayOutputStream fos = new ByteArrayOutputStream();
|
|
|
|
if (t instanceof DefineSoundTag) {
|
|
DefineSoundTag st = (DefineSoundTag) t;
|
|
if ((st.soundFormat == DefineSoundTag.FORMAT_ADPCM) && wave) {
|
|
fos = new ByteArrayOutputStream();
|
|
createWavFromAdpcm(fos, st.soundRate, st.soundSize, st.soundType, st.soundData);
|
|
} else if ((st.soundFormat == DefineSoundTag.FORMAT_MP3) && mp3) {
|
|
fos = new ByteArrayOutputStream();
|
|
fos.write(st.soundData, 2, st.soundData.length - 2);
|
|
} else {
|
|
fos = new ByteArrayOutputStream();
|
|
FLVOutputStream flv = new FLVOutputStream(fos);
|
|
flv.writeHeader(true, false);
|
|
flv.writeTag(new FLVTAG(0, new AUDIODATA(st.soundFormat, st.soundRate, st.soundSize, st.soundType, st.soundData)));
|
|
}
|
|
}
|
|
if (t instanceof SoundStreamHeadTypeTag) {
|
|
SoundStreamHeadTypeTag shead = (SoundStreamHeadTypeTag) t;
|
|
List<SoundStreamBlockTag> blocks = new ArrayList<>();
|
|
List<Object> objs = new ArrayList<Object>(this.tags);
|
|
populateSoundStreamBlocks(objs, t, blocks);
|
|
if ((shead.getSoundFormat() == DefineSoundTag.FORMAT_ADPCM) && wave) {
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
for (int b = 0; b < blocks.size(); b++) {
|
|
byte data[] = blocks.get(b).getData(SWF.DEFAULT_VERSION);
|
|
baos.write(data);
|
|
}
|
|
fos = new ByteArrayOutputStream();
|
|
createWavFromAdpcm(fos, shead.getSoundRate(), shead.getSoundSize(), shead.getSoundType(), baos.toByteArray());
|
|
} else if ((shead.getSoundFormat() == DefineSoundTag.FORMAT_MP3) && mp3) {
|
|
fos = new ByteArrayOutputStream();
|
|
for (int b = 0; b < blocks.size(); b++) {
|
|
byte data[] = blocks.get(b).getData(SWF.DEFAULT_VERSION);
|
|
fos.write(data, 4, data.length - 4);
|
|
}
|
|
} else {
|
|
fos = new ByteArrayOutputStream();
|
|
FLVOutputStream flv = new FLVOutputStream(fos);
|
|
flv.writeHeader(true, false);
|
|
|
|
int ms = (int) (1000.0f / ((float) frameRate));
|
|
for (int b = 0; b < blocks.size(); b++) {
|
|
byte data[] = blocks.get(b).getData(SWF.DEFAULT_VERSION);
|
|
if (shead.getSoundFormat() == 2) { //MP3
|
|
data = Arrays.copyOfRange(data, 4, data.length);
|
|
}
|
|
flv.writeTag(new FLVTAG(ms * b, new AUDIODATA(shead.getSoundFormat(), shead.getSoundRate(), shead.getSoundSize(), shead.getSoundType(), data)));
|
|
}
|
|
}
|
|
}
|
|
return fos.toByteArray();
|
|
}
|
|
|
|
private static void writeLE(OutputStream os, long val, int size) throws IOException {
|
|
for (int i = 0; i < size; i++) {
|
|
os.write((int) (val & 0xff));
|
|
val = val >> 8;
|
|
}
|
|
}
|
|
|
|
private static void createWavFromAdpcm(OutputStream fos, int soundRate, int soundSize, int soundType, byte data[]) throws IOException {
|
|
try {
|
|
byte pcmData[] = AdpcmDecoder.decode(data, soundType == 1 ? true : false);
|
|
|
|
ByteArrayOutputStream subChunk1Data = new ByteArrayOutputStream();
|
|
int audioFormat = 1; //PCM
|
|
writeLE(subChunk1Data, audioFormat, 2);
|
|
int numChannels = soundType == 1 ? 2 : 1;
|
|
writeLE(subChunk1Data, numChannels, 2);
|
|
int rateMap[] = {5512, 11025, 22050, 44100};
|
|
int sampleRate = rateMap[soundRate];
|
|
writeLE(subChunk1Data, sampleRate, 4);
|
|
int bitsPerSample = soundSize == 1 ? 16 : 8;
|
|
int byteRate = sampleRate * numChannels * bitsPerSample / 8;
|
|
writeLE(subChunk1Data, byteRate, 4);
|
|
int blockAlign = numChannels * bitsPerSample / 8;
|
|
writeLE(subChunk1Data, blockAlign, 2);
|
|
writeLE(subChunk1Data, bitsPerSample, 2);
|
|
|
|
ByteArrayOutputStream chunks = new ByteArrayOutputStream();
|
|
chunks.write("fmt ".getBytes("utf-8"));
|
|
byte subChunk1DataBytes[] = subChunk1Data.toByteArray();
|
|
writeLE(chunks, subChunk1DataBytes.length, 4);
|
|
chunks.write(subChunk1DataBytes);
|
|
|
|
|
|
chunks.write("data".getBytes("utf-8"));
|
|
writeLE(chunks, pcmData.length, 4);
|
|
chunks.write(pcmData);
|
|
|
|
fos.write("RIFF".getBytes("utf-8"));
|
|
byte chunkBytes[] = chunks.toByteArray();
|
|
writeLE(fos, 4 + chunkBytes.length, 4);
|
|
fos.write("WAVE".getBytes("utf-8"));
|
|
fos.write(chunkBytes);
|
|
//size1=>16bit*/
|
|
} finally {
|
|
if (fos != null) {
|
|
try {
|
|
fos.close();
|
|
} catch (IOException ex) {
|
|
//ignore
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public List<File> exportSounds(AbortRetryIgnoreHandler handler, String outdir, List<Tag> tags, boolean mp3, boolean wave) throws IOException {
|
|
List<File> ret = new ArrayList<>();
|
|
if (tags.isEmpty()) {
|
|
return ret;
|
|
}
|
|
File foutdir = new File(outdir);
|
|
if (!foutdir.exists()) {
|
|
if (!foutdir.mkdirs()) {
|
|
if (!foutdir.exists()) {
|
|
throw new IOException("Cannot create directory " + outdir);
|
|
}
|
|
}
|
|
}
|
|
for (Tag t : tags) {
|
|
File newfile = null;
|
|
FileOutputStream fos = null;
|
|
//File file = null;
|
|
try {
|
|
int id = 0;
|
|
if (t instanceof DefineSoundTag) {
|
|
id = ((DefineSoundTag) t).soundId;
|
|
}
|
|
|
|
|
|
if (t instanceof DefineSoundTag) {
|
|
final DefineSoundTag st = (DefineSoundTag) t;
|
|
|
|
if ((st.soundFormat == DefineSoundTag.FORMAT_ADPCM) && wave) {
|
|
final File file = new File(outdir + File.separator + st.getCharacterExportFileName() + ".wav");
|
|
newfile = file;
|
|
new RetryTask(new RunnableIOEx() {
|
|
@Override
|
|
public void run() throws IOException {
|
|
FileOutputStream fos = new FileOutputStream(file);
|
|
createWavFromAdpcm(fos, st.soundRate, st.soundSize, st.soundType, st.soundData);
|
|
}
|
|
}, handler).run();
|
|
} else if ((st.soundFormat == DefineSoundTag.FORMAT_MP3) && mp3) {
|
|
final File file = new File(outdir + File.separator + st.getCharacterExportFileName() + ".mp3");
|
|
newfile = file;
|
|
new RetryTask(new RunnableIOEx() {
|
|
@Override
|
|
public void run() throws IOException {
|
|
FileOutputStream fos = new FileOutputStream(file);
|
|
fos.write(st.soundData, 2, st.soundData.length - 2);
|
|
}
|
|
}, handler).run();
|
|
} else {
|
|
final File file = new File(outdir + File.separator + st.getCharacterExportFileName() + ".flv");
|
|
newfile = file;
|
|
new RetryTask(new RunnableIOEx() {
|
|
@Override
|
|
public void run() throws IOException {
|
|
FileOutputStream fos = new FileOutputStream(file);
|
|
try (FLVOutputStream flv = new FLVOutputStream(fos)) {
|
|
flv.writeHeader(true, false);
|
|
flv.writeTag(new FLVTAG(0, new AUDIODATA(st.soundFormat, st.soundRate, st.soundSize, st.soundType, st.soundData)));
|
|
}
|
|
}
|
|
}, handler).run();
|
|
}
|
|
}
|
|
if (t instanceof SoundStreamHeadTypeTag) {
|
|
final SoundStreamHeadTypeTag shead = (SoundStreamHeadTypeTag) t;
|
|
final List<SoundStreamBlockTag> blocks = new ArrayList<>();
|
|
List<Object> objs = new ArrayList<Object>(this.tags);
|
|
populateSoundStreamBlocks(objs, t, blocks);
|
|
if ((shead.getSoundFormat() == DefineSoundTag.FORMAT_ADPCM) && wave) {
|
|
final ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
for (int b = 0; b < blocks.size(); b++) {
|
|
byte data[] = blocks.get(b).getData(SWF.DEFAULT_VERSION);
|
|
baos.write(data);
|
|
}
|
|
final File file = new File(outdir + File.separator + id + ".wav");
|
|
newfile = file;
|
|
new RetryTask(new RunnableIOEx() {
|
|
@Override
|
|
public void run() throws IOException {
|
|
FileOutputStream fos = new FileOutputStream(file);
|
|
createWavFromAdpcm(fos, shead.getSoundRate(), shead.getSoundSize(), shead.getSoundType(), baos.toByteArray());
|
|
}
|
|
}, handler).run();
|
|
} else if ((shead.getSoundFormat() == DefineSoundTag.FORMAT_MP3) && mp3) {
|
|
final File file = new File(outdir + File.separator + id + ".mp3");
|
|
newfile = file;
|
|
new RetryTask(new RunnableIOEx() {
|
|
@Override
|
|
public void run() throws IOException {
|
|
FileOutputStream fos = new FileOutputStream(file);
|
|
for (int b = 0; b < blocks.size(); b++) {
|
|
byte data[] = blocks.get(b).getData(SWF.DEFAULT_VERSION);
|
|
fos.write(data, 2, data.length - 2);
|
|
}
|
|
}
|
|
}, handler).run();
|
|
} else {
|
|
final File file = new File(outdir + File.separator + id + ".flv");
|
|
newfile = file;
|
|
new RetryTask(new RunnableIOEx() {
|
|
@Override
|
|
public void run() throws IOException {
|
|
FileOutputStream fos = new FileOutputStream(file);
|
|
FLVOutputStream flv = new FLVOutputStream(fos);
|
|
flv.writeHeader(true, false);
|
|
|
|
int ms = (int) (1000.0f / ((float) frameRate));
|
|
for (int b = 0; b < blocks.size(); b++) {
|
|
byte data[] = blocks.get(b).getData(SWF.DEFAULT_VERSION);
|
|
if (shead.getSoundFormat() == 2) { //MP3
|
|
data = Arrays.copyOfRange(data, 4, data.length);
|
|
}
|
|
flv.writeTag(new FLVTAG(ms * b, new AUDIODATA(shead.getSoundFormat(), shead.getSoundRate(), shead.getSoundSize(), shead.getSoundType(), data)));
|
|
}
|
|
}
|
|
}, handler).run();
|
|
}
|
|
}
|
|
} finally {
|
|
if (fos != null) {
|
|
try {
|
|
fos.close();
|
|
} catch (Exception ignore) {
|
|
//ignore
|
|
}
|
|
}
|
|
}
|
|
if (newfile != null) {
|
|
ret.add(newfile);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public byte[] exportMovie(DefineVideoStreamTag videoStream) throws IOException {
|
|
HashMap<Integer, VideoFrameTag> frames = new HashMap<>();
|
|
List<Object> os = new ArrayList<Object>(this.tags);
|
|
populateVideoFrames(videoStream.characterID, os, frames);
|
|
if (frames.isEmpty()) {
|
|
return new byte[0];
|
|
}
|
|
|
|
|
|
//double ms = 1000.0f / ((float) frameRate);
|
|
|
|
ByteArrayOutputStream fos = new ByteArrayOutputStream();
|
|
//CopyOutputStream cos = new CopyOutputStream(fos, new FileInputStream("f:\\trunk\\testdata\\xfl\\xfl\\_obj\\streamvideo 7.flv"));
|
|
OutputStream tos = fos;
|
|
FLVOutputStream flv = new FLVOutputStream(tos);
|
|
flv.writeHeader(false, true);
|
|
//flv.writeTag(new FLVTAG(0, SCRIPTDATA.onMetaData(ms * frames.size() / 1000.0, videoStream.width, videoStream.height, 0, frameRate, videoStream.codecID, 0, 0, false, 0, fileSize)));
|
|
int horizontalAdjustment = 0;
|
|
int verticalAdjustment = 0;
|
|
for (int i = 0; i < frames.size(); i++) {
|
|
VideoFrameTag tag = frames.get(i);
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
|
|
int frameType = 1;
|
|
|
|
if ((videoStream.codecID == DefineVideoStreamTag.CODEC_VP6)
|
|
|| (videoStream.codecID == DefineVideoStreamTag.CODEC_VP6_ALPHA)) {
|
|
SWFInputStream sis = new SWFInputStream(new ByteArrayInputStream(tag.videoData), SWF.DEFAULT_VERSION);
|
|
if (videoStream.codecID == DefineVideoStreamTag.CODEC_VP6_ALPHA) {
|
|
sis.readUI24(); //offsetToAlpha
|
|
}
|
|
int frameMode = (int) sis.readUB(1);
|
|
|
|
if (frameMode == 0) {
|
|
frameType = 1; //intra
|
|
} else {
|
|
frameType = 2; //inter
|
|
}
|
|
sis.readUB(6); //qp
|
|
int marker = (int) sis.readUB(1);
|
|
if (frameMode == 0) {
|
|
int version = (int) sis.readUB(5);
|
|
int version2 = (int) sis.readUB(2);
|
|
sis.readUB(1);//interlace
|
|
if (marker == 1 || version2 == 0) {
|
|
sis.readUI16();//offset
|
|
}
|
|
int dim_y = sis.readUI8();
|
|
int dim_x = sis.readUI8();
|
|
sis.readUI8(); //render_y
|
|
sis.readUI8(); //render_x
|
|
horizontalAdjustment = (int) (dim_x * Math.ceil(((double) videoStream.width) / (double) dim_x)) - videoStream.width;
|
|
verticalAdjustment = (int) (dim_y * Math.ceil(((double) videoStream.height) / (double) dim_y)) - videoStream.height;
|
|
|
|
}
|
|
|
|
SWFOutputStream sos = new SWFOutputStream(baos, SWF.DEFAULT_VERSION);
|
|
sos.writeUB(4, horizontalAdjustment);
|
|
sos.writeUB(4, verticalAdjustment);
|
|
}
|
|
if (videoStream.codecID == DefineVideoStreamTag.CODEC_SORENSON_H263) {
|
|
SWFInputStream sis = new SWFInputStream(new ByteArrayInputStream(tag.videoData), SWF.DEFAULT_VERSION);
|
|
sis.readUB(17);//pictureStartCode
|
|
sis.readUB(5); //version
|
|
sis.readUB(8); //temporalReference
|
|
int pictureSize = (int) sis.readUB(3); //pictureSize
|
|
if (pictureSize == 0) {
|
|
sis.readUB(8); //customWidth
|
|
sis.readUB(8); //customHeight
|
|
}
|
|
if (pictureSize == 1) {
|
|
sis.readUB(16); //customWidth
|
|
sis.readUB(16); //customHeight
|
|
}
|
|
int pictureType = (int) sis.readUB(2);
|
|
switch (pictureType) {
|
|
case 0: //intra
|
|
frameType = 1; //keyframe
|
|
break;
|
|
case 1://inter
|
|
frameType = 2;
|
|
break;
|
|
case 2: //disposable
|
|
frameType = 3;
|
|
break;
|
|
}
|
|
}
|
|
|
|
baos.write(tag.videoData);
|
|
flv.writeTag(new FLVTAG((int) Math.floor(i * 1000.0f / ((float) frameRate)), new VIDEODATA(frameType, videoStream.codecID, baos.toByteArray())));
|
|
}
|
|
return fos.toByteArray();
|
|
}
|
|
|
|
public List<File> exportMovies(AbortRetryIgnoreHandler handler, String outdir, List<Tag> tags) throws IOException {
|
|
List<File> ret = new ArrayList<>();
|
|
if (tags.isEmpty()) {
|
|
return ret;
|
|
}
|
|
File foutdir = new File(outdir);
|
|
if (!foutdir.exists()) {
|
|
if (!foutdir.mkdirs()) {
|
|
if (!foutdir.exists()) {
|
|
throw new IOException("Cannot create directory " + outdir);
|
|
}
|
|
}
|
|
}
|
|
for (Tag t : tags) {
|
|
if (t instanceof DefineVideoStreamTag) {
|
|
final DefineVideoStreamTag videoStream = (DefineVideoStreamTag) t;
|
|
final File file = new File(outdir + File.separator + ((DefineVideoStreamTag) t).getCharacterExportFileName() + ".flv");
|
|
new RetryTask(new RunnableIOEx() {
|
|
@Override
|
|
public void run() throws IOException {
|
|
try (FileOutputStream fos = new FileOutputStream(file)) {
|
|
fos.write(exportMovie(videoStream));
|
|
}
|
|
}
|
|
}, handler).run();
|
|
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public List<File> exportTexts(AbortRetryIgnoreHandler handler, String outdir, List<Tag> tags, final boolean formatted) throws IOException {
|
|
List<File> ret = new ArrayList<>();
|
|
if (tags.isEmpty()) {
|
|
return ret;
|
|
}
|
|
File foutdir = new File(outdir);
|
|
if (!foutdir.exists()) {
|
|
if (!foutdir.mkdirs()) {
|
|
if (!foutdir.exists()) {
|
|
throw new IOException("Cannot create directory " + outdir);
|
|
}
|
|
}
|
|
}
|
|
for (final Tag t : tags) {
|
|
if (t instanceof TextTag) {
|
|
final File file = new File(outdir + File.separator + ((TextTag) t).getCharacterId() + ".txt");
|
|
final List<Tag> ttags = this.tags;
|
|
new RetryTask(new RunnableIOEx() {
|
|
@Override
|
|
public void run() throws IOException {
|
|
try (FileOutputStream fos = new FileOutputStream(file)) {
|
|
if (formatted) {
|
|
fos.write(((TextTag) t).getFormattedText(ttags).getBytes("UTF-8"));
|
|
} else {
|
|
fos.write(((TextTag) t).getText(ttags).getBytes("UTF-8"));
|
|
}
|
|
}
|
|
}
|
|
}, handler).run();
|
|
ret.add(file);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public void exportTexts(AbortRetryIgnoreHandler handler, String outdir, boolean formatted) throws IOException {
|
|
exportTexts(handler, outdir, tags, formatted);
|
|
}
|
|
|
|
public static List<File> exportShapes(AbortRetryIgnoreHandler handler, String outdir, List<Tag> tags) throws IOException {
|
|
List<File> ret = new ArrayList<>();
|
|
if (tags.isEmpty()) {
|
|
return ret;
|
|
}
|
|
File foutdir = new File(outdir);
|
|
if (!foutdir.exists()) {
|
|
if (!foutdir.mkdirs()) {
|
|
if (!foutdir.exists()) {
|
|
throw new IOException("Cannot create directory " + outdir);
|
|
}
|
|
}
|
|
}
|
|
loopb:
|
|
for (final Tag t : tags) {
|
|
if (t instanceof ShapeTag) {
|
|
int characterID = 0;
|
|
if (t instanceof CharacterTag) {
|
|
characterID = ((CharacterTag) t).getCharacterId();
|
|
}
|
|
final File file = new File(outdir + File.separator + characterID + ".svg");
|
|
new RetryTask(new RunnableIOEx() {
|
|
@Override
|
|
public void run() throws IOException {
|
|
try (FileOutputStream fos = new FileOutputStream(file)) {
|
|
fos.write(((ShapeTag) t).toSVG().getBytes("utf-8"));
|
|
}
|
|
}
|
|
}, handler).run();
|
|
ret.add(file);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public static List<File> exportBinaryData(AbortRetryIgnoreHandler handler, String outdir, List<Tag> tags) throws IOException {
|
|
List<File> ret = new ArrayList<>();
|
|
if (tags.isEmpty()) {
|
|
return ret;
|
|
}
|
|
File foutdir = new File(outdir);
|
|
if (!foutdir.exists()) {
|
|
if (!foutdir.mkdirs()) {
|
|
if (!foutdir.exists()) {
|
|
throw new IOException("Cannot create directory " + outdir);
|
|
}
|
|
}
|
|
}
|
|
loopb:
|
|
for (final Tag t : tags) {
|
|
if (t instanceof DefineBinaryDataTag) {
|
|
int characterID = ((DefineBinaryDataTag) t).getCharacterId();
|
|
final File file = new File(outdir + File.separator + characterID + ".bin");
|
|
new RetryTask(new RunnableIOEx() {
|
|
@Override
|
|
public void run() throws IOException {
|
|
try (FileOutputStream fos = new FileOutputStream(file)) {
|
|
fos.write(((DefineBinaryDataTag) t).binaryData);
|
|
}
|
|
}
|
|
}, handler).run();
|
|
ret.add(file);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public List<File> exportImages(AbortRetryIgnoreHandler handler, String outdir, List<Tag> tags) throws IOException {
|
|
List<File> ret = new ArrayList<>();
|
|
if (tags.isEmpty()) {
|
|
return ret;
|
|
}
|
|
File foutdir = new File(outdir);
|
|
if (!foutdir.exists()) {
|
|
if (!foutdir.mkdirs()) {
|
|
if (!foutdir.exists()) {
|
|
throw new IOException("Cannot create directory " + outdir);
|
|
}
|
|
}
|
|
}
|
|
for (final Tag t : tags) {
|
|
if (t instanceof ImageTag) {
|
|
final File file = new File(outdir + File.separator + ((ImageTag) t).getCharacterId() + "." + ((ImageTag) t).getImageFormat());
|
|
final List<Tag> ttags = this.tags;
|
|
new RetryTask(new RunnableIOEx() {
|
|
@Override
|
|
public void run() throws IOException {
|
|
ImageIO.write(((ImageTag) t).getImage(ttags), ((ImageTag) t).getImageFormat().toUpperCase(Locale.ENGLISH), file);
|
|
}
|
|
}, handler).run();
|
|
ret.add(file);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public void exportImages(AbortRetryIgnoreHandler handler, String outdir) throws IOException {
|
|
exportImages(handler, outdir, tags);
|
|
}
|
|
|
|
public void exportShapes(AbortRetryIgnoreHandler handler, String outdir) throws IOException {
|
|
exportShapes(handler, outdir, tags);
|
|
}
|
|
|
|
public void exportBinaryData(AbortRetryIgnoreHandler handler, String outdir) throws IOException {
|
|
exportBinaryData(handler, outdir, tags);
|
|
}
|
|
public static final String[] reservedWords = {
|
|
"as", "break", "case", "catch", "class", "const", "continue", "default", "delete", "do", "each", "else",
|
|
"extends", "false", "finally", "for", "function", "get", "if", "implements", "import", "in", "instanceof",
|
|
"interface", "internal", "is", "native", "new", "null", "override", "package", "private", "protected", "public",
|
|
"return", "set", "super", "switch", "this", "throw", "true", "try", "typeof", "use", "var", /*"void",*/ "while",
|
|
"with", "dynamic", "default", "final", "in"};
|
|
|
|
private boolean isReserved(String s) {
|
|
for (String rw : reservedWords) {
|
|
if (rw.equals(s.trim())) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
private HashMap<String, String> deobfuscated = new HashMap<>();
|
|
private Random rnd = new Random();
|
|
private final int DEFAULT_FOO_SIZE = 10;
|
|
public static final String validFirstCharacters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ_";
|
|
public static final String validNextCharacters = validFirstCharacters + "0123456789";
|
|
public static final String fooCharacters = "bcdfghjklmnpqrstvwz";
|
|
public static final String fooJoinCharacters = "aeiouy";
|
|
private List<MyEntry<DirectValueActionItem, ConstantPool>> allVariableNames = new ArrayList<>();
|
|
private HashSet<String> allVariableNamesStr = new HashSet<>();
|
|
private List<GraphSourceItem> allFunctions = new ArrayList<>();
|
|
private HashMap<DirectValueActionItem, ConstantPool> allStrings = new HashMap<>();
|
|
private HashMap<DirectValueActionItem, String> usageTypes = new HashMap<>();
|
|
|
|
private String fooString(String orig, boolean firstUppercase, int rndSize) {
|
|
boolean exists;
|
|
String ret;
|
|
loopfoo:
|
|
do {
|
|
exists = false;
|
|
int len = 3 + rnd.nextInt(rndSize - 3);
|
|
ret = "";
|
|
for (int i = 0; i < len; i++) {
|
|
String c = "";
|
|
if ((i % 2) == 0) {
|
|
c = "" + fooCharacters.charAt(rnd.nextInt(fooCharacters.length()));
|
|
} else {
|
|
c = "" + fooJoinCharacters.charAt(rnd.nextInt(fooJoinCharacters.length()));
|
|
}
|
|
if (i == 0 && firstUppercase) {
|
|
c = c.toUpperCase(Locale.ENGLISH);
|
|
}
|
|
ret += c;
|
|
}
|
|
if (allVariableNamesStr.contains(ret)) {
|
|
exists = true;
|
|
rndSize = rndSize + 1;
|
|
continue loopfoo;
|
|
}
|
|
if (isReserved(ret)) {
|
|
exists = true;
|
|
rndSize = rndSize + 1;
|
|
continue;
|
|
}
|
|
if (deobfuscated.containsValue(ret)) {
|
|
exists = true;
|
|
rndSize = rndSize + 1;
|
|
continue;
|
|
}
|
|
} while (exists);
|
|
return ret;
|
|
}
|
|
|
|
public String deobfuscateName(String s, boolean firstUppercase, String usageType, RenameType renameType, Map<String, String> selected) {
|
|
boolean isValid = true;
|
|
if (usageType == null) {
|
|
usageType = "name";
|
|
}
|
|
|
|
if (selected != null) {
|
|
if (selected.containsKey(s)) {
|
|
return selected.get(s);
|
|
}
|
|
}
|
|
|
|
if (isReserved(s)) {
|
|
isValid = false;
|
|
}
|
|
|
|
if (isValid) {
|
|
for (int i = 0; i < s.length(); i++) {
|
|
if (s.charAt(i) > 127) {
|
|
isValid = false;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (isValid) {
|
|
Pattern pat = Pattern.compile("^[" + Pattern.quote(validFirstCharacters) + "]" + "[" + Pattern.quote(validFirstCharacters + validNextCharacters) + "]*$");
|
|
if (!pat.matcher(s).matches()) {
|
|
isValid = false;
|
|
}
|
|
}
|
|
if (!isValid) {
|
|
if (deobfuscated.containsKey(s)) {
|
|
return deobfuscated.get(s);
|
|
} else {
|
|
Integer cnt = typeCounts.get(usageType);
|
|
if (cnt == null) {
|
|
cnt = 0;
|
|
}
|
|
|
|
String ret = null;
|
|
if (renameType == RenameType.TYPENUMBER) {
|
|
|
|
boolean found;
|
|
do {
|
|
found = false;
|
|
cnt++;
|
|
ret = usageType + "_" + cnt;
|
|
found = allVariableNamesStr.contains(ret);
|
|
} while (found);
|
|
typeCounts.put(usageType, cnt);
|
|
} else if (renameType == RenameType.RANDOMWORD) {
|
|
ret = fooString(s, firstUppercase, DEFAULT_FOO_SIZE);
|
|
}
|
|
deobfuscated.put(s, ret);
|
|
return ret;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
private static void getVariables(ConstantPool constantPool, List<Object> localData, Stack<GraphTargetItem> stack, List<GraphTargetItem> output, ActionGraphSource code, int ip, List<MyEntry<DirectValueActionItem, ConstantPool>> variables, List<GraphSourceItem> functions, HashMap<DirectValueActionItem, ConstantPool> strings, List<Integer> visited, HashMap<DirectValueActionItem, String> usageTypes, String path) {
|
|
boolean debugMode = false;
|
|
while ((ip > -1) && ip < code.size()) {
|
|
if (visited.contains(ip)) {
|
|
break;
|
|
}
|
|
GraphSourceItem ins = code.get(ip);
|
|
|
|
if (debugMode) {
|
|
System.err.println("Visit " + ip + ": ofs" + Helper.formatAddress(((Action) ins).getAddress()) + ":" + ((Action) ins).getASMSource(new ArrayList<GraphSourceItem>(), new ArrayList<Long>(), new ArrayList<String>(), code.version, false) + " stack:" + Helper.stackToString(stack, Helper.toList(new ConstantPool())));
|
|
}
|
|
if (ins.isExit()) {
|
|
break;
|
|
}
|
|
if (ins.isIgnored()) {
|
|
ip++;
|
|
continue;
|
|
}
|
|
|
|
String usageType = "name";
|
|
GraphTargetItem name = null;
|
|
if ((ins instanceof ActionGetVariable)
|
|
|| (ins instanceof ActionGetMember)
|
|
|| (ins instanceof ActionDefineLocal2)
|
|
|| (ins instanceof ActionNewMethod)
|
|
|| (ins instanceof ActionNewObject)
|
|
|| (ins instanceof ActionCallMethod)
|
|
|| (ins instanceof ActionCallFunction)) {
|
|
name = stack.peek();
|
|
}
|
|
|
|
if ((ins instanceof ActionGetVariable) || (ins instanceof ActionDefineLocal2)) {
|
|
usageType = "variable";
|
|
}
|
|
if (ins instanceof ActionGetMember) {
|
|
usageType = "member";
|
|
}
|
|
if ((ins instanceof ActionNewMethod) || (ins instanceof ActionNewObject)) {
|
|
usageType = "class";
|
|
}
|
|
if (ins instanceof ActionCallMethod) {
|
|
usageType = "function"; //can there be method?
|
|
}
|
|
if (ins instanceof ActionCallFunction) {
|
|
usageType = "function";
|
|
}
|
|
|
|
if ((ins instanceof ActionDefineFunction) || (ins instanceof ActionDefineFunction2)) {
|
|
functions.add(ins);
|
|
}
|
|
|
|
if (ins instanceof GraphSourceItemContainer) {
|
|
GraphSourceItemContainer cnt = (GraphSourceItemContainer) ins;
|
|
List<Long> cntSizes = cnt.getContainerSizes();
|
|
long addr = code.pos2adr(ip + 1);
|
|
ip = code.adr2pos(addr);
|
|
String cntName = cnt.getName();
|
|
for (Long size : cntSizes) {
|
|
if (size == 0) {
|
|
continue;
|
|
}
|
|
ip = code.adr2pos(addr);
|
|
addr += size;
|
|
int nextip = code.adr2pos(addr);
|
|
getVariables(variables, functions, strings, usageTypes, new ActionGraphSource(code.getActions().subList(ip, nextip), code.version, new HashMap<Integer, String>(), new HashMap<String, GraphTargetItem>(), new HashMap<String, GraphTargetItem>()), 0, path + (cntName == null ? "" : "/" + cntName));
|
|
ip = nextip;
|
|
}
|
|
List<List<GraphTargetItem>> r = new ArrayList<>();
|
|
r.add(new ArrayList<GraphTargetItem>());
|
|
r.add(new ArrayList<GraphTargetItem>());
|
|
r.add(new ArrayList<GraphTargetItem>());
|
|
((GraphSourceItemContainer) ins).translateContainer(r, stack, output, new HashMap<Integer, String>(), new HashMap<String, GraphTargetItem>(), new HashMap<String, GraphTargetItem>());
|
|
//ip++;
|
|
continue;
|
|
}
|
|
|
|
if ((ins instanceof ActionSetVariable) || (ins instanceof ActionSetMember) || (ins instanceof ActionDefineLocal)) {
|
|
name = stack.get(stack.size() - 2);
|
|
}
|
|
|
|
if ((ins instanceof ActionSetVariable) || (ins instanceof ActionDefineLocal)) {
|
|
usageType = "variable";
|
|
}
|
|
|
|
if (ins instanceof ActionSetMember) {
|
|
usageType = "member";
|
|
}
|
|
|
|
if (name instanceof DirectValueActionItem) {
|
|
variables.add(new MyEntry<>((DirectValueActionItem) name, constantPool));
|
|
usageTypes.put((DirectValueActionItem) name, usageType);
|
|
}
|
|
|
|
//for..in return
|
|
if (((ins instanceof ActionEquals) || (ins instanceof ActionEquals2)) && (stack.size() == 1) && (stack.peek() instanceof DirectValueActionItem)) {
|
|
stack.push(new DirectValueActionItem(null, 0, new Null(), new ArrayList<String>()));
|
|
}
|
|
|
|
if (ins instanceof ActionConstantPool) {
|
|
constantPool = new ConstantPool(((ActionConstantPool) ins).constantPool);
|
|
}
|
|
int staticOperation = Graph.SOP_USE_STATIC; //(Boolean) Configuration.getConfig("autoDeobfuscate", true) ? Graph.SOP_SKIP_STATIC : Graph.SOP_USE_STATIC;
|
|
|
|
try {
|
|
ins.translate(localData, stack, output, staticOperation, path);
|
|
} catch (Exception ex) {
|
|
Logger.getLogger(SWF.class.getName()).log(Level.SEVERE, "Error during getting variables", ex);
|
|
}
|
|
if (ins.isExit()) {
|
|
break;
|
|
}
|
|
|
|
if (ins instanceof ActionPush) {
|
|
if (!stack.isEmpty()) {
|
|
GraphTargetItem top = stack.peek();
|
|
if (top instanceof DirectValueActionItem) {
|
|
DirectValueActionItem dvt = (DirectValueActionItem) top;
|
|
if ((dvt.value instanceof String) || (dvt.value instanceof ConstantIndex)) {
|
|
if (constantPool == null) {
|
|
constantPool = new ConstantPool(dvt.constants);
|
|
}
|
|
strings.put(dvt, constantPool);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (ins.isBranch() || ins.isJump()) {
|
|
if (ins instanceof ActionIf) {
|
|
stack.pop();
|
|
}
|
|
visited.add(ip);
|
|
List<Integer> branches = ins.getBranches(code);
|
|
for (int b : branches) {
|
|
@SuppressWarnings("unchecked")
|
|
Stack<GraphTargetItem> brStack = (Stack<GraphTargetItem>) stack.clone();
|
|
if (b >= 0) {
|
|
getVariables(constantPool, localData, brStack, output, code, b, variables, functions, strings, visited, usageTypes, path);
|
|
} else {
|
|
if (debugMode) {
|
|
System.out.println("Negative branch:" + b);
|
|
}
|
|
}
|
|
}
|
|
// }
|
|
break;
|
|
}
|
|
ip++;
|
|
};
|
|
}
|
|
|
|
private static void getVariables(List<MyEntry<DirectValueActionItem, ConstantPool>> variables, List<GraphSourceItem> functions, HashMap<DirectValueActionItem, ConstantPool> strings, HashMap<DirectValueActionItem, String> usageType, ActionGraphSource code, int addr, String path) {
|
|
List<Object> localData = Helper.toList(new HashMap<Integer, String>(), new HashMap<String, GraphTargetItem>(), new HashMap<String, GraphTargetItem>());
|
|
try {
|
|
getVariables(null, localData, new Stack<GraphTargetItem>(), new ArrayList<GraphTargetItem>(), code, code.adr2pos(addr), variables, functions, strings, new ArrayList<Integer>(), usageType, path);
|
|
} catch (Exception ex) {
|
|
Logger.getLogger(SWF.class.getName()).log(Level.SEVERE, "Getting variables error", ex);
|
|
}
|
|
}
|
|
|
|
private List<MyEntry<DirectValueActionItem, ConstantPool>> getVariables(List<MyEntry<DirectValueActionItem, ConstantPool>> variables, List<GraphSourceItem> functions, HashMap<DirectValueActionItem, ConstantPool> strings, HashMap<DirectValueActionItem, String> usageType, ASMSource src, String path) {
|
|
List<MyEntry<DirectValueActionItem, ConstantPool>> ret = new ArrayList<>();
|
|
List<Action> actions = src.getActions(version);
|
|
actionsMap.put(src, actions);
|
|
getVariables(variables, functions, strings, usageType, new ActionGraphSource(actions, version, new HashMap<Integer, String>(), new HashMap<String, GraphTargetItem>(), new HashMap<String, GraphTargetItem>()), 0, path);
|
|
return ret;
|
|
}
|
|
private HashMap<ASMSource, List<Action>> actionsMap = new HashMap<>();
|
|
|
|
private void getVariables(List<Object> objs, String path) {
|
|
List<String> processed = new ArrayList<>();
|
|
for (Object o : objs) {
|
|
if (o instanceof ASMSource) {
|
|
String infPath = path + "/" + o.toString();
|
|
int pos = 1;
|
|
String infPath2 = infPath;
|
|
while (processed.contains(infPath2)) {
|
|
pos++;
|
|
infPath2 = infPath + "[" + pos + "]";
|
|
}
|
|
processed.add(infPath2);
|
|
informListeners("getVariables", infPath2);
|
|
getVariables(allVariableNames, allFunctions, allStrings, usageTypes, (ASMSource) o, path);
|
|
}
|
|
if (o instanceof Container) {
|
|
getVariables(((Container) o).getSubItems(), path + "/" + o.toString());
|
|
}
|
|
}
|
|
}
|
|
|
|
public void deobfuscateInstanceNames(RenameType renameType, List<Tag> tags, Map<String, String> selected) {
|
|
for (Tag t : tags) {
|
|
if (t instanceof DefineSpriteTag) {
|
|
deobfuscateInstanceNames(renameType, ((DefineSpriteTag) t).subTags, selected);
|
|
}
|
|
if (t instanceof PlaceObjectTypeTag) {
|
|
PlaceObjectTypeTag po = (PlaceObjectTypeTag) t;
|
|
String name = po.getInstanceName();
|
|
if (name != null) {
|
|
String changedName = deobfuscateName(name, false, "instance", renameType, selected);
|
|
if (changedName != null) {
|
|
po.setInstanceName(changedName);
|
|
}
|
|
}
|
|
String className = po.getClassName();
|
|
if (className != null) {
|
|
String changedClassName = deobfuscateNameWithPackage(className, renameType, selected);
|
|
if (changedClassName != null) {
|
|
po.setClassName(changedClassName);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public String deobfuscatePackage(String pkg, RenameType renameType, Map<String, String> selected) {
|
|
if (deobfuscated.containsKey(pkg)) {
|
|
return deobfuscated.get(pkg);
|
|
}
|
|
String parts[] = null;
|
|
if (pkg.contains(".")) {
|
|
parts = pkg.split("\\.");
|
|
} else {
|
|
parts = new String[]{pkg};
|
|
}
|
|
String ret = "";
|
|
boolean isChanged = false;
|
|
for (int p = 0; p < parts.length; p++) {
|
|
if (p > 0) {
|
|
ret += ".";
|
|
}
|
|
String partChanged = deobfuscateName(parts[p], false, "package", renameType, selected);
|
|
if (partChanged != null) {
|
|
ret += partChanged;
|
|
isChanged = true;
|
|
} else {
|
|
ret += parts[p];
|
|
}
|
|
}
|
|
if (isChanged) {
|
|
deobfuscated.put(pkg, ret);
|
|
return ret;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public String deobfuscateNameWithPackage(String n, RenameType renameType, Map<String, String> selected) {
|
|
String pkg = null;
|
|
String name = "";
|
|
if (n.contains(".")) {
|
|
pkg = n.substring(0, n.lastIndexOf("."));
|
|
name = n.substring(n.lastIndexOf(".") + 1);
|
|
} else {
|
|
name = n;
|
|
}
|
|
boolean changed = false;
|
|
if ((pkg != null) && (!pkg.equals(""))) {
|
|
String changedPkg = deobfuscatePackage(pkg, renameType, selected);
|
|
if (changedPkg != null) {
|
|
changed = true;
|
|
pkg = changedPkg;
|
|
}
|
|
}
|
|
String changedName = deobfuscateName(name, true, "class", renameType, selected);
|
|
if (changedName != null) {
|
|
changed = true;
|
|
name = changedName;
|
|
}
|
|
if (changed) {
|
|
String newClassName = "";
|
|
if (pkg == null) {
|
|
newClassName = name;
|
|
} else {
|
|
newClassName = pkg + "." + name;
|
|
}
|
|
return newClassName;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public int deobfuscateAS3Identifiers(RenameType renameType) {
|
|
for (Tag tag : tags) {
|
|
if (tag instanceof ABCContainerTag) {
|
|
((ABCContainerTag) tag).getABC().deobfuscateIdentifiers(deobfuscated, renameType, true);
|
|
}
|
|
}
|
|
for (Tag tag : tags) {
|
|
if (tag instanceof ABCContainerTag) {
|
|
((ABCContainerTag) tag).getABC().deobfuscateIdentifiers(deobfuscated, renameType, false);
|
|
}
|
|
}
|
|
for (Tag tag : tags) {
|
|
if (tag instanceof SymbolClassTag) {
|
|
SymbolClassTag sc = (SymbolClassTag) tag;
|
|
for (int i = 0; i < sc.classNames.length; i++) {
|
|
String newname = deobfuscateNameWithPackage(sc.classNames[i], renameType, deobfuscated);
|
|
if (newname != null) {
|
|
sc.classNames[i] = newname;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
deobfuscateInstanceNames(renameType, tags, new HashMap<String, String>());
|
|
return deobfuscated.size();
|
|
}
|
|
HashMap<String, Integer> typeCounts = new HashMap<>();
|
|
|
|
public int deobfuscateIdentifiers(RenameType renameType) {
|
|
findFileAttributes();
|
|
if (fileAttributes == null) {
|
|
int cnt = 0;
|
|
cnt += deobfuscateAS2Identifiers(renameType);
|
|
cnt += deobfuscateAS3Identifiers(renameType);
|
|
return cnt;
|
|
} else {
|
|
if (fileAttributes.actionScript3) {
|
|
return deobfuscateAS3Identifiers(renameType);
|
|
} else {
|
|
return deobfuscateAS2Identifiers(renameType);
|
|
}
|
|
}
|
|
}
|
|
|
|
public void renameAS2Identifier(String identifier, String newname) {
|
|
Map<String, String> selected = new HashMap<>();
|
|
selected.put(identifier, newname);
|
|
renameAS2Identifiers(null, selected);
|
|
}
|
|
|
|
public int deobfuscateAS2Identifiers(RenameType renameType) {
|
|
return renameAS2Identifiers(renameType, null);
|
|
}
|
|
|
|
private int renameAS2Identifiers(RenameType renameType, Map<String, String> selected) {
|
|
actionsMap = new HashMap<>();
|
|
allFunctions = new ArrayList<>();
|
|
allVariableNames = new ArrayList<>();
|
|
allStrings = new HashMap<>();
|
|
|
|
List<Object> objs = new ArrayList<>();
|
|
int ret = 0;
|
|
objs.addAll(tags);
|
|
getVariables(objs, "");
|
|
informListeners("rename", "");
|
|
int fc = 0;
|
|
for (MyEntry<DirectValueActionItem, ConstantPool> it : allVariableNames) {
|
|
String name = it.key.toStringNoH(it.value);
|
|
allVariableNamesStr.add(name);
|
|
}
|
|
|
|
informListeners("rename", "classes");
|
|
int classCount = 0;
|
|
for (Tag t : tags) {
|
|
if (t instanceof DoInitActionTag) {
|
|
classCount++;
|
|
}
|
|
}
|
|
int cnt = 0;
|
|
for (Tag t : tags) {
|
|
if (t instanceof DoInitActionTag) {
|
|
cnt++;
|
|
informListeners("rename", "class " + cnt + "/" + classCount);
|
|
DoInitActionTag dia = (DoInitActionTag) t;
|
|
String exportName = dia.getExportName();
|
|
final String pkgPrefix = "__Packages.";
|
|
String classNameParts[] = null;
|
|
if ((exportName != null) && exportName.startsWith(pkgPrefix)) {
|
|
String className = exportName.substring(pkgPrefix.length());
|
|
if (className.contains(".")) {
|
|
classNameParts = className.split("\\.");
|
|
} else {
|
|
classNameParts = new String[]{className};
|
|
}
|
|
}
|
|
int staticOperation = Graph.SOP_USE_STATIC; //(Boolean) Configuration.getConfig("autoDeobfuscate", true) ? Graph.SOP_SKIP_STATIC : Graph.SOP_USE_STATIC;
|
|
List<GraphTargetItem> dec = Action.actionsToTree(dia.getActions(version), version, staticOperation, ""/*FIXME*/);
|
|
GraphTargetItem name = null;
|
|
for (GraphTargetItem it : dec) {
|
|
if (it instanceof ClassActionItem) {
|
|
ClassActionItem cti = (ClassActionItem) it;
|
|
List<GraphTargetItem> methods = new ArrayList<>();
|
|
methods.addAll(cti.functions);
|
|
methods.addAll(cti.staticFunctions);
|
|
|
|
for (GraphTargetItem gti : methods) {
|
|
if (gti instanceof FunctionActionItem) {
|
|
FunctionActionItem fun = (FunctionActionItem) gti;
|
|
if (fun.calculatedFunctionName instanceof DirectValueActionItem) {
|
|
DirectValueActionItem dvf = (DirectValueActionItem) fun.calculatedFunctionName;
|
|
String fname = dvf.toStringNoH(null);
|
|
String changed = deobfuscateName(fname, false, "method", renameType, selected);
|
|
if (changed != null) {
|
|
deobfuscated.put(fname, changed);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
List<GraphTargetItem> vars = new ArrayList<>();
|
|
for (MyEntry<GraphTargetItem, GraphTargetItem> item : cti.vars) {
|
|
vars.add(item.key);
|
|
}
|
|
for (MyEntry<GraphTargetItem, GraphTargetItem> item : cti.staticVars) {
|
|
vars.add(item.key);
|
|
}
|
|
for (GraphTargetItem gti : vars) {
|
|
if (gti instanceof DirectValueActionItem) {
|
|
DirectValueActionItem dvf = (DirectValueActionItem) gti;
|
|
String vname = dvf.toStringNoH(null);
|
|
String changed = deobfuscateName(vname, false, "attribute", renameType, selected);
|
|
if (changed != null) {
|
|
deobfuscated.put(vname, changed);
|
|
}
|
|
}
|
|
}
|
|
|
|
name = cti.className;
|
|
break;
|
|
}
|
|
if (it instanceof InterfaceActionItem) {
|
|
InterfaceActionItem ift = (InterfaceActionItem) it;
|
|
name = ift.name;
|
|
}
|
|
}
|
|
|
|
|
|
if (name != null) {
|
|
int pos = 0;
|
|
while (name instanceof GetMemberActionItem) {
|
|
GetMemberActionItem mem = (GetMemberActionItem) name;
|
|
GraphTargetItem memberName = mem.memberName;
|
|
if (memberName instanceof DirectValueActionItem) {
|
|
DirectValueActionItem dvt = (DirectValueActionItem) memberName;
|
|
String nameStr = dvt.toStringNoH(null);
|
|
if (classNameParts != null) {
|
|
if (classNameParts.length - 1 - pos < 0) {
|
|
break;
|
|
}
|
|
}
|
|
String changedNameStr = nameStr;
|
|
if (classNameParts != null) {
|
|
changedNameStr = classNameParts[classNameParts.length - 1 - pos];
|
|
}
|
|
String changedNameStr2 = deobfuscateName(changedNameStr, pos == 0, pos == 0 ? "class" : "package", renameType, selected);
|
|
if (changedNameStr2 != null) {
|
|
changedNameStr = changedNameStr2;
|
|
}
|
|
ret++;
|
|
deobfuscated.put(nameStr, changedNameStr);
|
|
pos++;
|
|
}
|
|
name = mem.object;
|
|
}
|
|
if (name instanceof GetVariableActionItem) {
|
|
GetVariableActionItem var = (GetVariableActionItem) name;
|
|
if (var.name instanceof DirectValueActionItem) {
|
|
DirectValueActionItem dvt = (DirectValueActionItem) var.name;
|
|
String nameStr = dvt.toStringNoH(null);
|
|
if (classNameParts != null) {
|
|
if (classNameParts.length - 1 - pos < 0) {
|
|
break;
|
|
}
|
|
}
|
|
String changedNameStr = nameStr;
|
|
if (classNameParts != null) {
|
|
changedNameStr = classNameParts[classNameParts.length - 1 - pos];
|
|
}
|
|
String changedNameStr2 = deobfuscateName(changedNameStr, pos == 0, pos == 0 ? "class" : "package", renameType, selected);
|
|
if (changedNameStr2 != null) {
|
|
changedNameStr = changedNameStr2;
|
|
}
|
|
ret++;
|
|
deobfuscated.put(nameStr, changedNameStr);
|
|
pos++;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
for (GraphSourceItem fun : allFunctions) {
|
|
fc++;
|
|
informListeners("rename", "function " + fc + "/" + allFunctions.size());
|
|
if (fun instanceof ActionDefineFunction) {
|
|
ActionDefineFunction f = (ActionDefineFunction) fun;
|
|
if (f.functionName.equals("")) { //anonymous function, leave as is
|
|
continue;
|
|
}
|
|
String changed = deobfuscateName(f.functionName, false, "function", renameType, selected);
|
|
if (changed != null) {
|
|
f.replacedFunctionName = changed;
|
|
ret++;
|
|
}
|
|
}
|
|
if (fun instanceof ActionDefineFunction2) {
|
|
ActionDefineFunction2 f = (ActionDefineFunction2) fun;
|
|
if (f.functionName.equals("")) { //anonymous function, leave as is
|
|
continue;
|
|
}
|
|
String changed = deobfuscateName(f.functionName, false, "function", renameType, selected);
|
|
if (changed != null) {
|
|
f.replacedFunctionName = changed;
|
|
ret++;
|
|
}
|
|
}
|
|
}
|
|
|
|
HashSet<String> stringsNoVarH = new HashSet<>();
|
|
List<DirectValueActionItem> allVariableNamesDv = new ArrayList<>();
|
|
for (MyEntry<DirectValueActionItem, ConstantPool> it : allVariableNames) {
|
|
allVariableNamesDv.add(it.key);
|
|
}
|
|
for (DirectValueActionItem ti : allStrings.keySet()) {
|
|
if (!allVariableNamesDv.contains(ti)) {
|
|
stringsNoVarH.add(System.identityHashCode(allStrings.get(ti)) + "_" + ti.toStringNoH(allStrings.get(ti)));
|
|
}
|
|
}
|
|
|
|
int vc = 0;
|
|
for (MyEntry<DirectValueActionItem, ConstantPool> it : allVariableNames) {
|
|
vc++;
|
|
String name = it.key.toStringNoH(it.value);
|
|
String changed = deobfuscateName(name, false, usageTypes.get(it.key), renameType, selected);
|
|
if (changed != null) {
|
|
boolean addNew = false;
|
|
String h = System.identityHashCode(it.key) + "_" + name;
|
|
if (stringsNoVarH.contains(h)) {
|
|
addNew = true;
|
|
}
|
|
ActionPush pu = (ActionPush) it.key.src;
|
|
if (pu.replacement == null) {
|
|
pu.replacement = new ArrayList<>();
|
|
pu.replacement.addAll(pu.values);
|
|
}
|
|
if (pu.replacement.get(it.key.pos) instanceof ConstantIndex) {
|
|
ConstantIndex ci = (ConstantIndex) pu.replacement.get(it.key.pos);
|
|
ConstantPool pool = it.value;
|
|
if (pool == null) {
|
|
continue;
|
|
}
|
|
if (pool.constants == null) {
|
|
continue;
|
|
}
|
|
if (addNew) {
|
|
pool.constants.add(changed);
|
|
ci.index = pool.constants.size() - 1;
|
|
} else {
|
|
pool.constants.set(ci.index, changed);
|
|
}
|
|
} else {
|
|
pu.replacement.set(it.key.pos, changed);
|
|
}
|
|
ret++;
|
|
}
|
|
}
|
|
for (ASMSource src : actionsMap.keySet()) {
|
|
actionsMap.put(src, Action.removeNops(0, actionsMap.get(src), version, 0, ""/*FIXME path*/));
|
|
src.setActions(actionsMap.get(src), version);
|
|
}
|
|
deobfuscateInstanceNames(renameType, tags, selected);
|
|
return ret;
|
|
}
|
|
|
|
public void exportFla(AbortRetryIgnoreHandler handler, String outfile, String swfName, String generator, String generatorVerName, String generatorVersion, boolean parallel) throws IOException {
|
|
XFLConverter.convertSWF(handler, this, swfName, outfile, true, generator, generatorVerName, generatorVersion, parallel);
|
|
}
|
|
|
|
public void exportXfl(AbortRetryIgnoreHandler handler, String outfile, String swfName, String generator, String generatorVerName, String generatorVersion, boolean parallel) throws IOException {
|
|
XFLConverter.convertSWF(handler, this, swfName, outfile, false, generator, generatorVerName, generatorVersion, parallel);
|
|
}
|
|
|
|
public static float twipToPixel(int twip) {
|
|
return ((float) twip) / 20.0f;
|
|
}
|
|
|
|
private static class CachedImage implements Serializable {
|
|
|
|
private int data[];
|
|
private int width;
|
|
private int height;
|
|
private int type;
|
|
public static final long serialVersionUID = 1L;
|
|
|
|
public CachedImage(BufferedImage img) {
|
|
width = img.getWidth();
|
|
height = img.getHeight();
|
|
type = img.getType();
|
|
data = Filtering.getRGB(img, 0, 0, width, height);
|
|
}
|
|
|
|
public BufferedImage getImage() {
|
|
BufferedImage ret = new BufferedImage(width, height, type);
|
|
Filtering.setRGB(ret, 0, 0, width, height, data);
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
public static AffineTransform matrixToTransform(MATRIX mat) {
|
|
return new AffineTransform(mat.getScaleXFloat(), mat.getRotateSkew0Float(),
|
|
mat.getRotateSkew1Float(), mat.getScaleYFloat(),
|
|
mat.translateX, mat.translateY);
|
|
}
|
|
private static Cache cache = Cache.getInstance(false);
|
|
|
|
public void clearImageCache() {
|
|
cache.clear();
|
|
SHAPERECORD.clearShapeCache();
|
|
}
|
|
|
|
public static RECT fixRect(RECT rect) {
|
|
RECT ret = new RECT();
|
|
ret.Xmin = rect.Xmin;
|
|
ret.Xmax = rect.Xmax;
|
|
ret.Ymin = rect.Ymin;
|
|
ret.Ymax = rect.Ymax;
|
|
|
|
|
|
|
|
if (ret.Xmax <= 0) {
|
|
ret.Xmax = ret.getWidth();
|
|
ret.Xmin = 0;
|
|
}
|
|
if (ret.Ymax <= 0) {
|
|
ret.Ymax = ret.getHeight();
|
|
ret.Ymin = 0;
|
|
}
|
|
if (ret.Xmin < 0) {
|
|
ret.Xmax += (-ret.Xmin);
|
|
ret.Xmin = 0;
|
|
}
|
|
if (ret.Ymin < 0) {
|
|
ret.Ymax += (-ret.Ymin);
|
|
ret.Ymin = 0;
|
|
}
|
|
|
|
if (ret.getWidth() < 1 || ret.getHeight() < 1) {
|
|
ret.Xmin = 0;
|
|
ret.Ymin = 0;
|
|
ret.Xmax = 20;
|
|
ret.Ymax = 20;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public static BufferedImage frameToImage(int containerId, int maxDepth, HashMap<Integer, Layer> layers, Color backgroundColor, HashMap<Integer, CharacterTag> characters, int frame, List<Tag> allTags, List<Tag> controlTags, RECT displayRect, Stack<Integer> visited) {
|
|
int fixX = 0;
|
|
int fixY = 0;
|
|
fixX = -displayRect.Xmin / 20;
|
|
fixY = -displayRect.Ymin / 20;
|
|
displayRect = fixRect(displayRect);
|
|
|
|
String key = "frame_" + frame + "_" + containerId;
|
|
if (cache.contains(key)) {
|
|
return ((CachedImage) cache.get(key)).getImage();
|
|
}
|
|
float unzoom = 20;
|
|
BufferedImage ret = new BufferedImage((int) (displayRect.Xmax / unzoom), (int) (displayRect.Ymax / unzoom), BufferedImage.TYPE_INT_ARGB);
|
|
|
|
Graphics2D g = (Graphics2D) ret.getGraphics();
|
|
g.setPaint(backgroundColor);
|
|
g.fill(new Rectangle(ret.getWidth(), ret.getHeight()));
|
|
g.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR);
|
|
g.setRenderingHint(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_QUALITY);
|
|
g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
|
|
|
|
for (int i = 1; i <= maxDepth; i++) {
|
|
if (!layers.containsKey(i)) {
|
|
continue;
|
|
}
|
|
Layer layer = layers.get(i);
|
|
if (!characters.containsKey(layer.characterId)) {
|
|
continue;
|
|
}
|
|
if (!layer.visible) {
|
|
continue;
|
|
}
|
|
CharacterTag character = characters.get(layer.characterId);
|
|
MATRIX mat = new MATRIX(layer.matrix);
|
|
mat.translateX /= 20;
|
|
mat.translateY /= 20;
|
|
mat.translateX += fixX;
|
|
mat.translateY += fixY;
|
|
|
|
AffineTransform trans = matrixToTransform(mat);
|
|
|
|
//trans.translate(transX, transY);
|
|
g.setTransform(trans);
|
|
if (character instanceof DrawableTag) {
|
|
DrawableTag drawable = (DrawableTag) character;
|
|
BufferedImage img = drawable.toImage(layer.ratio < 0 ? 0 : layer.ratio/*layer.duration*/, allTags, displayRect, characters, visited);
|
|
if (layer.filters != null) {
|
|
for (FILTER filter : layer.filters) {
|
|
img = filter.apply(img);
|
|
}
|
|
}
|
|
if (layer.colorTransForm != null) {
|
|
img = layer.colorTransForm.apply(img);
|
|
}
|
|
|
|
if (layer.colorTransFormAlpha != null) {
|
|
img = layer.colorTransFormAlpha.apply(img);
|
|
}
|
|
Point imgPos = drawable.getImagePos(layer.ratio < 0 ? 0 : layer.ratio, characters, visited);
|
|
switch (layer.blendMode) {
|
|
case 0:
|
|
case 1:
|
|
g.setComposite(AlphaComposite.SrcOver);
|
|
break;
|
|
case 2: //TODO:Layer
|
|
g.setComposite(AlphaComposite.SrcOver);
|
|
break;
|
|
case 3:
|
|
g.setComposite(BlendComposite.Multiply);
|
|
break;
|
|
case 4:
|
|
g.setComposite(BlendComposite.Screen);
|
|
break;
|
|
case 5:
|
|
g.setComposite(BlendComposite.Lighten);
|
|
break;
|
|
case 6:
|
|
g.setComposite(BlendComposite.Darken);
|
|
break;
|
|
case 7:
|
|
g.setComposite(BlendComposite.Difference);
|
|
break;
|
|
case 8:
|
|
g.setComposite(BlendComposite.Add);
|
|
break;
|
|
case 9:
|
|
g.setComposite(BlendComposite.Subtract);
|
|
break;
|
|
case 10:
|
|
g.setComposite(BlendComposite.Invert);
|
|
break;
|
|
case 11:
|
|
g.setComposite(BlendComposite.Alpha);
|
|
break;
|
|
case 12:
|
|
g.setComposite(BlendComposite.Erase);
|
|
break;
|
|
case 13:
|
|
g.setComposite(BlendComposite.Overlay);
|
|
break;
|
|
case 14:
|
|
g.setComposite(BlendComposite.HardLight);
|
|
break;
|
|
default: //Not implemented
|
|
g.setComposite(AlphaComposite.SrcOver);
|
|
break;
|
|
}
|
|
|
|
g.drawImage(img, imgPos.x, imgPos.y, null);
|
|
} else if (character instanceof BoundedTag) {
|
|
BoundedTag b = (BoundedTag) character;
|
|
g.setPaint(new Color(255, 255, 255, 128));
|
|
g.setComposite(BlendComposite.Invert);
|
|
RECT r = b.getRect(characters, visited);
|
|
g.drawString(character.toString(), (r.Xmin) / 20 + 3, (r.Ymin) / 20 + 15);
|
|
g.draw(new Rectangle(r.Xmin / 20, r.Ymin / 20, r.getWidth() / 20, r.getHeight() / 20));
|
|
g.drawLine(r.Xmin / 20, r.Ymin / 20, r.Xmax / 20, r.Ymax / 20);
|
|
g.drawLine(r.Xmax / 20, r.Ymin / 20, r.Xmin / 20, r.Ymax / 20);
|
|
g.setComposite(AlphaComposite.Dst);
|
|
}
|
|
}
|
|
g.setTransform(AffineTransform.getScaleInstance(1, 1));
|
|
/*g.setPaint(Color.yellow);
|
|
g.draw(new Rectangle(ret.getWidth()-1,ret.getHeight()-1));*/
|
|
cache.put(key, new CachedImage(ret));
|
|
|
|
/*try {
|
|
ImageIO.write(ret, "png", new File("tst_id_" + containerId + "_time_" + System.currentTimeMillis() + ".png"));
|
|
} catch (IOException ex) {
|
|
Logger.getLogger(SWF.class.getName()).log(Level.SEVERE, null, ex);
|
|
}*/
|
|
return ret;
|
|
}
|
|
|
|
public static BufferedImage frameToImage(int containerId, int frame, List<Tag> allTags, List<Tag> controlTags, RECT displayRect, int totalFrameCount, Stack<Integer> visited) {
|
|
List<BufferedImage> ret = new ArrayList<>();
|
|
framesToImage(containerId, ret, frame, frame, allTags, controlTags, displayRect, totalFrameCount, visited);
|
|
if (ret.isEmpty()) {
|
|
return new BufferedImage(1, 1, BufferedImage.TYPE_INT_ARGB);
|
|
}
|
|
return ret.get(0);
|
|
}
|
|
|
|
public static void framesToImage(int containerId, List<BufferedImage> ret, int startFrame, int stopFrame, List<Tag> allTags, List<Tag> controlTags, RECT displayRect, int totalFrameCount, Stack<Integer> visited) {
|
|
for (int i = startFrame; i <= stopFrame; i++) {
|
|
String key = "frame_" + i + "_" + containerId;
|
|
if (cache.contains(key)) {
|
|
CachedImage g = (CachedImage) cache.get(key);
|
|
if (g == null) {
|
|
break;
|
|
}
|
|
ret.add(g.getImage());
|
|
startFrame++;
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
if (startFrame > stopFrame) {
|
|
return;
|
|
}
|
|
if (totalFrameCount == 0) {
|
|
return;
|
|
}
|
|
|
|
while (startFrame >= totalFrameCount) {
|
|
startFrame -= totalFrameCount;
|
|
}
|
|
|
|
while (stopFrame >= totalFrameCount) {
|
|
stopFrame -= totalFrameCount;
|
|
}
|
|
|
|
|
|
HashMap<Integer, CharacterTag> characters = new HashMap<>();
|
|
for (Tag t : allTags) {
|
|
if (t instanceof CharacterTag) {
|
|
CharacterTag ch = (CharacterTag) t;
|
|
characters.put(ch.getCharacterId(), ch);
|
|
}
|
|
}
|
|
|
|
HashMap<Integer, Layer> layers = new HashMap<>();
|
|
|
|
int maxDepth = 0;
|
|
int f = 0;
|
|
Color backgroundColor = new Color(0, 0, 0, 0);
|
|
for (Tag t : controlTags) {
|
|
if (t instanceof SetBackgroundColorTag) {
|
|
SetBackgroundColorTag c = (SetBackgroundColorTag) t;
|
|
backgroundColor = new Color(c.backgroundColor.red, c.backgroundColor.green, c.backgroundColor.blue);
|
|
}
|
|
|
|
if (t instanceof PlaceObjectTypeTag) {
|
|
PlaceObjectTypeTag po = (PlaceObjectTypeTag) t;
|
|
int depth = po.getDepth();
|
|
if (depth > maxDepth) {
|
|
maxDepth = depth;
|
|
}
|
|
|
|
if (!layers.containsKey(depth)) {
|
|
layers.put(depth, new Layer());
|
|
}
|
|
Layer layer = layers.get(depth);
|
|
int characterId = po.getCharacterId();
|
|
if (characterId != -1) {
|
|
layer.characterId = characterId;
|
|
}
|
|
layer.visible = po.isVisible();
|
|
if (po.flagMove()) {
|
|
MATRIX matrix2 = po.getMatrix();
|
|
if (matrix2 != null) {
|
|
layer.matrix = matrix2;
|
|
}
|
|
String instanceName = po.getInstanceName();
|
|
if (instanceName != null) {
|
|
layer.instanceName = instanceName;
|
|
}
|
|
CXFORM colorTransForm = po.getColorTransform();
|
|
if (colorTransForm != null) {
|
|
layer.colorTransForm = colorTransForm;
|
|
}
|
|
CXFORMWITHALPHA colorTransFormAlpha = po.getColorTransformWithAlpha();
|
|
if (colorTransFormAlpha != null) {
|
|
layer.colorTransFormAlpha = colorTransFormAlpha;
|
|
}
|
|
if (po.cacheAsBitmap()) {
|
|
layer.cacheAsBitmap = true;
|
|
}
|
|
int blendMode = po.getBlendMode();
|
|
if (blendMode != 0) {
|
|
layer.blendMode = blendMode;
|
|
}
|
|
List<FILTER> filters = po.getFilters();
|
|
if (filters != null) {
|
|
layer.filters = filters;
|
|
}
|
|
int ratio = po.getRatio();
|
|
if (ratio != -1) {
|
|
layer.ratio = ratio;
|
|
}
|
|
} else {
|
|
layer.matrix = po.getMatrix();
|
|
layer.instanceName = po.getInstanceName();
|
|
layer.colorTransForm = po.getColorTransform();
|
|
layer.colorTransFormAlpha = po.getColorTransformWithAlpha();
|
|
layer.cacheAsBitmap = po.cacheAsBitmap();
|
|
layer.blendMode = po.getBlendMode();
|
|
layer.filters = po.getFilters();
|
|
layer.ratio = po.getRatio();
|
|
}
|
|
}
|
|
|
|
if (t instanceof RemoveTag) {
|
|
RemoveTag rt = (RemoveTag) t;
|
|
layers.remove(rt.getDepth());
|
|
|
|
}
|
|
for (Layer l : layers.values()) {
|
|
l.duration++;
|
|
}
|
|
if (t instanceof ShowFrameTag) {
|
|
if (f > stopFrame) {
|
|
break;
|
|
}
|
|
if ((f >= startFrame) && (f <= stopFrame)) {
|
|
ret.add(frameToImage(containerId, maxDepth, layers, backgroundColor, characters, f, allTags, controlTags, displayRect, visited));
|
|
}
|
|
f++;
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
public void removeTagFromTimeline(Tag toRemove, List<Tag> timeline) {
|
|
int characterId = 0;
|
|
if (toRemove instanceof CharacterTag) {
|
|
characterId = ((CharacterTag) toRemove).getCharacterId();
|
|
}
|
|
Map<Integer, Integer> stage = new HashMap<>();
|
|
|
|
Set<Integer> dependingChars = new HashSet<>();
|
|
if (characterId != 0) {
|
|
dependingChars.add(characterId);
|
|
for (int i = 0; i < timeline.size(); i++) {
|
|
Tag t = timeline.get(i);
|
|
if (t instanceof CharacterIdTag) {
|
|
CharacterIdTag c = (CharacterIdTag) t;
|
|
Set<Integer> needed = t.getNeededCharacters();
|
|
if (needed.contains(characterId)) {
|
|
dependingChars.add(c.getCharacterId());
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
for (int i = 0; i < timeline.size(); i++) {
|
|
Tag t = timeline.get(i);
|
|
if (t instanceof RemoveTag) {
|
|
RemoveTag rt = (RemoveTag) t;
|
|
int currentCharId = stage.get(rt.getDepth());
|
|
stage.remove(rt.getDepth());
|
|
if (dependingChars.contains(currentCharId)) {
|
|
timeline.remove(i);
|
|
i--;
|
|
continue;
|
|
}
|
|
}
|
|
if (t instanceof PlaceObjectTypeTag) {
|
|
PlaceObjectTypeTag po = (PlaceObjectTypeTag) t;
|
|
int placeCharId = po.getCharacterId();
|
|
int placeDepth = po.getDepth();
|
|
if (placeCharId != 0) {
|
|
stage.put(placeDepth, placeCharId);
|
|
}
|
|
int currentCharId = stage.get(placeDepth);
|
|
if (dependingChars.contains(currentCharId)) {
|
|
timeline.remove(i);
|
|
i--;
|
|
continue;
|
|
}
|
|
}
|
|
if (t instanceof CharacterIdTag) {
|
|
CharacterIdTag c = (CharacterIdTag) t;
|
|
if (dependingChars.contains(c.getCharacterId())) {
|
|
timeline.remove(i);
|
|
i--;
|
|
continue;
|
|
}
|
|
}
|
|
Set<Integer> needed = t.getNeededCharacters();
|
|
for (int dep : dependingChars) {
|
|
if (needed.contains(dep)) {
|
|
timeline.remove(i);
|
|
i--;
|
|
continue;
|
|
}
|
|
}
|
|
if (t == toRemove) {
|
|
timeline.remove(i);
|
|
i--;
|
|
continue;
|
|
}
|
|
if (t instanceof DefineSpriteTag) {
|
|
DefineSpriteTag spr = (DefineSpriteTag) t;
|
|
removeTagFromTimeline(toRemove, spr.subTags);
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
public void removeTag(Tag t) {
|
|
removeTagFromTimeline(t, tags);
|
|
}
|
|
}
|