Statistics
| Branch: | Tag: | Revision:

root / lib / config.py @ ea0f78c8

History | View | Annotate | Download (74.5 kB)

1 2f31098c Iustin Pop
#
2 a8083063 Iustin Pop
#
3 a8083063 Iustin Pop
4 12378fe3 Iustin Pop
# Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011, 2012 Google Inc.
5 a8083063 Iustin Pop
#
6 a8083063 Iustin Pop
# This program is free software; you can redistribute it and/or modify
7 a8083063 Iustin Pop
# it under the terms of the GNU General Public License as published by
8 a8083063 Iustin Pop
# the Free Software Foundation; either version 2 of the License, or
9 a8083063 Iustin Pop
# (at your option) any later version.
10 a8083063 Iustin Pop
#
11 a8083063 Iustin Pop
# This program is distributed in the hope that it will be useful, but
12 a8083063 Iustin Pop
# WITHOUT ANY WARRANTY; without even the implied warranty of
13 a8083063 Iustin Pop
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 a8083063 Iustin Pop
# General Public License for more details.
15 a8083063 Iustin Pop
#
16 a8083063 Iustin Pop
# You should have received a copy of the GNU General Public License
17 a8083063 Iustin Pop
# along with this program; if not, write to the Free Software
18 a8083063 Iustin Pop
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
19 a8083063 Iustin Pop
# 02110-1301, USA.
20 a8083063 Iustin Pop
21 a8083063 Iustin Pop
22 a8083063 Iustin Pop
"""Configuration management for Ganeti
23 a8083063 Iustin Pop

24 319856a9 Michael Hanselmann
This module provides the interface to the Ganeti cluster configuration.
25 a8083063 Iustin Pop

26 319856a9 Michael Hanselmann
The configuration data is stored on every node but is updated on the master
27 319856a9 Michael Hanselmann
only. After each update, the master distributes the data to the other nodes.
28 a8083063 Iustin Pop

29 319856a9 Michael Hanselmann
Currently, the data storage format is JSON. YAML was slow and consuming too
30 319856a9 Michael Hanselmann
much memory.
31 a8083063 Iustin Pop

32 a8083063 Iustin Pop
"""
33 a8083063 Iustin Pop
34 b459a848 Andrea Spadaccini
# pylint: disable=R0904
35 d367b66c Manuel Franceschini
# R0904: Too many public methods
36 d367b66c Manuel Franceschini
37 a8083063 Iustin Pop
import os
38 a8083063 Iustin Pop
import random
39 d8470559 Michael Hanselmann
import logging
40 d693c864 Iustin Pop
import time
41 54c31fd3 Michael Hanselmann
import itertools
42 a8083063 Iustin Pop
43 a8083063 Iustin Pop
from ganeti import errors
44 f78ede4e Guido Trotter
from ganeti import locking
45 a8083063 Iustin Pop
from ganeti import utils
46 a8083063 Iustin Pop
from ganeti import constants
47 a8083063 Iustin Pop
from ganeti import rpc
48 a8083063 Iustin Pop
from ganeti import objects
49 8d14b30d Iustin Pop
from ganeti import serializer
50 0fbae49a Balazs Lecz
from ganeti import uidpool
51 a744b676 Manuel Franceschini
from ganeti import netutils
52 e60c73a1 René Nussbaumer
from ganeti import runtime
53 243cdbcc Michael Hanselmann
54 243cdbcc Michael Hanselmann
55 7f93570a Iustin Pop
_config_lock = locking.SharedLock("ConfigWriter")
56 f78ede4e Guido Trotter
57 4fae38c5 Guido Trotter
# job id used for resource management at config upgrade time
58 8d9c3bef Michael Hanselmann
_UPGRADE_CONFIG_JID = "jid-cfg-upgrade"
59 4fae38c5 Guido Trotter
60 f78ede4e Guido Trotter
61 5b263ed7 Michael Hanselmann
def _ValidateConfig(data):
62 c41eea6e Iustin Pop
  """Verifies that a configuration objects looks valid.
63 c41eea6e Iustin Pop

64 c41eea6e Iustin Pop
  This only verifies the version of the configuration.
65 c41eea6e Iustin Pop

66 c41eea6e Iustin Pop
  @raise errors.ConfigurationError: if the version differs from what
67 c41eea6e Iustin Pop
      we expect
68 c41eea6e Iustin Pop

69 c41eea6e Iustin Pop
  """
70 5b263ed7 Michael Hanselmann
  if data.version != constants.CONFIG_VERSION:
71 4b63dc7a Iustin Pop
    raise errors.ConfigVersionMismatch(constants.CONFIG_VERSION, data.version)
72 a8083063 Iustin Pop
73 319856a9 Michael Hanselmann
74 013da361 Guido Trotter
class TemporaryReservationManager:
75 013da361 Guido Trotter
  """A temporary resource reservation manager.
76 013da361 Guido Trotter

77 013da361 Guido Trotter
  This is used to reserve resources in a job, before using them, making sure
78 013da361 Guido Trotter
  other jobs cannot get them in the meantime.
79 013da361 Guido Trotter

80 013da361 Guido Trotter
  """
81 013da361 Guido Trotter
  def __init__(self):
82 013da361 Guido Trotter
    self._ec_reserved = {}
83 013da361 Guido Trotter
84 013da361 Guido Trotter
  def Reserved(self, resource):
85 a7359d91 David Knowles
    for holder_reserved in self._ec_reserved.values():
86 013da361 Guido Trotter
      if resource in holder_reserved:
87 013da361 Guido Trotter
        return True
88 013da361 Guido Trotter
    return False
89 013da361 Guido Trotter
90 013da361 Guido Trotter
  def Reserve(self, ec_id, resource):
91 013da361 Guido Trotter
    if self.Reserved(resource):
92 28a7318f Iustin Pop
      raise errors.ReservationError("Duplicate reservation for resource '%s'"
93 28a7318f Iustin Pop
                                    % str(resource))
94 013da361 Guido Trotter
    if ec_id not in self._ec_reserved:
95 013da361 Guido Trotter
      self._ec_reserved[ec_id] = set([resource])
96 013da361 Guido Trotter
    else:
97 013da361 Guido Trotter
      self._ec_reserved[ec_id].add(resource)
98 013da361 Guido Trotter
99 013da361 Guido Trotter
  def DropECReservations(self, ec_id):
100 013da361 Guido Trotter
    if ec_id in self._ec_reserved:
101 013da361 Guido Trotter
      del self._ec_reserved[ec_id]
102 013da361 Guido Trotter
103 013da361 Guido Trotter
  def GetReserved(self):
104 013da361 Guido Trotter
    all_reserved = set()
105 013da361 Guido Trotter
    for holder_reserved in self._ec_reserved.values():
106 013da361 Guido Trotter
      all_reserved.update(holder_reserved)
107 013da361 Guido Trotter
    return all_reserved
108 013da361 Guido Trotter
109 013da361 Guido Trotter
  def Generate(self, existing, generate_one_fn, ec_id):
110 013da361 Guido Trotter
    """Generate a new resource of this type
111 013da361 Guido Trotter

112 013da361 Guido Trotter
    """
113 013da361 Guido Trotter
    assert callable(generate_one_fn)
114 013da361 Guido Trotter
115 013da361 Guido Trotter
    all_elems = self.GetReserved()
116 013da361 Guido Trotter
    all_elems.update(existing)
117 013da361 Guido Trotter
    retries = 64
118 013da361 Guido Trotter
    while retries > 0:
119 013da361 Guido Trotter
      new_resource = generate_one_fn()
120 013da361 Guido Trotter
      if new_resource is not None and new_resource not in all_elems:
121 013da361 Guido Trotter
        break
122 013da361 Guido Trotter
    else:
123 013da361 Guido Trotter
      raise errors.ConfigurationError("Not able generate new resource"
124 013da361 Guido Trotter
                                      " (last tried: %s)" % new_resource)
125 013da361 Guido Trotter
    self.Reserve(ec_id, new_resource)
126 013da361 Guido Trotter
    return new_resource
127 013da361 Guido Trotter
128 013da361 Guido Trotter
129 fe698b38 Michael Hanselmann
def _MatchNameComponentIgnoreCase(short_name, names):
130 3a93eebb Michael Hanselmann
  """Wrapper around L{utils.text.MatchNameComponent}.
131 fe698b38 Michael Hanselmann

132 fe698b38 Michael Hanselmann
  """
133 fe698b38 Michael Hanselmann
  return utils.MatchNameComponent(short_name, names, case_sensitive=False)
134 fe698b38 Michael Hanselmann
135 fe698b38 Michael Hanselmann
136 82c54b5b Michael Hanselmann
def _CheckInstanceDiskIvNames(disks):
137 82c54b5b Michael Hanselmann
  """Checks if instance's disks' C{iv_name} attributes are in order.
138 82c54b5b Michael Hanselmann

139 82c54b5b Michael Hanselmann
  @type disks: list of L{objects.Disk}
140 82c54b5b Michael Hanselmann
  @param disks: List of disks
141 82c54b5b Michael Hanselmann
  @rtype: list of tuples; (int, string, string)
142 82c54b5b Michael Hanselmann
  @return: List of wrongly named disks, each tuple contains disk index,
143 82c54b5b Michael Hanselmann
    expected and actual name
144 82c54b5b Michael Hanselmann

145 82c54b5b Michael Hanselmann
  """
146 82c54b5b Michael Hanselmann
  result = []
147 82c54b5b Michael Hanselmann
148 82c54b5b Michael Hanselmann
  for (idx, disk) in enumerate(disks):
149 82c54b5b Michael Hanselmann
    exp_iv_name = "disk/%s" % idx
150 82c54b5b Michael Hanselmann
    if disk.iv_name != exp_iv_name:
151 82c54b5b Michael Hanselmann
      result.append((idx, exp_iv_name, disk.iv_name))
152 82c54b5b Michael Hanselmann
153 82c54b5b Michael Hanselmann
  return result
154 82c54b5b Michael Hanselmann
155 82c54b5b Michael Hanselmann
156 a8083063 Iustin Pop
class ConfigWriter:
157 098c0958 Michael Hanselmann
  """The interface to the cluster configuration.
158 a8083063 Iustin Pop

159 d8aee57e Iustin Pop
  @ivar _temporary_lvs: reservation manager for temporary LVs
160 d8aee57e Iustin Pop
  @ivar _all_rms: a list of all temporary reservation managers
161 d8aee57e Iustin Pop

162 098c0958 Michael Hanselmann
  """
163 eb180fe2 Iustin Pop
  def __init__(self, cfg_file=None, offline=False, _getents=runtime.GetEnts,
164 eb180fe2 Iustin Pop
               accept_foreign=False):
165 14e15659 Iustin Pop
    self.write_count = 0
166 f78ede4e Guido Trotter
    self._lock = _config_lock
167 a8083063 Iustin Pop
    self._config_data = None
168 a8083063 Iustin Pop
    self._offline = offline
169 a8083063 Iustin Pop
    if cfg_file is None:
170 a8083063 Iustin Pop
      self._cfg_file = constants.CLUSTER_CONF_FILE
171 a8083063 Iustin Pop
    else:
172 a8083063 Iustin Pop
      self._cfg_file = cfg_file
173 e60c73a1 René Nussbaumer
    self._getents = _getents
174 4fae38c5 Guido Trotter
    self._temporary_ids = TemporaryReservationManager()
175 a81c53c9 Iustin Pop
    self._temporary_drbds = {}
176 36b66e6e Guido Trotter
    self._temporary_macs = TemporaryReservationManager()
177 afa1386e Guido Trotter
    self._temporary_secrets = TemporaryReservationManager()
178 d8aee57e Iustin Pop
    self._temporary_lvs = TemporaryReservationManager()
179 d8aee57e Iustin Pop
    self._all_rms = [self._temporary_ids, self._temporary_macs,
180 d8aee57e Iustin Pop
                     self._temporary_secrets, self._temporary_lvs]
181 89e1fc26 Iustin Pop
    # Note: in order to prevent errors when resolving our name in
182 89e1fc26 Iustin Pop
    # _DistributeConfig, we compute it here once and reuse it; it's
183 89e1fc26 Iustin Pop
    # better to raise an error before starting to modify the config
184 89e1fc26 Iustin Pop
    # file than after it was modified
185 b705c7a6 Manuel Franceschini
    self._my_hostname = netutils.Hostname.GetSysName()
186 3c7f6c44 Iustin Pop
    self._last_cluster_serial = -1
187 bd407597 Iustin Pop
    self._cfg_id = None
188 b2acdbdc Michael Hanselmann
    self._context = None
189 eb180fe2 Iustin Pop
    self._OpenConfig(accept_foreign)
190 a8083063 Iustin Pop
191 b2acdbdc Michael Hanselmann
  def _GetRpc(self, address_list):
192 b2acdbdc Michael Hanselmann
    """Returns RPC runner for configuration.
193 b2acdbdc Michael Hanselmann

194 b2acdbdc Michael Hanselmann
    """
195 b2acdbdc Michael Hanselmann
    return rpc.ConfigRunner(self._context, address_list)
196 b2acdbdc Michael Hanselmann
197 b2acdbdc Michael Hanselmann
  def SetContext(self, context):
198 b2acdbdc Michael Hanselmann
    """Sets Ganeti context.
199 b2acdbdc Michael Hanselmann

200 b2acdbdc Michael Hanselmann
    """
201 b2acdbdc Michael Hanselmann
    self._context = context
202 b2acdbdc Michael Hanselmann
203 a8083063 Iustin Pop
  # this method needs to be static, so that we can call it on the class
204 a8083063 Iustin Pop
  @staticmethod
205 a8083063 Iustin Pop
  def IsCluster():
206 a8083063 Iustin Pop
    """Check if the cluster is configured.
207 a8083063 Iustin Pop

208 a8083063 Iustin Pop
    """
209 a8083063 Iustin Pop
    return os.path.exists(constants.CLUSTER_CONF_FILE)
210 a8083063 Iustin Pop
211 36b66e6e Guido Trotter
  def _GenerateOneMAC(self):
212 36b66e6e Guido Trotter
    """Generate one mac address
213 36b66e6e Guido Trotter

214 36b66e6e Guido Trotter
    """
215 36b66e6e Guido Trotter
    prefix = self._config_data.cluster.mac_prefix
216 36b66e6e Guido Trotter
    byte1 = random.randrange(0, 256)
217 36b66e6e Guido Trotter
    byte2 = random.randrange(0, 256)
218 36b66e6e Guido Trotter
    byte3 = random.randrange(0, 256)
219 36b66e6e Guido Trotter
    mac = "%s:%02x:%02x:%02x" % (prefix, byte1, byte2, byte3)
220 36b66e6e Guido Trotter
    return mac
221 36b66e6e Guido Trotter
222 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
223 5768e6a6 René Nussbaumer
  def GetNdParams(self, node):
224 5768e6a6 René Nussbaumer
    """Get the node params populated with cluster defaults.
225 5768e6a6 René Nussbaumer

226 ce523de1 Michael Hanselmann
    @type node: L{objects.Node}
227 5768e6a6 René Nussbaumer
    @param node: The node we want to know the params for
228 5768e6a6 René Nussbaumer
    @return: A dict with the filled in node params
229 5768e6a6 René Nussbaumer

230 5768e6a6 René Nussbaumer
    """
231 5768e6a6 René Nussbaumer
    nodegroup = self._UnlockedGetNodeGroup(node.group)
232 5768e6a6 René Nussbaumer
    return self._config_data.cluster.FillND(node, nodegroup)
233 5768e6a6 René Nussbaumer
234 5768e6a6 René Nussbaumer
  @locking.ssynchronized(_config_lock, shared=1)
235 8a147bba René Nussbaumer
  def GetInstanceDiskParams(self, instance):
236 8a147bba René Nussbaumer
    """Get the disk params populated with inherit chain.
237 8a147bba René Nussbaumer

238 8a147bba René Nussbaumer
    @type instance: L{objects.Instance}
239 8a147bba René Nussbaumer
    @param instance: The instance we want to know the params for
240 8a147bba René Nussbaumer
    @return: A dict with the filled in disk params
241 8a147bba René Nussbaumer

242 8a147bba René Nussbaumer
    """
243 8a147bba René Nussbaumer
    node = self._UnlockedGetNodeInfo(instance.primary_node)
244 8a147bba René Nussbaumer
    nodegroup = self._UnlockedGetNodeGroup(node.group)
245 99ccf8b9 René Nussbaumer
    return self._UnlockedGetGroupDiskParams(nodegroup)
246 99ccf8b9 René Nussbaumer
247 99ccf8b9 René Nussbaumer
  @locking.ssynchronized(_config_lock, shared=1)
248 99ccf8b9 René Nussbaumer
  def GetGroupDiskParams(self, group):
249 99ccf8b9 René Nussbaumer
    """Get the disk params populated with inherit chain.
250 99ccf8b9 René Nussbaumer

251 af9fb4cc René Nussbaumer
    @type group: L{objects.NodeGroup}
252 99ccf8b9 René Nussbaumer
    @param group: The group we want to know the params for
253 99ccf8b9 René Nussbaumer
    @return: A dict with the filled in disk params
254 99ccf8b9 René Nussbaumer

255 99ccf8b9 René Nussbaumer
    """
256 99ccf8b9 René Nussbaumer
    return self._UnlockedGetGroupDiskParams(group)
257 99ccf8b9 René Nussbaumer
258 99ccf8b9 René Nussbaumer
  def _UnlockedGetGroupDiskParams(self, group):
259 99ccf8b9 René Nussbaumer
    """Get the disk params populated with inherit chain down to node-group.
260 99ccf8b9 René Nussbaumer

261 af9fb4cc René Nussbaumer
    @type group: L{objects.NodeGroup}
262 99ccf8b9 René Nussbaumer
    @param group: The group we want to know the params for
263 99ccf8b9 René Nussbaumer
    @return: A dict with the filled in disk params
264 99ccf8b9 René Nussbaumer

265 99ccf8b9 René Nussbaumer
    """
266 99ccf8b9 René Nussbaumer
    return self._config_data.cluster.SimpleFillDP(group.diskparams)
267 8a147bba René Nussbaumer
268 8a147bba René Nussbaumer
  @locking.ssynchronized(_config_lock, shared=1)
269 36b66e6e Guido Trotter
  def GenerateMAC(self, ec_id):
270 a8083063 Iustin Pop
    """Generate a MAC for an instance.
271 a8083063 Iustin Pop

272 a8083063 Iustin Pop
    This should check the current instances for duplicates.
273 a8083063 Iustin Pop

274 a8083063 Iustin Pop
    """
275 36b66e6e Guido Trotter
    existing = self._AllMACs()
276 36b66e6e Guido Trotter
    return self._temporary_ids.Generate(existing, self._GenerateOneMAC, ec_id)
277 a8083063 Iustin Pop
278 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
279 36b66e6e Guido Trotter
  def ReserveMAC(self, mac, ec_id):
280 36b66e6e Guido Trotter
    """Reserve a MAC for an instance.
281 1862d460 Alexander Schreiber

282 1862d460 Alexander Schreiber
    This only checks instances managed by this cluster, it does not
283 1862d460 Alexander Schreiber
    check for potential collisions elsewhere.
284 1862d460 Alexander Schreiber

285 1862d460 Alexander Schreiber
    """
286 1862d460 Alexander Schreiber
    all_macs = self._AllMACs()
287 36b66e6e Guido Trotter
    if mac in all_macs:
288 36b66e6e Guido Trotter
      raise errors.ReservationError("mac already in use")
289 36b66e6e Guido Trotter
    else:
290 8785b71b Apollon Oikonomopoulos
      self._temporary_macs.Reserve(ec_id, mac)
291 1862d460 Alexander Schreiber
292 f9518d38 Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
293 d8aee57e Iustin Pop
  def ReserveLV(self, lv_name, ec_id):
294 d8aee57e Iustin Pop
    """Reserve an VG/LV pair for an instance.
295 d8aee57e Iustin Pop

296 d8aee57e Iustin Pop
    @type lv_name: string
297 d8aee57e Iustin Pop
    @param lv_name: the logical volume name to reserve
298 d8aee57e Iustin Pop

299 d8aee57e Iustin Pop
    """
300 d8aee57e Iustin Pop
    all_lvs = self._AllLVs()
301 d8aee57e Iustin Pop
    if lv_name in all_lvs:
302 d8aee57e Iustin Pop
      raise errors.ReservationError("LV already in use")
303 d8aee57e Iustin Pop
    else:
304 8785b71b Apollon Oikonomopoulos
      self._temporary_lvs.Reserve(ec_id, lv_name)
305 d8aee57e Iustin Pop
306 d8aee57e Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
307 afa1386e Guido Trotter
  def GenerateDRBDSecret(self, ec_id):
308 f9518d38 Iustin Pop
    """Generate a DRBD secret.
309 f9518d38 Iustin Pop

310 f9518d38 Iustin Pop
    This checks the current disks for duplicates.
311 f9518d38 Iustin Pop

312 f9518d38 Iustin Pop
    """
313 afa1386e Guido Trotter
    return self._temporary_secrets.Generate(self._AllDRBDSecrets(),
314 afa1386e Guido Trotter
                                            utils.GenerateSecret,
315 afa1386e Guido Trotter
                                            ec_id)
316 8d9c3bef Michael Hanselmann
317 34e54ebc Iustin Pop
  def _AllLVs(self):
318 923b1523 Iustin Pop
    """Compute the list of all LVs.
319 923b1523 Iustin Pop

320 923b1523 Iustin Pop
    """
321 923b1523 Iustin Pop
    lvnames = set()
322 923b1523 Iustin Pop
    for instance in self._config_data.instances.values():
323 923b1523 Iustin Pop
      node_data = instance.MapLVsByNode()
324 923b1523 Iustin Pop
      for lv_list in node_data.values():
325 923b1523 Iustin Pop
        lvnames.update(lv_list)
326 923b1523 Iustin Pop
    return lvnames
327 923b1523 Iustin Pop
328 34e54ebc Iustin Pop
  def _AllIDs(self, include_temporary):
329 34e54ebc Iustin Pop
    """Compute the list of all UUIDs and names we have.
330 34e54ebc Iustin Pop

331 34e54ebc Iustin Pop
    @type include_temporary: boolean
332 34e54ebc Iustin Pop
    @param include_temporary: whether to include the _temporary_ids set
333 34e54ebc Iustin Pop
    @rtype: set
334 34e54ebc Iustin Pop
    @return: a set of IDs
335 34e54ebc Iustin Pop

336 34e54ebc Iustin Pop
    """
337 34e54ebc Iustin Pop
    existing = set()
338 34e54ebc Iustin Pop
    if include_temporary:
339 4fae38c5 Guido Trotter
      existing.update(self._temporary_ids.GetReserved())
340 34e54ebc Iustin Pop
    existing.update(self._AllLVs())
341 34e54ebc Iustin Pop
    existing.update(self._config_data.instances.keys())
342 34e54ebc Iustin Pop
    existing.update(self._config_data.nodes.keys())
343 76d5d3a3 Iustin Pop
    existing.update([i.uuid for i in self._AllUUIDObjects() if i.uuid])
344 34e54ebc Iustin Pop
    return existing
345 34e54ebc Iustin Pop
346 4fae38c5 Guido Trotter
  def _GenerateUniqueID(self, ec_id):
347 430b923c Iustin Pop
    """Generate an unique UUID.
348 923b1523 Iustin Pop

349 923b1523 Iustin Pop
    This checks the current node, instances and disk names for
350 923b1523 Iustin Pop
    duplicates.
351 923b1523 Iustin Pop

352 c41eea6e Iustin Pop
    @rtype: string
353 c41eea6e Iustin Pop
    @return: the unique id
354 923b1523 Iustin Pop

355 923b1523 Iustin Pop
    """
356 4fae38c5 Guido Trotter
    existing = self._AllIDs(include_temporary=False)
357 4fae38c5 Guido Trotter
    return self._temporary_ids.Generate(existing, utils.NewUUID, ec_id)
358 923b1523 Iustin Pop
359 430b923c Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
360 4fae38c5 Guido Trotter
  def GenerateUniqueID(self, ec_id):
361 430b923c Iustin Pop
    """Generate an unique ID.
362 430b923c Iustin Pop

363 430b923c Iustin Pop
    This is just a wrapper over the unlocked version.
364 430b923c Iustin Pop

365 4fae38c5 Guido Trotter
    @type ec_id: string
366 4fae38c5 Guido Trotter
    @param ec_id: unique id for the job to reserve the id to
367 34d657ba Iustin Pop

368 34d657ba Iustin Pop
    """
369 4fae38c5 Guido Trotter
    return self._GenerateUniqueID(ec_id)
370 34d657ba Iustin Pop
371 a8083063 Iustin Pop
  def _AllMACs(self):
372 a8083063 Iustin Pop
    """Return all MACs present in the config.
373 a8083063 Iustin Pop

374 c41eea6e Iustin Pop
    @rtype: list
375 c41eea6e Iustin Pop
    @return: the list of all MACs
376 c41eea6e Iustin Pop

377 a8083063 Iustin Pop
    """
378 a8083063 Iustin Pop
    result = []
379 a8083063 Iustin Pop
    for instance in self._config_data.instances.values():
380 a8083063 Iustin Pop
      for nic in instance.nics:
381 a8083063 Iustin Pop
        result.append(nic.mac)
382 a8083063 Iustin Pop
383 a8083063 Iustin Pop
    return result
384 a8083063 Iustin Pop
385 f9518d38 Iustin Pop
  def _AllDRBDSecrets(self):
386 f9518d38 Iustin Pop
    """Return all DRBD secrets present in the config.
387 f9518d38 Iustin Pop

388 c41eea6e Iustin Pop
    @rtype: list
389 c41eea6e Iustin Pop
    @return: the list of all DRBD secrets
390 c41eea6e Iustin Pop

391 f9518d38 Iustin Pop
    """
392 f9518d38 Iustin Pop
    def helper(disk, result):
393 f9518d38 Iustin Pop
      """Recursively gather secrets from this disk."""
394 f9518d38 Iustin Pop
      if disk.dev_type == constants.DT_DRBD8:
395 f9518d38 Iustin Pop
        result.append(disk.logical_id[5])
396 f9518d38 Iustin Pop
      if disk.children:
397 f9518d38 Iustin Pop
        for child in disk.children:
398 f9518d38 Iustin Pop
          helper(child, result)
399 f9518d38 Iustin Pop
400 f9518d38 Iustin Pop
    result = []
401 f9518d38 Iustin Pop
    for instance in self._config_data.instances.values():
402 f9518d38 Iustin Pop
      for disk in instance.disks:
403 f9518d38 Iustin Pop
        helper(disk, result)
404 f9518d38 Iustin Pop
405 f9518d38 Iustin Pop
    return result
406 f9518d38 Iustin Pop
407 4b98ac29 Iustin Pop
  def _CheckDiskIDs(self, disk, l_ids, p_ids):
408 4b98ac29 Iustin Pop
    """Compute duplicate disk IDs
409 4b98ac29 Iustin Pop

410 4b98ac29 Iustin Pop
    @type disk: L{objects.Disk}
411 4b98ac29 Iustin Pop
    @param disk: the disk at which to start searching
412 4b98ac29 Iustin Pop
    @type l_ids: list
413 4b98ac29 Iustin Pop
    @param l_ids: list of current logical ids
414 4b98ac29 Iustin Pop
    @type p_ids: list
415 4b98ac29 Iustin Pop
    @param p_ids: list of current physical ids
416 4b98ac29 Iustin Pop
    @rtype: list
417 4b98ac29 Iustin Pop
    @return: a list of error messages
418 4b98ac29 Iustin Pop

419 4b98ac29 Iustin Pop
    """
420 4b98ac29 Iustin Pop
    result = []
421 25ae22e4 Iustin Pop
    if disk.logical_id is not None:
422 25ae22e4 Iustin Pop
      if disk.logical_id in l_ids:
423 25ae22e4 Iustin Pop
        result.append("duplicate logical id %s" % str(disk.logical_id))
424 25ae22e4 Iustin Pop
      else:
425 25ae22e4 Iustin Pop
        l_ids.append(disk.logical_id)
426 25ae22e4 Iustin Pop
    if disk.physical_id is not None:
427 25ae22e4 Iustin Pop
      if disk.physical_id in p_ids:
428 25ae22e4 Iustin Pop
        result.append("duplicate physical id %s" % str(disk.physical_id))
429 25ae22e4 Iustin Pop
      else:
430 25ae22e4 Iustin Pop
        p_ids.append(disk.physical_id)
431 4b98ac29 Iustin Pop
432 4b98ac29 Iustin Pop
    if disk.children:
433 4b98ac29 Iustin Pop
      for child in disk.children:
434 4b98ac29 Iustin Pop
        result.extend(self._CheckDiskIDs(child, l_ids, p_ids))
435 4b98ac29 Iustin Pop
    return result
436 4b98ac29 Iustin Pop
437 4a89c54a Iustin Pop
  def _UnlockedVerifyConfig(self):
438 a8efbb40 Iustin Pop
    """Verify function.
439 a8efbb40 Iustin Pop

440 4a89c54a Iustin Pop
    @rtype: list
441 4a89c54a Iustin Pop
    @return: a list of error messages; a non-empty list signifies
442 4a89c54a Iustin Pop
        configuration errors
443 4a89c54a Iustin Pop

444 a8083063 Iustin Pop
    """
445 b459a848 Andrea Spadaccini
    # pylint: disable=R0914
446 a8083063 Iustin Pop
    result = []
447 a8083063 Iustin Pop
    seen_macs = []
448 48ce9fd9 Iustin Pop
    ports = {}
449 a8083063 Iustin Pop
    data = self._config_data
450 7e01d204 Iustin Pop
    cluster = data.cluster
451 4b98ac29 Iustin Pop
    seen_lids = []
452 4b98ac29 Iustin Pop
    seen_pids = []
453 9a5fba23 Guido Trotter
454 9a5fba23 Guido Trotter
    # global cluster checks
455 7e01d204 Iustin Pop
    if not cluster.enabled_hypervisors:
456 9a5fba23 Guido Trotter
      result.append("enabled hypervisors list doesn't have any entries")
457 7e01d204 Iustin Pop
    invalid_hvs = set(cluster.enabled_hypervisors) - constants.HYPER_TYPES
458 9a5fba23 Guido Trotter
    if invalid_hvs:
459 9a5fba23 Guido Trotter
      result.append("enabled hypervisors contains invalid entries: %s" %
460 9a5fba23 Guido Trotter
                    invalid_hvs)
461 7e01d204 Iustin Pop
    missing_hvp = (set(cluster.enabled_hypervisors) -
462 7e01d204 Iustin Pop
                   set(cluster.hvparams.keys()))
463 9f3ac970 Iustin Pop
    if missing_hvp:
464 9f3ac970 Iustin Pop
      result.append("hypervisor parameters missing for the enabled"
465 9f3ac970 Iustin Pop
                    " hypervisor(s) %s" % utils.CommaJoin(missing_hvp))
466 9a5fba23 Guido Trotter
467 7e01d204 Iustin Pop
    if cluster.master_node not in data.nodes:
468 9a5fba23 Guido Trotter
      result.append("cluster has invalid primary node '%s'" %
469 7e01d204 Iustin Pop
                    cluster.master_node)
470 9a5fba23 Guido Trotter
471 26f2fd8d Iustin Pop
    def _helper(owner, attr, value, template):
472 26f2fd8d Iustin Pop
      try:
473 26f2fd8d Iustin Pop
        utils.ForceDictType(value, template)
474 26f2fd8d Iustin Pop
      except errors.GenericError, err:
475 26f2fd8d Iustin Pop
        result.append("%s has invalid %s: %s" % (owner, attr, err))
476 26f2fd8d Iustin Pop
477 26f2fd8d Iustin Pop
    def _helper_nic(owner, params):
478 26f2fd8d Iustin Pop
      try:
479 26f2fd8d Iustin Pop
        objects.NIC.CheckParameterSyntax(params)
480 26f2fd8d Iustin Pop
      except errors.ConfigurationError, err:
481 26f2fd8d Iustin Pop
        result.append("%s has invalid nicparams: %s" % (owner, err))
482 26f2fd8d Iustin Pop
483 918eb80b Agata Murawska
    def _helper_ipolicy(owner, params):
484 918eb80b Agata Murawska
      try:
485 918eb80b Agata Murawska
        objects.InstancePolicy.CheckParameterSyntax(params)
486 918eb80b Agata Murawska
      except errors.ConfigurationError, err:
487 918eb80b Agata Murawska
        result.append("%s has invalid instance policy: %s" % (owner, err))
488 918eb80b Agata Murawska
489 918eb80b Agata Murawska
    def _helper_ispecs(owner, params):
490 2cc673a3 Iustin Pop
      for key, value in params.items():
491 12378fe3 Iustin Pop
        if key in constants.IPOLICY_ISPECS:
492 2cc673a3 Iustin Pop
          fullkey = "ipolicy/" + key
493 2cc673a3 Iustin Pop
          _helper(owner, fullkey, value, constants.ISPECS_PARAMETER_TYPES)
494 2cc673a3 Iustin Pop
        else:
495 2cc673a3 Iustin Pop
          # FIXME: assuming list type
496 ff6c5e55 Iustin Pop
          if key in constants.IPOLICY_PARAMETERS:
497 ff6c5e55 Iustin Pop
            exp_type = float
498 ff6c5e55 Iustin Pop
          else:
499 ff6c5e55 Iustin Pop
            exp_type = list
500 ff6c5e55 Iustin Pop
          if not isinstance(value, exp_type):
501 2cc673a3 Iustin Pop
            result.append("%s has invalid instance policy: for %s,"
502 ff6c5e55 Iustin Pop
                          " expecting %s, got %s" %
503 ff6c5e55 Iustin Pop
                          (owner, key, exp_type.__name__, type(value)))
504 918eb80b Agata Murawska
505 26f2fd8d Iustin Pop
    # check cluster parameters
506 26f2fd8d Iustin Pop
    _helper("cluster", "beparams", cluster.SimpleFillBE({}),
507 26f2fd8d Iustin Pop
            constants.BES_PARAMETER_TYPES)
508 26f2fd8d Iustin Pop
    _helper("cluster", "nicparams", cluster.SimpleFillNIC({}),
509 26f2fd8d Iustin Pop
            constants.NICS_PARAMETER_TYPES)
510 26f2fd8d Iustin Pop
    _helper_nic("cluster", cluster.SimpleFillNIC({}))
511 26f2fd8d Iustin Pop
    _helper("cluster", "ndparams", cluster.SimpleFillND({}),
512 26f2fd8d Iustin Pop
            constants.NDS_PARAMETER_TYPES)
513 918eb80b Agata Murawska
    _helper_ipolicy("cluster", cluster.SimpleFillIPolicy({}))
514 918eb80b Agata Murawska
    _helper_ispecs("cluster", cluster.SimpleFillIPolicy({}))
515 26f2fd8d Iustin Pop
516 9a5fba23 Guido Trotter
    # per-instance checks
517 a8083063 Iustin Pop
    for instance_name in data.instances:
518 a8083063 Iustin Pop
      instance = data.instances[instance_name]
519 81196341 Iustin Pop
      if instance.name != instance_name:
520 81196341 Iustin Pop
        result.append("instance '%s' is indexed by wrong name '%s'" %
521 81196341 Iustin Pop
                      (instance.name, instance_name))
522 a8083063 Iustin Pop
      if instance.primary_node not in data.nodes:
523 8522ceeb Iustin Pop
        result.append("instance '%s' has invalid primary node '%s'" %
524 a8083063 Iustin Pop
                      (instance_name, instance.primary_node))
525 a8083063 Iustin Pop
      for snode in instance.secondary_nodes:
526 a8083063 Iustin Pop
        if snode not in data.nodes:
527 8522ceeb Iustin Pop
          result.append("instance '%s' has invalid secondary node '%s'" %
528 a8083063 Iustin Pop
                        (instance_name, snode))
529 a8083063 Iustin Pop
      for idx, nic in enumerate(instance.nics):
530 a8083063 Iustin Pop
        if nic.mac in seen_macs:
531 8522ceeb Iustin Pop
          result.append("instance '%s' has NIC %d mac %s duplicate" %
532 a8083063 Iustin Pop
                        (instance_name, idx, nic.mac))
533 a8083063 Iustin Pop
        else:
534 a8083063 Iustin Pop
          seen_macs.append(nic.mac)
535 26f2fd8d Iustin Pop
        if nic.nicparams:
536 26f2fd8d Iustin Pop
          filled = cluster.SimpleFillNIC(nic.nicparams)
537 26f2fd8d Iustin Pop
          owner = "instance %s nic %d" % (instance.name, idx)
538 26f2fd8d Iustin Pop
          _helper(owner, "nicparams",
539 26f2fd8d Iustin Pop
                  filled, constants.NICS_PARAMETER_TYPES)
540 26f2fd8d Iustin Pop
          _helper_nic(owner, filled)
541 26f2fd8d Iustin Pop
542 26f2fd8d Iustin Pop
      # parameter checks
543 26f2fd8d Iustin Pop
      if instance.beparams:
544 26f2fd8d Iustin Pop
        _helper("instance %s" % instance.name, "beparams",
545 26f2fd8d Iustin Pop
                cluster.FillBE(instance), constants.BES_PARAMETER_TYPES)
546 48ce9fd9 Iustin Pop
547 48ce9fd9 Iustin Pop
      # gather the drbd ports for duplicate checks
548 e2569c1d Michael Hanselmann
      for (idx, dsk) in enumerate(instance.disks):
549 48ce9fd9 Iustin Pop
        if dsk.dev_type in constants.LDS_DRBD:
550 48ce9fd9 Iustin Pop
          tcp_port = dsk.logical_id[2]
551 48ce9fd9 Iustin Pop
          if tcp_port not in ports:
552 48ce9fd9 Iustin Pop
            ports[tcp_port] = []
553 e2569c1d Michael Hanselmann
          ports[tcp_port].append((instance.name, "drbd disk %s" % idx))
554 48ce9fd9 Iustin Pop
      # gather network port reservation
555 48ce9fd9 Iustin Pop
      net_port = getattr(instance, "network_port", None)
556 48ce9fd9 Iustin Pop
      if net_port is not None:
557 48ce9fd9 Iustin Pop
        if net_port not in ports:
558 48ce9fd9 Iustin Pop
          ports[net_port] = []
559 48ce9fd9 Iustin Pop
        ports[net_port].append((instance.name, "network port"))
560 48ce9fd9 Iustin Pop
561 332d0e37 Iustin Pop
      # instance disk verify
562 332d0e37 Iustin Pop
      for idx, disk in enumerate(instance.disks):
563 332d0e37 Iustin Pop
        result.extend(["instance '%s' disk %d error: %s" %
564 332d0e37 Iustin Pop
                       (instance.name, idx, msg) for msg in disk.Verify()])
565 4b98ac29 Iustin Pop
        result.extend(self._CheckDiskIDs(disk, seen_lids, seen_pids))
566 332d0e37 Iustin Pop
567 82c54b5b Michael Hanselmann
      wrong_names = _CheckInstanceDiskIvNames(instance.disks)
568 82c54b5b Michael Hanselmann
      if wrong_names:
569 82c54b5b Michael Hanselmann
        tmp = "; ".join(("name of disk %s should be '%s', but is '%s'" %
570 82c54b5b Michael Hanselmann
                         (idx, exp_name, actual_name))
571 82c54b5b Michael Hanselmann
                        for (idx, exp_name, actual_name) in wrong_names)
572 82c54b5b Michael Hanselmann
573 82c54b5b Michael Hanselmann
        result.append("Instance '%s' has wrongly named disks: %s" %
574 82c54b5b Michael Hanselmann
                      (instance.name, tmp))
575 82c54b5b Michael Hanselmann
576 48ce9fd9 Iustin Pop
    # cluster-wide pool of free ports
577 7e01d204 Iustin Pop
    for free_port in cluster.tcpudp_port_pool:
578 48ce9fd9 Iustin Pop
      if free_port not in ports:
579 48ce9fd9 Iustin Pop
        ports[free_port] = []
580 48ce9fd9 Iustin Pop
      ports[free_port].append(("cluster", "port marked as free"))
581 48ce9fd9 Iustin Pop
582 48ce9fd9 Iustin Pop
    # compute tcp/udp duplicate ports
583 48ce9fd9 Iustin Pop
    keys = ports.keys()
584 48ce9fd9 Iustin Pop
    keys.sort()
585 48ce9fd9 Iustin Pop
    for pnum in keys:
586 48ce9fd9 Iustin Pop
      pdata = ports[pnum]
587 48ce9fd9 Iustin Pop
      if len(pdata) > 1:
588 1f864b60 Iustin Pop
        txt = utils.CommaJoin(["%s/%s" % val for val in pdata])
589 48ce9fd9 Iustin Pop
        result.append("tcp/udp port %s has duplicates: %s" % (pnum, txt))
590 48ce9fd9 Iustin Pop
591 48ce9fd9 Iustin Pop
    # highest used tcp port check
592 48ce9fd9 Iustin Pop
    if keys:
593 7e01d204 Iustin Pop
      if keys[-1] > cluster.highest_used_port:
594 48ce9fd9 Iustin Pop
        result.append("Highest used port mismatch, saved %s, computed %s" %
595 7e01d204 Iustin Pop
                      (cluster.highest_used_port, keys[-1]))
596 a8efbb40 Iustin Pop
597 7e01d204 Iustin Pop
    if not data.nodes[cluster.master_node].master_candidate:
598 3a26773f Iustin Pop
      result.append("Master node is not a master candidate")
599 3a26773f Iustin Pop
600 4a89c54a Iustin Pop
    # master candidate checks
601 e623dbe3 Guido Trotter
    mc_now, mc_max, _ = self._UnlockedGetMasterCandidateStats()
602 ec0292f1 Iustin Pop
    if mc_now < mc_max:
603 ec0292f1 Iustin Pop
      result.append("Not enough master candidates: actual %d, target %d" %
604 ec0292f1 Iustin Pop
                    (mc_now, mc_max))
605 48ce9fd9 Iustin Pop
606 5bf07049 Iustin Pop
    # node checks
607 81196341 Iustin Pop
    for node_name, node in data.nodes.items():
608 81196341 Iustin Pop
      if node.name != node_name:
609 81196341 Iustin Pop
        result.append("Node '%s' is indexed by wrong name '%s'" %
610 81196341 Iustin Pop
                      (node.name, node_name))
611 5bf07049 Iustin Pop
      if [node.master_candidate, node.drained, node.offline].count(True) > 1:
612 5bf07049 Iustin Pop
        result.append("Node %s state is invalid: master_candidate=%s,"
613 5bf07049 Iustin Pop
                      " drain=%s, offline=%s" %
614 3d889a7d Michael Hanselmann
                      (node.name, node.master_candidate, node.drained,
615 5bf07049 Iustin Pop
                       node.offline))
616 26f2fd8d Iustin Pop
      if node.group not in data.nodegroups:
617 26f2fd8d Iustin Pop
        result.append("Node '%s' has invalid group '%s'" %
618 26f2fd8d Iustin Pop
                      (node.name, node.group))
619 26f2fd8d Iustin Pop
      else:
620 26f2fd8d Iustin Pop
        _helper("node %s" % node.name, "ndparams",
621 26f2fd8d Iustin Pop
                cluster.FillND(node, data.nodegroups[node.group]),
622 26f2fd8d Iustin Pop
                constants.NDS_PARAMETER_TYPES)
623 5bf07049 Iustin Pop
624 6520ba14 Guido Trotter
    # nodegroups checks
625 ace16501 Guido Trotter
    nodegroups_names = set()
626 6520ba14 Guido Trotter
    for nodegroup_uuid in data.nodegroups:
627 6520ba14 Guido Trotter
      nodegroup = data.nodegroups[nodegroup_uuid]
628 6520ba14 Guido Trotter
      if nodegroup.uuid != nodegroup_uuid:
629 913cc25e Adeodato Simo
        result.append("node group '%s' (uuid: '%s') indexed by wrong uuid '%s'"
630 6520ba14 Guido Trotter
                      % (nodegroup.name, nodegroup.uuid, nodegroup_uuid))
631 485ba212 Guido Trotter
      if utils.UUID_RE.match(nodegroup.name.lower()):
632 913cc25e Adeodato Simo
        result.append("node group '%s' (uuid: '%s') has uuid-like name" %
633 485ba212 Guido Trotter
                      (nodegroup.name, nodegroup.uuid))
634 ace16501 Guido Trotter
      if nodegroup.name in nodegroups_names:
635 913cc25e Adeodato Simo
        result.append("duplicate node group name '%s'" % nodegroup.name)
636 ace16501 Guido Trotter
      else:
637 ace16501 Guido Trotter
        nodegroups_names.add(nodegroup.name)
638 81e3ab4f Agata Murawska
      group_name = "group %s" % nodegroup.name
639 81e3ab4f Agata Murawska
      _helper_ipolicy(group_name, cluster.SimpleFillIPolicy(nodegroup.ipolicy))
640 81e3ab4f Agata Murawska
      _helper_ispecs(group_name, cluster.SimpleFillIPolicy(nodegroup.ipolicy))
641 26f2fd8d Iustin Pop
      if nodegroup.ndparams:
642 81e3ab4f Agata Murawska
        _helper(group_name, "ndparams",
643 26f2fd8d Iustin Pop
                cluster.SimpleFillND(nodegroup.ndparams),
644 26f2fd8d Iustin Pop
                constants.NDS_PARAMETER_TYPES)
645 26f2fd8d Iustin Pop
646 4a89c54a Iustin Pop
    # drbd minors check
647 1122eb25 Iustin Pop
    _, duplicates = self._UnlockedComputeDRBDMap()
648 4a89c54a Iustin Pop
    for node, minor, instance_a, instance_b in duplicates:
649 4a89c54a Iustin Pop
      result.append("DRBD minor %d on node %s is assigned twice to instances"
650 4a89c54a Iustin Pop
                    " %s and %s" % (minor, node, instance_a, instance_b))
651 4a89c54a Iustin Pop
652 0ce8f948 Iustin Pop
    # IP checks
653 7e01d204 Iustin Pop
    default_nicparams = cluster.nicparams[constants.PP_DEFAULT]
654 b8716596 Michael Hanselmann
    ips = {}
655 b8716596 Michael Hanselmann
656 b8716596 Michael Hanselmann
    def _AddIpAddress(ip, name):
657 b8716596 Michael Hanselmann
      ips.setdefault(ip, []).append(name)
658 b8716596 Michael Hanselmann
659 7e01d204 Iustin Pop
    _AddIpAddress(cluster.master_ip, "cluster_ip")
660 0ce8f948 Iustin Pop
661 0ce8f948 Iustin Pop
    for node in data.nodes.values():
662 b8716596 Michael Hanselmann
      _AddIpAddress(node.primary_ip, "node:%s/primary" % node.name)
663 0ce8f948 Iustin Pop
      if node.secondary_ip != node.primary_ip:
664 b8716596 Michael Hanselmann
        _AddIpAddress(node.secondary_ip, "node:%s/secondary" % node.name)
665 b8716596 Michael Hanselmann
666 b8716596 Michael Hanselmann
    for instance in data.instances.values():
667 b8716596 Michael Hanselmann
      for idx, nic in enumerate(instance.nics):
668 b8716596 Michael Hanselmann
        if nic.ip is None:
669 b8716596 Michael Hanselmann
          continue
670 b8716596 Michael Hanselmann
671 b8716596 Michael Hanselmann
        nicparams = objects.FillDict(default_nicparams, nic.nicparams)
672 b8716596 Michael Hanselmann
        nic_mode = nicparams[constants.NIC_MODE]
673 b8716596 Michael Hanselmann
        nic_link = nicparams[constants.NIC_LINK]
674 b8716596 Michael Hanselmann
675 b8716596 Michael Hanselmann
        if nic_mode == constants.NIC_MODE_BRIDGED:
676 b8716596 Michael Hanselmann
          link = "bridge:%s" % nic_link
677 b8716596 Michael Hanselmann
        elif nic_mode == constants.NIC_MODE_ROUTED:
678 b8716596 Michael Hanselmann
          link = "route:%s" % nic_link
679 b8716596 Michael Hanselmann
        else:
680 b8716596 Michael Hanselmann
          raise errors.ProgrammerError("NIC mode '%s' not handled" % nic_mode)
681 b8716596 Michael Hanselmann
682 b8716596 Michael Hanselmann
        _AddIpAddress("%s/%s" % (link, nic.ip),
683 b8716596 Michael Hanselmann
                      "instance:%s/nic:%d" % (instance.name, idx))
684 0ce8f948 Iustin Pop
685 0ce8f948 Iustin Pop
    for ip, owners in ips.items():
686 0ce8f948 Iustin Pop
      if len(owners) > 1:
687 0ce8f948 Iustin Pop
        result.append("IP address %s is used by multiple owners: %s" %
688 1f864b60 Iustin Pop
                      (ip, utils.CommaJoin(owners)))
689 b8716596 Michael Hanselmann
690 a8083063 Iustin Pop
    return result
691 a8083063 Iustin Pop
692 4a89c54a Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
693 4a89c54a Iustin Pop
  def VerifyConfig(self):
694 4a89c54a Iustin Pop
    """Verify function.
695 4a89c54a Iustin Pop

696 4a89c54a Iustin Pop
    This is just a wrapper over L{_UnlockedVerifyConfig}.
697 4a89c54a Iustin Pop

698 4a89c54a Iustin Pop
    @rtype: list
699 4a89c54a Iustin Pop
    @return: a list of error messages; a non-empty list signifies
700 4a89c54a Iustin Pop
        configuration errors
701 4a89c54a Iustin Pop

702 4a89c54a Iustin Pop
    """
703 4a89c54a Iustin Pop
    return self._UnlockedVerifyConfig()
704 4a89c54a Iustin Pop
705 f78ede4e Guido Trotter
  def _UnlockedSetDiskID(self, disk, node_name):
706 a8083063 Iustin Pop
    """Convert the unique ID to the ID needed on the target nodes.
707 a8083063 Iustin Pop

708 a8083063 Iustin Pop
    This is used only for drbd, which needs ip/port configuration.
709 a8083063 Iustin Pop

710 a8083063 Iustin Pop
    The routine descends down and updates its children also, because
711 a8083063 Iustin Pop
    this helps when the only the top device is passed to the remote
712 a8083063 Iustin Pop
    node.
713 a8083063 Iustin Pop

714 f78ede4e Guido Trotter
    This function is for internal use, when the config lock is already held.
715 f78ede4e Guido Trotter

716 a8083063 Iustin Pop
    """
717 a8083063 Iustin Pop
    if disk.children:
718 a8083063 Iustin Pop
      for child in disk.children:
719 f78ede4e Guido Trotter
        self._UnlockedSetDiskID(child, node_name)
720 a8083063 Iustin Pop
721 a8083063 Iustin Pop
    if disk.logical_id is None and disk.physical_id is not None:
722 a8083063 Iustin Pop
      return
723 ffa1c0dc Iustin Pop
    if disk.dev_type == constants.LD_DRBD8:
724 f9518d38 Iustin Pop
      pnode, snode, port, pminor, sminor, secret = disk.logical_id
725 a8083063 Iustin Pop
      if node_name not in (pnode, snode):
726 3ecf6786 Iustin Pop
        raise errors.ConfigurationError("DRBD device not knowing node %s" %
727 3ecf6786 Iustin Pop
                                        node_name)
728 f78ede4e Guido Trotter
      pnode_info = self._UnlockedGetNodeInfo(pnode)
729 f78ede4e Guido Trotter
      snode_info = self._UnlockedGetNodeInfo(snode)
730 a8083063 Iustin Pop
      if pnode_info is None or snode_info is None:
731 a8083063 Iustin Pop
        raise errors.ConfigurationError("Can't find primary or secondary node"
732 a8083063 Iustin Pop
                                        " for %s" % str(disk))
733 ffa1c0dc Iustin Pop
      p_data = (pnode_info.secondary_ip, port)
734 ffa1c0dc Iustin Pop
      s_data = (snode_info.secondary_ip, port)
735 a8083063 Iustin Pop
      if pnode == node_name:
736 f9518d38 Iustin Pop
        disk.physical_id = p_data + s_data + (pminor, secret)
737 a8083063 Iustin Pop
      else: # it must be secondary, we tested above
738 f9518d38 Iustin Pop
        disk.physical_id = s_data + p_data + (sminor, secret)
739 a8083063 Iustin Pop
    else:
740 a8083063 Iustin Pop
      disk.physical_id = disk.logical_id
741 a8083063 Iustin Pop
    return
742 a8083063 Iustin Pop
743 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock)
744 f78ede4e Guido Trotter
  def SetDiskID(self, disk, node_name):
745 f78ede4e Guido Trotter
    """Convert the unique ID to the ID needed on the target nodes.
746 f78ede4e Guido Trotter

747 f78ede4e Guido Trotter
    This is used only for drbd, which needs ip/port configuration.
748 f78ede4e Guido Trotter

749 f78ede4e Guido Trotter
    The routine descends down and updates its children also, because
750 f78ede4e Guido Trotter
    this helps when the only the top device is passed to the remote
751 f78ede4e Guido Trotter
    node.
752 f78ede4e Guido Trotter

753 f78ede4e Guido Trotter
    """
754 f78ede4e Guido Trotter
    return self._UnlockedSetDiskID(disk, node_name)
755 f78ede4e Guido Trotter
756 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock)
757 b2fddf63 Iustin Pop
  def AddTcpUdpPort(self, port):
758 b2fddf63 Iustin Pop
    """Adds a new port to the available port pool.
759 b2fddf63 Iustin Pop

760 3b3b1bca Dimitris Aragiorgis
    @warning: this method does not "flush" the configuration (via
761 3b3b1bca Dimitris Aragiorgis
        L{_WriteConfig}); callers should do that themselves once the
762 3b3b1bca Dimitris Aragiorgis
        configuration is stable
763 3b3b1bca Dimitris Aragiorgis

764 b2fddf63 Iustin Pop
    """
765 264bb3c5 Michael Hanselmann
    if not isinstance(port, int):
766 3ecf6786 Iustin Pop
      raise errors.ProgrammerError("Invalid type passed for port")
767 264bb3c5 Michael Hanselmann
768 b2fddf63 Iustin Pop
    self._config_data.cluster.tcpudp_port_pool.add(port)
769 264bb3c5 Michael Hanselmann
770 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
771 b2fddf63 Iustin Pop
  def GetPortList(self):
772 264bb3c5 Michael Hanselmann
    """Returns a copy of the current port list.
773 264bb3c5 Michael Hanselmann

774 264bb3c5 Michael Hanselmann
    """
775 b2fddf63 Iustin Pop
    return self._config_data.cluster.tcpudp_port_pool.copy()
776 264bb3c5 Michael Hanselmann
777 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock)
778 a8083063 Iustin Pop
  def AllocatePort(self):
779 a8083063 Iustin Pop
    """Allocate a port.
780 a8083063 Iustin Pop

781 b2fddf63 Iustin Pop
    The port will be taken from the available port pool or from the
782 b2fddf63 Iustin Pop
    default port range (and in this case we increase
783 b2fddf63 Iustin Pop
    highest_used_port).
784 a8083063 Iustin Pop

785 a8083063 Iustin Pop
    """
786 264bb3c5 Michael Hanselmann
    # If there are TCP/IP ports configured, we use them first.
787 b2fddf63 Iustin Pop
    if self._config_data.cluster.tcpudp_port_pool:
788 b2fddf63 Iustin Pop
      port = self._config_data.cluster.tcpudp_port_pool.pop()
789 264bb3c5 Michael Hanselmann
    else:
790 264bb3c5 Michael Hanselmann
      port = self._config_data.cluster.highest_used_port + 1
791 264bb3c5 Michael Hanselmann
      if port >= constants.LAST_DRBD_PORT:
792 3ecf6786 Iustin Pop
        raise errors.ConfigurationError("The highest used port is greater"
793 3ecf6786 Iustin Pop
                                        " than %s. Aborting." %
794 3ecf6786 Iustin Pop
                                        constants.LAST_DRBD_PORT)
795 264bb3c5 Michael Hanselmann
      self._config_data.cluster.highest_used_port = port
796 a8083063 Iustin Pop
797 a8083063 Iustin Pop
    self._WriteConfig()
798 a8083063 Iustin Pop
    return port
799 a8083063 Iustin Pop
800 6d2e83d5 Iustin Pop
  def _UnlockedComputeDRBDMap(self):
801 a81c53c9 Iustin Pop
    """Compute the used DRBD minor/nodes.
802 a81c53c9 Iustin Pop

803 4a89c54a Iustin Pop
    @rtype: (dict, list)
804 c41eea6e Iustin Pop
    @return: dictionary of node_name: dict of minor: instance_name;
805 c41eea6e Iustin Pop
        the returned dict will have all the nodes in it (even if with
806 4a89c54a Iustin Pop
        an empty list), and a list of duplicates; if the duplicates
807 4a89c54a Iustin Pop
        list is not empty, the configuration is corrupted and its caller
808 4a89c54a Iustin Pop
        should raise an exception
809 a81c53c9 Iustin Pop

810 a81c53c9 Iustin Pop
    """
811 a81c53c9 Iustin Pop
    def _AppendUsedPorts(instance_name, disk, used):
812 4a89c54a Iustin Pop
      duplicates = []
813 f9518d38 Iustin Pop
      if disk.dev_type == constants.LD_DRBD8 and len(disk.logical_id) >= 5:
814 7c4d6c7b Michael Hanselmann
        node_a, node_b, _, minor_a, minor_b = disk.logical_id[:5]
815 7c4d6c7b Michael Hanselmann
        for node, port in ((node_a, minor_a), (node_b, minor_b)):
816 4a89c54a Iustin Pop
          assert node in used, ("Node '%s' of instance '%s' not found"
817 4a89c54a Iustin Pop
                                " in node list" % (node, instance_name))
818 a81c53c9 Iustin Pop
          if port in used[node]:
819 4a89c54a Iustin Pop
            duplicates.append((node, port, instance_name, used[node][port]))
820 4a89c54a Iustin Pop
          else:
821 4a89c54a Iustin Pop
            used[node][port] = instance_name
822 a81c53c9 Iustin Pop
      if disk.children:
823 a81c53c9 Iustin Pop
        for child in disk.children:
824 4a89c54a Iustin Pop
          duplicates.extend(_AppendUsedPorts(instance_name, child, used))
825 4a89c54a Iustin Pop
      return duplicates
826 a81c53c9 Iustin Pop
827 4a89c54a Iustin Pop
    duplicates = []
828 a81c53c9 Iustin Pop
    my_dict = dict((node, {}) for node in self._config_data.nodes)
829 79b26a7a Iustin Pop
    for instance in self._config_data.instances.itervalues():
830 79b26a7a Iustin Pop
      for disk in instance.disks:
831 79b26a7a Iustin Pop
        duplicates.extend(_AppendUsedPorts(instance.name, disk, my_dict))
832 a81c53c9 Iustin Pop
    for (node, minor), instance in self._temporary_drbds.iteritems():
833 79b26a7a Iustin Pop
      if minor in my_dict[node] and my_dict[node][minor] != instance:
834 4a89c54a Iustin Pop
        duplicates.append((node, minor, instance, my_dict[node][minor]))
835 4a89c54a Iustin Pop
      else:
836 4a89c54a Iustin Pop
        my_dict[node][minor] = instance
837 4a89c54a Iustin Pop
    return my_dict, duplicates
838 a81c53c9 Iustin Pop
839 a81c53c9 Iustin Pop
  @locking.ssynchronized(_config_lock)
840 6d2e83d5 Iustin Pop
  def ComputeDRBDMap(self):
841 6d2e83d5 Iustin Pop
    """Compute the used DRBD minor/nodes.
842 6d2e83d5 Iustin Pop

843 6d2e83d5 Iustin Pop
    This is just a wrapper over L{_UnlockedComputeDRBDMap}.
844 6d2e83d5 Iustin Pop

845 6d2e83d5 Iustin Pop
    @return: dictionary of node_name: dict of minor: instance_name;
846 6d2e83d5 Iustin Pop
        the returned dict will have all the nodes in it (even if with
847 6d2e83d5 Iustin Pop
        an empty list).
848 6d2e83d5 Iustin Pop

849 6d2e83d5 Iustin Pop
    """
850 4a89c54a Iustin Pop
    d_map, duplicates = self._UnlockedComputeDRBDMap()
851 4a89c54a Iustin Pop
    if duplicates:
852 4a89c54a Iustin Pop
      raise errors.ConfigurationError("Duplicate DRBD ports detected: %s" %
853 4a89c54a Iustin Pop
                                      str(duplicates))
854 4a89c54a Iustin Pop
    return d_map
855 6d2e83d5 Iustin Pop
856 6d2e83d5 Iustin Pop
  @locking.ssynchronized(_config_lock)
857 a81c53c9 Iustin Pop
  def AllocateDRBDMinor(self, nodes, instance):
858 a81c53c9 Iustin Pop
    """Allocate a drbd minor.
859 a81c53c9 Iustin Pop

860 a81c53c9 Iustin Pop
    The free minor will be automatically computed from the existing
861 a81c53c9 Iustin Pop
    devices. A node can be given multiple times in order to allocate
862 a81c53c9 Iustin Pop
    multiple minors. The result is the list of minors, in the same
863 a81c53c9 Iustin Pop
    order as the passed nodes.
864 a81c53c9 Iustin Pop

865 32388e6d Iustin Pop
    @type instance: string
866 32388e6d Iustin Pop
    @param instance: the instance for which we allocate minors
867 32388e6d Iustin Pop

868 a81c53c9 Iustin Pop
    """
869 32388e6d Iustin Pop
    assert isinstance(instance, basestring), \
870 4a89c54a Iustin Pop
           "Invalid argument '%s' passed to AllocateDRBDMinor" % instance
871 32388e6d Iustin Pop
872 4a89c54a Iustin Pop
    d_map, duplicates = self._UnlockedComputeDRBDMap()
873 4a89c54a Iustin Pop
    if duplicates:
874 4a89c54a Iustin Pop
      raise errors.ConfigurationError("Duplicate DRBD ports detected: %s" %
875 4a89c54a Iustin Pop
                                      str(duplicates))
876 a81c53c9 Iustin Pop
    result = []
877 a81c53c9 Iustin Pop
    for nname in nodes:
878 a81c53c9 Iustin Pop
      ndata = d_map[nname]
879 a81c53c9 Iustin Pop
      if not ndata:
880 a81c53c9 Iustin Pop
        # no minors used, we can start at 0
881 a81c53c9 Iustin Pop
        result.append(0)
882 a81c53c9 Iustin Pop
        ndata[0] = instance
883 d48663e4 Iustin Pop
        self._temporary_drbds[(nname, 0)] = instance
884 a81c53c9 Iustin Pop
        continue
885 a81c53c9 Iustin Pop
      keys = ndata.keys()
886 a81c53c9 Iustin Pop
      keys.sort()
887 a81c53c9 Iustin Pop
      ffree = utils.FirstFree(keys)
888 a81c53c9 Iustin Pop
      if ffree is None:
889 a81c53c9 Iustin Pop
        # return the next minor
890 a81c53c9 Iustin Pop
        # TODO: implement high-limit check
891 a81c53c9 Iustin Pop
        minor = keys[-1] + 1
892 a81c53c9 Iustin Pop
      else:
893 a81c53c9 Iustin Pop
        minor = ffree
894 4a89c54a Iustin Pop
      # double-check minor against current instances
895 4a89c54a Iustin Pop
      assert minor not in d_map[nname], \
896 4a89c54a Iustin Pop
             ("Attempt to reuse allocated DRBD minor %d on node %s,"
897 4a89c54a Iustin Pop
              " already allocated to instance %s" %
898 4a89c54a Iustin Pop
              (minor, nname, d_map[nname][minor]))
899 a81c53c9 Iustin Pop
      ndata[minor] = instance
900 4a89c54a Iustin Pop
      # double-check minor against reservation
901 4a89c54a Iustin Pop
      r_key = (nname, minor)
902 4a89c54a Iustin Pop
      assert r_key not in self._temporary_drbds, \
903 4a89c54a Iustin Pop
             ("Attempt to reuse reserved DRBD minor %d on node %s,"
904 4a89c54a Iustin Pop
              " reserved for instance %s" %
905 4a89c54a Iustin Pop
              (minor, nname, self._temporary_drbds[r_key]))
906 4a89c54a Iustin Pop
      self._temporary_drbds[r_key] = instance
907 4a89c54a Iustin Pop
      result.append(minor)
908 a81c53c9 Iustin Pop
    logging.debug("Request to allocate drbd minors, input: %s, returning %s",
909 a81c53c9 Iustin Pop
                  nodes, result)
910 a81c53c9 Iustin Pop
    return result
911 a81c53c9 Iustin Pop
912 61cf6b5e Iustin Pop
  def _UnlockedReleaseDRBDMinors(self, instance):
913 a81c53c9 Iustin Pop
    """Release temporary drbd minors allocated for a given instance.
914 a81c53c9 Iustin Pop

915 a81c53c9 Iustin Pop
    @type instance: string
916 a81c53c9 Iustin Pop
    @param instance: the instance for which temporary minors should be
917 a81c53c9 Iustin Pop
                     released
918 a81c53c9 Iustin Pop

919 a81c53c9 Iustin Pop
    """
920 32388e6d Iustin Pop
    assert isinstance(instance, basestring), \
921 32388e6d Iustin Pop
           "Invalid argument passed to ReleaseDRBDMinors"
922 a81c53c9 Iustin Pop
    for key, name in self._temporary_drbds.items():
923 a81c53c9 Iustin Pop
      if name == instance:
924 a81c53c9 Iustin Pop
        del self._temporary_drbds[key]
925 a81c53c9 Iustin Pop
926 61cf6b5e Iustin Pop
  @locking.ssynchronized(_config_lock)
927 61cf6b5e Iustin Pop
  def ReleaseDRBDMinors(self, instance):
928 61cf6b5e Iustin Pop
    """Release temporary drbd minors allocated for a given instance.
929 61cf6b5e Iustin Pop

930 61cf6b5e Iustin Pop
    This should be called on the error paths, on the success paths
931 61cf6b5e Iustin Pop
    it's automatically called by the ConfigWriter add and update
932 61cf6b5e Iustin Pop
    functions.
933 61cf6b5e Iustin Pop

934 61cf6b5e Iustin Pop
    This function is just a wrapper over L{_UnlockedReleaseDRBDMinors}.
935 61cf6b5e Iustin Pop

936 61cf6b5e Iustin Pop
    @type instance: string
937 61cf6b5e Iustin Pop
    @param instance: the instance for which temporary minors should be
938 61cf6b5e Iustin Pop
                     released
939 61cf6b5e Iustin Pop

940 61cf6b5e Iustin Pop
    """
941 61cf6b5e Iustin Pop
    self._UnlockedReleaseDRBDMinors(instance)
942 61cf6b5e Iustin Pop
943 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
944 4a8b186a Michael Hanselmann
  def GetConfigVersion(self):
945 4a8b186a Michael Hanselmann
    """Get the configuration version.
946 4a8b186a Michael Hanselmann

947 4a8b186a Michael Hanselmann
    @return: Config version
948 4a8b186a Michael Hanselmann

949 4a8b186a Michael Hanselmann
    """
950 4a8b186a Michael Hanselmann
    return self._config_data.version
951 4a8b186a Michael Hanselmann
952 4a8b186a Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
953 4a8b186a Michael Hanselmann
  def GetClusterName(self):
954 4a8b186a Michael Hanselmann
    """Get cluster name.
955 4a8b186a Michael Hanselmann

956 4a8b186a Michael Hanselmann
    @return: Cluster name
957 4a8b186a Michael Hanselmann

958 4a8b186a Michael Hanselmann
    """
959 4a8b186a Michael Hanselmann
    return self._config_data.cluster.cluster_name
960 4a8b186a Michael Hanselmann
961 4a8b186a Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
962 4a8b186a Michael Hanselmann
  def GetMasterNode(self):
963 4a8b186a Michael Hanselmann
    """Get the hostname of the master node for this cluster.
964 4a8b186a Michael Hanselmann

965 4a8b186a Michael Hanselmann
    @return: Master hostname
966 4a8b186a Michael Hanselmann

967 4a8b186a Michael Hanselmann
    """
968 4a8b186a Michael Hanselmann
    return self._config_data.cluster.master_node
969 4a8b186a Michael Hanselmann
970 4a8b186a Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
971 4a8b186a Michael Hanselmann
  def GetMasterIP(self):
972 4a8b186a Michael Hanselmann
    """Get the IP of the master node for this cluster.
973 4a8b186a Michael Hanselmann

974 4a8b186a Michael Hanselmann
    @return: Master IP
975 4a8b186a Michael Hanselmann

976 4a8b186a Michael Hanselmann
    """
977 4a8b186a Michael Hanselmann
    return self._config_data.cluster.master_ip
978 4a8b186a Michael Hanselmann
979 4a8b186a Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
980 4a8b186a Michael Hanselmann
  def GetMasterNetdev(self):
981 4a8b186a Michael Hanselmann
    """Get the master network device for this cluster.
982 4a8b186a Michael Hanselmann

983 4a8b186a Michael Hanselmann
    """
984 4a8b186a Michael Hanselmann
    return self._config_data.cluster.master_netdev
985 4a8b186a Michael Hanselmann
986 4a8b186a Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
987 5a8648eb Andrea Spadaccini
  def GetMasterNetmask(self):
988 5a8648eb Andrea Spadaccini
    """Get the netmask of the master node for this cluster.
989 5a8648eb Andrea Spadaccini

990 5a8648eb Andrea Spadaccini
    """
991 5a8648eb Andrea Spadaccini
    return self._config_data.cluster.master_netmask
992 5a8648eb Andrea Spadaccini
993 5a8648eb Andrea Spadaccini
  @locking.ssynchronized(_config_lock, shared=1)
994 33be7576 Andrea Spadaccini
  def GetUseExternalMipScript(self):
995 33be7576 Andrea Spadaccini
    """Get flag representing whether to use the external master IP setup script.
996 33be7576 Andrea Spadaccini

997 33be7576 Andrea Spadaccini
    """
998 33be7576 Andrea Spadaccini
    return self._config_data.cluster.use_external_mip_script
999 33be7576 Andrea Spadaccini
1000 33be7576 Andrea Spadaccini
  @locking.ssynchronized(_config_lock, shared=1)
1001 4a8b186a Michael Hanselmann
  def GetFileStorageDir(self):
1002 4a8b186a Michael Hanselmann
    """Get the file storage dir for this cluster.
1003 4a8b186a Michael Hanselmann

1004 4a8b186a Michael Hanselmann
    """
1005 4a8b186a Michael Hanselmann
    return self._config_data.cluster.file_storage_dir
1006 4a8b186a Michael Hanselmann
1007 4a8b186a Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
1008 4b97f902 Apollon Oikonomopoulos
  def GetSharedFileStorageDir(self):
1009 4b97f902 Apollon Oikonomopoulos
    """Get the shared file storage dir for this cluster.
1010 4b97f902 Apollon Oikonomopoulos

1011 4b97f902 Apollon Oikonomopoulos
    """
1012 4b97f902 Apollon Oikonomopoulos
    return self._config_data.cluster.shared_file_storage_dir
1013 4b97f902 Apollon Oikonomopoulos
1014 4b97f902 Apollon Oikonomopoulos
  @locking.ssynchronized(_config_lock, shared=1)
1015 4a8b186a Michael Hanselmann
  def GetHypervisorType(self):
1016 4a8b186a Michael Hanselmann
    """Get the hypervisor type for this cluster.
1017 4a8b186a Michael Hanselmann

1018 4a8b186a Michael Hanselmann
    """
1019 066f465d Guido Trotter
    return self._config_data.cluster.enabled_hypervisors[0]
1020 4a8b186a Michael Hanselmann
1021 4a8b186a Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
1022 a8083063 Iustin Pop
  def GetHostKey(self):
1023 a8083063 Iustin Pop
    """Return the rsa hostkey from the config.
1024 a8083063 Iustin Pop

1025 c41eea6e Iustin Pop
    @rtype: string
1026 c41eea6e Iustin Pop
    @return: the rsa hostkey
1027 a8083063 Iustin Pop

1028 a8083063 Iustin Pop
    """
1029 a8083063 Iustin Pop
    return self._config_data.cluster.rsahostkeypub
1030 a8083063 Iustin Pop
1031 bf4af505 Apollon Oikonomopoulos
  @locking.ssynchronized(_config_lock, shared=1)
1032 bf4af505 Apollon Oikonomopoulos
  def GetDefaultIAllocator(self):
1033 bf4af505 Apollon Oikonomopoulos
    """Get the default instance allocator for this cluster.
1034 bf4af505 Apollon Oikonomopoulos

1035 bf4af505 Apollon Oikonomopoulos
    """
1036 bf4af505 Apollon Oikonomopoulos
    return self._config_data.cluster.default_iallocator
1037 bf4af505 Apollon Oikonomopoulos
1038 868a98ca Manuel Franceschini
  @locking.ssynchronized(_config_lock, shared=1)
1039 868a98ca Manuel Franceschini
  def GetPrimaryIPFamily(self):
1040 868a98ca Manuel Franceschini
    """Get cluster primary ip family.
1041 868a98ca Manuel Franceschini

1042 868a98ca Manuel Franceschini
    @return: primary ip family
1043 868a98ca Manuel Franceschini

1044 868a98ca Manuel Franceschini
    """
1045 868a98ca Manuel Franceschini
    return self._config_data.cluster.primary_ip_family
1046 868a98ca Manuel Franceschini
1047 c9f4b8e6 Andrea Spadaccini
  @locking.ssynchronized(_config_lock, shared=1)
1048 c9f4b8e6 Andrea Spadaccini
  def GetMasterNetworkParameters(self):
1049 c9f4b8e6 Andrea Spadaccini
    """Get network parameters of the master node.
1050 c9f4b8e6 Andrea Spadaccini

1051 f9d20654 Andrea Spadaccini
    @rtype: L{object.MasterNetworkParameters}
1052 f9d20654 Andrea Spadaccini
    @return: network parameters of the master node
1053 c9f4b8e6 Andrea Spadaccini

1054 c9f4b8e6 Andrea Spadaccini
    """
1055 c9f4b8e6 Andrea Spadaccini
    cluster = self._config_data.cluster
1056 c79198a0 Andrea Spadaccini
    result = objects.MasterNetworkParameters(name=cluster.master_node,
1057 c79198a0 Andrea Spadaccini
      ip=cluster.master_ip,
1058 c79198a0 Andrea Spadaccini
      netmask=cluster.master_netmask,
1059 c79198a0 Andrea Spadaccini
      netdev=cluster.master_netdev,
1060 c79198a0 Andrea Spadaccini
      ip_family=cluster.primary_ip_family)
1061 c9f4b8e6 Andrea Spadaccini
1062 f9d20654 Andrea Spadaccini
    return result
1063 f9d20654 Andrea Spadaccini
1064 e11a1b77 Adeodato Simo
  @locking.ssynchronized(_config_lock)
1065 e11a1b77 Adeodato Simo
  def AddNodeGroup(self, group, ec_id, check_uuid=True):
1066 e11a1b77 Adeodato Simo
    """Add a node group to the configuration.
1067 e11a1b77 Adeodato Simo

1068 90e99856 Adeodato Simo
    This method calls group.UpgradeConfig() to fill any missing attributes
1069 90e99856 Adeodato Simo
    according to their default values.
1070 90e99856 Adeodato Simo

1071 e11a1b77 Adeodato Simo
    @type group: L{objects.NodeGroup}
1072 e11a1b77 Adeodato Simo
    @param group: the NodeGroup object to add
1073 e11a1b77 Adeodato Simo
    @type ec_id: string
1074 e11a1b77 Adeodato Simo
    @param ec_id: unique id for the job to use when creating a missing UUID
1075 e11a1b77 Adeodato Simo
    @type check_uuid: bool
1076 e11a1b77 Adeodato Simo
    @param check_uuid: add an UUID to the group if it doesn't have one or, if
1077 e11a1b77 Adeodato Simo
                       it does, ensure that it does not exist in the
1078 e11a1b77 Adeodato Simo
                       configuration already
1079 e11a1b77 Adeodato Simo

1080 e11a1b77 Adeodato Simo
    """
1081 e11a1b77 Adeodato Simo
    self._UnlockedAddNodeGroup(group, ec_id, check_uuid)
1082 e11a1b77 Adeodato Simo
    self._WriteConfig()
1083 e11a1b77 Adeodato Simo
1084 e11a1b77 Adeodato Simo
  def _UnlockedAddNodeGroup(self, group, ec_id, check_uuid):
1085 e11a1b77 Adeodato Simo
    """Add a node group to the configuration.
1086 e11a1b77 Adeodato Simo

1087 e11a1b77 Adeodato Simo
    """
1088 e11a1b77 Adeodato Simo
    logging.info("Adding node group %s to configuration", group.name)
1089 e11a1b77 Adeodato Simo
1090 e11a1b77 Adeodato Simo
    # Some code might need to add a node group with a pre-populated UUID
1091 e11a1b77 Adeodato Simo
    # generated with ConfigWriter.GenerateUniqueID(). We allow them to bypass
1092 e11a1b77 Adeodato Simo
    # the "does this UUID" exist already check.
1093 e11a1b77 Adeodato Simo
    if check_uuid:
1094 e11a1b77 Adeodato Simo
      self._EnsureUUID(group, ec_id)
1095 e11a1b77 Adeodato Simo
1096 18ffc0fe Stephen Shirley
    try:
1097 18ffc0fe Stephen Shirley
      existing_uuid = self._UnlockedLookupNodeGroup(group.name)
1098 18ffc0fe Stephen Shirley
    except errors.OpPrereqError:
1099 18ffc0fe Stephen Shirley
      pass
1100 18ffc0fe Stephen Shirley
    else:
1101 18ffc0fe Stephen Shirley
      raise errors.OpPrereqError("Desired group name '%s' already exists as a"
1102 18ffc0fe Stephen Shirley
                                 " node group (UUID: %s)" %
1103 18ffc0fe Stephen Shirley
                                 (group.name, existing_uuid),
1104 18ffc0fe Stephen Shirley
                                 errors.ECODE_EXISTS)
1105 18ffc0fe Stephen Shirley
1106 e11a1b77 Adeodato Simo
    group.serial_no = 1
1107 e11a1b77 Adeodato Simo
    group.ctime = group.mtime = time.time()
1108 90e99856 Adeodato Simo
    group.UpgradeConfig()
1109 e11a1b77 Adeodato Simo
1110 e11a1b77 Adeodato Simo
    self._config_data.nodegroups[group.uuid] = group
1111 e11a1b77 Adeodato Simo
    self._config_data.cluster.serial_no += 1
1112 e11a1b77 Adeodato Simo
1113 e11a1b77 Adeodato Simo
  @locking.ssynchronized(_config_lock)
1114 e11a1b77 Adeodato Simo
  def RemoveNodeGroup(self, group_uuid):
1115 e11a1b77 Adeodato Simo
    """Remove a node group from the configuration.
1116 e11a1b77 Adeodato Simo

1117 e11a1b77 Adeodato Simo
    @type group_uuid: string
1118 e11a1b77 Adeodato Simo
    @param group_uuid: the UUID of the node group to remove
1119 e11a1b77 Adeodato Simo

1120 e11a1b77 Adeodato Simo
    """
1121 e11a1b77 Adeodato Simo
    logging.info("Removing node group %s from configuration", group_uuid)
1122 e11a1b77 Adeodato Simo
1123 e11a1b77 Adeodato Simo
    if group_uuid not in self._config_data.nodegroups:
1124 e11a1b77 Adeodato Simo
      raise errors.ConfigurationError("Unknown node group '%s'" % group_uuid)
1125 e11a1b77 Adeodato Simo
1126 0389c42a Stephen Shirley
    assert len(self._config_data.nodegroups) != 1, \
1127 0389c42a Stephen Shirley
            "Group '%s' is the only group, cannot be removed" % group_uuid
1128 0389c42a Stephen Shirley
1129 e11a1b77 Adeodato Simo
    del self._config_data.nodegroups[group_uuid]
1130 e11a1b77 Adeodato Simo
    self._config_data.cluster.serial_no += 1
1131 e11a1b77 Adeodato Simo
    self._WriteConfig()
1132 e11a1b77 Adeodato Simo
1133 e85d8982 Stephen Shirley
  def _UnlockedLookupNodeGroup(self, target):
1134 412b3531 Guido Trotter
    """Lookup a node group's UUID.
1135 eaa98a04 Guido Trotter

1136 eaa98a04 Guido Trotter
    @type target: string or None
1137 412b3531 Guido Trotter
    @param target: group name or UUID or None to look for the default
1138 eaa98a04 Guido Trotter
    @rtype: string
1139 412b3531 Guido Trotter
    @return: nodegroup UUID
1140 eaa98a04 Guido Trotter
    @raises errors.OpPrereqError: when the target group cannot be found
1141 eaa98a04 Guido Trotter

1142 eaa98a04 Guido Trotter
    """
1143 eaa98a04 Guido Trotter
    if target is None:
1144 eaa98a04 Guido Trotter
      if len(self._config_data.nodegroups) != 1:
1145 913cc25e Adeodato Simo
        raise errors.OpPrereqError("More than one node group exists. Target"
1146 eaa98a04 Guido Trotter
                                   " group must be specified explicitely.")
1147 eaa98a04 Guido Trotter
      else:
1148 eaa98a04 Guido Trotter
        return self._config_data.nodegroups.keys()[0]
1149 eaa98a04 Guido Trotter
    if target in self._config_data.nodegroups:
1150 eaa98a04 Guido Trotter
      return target
1151 eaa98a04 Guido Trotter
    for nodegroup in self._config_data.nodegroups.values():
1152 eaa98a04 Guido Trotter
      if nodegroup.name == target:
1153 eaa98a04 Guido Trotter
        return nodegroup.uuid
1154 e0f9ed64 Adeodato Simo
    raise errors.OpPrereqError("Node group '%s' not found" % target,
1155 e0f9ed64 Adeodato Simo
                               errors.ECODE_NOENT)
1156 eaa98a04 Guido Trotter
1157 e85d8982 Stephen Shirley
  @locking.ssynchronized(_config_lock, shared=1)
1158 e85d8982 Stephen Shirley
  def LookupNodeGroup(self, target):
1159 e85d8982 Stephen Shirley
    """Lookup a node group's UUID.
1160 e85d8982 Stephen Shirley

1161 e85d8982 Stephen Shirley
    This function is just a wrapper over L{_UnlockedLookupNodeGroup}.
1162 e85d8982 Stephen Shirley

1163 e85d8982 Stephen Shirley
    @type target: string or None
1164 e85d8982 Stephen Shirley
    @param target: group name or UUID or None to look for the default
1165 e85d8982 Stephen Shirley
    @rtype: string
1166 e85d8982 Stephen Shirley
    @return: nodegroup UUID
1167 e85d8982 Stephen Shirley

1168 e85d8982 Stephen Shirley
    """
1169 e85d8982 Stephen Shirley
    return self._UnlockedLookupNodeGroup(target)
1170 e85d8982 Stephen Shirley
1171 5768e6a6 René Nussbaumer
  def _UnlockedGetNodeGroup(self, uuid):
1172 648e4196 Guido Trotter
    """Lookup a node group.
1173 648e4196 Guido Trotter

1174 648e4196 Guido Trotter
    @type uuid: string
1175 648e4196 Guido Trotter
    @param uuid: group UUID
1176 648e4196 Guido Trotter
    @rtype: L{objects.NodeGroup} or None
1177 648e4196 Guido Trotter
    @return: nodegroup object, or None if not found
1178 648e4196 Guido Trotter

1179 648e4196 Guido Trotter
    """
1180 648e4196 Guido Trotter
    if uuid not in self._config_data.nodegroups:
1181 648e4196 Guido Trotter
      return None
1182 648e4196 Guido Trotter
1183 648e4196 Guido Trotter
    return self._config_data.nodegroups[uuid]
1184 648e4196 Guido Trotter
1185 648e4196 Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
1186 5768e6a6 René Nussbaumer
  def GetNodeGroup(self, uuid):
1187 5768e6a6 René Nussbaumer
    """Lookup a node group.
1188 5768e6a6 René Nussbaumer

1189 5768e6a6 René Nussbaumer
    @type uuid: string
1190 5768e6a6 René Nussbaumer
    @param uuid: group UUID
1191 5768e6a6 René Nussbaumer
    @rtype: L{objects.NodeGroup} or None
1192 5768e6a6 René Nussbaumer
    @return: nodegroup object, or None if not found
1193 5768e6a6 René Nussbaumer

1194 5768e6a6 René Nussbaumer
    """
1195 5768e6a6 René Nussbaumer
    return self._UnlockedGetNodeGroup(uuid)
1196 5768e6a6 René Nussbaumer
1197 5768e6a6 René Nussbaumer
  @locking.ssynchronized(_config_lock, shared=1)
1198 622444e5 Iustin Pop
  def GetAllNodeGroupsInfo(self):
1199 622444e5 Iustin Pop
    """Get the configuration of all node groups.
1200 622444e5 Iustin Pop

1201 622444e5 Iustin Pop
    """
1202 622444e5 Iustin Pop
    return dict(self._config_data.nodegroups)
1203 622444e5 Iustin Pop
1204 1ac6f2ad Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
1205 1ac6f2ad Guido Trotter
  def GetNodeGroupList(self):
1206 1ac6f2ad Guido Trotter
    """Get a list of node groups.
1207 1ac6f2ad Guido Trotter

1208 1ac6f2ad Guido Trotter
    """
1209 1ac6f2ad Guido Trotter
    return self._config_data.nodegroups.keys()
1210 1ac6f2ad Guido Trotter
1211 dac81741 Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
1212 dac81741 Michael Hanselmann
  def GetNodeGroupMembersByNodes(self, nodes):
1213 dac81741 Michael Hanselmann
    """Get nodes which are member in the same nodegroups as the given nodes.
1214 dac81741 Michael Hanselmann

1215 dac81741 Michael Hanselmann
    """
1216 dac81741 Michael Hanselmann
    ngfn = lambda node_name: self._UnlockedGetNodeInfo(node_name).group
1217 dac81741 Michael Hanselmann
    return frozenset(member_name
1218 dac81741 Michael Hanselmann
                     for node_name in nodes
1219 dac81741 Michael Hanselmann
                     for member_name in
1220 dac81741 Michael Hanselmann
                       self._UnlockedGetNodeGroup(ngfn(node_name)).members)
1221 dac81741 Michael Hanselmann
1222 080fbeea Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
1223 080fbeea Michael Hanselmann
  def GetMultiNodeGroupInfo(self, group_uuids):
1224 080fbeea Michael Hanselmann
    """Get the configuration of multiple node groups.
1225 080fbeea Michael Hanselmann

1226 080fbeea Michael Hanselmann
    @param group_uuids: List of node group UUIDs
1227 080fbeea Michael Hanselmann
    @rtype: list
1228 080fbeea Michael Hanselmann
    @return: List of tuples of (group_uuid, group_info)
1229 080fbeea Michael Hanselmann

1230 080fbeea Michael Hanselmann
    """
1231 080fbeea Michael Hanselmann
    return [(uuid, self._UnlockedGetNodeGroup(uuid)) for uuid in group_uuids]
1232 080fbeea Michael Hanselmann
1233 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock)
1234 0debfb35 Guido Trotter
  def AddInstance(self, instance, ec_id):
1235 a8083063 Iustin Pop
    """Add an instance to the config.
1236 a8083063 Iustin Pop

1237 a8083063 Iustin Pop
    This should be used after creating a new instance.
1238 a8083063 Iustin Pop

1239 c41eea6e Iustin Pop
    @type instance: L{objects.Instance}
1240 c41eea6e Iustin Pop
    @param instance: the instance object
1241 c41eea6e Iustin Pop

1242 a8083063 Iustin Pop
    """
1243 a8083063 Iustin Pop
    if not isinstance(instance, objects.Instance):
1244 a8083063 Iustin Pop
      raise errors.ProgrammerError("Invalid type passed to AddInstance")
1245 a8083063 Iustin Pop
1246 e00fb268 Iustin Pop
    if instance.disk_template != constants.DT_DISKLESS:
1247 e00fb268 Iustin Pop
      all_lvs = instance.MapLVsByNode()
1248 74a48621 Iustin Pop
      logging.info("Instance '%s' DISK_LAYOUT: %s", instance.name, all_lvs)
1249 923b1523 Iustin Pop
1250 e4640214 Guido Trotter
    all_macs = self._AllMACs()
1251 e4640214 Guido Trotter
    for nic in instance.nics:
1252 e4640214 Guido Trotter
      if nic.mac in all_macs:
1253 e4640214 Guido Trotter
        raise errors.ConfigurationError("Cannot add instance %s:"
1254 430b923c Iustin Pop
                                        " MAC address '%s' already in use." %
1255 430b923c Iustin Pop
                                        (instance.name, nic.mac))
1256 430b923c Iustin Pop
1257 0debfb35 Guido Trotter
    self._EnsureUUID(instance, ec_id)
1258 e4640214 Guido Trotter
1259 b989e85d Iustin Pop
    instance.serial_no = 1
1260 d693c864 Iustin Pop
    instance.ctime = instance.mtime = time.time()
1261 a8083063 Iustin Pop
    self._config_data.instances[instance.name] = instance
1262 81a49123 Iustin Pop
    self._config_data.cluster.serial_no += 1
1263 61cf6b5e Iustin Pop
    self._UnlockedReleaseDRBDMinors(instance.name)
1264 a8083063 Iustin Pop
    self._WriteConfig()
1265 a8083063 Iustin Pop
1266 0debfb35 Guido Trotter
  def _EnsureUUID(self, item, ec_id):
1267 430b923c Iustin Pop
    """Ensures a given object has a valid UUID.
1268 430b923c Iustin Pop

1269 430b923c Iustin Pop
    @param item: the instance or node to be checked
1270 0debfb35 Guido Trotter
    @param ec_id: the execution context id for the uuid reservation
1271 430b923c Iustin Pop

1272 430b923c Iustin Pop
    """
1273 430b923c Iustin Pop
    if not item.uuid:
1274 4fae38c5 Guido Trotter
      item.uuid = self._GenerateUniqueID(ec_id)
1275 be0fc05d Iustin Pop
    elif item.uuid in self._AllIDs(include_temporary=True):
1276 be0fc05d Iustin Pop
      raise errors.ConfigurationError("Cannot add '%s': UUID %s already"
1277 be0fc05d Iustin Pop
                                      " in use" % (item.name, item.uuid))
1278 430b923c Iustin Pop
1279 6a408fb2 Iustin Pop
  def _SetInstanceStatus(self, instance_name, status):
1280 6a408fb2 Iustin Pop
    """Set the instance's status to a given value.
1281 a8083063 Iustin Pop

1282 a8083063 Iustin Pop
    """
1283 2e04d454 Agata Murawska
    assert status in constants.ADMINST_ALL, \
1284 0d68c45d Iustin Pop
           "Invalid status '%s' passed to SetInstanceStatus" % (status,)
1285 a8083063 Iustin Pop
1286 a8083063 Iustin Pop
    if instance_name not in self._config_data.instances:
1287 3ecf6786 Iustin Pop
      raise errors.ConfigurationError("Unknown instance '%s'" %
1288 3ecf6786 Iustin Pop
                                      instance_name)
1289 a8083063 Iustin Pop
    instance = self._config_data.instances[instance_name]
1290 9ca8a7c5 Agata Murawska
    if instance.admin_state != status:
1291 9ca8a7c5 Agata Murawska
      instance.admin_state = status
1292 b989e85d Iustin Pop
      instance.serial_no += 1
1293 d693c864 Iustin Pop
      instance.mtime = time.time()
1294 455a3445 Iustin Pop
      self._WriteConfig()
1295 a8083063 Iustin Pop
1296 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock)
1297 6a408fb2 Iustin Pop
  def MarkInstanceUp(self, instance_name):
1298 6a408fb2 Iustin Pop
    """Mark the instance status to up in the config.
1299 6a408fb2 Iustin Pop

1300 6a408fb2 Iustin Pop
    """
1301 57de31c0 Agata Murawska
    self._SetInstanceStatus(instance_name, constants.ADMINST_UP)
1302 57de31c0 Agata Murawska
1303 57de31c0 Agata Murawska
  @locking.ssynchronized(_config_lock)
1304 57de31c0 Agata Murawska
  def MarkInstanceOffline(self, instance_name):
1305 57de31c0 Agata Murawska
    """Mark the instance status to down in the config.
1306 57de31c0 Agata Murawska

1307 57de31c0 Agata Murawska
    """
1308 57de31c0 Agata Murawska
    self._SetInstanceStatus(instance_name, constants.ADMINST_OFFLINE)
1309 6a408fb2 Iustin Pop
1310 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock)
1311 a8083063 Iustin Pop
  def RemoveInstance(self, instance_name):
1312 a8083063 Iustin Pop
    """Remove the instance from the configuration.
1313 a8083063 Iustin Pop

1314 a8083063 Iustin Pop
    """
1315 a8083063 Iustin Pop
    if instance_name not in self._config_data.instances:
1316 3ecf6786 Iustin Pop
      raise errors.ConfigurationError("Unknown instance '%s'" % instance_name)
1317 f396ad8c Vangelis Koukis
1318 f396ad8c Vangelis Koukis
    # If a network port has been allocated to the instance,
1319 f396ad8c Vangelis Koukis
    # return it to the pool of free ports.
1320 f396ad8c Vangelis Koukis
    inst = self._config_data.instances[instance_name]
1321 f396ad8c Vangelis Koukis
    network_port = getattr(inst, "network_port", None)
1322 f396ad8c Vangelis Koukis
    if network_port is not None:
1323 f396ad8c Vangelis Koukis
      self._config_data.cluster.tcpudp_port_pool.add(network_port)
1324 f396ad8c Vangelis Koukis
1325 a8083063 Iustin Pop
    del self._config_data.instances[instance_name]
1326 81a49123 Iustin Pop
    self._config_data.cluster.serial_no += 1
1327 a8083063 Iustin Pop
    self._WriteConfig()
1328 a8083063 Iustin Pop
1329 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock)
1330 fc95f88f Iustin Pop
  def RenameInstance(self, old_name, new_name):
1331 fc95f88f Iustin Pop
    """Rename an instance.
1332 fc95f88f Iustin Pop

1333 fc95f88f Iustin Pop
    This needs to be done in ConfigWriter and not by RemoveInstance
1334 fc95f88f Iustin Pop
    combined with AddInstance as only we can guarantee an atomic
1335 fc95f88f Iustin Pop
    rename.
1336 fc95f88f Iustin Pop

1337 fc95f88f Iustin Pop
    """
1338 fc95f88f Iustin Pop
    if old_name not in self._config_data.instances:
1339 fc95f88f Iustin Pop
      raise errors.ConfigurationError("Unknown instance '%s'" % old_name)
1340 ea642319 Michael Hanselmann
1341 ea642319 Michael Hanselmann
    # Operate on a copy to not loose instance object in case of a failure
1342 ea642319 Michael Hanselmann
    inst = self._config_data.instances[old_name].Copy()
1343 fc95f88f Iustin Pop
    inst.name = new_name
1344 b23c4333 Manuel Franceschini
1345 ea642319 Michael Hanselmann
    for (idx, disk) in enumerate(inst.disks):
1346 b23c4333 Manuel Franceschini
      if disk.dev_type == constants.LD_FILE:
1347 b23c4333 Manuel Franceschini
        # rename the file paths in logical and physical id
1348 b23c4333 Manuel Franceschini
        file_storage_dir = os.path.dirname(os.path.dirname(disk.logical_id[1]))
1349 ea642319 Michael Hanselmann
        disk.logical_id = (disk.logical_id[0],
1350 ea642319 Michael Hanselmann
                           utils.PathJoin(file_storage_dir, inst.name,
1351 ea642319 Michael Hanselmann
                                          "disk%s" % idx))
1352 ea642319 Michael Hanselmann
        disk.physical_id = disk.logical_id
1353 ea642319 Michael Hanselmann
1354 ea642319 Michael Hanselmann
    # Actually replace instance object
1355 ea642319 Michael Hanselmann
    del self._config_data.instances[old_name]
1356 ea642319 Michael Hanselmann
    self._config_data.instances[inst.name] = inst
1357 b23c4333 Manuel Franceschini
1358 1fc34c48 Michael Hanselmann
    # Force update of ssconf files
1359 1fc34c48 Michael Hanselmann
    self._config_data.cluster.serial_no += 1
1360 1fc34c48 Michael Hanselmann
1361 fc95f88f Iustin Pop
    self._WriteConfig()
1362 fc95f88f Iustin Pop
1363 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock)
1364 a8083063 Iustin Pop
  def MarkInstanceDown(self, instance_name):
1365 a8083063 Iustin Pop
    """Mark the status of an instance to down in the configuration.
1366 a8083063 Iustin Pop

1367 a8083063 Iustin Pop
    """
1368 2e04d454 Agata Murawska
    self._SetInstanceStatus(instance_name, constants.ADMINST_DOWN)
1369 a8083063 Iustin Pop
1370 94bbfece Iustin Pop
  def _UnlockedGetInstanceList(self):
1371 94bbfece Iustin Pop
    """Get the list of instances.
1372 94bbfece Iustin Pop

1373 94bbfece Iustin Pop
    This function is for internal use, when the config lock is already held.
1374 94bbfece Iustin Pop

1375 94bbfece Iustin Pop
    """
1376 94bbfece Iustin Pop
    return self._config_data.instances.keys()
1377 94bbfece Iustin Pop
1378 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
1379 a8083063 Iustin Pop
  def GetInstanceList(self):
1380 a8083063 Iustin Pop
    """Get the list of instances.
1381 a8083063 Iustin Pop

1382 c41eea6e Iustin Pop
    @return: array of instances, ex. ['instance2.example.com',
1383 c41eea6e Iustin Pop
        'instance1.example.com']
1384 a8083063 Iustin Pop

1385 a8083063 Iustin Pop
    """
1386 94bbfece Iustin Pop
    return self._UnlockedGetInstanceList()
1387 a8083063 Iustin Pop
1388 a8083063 Iustin Pop
  def ExpandInstanceName(self, short_name):
1389 a8083063 Iustin Pop
    """Attempt to expand an incomplete instance name.
1390 a8083063 Iustin Pop

1391 a8083063 Iustin Pop
    """
1392 fe698b38 Michael Hanselmann
    # Locking is done in L{ConfigWriter.GetInstanceList}
1393 fe698b38 Michael Hanselmann
    return _MatchNameComponentIgnoreCase(short_name, self.GetInstanceList())
1394 a8083063 Iustin Pop
1395 94bbfece Iustin Pop
  def _UnlockedGetInstanceInfo(self, instance_name):
1396 5bbd3f7f Michael Hanselmann
    """Returns information about an instance.
1397 94bbfece Iustin Pop

1398 94bbfece Iustin Pop
    This function is for internal use, when the config lock is already held.
1399 94bbfece Iustin Pop

1400 94bbfece Iustin Pop
    """
1401 94bbfece Iustin Pop
    if instance_name not in self._config_data.instances:
1402 94bbfece Iustin Pop
      return None
1403 94bbfece Iustin Pop
1404 94bbfece Iustin Pop
    return self._config_data.instances[instance_name]
1405 94bbfece Iustin Pop
1406 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
1407 a8083063 Iustin Pop
  def GetInstanceInfo(self, instance_name):
1408 5bbd3f7f Michael Hanselmann
    """Returns information about an instance.
1409 a8083063 Iustin Pop

1410 5bbd3f7f Michael Hanselmann
    It takes the information from the configuration file. Other information of
1411 a8083063 Iustin Pop
    an instance are taken from the live systems.
1412 a8083063 Iustin Pop

1413 c41eea6e Iustin Pop
    @param instance_name: name of the instance, e.g.
1414 c41eea6e Iustin Pop
        I{instance1.example.com}
1415 a8083063 Iustin Pop

1416 c41eea6e Iustin Pop
    @rtype: L{objects.Instance}
1417 c41eea6e Iustin Pop
    @return: the instance object
1418 a8083063 Iustin Pop

1419 a8083063 Iustin Pop
    """
1420 94bbfece Iustin Pop
    return self._UnlockedGetInstanceInfo(instance_name)
1421 a8083063 Iustin Pop
1422 0b2de758 Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
1423 2674690b Michael Hanselmann
  def GetInstanceNodeGroups(self, instance_name, primary_only=False):
1424 2674690b Michael Hanselmann
    """Returns set of node group UUIDs for instance's nodes.
1425 2674690b Michael Hanselmann

1426 2674690b Michael Hanselmann
    @rtype: frozenset
1427 2674690b Michael Hanselmann

1428 2674690b Michael Hanselmann
    """
1429 2674690b Michael Hanselmann
    instance = self._UnlockedGetInstanceInfo(instance_name)
1430 2674690b Michael Hanselmann
    if not instance:
1431 2674690b Michael Hanselmann
      raise errors.ConfigurationError("Unknown instance '%s'" % instance_name)
1432 2674690b Michael Hanselmann
1433 2674690b Michael Hanselmann
    if primary_only:
1434 2674690b Michael Hanselmann
      nodes = [instance.primary_node]
1435 2674690b Michael Hanselmann
    else:
1436 2674690b Michael Hanselmann
      nodes = instance.all_nodes
1437 2674690b Michael Hanselmann
1438 2674690b Michael Hanselmann
    return frozenset(self._UnlockedGetNodeInfo(node_name).group
1439 2674690b Michael Hanselmann
                     for node_name in nodes)
1440 2674690b Michael Hanselmann
1441 2674690b Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
1442 71333cb9 Iustin Pop
  def GetMultiInstanceInfo(self, instances):
1443 71333cb9 Iustin Pop
    """Get the configuration of multiple instances.
1444 71333cb9 Iustin Pop

1445 71333cb9 Iustin Pop
    @param instances: list of instance names
1446 71333cb9 Iustin Pop
    @rtype: list
1447 71333cb9 Iustin Pop
    @return: list of tuples (instance, instance_info), where
1448 71333cb9 Iustin Pop
        instance_info is what would GetInstanceInfo return for the
1449 71333cb9 Iustin Pop
        node, while keeping the original order
1450 71333cb9 Iustin Pop

1451 71333cb9 Iustin Pop
    """
1452 71333cb9 Iustin Pop
    return [(name, self._UnlockedGetInstanceInfo(name)) for name in instances]
1453 71333cb9 Iustin Pop
1454 71333cb9 Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
1455 0b2de758 Iustin Pop
  def GetAllInstancesInfo(self):
1456 0b2de758 Iustin Pop
    """Get the configuration of all instances.
1457 0b2de758 Iustin Pop

1458 0b2de758 Iustin Pop
    @rtype: dict
1459 5fcc718f Iustin Pop
    @return: dict of (instance, instance_info), where instance_info is what
1460 0b2de758 Iustin Pop
              would GetInstanceInfo return for the node
1461 0b2de758 Iustin Pop

1462 0b2de758 Iustin Pop
    """
1463 64d3bd52 Guido Trotter
    my_dict = dict([(instance, self._UnlockedGetInstanceInfo(instance))
1464 64d3bd52 Guido Trotter
                    for instance in self._UnlockedGetInstanceList()])
1465 0b2de758 Iustin Pop
    return my_dict
1466 0b2de758 Iustin Pop
1467 cc19798f Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
1468 cc19798f Michael Hanselmann
  def GetInstancesInfoByFilter(self, filter_fn):
1469 cc19798f Michael Hanselmann
    """Get instance configuration with a filter.
1470 cc19798f Michael Hanselmann

1471 cc19798f Michael Hanselmann
    @type filter_fn: callable
1472 cc19798f Michael Hanselmann
    @param filter_fn: Filter function receiving instance object as parameter,
1473 cc19798f Michael Hanselmann
      returning boolean. Important: this function is called while the
1474 cc19798f Michael Hanselmann
      configuration locks is held. It must not do any complex work or call
1475 cc19798f Michael Hanselmann
      functions potentially leading to a deadlock. Ideally it doesn't call any
1476 cc19798f Michael Hanselmann
      other functions and just compares instance attributes.
1477 cc19798f Michael Hanselmann

1478 cc19798f Michael Hanselmann
    """
1479 cc19798f Michael Hanselmann
    return dict((name, inst)
1480 cc19798f Michael Hanselmann
                for (name, inst) in self._config_data.instances.items()
1481 cc19798f Michael Hanselmann
                if filter_fn(inst))
1482 cc19798f Michael Hanselmann
1483 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock)
1484 0debfb35 Guido Trotter
  def AddNode(self, node, ec_id):
1485 a8083063 Iustin Pop
    """Add a node to the configuration.
1486 a8083063 Iustin Pop

1487 c41eea6e Iustin Pop
    @type node: L{objects.Node}
1488 c41eea6e Iustin Pop
    @param node: a Node instance
1489 a8083063 Iustin Pop

1490 a8083063 Iustin Pop
    """
1491 099c52ad Iustin Pop
    logging.info("Adding node %s to configuration", node.name)
1492 d8470559 Michael Hanselmann
1493 0debfb35 Guido Trotter
    self._EnsureUUID(node, ec_id)
1494 430b923c Iustin Pop
1495 b989e85d Iustin Pop
    node.serial_no = 1
1496 d693c864 Iustin Pop
    node.ctime = node.mtime = time.time()
1497 f936c153 Iustin Pop
    self._UnlockedAddNodeToGroup(node.name, node.group)
1498 a8083063 Iustin Pop
    self._config_data.nodes[node.name] = node
1499 b9f72b4e Iustin Pop
    self._config_data.cluster.serial_no += 1
1500 a8083063 Iustin Pop
    self._WriteConfig()
1501 a8083063 Iustin Pop
1502 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock)
1503 a8083063 Iustin Pop
  def RemoveNode(self, node_name):
1504 a8083063 Iustin Pop
    """Remove a node from the configuration.
1505 a8083063 Iustin Pop

1506 a8083063 Iustin Pop
    """
1507 099c52ad Iustin Pop
    logging.info("Removing node %s from configuration", node_name)
1508 d8470559 Michael Hanselmann
1509 a8083063 Iustin Pop
    if node_name not in self._config_data.nodes:
1510 3ecf6786 Iustin Pop
      raise errors.ConfigurationError("Unknown node '%s'" % node_name)
1511 a8083063 Iustin Pop
1512 190e3cb6 Guido Trotter
    self._UnlockedRemoveNodeFromGroup(self._config_data.nodes[node_name])
1513 a8083063 Iustin Pop
    del self._config_data.nodes[node_name]
1514 b9f72b4e Iustin Pop
    self._config_data.cluster.serial_no += 1
1515 a8083063 Iustin Pop
    self._WriteConfig()
1516 a8083063 Iustin Pop
1517 a8083063 Iustin Pop
  def ExpandNodeName(self, short_name):
1518 fe698b38 Michael Hanselmann
    """Attempt to expand an incomplete node name.
1519 a8083063 Iustin Pop

1520 a8083063 Iustin Pop
    """
1521 fe698b38 Michael Hanselmann
    # Locking is done in L{ConfigWriter.GetNodeList}
1522 fe698b38 Michael Hanselmann
    return _MatchNameComponentIgnoreCase(short_name, self.GetNodeList())
1523 a8083063 Iustin Pop
1524 f78ede4e Guido Trotter
  def _UnlockedGetNodeInfo(self, node_name):
1525 a8083063 Iustin Pop
    """Get the configuration of a node, as stored in the config.
1526 a8083063 Iustin Pop

1527 c41eea6e Iustin Pop
    This function is for internal use, when the config lock is already
1528 c41eea6e Iustin Pop
    held.
1529 f78ede4e Guido Trotter

1530 c41eea6e Iustin Pop
    @param node_name: the node name, e.g. I{node1.example.com}
1531 a8083063 Iustin Pop

1532 c41eea6e Iustin Pop
    @rtype: L{objects.Node}
1533 c41eea6e Iustin Pop
    @return: the node object
1534 a8083063 Iustin Pop

1535 a8083063 Iustin Pop
    """
1536 a8083063 Iustin Pop
    if node_name not in self._config_data.nodes:
1537 a8083063 Iustin Pop
      return None
1538 a8083063 Iustin Pop
1539 a8083063 Iustin Pop
    return self._config_data.nodes[node_name]
1540 a8083063 Iustin Pop
1541 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
1542 f78ede4e Guido Trotter
  def GetNodeInfo(self, node_name):
1543 f78ede4e Guido Trotter
    """Get the configuration of a node, as stored in the config.
1544 f78ede4e Guido Trotter

1545 c41eea6e Iustin Pop
    This is just a locked wrapper over L{_UnlockedGetNodeInfo}.
1546 f78ede4e Guido Trotter

1547 c41eea6e Iustin Pop
    @param node_name: the node name, e.g. I{node1.example.com}
1548 c41eea6e Iustin Pop

1549 c41eea6e Iustin Pop
    @rtype: L{objects.Node}
1550 c41eea6e Iustin Pop
    @return: the node object
1551 f78ede4e Guido Trotter

1552 f78ede4e Guido Trotter
    """
1553 f78ede4e Guido Trotter
    return self._UnlockedGetNodeInfo(node_name)
1554 f78ede4e Guido Trotter
1555 8bf9e9a5 Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
1556 8bf9e9a5 Iustin Pop
  def GetNodeInstances(self, node_name):
1557 8bf9e9a5 Iustin Pop
    """Get the instances of a node, as stored in the config.
1558 8bf9e9a5 Iustin Pop

1559 8bf9e9a5 Iustin Pop
    @param node_name: the node name, e.g. I{node1.example.com}
1560 8bf9e9a5 Iustin Pop

1561 8bf9e9a5 Iustin Pop
    @rtype: (list, list)
1562 8bf9e9a5 Iustin Pop
    @return: a tuple with two lists: the primary and the secondary instances
1563 8bf9e9a5 Iustin Pop

1564 8bf9e9a5 Iustin Pop
    """
1565 8bf9e9a5 Iustin Pop
    pri = []
1566 8bf9e9a5 Iustin Pop
    sec = []
1567 8bf9e9a5 Iustin Pop
    for inst in self._config_data.instances.values():
1568 8bf9e9a5 Iustin Pop
      if inst.primary_node == node_name:
1569 8bf9e9a5 Iustin Pop
        pri.append(inst.name)
1570 8bf9e9a5 Iustin Pop
      if node_name in inst.secondary_nodes:
1571 8bf9e9a5 Iustin Pop
        sec.append(inst.name)
1572 8bf9e9a5 Iustin Pop
    return (pri, sec)
1573 8bf9e9a5 Iustin Pop
1574 c71b049c Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
1575 c71b049c Michael Hanselmann
  def GetNodeGroupInstances(self, uuid, primary_only=False):
1576 c71b049c Michael Hanselmann
    """Get the instances of a node group.
1577 c71b049c Michael Hanselmann

1578 c71b049c Michael Hanselmann
    @param uuid: Node group UUID
1579 c71b049c Michael Hanselmann
    @param primary_only: Whether to only consider primary nodes
1580 c71b049c Michael Hanselmann
    @rtype: frozenset
1581 c71b049c Michael Hanselmann
    @return: List of instance names in node group
1582 c71b049c Michael Hanselmann

1583 c71b049c Michael Hanselmann
    """
1584 c71b049c Michael Hanselmann
    if primary_only:
1585 c71b049c Michael Hanselmann
      nodes_fn = lambda inst: [inst.primary_node]
1586 c71b049c Michael Hanselmann
    else:
1587 c71b049c Michael Hanselmann
      nodes_fn = lambda inst: inst.all_nodes
1588 c71b049c Michael Hanselmann
1589 c71b049c Michael Hanselmann
    return frozenset(inst.name
1590 c71b049c Michael Hanselmann
                     for inst in self._config_data.instances.values()
1591 c71b049c Michael Hanselmann
                     for node_name in nodes_fn(inst)
1592 c71b049c Michael Hanselmann
                     if self._UnlockedGetNodeInfo(node_name).group == uuid)
1593 c71b049c Michael Hanselmann
1594 f78ede4e Guido Trotter
  def _UnlockedGetNodeList(self):
1595 a8083063 Iustin Pop
    """Return the list of nodes which are in the configuration.
1596 a8083063 Iustin Pop

1597 c41eea6e Iustin Pop
    This function is for internal use, when the config lock is already
1598 c41eea6e Iustin Pop
    held.
1599 c41eea6e Iustin Pop

1600 c41eea6e Iustin Pop
    @rtype: list
1601 f78ede4e Guido Trotter

1602 a8083063 Iustin Pop
    """
1603 a8083063 Iustin Pop
    return self._config_data.nodes.keys()
1604 a8083063 Iustin Pop
1605 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
1606 f78ede4e Guido Trotter
  def GetNodeList(self):
1607 f78ede4e Guido Trotter
    """Return the list of nodes which are in the configuration.
1608 f78ede4e Guido Trotter

1609 f78ede4e Guido Trotter
    """
1610 f78ede4e Guido Trotter
    return self._UnlockedGetNodeList()
1611 f78ede4e Guido Trotter
1612 6819dc49 Iustin Pop
  def _UnlockedGetOnlineNodeList(self):
1613 94a02bb5 Iustin Pop
    """Return the list of nodes which are online.
1614 94a02bb5 Iustin Pop

1615 94a02bb5 Iustin Pop
    """
1616 94a02bb5 Iustin Pop
    all_nodes = [self._UnlockedGetNodeInfo(node)
1617 94a02bb5 Iustin Pop
                 for node in self._UnlockedGetNodeList()]
1618 94a02bb5 Iustin Pop
    return [node.name for node in all_nodes if not node.offline]
1619 94a02bb5 Iustin Pop
1620 94a02bb5 Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
1621 6819dc49 Iustin Pop
  def GetOnlineNodeList(self):
1622 6819dc49 Iustin Pop
    """Return the list of nodes which are online.
1623 6819dc49 Iustin Pop

1624 6819dc49 Iustin Pop
    """
1625 6819dc49 Iustin Pop
    return self._UnlockedGetOnlineNodeList()
1626 6819dc49 Iustin Pop
1627 6819dc49 Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
1628 075b62ca Iustin Pop
  def GetVmCapableNodeList(self):
1629 075b62ca Iustin Pop
    """Return the list of nodes which are not vm capable.
1630 075b62ca Iustin Pop

1631 075b62ca Iustin Pop
    """
1632 075b62ca Iustin Pop
    all_nodes = [self._UnlockedGetNodeInfo(node)
1633 075b62ca Iustin Pop
                 for node in self._UnlockedGetNodeList()]
1634 075b62ca Iustin Pop
    return [node.name for node in all_nodes if node.vm_capable]
1635 075b62ca Iustin Pop
1636 075b62ca Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
1637 8bf9e9a5 Iustin Pop
  def GetNonVmCapableNodeList(self):
1638 8bf9e9a5 Iustin Pop
    """Return the list of nodes which are not vm capable.
1639 8bf9e9a5 Iustin Pop

1640 8bf9e9a5 Iustin Pop
    """
1641 8bf9e9a5 Iustin Pop
    all_nodes = [self._UnlockedGetNodeInfo(node)
1642 8bf9e9a5 Iustin Pop
                 for node in self._UnlockedGetNodeList()]
1643 8bf9e9a5 Iustin Pop
    return [node.name for node in all_nodes if not node.vm_capable]
1644 8bf9e9a5 Iustin Pop
1645 8bf9e9a5 Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
1646 f5eaa3c1 Iustin Pop
  def GetMultiNodeInfo(self, nodes):
1647 f5eaa3c1 Iustin Pop
    """Get the configuration of multiple nodes.
1648 f5eaa3c1 Iustin Pop

1649 f5eaa3c1 Iustin Pop
    @param nodes: list of node names
1650 f5eaa3c1 Iustin Pop
    @rtype: list
1651 f5eaa3c1 Iustin Pop
    @return: list of tuples of (node, node_info), where node_info is
1652 f5eaa3c1 Iustin Pop
        what would GetNodeInfo return for the node, in the original
1653 f5eaa3c1 Iustin Pop
        order
1654 f5eaa3c1 Iustin Pop

1655 f5eaa3c1 Iustin Pop
    """
1656 f5eaa3c1 Iustin Pop
    return [(name, self._UnlockedGetNodeInfo(name)) for name in nodes]
1657 f5eaa3c1 Iustin Pop
1658 f5eaa3c1 Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
1659 d65e5776 Iustin Pop
  def GetAllNodesInfo(self):
1660 d65e5776 Iustin Pop
    """Get the configuration of all nodes.
1661 d65e5776 Iustin Pop

1662 d65e5776 Iustin Pop
    @rtype: dict
1663 ec0292f1 Iustin Pop
    @return: dict of (node, node_info), where node_info is what
1664 d65e5776 Iustin Pop
              would GetNodeInfo return for the node
1665 d65e5776 Iustin Pop

1666 d65e5776 Iustin Pop
    """
1667 ee14d800 Michael Hanselmann
    return self._UnlockedGetAllNodesInfo()
1668 ee14d800 Michael Hanselmann
1669 ee14d800 Michael Hanselmann
  def _UnlockedGetAllNodesInfo(self):
1670 ee14d800 Michael Hanselmann
    """Gets configuration of all nodes.
1671 ee14d800 Michael Hanselmann

1672 ee14d800 Michael Hanselmann
    @note: See L{GetAllNodesInfo}
1673 ee14d800 Michael Hanselmann

1674 ee14d800 Michael Hanselmann
    """
1675 ee14d800 Michael Hanselmann
    return dict([(node, self._UnlockedGetNodeInfo(node))
1676 ee14d800 Michael Hanselmann
                 for node in self._UnlockedGetNodeList()])
1677 d65e5776 Iustin Pop
1678 9d5b1371 Michael Hanselmann
  @locking.ssynchronized(_config_lock, shared=1)
1679 9d5b1371 Michael Hanselmann
  def GetNodeGroupsFromNodes(self, nodes):
1680 9d5b1371 Michael Hanselmann
    """Returns groups for a list of nodes.
1681 9d5b1371 Michael Hanselmann

1682 9d5b1371 Michael Hanselmann
    @type nodes: list of string
1683 9d5b1371 Michael Hanselmann
    @param nodes: List of node names
1684 9d5b1371 Michael Hanselmann
    @rtype: frozenset
1685 9d5b1371 Michael Hanselmann

1686 9d5b1371 Michael Hanselmann
    """
1687 9d5b1371 Michael Hanselmann
    return frozenset(self._UnlockedGetNodeInfo(name).group for name in nodes)
1688 9d5b1371 Michael Hanselmann
1689 23f06b2b Iustin Pop
  def _UnlockedGetMasterCandidateStats(self, exceptions=None):
1690 ec0292f1 Iustin Pop
    """Get the number of current and maximum desired and possible candidates.
1691 ec0292f1 Iustin Pop

1692 23f06b2b Iustin Pop
    @type exceptions: list
1693 23f06b2b Iustin Pop
    @param exceptions: if passed, list of nodes that should be ignored
1694 ec0292f1 Iustin Pop
    @rtype: tuple
1695 e623dbe3 Guido Trotter
    @return: tuple of (current, desired and possible, possible)
1696 ec0292f1 Iustin Pop

1697 ec0292f1 Iustin Pop
    """
1698 e623dbe3 Guido Trotter
    mc_now = mc_should = mc_max = 0
1699 23f06b2b Iustin Pop
    for node in self._config_data.nodes.values():
1700 23f06b2b Iustin Pop
      if exceptions and node.name in exceptions:
1701 23f06b2b Iustin Pop
        continue
1702 490acd18 Iustin Pop
      if not (node.offline or node.drained) and node.master_capable:
1703 ec0292f1 Iustin Pop
        mc_max += 1
1704 ec0292f1 Iustin Pop
      if node.master_candidate:
1705 ec0292f1 Iustin Pop
        mc_now += 1
1706 e623dbe3 Guido Trotter
    mc_should = min(mc_max, self._config_data.cluster.candidate_pool_size)
1707 e623dbe3 Guido Trotter
    return (mc_now, mc_should, mc_max)
1708 ec0292f1 Iustin Pop
1709 ec0292f1 Iustin Pop
  @locking.ssynchronized(_config_lock, shared=1)
1710 23f06b2b Iustin Pop
  def GetMasterCandidateStats(self, exceptions=None):
1711 ec0292f1 Iustin Pop
    """Get the number of current and maximum possible candidates.
1712 ec0292f1 Iustin Pop

1713 ec0292f1 Iustin Pop
    This is just a wrapper over L{_UnlockedGetMasterCandidateStats}.
1714 ec0292f1 Iustin Pop

1715 23f06b2b Iustin Pop
    @type exceptions: list
1716 23f06b2b Iustin Pop
    @param exceptions: if passed, list of nodes that should be ignored
1717 ec0292f1 Iustin Pop
    @rtype: tuple
1718 ec0292f1 Iustin Pop
    @return: tuple of (current, max)
1719 ec0292f1 Iustin Pop

1720 ec0292f1 Iustin Pop
    """
1721 23f06b2b Iustin Pop
    return self._UnlockedGetMasterCandidateStats(exceptions)
1722 ec0292f1 Iustin Pop
1723 ec0292f1 Iustin Pop
  @locking.ssynchronized(_config_lock)
1724 44485f49 Guido Trotter
  def MaintainCandidatePool(self, exceptions):
1725 ec0292f1 Iustin Pop
    """Try to grow the candidate pool to the desired size.
1726 ec0292f1 Iustin Pop

1727 44485f49 Guido Trotter
    @type exceptions: list
1728 44485f49 Guido Trotter
    @param exceptions: if passed, list of nodes that should be ignored
1729 ec0292f1 Iustin Pop
    @rtype: list
1730 ee513a66 Iustin Pop
    @return: list with the adjusted nodes (L{objects.Node} instances)
1731 ec0292f1 Iustin Pop

1732 ec0292f1 Iustin Pop
    """
1733 44485f49 Guido Trotter
    mc_now, mc_max, _ = self._UnlockedGetMasterCandidateStats(exceptions)
1734 ec0292f1 Iustin Pop
    mod_list = []
1735 ec0292f1 Iustin Pop
    if mc_now < mc_max:
1736 ec0292f1 Iustin Pop
      node_list = self._config_data.nodes.keys()
1737 ec0292f1 Iustin Pop
      random.shuffle(node_list)
1738 ec0292f1 Iustin Pop
      for name in node_list:
1739 ec0292f1 Iustin Pop
        if mc_now >= mc_max:
1740 ec0292f1 Iustin Pop
          break
1741 ec0292f1 Iustin Pop
        node = self._config_data.nodes[name]
1742 44485f49 Guido Trotter
        if (node.master_candidate or node.offline or node.drained or
1743 490acd18 Iustin Pop
            node.name in exceptions or not node.master_capable):
1744 ec0292f1 Iustin Pop
          continue
1745 ee513a66 Iustin Pop
        mod_list.append(node)
1746 ec0292f1 Iustin Pop
        node.master_candidate = True
1747 ec0292f1 Iustin Pop
        node.serial_no += 1
1748 ec0292f1 Iustin Pop
        mc_now += 1
1749 ec0292f1 Iustin Pop
      if mc_now != mc_max:
1750 ec0292f1 Iustin Pop
        # this should not happen
1751 ec0292f1 Iustin Pop
        logging.warning("Warning: MaintainCandidatePool didn't manage to"
1752 ec0292f1 Iustin Pop
                        " fill the candidate pool (%d/%d)", mc_now, mc_max)
1753 ec0292f1 Iustin Pop
      if mod_list:
1754 ec0292f1 Iustin Pop
        self._config_data.cluster.serial_no += 1
1755 ec0292f1 Iustin Pop
        self._WriteConfig()
1756 ec0292f1 Iustin Pop
1757 ec0292f1 Iustin Pop
    return mod_list
1758 ec0292f1 Iustin Pop
1759 190e3cb6 Guido Trotter
  def _UnlockedAddNodeToGroup(self, node_name, nodegroup_uuid):
1760 190e3cb6 Guido Trotter
    """Add a given node to the specified group.
1761 190e3cb6 Guido Trotter

1762 190e3cb6 Guido Trotter
    """
1763 190e3cb6 Guido Trotter
    if nodegroup_uuid not in self._config_data.nodegroups:
1764 190e3cb6 Guido Trotter
      # This can happen if a node group gets deleted between its lookup and
1765 190e3cb6 Guido Trotter
      # when we're adding the first node to it, since we don't keep a lock in
1766 190e3cb6 Guido Trotter
      # the meantime. It's ok though, as we'll fail cleanly if the node group
1767 190e3cb6 Guido Trotter
      # is not found anymore.
1768 f936c153 Iustin Pop
      raise errors.OpExecError("Unknown node group: %s" % nodegroup_uuid)
1769 190e3cb6 Guido Trotter
    if node_name not in self._config_data.nodegroups[nodegroup_uuid].members:
1770 190e3cb6 Guido Trotter
      self._config_data.nodegroups[nodegroup_uuid].members.append(node_name)
1771 190e3cb6 Guido Trotter
1772 190e3cb6 Guido Trotter
  def _UnlockedRemoveNodeFromGroup(self, node):
1773 190e3cb6 Guido Trotter
    """Remove a given node from its group.
1774 190e3cb6 Guido Trotter

1775 190e3cb6 Guido Trotter
    """
1776 f936c153 Iustin Pop
    nodegroup = node.group
1777 190e3cb6 Guido Trotter
    if nodegroup not in self._config_data.nodegroups:
1778 f936c153 Iustin Pop
      logging.warning("Warning: node '%s' has unknown node group '%s'"
1779 190e3cb6 Guido Trotter
                      " (while being removed from it)", node.name, nodegroup)
1780 190e3cb6 Guido Trotter
    nodegroup_obj = self._config_data.nodegroups[nodegroup]
1781 190e3cb6 Guido Trotter
    if node.name not in nodegroup_obj.members:
1782 f936c153 Iustin Pop
      logging.warning("Warning: node '%s' not a member of its node group '%s'"
1783 190e3cb6 Guido Trotter
                      " (while being removed from it)", node.name, nodegroup)
1784 190e3cb6 Guido Trotter
    else:
1785 190e3cb6 Guido Trotter
      nodegroup_obj.members.remove(node.name)
1786 190e3cb6 Guido Trotter
1787 54c31fd3 Michael Hanselmann
  @locking.ssynchronized(_config_lock)
1788 54c31fd3 Michael Hanselmann
  def AssignGroupNodes(self, mods):
1789 54c31fd3 Michael Hanselmann
    """Changes the group of a number of nodes.
1790 54c31fd3 Michael Hanselmann

1791 54c31fd3 Michael Hanselmann
    @type mods: list of tuples; (node name, new group UUID)
1792 1d4930b9 Michael Hanselmann
    @param mods: Node membership modifications
1793 54c31fd3 Michael Hanselmann

1794 54c31fd3 Michael Hanselmann
    """
1795 54c31fd3 Michael Hanselmann
    groups = self._config_data.nodegroups
1796 54c31fd3 Michael Hanselmann
    nodes = self._config_data.nodes
1797 54c31fd3 Michael Hanselmann
1798 54c31fd3 Michael Hanselmann
    resmod = []
1799 54c31fd3 Michael Hanselmann
1800 54c31fd3 Michael Hanselmann
    # Try to resolve names/UUIDs first
1801 54c31fd3 Michael Hanselmann
    for (node_name, new_group_uuid) in mods:
1802 54c31fd3 Michael Hanselmann
      try:
1803 54c31fd3 Michael Hanselmann
        node = nodes[node_name]
1804 54c31fd3 Michael Hanselmann
      except KeyError:
1805 54c31fd3 Michael Hanselmann
        raise errors.ConfigurationError("Unable to find node '%s'" % node_name)
1806 54c31fd3 Michael Hanselmann
1807 54c31fd3 Michael Hanselmann
      if node.group == new_group_uuid:
1808 54c31fd3 Michael Hanselmann
        # Node is being assigned to its current group
1809 54c31fd3 Michael Hanselmann
        logging.debug("Node '%s' was assigned to its current group (%s)",
1810 54c31fd3 Michael Hanselmann
                      node_name, node.group)
1811 54c31fd3 Michael Hanselmann
        continue
1812 54c31fd3 Michael Hanselmann
1813 54c31fd3 Michael Hanselmann
      # Try to find current group of node
1814 54c31fd3 Michael Hanselmann
      try:
1815 54c31fd3 Michael Hanselmann
        old_group = groups[node.group]
1816 54c31fd3 Michael Hanselmann
      except KeyError:
1817 54c31fd3 Michael Hanselmann
        raise errors.ConfigurationError("Unable to find old group '%s'" %
1818 54c31fd3 Michael Hanselmann
                                        node.group)
1819 54c31fd3 Michael Hanselmann
1820 54c31fd3 Michael Hanselmann
      # Try to find new group for node
1821 54c31fd3 Michael Hanselmann
      try:
1822 54c31fd3 Michael Hanselmann
        new_group = groups[new_group_uuid]
1823 54c31fd3 Michael Hanselmann
      except KeyError:
1824 54c31fd3 Michael Hanselmann
        raise errors.ConfigurationError("Unable to find new group '%s'" %
1825 54c31fd3 Michael Hanselmann
                                        new_group_uuid)
1826 54c31fd3 Michael Hanselmann
1827 54c31fd3 Michael Hanselmann
      assert node.name in old_group.members, \
1828 54c31fd3 Michael Hanselmann
        ("Inconsistent configuration: node '%s' not listed in members for its"
1829 54c31fd3 Michael Hanselmann
         " old group '%s'" % (node.name, old_group.uuid))
1830 54c31fd3 Michael Hanselmann
      assert node.name not in new_group.members, \
1831 54c31fd3 Michael Hanselmann
        ("Inconsistent configuration: node '%s' already listed in members for"
1832 54c31fd3 Michael Hanselmann
         " its new group '%s'" % (node.name, new_group.uuid))
1833 54c31fd3 Michael Hanselmann
1834 54c31fd3 Michael Hanselmann
      resmod.append((node, old_group, new_group))
1835 54c31fd3 Michael Hanselmann
1836 54c31fd3 Michael Hanselmann
    # Apply changes
1837 54c31fd3 Michael Hanselmann
    for (node, old_group, new_group) in resmod:
1838 54c31fd3 Michael Hanselmann
      assert node.uuid != new_group.uuid and old_group.uuid != new_group.uuid, \
1839 54c31fd3 Michael Hanselmann
        "Assigning to current group is not possible"
1840 54c31fd3 Michael Hanselmann
1841 54c31fd3 Michael Hanselmann
      node.group = new_group.uuid
1842 54c31fd3 Michael Hanselmann
1843 54c31fd3 Michael Hanselmann
      # Update members of involved groups
1844 54c31fd3 Michael Hanselmann
      if node.name in old_group.members:
1845 54c31fd3 Michael Hanselmann
        old_group.members.remove(node.name)
1846 54c31fd3 Michael Hanselmann
      if node.name not in new_group.members:
1847 54c31fd3 Michael Hanselmann
        new_group.members.append(node.name)
1848 54c31fd3 Michael Hanselmann
1849 54c31fd3 Michael Hanselmann
    # Update timestamps and serials (only once per node/group object)
1850 54c31fd3 Michael Hanselmann
    now = time.time()
1851 75191077 Michael Hanselmann
    for obj in frozenset(itertools.chain(*resmod)): # pylint: disable=W0142
1852 54c31fd3 Michael Hanselmann
      obj.serial_no += 1
1853 54c31fd3 Michael Hanselmann
      obj.mtime = now
1854 54c31fd3 Michael Hanselmann
1855 54c31fd3 Michael Hanselmann
    # Force ssconf update
1856 54c31fd3 Michael Hanselmann
    self._config_data.cluster.serial_no += 1
1857 54c31fd3 Michael Hanselmann
1858 54c31fd3 Michael Hanselmann
    self._WriteConfig()
1859 54c31fd3 Michael Hanselmann
1860 a8083063 Iustin Pop
  def _BumpSerialNo(self):
1861 a8083063 Iustin Pop
    """Bump up the serial number of the config.
1862 a8083063 Iustin Pop

1863 a8083063 Iustin Pop
    """
1864 9d38c6e1 Iustin Pop
    self._config_data.serial_no += 1
1865 d693c864 Iustin Pop
    self._config_data.mtime = time.time()
1866 a8083063 Iustin Pop
1867 76d5d3a3 Iustin Pop
  def _AllUUIDObjects(self):
1868 76d5d3a3 Iustin Pop
    """Returns all objects with uuid attributes.
1869 76d5d3a3 Iustin Pop

1870 76d5d3a3 Iustin Pop
    """
1871 76d5d3a3 Iustin Pop
    return (self._config_data.instances.values() +
1872 76d5d3a3 Iustin Pop
            self._config_data.nodes.values() +
1873 3df43542 Guido Trotter
            self._config_data.nodegroups.values() +
1874 76d5d3a3 Iustin Pop
            [self._config_data.cluster])
1875 76d5d3a3 Iustin Pop
1876 eb180fe2 Iustin Pop
  def _OpenConfig(self, accept_foreign):
1877 a8083063 Iustin Pop
    """Read the config data from disk.
1878 a8083063 Iustin Pop

1879 a8083063 Iustin Pop
    """
1880 13998ef2 Michael Hanselmann
    raw_data = utils.ReadFile(self._cfg_file)
1881 13998ef2 Michael Hanselmann
1882 a8083063 Iustin Pop
    try:
1883 13998ef2 Michael Hanselmann
      data = objects.ConfigData.FromDict(serializer.Load(raw_data))
1884 13998ef2 Michael Hanselmann
    except Exception, err:
1885 13998ef2 Michael Hanselmann
      raise errors.ConfigurationError(err)
1886 5b263ed7 Michael Hanselmann
1887 5b263ed7 Michael Hanselmann
    # Make sure the configuration has the right version
1888 5b263ed7 Michael Hanselmann
    _ValidateConfig(data)
1889 5b263ed7 Michael Hanselmann
1890 3ccb3a64 Michael Hanselmann
    if (not hasattr(data, "cluster") or
1891 3ccb3a64 Michael Hanselmann
        not hasattr(data.cluster, "rsahostkeypub")):
1892 3ecf6786 Iustin Pop
      raise errors.ConfigurationError("Incomplete configuration"
1893 243cdbcc Michael Hanselmann
                                      " (missing cluster.rsahostkeypub)")
1894 90d726a8 Iustin Pop
1895 eb180fe2 Iustin Pop
    if data.cluster.master_node != self._my_hostname and not accept_foreign:
1896 eb180fe2 Iustin Pop
      msg = ("The configuration denotes node %s as master, while my"
1897 eb180fe2 Iustin Pop
             " hostname is %s; opening a foreign configuration is only"
1898 eb180fe2 Iustin Pop
             " possible in accept_foreign mode" %
1899 eb180fe2 Iustin Pop
             (data.cluster.master_node, self._my_hostname))
1900 eb180fe2 Iustin Pop
      raise errors.ConfigurationError(msg)
1901 eb180fe2 Iustin Pop
1902 90d726a8 Iustin Pop
    # Upgrade configuration if needed
1903 90d726a8 Iustin Pop
    data.UpgradeConfig()
1904 90d726a8 Iustin Pop
1905 a8083063 Iustin Pop
    self._config_data = data
1906 3c7f6c44 Iustin Pop
    # reset the last serial as -1 so that the next write will cause
1907 0779e3aa Iustin Pop
    # ssconf update
1908 0779e3aa Iustin Pop
    self._last_cluster_serial = -1
1909 a8083063 Iustin Pop
1910 76d5d3a3 Iustin Pop
    # And finally run our (custom) config upgrade sequence
1911 76d5d3a3 Iustin Pop
    self._UpgradeConfig()
1912 76d5d3a3 Iustin Pop
1913 bd407597 Iustin Pop
    self._cfg_id = utils.GetFileID(path=self._cfg_file)
1914 bd407597 Iustin Pop
1915 76d5d3a3 Iustin Pop
  def _UpgradeConfig(self):
1916 76d5d3a3 Iustin Pop
    """Run upgrade steps that cannot be done purely in the objects.
1917 76d5d3a3 Iustin Pop

1918 76d5d3a3 Iustin Pop
    This is because some data elements need uniqueness across the
1919 76d5d3a3 Iustin Pop
    whole configuration, etc.
1920 76d5d3a3 Iustin Pop

1921 111c4e2f Guido Trotter
    @warning: this function will call L{_WriteConfig()}, but also
1922 111c4e2f Guido Trotter
        L{DropECReservations} so it needs to be called only from a
1923 111c4e2f Guido Trotter
        "safe" place (the constructor). If one wanted to call it with
1924 111c4e2f Guido Trotter
        the lock held, a DropECReservationUnlocked would need to be
1925 111c4e2f Guido Trotter
        created first, to avoid causing deadlock.
1926 76d5d3a3 Iustin Pop

1927 76d5d3a3 Iustin Pop
    """
1928 76d5d3a3 Iustin Pop
    modified = False
1929 76d5d3a3 Iustin Pop
    for item in self._AllUUIDObjects():
1930 76d5d3a3 Iustin Pop
      if item.uuid is None:
1931 4fae38c5 Guido Trotter
        item.uuid = self._GenerateUniqueID(_UPGRADE_CONFIG_JID)
1932 76d5d3a3 Iustin Pop
        modified = True
1933 f9e81396 Guido Trotter
    if not self._config_data.nodegroups:
1934 75cf411a Adeodato Simo
      default_nodegroup_name = constants.INITIAL_NODE_GROUP_NAME
1935 75cf411a Adeodato Simo
      default_nodegroup = objects.NodeGroup(name=default_nodegroup_name,
1936 75cf411a Adeodato Simo
                                            members=[])
1937 e11a1b77 Adeodato Simo
      self._UnlockedAddNodeGroup(default_nodegroup, _UPGRADE_CONFIG_JID, True)
1938 f9e81396 Guido Trotter
      modified = True
1939 190e3cb6 Guido Trotter
    for node in self._config_data.nodes.values():
1940 f936c153 Iustin Pop
      if not node.group:
1941 f936c153 Iustin Pop
        node.group = self.LookupNodeGroup(None)
1942 190e3cb6 Guido Trotter
        modified = True
1943 190e3cb6 Guido Trotter
      # This is technically *not* an upgrade, but needs to be done both when
1944 190e3cb6 Guido Trotter
      # nodegroups are being added, and upon normally loading the config,
1945 190e3cb6 Guido Trotter
      # because the members list of a node group is discarded upon
1946 190e3cb6 Guido Trotter
      # serializing/deserializing the object.
1947 f936c153 Iustin Pop
      self._UnlockedAddNodeToGroup(node.name, node.group)
1948 76d5d3a3 Iustin Pop
    if modified:
1949 76d5d3a3 Iustin Pop
      self._WriteConfig()
1950 4fae38c5 Guido Trotter
      # This is ok even if it acquires the internal lock, as _UpgradeConfig is
1951 4fae38c5 Guido Trotter
      # only called at config init time, without the lock held
1952 4fae38c5 Guido Trotter
      self.DropECReservations(_UPGRADE_CONFIG_JID)
1953 4fae38c5 Guido Trotter
1954 a4eae71f Michael Hanselmann
  def _DistributeConfig(self, feedback_fn):
1955 a8083063 Iustin Pop
    """Distribute the configuration to the other nodes.
1956 a8083063 Iustin Pop

1957 a8083063 Iustin Pop
    Currently, this only copies the configuration file. In the future,
1958 a8083063 Iustin Pop
    it could be used to encapsulate the 2/3-phase update mechanism.
1959 a8083063 Iustin Pop

1960 a8083063 Iustin Pop
    """
1961 a8083063 Iustin Pop
    if self._offline:
1962 a8083063 Iustin Pop
      return True
1963 a4eae71f Michael Hanselmann
1964 a8083063 Iustin Pop
    bad = False
1965 a8083063 Iustin Pop
1966 6a5b8b4b Iustin Pop
    node_list = []
1967 6a5b8b4b Iustin Pop
    addr_list = []
1968 6a5b8b4b Iustin Pop
    myhostname = self._my_hostname
1969 6b294c53 Iustin Pop
    # we can skip checking whether _UnlockedGetNodeInfo returns None
1970 6b294c53 Iustin Pop
    # since the node list comes from _UnlocketGetNodeList, and we are
1971 6b294c53 Iustin Pop
    # called with the lock held, so no modifications should take place
1972 6b294c53 Iustin Pop
    # in between
1973 6a5b8b4b Iustin Pop
    for node_name in self._UnlockedGetNodeList():
1974 6a5b8b4b Iustin Pop
      if node_name == myhostname:
1975 6a5b8b4b Iustin Pop
        continue
1976 6a5b8b4b Iustin Pop
      node_info = self._UnlockedGetNodeInfo(node_name)
1977 6a5b8b4b Iustin Pop
      if not node_info.master_candidate:
1978 6a5b8b4b Iustin Pop
        continue
1979 6a5b8b4b Iustin Pop
      node_list.append(node_info.name)
1980 6a5b8b4b Iustin Pop
      addr_list.append(node_info.primary_ip)
1981 6b294c53 Iustin Pop
1982 415a7304 Michael Hanselmann
    # TODO: Use dedicated resolver talking to config writer for name resolution
1983 415a7304 Michael Hanselmann
    result = \
1984 b2acdbdc Michael Hanselmann
      self._GetRpc(addr_list).call_upload_file(node_list, self._cfg_file)
1985 1b54fc6c Guido Trotter
    for to_node, to_result in result.items():
1986 3cebe102 Michael Hanselmann
      msg = to_result.fail_msg
1987 1b54fc6c Guido Trotter
      if msg:
1988 1b54fc6c Guido Trotter
        msg = ("Copy of file %s to node %s failed: %s" %
1989 dd7db360 Iustin Pop
               (self._cfg_file, to_node, msg))
1990 1b54fc6c Guido Trotter
        logging.error(msg)
1991 a4eae71f Michael Hanselmann
1992 a4eae71f Michael Hanselmann
        if feedback_fn:
1993 a4eae71f Michael Hanselmann
          feedback_fn(msg)
1994 a4eae71f Michael Hanselmann
1995 a8083063 Iustin Pop
        bad = True
1996 a4eae71f Michael Hanselmann
1997 a8083063 Iustin Pop
    return not bad
1998 a8083063 Iustin Pop
1999 a4eae71f Michael Hanselmann
  def _WriteConfig(self, destination=None, feedback_fn=None):
2000 a8083063 Iustin Pop
    """Write the configuration data to persistent storage.
2001 a8083063 Iustin Pop

2002 a8083063 Iustin Pop
    """
2003 a4eae71f Michael Hanselmann
    assert feedback_fn is None or callable(feedback_fn)
2004 a4eae71f Michael Hanselmann
2005 d2231b8c Iustin Pop
    # Warn on config errors, but don't abort the save - the
2006 d2231b8c Iustin Pop
    # configuration has already been modified, and we can't revert;
2007 d2231b8c Iustin Pop
    # the best we can do is to warn the user and save as is, leaving
2008 d2231b8c Iustin Pop
    # recovery to the user
2009 4a89c54a Iustin Pop
    config_errors = self._UnlockedVerifyConfig()
2010 4a89c54a Iustin Pop
    if config_errors:
2011 d2231b8c Iustin Pop
      errmsg = ("Configuration data is not consistent: %s" %
2012 1f864b60 Iustin Pop
                (utils.CommaJoin(config_errors)))
2013 d2231b8c Iustin Pop
      logging.critical(errmsg)
2014 d2231b8c Iustin Pop
      if feedback_fn:
2015 d2231b8c Iustin Pop
        feedback_fn(errmsg)
2016 d2231b8c Iustin Pop
2017 a8083063 Iustin Pop
    if destination is None:
2018 a8083063 Iustin Pop
      destination = self._cfg_file
2019 a8083063 Iustin Pop
    self._BumpSerialNo()
2020 8d14b30d Iustin Pop
    txt = serializer.Dump(self._config_data.ToDict())
2021 13998ef2 Michael Hanselmann
2022 e60c73a1 René Nussbaumer
    getents = self._getents()
2023 bd407597 Iustin Pop
    try:
2024 bd407597 Iustin Pop
      fd = utils.SafeWriteFile(destination, self._cfg_id, data=txt,
2025 bd407597 Iustin Pop
                               close=False, gid=getents.confd_gid, mode=0640)
2026 bd407597 Iustin Pop
    except errors.LockError:
2027 bd407597 Iustin Pop
      raise errors.ConfigurationError("The configuration file has been"
2028 bd407597 Iustin Pop
                                      " modified since the last write, cannot"
2029 bd407597 Iustin Pop
                                      " update")
2030 bd407597 Iustin Pop
    try:
2031 bd407597 Iustin Pop
      self._cfg_id = utils.GetFileID(fd=fd)
2032 bd407597 Iustin Pop
    finally:
2033 bd407597 Iustin Pop
      os.close(fd)
2034 13998ef2 Michael Hanselmann
2035 14e15659 Iustin Pop
    self.write_count += 1
2036 3d3a04bc Iustin Pop
2037 f56618e0 Iustin Pop
    # and redistribute the config file to master candidates
2038 a4eae71f Michael Hanselmann
    self._DistributeConfig(feedback_fn)
2039 a8083063 Iustin Pop
2040 54d1a06e Michael Hanselmann
    # Write ssconf files on all nodes (including locally)
2041 0779e3aa Iustin Pop
    if self._last_cluster_serial < self._config_data.cluster.serial_no:
2042 d9a855f1 Michael Hanselmann
      if not self._offline:
2043 b2acdbdc Michael Hanselmann
        result = self._GetRpc(None).call_write_ssconf_files(
2044 6819dc49 Iustin Pop
          self._UnlockedGetOnlineNodeList(),
2045 e1e75d00 Iustin Pop
          self._UnlockedGetSsconfValues())
2046 a4eae71f Michael Hanselmann
2047 e1e75d00 Iustin Pop
        for nname, nresu in result.items():
2048 3cebe102 Michael Hanselmann
          msg = nresu.fail_msg
2049 e1e75d00 Iustin Pop
          if msg:
2050 a4eae71f Michael Hanselmann
            errmsg = ("Error while uploading ssconf files to"
2051 a4eae71f Michael Hanselmann
                      " node %s: %s" % (nname, msg))
2052 a4eae71f Michael Hanselmann
            logging.warning(errmsg)
2053 a4eae71f Michael Hanselmann
2054 a4eae71f Michael Hanselmann
            if feedback_fn:
2055 a4eae71f Michael Hanselmann
              feedback_fn(errmsg)
2056 a4eae71f Michael Hanselmann
2057 0779e3aa Iustin Pop
      self._last_cluster_serial = self._config_data.cluster.serial_no
2058 54d1a06e Michael Hanselmann
2059 03d1dba2 Michael Hanselmann
  def _UnlockedGetSsconfValues(self):
2060 054596f0 Iustin Pop
    """Return the values needed by ssconf.
2061 054596f0 Iustin Pop

2062 054596f0 Iustin Pop
    @rtype: dict
2063 054596f0 Iustin Pop
    @return: a dictionary with keys the ssconf names and values their
2064 054596f0 Iustin Pop
        associated value
2065 054596f0 Iustin Pop

2066 054596f0 Iustin Pop
    """
2067 a3316e4a Iustin Pop
    fn = "\n".join
2068 81a49123 Iustin Pop
    instance_names = utils.NiceSort(self._UnlockedGetInstanceList())
2069 a3316e4a Iustin Pop
    node_names = utils.NiceSort(self._UnlockedGetNodeList())
2070 a3316e4a Iustin Pop
    node_info = [self._UnlockedGetNodeInfo(name) for name in node_names]
2071 c3029d0a Luca Bigliardi
    node_pri_ips = ["%s %s" % (ninfo.name, ninfo.primary_ip)
2072 5909fb97 Luca Bigliardi
                    for ninfo in node_info]
2073 c3029d0a Luca Bigliardi
    node_snd_ips = ["%s %s" % (ninfo.name, ninfo.secondary_ip)
2074 5909fb97 Luca Bigliardi
                    for ninfo in node_info]
2075 a3316e4a Iustin Pop
2076 81a49123 Iustin Pop
    instance_data = fn(instance_names)
2077 a3316e4a Iustin Pop
    off_data = fn(node.name for node in node_info if node.offline)
2078 81a49123 Iustin Pop
    on_data = fn(node.name for node in node_info if not node.offline)
2079 a3316e4a Iustin Pop
    mc_data = fn(node.name for node in node_info if node.master_candidate)
2080 8113a52e Luca Bigliardi
    mc_ips_data = fn(node.primary_ip for node in node_info
2081 8113a52e Luca Bigliardi
                     if node.master_candidate)
2082 a3316e4a Iustin Pop
    node_data = fn(node_names)
2083 f9780ccd Luca Bigliardi
    node_pri_ips_data = fn(node_pri_ips)
2084 f9780ccd Luca Bigliardi
    node_snd_ips_data = fn(node_snd_ips)
2085 f56618e0 Iustin Pop
2086 054596f0 Iustin Pop
    cluster = self._config_data.cluster
2087 5d60b3bd Iustin Pop
    cluster_tags = fn(cluster.GetTags())
2088 4f7a6a10 Iustin Pop
2089 4f7a6a10 Iustin Pop
    hypervisor_list = fn(cluster.enabled_hypervisors)
2090 4f7a6a10 Iustin Pop
2091 0fbae49a Balazs Lecz
    uid_pool = uidpool.FormatUidPool(cluster.uid_pool, separator="\n")
2092 0fbae49a Balazs Lecz
2093 6f076453 Guido Trotter
    nodegroups = ["%s %s" % (nodegroup.uuid, nodegroup.name) for nodegroup in
2094 6f076453 Guido Trotter
                  self._config_data.nodegroups.values()]
2095 6f076453 Guido Trotter
    nodegroups_data = fn(utils.NiceSort(nodegroups))
2096 6f076453 Guido Trotter
2097 2afc9238 Iustin Pop
    ssconf_values = {
2098 054596f0 Iustin Pop
      constants.SS_CLUSTER_NAME: cluster.cluster_name,
2099 5d60b3bd Iustin Pop
      constants.SS_CLUSTER_TAGS: cluster_tags,
2100 054596f0 Iustin Pop
      constants.SS_FILE_STORAGE_DIR: cluster.file_storage_dir,
2101 4b97f902 Apollon Oikonomopoulos
      constants.SS_SHARED_FILE_STORAGE_DIR: cluster.shared_file_storage_dir,
2102 a3316e4a Iustin Pop
      constants.SS_MASTER_CANDIDATES: mc_data,
2103 8113a52e Luca Bigliardi
      constants.SS_MASTER_CANDIDATES_IPS: mc_ips_data,
2104 054596f0 Iustin Pop
      constants.SS_MASTER_IP: cluster.master_ip,
2105 054596f0 Iustin Pop
      constants.SS_MASTER_NETDEV: cluster.master_netdev,
2106 5a8648eb Andrea Spadaccini
      constants.SS_MASTER_NETMASK: str(cluster.master_netmask),
2107 054596f0 Iustin Pop
      constants.SS_MASTER_NODE: cluster.master_node,
2108 a3316e4a Iustin Pop
      constants.SS_NODE_LIST: node_data,
2109 f9780ccd Luca Bigliardi
      constants.SS_NODE_PRIMARY_IPS: node_pri_ips_data,
2110 f9780ccd Luca Bigliardi
      constants.SS_NODE_SECONDARY_IPS: node_snd_ips_data,
2111 a3316e4a Iustin Pop
      constants.SS_OFFLINE_NODES: off_data,
2112 81a49123 Iustin Pop
      constants.SS_ONLINE_NODES: on_data,
2113 868a98ca Manuel Franceschini
      constants.SS_PRIMARY_IP_FAMILY: str(cluster.primary_ip_family),
2114 81a49123 Iustin Pop
      constants.SS_INSTANCE_LIST: instance_data,
2115 8a113c7a Iustin Pop
      constants.SS_RELEASE_VERSION: constants.RELEASE_VERSION,
2116 4f7a6a10 Iustin Pop
      constants.SS_HYPERVISOR_LIST: hypervisor_list,
2117 5c465a95 Iustin Pop
      constants.SS_MAINTAIN_NODE_HEALTH: str(cluster.maintain_node_health),
2118 0fbae49a Balazs Lecz
      constants.SS_UID_POOL: uid_pool,
2119 6f076453 Guido Trotter
      constants.SS_NODEGROUPS: nodegroups_data,
2120 03d1dba2 Michael Hanselmann
      }
2121 2afc9238 Iustin Pop
    bad_values = [(k, v) for k, v in ssconf_values.items()
2122 2afc9238 Iustin Pop
                  if not isinstance(v, (str, basestring))]
2123 2afc9238 Iustin Pop
    if bad_values:
2124 2afc9238 Iustin Pop
      err = utils.CommaJoin("%s=%s" % (k, v) for k, v in bad_values)
2125 2afc9238 Iustin Pop
      raise errors.ConfigurationError("Some ssconf key(s) have non-string"
2126 2afc9238 Iustin Pop
                                      " values: %s" % err)
2127 2afc9238 Iustin Pop
    return ssconf_values
2128 03d1dba2 Michael Hanselmann
2129 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
2130 d367b66c Manuel Franceschini
  def GetSsconfValues(self):
2131 d367b66c Manuel Franceschini
    """Wrapper using lock around _UnlockedGetSsconf().
2132 d367b66c Manuel Franceschini

2133 d367b66c Manuel Franceschini
    """
2134 d367b66c Manuel Franceschini
    return self._UnlockedGetSsconfValues()
2135 d367b66c Manuel Franceschini
2136 d367b66c Manuel Franceschini
  @locking.ssynchronized(_config_lock, shared=1)
2137 a8083063 Iustin Pop
  def GetVGName(self):
2138 a8083063 Iustin Pop
    """Return the volume group name.
2139 a8083063 Iustin Pop

2140 a8083063 Iustin Pop
    """
2141 a8083063 Iustin Pop
    return self._config_data.cluster.volume_group_name
2142 a8083063 Iustin Pop
2143 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock)
2144 89ff8e15 Manuel Franceschini
  def SetVGName(self, vg_name):
2145 89ff8e15 Manuel Franceschini
    """Set the volume group name.
2146 89ff8e15 Manuel Franceschini

2147 89ff8e15 Manuel Franceschini
    """
2148 2d4011cd Manuel Franceschini
    self._config_data.cluster.volume_group_name = vg_name
2149 b9f72b4e Iustin Pop
    self._config_data.cluster.serial_no += 1
2150 89ff8e15 Manuel Franceschini
    self._WriteConfig()
2151 89ff8e15 Manuel Franceschini
2152 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
2153 9e33896b Luca Bigliardi
  def GetDRBDHelper(self):
2154 9e33896b Luca Bigliardi
    """Return DRBD usermode helper.
2155 9e33896b Luca Bigliardi

2156 9e33896b Luca Bigliardi
    """
2157 9e33896b Luca Bigliardi
    return self._config_data.cluster.drbd_usermode_helper
2158 9e33896b Luca Bigliardi
2159 9e33896b Luca Bigliardi
  @locking.ssynchronized(_config_lock)
2160 9e33896b Luca Bigliardi
  def SetDRBDHelper(self, drbd_helper):
2161 9e33896b Luca Bigliardi
    """Set DRBD usermode helper.
2162 9e33896b Luca Bigliardi

2163 9e33896b Luca Bigliardi
    """
2164 9e33896b Luca Bigliardi
    self._config_data.cluster.drbd_usermode_helper = drbd_helper
2165 9e33896b Luca Bigliardi
    self._config_data.cluster.serial_no += 1
2166 9e33896b Luca Bigliardi
    self._WriteConfig()
2167 9e33896b Luca Bigliardi
2168 9e33896b Luca Bigliardi
  @locking.ssynchronized(_config_lock, shared=1)
2169 a8083063 Iustin Pop
  def GetMACPrefix(self):
2170 a8083063 Iustin Pop
    """Return the mac prefix.
2171 a8083063 Iustin Pop

2172 a8083063 Iustin Pop
    """
2173 a8083063 Iustin Pop
    return self._config_data.cluster.mac_prefix
2174 62779dd0 Iustin Pop
2175 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock, shared=1)
2176 62779dd0 Iustin Pop
  def GetClusterInfo(self):
2177 5bbd3f7f Michael Hanselmann
    """Returns information about the cluster
2178 62779dd0 Iustin Pop

2179 c41eea6e Iustin Pop
    @rtype: L{objects.Cluster}
2180 c41eea6e Iustin Pop
    @return: the cluster object
2181 62779dd0 Iustin Pop

2182 62779dd0 Iustin Pop
    """
2183 62779dd0 Iustin Pop
    return self._config_data.cluster
2184 e00fb268 Iustin Pop
2185 51cb1581 Luca Bigliardi
  @locking.ssynchronized(_config_lock, shared=1)
2186 51cb1581 Luca Bigliardi
  def HasAnyDiskOfType(self, dev_type):
2187 51cb1581 Luca Bigliardi
    """Check if in there is at disk of the given type in the configuration.
2188 51cb1581 Luca Bigliardi

2189 51cb1581 Luca Bigliardi
    """
2190 51cb1581 Luca Bigliardi
    return self._config_data.HasAnyDiskOfType(dev_type)
2191 51cb1581 Luca Bigliardi
2192 f78ede4e Guido Trotter
  @locking.ssynchronized(_config_lock)
2193 a4eae71f Michael Hanselmann
  def Update(self, target, feedback_fn):
2194 e00fb268 Iustin Pop
    """Notify function to be called after updates.
2195 e00fb268 Iustin Pop

2196 e00fb268 Iustin Pop
    This function must be called when an object (as returned by
2197 e00fb268 Iustin Pop
    GetInstanceInfo, GetNodeInfo, GetCluster) has been updated and the
2198 e00fb268 Iustin Pop
    caller wants the modifications saved to the backing store. Note
2199 e00fb268 Iustin Pop
    that all modified objects will be saved, but the target argument
2200 e00fb268 Iustin Pop
    is the one the caller wants to ensure that it's saved.
2201 e00fb268 Iustin Pop

2202 c41eea6e Iustin Pop
    @param target: an instance of either L{objects.Cluster},
2203 c41eea6e Iustin Pop
        L{objects.Node} or L{objects.Instance} which is existing in
2204 c41eea6e Iustin Pop
        the cluster
2205 a4eae71f Michael Hanselmann
    @param feedback_fn: Callable feedback function
2206 c41eea6e Iustin Pop

2207 e00fb268 Iustin Pop
    """
2208 e00fb268 Iustin Pop
    if self._config_data is None:
2209 3ecf6786 Iustin Pop
      raise errors.ProgrammerError("Configuration file not read,"
2210 3ecf6786 Iustin Pop
                                   " cannot save.")
2211 f34901f8 Iustin Pop
    update_serial = False
2212 e00fb268 Iustin Pop
    if isinstance(target, objects.Cluster):
2213 e00fb268 Iustin Pop
      test = target == self._config_data.cluster
2214 e00fb268 Iustin Pop
    elif isinstance(target, objects.Node):
2215 e00fb268 Iustin Pop
      test = target in self._config_data.nodes.values()
2216 f34901f8 Iustin Pop
      update_serial = True
2217 e00fb268 Iustin Pop
    elif isinstance(target, objects.Instance):
2218 e00fb268 Iustin Pop
      test = target in self._config_data.instances.values()
2219 e11a1b77 Adeodato Simo
    elif isinstance(target, objects.NodeGroup):
2220 e11a1b77 Adeodato Simo
      test = target in self._config_data.nodegroups.values()
2221 e00fb268 Iustin Pop
    else:
2222 3ecf6786 Iustin Pop
      raise errors.ProgrammerError("Invalid object type (%s) passed to"
2223 3ecf6786 Iustin Pop
                                   " ConfigWriter.Update" % type(target))
2224 e00fb268 Iustin Pop
    if not test:
2225 3ecf6786 Iustin Pop
      raise errors.ConfigurationError("Configuration updated since object"
2226 3ecf6786 Iustin Pop
                                      " has been read or unknown object")
2227 f34901f8 Iustin Pop
    target.serial_no += 1
2228 d693c864 Iustin Pop
    target.mtime = now = time.time()
2229 f34901f8 Iustin Pop
2230 cff4c037 Iustin Pop
    if update_serial:
2231 f34901f8 Iustin Pop
      # for node updates, we need to increase the cluster serial too
2232 f34901f8 Iustin Pop
      self._config_data.cluster.serial_no += 1
2233 d693c864 Iustin Pop
      self._config_data.cluster.mtime = now
2234 b989e85d Iustin Pop
2235 61cf6b5e Iustin Pop
    if isinstance(target, objects.Instance):
2236 61cf6b5e Iustin Pop
      self._UnlockedReleaseDRBDMinors(target.name)
2237 61cf6b5e Iustin Pop
2238 a4eae71f Michael Hanselmann
    self._WriteConfig(feedback_fn=feedback_fn)
2239 73064714 Guido Trotter
2240 73064714 Guido Trotter
  @locking.ssynchronized(_config_lock)
2241 73064714 Guido Trotter
  def DropECReservations(self, ec_id):
2242 73064714 Guido Trotter
    """Drop per-execution-context reservations
2243 73064714 Guido Trotter

2244 73064714 Guido Trotter
    """
2245 d8aee57e Iustin Pop
    for rm in self._all_rms:
2246 d8aee57e Iustin Pop
      rm.DropECReservations(ec_id)