def generate_instance(self, id, ip):
i = Instance()
i.id = id
i.private_ip_address = ip
i.private_dns_name = 'ip-{}.eu-west-1.compute.internal'.format(ip.replace('.', '-'))
i.tags = {'aws:cloudformation:stack-name': 'etc-cluster', 'aws:autoscaling:groupName': 'etc-cluster-postgres'}
return i
def setUp(self):
ec2.credentials.ACCESS_KEY_ID = 'abc'
ec2.credentials.SECRET_ACCESS_KEY = 'xyz'
# Build up two reservations, with two instances each, totalling 4 instances
# Two running, two stopped
reservations = []
instance_count = 0
for i in xrange(2):
i1 = Instance()
i1.id = 'i-abc%d' % instance_count
i1.state = 'running'
i1.tags = {'Name': 'instance-%d' % instance_count}
instance_count += 1
i2 = Instance()
i2.id = 'i-abc%d' % instance_count
i2.state = 'stopped'
i2.tags = {'Name': 'instance-%d' % instance_count}
instance_count += 1
reservation = MagicMock()
reservation.instances.__iter__ = MagicMock(
return_value=iter([i1, i2]))
reservations.append(reservation)
self.connection = MagicMock()
self.connection.get_all_instances = MagicMock(
return_value=reservations)
def run_instance(self,
image_id,
security_group_ids=None,
instance_type='m3.medium',
block_device_map=None):
instance = Instance()
instance.id = _new_id()
instance.root_device_name = '/dev/sda1'
instance._state.code = 0
instance._state.name = 'pending'
# Create volumes based on block device data from the image.
image = self.get_image(image_id)
instance_bdm = BlockDeviceMapping()
for device_name, bdm in image.block_device_mapping.iteritems():
# Create a new volume and attach it to the instance.
volume = Volume()
volume.size = 8
volume.id = _new_id()
self.volumes[volume.id] = volume
bdt = BlockDeviceType(volume_id=volume.id, size=8)
instance_bdm[device_name] = bdt
instance.block_device_mapping = instance_bdm
self.instances[instance.id] = instance
return instance
def setUp(self):
(flexmock(EC2Connection)
.should_receive('get_key_pair')
.and_return(None))
(flexmock(EC2Connection)
.should_receive('create_key_pair')
.with_args('bookeyname')
.and_return(KeyPair()))
(flexmock(EC2Connection)
.should_receive('get_all_security_groups')
.and_return([]))
(flexmock(EC2Connection)
.should_receive('create_security_group')
.with_args('boogroup', 'AppScale security group')
.and_return(SecurityGroup()))
(flexmock(EC2Connection)
.should_receive('authorize_security_group')
.and_return())
reservation = Reservation()
# the old implementation had a regression where public and private IPs
# were getting sorted, and thus public ip1 would point to private ip 2.
# to prevent this regression from resurfacing, set up the dns names so
# that a sort would mess them up again.
instance1 = Instance()
instance1.public_dns_name = 'ABC-public-ip1'
instance1.private_dns_name = 'DEF-private-ip1'
instance1.id = 'i-id1'
instance1._state.name = 'running'
instance1.key_name = 'bookeyname'
instance2 = Instance()
instance2.public_dns_name = 'DEF-public-ip2'
instance2.private_dns_name = 'ABC-private-ip2'
instance2.id = 'i-id2'
instance2._state.name = 'running'
instance2.key_name = 'bookeyname'
reservation.instances = [instance1, instance2]
(flexmock(EC2Connection)
.should_receive('get_all_instances')
.and_return([])
.and_return([reservation]))
(flexmock(EC2Connection)
.should_receive('run_instances')
.and_return())
(flexmock(utils)
.should_receive('get_secret')
.and_return('secret'))
(flexmock(utils)
.should_receive('sleep')
.and_return())
(flexmock(utils)
.should_receive('get_random_alphanumeric')
.and_return('0000000000'))
(flexmock(utils)
.should_receive('write_key_file')
.and_return())
def terminate_instances(self, instance_ids):
self.terminated.extend(instance_ids)
instances = []
for instance_id in instance_ids:
instance = Instance()
instance.id = instance_id
instances.append(instance)
return instances
def build_pending_reservations(instance_id):
instance = Instance()
instance.id = instance_id
instance.state = 'running'
instance.ip_address = '172.16.52.10'
instance.private_ip_address = '172.16.52.10'
r = Reservation()
r.instances = [instance]
return [r]
def run_instances(self, ami, *args, **kwargs):
self.instances.append("instance with ami %s and key %s and groups %s" % (
ami,
kwargs["key_name"],
", ".join(kwargs["security_groups"])
))
instance = Instance()
instance.id = 'i-00000302'
reservation = Reservation()
reservation.instances = [instance]
return reservation
def setUp(self):
ec2.credentials.ACCESS_KEY_ID = 'abc'
ec2.credentials.SECRET_ACCESS_KEY = 'xyz'
# Build up two reservations, with two instances each, totalling 4 instances
# Two running, two stopped
reservations = []
instance_count = 0
for i in xrange(2):
i1 = Instance()
i1.id = 'i-abc%d' % instance_count
i1._state = RUNNING_STATE
i1.tags = {'Name': 'instance-%d' % instance_count}
instance_count += 1
i2 = Instance()
i2.id = 'i-abc%d' % instance_count
i2._state = STOPPED_STATE
i2.tags = {'Name': 'instance-%d' % instance_count}
instance_count += 1
reservation = MagicMock()
reservation.instances.__iter__ = MagicMock(return_value=iter([i1, i2]))
reservations.append(reservation)
security_groups = []
for i in xrange(2):
sg = SecurityGroup()
sg.id = 'sg-abc%d' % i
sg.name = 'group-%d' % i
sg.description = 'Group %d' % i
security_groups.append(sg)
vpcs = []
for i in xrange(2):
vpc = VPC()
vpc.id = 'vpc-abc%d' % i
if i % 2:
vpc.state = 'pending'
vpc.is_default = False
vpc.instance_tenancy = 'default'
else:
vpc.state = 'available'
vpc.is_default = True
vpc.instance_tenancy = 'dedicated'
vpc.cidr_block = '10.%d.0.0/16' % i
vpc.dhcp_options_id = 'dopt-abc%d' % i
vpcs.append(vpc)
self.connection = MagicMock()
self.connection.get_all_instances = MagicMock(return_value=reservations)
self.connection.get_all_security_groups = MagicMock(return_value=security_groups)
self.vpc_connection = MagicMock()
self.vpc_connection.get_all_vpcs = MagicMock(return_value=vpcs)
def setUp(self):
ec2.credentials.ACCESS_KEY_ID = 'abc'
ec2.credentials.SECRET_ACCESS_KEY = 'xyz'
# Build up two reservations, with two instances each, totalling 4 instances
# Two running, two stopped
reservations = []
instance_count = 0
for i in xrange(2):
i1 = Instance()
i1.id = 'i-abc%d' % instance_count
i1._state = RUNNING_STATE
i1.tags = {'Name': 'instance-%d' % instance_count}
instance_count += 1
i2 = Instance()
i2.id = 'i-abc%d' % instance_count
i2._state = STOPPED_STATE
i2.tags = {'Name': 'instance-%d' % instance_count}
instance_count += 1
reservation = MagicMock()
reservation.instances.__iter__ = MagicMock(return_value=iter([i1, i2]))
reservations.append(reservation)
security_groups = []
for i in xrange(2):
sg = SecurityGroup()
sg.id = 'sg-abc%d' % i
sg.name = 'group-%d' % i
sg.description = 'Group %d' % i
security_groups.append(sg)
self.connection = MagicMock()
self.connection.get_all_instances = MagicMock(return_value=reservations)
self.connection.get_all_security_groups = MagicMock(return_value=security_groups)
def setUp(self):
ec2.credentials.ACCESS_KEY_ID = 'abc'
ec2.credentials.SECRET_ACCESS_KEY = 'xyz'
# Build up two reservations, with two instances each, totalling 4 instances
# Two running, two stopped
reservations = []
instance_count = 0
for i in xrange(2):
i1 = Instance()
i1.id = 'i-abc%d' % instance_count
i1.state = 'running'
i1.tags = {'Name': 'instance-%d' % instance_count}
instance_count += 1
i2 = Instance()
i2.id = 'i-abc%d' % instance_count
i2.state = 'stopped'
i2.tags = {'Name': 'instance-%d' % instance_count}
instance_count += 1
reservation = MagicMock()
reservation.instances.__iter__ = MagicMock(return_value=iter([i1, i2]))
reservations.append(reservation)
self.connection = MagicMock()
self.connection.get_all_instances = MagicMock(return_value=reservations)
def setUp(self):
(flexmock(EC2Connection)
.should_receive('get_key_pair')
.and_return(None))
(flexmock(EC2Connection)
.should_receive('create_key_pair')
.with_args('bookeyname')
.and_return(KeyPair()))
(flexmock(EC2Connection)
.should_receive('get_all_security_groups')
.and_return([]))
(flexmock(EC2Connection)
.should_receive('create_security_group')
.with_args('boogroup', 'AppScale security group')
.and_return(SecurityGroup()))
(flexmock(EC2Connection)
.should_receive('authorize_security_group')
.and_return())
reservation = Reservation()
instance = Instance()
instance.private_dns_name = 'private-ip'
instance.public_dns_name = 'public-ip'
instance.id = 'i-id'
instance._state.name = 'running'
instance.key_name = 'bookeyname'
reservation.instances = [instance]
(flexmock(EC2Connection)
.should_receive('get_all_instances')
.and_return([])
.and_return([reservation]))
(flexmock(EC2Connection)
.should_receive('terminate_instances')
.and_return([instance]))
(flexmock(EC2Connection)
.should_receive('run_instances')
.and_return())
(flexmock(utils)
.should_receive('get_secret')
.and_return('secret'))
(flexmock(utils)
.should_receive('sleep')
.and_return())
(flexmock(utils)
.should_receive('get_random_alphanumeric')
.and_return('0000000000'))
(flexmock(utils)
.should_receive('write_key_file')
.and_return())
def set_up(self):
self.old_requests_post = requests.post
self.old_requests_get = requests.get
self.old_requests_delete = requests.delete
requests.post = requests_post
requests.get = requests_get
requests.delete = requests_delete
self.ec2 = Instance()
self.ec2.id = 'i-foobar'
self.ec2.private_ip_address = '127.0.0.1'
self.ec2.private_dns_name = 'ip-127-0-0-1.eu-west-1.compute.internal'
self.ec2.tags = {
'aws:cloudformation:stack-name': 'etc-cluster',
'aws:autoscaling:groupName': 'etc-cluster-postgres'
}
self.ec2_member = EtcdMember(self.ec2)
self.etcd = {
'id':
'deadbeef',
'name':
'i-foobar2',
'clientURLs': [],
'peerURLs':
['http://127.0.0.2:{}'.format(EtcdMember.DEFAULT_PEER_PORT)],
}
self.etcd_member = EtcdMember(self.etcd)
def __enter__(self):
self.connect()
self._resolve_baseami()
self._instance = Instance(connection=self._connection)
self._instance.id = get_instance_metadata()['instance-id']
self._instance.update()
context = self._config.context
if context.ami.get("base_ami_name", None):
environ["AMINATOR_BASE_AMI_NAME"] = context.ami.base_ami_name
if context.ami.get("base_ami_id", None):
environ["AMINATOR_BASE_AMI_ID"] = context.ami.base_ami_id
if context.cloud.get("region", None):
environ["AMINATOR_REGION"] = context.cloud.region
return self
def setUp(self):
fake_ec2 = flexmock(name='fake_ec2')
fake_ec2.should_receive('get_key_pair')
fake_ec2.should_receive('create_key_pair').with_args('bookeyname') \
.and_return(KeyPair())
fake_ec2.should_receive('get_all_security_groups').and_return([])
fake_ec2.should_receive('create_security_group') \
.with_args('boogroup', 'AppScale security group') \
.and_return(SecurityGroup())
fake_ec2.should_receive('authorize_security_group')
reservation = Reservation()
instance = Instance()
instance.private_dns_name = 'private-ip'
instance.public_dns_name = 'public-ip'
instance.id = 'i-id'
instance._state.name = 'running'
instance.key_name = 'bookeyname'
reservation.instances = [instance]
fake_ec2.should_receive('get_all_instances').and_return([]) \
.and_return([reservation])
fake_ec2.should_receive('terminate_instances').and_return([instance])
fake_ec2.should_receive('request_spot_instances')
flexmock(boto.ec2)
boto.ec2.should_receive('connect_to_region').and_return(fake_ec2)
(flexmock(utils)
.should_receive('get_secret')
.and_return('secret'))
(flexmock(utils)
.should_receive('sleep')
.and_return())
(flexmock(utils)
.should_receive('get_random_alphanumeric')
.and_return('0000000000'))
(flexmock(utils)
.should_receive('write_key_file')
.and_return())
def generate_instance(self, id, ip):
i = Instance()
i.id = id
i.private_ip_address = ip
i.private_dns_name = 'ip-{}.eu-west-1.compute.internal'.format(ip.replace('.', '-'))
i.tags = {'aws:cloudformation:stack-name': 'etc-cluster', 'aws:autoscaling:groupName': 'etc-cluster-postgres'}
return i
def setUp(self):
(flexmock(EC2Connection).should_receive('get_key_pair').and_return(
None))
(flexmock(EC2Connection).should_receive('create_key_pair').with_args(
'bookeyname').and_return(KeyPair()))
(flexmock(EC2Connection).should_receive(
'get_all_security_groups').and_return([]))
(flexmock(
EC2Connection).should_receive('create_security_group').with_args(
'boogroup',
'AppScale security group').and_return(SecurityGroup()))
(flexmock(EC2Connection).should_receive(
'authorize_security_group').and_return())
reservation = Reservation()
instance = Instance()
instance.private_dns_name = 'private-ip'
instance.public_dns_name = 'public-ip'
instance.id = 'i-id'
instance._state.name = 'running'
instance.key_name = 'bookeyname'
reservation.instances = [instance]
(flexmock(EC2Connection).should_receive(
'get_all_instances').and_return([]).and_return([reservation]))
(flexmock(EC2Connection).should_receive(
'terminate_instances').and_return([instance]))
(flexmock(EC2Connection).should_receive('run_instances').and_return())
(flexmock(utils).should_receive('get_secret').and_return('secret'))
(flexmock(utils).should_receive('sleep').and_return())
(flexmock(utils).should_receive('get_random_alphanumeric').and_return(
'0000000000'))
(flexmock(utils).should_receive('write_key_file').and_return())
def __enter__(self):
self.connect()
self._resolve_baseami()
self._instance = Instance(connection=self._connection)
self._instance.id = get_instance_metadata()['instance-id']
self._instance.update()
context = self._config.context
if context.ami.get("base_ami_name", None):
environ["AMINATOR_BASE_AMI_NAME"] = context.ami.base_ami_name
if context.ami.get("base_ami_id", None):
environ["AMINATOR_BASE_AMI_ID"] = context.ami.base_ami_id
if context.cloud.get("region", None):
environ["AMINATOR_REGION"] = context.cloud.region
return self
def run_instance(self,
image_id,
security_group_ids=None,
instance_type='c3.xlarge',
placement=None,
block_device_map=None,
subnet_id=None,
user_data=None,
ebs_optimized=True,
instance_profile_name=None):
instance = Instance()
instance.id = new_id()
instance.image_id = image_id
instance.root_device_name = '/dev/sda1'
instance._state.code = 0
instance._state.name = 'pending'
# Create volumes based on block device data from the image.
image = self.get_image(image_id)
instance_bdm = BlockDeviceMapping()
for device_name, bdm in image.block_device_mapping.iteritems():
# Create a new volume and attach it to the instance.
volume = Volume()
volume.size = 8
volume.id = new_id()
self.volumes[volume.id] = volume
bdt = BlockDeviceType(volume_id=volume.id, size=8)
instance_bdm[device_name] = bdt
instance.block_device_mapping = instance_bdm
self.instances[instance.id] = instance
if self.run_instance_callback:
args = RunInstanceArgs()
args.image_id = image_id
args.instance_type = instance_type
args.ebs_optimized = ebs_optimized
args.security_group_ids = security_group_ids
args.subnet_id = subnet_id
args.user_data = user_data
args.instance = instance
self.run_instance_callback(args)
return instance
class EC2CloudPlugin(BaseCloudPlugin):
_name = 'ec2'
def add_metrics(self, metric_base_name, cls, func_name):
newfunc = succeeds("{0}.count".format(metric_base_name), self)(raises(
"{0}.error".format(metric_base_name),
self)(timer("{0}.duration".format(metric_base_name),
self)(getattr(cls, func_name))))
setattr(cls, func_name, newfunc)
def __init__(self):
super(EC2CloudPlugin, self).__init__()
# wrap each of the functions so we can get timer and error metrics
for ec2func in [
"create_volume", "create_tags", "register_image",
"get_all_images"
]:
self.add_metrics(
"aminator.cloud.ec2.connection.{0}".format(ec2func),
EC2Connection, ec2func)
for volfunc in [
"add_tag", "attach", "create_snapshot", "delete", "detach",
"update"
]:
self.add_metrics("aminator.cloud.ec2.volume.{0}".format(volfunc),
Volume, volfunc)
for imgfunc in ["update"]:
self.add_metrics("aminator.cloud.ec2.image.{0}".format(imgfunc),
Image, imgfunc)
for insfunc in ["update"]:
self.add_metrics("aminator.cloud.ec2.instance.{0}".format(insfunc),
Instance, insfunc)
def add_plugin_args(self, *args, **kwargs):
context = self._config.context
base_ami = self._parser.add_argument_group(
title='Base AMI', description='EITHER AMI id OR name, not both!')
base_ami_mutex = base_ami.add_mutually_exclusive_group(required=True)
base_ami_mutex.add_argument(
'-b',
'--base-ami-name',
dest='base_ami_name',
action=conf_action(config=context.ami),
help='The name of the base AMI used in provisioning')
base_ami_mutex.add_argument(
'-B',
'--base-ami-id',
dest='base_ami_id',
action=conf_action(config=context.ami),
help='The id of the base AMI used in provisioning')
cloud = self._parser.add_argument_group(
title='EC2 Options', description='EC2 Connection Information')
cloud.add_argument('-r',
'--region',
dest='region',
help='EC2 region (default: us-east-1)',
action=conf_action(config=context.cloud))
cloud.add_argument('--boto-secure',
dest='is_secure',
help='Connect via https',
action=conf_action(config=context.cloud,
action='store_true'))
cloud.add_argument('--boto-debug',
dest='boto_debug',
help='Boto debug output',
action=conf_action(config=context.cloud,
action='store_true'))
volume_mutex = cloud.add_mutually_exclusive_group()
volume_mutex.add_argument(
'-V',
'--volume-id',
dest='volume_id',
action=conf_action(config=context.ami),
help='The Base AMI volume id already attached to the system')
volume_mutex.add_argument(
'--provisioner-ebs-type',
dest='provisioner_ebs_type',
action=conf_action(config=context.cloud),
help='The type of EBS volume to create from the Base AMI snapshot')
cloud.add_argument(
'--register-ebs-type',
dest='register_ebs_type',
action=conf_action(config=context.cloud),
help='The root volume EBS type for AMI registration')
cloud.add_argument(
'--root-volume-size',
dest='root_volume_size',
action=conf_action(config=context.ami),
help=
'Root volume size (in GB). The default is to inherit from the base AMI.'
)
def configure(self, config, parser):
super(EC2CloudPlugin, self).configure(config, parser)
host = config.context.web_log.get('host', False)
if not host:
md = get_instance_metadata()
pub, ipv4 = 'public-hostname', 'local-ipv4'
config.context.web_log['host'] = md[pub] if pub in md else md[ipv4]
def connect(self, **kwargs):
if self._connection:
log.warn('Already connected to EC2')
else:
log.info('Connecting to EC2')
self._connect(**kwargs)
def _connect(self, **kwargs):
cloud_config = self._config.plugins[self.full_name]
context = self._config.context
self._instance_metadata = get_instance_metadata()
instance_region = self._instance_metadata['placement'][
'availability-zone'][:-1]
region = kwargs.pop(
'region',
context.get('region', cloud_config.get('region', instance_region)))
log.debug('Establishing connection to region: {0}'.format(region))
context.cloud.setdefault('boto_debug', False)
if context.cloud.boto_debug:
from aminator.config import configure_datetime_logfile
configure_datetime_logfile(self._config, 'boto')
kwargs['debug'] = 1
log.debug('Boto debug logging enabled')
else:
logging.getLogger('boto').setLevel(logging.INFO)
if 'is_secure' not in kwargs:
kwargs['is_secure'] = context.get(
'is_secure', cloud_config.get('is_secure', True))
self._connection = connect_to_region(region, **kwargs)
log.info('Aminating in region {0}'.format(region))
def allocate_base_volume(self, tag=True):
cloud_config = self._config.plugins[self.full_name]
context = self._config.context
self._volume = Volume(connection=self._connection)
rootdev = context.base_ami.block_device_mapping[
context.base_ami.root_device_name]
volume_type = context.cloud.get(
'provisioner_ebs_type',
cloud_config.get('provisioner_ebs_type', 'standard'))
volume_size = context.ami.get('root_volume_size', None)
if volume_size is None:
volume_size = cloud_config.get('root_volume_size', None)
if volume_size is None:
volume_size = rootdev.size
volume_size = int(volume_size)
if volume_size < 1:
raise VolumeException(
'root_volume_size must be a positive integer, received {}'.
format(volume_size))
if volume_size < rootdev.size:
raise VolumeException(
'root_volume_size ({}) must be at least as large as the root '
'volume of the base AMI ({})'.format(volume_size,
rootdev.size))
self._volume.id = self._connection.create_volume(
size=volume_size,
zone=self._instance.placement,
volume_type=volume_type,
snapshot=rootdev.snapshot_id).id
if not self._volume_available():
log.critical('{0}: unavailable.')
return False
if tag:
tags = {
'purpose': cloud_config.get('tag_ami_purpose', 'amination'),
'status': 'busy',
'ami': context.base_ami.id,
'ami-name': context.base_ami.name,
'arch': context.base_ami.architecture,
}
self._connection.create_tags([self._volume.id], tags)
self._volume.update()
log.debug('Volume {0} created'.format(self._volume.id))
@retry(VolumeException, tries=2, delay=1, backoff=2, logger=log)
def attach_volume(self, blockdevice, tag=True):
context = self._config.context
if "volume_id" in context.ami:
volumes = self._connection.get_all_volumes(
volume_ids=[context.ami.volume_id])
if not volumes:
raise VolumeException('Failed to find volume: {0}'.format(
context.ami.volume_id))
self._volume = volumes[0]
return
self.allocate_base_volume(tag=tag)
# must do this as amazon still wants /dev/sd*
ec2_device_name = blockdevice.replace('xvd', 'sd')
log.debug('Attaching volume {0} to {1}:{2}({3})'.format(
self._volume.id, self._instance.id, ec2_device_name, blockdevice))
self._volume.attach(self._instance.id, ec2_device_name)
if not self.is_volume_attached(blockdevice):
log.debug('{0} attachment to {1}:{2}({3}) timed out'.format(
self._volume.id, self._instance.id, ec2_device_name,
blockdevice))
self._volume.add_tag('status', 'used')
# trigger a retry
raise VolumeException(
'Timed out waiting for {0} to attach to {1}:{2}'.format(
self._volume.id, self._instance.id, blockdevice))
log.debug('Volume {0} attached to {1}:{2}'.format(
self._volume.id, self._instance.id, blockdevice))
def is_volume_attached(self, blockdevice):
context = self._config.context
if "volume_id" in context.ami:
return True
try:
self._volume_attached(blockdevice)
except VolumeException:
log.debug(
'Timed out waiting for volume {0} to attach to {1}:{2}'.format(
self._volume.id, self._instance.id, blockdevice))
return False
return True
@retry(VolumeException, tries=10, delay=1, backoff=2, logger=log)
def _volume_attached(self, blockdevice):
status = self._volume.update()
if status != 'in-use':
raise VolumeException(
'Volume {0} not yet attached to {1}:{2}'.format(
self._volume.id, self._instance.id, blockdevice))
elif not os_node_exists(blockdevice):
raise VolumeException(
'{0} does not exist yet.'.format(blockdevice))
else:
return True
def snapshot_volume(self, description=None):
context = self._config.context
if not description:
description = context.snapshot.get('description', '')
log.debug('Creating snapshot with description {0}'.format(description))
self._snapshot = self._volume.create_snapshot(description)
if not self._snapshot_complete():
log.critical('Failed to create snapshot')
return False
else:
log.debug('Snapshot complete. id: {0}'.format(self._snapshot.id))
return True
def _state_check(self, obj, state):
obj.update()
classname = obj.__class__.__name__
if classname in ('Snapshot', 'Volume'):
if classname == 'Snapshot':
log.debug("Snapshot {0} state: {1}, progress: {2}".format(
obj.id, obj.status, obj.progress))
return obj.status == state
else:
return obj.state == state
@retry(VolumeException,
tries=600,
delay=0.5,
backoff=1.5,
logger=log,
maxdelay=10)
def _wait_for_state(self, resource, state):
if self._state_check(resource, state):
log.debug('{0} reached state {1}'.format(
resource.__class__.__name__, state))
return True
else:
raise VolumeException(
'Timed out waiting for {0} to get to {1}({2})'.format(
resource.id, state, resource.status))
@lapse("aminator.cloud.ec2.ami_available.duration")
def _ami_available(self):
return self._wait_for_state(self._ami, 'available')
@lapse("aminator.cloud.ec2.snapshot_completed.duration")
def _snapshot_complete(self):
return self._wait_for_state(self._snapshot, 'completed')
@lapse("aminator.cloud.ec2.volume_available.duration")
def _volume_available(self):
return self._wait_for_state(self._volume, 'available')
def detach_volume(self, blockdevice):
context = self._config.context
if "volume_id" in context.ami:
return
log.debug('Detaching volume {0} from {1}'.format(
self._volume.id, self._instance.id))
self._volume.detach()
if not self._volume_detached(blockdevice):
raise VolumeException(
'Time out waiting for {0} to detach from {1}'.format(
self._volume.id, self._instance.id))
log.debug('Successfully detached volume {0} from {1}'.format(
self._volume.id, self._instance.id))
@retry(VolumeException, tries=7, delay=1, backoff=2, logger=log)
def _volume_detached(self, blockdevice):
status = self._volume.update()
if status != 'available':
raise VolumeException(
'Volume {0} not yet detached from {1}'.format(
self._volume.id, self._instance.id))
elif os_node_exists(blockdevice):
raise VolumeException(
'Device node {0} still exists'.format(blockdevice))
else:
return True
def delete_volume(self):
context = self._config.context
if "volume_id" in context.ami:
return True
log.debug('Deleting volume {0}'.format(self._volume.id))
result = self._volume.delete()
if not result:
log.debug(
'Volume {0} delete returned False, may require manual cleanup'.
format(self._volume.id))
else:
log.debug('Volume {0} successfully deleted'.format(
self._volume.id))
return result
def is_stale_attachment(self, dev, prefix):
log.debug(
'Checking for stale attachment. dev: {0}, prefix: {1}'.format(
dev, prefix))
if dev in self.attached_block_devices(
prefix) and not os_node_exists(dev):
log.debug('{0} is stale, rejecting'.format(dev))
return True
log.debug('{0} not stale, using'.format(dev))
return False
@registration_retry(tries=3, delay=1, backoff=1)
def _register_image(self, **ami_metadata):
"""Register the AMI using boto3/botocore components which supports ENA
This is the only use of boto3 in aminator currently"""
# construct AMI registration payload boto3 style
request = {}
request['Name'] = ami_metadata.get('name', None)
request['Description'] = ami_metadata.get('description', None)
request['Architecture'] = ami_metadata.get('architecture', None)
request['EnaSupport'] = ami_metadata.get('ena_networking', False)
request['VirtualizationType'] = ami_metadata.get(
'virtualization_type', None)
# when instance store, don't provide botocore expects a string value
if ami_metadata.get('block_device_map') is not None:
request['BlockDeviceMappings'] = ami_metadata.get(
'block_device_map')
if ami_metadata.get('root_device_name') is not None:
request['RootDeviceName'] = ami_metadata.get('root_device_name')
# only present for instance store
if ami_metadata.get('image_location') is not None:
request['ImageLocation'] = ami_metadata.get('image_location')
# can only be set to 'simple' for hvm. don't include otherwise
if ami_metadata.get('sriov_net_support') is not None:
request['SriovNetSupport'] = ami_metadata.get('sriov_net_support')
if (ami_metadata.get('virtualization_type') == 'paravirtual'):
# KernelId required
request['KernelId'] = ami_metadata.get('kernel_id', None)
if ami_metadata.get('ramdisk_id') is not None:
request['RamdiskId'] = ami_metadata.get('ramdisk_id', None)
# assert we have all the key params. Nothing to _here_ should be None
for key, value in request.items():
if request[key] is None:
raise FinalizerException('{} cannot be None'.format(key))
log.debug('Boto3 registration request data [{}]'.format(request))
try:
client = boto3.client('ec2',
region_name=ami_metadata.get('region'))
response = client.register_image(**request)
log.debug('Registration response data [{}]'.format(response))
ami_id = response['ImageId']
if ami_id is None:
return False
log.info('Waiting for [{}] to become available'.format(ami_id))
waiter = client.get_waiter('image_available')
wait_request = {}
wait_request['ImageIds'] = []
wait_request['ImageIds'].append(ami_id)
waiter.wait(**wait_request)
# Now, using boto2, load the Image so downstream tagging operations work
# using boto2 classes
log.debug('Image available! Loading boto2.Image for [{}]'.format(
ami_id))
self._ami = self._connection.get_image(ami_id)
except ClientError as e:
if e.response['Error']['Code'] == 'InvalidAMIID.NotFound':
log.debug(
'{0} was not found while waiting for it to become available'
.format(ami_id))
log.error('Error during register_image: {}'.format(e))
return False
else:
# defer to registration_retry decorator
raise e
log.info('AMI registered: {0} {1}'.format(self._ami.id,
self._ami.name))
self._config.context.ami.image = self._ami
return True
def register_image(self, *args, **kwargs):
context = self._config.context
vm_type = context.ami.get("vm_type", "paravirtual")
architecture = context.ami.get("architecture", "x86_64")
cloud_config = self._config.plugins[self.full_name]
self._instance_metadata = get_instance_metadata()
instance_region = self._instance_metadata['placement'][
'availability-zone'][:-1]
region = kwargs.pop(
'region',
context.get('region', cloud_config.get('region', instance_region)))
ami_metadata = {
'name': context.ami.name,
'description': context.ami.description,
'virtualization_type': vm_type,
'architecture': architecture,
'kernel_id': context.base_ami.kernel_id,
'ramdisk_id': context.base_ami.ramdisk_id,
'region': region
}
if 'manifest' in kwargs:
# it's an instance store AMI and needs bucket location
ami_metadata['image_location'] = kwargs['manifest']
else:
# args will be [block_device_map, root_block_device]
block_device_map, root_block_device = args[:2]
bdm = self._make_block_device_map(block_device_map,
root_block_device)
ami_metadata['block_device_map'] = bdm
ami_metadata['block_device_map_list'] = block_device_map
ami_metadata['root_device_name'] = root_block_device
if vm_type == 'hvm':
del ami_metadata['kernel_id']
del ami_metadata['ramdisk_id']
if context.ami.get("enhanced_networking", False):
ami_metadata['sriov_net_support'] = 'simple'
ami_metadata['ena_networking'] = context.ami.get(
'ena_networking', False)
if not self._register_image(**ami_metadata):
return False
return True
def _make_block_device_map(self,
block_device_map,
root_block_device,
delete_on_termination=True):
""" construct boto3 style BlockDeviceMapping """
bdm = []
volume_type = self.context.cloud.get('register_ebs_type', None)
if volume_type is None:
volume_type = self.plugin_config.get('register_ebs_type',
'standard')
rootdev = self.context.base_ami.block_device_mapping[
self.context.base_ami.root_device_name]
volume_size = self.context.ami.get('root_volume_size', None)
if volume_size is None:
volume_size = self.plugin_config.get('root_volume_size', None)
if volume_size is None:
volume_size = rootdev.size
volume_size = int(volume_size)
# root device
root_mapping = {}
root_mapping['DeviceName'] = root_block_device
root_mapping['Ebs'] = {}
root_mapping['Ebs']['SnapshotId'] = self._snapshot.id
root_mapping['Ebs']['VolumeSize'] = volume_size
root_mapping['Ebs']['VolumeType'] = volume_type
root_mapping['Ebs']['DeleteOnTermination'] = delete_on_termination
bdm.append(root_mapping)
# ephemerals
for (os_dev, ec2_dev) in block_device_map:
mapping = {}
mapping['VirtualName'] = ec2_dev
mapping['DeviceName'] = os_dev
bdm.append(mapping)
log.debug('Created BlockDeviceMapping [{}]'.format(bdm))
return bdm
@retry(FinalizerException, tries=3, delay=1, backoff=2, logger=log)
def add_tags(self, resource_type):
context = self._config.context
log.debug('Adding tags for resource type {0}'.format(resource_type))
tags = context[resource_type].get('tags', None)
if not tags:
log.critical('Unable to locate tags for {0}'.format(resource_type))
return False
instance_var = '_' + resource_type
try:
instance = getattr(self, instance_var)
except Exception:
errstr = 'Tagging failed: Unable to find local instance var {0}'.format(
instance_var)
log.debug(errstr, exc_info=True)
log.critical(errstr)
return False
else:
try:
self._connection.create_tags([instance.id], tags)
except EC2ResponseError:
errstr = 'Error creating tags for resource type {0}, id {1}'
errstr = errstr.format(resource_type, instance.id)
log.critical(errstr)
raise FinalizerException(errstr)
else:
log.debug('Successfully tagged {0}({1})'.format(
resource_type, instance.id))
instance.update()
tagstring = '\n'.join('='.join((key, val))
for (key, val) in tags.iteritems())
log.debug('Tags: \n{0}'.format(tagstring))
return True
def attached_block_devices(self, prefix):
log.debug('Checking for currently attached block devices. prefix: {0}'.
format(prefix))
self._instance.update()
if device_prefix(
self._instance.block_device_mapping.keys()[0]) != prefix:
return dict((native_block_device(dev, prefix), mapping) for (
dev,
mapping) in self._instance.block_device_mapping.iteritems())
return self._instance.block_device_mapping
def _resolve_baseami(self):
log.info('Resolving base AMI')
context = self._config.context
cloud_config = self._config.plugins[self.full_name]
try:
ami_id = context.ami.get('base_ami_name',
cloud_config.get('base_ami_name', None))
if ami_id is None:
ami_id = context.ami.get('base_ami_id',
cloud_config.get('base_ami_id', None))
if ami_id is None:
raise RuntimeError(
'Must configure or provide either a base ami name or id'
)
else:
context.ami['ami_id'] = ami_id
baseami = self._lookup_ami_by_id(ami_id)
else:
baseami = self._lookup_ami_by_name(ami_id)
except IndexError:
raise RuntimeError(
'Could not locate base AMI with identifier: {0}'.format(
ami_id))
log.info('Successfully resolved {0.name}({0.id})'.format(baseami))
context['base_ami'] = baseami
def _lookup_ami_by_name(self, ami_name):
ami_details = self._lookup_image_cache(ami_name)
if ami_details:
return ami_details
log.info('looking up base AMI with name {0}'.format(ami_name))
ami_details = self._connection.get_all_images(
filters={'name': ami_name})[0]
self._save_image_cache(ami_name, ami_details)
return ami_details
def _lookup_ami_by_id(self, ami_id):
ami_details = self._lookup_image_cache(ami_id)
if ami_details:
return ami_details
log.info('looking up base AMI with ID {0}'.format(ami_id))
ami_details = self._connection.get_all_images(image_ids=[ami_id])[0]
self._save_image_cache(ami_id, ami_details)
return ami_details
def _lookup_image_cache(self, filename):
cache_file = os.path.join(self._config.aminator_root, "image-cache",
filename)
if os.path.isfile(cache_file):
try:
log.info("loading cached ami details for {0}".format(filename))
with open(cache_file, 'r') as f:
return dill.load(f)
except Exception as e:
log.warning("Failed to parse {0}: {1}".format(cache_file, e))
return None
def _save_image_cache(self, filename, details):
cache_dir = os.path.join(self._config.aminator_root, "image-cache")
cache_file = os.path.join(cache_dir, filename)
mkdir_p(cache_dir)
with open(cache_file, 'w') as f:
dill.dump(details, f)
def __enter__(self):
self.connect()
self._resolve_baseami()
self._instance = Instance(connection=self._connection)
self._instance.id = get_instance_metadata()['instance-id']
self._instance.update()
context = self._config.context
if context.ami.get("base_ami_name", None):
environ["AMINATOR_BASE_AMI_NAME"] = context.ami.base_ami_name
if context.ami.get("base_ami_id", None):
environ["AMINATOR_BASE_AMI_ID"] = context.ami.base_ami_id
if context.cloud.get("region", None):
environ["AMINATOR_REGION"] = context.cloud.region
return self
def setUp(self):
(flexmock(EC2Connection).should_receive('get_key_pair').and_return(
None))
(flexmock(EC2Connection).should_receive('create_key_pair').with_args(
'bookeyname').and_return(KeyPair()))
(flexmock(EC2Connection).should_receive(
'get_all_security_groups').and_return([]))
(flexmock(
EC2Connection).should_receive('create_security_group').with_args(
'boogroup',
'AppScale security group').and_return(SecurityGroup()))
(flexmock(EC2Connection).should_receive(
'authorize_security_group').and_return())
reservation = Reservation()
# the old implementation had a regression where public and private IPs
# were getting sorted, and thus public ip1 would point to private ip 2.
# to prevent this regression from resurfacing, set up the dns names so
# that a sort would mess them up again.
instance1 = Instance()
instance1.public_dns_name = 'ABC-public-ip1'
instance1.private_dns_name = 'DEF-private-ip1'
instance1.id = 'i-id1'
instance1._state.name = 'running'
instance1.key_name = 'bookeyname'
instance2 = Instance()
instance2.public_dns_name = 'DEF-public-ip2'
instance2.private_dns_name = 'ABC-private-ip2'
instance2.id = 'i-id2'
instance2._state.name = 'running'
instance2.key_name = 'bookeyname'
reservation.instances = [instance1, instance2]
(flexmock(EC2Connection).should_receive(
'get_all_instances').and_return([]).and_return([reservation]))
(flexmock(EC2Connection).should_receive('run_instances').and_return())
(flexmock(utils).should_receive('get_secret').and_return('secret'))
(flexmock(utils).should_receive('sleep').and_return())
(flexmock(utils).should_receive('get_random_alphanumeric').and_return(
'0000000000'))
(flexmock(utils).should_receive('write_key_file').and_return())
def setUp(self):
self.instance = Instance()
self.instance.state = 'running'
self.instance.id = 'i-abc'
self.instance.tags = {'Name': 'awesome'}
def endElement(self, name, value, connection):
Instance.endElement(self, name, value, connection)
if name == 'dnsNameV6':
self.dns_name_v6 = value
def __init__(self, connection=None):
Instance.__init__(self, connection)
self.dns_name_v6 = None
class EC2CloudPlugin(BaseCloudPlugin):
_name = 'ec2'
def add_metrics(self, metric_base_name, cls, func_name):
newfunc = succeeds("{0}.count".format(metric_base_name), self)(raises("{0}.error".format(metric_base_name), self)(timer("{0}.duration".format(metric_base_name), self)(getattr(cls, func_name))))
setattr(cls, func_name, newfunc)
def __init__(self):
super(EC2CloudPlugin, self).__init__()
# wrap each of the functions so we can get timer and error metrics
for ec2func in ["create_volume", "create_tags", "register_image", "get_all_images"]:
self.add_metrics("aminator.cloud.ec2.connection.{0}".format(ec2func), EC2Connection, ec2func)
for volfunc in ["add_tag", "attach", "create_snapshot", "delete", "detach", "update"]:
self.add_metrics("aminator.cloud.ec2.volume.{0}".format(volfunc), Volume, volfunc)
for imgfunc in ["update"]:
self.add_metrics("aminator.cloud.ec2.image.{0}".format(imgfunc), Image, imgfunc)
for insfunc in ["update"]:
self.add_metrics("aminator.cloud.ec2.instance.{0}".format(insfunc), Instance, insfunc)
def add_plugin_args(self, *args, **kwargs):
context = self._config.context
base_ami = self._parser.add_argument_group(
title='Base AMI', description='EITHER AMI id OR name, not both!')
base_ami_mutex = base_ami.add_mutually_exclusive_group(required=True)
base_ami_mutex.add_argument(
'-b', '--base-ami-name', dest='base_ami_name',
action=conf_action(config=context.ami),
help='The name of the base AMI used in provisioning')
base_ami_mutex.add_argument(
'-B', '--base-ami-id', dest='base_ami_id',
action=conf_action(config=context.ami),
help='The id of the base AMI used in provisioning')
cloud = self._parser.add_argument_group(
title='EC2 Options', description='EC2 Connection Information')
cloud.add_argument(
'-r', '--region', dest='region',
help='EC2 region (default: us-east-1)',
action=conf_action(config=context.cloud))
cloud.add_argument(
'--boto-secure', dest='is_secure', help='Connect via https',
action=conf_action(config=context.cloud, action='store_true'))
cloud.add_argument(
'--boto-debug', dest='boto_debug', help='Boto debug output',
action=conf_action(config=context.cloud, action='store_true'))
volume_mutex = cloud.add_mutually_exclusive_group()
volume_mutex.add_argument(
'-V', '--volume-id', dest='volume_id',
action=conf_action(config=context.ami),
help='The Base AMI volume id already attached to the system')
volume_mutex.add_argument(
'--provisioner-ebs-type', dest='provisioner_ebs_type',
action=conf_action(config=context.cloud),
help='The type of EBS volume to create from the Base AMI snapshot')
cloud.add_argument(
'--register-ebs-type', dest='register_ebs_type',
action=conf_action(config=context.cloud),
help='The root volume EBS type for AMI registration')
cloud.add_argument(
'--root-volume-size', dest='root_volume_size',
action=conf_action(config=context.ami),
help='Root volume size (in GB). The default is to inherit from the base AMI.')
def configure(self, config, parser):
super(EC2CloudPlugin, self).configure(config, parser)
host = config.context.web_log.get('host', False)
if not host:
md = get_instance_metadata()
pub, ipv4 = 'public-hostname', 'local-ipv4'
config.context.web_log['host'] = md[pub] if pub in md else md[ipv4]
def connect(self, **kwargs):
if self._connection:
log.warn('Already connected to EC2')
else:
log.info('Connecting to EC2')
self._connect(**kwargs)
def _connect(self, **kwargs):
cloud_config = self._config.plugins[self.full_name]
context = self._config.context
self._instance_metadata = get_instance_metadata()
instance_region = self._instance_metadata['placement']['availability-zone'][:-1]
region = kwargs.pop('region', context.get('region', cloud_config.get('region', instance_region)))
log.debug('Establishing connection to region: {0}'.format(region))
context.cloud.setdefault('boto_debug', False)
if context.cloud.boto_debug:
from aminator.config import configure_datetime_logfile
configure_datetime_logfile(self._config, 'boto')
kwargs['debug'] = 1
log.debug('Boto debug logging enabled')
else:
logging.getLogger('boto').setLevel(logging.INFO)
if 'is_secure' not in kwargs:
kwargs['is_secure'] = context.get('is_secure', cloud_config.get('is_secure', True))
self._connection = connect_to_region(region, **kwargs)
log.info('Aminating in region {0}'.format(region))
def allocate_base_volume(self, tag=True):
cloud_config = self._config.plugins[self.full_name]
context = self._config.context
self._volume = Volume(connection=self._connection)
rootdev = context.base_ami.block_device_mapping[context.base_ami.root_device_name]
volume_type = context.cloud.get('provisioner_ebs_type', cloud_config.get('provisioner_ebs_type', 'standard'))
volume_size = context.ami.get('root_volume_size', None)
if volume_size is None:
volume_size = cloud_config.get('root_volume_size', None)
if volume_size is None:
volume_size = rootdev.size
volume_size = int(volume_size)
if volume_size < 1:
raise VolumeException('root_volume_size must be a positive integer, received {}'.format(volume_size))
if volume_size < rootdev.size:
raise VolumeException(
'root_volume_size ({}) must be at least as large as the root '
'volume of the base AMI ({})'.format(volume_size, rootdev.size))
self._volume.id = self._connection.create_volume(
size=volume_size, zone=self._instance.placement,
volume_type=volume_type, snapshot=rootdev.snapshot_id).id
if not self._volume_available():
log.critical('{0}: unavailable.')
return False
if tag:
tags = {
'purpose': cloud_config.get('tag_ami_purpose', 'amination'),
'status': 'busy',
'ami': context.base_ami.id,
'ami-name': context.base_ami.name,
'arch': context.base_ami.architecture,
}
self._connection.create_tags([self._volume.id], tags)
self._volume.update()
log.debug('Volume {0} created'.format(self._volume.id))
@retry(VolumeException, tries=2, delay=1, backoff=2, logger=log)
def attach_volume(self, blockdevice, tag=True):
context = self._config.context
if "volume_id" in context.ami:
volumes = self._connection.get_all_volumes(volume_ids=[context.ami.volume_id])
if not volumes:
raise VolumeException('Failed to find volume: {0}'.format(context.ami.volume_id))
self._volume = volumes[0]
return
self.allocate_base_volume(tag=tag)
# must do this as amazon still wants /dev/sd*
ec2_device_name = blockdevice.replace('xvd', 'sd')
log.debug('Attaching volume {0} to {1}:{2}({3})'.format(self._volume.id, self._instance.id, ec2_device_name, blockdevice))
self._volume.attach(self._instance.id, ec2_device_name)
if not self.is_volume_attached(blockdevice):
log.debug('{0} attachment to {1}:{2}({3}) timed out'.format(self._volume.id, self._instance.id, ec2_device_name, blockdevice))
self._volume.add_tag('status', 'used')
# trigger a retry
raise VolumeException('Timed out waiting for {0} to attach to {1}:{2}'.format(self._volume.id, self._instance.id, blockdevice))
log.debug('Volume {0} attached to {1}:{2}'.format(self._volume.id, self._instance.id, blockdevice))
def is_volume_attached(self, blockdevice):
context = self._config.context
if "volume_id" in context.ami:
return True
try:
self._volume_attached(blockdevice)
except VolumeException:
log.debug('Timed out waiting for volume {0} to attach to {1}:{2}'.format(self._volume.id, self._instance.id, blockdevice))
return False
return True
@retry(VolumeException, tries=10, delay=1, backoff=2, logger=log)
def _volume_attached(self, blockdevice):
status = self._volume.update()
if status != 'in-use':
raise VolumeException('Volume {0} not yet attached to {1}:{2}'.format(self._volume.id, self._instance.id, blockdevice))
elif not os_node_exists(blockdevice):
raise VolumeException('{0} does not exist yet.'.format(blockdevice))
else:
return True
def snapshot_volume(self, description=None):
context = self._config.context
if not description:
description = context.snapshot.get('description', '')
log.debug('Creating snapshot with description {0}'.format(description))
self._snapshot = self._volume.create_snapshot(description)
if not self._snapshot_complete():
log.critical('Failed to create snapshot')
return False
else:
log.debug('Snapshot complete. id: {0}'.format(self._snapshot.id))
return True
def _state_check(self, obj, state):
obj.update()
classname = obj.__class__.__name__
if classname in ('Snapshot', 'Volume'):
if classname == 'Snapshot':
log.debug("Snapshot {0} state: {1}, progress: {2}".format(obj.id, obj.status, obj.progress))
return obj.status == state
else:
return obj.state == state
@retry(VolumeException, tries=600, delay=0.5, backoff=1.5, logger=log, maxdelay=10)
def _wait_for_state(self, resource, state):
if self._state_check(resource, state):
log.debug('{0} reached state {1}'.format(resource.__class__.__name__, state))
return True
else:
raise VolumeException('Timed out waiting for {0} to get to {1}({2})'.format(resource.id, state, resource.status))
@lapse("aminator.cloud.ec2.ami_available.duration")
def _ami_available(self):
return self._wait_for_state(self._ami, 'available')
@lapse("aminator.cloud.ec2.snapshot_completed.duration")
def _snapshot_complete(self):
return self._wait_for_state(self._snapshot, 'completed')
@lapse("aminator.cloud.ec2.volume_available.duration")
def _volume_available(self):
return self._wait_for_state(self._volume, 'available')
def detach_volume(self, blockdevice):
context = self._config.context
if "volume_id" in context.ami:
return
log.debug('Detaching volume {0} from {1}'.format(self._volume.id, self._instance.id))
self._volume.detach()
if not self._volume_detached(blockdevice):
raise VolumeException('Time out waiting for {0} to detach from {1}'.format(self._volume.id, self._instance.id))
log.debug('Successfully detached volume {0} from {1}'.format(self._volume.id, self._instance.id))
@retry(VolumeException, tries=7, delay=1, backoff=2, logger=log)
def _volume_detached(self, blockdevice):
status = self._volume.update()
if status != 'available':
raise VolumeException('Volume {0} not yet detached from {1}'.format(self._volume.id, self._instance.id))
elif os_node_exists(blockdevice):
raise VolumeException('Device node {0} still exists'.format(blockdevice))
else:
return True
def delete_volume(self):
context = self._config.context
if "volume_id" in context.ami:
return True
log.debug('Deleting volume {0}'.format(self._volume.id))
result = self._volume.delete()
if not result:
log.debug('Volume {0} delete returned False, may require manual cleanup'.format(self._volume.id))
else:
log.debug('Volume {0} successfully deleted'.format(self._volume.id))
return result
def is_stale_attachment(self, dev, prefix):
log.debug('Checking for stale attachment. dev: {0}, prefix: {1}'.format(dev, prefix))
if dev in self.attached_block_devices(prefix) and not os_node_exists(dev):
log.debug('{0} is stale, rejecting'.format(dev))
return True
log.debug('{0} not stale, using'.format(dev))
return False
@registration_retry(tries=3, delay=1, backoff=1)
def _register_image(self, **ami_metadata):
"""Register the AMI using boto3/botocore components which supports ENA
This is the only use of boto3 in aminator currently"""
# construct AMI registration payload boto3 style
request = {}
request['Name'] = ami_metadata.get('name', None)
request['Description'] = ami_metadata.get('description', None)
request['Architecture'] = ami_metadata.get('architecture', None)
request['EnaSupport'] = ami_metadata.get('ena_networking', False)
request['VirtualizationType'] = ami_metadata.get('virtualization_type', None)
# when instance store, don't provide botocore expects a string value
if ami_metadata.get('block_device_map') is not None:
request['BlockDeviceMappings'] = ami_metadata.get('block_device_map')
if ami_metadata.get('root_device_name') is not None:
request['RootDeviceName'] = ami_metadata.get('root_device_name')
# only present for instance store
if ami_metadata.get('image_location') is not None:
request['ImageLocation'] = ami_metadata.get('image_location')
# can only be set to 'simple' for hvm. don't include otherwise
if ami_metadata.get('sriov_net_support') is not None:
request['SriovNetSupport'] = ami_metadata.get('sriov_net_support')
if (ami_metadata.get('virtualization_type') == 'paravirtual'):
# KernelId required
request['KernelId'] = ami_metadata.get('kernel_id', None)
if ami_metadata.get('ramdisk_id') is not None:
request['RamdiskId'] = ami_metadata.get('ramdisk_id', None)
# assert we have all the key params. Nothing to _here_ should be None
for key, value in request.items():
if request[key] is None:
raise FinalizerException('{} cannot be None'.format(key))
log.debug('Boto3 registration request data [{}]'.format(request))
try:
client = boto3.client('ec2', region_name=ami_metadata.get('region'))
response = client.register_image(**request)
log.debug('Registration response data [{}]'.format(response))
ami_id = response['ImageId']
if ami_id is None:
return False
log.info('Waiting for [{}] to become available'.format(ami_id))
waiter = client.get_waiter('image_available')
wait_request = {}
wait_request['ImageIds'] = []
wait_request['ImageIds'].append(ami_id)
waiter.wait(**wait_request)
# Now, using boto2, load the Image so downstream tagging operations work
# using boto2 classes
log.debug('Image available! Loading boto2.Image for [{}]'.format(ami_id))
self._ami = self._connection.get_image(ami_id)
except ClientError as e:
if e.response['Error']['Code'] == 'InvalidAMIID.NotFound':
log.debug('{0} was not found while waiting for it to become available'.format(ami_id))
log.error('Error during register_image: {}'.format(e))
return False
else:
# defer to registration_retry decorator
raise e
log.info('AMI registered: {0} {1}'.format(self._ami.id, self._ami.name))
self._config.context.ami.image = self._ami
return True
def register_image(self, *args, **kwargs):
context = self._config.context
vm_type = context.ami.get("vm_type", "paravirtual")
architecture = context.ami.get("architecture", "x86_64")
cloud_config = self._config.plugins[self.full_name]
self._instance_metadata = get_instance_metadata()
instance_region = self._instance_metadata['placement']['availability-zone'][:-1]
region = kwargs.pop('region', context.get('region', cloud_config.get('region', instance_region)))
ami_metadata = {
'name': context.ami.name,
'description': context.ami.description,
'virtualization_type': vm_type,
'architecture': architecture,
'kernel_id': context.base_ami.kernel_id,
'ramdisk_id': context.base_ami.ramdisk_id,
'region': region
}
if 'manifest' in kwargs:
# it's an instance store AMI and needs bucket location
ami_metadata['image_location'] = kwargs['manifest']
else:
# args will be [block_device_map, root_block_device]
block_device_map, root_block_device = args[:2]
bdm = self._make_block_device_map(block_device_map, root_block_device)
ami_metadata['block_device_map'] = bdm
ami_metadata['block_device_map_list'] = block_device_map
ami_metadata['root_device_name'] = root_block_device
if vm_type == 'hvm':
del ami_metadata['kernel_id']
del ami_metadata['ramdisk_id']
if context.ami.get("enhanced_networking", False):
ami_metadata['sriov_net_support'] = 'simple'
ami_metadata['ena_networking'] = context.ami.get('ena_networking', False)
if not self._register_image(**ami_metadata):
return False
return True
def _make_block_device_map(self, block_device_map, root_block_device, delete_on_termination=True):
""" construct boto3 style BlockDeviceMapping """
bdm = []
volume_type = self.context.cloud.get('register_ebs_type', None)
if volume_type is None:
volume_type = self.plugin_config.get('register_ebs_type', 'standard')
rootdev = self.context.base_ami.block_device_mapping[self.context.base_ami.root_device_name]
volume_size = self.context.ami.get('root_volume_size', None)
if volume_size is None:
volume_size = self.plugin_config.get('root_volume_size', None)
if volume_size is None:
volume_size = rootdev.size
volume_size = int(volume_size)
# root device
root_mapping = {}
root_mapping['DeviceName'] = root_block_device
root_mapping['Ebs'] = {}
root_mapping['Ebs']['SnapshotId'] = self._snapshot.id
root_mapping['Ebs']['VolumeSize'] = volume_size
root_mapping['Ebs']['VolumeType'] = volume_type
root_mapping['Ebs']['DeleteOnTermination'] = delete_on_termination
bdm.append(root_mapping)
# ephemerals
for (os_dev, ec2_dev) in block_device_map:
mapping = {}
mapping['VirtualName'] = ec2_dev
mapping['DeviceName'] = os_dev
bdm.append(mapping)
log.debug('Created BlockDeviceMapping [{}]'.format(bdm))
return bdm
@retry(FinalizerException, tries=3, delay=1, backoff=2, logger=log)
def add_tags(self, resource_type):
context = self._config.context
log.debug('Adding tags for resource type {0}'.format(resource_type))
tags = context[resource_type].get('tags', None)
if not tags:
log.critical('Unable to locate tags for {0}'.format(resource_type))
return False
instance_var = '_' + resource_type
try:
instance = getattr(self, instance_var)
except Exception:
errstr = 'Tagging failed: Unable to find local instance var {0}'.format(instance_var)
log.debug(errstr, exc_info=True)
log.critical(errstr)
return False
else:
try:
self._connection.create_tags([instance.id], tags)
except EC2ResponseError:
errstr = 'Error creating tags for resource type {0}, id {1}'
errstr = errstr.format(resource_type, instance.id)
log.critical(errstr)
raise FinalizerException(errstr)
else:
log.debug('Successfully tagged {0}({1})'.format(resource_type, instance.id))
instance.update()
tagstring = '\n'.join('='.join((key, val)) for (key, val) in tags.iteritems())
log.debug('Tags: \n{0}'.format(tagstring))
return True
def attached_block_devices(self, prefix):
log.debug('Checking for currently attached block devices. prefix: {0}'.format(prefix))
self._instance.update()
if device_prefix(self._instance.block_device_mapping.keys()[0]) != prefix:
return dict((native_block_device(dev, prefix), mapping) for (dev, mapping) in self._instance.block_device_mapping.iteritems())
return self._instance.block_device_mapping
def _resolve_baseami(self):
log.info('Resolving base AMI')
context = self._config.context
cloud_config = self._config.plugins[self.full_name]
try:
ami_id = context.ami.get('base_ami_name', cloud_config.get('base_ami_name', None))
if ami_id is None:
ami_id = context.ami.get('base_ami_id', cloud_config.get('base_ami_id', None))
if ami_id is None:
raise RuntimeError('Must configure or provide either a base ami name or id')
else:
context.ami['ami_id'] = ami_id
baseami = self._lookup_ami_by_id(ami_id)
else:
baseami = self._lookup_ami_by_name(ami_id)
except IndexError:
raise RuntimeError('Could not locate base AMI with identifier: {0}'.format(ami_id))
log.info('Successfully resolved {0.name}({0.id})'.format(baseami))
context['base_ami'] = baseami
def _lookup_ami_by_name(self, ami_name):
ami_details = self._lookup_image_cache(ami_name)
if ami_details:
return ami_details
log.info('looking up base AMI with name {0}'.format(ami_name))
ami_details = self._connection.get_all_images(filters={'name': ami_name})[0]
self._save_image_cache(ami_name, ami_details)
return ami_details
def _lookup_ami_by_id(self, ami_id):
ami_details = self._lookup_image_cache(ami_id)
if ami_details:
return ami_details
log.info('looking up base AMI with ID {0}'.format(ami_id))
ami_details = self._connection.get_all_images(image_ids=[ami_id])[0]
self._save_image_cache(ami_id, ami_details)
return ami_details
def _lookup_image_cache(self, filename):
cache_file = os.path.join(self._config.aminator_root, "image-cache", filename)
if os.path.isfile(cache_file):
try:
log.info("loading cached ami details for {0}".format(filename))
with open(cache_file, 'r') as f:
return dill.load(f)
except Exception as e:
log.warning("Failed to parse {0}: {1}".format(cache_file, e))
return None
def _save_image_cache(self, filename, details):
cache_dir = os.path.join(self._config.aminator_root, "image-cache")
cache_file = os.path.join(cache_dir, filename)
mkdir_p(cache_dir)
with open(cache_file, 'w') as f:
dill.dump(details, f)
def __enter__(self):
self.connect()
self._resolve_baseami()
self._instance = Instance(connection=self._connection)
self._instance.id = get_instance_metadata()['instance-id']
self._instance.update()
context = self._config.context
if context.ami.get("base_ami_name", None):
environ["AMINATOR_BASE_AMI_NAME"] = context.ami.base_ami_name
if context.ami.get("base_ami_id", None):
environ["AMINATOR_BASE_AMI_ID"] = context.ami.base_ami_id
if context.cloud.get("region", None):
environ["AMINATOR_REGION"] = context.cloud.region
return self
def setUp(self):
self.instance = Instance()
self.instance._state = RUNNING_STATE
self.instance.id = 'i-abc'
self.instance.tags = {'Name': 'awesome'}
