go-sample-webpage/vendor/github.com/lestrrat-go/jwx/jwk/certchain.go

86 lines
1.8 KiB
Go
Raw Normal View History

2021-11-04 01:14:51 +00:00
package jwk
import (
"crypto/x509"
"github.com/lestrrat-go/jwx/internal/json"
"github.com/lestrrat-go/jwx/internal/base64"
"github.com/pkg/errors"
)
func (c CertificateChain) MarshalJSON() ([]byte, error) {
certs := c.Get()
encoded := make([]string, len(certs))
for i := 0; i < len(certs); i++ {
encoded[i] = base64.EncodeToStringStd(certs[i].Raw)
}
return json.Marshal(encoded)
}
func (c *CertificateChain) UnmarshalJSON(buf []byte) error {
var list []string
if err := json.Unmarshal(buf, &list); err != nil {
return errors.Wrap(err, `failed to unmarshal JSON into []string`)
}
var tmp CertificateChain
if err := tmp.Accept(list); err != nil {
return err
}
*c = tmp
return nil
}
func (c CertificateChain) Get() []*x509.Certificate {
return c.certs
}
func (c *CertificateChain) Accept(v interface{}) error {
var list []string
switch x := v.(type) {
case string:
list = []string{x}
case []interface{}:
list = make([]string, len(x))
for i, e := range x {
if es, ok := e.(string); ok {
list[i] = es
continue
}
return errors.Errorf(`invalid list element type: expected string, got %T at element %d`, e, i)
}
case []string:
list = x
case CertificateChain:
certs := make([]*x509.Certificate, len(x.certs))
copy(certs, x.certs)
*c = CertificateChain{
certs: certs,
}
return nil
default:
return errors.Errorf(`invalid type for CertificateChain: %T`, v)
}
certs := make([]*x509.Certificate, len(list))
for i, e := range list {
buf, err := base64.DecodeString(e)
if err != nil {
return errors.Wrap(err, `failed to base64 decode list element`)
}
cert, err := x509.ParseCertificate(buf)
if err != nil {
return errors.Wrap(err, `failed to parse certificate`)
}
certs[i] = cert
}
*c = CertificateChain{
certs: certs,
}
return nil
}