Make sure zip filesystems are closed on exit.

extracted/konan.serializer/src/org/jetbrains/kotlin/konan/library/KonanLibraryUtils.kt

@@ -1,10 +1,12 @@
 package org.jetbrains.kotlin.konan.library
 
 import org.jetbrains.kotlin.konan.file.File
+import org.jetbrains.kotlin.konan.file.file
+import org.jetbrains.kotlin.konan.file.withMutableZipFileSystem
+import org.jetbrains.kotlin.konan.file.withZipFileSystem
 import org.jetbrains.kotlin.konan.library.impl.DefaultMetadataReaderImpl
 import org.jetbrains.kotlin.konan.library.impl.KonanLibraryImpl
 import org.jetbrains.kotlin.konan.library.impl.zippedKonanLibraryChecks
-import org.jetbrains.kotlin.konan.library.impl.zippedKonanLibraryRoot
 import org.jetbrains.kotlin.konan.library.resolver.KonanLibraryResolver
 import org.jetbrains.kotlin.konan.library.resolver.impl.KonanLibraryResolverImpl
 import org.jetbrains.kotlin.konan.target.KonanTarget
@@ -21,7 +23,9 @@ fun File.unpackZippedKonanLibraryTo(newDir: File) {
             newDir.delete()
     }
 
-    zippedKonanLibraryRoot(this).recursiveCopyTo(newDir)
+    this.withMutableZipFileSystem {
+        it.file("/").recursiveCopyTo(newDir)
+    }
     check(newDir.exists) { "Could not unpack $this as $newDir." }
 }
 

extracted/konan.serializer/src/org/jetbrains/kotlin/konan/library/impl/KonanLibraryImpl.kt

@@ -1,10 +1,7 @@
 package org.jetbrains.kotlin.konan.library.impl
 
 import org.jetbrains.kotlin.konan.file.File
-import org.jetbrains.kotlin.konan.library.KLIB_PROPERTY_ABI_VERSION
-import org.jetbrains.kotlin.konan.library.KLIB_PROPERTY_LINKED_OPTS
-import org.jetbrains.kotlin.konan.library.KonanLibrary
-import org.jetbrains.kotlin.konan.library.MetadataReader
+import org.jetbrains.kotlin.konan.library.*
 import org.jetbrains.kotlin.konan.properties.Properties
 import org.jetbrains.kotlin.konan.properties.loadProperties
 import org.jetbrains.kotlin.konan.properties.propertyList
@@ -24,14 +21,13 @@ internal class KonanLibraryImpl(
     // whereas realFiles extracts them to /tmp.
     // For unzipped libraries inPlace and realFiles are the same
     // providing files in the library directory.
-    private val inPlace = createKonanLibraryLayout(libraryFile, target)
-    private val realFiles = inPlace.realFiles
 
-    override val libraryName
-        get() = inPlace.libraryName
+    private val layout = createKonanLibraryLayout(libraryFile, target)
+
+    override val libraryName: String by lazy { layout.inPlace { it.libraryName } }
 
     override val manifestProperties: Properties by lazy {
-        val properties = inPlace.manifestFile.loadProperties()
+        val properties = layout.inPlace { it.manifestFile.loadProperties() }
         if (target != null) substitute(properties, defaultTargetSubstitutions(target))
         properties
     }
@@ -49,18 +45,18 @@ internal class KonanLibraryImpl(
         get() = manifestProperties.propertyList(KLIB_PROPERTY_LINKED_OPTS, target!!.visibleName)
 
     override val bitcodePaths: List<String>
-        get() = (realFiles.kotlinDir.listFilesOrEmpty + realFiles.nativeDir.listFilesOrEmpty).map { it.absolutePath }
+        get() = layout.realFiles { (it.kotlinDir.listFilesOrEmpty + it.nativeDir.listFilesOrEmpty).map { it.absolutePath } }
 
     override val includedPaths: List<String>
-        get() = realFiles.includedDir.listFilesOrEmpty.map { it.absolutePath }
+        get() = layout.realFiles { it.includedDir.listFilesOrEmpty.map { it.absolutePath } }
 
-    override val targetList by lazy { inPlace.targetsDir.listFiles.map { it.name } }
+    override val targetList by lazy { layout.inPlace { it.targetsDir.listFiles.map { it.name } } }
 
-    override val dataFlowGraph by lazy { inPlace.dataFlowGraphFile.let { if (it.exists) it.readBytes() else null } }
+    override val dataFlowGraph by lazy { layout.inPlace { it.dataFlowGraphFile.let { if (it.exists) it.readBytes() else null } } }
 
-    override val moduleHeaderData: ByteArray by lazy { metadataReader.loadSerializedModule(inPlace) }
+    override val moduleHeaderData: ByteArray by lazy { layout.inPlace { metadataReader.loadSerializedModule(it) } }
 
-    override fun packageMetadata(fqName: String) = metadataReader.loadSerializedPackageFragment(inPlace, fqName)
+    override fun packageMetadata(fqName: String) = layout.inPlace { metadataReader.loadSerializedPackageFragment(it, fqName) }
 
     override fun toString() = "$libraryName[default=$isDefault]"
-}
+}

extracted/konan.serializer/src/org/jetbrains/kotlin/konan/library/impl/KonanLibraryLayoutImpl.kt

@@ -1,22 +1,35 @@
 package org.jetbrains.kotlin.konan.library.impl
 
-import org.jetbrains.kotlin.konan.file.File
-import org.jetbrains.kotlin.konan.file.asZipRoot
-import org.jetbrains.kotlin.konan.file.createTempDir
-import org.jetbrains.kotlin.konan.file.createTempFile
+import org.jetbrains.kotlin.konan.file.*
 import org.jetbrains.kotlin.konan.library.KLIB_FILE_EXTENSION
 import org.jetbrains.kotlin.konan.library.KLIB_FILE_EXTENSION_WITH_DOT
 import org.jetbrains.kotlin.konan.library.KonanLibraryLayout
 import org.jetbrains.kotlin.konan.target.KonanTarget
 import org.jetbrains.kotlin.konan.util.removeSuffixIfPresent
+import java.nio.file.FileSystem
 
-private class ZippedKonanLibraryLayout(val klibFile: File, override val target: KonanTarget?): KonanLibraryLayout {
+interface KonanLibraryLayoutImpl: KonanLibraryLayout {
+    fun <T> inPlace(action: (KonanLibraryLayout) -> T): T
+    fun <T> realFiles (action: (KonanLibraryLayout) -> T): T
+}
+
+private class ZippedKonanLibraryLayout(val klibFile: File, override val target: KonanTarget?): KonanLibraryLayoutImpl {
 
     init { zippedKonanLibraryChecks(klibFile) }
 
     override val libraryName = klibFile.path.removeSuffixIfPresent(KLIB_FILE_EXTENSION_WITH_DOT)
 
-    override val libDir by lazy { zippedKonanLibraryRoot(klibFile) }
+    override val libDir: File= File("/")
+
+    override fun <T> realFiles(action: (KonanLibraryLayout) -> T): T {
+        return action(FileExtractor(this))!!
+    }
+
+    override fun <T> inPlace(action: (KonanLibraryLayout) -> T): T {
+        return klibFile.withZipFileSystem { zipFileSystem ->
+            action(DirectFromZip(this, zipFileSystem))
+        }
+    }
 }
 
 internal fun zippedKonanLibraryChecks(klibFile: File) {
@@ -27,17 +40,23 @@ internal fun zippedKonanLibraryChecks(klibFile: File) {
     check(extension.isEmpty() || extension == KLIB_FILE_EXTENSION) { "Unexpected file extension: $extension" }
 }
 
-internal fun zippedKonanLibraryRoot(klibFile: File) = klibFile.asZipRoot
-
-private class UnzippedKonanLibraryLayout(override val libDir: File, override val target: KonanTarget?): KonanLibraryLayout {
+private class UnzippedKonanLibraryLayout(override val libDir: File, override val target: KonanTarget?): KonanLibraryLayoutImpl {
     override val libraryName = libDir.path
+
+    override fun <T> inPlace(action: (KonanLibraryLayout) -> T): T = action(this)
+    override fun <T> realFiles (action: (KonanLibraryLayout) -> T): T = inPlace(action)
+}
+
+private class DirectFromZip(zippedLayout: ZippedKonanLibraryLayout, val zipFileSystem: FileSystem): KonanLibraryLayout {
+    override val libraryName = zippedLayout.libraryName
+    override val libDir = zipFileSystem.file(zippedLayout.libDir)
 }
 
 /**
  * This class automatically extracts pieces of the library on first access. Use it if you need
- * to pass extracted files to an external tool. Otherwise, stick to [ZippedKonanLibraryLayout].
+ * to pass extracted files to an external tool. Otherwise, stick to [DirectFromZip].
  */
-private class FileExtractor(zippedLibraryLayout: KonanLibraryLayout): KonanLibraryLayout by zippedLibraryLayout {
+private class FileExtractor(val zippedLibraryLayout: ZippedKonanLibraryLayout): KonanLibraryLayout by zippedLibraryLayout {
 
     override val manifestFile: File by lazy { extract(super.manifestFile) }
 
@@ -51,29 +70,20 @@ private class FileExtractor(zippedLibraryLayout: KonanLibraryLayout): KonanLibra
 
     override val linkdataDir: File by lazy { extractDir(super.linkdataDir) }
 
-    fun extract(file: File): File {
+    fun extract(file: File): File = zippedLibraryLayout.klibFile.withZipFileSystem { zipFileSystem ->
         val temporary = createTempFile(file.name)
-        file.copyTo(temporary)
+        zipFileSystem.file(file).copyTo(temporary)
         temporary.deleteOnExit()
-        return temporary
+        temporary
     }
 
-    fun extractDir(directory: File): File {
+    fun extractDir(directory: File): File = zippedLibraryLayout.klibFile.withZipFileSystem { zipFileSystem ->
         val temporary = createTempDir(directory.name)
-        directory.recursiveCopyTo(temporary)
+        zipFileSystem.file(directory).recursiveCopyTo(temporary)
         temporary.deleteOnExitRecursively()
-        return temporary
+        temporary
     }
 }
 
 internal fun createKonanLibraryLayout(klib: File, target: KonanTarget? = null) =
         if (klib.isFile) ZippedKonanLibraryLayout(klib, target) else UnzippedKonanLibraryLayout(klib, target)
-
-internal val KonanLibraryLayout.realFiles
-    get() = when (this) {
-        is ZippedKonanLibraryLayout -> FileExtractor(this)
-        // Unpacked library just provides its own files.
-        is UnzippedKonanLibraryLayout -> this
-        else -> error("Provide an extractor for your container.")
-    }
-