root / snf-cyclades-app / synnefo / api / floating_ips.py @ 326c3ec8
History | View | Annotate | Download (9.5 kB)
1 |
# Copyright 2013 GRNET S.A. All rights reserved.
|
---|---|
2 |
#
|
3 |
# Redistribution and use in source and binary forms, with or
|
4 |
# without modification, are permitted provided that the following
|
5 |
# conditions are met:
|
6 |
#
|
7 |
# 1. Redistributions of source code must retain the above
|
8 |
# copyright notice, this list of conditions and the following
|
9 |
# disclaimer.
|
10 |
#
|
11 |
# 2. Redistributions in binary form must reproduce the above
|
12 |
# copyright notice, this list of conditions and the following
|
13 |
# disclaimer in the documentation and/or other materials
|
14 |
# provided with the distribution.
|
15 |
#
|
16 |
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
|
17 |
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
18 |
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
19 |
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
|
20 |
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
21 |
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
22 |
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
|
23 |
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
|
24 |
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
25 |
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
|
26 |
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
27 |
# POSSIBILITY OF SUCH DAMAGE.
|
28 |
#
|
29 |
# The views and conclusions contained in the software and
|
30 |
# documentation are those of the authors and should not be
|
31 |
# interpreted as representing official policies, either expressed
|
32 |
# or implied, of GRNET S.A.
|
33 |
|
34 |
from django.conf.urls.defaults import patterns |
35 |
from django.db import transaction |
36 |
from django.http import HttpResponse |
37 |
from django.utils import simplejson as json |
38 |
|
39 |
from snf_django.lib import api |
40 |
from snf_django.lib.api import faults, utils |
41 |
from synnefo.api import util |
42 |
from synnefo import quotas |
43 |
from synnefo.db.models import Network, IPAddress, NetworkInterface |
44 |
|
45 |
|
46 |
from logging import getLogger |
47 |
log = getLogger(__name__) |
48 |
|
49 |
ips_urlpatterns = patterns( |
50 |
'synnefo.api.floating_ips',
|
51 |
(r'^(?:/|.json|.xml)?$', 'demux'), |
52 |
(r'^/(\w+)(?:.json|.xml)?$', 'floating_ip_demux'), |
53 |
) |
54 |
|
55 |
pools_urlpatterns = patterns( |
56 |
"synnefo.api.floating_ips",
|
57 |
(r'^(?:/|.json|.xml)?$', 'list_floating_ip_pools'), |
58 |
) |
59 |
|
60 |
|
61 |
def demux(request): |
62 |
if request.method == 'GET': |
63 |
return list_floating_ips(request)
|
64 |
elif request.method == 'POST': |
65 |
return allocate_floating_ip(request)
|
66 |
else:
|
67 |
return api.api_method_not_allowed(request)
|
68 |
|
69 |
|
70 |
def floating_ip_demux(request, floating_ip_id): |
71 |
if request.method == 'GET': |
72 |
return get_floating_ip(request, floating_ip_id)
|
73 |
elif request.method == 'DELETE': |
74 |
return release_floating_ip(request, floating_ip_id)
|
75 |
else:
|
76 |
return api.api_method_not_allowed(request)
|
77 |
|
78 |
|
79 |
def ip_to_dict(floating_ip): |
80 |
machine_id = floating_ip.machine_id |
81 |
return {"fixed_ip": None, |
82 |
"id": str(floating_ip.id), |
83 |
"instance_id": str(machine_id) if machine_id else None, |
84 |
"ip": floating_ip.ipv4,
|
85 |
"pool": str(floating_ip.network_id)} |
86 |
|
87 |
|
88 |
@api.api_method(http_method="GET", user_required=True, logger=log, |
89 |
serializations=["json"])
|
90 |
def list_floating_ips(request): |
91 |
"""Return user reserved floating IPs"""
|
92 |
log.debug("list_floating_ips")
|
93 |
|
94 |
userid = request.user_uniq |
95 |
floating_ips = IPAddress.objects.filter(userid=userid).order_by("id")
|
96 |
floating_ips = utils.filter_modified_since(request, objects=floating_ips) |
97 |
|
98 |
floating_ips = map(ip_to_dict, floating_ips)
|
99 |
|
100 |
request.serialization = "json"
|
101 |
data = json.dumps({"floating_ips": floating_ips})
|
102 |
|
103 |
return HttpResponse(data, status=200) |
104 |
|
105 |
|
106 |
@api.api_method(http_method="GET", user_required=True, logger=log, |
107 |
serializations=["json"])
|
108 |
def get_floating_ip(request, floating_ip_id): |
109 |
"""Return information for a floating IP."""
|
110 |
userid = request.user_uniq |
111 |
try:
|
112 |
floating_ip = IPAddress.objects.get(id=floating_ip_id, |
113 |
deleted=False,
|
114 |
userid=userid) |
115 |
except IPAddress.DoesNotExist:
|
116 |
raise faults.ItemNotFound("Floating IP '%s' does not exist" % |
117 |
floating_ip_id) |
118 |
request.serialization = "json"
|
119 |
data = json.dumps({"floating_ip": ip_to_dict(floating_ip)})
|
120 |
return HttpResponse(data, status=200) |
121 |
|
122 |
|
123 |
@api.api_method(http_method='POST', user_required=True, logger=log, |
124 |
serializations=["json"])
|
125 |
@transaction.commit_manually
|
126 |
def allocate_floating_ip(request): |
127 |
"""Allocate a floating IP."""
|
128 |
req = utils.get_request_dict(request) |
129 |
log.info('allocate_floating_ip %s', req)
|
130 |
|
131 |
userid = request.user_uniq |
132 |
pool = req.get("pool", None) |
133 |
address = req.get("address", None) |
134 |
machine = None
|
135 |
net_objects = Network.objects.select_for_update()\ |
136 |
.filter(public=True, floating_ip_pool=True, |
137 |
deleted=False)
|
138 |
try:
|
139 |
if pool is None: |
140 |
# User did not specified a pool. Choose a random public IP
|
141 |
network, address = util.get_free_ip(net_objects) |
142 |
else:
|
143 |
try:
|
144 |
network_id = int(pool)
|
145 |
except ValueError: |
146 |
raise faults.BadRequest("Invalid pool ID.") |
147 |
network = next((n for n in net_objects if n.id == network_id), |
148 |
None)
|
149 |
if network is None: |
150 |
raise faults.ItemNotFound("Pool '%s' does not exist." % pool) |
151 |
if address is None: |
152 |
# User did not specified an IP address. Choose a random one
|
153 |
# Gets X-Lock on IP pool
|
154 |
address = util.get_network_free_address(network) |
155 |
else:
|
156 |
# User specified an IP address. Check that it is not a used
|
157 |
# floating IP
|
158 |
if IPAddress.objects.filter(network=network,
|
159 |
deleted=False,
|
160 |
ipv4=address).exists(): |
161 |
msg = "Floating IP '%s' is reserved" % address
|
162 |
raise faults.Conflict(msg)
|
163 |
pool = network.get_pool() # Gets X-Lock
|
164 |
# Check address belongs to pool
|
165 |
if not pool.contains(address): |
166 |
raise faults.BadRequest("Invalid address") |
167 |
if pool.is_available(address):
|
168 |
pool.reserve(address) |
169 |
pool.save() |
170 |
# If address is not available, check that it belongs to the
|
171 |
# same user
|
172 |
else:
|
173 |
try:
|
174 |
nic = network.nics.get(ipv4=address, |
175 |
machine__userid=userid) |
176 |
nic.ip_type = "FLOATING"
|
177 |
nic.save() |
178 |
except NetworkInterface.DoesNotExist:
|
179 |
msg = "Address '%s' is already in use" % address
|
180 |
raise faults.Conflict(msg)
|
181 |
floating_ip = IPAddress.objects.create(ipv4=address, network=network, |
182 |
userid=userid, machine=machine) |
183 |
quotas.issue_and_accept_commission(floating_ip) |
184 |
except:
|
185 |
transaction.rollback() |
186 |
raise
|
187 |
else:
|
188 |
transaction.commit() |
189 |
|
190 |
log.info("User '%s' allocated floating IP '%s", userid, floating_ip)
|
191 |
|
192 |
request.serialization = "json"
|
193 |
data = json.dumps({"floating_ip": ip_to_dict(floating_ip)})
|
194 |
return HttpResponse(data, status=200) |
195 |
|
196 |
|
197 |
@api.api_method(http_method='DELETE', user_required=True, logger=log, |
198 |
serializations=["json"])
|
199 |
@transaction.commit_on_success
|
200 |
def release_floating_ip(request, floating_ip_id): |
201 |
"""Release a floating IP."""
|
202 |
userid = request.user_uniq |
203 |
log.info("release_floating_ip '%s'. User '%s'.", floating_ip_id, userid)
|
204 |
try:
|
205 |
floating_ip = IPAddress.objects.select_for_update()\ |
206 |
.get(id=floating_ip_id, |
207 |
deleted=False,
|
208 |
userid=userid) |
209 |
except IPAddress.DoesNotExist:
|
210 |
raise faults.ItemNotFound("Floating IP '%s' does not exist" % |
211 |
floating_ip_id) |
212 |
|
213 |
# Since we have got an exlusively lock in floating IP, and since
|
214 |
# to remove a floating IP you need the same lock, the in_use() query
|
215 |
# is safe
|
216 |
if floating_ip.in_use():
|
217 |
msg = "Floating IP '%s' is used" % floating_ip.id
|
218 |
raise faults.Conflict(message=msg)
|
219 |
|
220 |
try:
|
221 |
floating_ip.network.release_address(floating_ip.ipv4) |
222 |
floating_ip.deleted = True
|
223 |
quotas.issue_and_accept_commission(floating_ip, delete=True)
|
224 |
except:
|
225 |
transaction.rollback() |
226 |
raise
|
227 |
else:
|
228 |
floating_ip.delete() |
229 |
transaction.commit() |
230 |
|
231 |
log.info("User '%s' released IP '%s", userid, floating_ip)
|
232 |
|
233 |
return HttpResponse(status=204) |
234 |
|
235 |
|
236 |
def network_to_pool(network): |
237 |
pool = network.get_pool(with_lock=False)
|
238 |
return {"name": str(network.id), |
239 |
"size": pool.pool_size,
|
240 |
"free": pool.count_available()}
|
241 |
|
242 |
|
243 |
@api.api_method(http_method='GET', user_required=True, logger=log, |
244 |
serializations=["json"])
|
245 |
def list_floating_ip_pools(request): |
246 |
networks = Network.objects.filter(public=True, floating_ip_pool=True) |
247 |
networks = utils.filter_modified_since(request, objects=networks) |
248 |
pools = map(network_to_pool, networks)
|
249 |
request.serialization = "json"
|
250 |
data = json.dumps({"floating_ip_pools": pools})
|
251 |
request.serialization = "json"
|
252 |
return HttpResponse(data, status=200) |