SerializedDataHeader now always has valid annotationData

This helps to avoid errors of checking 'instanceof SerializedDataHeader' but
forgetting to check if it 'isCompatibleKotlinCompiledFile'
This commit is contained in:
Alexander Udalov
2013-10-01 18:35:49 +04:00
parent eebe605771
commit fc4e13f573
8 changed files with 27 additions and 47 deletions
@@ -112,13 +112,14 @@ 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 instanceof SerializedDataHeader && header.isCompatibleKotlinCompiledFile()) { if (header instanceof SerializedDataHeader) {
return ((SerializedDataHeader) header).getAnnotationData(); return ((SerializedDataHeader) header).getAnnotationData();
} }
if (header != null) { if (header != null) {
errorReporter.reportIncompatibleAbiVersion(header.getFqName(), virtualFile, header.getVersion()); errorReporter.reportIncompatibleAbiVersion(header.getFqName(), virtualFile, header.getVersion());
} }
return null; return null;
} }
} }
@@ -17,11 +17,10 @@
package org.jetbrains.jet.lang.resolve.kotlin.header; package org.jetbrains.jet.lang.resolve.kotlin.header;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.NotNull;
import org.jetbrains.jet.lang.resolve.java.AbiVersionUtil;
import org.jetbrains.jet.lang.resolve.name.FqName; import org.jetbrains.jet.lang.resolve.name.FqName;
public class OldAnnotationHeader extends KotlinClassFileHeader { public class IncompatibleAnnotationHeader extends KotlinClassFileHeader {
protected OldAnnotationHeader(@NotNull FqName fqName) { protected IncompatibleAnnotationHeader(int version, @NotNull FqName fqName) {
super(AbiVersionUtil.INVALID_VERSION, fqName); super(version, fqName);
} }
} }
@@ -25,7 +25,6 @@ import org.jetbrains.jet.lang.resolve.name.FqName;
import java.io.IOException; import java.io.IOException;
import static org.jetbrains.asm4.ClassReader.*; import static org.jetbrains.asm4.ClassReader.*;
import static org.jetbrains.jet.lang.resolve.java.AbiVersionUtil.isAbiVersionCompatible;
public abstract class KotlinClassFileHeader { public abstract class KotlinClassFileHeader {
@Nullable @Nullable
@@ -60,11 +59,4 @@ public abstract class KotlinClassFileHeader {
public FqName getFqName() { public FqName getFqName() {
return fqName; return fqName;
} }
/**
* @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 boolean isCompatibleKotlinCompiledFile() {
return isAbiVersionCompatible(version);
}
} }
@@ -84,19 +84,29 @@ import static org.jetbrains.jet.lang.resolve.java.AbiVersionUtil.isAbiVersionCom
return null; return null;
} }
if (!AbiVersionUtil.isAbiVersionCompatible(version)) {
return new IncompatibleAnnotationHeader(version, fqName);
}
switch (foundType) { switch (foundType) {
case CLASS: case CLASS:
return SerializedDataHeader.create(version, annotationData, SerializedDataHeader.Kind.CLASS, fqName); return serializedDataHeader(SerializedDataHeader.Kind.CLASS, fqName);
case PACKAGE: case PACKAGE:
return SerializedDataHeader.create(version, annotationData, SerializedDataHeader.Kind.PACKAGE, fqName); return serializedDataHeader(SerializedDataHeader.Kind.PACKAGE, fqName);
case OLD_CLASS:
case OLD_PACKAGE:
return new OldAnnotationHeader(fqName);
default: default:
throw new UnsupportedOperationException("Unknown HeaderType: " + foundType); throw new UnsupportedOperationException("Unknown compatible HeaderType: " + foundType);
} }
} }
@Nullable
private SerializedDataHeader serializedDataHeader(@NotNull SerializedDataHeader.Kind kind, @NotNull FqName fqName) {
if (annotationData == null) {
LOG.error("Kotlin annotation " + foundType + " is incorrect for class: " + fqName);
return null;
}
return new SerializedDataHeader(version, annotationData, kind, fqName);
}
@Override @Override
public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) { public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {
fqName = JvmClassName.byInternalName(name).getFqName(); fqName = JvmClassName.byInternalName(name).getFqName();
@@ -16,16 +16,10 @@
package org.jetbrains.jet.lang.resolve.kotlin.header; package org.jetbrains.jet.lang.resolve.kotlin.header;
import com.intellij.openapi.diagnostic.Logger;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.jetbrains.jet.lang.resolve.name.FqName; import org.jetbrains.jet.lang.resolve.name.FqName;
import static org.jetbrains.jet.lang.resolve.java.AbiVersionUtil.isAbiVersionCompatible;
public class SerializedDataHeader extends KotlinClassFileHeader { public class SerializedDataHeader extends KotlinClassFileHeader {
private static final Logger LOG = Logger.getInstance(SerializedDataHeader.class);
public enum Kind { public enum Kind {
CLASS, CLASS,
PACKAGE PACKAGE
@@ -34,22 +28,13 @@ public class SerializedDataHeader extends KotlinClassFileHeader {
private final String[] data; private final String[] data;
private final Kind kind; private final Kind kind;
private SerializedDataHeader(int version, @Nullable String[] annotationData, @NotNull Kind kind, @NotNull FqName fqName) { protected SerializedDataHeader(int version, @NotNull String[] annotationData, @NotNull Kind kind, @NotNull FqName fqName) {
super(version, fqName); super(version, fqName);
this.data = annotationData; this.data = annotationData;
this.kind = kind; this.kind = kind;
} }
@Nullable @NotNull
public static SerializedDataHeader create(int version, @Nullable String[] annotationData, @NotNull Kind kind, @NotNull FqName fqName) {
if (isAbiVersionCompatible(version) && annotationData == null) {
LOG.error("Kotlin annotation " + kind + " is incorrect for class: " + fqName);
return null;
}
return new SerializedDataHeader(version, annotationData, kind, fqName);
}
@Nullable
public String[] getAnnotationData() { public String[] getAnnotationData() {
return data; return data;
} }
@@ -106,10 +106,7 @@ public class JetFromJavaDescriptorHelper {
if (virtualFile != null) { if (virtualFile != null) {
KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(virtualFile); KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(virtualFile);
if (header instanceof SerializedDataHeader) { if (header instanceof SerializedDataHeader) {
String[] data = ((SerializedDataHeader) header).getAnnotationData(); return JavaProtoBufUtil.readClassDataFrom(((SerializedDataHeader) header).getAnnotationData());
if (data != null) {
return JavaProtoBufUtil.readClassDataFrom(data);
}
} }
} }
return null; return null;
@@ -121,10 +118,7 @@ public class JetFromJavaDescriptorHelper {
if (virtualFile != null) { if (virtualFile != null) {
KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(virtualFile); KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(virtualFile);
if (header instanceof SerializedDataHeader) { if (header instanceof SerializedDataHeader) {
String[] data = ((SerializedDataHeader) header).getAnnotationData(); return JavaProtoBufUtil.readPackageDataFrom(((SerializedDataHeader) header).getAnnotationData());
if (data != null) {
return JavaProtoBufUtil.readPackageDataFrom(data);
}
} }
} }
return null; return null;
@@ -29,8 +29,7 @@ public final class DecompiledUtils {
return false; return false;
} }
//TODO: check index //TODO: check index
KotlinClassFileHeader header = KotlinClassFileHeader.readKotlinHeaderFromClassFile(file); return KotlinClassFileHeader.readKotlinHeaderFromClassFile(file) instanceof SerializedDataHeader;
return header instanceof SerializedDataHeader && header.isCompatibleKotlinCompiledFile();
} }
private DecompiledUtils() { private DecompiledUtils() {
@@ -59,7 +59,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 instanceof SerializedDataHeader && header.isCompatibleKotlinCompiledFile()) { if (header instanceof SerializedDataHeader) {
return Collections.singletonMap(header.getFqName(), null); return Collections.singletonMap(header.getFqName(), null);
} }
} }