| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 1 | # -*- coding: utf-8 -*- |
| 2 | |
| 3 | ## |
| 4 | # Copyright 2017 xFlow Research Pvt. Ltd |
| 5 | # This file is part of openmano |
| 6 | # All Rights Reserved. |
| 7 | # |
| 8 | # Licensed under the Apache License, Version 2.0 (the "License"); you may |
| 9 | # not use this file except in compliance with the License. You may obtain |
| 10 | # a copy of the License at |
| 11 | # |
| 12 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 13 | # |
| 14 | # Unless required by applicable law or agreed to in writing, software |
| 15 | # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| 16 | # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| 17 | # License for the specific language governing permissions and limitations |
| 18 | # under the License. |
| 19 | # |
| 20 | # For those usages not covered by the Apache License, Version 2.0 please |
| 21 | # contact with: saboor.ahmad@xflowresearch.com |
| 22 | ## |
| 23 | |
| 24 | ''' |
| 25 | AWS-connector implements all the methods to interact with AWS using the BOTO client |
| 26 | ''' |
| 27 | |
| 28 | __author__ = "Saboor Ahmad" |
| 29 | __date__ = "10-Apr-2017" |
| 30 | |
| 31 | import vimconn |
| 32 | import yaml |
| 33 | import logging |
| 34 | import netaddr |
| 35 | import time |
| 36 | |
| 37 | try: |
| 38 | import boto |
| 39 | import boto.ec2 |
| 40 | import boto.vpc |
| 41 | except: |
| 42 | exit("Boto not avialable. Try activating your virtualenv OR `pip install boto`") |
| 43 | |
| 44 | |
| 45 | class vimconnector(vimconn.vimconnector): |
| 46 | def __init__(self, uuid, name, tenant_id, tenant_name, url, url_admin=None, user=None, passwd=None, log_level=None, |
| Vance Shipley | ae26697 | 2017-05-12 01:58:43 +0530 | [diff] [blame] | 47 | config={}, persistent_info={}): |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 48 | """ Params: uuid - id asigned to this VIM |
| 49 | name - name assigned to this VIM, can be used for logging |
| 50 | tenant_id - ID to be used for tenant |
| 51 | tenant_name - name of tenant to be used VIM tenant to be used |
| 52 | url_admin - optional, url used for administrative tasks |
| 53 | user - credentials of the VIM user |
| 54 | passwd - credentials of the VIM user |
| 55 | log_level - if must use a different log_level than the general one |
| 56 | config - dictionary with misc VIM information |
| 57 | region_name - name of region to deploy the instances |
| 58 | vpc_cidr_block - default CIDR block for VPC |
| 59 | security_groups - default security group to specify this instance |
| Vance Shipley | ae26697 | 2017-05-12 01:58:43 +0530 | [diff] [blame] | 60 | persistent_info - dict where the class can store information that will be available among class |
| 61 | destroy/creation cycles. This info is unique per VIM/credential. At first call it will contain an |
| 62 | empty dict. Useful to store login/tokens information for speed up communication |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 63 | """ |
| 64 | |
| 65 | vimconn.vimconnector.__init__(self, uuid, name, tenant_id, tenant_name, url, url_admin, user, passwd, log_level, |
| Vance Shipley | ae26697 | 2017-05-12 01:58:43 +0530 | [diff] [blame] | 66 | config, persistent_info) |
| 67 | |
| 68 | self.persistent_info = persistent_info |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 69 | self.a_creds = {} |
| 70 | if user: |
| 71 | self.a_creds['aws_access_key_id'] = user |
| 72 | else: |
| 73 | raise vimconn.vimconnAuthException("Username is not specified") |
| 74 | if passwd: |
| 75 | self.a_creds['aws_secret_access_key'] = passwd |
| 76 | else: |
| 77 | raise vimconn.vimconnAuthException("Password is not specified") |
| 78 | if 'region_name' in config: |
| 79 | self.region = config.get('region_name') |
| 80 | else: |
| tierno | a357269 | 2018-05-14 13:09:33 +0200 | [diff] [blame] | 81 | raise vimconn.vimconnException("AWS region_name is not specified at config") |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 82 | |
| 83 | self.vpc_data = {} |
| 84 | self.subnet_data = {} |
| 85 | self.conn = None |
| 86 | self.conn_vpc = None |
| 87 | self.account_id = None |
| 88 | |
| 89 | self.vpc_id = self.get_tenant_list()[0]['id'] |
| 90 | # we take VPC CIDR block if specified, otherwise we use the default CIDR |
| 91 | # block suggested by AWS while creating instance |
| 92 | self.vpc_cidr_block = '10.0.0.0/24' |
| 93 | |
| 94 | if tenant_id: |
| 95 | self.vpc_id = tenant_id |
| 96 | if 'vpc_cidr_block' in config: |
| 97 | self.vpc_cidr_block = config['vpc_cidr_block'] |
| 98 | |
| 99 | self.security_groups = None |
| 100 | if 'security_groups' in config: |
| 101 | self.security_groups = config['security_groups'] |
| 102 | |
| 103 | self.key_pair = None |
| 104 | if 'key_pair' in config: |
| 105 | self.key_pair = config['key_pair'] |
| 106 | |
| 107 | self.flavor_info = None |
| 108 | if 'flavor_info' in config: |
| 109 | flavor_data = config.get('flavor_info') |
| 110 | if isinstance(flavor_data, str): |
| 111 | try: |
| tierno | a357269 | 2018-05-14 13:09:33 +0200 | [diff] [blame] | 112 | if flavor_data[0] == "@": # read from a file |
| 113 | with open(flavor_data[1:], 'r') as stream: |
| 114 | self.flavor_info = yaml.load(stream) |
| 115 | else: |
| 116 | self.flavor_info = yaml.load(flavor_data) |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 117 | except yaml.YAMLError as e: |
| 118 | self.flavor_info = None |
| 119 | raise vimconn.vimconnException("Bad format at file '{}': {}".format(flavor_data[1:], e)) |
| 120 | except IOError as e: |
| 121 | raise vimconn.vimconnException("Error reading file '{}': {}".format(flavor_data[1:], e)) |
| 122 | elif isinstance(flavor_data, dict): |
| tierno | f44e9e5 | 2017-06-13 18:48:30 +0200 | [diff] [blame] | 123 | self.flavor_info = flavor_data |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 124 | |
| 125 | self.logger = logging.getLogger('openmano.vim.aws') |
| 126 | if log_level: |
| 127 | self.logger.setLevel(getattr(logging, log_level)) |
| 128 | |
| 129 | def __setitem__(self, index, value): |
| 130 | """Params: index - name of value of set |
| 131 | value - value to set |
| 132 | """ |
| 133 | if index == 'user': |
| 134 | self.a_creds['aws_access_key_id'] = value |
| 135 | elif index == 'passwd': |
| 136 | self.a_creds['aws_secret_access_key'] = value |
| 137 | elif index == 'region': |
| 138 | self.region = value |
| 139 | else: |
| 140 | vimconn.vimconnector.__setitem__(self, index, value) |
| 141 | |
| 142 | def _reload_connection(self): |
| 143 | """Returns: sets boto.EC2 and boto.VPC connection to work with AWS services |
| 144 | """ |
| 145 | |
| 146 | try: |
| 147 | self.conn = boto.ec2.connect_to_region(self.region, aws_access_key_id=self.a_creds['aws_access_key_id'], |
| 148 | aws_secret_access_key=self.a_creds['aws_secret_access_key']) |
| 149 | self.conn_vpc = boto.vpc.connect_to_region(self.region, aws_access_key_id=self.a_creds['aws_access_key_id'], |
| 150 | aws_secret_access_key=self.a_creds['aws_secret_access_key']) |
| 151 | # client = boto3.client("sts", aws_access_key_id=self.a_creds['aws_access_key_id'], aws_secret_access_key=self.a_creds['aws_secret_access_key']) |
| 152 | # self.account_id = client.get_caller_identity()["Account"] |
| 153 | except Exception as e: |
| 154 | self.format_vimconn_exception(e) |
| 155 | |
| 156 | def format_vimconn_exception(self, e): |
| 157 | """Params: an Exception object |
| 158 | Returns: Raises the exception 'e' passed in mehtod parameters |
| 159 | """ |
| 160 | |
| 161 | self.conn = None |
| 162 | self.conn_vpc = None |
| 163 | raise vimconn.vimconnConnectionException(type(e).__name__ + ": " + str(e)) |
| 164 | |
| 165 | def get_availability_zones_list(self): |
| 166 | """Obtain AvailabilityZones from AWS |
| 167 | """ |
| 168 | |
| 169 | try: |
| 170 | self._reload_connection() |
| 171 | az_list = [] |
| 172 | for az in self.conn.get_all_zones(): |
| 173 | az_list.append(az.name) |
| 174 | return az_list |
| 175 | except Exception as e: |
| 176 | self.format_vimconn_exception(e) |
| 177 | |
| 178 | def get_tenant_list(self, filter_dict={}): |
| 179 | """Obtain tenants of VIM |
| 180 | filter_dict dictionary that can contain the following keys: |
| 181 | name: filter by tenant name |
| 182 | id: filter by tenant uuid/id |
| 183 | <other VIM specific> |
| 184 | Returns the tenant list of dictionaries, and empty list if no tenant match all the filers: |
| 185 | [{'name':'<name>, 'id':'<id>, ...}, ...] |
| 186 | """ |
| 187 | |
| 188 | try: |
| 189 | self._reload_connection() |
| 190 | vpc_ids = [] |
| 191 | tfilters = {} |
| 192 | if filter_dict != {}: |
| 193 | if 'id' in filter_dict: |
| 194 | vpc_ids.append(filter_dict['id']) |
| 195 | tfilters['name'] = filter_dict['id'] |
| 196 | tenants = self.conn_vpc.get_all_vpcs(vpc_ids, tfilters) |
| 197 | tenant_list = [] |
| 198 | for tenant in tenants: |
| 199 | tenant_list.append({'id': str(tenant.id), 'name': str(tenant.id), 'status': str(tenant.state), |
| 200 | 'cidr_block': str(tenant.cidr_block)}) |
| 201 | return tenant_list |
| 202 | except Exception as e: |
| 203 | self.format_vimconn_exception(e) |
| 204 | |
| 205 | def new_tenant(self, tenant_name, tenant_description): |
| 206 | """Adds a new tenant to VIM with this name and description, this is done using admin_url if provided |
| 207 | "tenant_name": string max lenght 64 |
| 208 | "tenant_description": string max length 256 |
| 209 | returns the tenant identifier or raise exception |
| 210 | """ |
| 211 | |
| 212 | self.logger.debug("Adding a new VPC") |
| 213 | try: |
| 214 | self._reload_connection() |
| 215 | vpc = self.conn_vpc.create_vpc(self.vpc_cidr_block) |
| 216 | self.conn_vpc.modify_vpc_attribute(vpc.id, enable_dns_support=True) |
| 217 | self.conn_vpc.modify_vpc_attribute(vpc.id, enable_dns_hostnames=True) |
| 218 | |
| 219 | gateway = self.conn_vpc.create_internet_gateway() |
| 220 | self.conn_vpc.attach_internet_gateway(gateway.id, vpc.id) |
| 221 | route_table = self.conn_vpc.create_route_table(vpc.id) |
| 222 | self.conn_vpc.create_route(route_table.id, '0.0.0.0/0', gateway.id) |
| 223 | |
| 224 | self.vpc_data[vpc.id] = {'gateway': gateway.id, 'route_table': route_table.id, |
| 225 | 'subnets': self.subnet_sizes(len(self.get_availability_zones_list()), |
| 226 | self.vpc_cidr_block)} |
| 227 | return vpc.id |
| 228 | except Exception as e: |
| 229 | self.format_vimconn_exception(e) |
| 230 | |
| 231 | def delete_tenant(self, tenant_id): |
| 232 | """Delete a tenant from VIM |
| 233 | tenant_id: returned VIM tenant_id on "new_tenant" |
| 234 | Returns None on success. Raises and exception of failure. If tenant is not found raises vimconnNotFoundException |
| 235 | """ |
| 236 | |
| 237 | self.logger.debug("Deleting specified VPC") |
| 238 | try: |
| 239 | self._reload_connection() |
| 240 | vpc = self.vpc_data.get(tenant_id) |
| 241 | if 'gateway' in vpc and 'route_table' in vpc: |
| 242 | gateway_id, route_table_id = vpc['gateway'], vpc['route_table'] |
| 243 | self.conn_vpc.detach_internet_gateway(gateway_id, tenant_id) |
| 244 | self.conn_vpc.delete_vpc(tenant_id) |
| 245 | self.conn_vpc.delete_route(route_table_id, '0.0.0.0/0') |
| 246 | else: |
| 247 | self.conn_vpc.delete_vpc(tenant_id) |
| 248 | except Exception as e: |
| 249 | self.format_vimconn_exception(e) |
| 250 | |
| 251 | def subnet_sizes(self, availability_zones, cidr): |
| 252 | """Calcualtes possible subnets given CIDR value of VPC |
| 253 | """ |
| 254 | |
| 255 | if availability_zones != 2 and availability_zones != 3: |
| 256 | self.logger.debug("Number of AZs should be 2 or 3") |
| 257 | raise vimconn.vimconnNotSupportedException("Number of AZs should be 2 or 3") |
| 258 | |
| 259 | netmasks = ('255.255.252.0', '255.255.254.0', '255.255.255.0', '255.255.255.128') |
| 260 | ip = netaddr.IPNetwork(cidr) |
| 261 | mask = ip.netmask |
| 262 | |
| 263 | if str(mask) not in netmasks: |
| 264 | self.logger.debug("Netmask " + str(mask) + " not found") |
| 265 | raise vimconn.vimconnNotFoundException("Netmask " + str(mask) + " not found") |
| 266 | |
| 267 | if availability_zones == 2: |
| 268 | for n, netmask in enumerate(netmasks): |
| 269 | if str(mask) == netmask: |
| 270 | subnets = list(ip.subnet(n + 24)) |
| 271 | else: |
| 272 | for n, netmask in enumerate(netmasks): |
| 273 | if str(mask) == netmask: |
| 274 | pub_net = list(ip.subnet(n + 24)) |
| 275 | pri_subs = pub_net[1:] |
| 276 | pub_mask = pub_net[0].netmask |
| 277 | pub_split = list(ip.subnet(26)) if (str(pub_mask) == '255.255.255.0') else list(ip.subnet(27)) |
| 278 | pub_subs = pub_split[:3] |
| 279 | subnets = pub_subs + pri_subs |
| 280 | |
| 281 | return map(str, subnets) |
| 282 | |
| 283 | def new_network(self, net_name, net_type, ip_profile=None, shared=False, vlan=None): |
| 284 | """Adds a tenant network to VIM |
| 285 | Params: |
| 286 | 'net_name': name of the network |
| 287 | 'net_type': one of: |
| 288 | 'bridge': overlay isolated network |
| 289 | 'data': underlay E-LAN network for Passthrough and SRIOV interfaces |
| 290 | 'ptp': underlay E-LINE network for Passthrough and SRIOV interfaces. |
| 291 | 'ip_profile': is a dict containing the IP parameters of the network (Currently only IPv4 is implemented) |
| 292 | 'ip-version': can be one of ["IPv4","IPv6"] |
| 293 | 'subnet-address': ip_prefix_schema, that is X.X.X.X/Y |
| 294 | 'gateway-address': (Optional) ip_schema, that is X.X.X.X |
| 295 | 'dns-address': (Optional) ip_schema, |
| 296 | 'dhcp': (Optional) dict containing |
| 297 | 'enabled': {"type": "boolean"}, |
| 298 | 'start-address': ip_schema, first IP to grant |
| 299 | 'count': number of IPs to grant. |
| 300 | 'shared': if this network can be seen/use by other tenants/organization |
| 301 | 'vlan': in case of a data or ptp net_type, the intended vlan tag to be used for the network |
| garciadeblas | ebd6672 | 2019-01-31 16:01:31 +0000 | [diff] [blame] | 302 | Returns a tuple with the network identifier and created_items, or raises an exception on error |
| 303 | created_items can be None or a dictionary where this method can include key-values that will be passed to |
| 304 | the method delete_network. Can be used to store created segments, created l2gw connections, etc. |
| 305 | Format is vimconnector dependent, but do not use nested dictionaries and a value of None should be the same |
| 306 | as not present. |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 307 | """ |
| 308 | |
| 309 | self.logger.debug("Adding a subnet to VPC") |
| 310 | try: |
| garciadeblas | ebd6672 | 2019-01-31 16:01:31 +0000 | [diff] [blame] | 311 | created_items = {} |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 312 | self._reload_connection() |
| 313 | subnet = None |
| 314 | vpc_id = self.vpc_id |
| 315 | if self.vpc_data.get(vpc_id, None): |
| 316 | cidr_block = list(set(self.vpc_data[vpc_id]['subnets']) - set(self.get_network_details({'tenant_id': vpc_id}, detail='cidr_block')))[0] |
| 317 | else: |
| 318 | vpc = self.get_tenant_list({'id': vpc_id})[0] |
| 319 | subnet_list = self.subnet_sizes(len(self.get_availability_zones_list()), vpc['cidr_block']) |
| 320 | cidr_block = list(set(subnet_list) - set(self.get_network_details({'tenant_id': vpc['id']}, detail='cidr_block')))[0] |
| 321 | subnet = self.conn_vpc.create_subnet(vpc_id, cidr_block) |
| garciadeblas | ebd6672 | 2019-01-31 16:01:31 +0000 | [diff] [blame] | 322 | return subnet.id, created_items |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 323 | except Exception as e: |
| 324 | self.format_vimconn_exception(e) |
| 325 | |
| 326 | def get_network_details(self, filters, detail): |
| 327 | """Get specified details related to a subnet |
| 328 | """ |
| 329 | detail_list = [] |
| 330 | subnet_list = self.get_network_list(filters) |
| 331 | for net in subnet_list: |
| 332 | detail_list.append(net[detail]) |
| 333 | return detail_list |
| 334 | |
| 335 | def get_network_list(self, filter_dict={}): |
| 336 | """Obtain tenant networks of VIM |
| 337 | Params: |
| 338 | 'filter_dict' (optional) contains entries to return only networks that matches ALL entries: |
| 339 | name: string => returns only networks with this name |
| 340 | id: string => returns networks with this VIM id, this imply returns one network at most |
| 341 | shared: boolean >= returns only networks that are (or are not) shared |
| 342 | tenant_id: sting => returns only networks that belong to this tenant/project |
| 343 | ,#(not used yet) admin_state_up: boolean => returns only networks that are (or are not) in admin state active |
| 344 | #(not used yet) status: 'ACTIVE','ERROR',... => filter networks that are on this status |
| 345 | Returns the network list of dictionaries. each dictionary contains: |
| 346 | 'id': (mandatory) VIM network id |
| 347 | 'name': (mandatory) VIM network name |
| 348 | 'status': (mandatory) can be 'ACTIVE', 'INACTIVE', 'DOWN', 'BUILD', 'ERROR', 'VIM_ERROR', 'OTHER' |
| 349 | 'error_msg': (optional) text that explains the ERROR status |
| 350 | other VIM specific fields: (optional) whenever possible using the same naming of filter_dict param |
| 351 | List can be empty if no network map the filter_dict. Raise an exception only upon VIM connectivity, |
| 352 | authorization, or some other unspecific error |
| 353 | """ |
| 354 | |
| 355 | self.logger.debug("Getting all subnets from VIM") |
| 356 | try: |
| 357 | self._reload_connection() |
| 358 | tfilters = {} |
| 359 | if filter_dict != {}: |
| 360 | if 'tenant_id' in filter_dict: |
| 361 | tfilters['vpcId'] = filter_dict['tenant_id'] |
| montesmoreno | c6f2681 | 2017-07-12 11:47:13 +0200 | [diff] [blame] | 362 | subnets = self.conn_vpc.get_all_subnets(subnet_ids=filter_dict.get('name', None), filters=tfilters) |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 363 | net_list = [] |
| 364 | for net in subnets: |
| 365 | net_list.append( |
| 366 | {'id': str(net.id), 'name': str(net.id), 'status': str(net.state), 'vpc_id': str(net.vpc_id), |
| montesmoreno | c6f2681 | 2017-07-12 11:47:13 +0200 | [diff] [blame] | 367 | 'cidr_block': str(net.cidr_block), 'type': 'bridge'}) |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 368 | return net_list |
| 369 | except Exception as e: |
| 370 | self.format_vimconn_exception(e) |
| 371 | |
| 372 | def get_network(self, net_id): |
| 373 | """Obtain network details from the 'net_id' VIM network |
| 374 | Return a dict that contains: |
| 375 | 'id': (mandatory) VIM network id, that is, net_id |
| 376 | 'name': (mandatory) VIM network name |
| 377 | 'status': (mandatory) can be 'ACTIVE', 'INACTIVE', 'DOWN', 'BUILD', 'ERROR', 'VIM_ERROR', 'OTHER' |
| 378 | 'error_msg': (optional) text that explains the ERROR status |
| 379 | other VIM specific fields: (optional) whenever possible using the same naming of filter_dict param |
| 380 | Raises an exception upon error or when network is not found |
| 381 | """ |
| 382 | |
| 383 | self.logger.debug("Getting Subnet from VIM") |
| 384 | try: |
| 385 | self._reload_connection() |
| 386 | subnet = self.conn_vpc.get_all_subnets(net_id)[0] |
| 387 | return {'id': str(subnet.id), 'name': str(subnet.id), 'status': str(subnet.state), |
| 388 | 'vpc_id': str(subnet.vpc_id), 'cidr_block': str(subnet.cidr_block)} |
| 389 | except Exception as e: |
| 390 | self.format_vimconn_exception(e) |
| 391 | |
| garciadeblas | ebd6672 | 2019-01-31 16:01:31 +0000 | [diff] [blame] | 392 | def delete_network(self, net_id, created_items=None): |
| 393 | """ |
| 394 | Removes a tenant network from VIM and its associated elements |
| 395 | :param net_id: VIM identifier of the network, provided by method new_network |
| 396 | :param created_items: dictionary with extra items to be deleted. provided by method new_network |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 397 | Returns the network identifier or raises an exception upon error or when network is not found |
| 398 | """ |
| 399 | |
| 400 | self.logger.debug("Deleting subnet from VIM") |
| 401 | try: |
| 402 | self._reload_connection() |
| 403 | self.logger.debug("DELETING NET_ID: " + str(net_id)) |
| 404 | self.conn_vpc.delete_subnet(net_id) |
| 405 | return net_id |
| 406 | except Exception as e: |
| 407 | self.format_vimconn_exception(e) |
| 408 | |
| 409 | def refresh_nets_status(self, net_list): |
| 410 | """Get the status of the networks |
| 411 | Params: |
| 412 | 'net_list': a list with the VIM network id to be get the status |
| 413 | Returns a dictionary with: |
| 414 | 'net_id': #VIM id of this network |
| 415 | status: #Mandatory. Text with one of: |
| 416 | # DELETED (not found at vim) |
| 417 | # VIM_ERROR (Cannot connect to VIM, authentication problems, VIM response error, ...) |
| 418 | # OTHER (Vim reported other status not understood) |
| 419 | # ERROR (VIM indicates an ERROR status) |
| 420 | # ACTIVE, INACTIVE, DOWN (admin down), |
| 421 | # BUILD (on building process) |
| 422 | error_msg: #Text with VIM error message, if any. Or the VIM connection ERROR |
| 423 | vim_info: #Text with plain information obtained from vim (yaml.safe_dump) |
| 424 | 'net_id2': ... |
| 425 | """ |
| 426 | |
| 427 | self._reload_connection() |
| 428 | try: |
| 429 | dict_entry = {} |
| 430 | for net_id in net_list: |
| 431 | subnet_dict = {} |
| 432 | subnet = None |
| 433 | try: |
| 434 | subnet = self.conn_vpc.get_all_subnets(net_id)[0] |
| 435 | if subnet.state == "pending": |
| 436 | subnet_dict['status'] = "BUILD" |
| 437 | elif subnet.state == "available": |
| 438 | subnet_dict['status'] = 'ACTIVE' |
| 439 | else: |
| 440 | subnet_dict['status'] = 'ERROR' |
| 441 | subnet_dict['error_msg'] = '' |
| 442 | except Exception as e: |
| 443 | subnet_dict['status'] = 'DELETED' |
| 444 | subnet_dict['error_msg'] = 'Network not found' |
| 445 | finally: |
| 446 | try: |
| 447 | subnet_dict['vim_info'] = yaml.safe_dump(subnet, default_flow_style=True, width=256) |
| 448 | except yaml.YAMLError as e: |
| 449 | subnet_dict['vim_info'] = str(subnet) |
| 450 | dict_entry[net_id] = subnet_dict |
| 451 | return dict_entry |
| 452 | except Exception as e: |
| 453 | self.format_vimconn_exception(e) |
| 454 | |
| 455 | def get_flavor(self, flavor_id): |
| 456 | """Obtain flavor details from the VIM |
| 457 | Returns the flavor dict details {'id':<>, 'name':<>, other vim specific } |
| 458 | Raises an exception upon error or if not found |
| 459 | """ |
| 460 | |
| 461 | self.logger.debug("Getting instance type") |
| 462 | try: |
| 463 | if flavor_id in self.flavor_info: |
| 464 | return self.flavor_info[flavor_id] |
| 465 | else: |
| 466 | raise vimconn.vimconnNotFoundException("Cannot find flavor with this flavor ID/Name") |
| 467 | except Exception as e: |
| 468 | self.format_vimconn_exception(e) |
| 469 | |
| 470 | def get_flavor_id_from_data(self, flavor_dict): |
| 471 | """Obtain flavor id that match the flavor description |
| 472 | Params: |
| 473 | 'flavor_dict': dictionary that contains: |
| 474 | 'disk': main hard disk in GB |
| 475 | 'ram': memory in MB |
| 476 | 'vcpus': number of virtual cpus |
| 477 | #todo: complete parameters for EPA |
| 478 | Returns the flavor_id or raises a vimconnNotFoundException |
| 479 | """ |
| 480 | |
| 481 | self.logger.debug("Getting flavor id from data") |
| 482 | try: |
| 483 | flavor = None |
| 484 | for key, values in self.flavor_info.iteritems(): |
| Vance Shipley | 8c4285d | 2017-05-12 04:02:50 +0530 | [diff] [blame] | 485 | if (values["ram"], values["cpus"], values["disk"]) == ( |
| 486 | flavor_dict["ram"], flavor_dict["vcpus"], flavor_dict["disk"]): |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 487 | flavor = (key, values) |
| 488 | break |
| Vance Shipley | 8c4285d | 2017-05-12 04:02:50 +0530 | [diff] [blame] | 489 | elif (values["ram"], values["cpus"], values["disk"]) >= ( |
| 490 | flavor_dict["ram"], flavor_dict["vcpus"], flavor_dict["disk"]): |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 491 | if not flavor: |
| 492 | flavor = (key, values) |
| 493 | else: |
| Vance Shipley | 8c4285d | 2017-05-12 04:02:50 +0530 | [diff] [blame] | 494 | if (flavor[1]["ram"], flavor[1]["cpus"], flavor[1]["disk"]) >= ( |
| 495 | values["ram"], values["cpus"], values["disk"]): |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 496 | flavor = (key, values) |
| 497 | if flavor: |
| 498 | return flavor[0] |
| 499 | raise vimconn.vimconnNotFoundException("Cannot find flavor with this flavor ID/Name") |
| 500 | except Exception as e: |
| 501 | self.format_vimconn_exception(e) |
| 502 | |
| 503 | def new_image(self, image_dict): |
| 504 | """ Adds a tenant image to VIM |
| 505 | Params: image_dict |
| 506 | name (string) - The name of the AMI. Valid only for EBS-based images. |
| 507 | description (string) - The description of the AMI. |
| 508 | image_location (string) - Full path to your AMI manifest in Amazon S3 storage. Only used for S3-based AMI’s. |
| 509 | architecture (string) - The architecture of the AMI. Valid choices are: * i386 * x86_64 |
| 510 | kernel_id (string) - The ID of the kernel with which to launch the instances |
| 511 | root_device_name (string) - The root device name (e.g. /dev/sdh) |
| 512 | block_device_map (boto.ec2.blockdevicemapping.BlockDeviceMapping) - A BlockDeviceMapping data structure describing the EBS volumes associated with the Image. |
| 513 | virtualization_type (string) - The virutalization_type of the image. Valid choices are: * paravirtual * hvm |
| 514 | sriov_net_support (string) - Advanced networking support. Valid choices are: * simple |
| 515 | snapshot_id (string) - A snapshot ID for the snapshot to be used as root device for the image. Mutually exclusive with block_device_map, requires root_device_name |
| 516 | delete_root_volume_on_termination (bool) - Whether to delete the root volume of the image after instance termination. Only applies when creating image from snapshot_id. Defaults to False. Note that leaving volumes behind after instance termination is not free |
| 517 | Returns: image_id - image ID of the newly created image |
| 518 | """ |
| 519 | |
| 520 | try: |
| 521 | self._reload_connection() |
| 522 | image_location = image_dict.get('image_location', None) |
| 523 | if image_location: |
| 524 | image_location = str(self.account_id) + str(image_location) |
| 525 | |
| 526 | image_id = self.conn.register_image(image_dict.get('name', None), image_dict.get('description', None), |
| 527 | image_location, image_dict.get('architecture', None), |
| 528 | image_dict.get('kernel_id', None), |
| 529 | image_dict.get('root_device_name', None), |
| 530 | image_dict.get('block_device_map', None), |
| 531 | image_dict.get('virtualization_type', None), |
| 532 | image_dict.get('sriov_net_support', None), |
| 533 | image_dict.get('snapshot_id', None), |
| 534 | image_dict.get('delete_root_volume_on_termination', None)) |
| 535 | return image_id |
| 536 | except Exception as e: |
| 537 | self.format_vimconn_exception(e) |
| 538 | |
| 539 | def delete_image(self, image_id): |
| 540 | """Deletes a tenant image from VIM |
| 541 | Returns the image_id if image is deleted or raises an exception on error""" |
| 542 | |
| 543 | try: |
| 544 | self._reload_connection() |
| 545 | self.conn.deregister_image(image_id) |
| 546 | return image_id |
| 547 | except Exception as e: |
| 548 | self.format_vimconn_exception(e) |
| 549 | |
| 550 | def get_image_id_from_path(self, path): |
| 551 | ''' |
| 552 | Params: path - location of the image |
| 553 | Returns: image_id - ID of the matching image |
| 554 | ''' |
| 555 | self._reload_connection() |
| 556 | try: |
| 557 | filters = {} |
| 558 | if path: |
| 559 | tokens = path.split('/') |
| 560 | filters['owner_id'] = tokens[0] |
| 561 | filters['name'] = '/'.join(tokens[1:]) |
| 562 | image = self.conn.get_all_images(filters=filters)[0] |
| 563 | return image.id |
| 564 | except Exception as e: |
| 565 | self.format_vimconn_exception(e) |
| 566 | |
| 567 | def get_image_list(self, filter_dict={}): |
| 568 | """Obtain tenant images from VIM |
| 569 | Filter_dict can be: |
| 570 | name: image name |
| 571 | id: image uuid |
| 572 | checksum: image checksum |
| 573 | location: image path |
| 574 | Returns the image list of dictionaries: |
| 575 | [{<the fields at Filter_dict plus some VIM specific>}, ...] |
| 576 | List can be empty |
| 577 | """ |
| 578 | |
| 579 | self.logger.debug("Getting image list from VIM") |
| 580 | try: |
| 581 | self._reload_connection() |
| 582 | image_id = None |
| 583 | filters = {} |
| 584 | if 'id' in filter_dict: |
| 585 | image_id = filter_dict['id'] |
| 586 | if 'name' in filter_dict: |
| 587 | filters['name'] = filter_dict['name'] |
| 588 | if 'location' in filter_dict: |
| 589 | filters['location'] = filter_dict['location'] |
| 590 | # filters['image_type'] = 'machine' |
| 591 | # filter_dict['owner_id'] = self.account_id |
| 592 | images = self.conn.get_all_images(image_id, filters=filters) |
| 593 | image_list = [] |
| 594 | for image in images: |
| 595 | image_list.append({'id': str(image.id), 'name': str(image.name), 'status': str(image.state), |
| 596 | 'owner': str(image.owner_id), 'location': str(image.location), |
| 597 | 'is_public': str(image.is_public), 'architecture': str(image.architecture), |
| 598 | 'platform': str(image.platform)}) |
| 599 | return image_list |
| 600 | except Exception as e: |
| 601 | self.format_vimconn_exception(e) |
| 602 | |
| 603 | def new_vminstance(self, name, description, start, image_id, flavor_id, net_list, cloud_config=None, |
| tierno | 5a3273c | 2017-08-29 11:43:46 +0200 | [diff] [blame] | 604 | disk_list=None, availability_zone_index=None, availability_zone_list=None): |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 605 | """Create a new VM/instance in AWS |
| 606 | Params: name |
| 607 | decription |
| 608 | start: (boolean) indicates if VM must start or created in pause mode. |
| 609 | image_id - image ID in AWS |
| 610 | flavor_id - instance type ID in AWS |
| 611 | net_list |
| 612 | name |
| 613 | net_id - subnet_id from AWS |
| 614 | vpci - (optional) virtual vPCI address to assign at the VM. Can be ignored depending on VIM capabilities |
| garciadeblas | c4f4d73 | 2018-10-25 18:17:24 +0200 | [diff] [blame] | 615 | model: (optional and only have sense for type==virtual) interface model: virtio, e1000, ... |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 616 | mac_address: (optional) mac address to assign to this interface |
| 617 | type: (mandatory) can be one of: |
| 618 | virtual, in this case always connected to a network of type 'net_type=bridge' |
| tierno | 66eba6e | 2017-11-10 17:09:18 +0100 | [diff] [blame] | 619 | 'PCI-PASSTHROUGH' or 'PF' (passthrough): depending on VIM capabilities it can be connected to a data/ptp network ot it |
| 620 | can created unconnected |
| 621 | 'SR-IOV' or 'VF' (SRIOV with VLAN tag): same as PF for network connectivity. |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 622 | VFnotShared - (SRIOV without VLAN tag) same as PF for network connectivity. VF where no other VFs |
| 623 | are allocated on the same physical NIC |
| 624 | bw': (optional) only for PF/VF/VFnotShared. Minimal Bandwidth required for the interface in GBPS |
| 625 | port_security': (optional) If False it must avoid any traffic filtering at this interface. If missing or True, it must apply the default VIM behaviour |
| 626 | vim_id': must be filled/added by this method with the VIM identifier generated by the VIM for this interface. 'net_list' is modified |
| 627 | elastic_ip - True/False to define if an elastic_ip is required |
| 628 | cloud_config': (optional) dictionary with: |
| 629 | key-pairs': (optional) list of strings with the public key to be inserted to the default user |
| 630 | users': (optional) list of users to be inserted, each item is a dict with: |
| 631 | name': (mandatory) user name, |
| 632 | key-pairs': (optional) list of strings with the public key to be inserted to the user |
| 633 | user-data': (optional) string is a text script to be passed directly to cloud-init |
| 634 | config-files': (optional). List of files to be transferred. Each item is a dict with: |
| 635 | dest': (mandatory) string with the destination absolute path |
| 636 | encoding': (optional, by default text). Can be one of: |
| 637 | b64', 'base64', 'gz', 'gz+b64', 'gz+base64', 'gzip+b64', 'gzip+base64' |
| 638 | content' (mandatory): string with the content of the file |
| 639 | permissions': (optional) string with file permissions, typically octal notation '0644' |
| 640 | owner: (optional) file owner, string with the format 'owner:group' |
| 641 | boot-data-drive: boolean to indicate if user-data must be passed using a boot drive (hard disk) |
| 642 | security-groups: |
| 643 | subnet_id |
| 644 | security_group_id |
| 645 | disk_list': (optional) list with additional disks to the VM. Each item is a dict with: |
| 646 | image_id': (optional). VIM id of an existing image. If not provided an empty disk must be mounted |
| 647 | size': (mandatory) string with the size of the disk in GB |
| tierno | 98e909c | 2017-10-14 13:27:03 +0200 | [diff] [blame] | 648 | Returns a tuple with the instance identifier and created_items or raises an exception on error |
| 649 | created_items can be None or a dictionary where this method can include key-values that will be passed to |
| 650 | the method delete_vminstance and action_vminstance. Can be used to store created ports, volumes, etc. |
| 651 | Format is vimconnector dependent, but do not use nested dictionaries and a value of None should be the same |
| 652 | as not present. |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 653 | """ |
| 654 | |
| 655 | self.logger.debug("Creating a new VM instance") |
| 656 | try: |
| 657 | self._reload_connection() |
| 658 | instance = None |
| tierno | 0a1437e | 2017-10-02 00:17:43 +0200 | [diff] [blame] | 659 | _, userdata = self._create_user_data(cloud_config) |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 660 | |
| 661 | if not net_list: |
| 662 | reservation = self.conn.run_instances( |
| 663 | image_id, |
| 664 | key_name=self.key_pair, |
| 665 | instance_type=flavor_id, |
| 666 | security_groups=self.security_groups, |
| 667 | user_data=userdata |
| 668 | ) |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 669 | else: |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 670 | for index, subnet in enumerate(net_list): |
| 671 | net_intr = boto.ec2.networkinterface.NetworkInterfaceSpecification(subnet_id=subnet.get('net_id'), |
| 672 | groups=None, |
| 673 | associate_public_ip_address=True) |
| 674 | |
| 675 | if subnet.get('elastic_ip'): |
| 676 | eip = self.conn.allocate_address() |
| 677 | self.conn.associate_address(allocation_id=eip.allocation_id, network_interface_id=net_intr.id) |
| 678 | |
| 679 | if index == 0: |
| 680 | reservation = self.conn.run_instances( |
| 681 | image_id, |
| 682 | key_name=self.key_pair, |
| 683 | instance_type=flavor_id, |
| 684 | security_groups=self.security_groups, |
| 685 | network_interfaces=boto.ec2.networkinterface.NetworkInterfaceCollection(net_intr), |
| 686 | user_data=userdata |
| 687 | ) |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 688 | else: |
| 689 | while True: |
| 690 | try: |
| 691 | self.conn.attach_network_interface( |
| 692 | network_interface_id=boto.ec2.networkinterface.NetworkInterfaceCollection(net_intr), |
| 693 | instance_id=instance.id, device_index=0) |
| 694 | break |
| 695 | except: |
| 696 | time.sleep(10) |
| montesmoreno | 3b299bb | 2017-10-02 18:19:05 +0200 | [diff] [blame] | 697 | net_list[index]['vim_id'] = reservation.instances[0].interfaces[index].id |
| 698 | |
| 699 | instance = reservation.instances[0] |
| tierno | 98e909c | 2017-10-14 13:27:03 +0200 | [diff] [blame] | 700 | return instance.id, None |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 701 | except Exception as e: |
| 702 | self.format_vimconn_exception(e) |
| 703 | |
| 704 | def get_vminstance(self, vm_id): |
| 705 | """Returns the VM instance information from VIM""" |
| 706 | |
| 707 | try: |
| 708 | self._reload_connection() |
| 709 | reservation = self.conn.get_all_instances(vm_id) |
| 710 | return reservation[0].instances[0].__dict__ |
| 711 | except Exception as e: |
| 712 | self.format_vimconn_exception(e) |
| 713 | |
| tierno | 98e909c | 2017-10-14 13:27:03 +0200 | [diff] [blame] | 714 | def delete_vminstance(self, vm_id, created_items=None): |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 715 | """Removes a VM instance from VIM |
| 716 | Returns the instance identifier""" |
| 717 | |
| 718 | try: |
| 719 | self._reload_connection() |
| 720 | self.logger.debug("DELETING VM_ID: " + str(vm_id)) |
| 721 | self.conn.terminate_instances(vm_id) |
| 722 | return vm_id |
| 723 | except Exception as e: |
| 724 | self.format_vimconn_exception(e) |
| 725 | |
| 726 | def refresh_vms_status(self, vm_list): |
| 727 | """ Get the status of the virtual machines and their interfaces/ports |
| 728 | Params: the list of VM identifiers |
| 729 | Returns a dictionary with: |
| 730 | vm_id: #VIM id of this Virtual Machine |
| 731 | status: #Mandatory. Text with one of: |
| 732 | # DELETED (not found at vim) |
| 733 | # VIM_ERROR (Cannot connect to VIM, VIM response error, ...) |
| 734 | # OTHER (Vim reported other status not understood) |
| 735 | # ERROR (VIM indicates an ERROR status) |
| 736 | # ACTIVE, PAUSED, SUSPENDED, INACTIVE (not running), |
| 737 | # BUILD (on building process), ERROR |
| 738 | # ACTIVE:NoMgmtIP (Active but any of its interface has an IP address |
| 739 | # |
| 740 | error_msg: #Text with VIM error message, if any. Or the VIM connection ERROR |
| 741 | vim_info: #Text with plain information obtained from vim (yaml.safe_dump) |
| 742 | interfaces: list with interface info. Each item a dictionary with: |
| 743 | vim_interface_id - The ID of the ENI. |
| 744 | vim_net_id - The ID of the VPC subnet. |
| 745 | mac_address - The MAC address of the interface. |
| 746 | ip_address - The IP address of the interface within the subnet. |
| 747 | """ |
| 748 | self.logger.debug("Getting VM instance information from VIM") |
| 749 | try: |
| 750 | self._reload_connection() |
| 751 | reservation = self.conn.get_all_instances(vm_list)[0] |
| 752 | instances = {} |
| 753 | instance_dict = {} |
| 754 | for instance in reservation.instances: |
| 755 | try: |
| 756 | if instance.state in ("pending"): |
| 757 | instance_dict['status'] = "BUILD" |
| 758 | elif instance.state in ("available", "running", "up"): |
| 759 | instance_dict['status'] = 'ACTIVE' |
| 760 | else: |
| 761 | instance_dict['status'] = 'ERROR' |
| 762 | instance_dict['error_msg'] = "" |
| 763 | instance_dict['interfaces'] = [] |
| 764 | interface_dict = {} |
| 765 | for interface in instance.interfaces: |
| 766 | interface_dict['vim_interface_id'] = interface.id |
| 767 | interface_dict['vim_net_id'] = interface.subnet_id |
| 768 | interface_dict['mac_address'] = interface.mac_address |
| montesmoreno | c6f2681 | 2017-07-12 11:47:13 +0200 | [diff] [blame] | 769 | if hasattr(interface, 'publicIp') and interface.publicIp != None: |
| 770 | interface_dict['ip_address'] = interface.publicIp + ";" + interface.private_ip_address |
| 771 | else: |
| 772 | interface_dict['ip_address'] = interface.private_ip_address |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 773 | instance_dict['interfaces'].append(interface_dict) |
| 774 | except Exception as e: |
| 775 | self.logger.error("Exception getting vm status: %s", str(e), exc_info=True) |
| 776 | instance_dict['status'] = "DELETED" |
| 777 | instance_dict['error_msg'] = str(e) |
| 778 | finally: |
| 779 | try: |
| 780 | instance_dict['vim_info'] = yaml.safe_dump(instance, default_flow_style=True, width=256) |
| 781 | except yaml.YAMLError as e: |
| 782 | # self.logger.error("Exception getting vm status: %s", str(e), exc_info=True) |
| 783 | instance_dict['vim_info'] = str(instance) |
| 784 | instances[instance.id] = instance_dict |
| 785 | return instances |
| 786 | except Exception as e: |
| 787 | self.logger.error("Exception getting vm status: %s", str(e), exc_info=True) |
| 788 | self.format_vimconn_exception(e) |
| 789 | |
| tierno | 98e909c | 2017-10-14 13:27:03 +0200 | [diff] [blame] | 790 | def action_vminstance(self, vm_id, action_dict, created_items={}): |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 791 | """Send and action over a VM instance from VIM |
| 792 | Returns the vm_id if the action was successfully sent to the VIM""" |
| 793 | |
| 794 | self.logger.debug("Action over VM '%s': %s", vm_id, str(action_dict)) |
| 795 | try: |
| 796 | self._reload_connection() |
| 797 | if "start" in action_dict: |
| 798 | self.conn.start_instances(vm_id) |
| 799 | elif "stop" in action_dict or "stop" in action_dict: |
| 800 | self.conn.stop_instances(vm_id) |
| 801 | elif "terminate" in action_dict: |
| 802 | self.conn.terminate_instances(vm_id) |
| 803 | elif "reboot" in action_dict: |
| 804 | self.conn.reboot_instances(vm_id) |
| tierno | 98e909c | 2017-10-14 13:27:03 +0200 | [diff] [blame] | 805 | return None |
| ahmadsa | 61ccd64 | 2017-04-20 18:17:28 -0700 | [diff] [blame] | 806 | except Exception as e: |
| 807 | self.format_vimconn_exception(e) |