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 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159
|
package cose
import (
"crypto"
"crypto/ecdsa"
"crypto/elliptic"
"crypto/rand"
"crypto/rsa"
"encoding/base64"
"math/big"
"reflect"
"testing"
)
func mustBase64ToBigInt(s string) *big.Int {
val, err := base64.RawURLEncoding.DecodeString(s)
if err != nil {
panic(err)
}
return new(big.Int).SetBytes(val)
}
func generateBogusECKey() *ecdsa.PublicKey {
return &ecdsa.PublicKey{
Curve: elliptic.P256(),
// x-coord is not on curve p-256
X: mustBase64ToBigInt("MKBCTNIcKUSDii11ySs3526iDZ8AiTo7Tu6KPAqx7D4"),
Y: mustBase64ToBigInt("4Etl6SRW2YiLUrN5vfvVHuhp7x8PxltmWWlbbM4IFyM"),
}
}
func TestNewVerifier(t *testing.T) {
// generate ecdsa key
ecdsaKey := generateTestECDSAKey(t).Public().(*ecdsa.PublicKey)
// generate ed25519 key
ed25519Key, _ := generateTestEd25519Key(t)
// generate rsa keys
rsaKey := generateTestRSAKey(t).Public().(*rsa.PublicKey)
var rsaKeyLowEntropy *rsa.PublicKey
if key, err := rsa.GenerateKey(rand.Reader, 1024); err != nil {
t.Fatalf("rsa.GenerateKey() error = %v", err)
} else {
rsaKeyLowEntropy = &key.PublicKey
}
// craft an EC public key with the x-coord not on curve
ecdsaKeyPointNotOnCurve := generateBogusECKey()
// craft an EC public key with a curve not supported by crypto/ecdh
ecdsaKeyUnsupportedCurve := &ecdsa.PublicKey{
Curve: ecdsaKey.Curve.Params(),
X: ecdsaKey.X,
Y: ecdsaKey.Y,
}
// run tests
tests := []struct {
name string
alg Algorithm
key crypto.PublicKey
want Verifier
wantErr string
}{
{
name: "ecdsa key verifier",
alg: AlgorithmES256,
key: ecdsaKey,
want: &ecdsaVerifier{
alg: AlgorithmES256,
key: ecdsaKey,
},
},
{
name: "ecdsa invalid public key",
alg: AlgorithmES256,
key: rsaKey,
wantErr: "ES256: invalid public key",
},
{
name: "ed25519 verifier",
alg: AlgorithmEdDSA,
key: ed25519Key,
want: &ed25519Verifier{
key: ed25519Key,
},
},
{
name: "ed25519 invalid public key",
alg: AlgorithmEdDSA,
key: rsaKey,
wantErr: "EdDSA: invalid public key",
},
{
name: "rsa verifier",
alg: AlgorithmPS256,
key: rsaKey,
want: &rsaVerifier{
alg: AlgorithmPS256,
key: rsaKey,
},
},
{
name: "rsa invalid public key",
alg: AlgorithmPS256,
key: ecdsaKey,
wantErr: "PS256: invalid public key",
},
{
name: "rsa key under minimum entropy",
alg: AlgorithmPS256,
key: rsaKeyLowEntropy,
wantErr: "RSA key must be at least 2048 bits long",
},
{
name: "unsupported rsa signing algorithm",
alg: AlgorithmRS256,
wantErr: "can't create new Verifier for RS256: no built-in implementation available: algorithm not supported",
},
{
name: "reserved algorithm",
alg: AlgorithmReserved,
wantErr: "can't create new Verifier for Reserved: can't be implemented: algorithm not supported",
},
{
name: "unassigned algorithm",
alg: -1,
wantErr: "can't create new Verifier for Algorithm(-1): unknown algorithm: algorithm not supported",
},
{
name: "bogus ecdsa public key (point not on curve)",
alg: AlgorithmES256,
key: ecdsaKeyPointNotOnCurve,
wantErr: "ES256: invalid public key",
},
{
name: "ecdsa public key with unsupported curve",
alg: AlgorithmES256,
key: ecdsaKeyUnsupportedCurve,
wantErr: "ES256: invalid public key: ecdsa: unsupported curve by crypto/ecdh",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got, err := NewVerifier(tt.alg, tt.key)
if err != nil && (err.Error() != tt.wantErr) {
t.Errorf("NewVerifier() error = %v, wantErr %v", err, tt.wantErr)
return
} else if err == nil && (tt.wantErr != "") {
t.Errorf("NewVerifier() error = %v, wantErr %v", err, tt.wantErr)
return
}
if !reflect.DeepEqual(got, tt.want) {
t.Errorf("NewVerifier() = %v, want %v", got, tt.want)
}
})
}
}
|