go-waku/waku/v2/protocol/requestId.go

47 lines
1.1 KiB
Go
Raw Normal View History

package protocol
import (
"crypto/rand"
"sync"
"github.com/cruxic/go-hmac-drbg/hmacdrbg"
"github.com/waku-org/go-waku/waku/v2/utils"
"go.uber.org/zap"
)
var brHmacDrbgPool = sync.Pool{New: func() interface{} {
seed := make([]byte, 48)
_, err := rand.Read(seed)
if err != nil {
utils.Logger().Fatal("rand.Read err", zap.Error(err))
}
return hmacdrbg.NewHmacDrbg(256, seed, nil)
}}
2021-10-09 18:18:53 +00:00
// GenerateRequestId generates a random 32 byte slice that can be used for
// creating requests inf the filter, store and lightpush protocols
func GenerateRequestId() []byte {
rng := brHmacDrbgPool.Get().(*hmacdrbg.HmacDrbg)
defer brHmacDrbgPool.Put(rng)
randData := make([]byte, 32)
if !rng.Generate(randData) {
//Reseed is required every 10,000 calls
seed := make([]byte, 48)
_, err := rand.Read(seed)
if err != nil {
utils.Logger().Fatal("rand.Read err", zap.Error(err))
}
err = rng.Reseed(seed)
if err != nil {
//only happens if seed < security-level
utils.Logger().Fatal("rng.Reseed err", zap.Error(err))
}
if !rng.Generate(randData) {
utils.Logger().Error("could not generate random request id")
}
}
return randData
}