274 lines
9.1 KiB
Python
274 lines
9.1 KiB
Python
"""
|
|
.. module: lemur.plugins.lemur_aws.aws
|
|
:platform: Unix
|
|
:copyright: (c) 2015 by Netflix Inc., see AUTHORS for more
|
|
:license: Apache, see LICENSE for more details.
|
|
|
|
Terraform example to setup the destination bucket:
|
|
resource "aws_s3_bucket" "certs_log_bucket" {
|
|
bucket = "certs-log-access-bucket"
|
|
acl = "log-delivery-write"
|
|
}
|
|
|
|
resource "aws_s3_bucket" "certs_lemur" {
|
|
bucket = "certs-lemur"
|
|
acl = "private"
|
|
|
|
logging {
|
|
target_bucket = "${aws_s3_bucket.certs_log_bucket.id}"
|
|
target_prefix = "log/lemur"
|
|
}
|
|
}
|
|
|
|
The IAM role Lemur is running as should have the following actions on the destination bucket:
|
|
|
|
"S3:PutObject",
|
|
"S3:PutObjectAcl"
|
|
|
|
The reader should have the following actions:
|
|
"s3:GetObject"
|
|
|
|
.. moduleauthor:: Kevin Glisson <kglisson@netflix.com>
|
|
.. moduleauthor:: Mikhail Khodorovskiy <mikhail.khodorovskiy@jivesoftware.com>
|
|
.. moduleauthor:: Harm Weites <harm@weites.com>
|
|
"""
|
|
from flask import current_app
|
|
from boto.exception import BotoServerError
|
|
|
|
from lemur.plugins.bases import DestinationPlugin, SourcePlugin
|
|
from lemur.plugins.lemur_aws.ec2 import get_regions
|
|
from lemur.plugins.lemur_aws.elb import get_all_elbs, describe_load_balancer_policies, attach_certificate
|
|
from lemur.plugins.lemur_aws import iam, s3
|
|
from lemur.plugins import lemur_aws as aws
|
|
|
|
|
|
class AWSDestinationPlugin(DestinationPlugin):
|
|
title = 'AWS'
|
|
slug = 'aws-destination'
|
|
description = 'Allow the uploading of certificates to AWS IAM'
|
|
version = aws.VERSION
|
|
|
|
author = 'Kevin Glisson'
|
|
author_url = 'https://github.com/netflix/lemur'
|
|
|
|
options = [
|
|
{
|
|
'name': 'accountNumber',
|
|
'type': 'str',
|
|
'required': True,
|
|
'validation': '/^[0-9]{12,12}$/',
|
|
'helpMessage': 'Must be a valid AWS account number!',
|
|
}
|
|
]
|
|
|
|
# 'elb': {
|
|
# 'name': {'type': 'name'},
|
|
# 'region': {'type': 'str'},
|
|
# 'port': {'type': 'int'}
|
|
# }
|
|
|
|
def upload(self, name, body, private_key, cert_chain, options, **kwargs):
|
|
try:
|
|
iam.upload_cert(self.get_option('accountNumber', options), name, body, private_key,
|
|
cert_chain=cert_chain)
|
|
except BotoServerError as e:
|
|
if e.error_code != 'EntityAlreadyExists':
|
|
raise Exception(e)
|
|
|
|
e = self.get_option('elb', options)
|
|
if e:
|
|
attach_certificate(kwargs['accountNumber'], ['region'], e['name'], e['port'], e['certificateId'])
|
|
|
|
|
|
class AWSSourcePlugin(SourcePlugin):
|
|
title = 'AWS'
|
|
slug = 'aws-source'
|
|
description = 'Discovers all SSL certificates and ELB endpoints in an AWS account'
|
|
version = aws.VERSION
|
|
|
|
author = 'Kevin Glisson'
|
|
author_url = 'https://github.com/netflix/lemur'
|
|
|
|
options = [
|
|
{
|
|
'name': 'accountNumber',
|
|
'type': 'str',
|
|
'required': True,
|
|
'validation': '/^[0-9]{12,12}$/',
|
|
'helpMessage': 'Must be a valid AWS account number!',
|
|
},
|
|
{
|
|
'name': 'regions',
|
|
'type': 'str',
|
|
'helpMessage': 'Comma separated list of regions to search in, if no region is specified we look in all regions.'
|
|
},
|
|
]
|
|
|
|
def get_certificates(self, options, **kwargs):
|
|
certs = []
|
|
arns = iam.get_all_server_certs(self.get_option('accountNumber', options))
|
|
for arn in arns:
|
|
cert_body, cert_chain = iam.get_cert_from_arn(arn)
|
|
cert_name = iam.get_name_from_arn(arn)
|
|
cert = dict(
|
|
body=cert_body,
|
|
chain=cert_chain,
|
|
name=cert_name
|
|
)
|
|
certs.append(cert)
|
|
return certs
|
|
|
|
def get_endpoints(self, options, **kwargs):
|
|
endpoints = []
|
|
account_number = self.get_option('accountNumber', options)
|
|
regions = self.get_option('regions', options)
|
|
|
|
if not regions:
|
|
regions = get_regions(account_number=account_number)
|
|
else:
|
|
regions = regions.split(',')
|
|
|
|
for region in regions:
|
|
elbs = get_all_elbs(account_number=account_number, region=region)
|
|
current_app.logger.info("Describing load balancers in {0}-{1}".format(account_number, region))
|
|
for elb in elbs:
|
|
for listener in elb['ListenerDescriptions']:
|
|
if not listener['Listener'].get('SSLCertificateId'):
|
|
continue
|
|
|
|
if listener['Listener']['SSLCertificateId'] == 'Invalid-Certificate':
|
|
continue
|
|
|
|
endpoint = dict(
|
|
name=elb['LoadBalancerName'],
|
|
dnsname=elb['DNSName'],
|
|
type='elb',
|
|
port=listener['Listener']['LoadBalancerPort'],
|
|
certificate_name=iam.get_name_from_arn(listener['Listener']['SSLCertificateId'])
|
|
)
|
|
|
|
if listener['PolicyNames']:
|
|
policy = describe_load_balancer_policies(elb['LoadBalancerName'], listener['PolicyNames'], account_number=account_number, region=region)
|
|
endpoint['policy'] = format_elb_cipher_policy(policy)
|
|
|
|
endpoints.append(endpoint)
|
|
|
|
return endpoints
|
|
|
|
def clean(self, options, **kwargs):
|
|
account_number = self.get_option('accountNumber', options)
|
|
certificates = self.get_certificates(options)
|
|
endpoints = self.get_endpoints(options)
|
|
|
|
orphaned = []
|
|
for certificate in certificates:
|
|
for endpoint in endpoints:
|
|
if certificate['name'] == endpoint['certificate_name']:
|
|
break
|
|
else:
|
|
orphaned.append(certificate['name'])
|
|
iam.delete_cert(account_number, certificate)
|
|
|
|
return orphaned
|
|
|
|
|
|
def format_elb_cipher_policy(policy):
|
|
"""
|
|
Attempts to format cipher policy information into a common format.
|
|
:param policy:
|
|
:return:
|
|
"""
|
|
ciphers = []
|
|
name = None
|
|
for descr in policy['PolicyDescriptions']:
|
|
for attr in descr['PolicyAttributeDescriptions']:
|
|
if attr['AttributeName'] == 'Reference-Security-Policy':
|
|
name = attr['AttributeValue']
|
|
continue
|
|
|
|
if attr['AttributeValue'] == 'true':
|
|
ciphers.append(attr['AttributeName'])
|
|
|
|
return dict(name=name, ciphers=ciphers)
|
|
|
|
|
|
class S3DestinationPlugin(DestinationPlugin):
|
|
title = 'AWS-S3'
|
|
slug = 'aws-s3'
|
|
description = 'Allow the uploading of certificates to Amazon S3'
|
|
|
|
author = 'Mikhail Khodorovskiy, Harm Weites <harm@weites.com>'
|
|
author_url = 'https://github.com/Netflix/lemur'
|
|
|
|
options = [
|
|
{
|
|
'name': 'bucket',
|
|
'type': 'str',
|
|
'required': True,
|
|
'validation': '/^$|\s+/',
|
|
'helpMessage': 'Must be a valid S3 bucket name!',
|
|
},
|
|
{
|
|
'name': 'accountNumber',
|
|
'type': 'str',
|
|
'required': True,
|
|
'validation': '/^[0-9]{12,12}$/',
|
|
'helpMessage': 'A valid AWS account number with permission to access S3',
|
|
},
|
|
{
|
|
'name': 'region',
|
|
'type': 'str',
|
|
'default': 'eu-west-1',
|
|
'required': False,
|
|
'validation': '/^\w+-\w+-\d+$/',
|
|
'helpMessage': 'Availability zone to use',
|
|
},
|
|
{
|
|
'name': 'encrypt',
|
|
'type': 'bool',
|
|
'required': False,
|
|
'helpMessage': 'Availability zone to use',
|
|
'default': True
|
|
},
|
|
{
|
|
'name': 'key',
|
|
'type': 'str',
|
|
'required': False,
|
|
'validation': '/^$|\s+/',
|
|
'helpMessage': 'Must be a valid S3 object key!',
|
|
},
|
|
{
|
|
'name': 'caKey',
|
|
'type': 'str',
|
|
'required': False,
|
|
'validation': '/^$|\s+/',
|
|
'helpMessage': 'Must be a valid S3 object key!',
|
|
},
|
|
{
|
|
'name': 'certKey',
|
|
'type': 'str',
|
|
'required': False,
|
|
'validation': '/^$|\s+/',
|
|
'helpMessage': 'Must be a valid S3 object key!',
|
|
}
|
|
]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(S3DestinationPlugin, self).__init__(*args, **kwargs)
|
|
|
|
def upload(self, name, body, private_key, cert_chain, options, **kwargs):
|
|
account_number = self.get_option('accountNumber', options)
|
|
encrypt = self.get_option('encrypt', options)
|
|
bucket = self.get_option('bucket', options)
|
|
key = self.get_option('key', options)
|
|
ca_key = self.get_option('caKey', options)
|
|
cert_key = self.get_option('certKey', options)
|
|
|
|
if key and ca_key and cert_key:
|
|
s3.write_to_s3(account_number, bucket, key, private_key, encrypt=encrypt)
|
|
s3.write_to_s3(account_number, bucket, ca_key, cert_chain, encrypt=encrypt)
|
|
s3.write_to_s3(account_number, bucket, cert_key, body, encrypt=encrypt)
|
|
else:
|
|
pem_body = key + '\n' + body + '\n' + cert_chain + '\n'
|
|
s3.write_to_s3(account_number, bucket, name, pem_body, encrypt=encrypt)
|