Statistics
| Branch: | Tag: | Revision:

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)