139 lines
4.1 KiB
Go
139 lines
4.1 KiB
Go
/*
|
|
Copyright 2014 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package cert
|
|
|
|
import (
|
|
"crypto/rsa"
|
|
"crypto/x509"
|
|
"encoding/pem"
|
|
"errors"
|
|
"fmt"
|
|
)
|
|
|
|
const (
|
|
// ECPrivateKeyBlockType is a possible value for pem.Block.Type.
|
|
ECPrivateKeyBlockType = "EC PRIVATE KEY"
|
|
// RSAPrivateKeyBlockType is a possible value for pem.Block.Type.
|
|
RSAPrivateKeyBlockType = "RSA PRIVATE KEY"
|
|
// CertificateBlockType is a possible value for pem.Block.Type.
|
|
CertificateBlockType = "CERTIFICATE"
|
|
// CertificateRequestBlockType is a possible value for pem.Block.Type.
|
|
CertificateRequestBlockType = "CERTIFICATE REQUEST"
|
|
// PrivateKeyBlockType is a possible value for pem.Block.Type.
|
|
PrivateKeyBlockType = "PRIVATE KEY"
|
|
// PublicKeyBlockType is a possible value for pem.Block.Type.
|
|
PublicKeyBlockType = "PUBLIC KEY"
|
|
)
|
|
|
|
// EncodePublicKeyPEM returns PEM-endcode public data
|
|
func EncodePublicKeyPEM(key *rsa.PublicKey) ([]byte, error) {
|
|
der, err := x509.MarshalPKIXPublicKey(key)
|
|
if err != nil {
|
|
return []byte{}, err
|
|
}
|
|
block := pem.Block{
|
|
Type: PublicKeyBlockType,
|
|
Bytes: der,
|
|
}
|
|
return pem.EncodeToMemory(&block), nil
|
|
}
|
|
|
|
// EncodePrivateKeyPEM returns PEM-encoded private key data
|
|
func EncodePrivateKeyPEM(key *rsa.PrivateKey) []byte {
|
|
block := pem.Block{
|
|
Type: RSAPrivateKeyBlockType,
|
|
Bytes: x509.MarshalPKCS1PrivateKey(key),
|
|
}
|
|
return pem.EncodeToMemory(&block)
|
|
}
|
|
|
|
// EncodeCertPEM returns PEM-endcoded certificate data
|
|
func EncodeCertPEM(cert *x509.Certificate) []byte {
|
|
block := pem.Block{
|
|
Type: CertificateBlockType,
|
|
Bytes: cert.Raw,
|
|
}
|
|
return pem.EncodeToMemory(&block)
|
|
}
|
|
|
|
// ParsePrivateKeyPEM returns a private key parsed from a PEM block in the supplied data.
|
|
// Recognizes PEM blocks for "EC PRIVATE KEY", "RSA PRIVATE KEY", or "PRIVATE KEY"
|
|
func ParsePrivateKeyPEM(keyData []byte) (interface{}, error) {
|
|
var privateKeyPemBlock *pem.Block
|
|
for {
|
|
privateKeyPemBlock, keyData = pem.Decode(keyData)
|
|
if privateKeyPemBlock == nil {
|
|
break
|
|
}
|
|
|
|
switch privateKeyPemBlock.Type {
|
|
case ECPrivateKeyBlockType:
|
|
// ECDSA Private Key in ASN.1 format
|
|
if key, err := x509.ParseECPrivateKey(privateKeyPemBlock.Bytes); err == nil {
|
|
return key, nil
|
|
}
|
|
case RSAPrivateKeyBlockType:
|
|
// RSA Private Key in PKCS#1 format
|
|
if key, err := x509.ParsePKCS1PrivateKey(privateKeyPemBlock.Bytes); err == nil {
|
|
return key, nil
|
|
}
|
|
case PrivateKeyBlockType:
|
|
// RSA or ECDSA Private Key in unencrypted PKCS#8 format
|
|
if key, err := x509.ParsePKCS8PrivateKey(privateKeyPemBlock.Bytes); err == nil {
|
|
return key, nil
|
|
}
|
|
}
|
|
|
|
// tolerate non-key PEM blocks for compatibility with things like "EC PARAMETERS" blocks
|
|
// originally, only the first PEM block was parsed and expected to be a key block
|
|
}
|
|
|
|
// we read all the PEM blocks and didn't recognize one
|
|
return nil, fmt.Errorf("data does not contain a valid RSA or ECDSA private key")
|
|
}
|
|
|
|
// ParseCertsPEM returns the x509.Certificates contained in the given PEM-encoded byte array
|
|
// Returns an error if a certificate could not be parsed, or if the data does not contain any certificates
|
|
func ParseCertsPEM(pemCerts []byte) ([]*x509.Certificate, error) {
|
|
ok := false
|
|
certs := []*x509.Certificate{}
|
|
for len(pemCerts) > 0 {
|
|
var block *pem.Block
|
|
block, pemCerts = pem.Decode(pemCerts)
|
|
if block == nil {
|
|
break
|
|
}
|
|
// Only use PEM "CERTIFICATE" blocks without extra headers
|
|
if block.Type != CertificateBlockType || len(block.Headers) != 0 {
|
|
continue
|
|
}
|
|
|
|
cert, err := x509.ParseCertificate(block.Bytes)
|
|
if err != nil {
|
|
return certs, err
|
|
}
|
|
|
|
certs = append(certs, cert)
|
|
ok = true
|
|
}
|
|
|
|
if !ok {
|
|
return certs, errors.New("could not read any certificates")
|
|
}
|
|
return certs, nil
|
|
}
|