mirror of
https://gitea.phreedom.club/localhost_frssoft/bloat.git
synced 2025-05-02 10:04:22 +00:00
Update random id generation algorithm
This commit is contained in:
parent
bf2cfaf0ed
commit
c9842c65b4
3 changed files with 23 additions and 9 deletions
17
util/rand.go
17
util/rand.go
|
@ -1,7 +1,8 @@
|
|||
package util
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"crypto/rand"
|
||||
"math/big"
|
||||
)
|
||||
|
||||
var (
|
||||
|
@ -9,18 +10,22 @@ var (
|
|||
runes_length = len(runes)
|
||||
)
|
||||
|
||||
func NewRandId(n int) string {
|
||||
func NewRandId(n int) (string, error) {
|
||||
data := make([]rune, n)
|
||||
for i := range data {
|
||||
data[i] = runes[rand.Intn(runes_length)]
|
||||
num, err := rand.Int(rand.Reader, big.NewInt(int64(runes_length)))
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
data[i] = runes[num.Int64()]
|
||||
}
|
||||
return string(data)
|
||||
return string(data), nil
|
||||
}
|
||||
|
||||
func NewSessionId() string {
|
||||
func NewSessionId() (string, error) {
|
||||
return NewRandId(24)
|
||||
}
|
||||
|
||||
func NewCSRFToken() string {
|
||||
func NewCSRFToken() (string, error) {
|
||||
return NewRandId(24)
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue