Statistics
| Branch: | Tag: | Revision:

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

History | View | Annotate | Download (32.9 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 9122ffab Christos Stavrakakis
        commission_name = ("client: dispatcher, resource: %s, ganeti_job: %s"
100 9122ffab Christos Stavrakakis
                           % (vm, job_id))
101 41a7fae7 Christos Stavrakakis
        serial = quotas.issue_commission(user=vm.userid,
102 41a7fae7 Christos Stavrakakis
                                         source=quotas.DEFAULT_SOURCE,
103 41a7fae7 Christos Stavrakakis
                                         provisions=commission_info,
104 9122ffab Christos Stavrakakis
                                         name=commission_name,
105 41a7fae7 Christos Stavrakakis
                                         force=True,
106 41a7fae7 Christos Stavrakakis
                                         auto_accept=True)
107 41a7fae7 Christos Stavrakakis
        # Clear VM's serial. Expected job may arrive later. However correlated
108 41a7fae7 Christos Stavrakakis
        # serial must not be accepted, since it reflects a previous VM state
109 41a7fae7 Christos Stavrakakis
        vm.serial = None
110 41a7fae7 Christos Stavrakakis
111 41a7fae7 Christos Stavrakakis
    return vm
112 41a7fae7 Christos Stavrakakis
113 41a7fae7 Christos Stavrakakis
114 093f9c53 Vangelis Koukis
@transaction.commit_on_success
115 ca4d59e3 Christos Stavrakakis
def process_op_status(vm, etime, jobid, opcode, status, logmsg, nics=None,
116 ca4d59e3 Christos Stavrakakis
                      beparams=None):
117 ad2d6807 Vangelis Koukis
    """Process a job progress notification from the backend
118 02feca11 Vassilios Karakoidas

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

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

216 ad2d6807 Vangelis Koukis
    Process an incoming message from the Ganeti backend,
217 ad2d6807 Vangelis Koukis
    detailing the NIC configuration of a VM instance.
218 ad2d6807 Vangelis Koukis

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

360 99af08a4 Christos Stavrakakis
    Update the state of a Network based on the operstate of the networks in the
361 99af08a4 Christos Stavrakakis
    backends that network exists.
362 99af08a4 Christos Stavrakakis

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

367 99af08a4 Christos Stavrakakis
    This function also releases the resources (MAC prefix or Bridge) and the
368 99af08a4 Christos Stavrakakis
    quotas for the network.
369 99af08a4 Christos Stavrakakis

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

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

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

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

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

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

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

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