ExternalVendorCode/corda
1
0
Fork 0
mirror of https://github.com/corda/corda.git synced 2025-01-29 15:43:55 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add system property to disable public key caching (#7438)

Browse Source
This commit is contained in:
Rick Parker 2023-07-21 08:56:22 +01:00 committed by GitHub
parent 7d1d2297e7
commit 6ec8855c6e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
core/src/main/kotlin/net/corda/core/crypto/internal/PublicKeyCache.kt
View File

@ -7,6 +7,8 @@ import java.security.PublicKey
import java.util.concurrent.ConcurrentHashMap
object PublicKeyCache {
private val DISABLE = java.lang.Boolean.getBoolean("net.corda.core.pubkeycache.disable")
private val collectedWeakPubKeys = ReferenceQueue<PublicKey>()
private class WeakPubKey(key: PublicKey, val bytes: ByteSequence? = null) : WeakReference<PublicKey>(key, collectedWeakPubKeys) {
@ -14,8 +16,8 @@ object PublicKeyCache {
override fun hashCode(): Int = hashCode
override fun equals(other: Any?): Boolean {
if(this === other) return true
if(other !is WeakPubKey) return false
if (this === other) return true
if (other !is WeakPubKey) return false
if(this.hashCode != other.hashCode) return false
val thisGet = this.get()
val otherGet = other.get()
@ -36,15 +38,18 @@ object PublicKeyCache {
}
fun bytesForCachedPublicKey(key: PublicKey): ByteSequence? {
if (DISABLE) return null
val weakPubKey = WeakPubKey(key)
return pubKeyToBytes[weakPubKey]
}
fun publicKeyForCachedBytes(bytes: ByteSequence): PublicKey? {
if (DISABLE) return null
return bytesToPubKey[bytes]?.get()
}
fun cachePublicKey(key: PublicKey): PublicKey {
if (DISABLE) return key
reapCollectedWeakPubKeys()
val weakPubKey = WeakPubKey(key, ByteSequence.of(key.encoded))
pubKeyToBytes.putIfAbsent(weakPubKey, weakPubKey.bytes!!)