coolaj86/go-mockid
1
0
Fork 0
Code Issues 3 Pull Requests Releases Wiki Activity

make key type deterministic with seed

Browse Source
This commit is contained in:
AJ ONeal 2020-08-02 00:11:50 +00:00
parent da712abbb2
commit bc7e9740d8
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mockid/api/common.go
View File

@ -41,13 +41,18 @@ func getOpts(r *http.Request) (*options, error) {
seed, _ = binary.ReadVarint(bytes.NewReader(b[0:8])) seed, _ = binary.ReadVarint(bytes.NewReader(b[0:8]))
} }
var n int
if 0 != seed { if 0 != seed {
rndReader = rand.New(rand.NewSource(seed)) rnd := rand.New(rand.NewSource(seed))
rndReader = rnd
n = rnd.Intn(2)
} else {
n = rand.Intn(2)
} }
kty, _ := tok["kty"].(string) kty, _ := tok["kty"].(string)
if "" == kty { if "" == kty {
if 0 == rand.Intn(2) { if 0 == n {
kty = "RSA" kty = "RSA"
} else { } else {
kty = "EC" kty = "EC"

4
mockid/mockid_test.go
View File

@ -121,7 +121,7 @@ func TestGenWithSeed(t *testing.T) {
res, err := client.Do(&http.Request{ res, err := client.Do(&http.Request{
Method: "POST", Method: "POST",
URL: urlstr, URL: urlstr,
Body: ioutil.NopCloser(bytes.NewReader([]byte(`{"seed":"test"}`))), Body: ioutil.NopCloser(bytes.NewReader([]byte(`{"seed":"c"}`))),
}) })
if nil != err { if nil != err {
//t.Fatal(err) //t.Fatal(err)
@ -141,7 +141,7 @@ func TestGenWithSeed(t *testing.T) {
res, err = client.Do(&http.Request{ res, err = client.Do(&http.Request{
Method: "POST", Method: "POST",
URL: urlstr, URL: urlstr,
Body: ioutil.NopCloser(bytes.NewReader([]byte(`{"seed":"test"}`))), Body: ioutil.NopCloser(bytes.NewReader([]byte(`{"seed":"c"}`))),
}) })
if nil != err { if nil != err {
//t.Fatal(err) //t.Fatal(err)