1 /*
2 * The Apache Software License, Version 1.1
3 *
4 * Copyright (c) 2002 The Apache Software Foundation. All rights
5 * reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 *
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 *
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in
16 * the documentation and/or other materials provided with the
17 * distribution.
18 *
19 * 3. The end-user documentation included with the redistribution,
20 * if any, must include the following acknowledgment:
21 * "This product includes software developed by the
22 * Apache Software Foundation (http://www.apache.org/)."
23 * Alternately, this acknowledgment may appear in the software itself,
24 * if and wherever such third-party acknowledgments normally appear.
25 *
26 * 4. The names "Apache" and "Apache Software Foundation" must
27 * not be used to endorse or promote products derived from this
28 * software without prior written permission. For written
29 * permission, please contact apache@apache.org.
30 *
31 * 5. Products derived from this software may not be called "Apache",
32 * nor may "Apache" appear in their name, without prior written
33 * permission of the Apache Software Foundation.
34 *
35 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
36 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
37 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
38 * DISCLAIMED. IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
39 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
40 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
41 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
42 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
43 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
44 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
45 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
46 * SUCH DAMAGE.
47 * ====================================================================
48 *
49 * This software consists of voluntary contributions made by many
50 * individuals on behalf of the Apache Software Foundation. For more
51 * information on the Apache Software Foundation, please see
52 * <http://www.apache.org/>.
53 */
54 package net.sf.cglib.core;
55
56 import org.objectweb.asm.ClassWriter;
57 import org.objectweb.asm.ClassReader;
58 import org.objectweb.asm.util.TraceClassVisitor;
59
60 import java.io.*;
61
62 public class DebuggingClassWriter extends ClassWriter {
63
64 public static final String DEBUG_LOCATION_PROPERTY = "cglib.debugLocation";
65
66 private static String debugLocation;
67 private static boolean traceEnabled;
68
69 private String className;
70 private String superName;
71
72 static {
73 debugLocation = System.getProperty(DEBUG_LOCATION_PROPERTY);
74 if (debugLocation != null) {
75 System.err.println("CGLIB debugging enabled, writing to '" + debugLocation + "'");
76 try {
77 Class.forName("org.objectweb.asm.util.TraceClassVisitor");
78 traceEnabled = true;
79 } catch (Throwable ignore) {
80 }
81 }
82 }
83
84 public DebuggingClassWriter(boolean computeMaxs) {
85 super(computeMaxs);
86 }
87
88 public void visit(int access, String name, String superName, String[] interfaces, String sourceFile) {
89 className = name.replace('/', '.');
90 this.superName = superName.replace('/', '.');
91 super.visit(access, name, superName, interfaces, sourceFile);
92 }
93
94 public String getClassName() {
95 return className;
96 }
97
98 public String getSuperName() {
99 return superName;
100 }
101
102 public byte[] toByteArray() {
103
104 return (byte[]) java.security.AccessController.doPrivileged(
105 new java.security.PrivilegedAction() {
106 public Object run() {
107
108
109 byte[] b = DebuggingClassWriter.super.toByteArray();
110 if (debugLocation != null) {
111 String dirs = className.replace('.', File.separatorChar);
112 try {
113 new File(debugLocation + File.separatorChar + dirs).getParentFile().mkdirs();
114
115 File file = new File(new File(debugLocation), dirs + ".class");
116 OutputStream out = new BufferedOutputStream(new FileOutputStream(file));
117 try {
118 out.write(b);
119 } finally {
120 out.close();
121 }
122
123 if (traceEnabled) {
124 file = new File(new File(debugLocation), dirs + ".asm");
125 out = new BufferedOutputStream(new FileOutputStream(file));
126 try {
127 ClassReader cr = new ClassReader(b);
128 PrintWriter pw = new PrintWriter(new OutputStreamWriter(out));
129 TraceClassVisitor tcv = new TraceClassVisitor(null, pw);
130 cr.accept(tcv, false);
131 pw.flush();
132 } finally {
133 out.close();
134 }
135 }
136 } catch (IOException e) {
137 throw new CodeGenerationException(e);
138 }
139 }
140 return b;
141 }
142 });
143
144 }
145 }
This page was automatically generated by Maven