com.samczsun.helios.transformers.assemblers.SmaliAssembler.java Source code

Introduction

Here is the source code for com.samczsun.helios.transformers.assemblers.SmaliAssembler.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.samczsun.helios.transformers.assemblers;

import com.samczsun.helios.Settings;
import com.samczsun.helios.handler.ExceptionHandler;
import com.samczsun.helios.transformers.converters.Converter;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.zeroturnaround.zip.ZipUtil;

import java.io.File;
import java.nio.file.Files;

public class SmaliAssembler extends Assembler {

    SmaliAssembler() {
        super("smali-assembler", "Smali Assembler");
    }

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

    @Override
    public byte[] assemble(String name, String contents) {
        File tempDir = null;
        File tempSmaliFolder = null;
        File tempSmali = null;
        File tempDex = null;
        File tempJar = null;
        File tempJarFolder = null;
        try {
            tempDir = Files.createTempDirectory("smali").toFile();
            tempSmaliFolder = new File(tempDir, "smalifolder");
            tempSmaliFolder.mkdir();

            tempSmali = new File(tempDir, "temp.smali");
            tempDex = new File(tempDir, "temp.dex");
            tempJar = new File(tempDir, "temp.jar");
            tempJarFolder = new File(tempDir, "temp-jar");

            FileUtils.write(tempSmali, contents, "UTF-8", false);
            try {
                org.jf.smali.main
                        .main(new String[] { tempSmaliFolder.getAbsolutePath(), "-o", tempDex.getAbsolutePath() });
            } catch (Exception e) {
                ExceptionHandler.handle(e);
            }

            if (Settings.APK_CONVERSION.get().asString().equals(Converter.ENJARIFY.getId())) {
                Converter.ENJARIFY.convert(tempDex, tempJar);
            } else if (Settings.APK_CONVERSION.get().asString().equals(Converter.DEX2JAR.getId())) {
                Converter.DEX2JAR.convert(tempDex, tempJar);
            }
            ZipUtil.unpack(tempJar, tempJarFolder);

            File outputClass = null;
            boolean found = false;
            File current = tempJarFolder;
            try {
                while (!found) {
                    File f = current.listFiles()[0];
                    if (f.isDirectory())
                        current = f;
                    else {
                        outputClass = f;
                        found = true;
                    }

                }

                return org.apache.commons.io.FileUtils.readFileToByteArray(outputClass);
            } catch (java.lang.NullPointerException e) {

            }
        } catch (Exception e) {
            ExceptionHandler.handle(e);
        } finally {
            FileUtils.deleteQuietly(tempDir);
            FileUtils.deleteQuietly(tempSmaliFolder);
            FileUtils.deleteQuietly(tempSmali);
            FileUtils.deleteQuietly(tempDex);
            FileUtils.deleteQuietly(tempJar);
            FileUtils.deleteQuietly(tempJarFolder);
        }
        return null;
    }
}