Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / logic / servers.py @ 883c1f94

History | View | Annotate | Download (28 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 41a7fae7 Christos Stavrakakis
                    quotas.reject_serial(vm.serial)
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 88fd91af Christos Stavrakakis
                quotas.accept_serial(vm.serial)
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 41a7fae7 Christos Stavrakakis
def destroy(vm):
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 41a7fae7 Christos Stavrakakis
    return backend.delete_instance(vm)
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 41a7fae7 Christos Stavrakakis
def stop(vm):
294 41a7fae7 Christos Stavrakakis
    log.info("Stopping VM %s", vm)
295 41a7fae7 Christos Stavrakakis
    return backend.shutdown_instance(vm)
296 41a7fae7 Christos Stavrakakis
297 41a7fae7 Christos Stavrakakis
298 41a7fae7 Christos Stavrakakis
@server_command("REBOOT")
299 41a7fae7 Christos Stavrakakis
def reboot(vm, reboot_type):
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 41a7fae7 Christos Stavrakakis
    return backend.reboot_instance(vm, reboot_type.lower())
306 41a7fae7 Christos Stavrakakis
307 41a7fae7 Christos Stavrakakis
308 41a7fae7 Christos Stavrakakis
def resize(vm, flavor):
309 64bca363 Giorgos Korfiatis
    action_fields = {"beparams": {"vcpus": flavor.cpu,
310 64bca363 Giorgos Korfiatis
                                  "maxmem": flavor.ram}}
311 64bca363 Giorgos Korfiatis
    comm = server_command("RESIZE", action_fields=action_fields)
312 64bca363 Giorgos Korfiatis
    return comm(_resize)(vm, flavor)
313 64bca363 Giorgos Korfiatis
314 64bca363 Giorgos Korfiatis
315 64bca363 Giorgos Korfiatis
def _resize(vm, flavor):
316 41a7fae7 Christos Stavrakakis
    old_flavor = vm.flavor
317 41a7fae7 Christos Stavrakakis
    # User requested the same flavor
318 41a7fae7 Christos Stavrakakis
    if old_flavor.id == flavor.id:
319 41a7fae7 Christos Stavrakakis
        raise faults.BadRequest("Server '%s' flavor is already '%s'."
320 41a7fae7 Christos Stavrakakis
                                % (vm, flavor))
321 41a7fae7 Christos Stavrakakis
    # Check that resize can be performed
322 41a7fae7 Christos Stavrakakis
    if old_flavor.disk != flavor.disk:
323 8d5795b4 Christos Stavrakakis
        raise faults.BadRequest("Cannot resize instance disk.")
324 41a7fae7 Christos Stavrakakis
    if old_flavor.disk_template != flavor.disk_template:
325 8d5795b4 Christos Stavrakakis
        raise faults.BadRequest("Cannot change instance disk template.")
326 41a7fae7 Christos Stavrakakis
327 41a7fae7 Christos Stavrakakis
    log.info("Resizing VM from flavor '%s' to '%s", old_flavor, flavor)
328 41a7fae7 Christos Stavrakakis
    return backend.resize_instance(vm, vcpus=flavor.cpu, memory=flavor.ram)
329 41a7fae7 Christos Stavrakakis
330 41a7fae7 Christos Stavrakakis
331 41a7fae7 Christos Stavrakakis
@server_command("SET_FIREWALL_PROFILE")
332 d0545590 Christos Stavrakakis
def set_firewall_profile(vm, profile, nic):
333 d0545590 Christos Stavrakakis
    log.info("Setting VM %s, NIC %s, firewall %s", vm, nic, profile)
334 41a7fae7 Christos Stavrakakis
335 41a7fae7 Christos Stavrakakis
    if profile not in [x[0] for x in NetworkInterface.FIREWALL_PROFILES]:
336 41a7fae7 Christos Stavrakakis
        raise faults.BadRequest("Unsupported firewall profile")
337 d0545590 Christos Stavrakakis
    backend.set_firewall_profile(vm, profile=profile, nic=nic)
338 41a7fae7 Christos Stavrakakis
    return None
339 41a7fae7 Christos Stavrakakis
340 41a7fae7 Christos Stavrakakis
341 41a7fae7 Christos Stavrakakis
@server_command("CONNECT")
342 6b8dc47c Christos Stavrakakis
def connect(vm, network, port=None):
343 6b8dc47c Christos Stavrakakis
    if port is None:
344 fae6e5f0 Christos Stavrakakis
        port = _create_port(vm.userid, network)
345 fae6e5f0 Christos Stavrakakis
    associate_port_with_machine(port, vm)
346 41a7fae7 Christos Stavrakakis
347 fae6e5f0 Christos Stavrakakis
    log.info("Creating NIC %s with IPv4 Address %s", port, port.ipv4_address)
348 41a7fae7 Christos Stavrakakis
349 fae6e5f0 Christos Stavrakakis
    return backend.connect_to_network(vm, port)
350 41a7fae7 Christos Stavrakakis
351 41a7fae7 Christos Stavrakakis
352 41a7fae7 Christos Stavrakakis
@server_command("DISCONNECT")
353 7c714455 Christos Stavrakakis
def disconnect(vm, nic):
354 7c714455 Christos Stavrakakis
    log.info("Removing NIC %s from VM %s", nic, vm)
355 41a7fae7 Christos Stavrakakis
    return backend.disconnect_from_network(vm, nic)
356 41a7fae7 Christos Stavrakakis
357 41a7fae7 Christos Stavrakakis
358 41a7fae7 Christos Stavrakakis
def console(vm, console_type):
359 41a7fae7 Christos Stavrakakis
    """Arrange for an OOB console of the specified type
360 41a7fae7 Christos Stavrakakis

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

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

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

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

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

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

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

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

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

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

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

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

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