baksmali/src/main/java/org/jf/baksmali/baksmali.java - platform/external/smali - Git at Google

 /*
  * [The "BSD licence"]
  * Copyright (c) 2010 Ben Gruver (JesusFreke)
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
  * 3. The name of the author may not be used to endorse or promote products
  *    derived from this software without specific prior written permission.
  *
  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */

 package org.jf.baksmali;

 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.Iterables;
 import com.google.common.collect.Lists;
 import com.google.common.collect.Ordering;
 import org.jf.baksmali.Adaptors.ClassDefinition;
 import org.jf.dexlib2.analysis.ClassPath;
 import org.jf.dexlib2.analysis.CustomInlineMethodResolver;
 import org.jf.dexlib2.iface.ClassDef;
 import org.jf.dexlib2.iface.DexFile;
 import org.jf.dexlib2.util.SyntheticAccessorResolver;
 import org.jf.util.ClassFileNameHandler;
 import org.jf.util.IndentingWriter;
 import org.xml.sax.Attributes;
 import org.xml.sax.SAXException;
 import org.xml.sax.helpers.DefaultHandler;

 import javax.xml.parsers.ParserConfigurationException;
 import javax.xml.parsers.SAXParser;
 import javax.xml.parsers.SAXParserFactory;
 import java.io.*;
 import java.util.List;
 import java.util.Map.Entry;
 import java.util.concurrent.*;

 public class baksmali {

     public static boolean disassembleDexFile(DexFile dexFile, final baksmaliOptions options) {
         if (options.registerInfo != 0 || options.deodex || options.normalizeVirtualMethods) {
             try {
                 Iterable<String> extraClassPathEntries;
                 if (options.extraClassPathEntries != null) {
                     extraClassPathEntries = options.extraClassPathEntries;
                 } else {
                     extraClassPathEntries = ImmutableList.of();
                 }

                 options.classPath = ClassPath.fromClassPath(options.bootClassPathDirs,
                         Iterables.concat(options.bootClassPathEntries, extraClassPathEntries), dexFile,
                         options.apiLevel, options.checkPackagePrivateAccess, options.experimental);

                 if (options.customInlineDefinitions != null) {
                     options.inlineResolver = new CustomInlineMethodResolver(options.classPath,
                             options.customInlineDefinitions);
                 }
             } catch (Exception ex) {
                 System.err.println("\n\nError occurred while loading boot class path files. Aborting.");
                 ex.printStackTrace(System.err);
                 return false;
             }
         }

         if (options.resourceIdFileEntries != null) {
             class PublicHandler extends DefaultHandler {
                 String prefix = null;
                 public PublicHandler(String prefix) {
                     super();
                     this.prefix = prefix;
                 }

                 public void startElement(String uri, String localName,
                         String qName, Attributes attr) throws SAXException {
                     if (qName.equals("public")) {
                         String type = attr.getValue("type");
                         String name = attr.getValue("name").replace('.', '_');
                         Integer public_key = Integer.decode(attr.getValue("id"));
                         String public_val = new StringBuffer()
                             .append(prefix)
                             .append(".")
                             .append(type)
                             .append(".")
                             .append(name)
                             .toString();
                         options.resourceIds.put(public_key, public_val);
                     }
                 }
             };

             for (Entry<String,String> entry: options.resourceIdFileEntries.entrySet()) {
                 try {
                     SAXParser saxp = SAXParserFactory.newInstance().newSAXParser();
                     String prefix = entry.getValue();
                     saxp.parse(entry.getKey(), new PublicHandler(prefix));
                 } catch (ParserConfigurationException e) {
                     continue;
                 } catch (SAXException e) {
                     continue;
                 } catch (IOException e) {
                     continue;
                 }
             }
         }

         File outputDirectoryFile = new File(options.outputDirectory);
         if (!outputDirectoryFile.exists()) {
             if (!outputDirectoryFile.mkdirs()) {
                 System.err.println("Can't create the output directory " + options.outputDirectory);
                 return false;
             }
         }

         //sort the classes, so that if we're on a case-insensitive file system and need to handle classes with file
         //name collisions, then we'll use the same name for each class, if the dex file goes through multiple
         //baksmali/smali cycles for some reason. If a class with a colliding name is added or removed, the filenames
         //may still change of course
         List<? extends ClassDef> classDefs = Ordering.natural().sortedCopy(dexFile.getClasses());

         if (!options.noAccessorComments) {
             options.syntheticAccessorResolver = new SyntheticAccessorResolver(dexFile.getOpcodes(), classDefs);
         }

         final ClassFileNameHandler fileNameHandler = new ClassFileNameHandler(outputDirectoryFile, ".smali");

         ExecutorService executor = Executors.newFixedThreadPool(options.jobs);
         List<Future<Boolean>> tasks = Lists.newArrayList();

         for (final ClassDef classDef: classDefs) {
             tasks.add(executor.submit(new Callable<Boolean>() {
                 @Override public Boolean call() throws Exception {
                     return disassembleClass(classDef, fileNameHandler, options);
                 }
             }));
         }

         boolean errorOccurred = false;
         try {
             for (Future<Boolean> task: tasks) {
                 while(true) {
                     try {
                         if (!task.get()) {
                             errorOccurred = true;
                         }
                     } catch (InterruptedException ex) {
                         continue;
                     } catch (ExecutionException ex) {
                         throw new RuntimeException(ex);
                     }
                     break;
                 }
             }
         } finally {
             executor.shutdown();
         }
         return !errorOccurred;
     }

     private static boolean disassembleClass(ClassDef classDef, ClassFileNameHandler fileNameHandler,
                                             baksmaliOptions options) {
         /**
          * The path for the disassembly file is based on the package name
          * The class descriptor will look something like:
          * Ljava/lang/Object;
          * Where the there is leading 'L' and a trailing ';', and the parts of the
          * package name are separated by '/'
          */
         String classDescriptor = classDef.getType();

         //validate that the descriptor is formatted like we expect
         if (classDescriptor.charAt(0) != 'L' ||
                 classDescriptor.charAt(classDescriptor.length()-1) != ';') {
             System.err.println("Unrecognized class descriptor - " + classDescriptor + " - skipping class");
             return false;
         }

         File smaliFile = fileNameHandler.getUniqueFilenameForClass(classDescriptor);

         //create and initialize the top level string template
         ClassDefinition classDefinition = new ClassDefinition(options, classDef);

         //write the disassembly
         Writer writer = null;
         try
         {
             File smaliParent = smaliFile.getParentFile();
             if (!smaliParent.exists()) {
                 if (!smaliParent.mkdirs()) {
                     // check again, it's likely it was created in a different thread
                     if (!smaliParent.exists()) {
                         System.err.println("Unable to create directory " + smaliParent.toString() + " - skipping class");
                         return false;
                     }
                 }
             }

             if (!smaliFile.exists()){
                 if (!smaliFile.createNewFile()) {
                     System.err.println("Unable to create file " + smaliFile.toString() + " - skipping class");
                     return false;
                 }
             }

             BufferedWriter bufWriter = new BufferedWriter(new OutputStreamWriter(
                     new FileOutputStream(smaliFile), "UTF8"));

             writer = new IndentingWriter(bufWriter);
             classDefinition.writeTo((IndentingWriter)writer);
         } catch (Exception ex) {
             System.err.println("\n\nError occurred while disassembling class " + classDescriptor.replace('/', '.') + " - skipping class");
             ex.printStackTrace();
             // noinspection ResultOfMethodCallIgnored
             smaliFile.delete();
             return false;
         }
         finally
         {
             if (writer != null) {
                 try {
                     writer.close();
                 } catch (Throwable ex) {
                     System.err.println("\n\nError occurred while closing file " + smaliFile.toString());
                     ex.printStackTrace();
                 }
             }
         }
         return true;
     }
 }
	/*
	* [The "BSD licence"]
	* Copyright (c) 2010 Ben Gruver (JesusFreke)
	* All rights reserved.
	*
	* Redistribution and use in source and binary forms, with or without
	* modification, are permitted provided that the following conditions
	* are met:
	* 1. Redistributions of source code must retain the above copyright
	* notice, this list of conditions and the following disclaimer.
	* 2. Redistributions in binary form must reproduce the above copyright
	* notice, this list of conditions and the following disclaimer in the
	* documentation and/or other materials provided with the distribution.
	* 3. The name of the author may not be used to endorse or promote products
	* derived from this software without specific prior written permission.
	*
	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
	* INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
	*/

	package org.jf.baksmali;

	import com.google.common.collect.ImmutableList;
	import com.google.common.collect.Iterables;
	import com.google.common.collect.Lists;
	import com.google.common.collect.Ordering;
	import org.jf.baksmali.Adaptors.ClassDefinition;
	import org.jf.dexlib2.analysis.ClassPath;
	import org.jf.dexlib2.analysis.CustomInlineMethodResolver;
	import org.jf.dexlib2.iface.ClassDef;
	import org.jf.dexlib2.iface.DexFile;
	import org.jf.dexlib2.util.SyntheticAccessorResolver;
	import org.jf.util.ClassFileNameHandler;
	import org.jf.util.IndentingWriter;
	import org.xml.sax.Attributes;
	import org.xml.sax.SAXException;
	import org.xml.sax.helpers.DefaultHandler;

	import javax.xml.parsers.ParserConfigurationException;
	import javax.xml.parsers.SAXParser;
	import javax.xml.parsers.SAXParserFactory;
	import java.io.*;
	import java.util.List;
	import java.util.Map.Entry;
	import java.util.concurrent.*;

	public class baksmali {

	public static boolean disassembleDexFile(DexFile dexFile, final baksmaliOptions options) {
	if (options.registerInfo != 0 \|\| options.deodex \|\| options.normalizeVirtualMethods) {
	try {
	Iterable<String> extraClassPathEntries;
	if (options.extraClassPathEntries != null) {
	extraClassPathEntries = options.extraClassPathEntries;
	} else {
	extraClassPathEntries = ImmutableList.of();
	}

	options.classPath = ClassPath.fromClassPath(options.bootClassPathDirs,
	Iterables.concat(options.bootClassPathEntries, extraClassPathEntries), dexFile,
	options.apiLevel, options.checkPackagePrivateAccess, options.experimental);

	if (options.customInlineDefinitions != null) {
	options.inlineResolver = new CustomInlineMethodResolver(options.classPath,
	options.customInlineDefinitions);
	}
	} catch (Exception ex) {
	System.err.println("\n\nError occurred while loading boot class path files. Aborting.");
	ex.printStackTrace(System.err);
	return false;
	}
	}

	if (options.resourceIdFileEntries != null) {
	class PublicHandler extends DefaultHandler {
	String prefix = null;
	public PublicHandler(String prefix) {
	super();
	this.prefix = prefix;
	}

	public void startElement(String uri, String localName,
	String qName, Attributes attr) throws SAXException {
	if (qName.equals("public")) {
	String type = attr.getValue("type");
	String name = attr.getValue("name").replace('.', '_');
	Integer public_key = Integer.decode(attr.getValue("id"));
	String public_val = new StringBuffer()
	.append(prefix)
	.append(".")
	.append(type)
	.append(".")
	.append(name)
	.toString();
	options.resourceIds.put(public_key, public_val);
	}
	}
	};

	for (Entry<String,String> entry: options.resourceIdFileEntries.entrySet()) {
	try {
	SAXParser saxp = SAXParserFactory.newInstance().newSAXParser();
	String prefix = entry.getValue();
	saxp.parse(entry.getKey(), new PublicHandler(prefix));
	} catch (ParserConfigurationException e) {
	continue;
	} catch (SAXException e) {
	continue;
	} catch (IOException e) {
	continue;
	}
	}
	}

	File outputDirectoryFile = new File(options.outputDirectory);
	if (!outputDirectoryFile.exists()) {
	if (!outputDirectoryFile.mkdirs()) {
	System.err.println("Can't create the output directory " + options.outputDirectory);
	return false;
	}
	}

	//sort the classes, so that if we're on a case-insensitive file system and need to handle classes with file
	//name collisions, then we'll use the same name for each class, if the dex file goes through multiple
	//baksmali/smali cycles for some reason. If a class with a colliding name is added or removed, the filenames
	//may still change of course
	List<? extends ClassDef> classDefs = Ordering.natural().sortedCopy(dexFile.getClasses());

	if (!options.noAccessorComments) {
	options.syntheticAccessorResolver = new SyntheticAccessorResolver(dexFile.getOpcodes(), classDefs);
	}

	final ClassFileNameHandler fileNameHandler = new ClassFileNameHandler(outputDirectoryFile, ".smali");

	ExecutorService executor = Executors.newFixedThreadPool(options.jobs);
	List<Future<Boolean>> tasks = Lists.newArrayList();

	for (final ClassDef classDef: classDefs) {
	tasks.add(executor.submit(new Callable<Boolean>() {
	@Override public Boolean call() throws Exception {
	return disassembleClass(classDef, fileNameHandler, options);
	}
	}));
	}

	boolean errorOccurred = false;
	try {
	for (Future<Boolean> task: tasks) {
	while(true) {
	try {
	if (!task.get()) {
	errorOccurred = true;
	}
	} catch (InterruptedException ex) {
	continue;
	} catch (ExecutionException ex) {
	throw new RuntimeException(ex);
	}
	break;
	}
	}
	} finally {
	executor.shutdown();
	}
	return !errorOccurred;
	}

	private static boolean disassembleClass(ClassDef classDef, ClassFileNameHandler fileNameHandler,
	baksmaliOptions options) {
	/**
	* The path for the disassembly file is based on the package name
	* The class descriptor will look something like:
	* Ljava/lang/Object;
	* Where the there is leading 'L' and a trailing ';', and the parts of the
	* package name are separated by '/'
	*/
	String classDescriptor = classDef.getType();

	//validate that the descriptor is formatted like we expect
	if (classDescriptor.charAt(0) != 'L' \|\|
	classDescriptor.charAt(classDescriptor.length()-1) != ';') {
	System.err.println("Unrecognized class descriptor - " + classDescriptor + " - skipping class");
	return false;
	}

	File smaliFile = fileNameHandler.getUniqueFilenameForClass(classDescriptor);

	//create and initialize the top level string template
	ClassDefinition classDefinition = new ClassDefinition(options, classDef);

	//write the disassembly
	Writer writer = null;
	try
	{
	File smaliParent = smaliFile.getParentFile();
	if (!smaliParent.exists()) {
	if (!smaliParent.mkdirs()) {
	// check again, it's likely it was created in a different thread
	if (!smaliParent.exists()) {
	System.err.println("Unable to create directory " + smaliParent.toString() + " - skipping class");
	return false;
	}
	}
	}

	if (!smaliFile.exists()){
	if (!smaliFile.createNewFile()) {
	System.err.println("Unable to create file " + smaliFile.toString() + " - skipping class");
	return false;
	}
	}

	BufferedWriter bufWriter = new BufferedWriter(new OutputStreamWriter(
	new FileOutputStream(smaliFile), "UTF8"));

	writer = new IndentingWriter(bufWriter);
	classDefinition.writeTo((IndentingWriter)writer);
	} catch (Exception ex) {
	System.err.println("\n\nError occurred while disassembling class " + classDescriptor.replace('/', '.') + " - skipping class");
	ex.printStackTrace();
	// noinspection ResultOfMethodCallIgnored
	smaliFile.delete();
	return false;
	}
	finally
	{
	if (writer != null) {
	try {
	writer.close();
	} catch (Throwable ex) {
	System.err.println("\n\nError occurred while closing file " + smaliFile.toString());
	ex.printStackTrace();
	}
	}
	}
	return true;
	}
	}