/testability-explorer/src/main/java/com/google/test/metric/JavaClassRepository.java
Java | 111 lines | 78 code | 14 blank | 19 comment | 16 complexity | 99c31401a4e26a9ca83e6325f8010a72 MD5 | raw file
Possible License(s): Apache-2.0
- /*
- * Copyright 2007 Google Inc.
- *
- * Licensed under the Apache License, Version 2.0 (the "License"); you may not
- * use this file except in compliance with the License. You may obtain a copy of
- * the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
- * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
- * License for the specific language governing permissions and limitations under
- * the License.
- */
- package com.google.test.metric;
- import java.io.IOException;
- import java.io.InputStream;
- import java.util.HashMap;
- import java.util.Map;
- import org.objectweb.asm.ClassReader;
- import com.google.classpath.ClassPath;
- import com.google.test.metric.asm.ClassInfoBuilderVisitor;
- public class JavaClassRepository implements ClassRepository {
- private final Map<String, ClassInfo> classes = new HashMap<String, ClassInfo>();
- private ClassPath classpathRoots;
- public JavaClassRepository() {
- }
- public JavaClassRepository(ClassPath classpathRoots) {
- this.classpathRoots = classpathRoots;
- }
- public ClassInfo getClass(String name) {
- if (name.startsWith("[")) {
- return getClass(Object.class.getCanonicalName());
- }
- if (name.contains("$") || name.contains("/")) {
- throw new IllegalStateException("Class name can not contain '$' or '/' in a name: " + name);
- }
- ClassInfo classInfo = classes.get(name);
- if (classInfo == null) {
- try {
- classInfo = parseClass(inputStreamForClass(name));
- } catch (ArrayIndexOutOfBoundsException e) {
- throw new ClassNotFoundException(name);
- } catch (ClassNotFoundException e) {
- throw new ClassNotFoundException(name, e);
- }
- }
- return classInfo;
- }
- private InputStream inputStreamForClass(String clazzName) {
- String resource = clazzName.replace(".", "/");
- InputStream classBytes = null;
- while (true) {
- classBytes = getResource(resource + ".class");
- if (classBytes != null) {
- return classBytes;
- }
- int index = resource.lastIndexOf('/');
- if (index == -1) {
- throw new ClassNotFoundException(clazzName);
- }
- resource = resource.substring(0, index) + "$" + resource.substring(index + 1);
- }
- }
- private InputStream getResource(String classResource) {
- InputStream classBytes = null;
- if (classpathRoots != null) {
- classBytes = classpathRoots.getResourceAsStream(classResource);
- }
- if (classBytes == null) {
- //Perhaps it is a JDK Class
- classBytes = ClassLoader.getSystemResourceAsStream(classResource);
- }
- return classBytes;
- }
- private ClassInfo parseClass(InputStream classBytes) {
- try {
- ClassReader classReader = new ClassReader(classBytes);
- ClassInfoBuilderVisitor visitor = new ClassInfoBuilderVisitor(this);
- classReader.accept(visitor, 0);
- return visitor.getClassInfo();
- } catch (IOException e) {
- throw new RuntimeException(e);
- }
- }
- /* (non-Javadoc)
- * @see com.google.test.metric.ClassRepository#addClass(com.google.test.metric.ClassInfo)
- */
- public void addClass(ClassInfo classInfo) {
- String name = classInfo.getName();
- if (name.contains("$") || name.contains("/")) {
- throw new IllegalStateException();
- }
- classes.put(name, classInfo);
- }
- }