Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / logic / backend.py @ b2791a77

History | View | Annotate | Download (32.7 kB)

1 41a7fae7 Christos Stavrakakis
# Copyright 2011-2013 GRNET S.A. All rights reserved.
2 37ca953f Christodoulos Psaltis
#
3 adee02b8 Giorgos Verigakis
# Redistribution and use in source and binary forms, with or
4 adee02b8 Giorgos Verigakis
# without modification, are permitted provided that the following
5 adee02b8 Giorgos Verigakis
# conditions are met:
6 37ca953f Christodoulos Psaltis
#
7 adee02b8 Giorgos Verigakis
#   1. Redistributions of source code must retain the above
8 adee02b8 Giorgos Verigakis
#      copyright notice, this list of conditions and the following
9 adee02b8 Giorgos Verigakis
#      disclaimer.
10 37ca953f Christodoulos Psaltis
#
11 adee02b8 Giorgos Verigakis
#   2. Redistributions in binary form must reproduce the above
12 adee02b8 Giorgos Verigakis
#      copyright notice, this list of conditions and the following
13 adee02b8 Giorgos Verigakis
#      disclaimer in the documentation and/or other materials
14 adee02b8 Giorgos Verigakis
#      provided with the distribution.
15 37ca953f Christodoulos Psaltis
#
16 adee02b8 Giorgos Verigakis
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
17 adee02b8 Giorgos Verigakis
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18 adee02b8 Giorgos Verigakis
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19 adee02b8 Giorgos Verigakis
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
20 adee02b8 Giorgos Verigakis
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21 adee02b8 Giorgos Verigakis
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22 adee02b8 Giorgos Verigakis
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
23 adee02b8 Giorgos Verigakis
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
24 adee02b8 Giorgos Verigakis
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 adee02b8 Giorgos Verigakis
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
26 adee02b8 Giorgos Verigakis
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27 adee02b8 Giorgos Verigakis
# POSSIBILITY OF SUCH DAMAGE.
28 37ca953f Christodoulos Psaltis
#
29 adee02b8 Giorgos Verigakis
# The views and conclusions contained in the software and
30 adee02b8 Giorgos Verigakis
# documentation are those of the authors and should not be
31 adee02b8 Giorgos Verigakis
# interpreted as representing official policies, either expressed
32 adee02b8 Giorgos Verigakis
# or implied, of GRNET S.A.
33 529178b1 Giorgos Verigakis
from django.conf import settings
34 207b70d5 Giorgos Verigakis
from django.db import transaction
35 1a894bfe Christos Stavrakakis
from datetime import datetime
36 207b70d5 Giorgos Verigakis
37 22ee6892 Christos Stavrakakis
from synnefo.db.models import (Backend, VirtualMachine, Network,
38 72dea98f Christos Stavrakakis
                               FloatingIP,
39 3524241a Christos Stavrakakis
                               BackendNetwork, BACKEND_STATUSES,
40 ca4d59e3 Christos Stavrakakis
                               pooled_rapi_client, VirtualMachineDiagnostic,
41 ca4d59e3 Christos Stavrakakis
                               Flavor)
42 03992c72 Christos Stavrakakis
from synnefo.logic import utils
43 cb4eee84 Christos Stavrakakis
from synnefo import quotas
44 b7d38981 Dimitris Aragiorgis
from synnefo.api.util import release_resource
45 fd95834e Christos Stavrakakis
from synnefo.util.mac2eui64 import mac2eui64
46 529178b1 Giorgos Verigakis
47 3524241a Christos Stavrakakis
from logging import getLogger
48 3524241a Christos Stavrakakis
log = getLogger(__name__)
49 9e98ba3c Giorgos Verigakis
50 529178b1 Giorgos Verigakis
51 efff6193 Giorgos Verigakis
_firewall_tags = {
52 efff6193 Giorgos Verigakis
    'ENABLED': settings.GANETI_FIREWALL_ENABLED_TAG,
53 efff6193 Giorgos Verigakis
    'DISABLED': settings.GANETI_FIREWALL_DISABLED_TAG,
54 efff6193 Giorgos Verigakis
    'PROTECTED': settings.GANETI_FIREWALL_PROTECTED_TAG}
55 efff6193 Giorgos Verigakis
56 efff6193 Giorgos Verigakis
_reverse_tags = dict((v.split(':')[3], k) for k, v in _firewall_tags.items())
57 efff6193 Giorgos Verigakis
58 02feca11 Vassilios Karakoidas
59 41a7fae7 Christos Stavrakakis
def handle_vm_quotas(vm, job_id, job_opcode, job_status, job_fields):
60 41a7fae7 Christos Stavrakakis
    """Handle quotas for updated VirtualMachine.
61 41a7fae7 Christos Stavrakakis

62 41a7fae7 Christos Stavrakakis
    Update quotas for the updated VirtualMachine based on the job that run on
63 41a7fae7 Christos Stavrakakis
    the Ganeti backend. If a commission has been already issued for this job,
64 41a7fae7 Christos Stavrakakis
    then this commission is just accepted or rejected based on the job status.
65 41a7fae7 Christos Stavrakakis
    Otherwise, a new commission for the given change is issued, that is also in
66 41a7fae7 Christos Stavrakakis
    force and auto-accept mode. In this case, previous commissions are
67 41a7fae7 Christos Stavrakakis
    rejected, since they reflect a previous state of the VM.
68 41a7fae7 Christos Stavrakakis

69 41a7fae7 Christos Stavrakakis
    """
70 41a7fae7 Christos Stavrakakis
    if job_status not in ["success", "error", "canceled"]:
71 41a7fae7 Christos Stavrakakis
        return
72 41a7fae7 Christos Stavrakakis
73 41a7fae7 Christos Stavrakakis
    # Check successful completion of a job will trigger any quotable change in
74 41a7fae7 Christos Stavrakakis
    # the VM state.
75 41a7fae7 Christos Stavrakakis
    action = utils.get_action_from_opcode(job_opcode, job_fields)
76 41a7fae7 Christos Stavrakakis
    commission_info = quotas.get_commission_info(vm, action=action,
77 41a7fae7 Christos Stavrakakis
                                                 action_fields=job_fields)
78 41a7fae7 Christos Stavrakakis
79 41a7fae7 Christos Stavrakakis
    if vm.task_job_id == job_id and vm.serial is not None:
80 41a7fae7 Christos Stavrakakis
        # Commission for this change has already been issued. So just
81 41a7fae7 Christos Stavrakakis
        # accept/reject it
82 41a7fae7 Christos Stavrakakis
        serial = vm.serial
83 41a7fae7 Christos Stavrakakis
        if job_status == "success":
84 41a7fae7 Christos Stavrakakis
            quotas.accept_serial(serial)
85 41a7fae7 Christos Stavrakakis
        elif job_status in ["error", "canceled"]:
86 41a7fae7 Christos Stavrakakis
            log.debug("Job %s failed. Rejecting related serial %s", job_id,
87 41a7fae7 Christos Stavrakakis
                      serial)
88 41a7fae7 Christos Stavrakakis
            quotas.reject_serial(serial)
89 41a7fae7 Christos Stavrakakis
        vm.serial = None
90 41a7fae7 Christos Stavrakakis
    elif job_status == "success" and commission_info is not None:
91 41a7fae7 Christos Stavrakakis
        log.debug("Expected job was %s. Processing job %s. Commission for"
92 41a7fae7 Christos Stavrakakis
                  " this job: %s", vm.task_job_id, job_id, commission_info)
93 41a7fae7 Christos Stavrakakis
        # Commission for this change has not been issued, or the issued
94 41a7fae7 Christos Stavrakakis
        # commission was unaware of the current change. Reject all previous
95 41a7fae7 Christos Stavrakakis
        # commissions and create a new one in forced mode!
96 41a7fae7 Christos Stavrakakis
        previous_serial = vm.serial
97 41a7fae7 Christos Stavrakakis
        if previous_serial and not previous_serial.resolved:
98 41a7fae7 Christos Stavrakakis
            quotas.resolve_vm_commission(previous_serial)
99 41a7fae7 Christos Stavrakakis
        serial = quotas.issue_commission(user=vm.userid,
100 41a7fae7 Christos Stavrakakis
                                         source=quotas.DEFAULT_SOURCE,
101 41a7fae7 Christos Stavrakakis
                                         provisions=commission_info,
102 41a7fae7 Christos Stavrakakis
                                         force=True,
103 41a7fae7 Christos Stavrakakis
                                         auto_accept=True)
104 41a7fae7 Christos Stavrakakis
        # Clear VM's serial. Expected job may arrive later. However correlated
105 41a7fae7 Christos Stavrakakis
        # serial must not be accepted, since it reflects a previous VM state
106 41a7fae7 Christos Stavrakakis
        vm.serial = None
107 41a7fae7 Christos Stavrakakis
108 41a7fae7 Christos Stavrakakis
    return vm
109 41a7fae7 Christos Stavrakakis
110 41a7fae7 Christos Stavrakakis
111 093f9c53 Vangelis Koukis
@transaction.commit_on_success
112 ca4d59e3 Christos Stavrakakis
def process_op_status(vm, etime, jobid, opcode, status, logmsg, nics=None,
113 ca4d59e3 Christos Stavrakakis
                      beparams=None):
114 ad2d6807 Vangelis Koukis
    """Process a job progress notification from the backend
115 02feca11 Vassilios Karakoidas

116 02feca11 Vassilios Karakoidas
    Process an incoming message from the backend (currently Ganeti).
117 02feca11 Vassilios Karakoidas
    Job notifications with a terminating status (sucess, error, or canceled),
118 02feca11 Vassilios Karakoidas
    also update the operating state of the VM.
119 02feca11 Vassilios Karakoidas

120 02feca11 Vassilios Karakoidas
    """
121 41303ed0 Vangelis Koukis
    # See #1492, #1031, #1111 why this line has been removed
122 41303ed0 Vangelis Koukis
    #if (opcode not in [x[0] for x in VirtualMachine.BACKEND_OPCODES] or
123 fd65ab41 Christos Stavrakakis
    if status not in [x[0] for x in BACKEND_STATUSES]:
124 02feca11 Vassilios Karakoidas
        raise VirtualMachine.InvalidBackendMsgError(opcode, status)
125 02feca11 Vassilios Karakoidas
126 dfd19c2d Vassilios Karakoidas
    vm.backendjobid = jobid
127 dfd19c2d Vassilios Karakoidas
    vm.backendjobstatus = status
128 dfd19c2d Vassilios Karakoidas
    vm.backendopcode = opcode
129 dfd19c2d Vassilios Karakoidas
    vm.backendlogmsg = logmsg
130 02feca11 Vassilios Karakoidas
131 41a7fae7 Christos Stavrakakis
    if status in ["queued", "waiting", "running"]:
132 41a7fae7 Christos Stavrakakis
        vm.save()
133 41a7fae7 Christos Stavrakakis
        return
134 41a7fae7 Christos Stavrakakis
135 41a7fae7 Christos Stavrakakis
    state_for_success = VirtualMachine.OPER_STATE_FROM_OPCODE.get(opcode)
136 02feca11 Vassilios Karakoidas
    # Notifications of success change the operating state
137 41a7fae7 Christos Stavrakakis
    if status == "success":
138 41a7fae7 Christos Stavrakakis
        if state_for_success is not None:
139 41a7fae7 Christos Stavrakakis
            vm.operstate = state_for_success
140 41a7fae7 Christos Stavrakakis
        if nics is not None:
141 41a7fae7 Christos Stavrakakis
            # Update the NICs of the VM
142 41a7fae7 Christos Stavrakakis
            _process_net_status(vm, etime, nics)
143 41a7fae7 Christos Stavrakakis
        if beparams:
144 41a7fae7 Christos Stavrakakis
            # Change the flavor of the VM
145 ca4d59e3 Christos Stavrakakis
            _process_resize(vm, beparams)
146 41a7fae7 Christos Stavrakakis
        # Update backendtime only for jobs that have been successfully
147 41a7fae7 Christos Stavrakakis
        # completed, since only these jobs update the state of the VM. Else a
148 41a7fae7 Christos Stavrakakis
        # "race condition" may occur when a successful job (e.g.
149 41a7fae7 Christos Stavrakakis
        # OP_INSTANCE_REMOVE) completes before an error job and messages arrive
150 41a7fae7 Christos Stavrakakis
        # in reversed order.
151 41a7fae7 Christos Stavrakakis
        vm.backendtime = etime
152 ca4d59e3 Christos Stavrakakis
153 91954b45 Christos Stavrakakis
    # Special case: if OP_INSTANCE_CREATE fails --> ERROR
154 cb4eee84 Christos Stavrakakis
    if opcode == 'OP_INSTANCE_CREATE' and status in ('canceled', 'error'):
155 c4ce868e Christos Stavrakakis
        vm.operstate = 'ERROR'
156 c4ce868e Christos Stavrakakis
        vm.backendtime = etime
157 cb4eee84 Christos Stavrakakis
    elif opcode == 'OP_INSTANCE_REMOVE':
158 e97288bc Christos Stavrakakis
        # Set the deleted flag explicitly, cater for admin-initiated removals
159 e97288bc Christos Stavrakakis
        # Special case: OP_INSTANCE_REMOVE fails for machines in ERROR,
160 e97288bc Christos Stavrakakis
        # when no instance exists at the Ganeti backend.
161 e97288bc Christos Stavrakakis
        # See ticket #799 for all the details.
162 07602322 Christos Stavrakakis
        if (status == 'success' or
163 07602322 Christos Stavrakakis
           (status == 'error' and (vm.operstate == 'ERROR' or
164 07602322 Christos Stavrakakis
                                   vm.action == 'DESTROY'))):
165 72dea98f Christos Stavrakakis
            # VM has been deleted. Release the instance IPs
166 72dea98f Christos Stavrakakis
            release_instance_ips(vm, [])
167 72dea98f Christos Stavrakakis
            # And delete the releated NICs (must be performed after release!)
168 72dea98f Christos Stavrakakis
            vm.nics.all().delete()
169 e97288bc Christos Stavrakakis
            vm.deleted = True
170 e97288bc Christos Stavrakakis
            vm.operstate = state_for_success
171 e97288bc Christos Stavrakakis
            vm.backendtime = etime
172 41a7fae7 Christos Stavrakakis
            status = "success"
173 41a7fae7 Christos Stavrakakis
174 41a7fae7 Christos Stavrakakis
    if status in ["success", "error", "canceled"]:
175 41a7fae7 Christos Stavrakakis
        # Job is finalized: Handle quotas/commissioning
176 41a7fae7 Christos Stavrakakis
        job_fields = {"nics": nics, "beparams": beparams}
177 41a7fae7 Christos Stavrakakis
        vm = handle_vm_quotas(vm, job_id=jobid, job_opcode=opcode,
178 41a7fae7 Christos Stavrakakis
                              job_status=status, job_fields=job_fields)
179 41a7fae7 Christos Stavrakakis
        # and clear task fields
180 41a7fae7 Christos Stavrakakis
        if vm.task_job_id == jobid:
181 41a7fae7 Christos Stavrakakis
            vm.task = None
182 41a7fae7 Christos Stavrakakis
            vm.task_job_id = None
183 02feca11 Vassilios Karakoidas
184 02feca11 Vassilios Karakoidas
    vm.save()
185 22e52ede Vassilios Karakoidas
186 ad2d6807 Vangelis Koukis
187 ca4d59e3 Christos Stavrakakis
def _process_resize(vm, beparams):
188 ca4d59e3 Christos Stavrakakis
    """Change flavor of a VirtualMachine based on new beparams."""
189 ca4d59e3 Christos Stavrakakis
    old_flavor = vm.flavor
190 41a7fae7 Christos Stavrakakis
    vcpus = beparams.get("vcpus", old_flavor.cpu)
191 41a7fae7 Christos Stavrakakis
    ram = beparams.get("maxmem", old_flavor.ram)
192 41a7fae7 Christos Stavrakakis
    if vcpus == old_flavor.cpu and ram == old_flavor.ram:
193 ca4d59e3 Christos Stavrakakis
        return
194 ca4d59e3 Christos Stavrakakis
    try:
195 ca4d59e3 Christos Stavrakakis
        new_flavor = Flavor.objects.get(cpu=vcpus, ram=ram,
196 ca4d59e3 Christos Stavrakakis
                                        disk=old_flavor.disk,
197 ca4d59e3 Christos Stavrakakis
                                        disk_template=old_flavor.disk_template)
198 ca4d59e3 Christos Stavrakakis
    except Flavor.DoesNotExist:
199 ca4d59e3 Christos Stavrakakis
        raise Exception("Can not find flavor for VM")
200 ca4d59e3 Christos Stavrakakis
    vm.flavor = new_flavor
201 ca4d59e3 Christos Stavrakakis
    vm.save()
202 ca4d59e3 Christos Stavrakakis
203 ca4d59e3 Christos Stavrakakis
204 207b70d5 Giorgos Verigakis
@transaction.commit_on_success
205 c4e55622 Christos Stavrakakis
def process_net_status(vm, etime, nics):
206 fd95834e Christos Stavrakakis
    """Wrap _process_net_status inside transaction."""
207 fd95834e Christos Stavrakakis
    _process_net_status(vm, etime, nics)
208 fd95834e Christos Stavrakakis
209 fd95834e Christos Stavrakakis
210 fd95834e Christos Stavrakakis
def _process_net_status(vm, etime, nics):
211 ad2d6807 Vangelis Koukis
    """Process a net status notification from the backend
212 ad2d6807 Vangelis Koukis

213 ad2d6807 Vangelis Koukis
    Process an incoming message from the Ganeti backend,
214 ad2d6807 Vangelis Koukis
    detailing the NIC configuration of a VM instance.
215 ad2d6807 Vangelis Koukis

216 ad2d6807 Vangelis Koukis
    Update the state of the VM in the DB accordingly.
217 ad2d6807 Vangelis Koukis
    """
218 37ca953f Christodoulos Psaltis
219 b578d9e7 Christos Stavrakakis
    ganeti_nics = process_ganeti_nics(nics)
220 b578d9e7 Christos Stavrakakis
    if not nics_changed(vm.nics.order_by('index'), ganeti_nics):
221 b578d9e7 Christos Stavrakakis
        log.debug("NICs for VM %s have not changed", vm)
222 40d53b77 Christos Stavrakakis
        return
223 40d53b77 Christos Stavrakakis
224 40d53b77 Christos Stavrakakis
    # Get X-Lock on backend before getting X-Lock on network IP pools, to
225 40d53b77 Christos Stavrakakis
    # guarantee that no deadlock will occur with Backend allocator.
226 40d53b77 Christos Stavrakakis
    Backend.objects.select_for_update().get(id=vm.backend_id)
227 b578d9e7 Christos Stavrakakis
228 72dea98f Christos Stavrakakis
    # NICs have changed. Release the instance IPs
229 72dea98f Christos Stavrakakis
    release_instance_ips(vm, ganeti_nics)
230 72dea98f Christos Stavrakakis
    # And delete the releated NICs (must be performed after release!)
231 72dea98f Christos Stavrakakis
    vm.nics.all().delete()
232 77f0fa63 Christos Stavrakakis
233 b578d9e7 Christos Stavrakakis
    for nic in ganeti_nics:
234 8d325d4b Christos Stavrakakis
        ipv4 = nic["ipv4"]
235 f45a7ac4 Christos Stavrakakis
        net = nic['network']
236 b578d9e7 Christos Stavrakakis
        if ipv4:
237 b578d9e7 Christos Stavrakakis
            net.reserve_address(ipv4)
238 b578d9e7 Christos Stavrakakis
239 b578d9e7 Christos Stavrakakis
        nic['dirty'] = False
240 b578d9e7 Christos Stavrakakis
        vm.nics.create(**nic)
241 b578d9e7 Christos Stavrakakis
        # Dummy save the network, because UI uses changed-since for VMs
242 b578d9e7 Christos Stavrakakis
        # and Networks in order to show the VM NICs
243 b578d9e7 Christos Stavrakakis
        net.save()
244 b578d9e7 Christos Stavrakakis
245 b578d9e7 Christos Stavrakakis
    vm.backendtime = etime
246 b578d9e7 Christos Stavrakakis
    vm.save()
247 b578d9e7 Christos Stavrakakis
248 b578d9e7 Christos Stavrakakis
249 b578d9e7 Christos Stavrakakis
def process_ganeti_nics(ganeti_nics):
250 b578d9e7 Christos Stavrakakis
    """Process NIC dict from ganeti hooks."""
251 b578d9e7 Christos Stavrakakis
    new_nics = []
252 b578d9e7 Christos Stavrakakis
    for i, new_nic in enumerate(ganeti_nics):
253 77f0fa63 Christos Stavrakakis
        network = new_nic.get('network', '')
254 22ee6892 Christos Stavrakakis
        n = str(network)
255 77f0fa63 Christos Stavrakakis
        pk = utils.id_from_network_name(n)
256 77f0fa63 Christos Stavrakakis
257 fdc94944 Christos Stavrakakis
        net = Network.objects.get(pk=pk)
258 77f0fa63 Christos Stavrakakis
259 77f0fa63 Christos Stavrakakis
        # Get the new nic info
260 8d325d4b Christos Stavrakakis
        mac = new_nic.get('mac')
261 8d325d4b Christos Stavrakakis
        ipv4 = new_nic.get('ip')
262 8d325d4b Christos Stavrakakis
        ipv6 = mac2eui64(mac, net.subnet6) if net.subnet6 is not None else None
263 8d325d4b Christos Stavrakakis
264 8d325d4b Christos Stavrakakis
        firewall = new_nic.get('firewall')
265 8d325d4b Christos Stavrakakis
        firewall_profile = _reverse_tags.get(firewall)
266 658a825a Giorgos Verigakis
        if not firewall_profile and net.public:
267 658a825a Giorgos Verigakis
            firewall_profile = settings.DEFAULT_FIREWALL_PROFILE
268 9afeb669 Kostas Papadimitriou
269 b578d9e7 Christos Stavrakakis
        nic = {
270 cc92b70f Christos Stavrakakis
            'index': i,
271 cc92b70f Christos Stavrakakis
            'network': net,
272 cc92b70f Christos Stavrakakis
            'mac': mac,
273 cc92b70f Christos Stavrakakis
            'ipv4': ipv4,
274 cc92b70f Christos Stavrakakis
            'ipv6': ipv6,
275 939d71dd Christos Stavrakakis
            'firewall_profile': firewall_profile,
276 939d71dd Christos Stavrakakis
            'state': 'ACTIVE'}
277 b578d9e7 Christos Stavrakakis
278 b578d9e7 Christos Stavrakakis
        new_nics.append(nic)
279 b578d9e7 Christos Stavrakakis
    return new_nics
280 b578d9e7 Christos Stavrakakis
281 b578d9e7 Christos Stavrakakis
282 b578d9e7 Christos Stavrakakis
def nics_changed(old_nics, new_nics):
283 b578d9e7 Christos Stavrakakis
    """Return True if NICs have changed in any way."""
284 b578d9e7 Christos Stavrakakis
    if len(old_nics) != len(new_nics):
285 b578d9e7 Christos Stavrakakis
        return True
286 be4d8aed Christos Stavrakakis
    fields = ["ipv4", "ipv6", "mac", "firewall_profile", "index", "network"]
287 b578d9e7 Christos Stavrakakis
    for old_nic, new_nic in zip(old_nics, new_nics):
288 be4d8aed Christos Stavrakakis
        for field in fields:
289 be4d8aed Christos Stavrakakis
            if getattr(old_nic, field) != new_nic[field]:
290 be4d8aed Christos Stavrakakis
                return True
291 b578d9e7 Christos Stavrakakis
    return False
292 22ee6892 Christos Stavrakakis
293 22ee6892 Christos Stavrakakis
294 72dea98f Christos Stavrakakis
def release_instance_ips(vm, ganeti_nics):
295 72dea98f Christos Stavrakakis
    old_addresses = set(vm.nics.values_list("network", "ipv4"))
296 f8675683 Christos Stavrakakis
    new_addresses = set(map(lambda nic: (nic["network"].id, nic["ipv4"]),
297 72dea98f Christos Stavrakakis
                            ganeti_nics))
298 72dea98f Christos Stavrakakis
    to_release = old_addresses - new_addresses
299 72dea98f Christos Stavrakakis
    for (network_id, ipv4) in to_release:
300 72dea98f Christos Stavrakakis
        if ipv4:
301 0c50c760 Christos Stavrakakis
            # Get X-Lock before searching floating IP, to exclusively search
302 0c50c760 Christos Stavrakakis
            # and release floating IP. Otherwise you may release a floating IP
303 0c50c760 Christos Stavrakakis
            # that has been just reserved.
304 0c50c760 Christos Stavrakakis
            net = Network.objects.select_for_update().get(id=network_id)
305 0c50c760 Christos Stavrakakis
            if net.floating_ip_pool:
306 0c50c760 Christos Stavrakakis
                try:
307 0c50c760 Christos Stavrakakis
                    floating_ip = net.floating_ips.select_for_update()\
308 0c50c760 Christos Stavrakakis
                                                  .get(ipv4=ipv4, machine=vm,
309 0c50c760 Christos Stavrakakis
                                                       deleted=False)
310 0c50c760 Christos Stavrakakis
                    floating_ip.machine = None
311 0c50c760 Christos Stavrakakis
                    floating_ip.save()
312 0c50c760 Christos Stavrakakis
                except FloatingIP.DoesNotExist:
313 0c50c760 Christos Stavrakakis
                    net.release_address(ipv4)
314 0c50c760 Christos Stavrakakis
            else:
315 72dea98f Christos Stavrakakis
                net.release_address(ipv4)
316 bd392934 Christos Stavrakakis
317 bd392934 Christos Stavrakakis
318 22ee6892 Christos Stavrakakis
@transaction.commit_on_success
319 22ee6892 Christos Stavrakakis
def process_network_status(back_network, etime, jobid, opcode, status, logmsg):
320 22ee6892 Christos Stavrakakis
    if status not in [x[0] for x in BACKEND_STATUSES]:
321 fd2bdbb2 Christos Stavrakakis
        raise Network.InvalidBackendMsgError(opcode, status)
322 22ee6892 Christos Stavrakakis
323 22ee6892 Christos Stavrakakis
    back_network.backendjobid = jobid
324 22ee6892 Christos Stavrakakis
    back_network.backendjobstatus = status
325 22ee6892 Christos Stavrakakis
    back_network.backendopcode = opcode
326 22ee6892 Christos Stavrakakis
    back_network.backendlogmsg = logmsg
327 22ee6892 Christos Stavrakakis
328 05146623 Christos Stavrakakis
    network = back_network.network
329 05146623 Christos Stavrakakis
330 22ee6892 Christos Stavrakakis
    # Notifications of success change the operating state
331 22ee6892 Christos Stavrakakis
    state_for_success = BackendNetwork.OPER_STATE_FROM_OPCODE.get(opcode, None)
332 22ee6892 Christos Stavrakakis
    if status == 'success' and state_for_success is not None:
333 22ee6892 Christos Stavrakakis
        back_network.operstate = state_for_success
334 22ee6892 Christos Stavrakakis
335 e6f6627c Christos Stavrakakis
    if status in ('canceled', 'error') and opcode == 'OP_NETWORK_ADD':
336 5480daec Christos Stavrakakis
        back_network.operstate = 'ERROR'
337 e97288bc Christos Stavrakakis
        back_network.backendtime = etime
338 22ee6892 Christos Stavrakakis
339 e97288bc Christos Stavrakakis
    if opcode == 'OP_NETWORK_REMOVE':
340 05146623 Christos Stavrakakis
        if (status == 'success' or
341 05146623 Christos Stavrakakis
           (status == 'error' and (back_network.operstate == 'ERROR' or
342 05146623 Christos Stavrakakis
                                   network.action == 'DESTROY'))):
343 e97288bc Christos Stavrakakis
            back_network.operstate = state_for_success
344 e97288bc Christos Stavrakakis
            back_network.deleted = True
345 e97288bc Christos Stavrakakis
            back_network.backendtime = etime
346 22ee6892 Christos Stavrakakis
347 e97288bc Christos Stavrakakis
    if status == 'success':
348 e97288bc Christos Stavrakakis
        back_network.backendtime = etime
349 22ee6892 Christos Stavrakakis
    back_network.save()
350 4d5d0b9c Christos Stavrakakis
    # Also you must update the state of the Network!!
351 05146623 Christos Stavrakakis
    update_network_state(network)
352 fd2bdbb2 Christos Stavrakakis
353 fd2bdbb2 Christos Stavrakakis
354 2509ce17 Christos Stavrakakis
def update_network_state(network):
355 99af08a4 Christos Stavrakakis
    """Update the state of a Network based on BackendNetwork states.
356 cb4eee84 Christos Stavrakakis

357 99af08a4 Christos Stavrakakis
    Update the state of a Network based on the operstate of the networks in the
358 99af08a4 Christos Stavrakakis
    backends that network exists.
359 99af08a4 Christos Stavrakakis

360 99af08a4 Christos Stavrakakis
    The state of the network is:
361 99af08a4 Christos Stavrakakis
    * ACTIVE: If it is 'ACTIVE' in at least one backend.
362 99af08a4 Christos Stavrakakis
    * DELETED: If it is is 'DELETED' in all backends that have been created.
363 99af08a4 Christos Stavrakakis

364 99af08a4 Christos Stavrakakis
    This function also releases the resources (MAC prefix or Bridge) and the
365 99af08a4 Christos Stavrakakis
    quotas for the network.
366 99af08a4 Christos Stavrakakis

367 99af08a4 Christos Stavrakakis
    """
368 99af08a4 Christos Stavrakakis
    if network.deleted:
369 99af08a4 Christos Stavrakakis
        # Network has already been deleted. Just assert that state is also
370 99af08a4 Christos Stavrakakis
        # DELETED
371 99af08a4 Christos Stavrakakis
        if not network.state == "DELETED":
372 99af08a4 Christos Stavrakakis
            network.state = "DELETED"
373 99af08a4 Christos Stavrakakis
            network.save()
374 cb4eee84 Christos Stavrakakis
        return
375 cb4eee84 Christos Stavrakakis
376 99af08a4 Christos Stavrakakis
    backend_states = [s.operstate for s in network.backend_networks.all()]
377 27cda06b Christos Stavrakakis
    if not backend_states and network.action != "DESTROY":
378 99af08a4 Christos Stavrakakis
        if network.state != "ACTIVE":
379 99af08a4 Christos Stavrakakis
            network.state = "ACTIVE"
380 99af08a4 Christos Stavrakakis
            network.save()
381 99af08a4 Christos Stavrakakis
            return
382 99af08a4 Christos Stavrakakis
383 99af08a4 Christos Stavrakakis
    # Network is deleted when all BackendNetworks go to "DELETED" operstate
384 27cda06b Christos Stavrakakis
    deleted = reduce(lambda x, y: x == y and "DELETED", backend_states,
385 27cda06b Christos Stavrakakis
                     "DELETED")
386 99af08a4 Christos Stavrakakis
387 cb4eee84 Christos Stavrakakis
    # Release the resources on the deletion of the Network
388 99af08a4 Christos Stavrakakis
    if deleted:
389 cb4eee84 Christos Stavrakakis
        log.info("Network %r deleted. Releasing link %r mac_prefix %r",
390 cb4eee84 Christos Stavrakakis
                 network.id, network.mac_prefix, network.link)
391 cb4eee84 Christos Stavrakakis
        network.deleted = True
392 99af08a4 Christos Stavrakakis
        network.state = "DELETED"
393 b7d38981 Dimitris Aragiorgis
        if network.mac_prefix:
394 b7d38981 Dimitris Aragiorgis
            if network.FLAVORS[network.flavor]["mac_prefix"] == "pool":
395 b7d38981 Dimitris Aragiorgis
                release_resource(res_type="mac_prefix",
396 b7d38981 Dimitris Aragiorgis
                                 value=network.mac_prefix)
397 b7d38981 Dimitris Aragiorgis
        if network.link:
398 b7d38981 Dimitris Aragiorgis
            if network.FLAVORS[network.flavor]["link"] == "pool":
399 b7d38981 Dimitris Aragiorgis
                release_resource(res_type="bridge", value=network.link)
400 cb4eee84 Christos Stavrakakis
401 cb4eee84 Christos Stavrakakis
        # Issue commission
402 e18c1749 Christos Stavrakakis
        if network.userid:
403 2509ce17 Christos Stavrakakis
            quotas.issue_and_accept_commission(network, delete=True)
404 e18c1749 Christos Stavrakakis
        elif not network.public:
405 e18c1749 Christos Stavrakakis
            log.warning("Network %s does not have an owner!", network.id)
406 cb4eee84 Christos Stavrakakis
    network.save()
407 cb4eee84 Christos Stavrakakis
408 cb4eee84 Christos Stavrakakis
409 fd2bdbb2 Christos Stavrakakis
@transaction.commit_on_success
410 fd2bdbb2 Christos Stavrakakis
def process_network_modify(back_network, etime, jobid, opcode, status,
411 fd2bdbb2 Christos Stavrakakis
                           add_reserved_ips, remove_reserved_ips):
412 fd2bdbb2 Christos Stavrakakis
    assert (opcode == "OP_NETWORK_SET_PARAMS")
413 fd2bdbb2 Christos Stavrakakis
    if status not in [x[0] for x in BACKEND_STATUSES]:
414 fd2bdbb2 Christos Stavrakakis
        raise Network.InvalidBackendMsgError(opcode, status)
415 fd2bdbb2 Christos Stavrakakis
416 fd2bdbb2 Christos Stavrakakis
    back_network.backendjobid = jobid
417 fd2bdbb2 Christos Stavrakakis
    back_network.backendjobstatus = status
418 fd2bdbb2 Christos Stavrakakis
    back_network.opcode = opcode
419 fd2bdbb2 Christos Stavrakakis
420 fd2bdbb2 Christos Stavrakakis
    if add_reserved_ips or remove_reserved_ips:
421 fd2bdbb2 Christos Stavrakakis
        net = back_network.network
422 fd2bdbb2 Christos Stavrakakis
        pool = net.get_pool()
423 fd2bdbb2 Christos Stavrakakis
        if add_reserved_ips:
424 fd2bdbb2 Christos Stavrakakis
            for ip in add_reserved_ips:
425 fd2bdbb2 Christos Stavrakakis
                pool.reserve(ip, external=True)
426 fd2bdbb2 Christos Stavrakakis
        if remove_reserved_ips:
427 fd2bdbb2 Christos Stavrakakis
            for ip in remove_reserved_ips:
428 fd2bdbb2 Christos Stavrakakis
                pool.put(ip, external=True)
429 fd2bdbb2 Christos Stavrakakis
        pool.save()
430 fd2bdbb2 Christos Stavrakakis
431 fd2bdbb2 Christos Stavrakakis
    if status == 'success':
432 fd2bdbb2 Christos Stavrakakis
        back_network.backendtime = etime
433 fd2bdbb2 Christos Stavrakakis
    back_network.save()
434 ad2d6807 Vangelis Koukis
435 c25cc9ec Vangelis Koukis
436 9068cd85 Georgios Gousios
@transaction.commit_on_success
437 0827883e Nikos Skalkotos
def process_create_progress(vm, etime, progress):
438 9068cd85 Georgios Gousios
439 0827883e Nikos Skalkotos
    percentage = int(progress)
440 9068cd85 Georgios Gousios
441 af90d919 Vangelis Koukis
    # The percentage may exceed 100%, due to the way
442 0827883e Nikos Skalkotos
    # snf-image:copy-progress tracks bytes read by image handling processes
443 af90d919 Vangelis Koukis
    percentage = 100 if percentage > 100 else percentage
444 af90d919 Vangelis Koukis
    if percentage < 0:
445 af90d919 Vangelis Koukis
        raise ValueError("Percentage cannot be negative")
446 9068cd85 Georgios Gousios
447 7ec9558b Vangelis Koukis
    # FIXME: log a warning here, see #1033
448 7ec9558b Vangelis Koukis
#   if last_update > percentage:
449 7ec9558b Vangelis Koukis
#       raise ValueError("Build percentage should increase monotonically " \
450 7ec9558b Vangelis Koukis
#                        "(old = %d, new = %d)" % (last_update, percentage))
451 9068cd85 Georgios Gousios
452 c25cc9ec Vangelis Koukis
    # This assumes that no message of type 'ganeti-create-progress' is going to
453 c25cc9ec Vangelis Koukis
    # arrive once OP_INSTANCE_CREATE has succeeded for a Ganeti instance and
454 c25cc9ec Vangelis Koukis
    # the instance is STARTED.  What if the two messages are processed by two
455 c25cc9ec Vangelis Koukis
    # separate dispatcher threads, and the 'ganeti-op-status' message for
456 c25cc9ec Vangelis Koukis
    # successful creation gets processed before the 'ganeti-create-progress'
457 c25cc9ec Vangelis Koukis
    # message? [vkoukis]
458 c25cc9ec Vangelis Koukis
    #
459 c25cc9ec Vangelis Koukis
    #if not vm.operstate == 'BUILD':
460 c25cc9ec Vangelis Koukis
    #    raise VirtualMachine.IllegalState("VM is not in building state")
461 9068cd85 Georgios Gousios
462 c25cc9ec Vangelis Koukis
    vm.buildpercentage = percentage
463 c4e55622 Christos Stavrakakis
    vm.backendtime = etime
464 9068cd85 Georgios Gousios
    vm.save()
465 ad2d6807 Vangelis Koukis
466 c25cc9ec Vangelis Koukis
467 952b2a48 Christos Stavrakakis
@transaction.commit_on_success
468 7d43565f Kostas Papadimitriou
def create_instance_diagnostic(vm, message, source, level="DEBUG", etime=None,
469 cc92b70f Christos Stavrakakis
                               details=None):
470 7d43565f Kostas Papadimitriou
    """
471 7d43565f Kostas Papadimitriou
    Create virtual machine instance diagnostic entry.
472 7d43565f Kostas Papadimitriou

473 7d43565f Kostas Papadimitriou
    :param vm: VirtualMachine instance to create diagnostic for.
474 7d43565f Kostas Papadimitriou
    :param message: Diagnostic message.
475 7d43565f Kostas Papadimitriou
    :param source: Diagnostic source identifier (e.g. image-helper).
476 7d43565f Kostas Papadimitriou
    :param level: Diagnostic level (`DEBUG`, `INFO`, `WARNING`, `ERROR`).
477 7d43565f Kostas Papadimitriou
    :param etime: The time the message occured (if available).
478 7d43565f Kostas Papadimitriou
    :param details: Additional details or debug information.
479 7d43565f Kostas Papadimitriou
    """
480 7d43565f Kostas Papadimitriou
    VirtualMachineDiagnostic.objects.create_for_vm(vm, level, source=source,
481 cc92b70f Christos Stavrakakis
                                                   source_date=etime,
482 cc92b70f Christos Stavrakakis
                                                   message=message,
483 cc92b70f Christos Stavrakakis
                                                   details=details)
484 7d43565f Kostas Papadimitriou
485 7d43565f Kostas Papadimitriou
486 2c022086 Christos Stavrakakis
def create_instance(vm, nics, flavor, image):
487 3a9b3cde Giorgos Verigakis
    """`image` is a dictionary which should contain the keys:
488 3a9b3cde Giorgos Verigakis
            'backend_id', 'format' and 'metadata'
489 7f691719 Christos Stavrakakis

490 3a9b3cde Giorgos Verigakis
        metadata value should be a dictionary.
491 3a9b3cde Giorgos Verigakis
    """
492 864bed43 Christos Stavrakakis
493 1c382247 Vangelis Koukis
    # Handle arguments to CreateInstance() as a dictionary,
494 1c382247 Vangelis Koukis
    # initialize it based on a deployment-specific value.
495 1c382247 Vangelis Koukis
    # This enables the administrator to override deployment-specific
496 c25cc9ec Vangelis Koukis
    # arguments, such as the disk template to use, name of os provider
497 1c382247 Vangelis Koukis
    # and hypervisor-specific parameters at will (see Synnefo #785, #835).
498 1c382247 Vangelis Koukis
    #
499 bd87213f Christos Stavrakakis
    kw = vm.backend.get_create_params()
500 1c382247 Vangelis Koukis
    kw['mode'] = 'create'
501 924d8085 Christos Stavrakakis
    kw['name'] = vm.backend_vm_id
502 5949b704 Vangelis Koukis
    # Defined in settings.GANETI_CREATEINSTANCE_KWARGS
503 296682fe Kostas Papadimitriou
504 2a599282 Christos Stavrakakis
    kw['disk_template'] = flavor.disk_template
505 d7841399 Christos Stavrakakis
    kw['disks'] = [{"size": flavor.disk * 1024}]
506 006c6249 Christos Stavrakakis
    provider = flavor.disk_provider
507 296682fe Kostas Papadimitriou
    if provider:
508 296682fe Kostas Papadimitriou
        kw['disks'][0]['provider'] = provider
509 b2222a7f Christos Stavrakakis
        kw['disks'][0]['origin'] = flavor.disk_origin
510 296682fe Kostas Papadimitriou
511 cb66110b Christos Stavrakakis
    kw['nics'] = [{"network": nic.network.backend_id, "ip": nic.ipv4}
512 cb66110b Christos Stavrakakis
                  for nic in nics]
513 cb66110b Christos Stavrakakis
    backend = vm.backend
514 cb66110b Christos Stavrakakis
    depend_jobs = []
515 cb66110b Christos Stavrakakis
    for nic in nics:
516 cb66110b Christos Stavrakakis
        network = Network.objects.select_for_update().get(id=nic.network.id)
517 cb66110b Christos Stavrakakis
        bnet, created = BackendNetwork.objects.get_or_create(backend=backend,
518 cb66110b Christos Stavrakakis
                                                             network=network)
519 cb66110b Christos Stavrakakis
        if bnet.operstate != "ACTIVE":
520 cb66110b Christos Stavrakakis
            if network.public:
521 736b9244 Christos Stavrakakis
                msg = "Can not connect instance to network %s. Network is not"\
522 736b9244 Christos Stavrakakis
                      " ACTIVE in backend %s." % (network, backend)
523 736b9244 Christos Stavrakakis
                raise Exception(msg)
524 cb66110b Christos Stavrakakis
            else:
525 527c7a7b Christos Stavrakakis
                jobs = create_network(network, backend, connect=True)
526 527c7a7b Christos Stavrakakis
                if isinstance(jobs, list):
527 527c7a7b Christos Stavrakakis
                    depend_jobs.extend(jobs)
528 527c7a7b Christos Stavrakakis
                else:
529 527c7a7b Christos Stavrakakis
                    depend_jobs.append(jobs)
530 cb66110b Christos Stavrakakis
    kw["depends"] = [[job, ["success", "error", "canceled"]]
531 cb66110b Christos Stavrakakis
                     for job in depend_jobs]
532 cb66110b Christos Stavrakakis
533 bd87213f Christos Stavrakakis
    if vm.backend.use_hotplug():
534 341c818e Dimitris Aragiorgis
        kw['hotplug'] = True
535 5949b704 Vangelis Koukis
    # Defined in settings.GANETI_CREATEINSTANCE_KWARGS
536 1c382247 Vangelis Koukis
    # kw['os'] = settings.GANETI_OS_PROVIDER
537 1c382247 Vangelis Koukis
    kw['ip_check'] = False
538 1c382247 Vangelis Koukis
    kw['name_check'] = False
539 79b7dbb7 Christos Stavrakakis
540 1c382247 Vangelis Koukis
    # Do not specific a node explicitly, have
541 1c382247 Vangelis Koukis
    # Ganeti use an iallocator instead
542 f8d1eaf4 Kostas Papadimitriou
    #kw['pnode'] = rapi.GetNodes()[0]
543 79b7dbb7 Christos Stavrakakis
544 1c382247 Vangelis Koukis
    kw['dry_run'] = settings.TEST
545 41303ed0 Vangelis Koukis
546 2b1db26f Giorgos Verigakis
    kw['beparams'] = {
547 cc92b70f Christos Stavrakakis
        'auto_balance': True,
548 cc92b70f Christos Stavrakakis
        'vcpus': flavor.cpu,
549 cc92b70f Christos Stavrakakis
        'memory': flavor.ram}
550 79b7dbb7 Christos Stavrakakis
551 e3b5be49 Giorgos Verigakis
    kw['osparams'] = {
552 79b7dbb7 Christos Stavrakakis
        'config_url': vm.config_url,
553 79b7dbb7 Christos Stavrakakis
        # Store image id and format to Ganeti
554 2a599282 Christos Stavrakakis
        'img_id': image['backend_id'],
555 3a9b3cde Giorgos Verigakis
        'img_format': image['format']}
556 d1eaa651 Christos Stavrakakis
557 5949b704 Vangelis Koukis
    # Defined in settings.GANETI_CREATEINSTANCE_KWARGS
558 1c382247 Vangelis Koukis
    # kw['hvparams'] = dict(serial_console=False)
559 79b7dbb7 Christos Stavrakakis
560 6afeb85d Christos Stavrakakis
    log.debug("Creating instance %s", utils.hide_pass(kw))
561 3524241a Christos Stavrakakis
    with pooled_rapi_client(vm) as client:
562 3524241a Christos Stavrakakis
        return client.CreateInstance(**kw)
563 f533f224 Vangelis Koukis
564 529178b1 Giorgos Verigakis
565 529178b1 Giorgos Verigakis
def delete_instance(vm):
566 3524241a Christos Stavrakakis
    with pooled_rapi_client(vm) as client:
567 3524241a Christos Stavrakakis
        return client.DeleteInstance(vm.backend_vm_id, dry_run=settings.TEST)
568 529178b1 Giorgos Verigakis
569 ad2d6807 Vangelis Koukis
570 529178b1 Giorgos Verigakis
def reboot_instance(vm, reboot_type):
571 529178b1 Giorgos Verigakis
    assert reboot_type in ('soft', 'hard')
572 3524241a Christos Stavrakakis
    with pooled_rapi_client(vm) as client:
573 bf5c82dc Christos Stavrakakis
        return client.RebootInstance(vm.backend_vm_id, reboot_type,
574 bf5c82dc Christos Stavrakakis
                                     dry_run=settings.TEST)
575 529178b1 Giorgos Verigakis
576 ad2d6807 Vangelis Koukis
577 529178b1 Giorgos Verigakis
def startup_instance(vm):
578 3524241a Christos Stavrakakis
    with pooled_rapi_client(vm) as client:
579 3524241a Christos Stavrakakis
        return client.StartupInstance(vm.backend_vm_id, dry_run=settings.TEST)
580 529178b1 Giorgos Verigakis
581 ad2d6807 Vangelis Koukis
582 529178b1 Giorgos Verigakis
def shutdown_instance(vm):
583 3524241a Christos Stavrakakis
    with pooled_rapi_client(vm) as client:
584 3524241a Christos Stavrakakis
        return client.ShutdownInstance(vm.backend_vm_id, dry_run=settings.TEST)
585 529178b1 Giorgos Verigakis
586 ad2d6807 Vangelis Koukis
587 2cd3f389 Christos Stavrakakis
def resize_instance(vm, vcpus, memory):
588 2cd3f389 Christos Stavrakakis
    beparams = {"vcpus": int(vcpus),
589 2cd3f389 Christos Stavrakakis
                "minmem": int(memory),
590 2cd3f389 Christos Stavrakakis
                "maxmem": int(memory)}
591 2cd3f389 Christos Stavrakakis
    with pooled_rapi_client(vm) as client:
592 2cd3f389 Christos Stavrakakis
        return client.ModifyInstance(vm.backend_vm_id, beparams=beparams)
593 2cd3f389 Christos Stavrakakis
594 2cd3f389 Christos Stavrakakis
595 529178b1 Giorgos Verigakis
def get_instance_console(vm):
596 71099804 Vangelis Koukis
    # RAPI GetInstanceConsole() returns endpoints to the vnc_bind_address,
597 71099804 Vangelis Koukis
    # which is a cluster-wide setting, either 0.0.0.0 or 127.0.0.1, and pretty
598 71099804 Vangelis Koukis
    # useless (see #783).
599 71099804 Vangelis Koukis
    #
600 71099804 Vangelis Koukis
    # Until this is fixed on the Ganeti side, construct a console info reply
601 71099804 Vangelis Koukis
    # directly.
602 9afeb669 Kostas Papadimitriou
    #
603 71099804 Vangelis Koukis
    # WARNING: This assumes that VNC runs on port network_port on
604 71099804 Vangelis Koukis
    #          the instance's primary node, and is probably
605 71099804 Vangelis Koukis
    #          hypervisor-specific.
606 71099804 Vangelis Koukis
    #
607 bf5c82dc Christos Stavrakakis
    log.debug("Getting console for vm %s", vm)
608 bf5c82dc Christos Stavrakakis
609 71099804 Vangelis Koukis
    console = {}
610 71099804 Vangelis Koukis
    console['kind'] = 'vnc'
611 3524241a Christos Stavrakakis
612 3524241a Christos Stavrakakis
    with pooled_rapi_client(vm) as client:
613 3524241a Christos Stavrakakis
        i = client.GetInstance(vm.backend_vm_id)
614 3524241a Christos Stavrakakis
615 bd87213f Christos Stavrakakis
    if vm.backend.hypervisor == "kvm" and i['hvparams']['serial_console']:
616 71099804 Vangelis Koukis
        raise Exception("hv parameter serial_console cannot be true")
617 71099804 Vangelis Koukis
    console['host'] = i['pnode']
618 71099804 Vangelis Koukis
    console['port'] = i['network_port']
619 9afeb669 Kostas Papadimitriou
620 71099804 Vangelis Koukis
    return console
621 604b2bf8 Georgios Gousios
622 604b2bf8 Georgios Gousios
623 3524241a Christos Stavrakakis
def get_instance_info(vm):
624 3524241a Christos Stavrakakis
    with pooled_rapi_client(vm) as client:
625 3524241a Christos Stavrakakis
        return client.GetInstanceInfo(vm.backend_vm_id)
626 f533f224 Vangelis Koukis
627 c25cc9ec Vangelis Koukis
628 99af08a4 Christos Stavrakakis
def create_network(network, backend, connect=True):
629 99af08a4 Christos Stavrakakis
    """Create a network in a Ganeti backend"""
630 99af08a4 Christos Stavrakakis
    log.debug("Creating network %s in backend %s", network, backend)
631 64938cb0 Giorgos Verigakis
632 99af08a4 Christos Stavrakakis
    job_id = _create_network(network, backend)
633 c25cc9ec Vangelis Koukis
634 99af08a4 Christos Stavrakakis
    if connect:
635 99af08a4 Christos Stavrakakis
        job_ids = connect_network(network, backend, depends=[job_id])
636 99af08a4 Christos Stavrakakis
        return job_ids
637 99af08a4 Christos Stavrakakis
    else:
638 99af08a4 Christos Stavrakakis
        return [job_id]
639 37ca953f Christodoulos Psaltis
640 64938cb0 Giorgos Verigakis
641 3524241a Christos Stavrakakis
def _create_network(network, backend):
642 3524241a Christos Stavrakakis
    """Create a network."""
643 c25cc9ec Vangelis Koukis
644 22ee6892 Christos Stavrakakis
    network_type = network.public and 'public' or 'private'
645 22ee6892 Christos Stavrakakis
646 22ee6892 Christos Stavrakakis
    tags = network.backend_tag
647 22ee6892 Christos Stavrakakis
    if network.dhcp:
648 22ee6892 Christos Stavrakakis
        tags.append('nfdhcpd')
649 2d762302 Dimitris Aragiorgis
650 2d762302 Dimitris Aragiorgis
    if network.public:
651 2d762302 Dimitris Aragiorgis
        conflicts_check = True
652 2d762302 Dimitris Aragiorgis
    else:
653 2d762302 Dimitris Aragiorgis
        conflicts_check = False
654 22ee6892 Christos Stavrakakis
655 5aeb4e93 Christos Stavrakakis
    # Use a dummy network subnet for IPv6 only networks. Currently Ganeti does
656 5aeb4e93 Christos Stavrakakis
    # not support IPv6 only networks. To bypass this limitation, we create the
657 5aeb4e93 Christos Stavrakakis
    # network with a dummy network subnet, and make Cyclades connect instances
658 5aeb4e93 Christos Stavrakakis
    # to such networks, with address=None.
659 5aeb4e93 Christos Stavrakakis
    subnet = network.subnet
660 5aeb4e93 Christos Stavrakakis
    if subnet is None:
661 5aeb4e93 Christos Stavrakakis
        subnet = "10.0.0.0/24"
662 5aeb4e93 Christos Stavrakakis
663 3524241a Christos Stavrakakis
    try:
664 3524241a Christos Stavrakakis
        bn = BackendNetwork.objects.get(network=network, backend=backend)
665 3524241a Christos Stavrakakis
        mac_prefix = bn.mac_prefix
666 3524241a Christos Stavrakakis
    except BackendNetwork.DoesNotExist:
667 cc92b70f Christos Stavrakakis
        raise Exception("BackendNetwork for network '%s' in backend '%s'"
668 3524241a Christos Stavrakakis
                        " does not exist" % (network.id, backend.id))
669 3524241a Christos Stavrakakis
670 3524241a Christos Stavrakakis
    with pooled_rapi_client(backend) as client:
671 3524241a Christos Stavrakakis
        return client.CreateNetwork(network_name=network.backend_id,
672 5aeb4e93 Christos Stavrakakis
                                    network=subnet,
673 6cc3a31c Christos Stavrakakis
                                    network6=network.subnet6,
674 3524241a Christos Stavrakakis
                                    gateway=network.gateway,
675 6cc3a31c Christos Stavrakakis
                                    gateway6=network.gateway6,
676 3524241a Christos Stavrakakis
                                    network_type=network_type,
677 3524241a Christos Stavrakakis
                                    mac_prefix=mac_prefix,
678 2d762302 Dimitris Aragiorgis
                                    conflicts_check=conflicts_check,
679 3524241a Christos Stavrakakis
                                    tags=tags)
680 3524241a Christos Stavrakakis
681 3524241a Christos Stavrakakis
682 99af08a4 Christos Stavrakakis
def connect_network(network, backend, depends=[], group=None):
683 3524241a Christos Stavrakakis
    """Connect a network to nodegroups."""
684 bf5c82dc Christos Stavrakakis
    log.debug("Connecting network %s to backend %s", network, backend)
685 bf5c82dc Christos Stavrakakis
686 2d762302 Dimitris Aragiorgis
    if network.public:
687 2d762302 Dimitris Aragiorgis
        conflicts_check = True
688 2d762302 Dimitris Aragiorgis
    else:
689 2d762302 Dimitris Aragiorgis
        conflicts_check = False
690 2d762302 Dimitris Aragiorgis
691 99af08a4 Christos Stavrakakis
    depends = [[job, ["success", "error", "canceled"]] for job in depends]
692 3524241a Christos Stavrakakis
    with pooled_rapi_client(backend) as client:
693 99af08a4 Christos Stavrakakis
        groups = [group] if group is not None else client.GetGroups()
694 99af08a4 Christos Stavrakakis
        job_ids = []
695 99af08a4 Christos Stavrakakis
        for group in groups:
696 99af08a4 Christos Stavrakakis
            job_id = client.ConnectNetwork(network.backend_id, group,
697 99af08a4 Christos Stavrakakis
                                           network.mode, network.link,
698 99af08a4 Christos Stavrakakis
                                           conflicts_check,
699 99af08a4 Christos Stavrakakis
                                           depends=depends)
700 99af08a4 Christos Stavrakakis
            job_ids.append(job_id)
701 99af08a4 Christos Stavrakakis
    return job_ids
702 22ee6892 Christos Stavrakakis
703 22ee6892 Christos Stavrakakis
704 99af08a4 Christos Stavrakakis
def delete_network(network, backend, disconnect=True):
705 99af08a4 Christos Stavrakakis
    log.debug("Deleting network %s from backend %s", network, backend)
706 22ee6892 Christos Stavrakakis
707 99af08a4 Christos Stavrakakis
    depends = []
708 99af08a4 Christos Stavrakakis
    if disconnect:
709 99af08a4 Christos Stavrakakis
        depends = disconnect_network(network, backend)
710 99af08a4 Christos Stavrakakis
    _delete_network(network, backend, depends=depends)
711 bf5c82dc Christos Stavrakakis
712 22ee6892 Christos Stavrakakis
713 99af08a4 Christos Stavrakakis
def _delete_network(network, backend, depends=[]):
714 99af08a4 Christos Stavrakakis
    depends = [[job, ["success", "error", "canceled"]] for job in depends]
715 3524241a Christos Stavrakakis
    with pooled_rapi_client(backend) as client:
716 99af08a4 Christos Stavrakakis
        return client.DeleteNetwork(network.backend_id, depends)
717 22ee6892 Christos Stavrakakis
718 22ee6892 Christos Stavrakakis
719 3524241a Christos Stavrakakis
def disconnect_network(network, backend, group=None):
720 bf5c82dc Christos Stavrakakis
    log.debug("Disconnecting network %s to backend %s", network, backend)
721 22ee6892 Christos Stavrakakis
722 3524241a Christos Stavrakakis
    with pooled_rapi_client(backend) as client:
723 99af08a4 Christos Stavrakakis
        groups = [group] if group is not None else client.GetGroups()
724 99af08a4 Christos Stavrakakis
        job_ids = []
725 99af08a4 Christos Stavrakakis
        for group in groups:
726 99af08a4 Christos Stavrakakis
            job_id = client.DisconnectNetwork(network.backend_id, group)
727 99af08a4 Christos Stavrakakis
            job_ids.append(job_id)
728 99af08a4 Christos Stavrakakis
    return job_ids
729 36f4cb29 Christos Stavrakakis
730 36f4cb29 Christos Stavrakakis
731 87920bc3 Christos Stavrakakis
def connect_to_network(vm, network, address=None):
732 99af08a4 Christos Stavrakakis
    backend = vm.backend
733 acfc71ef Christos Stavrakakis
    network = Network.objects.select_for_update().get(id=network.id)
734 99af08a4 Christos Stavrakakis
    bnet, created = BackendNetwork.objects.get_or_create(backend=backend,
735 99af08a4 Christos Stavrakakis
                                                         network=network)
736 99af08a4 Christos Stavrakakis
    depend_jobs = []
737 99af08a4 Christos Stavrakakis
    if bnet.operstate != "ACTIVE":
738 99af08a4 Christos Stavrakakis
        depend_jobs = create_network(network, backend, connect=True)
739 99af08a4 Christos Stavrakakis
740 99af08a4 Christos Stavrakakis
    depends = [[job, ["success", "error", "canceled"]] for job in depend_jobs]
741 99af08a4 Christos Stavrakakis
742 3524241a Christos Stavrakakis
    nic = {'ip': address, 'network': network.backend_id}
743 22ee6892 Christos Stavrakakis
744 bf5c82dc Christos Stavrakakis
    log.debug("Connecting vm %s to network %s(%s)", vm, network, address)
745 22ee6892 Christos Stavrakakis
746 3524241a Christos Stavrakakis
    with pooled_rapi_client(vm) as client:
747 3524241a Christos Stavrakakis
        return client.ModifyInstance(vm.backend_vm_id, nics=[('add',  nic)],
748 bd87213f Christos Stavrakakis
                                     hotplug=vm.backend.use_hotplug(),
749 99af08a4 Christos Stavrakakis
                                     depends=depends,
750 3524241a Christos Stavrakakis
                                     dry_run=settings.TEST)
751 22ee6892 Christos Stavrakakis
752 22ee6892 Christos Stavrakakis
753 3524241a Christos Stavrakakis
def disconnect_from_network(vm, nic):
754 3524241a Christos Stavrakakis
    op = [('remove', nic.index, {})]
755 22ee6892 Christos Stavrakakis
756 bf5c82dc Christos Stavrakakis
    log.debug("Removing nic of VM %s, with index %s", vm, str(nic.index))
757 22ee6892 Christos Stavrakakis
758 3524241a Christos Stavrakakis
    with pooled_rapi_client(vm) as client:
759 3524241a Christos Stavrakakis
        return client.ModifyInstance(vm.backend_vm_id, nics=op,
760 bd87213f Christos Stavrakakis
                                     hotplug=vm.backend.use_hotplug(),
761 3524241a Christos Stavrakakis
                                     dry_run=settings.TEST)
762 91826390 Giorgos Verigakis
763 c25cc9ec Vangelis Koukis
764 b2791a77 Christos Stavrakakis
def set_firewall_profile(vm, profile, index=0):
765 26563957 Giorgos Verigakis
    try:
766 b2791a77 Christos Stavrakakis
        tag = _firewall_tags[profile] % index
767 26563957 Giorgos Verigakis
    except KeyError:
768 91826390 Giorgos Verigakis
        raise ValueError("Unsopported Firewall Profile: %s" % profile)
769 37ca953f Christodoulos Psaltis
770 b2791a77 Christos Stavrakakis
    log.debug("Setting tag of VM %s, NIC index %d, to %s", vm, index, profile)
771 bf5c82dc Christos Stavrakakis
772 3524241a Christos Stavrakakis
    with pooled_rapi_client(vm) as client:
773 b2791a77 Christos Stavrakakis
        # Delete previous firewall tags
774 b2791a77 Christos Stavrakakis
        old_tags = client.GetInstanceTags(vm.backend_vm_id)
775 b2791a77 Christos Stavrakakis
        delete_tags = [(t % index) for t in _firewall_tags.values()
776 b2791a77 Christos Stavrakakis
                       if (t % index) in old_tags]
777 b2791a77 Christos Stavrakakis
        if delete_tags:
778 b2791a77 Christos Stavrakakis
            client.DeleteInstanceTags(vm.backend_vm_id, delete_tags,
779 3524241a Christos Stavrakakis
                                      dry_run=settings.TEST)
780 37ca953f Christodoulos Psaltis
781 3524241a Christos Stavrakakis
        client.AddInstanceTags(vm.backend_vm_id, [tag], dry_run=settings.TEST)
782 9afeb669 Kostas Papadimitriou
783 3524241a Christos Stavrakakis
        # XXX NOP ModifyInstance call to force process_net_status to run
784 3524241a Christos Stavrakakis
        # on the dispatcher
785 cc92b70f Christos Stavrakakis
        os_name = settings.GANETI_CREATEINSTANCE_KWARGS['os']
786 3524241a Christos Stavrakakis
        client.ModifyInstance(vm.backend_vm_id,
787 cc92b70f Christos Stavrakakis
                              os_name=os_name)
788 41a7fae7 Christos Stavrakakis
    return None
789 5eedb0e4 Vangelis Koukis
790 41303ed0 Vangelis Koukis
791 e77a29ab Christos Stavrakakis
def get_instances(backend, bulk=True):
792 e77a29ab Christos Stavrakakis
    with pooled_rapi_client(backend) as c:
793 e77a29ab Christos Stavrakakis
        return c.GetInstances(bulk=bulk)
794 d986cb32 Christos Stavrakakis
795 f5b4f2a3 Christos Stavrakakis
796 e77a29ab Christos Stavrakakis
def get_nodes(backend, bulk=True):
797 e77a29ab Christos Stavrakakis
    with pooled_rapi_client(backend) as c:
798 e77a29ab Christos Stavrakakis
        return c.GetNodes(bulk=bulk)
799 1a894bfe Christos Stavrakakis
800 1a894bfe Christos Stavrakakis
801 e77a29ab Christos Stavrakakis
def get_jobs(backend):
802 e77a29ab Christos Stavrakakis
    with pooled_rapi_client(backend) as c:
803 e77a29ab Christos Stavrakakis
        return c.GetJobs()
804 f5b4f2a3 Christos Stavrakakis
805 17852fe9 Giorgos Verigakis
806 1a894bfe Christos Stavrakakis
def get_physical_resources(backend):
807 1a894bfe Christos Stavrakakis
    """ Get the physical resources of a backend.
808 1a894bfe Christos Stavrakakis

809 1a894bfe Christos Stavrakakis
    Get the resources of a backend as reported by the backend (not the db).
810 41303ed0 Vangelis Koukis

811 1a894bfe Christos Stavrakakis
    """
812 e77a29ab Christos Stavrakakis
    nodes = get_nodes(backend, bulk=True)
813 1a894bfe Christos Stavrakakis
    attr = ['mfree', 'mtotal', 'dfree', 'dtotal', 'pinst_cnt', 'ctotal']
814 1a894bfe Christos Stavrakakis
    res = {}
815 1a894bfe Christos Stavrakakis
    for a in attr:
816 1a894bfe Christos Stavrakakis
        res[a] = 0
817 1a894bfe Christos Stavrakakis
    for n in nodes:
818 1a894bfe Christos Stavrakakis
        # Filter out drained, offline and not vm_capable nodes since they will
819 1a894bfe Christos Stavrakakis
        # not take part in the vm allocation process
820 cc92b70f Christos Stavrakakis
        can_host_vms = n['vm_capable'] and not (n['drained'] or n['offline'])
821 cc92b70f Christos Stavrakakis
        if can_host_vms and n['cnodes']:
822 1a894bfe Christos Stavrakakis
            for a in attr:
823 1a894bfe Christos Stavrakakis
                res[a] += int(n[a])
824 1a894bfe Christos Stavrakakis
    return res
825 1a894bfe Christos Stavrakakis
826 1a894bfe Christos Stavrakakis
827 1a894bfe Christos Stavrakakis
def update_resources(backend, resources=None):
828 1a894bfe Christos Stavrakakis
    """ Update the state of the backend resources in db.
829 1a894bfe Christos Stavrakakis

830 1a894bfe Christos Stavrakakis
    """
831 17852fe9 Giorgos Verigakis
832 1a894bfe Christos Stavrakakis
    if not resources:
833 1a894bfe Christos Stavrakakis
        resources = get_physical_resources(backend)
834 41303ed0 Vangelis Koukis
835 1a894bfe Christos Stavrakakis
    backend.mfree = resources['mfree']
836 1a894bfe Christos Stavrakakis
    backend.mtotal = resources['mtotal']
837 1a894bfe Christos Stavrakakis
    backend.dfree = resources['dfree']
838 1a894bfe Christos Stavrakakis
    backend.dtotal = resources['dtotal']
839 1a894bfe Christos Stavrakakis
    backend.pinst_cnt = resources['pinst_cnt']
840 1a894bfe Christos Stavrakakis
    backend.ctotal = resources['ctotal']
841 1a894bfe Christos Stavrakakis
    backend.updated = datetime.now()
842 1a894bfe Christos Stavrakakis
    backend.save()
843 1a894bfe Christos Stavrakakis
844 1a894bfe Christos Stavrakakis
845 1a894bfe Christos Stavrakakis
def get_memory_from_instances(backend):
846 1a894bfe Christos Stavrakakis
    """ Get the memory that is used from instances.
847 1a894bfe Christos Stavrakakis

848 1a894bfe Christos Stavrakakis
    Get the used memory of a backend. Note: This is different for
849 1a894bfe Christos Stavrakakis
    the real memory used, due to kvm's memory de-duplication.
850 1a894bfe Christos Stavrakakis

851 1a894bfe Christos Stavrakakis
    """
852 3524241a Christos Stavrakakis
    with pooled_rapi_client(backend) as client:
853 3524241a Christos Stavrakakis
        instances = client.GetInstances(bulk=True)
854 1a894bfe Christos Stavrakakis
    mem = 0
855 1a894bfe Christos Stavrakakis
    for i in instances:
856 1a894bfe Christos Stavrakakis
        mem += i['oper_ram']
857 1a894bfe Christos Stavrakakis
    return mem
858 b3d28af2 Christos Stavrakakis
859 b3d28af2 Christos Stavrakakis
##
860 b3d28af2 Christos Stavrakakis
## Synchronized operations for reconciliation
861 b3d28af2 Christos Stavrakakis
##
862 b3d28af2 Christos Stavrakakis
863 b3d28af2 Christos Stavrakakis
864 b3d28af2 Christos Stavrakakis
def create_network_synced(network, backend):
865 b3d28af2 Christos Stavrakakis
    result = _create_network_synced(network, backend)
866 b3d28af2 Christos Stavrakakis
    if result[0] != 'success':
867 b3d28af2 Christos Stavrakakis
        return result
868 b3d28af2 Christos Stavrakakis
    result = connect_network_synced(network, backend)
869 b3d28af2 Christos Stavrakakis
    return result
870 b3d28af2 Christos Stavrakakis
871 b3d28af2 Christos Stavrakakis
872 b3d28af2 Christos Stavrakakis
def _create_network_synced(network, backend):
873 3524241a Christos Stavrakakis
    with pooled_rapi_client(backend) as client:
874 e60b4630 Christos Stavrakakis
        job = _create_network(network, backend)
875 3524241a Christos Stavrakakis
        result = wait_for_job(client, job)
876 3524241a Christos Stavrakakis
    return result
877 b3d28af2 Christos Stavrakakis
878 b3d28af2 Christos Stavrakakis
879 b3d28af2 Christos Stavrakakis
def connect_network_synced(network, backend):
880 3524241a Christos Stavrakakis
    with pooled_rapi_client(backend) as client:
881 3524241a Christos Stavrakakis
        for group in client.GetGroups():
882 b7d38981 Dimitris Aragiorgis
            job = client.ConnectNetwork(network.backend_id, group,
883 b7d38981 Dimitris Aragiorgis
                                        network.mode, network.link)
884 3524241a Christos Stavrakakis
            result = wait_for_job(client, job)
885 3524241a Christos Stavrakakis
            if result[0] != 'success':
886 3524241a Christos Stavrakakis
                return result
887 b3d28af2 Christos Stavrakakis
888 b3d28af2 Christos Stavrakakis
    return result
889 b3d28af2 Christos Stavrakakis
890 b3d28af2 Christos Stavrakakis
891 b3d28af2 Christos Stavrakakis
def wait_for_job(client, jobid):
892 b3d28af2 Christos Stavrakakis
    result = client.WaitForJobChange(jobid, ['status', 'opresult'], None, None)
893 b3d28af2 Christos Stavrakakis
    status = result['job_info'][0]
894 b3d28af2 Christos Stavrakakis
    while status not in ['success', 'error', 'cancel']:
895 b3d28af2 Christos Stavrakakis
        result = client.WaitForJobChange(jobid, ['status', 'opresult'],
896 cc92b70f Christos Stavrakakis
                                         [result], None)
897 b3d28af2 Christos Stavrakakis
        status = result['job_info'][0]
898 b3d28af2 Christos Stavrakakis
899 b3d28af2 Christos Stavrakakis
    if status == 'success':
900 b3d28af2 Christos Stavrakakis
        return (status, None)
901 b3d28af2 Christos Stavrakakis
    else:
902 b3d28af2 Christos Stavrakakis
        error = result['job_info'][1]
903 b3d28af2 Christos Stavrakakis
        return (status, error)