PostInstructionRegisterInfoMethodItem.java revision 4b72225e9d81201838f387171a68a832486903f9 
1/*
2 * [The "BSD licence"]
3 * Copyright (c) 2010 Ben Gruver
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.Adaptors;
30
31import org.jf.util.IndentingWriter;
32import org.jf.baksmali.baksmali;
33import org.jf.baksmali.main;
34import org.jf.dexlib.ClassDataItem;
35import org.jf.dexlib.Code.Analysis.AnalyzedInstruction;
36import org.jf.dexlib.Code.Analysis.MethodAnalyzer;
37import org.jf.dexlib.Code.Analysis.RegisterType;
38
39import java.io.IOException;
40import java.util.BitSet;
41
42public class PostInstructionRegisterInfoMethodItem extends MethodItem {
43    private final AnalyzedInstruction analyzedInstruction;
44    private final MethodAnalyzer methodAnalyzer;
45
46    public PostInstructionRegisterInfoMethodItem(AnalyzedInstruction analyzedInstruction, MethodAnalyzer methodAnalyzer,
47                                                int codeAddress) {
48        super(codeAddress);
49        this.analyzedInstruction = analyzedInstruction;
50        this.methodAnalyzer = methodAnalyzer;
51    }
52
53    @Override
54    public double getSortOrder() {
55        return 100.1;
56    }
57
58    @Override
59    public boolean writeTo(IndentingWriter writer) throws IOException {
60        int registerInfo = baksmali.registerInfo;
61        int registerCount = analyzedInstruction.getRegisterCount();
62        BitSet registers = new BitSet(registerCount);
63
64        if ((registerInfo & main.ALL) != 0) {
65            registers.set(0, registerCount);
66        } else {
67            if ((registerInfo & main.ALLPOST) != 0) {
68                registers.set(0, registerCount);
69            } else if ((registerInfo & main.DEST) != 0) {
70                addDestRegs(registers, registerCount);
71            }
72        }
73
74        return writeRegisterInfo(writer, registers);
75    }
76
77    private void addDestRegs(BitSet printPostRegister, int registerCount) {
78        for (int registerNum=0; registerNum<registerCount; registerNum++) {
79            if (analyzedInstruction.getPreInstructionRegisterType(registerNum) !=
80                    analyzedInstruction.getPostInstructionRegisterType(registerNum)) {
81                printPostRegister.set(registerNum);
82            }
83        }
84    }
85
86    private boolean writeRegisterInfo(IndentingWriter writer, BitSet registers) throws IOException {
87        ClassDataItem.EncodedMethod encodedMethod = methodAnalyzer.getMethod();
88
89        int registerNum = registers.nextSetBit(0);
90        if (registerNum < 0) {
91            return false;
92        }
93
94        writer.write('#');
95        for (; registerNum >= 0; registerNum = registers.nextSetBit(registerNum + 1)) {
96
97            RegisterType registerType = analyzedInstruction.getPostInstructionRegisterType(registerNum);
98
99            RegisterFormatter.writeTo(writer, encodedMethod.codeItem, registerNum);
100            writer.write('=');
101
102            if (registerType == null) {
103                writer.write("null");
104            } else {
105                registerType.writeTo(writer);
106            }
107            writer.write(';');
108        }
109        return true;
110    }
111}
112