root / snf-cyclades-app / synnefo / api / floating_ips.py @ 9446e7e5
History | View | Annotate | Download (9 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, FloatingIP |
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.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.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 = FloatingIP.objects.filter(userid=userid, deleted=False)\
|
96 |
.order_by("id")
|
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 = FloatingIP.objects.get(id=floating_ip_id, |
113 |
deleted=False,
|
114 |
userid=userid) |
115 |
except FloatingIP.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 ValueErrorx:
|
146 |
raise faults.BadRequest("Invalid pool ID.") |
147 |
network = next((n for n in net_objects if n.id==pool), None) |
148 |
if network is None: |
149 |
raise faults.ItemNotFound("Pool '%s' does not exist." % pool) |
150 |
if address is None: |
151 |
# User did not specified an IP address. Choose a random one
|
152 |
# Gets X-Lock on IP pool
|
153 |
address = util.get_network_free_address(network) |
154 |
else:
|
155 |
# User specified an IP address. Check that it is not a used
|
156 |
# floating IP
|
157 |
if FloatingIP.objects.filter(network=network,
|
158 |
deleted=False,
|
159 |
ipv4=address).exists(): |
160 |
msg = "Floating IP '%s' is reserved" % address
|
161 |
raise faults.Conflict(msg)
|
162 |
pool = network.get_pool() # Gets X-Lock
|
163 |
# Check address belongs to pool
|
164 |
if not pool.contains(address): |
165 |
raise faults.BadRequest("Invalid address") |
166 |
if pool.is_available(address):
|
167 |
pool.reserve(address) |
168 |
pool.save() |
169 |
# If address is not available, check that it belongs to the
|
170 |
# same user
|
171 |
elif not network.nics.filter(ipv4=address, |
172 |
machine__userid=userid).exists(): |
173 |
msg = "Address '%s' is already in use" % address
|
174 |
raise faults.Conflict(msg)
|
175 |
floating_ip = FloatingIP.objects.create(ipv4=address, network=network, |
176 |
userid=userid, machine=machine) |
177 |
quotas.issue_and_accept_commission(floating_ip) |
178 |
except:
|
179 |
transaction.rollback() |
180 |
raise
|
181 |
else:
|
182 |
transaction.commit() |
183 |
|
184 |
log.info("User '%s' allocated floating IP '%s", userid, floating_ip)
|
185 |
|
186 |
request.serialization = "json"
|
187 |
data = json.dumps({"floating_ip": ip_to_dict(floating_ip)})
|
188 |
return HttpResponse(data, status=200) |
189 |
|
190 |
|
191 |
@api.api_method(http_method='DELETE', user_required=True, logger=log, |
192 |
serializations=["json"])
|
193 |
@transaction.commit_on_success
|
194 |
def release_floating_ip(request, floating_ip_id): |
195 |
"""Release a floating IP."""
|
196 |
userid = request.user_uniq |
197 |
log.info("release_floating_ip '%s'. User '%s'.", floating_ip_id, userid)
|
198 |
try:
|
199 |
floating_ip = FloatingIP.objects.select_for_update()\ |
200 |
.get(id=floating_ip_id, |
201 |
deleted=False,
|
202 |
userid=userid) |
203 |
except FloatingIP.DoesNotExist:
|
204 |
raise faults.ItemNotFound("Floating IP '%s' does not exist" % |
205 |
floating_ip_id) |
206 |
|
207 |
# Since we have got an exlusively lock in floating IP, and since
|
208 |
# to remove a floating IP you need the same lock, the in_use() query
|
209 |
# is safe
|
210 |
if floating_ip.in_use():
|
211 |
msg = "Floating IP '%s' is used" % floating_ip.id
|
212 |
raise faults.Conflict(message=msg)
|
213 |
|
214 |
try:
|
215 |
floating_ip.network.release_address(floating_ip.ipv4) |
216 |
floating_ip.deleted = True
|
217 |
quotas.issue_and_accept_commission(floating_ip, delete=True)
|
218 |
except:
|
219 |
transaction.rollback() |
220 |
raise
|
221 |
else:
|
222 |
floating_ip.delete() |
223 |
transaction.commit() |
224 |
|
225 |
log.info("User '%s' released IP '%s", userid, floating_ip)
|
226 |
|
227 |
return HttpResponse(status=204) |
228 |
|
229 |
|
230 |
@api.api_method(http_method='GET', user_required=True, logger=log, |
231 |
serializations=["json"])
|
232 |
def list_floating_ip_pools(request): |
233 |
networks = Network.objects.filter(public=True, deleted=False, |
234 |
floating_ip_pool=True)
|
235 |
pools = [{"name": str(net.id)} for net in networks] |
236 |
request.serialization = "json"
|
237 |
data = json.dumps({"floating_ip_pools": pools})
|
238 |
request.serialization = "json"
|
239 |
return HttpResponse(data, status=200) |