A common library that interfaces with VMware NSX.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
vmware-nsxlib/vmware_nsxlib/v3/client_cert.py

362 lines
12 KiB

# Copyright 2016 VMware, Inc.
# All Rights Reserved
#
# 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.
import datetime
from time import time
import uuid
from OpenSSL import crypto
from oslo_log import log
from vmware_nsxlib._i18n import _
from vmware_nsxlib.v3 import exceptions as nsxlib_exceptions
LOG = log.getLogger(__name__)
CERT_SUBJECT_COUNTRY = 'country'
CERT_SUBJECT_STATE = 'state'
CERT_SUBJECT_ORG = 'organization'
CERT_SUBJECT_UNIT = 'unit'
CERT_SUBJECT_HOST = 'hostname'
def validate_cert_params(key_size, valid_for_days,
signature_alg, subject):
"""Validate parameters for certificate"""
expected_key_sizes = (2048, 4096)
if key_size not in expected_key_sizes:
raise nsxlib_exceptions.NsxLibInvalidInput(
error_message=_('Invalid key size %(value)d'
'(must be one of %(list)s)') %
{'value': key_size,
'list': expected_key_sizes})
expected_signature_algs = ('sha224', 'sha256')
if signature_alg not in expected_signature_algs:
raise nsxlib_exceptions.NsxLibInvalidInput(
error_message=_('Invalid signature algorithm %(value)s'
'(must be one of %(list)s)') %
{'value': signature_alg,
'list': expected_signature_algs})
if (CERT_SUBJECT_COUNTRY in subject and
(len(subject[CERT_SUBJECT_COUNTRY]) != 2)):
raise nsxlib_exceptions.NsxLibInvalidInput(
error_message=_('Invalid country %s: '
'must be exactly 2 characters') %
subject[CERT_SUBJECT_COUNTRY])
# values defined in rfc5280
max_len_constraints = {CERT_SUBJECT_STATE: 128,
CERT_SUBJECT_ORG: 64,
CERT_SUBJECT_UNIT: 64,
CERT_SUBJECT_HOST: 64}
for field, max_len in max_len_constraints.items():
if field in subject and (len(subject[field]) > max_len):
raise nsxlib_exceptions.NsxLibInvalidInput(
error_message=_('Invalid %(field)s [%(value)s]: '
'must not exceed %(max)d characters') %
{'field': field,
'value': subject[field],
'max': max_len})
def generate_self_signed_cert_pair(key_size, valid_for_days,
signature_alg, subject):
"""Generate self signed certificate and key pair"""
validate_cert_params(key_size, valid_for_days,
signature_alg, subject)
# generate key pair
key = crypto.PKey()
key.generate_key(crypto.TYPE_RSA, key_size)
# generate certificate
cert = crypto.X509()
cert.get_subject().C = subject.get(CERT_SUBJECT_COUNTRY, 'US')
cert.get_subject().ST = subject.get(CERT_SUBJECT_STATE, 'California')
cert.get_subject().O = subject.get(CERT_SUBJECT_ORG, 'MyOrg')
cert.get_subject().OU = subject.get(CERT_SUBJECT_UNIT, 'MyUnit')
cert.get_subject().CN = subject.get(CERT_SUBJECT_HOST, 'myorg.com')
cert.gmtime_adj_notBefore(0)
cert.gmtime_adj_notAfter(valid_for_days * 24 * 60 * 60)
cert.set_issuer(cert.get_subject())
cert.set_pubkey(key)
cert.set_serial_number(int(time()))
cert.sign(key, signature_alg)
return cert, key
class ClientCertificateManager(object):
"""Manage Client Certificate for backend authentication
There should be single client certificate associated
with certain principal identity. Certificate and PK storage
is pluggable. Storage API (similar to neutron-lbaas barbican API):
store_cert(purpose, certificate, private_key)
get_cert(purpose)
delete_cert(purpose)
"""
def __init__(self, identity, nsx_trust_management, storage_driver):
self._cert = None
self._key = None
self._storage_driver = storage_driver
self._identity = identity
self._nsx_trust_management = nsx_trust_management
def __enter__(self):
"""Load cert from storage
This is an optimization to avoid repeated storage access.
Usage example:
with cert_manager as c:
if c.exists():
date = c.expires_on()
days = c.exires_in_days()
"""
self._cert, self._key = self.get_cert_and_key()
return self
def __exit__(self, type, value, traceback):
self._cert = None
self._key = None
def generate(self, subject, key_size=2048, valid_for_days=3650,
signature_alg='sha256', node_id=None):
"""Generate new certificate and register it in the system
Generate certificate with RSA key based on arguments provided,
register and associate it to principal identity on backend,
and store it in storage. If certificate already exists, fail.
"""
self._validate_empty()
cert, key = generate_self_signed_cert_pair(key_size,
valid_for_days,
signature_alg,
subject)
# register on backend
self._register_cert(cert, node_id or uuid.uuid4())
# save in storage
cert_pem = crypto.dump_certificate(crypto.FILETYPE_PEM, cert)
key_pem = crypto.dump_privatekey(crypto.FILETYPE_PEM, key)
self._storage_driver.store_cert(self._identity, cert_pem, key_pem)
LOG.debug("Client certificate generated successfully")
def delete(self):
"""Delete existing certificate from storage and backend"""
cert_pem, key_pem = self.get_pem()
if not cert_pem:
return
ok = True
try:
self._nsx_trust_management.delete_cert_and_identity(
self._identity, cert_pem)
except nsxlib_exceptions.ManagerError as e:
LOG.error("Failed to clear certificate on backend: %s", e)
ok = False
try:
self._storage_driver.delete_cert(self._identity)
except Exception:
LOG.error("Failed to clear certificate in storage: %s", e)
ok = False
self._cert = None
self._key = None
if ok:
LOG.debug("Client certificate removed successfully")
def exists(self):
"""Check if certificate was created for given identity"""
if self._cert:
return True
cert_pem, key_pem = self._storage_driver.get_cert(self._identity)
return cert_pem is not None
def _get_cert_from_file(self, filename):
with open(filename, 'r') as f:
cert_pem = f.read()
if not cert_pem:
raise nsxlib_exceptions.CertificateError(
msg=_("Failed to read certificate from %s") % filename)
# validate correct crypto
try:
cert = crypto.load_certificate(crypto.FILETYPE_PEM, cert_pem)
except crypto.Error:
raise nsxlib_exceptions.CertificateError(
msg=_("Failed to import client certificate"))
return cert
def import_pem(self, filename, node_id=None):
"""Import and register existing certificate in PEM format"""
# TODO(annak): support PK import as well
self._validate_empty()
cert = self._get_cert_from_file(filename)
# register on backend
self._register_cert(cert, node_id or uuid.uuid4())
cert_pem = crypto.dump_certificate(crypto.FILETYPE_PEM, cert)
self._storage_driver.store_cert(self._identity, cert_pem, None)
LOG.debug("Client certificate imported successfully")
def delete_pem(self, filename):
"""Delete specified client certificate without storage verification"""
# This file may contain private key
# passing the pem through crypto will perform validation and
# strip off the key
cert = self._get_cert_from_file(filename)
cert_pem = crypto.dump_certificate(crypto.FILETYPE_PEM, cert)
self._nsx_trust_management.delete_cert_and_identity(self._identity,
cert_pem)
self._storage_driver.delete_cert(self._identity)
def _load_from_storage(self):
"""Returns certificate and key pair in PEM format"""
cert_pem, key_pem = self._storage_driver.get_cert(self._identity)
if cert_pem is None:
return None, None
return (cert_pem, key_pem)
def get_pem(self):
return self._load_from_storage()
def export_pem(self, filename):
"""Exports certificate and key pair to file"""
self._validate_exists()
cert_pem, key_pem = self._load_from_storage()
with open(filename, 'w') as f:
f.write(cert_pem)
f.write(key_pem)
def expires_on(self):
"""Returns certificate expiration timestamp"""
self._validate_exists()
cert, key = self.get_cert_and_key()
converted = datetime.datetime.strptime(
cert.get_notAfter().decode(),
"%Y%m%d%H%M%SZ")
return converted
def expires_in_days(self):
"""Returns in how many days the certificate expires"""
delta = self.expires_on() - datetime.datetime.utcnow()
return delta.days
def get_subject(self):
self._validate_exists()
cert, key = self.get_cert_and_key()
return {CERT_SUBJECT_COUNTRY: cert.get_subject().C,
CERT_SUBJECT_STATE: cert.get_subject().ST,
CERT_SUBJECT_ORG: cert.get_subject().O,
CERT_SUBJECT_UNIT: cert.get_subject().OU,
CERT_SUBJECT_HOST: cert.get_subject().CN}
def get_signature_alg(self):
self._validate_exists()
cert, key = self.get_cert_and_key()
return cert.get_signature_algorithm()
def get_key_size(self):
self._validate_exists()
cert, key = self.get_cert_and_key()
return key.bits()
def _validate_empty(self):
if self.exists():
raise nsxlib_exceptions.ObjectAlreadyExists(
object_type='Client Certificate')
def _validate_exists(self):
if not self.exists():
raise nsxlib_exceptions.ObjectNotGenerated(
object_type='Client Certificate')
def get_cert_and_key(self):
"""Load cert and key from storage"""
if self._cert and self._key:
return self._cert, self._key
cert_pem, key_pem = self._load_from_storage()
if cert_pem is None:
return None, None
try:
cert = crypto.load_certificate(crypto.FILETYPE_PEM, cert_pem)
key = crypto.load_privatekey(crypto.FILETYPE_PEM, key_pem)
except crypto.Error:
raise nsxlib_exceptions.CertificateError(
msg="Failed to load client certificate")
return cert, key
def _register_cert(self, cert, node_id):
cert_pem = crypto.dump_certificate(crypto.FILETYPE_PEM, cert)
self._nsx_trust_management.create_cert_and_identity(self._identity,
cert_pem,
node_id)
class ClientCertProvider(object):
"""Basic implementation for client certificate provider
Responsible for preparing, providing and disposing client certificate
file. Basic implementation assumes the file exists in the file system
and does not take responsibility of deleting this sensitive information
after use.
Inheriting objects should make use of __enter__ and __exit__ APIs to
prepare and dispose the certificate file data.
"""
def __init__(self, filename):
self._filename = filename
def __enter__(self):
return self
def __exit__(self, type, value, traceback):
pass
def filename(self):
return self._filename