com.ackbox.totp.ReseedingSecureRandom.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ackbox-totp Show documentation

TOTP authenticator written in Kotlin

There is a newer version: 1.1.0

package com.ackbox.totp

import java.security.SecureRandom
import java.util.concurrent.atomic.AtomicInteger

class ReseedingSecureRandom {

    private val count = AtomicInteger(0)
    private var secureRandom = createSecureRandom()

    fun nextBytes(bytes: ByteArray) {
        if (count.incrementAndGet() > MAX_OPERATIONS) {
            synchronized(this) {
                if (count.get() > MAX_OPERATIONS) {
                    secureRandom = createSecureRandom()
                    count.set(0)
                }
            }
        }
        secureRandom.nextBytes(bytes)
    }

    private fun createSecureRandom() = SecureRandom()

    companion object {

        private val MAX_OPERATIONS = 1000000
    }
}