rpc: remove silly use of ReadVarint in subscription ID generator (#21391)

Found by @protolambda
pull/21395/head
Felix Lange 4 years ago committed by GitHub
parent a00dc5095b
commit 9c2ac6fbd5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 9
      rpc/subscription.go

@ -17,7 +17,6 @@
package rpc package rpc
import ( import (
"bufio"
"container/list" "container/list"
"context" "context"
crand "crypto/rand" crand "crypto/rand"
@ -51,10 +50,14 @@ func NewID() ID {
// randomIDGenerator returns a function generates a random IDs. // randomIDGenerator returns a function generates a random IDs.
func randomIDGenerator() func() ID { func randomIDGenerator() func() ID {
seed, err := binary.ReadVarint(bufio.NewReader(crand.Reader)) var buf = make([]byte, 8)
if err != nil { var seed int64
if _, err := crand.Read(buf); err == nil {
seed = int64(binary.BigEndian.Uint64(buf))
} else {
seed = int64(time.Now().Nanosecond()) seed = int64(time.Now().Nanosecond())
} }
var ( var (
mu sync.Mutex mu sync.Mutex
rng = rand.New(rand.NewSource(seed)) rng = rand.New(rand.NewSource(seed))

Loading…
Cancel
Save