mirror of
https://github.com/smallstep/certificates.git
synced 2024-10-31 03:20:16 +00:00
2111 lines
61 KiB
Go
2111 lines
61 KiB
Go
package acme
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"crypto"
|
|
"crypto/sha256"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/smallstep/assert"
|
|
"go.step.sm/crypto/jose"
|
|
)
|
|
|
|
func TestKeyAuthorization(t *testing.T) {
|
|
type test struct {
|
|
token string
|
|
jwk *jose.JSONWebKey
|
|
exp string
|
|
err *Error
|
|
}
|
|
tests := map[string]func(t *testing.T) test{
|
|
"fail/jwk-thumbprint-error": func(t *testing.T) test {
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
jwk.Key = "foo"
|
|
return test{
|
|
token: "1234",
|
|
jwk: jwk,
|
|
err: NewErrorISE("error generating JWK thumbprint: square/go-jose: unknown key type 'string'"),
|
|
}
|
|
},
|
|
"ok": func(t *testing.T) test {
|
|
token := "1234"
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
thumbprint, err := jwk.Thumbprint(crypto.SHA256)
|
|
assert.FatalError(t, err)
|
|
encPrint := base64.RawURLEncoding.EncodeToString(thumbprint)
|
|
return test{
|
|
token: token,
|
|
jwk: jwk,
|
|
exp: fmt.Sprintf("%s.%s", token, encPrint),
|
|
}
|
|
},
|
|
}
|
|
for name, run := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
tc := run(t)
|
|
if ka, err := KeyAuthorization(tc.token, tc.jwk); err != nil {
|
|
if assert.NotNil(t, tc.err) {
|
|
switch k := err.(type) {
|
|
case *Error:
|
|
assert.Equals(t, k.Type, tc.err.Type)
|
|
assert.Equals(t, k.Detail, tc.err.Detail)
|
|
assert.Equals(t, k.Status, tc.err.Status)
|
|
assert.Equals(t, k.Err.Error(), tc.err.Err.Error())
|
|
assert.Equals(t, k.Detail, tc.err.Detail)
|
|
default:
|
|
assert.FatalError(t, errors.New("unexpected error type"))
|
|
}
|
|
}
|
|
} else {
|
|
if assert.Nil(t, tc.err) {
|
|
assert.Equals(t, tc.exp, ka)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
type errReader int
|
|
|
|
func (errReader) Read(p []byte) (n int, err error) {
|
|
return 0, errors.New("force")
|
|
}
|
|
func (errReader) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func TestHTTP01Validate(t *testing.T) {
|
|
type test struct {
|
|
vo *ValidateChallengeOptions
|
|
ch *Challenge
|
|
jwk *jose.JSONWebKey
|
|
db DB
|
|
err *Error
|
|
}
|
|
tests := map[string]func(t *testing.T) test{
|
|
"fail/http-get-error-store-error": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: "zap.internal",
|
|
}
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
HTTPGet: func(url string) (*http.Response, error) {
|
|
return nil, errors.New("force")
|
|
},
|
|
},
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
|
|
err := NewError(ErrorConnectionType, "error doing http GET for url http://zap.internal/.well-known/acme-challenge/%s: force", ch.Token)
|
|
assert.HasPrefix(t, updch.Error.Err.Error(), err.Err.Error())
|
|
assert.Equals(t, updch.Error.Type, err.Type)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
assert.Equals(t, updch.Error.Status, err.Status)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
return errors.New("force")
|
|
},
|
|
},
|
|
err: NewErrorISE("failure saving error to acme challenge: force"),
|
|
}
|
|
},
|
|
"ok/http-get-error": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: "zap.internal",
|
|
}
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
HTTPGet: func(url string) (*http.Response, error) {
|
|
return nil, errors.New("force")
|
|
},
|
|
},
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
|
|
err := NewError(ErrorConnectionType, "error doing http GET for url http://zap.internal/.well-known/acme-challenge/%s: force", ch.Token)
|
|
assert.HasPrefix(t, updch.Error.Err.Error(), err.Err.Error())
|
|
assert.Equals(t, updch.Error.Type, err.Type)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
assert.Equals(t, updch.Error.Status, err.Status)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
return nil
|
|
},
|
|
},
|
|
}
|
|
},
|
|
"fail/http-get->=400-store-error": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: "zap.internal",
|
|
}
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
HTTPGet: func(url string) (*http.Response, error) {
|
|
return &http.Response{
|
|
StatusCode: http.StatusBadRequest,
|
|
}, nil
|
|
},
|
|
},
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
|
|
err := NewError(ErrorConnectionType, "error doing http GET for url http://zap.internal/.well-known/acme-challenge/%s with status code 400", ch.Token)
|
|
assert.HasPrefix(t, updch.Error.Err.Error(), err.Err.Error())
|
|
assert.Equals(t, updch.Error.Type, err.Type)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
assert.Equals(t, updch.Error.Status, err.Status)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
return errors.New("force")
|
|
},
|
|
},
|
|
err: NewErrorISE("failure saving error to acme challenge: force"),
|
|
}
|
|
},
|
|
"ok/http-get->=400": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: "zap.internal",
|
|
}
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
HTTPGet: func(url string) (*http.Response, error) {
|
|
return &http.Response{
|
|
StatusCode: http.StatusBadRequest,
|
|
}, nil
|
|
},
|
|
},
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
|
|
err := NewError(ErrorConnectionType, "error doing http GET for url http://zap.internal/.well-known/acme-challenge/%s with status code 400", ch.Token)
|
|
assert.HasPrefix(t, updch.Error.Err.Error(), err.Err.Error())
|
|
assert.Equals(t, updch.Error.Type, err.Type)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
assert.Equals(t, updch.Error.Status, err.Status)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
return nil
|
|
},
|
|
},
|
|
}
|
|
},
|
|
"fail/read-body": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: "zap.internal",
|
|
}
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
HTTPGet: func(url string) (*http.Response, error) {
|
|
return &http.Response{
|
|
Body: errReader(0),
|
|
}, nil
|
|
},
|
|
},
|
|
err: NewErrorISE("error reading response body for url http://zap.internal/.well-known/acme-challenge/%s: force", ch.Token),
|
|
}
|
|
},
|
|
"fail/key-auth-gen-error": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: "zap.internal",
|
|
}
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
jwk.Key = "foo"
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
HTTPGet: func(url string) (*http.Response, error) {
|
|
return &http.Response{
|
|
Body: ioutil.NopCloser(bytes.NewBufferString("foo")),
|
|
}, nil
|
|
},
|
|
},
|
|
jwk: jwk,
|
|
err: NewErrorISE("error generating JWK thumbprint: square/go-jose: unknown key type 'string'"),
|
|
}
|
|
},
|
|
"ok/key-auth-mismatch": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: "zap.internal",
|
|
}
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.Token, jwk)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
HTTPGet: func(url string) (*http.Response, error) {
|
|
return &http.Response{
|
|
Body: ioutil.NopCloser(bytes.NewBufferString("foo")),
|
|
}, nil
|
|
},
|
|
},
|
|
jwk: jwk,
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
assert.Equals(t, updch.Value, ch.Value)
|
|
|
|
err := NewError(ErrorRejectedIdentifierType,
|
|
"keyAuthorization does not match; expected %s, but got foo", expKeyAuth)
|
|
assert.HasPrefix(t, updch.Error.Err.Error(), err.Err.Error())
|
|
assert.Equals(t, updch.Error.Type, err.Type)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
assert.Equals(t, updch.Error.Status, err.Status)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
return nil
|
|
},
|
|
},
|
|
}
|
|
},
|
|
"fail/key-auth-mismatch-store-error": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: "zap.internal",
|
|
}
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.Token, jwk)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
HTTPGet: func(url string) (*http.Response, error) {
|
|
return &http.Response{
|
|
Body: ioutil.NopCloser(bytes.NewBufferString("foo")),
|
|
}, nil
|
|
},
|
|
},
|
|
jwk: jwk,
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
assert.Equals(t, updch.Value, ch.Value)
|
|
|
|
err := NewError(ErrorRejectedIdentifierType,
|
|
"keyAuthorization does not match; expected %s, but got foo", expKeyAuth)
|
|
assert.HasPrefix(t, updch.Error.Err.Error(), err.Err.Error())
|
|
assert.Equals(t, updch.Error.Type, err.Type)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
assert.Equals(t, updch.Error.Status, err.Status)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
return errors.New("force")
|
|
},
|
|
},
|
|
err: NewErrorISE("failure saving error to acme challenge: force"),
|
|
}
|
|
},
|
|
"fail/update-challenge-error": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: "zap.internal",
|
|
}
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.Token, jwk)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
HTTPGet: func(url string) (*http.Response, error) {
|
|
return &http.Response{
|
|
Body: ioutil.NopCloser(bytes.NewBufferString(expKeyAuth)),
|
|
}, nil
|
|
},
|
|
},
|
|
jwk: jwk,
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
assert.Equals(t, updch.Value, ch.Value)
|
|
assert.Equals(t, updch.Status, StatusValid)
|
|
assert.Equals(t, updch.Error, nil)
|
|
va, err := time.Parse(time.RFC3339, updch.ValidatedAt)
|
|
assert.FatalError(t, err)
|
|
now := clock.Now()
|
|
assert.True(t, va.Add(-time.Minute).Before(now))
|
|
assert.True(t, va.Add(time.Minute).After(now))
|
|
|
|
return errors.New("force")
|
|
},
|
|
},
|
|
err: NewErrorISE("error updating challenge: force"),
|
|
}
|
|
},
|
|
"ok": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: "zap.internal",
|
|
}
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.Token, jwk)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
HTTPGet: func(url string) (*http.Response, error) {
|
|
return &http.Response{
|
|
Body: ioutil.NopCloser(bytes.NewBufferString(expKeyAuth)),
|
|
}, nil
|
|
},
|
|
},
|
|
jwk: jwk,
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
assert.Equals(t, updch.Value, ch.Value)
|
|
|
|
assert.Equals(t, updch.Status, StatusValid)
|
|
assert.Equals(t, updch.Error, nil)
|
|
va, err := time.Parse(time.RFC3339, updch.ValidatedAt)
|
|
assert.FatalError(t, err)
|
|
now := clock.Now()
|
|
assert.True(t, va.Add(-time.Minute).Before(now))
|
|
assert.True(t, va.Add(time.Minute).After(now))
|
|
return nil
|
|
},
|
|
},
|
|
}
|
|
},
|
|
}
|
|
for name, run := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
tc := run(t)
|
|
if err := http01Validate(context.Background(), tc.ch, tc.db, tc.jwk, tc.vo); err != nil {
|
|
if assert.NotNil(t, tc.err) {
|
|
switch k := err.(type) {
|
|
case *Error:
|
|
assert.Equals(t, k.Type, tc.err.Type)
|
|
assert.Equals(t, k.Detail, tc.err.Detail)
|
|
assert.Equals(t, k.Status, tc.err.Status)
|
|
assert.Equals(t, k.Err.Error(), tc.err.Err.Error())
|
|
assert.Equals(t, k.Detail, tc.err.Detail)
|
|
default:
|
|
assert.FatalError(t, errors.New("unexpected error type"))
|
|
}
|
|
}
|
|
} else {
|
|
assert.Nil(t, tc.err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDNS01Validate(t *testing.T) {
|
|
fulldomain := "*.zap.internal"
|
|
domain := strings.TrimPrefix(fulldomain, "*.")
|
|
type test struct {
|
|
vo *ValidateChallengeOptions
|
|
ch *Challenge
|
|
jwk *jose.JSONWebKey
|
|
db DB
|
|
err *Error
|
|
}
|
|
tests := map[string]func(t *testing.T) test{
|
|
"fail/lookupTXT-store-error": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: fulldomain,
|
|
}
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
LookupTxt: func(url string) ([]string, error) {
|
|
return nil, errors.New("force")
|
|
},
|
|
},
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
|
|
err := NewError(ErrorDNSType, "error looking up TXT records for domain %s: force", domain)
|
|
|
|
assert.HasPrefix(t, updch.Error.Err.Error(), err.Err.Error())
|
|
assert.Equals(t, updch.Error.Type, err.Type)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
assert.Equals(t, updch.Error.Status, err.Status)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
return errors.New("force")
|
|
},
|
|
},
|
|
err: NewErrorISE("failure saving error to acme challenge: force"),
|
|
}
|
|
},
|
|
"ok/lookupTXT-error": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: fulldomain,
|
|
}
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
LookupTxt: func(url string) ([]string, error) {
|
|
return nil, errors.New("force")
|
|
},
|
|
},
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
|
|
err := NewError(ErrorDNSType, "error looking up TXT records for domain %s: force", domain)
|
|
|
|
assert.HasPrefix(t, updch.Error.Err.Error(), err.Err.Error())
|
|
assert.Equals(t, updch.Error.Type, err.Type)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
assert.Equals(t, updch.Error.Status, err.Status)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
return nil
|
|
},
|
|
},
|
|
}
|
|
},
|
|
"fail/key-auth-gen-error": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: fulldomain,
|
|
}
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
jwk.Key = "foo"
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
LookupTxt: func(url string) ([]string, error) {
|
|
return []string{"foo"}, nil
|
|
},
|
|
},
|
|
jwk: jwk,
|
|
err: NewErrorISE("error generating JWK thumbprint: square/go-jose: unknown key type 'string'"),
|
|
}
|
|
},
|
|
"fail/key-auth-mismatch-store-error": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: fulldomain,
|
|
}
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.Token, jwk)
|
|
assert.FatalError(t, err)
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
LookupTxt: func(url string) ([]string, error) {
|
|
return []string{"foo", "bar"}, nil
|
|
},
|
|
},
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
|
|
err := NewError(ErrorRejectedIdentifierType, "keyAuthorization does not match; expected %s, but got %s", expKeyAuth, []string{"foo", "bar"})
|
|
|
|
assert.HasPrefix(t, updch.Error.Err.Error(), err.Err.Error())
|
|
assert.Equals(t, updch.Error.Type, err.Type)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
assert.Equals(t, updch.Error.Status, err.Status)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
return errors.New("force")
|
|
},
|
|
},
|
|
jwk: jwk,
|
|
err: NewErrorISE("failure saving error to acme challenge: force"),
|
|
}
|
|
},
|
|
"ok/key-auth-mismatch-store-error": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: fulldomain,
|
|
}
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.Token, jwk)
|
|
assert.FatalError(t, err)
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
LookupTxt: func(url string) ([]string, error) {
|
|
return []string{"foo", "bar"}, nil
|
|
},
|
|
},
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
|
|
err := NewError(ErrorRejectedIdentifierType, "keyAuthorization does not match; expected %s, but got %s", expKeyAuth, []string{"foo", "bar"})
|
|
|
|
assert.HasPrefix(t, updch.Error.Err.Error(), err.Err.Error())
|
|
assert.Equals(t, updch.Error.Type, err.Type)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
assert.Equals(t, updch.Error.Status, err.Status)
|
|
assert.Equals(t, updch.Error.Detail, err.Detail)
|
|
return nil
|
|
},
|
|
},
|
|
jwk: jwk,
|
|
}
|
|
},
|
|
"fail/update-challenge-error": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: fulldomain,
|
|
}
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.Token, jwk)
|
|
assert.FatalError(t, err)
|
|
h := sha256.Sum256([]byte(expKeyAuth))
|
|
expected := base64.RawURLEncoding.EncodeToString(h[:])
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
LookupTxt: func(url string) ([]string, error) {
|
|
return []string{"foo", expected}, nil
|
|
},
|
|
},
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
assert.Equals(t, updch.Value, ch.Value)
|
|
|
|
assert.Equals(t, updch.Status, StatusValid)
|
|
assert.Equals(t, updch.Error, nil)
|
|
va, err := time.Parse(time.RFC3339, updch.ValidatedAt)
|
|
assert.FatalError(t, err)
|
|
now := clock.Now()
|
|
assert.True(t, va.Add(-time.Minute).Before(now))
|
|
assert.True(t, va.Add(time.Minute).After(now))
|
|
|
|
return errors.New("force")
|
|
},
|
|
},
|
|
jwk: jwk,
|
|
err: NewErrorISE("error updating challenge: force"),
|
|
}
|
|
},
|
|
"ok": func(t *testing.T) test {
|
|
ch := &Challenge{
|
|
ID: "chID",
|
|
AuthzID: "azID",
|
|
Token: "token",
|
|
Value: fulldomain,
|
|
}
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.Token, jwk)
|
|
assert.FatalError(t, err)
|
|
h := sha256.Sum256([]byte(expKeyAuth))
|
|
expected := base64.RawURLEncoding.EncodeToString(h[:])
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: &ValidateChallengeOptions{
|
|
LookupTxt: func(url string) ([]string, error) {
|
|
return []string{"foo", expected}, nil
|
|
},
|
|
},
|
|
db: &MockDB{
|
|
MockUpdateChallenge: func(ctx context.Context, updch *Challenge) error {
|
|
assert.Equals(t, updch.ID, ch.ID)
|
|
assert.Equals(t, updch.AuthzID, ch.AuthzID)
|
|
assert.Equals(t, updch.Token, ch.Token)
|
|
assert.Equals(t, updch.Value, ch.Value)
|
|
|
|
assert.Equals(t, updch.Status, StatusValid)
|
|
assert.Equals(t, updch.Error, nil)
|
|
va, err := time.Parse(time.RFC3339, updch.ValidatedAt)
|
|
assert.FatalError(t, err)
|
|
now := clock.Now()
|
|
assert.True(t, va.Add(-time.Minute).Before(now))
|
|
assert.True(t, va.Add(time.Minute).After(now))
|
|
|
|
return nil
|
|
},
|
|
},
|
|
jwk: jwk,
|
|
}
|
|
},
|
|
}
|
|
for name, run := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
tc := run(t)
|
|
if err := dns01Validate(context.Background(), tc.ch, tc.db, tc.jwk, tc.vo); err != nil {
|
|
if assert.NotNil(t, tc.err) {
|
|
switch k := err.(type) {
|
|
case *Error:
|
|
assert.Equals(t, k.Type, tc.err.Type)
|
|
assert.Equals(t, k.Detail, tc.err.Detail)
|
|
assert.Equals(t, k.Status, tc.err.Status)
|
|
assert.Equals(t, k.Err.Error(), tc.err.Err.Error())
|
|
assert.Equals(t, k.Detail, tc.err.Detail)
|
|
default:
|
|
assert.FatalError(t, errors.New("unexpected error type"))
|
|
}
|
|
}
|
|
} else {
|
|
assert.Nil(t, tc.err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
/*
|
|
func TestTLSALPN01Validate(t *testing.T) {
|
|
type test struct {
|
|
srv *httptest.Server
|
|
vo validateOptions
|
|
ch challenge
|
|
res challenge
|
|
jwk *jose.JSONWebKey
|
|
db nosql.DB
|
|
err *Error
|
|
}
|
|
tests := map[string]func(t *testing.T) test{
|
|
"ok/status-already-valid": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
_ch, ok := ch.(*tlsALPN01Challenge)
|
|
assert.Fatal(t, ok)
|
|
_ch.baseChallenge.Status = StatusValid
|
|
|
|
return test{
|
|
ch: ch,
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/status-already-invalid": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
_ch, ok := ch.(*tlsALPN01Challenge)
|
|
assert.Fatal(t, ok)
|
|
_ch.baseChallenge.Status = StatusInvalid
|
|
|
|
return test{
|
|
ch: ch,
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/tls-dial-error": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
expErr := ConnectionErr(errors.Errorf("error doing TLS dial for %v:443: force", ch.getValue()))
|
|
baseClone := ch.clone()
|
|
baseClone.Error = expErr.ToACME()
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
newb, err := json.Marshal(newCh)
|
|
assert.FatalError(t, err)
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: func(network, addr string, config *tls.Config) (*tls.Conn, error) {
|
|
return nil, errors.New("force")
|
|
},
|
|
},
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, newval, newb)
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/timeout": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
expErr := ConnectionErr(errors.Errorf("error doing TLS dial for %v:443: tls: DialWithDialer timed out", ch.getValue()))
|
|
baseClone := ch.clone()
|
|
baseClone.Error = expErr.ToACME()
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
newb, err := json.Marshal(newCh)
|
|
assert.FatalError(t, err)
|
|
|
|
srv, tlsDial := newTestTLSALPNServer(nil)
|
|
// srv.Start() - do not start server to cause timeout
|
|
|
|
return test{
|
|
srv: srv,
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: tlsDial,
|
|
},
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, string(newval), string(newb))
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/no-certificates": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
expErr := RejectedIdentifierErr(errors.Errorf("tls-alpn-01 challenge for %v resulted in no certificates", ch.getValue()))
|
|
baseClone := ch.clone()
|
|
baseClone.Error = expErr.ToACME()
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
newb, err := json.Marshal(newCh)
|
|
assert.FatalError(t, err)
|
|
|
|
return test{
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: func(network, addr string, config *tls.Config) (*tls.Conn, error) {
|
|
return tls.Client(&noopConn{}, config), nil
|
|
},
|
|
},
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, string(newval), string(newb))
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/no-names": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
expErr := RejectedIdentifierErr(errors.Errorf("incorrect certificate for tls-alpn-01 challenge: leaf certificate must contain a single DNS name, %v", ch.getValue()))
|
|
baseClone := ch.clone()
|
|
baseClone.Error = expErr.ToACME()
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
newb, err := json.Marshal(newCh)
|
|
assert.FatalError(t, err)
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.getToken(), jwk)
|
|
assert.FatalError(t, err)
|
|
expKeyAuthHash := sha256.Sum256([]byte(expKeyAuth))
|
|
|
|
cert, err := newTLSALPNValidationCert(expKeyAuthHash[:], false, true)
|
|
assert.FatalError(t, err)
|
|
|
|
srv, tlsDial := newTestTLSALPNServer(cert)
|
|
srv.Start()
|
|
|
|
return test{
|
|
srv: srv,
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: tlsDial,
|
|
},
|
|
jwk: jwk,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, string(newval), string(newb))
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/too-many-names": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
expErr := RejectedIdentifierErr(errors.Errorf("incorrect certificate for tls-alpn-01 challenge: leaf certificate must contain a single DNS name, %v", ch.getValue()))
|
|
baseClone := ch.clone()
|
|
baseClone.Error = expErr.ToACME()
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
newb, err := json.Marshal(newCh)
|
|
assert.FatalError(t, err)
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.getToken(), jwk)
|
|
assert.FatalError(t, err)
|
|
expKeyAuthHash := sha256.Sum256([]byte(expKeyAuth))
|
|
|
|
cert, err := newTLSALPNValidationCert(expKeyAuthHash[:], false, true, ch.getValue(), "other.internal")
|
|
assert.FatalError(t, err)
|
|
|
|
srv, tlsDial := newTestTLSALPNServer(cert)
|
|
srv.Start()
|
|
|
|
return test{
|
|
srv: srv,
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: tlsDial,
|
|
},
|
|
jwk: jwk,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, string(newval), string(newb))
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/wrong-name": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
expErr := RejectedIdentifierErr(errors.Errorf("incorrect certificate for tls-alpn-01 challenge: leaf certificate must contain a single DNS name, %v", ch.getValue()))
|
|
baseClone := ch.clone()
|
|
baseClone.Error = expErr.ToACME()
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
newb, err := json.Marshal(newCh)
|
|
assert.FatalError(t, err)
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.getToken(), jwk)
|
|
assert.FatalError(t, err)
|
|
expKeyAuthHash := sha256.Sum256([]byte(expKeyAuth))
|
|
|
|
cert, err := newTLSALPNValidationCert(expKeyAuthHash[:], false, true, "other.internal")
|
|
assert.FatalError(t, err)
|
|
|
|
srv, tlsDial := newTestTLSALPNServer(cert)
|
|
srv.Start()
|
|
|
|
return test{
|
|
srv: srv,
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: tlsDial,
|
|
},
|
|
jwk: jwk,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, string(newval), string(newb))
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/no-extension": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
expErr := RejectedIdentifierErr(errors.New("incorrect certificate for tls-alpn-01 challenge: missing acmeValidationV1 extension"))
|
|
baseClone := ch.clone()
|
|
baseClone.Error = expErr.ToACME()
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
newb, err := json.Marshal(newCh)
|
|
assert.FatalError(t, err)
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
cert, err := newTLSALPNValidationCert(nil, false, true, ch.getValue())
|
|
assert.FatalError(t, err)
|
|
|
|
srv, tlsDial := newTestTLSALPNServer(cert)
|
|
srv.Start()
|
|
|
|
return test{
|
|
srv: srv,
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: tlsDial,
|
|
},
|
|
jwk: jwk,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, string(newval), string(newb))
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/extension-not-critical": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
expErr := RejectedIdentifierErr(errors.New("incorrect certificate for tls-alpn-01 challenge: acmeValidationV1 extension not critical"))
|
|
baseClone := ch.clone()
|
|
baseClone.Error = expErr.ToACME()
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
newb, err := json.Marshal(newCh)
|
|
assert.FatalError(t, err)
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.getToken(), jwk)
|
|
assert.FatalError(t, err)
|
|
expKeyAuthHash := sha256.Sum256([]byte(expKeyAuth))
|
|
|
|
cert, err := newTLSALPNValidationCert(expKeyAuthHash[:], false, false, ch.getValue())
|
|
assert.FatalError(t, err)
|
|
|
|
srv, tlsDial := newTestTLSALPNServer(cert)
|
|
srv.Start()
|
|
|
|
return test{
|
|
srv: srv,
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: tlsDial,
|
|
},
|
|
jwk: jwk,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, string(newval), string(newb))
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/extension-malformed": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
expErr := RejectedIdentifierErr(errors.New("incorrect certificate for tls-alpn-01 challenge: malformed acmeValidationV1 extension value"))
|
|
baseClone := ch.clone()
|
|
baseClone.Error = expErr.ToACME()
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
newb, err := json.Marshal(newCh)
|
|
assert.FatalError(t, err)
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
cert, err := newTLSALPNValidationCert([]byte{1, 2, 3}, false, true, ch.getValue())
|
|
assert.FatalError(t, err)
|
|
|
|
srv, tlsDial := newTestTLSALPNServer(cert)
|
|
srv.Start()
|
|
|
|
return test{
|
|
srv: srv,
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: tlsDial,
|
|
},
|
|
jwk: jwk,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, string(newval), string(newb))
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/no-protocol": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expErr := RejectedIdentifierErr(errors.New("cannot negotiate ALPN acme-tls/1 protocol for tls-alpn-01 challenge"))
|
|
baseClone := ch.clone()
|
|
baseClone.Error = expErr.ToACME()
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
newb, err := json.Marshal(newCh)
|
|
assert.FatalError(t, err)
|
|
|
|
srv := httptest.NewTLSServer(nil)
|
|
|
|
return test{
|
|
srv: srv,
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: func(network, addr string, config *tls.Config) (*tls.Conn, error) {
|
|
return tls.DialWithDialer(&net.Dialer{Timeout: time.Second}, "tcp", srv.Listener.Addr().String(), config)
|
|
},
|
|
},
|
|
jwk: jwk,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, string(newval), string(newb))
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/mismatched-token": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.getToken(), jwk)
|
|
assert.FatalError(t, err)
|
|
expKeyAuthHash := sha256.Sum256([]byte(expKeyAuth))
|
|
incorrectTokenHash := sha256.Sum256([]byte("mismatched"))
|
|
|
|
expErr := RejectedIdentifierErr(errors.Errorf("incorrect certificate for tls-alpn-01 challenge: "+
|
|
"expected acmeValidationV1 extension value %s for this challenge but got %s",
|
|
hex.EncodeToString(expKeyAuthHash[:]), hex.EncodeToString(incorrectTokenHash[:])))
|
|
baseClone := ch.clone()
|
|
baseClone.Error = expErr.ToACME()
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
newb, err := json.Marshal(newCh)
|
|
assert.FatalError(t, err)
|
|
|
|
cert, err := newTLSALPNValidationCert(incorrectTokenHash[:], false, true, ch.getValue())
|
|
assert.FatalError(t, err)
|
|
|
|
srv, tlsDial := newTestTLSALPNServer(cert)
|
|
srv.Start()
|
|
|
|
return test{
|
|
srv: srv,
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: tlsDial,
|
|
},
|
|
jwk: jwk,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, string(newval), string(newb))
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok/obsolete-oid": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expErr := RejectedIdentifierErr(errors.New("incorrect certificate for tls-alpn-01 challenge: " +
|
|
"obsolete id-pe-acmeIdentifier in acmeValidationV1 extension"))
|
|
baseClone := ch.clone()
|
|
baseClone.Error = expErr.ToACME()
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
newb, err := json.Marshal(newCh)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.getToken(), jwk)
|
|
assert.FatalError(t, err)
|
|
expKeyAuthHash := sha256.Sum256([]byte(expKeyAuth))
|
|
|
|
cert, err := newTLSALPNValidationCert(expKeyAuthHash[:], true, true, ch.getValue())
|
|
assert.FatalError(t, err)
|
|
|
|
srv, tlsDial := newTestTLSALPNServer(cert)
|
|
srv.Start()
|
|
|
|
return test{
|
|
srv: srv,
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: tlsDial,
|
|
},
|
|
jwk: jwk,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, string(newval), string(newb))
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: ch,
|
|
}
|
|
},
|
|
"ok": func(t *testing.T) test {
|
|
ch, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
_ch, ok := ch.(*tlsALPN01Challenge)
|
|
assert.Fatal(t, ok)
|
|
_ch.baseChallenge.Error = MalformedErr(nil).ToACME()
|
|
oldb, err := json.Marshal(ch)
|
|
assert.FatalError(t, err)
|
|
|
|
baseClone := ch.clone()
|
|
baseClone.Status = StatusValid
|
|
baseClone.Error = nil
|
|
newCh := &tlsALPN01Challenge{baseClone}
|
|
|
|
jwk, err := jose.GenerateJWK("EC", "P-256", "ES256", "sig", "", 0)
|
|
assert.FatalError(t, err)
|
|
|
|
expKeyAuth, err := KeyAuthorization(ch.getToken(), jwk)
|
|
assert.FatalError(t, err)
|
|
expKeyAuthHash := sha256.Sum256([]byte(expKeyAuth))
|
|
|
|
cert, err := newTLSALPNValidationCert(expKeyAuthHash[:], false, true, ch.getValue())
|
|
assert.FatalError(t, err)
|
|
|
|
srv, tlsDial := newTestTLSALPNServer(cert)
|
|
srv.Start()
|
|
|
|
return test{
|
|
srv: srv,
|
|
ch: ch,
|
|
vo: validateOptions{
|
|
tlsDial: func(network, addr string, config *tls.Config) (conn *tls.Conn, err error) {
|
|
assert.Equals(t, network, "tcp")
|
|
assert.Equals(t, addr, net.JoinHostPort(newCh.getValue(), "443"))
|
|
assert.Equals(t, config.NextProtos, []string{"acme-tls/1"})
|
|
assert.Equals(t, config.ServerName, newCh.getValue())
|
|
assert.True(t, config.InsecureSkipVerify)
|
|
|
|
return tlsDial(network, addr, config)
|
|
},
|
|
},
|
|
jwk: jwk,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(ch.getID()))
|
|
assert.Equals(t, old, oldb)
|
|
|
|
alpnCh, err := unmarshalChallenge(newval)
|
|
assert.FatalError(t, err)
|
|
assert.Equals(t, alpnCh.getStatus(), StatusValid)
|
|
assert.True(t, alpnCh.getValidated().Before(time.Now().UTC().Add(time.Minute)))
|
|
assert.True(t, alpnCh.getValidated().After(time.Now().UTC().Add(-1*time.Second)))
|
|
|
|
baseClone.Validated = alpnCh.getValidated()
|
|
|
|
return nil, true, nil
|
|
},
|
|
},
|
|
res: newCh,
|
|
}
|
|
},
|
|
}
|
|
for name, run := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
tc := run(t)
|
|
|
|
if tc.srv != nil {
|
|
defer tc.srv.Close()
|
|
}
|
|
|
|
if ch, err := tc.ch.validate(tc.db, tc.jwk, tc.vo); err != nil {
|
|
if assert.NotNil(t, tc.err) {
|
|
ae, ok := err.(*Error)
|
|
assert.True(t, ok)
|
|
assert.HasPrefix(t, ae.Error(), tc.err.Error())
|
|
assert.Equals(t, ae.StatusCode(), tc.err.StatusCode())
|
|
assert.Equals(t, ae.Type, tc.err.Type)
|
|
}
|
|
} else {
|
|
if assert.Nil(t, tc.err) {
|
|
assert.Equals(t, tc.res.getID(), ch.getID())
|
|
assert.Equals(t, tc.res.getAccountID(), ch.getAccountID())
|
|
assert.Equals(t, tc.res.getAuthzID(), ch.getAuthzID())
|
|
assert.Equals(t, tc.res.getStatus(), ch.getStatus())
|
|
assert.Equals(t, tc.res.getToken(), ch.getToken())
|
|
assert.Equals(t, tc.res.getCreated(), ch.getCreated())
|
|
assert.Equals(t, tc.res.getValidated(), ch.getValidated())
|
|
assert.Equals(t, tc.res.getError(), ch.getError())
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func newTestTLSALPNServer(validationCert *tls.Certificate) (*httptest.Server, tlsDialer) {
|
|
srv := httptest.NewUnstartedServer(http.NewServeMux())
|
|
|
|
srv.Config.TLSNextProto = map[string]func(*http.Server, *tls.Conn, http.Handler){
|
|
"acme-tls/1": func(_ *http.Server, conn *tls.Conn, _ http.Handler) {
|
|
// no-op
|
|
},
|
|
"http/1.1": func(_ *http.Server, conn *tls.Conn, _ http.Handler) {
|
|
panic("unexpected http/1.1 next proto")
|
|
},
|
|
}
|
|
|
|
srv.TLS = &tls.Config{
|
|
GetCertificate: func(hello *tls.ClientHelloInfo) (*tls.Certificate, error) {
|
|
if len(hello.SupportedProtos) == 1 && hello.SupportedProtos[0] == "acme-tls/1" {
|
|
return validationCert, nil
|
|
}
|
|
return nil, nil
|
|
},
|
|
NextProtos: []string{
|
|
"acme-tls/1",
|
|
"http/1.1",
|
|
},
|
|
}
|
|
|
|
srv.Listener = tls.NewListener(srv.Listener, srv.TLS)
|
|
//srv.Config.ErrorLog = log.New(ioutil.Discard, "", 0) // hush
|
|
|
|
return srv, func(network, addr string, config *tls.Config) (conn *tls.Conn, err error) {
|
|
return tls.DialWithDialer(&net.Dialer{Timeout: time.Second}, "tcp", srv.Listener.Addr().String(), config)
|
|
}
|
|
}
|
|
|
|
// noopConn is a mock net.Conn that does nothing.
|
|
type noopConn struct{}
|
|
|
|
func (c *noopConn) Read(_ []byte) (n int, err error) { return 0, io.EOF }
|
|
func (c *noopConn) Write(_ []byte) (n int, err error) { return 0, io.EOF }
|
|
func (c *noopConn) Close() error { return nil }
|
|
func (c *noopConn) LocalAddr() net.Addr { return &net.IPAddr{IP: net.IPv4zero, Zone: ""} }
|
|
func (c *noopConn) RemoteAddr() net.Addr { return &net.IPAddr{IP: net.IPv4zero, Zone: ""} }
|
|
func (c *noopConn) SetDeadline(t time.Time) error { return nil }
|
|
func (c *noopConn) SetReadDeadline(t time.Time) error { return nil }
|
|
func (c *noopConn) SetWriteDeadline(t time.Time) error { return nil }
|
|
|
|
func newTLSALPNValidationCert(keyAuthHash []byte, obsoleteOID, critical bool, names ...string) (*tls.Certificate, error) {
|
|
privateKey, err := rsa.GenerateKey(rand.Reader, 2048)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
certTemplate := &x509.Certificate{
|
|
SerialNumber: big.NewInt(1337),
|
|
Subject: pkix.Name{
|
|
Organization: []string{"Test"},
|
|
},
|
|
NotBefore: time.Now(),
|
|
NotAfter: time.Now().AddDate(0, 0, 1),
|
|
KeyUsage: x509.KeyUsageDigitalSignature | x509.KeyUsageCertSign,
|
|
ExtKeyUsage: []x509.ExtKeyUsage{x509.ExtKeyUsageServerAuth},
|
|
BasicConstraintsValid: true,
|
|
DNSNames: names,
|
|
}
|
|
|
|
if keyAuthHash != nil {
|
|
oid := asn1.ObjectIdentifier{1, 3, 6, 1, 5, 5, 7, 1, 31}
|
|
if obsoleteOID {
|
|
oid = asn1.ObjectIdentifier{1, 3, 6, 1, 5, 5, 7, 1, 30, 1}
|
|
}
|
|
|
|
keyAuthHashEnc, _ := asn1.Marshal(keyAuthHash[:])
|
|
|
|
certTemplate.ExtraExtensions = []pkix.Extension{
|
|
{
|
|
Id: oid,
|
|
Critical: critical,
|
|
Value: keyAuthHashEnc,
|
|
},
|
|
}
|
|
}
|
|
|
|
cert, err := x509.CreateCertificate(rand.Reader, certTemplate, certTemplate, privateKey.Public(), privateKey)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &tls.Certificate{
|
|
PrivateKey: privateKey,
|
|
Certificate: [][]byte{cert},
|
|
}, nil
|
|
}
|
|
|
|
/*
|
|
var testOps = ChallengeOptions{
|
|
AccountID: "accID",
|
|
AuthzID: "authzID",
|
|
Identifier: Identifier{
|
|
Type: "", // will get set correctly depending on the "new.." method.
|
|
Value: "zap.internal",
|
|
},
|
|
}
|
|
|
|
func newDNSCh() (Challenge, error) {
|
|
mockdb := &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
return []byte("foo"), true, nil
|
|
},
|
|
}
|
|
return newDNS01Challenge(mockdb, testOps)
|
|
}
|
|
|
|
func newTLSALPNCh() (Challenge, error) {
|
|
mockdb := &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
return []byte("foo"), true, nil
|
|
},
|
|
}
|
|
return newTLSALPN01Challenge(mockdb, testOps)
|
|
}
|
|
|
|
func newHTTPCh() (Challenge, error) {
|
|
mockdb := &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
return []byte("foo"), true, nil
|
|
},
|
|
}
|
|
return newHTTP01Challenge(mockdb, testOps)
|
|
}
|
|
|
|
func newHTTPChWithServer(host string) (Challenge, error) {
|
|
mockdb := &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
return []byte("foo"), true, nil
|
|
},
|
|
}
|
|
return newHTTP01Challenge(mockdb, ChallengeOptions{
|
|
AccountID: "accID",
|
|
AuthzID: "authzID",
|
|
Identifier: Identifier{
|
|
Type: "", // will get set correctly depending on the "new.." method.
|
|
Value: host,
|
|
},
|
|
})
|
|
}
|
|
|
|
func TestNewHTTP01Challenge(t *testing.T) {
|
|
ops := ChallengeOptions{
|
|
AccountID: "accID",
|
|
AuthzID: "authzID",
|
|
Identifier: Identifier{
|
|
Type: "http",
|
|
Value: "zap.internal",
|
|
},
|
|
}
|
|
type test struct {
|
|
ops ChallengeOptions
|
|
db nosql.DB
|
|
err *Error
|
|
}
|
|
tests := map[string]test{
|
|
"fail/store-error": {
|
|
ops: ops,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
return nil, false, errors.New("force")
|
|
},
|
|
},
|
|
err: ServerInternalErr(errors.New("error saving acme challenge: force")),
|
|
},
|
|
"ok": {
|
|
ops: ops,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
return []byte("foo"), true, nil
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for name, tc := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
ch, err := newHTTP01Challenge(tc.db, tc.ops)
|
|
if err != nil {
|
|
if assert.NotNil(t, tc.err) {
|
|
ae, ok := err.(*Error)
|
|
assert.True(t, ok)
|
|
assert.HasPrefix(t, ae.Error(), tc.err.Error())
|
|
assert.Equals(t, ae.StatusCode(), tc.err.StatusCode())
|
|
assert.Equals(t, ae.Type, tc.err.Type)
|
|
}
|
|
} else {
|
|
if assert.Nil(t, tc.err) {
|
|
assert.Equals(t, ch.getAccountID(), ops.AccountID)
|
|
assert.Equals(t, ch.getAuthzID(), ops.AuthzID)
|
|
assert.Equals(t, ch.getType(), "http-01")
|
|
assert.Equals(t, ch.getValue(), "zap.internal")
|
|
assert.Equals(t, ch.getStatus(), StatusPending)
|
|
|
|
assert.True(t, ch.getValidated().IsZero())
|
|
assert.True(t, ch.getCreated().Before(time.Now().UTC().Add(time.Minute)))
|
|
assert.True(t, ch.getCreated().After(time.Now().UTC().Add(-1*time.Minute)))
|
|
|
|
assert.True(t, ch.getID() != "")
|
|
assert.True(t, ch.getToken() != "")
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestNewTLSALPN01Challenge(t *testing.T) {
|
|
ops := ChallengeOptions{
|
|
AccountID: "accID",
|
|
AuthzID: "authzID",
|
|
Identifier: Identifier{
|
|
Type: "http",
|
|
Value: "zap.internal",
|
|
},
|
|
}
|
|
type test struct {
|
|
ops ChallengeOptions
|
|
db nosql.DB
|
|
err *Error
|
|
}
|
|
tests := map[string]test{
|
|
"fail/store-error": {
|
|
ops: ops,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
return nil, false, errors.New("force")
|
|
},
|
|
},
|
|
err: ServerInternalErr(errors.New("error saving acme challenge: force")),
|
|
},
|
|
"ok": {
|
|
ops: ops,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
return []byte("foo"), true, nil
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for name, tc := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
ch, err := newTLSALPN01Challenge(tc.db, tc.ops)
|
|
if err != nil {
|
|
if assert.NotNil(t, tc.err) {
|
|
ae, ok := err.(*Error)
|
|
assert.True(t, ok)
|
|
assert.HasPrefix(t, ae.Error(), tc.err.Error())
|
|
assert.Equals(t, ae.StatusCode(), tc.err.StatusCode())
|
|
assert.Equals(t, ae.Type, tc.err.Type)
|
|
}
|
|
} else {
|
|
if assert.Nil(t, tc.err) {
|
|
assert.Equals(t, ch.getAccountID(), ops.AccountID)
|
|
assert.Equals(t, ch.getAuthzID(), ops.AuthzID)
|
|
assert.Equals(t, ch.getType(), "tls-alpn-01")
|
|
assert.Equals(t, ch.getValue(), "zap.internal")
|
|
assert.Equals(t, ch.getStatus(), StatusPending)
|
|
|
|
assert.True(t, ch.getValidated().IsZero())
|
|
assert.True(t, ch.getCreated().Before(time.Now().UTC().Add(time.Minute)))
|
|
assert.True(t, ch.getCreated().After(time.Now().UTC().Add(-1*time.Minute)))
|
|
|
|
assert.True(t, ch.getID() != "")
|
|
assert.True(t, ch.getToken() != "")
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestNewDNS01Challenge(t *testing.T) {
|
|
ops := ChallengeOptions{
|
|
AccountID: "accID",
|
|
AuthzID: "authzID",
|
|
Identifier: Identifier{
|
|
Type: "dns",
|
|
Value: "zap.internal",
|
|
},
|
|
}
|
|
type test struct {
|
|
ops ChallengeOptions
|
|
db nosql.DB
|
|
err *Error
|
|
}
|
|
tests := map[string]test{
|
|
"fail/store-error": {
|
|
ops: ops,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
return nil, false, errors.New("force")
|
|
},
|
|
},
|
|
err: ServerInternalErr(errors.New("error saving acme challenge: force")),
|
|
},
|
|
"ok": {
|
|
ops: ops,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
return []byte("foo"), true, nil
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for name, tc := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
ch, err := newDNS01Challenge(tc.db, tc.ops)
|
|
if err != nil {
|
|
if assert.NotNil(t, tc.err) {
|
|
ae, ok := err.(*Error)
|
|
assert.True(t, ok)
|
|
assert.HasPrefix(t, ae.Error(), tc.err.Error())
|
|
assert.Equals(t, ae.StatusCode(), tc.err.StatusCode())
|
|
assert.Equals(t, ae.Type, tc.err.Type)
|
|
}
|
|
} else {
|
|
if assert.Nil(t, tc.err) {
|
|
assert.Equals(t, ch.getAccountID(), ops.AccountID)
|
|
assert.Equals(t, ch.getAuthzID(), ops.AuthzID)
|
|
assert.Equals(t, ch.getType(), "dns-01")
|
|
assert.Equals(t, ch.getValue(), "zap.internal")
|
|
assert.Equals(t, ch.getStatus(), StatusPending)
|
|
|
|
assert.True(t, ch.getValidated().IsZero())
|
|
assert.True(t, ch.getCreated().Before(time.Now().UTC().Add(time.Minute)))
|
|
assert.True(t, ch.getCreated().After(time.Now().UTC().Add(-1*time.Minute)))
|
|
|
|
assert.True(t, ch.getID() != "")
|
|
assert.True(t, ch.getToken() != "")
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestChallengeToACME(t *testing.T) {
|
|
dir := newDirectory("ca.smallstep.com", "acme")
|
|
|
|
httpCh, err := newHTTPCh()
|
|
assert.FatalError(t, err)
|
|
_httpCh, ok := httpCh.(*http01Challenge)
|
|
assert.Fatal(t, ok)
|
|
_httpCh.baseChallenge.Validated = clock.Now()
|
|
dnsCh, err := newDNSCh()
|
|
assert.FatalError(t, err)
|
|
tlsALPNCh, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
|
|
prov := newProv()
|
|
provName := url.PathEscape(prov.GetName())
|
|
baseURL := &url.URL{Scheme: "https", Host: "test.ca.smallstep.com"}
|
|
ctx := context.WithValue(context.Background(), ProvisionerContextKey, prov)
|
|
ctx = context.WithValue(ctx, BaseURLContextKey, baseURL)
|
|
tests := map[string]challenge{
|
|
"dns": dnsCh,
|
|
"http": httpCh,
|
|
"tls-alpn": tlsALPNCh,
|
|
}
|
|
for name, ch := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
ach, err := ch.toACME(ctx, nil, dir)
|
|
assert.FatalError(t, err)
|
|
|
|
assert.Equals(t, ach.Type, ch.getType())
|
|
assert.Equals(t, ach.Status, ch.getStatus())
|
|
assert.Equals(t, ach.Token, ch.getToken())
|
|
assert.Equals(t, ach.URL,
|
|
fmt.Sprintf("%s/acme/%s/challenge/%s",
|
|
baseURL.String(), provName, ch.getID()))
|
|
assert.Equals(t, ach.ID, ch.getID())
|
|
assert.Equals(t, ach.AuthzID, ch.getAuthzID())
|
|
|
|
if ach.Type == "http-01" {
|
|
v, err := time.Parse(time.RFC3339, ach.Validated)
|
|
assert.FatalError(t, err)
|
|
assert.Equals(t, v.String(), _httpCh.baseChallenge.Validated.String())
|
|
} else {
|
|
assert.Equals(t, ach.Validated, "")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestChallengeSave(t *testing.T) {
|
|
type test struct {
|
|
ch challenge
|
|
old challenge
|
|
db nosql.DB
|
|
err *Error
|
|
}
|
|
tests := map[string]func(t *testing.T) test{
|
|
"fail/old-nil/swap-error": func(t *testing.T) test {
|
|
httpCh, err := newHTTPCh()
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: httpCh,
|
|
old: nil,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
return nil, false, errors.New("force")
|
|
},
|
|
},
|
|
err: ServerInternalErr(errors.New("error saving acme challenge: force")),
|
|
}
|
|
},
|
|
"fail/old-nil/swap-false": func(t *testing.T) test {
|
|
httpCh, err := newHTTPCh()
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: httpCh,
|
|
old: nil,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
return []byte("foo"), false, nil
|
|
},
|
|
},
|
|
err: ServerInternalErr(errors.New("error saving acme challenge; acme challenge has changed since last read")),
|
|
}
|
|
},
|
|
"ok/old-nil": func(t *testing.T) test {
|
|
httpCh, err := newHTTPCh()
|
|
assert.FatalError(t, err)
|
|
b, err := json.Marshal(httpCh)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: httpCh,
|
|
old: nil,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, old, nil)
|
|
assert.Equals(t, b, newval)
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, []byte(httpCh.getID()), key)
|
|
return []byte("foo"), true, nil
|
|
},
|
|
},
|
|
}
|
|
},
|
|
"ok/old-not-nil": func(t *testing.T) test {
|
|
oldHTTPCh, err := newHTTPCh()
|
|
assert.FatalError(t, err)
|
|
httpCh, err := newHTTPCh()
|
|
assert.FatalError(t, err)
|
|
|
|
oldb, err := json.Marshal(oldHTTPCh)
|
|
assert.FatalError(t, err)
|
|
b, err := json.Marshal(httpCh)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: httpCh,
|
|
old: oldHTTPCh,
|
|
db: &db.MockNoSQLDB{
|
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
assert.Equals(t, old, oldb)
|
|
assert.Equals(t, b, newval)
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, []byte(httpCh.getID()), key)
|
|
return []byte("foo"), true, nil
|
|
},
|
|
},
|
|
}
|
|
},
|
|
}
|
|
for name, run := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
tc := run(t)
|
|
if err := tc.ch.save(tc.db, tc.old); err != nil {
|
|
if assert.NotNil(t, tc.err) {
|
|
ae, ok := err.(*Error)
|
|
assert.True(t, ok)
|
|
assert.HasPrefix(t, ae.Error(), tc.err.Error())
|
|
assert.Equals(t, ae.StatusCode(), tc.err.StatusCode())
|
|
assert.Equals(t, ae.Type, tc.err.Type)
|
|
}
|
|
} else {
|
|
assert.Nil(t, tc.err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestChallengeClone(t *testing.T) {
|
|
ch, err := newHTTPCh()
|
|
assert.FatalError(t, err)
|
|
|
|
clone := ch.clone()
|
|
|
|
assert.Equals(t, clone.getID(), ch.getID())
|
|
assert.Equals(t, clone.getAccountID(), ch.getAccountID())
|
|
assert.Equals(t, clone.getAuthzID(), ch.getAuthzID())
|
|
assert.Equals(t, clone.getStatus(), ch.getStatus())
|
|
assert.Equals(t, clone.getToken(), ch.getToken())
|
|
assert.Equals(t, clone.getCreated(), ch.getCreated())
|
|
assert.Equals(t, clone.getValidated(), ch.getValidated())
|
|
|
|
clone.Status = StatusValid
|
|
|
|
assert.NotEquals(t, clone.getStatus(), ch.getStatus())
|
|
}
|
|
|
|
func TestChallengeUnmarshal(t *testing.T) {
|
|
type test struct {
|
|
ch challenge
|
|
chb []byte
|
|
err *Error
|
|
}
|
|
tests := map[string]func(t *testing.T) test{
|
|
"fail/nil": func(t *testing.T) test {
|
|
return test{
|
|
chb: nil,
|
|
err: ServerInternalErr(errors.New("error unmarshaling challenge type: unexpected end of JSON input")),
|
|
}
|
|
},
|
|
"fail/unexpected-type-http": func(t *testing.T) test {
|
|
httpCh, err := newHTTPCh()
|
|
assert.FatalError(t, err)
|
|
_httpCh, ok := httpCh.(*http01Challenge)
|
|
assert.Fatal(t, ok)
|
|
_httpCh.baseChallenge.Type = "foo"
|
|
b, err := json.Marshal(httpCh)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
chb: b,
|
|
err: ServerInternalErr(errors.New("unexpected challenge type foo")),
|
|
}
|
|
},
|
|
"fail/unexpected-type-alpn": func(t *testing.T) test {
|
|
tlsALPNCh, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
_tlsALPNCh, ok := tlsALPNCh.(*tlsALPN01Challenge)
|
|
assert.Fatal(t, ok)
|
|
_tlsALPNCh.baseChallenge.Type = "foo"
|
|
b, err := json.Marshal(tlsALPNCh)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
chb: b,
|
|
err: ServerInternalErr(errors.New("unexpected challenge type foo")),
|
|
}
|
|
},
|
|
"fail/unexpected-type-dns": func(t *testing.T) test {
|
|
dnsCh, err := newDNSCh()
|
|
assert.FatalError(t, err)
|
|
_dnsCh, ok := dnsCh.(*dns01Challenge)
|
|
assert.Fatal(t, ok)
|
|
_dnsCh.baseChallenge.Type = "foo"
|
|
b, err := json.Marshal(dnsCh)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
chb: b,
|
|
err: ServerInternalErr(errors.New("unexpected challenge type foo")),
|
|
}
|
|
},
|
|
"ok/dns": func(t *testing.T) test {
|
|
dnsCh, err := newDNSCh()
|
|
assert.FatalError(t, err)
|
|
b, err := json.Marshal(dnsCh)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: dnsCh,
|
|
chb: b,
|
|
}
|
|
},
|
|
"ok/http": func(t *testing.T) test {
|
|
httpCh, err := newHTTPCh()
|
|
assert.FatalError(t, err)
|
|
b, err := json.Marshal(httpCh)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: httpCh,
|
|
chb: b,
|
|
}
|
|
},
|
|
"ok/alpn": func(t *testing.T) test {
|
|
tlsALPNCh, err := newTLSALPNCh()
|
|
assert.FatalError(t, err)
|
|
b, err := json.Marshal(tlsALPNCh)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: tlsALPNCh,
|
|
chb: b,
|
|
}
|
|
},
|
|
"ok/err": func(t *testing.T) test {
|
|
httpCh, err := newHTTPCh()
|
|
assert.FatalError(t, err)
|
|
_httpCh, ok := httpCh.(*http01Challenge)
|
|
assert.Fatal(t, ok)
|
|
_httpCh.baseChallenge.Error = ServerInternalErr(errors.New("force")).ToACME()
|
|
b, err := json.Marshal(httpCh)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: httpCh,
|
|
chb: b,
|
|
}
|
|
},
|
|
}
|
|
for name, run := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
tc := run(t)
|
|
if ch, err := unmarshalChallenge(tc.chb); err != nil {
|
|
if assert.NotNil(t, tc.err) {
|
|
ae, ok := err.(*Error)
|
|
assert.True(t, ok)
|
|
assert.HasPrefix(t, ae.Error(), tc.err.Error())
|
|
assert.Equals(t, ae.StatusCode(), tc.err.StatusCode())
|
|
assert.Equals(t, ae.Type, tc.err.Type)
|
|
}
|
|
} else {
|
|
if assert.Nil(t, tc.err) {
|
|
assert.Equals(t, tc.ch.getID(), ch.getID())
|
|
assert.Equals(t, tc.ch.getAccountID(), ch.getAccountID())
|
|
assert.Equals(t, tc.ch.getAuthzID(), ch.getAuthzID())
|
|
assert.Equals(t, tc.ch.getStatus(), ch.getStatus())
|
|
assert.Equals(t, tc.ch.getToken(), ch.getToken())
|
|
assert.Equals(t, tc.ch.getCreated(), ch.getCreated())
|
|
assert.Equals(t, tc.ch.getValidated(), ch.getValidated())
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
func TestGetChallenge(t *testing.T) {
|
|
type test struct {
|
|
id string
|
|
db nosql.DB
|
|
ch challenge
|
|
err *Error
|
|
}
|
|
tests := map[string]func(t *testing.T) test{
|
|
"fail/not-found": func(t *testing.T) test {
|
|
dnsCh, err := newDNSCh()
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: dnsCh,
|
|
id: dnsCh.getID(),
|
|
db: &db.MockNoSQLDB{
|
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
return nil, database.ErrNotFound
|
|
},
|
|
},
|
|
err: MalformedErr(errors.Errorf("challenge %s not found: not found", dnsCh.getID())),
|
|
}
|
|
},
|
|
"fail/db-error": func(t *testing.T) test {
|
|
dnsCh, err := newDNSCh()
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: dnsCh,
|
|
id: dnsCh.getID(),
|
|
db: &db.MockNoSQLDB{
|
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
return nil, errors.New("force")
|
|
},
|
|
},
|
|
err: ServerInternalErr(errors.Errorf("error loading challenge %s: force", dnsCh.getID())),
|
|
}
|
|
},
|
|
"fail/unmarshal-error": func(t *testing.T) test {
|
|
dnsCh, err := newDNSCh()
|
|
assert.FatalError(t, err)
|
|
_dnsCh, ok := dnsCh.(*dns01Challenge)
|
|
assert.Fatal(t, ok)
|
|
_dnsCh.baseChallenge.Type = "foo"
|
|
b, err := json.Marshal(dnsCh)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: dnsCh,
|
|
id: dnsCh.getID(),
|
|
db: &db.MockNoSQLDB{
|
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(dnsCh.getID()))
|
|
return b, nil
|
|
},
|
|
},
|
|
err: ServerInternalErr(errors.New("unexpected challenge type foo")),
|
|
}
|
|
},
|
|
"ok": func(t *testing.T) test {
|
|
dnsCh, err := newDNSCh()
|
|
assert.FatalError(t, err)
|
|
b, err := json.Marshal(dnsCh)
|
|
assert.FatalError(t, err)
|
|
return test{
|
|
ch: dnsCh,
|
|
id: dnsCh.getID(),
|
|
db: &db.MockNoSQLDB{
|
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
assert.Equals(t, bucket, challengeTable)
|
|
assert.Equals(t, key, []byte(dnsCh.getID()))
|
|
return b, nil
|
|
},
|
|
},
|
|
}
|
|
},
|
|
}
|
|
for name, run := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
tc := run(t)
|
|
if ch, err := getChallenge(tc.db, tc.id); err != nil {
|
|
if assert.NotNil(t, tc.err) {
|
|
ae, ok := err.(*Error)
|
|
assert.True(t, ok)
|
|
assert.HasPrefix(t, ae.Error(), tc.err.Error())
|
|
assert.Equals(t, ae.StatusCode(), tc.err.StatusCode())
|
|
assert.Equals(t, ae.Type, tc.err.Type)
|
|
}
|
|
} else {
|
|
if assert.Nil(t, tc.err) {
|
|
assert.Equals(t, tc.ch.getID(), ch.getID())
|
|
assert.Equals(t, tc.ch.getAccountID(), ch.getAccountID())
|
|
assert.Equals(t, tc.ch.getAuthzID(), ch.getAuthzID())
|
|
assert.Equals(t, tc.ch.getStatus(), ch.getStatus())
|
|
assert.Equals(t, tc.ch.getToken(), ch.getToken())
|
|
assert.Equals(t, tc.ch.getCreated(), ch.getCreated())
|
|
assert.Equals(t, tc.ch.getValidated(), ch.getValidated())
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
*/
|