2017-10-07 00:16:47 +00:00
|
|
|
// Copyright 2010 The Go Authors. All rights reserved.
|
2012-03-27 23:13:14 +00:00
|
|
|
// Use of this source code is governed by a BSD-style
|
|
|
|
// license that can be found in the LICENSE file.
|
|
|
|
|
|
|
|
// Windows cryptographically secure pseudorandom number
|
|
|
|
// generator.
|
|
|
|
|
|
|
|
package rand
|
|
|
|
|
|
|
|
import (
|
|
|
|
"os"
|
|
|
|
"sync"
|
2019-06-02 15:48:37 +00:00
|
|
|
"sync/atomic"
|
2012-03-27 23:13:14 +00:00
|
|
|
"syscall"
|
2019-06-02 15:48:37 +00:00
|
|
|
"time"
|
2012-03-27 23:13:14 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
// Implemented by using Windows CryptoAPI 2.0.
|
|
|
|
|
|
|
|
func init() { Reader = &rngReader{} }
|
|
|
|
|
|
|
|
// A rngReader satisfies reads by reading from the Windows CryptGenRandom API.
|
|
|
|
type rngReader struct {
|
2019-06-02 15:48:37 +00:00
|
|
|
used int32 // atomic; whether this rngReader has been used
|
2012-03-27 23:13:14 +00:00
|
|
|
prov syscall.Handle
|
|
|
|
mu sync.Mutex
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *rngReader) Read(b []byte) (n int, err error) {
|
2019-06-02 15:48:37 +00:00
|
|
|
if atomic.CompareAndSwapInt32(&r.used, 0, 1) {
|
|
|
|
// First use of randomness. Start timer to warn about
|
|
|
|
// being blocked on entropy not being available.
|
|
|
|
t := time.AfterFunc(60*time.Second, warnBlocked)
|
|
|
|
defer t.Stop()
|
|
|
|
}
|
2012-03-27 23:13:14 +00:00
|
|
|
r.mu.Lock()
|
|
|
|
if r.prov == 0 {
|
|
|
|
const provType = syscall.PROV_RSA_FULL
|
|
|
|
const flags = syscall.CRYPT_VERIFYCONTEXT | syscall.CRYPT_SILENT
|
|
|
|
err := syscall.CryptAcquireContext(&r.prov, nil, nil, provType, flags)
|
|
|
|
if err != nil {
|
|
|
|
r.mu.Unlock()
|
|
|
|
return 0, os.NewSyscallError("CryptAcquireContext", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
r.mu.Unlock()
|
2014-09-21 17:33:12 +00:00
|
|
|
|
|
|
|
if len(b) == 0 {
|
|
|
|
return 0, nil
|
|
|
|
}
|
2012-03-27 23:13:14 +00:00
|
|
|
err = syscall.CryptGenRandom(r.prov, uint32(len(b)), &b[0])
|
|
|
|
if err != nil {
|
|
|
|
return 0, os.NewSyscallError("CryptGenRandom", err)
|
|
|
|
}
|
|
|
|
return len(b), nil
|
|
|
|
}
|