/* * This program and the accompanying materials * are made available under the terms of the Eclipse Public License v1.0 * which accompanies this distribution, and is available at * http://www.eclipse.org/legal/epl-v10.html. * * This file is a derivative of code released under the terms listed below. * */ /* * Copyright (c) 2013, * Tobias Blaschke <code@tobiasblaschke.de> * All rights reserved. * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * 1. Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * * 2. 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. * * 3. The names of the contributors may not 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 com.ibm.wala.dalvik.ipa.callgraph.androidModel.parameters; import java.io.IOException; import java.io.Serializable; import java.util.HashMap; import java.util.Map; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import com.ibm.wala.classLoader.IClass; import com.ibm.wala.classLoader.IClassLoader; import com.ibm.wala.ipa.cha.IClassHierarchy; import com.ibm.wala.types.MethodReference; import com.ibm.wala.types.TypeName; import com.ibm.wala.util.strings.Atom; /** * Behavior loaded from a file. * * This class generates an empty mutable IInstantiationBehavior. * * @author Tobias Blaschke <code@tobiasblaschke.de> * @since 2013-10-25 */ public class LoadedInstantiationBehavior extends IInstantiationBehavior implements Serializable { private static Logger logger = LoggerFactory.getLogger(LoadedInstantiationBehavior.class); private static final class BehviourValue implements Serializable { public final InstanceBehavior behaviour; public final Exactness exactness; public final BehviourValue cacheFrom; public BehviourValue(final InstanceBehavior behaviour, final Exactness exactness, final BehviourValue cacheFrom) { this.behaviour = behaviour; this.exactness = exactness; this.cacheFrom = cacheFrom; } @Override public String toString() { return "<BehaviorValue " + behaviour + " - " + exactness + " = " + cacheFrom + ">"; } /** * If the value can be derived using an other mapping. */ public boolean isCached() { return this.cacheFrom != null; } } private static final class BehaviorKey<T> implements Serializable { // T is expected to be TypeName or Atom final T base; public BehaviorKey(T base) { this.base = base; } public static BehaviorKey<TypeName> mk(TypeName base) { return new BehaviorKey<TypeName>(base); } public static BehaviorKey<Atom> mk(Atom base) { return new BehaviorKey<Atom>(base); } @Override public boolean equals(Object o) { if (o instanceof BehaviorKey) { BehaviorKey other = (BehaviorKey) o; return base.equals(other.base); } else { return false; } } @Override public int hashCode() { return this.base.hashCode(); } @Override public String toString() { return "<BehaviorKey of " + base.getClass() + " " + base.toString() + " hash=" + this.hashCode() + "/>"; } } private InstanceBehavior defaultBehavior = null; private final Map<BehaviorKey, BehviourValue> behaviours = new HashMap<BehaviorKey, BehviourValue>(); private final IClassHierarchy cha; public LoadedInstantiationBehavior(IClassHierarchy cha) { this.cha = cha; } public void setDefaultBehavior(InstanceBehavior defaultBehavior) { this.defaultBehavior = defaultBehavior; } /** * {@inheritDoc} * * @param asParameterTo not considered * @param inCall not considered * @param withName not considered */ @Override public InstanceBehavior getBehavior(final TypeName type, final TypeName asParameterTo, final MethodReference inCall, final String withName) { if (type == null) { throw new IllegalArgumentException("type may not be null"); } final BehaviorKey<TypeName> typeK = BehaviorKey.mk(type); if (behaviours.containsKey(typeK)) { BehviourValue typeV = behaviours.get(typeK); while (typeV.cacheFrom != null) { typeV = typeV.cacheFrom; } return typeV.behaviour; } //System.out.println(typeK.toString() + " not in " + behaviours); // Search based on package { final Atom pack = type.getPackage(); if (pack != null) { final BehaviorKey<Atom> packK= BehaviorKey.mk(pack); if (behaviours.containsKey(packK)) { // Add (cache) the result final BehviourValue packV = behaviours.get(packK); final InstanceBehavior beh = packV.behaviour; behaviours.put(typeK, new BehviourValue(beh, Exactness.PACKAGE, packV)); return beh; } } } // Search the super-classes { if (this.cha != null) { IClass testClass = null; for (final IClassLoader loader : this.cha.getLoaders()) { testClass = loader.lookupClass(type); if (testClass != null) { testClass = testClass.getSuperclass(); break; } } while (testClass != null) { final BehaviorKey<TypeName> testKey = BehaviorKey.mk(testClass.getName()); if (behaviours.containsKey(testKey)) { // Add (cache) the result final BehviourValue value = behaviours.get(testKey); final InstanceBehavior beh = value.behaviour; behaviours.put(typeK, new BehviourValue(beh, Exactness.INHERITED, value)); return beh; } testClass = testClass.getSuperclass(); } } else { logger.warn("I got no classhierarchy. Can't test super classes"); } } // Search based on prefix { String prefix = type.toString(); while (prefix.contains("/")) { prefix = prefix.substring(0, prefix.lastIndexOf("/") - 1); final BehaviorKey<Atom> prefixKey= BehaviorKey.mk(Atom.findOrCreateAsciiAtom(prefix)); if (behaviours.containsKey(prefixKey)) { // cache final BehviourValue value = behaviours.get(prefixKey); final InstanceBehavior beh = value.behaviour; behaviours.put(typeK, new BehviourValue(beh, Exactness.PREFIX, value)); return beh; } } } // */ // Fall back to default { final InstanceBehavior beh = getDafultBehavior(); final BehviourValue packV = new BehviourValue(beh, Exactness.DEFAULT, null); behaviours.put(typeK, packV); return beh; } } /** * {@inheritDoc} * * The DefaultInstanciationBehavior only knows EXACT, PACKAGE, PREFIX and DEFAULT */ @Override public Exactness getExactness(final TypeName type, final TypeName asParameterTo, final MethodReference inCall, final String withName) { if (type == null) { throw new IllegalArgumentException("type may not be null"); } final BehaviorKey<TypeName> typeK = BehaviorKey.mk(type); if (! behaviours.containsKey(typeK)) { // Use sideeffect: caches. getBehavior(type, asParameterTo, inCall, withName); } return behaviours.get(typeK).exactness; } /** * */ @Override public InstanceBehavior getDafultBehavior() { if (this.defaultBehavior == null) { return InstanceBehavior.REUSE; } else { return this.defaultBehavior; } } public void setBehavior(final TypeName type, final TypeName asParameterTo, final MethodReference inCall, final String withName, final InstanceBehavior beh, final Exactness exactness) { final BehaviorKey<TypeName> typeK= BehaviorKey.mk(type); final BehviourValue val = new BehviourValue(beh, exactness, null); behaviours.put(typeK, val); } public void setBehavior(final Atom pack, final TypeName asParameterTo, final MethodReference inCall, final String withName, final InstanceBehavior beh, final Exactness exactness) { final BehaviorKey<Atom> typeK= BehaviorKey.mk(pack); final BehviourValue val = new BehviourValue(beh, exactness, null); behaviours.put(typeK, val); } /** * Convert a TypeName back to an Atom. */ protected static Atom type2atom(TypeName type) { return Atom.findOrCreateAsciiAtom(type.toString()); } // // (De-)Serialization stuff follows // /** * The last eight digits encode the date. */ private static final long serialVersionUID = 810020131212L; /** * Including the cache may be useful to get all seen types. */ public transient boolean serializationIncludesCache = true; private void writeObject(java.io.ObjectOutputStream stream) throws IOException { if (this.serializationIncludesCache) { stream.writeObject(this.behaviours); } else { final Map<BehaviorKey, BehviourValue> strippedBehaviours = new HashMap<BehaviorKey, BehviourValue>(); for (final BehaviorKey key : this.behaviours.keySet()) { final BehviourValue val = this.behaviours.get(key); if (! val.isCached() ) { strippedBehaviours.put(key, val); } } stream.writeObject(strippedBehaviours); } } private void readObject(java.io.ObjectInputStream stream) throws IOException, ClassNotFoundException { this.behaviours.clear(); this.behaviours.putAll( (Map<BehaviorKey, BehviourValue>) stream.readObject()); } }