Infendro/prng.kt
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Wiki Activity

implement optimizations

Browse Source
This commit is contained in:
Infendro
2026-01-30 13:41:43 +01:00
parent 7144c21074
commit 0f890c923c
10 changed files with 99 additions and 168 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
README.md
View File

@@ -6,16 +6,7 @@ Supported [PRNG](https://en.wikipedia.org/wiki/Pseudorandom_number_generator) al
- LCG
Supported [CSPRNG](https://en.wikipedia.org/wiki/Cryptographically_secure_pseudorandom_number_generator) algorithms:
- HMAC DRBG
## Targets
| Target | Status |
|------------------|---------------|
| JVM | Supported |
| JavaScript | Supported |
| Native (Linux) | Supported |
| Native (Windows) | Not Supported |
- HMAC-DRBG
## Installation
@@ -27,34 +18,30 @@ repositories {
}
dependencies {
implementation("com.infendro:prng:1.2.0")
implementation("com.infendro:prng:1.3.0")
}
```
## Usage
```kotlin
import com.infendro.hash.sha2.`SHA-256`
import com.infendro.random.csprng.CSPRNG
fun main() {
// create a PRNG instance
// create PRNG instance
val rng = CSPRNG.HMAC(`SHA-256`)
// seed the PRNG to receive predictable results
rng.seed(/* some seed */)
// seed PRNG to receive deterministic numbers
rng.seed(/* seed */)
val a = rng.nextInt()
val b = rng.nextInt()
val c = rng.nextInt()
// reseed the PRNG to add entropy to the following results
rng.reseed(/* some random entropy */)
// reseed PRNG to add additional entropy
rng.reseed(/* entropy */)
val d = rng.nextInt()
val e = rng.nextInt()
val f = rng.nextInt()
// each run:
// a, b, and c will be predictable
// d, e, and f will be random (assuming the entropy is also random)
// a, b, and c will be deterministic
// d, e, and f will be random (assuming entropy is random)
}
```

21
build.gradle.kts
View File

@@ -1,8 +1,11 @@
@file:OptIn(ExperimentalWasmDsl::class)
import com.infendro.plugin.infendro
import com.infendro.plugin.token
import org.jetbrains.kotlin.gradle.ExperimentalWasmDsl
group = "com.infendro"
version = "1.2.0"
version = "1.3.0"
plugins {
alias(libs.plugins.infendro)
@@ -13,9 +16,13 @@ plugins {
kotlin {
jvm()
linuxX64()
js {
nodejs()
}
linuxArm64()
mingwX64()
macosX64()
macosArm64()
js { nodejs() }
wasmJs { nodejs() }
wasmWasi { nodejs() }
repositories {
infendro()
@@ -26,13 +33,9 @@ kotlin {
commonMain.dependencies {
implementation(libs.hash)
implementation(libs.mac)
implementation(libs.bytearray)
implementation(libs.bytes)
}
}
compilerOptions {
freeCompilerArgs.add("-opt-in=kotlin.ExperimentalUnsignedTypes")
}
}
publishing.repositories {

14
gradle/libs.versions.toml
View File

@@ -1,15 +1,15 @@
[versions]
infendro = "1.0.0"
kotlin = "2.2.21"
hash = "1.3.2"
mac = "1.2.0"
bytearray = "1.2.4"
infendro = "1.1.0"
kotlin = "2.3.0"
mac = "1.3.0"
hash = "1.5.1"
bytes = "1.3.3"
[plugins]
infendro = { id = "com.infendro.plugin", version.ref = "infendro" }
multiplatform = { id = "org.jetbrains.kotlin.multiplatform", version.ref = "kotlin" }
[libraries]
hash = { module = "com.infendro:hash", version.ref = "hash" }
mac = { module = "com.infendro:mac", version.ref = "mac" }
bytearray = { module = "com.infendro:bytearray", version.ref = "bytearray" }
hash = { module = "com.infendro:hash", version.ref = "hash" }
bytes = { module = "com.infendro:bytes", version.ref = "bytes" }

5
src/commonMain/kotlin/com/infendro/random/CSPRNG.kt Normal file
View File

@@ -0,0 +1,5 @@
package com.infendro.random
abstract class CSPRNG : PRNG() {
abstract fun reseed(entropy: ByteArray)
}

16
src/commonMain/kotlin/com/infendro/random/PRNG.kt Normal file
View File

@@ -0,0 +1,16 @@
package com.infendro.random
import kotlin.random.Random
abstract class PRNG : Random() {
abstract fun seed(seed: ByteArray)
protected abstract fun generate(): Int
override fun nextBits(bitCount: Int): Int {
require(bitCount in 0..32)
if (bitCount == 0) return 0
return generate() ushr (32 - bitCount)
}
}

16
src/commonMain/kotlin/com/infendro/random/csprng/CSPRNG.kt
View File

@@ -1,16 +0,0 @@
package com.infendro.random.csprng
import com.infendro.hash.HashFunction
import com.infendro.random.prng.PRNG
abstract class CSPRNG : PRNG() {
abstract fun reseed(
entropy: UByteArray,
)
companion object {
fun HMAC(
function: HashFunction,
) = com.infendro.random.csprng.HMAC(function)
}
}

37
src/commonMain/kotlin/com/infendro/random/csprng/HMAC-DRBG.kt Normal file
View File

@@ -0,0 +1,37 @@
package com.infendro.random.csprng
import com.infendro.bytes.bytearray.toInt
import com.infendro.hash.HashFunction
import com.infendro.mac.HMAC
import com.infendro.random.CSPRNG
import kotlin.properties.Delegates.notNull
class `HMAC-DRBG` internal constructor(
private val function: HashFunction,
) : CSPRNG() {
private val hmac = HMAC(function)
private var k: ByteArray by notNull()
private var v: ByteArray by notNull()
override fun seed(seed: ByteArray) {
k = ByteArray(function.bytes) { 0x00 }
v = ByteArray(function.bytes) { 0x01 }
reseed(seed)
}
override fun reseed(entropy: ByteArray) {
k = hmac.hash(k, v + byteArrayOf(0x00) + entropy)
v = hmac.hash(k, v)
k = hmac.hash(k, v + byteArrayOf(0x01) + entropy)
v = hmac.hash(k, v)
}
override fun generate(): Int {
v = hmac.hash(k, v)
val result = v.toInt()
k = hmac.hash(k, v + byteArrayOf(0x00))
v = hmac.hash(k, v)
return result
}
}

47
src/commonMain/kotlin/com/infendro/random/csprng/HMAC.kt
View File

@@ -1,47 +0,0 @@
package com.infendro.random.csprng
import com.infendro.hash.HashFunction
import kotlin.math.min
import kotlin.properties.Delegates.notNull
class HMAC internal constructor(
private val function: HashFunction,
) : CSPRNG() {
private val hmac = com.infendro.mac.HMAC(function)
private var k: UByteArray by notNull()
private var v: UByteArray by notNull()
override fun seed(
seed: UByteArray,
) {
k = UByteArray(function.bytes) { 0x00U }
v = UByteArray(function.bytes) { 0x01U }
reseed(seed)
}
override fun reseed(
entropy: UByteArray,
) {
k = hmac.hash(k, v + ubyteArrayOf(0x00U) + entropy)
v = hmac.hash(k, v)
k = hmac.hash(k, v + ubyteArrayOf(0x01U) + entropy)
v = hmac.hash(k, v)
}
override fun generate(
count: Int,
): UByteArray {
val bytes = buildList {
while (size < count) {
v = hmac.hash(k, v)
addAll(v.take(min(count - size, function.bytes)))
}
}.toUByteArray()
k = hmac.hash(k, v + ubyteArrayOf(0x00U))
v = hmac.hash(k, v)
return bytes
}
}

36
src/commonMain/kotlin/com/infendro/random/prng/LCG.kt
View File

@@ -1,33 +1,23 @@
package com.infendro.random.prng
import com.infendro.bytearray.padStart
import com.infendro.bytearray.toUByteArray
import com.infendro.bytearray.toULong
import kotlin.math.min
import kotlin.properties.Delegates
import com.infendro.bytes.bytearray.toLong
import com.infendro.hash.sha2.`SHA-256`
import com.infendro.random.PRNG
import kotlin.properties.Delegates.notNull
class LCG internal constructor() : PRNG() {
private val a: ULong = 25214903917UL
private val c: ULong = 11UL
private val m: ULong = 1UL shl 48
private val a: Long = 25214903917L
private val c: Long = 11L
private val mask: Long = (1L shl 48) - 1
private var x: ULong by Delegates.notNull()
private var x: Long by notNull()
override fun seed(
seed: UByteArray,
) {
val s = seed.padStart(8, 0x00U).takeLast(8).toUByteArray()
x = s.toULong() % m
override fun seed(seed: ByteArray) {
x = `SHA-256`.hash(seed).toLong() and mask
}
override fun generate(
count: Int,
): UByteArray {
return buildList {
while (size < count) {
x = (a * x + c) % m
addAll(x.toUByteArray().takeLast(min(count - size, 6)))
}
}.toUByteArray()
override fun generate(): Int {
x = (a * x + c) and mask
return (x ushr 16).toInt()
}
}

44
src/commonMain/kotlin/com/infendro/random/prng/PRNG.kt
View File

@@ -1,44 +0,0 @@
package com.infendro.random.prng
import com.infendro.bytearray.toInt
import kotlin.random.Random
abstract class PRNG : Random() {
abstract fun seed(
seed: UByteArray,
)
abstract fun generate(
count: Int,
): UByteArray
override fun nextBits(
bitCount: Int,
): Int {
require(bitCount in 0..32)
if (bitCount == 0) return 0
return generate(4).toInt() ushr (32 - bitCount)
}
override fun nextBytes(
array: ByteArray,
fromIndex: Int,
toIndex: Int,
): ByteArray {
require(fromIndex in 0..array.size && toIndex in 0..array.size)
require(fromIndex <= toIndex)
val size = toIndex - fromIndex
val bytes = generate(size).asByteArray()
for (i in 0..<size) {
array[i + fromIndex] = bytes[i]
}
return array
}
companion object {
fun LCG() = com.infendro.random.prng.LCG()
}
}