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:
@@ -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;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+4
-1
@@ -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
-19
@@ -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);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user