Brad Fitzpatrick | 5194744 | 2016-03-01 22:57:46 +0000 | [diff] [blame] | 1 | // Copyright 2010 The Go Authors. All rights reserved. |
Peter Mundy | ccd28e8 | 2010-07-12 16:37:53 -0700 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style |
| 3 | // license that can be found in the LICENSE file. |
| 4 | |
| 5 | // Windows cryptographically secure pseudorandom number |
| 6 | // generator. |
| 7 | |
| 8 | package rand |
| 9 | |
| 10 | import ( |
Jason A. Donenfeld | bb5075a | 2021-01-15 00:04:10 +0100 | [diff] [blame] | 11 | "internal/syscall/windows" |
Peter Mundy | ccd28e8 | 2010-07-12 16:37:53 -0700 | [diff] [blame] | 12 | "os" |
Peter Mundy | ccd28e8 | 2010-07-12 16:37:53 -0700 | [diff] [blame] | 13 | ) |
| 14 | |
Peter Mundy | ccd28e8 | 2010-07-12 16:37:53 -0700 | [diff] [blame] | 15 | func init() { Reader = &rngReader{} } |
| 16 | |
Jason A. Donenfeld | 333e904 | 2019-12-05 18:48:21 +0100 | [diff] [blame] | 17 | type rngReader struct{} |
Peter Mundy | ccd28e8 | 2010-07-12 16:37:53 -0700 | [diff] [blame] | 18 | |
Russ Cox | c2049d2 | 2011-11-01 22:04:37 -0400 | [diff] [blame] | 19 | func (r *rngReader) Read(b []byte) (n int, err error) { |
Jason A. Donenfeld | 333e904 | 2019-12-05 18:48:21 +0100 | [diff] [blame] | 20 | // RtlGenRandom only accepts 2**32-1 bytes at a time, so truncate. |
| 21 | inputLen := uint32(len(b)) |
Adam Langley | b752ab2 | 2012-09-09 20:07:09 -0400 | [diff] [blame] | 22 | |
Jason A. Donenfeld | 333e904 | 2019-12-05 18:48:21 +0100 | [diff] [blame] | 23 | if inputLen == 0 { |
Adam Langley | b752ab2 | 2012-09-09 20:07:09 -0400 | [diff] [blame] | 24 | return 0, nil |
| 25 | } |
Jason A. Donenfeld | 333e904 | 2019-12-05 18:48:21 +0100 | [diff] [blame] | 26 | |
Jason A. Donenfeld | bb5075a | 2021-01-15 00:04:10 +0100 | [diff] [blame] | 27 | err = windows.RtlGenRandom(b) |
Russ Cox | c017a82 | 2011-11-13 22:44:52 -0500 | [diff] [blame] | 28 | if err != nil { |
Jason A. Donenfeld | 333e904 | 2019-12-05 18:48:21 +0100 | [diff] [blame] | 29 | return 0, os.NewSyscallError("RtlGenRandom", err) |
Peter Mundy | ccd28e8 | 2010-07-12 16:37:53 -0700 | [diff] [blame] | 30 | } |
Jason A. Donenfeld | 333e904 | 2019-12-05 18:48:21 +0100 | [diff] [blame] | 31 | return int(inputLen), nil |
Peter Mundy | ccd28e8 | 2010-07-12 16:37:53 -0700 | [diff] [blame] | 32 | } |