com.heliosdecompiler.helios.transformers.disassemblers.KrakatauDisassembler.java Source code

Introduction

Here is the source code for com.heliosdecompiler.helios.transformers.disassemblers.KrakatauDisassembler.java

Source

/*
 * Copyright 2016 Sam Sun <me@samczsun.com>
 *
 *    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 com.heliosdecompiler.helios.transformers.disassemblers;

import com.heliosdecompiler.helios.transformers.TransformerSettings;
import com.heliosdecompiler.helios.Constants;
import com.heliosdecompiler.helios.Helios;
import com.heliosdecompiler.helios.utils.Utils;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.objectweb.asm.tree.ClassNode;

import java.io.File;
import java.nio.file.Files;
import java.util.Enumeration;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;

public class KrakatauDisassembler extends Disassembler {
    public KrakatauDisassembler() {
        super("krakatau", "Krakatau", Settings.class);
    }

    @Override
    public boolean isApplicable(String className) {
        return className.endsWith(".class");
    }

    public boolean disassembleClassNode(ClassNode cn, byte[] b, StringBuilder output) {
        if (Helios.ensurePython2Set()) {
            File inputJar = null;
            File outputZip = null;
            String processLog = null;

            try {
                inputJar = Files.createTempFile("kdisin", ".jar").toFile();
                outputZip = Files.createTempFile("kdisout", ".zip").toFile();

                Map<String, byte[]> data = Helios.getAllLoadedData();
                data.put(cn.name + ".class", b);

                Utils.saveClasses(inputJar, data);

                Process process = Helios.launchProcess(new ProcessBuilder(
                        com.heliosdecompiler.helios.Settings.PYTHON2_LOCATION.get().asString(), "-O",
                        "disassemble.py", "-path", inputJar.getAbsolutePath(), "-out", outputZip.getAbsolutePath(),
                        cn.name + ".class", Settings.ROUNDTRIP.isEnabled() ? "-roundtrip" : "")
                                .directory(Constants.KRAKATAU_DIR));

                processLog = Utils.readProcess(process);

                ZipFile zipFile = new ZipFile(outputZip);
                Enumeration<? extends ZipEntry> entries = zipFile.entries();
                byte[] disassembled = null;
                while (entries.hasMoreElements()) {
                    ZipEntry next = entries.nextElement();
                    if (next.getName().equals(cn.name + ".j")) {
                        disassembled = IOUtils.toByteArray(zipFile.getInputStream(next));
                    }
                }
                zipFile.close();
                output.append(new String(disassembled, "UTF-8"));
                return true;
            } catch (Exception e) {
                output.append(parseException(e)).append(processLog);
                return false;
            } finally {
                FileUtils.deleteQuietly(inputJar);
                FileUtils.deleteQuietly(outputZip);
            }
        } else {
            output.append("You need to set the location of Python 2.x");
        }
        return false;
    }

    public enum Settings implements TransformerSettings.Setting {
        ROUNDTRIP("roundtrip", "Disassemble for roundtrip assembly");

        private final String name;
        private final String param;
        private boolean on;

        Settings(String param, String name) {
            this(param, name, false);
        }

        Settings(String param, String name, boolean on) {
            this.name = name;
            this.param = param;
            this.on = on;
        }

        public String getParam() {
            return param;
        }

        public String getText() {
            return name;
        }

        public boolean isEnabled() {
            return on;
        }

        public void setEnabled(boolean enabled) {
            this.on = enabled;
        }
    }
}