Home | History | Annotate | Download | only in Adaptors 
      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 
     29 package org.jf.baksmali.Adaptors;
     30 
     31 import org.jf.baksmali.BaksmaliOptions;
     32 import org.jf.dexlib2.analysis.AnalyzedInstruction;
     33 import org.jf.dexlib2.analysis.RegisterType;
     34 import org.jf.util.IndentingWriter;
     35 
     36 import javax.annotation.Nonnull;
     37 import java.io.IOException;
     38 import java.util.BitSet;
     39 
     40 public class PostInstructionRegisterInfoMethodItem extends MethodItem {
     41     @Nonnull private final RegisterFormatter registerFormatter;
     42     @Nonnull private final AnalyzedInstruction analyzedInstruction;
     43 
     44     public PostInstructionRegisterInfoMethodItem(@Nonnull RegisterFormatter registerFormatter,
     45                                                  @Nonnull AnalyzedInstruction analyzedInstruction,
     46                                                  int codeAddress) {
     47         super(codeAddress);
     48         this.registerFormatter = registerFormatter;
     49         this.analyzedInstruction = analyzedInstruction;
     50     }
     51 
     52     @Override
     53     public double getSortOrder() {
     54         return 100.1;
     55     }
     56 
     57     @Override
     58     public boolean writeTo(IndentingWriter writer) throws IOException {
     59         int registerInfo = registerFormatter.options.registerInfo;
     60         int registerCount = analyzedInstruction.getRegisterCount();
     61         BitSet registers = new BitSet(registerCount);
     62 
     63         if ((registerInfo & BaksmaliOptions.ALL) != 0) {
     64             registers.set(0, registerCount);
     65         } else {
     66             if ((registerInfo & BaksmaliOptions.ALLPOST) != 0) {
     67                 registers.set(0, registerCount);
     68             } else if ((registerInfo & BaksmaliOptions.DEST) != 0) {
     69                 addDestRegs(registers, registerCount);
     70             }
     71         }
     72 
     73         return writeRegisterInfo(writer, registers);
     74     }
     75 
     76     private void addDestRegs(BitSet printPostRegister, int registerCount) {
     77         for (int registerNum=0; registerNum<registerCount; registerNum++) {
     78             if (!analyzedInstruction.getPreInstructionRegisterType(registerNum).equals(
     79                     analyzedInstruction.getPostInstructionRegisterType(registerNum))) {
     80                 printPostRegister.set(registerNum);
     81             }
     82         }
     83     }
     84 
     85     private boolean writeRegisterInfo(IndentingWriter writer, BitSet registers) throws IOException {
     86         int registerNum = registers.nextSetBit(0);
     87         if (registerNum < 0) {
     88             return false;
     89         }
     90 
     91         writer.write('#');
     92         for (; registerNum >= 0; registerNum = registers.nextSetBit(registerNum + 1)) {
     93             RegisterType registerType = analyzedInstruction.getPostInstructionRegisterType(registerNum);
     94 
     95             registerFormatter.writeTo(writer, registerNum);
     96             writer.write('=');
     97             registerType.writeTo(writer);
     98             writer.write(';');
     99         }
    100         return true;
    101     }
    102 }
    103