Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / logic / reconciliation.py @ a9bb2a1a

History | View | Annotate | Download (29.6 kB)

1 9fea53cc Vangelis Koukis
# -*- coding: utf-8 -*-
2 9fea53cc Vangelis Koukis
#
3 75dc539e Christos Stavrakakis
# Copyright 2011-2013 GRNET S.A. All rights reserved.
4 9fea53cc Vangelis Koukis
#
5 9fea53cc Vangelis Koukis
# Redistribution and use in source and binary forms, with or
6 9fea53cc Vangelis Koukis
# without modification, are permitted provided that the following
7 9fea53cc Vangelis Koukis
# conditions are met:
8 9fea53cc Vangelis Koukis
#
9 9fea53cc Vangelis Koukis
#   1. Redistributions of source code must retain the above
10 9fea53cc Vangelis Koukis
#      copyright notice, this list of conditions and the following
11 9fea53cc Vangelis Koukis
#      disclaimer.
12 9fea53cc Vangelis Koukis
#
13 9fea53cc Vangelis Koukis
#   2. Redistributions in binary form must reproduce the above
14 9fea53cc Vangelis Koukis
#      copyright notice, this list of conditions and the following
15 9fea53cc Vangelis Koukis
#      disclaimer in the documentation and/or other materials
16 9fea53cc Vangelis Koukis
#      provided with the distribution.
17 9fea53cc Vangelis Koukis
#
18 9fea53cc Vangelis Koukis
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
19 9fea53cc Vangelis Koukis
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20 9fea53cc Vangelis Koukis
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
21 9fea53cc Vangelis Koukis
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
22 9fea53cc Vangelis Koukis
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23 9fea53cc Vangelis Koukis
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24 9fea53cc Vangelis Koukis
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
25 9fea53cc Vangelis Koukis
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
26 9fea53cc Vangelis Koukis
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 9fea53cc Vangelis Koukis
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
28 9fea53cc Vangelis Koukis
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 9fea53cc Vangelis Koukis
# POSSIBILITY OF SUCH DAMAGE.
30 9fea53cc Vangelis Koukis
#
31 9fea53cc Vangelis Koukis
# The views and conclusions contained in the software and
32 9fea53cc Vangelis Koukis
# documentation are those of the authors and should not be
33 9fea53cc Vangelis Koukis
# interpreted as representing official policies, either expressed
34 9fea53cc Vangelis Koukis
# or implied, of GRNET S.A.
35 9fea53cc Vangelis Koukis
#
36 9fea53cc Vangelis Koukis
"""Business logic for reconciliation
37 9fea53cc Vangelis Koukis

38 9fea53cc Vangelis Koukis
Reconcile the contents of the DB with the actual state of the
39 9fea53cc Vangelis Koukis
Ganeti backend.
40 9fea53cc Vangelis Koukis

41 9fea53cc Vangelis Koukis
Let D be the set of VMs in the DB, G the set of VMs in Ganeti.
42 9fea53cc Vangelis Koukis
RULES:
43 9fea53cc Vangelis Koukis
    R1. Stale servers in DB:
44 9fea53cc Vangelis Koukis
            For any v in D but not in G:
45 9fea53cc Vangelis Koukis
            Set deleted=True.
46 9fea53cc Vangelis Koukis
    R2. Orphan instances in Ganet:
47 9fea53cc Vangelis Koukis
            For any v in G with deleted=True in D:
48 9fea53cc Vangelis Koukis
            Issue OP_INSTANCE_DESTROY.
49 9fea53cc Vangelis Koukis
    R3. Unsynced operstate:
50 9fea53cc Vangelis Koukis
            For any v whose operating state differs between G and V:
51 9fea53cc Vangelis Koukis
            Set the operating state in D based on the state in G.
52 9fea53cc Vangelis Koukis
In the code, D, G are Python dicts mapping instance ids to operating state.
53 9fea53cc Vangelis Koukis
For D, the operating state is chosen from VirtualMachine.OPER_STATES.
54 9fea53cc Vangelis Koukis
For G, the operating state is True if the machine is up, False otherwise.
55 9fea53cc Vangelis Koukis

56 9fea53cc Vangelis Koukis
"""
57 9fea53cc Vangelis Koukis
58 9fea53cc Vangelis Koukis
59 0c09b1c0 Christos Stavrakakis
from django.conf import settings
60 4161cb41 Christos Stavrakakis
61 75dc539e Christos Stavrakakis
import logging
62 75dc539e Christos Stavrakakis
import itertools
63 89b2b908 Christos Stavrakakis
import bitarray
64 63f9eb8e Christos Stavrakakis
from datetime import datetime
65 4161cb41 Christos Stavrakakis
66 75dc539e Christos Stavrakakis
from django.db import transaction
67 75dc539e Christos Stavrakakis
from synnefo.db.models import (Backend, VirtualMachine, Flavor,
68 89b2b908 Christos Stavrakakis
                               pooled_rapi_client, Network,
69 0ccb6461 Christos Stavrakakis
                               BackendNetwork, BridgePoolTable,
70 0ccb6461 Christos Stavrakakis
                               MacPrefixPoolTable)
71 0ccb6461 Christos Stavrakakis
from synnefo.db import pools
72 75dc539e Christos Stavrakakis
from synnefo.logic import utils, backend as backend_mod
73 9fea53cc Vangelis Koukis
74 75dc539e Christos Stavrakakis
logger = logging.getLogger()
75 75dc539e Christos Stavrakakis
logging.basicConfig()
76 9e98ba3c Giorgos Verigakis
77 e63050ca Christos Stavrakakis
try:
78 e63050ca Christos Stavrakakis
    CHECK_INTERVAL = settings.RECONCILIATION_CHECK_INTERVAL
79 e63050ca Christos Stavrakakis
except AttributeError:
80 e63050ca Christos Stavrakakis
    CHECK_INTERVAL = 60
81 e63050ca Christos Stavrakakis
82 63f9eb8e Christos Stavrakakis
GANETI_JOB_ERROR = "error"
83 63f9eb8e Christos Stavrakakis
GANETI_JOBS_PENDING = ["queued", "waiting", "running", "canceling"]
84 63f9eb8e Christos Stavrakakis
GANETI_JOBS_FINALIZED = ["success", "error", "canceled"]
85 63f9eb8e Christos Stavrakakis
86 e63050ca Christos Stavrakakis
87 75dc539e Christos Stavrakakis
class BackendReconciler(object):
88 75dc539e Christos Stavrakakis
    def __init__(self, backend, logger, options=None):
89 75dc539e Christos Stavrakakis
        self.backend = backend
90 75dc539e Christos Stavrakakis
        self.log = logger
91 75dc539e Christos Stavrakakis
        self.client = backend.get_client()
92 75dc539e Christos Stavrakakis
        if options is None:
93 75dc539e Christos Stavrakakis
            self.options = {}
94 4161cb41 Christos Stavrakakis
        else:
95 75dc539e Christos Stavrakakis
            self.options = options
96 75dc539e Christos Stavrakakis
97 75dc539e Christos Stavrakakis
    def close(self):
98 75dc539e Christos Stavrakakis
        self.backend.put_client(self.client)
99 75dc539e Christos Stavrakakis
100 75dc539e Christos Stavrakakis
    @transaction.commit_on_success
101 75dc539e Christos Stavrakakis
    def reconcile(self):
102 75dc539e Christos Stavrakakis
        log = self.log
103 75dc539e Christos Stavrakakis
        backend = self.backend
104 75dc539e Christos Stavrakakis
        log.debug("Reconciling backend %s", backend)
105 75dc539e Christos Stavrakakis
106 75dc539e Christos Stavrakakis
        self.db_servers = get_database_servers(backend)
107 75dc539e Christos Stavrakakis
        self.db_servers_keys = set(self.db_servers.keys())
108 75dc539e Christos Stavrakakis
        log.debug("Got servers info from database.")
109 75dc539e Christos Stavrakakis
110 75dc539e Christos Stavrakakis
        self.gnt_servers = get_ganeti_servers(backend)
111 75dc539e Christos Stavrakakis
        self.gnt_servers_keys = set(self.gnt_servers.keys())
112 75dc539e Christos Stavrakakis
        log.debug("Got servers info from Ganeti backend.")
113 75dc539e Christos Stavrakakis
114 63f9eb8e Christos Stavrakakis
        self.gnt_jobs = get_ganeti_jobs(backend)
115 63f9eb8e Christos Stavrakakis
        log.debug("Got jobs from Ganeti backend")
116 63f9eb8e Christos Stavrakakis
117 75dc539e Christos Stavrakakis
        self.event_time = datetime.now()
118 75dc539e Christos Stavrakakis
119 75dc539e Christos Stavrakakis
        self.stale_servers = self.reconcile_stale_servers()
120 75dc539e Christos Stavrakakis
        self.orphan_servers = self.reconcile_orphan_servers()
121 75dc539e Christos Stavrakakis
        self.unsynced_servers = self.reconcile_unsynced_servers()
122 75dc539e Christos Stavrakakis
        self.close()
123 75dc539e Christos Stavrakakis
124 75dc539e Christos Stavrakakis
    def get_build_status(self, db_server):
125 63f9eb8e Christos Stavrakakis
        job_id = db_server.backendjobid
126 63f9eb8e Christos Stavrakakis
        if job_id in self.gnt_jobs:
127 63f9eb8e Christos Stavrakakis
            gnt_job_status = self.gnt_jobs[job_id]["status"]
128 63f9eb8e Christos Stavrakakis
            if gnt_job_status == GANETI_JOB_ERROR:
129 63f9eb8e Christos Stavrakakis
                return "ERROR"
130 63f9eb8e Christos Stavrakakis
            elif gnt_job_status not in GANETI_JOBS_FINALIZED:
131 75dc539e Christos Stavrakakis
                return "RUNNING"
132 75dc539e Christos Stavrakakis
            else:
133 63f9eb8e Christos Stavrakakis
                return "FINALIZED"
134 75dc539e Christos Stavrakakis
        else:
135 63f9eb8e Christos Stavrakakis
            return "ERROR"
136 75dc539e Christos Stavrakakis
137 75dc539e Christos Stavrakakis
    def reconcile_stale_servers(self):
138 75dc539e Christos Stavrakakis
        # Detect stale servers
139 75dc539e Christos Stavrakakis
        stale = []
140 75dc539e Christos Stavrakakis
        stale_keys = self.db_servers_keys - self.gnt_servers_keys
141 75dc539e Christos Stavrakakis
        for server_id in stale_keys:
142 75dc539e Christos Stavrakakis
            db_server = self.db_servers[server_id]
143 75dc539e Christos Stavrakakis
            if db_server.operstate == "BUILD":
144 75dc539e Christos Stavrakakis
                build_status = self.get_build_status(db_server)
145 75dc539e Christos Stavrakakis
                if build_status == "ERROR":
146 75dc539e Christos Stavrakakis
                    # Special handling of BUILD eerrors
147 75dc539e Christos Stavrakakis
                    self.reconcile_building_server(db_server)
148 75dc539e Christos Stavrakakis
                elif build_status != "RUNNING":
149 75dc539e Christos Stavrakakis
                    stale.append(server_id)
150 1464a17b Christos Stavrakakis
            elif (db_server.operstate == "ERROR" and
151 1464a17b Christos Stavrakakis
                  db_server.action != "DESTROY"):
152 1464a17b Christos Stavrakakis
                # Servers at building ERROR are stale only if the user has
153 1464a17b Christos Stavrakakis
                # asked to destroy them.
154 1464a17b Christos Stavrakakis
                pass
155 75dc539e Christos Stavrakakis
            else:
156 75dc539e Christos Stavrakakis
                stale.append(server_id)
157 75dc539e Christos Stavrakakis
158 75dc539e Christos Stavrakakis
        # Report them
159 75dc539e Christos Stavrakakis
        if stale:
160 75dc539e Christos Stavrakakis
            self.log.info("Found stale servers %s at backend %s",
161 75dc539e Christos Stavrakakis
                          ", ".join(map(str, stale)), self.backend)
162 75dc539e Christos Stavrakakis
        else:
163 75dc539e Christos Stavrakakis
            self.log.debug("No stale servers at backend %s", self.backend)
164 75dc539e Christos Stavrakakis
165 75dc539e Christos Stavrakakis
        # Fix them
166 75dc539e Christos Stavrakakis
        if stale and self.options["fix_stale"]:
167 75dc539e Christos Stavrakakis
            for server_id in stale:
168 75dc539e Christos Stavrakakis
                db_server = self.db_servers[server_id]
169 75dc539e Christos Stavrakakis
                backend_mod.process_op_status(
170 75dc539e Christos Stavrakakis
                    vm=db_server,
171 75dc539e Christos Stavrakakis
                    etime=self.event_time,
172 75dc539e Christos Stavrakakis
                    jobid=-0,
173 75dc539e Christos Stavrakakis
                    opcode='OP_INSTANCE_REMOVE', status='success',
174 75dc539e Christos Stavrakakis
                    logmsg='Reconciliation: simulated Ganeti event')
175 75dc539e Christos Stavrakakis
            self.log.debug("Simulated Ganeti removal for stale servers.")
176 75dc539e Christos Stavrakakis
177 75dc539e Christos Stavrakakis
    def reconcile_orphan_servers(self):
178 75dc539e Christos Stavrakakis
        orphans = self.gnt_servers_keys - self.db_servers_keys
179 75dc539e Christos Stavrakakis
        if orphans:
180 75dc539e Christos Stavrakakis
            self.log.info("Found orphan servers %s at backend %s",
181 75dc539e Christos Stavrakakis
                          ", ".join(map(str, orphans)), self.backend)
182 75dc539e Christos Stavrakakis
        else:
183 75dc539e Christos Stavrakakis
            self.log.debug("No orphan servers at backend %s", self.backend)
184 75dc539e Christos Stavrakakis
185 75dc539e Christos Stavrakakis
        if orphans and self.options["fix_orphans"]:
186 75dc539e Christos Stavrakakis
            for server_id in orphans:
187 75dc539e Christos Stavrakakis
                server_name = utils.id_to_instance_name(server_id)
188 75dc539e Christos Stavrakakis
                self.client.DeleteInstance(server_name)
189 75dc539e Christos Stavrakakis
            self.log.debug("Issued OP_INSTANCE_REMOVE for orphan servers.")
190 75dc539e Christos Stavrakakis
191 75dc539e Christos Stavrakakis
    def reconcile_unsynced_servers(self):
192 75dc539e Christos Stavrakakis
        #log = self.log
193 75dc539e Christos Stavrakakis
        for server_id in self.db_servers_keys & self.gnt_servers_keys:
194 75dc539e Christos Stavrakakis
            db_server = self.db_servers[server_id]
195 75dc539e Christos Stavrakakis
            gnt_server = self.gnt_servers[server_id]
196 75dc539e Christos Stavrakakis
            if db_server.operstate == "BUILD":
197 75dc539e Christos Stavrakakis
                build_status = self.get_build_status(db_server)
198 75dc539e Christos Stavrakakis
                if build_status == "RUNNING":
199 75dc539e Christos Stavrakakis
                    # Do not reconcile building VMs
200 75dc539e Christos Stavrakakis
                    continue
201 75dc539e Christos Stavrakakis
                elif build_status == "ERROR":
202 75dc539e Christos Stavrakakis
                    # Special handling of build errors
203 75dc539e Christos Stavrakakis
                    self.reconcile_building_server(db_server)
204 75dc539e Christos Stavrakakis
                    continue
205 75dc539e Christos Stavrakakis
206 75dc539e Christos Stavrakakis
            self.reconcile_unsynced_operstate(server_id, db_server,
207 75dc539e Christos Stavrakakis
                                              gnt_server)
208 75dc539e Christos Stavrakakis
            self.reconcile_unsynced_flavor(server_id, db_server,
209 75dc539e Christos Stavrakakis
                                           gnt_server)
210 75dc539e Christos Stavrakakis
            self.reconcile_unsynced_nics(server_id, db_server, gnt_server)
211 75dc539e Christos Stavrakakis
            self.reconcile_unsynced_disks(server_id, db_server, gnt_server)
212 63f9eb8e Christos Stavrakakis
            if db_server.task is not None:
213 63f9eb8e Christos Stavrakakis
                self.reconcile_pending_task(server_id, db_server)
214 75dc539e Christos Stavrakakis
215 75dc539e Christos Stavrakakis
    def reconcile_building_server(self, db_server):
216 75dc539e Christos Stavrakakis
        self.log.info("Server '%s' is BUILD in DB, but 'ERROR' in Ganeti.",
217 75dc539e Christos Stavrakakis
                      db_server.id)
218 75dc539e Christos Stavrakakis
        if self.options["fix_unsynced"]:
219 75dc539e Christos Stavrakakis
            fix_opcode = "OP_INSTANCE_CREATE"
220 75dc539e Christos Stavrakakis
            backend_mod.process_op_status(
221 75dc539e Christos Stavrakakis
                vm=db_server,
222 75dc539e Christos Stavrakakis
                etime=self.event_time,
223 75dc539e Christos Stavrakakis
                jobid=-0,
224 75dc539e Christos Stavrakakis
                opcode=fix_opcode, status='error',
225 75dc539e Christos Stavrakakis
                logmsg='Reconciliation: simulated Ganeti event')
226 75dc539e Christos Stavrakakis
            self.log.debug("Simulated Ganeti error build event for"
227 75dc539e Christos Stavrakakis
                           " server '%s'", db_server.id)
228 75dc539e Christos Stavrakakis
229 75dc539e Christos Stavrakakis
    def reconcile_unsynced_operstate(self, server_id, db_server, gnt_server):
230 75dc539e Christos Stavrakakis
        if db_server.operstate != gnt_server["state"]:
231 75dc539e Christos Stavrakakis
            self.log.info("Server '%s' is '%s' in DB and '%s' in Ganeti.",
232 75dc539e Christos Stavrakakis
                          server_id, db_server.operstate, gnt_server["state"])
233 75dc539e Christos Stavrakakis
            if self.options["fix_unsynced"]:
234 5d3e597a Christos Stavrakakis
                # If server is in building state, you will have first to
235 5d3e597a Christos Stavrakakis
                # reconcile it's creation, to avoid wrong quotas
236 5d3e597a Christos Stavrakakis
                if db_server.operstate == "BUILD":
237 5d3e597a Christos Stavrakakis
                    backend_mod.process_op_status(
238 5d3e597a Christos Stavrakakis
                        vm=db_server, etime=self.event_time, jobid=-0,
239 5d3e597a Christos Stavrakakis
                        opcode="OP_INSTANCE_CREATE", status='success',
240 5d3e597a Christos Stavrakakis
                        logmsg='Reconciliation: simulated Ganeti event')
241 5d3e597a Christos Stavrakakis
                fix_opcode = "OP_INSTANCE_STARTUP"\
242 5d3e597a Christos Stavrakakis
                    if gnt_server["state"] == "STARTED"\
243 75dc539e Christos Stavrakakis
                    else "OP_INSTANCE_SHUTDOWN"
244 75dc539e Christos Stavrakakis
                backend_mod.process_op_status(
245 5d3e597a Christos Stavrakakis
                    vm=db_server, etime=self.event_time, jobid=-0,
246 75dc539e Christos Stavrakakis
                    opcode=fix_opcode, status='success',
247 75dc539e Christos Stavrakakis
                    logmsg='Reconciliation: simulated Ganeti event')
248 75dc539e Christos Stavrakakis
                self.log.debug("Simulated Ganeti state event for server '%s'",
249 75dc539e Christos Stavrakakis
                               server_id)
250 75dc539e Christos Stavrakakis
251 75dc539e Christos Stavrakakis
    def reconcile_unsynced_flavor(self, server_id, db_server, gnt_server):
252 75dc539e Christos Stavrakakis
        db_flavor = db_server.flavor
253 75dc539e Christos Stavrakakis
        gnt_flavor = gnt_server["flavor"]
254 75dc539e Christos Stavrakakis
        if (db_flavor.ram != gnt_flavor["ram"] or
255 5d3e597a Christos Stavrakakis
           db_flavor.cpu != gnt_flavor["vcpus"]):
256 a67419d8 Christos Stavrakakis
            try:
257 a67419d8 Christos Stavrakakis
                gnt_flavor = Flavor.objects.get(
258 75dc539e Christos Stavrakakis
                    ram=gnt_flavor["ram"],
259 75dc539e Christos Stavrakakis
                    cpu=gnt_flavor["vcpus"],
260 75dc539e Christos Stavrakakis
                    disk=db_flavor.disk,
261 75dc539e Christos Stavrakakis
                    disk_template=db_flavor.disk_template)
262 a67419d8 Christos Stavrakakis
            except Flavor.DoesNotExist:
263 75dc539e Christos Stavrakakis
                self.log.warning("Server '%s' has unknown flavor.", server_id)
264 75dc539e Christos Stavrakakis
                return
265 75dc539e Christos Stavrakakis
266 4111c53e Christos Stavrakakis
            self.log.info("Server '%s' has flavor '%s' in DB and '%s' in"
267 75dc539e Christos Stavrakakis
                          " Ganeti", server_id, db_flavor, gnt_flavor)
268 75dc539e Christos Stavrakakis
            if self.options["fix_unsynced_flavors"]:
269 75dc539e Christos Stavrakakis
                old_state = db_server.operstate
270 75dc539e Christos Stavrakakis
                opcode = "OP_INSTANCE_SET_PARAMS"
271 75dc539e Christos Stavrakakis
                beparams = {"vcpus": gnt_flavor.cpu,
272 75dc539e Christos Stavrakakis
                            "minmem": gnt_flavor.ram,
273 75dc539e Christos Stavrakakis
                            "maxmem": gnt_flavor.ram}
274 75dc539e Christos Stavrakakis
                backend_mod.process_op_status(
275 75dc539e Christos Stavrakakis
                    vm=db_server, etime=self.event_time, jobid=-0,
276 75dc539e Christos Stavrakakis
                    opcode=opcode, status='success',
277 75dc539e Christos Stavrakakis
                    beparams=beparams,
278 75dc539e Christos Stavrakakis
                    logmsg='Reconciliation: simulated Ganeti event')
279 75dc539e Christos Stavrakakis
                # process_op_status with beparams will set the vmstate to
280 75dc539e Christos Stavrakakis
                # shutdown. Fix this be returning it to old state
281 75dc539e Christos Stavrakakis
                vm = VirtualMachine.objects.get(pk=server_id)
282 75dc539e Christos Stavrakakis
                vm.operstate = old_state
283 75dc539e Christos Stavrakakis
                vm.save()
284 75dc539e Christos Stavrakakis
                self.log.debug("Simulated Ganeti flavor event for server '%s'",
285 75dc539e Christos Stavrakakis
                               server_id)
286 75dc539e Christos Stavrakakis
287 75dc539e Christos Stavrakakis
    def reconcile_unsynced_nics(self, server_id, db_server, gnt_server):
288 75dc539e Christos Stavrakakis
        db_nics = db_server.nics.order_by("index")
289 75dc539e Christos Stavrakakis
        gnt_nics = gnt_server["nics"]
290 75dc539e Christos Stavrakakis
        gnt_nics_parsed = backend_mod.process_ganeti_nics(gnt_nics)
291 75dc539e Christos Stavrakakis
        if backend_mod.nics_changed(db_nics, gnt_nics_parsed):
292 75dc539e Christos Stavrakakis
            msg = "Found unsynced NICs for server '%s'.\n\t"\
293 75dc539e Christos Stavrakakis
                  "DB: %s\n\tGaneti: %s"
294 75dc539e Christos Stavrakakis
            db_nics_str = ", ".join(map(format_db_nic, db_nics))
295 75dc539e Christos Stavrakakis
            gnt_nics_str = ", ".join(map(format_gnt_nic, gnt_nics_parsed))
296 75dc539e Christos Stavrakakis
            self.log.info(msg, server_id, db_nics_str, gnt_nics_str)
297 75dc539e Christos Stavrakakis
            if self.options["fix_unsynced_nics"]:
298 75dc539e Christos Stavrakakis
                backend_mod.process_net_status(vm=db_server,
299 75dc539e Christos Stavrakakis
                                               etime=self.event_time,
300 75dc539e Christos Stavrakakis
                                               nics=gnt_nics)
301 75dc539e Christos Stavrakakis
302 75dc539e Christos Stavrakakis
    def reconcile_unsynced_disks(self, server_id, db_server, gnt_server):
303 75dc539e Christos Stavrakakis
        pass
304 75dc539e Christos Stavrakakis
305 63f9eb8e Christos Stavrakakis
    def reconcile_pending_task(self, server_id, db_server):
306 63f9eb8e Christos Stavrakakis
        job_id = db_server.task_job_id
307 63f9eb8e Christos Stavrakakis
        pending_task = False
308 63f9eb8e Christos Stavrakakis
        if job_id not in self.gnt_jobs:
309 63f9eb8e Christos Stavrakakis
            pending_task = True
310 63f9eb8e Christos Stavrakakis
        else:
311 ce55f724 Christos Stavrakakis
            gnt_job_status = self.gnt_jobs[job_id]["status"]
312 63f9eb8e Christos Stavrakakis
            if gnt_job_status in GANETI_JOBS_FINALIZED:
313 63f9eb8e Christos Stavrakakis
                pending_task = True
314 63f9eb8e Christos Stavrakakis
315 63f9eb8e Christos Stavrakakis
        if pending_task:
316 63f9eb8e Christos Stavrakakis
            self.log.info("Found server '%s' with pending task: '%s'",
317 63f9eb8e Christos Stavrakakis
                          server_id, db_server.task)
318 160c81a1 Christos Stavrakakis
            if self.options["fix_pending_tasks"]:
319 63f9eb8e Christos Stavrakakis
                db_server.task = None
320 63f9eb8e Christos Stavrakakis
                db_server.task_job_id = None
321 63f9eb8e Christos Stavrakakis
                db_server.save()
322 63f9eb8e Christos Stavrakakis
                self.log.info("Cleared pending task for server '%s", server_id)
323 63f9eb8e Christos Stavrakakis
324 75dc539e Christos Stavrakakis
325 75dc539e Christos Stavrakakis
def format_db_nic(nic):
326 4111c53e Christos Stavrakakis
    return "Index: %s, IP: %s Network: %s MAC: %s Firewall: %s" % (nic.index,
327 75dc539e Christos Stavrakakis
           nic.ipv4, nic.network_id, nic.mac, nic.firewall_profile)
328 75dc539e Christos Stavrakakis
329 75dc539e Christos Stavrakakis
330 75dc539e Christos Stavrakakis
def format_gnt_nic(nic):
331 75dc539e Christos Stavrakakis
    return "Index: %s IP: %s Network: %s MAC: %s Firewall: %s" %\
332 75dc539e Christos Stavrakakis
           (nic["index"], nic["ipv4"], nic["network"], nic["mac"],
333 75dc539e Christos Stavrakakis
            nic["firewall_profile"])
334 9fea53cc Vangelis Koukis
335 cc92b70f Christos Stavrakakis
336 0e9a423f Christos Stavrakakis
#
337 0e9a423f Christos Stavrakakis
# Networks
338 0e9a423f Christos Stavrakakis
#
339 3524241a Christos Stavrakakis
340 3524241a Christos Stavrakakis
341 0e9a423f Christos Stavrakakis
def get_networks_from_ganeti(backend):
342 44e2c577 Christos Stavrakakis
    prefix = settings.BACKEND_PREFIX_ID + 'net-'
343 0e9a423f Christos Stavrakakis
344 0e9a423f Christos Stavrakakis
    networks = {}
345 3524241a Christos Stavrakakis
    with pooled_rapi_client(backend) as c:
346 3524241a Christos Stavrakakis
        for net in c.GetNetworks(bulk=True):
347 3524241a Christos Stavrakakis
            if net['name'].startswith(prefix):
348 3524241a Christos Stavrakakis
                id = utils.id_from_network_name(net['name'])
349 3524241a Christos Stavrakakis
                networks[id] = net
350 0e9a423f Christos Stavrakakis
351 0e9a423f Christos Stavrakakis
    return networks
352 0e9a423f Christos Stavrakakis
353 0e9a423f Christos Stavrakakis
354 0e9a423f Christos Stavrakakis
def hanging_networks(backend, GNets):
355 0e9a423f Christos Stavrakakis
    """Get networks that are not connected to all Nodegroups.
356 0e9a423f Christos Stavrakakis

357 0e9a423f Christos Stavrakakis
    """
358 0e9a423f Christos Stavrakakis
    def get_network_groups(group_list):
359 0e9a423f Christos Stavrakakis
        groups = set()
360 a9bb2a1a Christos Stavrakakis
        for (name, mode, link) in group_list:
361 a9bb2a1a Christos Stavrakakis
            groups.add(name)
362 0e9a423f Christos Stavrakakis
        return groups
363 0e9a423f Christos Stavrakakis
364 3524241a Christos Stavrakakis
    with pooled_rapi_client(backend) as c:
365 3524241a Christos Stavrakakis
        groups = set(c.GetGroups())
366 0e9a423f Christos Stavrakakis
367 0e9a423f Christos Stavrakakis
    hanging = {}
368 0e9a423f Christos Stavrakakis
    for id, info in GNets.items():
369 0e9a423f Christos Stavrakakis
        group_list = get_network_groups(info['group_list'])
370 0e9a423f Christos Stavrakakis
        if group_list != groups:
371 0e9a423f Christos Stavrakakis
            hanging[id] = groups - group_list
372 0e9a423f Christos Stavrakakis
    return hanging
373 0e9a423f Christos Stavrakakis
374 9fea53cc Vangelis Koukis
375 75dc539e Christos Stavrakakis
def get_online_backends():
376 75dc539e Christos Stavrakakis
    return Backend.objects.filter(offline=False)
377 75dc539e Christos Stavrakakis
378 75dc539e Christos Stavrakakis
379 75dc539e Christos Stavrakakis
def get_database_servers(backend):
380 75dc539e Christos Stavrakakis
    servers = backend.virtual_machines.select_related("nics", "flavor")\
381 75dc539e Christos Stavrakakis
                                      .filter(deleted=False)
382 75dc539e Christos Stavrakakis
    return dict([(s.id, s) for s in servers])
383 75dc539e Christos Stavrakakis
384 75dc539e Christos Stavrakakis
385 75dc539e Christos Stavrakakis
def get_ganeti_servers(backend):
386 75dc539e Christos Stavrakakis
    gnt_instances = backend_mod.get_instances(backend)
387 75dc539e Christos Stavrakakis
    # Filter out non-synnefo instances
388 75dc539e Christos Stavrakakis
    snf_backend_prefix = settings.BACKEND_PREFIX_ID
389 75dc539e Christos Stavrakakis
    gnt_instances = filter(lambda i: i["name"].startswith(snf_backend_prefix),
390 75dc539e Christos Stavrakakis
                           gnt_instances)
391 75dc539e Christos Stavrakakis
    gnt_instances = map(parse_gnt_instance, gnt_instances)
392 75dc539e Christos Stavrakakis
    return dict([(i["id"], i) for i in gnt_instances if i["id"] is not None])
393 75dc539e Christos Stavrakakis
394 75dc539e Christos Stavrakakis
395 75dc539e Christos Stavrakakis
def parse_gnt_instance(instance):
396 75dc539e Christos Stavrakakis
    try:
397 75dc539e Christos Stavrakakis
        instance_id = utils.id_from_instance_name(instance['name'])
398 75dc539e Christos Stavrakakis
    except Exception:
399 75dc539e Christos Stavrakakis
        logger.error("Ignoring instance with malformed name %s",
400 75dc539e Christos Stavrakakis
                     instance['name'])
401 75dc539e Christos Stavrakakis
        return (None, None)
402 75dc539e Christos Stavrakakis
403 75dc539e Christos Stavrakakis
    beparams = instance["beparams"]
404 75dc539e Christos Stavrakakis
405 75dc539e Christos Stavrakakis
    vcpus = beparams["vcpus"]
406 75dc539e Christos Stavrakakis
    ram = beparams["maxmem"]
407 75dc539e Christos Stavrakakis
    state = instance["oper_state"] and "STARTED" or "STOPPED"
408 75dc539e Christos Stavrakakis
409 75dc539e Christos Stavrakakis
    return {
410 75dc539e Christos Stavrakakis
        "id": instance_id,
411 75dc539e Christos Stavrakakis
        "state": state,  # FIX
412 75dc539e Christos Stavrakakis
        "updated": datetime.fromtimestamp(instance["mtime"]),
413 75dc539e Christos Stavrakakis
        "disks": disks_from_instance(instance),
414 75dc539e Christos Stavrakakis
        "nics": nics_from_instance(instance),
415 75dc539e Christos Stavrakakis
        "flavor": {"vcpus": vcpus,
416 75dc539e Christos Stavrakakis
                   "ram": ram},
417 75dc539e Christos Stavrakakis
        "tags": instance["tags"]
418 75dc539e Christos Stavrakakis
    }
419 75dc539e Christos Stavrakakis
420 75dc539e Christos Stavrakakis
421 75dc539e Christos Stavrakakis
def nics_from_instance(i):
422 75dc539e Christos Stavrakakis
    ips = zip(itertools.repeat('ip'), i['nic.ips'])
423 75dc539e Christos Stavrakakis
    macs = zip(itertools.repeat('mac'), i['nic.macs'])
424 75dc539e Christos Stavrakakis
    networks = zip(itertools.repeat('network'), i['nic.networks'])
425 75dc539e Christos Stavrakakis
    # modes = zip(itertools.repeat('mode'), i['nic.modes'])
426 75dc539e Christos Stavrakakis
    # links = zip(itertools.repeat('link'), i['nic.links'])
427 75dc539e Christos Stavrakakis
    # nics = zip(ips,macs,modes,networks,links)
428 75dc539e Christos Stavrakakis
    nics = zip(ips, macs, networks)
429 75dc539e Christos Stavrakakis
    nics = map(lambda x: dict(x), nics)
430 75dc539e Christos Stavrakakis
    #nics = dict(enumerate(nics))
431 75dc539e Christos Stavrakakis
    tags = i["tags"]
432 75dc539e Christos Stavrakakis
    for tag in tags:
433 75dc539e Christos Stavrakakis
        t = tag.split(":")
434 75dc539e Christos Stavrakakis
        if t[0:2] == ["synnefo", "network"]:
435 75dc539e Christos Stavrakakis
            if len(t) != 4:
436 75dc539e Christos Stavrakakis
                logger.error("Malformed synefo tag %s", tag)
437 75dc539e Christos Stavrakakis
                continue
438 75dc539e Christos Stavrakakis
            try:
439 75dc539e Christos Stavrakakis
                index = int(t[2])
440 75dc539e Christos Stavrakakis
                nics[index]['firewall'] = t[3]
441 75dc539e Christos Stavrakakis
            except ValueError:
442 75dc539e Christos Stavrakakis
                logger.error("Malformed synnefo tag %s", tag)
443 75dc539e Christos Stavrakakis
            except IndexError:
444 75dc539e Christos Stavrakakis
                logger.error("Found tag %s for non-existent NIC %d",
445 75dc539e Christos Stavrakakis
                             tag, index)
446 75dc539e Christos Stavrakakis
    return nics
447 9fea53cc Vangelis Koukis
448 9fea53cc Vangelis Koukis
449 63f9eb8e Christos Stavrakakis
def get_ganeti_jobs(backend):
450 63f9eb8e Christos Stavrakakis
    gnt_jobs = backend_mod.get_jobs(backend)
451 63f9eb8e Christos Stavrakakis
    return dict([(int(j["id"]), j) for j in gnt_jobs])
452 63f9eb8e Christos Stavrakakis
453 63f9eb8e Christos Stavrakakis
454 75dc539e Christos Stavrakakis
def disks_from_instance(i):
455 75dc539e Christos Stavrakakis
    return dict([(index, {"size": size})
456 75dc539e Christos Stavrakakis
                 for index, size in enumerate(i["disk.sizes"])])
457 89b2b908 Christos Stavrakakis
458 89b2b908 Christos Stavrakakis
459 89b2b908 Christos Stavrakakis
class NetworkReconciler(object):
460 0ccb6461 Christos Stavrakakis
    def __init__(self, logger, fix=False):
461 89b2b908 Christos Stavrakakis
        self.log = logger
462 89b2b908 Christos Stavrakakis
        self.fix = fix
463 89b2b908 Christos Stavrakakis
464 89b2b908 Christos Stavrakakis
    @transaction.commit_on_success
465 89b2b908 Christos Stavrakakis
    def reconcile_networks(self):
466 89b2b908 Christos Stavrakakis
        # Get models from DB
467 0ccb6461 Christos Stavrakakis
        self.backends = Backend.objects.exclude(offline=True)
468 0ccb6461 Christos Stavrakakis
        self.networks = Network.objects.filter(deleted=False)
469 89b2b908 Christos Stavrakakis
470 89b2b908 Christos Stavrakakis
        self.event_time = datetime.now()
471 89b2b908 Christos Stavrakakis
472 89b2b908 Christos Stavrakakis
        # Get info from all ganeti backends
473 0ccb6461 Christos Stavrakakis
        self.ganeti_networks = {}
474 0ccb6461 Christos Stavrakakis
        self.ganeti_hanging_networks = {}
475 0ccb6461 Christos Stavrakakis
        for b in self.backends:
476 89b2b908 Christos Stavrakakis
            g_nets = get_networks_from_ganeti(b)
477 0ccb6461 Christos Stavrakakis
            self.ganeti_networks[b] = g_nets
478 89b2b908 Christos Stavrakakis
            g_hanging_nets = hanging_networks(b, g_nets)
479 0ccb6461 Christos Stavrakakis
            self.ganeti_hanging_networks[b] = g_hanging_nets
480 89b2b908 Christos Stavrakakis
481 0ccb6461 Christos Stavrakakis
        self._reconcile_orphan_networks()
482 89b2b908 Christos Stavrakakis
483 0ccb6461 Christos Stavrakakis
        for network in self.networks:
484 0ccb6461 Christos Stavrakakis
            self._reconcile_network(network)
485 89b2b908 Christos Stavrakakis
486 0ccb6461 Christos Stavrakakis
    @transaction.commit_on_success
487 0ccb6461 Christos Stavrakakis
    def _reconcile_network(self, network):
488 0ccb6461 Christos Stavrakakis
        """Reconcile a network with corresponging Ganeti networks.
489 0ccb6461 Christos Stavrakakis

490 0ccb6461 Christos Stavrakakis
        Reconcile a Network and the associated BackendNetworks with the
491 0ccb6461 Christos Stavrakakis
        corresponding Ganeti networks in all Ganeti backends.
492 0ccb6461 Christos Stavrakakis

493 0ccb6461 Christos Stavrakakis
        """
494 0ccb6461 Christos Stavrakakis
        network_ip_pool = network.get_pool()  # X-Lock on IP Pool
495 0ccb6461 Christos Stavrakakis
        for bend in self.backends:
496 0ccb6461 Christos Stavrakakis
            bnet = get_backend_network(network, bend)
497 0ccb6461 Christos Stavrakakis
            gnet = self.ganeti_networks[bend].get(network.id)
498 0ccb6461 Christos Stavrakakis
            if not bnet:
499 0ccb6461 Christos Stavrakakis
                if network.floating_ip_pool:
500 0ccb6461 Christos Stavrakakis
                    # Network is a floating IP pool and does not exist in
501 0ccb6461 Christos Stavrakakis
                    # backend. We need to create it
502 0ccb6461 Christos Stavrakakis
                    bnet = self.reconcile_parted_network(network, bend)
503 0ccb6461 Christos Stavrakakis
                elif not gnet:
504 0ccb6461 Christos Stavrakakis
                    # Network does not exist either in Ganeti nor in BD.
505 0ccb6461 Christos Stavrakakis
                    continue
506 0ccb6461 Christos Stavrakakis
                else:
507 0ccb6461 Christos Stavrakakis
                    # Network exists in Ganeti and not in DB.
508 0ccb6461 Christos Stavrakakis
                    if network.action != "DESTROY" and not network.public:
509 0ccb6461 Christos Stavrakakis
                        bnet = self.reconcile_parted_network(network, bend)
510 0ccb6461 Christos Stavrakakis
                    else:
511 0ccb6461 Christos Stavrakakis
                        continue
512 89b2b908 Christos Stavrakakis
513 0ccb6461 Christos Stavrakakis
            if not gnet:
514 0ccb6461 Christos Stavrakakis
                # Network does not exist in Ganeti. If the network action
515 0ccb6461 Christos Stavrakakis
                # is DESTROY, we have to mark as deleted in DB, else we
516 0ccb6461 Christos Stavrakakis
                # have to create it in Ganeti.
517 0ccb6461 Christos Stavrakakis
                if network.action == "DESTROY":
518 0ccb6461 Christos Stavrakakis
                    if bnet.operstate != "DELETED":
519 0ccb6461 Christos Stavrakakis
                        self.reconcile_stale_network(bnet)
520 0ccb6461 Christos Stavrakakis
                else:
521 0ccb6461 Christos Stavrakakis
                    self.reconcile_missing_network(network, bend)
522 0ccb6461 Christos Stavrakakis
                # Skip rest reconciliation!
523 0ccb6461 Christos Stavrakakis
                continue
524 89b2b908 Christos Stavrakakis
525 0ccb6461 Christos Stavrakakis
            try:
526 0ccb6461 Christos Stavrakakis
                hanging_groups = self.ganeti_hanging_networks[bend][network.id]
527 0ccb6461 Christos Stavrakakis
            except KeyError:
528 0ccb6461 Christos Stavrakakis
                # Network is connected to all nodegroups
529 0ccb6461 Christos Stavrakakis
                hanging_groups = []
530 0ccb6461 Christos Stavrakakis
531 0ccb6461 Christos Stavrakakis
            if hanging_groups:
532 0ccb6461 Christos Stavrakakis
                # CASE-3: Ganeti networks not connected to all nodegroups
533 0ccb6461 Christos Stavrakakis
                self.reconcile_hanging_groups(network, bend,
534 0ccb6461 Christos Stavrakakis
                                              hanging_groups)
535 0ccb6461 Christos Stavrakakis
                continue
536 89b2b908 Christos Stavrakakis
537 0ccb6461 Christos Stavrakakis
            if bnet.operstate != 'ACTIVE':
538 0ccb6461 Christos Stavrakakis
                # CASE-4: Unsynced network state. At this point the network
539 0ccb6461 Christos Stavrakakis
                # exists and is connected to all nodes so is must be
540 0ccb6461 Christos Stavrakakis
                # active!
541 0ccb6461 Christos Stavrakakis
                self.reconcile_unsynced_network(network, bend, bnet)
542 0ccb6461 Christos Stavrakakis
543 0ccb6461 Christos Stavrakakis
            # Check that externally reserved IPs of the network in Ganeti are
544 0ccb6461 Christos Stavrakakis
            # also externally reserved to the IP pool
545 0ccb6461 Christos Stavrakakis
            externally_reserved = gnet['external_reservations']
546 98a01362 Christos Stavrakakis
            if externally_reserved:
547 98a01362 Christos Stavrakakis
                for ip in externally_reserved.split(","):
548 98a01362 Christos Stavrakakis
                    ip = ip.strip()
549 98a01362 Christos Stavrakakis
                    if not network_ip_pool.is_reserved(ip):
550 98a01362 Christos Stavrakakis
                        msg = ("D: IP '%s' is reserved for network '%s' in"
551 98a01362 Christos Stavrakakis
                               " backend '%s' but not in DB.")
552 98a01362 Christos Stavrakakis
                        self.log.info(msg, ip, network, bend)
553 98a01362 Christos Stavrakakis
                        if self.fix:
554 98a01362 Christos Stavrakakis
                            network_ip_pool.reserve(ip, external=True)
555 98a01362 Christos Stavrakakis
                            network_ip_pool.save()
556 98a01362 Christos Stavrakakis
                            self.log.info("F: Reserved IP '%s'", ip)
557 89b2b908 Christos Stavrakakis
558 89b2b908 Christos Stavrakakis
    def reconcile_parted_network(self, network, backend):
559 89b2b908 Christos Stavrakakis
        self.log.info("D: Missing DB entry for network %s in backend %s",
560 89b2b908 Christos Stavrakakis
                      network, backend)
561 89b2b908 Christos Stavrakakis
        if self.fix:
562 89b2b908 Christos Stavrakakis
            network.create_backend_network(backend)
563 89b2b908 Christos Stavrakakis
            self.log.info("F: Created DB entry")
564 89b2b908 Christos Stavrakakis
            bnet = get_backend_network(network, backend)
565 89b2b908 Christos Stavrakakis
            return bnet
566 89b2b908 Christos Stavrakakis
567 89b2b908 Christos Stavrakakis
    def reconcile_stale_network(self, backend_network):
568 89b2b908 Christos Stavrakakis
        self.log.info("D: Stale DB entry for network %s in backend %s",
569 89b2b908 Christos Stavrakakis
                      backend_network.network, backend_network.backend)
570 89b2b908 Christos Stavrakakis
        if self.fix:
571 89b2b908 Christos Stavrakakis
            backend_mod.process_network_status(
572 89b2b908 Christos Stavrakakis
                backend_network, self.event_time, 0,
573 89b2b908 Christos Stavrakakis
                "OP_NETWORK_REMOVE",
574 89b2b908 Christos Stavrakakis
                "success",
575 89b2b908 Christos Stavrakakis
                "Reconciliation simulated event")
576 89b2b908 Christos Stavrakakis
            self.log.info("F: Reconciled event: OP_NETWORK_REMOVE")
577 89b2b908 Christos Stavrakakis
578 89b2b908 Christos Stavrakakis
    def reconcile_missing_network(self, network, backend):
579 89b2b908 Christos Stavrakakis
        self.log.info("D: Missing Ganeti network %s in backend %s",
580 89b2b908 Christos Stavrakakis
                      network, backend)
581 89b2b908 Christos Stavrakakis
        if self.fix:
582 89b2b908 Christos Stavrakakis
            backend_mod.create_network(network, backend)
583 89b2b908 Christos Stavrakakis
            self.log.info("F: Issued OP_NETWORK_CONNECT")
584 89b2b908 Christos Stavrakakis
585 89b2b908 Christos Stavrakakis
    def reconcile_hanging_groups(self, network, backend, hanging_groups):
586 89b2b908 Christos Stavrakakis
        self.log.info('D: Network %s in backend %s is not connected to '
587 89b2b908 Christos Stavrakakis
                      'the following groups:', network, backend)
588 89b2b908 Christos Stavrakakis
        self.log.info('-  ' + '\n-  '.join(hanging_groups))
589 89b2b908 Christos Stavrakakis
        if self.fix:
590 89b2b908 Christos Stavrakakis
            for group in hanging_groups:
591 89b2b908 Christos Stavrakakis
                self.log.info('F: Connecting network %s to nodegroup %s',
592 89b2b908 Christos Stavrakakis
                              network, group)
593 89b2b908 Christos Stavrakakis
                backend_mod.connect_network(network, backend, depends=[],
594 89b2b908 Christos Stavrakakis
                                            group=group)
595 89b2b908 Christos Stavrakakis
596 89b2b908 Christos Stavrakakis
    def reconcile_unsynced_network(self, network, backend, backend_network):
597 89b2b908 Christos Stavrakakis
        self.log.info("D: Unsynced network %s in backend %s", network, backend)
598 89b2b908 Christos Stavrakakis
        if self.fix:
599 89b2b908 Christos Stavrakakis
            self.log.info("F: Issuing OP_NETWORK_CONNECT")
600 89b2b908 Christos Stavrakakis
            backend_mod.process_network_status(
601 89b2b908 Christos Stavrakakis
                backend_network, self.event_time, 0,
602 89b2b908 Christos Stavrakakis
                "OP_NETWORK_CONNECT",
603 89b2b908 Christos Stavrakakis
                "success",
604 89b2b908 Christos Stavrakakis
                "Reconciliation simulated eventd")
605 89b2b908 Christos Stavrakakis
606 0ccb6461 Christos Stavrakakis
    def _reconcile_orphan_networks(self):
607 0ccb6461 Christos Stavrakakis
        db_networks = self.networks
608 0ccb6461 Christos Stavrakakis
        ganeti_networks = self.ganeti_networks
609 89b2b908 Christos Stavrakakis
        # Detect Orphan Networks in Ganeti
610 89b2b908 Christos Stavrakakis
        db_network_ids = set([net.id for net in db_networks])
611 89b2b908 Christos Stavrakakis
        for back_end, ganeti_networks in ganeti_networks.items():
612 89b2b908 Christos Stavrakakis
            ganeti_network_ids = set(ganeti_networks.keys())
613 89b2b908 Christos Stavrakakis
            orphans = ganeti_network_ids - db_network_ids
614 89b2b908 Christos Stavrakakis
615 89b2b908 Christos Stavrakakis
            if len(orphans) > 0:
616 89b2b908 Christos Stavrakakis
                self.log.info('D: Orphan Networks in backend %s:',
617 89b2b908 Christos Stavrakakis
                              back_end.clustername)
618 89b2b908 Christos Stavrakakis
                self.log.info('-  ' + '\n-  '.join([str(o) for o in orphans]))
619 89b2b908 Christos Stavrakakis
                if self.fix:
620 89b2b908 Christos Stavrakakis
                    for net_id in orphans:
621 89b2b908 Christos Stavrakakis
                        self.log.info('Disconnecting and deleting network %d',
622 89b2b908 Christos Stavrakakis
                                      net_id)
623 89b2b908 Christos Stavrakakis
                        try:
624 89b2b908 Christos Stavrakakis
                            network = Network.objects.get(id=net_id)
625 89b2b908 Christos Stavrakakis
                            backend_mod.delete_network(network,
626 89b2b908 Christos Stavrakakis
                                                       backend=back_end)
627 89b2b908 Christos Stavrakakis
                        except Network.DoesNotExist:
628 89b2b908 Christos Stavrakakis
                            self.log.info("Not entry for network %s in DB !!",
629 89b2b908 Christos Stavrakakis
                                          net_id)
630 89b2b908 Christos Stavrakakis
631 89b2b908 Christos Stavrakakis
632 89b2b908 Christos Stavrakakis
def get_backend_network(network, backend):
633 89b2b908 Christos Stavrakakis
    try:
634 89b2b908 Christos Stavrakakis
        return BackendNetwork.objects.get(network=network, backend=backend)
635 89b2b908 Christos Stavrakakis
    except BackendNetwork.DoesNotExist:
636 89b2b908 Christos Stavrakakis
        return None
637 89b2b908 Christos Stavrakakis
638 89b2b908 Christos Stavrakakis
639 0ccb6461 Christos Stavrakakis
class PoolReconciler(object):
640 0ccb6461 Christos Stavrakakis
    def __init__(self, logger, fix=False):
641 0ccb6461 Christos Stavrakakis
        self.log = logger
642 0ccb6461 Christos Stavrakakis
        self.fix = fix
643 0ccb6461 Christos Stavrakakis
644 0ccb6461 Christos Stavrakakis
    def reconcile(self):
645 0ccb6461 Christos Stavrakakis
        self.reconcile_bridges()
646 0ccb6461 Christos Stavrakakis
        self.reconcile_mac_prefixes()
647 0ccb6461 Christos Stavrakakis
        for network in Network.objects.filter(deleted=False):
648 0ccb6461 Christos Stavrakakis
            self.reconcile_ip_pool(network)
649 0ccb6461 Christos Stavrakakis
650 0ccb6461 Christos Stavrakakis
    @transaction.commit_on_success
651 0ccb6461 Christos Stavrakakis
    def reconcile_bridges(self):
652 0ccb6461 Christos Stavrakakis
        networks = Network.objects.filter(deleted=False,
653 0ccb6461 Christos Stavrakakis
                                          flavor="PHYSICAL_VLAN")
654 0ccb6461 Christos Stavrakakis
        check_unique_values(objects=networks, field='link', logger=self.log)
655 0ccb6461 Christos Stavrakakis
        try:
656 0ccb6461 Christos Stavrakakis
            pool = BridgePoolTable.get_pool()
657 0ccb6461 Christos Stavrakakis
        except pools.EmptyPool:
658 0ccb6461 Christos Stavrakakis
            self.log.info("There is no available pool for bridges.")
659 0ccb6461 Christos Stavrakakis
            return
660 0ccb6461 Christos Stavrakakis
661 0ccb6461 Christos Stavrakakis
        used_bridges = set(networks.values_list('link', flat=True))
662 0ccb6461 Christos Stavrakakis
        check_pool_consistent(pool=pool, pool_class=pools.BridgePool,
663 0ccb6461 Christos Stavrakakis
                              used_values=used_bridges, fix=self.fix,
664 0ccb6461 Christos Stavrakakis
                              logger=self.log)
665 89b2b908 Christos Stavrakakis
666 0ccb6461 Christos Stavrakakis
    @transaction.commit_on_success
667 0ccb6461 Christos Stavrakakis
    def reconcile_mac_prefixes(self):
668 0ccb6461 Christos Stavrakakis
        networks = Network.objects.filter(deleted=False, flavor="MAC_FILTERED")
669 0ccb6461 Christos Stavrakakis
        check_unique_values(objects=networks, field='mac_prefix',
670 0ccb6461 Christos Stavrakakis
                            logger=self.log)
671 0ccb6461 Christos Stavrakakis
        try:
672 0ccb6461 Christos Stavrakakis
            pool = MacPrefixPoolTable.get_pool()
673 0ccb6461 Christos Stavrakakis
        except pools.EmptyPool:
674 0ccb6461 Christos Stavrakakis
            self.log.info("There is no available pool for MAC prefixes.")
675 0ccb6461 Christos Stavrakakis
            return
676 0ccb6461 Christos Stavrakakis
677 0ccb6461 Christos Stavrakakis
        used_mac_prefixes = set(networks.values_list('mac_prefix', flat=True))
678 0ccb6461 Christos Stavrakakis
        check_pool_consistent(pool=pool, pool_class=pools.MacPrefixPool,
679 0ccb6461 Christos Stavrakakis
                              used_values=used_mac_prefixes, fix=self.fix,
680 0ccb6461 Christos Stavrakakis
                              logger=self.log)
681 89b2b908 Christos Stavrakakis
682 0ccb6461 Christos Stavrakakis
    @transaction.commit_on_success
683 0ccb6461 Christos Stavrakakis
    def reconcile_ip_pool(self, network):
684 0ccb6461 Christos Stavrakakis
        # Check that all NICs have unique IPv4 address
685 0ccb6461 Christos Stavrakakis
        nics = network.nics.filter(ipv4__isnull=False)
686 0ccb6461 Christos Stavrakakis
        check_unique_values(objects=nics, field='ipv4', logger=self.log)
687 0ccb6461 Christos Stavrakakis
688 0ccb6461 Christos Stavrakakis
        # Check that all Floating IPs have unique IPv4 address
689 0ccb6461 Christos Stavrakakis
        floating_ips = network.floating_ips.filter(deleted=False)
690 0ccb6461 Christos Stavrakakis
        check_unique_values(objects=floating_ips, field='ipv4',
691 0ccb6461 Christos Stavrakakis
                            logger=self.log)
692 0ccb6461 Christos Stavrakakis
693 0ccb6461 Christos Stavrakakis
        # First get(lock) the IP pool of the network to prevent new NICs
694 0ccb6461 Christos Stavrakakis
        # from being created.
695 0ccb6461 Christos Stavrakakis
        network_ip_pool = network.get_pool()
696 0ccb6461 Christos Stavrakakis
        used_ips = set(list(nics.values_list("ipv4", flat=True)) +
697 0ccb6461 Christos Stavrakakis
                       list(floating_ips.values_list("ipv4", flat=True)))
698 0ccb6461 Christos Stavrakakis
699 0ccb6461 Christos Stavrakakis
        check_pool_consistent(pool=network_ip_pool,
700 0ccb6461 Christos Stavrakakis
                              pool_class=pools.IPPool,
701 0ccb6461 Christos Stavrakakis
                              used_values=used_ips,
702 0ccb6461 Christos Stavrakakis
                              fix=self.fix, logger=self.log)
703 0ccb6461 Christos Stavrakakis
704 0ccb6461 Christos Stavrakakis
705 0ccb6461 Christos Stavrakakis
def check_unique_values(objects, field, logger):
706 0ccb6461 Christos Stavrakakis
    used_values = list(objects.values_list(field, flat=True))
707 0ccb6461 Christos Stavrakakis
    if len(used_values) != len(set(used_values)):
708 0ccb6461 Christos Stavrakakis
        duplicate_values = [v for v in used_values if used_values.count(v) > 1]
709 0ccb6461 Christos Stavrakakis
        for value in duplicate_values:
710 0ccb6461 Christos Stavrakakis
            filter_args = {field: value}
711 0ccb6461 Christos Stavrakakis
            using_objects = objects.filter(**filter_args)
712 0ccb6461 Christos Stavrakakis
            msg = "Value '%s' is used as %s for more than one objects: %s"
713 0ccb6461 Christos Stavrakakis
            logger.error(msg, value, field, ",".join(map(str, using_objects)))
714 0ccb6461 Christos Stavrakakis
        return False
715 0ccb6461 Christos Stavrakakis
    logger.debug("Values for field '%s' are unique.", field)
716 0ccb6461 Christos Stavrakakis
    return True
717 0ccb6461 Christos Stavrakakis
718 0ccb6461 Christos Stavrakakis
719 0ccb6461 Christos Stavrakakis
def check_pool_consistent(pool, pool_class, used_values, fix, logger):
720 0ccb6461 Christos Stavrakakis
    dummy_pool = create_empty_pool(pool, pool_class)
721 0ccb6461 Christos Stavrakakis
    [dummy_pool.reserve(value) for value in used_values]
722 0ccb6461 Christos Stavrakakis
    if dummy_pool.available != pool.available:
723 0ccb6461 Christos Stavrakakis
        msg = "'%s' is not consistent!\nPool: %s\nUsed: %s"
724 0ccb6461 Christos Stavrakakis
        pool_diff = dummy_pool.available ^ pool.available
725 0ccb6461 Christos Stavrakakis
        for index in pool_diff.itersearch(bitarray.bitarray("1")):
726 0ccb6461 Christos Stavrakakis
            value = pool.index_to_value(int(index))
727 0ccb6461 Christos Stavrakakis
            msg = "%s is incosistent! Value '%s' is %s but should be %s."
728 0ccb6461 Christos Stavrakakis
            value1 = pool.is_available(value) and "available" or "unavailable"
729 0ccb6461 Christos Stavrakakis
            value2 = dummy_pool.is_available(value) and "available"\
730 0ccb6461 Christos Stavrakakis
                or "unavailable"
731 0ccb6461 Christos Stavrakakis
            logger.error(msg, pool, value, value1, value2)
732 0ccb6461 Christos Stavrakakis
        if fix:
733 0ccb6461 Christos Stavrakakis
            pool.available = dummy_pool.available
734 0ccb6461 Christos Stavrakakis
            pool.save()
735 0ccb6461 Christos Stavrakakis
            logger.info("Fixed available map of pool '%s'", pool)
736 0ccb6461 Christos Stavrakakis
737 0ccb6461 Christos Stavrakakis
738 0ccb6461 Christos Stavrakakis
def create_empty_pool(pool, pool_class):
739 0ccb6461 Christos Stavrakakis
    pool_row = pool.pool_table
740 0ccb6461 Christos Stavrakakis
    pool_row.available_map = ""
741 0ccb6461 Christos Stavrakakis
    pool_row.reserved_map = ""
742 0ccb6461 Christos Stavrakakis
    return pool_class(pool_row)