Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / logic / servers.py @ 14402edc

History | View | Annotate | Download (28.5 kB)

1 91884d63 Giorgos Korfiatis
# Copyright 2011, 2012, 2013 GRNET S.A. All rights reserved.
2 91884d63 Giorgos Korfiatis
#
3 91884d63 Giorgos Korfiatis
# Redistribution and use in source and binary forms, with or without
4 91884d63 Giorgos Korfiatis
# modification, are permitted provided that the following conditions
5 91884d63 Giorgos Korfiatis
# are met:
6 91884d63 Giorgos Korfiatis
#
7 91884d63 Giorgos Korfiatis
#   1. Redistributions of source code must retain the above copyright
8 91884d63 Giorgos Korfiatis
#      notice, this list of conditions and the following disclaimer.
9 91884d63 Giorgos Korfiatis
#
10 91884d63 Giorgos Korfiatis
#  2. Redistributions in binary form must reproduce the above copyright
11 91884d63 Giorgos Korfiatis
#     notice, this list of conditions and the following disclaimer in the
12 91884d63 Giorgos Korfiatis
#     documentation and/or other materials provided with the distribution.
13 91884d63 Giorgos Korfiatis
#
14 91884d63 Giorgos Korfiatis
# THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
15 91884d63 Giorgos Korfiatis
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 91884d63 Giorgos Korfiatis
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 91884d63 Giorgos Korfiatis
# ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
18 91884d63 Giorgos Korfiatis
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 91884d63 Giorgos Korfiatis
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 91884d63 Giorgos Korfiatis
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 91884d63 Giorgos Korfiatis
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 91884d63 Giorgos Korfiatis
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 91884d63 Giorgos Korfiatis
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 91884d63 Giorgos Korfiatis
# SUCH DAMAGE.
25 91884d63 Giorgos Korfiatis
#
26 91884d63 Giorgos Korfiatis
# The views and conclusions contained in the software and documentation are
27 91884d63 Giorgos Korfiatis
# those of the authors and should not be interpreted as representing official
28 91884d63 Giorgos Korfiatis
# policies, either expressed or implied, of GRNET S.A.
29 91884d63 Giorgos Korfiatis
30 41a7fae7 Christos Stavrakakis
import logging
31 41a7fae7 Christos Stavrakakis
32 41a7fae7 Christos Stavrakakis
from socket import getfqdn
33 41a7fae7 Christos Stavrakakis
from functools import wraps
34 41a7fae7 Christos Stavrakakis
from django import dispatch
35 41a7fae7 Christos Stavrakakis
from django.db import transaction
36 41a7fae7 Christos Stavrakakis
from django.utils import simplejson as json
37 41a7fae7 Christos Stavrakakis
38 41a7fae7 Christos Stavrakakis
from snf_django.lib.api import faults
39 0c09b1c0 Christos Stavrakakis
from django.conf import settings
40 41a7fae7 Christos Stavrakakis
from synnefo import quotas
41 41a7fae7 Christos Stavrakakis
from synnefo.api import util
42 ba6ad346 Dionysis Grigoropoulos
from synnefo.logic import backend, ips, utils
43 41a7fae7 Christos Stavrakakis
from synnefo.logic.backend_allocator import BackendAllocator
44 710b1c43 Christos Stavrakakis
from synnefo.db.models import (NetworkInterface, VirtualMachine,
45 3aecadc8 Christos Stavrakakis
                               VirtualMachineMetadata, IPAddressLog, Network)
46 41a7fae7 Christos Stavrakakis
from vncauthproxy.client import request_forwarding as request_vnc_forwarding
47 af1832fe Christos Stavrakakis
from synnefo.logic import rapi
48 41a7fae7 Christos Stavrakakis
49 41a7fae7 Christos Stavrakakis
log = logging.getLogger(__name__)
50 41a7fae7 Christos Stavrakakis
51 41a7fae7 Christos Stavrakakis
# server creation signal
52 41a7fae7 Christos Stavrakakis
server_created = dispatch.Signal(providing_args=["created_vm_params"])
53 41a7fae7 Christos Stavrakakis
54 41a7fae7 Christos Stavrakakis
55 41a7fae7 Christos Stavrakakis
def validate_server_action(vm, action):
56 41a7fae7 Christos Stavrakakis
    if vm.deleted:
57 41a7fae7 Christos Stavrakakis
        raise faults.BadRequest("Server '%s' has been deleted." % vm.id)
58 41a7fae7 Christos Stavrakakis
59 41a7fae7 Christos Stavrakakis
    # Destroyin a server should always be permitted
60 41a7fae7 Christos Stavrakakis
    if action == "DESTROY":
61 41a7fae7 Christos Stavrakakis
        return
62 41a7fae7 Christos Stavrakakis
63 41a7fae7 Christos Stavrakakis
    # Check that there is no pending action
64 41a7fae7 Christos Stavrakakis
    pending_action = vm.task
65 41a7fae7 Christos Stavrakakis
    if pending_action:
66 41a7fae7 Christos Stavrakakis
        if pending_action == "BUILD":
67 41a7fae7 Christos Stavrakakis
            raise faults.BuildInProgress("Server '%s' is being build." % vm.id)
68 8d5795b4 Christos Stavrakakis
        raise faults.BadRequest("Cannot perform '%s' action while there is a"
69 41a7fae7 Christos Stavrakakis
                                " pending '%s'." % (action, pending_action))
70 41a7fae7 Christos Stavrakakis
71 41a7fae7 Christos Stavrakakis
    # Check if action can be performed to VM's operstate
72 41a7fae7 Christos Stavrakakis
    operstate = vm.operstate
73 ee688a71 Christos Stavrakakis
    if operstate == "ERROR":
74 8d5795b4 Christos Stavrakakis
        raise faults.BadRequest("Cannot perform '%s' action while server is"
75 ee688a71 Christos Stavrakakis
                                " in 'ERROR' state." % action)
76 ee688a71 Christos Stavrakakis
    elif operstate == "BUILD" and action != "BUILD":
77 41a7fae7 Christos Stavrakakis
        raise faults.BuildInProgress("Server '%s' is being build." % vm.id)
78 9599e997 Christos Stavrakakis
    elif (action == "START" and operstate != "STOPPED") or\
79 9599e997 Christos Stavrakakis
         (action == "STOP" and operstate != "STARTED") or\
80 9599e997 Christos Stavrakakis
         (action == "RESIZE" and operstate != "STOPPED") or\
81 9599e997 Christos Stavrakakis
         (action in ["CONNECT", "DISCONNECT"] and operstate != "STOPPED"
82 9599e997 Christos Stavrakakis
          and not settings.GANETI_USE_HOTPLUG):
83 8d5795b4 Christos Stavrakakis
        raise faults.BadRequest("Cannot perform '%s' action while server is"
84 41a7fae7 Christos Stavrakakis
                                " in '%s' state." % (action, operstate))
85 41a7fae7 Christos Stavrakakis
    return
86 41a7fae7 Christos Stavrakakis
87 41a7fae7 Christos Stavrakakis
88 64bca363 Giorgos Korfiatis
def server_command(action, action_fields=None):
89 41a7fae7 Christos Stavrakakis
    """Handle execution of a server action.
90 41a7fae7 Christos Stavrakakis

91 41a7fae7 Christos Stavrakakis
    Helper function to validate and execute a server action, handle quota
92 41a7fae7 Christos Stavrakakis
    commission and update the 'task' of the VM in the DB.
93 41a7fae7 Christos Stavrakakis

94 41a7fae7 Christos Stavrakakis
    1) Check if action can be performed. If it can, then there must be no
95 41a7fae7 Christos Stavrakakis
       pending task (with the exception of DESTROY).
96 41a7fae7 Christos Stavrakakis
    2) Handle previous commission if unresolved:
97 41a7fae7 Christos Stavrakakis
       * If it is not pending and it to accept, then accept
98 41a7fae7 Christos Stavrakakis
       * If it is not pending and to reject or is pending then reject it. Since
99 41a7fae7 Christos Stavrakakis
       the action can be performed only if there is no pending task, then there
100 41a7fae7 Christos Stavrakakis
       can be no pending commission. The exception is DESTROY, but in this case
101 41a7fae7 Christos Stavrakakis
       the commission can safely be rejected, and the dispatcher will generate
102 41a7fae7 Christos Stavrakakis
       the correct ones!
103 41a7fae7 Christos Stavrakakis
    3) Issue new commission and associate it with the VM. Also clear the task.
104 41a7fae7 Christos Stavrakakis
    4) Send job to ganeti
105 41a7fae7 Christos Stavrakakis
    5) Update task and commit
106 41a7fae7 Christos Stavrakakis
    """
107 41a7fae7 Christos Stavrakakis
    def decorator(func):
108 41a7fae7 Christos Stavrakakis
        @wraps(func)
109 41a7fae7 Christos Stavrakakis
        @transaction.commit_on_success
110 41a7fae7 Christos Stavrakakis
        def wrapper(vm, *args, **kwargs):
111 41a7fae7 Christos Stavrakakis
            user_id = vm.userid
112 41a7fae7 Christos Stavrakakis
            validate_server_action(vm, action)
113 c9fefba3 Christos Stavrakakis
            vm.action = action
114 41a7fae7 Christos Stavrakakis
115 5c8076b6 Christos Stavrakakis
            commission_name = "client: api, resource: %s" % vm
116 5c8076b6 Christos Stavrakakis
            quotas.handle_resource_commission(vm, action=action,
117 64bca363 Giorgos Korfiatis
                                              action_fields=action_fields,
118 5c8076b6 Christos Stavrakakis
                                              commission_name=commission_name)
119 5c8076b6 Christos Stavrakakis
            vm.save()
120 41a7fae7 Christos Stavrakakis
121 562bf712 Christos Stavrakakis
            # XXX: Special case for server creation!
122 562bf712 Christos Stavrakakis
            if action == "BUILD":
123 562bf712 Christos Stavrakakis
                # Perform a commit, because the VirtualMachine must be saved to
124 562bf712 Christos Stavrakakis
                # DB before the OP_INSTANCE_CREATE job in enqueued in Ganeti.
125 562bf712 Christos Stavrakakis
                # Otherwise, messages will arrive from snf-dispatcher about
126 562bf712 Christos Stavrakakis
                # this instance, before the VM is stored in DB.
127 562bf712 Christos Stavrakakis
                transaction.commit()
128 562bf712 Christos Stavrakakis
                # After committing the locks are released. Refetch the instance
129 562bf712 Christos Stavrakakis
                # to guarantee x-lock.
130 562bf712 Christos Stavrakakis
                vm = VirtualMachine.objects.select_for_update().get(id=vm.id)
131 562bf712 Christos Stavrakakis
132 41a7fae7 Christos Stavrakakis
            # Send the job to Ganeti and get the associated jobID
133 41a7fae7 Christos Stavrakakis
            try:
134 41a7fae7 Christos Stavrakakis
                job_id = func(vm, *args, **kwargs)
135 41a7fae7 Christos Stavrakakis
            except Exception as e:
136 41a7fae7 Christos Stavrakakis
                if vm.serial is not None:
137 41a7fae7 Christos Stavrakakis
                    # Since the job never reached Ganeti, reject the commission
138 41a7fae7 Christos Stavrakakis
                    log.debug("Rejecting commission: '%s', could not perform"
139 41a7fae7 Christos Stavrakakis
                              " action '%s': %s" % (vm.serial,  action, e))
140 41a7fae7 Christos Stavrakakis
                    transaction.rollback()
141 ba777b02 Giorgos Korfiatis
                    quotas.reject_resource_serial(vm)
142 41a7fae7 Christos Stavrakakis
                    transaction.commit()
143 41a7fae7 Christos Stavrakakis
                raise
144 41a7fae7 Christos Stavrakakis
145 88fd91af Christos Stavrakakis
            if action == "BUILD" and vm.serial is not None:
146 88fd91af Christos Stavrakakis
                # XXX: Special case for server creation: we must accept the
147 88fd91af Christos Stavrakakis
                # commission because the VM has been stored in DB. Also, if
148 88fd91af Christos Stavrakakis
                # communication with Ganeti fails, the job will never reach
149 88fd91af Christos Stavrakakis
                # Ganeti, and the commission will never be resolved.
150 ba777b02 Giorgos Korfiatis
                quotas.accept_resource_serial(vm)
151 88fd91af Christos Stavrakakis
152 41a7fae7 Christos Stavrakakis
            log.info("user: %s, vm: %s, action: %s, job_id: %s, serial: %s",
153 41a7fae7 Christos Stavrakakis
                     user_id, vm.id, action, job_id, vm.serial)
154 41a7fae7 Christos Stavrakakis
155 41a7fae7 Christos Stavrakakis
            # store the new task in the VM
156 41a7fae7 Christos Stavrakakis
            if job_id is not None:
157 41a7fae7 Christos Stavrakakis
                vm.task = action
158 41a7fae7 Christos Stavrakakis
                vm.task_job_id = job_id
159 41a7fae7 Christos Stavrakakis
            vm.save()
160 41a7fae7 Christos Stavrakakis
161 41a7fae7 Christos Stavrakakis
            return vm
162 41a7fae7 Christos Stavrakakis
        return wrapper
163 41a7fae7 Christos Stavrakakis
    return decorator
164 41a7fae7 Christos Stavrakakis
165 41a7fae7 Christos Stavrakakis
166 562bf712 Christos Stavrakakis
@transaction.commit_on_success
167 41a7fae7 Christos Stavrakakis
def create(userid, name, password, flavor, image, metadata={},
168 3aecadc8 Christos Stavrakakis
           personality=[], networks=None, use_backend=None):
169 41a7fae7 Christos Stavrakakis
    if use_backend is None:
170 562bf712 Christos Stavrakakis
        # Allocate server to a Ganeti backend
171 562bf712 Christos Stavrakakis
        use_backend = allocate_new_server(userid, flavor)
172 41a7fae7 Christos Stavrakakis
173 ba6ad346 Dionysis Grigoropoulos
    utils.check_name_length(name, VirtualMachine.VIRTUAL_MACHINE_NAME_LENGTH,
174 ba6ad346 Dionysis Grigoropoulos
                            "Server name is too long")
175 ba6ad346 Dionysis Grigoropoulos
176 3aecadc8 Christos Stavrakakis
    # Create the ports for the server
177 16a7ced5 Christos Stavrakakis
    ports = create_instance_ports(userid, networks)
178 6193628f Christos Stavrakakis
179 41a7fae7 Christos Stavrakakis
    # Fix flavor for archipelago
180 41a7fae7 Christos Stavrakakis
    disk_template, provider = util.get_flavor_provider(flavor)
181 41a7fae7 Christos Stavrakakis
    if provider:
182 41a7fae7 Christos Stavrakakis
        flavor.disk_template = disk_template
183 41a7fae7 Christos Stavrakakis
        flavor.disk_provider = provider
184 41a7fae7 Christos Stavrakakis
        flavor.disk_origin = None
185 62232bba Christos Stavrakakis
        if provider in settings.GANETI_CLONE_PROVIDERS:
186 41a7fae7 Christos Stavrakakis
            flavor.disk_origin = image['checksum']
187 41a7fae7 Christos Stavrakakis
            image['backend_id'] = 'null'
188 41a7fae7 Christos Stavrakakis
    else:
189 41a7fae7 Christos Stavrakakis
        flavor.disk_provider = None
190 41a7fae7 Christos Stavrakakis
191 562bf712 Christos Stavrakakis
    # We must save the VM instance now, so that it gets a valid
192 562bf712 Christos Stavrakakis
    # vm.backend_vm_id.
193 562bf712 Christos Stavrakakis
    vm = VirtualMachine.objects.create(name=name,
194 562bf712 Christos Stavrakakis
                                       backend=use_backend,
195 562bf712 Christos Stavrakakis
                                       userid=userid,
196 562bf712 Christos Stavrakakis
                                       imageid=image["id"],
197 562bf712 Christos Stavrakakis
                                       flavor=flavor,
198 562bf712 Christos Stavrakakis
                                       operstate="BUILD")
199 562bf712 Christos Stavrakakis
    log.info("Created entry in DB for VM '%s'", vm)
200 562bf712 Christos Stavrakakis
201 3aecadc8 Christos Stavrakakis
    # Associate the ports with the server
202 3aecadc8 Christos Stavrakakis
    for index, port in enumerate(ports):
203 3aecadc8 Christos Stavrakakis
        associate_port_with_machine(port, vm)
204 3aecadc8 Christos Stavrakakis
        port.index = index
205 3aecadc8 Christos Stavrakakis
        port.save()
206 562bf712 Christos Stavrakakis
207 562bf712 Christos Stavrakakis
    for key, val in metadata.items():
208 562bf712 Christos Stavrakakis
        VirtualMachineMetadata.objects.create(
209 562bf712 Christos Stavrakakis
            meta_key=key,
210 562bf712 Christos Stavrakakis
            meta_value=val,
211 562bf712 Christos Stavrakakis
            vm=vm)
212 41a7fae7 Christos Stavrakakis
213 88fd91af Christos Stavrakakis
    # Create the server in Ganeti.
214 3aecadc8 Christos Stavrakakis
    vm = create_server(vm, ports, flavor, image, personality, password)
215 41a7fae7 Christos Stavrakakis
216 41a7fae7 Christos Stavrakakis
    return vm
217 41a7fae7 Christos Stavrakakis
218 41a7fae7 Christos Stavrakakis
219 562bf712 Christos Stavrakakis
@transaction.commit_on_success
220 562bf712 Christos Stavrakakis
def allocate_new_server(userid, flavor):
221 562bf712 Christos Stavrakakis
    """Allocate a new server to a Ganeti backend.
222 562bf712 Christos Stavrakakis

223 562bf712 Christos Stavrakakis
    Allocation is performed based on the owner of the server and the specified
224 562bf712 Christos Stavrakakis
    flavor. Also, backends that do not have a public IPv4 address are excluded
225 562bf712 Christos Stavrakakis
    from server allocation.
226 562bf712 Christos Stavrakakis

227 562bf712 Christos Stavrakakis
    This function runs inside a transaction, because after allocating the
228 562bf712 Christos Stavrakakis
    instance a commit must be performed in order to release all locks.
229 562bf712 Christos Stavrakakis

230 562bf712 Christos Stavrakakis
    """
231 562bf712 Christos Stavrakakis
    backend_allocator = BackendAllocator()
232 562bf712 Christos Stavrakakis
    use_backend = backend_allocator.allocate(userid, flavor)
233 562bf712 Christos Stavrakakis
    if use_backend is None:
234 562bf712 Christos Stavrakakis
        log.error("No available backend for VM with flavor %s", flavor)
235 562bf712 Christos Stavrakakis
        raise faults.ServiceUnavailable("No available backends")
236 562bf712 Christos Stavrakakis
    return use_backend
237 562bf712 Christos Stavrakakis
238 562bf712 Christos Stavrakakis
239 562bf712 Christos Stavrakakis
@server_command("BUILD")
240 562bf712 Christos Stavrakakis
def create_server(vm, nics, flavor, image, personality, password):
241 562bf712 Christos Stavrakakis
    # dispatch server created signal needed to trigger the 'vmapi', which
242 562bf712 Christos Stavrakakis
    # enriches the vm object with the 'config_url' attribute which must be
243 562bf712 Christos Stavrakakis
    # passed to the Ganeti job.
244 562bf712 Christos Stavrakakis
    server_created.send(sender=vm, created_vm_params={
245 562bf712 Christos Stavrakakis
        'img_id': image['backend_id'],
246 562bf712 Christos Stavrakakis
        'img_passwd': password,
247 562bf712 Christos Stavrakakis
        'img_format': str(image['format']),
248 562bf712 Christos Stavrakakis
        'img_personality': json.dumps(personality),
249 562bf712 Christos Stavrakakis
        'img_properties': json.dumps(image['metadata']),
250 562bf712 Christos Stavrakakis
    })
251 562bf712 Christos Stavrakakis
    # send job to Ganeti
252 88fd91af Christos Stavrakakis
    try:
253 88fd91af Christos Stavrakakis
        jobID = backend.create_instance(vm, nics, flavor, image)
254 88fd91af Christos Stavrakakis
    except:
255 88fd91af Christos Stavrakakis
        log.exception("Failed create instance '%s'", vm)
256 88fd91af Christos Stavrakakis
        jobID = None
257 88fd91af Christos Stavrakakis
        vm.operstate = "ERROR"
258 88fd91af Christos Stavrakakis
        vm.backendlogmsg = "Failed to send job to Ganeti."
259 88fd91af Christos Stavrakakis
        vm.save()
260 88fd91af Christos Stavrakakis
        vm.nics.all().update(state="ERROR")
261 88fd91af Christos Stavrakakis
262 562bf712 Christos Stavrakakis
    # At this point the job is enqueued in the Ganeti backend
263 80a548e3 Christos Stavrakakis
    vm.backendopcode = "OP_INSTANCE_CREATE"
264 562bf712 Christos Stavrakakis
    vm.backendjobid = jobID
265 562bf712 Christos Stavrakakis
    vm.save()
266 562bf712 Christos Stavrakakis
    log.info("User %s created VM %s, NICs %s, Backend %s, JobID %s",
267 2fa6faca Christos Stavrakakis
             vm.userid, vm, nics, vm.backend, str(jobID))
268 562bf712 Christos Stavrakakis
269 562bf712 Christos Stavrakakis
    return jobID
270 562bf712 Christos Stavrakakis
271 562bf712 Christos Stavrakakis
272 41a7fae7 Christos Stavrakakis
@server_command("DESTROY")
273 c397dbce Christos Stavrakakis
def destroy(vm, shutdown_timeout=None):
274 80a548e3 Christos Stavrakakis
    # XXX: Workaround for race where OP_INSTANCE_REMOVE starts executing on
275 80a548e3 Christos Stavrakakis
    # Ganeti before OP_INSTANCE_CREATE. This will be fixed when
276 80a548e3 Christos Stavrakakis
    # OP_INSTANCE_REMOVE supports the 'depends' request attribute.
277 80a548e3 Christos Stavrakakis
    if (vm.backendopcode == "OP_INSTANCE_CREATE" and
278 af1832fe Christos Stavrakakis
       vm.backendjobstatus not in rapi.JOB_STATUS_FINALIZED and
279 80a548e3 Christos Stavrakakis
       backend.job_is_still_running(vm) and
280 80a548e3 Christos Stavrakakis
       not backend.vm_exists_in_backend(vm)):
281 80a548e3 Christos Stavrakakis
            raise faults.BuildInProgress("Server is being build")
282 41a7fae7 Christos Stavrakakis
    log.info("Deleting VM %s", vm)
283 c915a198 Giorgos Korfiatis
    return backend.delete_instance(vm, shutdown_timeout=shutdown_timeout)
284 41a7fae7 Christos Stavrakakis
285 41a7fae7 Christos Stavrakakis
286 41a7fae7 Christos Stavrakakis
@server_command("START")
287 41a7fae7 Christos Stavrakakis
def start(vm):
288 41a7fae7 Christos Stavrakakis
    log.info("Starting VM %s", vm)
289 41a7fae7 Christos Stavrakakis
    return backend.startup_instance(vm)
290 41a7fae7 Christos Stavrakakis
291 41a7fae7 Christos Stavrakakis
292 41a7fae7 Christos Stavrakakis
@server_command("STOP")
293 c397dbce Christos Stavrakakis
def stop(vm, shutdown_timeout=None):
294 41a7fae7 Christos Stavrakakis
    log.info("Stopping VM %s", vm)
295 c915a198 Giorgos Korfiatis
    return backend.shutdown_instance(vm, shutdown_timeout=shutdown_timeout)
296 41a7fae7 Christos Stavrakakis
297 41a7fae7 Christos Stavrakakis
298 41a7fae7 Christos Stavrakakis
@server_command("REBOOT")
299 c397dbce Christos Stavrakakis
def reboot(vm, reboot_type, shutdown_timeout=None):
300 41a7fae7 Christos Stavrakakis
    if reboot_type not in ("SOFT", "HARD"):
301 41a7fae7 Christos Stavrakakis
        raise faults.BadRequest("Malformed request. Invalid reboot"
302 41a7fae7 Christos Stavrakakis
                                " type %s" % reboot_type)
303 41a7fae7 Christos Stavrakakis
    log.info("Rebooting VM %s. Type %s", vm, reboot_type)
304 41a7fae7 Christos Stavrakakis
305 c397dbce Christos Stavrakakis
    return backend.reboot_instance(vm, reboot_type.lower(),
306 c397dbce Christos Stavrakakis
                                   shutdown_timeout=shutdown_timeout)
307 41a7fae7 Christos Stavrakakis
308 41a7fae7 Christos Stavrakakis
309 41a7fae7 Christos Stavrakakis
def resize(vm, flavor):
310 64bca363 Giorgos Korfiatis
    action_fields = {"beparams": {"vcpus": flavor.cpu,
311 64bca363 Giorgos Korfiatis
                                  "maxmem": flavor.ram}}
312 64bca363 Giorgos Korfiatis
    comm = server_command("RESIZE", action_fields=action_fields)
313 64bca363 Giorgos Korfiatis
    return comm(_resize)(vm, flavor)
314 64bca363 Giorgos Korfiatis
315 64bca363 Giorgos Korfiatis
316 64bca363 Giorgos Korfiatis
def _resize(vm, flavor):
317 41a7fae7 Christos Stavrakakis
    old_flavor = vm.flavor
318 41a7fae7 Christos Stavrakakis
    # User requested the same flavor
319 41a7fae7 Christos Stavrakakis
    if old_flavor.id == flavor.id:
320 41a7fae7 Christos Stavrakakis
        raise faults.BadRequest("Server '%s' flavor is already '%s'."
321 41a7fae7 Christos Stavrakakis
                                % (vm, flavor))
322 41a7fae7 Christos Stavrakakis
    # Check that resize can be performed
323 41a7fae7 Christos Stavrakakis
    if old_flavor.disk != flavor.disk:
324 8d5795b4 Christos Stavrakakis
        raise faults.BadRequest("Cannot resize instance disk.")
325 41a7fae7 Christos Stavrakakis
    if old_flavor.disk_template != flavor.disk_template:
326 8d5795b4 Christos Stavrakakis
        raise faults.BadRequest("Cannot change instance disk template.")
327 41a7fae7 Christos Stavrakakis
328 41a7fae7 Christos Stavrakakis
    log.info("Resizing VM from flavor '%s' to '%s", old_flavor, flavor)
329 41a7fae7 Christos Stavrakakis
    return backend.resize_instance(vm, vcpus=flavor.cpu, memory=flavor.ram)
330 41a7fae7 Christos Stavrakakis
331 41a7fae7 Christos Stavrakakis
332 41a7fae7 Christos Stavrakakis
@server_command("SET_FIREWALL_PROFILE")
333 d0545590 Christos Stavrakakis
def set_firewall_profile(vm, profile, nic):
334 d0545590 Christos Stavrakakis
    log.info("Setting VM %s, NIC %s, firewall %s", vm, nic, profile)
335 41a7fae7 Christos Stavrakakis
336 41a7fae7 Christos Stavrakakis
    if profile not in [x[0] for x in NetworkInterface.FIREWALL_PROFILES]:
337 41a7fae7 Christos Stavrakakis
        raise faults.BadRequest("Unsupported firewall profile")
338 d0545590 Christos Stavrakakis
    backend.set_firewall_profile(vm, profile=profile, nic=nic)
339 41a7fae7 Christos Stavrakakis
    return None
340 41a7fae7 Christos Stavrakakis
341 41a7fae7 Christos Stavrakakis
342 41a7fae7 Christos Stavrakakis
@server_command("CONNECT")
343 6b8dc47c Christos Stavrakakis
def connect(vm, network, port=None):
344 6b8dc47c Christos Stavrakakis
    if port is None:
345 fae6e5f0 Christos Stavrakakis
        port = _create_port(vm.userid, network)
346 fae6e5f0 Christos Stavrakakis
    associate_port_with_machine(port, vm)
347 41a7fae7 Christos Stavrakakis
348 fae6e5f0 Christos Stavrakakis
    log.info("Creating NIC %s with IPv4 Address %s", port, port.ipv4_address)
349 41a7fae7 Christos Stavrakakis
350 fae6e5f0 Christos Stavrakakis
    return backend.connect_to_network(vm, port)
351 41a7fae7 Christos Stavrakakis
352 41a7fae7 Christos Stavrakakis
353 41a7fae7 Christos Stavrakakis
@server_command("DISCONNECT")
354 7c714455 Christos Stavrakakis
def disconnect(vm, nic):
355 7c714455 Christos Stavrakakis
    log.info("Removing NIC %s from VM %s", nic, vm)
356 41a7fae7 Christos Stavrakakis
    return backend.disconnect_from_network(vm, nic)
357 41a7fae7 Christos Stavrakakis
358 41a7fae7 Christos Stavrakakis
359 41a7fae7 Christos Stavrakakis
def console(vm, console_type):
360 41a7fae7 Christos Stavrakakis
    """Arrange for an OOB console of the specified type
361 41a7fae7 Christos Stavrakakis

362 41a7fae7 Christos Stavrakakis
    This method arranges for an OOB console of the specified type.
363 41a7fae7 Christos Stavrakakis
    Only consoles of type "vnc" are supported for now.
364 41a7fae7 Christos Stavrakakis

365 41a7fae7 Christos Stavrakakis
    It uses a running instance of vncauthproxy to setup proper
366 41a7fae7 Christos Stavrakakis
    VNC forwarding with a random password, then returns the necessary
367 41a7fae7 Christos Stavrakakis
    VNC connection info to the caller.
368 41a7fae7 Christos Stavrakakis

369 41a7fae7 Christos Stavrakakis
    """
370 41a7fae7 Christos Stavrakakis
    log.info("Get console  VM %s, type %s", vm, console_type)
371 41a7fae7 Christos Stavrakakis
372 41a7fae7 Christos Stavrakakis
    # Use RAPI to get VNC console information for this instance
373 41a7fae7 Christos Stavrakakis
    if vm.operstate != "STARTED":
374 41a7fae7 Christos Stavrakakis
        raise faults.BadRequest('Server not in ACTIVE state.')
375 41a7fae7 Christos Stavrakakis
376 41a7fae7 Christos Stavrakakis
    if settings.TEST:
377 41a7fae7 Christos Stavrakakis
        console_data = {'kind': 'vnc', 'host': 'ganeti_node', 'port': 1000}
378 41a7fae7 Christos Stavrakakis
    else:
379 41a7fae7 Christos Stavrakakis
        console_data = backend.get_instance_console(vm)
380 41a7fae7 Christos Stavrakakis
381 41a7fae7 Christos Stavrakakis
    if console_data['kind'] != 'vnc':
382 41a7fae7 Christos Stavrakakis
        message = 'got console of kind %s, not "vnc"' % console_data['kind']
383 41a7fae7 Christos Stavrakakis
        raise faults.ServiceUnavailable(message)
384 41a7fae7 Christos Stavrakakis
385 41a7fae7 Christos Stavrakakis
    # Let vncauthproxy decide on the source port.
386 41a7fae7 Christos Stavrakakis
    # The alternative: static allocation, e.g.
387 41a7fae7 Christos Stavrakakis
    # sport = console_data['port'] - 1000
388 41a7fae7 Christos Stavrakakis
    sport = 0
389 41a7fae7 Christos Stavrakakis
    daddr = console_data['host']
390 41a7fae7 Christos Stavrakakis
    dport = console_data['port']
391 41a7fae7 Christos Stavrakakis
    password = util.random_password()
392 41a7fae7 Christos Stavrakakis
393 41a7fae7 Christos Stavrakakis
    if settings.TEST:
394 41a7fae7 Christos Stavrakakis
        fwd = {'source_port': 1234, 'status': 'OK'}
395 41a7fae7 Christos Stavrakakis
    else:
396 f3c5f1df Stratos Psomadakis
        vnc_extra_opts = settings.CYCLADES_VNCAUTHPROXY_OPTS
397 f3c5f1df Stratos Psomadakis
        fwd = request_vnc_forwarding(sport, daddr, dport, password,
398 f3c5f1df Stratos Psomadakis
                                     **vnc_extra_opts)
399 41a7fae7 Christos Stavrakakis
400 41a7fae7 Christos Stavrakakis
    if fwd['status'] != "OK":
401 41a7fae7 Christos Stavrakakis
        raise faults.ServiceUnavailable('vncauthproxy returned error status')
402 41a7fae7 Christos Stavrakakis
403 41a7fae7 Christos Stavrakakis
    # Verify that the VNC server settings haven't changed
404 41a7fae7 Christos Stavrakakis
    if not settings.TEST:
405 41a7fae7 Christos Stavrakakis
        if console_data != backend.get_instance_console(vm):
406 41a7fae7 Christos Stavrakakis
            raise faults.ServiceUnavailable('VNC Server settings changed.')
407 41a7fae7 Christos Stavrakakis
408 41a7fae7 Christos Stavrakakis
    console = {
409 41a7fae7 Christos Stavrakakis
        'type': 'vnc',
410 41a7fae7 Christos Stavrakakis
        'host': getfqdn(),
411 41a7fae7 Christos Stavrakakis
        'port': fwd['source_port'],
412 41a7fae7 Christos Stavrakakis
        'password': password}
413 41a7fae7 Christos Stavrakakis
414 41a7fae7 Christos Stavrakakis
    return console
415 9ba6bb95 Christos Stavrakakis
416 9ba6bb95 Christos Stavrakakis
417 a52cc1b4 Christos Stavrakakis
def rename(server, new_name):
418 a52cc1b4 Christos Stavrakakis
    """Rename a VirtualMachine."""
419 a52cc1b4 Christos Stavrakakis
    old_name = server.name
420 a52cc1b4 Christos Stavrakakis
    server.name = new_name
421 a52cc1b4 Christos Stavrakakis
    server.save()
422 a52cc1b4 Christos Stavrakakis
    log.info("Renamed server '%s' from '%s' to '%s'", server, old_name,
423 a52cc1b4 Christos Stavrakakis
             new_name)
424 a52cc1b4 Christos Stavrakakis
    return server
425 816d7588 Christos Stavrakakis
426 816d7588 Christos Stavrakakis
427 fae6e5f0 Christos Stavrakakis
@transaction.commit_on_success
428 fae6e5f0 Christos Stavrakakis
def create_port(*args, **kwargs):
429 129b94d5 Christos Stavrakakis
    vm = kwargs.get("machine", None)
430 129b94d5 Christos Stavrakakis
    if vm is None and len(args) >= 3:
431 129b94d5 Christos Stavrakakis
        vm = args[2]
432 129b94d5 Christos Stavrakakis
    if vm is not None:
433 129b94d5 Christos Stavrakakis
        if vm.nics.count() == settings.GANETI_MAX_NICS_PER_INSTANCE:
434 129b94d5 Christos Stavrakakis
            raise faults.BadRequest("Maximum ports per server limit reached")
435 fae6e5f0 Christos Stavrakakis
    return _create_port(*args, **kwargs)
436 fae6e5f0 Christos Stavrakakis
437 fae6e5f0 Christos Stavrakakis
438 fae6e5f0 Christos Stavrakakis
def _create_port(userid, network, machine=None, use_ipaddress=None,
439 fae6e5f0 Christos Stavrakakis
                 address=None, name="", security_groups=None,
440 fae6e5f0 Christos Stavrakakis
                 device_owner=None):
441 fae6e5f0 Christos Stavrakakis
    """Create a new port on the specified network.
442 816d7588 Christos Stavrakakis

443 fae6e5f0 Christos Stavrakakis
    Create a new Port(NetworkInterface model) on the specified Network. If
444 fae6e5f0 Christos Stavrakakis
    'machine' is specified, the machine will be connected to the network using
445 fae6e5f0 Christos Stavrakakis
    this port. If 'use_ipaddress' argument is specified, the port will be
446 fae6e5f0 Christos Stavrakakis
    assigned this IPAddress. Otherwise, an IPv4 address from the IPv4 subnet
447 fae6e5f0 Christos Stavrakakis
    will be allocated.
448 816d7588 Christos Stavrakakis

449 816d7588 Christos Stavrakakis
    """
450 fae6e5f0 Christos Stavrakakis
    if network.state != "ACTIVE":
451 8f335041 Christos Stavrakakis
        raise faults.Conflict("Cannot create port while network '%s' is in"
452 8f335041 Christos Stavrakakis
                              " '%s' status" % (network.id, network.state))
453 8f335041 Christos Stavrakakis
    elif network.action == "DESTROY":
454 8d5795b4 Christos Stavrakakis
        msg = "Cannot create port. Network %s is being deleted."
455 3f18f035 Christos Stavrakakis
        raise faults.Conflict(msg % network.id)
456 32b1ed4a Christos Stavrakakis
    elif network.drained:
457 32b1ed4a Christos Stavrakakis
        raise faults.Conflict("Cannot create port while network %s is in"
458 32b1ed4a Christos Stavrakakis
                              " 'SNF:DRAINED' status" % network.id)
459 8f335041 Christos Stavrakakis
460 ba6ad346 Dionysis Grigoropoulos
    utils.check_name_length(name, NetworkInterface.NETWORK_IFACE_NAME_LENGTH,
461 ba6ad346 Dionysis Grigoropoulos
                            "Port name is too long")
462 ba6ad346 Dionysis Grigoropoulos
463 fae6e5f0 Christos Stavrakakis
    ipaddress = None
464 fae6e5f0 Christos Stavrakakis
    if use_ipaddress is not None:
465 fae6e5f0 Christos Stavrakakis
        # Use an existing IPAddress object.
466 fae6e5f0 Christos Stavrakakis
        ipaddress = use_ipaddress
467 fae6e5f0 Christos Stavrakakis
        if ipaddress and (ipaddress.network_id != network.id):
468 fae6e5f0 Christos Stavrakakis
            msg = "IP Address %s does not belong to network %s"
469 fae6e5f0 Christos Stavrakakis
            raise faults.Conflict(msg % (ipaddress.address, network.id))
470 fae6e5f0 Christos Stavrakakis
    else:
471 fae6e5f0 Christos Stavrakakis
        # If network has IPv4 subnets, try to allocate the address that the
472 fae6e5f0 Christos Stavrakakis
        # the user specified or a random one.
473 fae6e5f0 Christos Stavrakakis
        if network.subnets.filter(ipversion=4).exists():
474 0292883e Christos Stavrakakis
            ipaddress = ips.allocate_ip(network, userid=userid,
475 0292883e Christos Stavrakakis
                                        address=address)
476 fae6e5f0 Christos Stavrakakis
        elif address is not None:
477 fae6e5f0 Christos Stavrakakis
            raise faults.BadRequest("Address %s is not a valid IP for the"
478 fae6e5f0 Christos Stavrakakis
                                    " defined network subnets" % address)
479 fae6e5f0 Christos Stavrakakis
480 fae6e5f0 Christos Stavrakakis
    if ipaddress is not None and ipaddress.nic is not None:
481 fae6e5f0 Christos Stavrakakis
        raise faults.Conflict("IP address '%s' is already in use" %
482 fae6e5f0 Christos Stavrakakis
                              ipaddress.address)
483 fae6e5f0 Christos Stavrakakis
484 fae6e5f0 Christos Stavrakakis
    port = NetworkInterface.objects.create(network=network,
485 fae6e5f0 Christos Stavrakakis
                                           state="DOWN",
486 fae6e5f0 Christos Stavrakakis
                                           userid=userid,
487 fae6e5f0 Christos Stavrakakis
                                           device_owner=None,
488 fae6e5f0 Christos Stavrakakis
                                           name=name)
489 fae6e5f0 Christos Stavrakakis
490 fae6e5f0 Christos Stavrakakis
    # add the security groups if any
491 fae6e5f0 Christos Stavrakakis
    if security_groups:
492 fae6e5f0 Christos Stavrakakis
        port.security_groups.add(*security_groups)
493 fae6e5f0 Christos Stavrakakis
494 fae6e5f0 Christos Stavrakakis
    if ipaddress is not None:
495 fae6e5f0 Christos Stavrakakis
        # Associate IPAddress with the Port
496 fae6e5f0 Christos Stavrakakis
        ipaddress.nic = port
497 fae6e5f0 Christos Stavrakakis
        ipaddress.save()
498 fae6e5f0 Christos Stavrakakis
499 fae6e5f0 Christos Stavrakakis
    if machine is not None:
500 ae2da8a2 Christos Stavrakakis
        # Connect port to the instance.
501 fae6e5f0 Christos Stavrakakis
        machine = connect(machine, network, port)
502 fae6e5f0 Christos Stavrakakis
        jobID = machine.task_job_id
503 fae6e5f0 Christos Stavrakakis
        log.info("Created Port %s with IP %s. Ganeti Job: %s",
504 fae6e5f0 Christos Stavrakakis
                 port, ipaddress, jobID)
505 fae6e5f0 Christos Stavrakakis
    else:
506 fae6e5f0 Christos Stavrakakis
        log.info("Created Port %s with IP %s not attached to any instance",
507 fae6e5f0 Christos Stavrakakis
                 port, ipaddress)
508 9ba6bb95 Christos Stavrakakis
509 fae6e5f0 Christos Stavrakakis
    return port
510 9ba6bb95 Christos Stavrakakis
511 e7f74e7d Christos Stavrakakis
512 fae6e5f0 Christos Stavrakakis
def associate_port_with_machine(port, machine):
513 fae6e5f0 Christos Stavrakakis
    """Associate a Port with a VirtualMachine.
514 9ba6bb95 Christos Stavrakakis

515 fae6e5f0 Christos Stavrakakis
    Associate the port with the VirtualMachine and add an entry to the
516 fae6e5f0 Christos Stavrakakis
    IPAddressLog if the port has a public IPv4 address from a public network.
517 9ba6bb95 Christos Stavrakakis

518 fae6e5f0 Christos Stavrakakis
    """
519 ae2da8a2 Christos Stavrakakis
    if port.machine is not None:
520 ae2da8a2 Christos Stavrakakis
        raise faults.Conflict("Port %s is already in use." % port.id)
521 fae6e5f0 Christos Stavrakakis
    if port.network.public:
522 fae6e5f0 Christos Stavrakakis
        ipv4_address = port.ipv4_address
523 fae6e5f0 Christos Stavrakakis
        if ipv4_address is not None:
524 fae6e5f0 Christos Stavrakakis
            ip_log = IPAddressLog.objects.create(server_id=machine.id,
525 fae6e5f0 Christos Stavrakakis
                                                 network_id=port.network_id,
526 fae6e5f0 Christos Stavrakakis
                                                 address=ipv4_address,
527 fae6e5f0 Christos Stavrakakis
                                                 active=True)
528 fae6e5f0 Christos Stavrakakis
            log.debug("Created IP log entry %s", ip_log)
529 fae6e5f0 Christos Stavrakakis
    port.machine = machine
530 fae6e5f0 Christos Stavrakakis
    port.state = "BUILD"
531 fae6e5f0 Christos Stavrakakis
    port.device_owner = "vm"
532 fae6e5f0 Christos Stavrakakis
    port.save()
533 fae6e5f0 Christos Stavrakakis
    return port
534 9ba6bb95 Christos Stavrakakis
535 9ba6bb95 Christos Stavrakakis
536 fae6e5f0 Christos Stavrakakis
@transaction.commit_on_success
537 fae6e5f0 Christos Stavrakakis
def delete_port(port):
538 fae6e5f0 Christos Stavrakakis
    """Delete a port by removing the NIC card from the instance.
539 9ba6bb95 Christos Stavrakakis

540 fae6e5f0 Christos Stavrakakis
    Send a Job to remove the NIC card from the instance. The port
541 fae6e5f0 Christos Stavrakakis
    will be deleted and the associated IPv4 addressess will be released
542 6e73f499 Christos Stavrakakis
    when the job completes successfully.
543 a52cc1b4 Christos Stavrakakis

544 fae6e5f0 Christos Stavrakakis
    """
545 a52cc1b4 Christos Stavrakakis
546 371ab004 Christos Stavrakakis
    vm = port.machine
547 371ab004 Christos Stavrakakis
    if vm is not None and not vm.deleted:
548 fae6e5f0 Christos Stavrakakis
        vm = disconnect(port.machine, port)
549 fae6e5f0 Christos Stavrakakis
        log.info("Removing port %s, Job: %s", port, vm.task_job_id)
550 fae6e5f0 Christos Stavrakakis
    else:
551 fae6e5f0 Christos Stavrakakis
        backend.remove_nic_ips(port)
552 fae6e5f0 Christos Stavrakakis
        port.delete()
553 fae6e5f0 Christos Stavrakakis
        log.info("Removed port %s", port)
554 fae6e5f0 Christos Stavrakakis
555 fae6e5f0 Christos Stavrakakis
    return port
556 3aecadc8 Christos Stavrakakis
557 3aecadc8 Christos Stavrakakis
558 3aecadc8 Christos Stavrakakis
def create_instance_ports(user_id, networks=None):
559 3aecadc8 Christos Stavrakakis
    # First connect the instance to the networks defined by the admin
560 3aecadc8 Christos Stavrakakis
    forced_ports = create_ports_for_setting(user_id, category="admin")
561 3aecadc8 Christos Stavrakakis
    if networks is None:
562 3aecadc8 Christos Stavrakakis
        # If the user did not asked for any networks, connect instance to
563 3aecadc8 Christos Stavrakakis
        # default networks as defined by the admin
564 3aecadc8 Christos Stavrakakis
        ports = create_ports_for_setting(user_id, category="default")
565 3aecadc8 Christos Stavrakakis
    else:
566 3aecadc8 Christos Stavrakakis
        # Else just connect to the networks that the user defined
567 3aecadc8 Christos Stavrakakis
        ports = create_ports_for_request(user_id, networks)
568 129b94d5 Christos Stavrakakis
    total_ports = forced_ports + ports
569 129b94d5 Christos Stavrakakis
    if len(total_ports) > settings.GANETI_MAX_NICS_PER_INSTANCE:
570 129b94d5 Christos Stavrakakis
        raise faults.BadRequest("Maximum ports per server limit reached")
571 129b94d5 Christos Stavrakakis
    return total_ports
572 3aecadc8 Christos Stavrakakis
573 3aecadc8 Christos Stavrakakis
574 3aecadc8 Christos Stavrakakis
def create_ports_for_setting(user_id, category):
575 3aecadc8 Christos Stavrakakis
    if category == "admin":
576 3aecadc8 Christos Stavrakakis
        network_setting = settings.CYCLADES_FORCED_SERVER_NETWORKS
577 e74a5b4b Christos Stavrakakis
        exception = faults.ServiceUnavailable
578 3aecadc8 Christos Stavrakakis
    elif category == "default":
579 3aecadc8 Christos Stavrakakis
        network_setting = settings.CYCLADES_DEFAULT_SERVER_NETWORKS
580 e74a5b4b Christos Stavrakakis
        exception = faults.Conflict
581 3aecadc8 Christos Stavrakakis
    else:
582 3aecadc8 Christos Stavrakakis
        raise ValueError("Unknown category: %s" % category)
583 3aecadc8 Christos Stavrakakis
584 3aecadc8 Christos Stavrakakis
    ports = []
585 3aecadc8 Christos Stavrakakis
    for network_ids in network_setting:
586 3aecadc8 Christos Stavrakakis
        # Treat even simple network IDs as group of networks with one network
587 3aecadc8 Christos Stavrakakis
        if type(network_ids) not in (list, tuple):
588 3aecadc8 Christos Stavrakakis
            network_ids = [network_ids]
589 3aecadc8 Christos Stavrakakis
590 e74a5b4b Christos Stavrakakis
        error_msgs = []
591 3aecadc8 Christos Stavrakakis
        for network_id in network_ids:
592 c32c74d9 Christos Stavrakakis
            success = False
593 3aecadc8 Christos Stavrakakis
            try:
594 3aecadc8 Christos Stavrakakis
                ports.append(_port_from_setting(user_id, network_id, category))
595 e74a5b4b Christos Stavrakakis
                # Port successfully created in one of the networks. Skip the
596 e74a5b4b Christos Stavrakakis
                # the rest.
597 c32c74d9 Christos Stavrakakis
                success = True
598 3aecadc8 Christos Stavrakakis
                break
599 e74a5b4b Christos Stavrakakis
            except faults.Conflict as e:
600 e74a5b4b Christos Stavrakakis
                if len(network_ids) == 1:
601 e74a5b4b Christos Stavrakakis
                    raise exception(e.message)
602 e74a5b4b Christos Stavrakakis
                else:
603 e74a5b4b Christos Stavrakakis
                    error_msgs.append(e.message)
604 e74a5b4b Christos Stavrakakis
605 c32c74d9 Christos Stavrakakis
        if not success:
606 c32c74d9 Christos Stavrakakis
            if category == "admin":
607 c32c74d9 Christos Stavrakakis
                log.error("Cannot connect server to forced networks '%s': %s",
608 c32c74d9 Christos Stavrakakis
                          network_ids, error_msgs)
609 c32c74d9 Christos Stavrakakis
                raise exception("Cannot connect server to forced server"
610 c32c74d9 Christos Stavrakakis
                                " networks.")
611 c32c74d9 Christos Stavrakakis
            else:
612 c32c74d9 Christos Stavrakakis
                log.debug("Cannot connect server to default networks '%s': %s",
613 c32c74d9 Christos Stavrakakis
                          network_ids, error_msgs)
614 c32c74d9 Christos Stavrakakis
                raise exception("Cannot connect server to default server"
615 c32c74d9 Christos Stavrakakis
                                " networks.")
616 e74a5b4b Christos Stavrakakis
617 3aecadc8 Christos Stavrakakis
    return ports
618 3aecadc8 Christos Stavrakakis
619 3aecadc8 Christos Stavrakakis
620 3aecadc8 Christos Stavrakakis
def _port_from_setting(user_id, network_id, category):
621 3aecadc8 Christos Stavrakakis
    # TODO: Fix this..you need only IPv4 and only IPv6 network
622 3aecadc8 Christos Stavrakakis
    if network_id == "SNF:ANY_PUBLIC_IPV4":
623 3aecadc8 Christos Stavrakakis
        return create_public_ipv4_port(user_id, category=category)
624 3aecadc8 Christos Stavrakakis
    elif network_id == "SNF:ANY_PUBLIC_IPV6":
625 3aecadc8 Christos Stavrakakis
        return create_public_ipv6_port(user_id, category=category)
626 3aecadc8 Christos Stavrakakis
    elif network_id == "SNF:ANY_PUBLIC":
627 3aecadc8 Christos Stavrakakis
        try:
628 3aecadc8 Christos Stavrakakis
            return create_public_ipv4_port(user_id, category=category)
629 e74a5b4b Christos Stavrakakis
        except faults.Conflict as e1:
630 e74a5b4b Christos Stavrakakis
            try:
631 e74a5b4b Christos Stavrakakis
                return create_public_ipv6_port(user_id, category=category)
632 e74a5b4b Christos Stavrakakis
            except faults.Conflict as e2:
633 e74a5b4b Christos Stavrakakis
                log.error("Failed to connect server to a public IPv4 or IPv6"
634 e74a5b4b Christos Stavrakakis
                          " network. IPv4: %s, IPv6: %s", e1, e2)
635 e74a5b4b Christos Stavrakakis
                msg = ("Cannot connect server to a public IPv4 or IPv6"
636 e74a5b4b Christos Stavrakakis
                       " network.")
637 e74a5b4b Christos Stavrakakis
                raise faults.Conflict(msg)
638 3aecadc8 Christos Stavrakakis
    else:  # Case of network ID
639 3aecadc8 Christos Stavrakakis
        if category in ["user", "default"]:
640 3aecadc8 Christos Stavrakakis
            return _port_for_request(user_id, {"uuid": network_id})
641 3aecadc8 Christos Stavrakakis
        elif category == "admin":
642 3aecadc8 Christos Stavrakakis
            network = util.get_network(network_id, user_id, non_deleted=True)
643 3aecadc8 Christos Stavrakakis
            return _create_port(user_id, network)
644 3aecadc8 Christos Stavrakakis
        else:
645 3aecadc8 Christos Stavrakakis
            raise ValueError("Unknown category: %s" % category)
646 3aecadc8 Christos Stavrakakis
647 3aecadc8 Christos Stavrakakis
648 3aecadc8 Christos Stavrakakis
def create_public_ipv4_port(user_id, network=None, address=None,
649 3aecadc8 Christos Stavrakakis
                            category="user"):
650 3aecadc8 Christos Stavrakakis
    """Create a port in a public IPv4 network.
651 3aecadc8 Christos Stavrakakis

652 3aecadc8 Christos Stavrakakis
    Create a port in a public IPv4 network (that may also have an IPv6
653 3aecadc8 Christos Stavrakakis
    subnet). If the category is 'user' or 'default' this will try to use
654 3aecadc8 Christos Stavrakakis
    one of the users floating IPs. If the category is 'admin' will
655 3aecadc8 Christos Stavrakakis
    create a port to the public network (without floating IPs or quotas).
656 3aecadc8 Christos Stavrakakis

657 3aecadc8 Christos Stavrakakis
    """
658 3aecadc8 Christos Stavrakakis
    if category in ["user", "default"]:
659 3aecadc8 Christos Stavrakakis
        if address is None:
660 3aecadc8 Christos Stavrakakis
            ipaddress = ips.get_free_floating_ip(user_id, network)
661 3aecadc8 Christos Stavrakakis
        else:
662 3aecadc8 Christos Stavrakakis
            ipaddress = util.get_floating_ip_by_address(user_id, address,
663 3aecadc8 Christos Stavrakakis
                                                        for_update=True)
664 3aecadc8 Christos Stavrakakis
    elif category == "admin":
665 3aecadc8 Christos Stavrakakis
        if network is None:
666 3aecadc8 Christos Stavrakakis
            ipaddress = ips.allocate_public_ip(user_id)
667 3aecadc8 Christos Stavrakakis
        else:
668 3aecadc8 Christos Stavrakakis
            ipaddress = ips.allocate_ip(network, user_id)
669 3aecadc8 Christos Stavrakakis
    else:
670 3aecadc8 Christos Stavrakakis
        raise ValueError("Unknown category: %s" % category)
671 3aecadc8 Christos Stavrakakis
    if network is None:
672 3aecadc8 Christos Stavrakakis
        network = ipaddress.network
673 3aecadc8 Christos Stavrakakis
    return _create_port(user_id, network, use_ipaddress=ipaddress)
674 3aecadc8 Christos Stavrakakis
675 3aecadc8 Christos Stavrakakis
676 3aecadc8 Christos Stavrakakis
def create_public_ipv6_port(user_id, category=None):
677 3aecadc8 Christos Stavrakakis
    """Create a port in a public IPv6 only network."""
678 3aecadc8 Christos Stavrakakis
    networks = Network.objects.filter(public=True, deleted=False,
679 3aecadc8 Christos Stavrakakis
                                      drained=False, subnets__ipversion=6)\
680 3aecadc8 Christos Stavrakakis
                              .exclude(subnets__ipversion=4)
681 3aecadc8 Christos Stavrakakis
    if networks:
682 3aecadc8 Christos Stavrakakis
        return _create_port(user_id, networks[0])
683 3aecadc8 Christos Stavrakakis
    else:
684 3aecadc8 Christos Stavrakakis
        msg = "No available IPv6 only network!"
685 3aecadc8 Christos Stavrakakis
        log.error(msg)
686 3aecadc8 Christos Stavrakakis
        raise faults.Conflict(msg)
687 3aecadc8 Christos Stavrakakis
688 3aecadc8 Christos Stavrakakis
689 3aecadc8 Christos Stavrakakis
def create_ports_for_request(user_id, networks):
690 3aecadc8 Christos Stavrakakis
    """Create the server ports requested by the user.
691 3aecadc8 Christos Stavrakakis

692 3aecadc8 Christos Stavrakakis
    Create the ports for the new servers as requested in the 'networks'
693 3aecadc8 Christos Stavrakakis
    attribute. The networks attribute contains either a list of network IDs
694 3aecadc8 Christos Stavrakakis
    ('uuid') or a list of ports IDs ('port'). In case of network IDs, the user
695 3aecadc8 Christos Stavrakakis
    can also specify an IPv4 address ('fixed_ip'). In order to connect to a
696 3aecadc8 Christos Stavrakakis
    public network, the 'fixed_ip' attribute must contain the IPv4 address of a
697 3aecadc8 Christos Stavrakakis
    floating IP. If the network is public but the 'fixed_ip' attribute is not
698 3aecadc8 Christos Stavrakakis
    specified, the system will automatically reserve one of the users floating
699 3aecadc8 Christos Stavrakakis
    IPs.
700 3aecadc8 Christos Stavrakakis

701 3aecadc8 Christos Stavrakakis
    """
702 14402edc Christos Stavrakakis
    if not isinstance(networks, list):
703 14402edc Christos Stavrakakis
        raise faults.BadRequest("Malformed request. Invalid 'networks' field")
704 3aecadc8 Christos Stavrakakis
    return [_port_for_request(user_id, network) for network in networks]
705 3aecadc8 Christos Stavrakakis
706 3aecadc8 Christos Stavrakakis
707 3aecadc8 Christos Stavrakakis
def _port_for_request(user_id, network_dict):
708 14402edc Christos Stavrakakis
    if not isinstance(network_dict, dict):
709 14402edc Christos Stavrakakis
        raise faults.BadRequest("Malformed request. Invalid 'networks' field")
710 3aecadc8 Christos Stavrakakis
    port_id = network_dict.get("port")
711 3aecadc8 Christos Stavrakakis
    network_id = network_dict.get("uuid")
712 3aecadc8 Christos Stavrakakis
    if port_id is not None:
713 3aecadc8 Christos Stavrakakis
        return util.get_port(port_id, user_id, for_update=True)
714 3aecadc8 Christos Stavrakakis
    elif network_id is not None:
715 3aecadc8 Christos Stavrakakis
        address = network_dict.get("fixed_ip")
716 3aecadc8 Christos Stavrakakis
        network = util.get_network(network_id, user_id, non_deleted=True)
717 3aecadc8 Christos Stavrakakis
        if network.public:
718 3aecadc8 Christos Stavrakakis
            if network.subnet4 is not None:
719 3aecadc8 Christos Stavrakakis
                if not "fixed_ip" in network_dict:
720 3aecadc8 Christos Stavrakakis
                    return create_public_ipv4_port(user_id, network)
721 3aecadc8 Christos Stavrakakis
                elif address is None:
722 3aecadc8 Christos Stavrakakis
                    msg = "Cannot connect to public network"
723 3aecadc8 Christos Stavrakakis
                    raise faults.BadRequest(msg % network.id)
724 3aecadc8 Christos Stavrakakis
                else:
725 3aecadc8 Christos Stavrakakis
                    return create_public_ipv4_port(user_id, network, address)
726 3aecadc8 Christos Stavrakakis
            else:
727 3aecadc8 Christos Stavrakakis
                raise faults.Forbidden("Cannot connect to IPv6 only public"
728 3aecadc8 Christos Stavrakakis
                                       " network %" % network.id)
729 3aecadc8 Christos Stavrakakis
        else:
730 3aecadc8 Christos Stavrakakis
            return _create_port(user_id, network, address=address)
731 3aecadc8 Christos Stavrakakis
    else:
732 3aecadc8 Christos Stavrakakis
        raise faults.BadRequest("Network 'uuid' or 'port' attribute"
733 3aecadc8 Christos Stavrakakis
                                " is required.")