You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
211 lines
5.8 KiB
211 lines
5.8 KiB
10 years ago
|
/*
|
||
|
* Fernflower - The Analytical Java Decompiler
|
||
|
* http://www.reversed-java.com
|
||
|
*
|
||
|
* (C) 2008 - 2010, Stiver
|
||
|
*
|
||
|
* This software is NEITHER public domain NOR free software
|
||
|
* as per GNU License. See license.txt for more details.
|
||
|
*
|
||
|
* This software is distributed WITHOUT ANY WARRANTY; without
|
||
|
* even the implied warranty of MERCHANTABILITY or FITNESS FOR
|
||
|
* A PARTICULAR PURPOSE.
|
||
|
*/
|
||
|
|
||
10 years ago
|
package org.jetbrains.java.decompiler.struct;
|
||
10 years ago
|
|
||
|
import java.io.IOException;
|
||
|
import java.util.ArrayList;
|
||
|
import java.util.List;
|
||
|
import java.util.jar.Manifest;
|
||
|
|
||
10 years ago
|
import org.jetbrains.java.decompiler.main.extern.IDecompilatSaver;
|
||
|
import org.jetbrains.java.decompiler.struct.lazy.LazyLoader;
|
||
|
import org.jetbrains.java.decompiler.struct.lazy.LazyLoader.Link;
|
||
10 years ago
|
|
||
|
public class ContextUnit {
|
||
|
|
||
|
public static final int TYPE_FOLDER = 0;
|
||
|
public static final int TYPE_JAR = 1;
|
||
|
public static final int TYPE_ZIP = 2;
|
||
|
|
||
|
private static final String MANIFEST_ENTRY = "META-INF/MANIFEST.MF";
|
||
|
|
||
|
// *****************************************************************************
|
||
|
// private fields
|
||
|
// *****************************************************************************
|
||
|
|
||
|
private int type;
|
||
|
|
||
|
// relative path to jar/zip
|
||
|
private String archivepath;
|
||
|
|
||
|
// folder: relative path
|
||
|
// archive: file name
|
||
|
private String filename;
|
||
|
|
||
|
private List<StructClass> classes = new ArrayList<StructClass>();
|
||
|
|
||
|
// class file or jar/zip entry. Should, but doesn't have to be the same as qualifiedName of the class
|
||
|
private List<String> classentries = new ArrayList<String>();
|
||
|
|
||
|
private List<String> direntries = new ArrayList<String>();
|
||
|
|
||
|
private List<String[]> otherentries = new ArrayList<String[]>();
|
||
|
|
||
|
private Manifest manifest;
|
||
|
|
||
|
private IDecompilatSaver decompilatSaver;
|
||
|
|
||
|
private IDecompiledData decompiledData;
|
||
|
|
||
|
private boolean own = true;
|
||
|
|
||
|
// *****************************************************************************
|
||
|
// constructors
|
||
|
// *****************************************************************************
|
||
|
|
||
|
public ContextUnit(int type, String archivepath, String filename, boolean own,
|
||
|
IDecompilatSaver decompilatSaver, IDecompiledData decompiledData) {
|
||
|
this.type = type;
|
||
|
this.own = own;
|
||
|
this.archivepath = archivepath;
|
||
|
this.filename = filename;
|
||
|
this.decompilatSaver = decompilatSaver;
|
||
|
this.decompiledData = decompiledData;
|
||
|
}
|
||
|
|
||
|
// *****************************************************************************
|
||
|
// public methods
|
||
|
// *****************************************************************************
|
||
|
|
||
|
public void addClass(StructClass cl, String entryname) {
|
||
|
classes.add(cl);
|
||
|
classentries.add(entryname);
|
||
|
}
|
||
|
|
||
|
public void addDirEntry(String entry) {
|
||
|
direntries.add(entry);
|
||
|
}
|
||
|
|
||
|
public void addOtherEntry(String fullpath, String entry) {
|
||
|
otherentries.add(new String[]{fullpath, entry});
|
||
|
}
|
||
|
|
||
|
public void reload(LazyLoader loader) throws IOException {
|
||
|
|
||
|
List<StructClass> lstClasses = new ArrayList<StructClass>();
|
||
|
for(StructClass cl : classes) {
|
||
|
String oldname = cl.qualifiedName;
|
||
|
StructClass newcl = new StructClass(loader.getClassStream(oldname), cl.isOwn(), loader);
|
||
|
|
||
|
lstClasses.add(newcl);
|
||
|
|
||
|
Link lnk = loader.getClassLink(oldname);
|
||
|
loader.removeClassLink(oldname);
|
||
|
loader.addClassLink(newcl.qualifiedName, lnk);
|
||
|
}
|
||
|
|
||
|
classes = lstClasses;
|
||
|
}
|
||
|
|
||
|
public void save() {
|
||
|
|
||
|
switch(type) {
|
||
|
case TYPE_FOLDER:
|
||
|
|
||
|
// create folder
|
||
|
decompilatSaver.saveFolder(filename);
|
||
|
|
||
|
// non-class files
|
||
|
for(String[] arr: otherentries) {
|
||
|
decompilatSaver.copyFile(arr[0], filename, arr[0]);
|
||
|
}
|
||
|
|
||
|
// classes
|
||
|
for(int i=0;i<classes.size();i++) {
|
||
|
|
||
|
StructClass cl = classes.get(i);
|
||
|
String entryname = classentries.get(i);
|
||
|
|
||
|
entryname = decompiledData.getClassEntryName(cl, entryname);
|
||
|
if(entryname != null) {
|
||
|
String content = decompiledData.getClassContent(cl);
|
||
|
if(content != null) {
|
||
|
decompilatSaver.saveClassFile(filename, cl.qualifiedName, entryname, content);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
break;
|
||
|
case TYPE_JAR:
|
||
|
case TYPE_ZIP:
|
||
|
|
||
|
// create archive file
|
||
|
decompilatSaver.saveFolder(archivepath);
|
||
|
decompilatSaver.createArchive(archivepath, filename, manifest);
|
||
|
|
||
|
// directory entries
|
||
|
for(String direntry: direntries) {
|
||
|
decompilatSaver.saveEntry(archivepath, filename, direntry, null);
|
||
|
}
|
||
|
|
||
|
// non-class entries
|
||
|
for(String[] arr: otherentries) {
|
||
|
// manifest was defined by constructor invocation
|
||
|
if(type != TYPE_JAR || !MANIFEST_ENTRY.equalsIgnoreCase(arr[1])) {
|
||
|
decompilatSaver.copyEntry(arr[0], archivepath, filename, arr[1]);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// classes
|
||
|
for(int i=0;i<classes.size();i++) {
|
||
|
|
||
|
StructClass cl = classes.get(i);
|
||
|
String entryname = classentries.get(i);
|
||
|
|
||
|
entryname = decompiledData.getClassEntryName(cl, entryname);
|
||
|
if(entryname != null) {
|
||
|
String content = decompiledData.getClassContent(cl);
|
||
|
decompilatSaver.saveClassEntry(archivepath, filename, cl.qualifiedName, entryname, content);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
decompilatSaver.closeArchive(archivepath, filename);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// *****************************************************************************
|
||
|
// private methods
|
||
|
// *****************************************************************************
|
||
|
|
||
|
// *****************************************************************************
|
||
|
// getter and setter methods
|
||
|
// *****************************************************************************
|
||
|
|
||
|
public void setManifest(Manifest manifest) {
|
||
|
this.manifest = manifest;
|
||
|
}
|
||
|
|
||
|
public boolean isOwn() {
|
||
|
return own;
|
||
|
}
|
||
|
|
||
|
public List<StructClass> getClasses() {
|
||
|
return classes;
|
||
|
}
|
||
|
|
||
|
public int getType() {
|
||
|
return type;
|
||
|
}
|
||
|
|
||
|
public void setDecompilatSaver(IDecompilatSaver decompilatSaver) {
|
||
|
this.decompilatSaver = decompilatSaver;
|
||
|
}
|
||
|
|
||
|
public void setDecompiledData(IDecompiledData decompiledData) {
|
||
|
this.decompiledData = decompiledData;
|
||
|
}
|
||
|
|
||
|
}
|