/*
* Copyright (C) 2010-2016 JPEXS, All rights reserved.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 3.0 of the License, or (at your option) any later version.
*
* This library 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library.
*/
package com.jpexs.decompiler.flash;
import com.jpexs.helpers.Helper;
import com.jpexs.helpers.MemoryInputStream;
import com.jpexs.helpers.ReReadableInputStream;
import com.jpexs.helpers.streams.SeekableInputStream;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;
/**
*
* @author JPEXS
*/
public class ZippedSWFBundle implements SWFBundle {
protected Set<String> keySet = new HashSet<>();
protected FileInputStream fis;
protected ReReadableInputStream is;
protected File filename;
public ZippedSWFBundle(InputStream is) throws IOException {
this(is, null);
}
public ZippedSWFBundle(File filename) throws IOException {
this(null, filename);
}
protected ZippedSWFBundle(InputStream is, File filename) throws IOException {
initBundle(is, filename);
}
protected void initBundle(InputStream is, File filename) throws IOException {
if (filename != null) {
fis = new FileInputStream(filename);
is = fis;
}
this.filename = filename;
this.is = new ReReadableInputStream(is);
ZipInputStream zip = new ZipInputStream(this.is);
ZipEntry entry;
keySet.clear();
while ((entry = zip.getNextEntry()) != null) {
if (entry.getName().toLowerCase().endsWith(".swf")
|| entry.getName().toLowerCase().endsWith(".gfx")) {
keySet.add(entry.getName());
}
}
}
@Override
public int length() {
return keySet.size();
}
@Override
public Set<String> getKeys() {
return keySet;
}
@Override
public SeekableInputStream getSWF(String key) throws IOException {
if (!keySet.contains(key)) {
return null;
}
//if (!cachedSWFs.containsKey(key)) {
SeekableInputStream ret = null;
this.is.reset();
ZipInputStream zip = new ZipInputStream(this.is);
ZipEntry entry;
while ((entry = zip.getNextEntry()) != null) {
if (entry.getName().equals(key)) {
MemoryInputStream mis = new MemoryInputStream(Helper.readStream(zip));
ret = mis;
//cachedSWFs.put(key, mis);
break;
}
zip.closeEntry();
}
return ret;
//return cachedSWFs.get(key);
}
@Override
public Map<String, SeekableInputStream> getAll() throws IOException {
Map<String, SeekableInputStream> ret = new HashMap<>();
for (String key : getKeys()) { // cache everything first
ret.put(key, getSWF(key));
}
return ret;
}
@Override
public String getExtension() {
return "zip";
}
@Override
public boolean isReadOnly() {
return this.filename == null || !this.filename.canWrite();
}
@Override
public boolean putSWF(String key, InputStream swfIs) throws IOException {
if (this.isReadOnly()) {
return false;
}
if (key == null) {
return false;
}
if (!getKeys().contains(key)) { //replace only existing keys
return false;
}
//Write to temp file first
File tempFile = new File((filename.getAbsolutePath()) + ".tmp");
try {
ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(tempFile));
this.is.reset();
ZipInputStream zis = new ZipInputStream(this.is);
ZipEntry entryIn;
ZipEntry entryOut;
byte[] swfData = Helper.readStream(swfIs);
try {
while ((entryIn = zis.getNextEntry()) != null) {
InputStream src;
if (entryIn.getName().equals(key)) {
entryOut = new ZipEntry(key);
src = new ByteArrayInputStream(swfData);
} else {
src = zis;
entryOut = entryIn;
}
zos.putNextEntry(entryOut);
Helper.copyStream(src, zos, entryOut.getSize() == -1 ? Long.MAX_VALUE : entryOut.getSize());
zos.closeEntry();
zis.closeEntry();
}
} finally {
zis.close();
zos.close();
}
this.is.close();
this.fis.close();
} catch (IOException ex) {
tempFile.delete();
throw ex;
}
filename.delete();
tempFile.renameTo(filename);
initBundle(null, filename);
return true;
}
}