82 lines
2.4 KiB
Go
82 lines
2.4 KiB
Go
// Written in 2011-2014 by Dmitry Chestnykh
|
|
//
|
|
// The author(s) have dedicated all copyright and related and
|
|
// neighboring rights to this software to the public domain
|
|
// worldwide. Distributed without any warranty.
|
|
// http://creativecommons.org/publicdomain/zero/1.0/
|
|
|
|
// Package uniuri generates random strings good for use in URIs to identify
|
|
// unique objects.
|
|
//
|
|
// Example usage:
|
|
//
|
|
// s := uniuri.New() // s is now "apHCJBl7L1OmC57n"
|
|
//
|
|
// A standard string created by New() is 16 bytes in length and consists of
|
|
// Latin upper and lowercase letters, and numbers (from the set of 62 allowed
|
|
// characters), which means that it has ~95 bits of entropy. To get more
|
|
// entropy, you can use NewLen(UUIDLen), which returns 20-byte string, giving
|
|
// ~119 bits of entropy, or any other desired length.
|
|
//
|
|
// Functions read from crypto/rand random source, and panic if they fail to
|
|
// read from it.
|
|
package uniuri
|
|
|
|
import "crypto/rand"
|
|
|
|
const (
|
|
// StdLen is a standard length of uniuri string to achive ~95 bits of entropy.
|
|
StdLen = 16
|
|
// UUIDLen is a length of uniuri string to achive ~119 bits of entropy, closest
|
|
// to what can be losslessly converted to UUIDv4 (122 bits).
|
|
UUIDLen = 20
|
|
)
|
|
|
|
// StdChars is a set of standard characters allowed in uniuri string.
|
|
var StdChars = []byte("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789")
|
|
|
|
// New returns a new random string of the standard length, consisting of
|
|
// standard characters.
|
|
func New() string {
|
|
return NewLenChars(StdLen, StdChars)
|
|
}
|
|
|
|
// NewLen returns a new random string of the provided length, consisting of
|
|
// standard characters.
|
|
func NewLen(length int) string {
|
|
return NewLenChars(length, StdChars)
|
|
}
|
|
|
|
// NewLenChars returns a new random string of the provided length, consisting
|
|
// of the provided byte slice of allowed characters (maximum 256).
|
|
func NewLenChars(length int, chars []byte) string {
|
|
if length == 0 {
|
|
return ""
|
|
}
|
|
clen := len(chars)
|
|
if clen < 2 || clen > 256 {
|
|
panic("uniuri: wrong charset length for NewLenChars")
|
|
}
|
|
maxrb := 255 - (256 % clen)
|
|
b := make([]byte, length)
|
|
r := make([]byte, length+(length/4)) // storage for random bytes.
|
|
i := 0
|
|
for {
|
|
if _, err := rand.Read(r); err != nil {
|
|
panic("error reading from random source: " + err.Error())
|
|
}
|
|
for _, rb := range r {
|
|
c := int(rb)
|
|
if c > maxrb {
|
|
// Skip this number to avoid modulo bias.
|
|
continue
|
|
}
|
|
b[i] = chars[c%clen]
|
|
i++
|
|
if i == length {
|
|
return string(b)
|
|
}
|
|
}
|
|
}
|
|
}
|