fix: ClassLoader not working with Java 9+

This commit is contained in:
danthe1st 2022-05-07 21:05:09 +02:00
parent 7a6025a278
commit 3a11e1135b
No known key found for this signature in database
GPG Key ID: 2F8074F370864D56
6 changed files with 26 additions and 37 deletions

View File

@ -1,6 +1,5 @@
package app.revanced.cli package app.revanced.cli
import app.revanced.patch.PatchLoader
import app.revanced.patch.Patches import app.revanced.patch.Patches
import app.revanced.utils.adb.Adb import app.revanced.utils.adb.Adb
import picocli.CommandLine.* import picocli.CommandLine.*
@ -49,8 +48,7 @@ internal object MainCommand : Runnable {
override fun run() { override fun run() {
if (listOnly) { if (listOnly) {
patchBundles.forEach { patchBundles.forEach {
PatchLoader.injectPatches(it) Patches.load(it).forEach {
Patches.loadPatches().forEach {
println(it().metadata) println(it().metadata)
} }
} }

View File

@ -1,6 +1,5 @@
package app.revanced.cli package app.revanced.cli
import app.revanced.patch.PatchLoader
import app.revanced.patch.Patches import app.revanced.patch.Patches
import app.revanced.patcher.data.base.Data import app.revanced.patcher.data.base.Data
import app.revanced.patcher.patch.base.Patch import app.revanced.patcher.patch.base.Patch
@ -59,9 +58,8 @@ internal class Patcher {
val checkInclude = MainCommand.includedPatches.isNotEmpty() val checkInclude = MainCommand.includedPatches.isNotEmpty()
MainCommand.patchBundles.forEach { bundle -> MainCommand.patchBundles.forEach { bundle ->
PatchLoader.injectPatches(bundle)
val includedPatches = mutableListOf<Patch<Data>>() val includedPatches = mutableListOf<Patch<Data>>()
Patches.loadPatches().forEach patch@{ Patches.load(bundle).forEach patch@{
val patch = it() val patch = it()
// TODO: filter out incompatible patches with package metadata // TODO: filter out incompatible patches with package metadata

View File

@ -1,25 +1,10 @@
package app.revanced.patch package app.revanced.patch
import java.io.File import java.io.File
import java.net.URL
import java.net.URLClassLoader import java.net.URLClassLoader
internal class PatchLoader { internal class PatchLoader {
internal companion object { internal companion object {
internal fun injectPatches(file: File) {
// This function will fail on Java 9 and above.
try {
val url = file.toURI().toURL()
val classLoader = Thread.currentThread().contextClassLoader as URLClassLoader
val method = URLClassLoader::class.java.getDeclaredMethod("addURL", URL::class.java)
method.isAccessible = true
method.invoke(classLoader, url)
} catch (e: Exception) {
throw Exception(
"Failed to inject patches! The CLI does NOT work on Java 9 and above, please use Java 8!",
e // propagate exception
)
}
}
} }
} }

View File

@ -1,15 +1,26 @@
package app.revanced.patch package app.revanced.patch
import app.revanced.patcher.data.base.Data
import app.revanced.patcher.patch.base.Patch
import app.revanced.patches.Index import app.revanced.patches.Index
import java.io.File
import java.net.URLClassLoader
internal class Patches { internal object Patches {
internal companion object {
// You may ask yourself, "why do this?".
// We do it like this, because we don't want the Index class /**
// to be loaded while the dependency hasn't been injected yet. * This method loads patches from a given patch file
// You can see this as "controlled class loading". * @return the loaded patches represented as a list of functions returning instances of [Patch]
// Whenever this class is loaded (because it is invoked), all the imports */
// will be loaded too. We don't want to do this until we've injected the class. internal fun load(patchFile: File): List<() -> Patch<Data>> {
internal fun loadPatches() = Index.patches val url = patchFile.toURI().toURL()
val classLoader = URLClassLoader(arrayOf(url))
return loadIndex(classLoader).patches
} }
} private fun loadIndex(classLoader: ClassLoader) = classLoader
.loadClass(Index::class.java.canonicalName)
.fields
.first()
.get(null) as Index
}

View File

@ -12,10 +12,7 @@ internal class FileSystemUtils(
private var fileSystem: FileSystem private var fileSystem: FileSystem
init { init {
fileSystem = FileSystems.newFileSystem( fileSystem = FileSystems.newFileSystem(file.toPath(), null as ClassLoader?)
file.toPath(),
null
)
} }
private fun deleteDirectory(dirPath: String) { private fun deleteDirectory(dirPath: String) {

View File

@ -91,7 +91,7 @@ object Signer {
(keyStore.getKey(alias, PASSWORD) as PrivateKey) (keyStore.getKey(alias, PASSWORD) as PrivateKey)
) )
val zip = FileSystems.newFileSystem(apkFile.toPath(), null) val zip = FileSystems.newFileSystem(apkFile.toPath(), null as ClassLoader?)
val dig = MessageDigest.getInstance("SHA1") val dig = MessageDigest.getInstance("SHA1")
val digests: MutableMap<String, String> = LinkedHashMap() val digests: MutableMap<String, String> = LinkedHashMap()