diff options
Diffstat (limited to 'pkg/rand')
-rw-r--r-- | pkg/rand/BUILD | 5 | ||||
-rw-r--r-- | pkg/rand/rand.go | 22 | ||||
-rw-r--r-- | pkg/rand/rand_linux.go | 39 |
3 files changed, 49 insertions, 17 deletions
diff --git a/pkg/rand/BUILD b/pkg/rand/BUILD index 12e6cf25a..97b9ba3ff 100644 --- a/pkg/rand/BUILD +++ b/pkg/rand/BUILD @@ -4,7 +4,10 @@ load("//tools/go_stateify:defs.bzl", "go_library") go_library( name = "rand", - srcs = ["rand.go"], + srcs = [ + "rand.go", + "rand_linux.go", + ], importpath = "gvisor.googlesource.com/gvisor/pkg/rand", visibility = ["//:sandbox"], deps = ["@org_golang_x_sys//unix:go_default_library"], diff --git a/pkg/rand/rand.go b/pkg/rand/rand.go index 37ac07620..e81f0f5db 100644 --- a/pkg/rand/rand.go +++ b/pkg/rand/rand.go @@ -12,28 +12,18 @@ // See the License for the specific language governing permissions and // limitations under the License. +// +build !linux + // Package rand implements a cryptographically secure pseudorandom number // generator. package rand -import ( - "io" - - "golang.org/x/sys/unix" -) - -// reader implements an io.Reader that returns pseudorandom bytes. -type reader struct{} - -// Read implements io.Reader.Read. -func (reader) Read(p []byte) (int, error) { - return unix.Getrandom(p, 0) -} +import "crypto/rand" // Reader is the default reader. -var Reader io.Reader = reader{} +var Reader = rand.Reader -// Read reads from the default reader. +// Read implements io.Reader.Read. func Read(b []byte) (int, error) { - return io.ReadFull(Reader, b) + return rand.Read(b) } diff --git a/pkg/rand/rand_linux.go b/pkg/rand/rand_linux.go new file mode 100644 index 000000000..37ac07620 --- /dev/null +++ b/pkg/rand/rand_linux.go @@ -0,0 +1,39 @@ +// Copyright 2018 Google Inc. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +// Package rand implements a cryptographically secure pseudorandom number +// generator. +package rand + +import ( + "io" + + "golang.org/x/sys/unix" +) + +// reader implements an io.Reader that returns pseudorandom bytes. +type reader struct{} + +// Read implements io.Reader.Read. +func (reader) Read(p []byte) (int, error) { + return unix.Getrandom(p, 0) +} + +// Reader is the default reader. +var Reader io.Reader = reader{} + +// Read reads from the default reader. +func Read(b []byte) (int, error) { + return io.ReadFull(Reader, b) +} |