KotlinClassFileHeader is now nullable everywhere

'null' means there's no header. This makes all of KotlinClassFileHeader's
fields non-null
This commit is contained in:
Alexander Udalov
2013-09-19 18:50:07 +04:00
parent 9e9b3c7284
commit 92cc0ddd1f
7 changed files with 55 additions and 30 deletions
@@ -79,7 +79,7 @@ public class CliVirtualFileFinder implements VirtualFileFinder {
return null; return null;
} }
//NOTE: currently we use VirtualFileFinder to find Kotlin binaries only //NOTE: currently we use VirtualFileFinder to find Kotlin binaries only
if (KotlinClassFileHeader.readKotlinHeaderFromClassFile(vFile).getType() != null) { if (KotlinClassFileHeader.readKotlinHeaderFromClassFile(vFile) != null) {
return vFile; return vFile;
} }
} }
@@ -126,8 +126,11 @@ public final class DeserializedDescriptorResolver {
@Nullable @Nullable
private String[] readData(@NotNull VirtualFile virtualFile) { private String[] readData(@NotNull VirtualFile virtualFile) {
KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(virtualFile); KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(virtualFile);
if (header == null) {
return null;
}
int version = header.getVersion(); int version = header.getVersion();
if (!isAbiVersionCompatible(version) && header.getType() != null) { if (!isAbiVersionCompatible(version)) {
errorReporter.reportIncompatibleAbiVersion(header.getFqName(), virtualFile, version); errorReporter.reportIncompatibleAbiVersion(header.getFqName(), virtualFile, version);
return null; return null;
} }
@@ -23,12 +23,19 @@ import static org.jetbrains.jet.lang.resolve.java.AbiVersionUtil.isAbiVersionCom
public final class KotlinClassFileHeader { public final class KotlinClassFileHeader {
private static final Logger LOG = Logger.getInstance(KotlinClassFileHeader.class); private static final Logger LOG = Logger.getInstance(KotlinClassFileHeader.class);
@NotNull @Nullable
public static KotlinClassFileHeader readKotlinHeaderFromClassFile(@NotNull VirtualFile virtualFile) { public static KotlinClassFileHeader readKotlinHeaderFromClassFile(@NotNull VirtualFile virtualFile) {
try { try {
ClassReader reader = new ClassReader(virtualFile.contentsToByteArray()); ClassReader reader = new ClassReader(virtualFile.contentsToByteArray());
ReadDataFromAnnotationVisitor visitor = new ReadDataFromAnnotationVisitor(); ReadDataFromAnnotationVisitor visitor = new ReadDataFromAnnotationVisitor();
reader.accept(visitor, SKIP_CODE | SKIP_FRAMES | SKIP_DEBUG); reader.accept(visitor, SKIP_CODE | SKIP_FRAMES | SKIP_DEBUG);
if (visitor.type == null) {
return null;
}
if (visitor.fqName == null) {
LOG.error("File doesn't have a class name: " + virtualFile);
return null;
}
return new KotlinClassFileHeader(visitor.version, visitor.annotationData, visitor.type, visitor.fqName); return new KotlinClassFileHeader(visitor.version, visitor.annotationData, visitor.type, visitor.fqName);
} }
catch (IOException e) { catch (IOException e) {
@@ -70,14 +77,11 @@ public final class KotlinClassFileHeader {
} }
private final int version; private final int version;
@Nullable
private final String[] annotationData; private final String[] annotationData;
@Nullable
private final HeaderType type; private final HeaderType type;
@Nullable
private final FqName fqName; private final FqName fqName;
private KotlinClassFileHeader(int version, @Nullable String[] annotationData, @Nullable HeaderType type, @Nullable FqName fqName) { private KotlinClassFileHeader(int version, @Nullable String[] annotationData, @NotNull HeaderType type, @NotNull FqName fqName) {
this.version = version; this.version = version;
this.annotationData = annotationData; this.annotationData = annotationData;
this.type = type; this.type = type;
@@ -89,15 +93,17 @@ public final class KotlinClassFileHeader {
} }
@Nullable @Nullable
public HeaderType getType() { public String[] getAnnotationData() {
return type; if (isCompatibleKotlinCompiledFile() && annotationData == null) {
LOG.error("Kotlin annotation " + type + " is incorrect for class: " + fqName);
return null;
}
return annotationData;
} }
/** @NotNull
* @return true if this is a header for compiled Kotlin file with correct abi version which can be processed by compiler or the IDE public HeaderType getType() {
*/ return type;
public boolean isCompatibleKotlinCompiledFile() {
return type != null && type.isValidAnnotation() && isAbiVersionCompatible(version);
} }
/** /**
@@ -105,16 +111,14 @@ public final class KotlinClassFileHeader {
*/ */
@NotNull @NotNull
public FqName getFqName() { public FqName getFqName() {
assert fqName != null;
return fqName; return fqName;
} }
@Nullable /**
public String[] getAnnotationData() { * @return true if this is a header for compiled Kotlin file with correct abi version which can be processed by compiler or the IDE
if (annotationData == null && type != null) { */
LOG.error("Data for annotations " + type.correspondingAnnotation + " was not read."); public boolean isCompatibleKotlinCompiledFile() {
} return type.isValidAnnotation() && isAbiVersionCompatible(version);
return annotationData;
} }
private static class ReadDataFromAnnotationVisitor extends ClassVisitor { private static class ReadDataFromAnnotationVisitor extends ClassVisitor {
@@ -102,17 +102,31 @@ public class JetFromJavaDescriptorHelper {
@Nullable @Nullable
private static ClassData getClassData(@NotNull PsiClass psiClass) { private static ClassData getClassData(@NotNull PsiClass psiClass) {
VirtualFile virtualFile = getVirtualFileForPsiClass(psiClass); VirtualFile virtualFile = getVirtualFileForPsiClass(psiClass);
if (virtualFile == null) return null; if (virtualFile != null) {
String[] data = KotlinClassFileHeader.readKotlinHeaderFromClassFile(virtualFile).getAnnotationData(); KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(virtualFile);
return data == null ? null : JavaProtoBufUtil.readClassDataFrom(data); if (header != null) {
String[] data = header.getAnnotationData();
if (data != null) {
return JavaProtoBufUtil.readClassDataFrom(data);
}
}
}
return null;
} }
@Nullable @Nullable
private static PackageData getPackageData(@NotNull PsiClass psiClass) { private static PackageData getPackageData(@NotNull PsiClass psiClass) {
VirtualFile virtualFile = getVirtualFileForPsiClass(psiClass); VirtualFile virtualFile = getVirtualFileForPsiClass(psiClass);
if (virtualFile == null) return null; if (virtualFile != null) {
String[] data = KotlinClassFileHeader.readKotlinHeaderFromClassFile(virtualFile).getAnnotationData(); KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(virtualFile);
return data == null ? null : JavaProtoBufUtil.readPackageDataFrom(data); if (header != null) {
String[] data = header.getAnnotationData();
if (data != null) {
return JavaProtoBufUtil.readPackageDataFrom(data);
}
}
}
return null;
} }
//TODO: common utility //TODO: common utility
@@ -60,7 +60,10 @@ public final class DecompiledDataFactory {
InjectorForJavaDescriptorResolver injector = InjectorForJavaDescriptorResolver injector =
new InjectorForJavaDescriptorResolver(project, new BindingTraceContext()); new InjectorForJavaDescriptorResolver(project, new BindingTraceContext());
this.javaDescriptorResolver = injector.getJavaDescriptorResolver(); this.javaDescriptorResolver = injector.getJavaDescriptorResolver();
this.kotlinClassFileHeader = KotlinClassFileHeader.readKotlinHeaderFromClassFile(classFile);
KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(classFile);
assert header != null : "Decompiled data factory shouldn't be called on an unsupported file: " + classFile;
this.kotlinClassFileHeader = header;
} }
@NotNull @NotNull
@@ -28,7 +28,8 @@ public final class DecompiledUtils {
return false; return false;
} }
//TODO: check index //TODO: check index
return KotlinClassFileHeader.readKotlinHeaderFromClassFile(file).isCompatibleKotlinCompiledFile(); KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(file);
return header != null && header.isCompatibleKotlinCompiledFile();
} }
private DecompiledUtils() { private DecompiledUtils() {
@@ -58,7 +58,7 @@ public final class KotlinClassFileIndex extends ScalarIndexExtension<FqName> {
public Map<FqName, Void> map(FileContent inputData) { public Map<FqName, Void> map(FileContent inputData) {
try { try {
KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(inputData.getFile()); KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(inputData.getFile());
if (header.isCompatibleKotlinCompiledFile()) { if (header != null && header.isCompatibleKotlinCompiledFile()) {
return Collections.singletonMap(header.getFqName(), null); return Collections.singletonMap(header.getFqName(), null);
} }
} }