1/* 2 * Copyright 2014, Google Inc. 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are 7 * met: 8 * 9 * * Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * * Redistributions in binary form must reproduce the above 12 * copyright notice, this list of conditions and the following disclaimer 13 * in the documentation and/or other materials provided with the 14 * distribution. 15 * * Neither the name of Google Inc. nor the names of its 16 * contributors may be used to endorse or promote products derived from 17 * this software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 20 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 22 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 23 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 24 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 25 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 29 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32package org.jf.dexlib2.rewriter; 33 34import com.google.common.collect.Iterators; 35import org.jf.dexlib2.base.reference.BaseTypeReference; 36import org.jf.dexlib2.iface.Annotation; 37import org.jf.dexlib2.iface.ClassDef; 38import org.jf.dexlib2.iface.Field; 39import org.jf.dexlib2.iface.Method; 40 41import javax.annotation.Nonnull; 42import javax.annotation.Nullable; 43import java.util.Iterator; 44import java.util.List; 45import java.util.Set; 46 47public class ClassDefRewriter implements Rewriter<ClassDef> { 48 @Nonnull protected final Rewriters rewriters; 49 50 public ClassDefRewriter(@Nonnull Rewriters rewriters) { 51 this.rewriters = rewriters; 52 } 53 54 @Nonnull @Override public ClassDef rewrite(@Nonnull ClassDef classDef) { 55 return new RewrittenClassDef(classDef); 56 } 57 58 protected class RewrittenClassDef extends BaseTypeReference implements ClassDef { 59 @Nonnull protected ClassDef classDef; 60 61 public RewrittenClassDef(@Nonnull ClassDef classdef) { 62 this.classDef = classdef; 63 } 64 65 @Override @Nonnull public String getType() { 66 return rewriters.getTypeRewriter().rewrite(classDef.getType()); 67 } 68 69 @Override public int getAccessFlags() { 70 return classDef.getAccessFlags(); 71 } 72 73 @Override @Nullable public String getSuperclass() { 74 return RewriterUtils.rewriteNullable(rewriters.getTypeRewriter(), classDef.getSuperclass()); 75 } 76 77 @Override @Nonnull public List<String> getInterfaces() { 78 return RewriterUtils.rewriteList(rewriters.getTypeRewriter(), classDef.getInterfaces()); 79 } 80 81 @Override @Nullable public String getSourceFile() { 82 return classDef.getSourceFile(); 83 } 84 85 @Override @Nonnull public Set<? extends Annotation> getAnnotations() { 86 return RewriterUtils.rewriteSet(rewriters.getAnnotationRewriter(), classDef.getAnnotations()); 87 } 88 89 @Override @Nonnull public Iterable<? extends Field> getStaticFields() { 90 return RewriterUtils.rewriteIterable(rewriters.getFieldRewriter(), classDef.getStaticFields()); 91 } 92 93 @Override @Nonnull public Iterable<? extends Field> getInstanceFields() { 94 return RewriterUtils.rewriteIterable(rewriters.getFieldRewriter(), classDef.getInstanceFields()); 95 } 96 97 @Nonnull 98 @Override 99 public Iterable<? extends Field> getFields() { 100 return new Iterable<Field>() { 101 @Nonnull 102 @Override 103 public Iterator<Field> iterator() { 104 return Iterators.concat(getStaticFields().iterator(), getInstanceFields().iterator()); 105 } 106 }; 107 } 108 109 @Override @Nonnull public Iterable<? extends Method> getDirectMethods() { 110 return RewriterUtils.rewriteIterable(rewriters.getMethodRewriter(), classDef.getDirectMethods()); 111 } 112 113 @Override @Nonnull public Iterable<? extends Method> getVirtualMethods() { 114 return RewriterUtils.rewriteIterable(rewriters.getMethodRewriter(), classDef.getVirtualMethods()); 115 } 116 117 @Nonnull 118 @Override 119 public Iterable<? extends Method> getMethods() { 120 return new Iterable<Method>() { 121 @Nonnull 122 @Override 123 public Iterator<Method> iterator() { 124 return Iterators.concat(getDirectMethods().iterator(), getVirtualMethods().iterator()); 125 } 126 }; 127 } 128 } 129} 130