001    /**
002     *  Licensed to the Apache Software Foundation (ASF) under one or more
003     *  contributor license agreements.  See the NOTICE file distributed with
004     *  this work for additional information regarding copyright ownership.
005     *  The ASF licenses this file to You under the Apache License, Version 2.0
006     *  (the "License"); you may not use this file except in compliance with
007     *  the License.  You may obtain a copy of the License at
008     *
009     *     http://www.apache.org/licenses/LICENSE-2.0
010     *
011     *  Unless required by applicable law or agreed to in writing, software
012     *  distributed under the License is distributed on an "AS IS" BASIS,
013     *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014     *  See the License for the specific language governing permissions and
015     *  limitations under the License.
016     */
017    package org.apache.geronimo.persistence;
018    
019    import java.lang.instrument.ClassFileTransformer;
020    import java.lang.instrument.IllegalClassFormatException;
021    import java.security.ProtectionDomain;
022    
023    import javax.persistence.spi.ClassTransformer;
024    
025    /**
026     * @version $Rev: 706640 $ $Date: 2008-10-21 14:44:05 +0000 (Tue, 21 Oct 2008) $
027     */
028    public class TransformerWrapper implements ClassFileTransformer {
029    
030        private final ClassTransformer classTransformer;
031        private final ClassLoader classLoader;
032    
033        public TransformerWrapper(ClassTransformer classTransformer, ClassLoader classLoader) {
034            this.classTransformer = classTransformer;
035            this.classLoader = classLoader;
036        }
037    
038        public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
039            if (loader != classLoader) {
040                return null;
041            }
042            try {
043                return classTransformer.transform(loader, className, classBeingRedefined,  protectionDomain, classfileBuffer);
044            } catch (IllegalClassFormatException e) {
045                throw e;
046            } catch (RuntimeException e) {
047                return null;
048            }
049        }
050    }