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.Set; 45 46public class ClassDefRewriter implements Rewriter<ClassDef> { 47 @Nonnull protected final Rewriters rewriters; 48 49 public ClassDefRewriter(@Nonnull Rewriters rewriters) { 50 this.rewriters = rewriters; 51 } 52 53 @Nonnull @Override public ClassDef rewrite(@Nonnull ClassDef classDef) { 54 return new RewrittenClassDef(classDef); 55 } 56 57 protected class RewrittenClassDef extends BaseTypeReference implements ClassDef { 58 @Nonnull protected ClassDef classDef; 59 60 public RewrittenClassDef(@Nonnull ClassDef classdef) { 61 this.classDef = classdef; 62 } 63 64 @Override @Nonnull public String getType() { 65 return rewriters.getTypeRewriter().rewrite(classDef.getType()); 66 } 67 68 @Override public int getAccessFlags() { 69 return classDef.getAccessFlags(); 70 } 71 72 @Override @Nullable public String getSuperclass() { 73 return RewriterUtils.rewriteNullable(rewriters.getTypeRewriter(), classDef.getSuperclass()); 74 } 75 76 @Override @Nonnull public Set<String> getInterfaces() { 77 return RewriterUtils.rewriteSet(rewriters.getTypeRewriter(), classDef.getInterfaces()); 78 } 79 80 @Override @Nullable public String getSourceFile() { 81 return classDef.getSourceFile(); 82 } 83 84 @Override @Nonnull public Set<? extends Annotation> getAnnotations() { 85 return RewriterUtils.rewriteSet(rewriters.getAnnotationRewriter(), classDef.getAnnotations()); 86 } 87 88 @Override @Nonnull public Iterable<? extends Field> getStaticFields() { 89 return RewriterUtils.rewriteIterable(rewriters.getFieldRewriter(), classDef.getStaticFields()); 90 } 91 92 @Override @Nonnull public Iterable<? extends Field> getInstanceFields() { 93 return RewriterUtils.rewriteIterable(rewriters.getFieldRewriter(), classDef.getInstanceFields()); 94 } 95 96 @Nonnull 97 @Override 98 public Iterable<? extends Field> getFields() { 99 return new Iterable<Field>() { 100 @Nonnull 101 @Override 102 public Iterator<Field> iterator() { 103 return Iterators.concat(getStaticFields().iterator(), getInstanceFields().iterator()); 104 } 105 }; 106 } 107 108 @Override @Nonnull public Iterable<? extends Method> getDirectMethods() { 109 return RewriterUtils.rewriteIterable(rewriters.getMethodRewriter(), classDef.getDirectMethods()); 110 } 111 112 @Override @Nonnull public Iterable<? extends Method> getVirtualMethods() { 113 return RewriterUtils.rewriteIterable(rewriters.getMethodRewriter(), classDef.getVirtualMethods()); 114 } 115 116 @Nonnull 117 @Override 118 public Iterable<? extends Method> getMethods() { 119 return new Iterable<Method>() { 120 @Nonnull 121 @Override 122 public Iterator<Method> iterator() { 123 return Iterators.concat(getDirectMethods().iterator(), getVirtualMethods().iterator()); 124 } 125 }; 126 } 127 } 128} 129