SmaliAssembler.java

package the.bytecode.club.bytecodeviewer.compilers;

import org.apache.commons.io.FileUtils;
import org.zeroturnaround.zip.ZipUtil;
import the.bytecode.club.bytecodeviewer.BytecodeViewer;
import the.bytecode.club.bytecodeviewer.Dex2Jar;
import the.bytecode.club.bytecodeviewer.Enjarify;
import the.bytecode.club.bytecodeviewer.MiscUtils;

import java.io.File;

/***************************************************************************
 * Bytecode Viewer (BCV) - Java & Android Reverse Engineering Suite        *
 * Copyright (C) 2014 Kalen 'Konloch' Kinloch - http://bytecodeviewer.com  *
 *                                                                         *
 * This program is free software: you can redistribute it and/or modify    *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation, either version 3 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program 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 General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>. *
 ***************************************************************************/

/**
 * Smali Assembler Wrapper for Java
 * 
 * @author Konloch
 *
 */

public class SmaliAssembler extends Compiler {

	@Override
	public byte[] compile(String contents, String name) {
		String fileStart = BytecodeViewer.tempDir.getAbsoluteFile() + BytecodeViewer.fs + "temp";
		int fileNumber = MiscUtils.getClassNumber(fileStart, ".dex");

		final File tempSmaliFolder = new File(fileStart + fileNumber + "-smalifolder"+BytecodeViewer.fs);
		tempSmaliFolder.mkdir();
		
		File tempSmali = new File(tempSmaliFolder.getAbsolutePath() +BytecodeViewer.fs + fileNumber + ".smali");
		File tempDex = new File(fileStart + fileNumber + ".dex");
		File tempJar = new File(fileStart + fileNumber + ".jar");
		File tempJarFolder = new File(fileStart + fileNumber + "-jar"+BytecodeViewer.fs);
		
		try {
			FileUtils.write(tempSmali, contents, "UTF-8", false);
		} catch (final Exception e) {
			new the.bytecode.club.bytecodeviewer.api.ExceptionUI(e);
		}
		
		try {
			org.jf.smali.main.main(new String[]{tempSmaliFolder.getAbsolutePath(), "-o", tempDex.getAbsolutePath()});
		} catch (Exception e) {
			new the.bytecode.club.bytecodeviewer.api.ExceptionUI(e);
		}
		

		if(BytecodeViewer.viewer.apkConversionGroup.isSelected(BytecodeViewer.viewer.apkConversionDex.getModel()))
			 Dex2Jar.dex2Jar(tempDex, tempJar);
		else if(BytecodeViewer.viewer.apkConversionGroup.isSelected(BytecodeViewer.viewer.apkConversionEnjarify.getModel()))
			 Enjarify.apk2Jar(tempDex, tempJar);
		
		try {
			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) {
			new the.bytecode.club.bytecodeviewer.api.ExceptionUI(e);
		}
		
		return null;
	}
	
}