services.py 53 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300
  1. """Services implemented by the AWS provider."""
  2. import ipaddress
  3. import logging
  4. import string
  5. from botocore.exceptions import ClientError
  6. import cachetools
  7. import requests
  8. import cloudbridge.base.helpers as cb_helpers
  9. from cloudbridge.base.middleware import dispatch
  10. from cloudbridge.base.resources import ClientPagedResultList
  11. from cloudbridge.base.services import BaseBucketObjectService
  12. from cloudbridge.base.services import BaseBucketService
  13. from cloudbridge.base.services import BaseComputeService
  14. from cloudbridge.base.services import BaseFloatingIPService
  15. from cloudbridge.base.services import BaseGatewayService
  16. from cloudbridge.base.services import BaseImageService
  17. from cloudbridge.base.services import BaseInstanceService
  18. from cloudbridge.base.services import BaseKeyPairService
  19. from cloudbridge.base.services import BaseNetworkService
  20. from cloudbridge.base.services import BaseNetworkingService
  21. from cloudbridge.base.services import BaseRegionService
  22. from cloudbridge.base.services import BaseRouterService
  23. from cloudbridge.base.services import BaseSecurityService
  24. from cloudbridge.base.services import BaseSnapshotService
  25. from cloudbridge.base.services import BaseStorageService
  26. from cloudbridge.base.services import BaseSubnetService
  27. from cloudbridge.base.services import BaseVMFirewallRuleService
  28. from cloudbridge.base.services import BaseVMFirewallService
  29. from cloudbridge.base.services import BaseVMTypeService
  30. from cloudbridge.base.services import BaseVolumeService
  31. from cloudbridge.interfaces.exceptions import DuplicateResourceException
  32. from cloudbridge.interfaces.exceptions import \
  33. InvalidConfigurationException
  34. from cloudbridge.interfaces.exceptions import InvalidParamException
  35. from cloudbridge.interfaces.exceptions import InvalidValueException
  36. from cloudbridge.interfaces.resources import KeyPair
  37. from cloudbridge.interfaces.resources import MachineImage
  38. from cloudbridge.interfaces.resources import Network
  39. from cloudbridge.interfaces.resources import Snapshot
  40. from cloudbridge.interfaces.resources import TrafficDirection
  41. from cloudbridge.interfaces.resources import VMFirewall
  42. from cloudbridge.interfaces.resources import VMType
  43. from cloudbridge.interfaces.resources import Volume
  44. from .helpers import BotoEC2Service
  45. from .helpers import BotoS3Service
  46. from .helpers import trim_empty_params
  47. from .resources import AWSBucket
  48. from .resources import AWSBucketObject
  49. from .resources import AWSFloatingIP
  50. from .resources import AWSInstance
  51. from .resources import AWSInternetGateway
  52. from .resources import AWSKeyPair
  53. from .resources import AWSLaunchConfig
  54. from .resources import AWSMachineImage
  55. from .resources import AWSNetwork
  56. from .resources import AWSRegion
  57. from .resources import AWSRouter
  58. from .resources import AWSSnapshot
  59. from .resources import AWSSubnet
  60. from .resources import AWSVMFirewall
  61. from .resources import AWSVMFirewallRule
  62. from .resources import AWSVMType
  63. from .resources import AWSVolume
  64. log = logging.getLogger(__name__)
  65. class AWSSecurityService(BaseSecurityService):
  66. def __init__(self, provider):
  67. super(AWSSecurityService, self).__init__(provider)
  68. # Initialize provider services
  69. self._key_pairs = AWSKeyPairService(provider)
  70. self._vm_firewalls = AWSVMFirewallService(provider)
  71. self._vm_firewall_rule_svc = AWSVMFirewallRuleService(provider)
  72. @property
  73. def key_pairs(self):
  74. return self._key_pairs
  75. @property
  76. def vm_firewalls(self):
  77. return self._vm_firewalls
  78. @property
  79. def _vm_firewall_rules(self):
  80. return self._vm_firewall_rule_svc
  81. class AWSKeyPairService(BaseKeyPairService):
  82. def __init__(self, provider):
  83. super(AWSKeyPairService, self).__init__(provider)
  84. self.svc = BotoEC2Service(provider=self.provider,
  85. cb_resource=AWSKeyPair,
  86. boto_collection_name='key_pairs')
  87. @dispatch(event="provider.security.key_pairs.get",
  88. priority=BaseKeyPairService.STANDARD_EVENT_PRIORITY)
  89. def get(self, key_pair_id):
  90. log.debug("Getting Key Pair Service %s", key_pair_id)
  91. return self.svc.get(key_pair_id)
  92. @dispatch(event="provider.security.key_pairs.list",
  93. priority=BaseKeyPairService.STANDARD_EVENT_PRIORITY)
  94. def list(self, limit=None, marker=None):
  95. return self.svc.list(limit=limit, marker=marker)
  96. @dispatch(event="provider.security.key_pairs.find",
  97. priority=BaseKeyPairService.STANDARD_EVENT_PRIORITY)
  98. def find(self, **kwargs):
  99. name = kwargs.pop('name', None)
  100. # All kwargs should have been popped at this time.
  101. if len(kwargs) > 0:
  102. raise InvalidParamException(
  103. "Unrecognised parameters for search: %s. Supported "
  104. "attributes: %s" % (kwargs, 'name'))
  105. log.debug("Searching for Key Pair %s", name)
  106. return self.svc.find(filters={'key-name': name})
  107. @dispatch(event="provider.security.key_pairs.create",
  108. priority=BaseKeyPairService.STANDARD_EVENT_PRIORITY)
  109. def create(self, name, public_key_material=None):
  110. AWSKeyPair.assert_valid_resource_name(name)
  111. private_key = None
  112. if not public_key_material:
  113. public_key_material, private_key = cb_helpers.generate_key_pair()
  114. try:
  115. kp = self.svc.create('import_key_pair', KeyName=name,
  116. PublicKeyMaterial=public_key_material)
  117. kp.material = private_key
  118. return kp
  119. except ClientError as e:
  120. if e.response['Error']['Code'] == 'InvalidKeyPair.Duplicate':
  121. raise DuplicateResourceException(
  122. 'Keypair already exists with name {0}'.format(name))
  123. else:
  124. raise e
  125. @dispatch(event="provider.security.key_pairs.delete",
  126. priority=BaseKeyPairService.STANDARD_EVENT_PRIORITY)
  127. def delete(self, key_pair):
  128. key_pair = (key_pair if isinstance(key_pair, AWSKeyPair) else
  129. self.get(key_pair))
  130. if key_pair:
  131. # pylint:disable=protected-access
  132. key_pair._key_pair.delete()
  133. class AWSVMFirewallService(BaseVMFirewallService):
  134. def __init__(self, provider):
  135. super(AWSVMFirewallService, self).__init__(provider)
  136. self.svc = BotoEC2Service(provider=self.provider,
  137. cb_resource=AWSVMFirewall,
  138. boto_collection_name='security_groups')
  139. @dispatch(event="provider.security.vm_firewalls.get",
  140. priority=BaseVMFirewallService.STANDARD_EVENT_PRIORITY)
  141. def get(self, vm_firewall_id):
  142. log.debug("Getting Firewall Service with the id: %s", vm_firewall_id)
  143. return self.svc.get(vm_firewall_id)
  144. @dispatch(event="provider.security.vm_firewalls.list",
  145. priority=BaseVMFirewallService.STANDARD_EVENT_PRIORITY)
  146. def list(self, limit=None, marker=None):
  147. return self.svc.list(limit=limit, marker=marker)
  148. @cb_helpers.deprecated_alias(network_id='network')
  149. @dispatch(event="provider.security.vm_firewalls.create",
  150. priority=BaseVMFirewallService.STANDARD_EVENT_PRIORITY)
  151. def create(self, label, network, description=None):
  152. AWSVMFirewall.assert_valid_resource_label(label)
  153. name = AWSVMFirewall._generate_name_from_label(label, 'cb-fw')
  154. network_id = network.id if isinstance(network, Network) else network
  155. obj = self.svc.create('create_security_group', GroupName=name,
  156. Description=name,
  157. VpcId=network_id)
  158. obj.label = label
  159. obj.description = description
  160. return obj
  161. @dispatch(event="provider.security.vm_firewalls.find",
  162. priority=BaseVMFirewallService.STANDARD_EVENT_PRIORITY)
  163. def find(self, **kwargs):
  164. # Filter by name or label
  165. label = kwargs.pop('label', None)
  166. log.debug("Searching for Firewall Service %s", label)
  167. # All kwargs should have been popped at this time.
  168. if len(kwargs) > 0:
  169. raise InvalidParamException(
  170. "Unrecognised parameters for search: %s. Supported "
  171. "attributes: %s" % (kwargs, 'label'))
  172. return self.svc.find(filters={'tag:Name': label})
  173. @dispatch(event="provider.security.vm_firewalls.delete",
  174. priority=BaseVMFirewallService.STANDARD_EVENT_PRIORITY)
  175. def delete(self, vm_firewall):
  176. firewall = (vm_firewall if isinstance(vm_firewall, AWSVMFirewall)
  177. else self.get(vm_firewall))
  178. if firewall:
  179. # pylint:disable=protected-access
  180. firewall._vm_firewall.delete()
  181. class AWSVMFirewallRuleService(BaseVMFirewallRuleService):
  182. def __init__(self, provider):
  183. super(AWSVMFirewallRuleService, self).__init__(provider)
  184. @dispatch(event="provider.security.vm_firewall_rules.list",
  185. priority=BaseVMFirewallRuleService.STANDARD_EVENT_PRIORITY)
  186. def list(self, firewall, limit=None, marker=None):
  187. # pylint:disable=protected-access
  188. rules = [AWSVMFirewallRule(firewall,
  189. TrafficDirection.INBOUND, r)
  190. for r in firewall._vm_firewall.ip_permissions]
  191. # pylint:disable=protected-access
  192. rules = rules + [
  193. AWSVMFirewallRule(
  194. firewall, TrafficDirection.OUTBOUND, r)
  195. for r in firewall._vm_firewall.ip_permissions_egress]
  196. return ClientPagedResultList(self.provider, rules,
  197. limit=limit, marker=marker)
  198. @dispatch(event="provider.security.vm_firewall_rules.create",
  199. priority=BaseVMFirewallRuleService.STANDARD_EVENT_PRIORITY)
  200. def create(self, firewall, direction, protocol=None, from_port=None,
  201. to_port=None, cidr=None, src_dest_fw=None):
  202. src_dest_fw_id = (
  203. src_dest_fw.id if isinstance(src_dest_fw, AWSVMFirewall)
  204. else src_dest_fw)
  205. # pylint:disable=protected-access
  206. ip_perm_entry = AWSVMFirewallRule._construct_ip_perms(
  207. protocol, from_port, to_port, cidr, src_dest_fw_id)
  208. # Filter out empty values to please Boto
  209. ip_perms = [trim_empty_params(ip_perm_entry)]
  210. try:
  211. if direction == TrafficDirection.INBOUND:
  212. # pylint:disable=protected-access
  213. firewall._vm_firewall.authorize_ingress(
  214. IpPermissions=ip_perms)
  215. elif direction == TrafficDirection.OUTBOUND:
  216. # pylint:disable=protected-access
  217. firewall._vm_firewall.authorize_egress(
  218. IpPermissions=ip_perms)
  219. else:
  220. raise InvalidValueException("direction", direction)
  221. firewall.refresh()
  222. return AWSVMFirewallRule(firewall, direction, ip_perm_entry)
  223. except ClientError as ec2e:
  224. if ec2e.response['Error']['Code'] == "InvalidPermission.Duplicate":
  225. return AWSVMFirewallRule(
  226. firewall, direction, ip_perm_entry)
  227. else:
  228. raise ec2e
  229. @dispatch(event="provider.security.vm_firewall_rules.delete",
  230. priority=BaseVMFirewallRuleService.STANDARD_EVENT_PRIORITY)
  231. def delete(self, firewall, rule):
  232. # pylint:disable=protected-access
  233. ip_perm_entry = rule._construct_ip_perms(
  234. rule.protocol, rule.from_port, rule.to_port,
  235. rule.cidr, rule.src_dest_fw_id)
  236. # Filter out empty values to please Boto
  237. ip_perms = [trim_empty_params(ip_perm_entry)]
  238. # pylint:disable=protected-access
  239. if rule.direction == TrafficDirection.INBOUND:
  240. firewall._vm_firewall.revoke_ingress(
  241. IpPermissions=ip_perms)
  242. else:
  243. # pylint:disable=protected-access
  244. firewall._vm_firewall.revoke_egress(
  245. IpPermissions=ip_perms)
  246. firewall.refresh()
  247. class AWSStorageService(BaseStorageService):
  248. def __init__(self, provider):
  249. super(AWSStorageService, self).__init__(provider)
  250. # Initialize provider services
  251. self._volume_svc = AWSVolumeService(self.provider)
  252. self._snapshot_svc = AWSSnapshotService(self.provider)
  253. self._bucket_svc = AWSBucketService(self.provider)
  254. self._bucket_obj_svc = AWSBucketObjectService(self.provider)
  255. @property
  256. def volumes(self):
  257. return self._volume_svc
  258. @property
  259. def snapshots(self):
  260. return self._snapshot_svc
  261. @property
  262. def buckets(self):
  263. return self._bucket_svc
  264. @property
  265. def _bucket_objects(self):
  266. return self._bucket_obj_svc
  267. class AWSVolumeService(BaseVolumeService):
  268. def __init__(self, provider):
  269. super(AWSVolumeService, self).__init__(provider)
  270. self.svc = BotoEC2Service(provider=self.provider,
  271. cb_resource=AWSVolume,
  272. boto_collection_name='volumes')
  273. @dispatch(event="provider.storage.volumes.get",
  274. priority=BaseVolumeService.STANDARD_EVENT_PRIORITY)
  275. def get(self, volume_id):
  276. return self.svc.get(volume_id)
  277. @dispatch(event="provider.storage.volumes.find",
  278. priority=BaseVolumeService.STANDARD_EVENT_PRIORITY)
  279. def find(self, **kwargs):
  280. label = kwargs.pop('label', None)
  281. # All kwargs should have been popped at this time.
  282. if len(kwargs) > 0:
  283. raise InvalidParamException(
  284. "Unrecognised parameters for search: %s. Supported "
  285. "attributes: %s" % (kwargs, 'label'))
  286. log.debug("Searching for AWS Volume Service %s", label)
  287. return self.svc.find(
  288. filters={'tag:Name': label,
  289. 'availability-zone': self.provider.zone_name})
  290. @dispatch(event="provider.storage.volumes.list",
  291. priority=BaseVolumeService.STANDARD_EVENT_PRIORITY)
  292. def list(self, limit=None, marker=None):
  293. return self.svc.find(
  294. filters={'availability-zone': self.provider.zone_name},
  295. limit=limit, marker=marker)
  296. @dispatch(event="provider.storage.volumes.create",
  297. priority=BaseVolumeService.STANDARD_EVENT_PRIORITY)
  298. def create(self, label, size, snapshot=None, description=None):
  299. AWSVolume.assert_valid_resource_label(label)
  300. zone_name = self.provider.zone_name
  301. snapshot_id = snapshot.id if isinstance(
  302. snapshot, AWSSnapshot) and snapshot else snapshot
  303. cb_vol = self.svc.create('create_volume', Size=size,
  304. AvailabilityZone=zone_name,
  305. SnapshotId=snapshot_id)
  306. # Wait until ready to tag instance
  307. cb_vol.wait_till_ready()
  308. cb_vol.label = label
  309. if description:
  310. cb_vol.description = description
  311. return cb_vol
  312. @dispatch(event="provider.storage.volumes.delete",
  313. priority=BaseVolumeService.STANDARD_EVENT_PRIORITY)
  314. def delete(self, vol):
  315. volume = vol if isinstance(vol, AWSVolume) else self.get(vol)
  316. if volume:
  317. # pylint:disable=protected-access
  318. volume._volume.delete()
  319. class AWSSnapshotService(BaseSnapshotService):
  320. def __init__(self, provider):
  321. super(AWSSnapshotService, self).__init__(provider)
  322. self.svc = BotoEC2Service(provider=self.provider,
  323. cb_resource=AWSSnapshot,
  324. boto_collection_name='snapshots')
  325. @dispatch(event="provider.storage.snapshots.get",
  326. priority=BaseSnapshotService.STANDARD_EVENT_PRIORITY)
  327. def get(self, snapshot_id):
  328. return self.svc.get(snapshot_id)
  329. @dispatch(event="provider.storage.snapshots.find",
  330. priority=BaseSnapshotService.STANDARD_EVENT_PRIORITY)
  331. def find(self, **kwargs):
  332. # Filter by description or label
  333. label = kwargs.get('label', None)
  334. obj_list = []
  335. if label:
  336. log.debug("Searching for AWS Snapshot with label %s", label)
  337. obj_list.extend(self.svc.find(filters={'tag:Name': label},
  338. OwnerIds=['self']))
  339. else:
  340. obj_list = list(self)
  341. filters = ['label']
  342. return cb_helpers.generic_find(filters, kwargs, obj_list)
  343. @dispatch(event="provider.storage.snapshots.list",
  344. priority=BaseSnapshotService.STANDARD_EVENT_PRIORITY)
  345. def list(self, limit=None, marker=None):
  346. return self.svc.list(limit=limit, marker=marker,
  347. OwnerIds=['self'])
  348. @dispatch(event="provider.storage.snapshots.create",
  349. priority=BaseSnapshotService.STANDARD_EVENT_PRIORITY)
  350. def create(self, label, volume, description=None):
  351. AWSSnapshot.assert_valid_resource_label(label)
  352. volume_id = volume.id if isinstance(volume, AWSVolume) else volume
  353. cb_snap = self.svc.create('create_snapshot', VolumeId=volume_id)
  354. # Wait until ready to tag instance
  355. cb_snap.wait_till_ready()
  356. cb_snap.label = label
  357. if cb_snap.description:
  358. cb_snap.description = description
  359. return cb_snap
  360. @dispatch(event="provider.storage.snapshots.delete",
  361. priority=BaseSnapshotService.STANDARD_EVENT_PRIORITY)
  362. def delete(self, snapshot):
  363. snapshot = (snapshot if isinstance(snapshot, AWSSnapshot) else
  364. self.get(snapshot))
  365. if snapshot:
  366. # pylint:disable=protected-access
  367. snapshot._snapshot.delete()
  368. class AWSBucketService(BaseBucketService):
  369. def __init__(self, provider):
  370. super(AWSBucketService, self).__init__(provider)
  371. self.svc = BotoS3Service(provider=self.provider,
  372. cb_resource=AWSBucket,
  373. boto_collection_name='buckets')
  374. @dispatch(event="provider.storage.buckets.get",
  375. priority=BaseBucketService.STANDARD_EVENT_PRIORITY)
  376. def get(self, bucket_id):
  377. """
  378. Returns a bucket given its ID. Returns ``None`` if the bucket
  379. does not exist.
  380. """
  381. try:
  382. # Make a call to make sure the bucket exists. There's an edge case
  383. # where a 403 response can occur when the bucket exists but the
  384. # user simply does not have permissions to access it. See below.
  385. self.provider.s3_conn.meta.client.head_bucket(Bucket=bucket_id)
  386. return AWSBucket(self.provider,
  387. self.provider.s3_conn.Bucket(bucket_id))
  388. except ClientError as e:
  389. # If 403, it means the bucket exists, but the user does not have
  390. # permissions to access the bucket. However, limited operations
  391. # may be permitted (with a session token for example), so return a
  392. # Bucket instance to allow further operations.
  393. # http://stackoverflow.com/questions/32331456/using-boto-upload-file-to-s3-
  394. # sub-folder-when-i-have-no-permissions-on-listing-fo
  395. if e.response['Error']['Code'] == "403":
  396. log.warning("AWS Bucket %s already exists but user doesn't "
  397. "have enough permissions to list its contents."
  398. "Other operations may be available.",
  399. bucket_id)
  400. return AWSBucket(self.provider,
  401. self.provider.s3_conn.Bucket(bucket_id))
  402. # For all other responses, it's assumed that the bucket does not exist.
  403. return None
  404. @dispatch(event="provider.storage.buckets.list",
  405. priority=BaseBucketService.STANDARD_EVENT_PRIORITY)
  406. def list(self, limit=None, marker=None):
  407. return self.svc.list(limit=limit, marker=marker)
  408. @dispatch(event="provider.storage.buckets.create",
  409. priority=BaseBucketService.STANDARD_EVENT_PRIORITY)
  410. def create(self, name, location=None):
  411. AWSBucket.assert_valid_resource_name(name)
  412. location = location or self.provider.region_name
  413. # Due to an API issue in S3, specifying us-east-1 as a
  414. # LocationConstraint results in an InvalidLocationConstraint.
  415. # Therefore, it must be special-cased and omitted altogether.
  416. # See: https://github.com/boto/boto3/issues/125
  417. # In addition, us-east-1 also behaves differently when it comes
  418. # to raising duplicate resource exceptions, so perform a manual
  419. # check
  420. if location == 'us-east-1':
  421. try:
  422. # check whether bucket already exists
  423. self.provider.s3_conn.meta.client.head_bucket(Bucket=name)
  424. except ClientError as e:
  425. if e.response['Error']['Code'] == "404":
  426. # bucket doesn't exist, go ahead and create it
  427. return self.svc.create('create_bucket', Bucket=name)
  428. raise DuplicateResourceException(
  429. 'Bucket already exists with name {0}'.format(name))
  430. else:
  431. try:
  432. return self.svc.create('create_bucket', Bucket=name,
  433. CreateBucketConfiguration={
  434. 'LocationConstraint': location
  435. })
  436. except ClientError as e:
  437. if e.response['Error']['Code'] == "BucketAlreadyOwnedByYou":
  438. raise DuplicateResourceException(
  439. 'Bucket already exists with name {0}'.format(name))
  440. else:
  441. raise
  442. @dispatch(event="provider.storage.buckets.delete",
  443. priority=BaseBucketService.STANDARD_EVENT_PRIORITY)
  444. def delete(self, bucket):
  445. b = bucket if isinstance(bucket, AWSBucket) else self.get(bucket)
  446. if b:
  447. # pylint:disable=protected-access
  448. b._bucket.delete()
  449. class AWSBucketObjectService(BaseBucketObjectService):
  450. def __init__(self, provider):
  451. super(AWSBucketObjectService, self).__init__(provider)
  452. def get(self, bucket, object_id):
  453. try:
  454. # pylint:disable=protected-access
  455. obj = bucket._bucket.Object(object_id)
  456. # load() throws an error if object does not exist
  457. obj.load()
  458. return AWSBucketObject(self.provider, obj)
  459. except ClientError:
  460. return None
  461. def list(self, bucket, limit=None, marker=None, prefix=None):
  462. if prefix:
  463. # pylint:disable=protected-access
  464. boto_objs = bucket._bucket.objects.filter(Prefix=prefix)
  465. else:
  466. # pylint:disable=protected-access
  467. boto_objs = bucket._bucket.objects.all()
  468. objects = [AWSBucketObject(self.provider, obj) for obj in boto_objs]
  469. return ClientPagedResultList(self.provider, objects,
  470. limit=limit, marker=marker)
  471. def find(self, bucket, **kwargs):
  472. # pylint:disable=protected-access
  473. obj_list = [AWSBucketObject(self.provider, o)
  474. for o in bucket._bucket.objects.all()]
  475. filters = ['name']
  476. matches = cb_helpers.generic_find(filters, kwargs, obj_list)
  477. return ClientPagedResultList(self.provider, list(matches),
  478. limit=None, marker=None)
  479. def create(self, bucket, name):
  480. # pylint:disable=protected-access
  481. obj = bucket._bucket.Object(name)
  482. return AWSBucketObject(self.provider, obj)
  483. class AWSComputeService(BaseComputeService):
  484. def __init__(self, provider):
  485. super(AWSComputeService, self).__init__(provider)
  486. self._vm_type_svc = AWSVMTypeService(self.provider)
  487. self._instance_svc = AWSInstanceService(self.provider)
  488. self._region_svc = AWSRegionService(self.provider)
  489. self._images_svc = AWSImageService(self.provider)
  490. @property
  491. def images(self):
  492. return self._images_svc
  493. @property
  494. def vm_types(self):
  495. return self._vm_type_svc
  496. @property
  497. def instances(self):
  498. return self._instance_svc
  499. @property
  500. def regions(self):
  501. return self._region_svc
  502. class AWSImageService(BaseImageService):
  503. def __init__(self, provider):
  504. super(AWSImageService, self).__init__(provider)
  505. self.svc = BotoEC2Service(provider=self.provider,
  506. cb_resource=AWSMachineImage,
  507. boto_collection_name='images')
  508. def get(self, image_id):
  509. log.debug("Getting AWS Image Service with the id: %s", image_id)
  510. return self.svc.get(image_id)
  511. def find(self, **kwargs):
  512. # Filter by name or label
  513. label = kwargs.pop('label', None)
  514. # Popped here, not used in the generic find
  515. owner = kwargs.pop('owners', None)
  516. # All kwargs should have been popped at this time.
  517. if len(kwargs) > 0:
  518. raise InvalidParamException(
  519. "Unrecognised parameters for search: %s. Supported "
  520. "attributes: %s" % (kwargs, 'label'))
  521. extra_args = {}
  522. if owner:
  523. extra_args.update(Owners=owner)
  524. # The original list is made by combining both searches by "tag:Name"
  525. # and "AMI name" to allow for searches of public images
  526. if label:
  527. log.debug("Searching for AWS Image Service %s", label)
  528. obj_list = []
  529. obj_list.extend(
  530. self.svc.find(filters={'name': label}, **extra_args))
  531. obj_list.extend(
  532. self.svc.find(filters={'tag:Name': label}, **extra_args))
  533. return obj_list
  534. else:
  535. return []
  536. def list(self, filter_by_owner=True, limit=None, marker=None):
  537. return self.svc.list(Owners=['self'] if filter_by_owner else
  538. ['amazon', 'self'],
  539. limit=limit, marker=marker)
  540. class AWSInstanceService(BaseInstanceService):
  541. def __init__(self, provider):
  542. super(AWSInstanceService, self).__init__(provider)
  543. self.svc = BotoEC2Service(provider=self.provider,
  544. cb_resource=AWSInstance,
  545. boto_collection_name='instances')
  546. def _resolve_launch_options(self, subnet=None, zone_id=None,
  547. vm_firewalls=None):
  548. """
  549. Work out interdependent launch options.
  550. Some launch options are required and interdependent so make sure
  551. they conform to the interface contract.
  552. :type subnet: ``Subnet``
  553. :param subnet: Subnet object within which to launch.
  554. :type zone_id: ``str``
  555. :param zone_id: ID of the zone where the launch should happen.
  556. :type vm_firewalls: ``list`` of ``id``
  557. :param vm_firewalls: List of firewall IDs.
  558. :rtype: triplet of ``str``
  559. :return: Subnet ID, zone ID and VM firewall IDs for launch.
  560. :raise ValueError: In case a conflicting combination is found.
  561. """
  562. if subnet:
  563. # subnet's zone takes precedence
  564. zone_id = subnet.zone.id
  565. if vm_firewalls and isinstance(vm_firewalls, list) and isinstance(
  566. vm_firewalls[0], VMFirewall):
  567. vm_firewall_ids = [fw.id for fw in vm_firewalls]
  568. else:
  569. vm_firewall_ids = vm_firewalls
  570. return subnet.id if subnet else None, zone_id, vm_firewall_ids
  571. def _process_block_device_mappings(self, launch_config):
  572. """
  573. Processes block device mapping information
  574. and returns a Boto BlockDeviceMapping object. If new volumes
  575. are requested (source is None and destination is VOLUME), they will be
  576. created and the relevant volume ids included in the mapping.
  577. """
  578. bdml = []
  579. # Assign letters from f onwards
  580. # http://docs.aws.amazon.com/AWSEC2/latest/UserGuide/device_naming.html
  581. next_letter = iter(list(string.ascii_lowercase[6:]))
  582. # assign ephemeral devices from 0 onwards
  583. ephemeral_counter = 0
  584. for device in launch_config.block_devices:
  585. bdm = {}
  586. if device.is_volume:
  587. # Generate the device path
  588. bdm['DeviceName'] = \
  589. '/dev/sd' + ('a1' if device.is_root else next(next_letter))
  590. ebs_def = {}
  591. if isinstance(device.source, Snapshot):
  592. ebs_def['SnapshotId'] = device.source.id
  593. elif isinstance(device.source, Volume):
  594. # TODO: We could create a snapshot from the volume
  595. # and use that instead.
  596. # Not supported
  597. pass
  598. elif isinstance(device.source, MachineImage):
  599. # Not supported
  600. pass
  601. else:
  602. # source is None, but destination is volume, therefore
  603. # create a blank volume. This requires a size though.
  604. if not device.size:
  605. raise InvalidConfigurationException(
  606. "The source is none and the destination is a"
  607. " volume. Therefore, you must specify a size.")
  608. ebs_def['DeleteOnTermination'] = device.delete_on_terminate \
  609. or True
  610. if device.size:
  611. ebs_def['VolumeSize'] = device.size
  612. if ebs_def:
  613. bdm['Ebs'] = ebs_def
  614. else: # device is ephemeral
  615. bdm['VirtualName'] = 'ephemeral%s' % ephemeral_counter
  616. # Append the config
  617. bdml.append(bdm)
  618. return bdml
  619. def create_launch_config(self):
  620. return AWSLaunchConfig(self.provider)
  621. @dispatch(event="provider.compute.instances.create",
  622. priority=BaseInstanceService.STANDARD_EVENT_PRIORITY)
  623. def create(self, label, image, vm_type, subnet,
  624. key_pair=None, vm_firewalls=None, user_data=None,
  625. launch_config=None, **kwargs):
  626. AWSInstance.assert_valid_resource_label(label)
  627. image_id = image.id if isinstance(image, MachineImage) else image
  628. vm_size = vm_type.id if \
  629. isinstance(vm_type, VMType) else vm_type
  630. subnet = (self.provider.networking.subnets.get(subnet)
  631. if isinstance(subnet, str) else subnet)
  632. zone_name = self.provider.zone_name
  633. key_pair_name = key_pair.name if isinstance(
  634. key_pair,
  635. KeyPair) else key_pair
  636. if launch_config:
  637. bdm = self._process_block_device_mappings(launch_config)
  638. else:
  639. bdm = None
  640. subnet_id, zone_id, vm_firewall_ids = \
  641. self._resolve_launch_options(subnet, zone_name, vm_firewalls)
  642. placement = {'AvailabilityZone': zone_id} if zone_id else None
  643. inst = self.svc.create('create_instances',
  644. ImageId=image_id,
  645. MinCount=1,
  646. MaxCount=1,
  647. KeyName=key_pair_name,
  648. SecurityGroupIds=vm_firewall_ids or None,
  649. UserData=str(user_data) or None,
  650. InstanceType=vm_size,
  651. Placement=placement,
  652. BlockDeviceMappings=bdm,
  653. SubnetId=subnet_id
  654. )
  655. if inst and len(inst) == 1:
  656. # Wait until the resource exists
  657. # pylint:disable=protected-access
  658. inst[0]._wait_till_exists()
  659. # Tag the instance w/ the name
  660. inst[0].label = label
  661. return inst[0]
  662. raise ValueError(
  663. 'Expected a single object response, got a list: %s' % inst)
  664. @dispatch(event="provider.compute.instances.get",
  665. priority=BaseInstanceService.STANDARD_EVENT_PRIORITY)
  666. def get(self, instance_id):
  667. return self.svc.get(instance_id)
  668. @dispatch(event="provider.compute.instances.find",
  669. priority=BaseInstanceService.STANDARD_EVENT_PRIORITY)
  670. def find(self, **kwargs):
  671. label = kwargs.pop('label', None)
  672. # All kwargs should have been popped at this time.
  673. if len(kwargs) > 0:
  674. raise InvalidParamException(
  675. "Unrecognised parameters for search: %s. Supported "
  676. "attributes: %s" % (kwargs, 'label'))
  677. return self.svc.find(
  678. filters={'tag:Name': label,
  679. 'availability-zone': self.provider.zone_name})
  680. @dispatch(event="provider.compute.instances.list",
  681. priority=BaseInstanceService.STANDARD_EVENT_PRIORITY)
  682. def list(self, limit=None, marker=None):
  683. return self.svc.find(
  684. filters={'availability-zone': self.provider.zone_name},
  685. limit=limit, marker=marker)
  686. @dispatch(event="provider.compute.instances.delete",
  687. priority=BaseInstanceService.STANDARD_EVENT_PRIORITY)
  688. def delete(self, instance):
  689. aws_inst = (instance if isinstance(instance, AWSInstance) else
  690. self.get(instance))
  691. if aws_inst:
  692. # pylint:disable=protected-access
  693. aws_inst._ec2_instance.terminate()
  694. class AWSVMTypeService(BaseVMTypeService):
  695. def __init__(self, provider):
  696. super(AWSVMTypeService, self).__init__(provider)
  697. @property
  698. @cachetools.cached(cachetools.TTLCache(maxsize=1, ttl=24*3600))
  699. def instance_data(self):
  700. """
  701. Fetch info about the available instances.
  702. To update this information, update the file pointed to by the
  703. ``provider.AWS_INSTANCE_DATA_DEFAULT_URL`` above. The content for this
  704. file should be obtained from this repo:
  705. https://github.com/powdahound/ec2instances.info (in particular, this
  706. file: https://raw.githubusercontent.com/powdahound/ec2instances.info/
  707. master/www/instances.json).
  708. """
  709. r = requests.get(self.provider.config.get(
  710. "aws_instance_info_url",
  711. self.provider.AWS_INSTANCE_DATA_DEFAULT_URL))
  712. # Some instances are only available in certain regions. Use pricing
  713. # info to determine and filter out instance types that are not
  714. # available in the current region
  715. vm_types_list = r.json()
  716. return [vm_type for vm_type in vm_types_list
  717. if vm_type.get('pricing', {}).get(self.provider.region_name)]
  718. @dispatch(event="provider.compute.vm_types.list",
  719. priority=BaseVMTypeService.STANDARD_EVENT_PRIORITY)
  720. def list(self, limit=None, marker=None):
  721. vm_types = [AWSVMType(self.provider, vm_type)
  722. for vm_type in self.instance_data]
  723. return ClientPagedResultList(self.provider, vm_types,
  724. limit=limit, marker=marker)
  725. class AWSRegionService(BaseRegionService):
  726. def __init__(self, provider):
  727. super(AWSRegionService, self).__init__(provider)
  728. @dispatch(event="provider.compute.regions.get",
  729. priority=BaseRegionService.STANDARD_EVENT_PRIORITY)
  730. def get(self, region_id):
  731. log.debug("Getting AWS Region Service with the id: %s",
  732. region_id)
  733. region = [r for r in self if r.id == region_id]
  734. if region:
  735. return region[0]
  736. else:
  737. return None
  738. @dispatch(event="provider.compute.regions.list",
  739. priority=BaseRegionService.STANDARD_EVENT_PRIORITY)
  740. def list(self, limit=None, marker=None):
  741. regions = [
  742. AWSRegion(self.provider, region) for region in
  743. self.provider.ec2_conn.meta.client.describe_regions()
  744. .get('Regions', [])]
  745. return ClientPagedResultList(self.provider, regions,
  746. limit=limit, marker=marker)
  747. @property
  748. def current(self):
  749. return self.get(self._provider.region_name)
  750. class AWSNetworkingService(BaseNetworkingService):
  751. def __init__(self, provider):
  752. super(AWSNetworkingService, self).__init__(provider)
  753. self._network_service = AWSNetworkService(self.provider)
  754. self._subnet_service = AWSSubnetService(self.provider)
  755. self._router_service = AWSRouterService(self.provider)
  756. self._gateway_service = AWSGatewayService(self.provider)
  757. self._floating_ip_service = AWSFloatingIPService(self.provider)
  758. @property
  759. def networks(self):
  760. return self._network_service
  761. @property
  762. def subnets(self):
  763. return self._subnet_service
  764. @property
  765. def routers(self):
  766. return self._router_service
  767. @property
  768. def _gateways(self):
  769. return self._gateway_service
  770. @property
  771. def _floating_ips(self):
  772. return self._floating_ip_service
  773. class AWSNetworkService(BaseNetworkService):
  774. def __init__(self, provider):
  775. super(AWSNetworkService, self).__init__(provider)
  776. self.svc = BotoEC2Service(provider=self.provider,
  777. cb_resource=AWSNetwork,
  778. boto_collection_name='vpcs')
  779. @dispatch(event="provider.networking.networks.get",
  780. priority=BaseNetworkService.STANDARD_EVENT_PRIORITY)
  781. def get(self, network_id):
  782. return self.svc.get(network_id)
  783. @dispatch(event="provider.networking.networks.list",
  784. priority=BaseNetworkService.STANDARD_EVENT_PRIORITY)
  785. def list(self, limit=None, marker=None):
  786. return self.svc.list(limit=limit, marker=marker)
  787. @dispatch(event="provider.networking.networks.find",
  788. priority=BaseNetworkService.STANDARD_EVENT_PRIORITY)
  789. def find(self, **kwargs):
  790. label = kwargs.pop('label', None)
  791. # All kwargs should have been popped at this time.
  792. if len(kwargs) > 0:
  793. raise InvalidParamException(
  794. "Unrecognised parameters for search: %s. Supported "
  795. "attributes: %s" % (kwargs, 'label'))
  796. log.debug("Searching for AWS Network Service %s", label)
  797. return self.svc.find(filters={'tag:Name': label})
  798. @dispatch(event="provider.networking.networks.create",
  799. priority=BaseNetworkService.STANDARD_EVENT_PRIORITY)
  800. def create(self, label, cidr_block):
  801. AWSNetwork.assert_valid_resource_label(label)
  802. cb_net = self.svc.create('create_vpc', CidrBlock=cidr_block)
  803. # Wait until ready to tag instance
  804. cb_net.wait_till_ready()
  805. if label:
  806. cb_net.label = label
  807. return cb_net
  808. @dispatch(event="provider.networking.networks.delete",
  809. priority=BaseNetworkService.STANDARD_EVENT_PRIORITY)
  810. def delete(self, network):
  811. network = (network if isinstance(network, AWSNetwork)
  812. else self.get(network))
  813. if network:
  814. # pylint:disable=protected-access
  815. network._vpc.delete()
  816. def get_or_create_default(self):
  817. # # Look for provided default network
  818. # for net in self.provider.networking.networks:
  819. # pylint:disable=protected-access
  820. # if net._vpc.is_default:
  821. # return net
  822. # No provider-default, try CB-default instead
  823. default_nets = self.provider.networking.networks.find(
  824. label=AWSNetwork.CB_DEFAULT_NETWORK_LABEL)
  825. if default_nets:
  826. return default_nets[0]
  827. else:
  828. log.info("Creating a CloudBridge-default network labeled %s",
  829. AWSNetwork.CB_DEFAULT_NETWORK_LABEL)
  830. return self.provider.networking.networks.create(
  831. label=AWSNetwork.CB_DEFAULT_NETWORK_LABEL,
  832. cidr_block=AWSNetwork.CB_DEFAULT_IPV4RANGE)
  833. class AWSSubnetService(BaseSubnetService):
  834. def __init__(self, provider):
  835. super(AWSSubnetService, self).__init__(provider)
  836. self.svc = BotoEC2Service(provider=self.provider,
  837. cb_resource=AWSSubnet,
  838. boto_collection_name='subnets')
  839. @dispatch(event="provider.networking.subnets.get",
  840. priority=BaseSubnetService.STANDARD_EVENT_PRIORITY)
  841. def get(self, subnet_id):
  842. return self.svc.get(subnet_id)
  843. @dispatch(event="provider.networking.subnets.list",
  844. priority=BaseSubnetService.STANDARD_EVENT_PRIORITY)
  845. def list(self, network=None, limit=None, marker=None):
  846. network_id = network.id if isinstance(network, AWSNetwork) else network
  847. if network_id:
  848. return self.svc.find(
  849. filters={'vpc-id': network_id,
  850. 'availability-zone': self.provider.zone_name},
  851. limit=limit, marker=marker)
  852. else:
  853. return self.svc.find(
  854. filters={'availability-zone': self.provider.zone_name},
  855. limit=limit, marker=marker)
  856. @dispatch(event="provider.networking.subnets.find",
  857. priority=BaseSubnetService.STANDARD_EVENT_PRIORITY)
  858. def find(self, network=None, **kwargs):
  859. label = kwargs.pop('label', None)
  860. # All kwargs should have been popped at this time.
  861. if len(kwargs) > 0:
  862. raise InvalidParamException(
  863. "Unrecognised parameters for search: %s. Supported "
  864. "attributes: %s" % (kwargs, 'label'))
  865. log.debug("Searching for AWS Subnet Service %s", label)
  866. return self.svc.find(
  867. filters={'tag:Name': label,
  868. 'availability-zone': self.provider.zone_name})
  869. @dispatch(event="provider.networking.subnets.create",
  870. priority=BaseSubnetService.STANDARD_EVENT_PRIORITY)
  871. def create(self, label, network, cidr_block):
  872. AWSSubnet.assert_valid_resource_label(label)
  873. zone_name = self.provider.zone_name
  874. network_id = network.id if isinstance(network, AWSNetwork) else network
  875. subnet = self.svc.create('create_subnet',
  876. VpcId=network_id,
  877. CidrBlock=cidr_block,
  878. AvailabilityZone=zone_name)
  879. if label:
  880. subnet.label = label
  881. return subnet
  882. @dispatch(event="provider.networking.subnets.delete",
  883. priority=BaseSubnetService.STANDARD_EVENT_PRIORITY)
  884. def delete(self, subnet):
  885. sn = subnet if isinstance(subnet, AWSSubnet) else self.get(subnet)
  886. if sn:
  887. # pylint:disable=protected-access
  888. sn._subnet.delete()
  889. def get_or_create_default(self):
  890. zone_name = self.provider.zone_name
  891. # # Look for provider default subnet in current zone
  892. # if zone_name:
  893. # snl = self.svc.find('availabilityZone', zone_name)
  894. #
  895. # else:
  896. # snl = self.svc.list()
  897. # # Find first available default subnet by sorted order
  898. # # of availability zone. Prefer zone us-east-1a over 1e,
  899. # # because newer zones tend to have less compatibility
  900. # # with different instance types (e.g. c5.large not available
  901. # # on us-east-1e as of 14 Dec. 2017).
  902. # # pylint:disable=protected-access
  903. # snl.sort(key=lambda sn: sn._subnet.availability_zone)
  904. #
  905. # for sn in snl:
  906. # # pylint:disable=protected-access
  907. # if sn._subnet.default_for_az:
  908. # return sn
  909. # If no provider-default subnet has been found, look for
  910. # cloudbridge-default by label. We suffix labels by availability zone,
  911. # thus we add the wildcard for the regular expression to find the
  912. # subnet
  913. snl = self.find(label=AWSSubnet.CB_DEFAULT_SUBNET_LABEL + "*")
  914. if snl:
  915. # pylint:disable=protected-access
  916. snl.sort(key=lambda sn: sn._subnet.availability_zone)
  917. for subnet in snl:
  918. if subnet.zone.name == zone_name:
  919. return subnet
  920. # No default Subnet exists, try to create a CloudBridge-specific
  921. # subnet. This involves creating the network, subnets, internet
  922. # gateway, and connecting it all together so that the network has
  923. # Internet connectivity.
  924. # Check if a default net already exists and get it or create on
  925. default_net = self.provider.networking.networks.get_or_create_default()
  926. # Get/create an internet gateway for the default network and a
  927. # corresponding router if it does not already exist.
  928. # NOTE: Comment this out because the docs instruct users to setup
  929. # network connectivity manually. There's a bit of discrepancy here
  930. # though because the provider-default network will have Internet
  931. # connectivity (unlike the CloudBridge-default network with this
  932. # being commented) and is hence left in the codebase.
  933. # default_gtw = default_net.gateways.get_or_create()
  934. # router_label = "{0}-router".format(
  935. # AWSNetwork.CB_DEFAULT_NETWORK_LABEL)
  936. # default_routers = self.provider.networking.routers.find(
  937. # label=router_label)
  938. # if len(default_routers) == 0:
  939. # default_router = self.provider.networking.routers.create(
  940. # router_label, default_net)
  941. # default_router.attach_gateway(default_gtw)
  942. # else:
  943. # default_router = default_routers[0]
  944. # Create a subnet in each of the region's zones
  945. region = self.provider.compute.regions.get(self.provider.region_name)
  946. default_sn = None
  947. # Determine how many subnets we'll need for the default network and the
  948. # number of available zones. We need to derive a non-overlapping
  949. # network size for each subnet within the parent net so figure those
  950. # subnets here. `<net>.subnets` method will do this but we need to give
  951. # it a prefix. Determining that prefix depends on the size of the
  952. # network and should be incorporate the number of zones. So iterate
  953. # over potential number of subnets until enough can be created to
  954. # accommodate the number of available zones. That is where the fixed
  955. # number comes from in the for loop as that many iterations will yield
  956. # more potential subnets than any region has zones.
  957. ip_net = ipaddress.ip_network(AWSNetwork.CB_DEFAULT_IPV4RANGE)
  958. for x in range(5):
  959. if len(region.zones) <= len(list(ip_net.subnets(
  960. prefixlen_diff=x))):
  961. prefixlen_diff = x
  962. break
  963. subnets = list(ip_net.subnets(prefixlen_diff=prefixlen_diff))
  964. for i, z in reversed(list(enumerate(region.zones))):
  965. if zone_name == z.name:
  966. sn_label = "{0}-{1}".format(AWSSubnet.CB_DEFAULT_SUBNET_LABEL,
  967. z.id[-1])
  968. log.info("Creating a default CloudBridge subnet %s: %s" %
  969. (sn_label, str(subnets[i])))
  970. sn = self.create(sn_label, default_net, str(subnets[i]))
  971. # Create a route table entry between the SN and the inet
  972. # gateway. See note above about why this is commented
  973. # default_router.attach_subnet(sn)
  974. default_sn = sn
  975. # No specific zone was supplied; return the last created subnet
  976. # The list was originally reversed to have the last subnet be in zone a
  977. if not default_sn:
  978. default_sn = sn
  979. return default_sn
  980. class AWSRouterService(BaseRouterService):
  981. """For AWS, a CloudBridge router corresponds to an AWS Route Table."""
  982. def __init__(self, provider):
  983. super(AWSRouterService, self).__init__(provider)
  984. self.svc = BotoEC2Service(provider=self.provider,
  985. cb_resource=AWSRouter,
  986. boto_collection_name='route_tables')
  987. @dispatch(event="provider.networking.routers.get",
  988. priority=BaseRouterService.STANDARD_EVENT_PRIORITY)
  989. def get(self, router_id):
  990. return self.svc.get(router_id)
  991. @dispatch(event="provider.networking.routers.find",
  992. priority=BaseRouterService.STANDARD_EVENT_PRIORITY)
  993. def find(self, **kwargs):
  994. label = kwargs.pop('label', None)
  995. # All kwargs should have been popped at this time.
  996. if len(kwargs) > 0:
  997. raise InvalidParamException(
  998. "Unrecognised parameters for search: %s. Supported "
  999. "attributes: %s" % (kwargs, 'label'))
  1000. log.debug("Searching for AWS Router Service %s", label)
  1001. return self.svc.find(filters={'tag:Name': label})
  1002. @dispatch(event="provider.networking.routers.list",
  1003. priority=BaseRouterService.STANDARD_EVENT_PRIORITY)
  1004. def list(self, limit=None, marker=None):
  1005. return self.svc.list(limit=limit, marker=marker)
  1006. @dispatch(event="provider.networking.routers.create",
  1007. priority=BaseRouterService.STANDARD_EVENT_PRIORITY)
  1008. def create(self, label, network):
  1009. network_id = network.id if isinstance(network, AWSNetwork) else network
  1010. cb_router = self.svc.create('create_route_table', VpcId=network_id)
  1011. if label:
  1012. cb_router.label = label
  1013. return cb_router
  1014. @dispatch(event="provider.networking.routers.delete",
  1015. priority=BaseRouterService.STANDARD_EVENT_PRIORITY)
  1016. def delete(self, router):
  1017. r = router if isinstance(router, AWSRouter) else self.get(router)
  1018. if r:
  1019. # pylint:disable=protected-access
  1020. r._route_table.delete()
  1021. class AWSGatewayService(BaseGatewayService):
  1022. def __init__(self, provider):
  1023. super(AWSGatewayService, self).__init__(provider)
  1024. self.svc = BotoEC2Service(provider=provider,
  1025. cb_resource=AWSInternetGateway,
  1026. boto_collection_name='internet_gateways')
  1027. @dispatch(event="provider.networking.gateways.get_or_create",
  1028. priority=BaseGatewayService.STANDARD_EVENT_PRIORITY)
  1029. def get_or_create(self, network):
  1030. network_id = network.id if isinstance(
  1031. network, AWSNetwork) else network
  1032. # Don't filter by label because it may conflict with at least the
  1033. # default VPC that most accounts have but that network is typically
  1034. # without a name.
  1035. gtw = self.svc.find(filters={'attachment.vpc-id': network_id})
  1036. if gtw:
  1037. return gtw[0] # There can be only one gtw attached to a VPC
  1038. # Gateway does not exist so create one and attach to the supplied net
  1039. cb_gateway = self.svc.create('create_internet_gateway')
  1040. cb_gateway._gateway.create_tags(
  1041. Tags=[{'Key': 'Name',
  1042. 'Value': AWSInternetGateway.CB_DEFAULT_INET_GATEWAY_NAME
  1043. }])
  1044. cb_gateway._gateway.attach_to_vpc(VpcId=network_id)
  1045. return cb_gateway
  1046. @dispatch(event="provider.networking.gateways.delete",
  1047. priority=BaseGatewayService.STANDARD_EVENT_PRIORITY)
  1048. def delete(self, network, gateway):
  1049. gw = (gateway if isinstance(gateway, AWSInternetGateway)
  1050. else self.svc.get(gateway))
  1051. try:
  1052. if gw.network_id:
  1053. # pylint:disable=protected-access
  1054. gw._gateway.detach_from_vpc(VpcId=gw.network_id)
  1055. except ClientError as e:
  1056. log.warn("Error deleting gateway {0}: {1}".format(self.id, e))
  1057. # pylint:disable=protected-access
  1058. gw._gateway.delete()
  1059. @dispatch(event="provider.networking.gateways.list",
  1060. priority=BaseGatewayService.STANDARD_EVENT_PRIORITY)
  1061. def list(self, network, limit=None, marker=None):
  1062. log.debug("Listing current AWS internet gateways for net %s.",
  1063. network.id)
  1064. fltr = [{'Name': 'attachment.vpc-id', 'Values': [network.id]}]
  1065. return self.svc.list(limit=None, marker=None, Filters=fltr)
  1066. class AWSFloatingIPService(BaseFloatingIPService):
  1067. def __init__(self, provider):
  1068. super(AWSFloatingIPService, self).__init__(provider)
  1069. self.svc = BotoEC2Service(provider=self.provider,
  1070. cb_resource=AWSFloatingIP,
  1071. boto_collection_name='vpc_addresses')
  1072. @dispatch(event="provider.networking.floating_ips.get",
  1073. priority=BaseFloatingIPService.STANDARD_EVENT_PRIORITY)
  1074. def get(self, gateway, fip_id):
  1075. log.debug("Getting AWS Floating IP Service with the id: %s", fip_id)
  1076. return self.svc.get(fip_id)
  1077. @dispatch(event="provider.networking.floating_ips.list",
  1078. priority=BaseFloatingIPService.STANDARD_EVENT_PRIORITY)
  1079. def list(self, gateway, limit=None, marker=None):
  1080. return self.svc.list(limit, marker)
  1081. @dispatch(event="provider.networking.floating_ips.create",
  1082. priority=BaseFloatingIPService.STANDARD_EVENT_PRIORITY)
  1083. def create(self, gateway):
  1084. log.debug("Creating a floating IP under gateway %s", gateway)
  1085. ip = self.provider.ec2_conn.meta.client.allocate_address(
  1086. Domain='vpc')
  1087. return AWSFloatingIP(
  1088. self.provider,
  1089. self.provider.ec2_conn.VpcAddress(ip.get('AllocationId')))
  1090. @dispatch(event="provider.networking.floating_ips.delete",
  1091. priority=BaseFloatingIPService.STANDARD_EVENT_PRIORITY)
  1092. def delete(self, gateway, fip):
  1093. if isinstance(fip, AWSFloatingIP):
  1094. # pylint:disable=protected-access
  1095. aws_fip = fip._ip
  1096. else:
  1097. aws_fip = self.svc.get_raw(fip)
  1098. aws_fip.release()