Statistics
| Branch: | Tag: | Revision:

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)