Tony Mack | 9b5dd75 | 2013-05-09 20:33:21 -0400 | [diff] [blame] | 1 | import commands |
Tony Mack | 0ae3fe3 | 2013-10-08 08:47:53 -0400 | [diff] [blame] | 2 | import hashlib |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 3 | from planetstack.config import Config |
Tony Mack | 336e0f9 | 2014-11-30 15:53:08 -0500 | [diff] [blame] | 4 | from core.models import Controller |
Tony Mack | 387a73f | 2013-09-18 07:59:14 -0400 | [diff] [blame] | 5 | |
| 6 | try: |
| 7 | from openstack.client import OpenStackClient |
Tony Mack | 387a73f | 2013-09-18 07:59:14 -0400 | [diff] [blame] | 8 | has_openstack = True |
| 9 | except: |
| 10 | has_openstack = False |
| 11 | |
| 12 | manager_enabled = Config().api_nova_enabled |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 13 | |
Tony Mack | a033ced | 2013-03-29 17:57:06 -0400 | [diff] [blame] | 14 | class OpenStackDriver: |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 15 | |
Tony Mack | 9446692 | 2014-06-19 20:09:40 -0400 | [diff] [blame] | 16 | def __init__(self, config = None, client=None): |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 17 | if config: |
| 18 | self.config = Config(config) |
| 19 | else: |
Tony Mack | e4be32f | 2014-03-11 20:45:25 -0400 | [diff] [blame] | 20 | self.config = Config() |
Tony Mack | ad8b093 | 2013-04-07 19:32:07 -0400 | [diff] [blame] | 21 | |
| 22 | if client: |
| 23 | self.shell = client |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 24 | |
Tony Mack | c12d5ca | 2013-09-24 10:30:39 -0400 | [diff] [blame] | 25 | self.enabled = manager_enabled |
| 26 | self.has_openstack = has_openstack |
Tony Mack | 336e0f9 | 2014-11-30 15:53:08 -0500 | [diff] [blame] | 27 | self.controller = None |
Tony Mack | a16cfb2 | 2014-06-19 22:00:22 -0400 | [diff] [blame] | 28 | self.admin_user = None |
Tony Mack | c12d5ca | 2013-09-24 10:30:39 -0400 | [diff] [blame] | 29 | |
Tony Mack | 336e0f9 | 2014-11-30 15:53:08 -0500 | [diff] [blame] | 30 | def client_driver(self, caller=None, tenant=None, controller=None): |
Tony Mack | 387a73f | 2013-09-18 07:59:14 -0400 | [diff] [blame] | 31 | if caller: |
| 32 | auth = {'username': caller.email, |
| 33 | 'password': hashlib.md5(caller.password).hexdigest()[:6], |
| 34 | 'tenant': tenant} |
Tony Mack | 336e0f9 | 2014-11-30 15:53:08 -0500 | [diff] [blame] | 35 | client = OpenStackClient(controller=controller, **auth) |
Tony Mack | 387a73f | 2013-09-18 07:59:14 -0400 | [diff] [blame] | 36 | else: |
Tony Mack | 336e0f9 | 2014-11-30 15:53:08 -0500 | [diff] [blame] | 37 | admin_driver = self.admin_driver(tenant=tenant, controller=controller) |
| 38 | client = OpenStackClient(tenant=tenant, controller=admin_driver.controller) |
Tony Mack | e4be32f | 2014-03-11 20:45:25 -0400 | [diff] [blame] | 39 | |
Tony Mack | 9446692 | 2014-06-19 20:09:40 -0400 | [diff] [blame] | 40 | driver = OpenStackDriver(client=client) |
Sapan Bhatia | 475c597 | 2014-11-05 10:32:41 -0500 | [diff] [blame] | 41 | #driver.admin_user = admin_driver.admin_user |
Tony Mack | 336e0f9 | 2014-11-30 15:53:08 -0500 | [diff] [blame] | 42 | #driver.controller = admin_driver.controller |
Tony Mack | 387a73f | 2013-09-18 07:59:14 -0400 | [diff] [blame] | 43 | return driver |
| 44 | |
Tony Mack | 336e0f9 | 2014-11-30 15:53:08 -0500 | [diff] [blame] | 45 | def admin_driver(self, tenant=None, controller=None): |
Tony Mack | 3375088 | 2015-01-06 18:27:48 -0500 | [diff] [blame^] | 46 | if isinstance(controller, int): |
| 47 | controller = Controller.objects.get(id=controller.id) |
Tony Mack | 336e0f9 | 2014-11-30 15:53:08 -0500 | [diff] [blame] | 48 | client = OpenStackClient(tenant=tenant, controller=controller) |
Tony Mack | 9446692 | 2014-06-19 20:09:40 -0400 | [diff] [blame] | 49 | driver = OpenStackDriver(client=client) |
Tony Mack | 336e0f9 | 2014-11-30 15:53:08 -0500 | [diff] [blame] | 50 | driver.admin_user = client.keystone.users.find(name=controller.admin_user) |
| 51 | driver.controller = controller |
Tony Mack | e4be32f | 2014-03-11 20:45:25 -0400 | [diff] [blame] | 52 | return driver |
Tony Mack | 387a73f | 2013-09-18 07:59:14 -0400 | [diff] [blame] | 53 | |
Scott Baker | 8a6d512 | 2013-08-07 18:57:15 -0700 | [diff] [blame] | 54 | def create_role(self, name): |
Tony Mack | b5f339f | 2013-04-08 16:29:38 -0400 | [diff] [blame] | 55 | roles = self.shell.keystone.roles.findall(name=name) |
Tony Mack | 0f26b27 | 2014-04-15 12:17:15 -0400 | [diff] [blame] | 56 | roles_title = self.shell.keystone.roles.findall(name=name.title()) |
| 57 | roles_found = roles + roles_title |
| 58 | if not roles_found: |
Tony Mack | b5f339f | 2013-04-08 16:29:38 -0400 | [diff] [blame] | 59 | role = self.shell.keystone.roles.create(name) |
| 60 | else: |
Tony Mack | 0f26b27 | 2014-04-15 12:17:15 -0400 | [diff] [blame] | 61 | role = roles_found[0] |
Tony Mack | b5f339f | 2013-04-08 16:29:38 -0400 | [diff] [blame] | 62 | return role |
| 63 | |
Tony Mack | a704053 | 2013-04-08 21:45:43 -0400 | [diff] [blame] | 64 | def delete_role(self, filter): |
| 65 | roles = self.shell.keystone.roles.findall(**filter) |
Tony Mack | b5f339f | 2013-04-08 16:29:38 -0400 | [diff] [blame] | 66 | for role in roles: |
| 67 | self.shell.keystone.roles.delete(role) |
| 68 | return 1 |
| 69 | |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 70 | def create_tenant(self, tenant_name, enabled, description): |
Tony Mack | 5ee2190 | 2013-03-29 20:36:42 -0400 | [diff] [blame] | 71 | """Create keystone tenant. Suggested fields: name, description, enabled""" |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 72 | tenants = self.shell.keystone.tenants.findall(name=tenant_name) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 73 | if not tenants: |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 74 | fields = {'tenant_name': tenant_name, 'enabled': enabled, |
Tony Mack | 488950a | 2013-04-07 20:35:50 -0400 | [diff] [blame] | 75 | 'description': description} |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 76 | tenant = self.shell.keystone.tenants.create(**fields) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 77 | else: |
| 78 | tenant = tenants[0] |
Tony Mack | 0dc2f56 | 2013-04-29 19:41:39 -0400 | [diff] [blame] | 79 | |
| 80 | # always give the admin user the admin role to any tenant created |
| 81 | # by the driver. |
| 82 | self.add_user_role(self.admin_user.id, tenant.id, 'admin') |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 83 | return tenant |
| 84 | |
| 85 | def update_tenant(self, id, **kwds): |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 86 | return self.shell.keystone.tenants.update(id, **kwds) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 87 | |
| 88 | def delete_tenant(self, id): |
Tony Mack | ac3e014 | 2013-06-26 11:28:57 -0400 | [diff] [blame] | 89 | ctx = self.shell.nova_db.ctx |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 90 | tenants = self.shell.keystone.tenants.findall(id=id) |
| 91 | for tenant in tenants: |
Tony Mack | ac3e014 | 2013-06-26 11:28:57 -0400 | [diff] [blame] | 92 | # nova does not automatically delete the tenant's instances |
| 93 | # so we manually delete instances before deleteing the tenant |
Scott Baker | 8a6d512 | 2013-08-07 18:57:15 -0700 | [diff] [blame] | 94 | instances = self.shell.nova_db.instance_get_all_by_filters(ctx, |
Tony Mack | ac3e014 | 2013-06-26 11:28:57 -0400 | [diff] [blame] | 95 | {'project_id': tenant.id}, 'id', 'asc') |
Tony Mack | 28897e1 | 2013-07-09 07:25:38 -0400 | [diff] [blame] | 96 | client = OpenStackClient(tenant=tenant.name) |
Tony Mack | ac3e014 | 2013-06-26 11:28:57 -0400 | [diff] [blame] | 97 | driver = OpenStackDriver(client=client) |
| 98 | for instance in instances: |
| 99 | driver.destroy_instance(instance.id) |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 100 | self.shell.keystone.tenants.delete(tenant) |
| 101 | return 1 |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 102 | |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 103 | def create_user(self, name, email, password, enabled): |
| 104 | users = self.shell.keystone.users.findall(email=email) |
| 105 | if not users: |
| 106 | fields = {'name': name, 'email': email, 'password': password, |
| 107 | 'enabled': enabled} |
Tony Mack | fdcbf83 | 2013-04-09 11:47:11 -0400 | [diff] [blame] | 108 | user = self.shell.keystone.users.create(**fields) |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 109 | else: |
| 110 | user = users[0] |
| 111 | return user |
| 112 | |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 113 | def delete_user(self, id): |
| 114 | users = self.shell.keystone.users.findall(id=id) |
| 115 | for user in users: |
Tony Mack | c59fcaf | 2013-06-10 11:14:04 -0400 | [diff] [blame] | 116 | # delete users keys |
| 117 | keys = self.shell.nova.keypairs.findall() |
| 118 | for key in keys: |
| 119 | self.shell.nova.keypairs.delete(key) |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 120 | self.shell.keystone.users.delete(user) |
Tony Mack | ac2ebaa | 2014-04-10 12:26:18 -0400 | [diff] [blame] | 121 | return 1 |
| 122 | |
| 123 | def get_admin_role(self): |
| 124 | role = None |
| 125 | for admin_role_name in ['admin', 'Admin']: |
| 126 | roles = self.shell.keystone.roles.findall(name=admin_role_name) |
| 127 | if roles: |
| 128 | role = roles[0] |
| 129 | break |
| 130 | return role |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 131 | |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 132 | def add_user_role(self, kuser_id, tenant_id, role_name): |
| 133 | user = self.shell.keystone.users.find(id=kuser_id) |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 134 | tenant = self.shell.keystone.tenants.find(id=tenant_id) |
Tony Mack | a4f1feb | 2014-03-28 20:09:21 -0400 | [diff] [blame] | 135 | # admin role can be lowercase or title. Look for both |
| 136 | role = None |
| 137 | if role_name.lower() == 'admin': |
Tony Mack | ac2ebaa | 2014-04-10 12:26:18 -0400 | [diff] [blame] | 138 | role = self.get_admin_role() |
| 139 | else: |
Tony Mack | a4f1feb | 2014-03-28 20:09:21 -0400 | [diff] [blame] | 140 | # look up non admin role or force exception when admin role isnt found |
| 141 | role = self.shell.keystone.roles.find(name=role_name) |
Tony Mack | 54304c6 | 2013-04-28 11:27:07 -0400 | [diff] [blame] | 142 | |
| 143 | role_found = False |
| 144 | user_roles = user.list_roles(tenant.id) |
| 145 | for user_role in user_roles: |
| 146 | if user_role.name == role.name: |
| 147 | role_found = True |
| 148 | if not role_found: |
| 149 | tenant.add_user(user, role) |
| 150 | |
| 151 | return 1 |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 152 | |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 153 | def delete_user_role(self, kuser_id, tenant_id, role_name): |
| 154 | user = self.shell.keystone.users.find(id=kuser_id) |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 155 | tenant = self.shell.keystone.tenants.find(id=tenant_id) |
Tony Mack | ac2ebaa | 2014-04-10 12:26:18 -0400 | [diff] [blame] | 156 | # admin role can be lowercase or title. Look for both |
| 157 | role = None |
| 158 | if role_name.lower() == 'admin': |
| 159 | role = self.get_admin_role() |
| 160 | else: |
| 161 | # look up non admin role or force exception when admin role isnt found |
| 162 | role = self.shell.keystone.roles.find(name=role_name) |
Tony Mack | 54304c6 | 2013-04-28 11:27:07 -0400 | [diff] [blame] | 163 | |
| 164 | role_found = False |
| 165 | user_roles = user.list_roles(tenant.id) |
| 166 | for user_role in user_roles: |
| 167 | if user_role.name == role.name: |
| 168 | role_found = True |
| 169 | if role_found: |
| 170 | tenant.remove_user(user, role) |
| 171 | |
| 172 | return 1 |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 173 | |
Tony Mack | 6795fbe | 2013-06-11 10:10:24 -0400 | [diff] [blame] | 174 | def update_user(self, id, fields): |
| 175 | if 'password' in fields: |
| 176 | self.shell.keystone.users.update_password(id, fields['password']) |
| 177 | if 'enabled' in fields: |
| 178 | self.shell.keystone.users.update_enabled(id, fields['enabled']) |
| 179 | return 1 |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 180 | |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 181 | def create_router(self, name, set_gateway=True): |
Tony Mack | 28be756 | 2013-04-10 08:45:55 -0400 | [diff] [blame] | 182 | routers = self.shell.quantum.list_routers(name=name)['routers'] |
| 183 | if routers: |
| 184 | router = routers[0] |
| 185 | else: |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 186 | router = self.shell.quantum.create_router({'router': {'name': name}})['router'] |
Tony Mack | c1665b1 | 2013-04-29 16:45:14 -0400 | [diff] [blame] | 187 | # add router to external network |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 188 | if set_gateway: |
Tony Mack | 257bedc | 2013-04-10 08:58:09 -0400 | [diff] [blame] | 189 | nets = self.shell.quantum.list_networks()['networks'] |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 190 | for net in nets: |
| 191 | if net['router:external'] == True: |
Tony Mack | 9dcc050 | 2013-04-10 09:07:45 -0400 | [diff] [blame] | 192 | self.shell.quantum.add_gateway_router(router['id'], |
| 193 | {'network_id': net['id']}) |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 194 | |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 195 | return router |
| 196 | |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 197 | def delete_router(self, id): |
| 198 | routers = self.shell.quantum.list_routers(id=id)['routers'] |
Tony Mack | 28be756 | 2013-04-10 08:45:55 -0400 | [diff] [blame] | 199 | for router in routers: |
| 200 | self.shell.quantum.delete_router(router['id']) |
Tony Mack | c1665b1 | 2013-04-29 16:45:14 -0400 | [diff] [blame] | 201 | # remove router form external network |
Tony Mack | 0c0e05d | 2013-04-29 17:52:37 -0400 | [diff] [blame] | 202 | #nets = self.shell.quantum.list_networks()['networks'] |
| 203 | #for net in nets: |
| 204 | # if net['router:external'] == True: |
| 205 | # self.shell.quantum.remove_gateway_router(router['id']) |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 206 | |
| 207 | def add_router_interface(self, router_id, subnet_id): |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 208 | router = self.shell.quantum.show_router(router_id)['router'] |
| 209 | subnet = self.shell.quantum.show_subnet(subnet_id)['subnet'] |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 210 | if router and subnet: |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 211 | self.shell.quantum.add_interface_router(router_id, {'subnet_id': subnet_id}) |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 212 | |
| 213 | def delete_router_interface(self, router_id, subnet_id): |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 214 | router = self.shell.quantum.show_router(router_id) |
| 215 | subnet = self.shell.quantum.show_subnet(subnet_id) |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 216 | if router and subnet: |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 217 | self.shell.quantum.remove_interface_router(router_id, {'subnet_id': subnet_id}) |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 218 | |
Scott Baker | 1ac5ddb | 2013-08-14 10:50:48 -0700 | [diff] [blame] | 219 | def create_network(self, name, shared=False): |
Tony Mack | 28be756 | 2013-04-10 08:45:55 -0400 | [diff] [blame] | 220 | nets = self.shell.quantum.list_networks(name=name)['networks'] |
| 221 | if nets: |
Tony Mack | 53c6334 | 2013-04-10 08:11:45 -0400 | [diff] [blame] | 222 | net = nets[0] |
Tony Mack | 82a688d | 2013-04-10 08:12:58 -0400 | [diff] [blame] | 223 | else: |
Scott Baker | 1ac5ddb | 2013-08-14 10:50:48 -0700 | [diff] [blame] | 224 | net = self.shell.quantum.create_network({'network': {'name': name, 'shared': shared}})['network'] |
Tony Mack | 53c6334 | 2013-04-10 08:11:45 -0400 | [diff] [blame] | 225 | return net |
| 226 | |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 227 | def delete_network(self, id): |
| 228 | nets = self.shell.quantum.list_networks()['networks'] |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 229 | for net in nets: |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 230 | if net['id'] == id: |
Tony Mack | 94c6ea0 | 2013-04-16 21:41:50 -0400 | [diff] [blame] | 231 | # delete_all ports |
| 232 | self.delete_network_ports(net['id']) |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 233 | # delete all subnets: |
| 234 | for subnet_id in net['subnets']: |
| 235 | self.delete_subnet(subnet_id) |
| 236 | self.shell.quantum.delete_network(net['id']) |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 237 | return 1 |
Tony Mack | 94c6ea0 | 2013-04-16 21:41:50 -0400 | [diff] [blame] | 238 | |
| 239 | def delete_network_ports(self, network_id): |
| 240 | ports = self.shell.quantum.list_ports()['ports'] |
| 241 | for port in ports: |
Tony Mack | c1665b1 | 2013-04-29 16:45:14 -0400 | [diff] [blame] | 242 | if port['network_id'] == network_id: |
Tony Mack | 94c6ea0 | 2013-04-16 21:41:50 -0400 | [diff] [blame] | 243 | self.shell.quantum.delete_port(port['id']) |
| 244 | return 1 |
| 245 | |
| 246 | def delete_subnet_ports(self, subnet_id): |
| 247 | ports = self.shell.quantum.list_ports()['ports'] |
| 248 | for port in ports: |
| 249 | delete = False |
| 250 | for fixed_ip in port['fixed_ips']: |
| 251 | if fixed_ip['subnet_id'] == subnet_id: |
| 252 | delete=True |
| 253 | break |
| 254 | if delete: |
| 255 | self.shell.quantum.delete_port(port['id']) |
| 256 | return 1 |
| 257 | |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 258 | def create_subnet(self, name, network_id, cidr_ip, ip_version, start, end): |
| 259 | #nets = self.shell.quantum.list_networks(name=network_name)['networks'] |
| 260 | #if not nets: |
| 261 | # raise Exception, "No such network: %s" % network_name |
| 262 | #net = nets[0] |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 263 | |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 264 | subnet = None |
| 265 | subnets = self.shell.quantum.list_subnets()['subnets'] |
| 266 | for snet in subnets: |
| 267 | if snet['cidr'] == cidr_ip and snet['network_id'] == network_id: |
| 268 | subnet = snet |
Scott Baker | 8a6d512 | 2013-08-07 18:57:15 -0700 | [diff] [blame] | 269 | |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 270 | if not subnet: |
Andy Bavier | d2b0c6d | 2014-04-25 11:44:15 -0400 | [diff] [blame] | 271 | # HACK: Add metadata route -- Neutron does not reliably supply this |
| 272 | metadata_ip = cidr_ip.replace("0/24", "3") |
| 273 | |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 274 | allocation_pools = [{'start': start, 'end': end}] |
| 275 | subnet = {'subnet': {'name': name, |
| 276 | 'network_id': network_id, |
| 277 | 'ip_version': ip_version, |
| 278 | 'cidr': cidr_ip, |
Tony Mack | f549eb4 | 2014-04-18 15:33:24 -0400 | [diff] [blame] | 279 | #'dns_nameservers': ['8.8.8.8', '8.8.4.4'], |
Tony Mack | 8b04773 | 2014-04-25 12:42:21 -0400 | [diff] [blame] | 280 | 'host_routes': [{'destination':'169.254.169.254/32','nexthop':metadata_ip}], |
Andy Bavier | d2b0c6d | 2014-04-25 11:44:15 -0400 | [diff] [blame] | 281 | 'gateway_ip': None, |
Scott Baker | 8a6d512 | 2013-08-07 18:57:15 -0700 | [diff] [blame] | 282 | 'allocation_pools': allocation_pools}} |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 283 | subnet = self.shell.quantum.create_subnet(subnet)['subnet'] |
Andy Bavier | d2b0c6d | 2014-04-25 11:44:15 -0400 | [diff] [blame] | 284 | # self.add_external_route(subnet) |
| 285 | |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 286 | return subnet |
| 287 | |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 288 | def update_subnet(self, id, fields): |
| 289 | return self.shell.quantum.update_subnet(id, fields) |
| 290 | |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 291 | def delete_subnet(self, id): |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 292 | #return self.shell.quantum.delete_subnet(id=id) |
| 293 | # inefficient but fault tolerant |
| 294 | subnets = self.shell.quantum.list_subnets()['subnets'] |
| 295 | for subnet in subnets: |
| 296 | if subnet['id'] == id: |
Tony Mack | 94c6ea0 | 2013-04-16 21:41:50 -0400 | [diff] [blame] | 297 | self.delete_subnet_ports(subnet['id']) |
Tony Mack | 31d3413 | 2013-04-15 00:02:06 -0400 | [diff] [blame] | 298 | self.shell.quantum.delete_subnet(id) |
Tony Mack | 9b5dd75 | 2013-05-09 20:33:21 -0400 | [diff] [blame] | 299 | self.delete_external_route(subnet) |
| 300 | return 1 |
| 301 | |
Tony Mack | 70780e9 | 2013-06-30 12:46:16 -0400 | [diff] [blame] | 302 | def get_external_routes(self): |
| 303 | status, output = commands.getstatusoutput('route') |
| 304 | routes = output.split('\n')[3:] |
| 305 | return routes |
| 306 | |
| 307 | def add_external_route(self, subnet, routes=[]): |
| 308 | if not routes: |
| 309 | routes = self.get_external_routes() |
| 310 | |
Tony Mack | 9b5dd75 | 2013-05-09 20:33:21 -0400 | [diff] [blame] | 311 | ports = self.shell.quantum.list_ports()['ports'] |
| 312 | |
| 313 | gw_ip = subnet['gateway_ip'] |
| 314 | subnet_id = subnet['id'] |
| 315 | |
| 316 | # 1. Find the port associated with the subnet's gateway |
| 317 | # 2. Find the router associated with that port |
| 318 | # 3. Find the port associated with this router and on the external net |
| 319 | # 4. Set up route to the subnet through the port from step 3 |
| 320 | ip_address = None |
| 321 | for port in ports: |
| 322 | for fixed_ip in port['fixed_ips']: |
| 323 | if fixed_ip['subnet_id'] == subnet_id and fixed_ip['ip_address'] == gw_ip: |
| 324 | gw_port = port |
| 325 | router_id = gw_port['device_id'] |
| 326 | router = self.shell.quantum.show_router(router_id)['router'] |
Tony Mack | 4418701 | 2013-07-09 07:15:17 -0400 | [diff] [blame] | 327 | if router and router.get('external_gateway_info'): |
Tony Mack | 16f51e0 | 2013-07-01 10:33:29 -0400 | [diff] [blame] | 328 | ext_net = router['external_gateway_info']['network_id'] |
| 329 | for port in ports: |
| 330 | if port['device_id'] == router_id and port['network_id'] == ext_net: |
| 331 | ip_address = port['fixed_ips'][0]['ip_address'] |
Tony Mack | 9b5dd75 | 2013-05-09 20:33:21 -0400 | [diff] [blame] | 332 | |
| 333 | if ip_address: |
Tony Mack | 70780e9 | 2013-06-30 12:46:16 -0400 | [diff] [blame] | 334 | # check if external route already exists |
| 335 | route_exists = False |
| 336 | if routes: |
| 337 | for route in routes: |
| 338 | if subnet['cidr'] in route and ip_address in route: |
| 339 | route_exists = True |
| 340 | if not route_exists: |
| 341 | cmd = "route add -net %s dev br-ex gw %s" % (subnet['cidr'], ip_address) |
Tony Mack | 16f51e0 | 2013-07-01 10:33:29 -0400 | [diff] [blame] | 342 | s, o = commands.getstatusoutput(cmd) |
| 343 | #print cmd, "\n", s, o |
Tony Mack | 9b5dd75 | 2013-05-09 20:33:21 -0400 | [diff] [blame] | 344 | |
| 345 | return 1 |
| 346 | |
| 347 | def delete_external_route(self, subnet): |
| 348 | ports = self.shell.quantum.list_ports()['ports'] |
| 349 | |
| 350 | gw_ip = subnet['gateway_ip'] |
| 351 | subnet_id = subnet['id'] |
| 352 | |
| 353 | # 1. Find the port associated with the subnet's gateway |
| 354 | # 2. Find the router associated with that port |
| 355 | # 3. Find the port associated with this router and on the external net |
| 356 | # 4. Set up route to the subnet through the port from step 3 |
| 357 | ip_address = None |
| 358 | for port in ports: |
| 359 | for fixed_ip in port['fixed_ips']: |
| 360 | if fixed_ip['subnet_id'] == subnet_id and fixed_ip['ip_address'] == gw_ip: |
| 361 | gw_port = port |
| 362 | router_id = gw_port['device_id'] |
| 363 | router = self.shell.quantum.show_router(router_id)['router'] |
| 364 | ext_net = router['external_gateway_info']['network_id'] |
| 365 | for port in ports: |
| 366 | if port['device_id'] == router_id and port['network_id'] == ext_net: |
| 367 | ip_address = port['fixed_ips'][0]['ip_address'] |
| 368 | |
| 369 | if ip_address: |
| 370 | cmd = "route delete -net %s" % (subnet['cidr']) |
| 371 | commands.getstatusoutput(cmd) |
| 372 | |
| 373 | return 1 |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 374 | |
Tony Mack | b0d9742 | 2013-06-10 09:57:45 -0400 | [diff] [blame] | 375 | def create_keypair(self, name, public_key): |
Tony Mack | c2a1c7c | 2013-04-09 13:06:34 -0400 | [diff] [blame] | 376 | keys = self.shell.nova.keypairs.findall(name=name) |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 377 | if keys: |
Tony Mack | 52cb930 | 2013-04-09 12:25:35 -0400 | [diff] [blame] | 378 | key = keys[0] |
Tony Mack | b0d9742 | 2013-06-10 09:57:45 -0400 | [diff] [blame] | 379 | # update key |
| 380 | if key.public_key != public_key: |
| 381 | self.delete_keypair(key.id) |
| 382 | key = self.shell.nova.keypairs.create(name=name, public_key=public_key) |
Tony Mack | 52cb930 | 2013-04-09 12:25:35 -0400 | [diff] [blame] | 383 | else: |
Tony Mack | b0d9742 | 2013-06-10 09:57:45 -0400 | [diff] [blame] | 384 | key = self.shell.nova.keypairs.create(name=name, public_key=public_key) |
Tony Mack | 52cb930 | 2013-04-09 12:25:35 -0400 | [diff] [blame] | 385 | return key |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 386 | |
Tony Mack | 759b57a | 2013-04-14 21:03:31 -0400 | [diff] [blame] | 387 | def delete_keypair(self, id): |
| 388 | keys = self.shell.nova.keypairs.findall(id=id) |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 389 | for key in keys: |
Tony Mack | c2a1c7c | 2013-04-09 13:06:34 -0400 | [diff] [blame] | 390 | self.shell.nova.keypairs.delete(key) |
Scott Baker | 8a6d512 | 2013-08-07 18:57:15 -0700 | [diff] [blame] | 391 | return 1 |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 392 | |
Scott Baker | 8a6d512 | 2013-08-07 18:57:15 -0700 | [diff] [blame] | 393 | def get_private_networks(self, tenant=None): |
| 394 | if not tenant: |
| 395 | tenant = self.shell.nova.tenant |
| 396 | tenant = self.shell.keystone.tenants.find(name=tenant) |
| 397 | search_opts = {"tenant_id": tenant.id, "shared": False} |
| 398 | private_networks = self.shell.quantum.list_networks(**search_opts) |
| 399 | return private_networks |
| 400 | |
| 401 | def get_shared_networks(self): |
| 402 | search_opts = {"shared": True} |
| 403 | shared_networks = self.shell.quantum.list_networks(**search_opts) |
| 404 | return shared_networks |
| 405 | |
Scott Baker | 0d5ea5c | 2013-08-09 14:49:56 -0700 | [diff] [blame] | 406 | def get_network_subnet(self, network_id): |
| 407 | subnet_id = None |
| 408 | subnet = None |
| 409 | if network_id: |
| 410 | os_networks = self.shell.quantum.list_networks(id=network_id)["networks"] |
| 411 | if os_networks: |
| 412 | os_network = os_networks[0] |
| 413 | if os_network['subnets']: |
| 414 | subnet_id = os_network['subnets'][0] |
| 415 | os_subnets = self.shell.quantum.list_subnets(id=subnet_id)['subnets'] |
| 416 | if os_subnets: |
| 417 | subnet = os_subnets[0]['cidr'] |
| 418 | |
| 419 | return (subnet_id, subnet) |
| 420 | |
Tony Mack | 046e3f0 | 2014-09-29 16:21:57 -0400 | [diff] [blame] | 421 | def spawn_instance(self, name, key_name=None, availability_zone=None, hostname=None, image_id=None, security_group=None, pubkeys=[], nics=None, metadata=None, userdata=None, flavor_name=None): |
Scott Baker | c132d09 | 2014-09-04 16:55:08 -0700 | [diff] [blame] | 422 | if not flavor_name: |
| 423 | flavor_name = self.config.nova_default_flavor |
| 424 | |
Tony Mack | e7e304f | 2013-04-30 18:18:03 -0400 | [diff] [blame] | 425 | flavor = self.shell.nova.flavors.find(name=flavor_name) |
Scott Baker | c132d09 | 2014-09-04 16:55:08 -0700 | [diff] [blame] | 426 | |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 427 | if not security_group: |
Scott Baker | 3a0e66b | 2013-09-12 11:21:34 -0700 | [diff] [blame] | 428 | security_group = self.config.nova_default_security_group |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 429 | |
Tony Mack | 8538b44 | 2014-04-16 07:49:41 -0400 | [diff] [blame] | 430 | files = {} |
Tony Mack | 38c158b | 2014-04-16 20:15:55 -0400 | [diff] [blame] | 431 | #if pubkeys: |
| 432 | # files["/root/.ssh/authorized_keys"] = "\n".join(pubkeys).encode('base64') |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 433 | hints = {} |
Tony Mack | 046e3f0 | 2014-09-29 16:21:57 -0400 | [diff] [blame] | 434 | |
| 435 | # determine availability zone and compute host |
| 436 | availability_zone_filter = None |
Tony Mack | 3023ba3 | 2014-11-14 00:56:04 -0500 | [diff] [blame] | 437 | if availability_zone is None or not availability_zone: |
Tony Mack | 046e3f0 | 2014-09-29 16:21:57 -0400 | [diff] [blame] | 438 | availability_zone_filter = 'nova' |
| 439 | else: |
| 440 | availability_zone_filter = availability_zone |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 441 | if hostname: |
Tony Mack | 3023ba3 | 2014-11-14 00:56:04 -0500 | [diff] [blame] | 442 | availability_zone_filter += ':%s' % hostname |
Tony Mack | 046e3f0 | 2014-09-29 16:21:57 -0400 | [diff] [blame] | 443 | |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 444 | server = self.shell.nova.servers.create( |
| 445 | name=name, |
| 446 | key_name = key_name, |
Tony Mack | e7e304f | 2013-04-30 18:18:03 -0400 | [diff] [blame] | 447 | flavor=flavor.id, |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 448 | image=image_id, |
| 449 | security_group = security_group, |
Tony Mack | 38c158b | 2014-04-16 20:15:55 -0400 | [diff] [blame] | 450 | #files = files, |
Tony Mack | 79b28ca | 2013-04-15 21:47:45 -0400 | [diff] [blame] | 451 | scheduler_hints=hints, |
Tony Mack | 046e3f0 | 2014-09-29 16:21:57 -0400 | [diff] [blame] | 452 | availability_zone=availability_zone_filter, |
Scott Baker | 3a0e66b | 2013-09-12 11:21:34 -0700 | [diff] [blame] | 453 | nics=nics, |
Tony Mack | 6719811 | 2014-04-09 01:46:03 -0400 | [diff] [blame] | 454 | networks=nics, |
Scott Baker | 9c7a1e1 | 2014-05-30 14:42:42 -0700 | [diff] [blame] | 455 | meta=metadata, |
| 456 | userdata=userdata) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 457 | return server |
Scott Baker | 6a50b02 | 2013-08-26 08:25:37 -0700 | [diff] [blame] | 458 | |
Tony Mack | cdec090 | 2013-04-15 00:38:49 -0400 | [diff] [blame] | 459 | def destroy_instance(self, id): |
Scott Baker | 6a50b02 | 2013-08-26 08:25:37 -0700 | [diff] [blame] | 460 | if (self.shell.nova.tenant=="admin"): |
| 461 | # findall() is implemented as a list() followed by a python search of the |
| 462 | # list. Since findall() doesn't accept "all_tenants", we do this using |
| 463 | # list() ourselves. This allows us to delete an instance as admin. |
| 464 | servers = self.shell.nova.servers.list(search_opts={"all_tenants": True}) |
| 465 | else: |
| 466 | servers = self.shell.nova.servers.list() |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 467 | for server in servers: |
Scott Baker | 6a50b02 | 2013-08-26 08:25:37 -0700 | [diff] [blame] | 468 | if server.id == id: |
| 469 | result=self.shell.nova.servers.delete(server) |
smbaker | 6ada7bf | 2013-04-30 18:15:48 -0700 | [diff] [blame] | 470 | |
| 471 | def update_instance_metadata(self, id, metadata): |
| 472 | servers = self.shell.nova.servers.findall(id=id) |
| 473 | for server in servers: |
| 474 | self.shell.nova.servers.set_meta(server, metadata) |
| 475 | # note: set_meta() returns a broken Server() object. Don't try to |
| 476 | # print it in the shell or it will fail in __repr__. |
| 477 | |
| 478 | def delete_instance_metadata(self, id, metadata): |
| 479 | # note: metadata is a dict. Only the keys matter, not the values. |
| 480 | servers = self.shell.nova.servers.findall(id=id) |
| 481 | for server in servers: |
| 482 | self.shell.nova.servers.delete_meta(server, metadata) |
| 483 | |