135 lines
4.4 KiB
Kotlin
135 lines
4.4 KiB
Kotlin
package dev.beefers.vendetta.manager.installer.util
|
|
|
|
import android.content.Context
|
|
import com.android.apksig.ApkSigner
|
|
import dev.beefers.vendetta.manager.utils.Constants
|
|
import org.bouncycastle.asn1.x500.X500Name
|
|
import org.bouncycastle.asn1.x509.SubjectPublicKeyInfo
|
|
import org.bouncycastle.cert.X509v3CertificateBuilder
|
|
import org.bouncycastle.cert.jcajce.JcaX509CertificateConverter
|
|
import org.bouncycastle.operator.jcajce.JcaContentSignerBuilder
|
|
import org.koin.core.component.KoinComponent
|
|
import org.koin.core.component.inject
|
|
import java.io.File
|
|
import java.math.BigInteger
|
|
import java.nio.file.Path
|
|
import java.security.KeyPairGenerator
|
|
import java.security.KeyStore
|
|
import java.security.PrivateKey
|
|
import java.security.SecureRandom
|
|
import java.security.cert.Certificate
|
|
import java.security.cert.X509Certificate
|
|
import java.util.Date
|
|
import java.util.Locale
|
|
import kotlin.io.path.Path
|
|
import kotlin.io.path.moveTo
|
|
|
|
object Signer : KoinComponent {
|
|
private val password = "password".toCharArray()
|
|
private val context by inject<Context>()
|
|
private val cacheDir = context.cacheDir
|
|
private val filesDir = context.filesDir
|
|
|
|
val keyStore: File by lazy {
|
|
val ks = filesDir.resolve("ks.keystore")
|
|
migrate(cacheDir, filesDir)
|
|
migrate(Constants.VENDETTA_DIR, filesDir)
|
|
ks.also {
|
|
if (!it.exists()) {
|
|
it.createNewFile()
|
|
newKeystore(it)
|
|
}
|
|
}
|
|
ks
|
|
}
|
|
|
|
private val signerConfig: ApkSigner.SignerConfig by lazy {
|
|
val keyStore = KeyStore.getInstance(KeyStore.getDefaultType())
|
|
|
|
this.keyStore.inputStream().use { stream ->
|
|
keyStore.load(stream, null)
|
|
}
|
|
|
|
val alias = keyStore.aliases().nextElement()
|
|
val certificate = keyStore.getCertificate(alias) as X509Certificate
|
|
|
|
ApkSigner.SignerConfig.Builder(
|
|
"Vendetta",
|
|
keyStore.getKey(alias, password) as PrivateKey,
|
|
listOf(certificate)
|
|
).build()
|
|
}
|
|
|
|
private fun newKeystore(out: File) {
|
|
val key = createKey()
|
|
|
|
with(KeyStore.getInstance(KeyStore.getDefaultType())) {
|
|
load(null, password)
|
|
setKeyEntry("alias", key.privateKey, password, arrayOf<Certificate>(key.publicKey))
|
|
store(out.outputStream(), password)
|
|
}
|
|
}
|
|
|
|
private fun createKey(): KeySet {
|
|
var serialNumber: BigInteger
|
|
|
|
do serialNumber = SecureRandom().nextInt().toBigInteger()
|
|
while (serialNumber < BigInteger.ZERO)
|
|
|
|
val x500Name = X500Name("CN=Vendetta Manager")
|
|
val pair = KeyPairGenerator.getInstance("RSA").run {
|
|
initialize(2048)
|
|
generateKeyPair()
|
|
}
|
|
val builder = X509v3CertificateBuilder(
|
|
/* issuer = */ x500Name,
|
|
/* serial = */
|
|
serialNumber,
|
|
/* notBefore = */
|
|
Date(System.currentTimeMillis() - 1000L * 60L * 60L * 24L * 30L),
|
|
/* notAfter = */
|
|
Date(System.currentTimeMillis() + 1000L * 60L * 60L * 24L * 366L * 30L),
|
|
/* dateLocale = */
|
|
Locale.ENGLISH,
|
|
/* subject = */
|
|
x500Name,
|
|
/* publicKeyInfo = */
|
|
SubjectPublicKeyInfo.getInstance(pair.public.encoded)
|
|
)
|
|
val signer = JcaContentSignerBuilder("SHA1withRSA").build(pair.private)
|
|
|
|
return KeySet(
|
|
JcaX509CertificateConverter().getCertificate(builder.build(signer)),
|
|
pair.private
|
|
)
|
|
}
|
|
|
|
fun signApk(apkFile: File, output: File) {
|
|
val outputApk = cacheDir.resolve(apkFile.name)
|
|
|
|
ApkSigner.Builder(listOf(signerConfig))
|
|
.setV1SigningEnabled(false) // TODO: enable so api <24 devices can work, however zip-alignment breaks
|
|
.setV2SigningEnabled(true)
|
|
.setV3SigningEnabled(true)
|
|
.setInputApk(apkFile)
|
|
.setOutputApk(output)
|
|
.build()
|
|
.sign()
|
|
|
|
outputApk.renameTo(apkFile)
|
|
}
|
|
|
|
private fun migrate(oldDir: File, newDir: File) {
|
|
oldDir.resolve("ks.keystore").also {
|
|
if (it.exists()) {
|
|
Path(it.absolutePath)
|
|
.moveTo(
|
|
Path(newDir.resolve("ks.keystore").absolutePath),
|
|
overwrite = true
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
private class KeySet(val publicKey: X509Certificate, val privateKey: PrivateKey)
|
|
} |