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.
167 lines
5.4 KiB
167 lines
5.4 KiB
/*
|
|
* Copyright 2000-2016 JetBrains s.r.o.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
package org.jetbrains.java.decompiler.struct;
|
|
|
|
import org.jetbrains.java.decompiler.main.DecompilerContext;
|
|
import org.jetbrains.java.decompiler.main.extern.IFernflowerPreferences;
|
|
import org.jetbrains.java.decompiler.main.extern.IResultSaver;
|
|
import org.jetbrains.java.decompiler.struct.lazy.LazyLoader;
|
|
import org.jetbrains.java.decompiler.struct.lazy.LazyLoader.Link;
|
|
import org.jetbrains.java.decompiler.util.DataInputFullStream;
|
|
|
|
import java.io.IOException;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.jar.JarFile;
|
|
import java.util.jar.Manifest;
|
|
|
|
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 final int type;
|
|
private final boolean own;
|
|
|
|
private final String archivePath; // relative path to jar/zip
|
|
private final String filename; // folder: relative path, archive: file name
|
|
private final IResultSaver resultSaver;
|
|
private final IDecompiledData decompiledData;
|
|
|
|
private final List<String> classEntries = new ArrayList<String>(); // class file or jar/zip entry
|
|
private final List<String> dirEntries = new ArrayList<String>();
|
|
private final List<String[]> otherEntries = new ArrayList<String[]>();
|
|
|
|
private List<StructClass> classes = new ArrayList<StructClass>();
|
|
private Manifest manifest;
|
|
|
|
public ContextUnit(int type, String archivePath, String filename, boolean own, IResultSaver resultSaver, IDecompiledData decompiledData) {
|
|
this.type = type;
|
|
this.own = own;
|
|
this.archivePath = archivePath;
|
|
this.filename = filename;
|
|
this.resultSaver = resultSaver;
|
|
this.decompiledData = decompiledData;
|
|
}
|
|
|
|
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;
|
|
try (DataInputFullStream in = loader.getClassStream(oldName)) {
|
|
newCl = new StructClass(in, 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
|
|
resultSaver.saveFolder(filename);
|
|
|
|
// non-class files
|
|
for (String[] pair : otherEntries) {
|
|
resultSaver.copyFile(pair[0], filename, pair[1]);
|
|
}
|
|
|
|
// classes
|
|
for (int i = 0; i < classes.size(); i++) {
|
|
StructClass cl = classes.get(i);
|
|
String entryName = decompiledData.getClassEntryName(cl, classEntries.get(i));
|
|
if (entryName != null) {
|
|
String content = decompiledData.getClassContent(cl);
|
|
if (content != null) {
|
|
int[] mapping = null;
|
|
if (DecompilerContext.getOption(IFernflowerPreferences.BYTECODE_SOURCE_MAPPING)) {
|
|
mapping = DecompilerContext.getBytecodeSourceMapper().getOriginalLinesMapping();
|
|
}
|
|
resultSaver.saveClassFile(filename, cl.qualifiedName, entryName, content, mapping);
|
|
}
|
|
}
|
|
}
|
|
|
|
break;
|
|
|
|
case TYPE_JAR:
|
|
case TYPE_ZIP:
|
|
// create archive file
|
|
resultSaver.saveFolder(archivePath);
|
|
resultSaver.createArchive(archivePath, filename, manifest);
|
|
|
|
// directory entries
|
|
for (String dirEntry : dirEntries) {
|
|
resultSaver.saveDirEntry(archivePath, filename, dirEntry);
|
|
}
|
|
|
|
// non-class entries
|
|
for (String[] pair : otherEntries) {
|
|
if (type != TYPE_JAR || !JarFile.MANIFEST_NAME.equalsIgnoreCase(pair[1])) {
|
|
resultSaver.copyEntry(pair[0], archivePath, filename, pair[1]);
|
|
}
|
|
}
|
|
|
|
// classes
|
|
for (int i = 0; i < classes.size(); i++) {
|
|
StructClass cl = classes.get(i);
|
|
String entryName = decompiledData.getClassEntryName(cl, classEntries.get(i));
|
|
if (entryName != null) {
|
|
String content = decompiledData.getClassContent(cl);
|
|
resultSaver.saveClassEntry(archivePath, filename, cl.qualifiedName, entryName, content);
|
|
}
|
|
}
|
|
|
|
resultSaver.closeArchive(archivePath, filename);
|
|
}
|
|
}
|
|
|
|
public void setManifest(Manifest manifest) {
|
|
this.manifest = manifest;
|
|
}
|
|
|
|
public boolean isOwn() {
|
|
return own;
|
|
}
|
|
|
|
public List<StructClass> getClasses() {
|
|
return classes;
|
|
}
|
|
} |