From 14a70624ce95dcf9e4505698b1174df583b13179 Mon Sep 17 00:00:00 2001
From: Tamir Duberstein <tamird@google.com>
Date: Wed, 26 May 2021 10:37:59 -0700
Subject: Alias most local import

PiperOrigin-RevId: 375977977
---
 pkg/tcpip/stack/rand.go  |  4 ++--
 pkg/tcpip/stack/stack.go | 24 ++++++++++++------------
 2 files changed, 14 insertions(+), 14 deletions(-)

(limited to 'pkg/tcpip/stack')

diff --git a/pkg/tcpip/stack/rand.go b/pkg/tcpip/stack/rand.go
index 421fb5c15..c8294eb6e 100644
--- a/pkg/tcpip/stack/rand.go
+++ b/pkg/tcpip/stack/rand.go
@@ -15,7 +15,7 @@
 package stack
 
 import (
-	mathrand "math/rand"
+	"math/rand"
 
 	"gvisor.dev/gvisor/pkg/sync"
 )
@@ -23,7 +23,7 @@ import (
 // lockedRandomSource provides a threadsafe rand.Source.
 type lockedRandomSource struct {
 	mu  sync.Mutex
-	src mathrand.Source
+	src rand.Source
 }
 
 func (r *lockedRandomSource) Int63() (n int64) {
diff --git a/pkg/tcpip/stack/stack.go b/pkg/tcpip/stack/stack.go
index 72760a4a7..c467f1266 100644
--- a/pkg/tcpip/stack/stack.go
+++ b/pkg/tcpip/stack/stack.go
@@ -24,13 +24,13 @@ import (
 	"encoding/binary"
 	"fmt"
 	"io"
-	mathrand "math/rand"
+	"math/rand"
 	"sync/atomic"
 	"time"
 
 	"golang.org/x/time/rate"
 	"gvisor.dev/gvisor/pkg/atomicbitops"
-	"gvisor.dev/gvisor/pkg/rand"
+	cryptorand "gvisor.dev/gvisor/pkg/rand"
 	"gvisor.dev/gvisor/pkg/sync"
 	"gvisor.dev/gvisor/pkg/tcpip"
 	"gvisor.dev/gvisor/pkg/tcpip/buffer"
@@ -138,7 +138,7 @@ type Stack struct {
 
 	// randomGenerator is an injectable pseudo random generator that can be
 	// used when a random number is required.
-	randomGenerator *mathrand.Rand
+	randomGenerator *rand.Rand
 
 	// secureRNG is a cryptographically secure random number generator.
 	secureRNG io.Reader
@@ -218,10 +218,10 @@ type Options struct {
 
 	// RandSource is an optional source to use to generate random
 	// numbers. If omitted it defaults to a Source seeded by the data
-	// returned by rand.Read().
+	// returned by the stack secure RNG.
 	//
 	// RandSource must be thread-safe.
-	RandSource mathrand.Source
+	RandSource rand.Source
 
 	// IPTables are the initial iptables rules. If nil, iptables will allow
 	// all traffic.
@@ -326,9 +326,9 @@ func New(opts Options) *Stack {
 
 	randSrc := opts.RandSource
 	if randSrc == nil {
-		// Source provided by mathrand.NewSource is not thread-safe so
+		// Source provided by rand.NewSource is not thread-safe so
 		// we wrap it in a simple thread-safe version.
-		randSrc = &lockedRandomSource{src: mathrand.NewSource(generateRandInt64())}
+		randSrc = &lockedRandomSource{src: rand.NewSource(generateRandInt64())}
 	}
 
 	if opts.IPTables == nil {
@@ -338,7 +338,7 @@ func New(opts Options) *Stack {
 	opts.NUDConfigs.resetInvalidFields()
 
 	if opts.SecureRNG == nil {
-		opts.SecureRNG = rand.Reader
+		opts.SecureRNG = cryptorand.Reader
 	}
 
 	s := &Stack{
@@ -357,7 +357,7 @@ func New(opts Options) *Stack {
 		nudConfigs:               opts.NUDConfigs,
 		uniqueIDGenerator:        opts.UniqueID,
 		nudDisp:                  opts.NUDDisp,
-		randomGenerator:          mathrand.New(randSrc),
+		randomGenerator:          rand.New(randSrc),
 		secureRNG:                opts.SecureRNG,
 		sendBufferSize: tcpip.SendBufferSizeOption{
 			Min:     MinBufferSize,
@@ -1812,7 +1812,7 @@ func (s *Stack) Seed() uint32 {
 
 // Rand returns a reference to a pseudo random generator that can be used
 // to generate random numbers as required.
-func (s *Stack) Rand() *mathrand.Rand {
+func (s *Stack) Rand() *rand.Rand {
 	return s.randomGenerator
 }
 
@@ -1824,7 +1824,7 @@ func (s *Stack) SecureRNG() io.Reader {
 
 func generateRandUint32() uint32 {
 	b := make([]byte, 4)
-	if _, err := rand.Read(b); err != nil {
+	if _, err := cryptorand.Read(b); err != nil {
 		panic(err)
 	}
 	return binary.LittleEndian.Uint32(b)
@@ -1832,7 +1832,7 @@ func generateRandUint32() uint32 {
 
 func generateRandInt64() int64 {
 	b := make([]byte, 8)
-	if _, err := rand.Read(b); err != nil {
+	if _, err := cryptorand.Read(b); err != nil {
 		panic(err)
 	}
 	buf := bytes.NewReader(b)
-- 
cgit v1.2.3