summaryrefslogtreecommitdiff
path: root/libgo/go/crypto/rand/rand_windows.go
blob: 4b2b7a26f37f214fe7f3b34636fc78641c7fbe33 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
// Copyright 2010 The Go Authors.  All rights reserved.
// 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"
	"syscall"
)

// 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 {
	prov uint32
	mu   sync.Mutex
}

func (r *rngReader) Read(b []byte) (n int, err os.Error) {
	r.mu.Lock()
	if r.prov == 0 {
		const provType = syscall.PROV_RSA_FULL
		const flags = syscall.CRYPT_VERIFYCONTEXT | syscall.CRYPT_SILENT
		ok, errno := syscall.CryptAcquireContext(&r.prov, nil, nil, provType, flags)
		if !ok {
			r.mu.Unlock()
			return 0, os.NewSyscallError("CryptAcquireContext", errno)
		}
	}
	r.mu.Unlock()
	ok, errno := syscall.CryptGenRandom(r.prov, uint32(len(b)), &b[0])
	if !ok {
		return 0, os.NewSyscallError("CryptGenRandom", errno)
	}
	return len(b), nil
}