01: /***
02: * ASM Guide
03: * Copyright (c) 2007 Eric Bruneton
04: * All rights reserved.
05: *
06: * Redistribution and use in source and binary forms, with or without
07: * modification, are permitted provided that the following conditions
08: * are met:
09: * 1. Redistributions of source code must retain the above copyright
10: * notice, this list of conditions and the following disclaimer.
11: * 2. Redistributions in binary form must reproduce the above copyright
12: * notice, this list of conditions and the following disclaimer in the
13: * documentation and/or other materials provided with the distribution.
14: * 3. Neither the name of the copyright holders nor the names of its
15: * contributors may be used to endorse or promote products derived from
16: * this software without specific prior written permission.
17: *
18: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
19: * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21: * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
22: * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23: * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24: * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25: * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26: * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27: * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
28: * THE POSSIBILITY OF SUCH DAMAGE.
29: */package ch7.sec2;
30:
31: import java.util.HashSet;
32: import java.util.Set;
33:
34: import org.objectweb.asm.tree.MethodNode;
35: import org.objectweb.asm.tree.analysis.Analyzer;
36: import org.objectweb.asm.tree.analysis.AnalyzerException;
37: import org.objectweb.asm.tree.analysis.BasicInterpreter;
38: import org.objectweb.asm.tree.analysis.Frame;
39:
40: /**
41: * ASM Guide example class.
42: *
43: * @author Eric Bruneton
44: */
45: public class CyclomaticComplexity {
46:
47: public int getCyclomaticComplexity(String owner, MethodNode mn)
48: throws AnalyzerException {
49: Analyzer a = new Analyzer(new BasicInterpreter()) {
50: protected Frame newFrame(int nLocals, int nStack) {
51: return new Node(nLocals, nStack);
52: }
53:
54: protected Frame newFrame(Frame src) {
55: return new Node(src);
56: }
57:
58: protected void newControlFlowEdge(int src, int dst) {
59: Node s = (Node) getFrames()[src];
60: s.successors.add((Node) getFrames()[dst]);
61: }
62: };
63: a.analyze(owner, mn);
64: Frame[] frames = a.getFrames();
65: int edges = 0;
66: int nodes = 0;
67: for (int i = 0; i < frames.length; ++i) {
68: if (frames[i] != null) {
69: edges += ((Node) frames[i]).successors.size();
70: nodes += 1;
71: }
72: }
73: return edges - nodes + 2;
74: }
75: }
76:
77: class Node extends Frame {
78:
79: Set<Node> successors = new HashSet<Node>();
80:
81: public Node(int nLocals, int nStack) {
82: super (nLocals, nStack);
83: }
84:
85: public Node(Frame src) {
86: super(src);
87: }
88: }
|