dump.java revision fbea4e751fa6f1748ded4379a4b64601cb53ba7b 
1/*
2 * [The "BSD licence"]
3 * Copyright (c) 2010 Ben Gruver (JesusFreke)
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 *    notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 *    notice, this list of conditions and the following disclaimer in the
13 *    documentation and/or other materials provided with the distribution.
14 * 3. The name of the author may not be used to endorse or promote products
15 *    derived from this software without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29package org.jf.baksmali;
30
31import org.jf.dexlib.DexFile;
32import org.jf.dexlib.Util.ByteArrayAnnotatedOutput;
33
34import java.io.FileOutputStream;
35import java.io.FileWriter;
36import java.io.IOException;
37
38public class dump {
39    public static void dump(DexFile dexFile, String dumpFileName, String outputDexFileName, boolean sort)
40            throws IOException {
41
42        if (sort) {
43            //sort all items, to guarantee a unique ordering
44            dexFile.setSortAllItems(true);
45        } else {
46            //don't change the order
47            dexFile.setInplace(true);
48        }
49
50        ByteArrayAnnotatedOutput out = new ByteArrayAnnotatedOutput();
51
52        if (dumpFileName != null) {
53            out.enableAnnotations(120, true);
54        }
55
56        dexFile.place();
57        dexFile.writeTo(out);
58
59        //write the dump
60        if (dumpFileName != null) {
61            out.finishAnnotating();
62            FileWriter writer = null;
63
64
65            try {
66                writer = new FileWriter(dumpFileName);
67                out.writeAnnotationsTo(writer);
68            } catch (IOException ex) {
69                System.err.println("\n\nThere was an error while dumping the dex file to " + dumpFileName);
70                ex.printStackTrace();
71            } finally {
72                if (writer != null) {
73                    try {
74                        writer.close();
75                    } catch (IOException ex) {
76                        System.err.println("\n\nThere was an error while closing the dump file " + dumpFileName);
77                        ex.printStackTrace();
78                    }
79                }
80            }
81        }
82
83        //rewrite the dex file
84        if (outputDexFileName != null) {
85            byte[] bytes = out.toByteArray();
86
87            DexFile.calcSignature(bytes);
88            DexFile.calcChecksum(bytes);
89
90            FileOutputStream fileOutputStream = null;
91            try {
92                fileOutputStream = new FileOutputStream(outputDexFileName);
93                fileOutputStream.write(bytes);
94            } catch (IOException ex) {
95                System.err.println("\n\nThere was an error while writing the dex file " + outputDexFileName);
96                ex.printStackTrace();
97            } finally {
98                if (fileOutputStream != null) {
99                    try {
100                        fileOutputStream.close();
101                    } catch (IOException ex) {
102                        System.err.println("\n\nThere was an error while closing the dex file " + outputDexFileName);
103                        ex.printStackTrace();
104                    }
105                }
106            }
107        }
108    }
109}
110