mirror of
https://github.com/ossrs/srs.git
synced 2025-03-09 15:49:59 +00:00
30 lines
685 B
Go
30 lines
685 B
Go
package randutil
|
|
|
|
import (
|
|
crand "crypto/rand"
|
|
"encoding/binary"
|
|
"math/big"
|
|
)
|
|
|
|
// GenerateCryptoRandomString generates a random string for cryptographic usage.
|
|
func GenerateCryptoRandomString(n int, runes string) (string, error) {
|
|
letters := []rune(runes)
|
|
b := make([]rune, n)
|
|
for i := range b {
|
|
v, err := crand.Int(crand.Reader, big.NewInt(int64(len(letters))))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
b[i] = letters[v.Int64()]
|
|
}
|
|
return string(b), nil
|
|
}
|
|
|
|
// CryptoUint64 returns cryptographic random uint64.
|
|
func CryptoUint64() (uint64, error) {
|
|
var v uint64
|
|
if err := binary.Read(crand.Reader, binary.LittleEndian, &v); err != nil {
|
|
return 0, err
|
|
}
|
|
return v, nil
|
|
}
|