163 lines
5.4 KiB
Java
163 lines
5.4 KiB
Java
|
/*******************************************************************************
|
||
|
* Copyright (c) 2011 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.test;
|
||
|
|
||
|
import java.util.ArrayList;
|
||
|
import java.util.Collection;
|
||
|
import java.util.Collections;
|
||
|
import java.util.Comparator;
|
||
|
import java.util.Map;
|
||
|
import java.util.Map.Entry;
|
||
|
|
||
|
import com.ibm.wala.cast.js.ipa.callgraph.correlations.extraction.NodeLabeller;
|
||
|
import com.ibm.wala.cast.tree.CAstControlFlowMap;
|
||
|
import com.ibm.wala.cast.tree.CAstEntity;
|
||
|
import com.ibm.wala.cast.tree.CAstNode;
|
||
|
import com.ibm.wala.cast.util.CAstPrinter;
|
||
|
import com.ibm.wala.util.collections.HashMapFactory;
|
||
|
|
||
|
/**
|
||
|
* A class for dumping a textual representation of a CAst syntax tree.
|
||
|
*
|
||
|
* <p>Similar to {@link CAstPrinter}, but additionally prints control flow information.</p>
|
||
|
*
|
||
|
* <p>It also suppresses certain kinds of spurious nodes such as empty statements within a block or
|
||
|
* block expressions with a single child, which are simply artifacts of the translation process. This
|
||
|
* is not nice, but needed for our tests.</p>
|
||
|
*
|
||
|
* @author mschaefer
|
||
|
*
|
||
|
*/
|
||
|
public class CAstDumper {
|
||
|
private final NodeLabeller labeller;
|
||
|
|
||
|
public CAstDumper() {
|
||
|
labeller = new NodeLabeller();
|
||
|
}
|
||
|
|
||
|
public CAstDumper(NodeLabeller labeller) {
|
||
|
this.labeller = labeller;
|
||
|
}
|
||
|
|
||
|
private static String indent(int indent) {
|
||
|
StringBuilder buf = new StringBuilder();
|
||
|
for(int i=0;i<indent;++i)
|
||
|
buf.append(' ');
|
||
|
return buf.toString();
|
||
|
}
|
||
|
|
||
|
public String dump(CAstEntity entity) {
|
||
|
StringBuilder buf = new StringBuilder();
|
||
|
dump(entity, 0, buf);
|
||
|
return buf.toString();
|
||
|
}
|
||
|
|
||
|
private void dump(CAstEntity entity, int indent, StringBuilder buf) {
|
||
|
Collection<CAstEntity> scopedEntities = Collections.emptySet();
|
||
|
if(entity.getKind() == CAstEntity.SCRIPT_ENTITY) {
|
||
|
buf.append(indent(indent) + entity.getName() + ":\n");
|
||
|
scopedEntities = dumpScopedEntities(entity, indent+2, buf);
|
||
|
dump(entity.getAST(), indent, buf, entity.getControlFlow());
|
||
|
} else if(entity.getKind() == CAstEntity.FUNCTION_ENTITY) {
|
||
|
buf.append(indent(indent) + "function " + entity.getName() + "(");
|
||
|
for(int i=0;i<entity.getArgumentCount();++i) {
|
||
|
if(i>0)
|
||
|
buf.append(", ");
|
||
|
buf.append(entity.getArgumentNames()[i]);
|
||
|
}
|
||
|
buf.append(") {\n");
|
||
|
scopedEntities = dumpScopedEntities(entity, indent+2, buf);
|
||
|
dump(entity.getAST(), indent+2, buf, entity.getControlFlow());
|
||
|
buf.append(indent(indent) + "}\n\n");
|
||
|
} else {
|
||
|
throw new Error("Unknown entity kind " + entity.getKind());
|
||
|
}
|
||
|
for(CAstEntity scopedEntity : scopedEntities)
|
||
|
dump(scopedEntity, indent, buf);
|
||
|
}
|
||
|
|
||
|
private Collection<CAstEntity> dumpScopedEntities(CAstEntity entity, int indent, StringBuilder buf) {
|
||
|
ArrayList<CAstEntity> scopedEntities = new ArrayList<CAstEntity>();
|
||
|
Map<CAstEntity, CAstNode> m = HashMapFactory.make();
|
||
|
for(Entry<CAstNode, Collection<CAstEntity>> e : entity.getAllScopedEntities().entrySet())
|
||
|
for(CAstEntity scopedEntity : e.getValue()) {
|
||
|
scopedEntities.add(scopedEntity);
|
||
|
m.put(scopedEntity, e.getKey());
|
||
|
}
|
||
|
Collections.sort(scopedEntities, new Comparator<CAstEntity>() {
|
||
|
public int compare(CAstEntity o1, CAstEntity o2) {
|
||
|
return o1.getName().compareTo(o2.getName());
|
||
|
}
|
||
|
});
|
||
|
|
||
|
buf.append(indent(indent) + "> ");
|
||
|
boolean first = true;
|
||
|
for(CAstEntity scopedEntity : scopedEntities) {
|
||
|
if(first)
|
||
|
first = false;
|
||
|
else
|
||
|
buf.append(", ");
|
||
|
buf.append(scopedEntity.getName() + "@" + labeller.addNode(m.get(scopedEntity)));
|
||
|
}
|
||
|
buf.append("\n");
|
||
|
return scopedEntities;
|
||
|
}
|
||
|
|
||
|
private void dump(CAstNode node, int indent, StringBuilder buf, CAstControlFlowMap cfg) {
|
||
|
// normalise away single-child block expressions
|
||
|
if(node.getKind() == CAstNode.BLOCK_EXPR && node.getChildCount() == 1) {
|
||
|
dump(node.getChild(0), indent, buf, cfg);
|
||
|
} else {
|
||
|
buf.append(indent(indent) + labeller.addNode(node) + ": ");
|
||
|
if(node.getKind() == CAstNode.CONSTANT) {
|
||
|
if(node.getValue() == null)
|
||
|
buf.append("null");
|
||
|
else if(node.getValue() instanceof Integer)
|
||
|
buf.append(node.getValue()+"");
|
||
|
else
|
||
|
buf.append("\"" + node.getValue() + "\"");
|
||
|
} else if(node.getKind() == CAstNode.OPERATOR) {
|
||
|
buf.append(node.getValue().toString());
|
||
|
} else {
|
||
|
buf.append(CAstPrinter.kindAsString(node.getKind()));
|
||
|
}
|
||
|
Collection<Object> labels = cfg.getTargetLabels(node);
|
||
|
if(!labels.isEmpty()) {
|
||
|
buf.append(" [");
|
||
|
boolean first = true;
|
||
|
for(Object label : labels) {
|
||
|
CAstNode target = cfg.getTarget(node, label);
|
||
|
if(first) {
|
||
|
first = false;
|
||
|
} else {
|
||
|
buf.append(", ");
|
||
|
}
|
||
|
if(label instanceof CAstNode)
|
||
|
buf.append("CAstNode@" + labeller.addNode((CAstNode)label) + ": ");
|
||
|
else
|
||
|
buf.append(label + ": ");
|
||
|
buf.append(labeller.addNode(target));
|
||
|
}
|
||
|
buf.append("]");
|
||
|
}
|
||
|
buf.append("\n");
|
||
|
for(int i=0;i<node.getChildCount();++i) {
|
||
|
CAstNode child = node.getChild(i);
|
||
|
// omit empty statements in a block
|
||
|
if(node.getKind() == CAstNode.BLOCK_STMT && child != null && child.getKind() == CAstNode.EMPTY)
|
||
|
continue;
|
||
|
dump(child, indent+2, buf, cfg);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|