| /* |
| * Copyright 2014, Google Inc. |
| * All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions are |
| * met: |
| * |
| * * Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * * 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. |
| * * Neither the name of Google Inc. nor the names of its |
| * contributors may be used to endorse or promote products derived from |
| * this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| * "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 COPYRIGHT |
| * OWNER OR CONTRIBUTORS 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.smalidea.dexlib; |
| |
| import com.google.common.base.Function; |
| import com.google.common.collect.ImmutableList; |
| import com.google.common.collect.ImmutableSet; |
| import com.google.common.collect.Lists; |
| import com.intellij.psi.PsiClass; |
| import com.intellij.psi.PsiParameter; |
| import org.jetbrains.annotations.NotNull; |
| import org.jetbrains.annotations.Nullable; |
| import org.jf.dexlib2.base.reference.BaseMethodReference; |
| import org.jf.dexlib2.iface.*; |
| import org.jf.dexlib2.iface.debug.DebugItem; |
| import org.jf.dexlib2.iface.instruction.Instruction; |
| import org.jf.smalidea.dexlib.instruction.SmalideaInstruction; |
| import org.jf.smalidea.psi.impl.SmaliCatchStatement; |
| import org.jf.smalidea.psi.impl.SmaliInstruction; |
| import org.jf.smalidea.psi.impl.SmaliMethod; |
| import org.jf.smalidea.psi.impl.SmaliMethodParameter; |
| import org.jf.smalidea.util.NameUtils; |
| |
| import javax.annotation.Nonnull; |
| import java.util.Arrays; |
| import java.util.List; |
| import java.util.Set; |
| |
| public class SmalideaMethod extends BaseMethodReference implements Method { |
| private final SmaliMethod psiMethod; |
| |
| public SmalideaMethod(@NotNull SmaliMethod psiMethod) { |
| this.psiMethod = psiMethod; |
| } |
| |
| @Nonnull @Override public String getDefiningClass() { |
| PsiClass cls = psiMethod.getContainingClass(); |
| assert cls != null; |
| String qualifiedName = cls.getQualifiedName(); |
| assert qualifiedName != null; |
| return NameUtils.javaToSmaliType(qualifiedName); |
| } |
| |
| @Nonnull @Override public List<? extends MethodParameter> getParameters() { |
| SmaliMethodParameter[] parameters = psiMethod.getParameterList().getParameters(); |
| |
| return Lists.transform(Arrays.asList(parameters), new Function<SmaliMethodParameter, MethodParameter>() { |
| @Nullable @Override |
| public MethodParameter apply(@Nullable SmaliMethodParameter smaliParameter) { |
| if (smaliParameter == null) { |
| return null; |
| } |
| return new SmalideaMethodParameter(smaliParameter); |
| } |
| }); |
| } |
| |
| @Override public int getAccessFlags() { |
| return psiMethod.getModifierList().getAccessFlags(); |
| } |
| |
| @Nonnull @Override public Set<? extends Annotation> getAnnotations() { |
| // TODO: implement this |
| return ImmutableSet.of(); |
| } |
| |
| @Nullable @Override public MethodImplementation getImplementation() { |
| List<SmaliInstruction> instructions = psiMethod.getInstructions(); |
| if (instructions.size() == 0) { |
| return null; |
| } |
| |
| // TODO: cache this? |
| return new MethodImplementation() { |
| @Override public int getRegisterCount() { |
| return psiMethod.getRegisterCount(); |
| } |
| |
| @Nonnull @Override public Iterable<? extends Instruction> getInstructions() { |
| return Lists.transform(psiMethod.getInstructions(), |
| new Function<SmaliInstruction, Instruction>() { |
| @Override |
| public Instruction apply(SmaliInstruction smaliInstruction) { |
| return SmalideaInstruction.of(smaliInstruction); |
| } |
| }); |
| } |
| |
| @Nonnull @Override public List<? extends TryBlock<? extends ExceptionHandler>> getTryBlocks() { |
| return Lists.transform(psiMethod.getCatchStatements(), |
| new Function<SmaliCatchStatement, TryBlock<? extends ExceptionHandler>>() { |
| @Override |
| public TryBlock<? extends ExceptionHandler> apply( |
| SmaliCatchStatement smaliCatchStatement) { |
| assert smaliCatchStatement != null; |
| return new SmalideaTryBlock(smaliCatchStatement); |
| } |
| }); |
| } |
| |
| @Nonnull @Override public Iterable<? extends DebugItem> getDebugItems() { |
| // TODO: implement this |
| return ImmutableList.of(); |
| } |
| }; |
| } |
| |
| @Nonnull @Override public String getName() { |
| return psiMethod.getName(); |
| } |
| |
| @Nonnull @Override public List<? extends CharSequence> getParameterTypes() { |
| PsiParameter[] parameters = psiMethod.getParameterList().getParameters(); |
| |
| return Lists.transform(Arrays.asList(parameters), new Function<PsiParameter, CharSequence>() { |
| @Nullable @Override |
| public CharSequence apply(@Nullable PsiParameter psiParameter) { |
| if (psiParameter == null) { |
| return null; |
| } |
| return psiParameter.getText(); |
| } |
| }); |
| } |
| |
| @Nonnull @Override public String getReturnType() { |
| return psiMethod.getReturnTypeElement().getText(); |
| } |
| } |