mirror of
https://github.com/octoleo/syncthing.git
synced 2024-11-06 05:17:49 +00:00
198028d627
rand.secureSource.Uint64 no longer allocates. rand.String uses a strings.Builder. Benchmark results on linux/amd64: name old time/op new time/op delta SecureSource-8 69.1ns ± 3% 51.7ns ± 3% -25.21% (p=0.000 n=20+10) String-8 2.66µs ± 2% 1.95µs ± 1% -26.61% (p=0.000 n=10+10) name old alloc/op new alloc/op delta SecureSource-8 8.00B ± 0% 0.00B -100.00% (p=0.000 n=20+10) String-8 288B ± 0% 32B ± 0% -88.89% (p=0.000 n=10+10) name old allocs/op new allocs/op delta SecureSource-8 1.00 ± 0% 0.00 -100.00% (p=0.000 n=20+10) String-8 33.0 ± 0% 1.0 ± 0% -96.97% (p=0.000 n=10+10)
76 lines
2.1 KiB
Go
76 lines
2.1 KiB
Go
// Copyright (C) 2014 The Syncthing Authors.
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
// You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
|
|
// Package rand implements functions similar to math/rand in the standard
|
|
// library, but on top of a secure random number generator.
|
|
package rand
|
|
|
|
import (
|
|
"io"
|
|
mathRand "math/rand"
|
|
"reflect"
|
|
"strings"
|
|
)
|
|
|
|
// Reader is the standard crypto/rand.Reader with added buffering.
|
|
var Reader = defaultSecureSource
|
|
|
|
func Read(p []byte) (int, error) {
|
|
return io.ReadFull(defaultSecureSource, p)
|
|
}
|
|
|
|
// randomCharset contains the characters that can make up a rand.String().
|
|
const randomCharset = "2345679abcdefghijkmnopqrstuvwxyzACDEFGHJKLMNPQRSTUVWXYZ"
|
|
|
|
var (
|
|
// defaultSecureSource is a concurrency-safe, cryptographically secure
|
|
// math/rand.Source.
|
|
defaultSecureSource = newSecureSource()
|
|
|
|
// defaultSecureRand is a math/rand.Rand based on the secure source.
|
|
defaultSecureRand = mathRand.New(defaultSecureSource)
|
|
)
|
|
|
|
// String returns a cryptographically secure random string of characters
|
|
// (taken from randomCharset) of the specified length. The returned string
|
|
// contains ~5.8 bits of entropy per character, due to the character set used.
|
|
func String(l int) string {
|
|
var sb strings.Builder
|
|
sb.Grow(l)
|
|
|
|
for i := 0; i < l; i++ {
|
|
sb.WriteByte(randomCharset[defaultSecureRand.Intn(len(randomCharset))])
|
|
}
|
|
return sb.String()
|
|
}
|
|
|
|
// Int63 returns a cryptographically secure random int63.
|
|
func Int63() int64 {
|
|
return defaultSecureSource.Int63()
|
|
}
|
|
|
|
// Uint64 returns a cryptographically secure strongly random uint64.
|
|
func Uint64() uint64 {
|
|
return defaultSecureSource.Uint64()
|
|
}
|
|
|
|
// Intn returns, as an int, a cryptographically secure non-negative
|
|
// random number in [0,n). It panics if n <= 0.
|
|
func Intn(n int) int {
|
|
return defaultSecureRand.Intn(n)
|
|
}
|
|
|
|
// Shuffle the order of elements in slice.
|
|
func Shuffle(slice interface{}) {
|
|
rv := reflect.ValueOf(slice)
|
|
swap := reflect.Swapper(slice)
|
|
length := rv.Len()
|
|
if length < 2 {
|
|
return
|
|
}
|
|
defaultSecureRand.Shuffle(length, swap)
|
|
}
|