94 lines
3.4 KiB
Java
94 lines
3.4 KiB
Java
/******************************************************************************
|
|
* Copyright (c) 2002 - 2006 IBM Corporation.
|
|
* All rights reserved. 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
|
|
*
|
|
* Contributors:
|
|
* IBM Corporation - initial API and implementation
|
|
*****************************************************************************/
|
|
package com.ibm.wala.cast.js.ipa.callgraph;
|
|
|
|
import java.util.Set;
|
|
|
|
import com.ibm.wala.cast.ipa.callgraph.AstCallGraph;
|
|
import com.ibm.wala.cast.js.cfg.JSInducedCFG;
|
|
import com.ibm.wala.cast.js.ssa.JavaScriptInvoke;
|
|
import com.ibm.wala.cast.js.types.JavaScriptMethods;
|
|
import com.ibm.wala.cast.js.types.JavaScriptTypes;
|
|
import com.ibm.wala.cast.loader.DynamicCallSiteReference;
|
|
import com.ibm.wala.cfg.InducedCFG;
|
|
import com.ibm.wala.classLoader.CallSiteReference;
|
|
import com.ibm.wala.classLoader.IMethod;
|
|
import com.ibm.wala.ipa.callgraph.AnalysisOptions;
|
|
import com.ibm.wala.ipa.callgraph.CGNode;
|
|
import com.ibm.wala.ipa.callgraph.IAnalysisCacheView;
|
|
import com.ibm.wala.ipa.callgraph.impl.Everywhere;
|
|
import com.ibm.wala.ipa.callgraph.impl.FakeRootMethod;
|
|
import com.ibm.wala.ipa.cha.IClassHierarchy;
|
|
import com.ibm.wala.ssa.SSAAbstractInvokeInstruction;
|
|
import com.ibm.wala.ssa.SSAInstruction;
|
|
import com.ibm.wala.types.MethodReference;
|
|
import com.ibm.wala.util.collections.HashSetFactory;
|
|
|
|
public class JSCallGraph extends AstCallGraph {
|
|
|
|
public JSCallGraph(IMethod fakeRootClass, AnalysisOptions options, IAnalysisCacheView cache) {
|
|
super(fakeRootClass, options, cache);
|
|
}
|
|
|
|
public final static MethodReference fakeRoot = MethodReference.findOrCreate(JavaScriptTypes.FakeRoot, FakeRootMethod.name,
|
|
FakeRootMethod.descr);
|
|
|
|
public static class JSFakeRoot extends ScriptFakeRoot {
|
|
|
|
public JSFakeRoot(IClassHierarchy cha, AnalysisOptions options, IAnalysisCacheView cache) {
|
|
super(fakeRoot, cha.lookupClass(JavaScriptTypes.FakeRoot), cha, options, cache);
|
|
}
|
|
|
|
@Override
|
|
public InducedCFG makeControlFlowGraph(SSAInstruction[] instructions) {
|
|
return new JSInducedCFG(instructions, this, Everywhere.EVERYWHERE);
|
|
}
|
|
|
|
@Override
|
|
public SSAAbstractInvokeInstruction addDirectCall(int function, int[] params, CallSiteReference site) {
|
|
CallSiteReference newSite = new DynamicCallSiteReference(JavaScriptTypes.CodeBody, statements.size());
|
|
|
|
JavaScriptInvoke s = new JavaScriptInvoke(statements.size(), function, nextLocal++, params, nextLocal++, newSite);
|
|
statements.add(s);
|
|
|
|
return s;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected CGNode makeFakeWorldClinitNode() {
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
protected CGNode makeFakeRootNode() throws com.ibm.wala.util.CancelException {
|
|
return findOrCreateNode(new JSFakeRoot(cha, options, getAnalysisCache()), Everywhere.EVERYWHERE);
|
|
}
|
|
|
|
@Override
|
|
public Set<CGNode> getNodes(MethodReference m) {
|
|
if (m.getName().equals(JavaScriptMethods.ctorAtom)) {
|
|
// TODO cache this?
|
|
Set<CGNode> result = HashSetFactory.make(1);
|
|
for (CGNode n : this) {
|
|
IMethod method = n.getMethod();
|
|
if (method.getName().equals(JavaScriptMethods.ctorAtom) && method.getDeclaringClass().getReference().equals(m.getDeclaringClass())) {
|
|
result.add(n);
|
|
}
|
|
}
|
|
return result;
|
|
} else {
|
|
return super.getNodes(m);
|
|
}
|
|
}
|
|
|
|
}
|