Statistics
| Branch: | Tag: | Revision:

root / lib / bootstrap.py @ a5c30dc2

History | View | Annotate | Download (29.1 kB)

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

24 a0c9f010 Michael Hanselmann
"""
25 a0c9f010 Michael Hanselmann
26 a0c9f010 Michael Hanselmann
import os
27 a0c9f010 Michael Hanselmann
import os.path
28 a0c9f010 Michael Hanselmann
import re
29 b1b6ea87 Iustin Pop
import logging
30 d693c864 Iustin Pop
import time
31 a0c9f010 Michael Hanselmann
32 a0c9f010 Michael Hanselmann
from ganeti import rpc
33 a0c9f010 Michael Hanselmann
from ganeti import ssh
34 a0c9f010 Michael Hanselmann
from ganeti import utils
35 a0c9f010 Michael Hanselmann
from ganeti import errors
36 a0c9f010 Michael Hanselmann
from ganeti import config
37 a0c9f010 Michael Hanselmann
from ganeti import constants
38 b9eeeb02 Michael Hanselmann
from ganeti import objects
39 a0c9f010 Michael Hanselmann
from ganeti import ssconf
40 a33848a5 Guido Trotter
from ganeti import serializer
41 a5728081 Guido Trotter
from ganeti import hypervisor
42 a721e23a Luca Bigliardi
from ganeti import bdev
43 a744b676 Manuel Franceschini
from ganeti import netutils
44 d367b66c Manuel Franceschini
from ganeti import backend
45 3b6b6129 Michael Hanselmann
from ganeti import luxi
46 ff699aa9 Michael Hanselmann
from ganeti import jstore
47 3b6b6129 Michael Hanselmann
48 a0c9f010 Michael Hanselmann
49 88b92fe3 Guido Trotter
# ec_id for InitConfig's temporary reservation manager
50 88b92fe3 Guido Trotter
_INITCONF_ECID = "initconfig-ecid"
51 88b92fe3 Guido Trotter
52 3b6b6129 Michael Hanselmann
#: After how many seconds daemon must be responsive
53 3b6b6129 Michael Hanselmann
_DAEMON_READY_TIMEOUT = 10.0
54 3b6b6129 Michael Hanselmann
55 e38220e4 Michael Hanselmann
56 531baf8e Iustin Pop
def _InitSSHSetup():
57 a0c9f010 Michael Hanselmann
  """Setup the SSH configuration for the cluster.
58 a0c9f010 Michael Hanselmann

59 a0c9f010 Michael Hanselmann
  This generates a dsa keypair for root, adds the pub key to the
60 a0c9f010 Michael Hanselmann
  permitted hosts and adds the hostkey to its own known hosts.
61 a0c9f010 Michael Hanselmann

62 a0c9f010 Michael Hanselmann
  """
63 a0c9f010 Michael Hanselmann
  priv_key, pub_key, auth_keys = ssh.GetUserFiles(constants.GANETI_RUNAS)
64 a0c9f010 Michael Hanselmann
65 a0c9f010 Michael Hanselmann
  for name in priv_key, pub_key:
66 a0c9f010 Michael Hanselmann
    if os.path.exists(name):
67 a0c9f010 Michael Hanselmann
      utils.CreateBackup(name)
68 a0c9f010 Michael Hanselmann
    utils.RemoveFile(name)
69 a0c9f010 Michael Hanselmann
70 a0c9f010 Michael Hanselmann
  result = utils.RunCmd(["ssh-keygen", "-t", "dsa",
71 a0c9f010 Michael Hanselmann
                         "-f", priv_key,
72 a0c9f010 Michael Hanselmann
                         "-q", "-N", ""])
73 a0c9f010 Michael Hanselmann
  if result.failed:
74 a0c9f010 Michael Hanselmann
    raise errors.OpExecError("Could not generate ssh keypair, error %s" %
75 a0c9f010 Michael Hanselmann
                             result.output)
76 a0c9f010 Michael Hanselmann
77 7a0156dc Luca Bigliardi
  utils.AddAuthorizedKey(auth_keys, utils.ReadFile(pub_key))
78 a0c9f010 Michael Hanselmann
79 a0c9f010 Michael Hanselmann
80 c008906b Michael Hanselmann
def GenerateHmacKey(file_name):
81 c008906b Michael Hanselmann
  """Writes a new HMAC key.
82 c008906b Michael Hanselmann

83 c008906b Michael Hanselmann
  @type file_name: str
84 c008906b Michael Hanselmann
  @param file_name: Path to output file
85 c008906b Michael Hanselmann

86 c008906b Michael Hanselmann
  """
87 43575108 Michael Hanselmann
  utils.WriteFile(file_name, data="%s\n" % utils.GenerateSecret(), mode=0400,
88 43575108 Michael Hanselmann
                  backup=True)
89 43575108 Michael Hanselmann
90 43575108 Michael Hanselmann
91 6b7d5878 Michael Hanselmann
def GenerateClusterCrypto(new_cluster_cert, new_rapi_cert, new_confd_hmac_key,
92 af2ae1c0 Iustin Pop
                          new_cds, rapi_cert_pem=None, cds=None,
93 aeefe835 Iustin Pop
                          nodecert_file=constants.NODED_CERT_FILE,
94 aeefe835 Iustin Pop
                          rapicert_file=constants.RAPI_CERT_FILE,
95 fc0726b9 Michael Hanselmann
                          hmackey_file=constants.CONFD_HMAC_KEY,
96 fc0726b9 Michael Hanselmann
                          cds_file=constants.CLUSTER_DOMAIN_SECRET_FILE):
97 43575108 Michael Hanselmann
  """Updates the cluster certificates, keys and secrets.
98 43575108 Michael Hanselmann

99 43575108 Michael Hanselmann
  @type new_cluster_cert: bool
100 43575108 Michael Hanselmann
  @param new_cluster_cert: Whether to generate a new cluster certificate
101 43575108 Michael Hanselmann
  @type new_rapi_cert: bool
102 43575108 Michael Hanselmann
  @param new_rapi_cert: Whether to generate a new RAPI certificate
103 6b7d5878 Michael Hanselmann
  @type new_confd_hmac_key: bool
104 6b7d5878 Michael Hanselmann
  @param new_confd_hmac_key: Whether to generate a new HMAC key
105 3db3eb2a Michael Hanselmann
  @type new_cds: bool
106 3db3eb2a Michael Hanselmann
  @param new_cds: Whether to generate a new cluster domain secret
107 43575108 Michael Hanselmann
  @type rapi_cert_pem: string
108 43575108 Michael Hanselmann
  @param rapi_cert_pem: New RAPI certificate in PEM format
109 3db3eb2a Michael Hanselmann
  @type cds: string
110 3db3eb2a Michael Hanselmann
  @param cds: New cluster domain secret
111 aeefe835 Iustin Pop
  @type nodecert_file: string
112 aeefe835 Iustin Pop
  @param nodecert_file: optional override of the node cert file path
113 aeefe835 Iustin Pop
  @type rapicert_file: string
114 aeefe835 Iustin Pop
  @param rapicert_file: optional override of the rapi cert file path
115 aeefe835 Iustin Pop
  @type hmackey_file: string
116 aeefe835 Iustin Pop
  @param hmackey_file: optional override of the hmac key file path
117 43575108 Michael Hanselmann

118 43575108 Michael Hanselmann
  """
119 168c1de2 Michael Hanselmann
  # noded SSL certificate
120 aeefe835 Iustin Pop
  cluster_cert_exists = os.path.exists(nodecert_file)
121 43575108 Michael Hanselmann
  if new_cluster_cert or not cluster_cert_exists:
122 43575108 Michael Hanselmann
    if cluster_cert_exists:
123 aeefe835 Iustin Pop
      utils.CreateBackup(nodecert_file)
124 43575108 Michael Hanselmann
125 aeefe835 Iustin Pop
    logging.debug("Generating new cluster certificate at %s", nodecert_file)
126 af2ae1c0 Iustin Pop
    utils.GenerateSelfSignedSslCert(nodecert_file)
127 43575108 Michael Hanselmann
128 6b7d5878 Michael Hanselmann
  # confd HMAC key
129 aeefe835 Iustin Pop
  if new_confd_hmac_key or not os.path.exists(hmackey_file):
130 aeefe835 Iustin Pop
    logging.debug("Writing new confd HMAC key to %s", hmackey_file)
131 aeefe835 Iustin Pop
    GenerateHmacKey(hmackey_file)
132 43575108 Michael Hanselmann
133 43575108 Michael Hanselmann
  # RAPI
134 aeefe835 Iustin Pop
  rapi_cert_exists = os.path.exists(rapicert_file)
135 43575108 Michael Hanselmann
136 43575108 Michael Hanselmann
  if rapi_cert_pem:
137 43575108 Michael Hanselmann
    # Assume rapi_pem contains a valid PEM-formatted certificate and key
138 aeefe835 Iustin Pop
    logging.debug("Writing RAPI certificate at %s", rapicert_file)
139 aeefe835 Iustin Pop
    utils.WriteFile(rapicert_file, data=rapi_cert_pem, backup=True)
140 43575108 Michael Hanselmann
141 43575108 Michael Hanselmann
  elif new_rapi_cert or not rapi_cert_exists:
142 43575108 Michael Hanselmann
    if rapi_cert_exists:
143 aeefe835 Iustin Pop
      utils.CreateBackup(rapicert_file)
144 43575108 Michael Hanselmann
145 aeefe835 Iustin Pop
    logging.debug("Generating new RAPI certificate at %s", rapicert_file)
146 af2ae1c0 Iustin Pop
    utils.GenerateSelfSignedSslCert(rapicert_file)
147 c008906b Michael Hanselmann
148 3db3eb2a Michael Hanselmann
  # Cluster domain secret
149 3db3eb2a Michael Hanselmann
  if cds:
150 fc0726b9 Michael Hanselmann
    logging.debug("Writing cluster domain secret to %s", cds_file)
151 fc0726b9 Michael Hanselmann
    utils.WriteFile(cds_file, data=cds, backup=True)
152 fc0726b9 Michael Hanselmann
153 fc0726b9 Michael Hanselmann
  elif new_cds or not os.path.exists(cds_file):
154 fc0726b9 Michael Hanselmann
    logging.debug("Generating new cluster domain secret at %s", cds_file)
155 fc0726b9 Michael Hanselmann
    GenerateHmacKey(cds_file)
156 3db3eb2a Michael Hanselmann
157 c008906b Michael Hanselmann
158 8f215968 Michael Hanselmann
def _InitGanetiServerSetup(master_name):
159 40a97d80 Michael Hanselmann
  """Setup the necessary configuration for the initial node daemon.
160 40a97d80 Michael Hanselmann

161 40a97d80 Michael Hanselmann
  This creates the nodepass file containing the shared password for
162 600535f0 Manuel Franceschini
  the cluster, generates the SSL certificate and starts the node daemon.
163 600535f0 Manuel Franceschini

164 600535f0 Manuel Franceschini
  @type master_name: str
165 600535f0 Manuel Franceschini
  @param master_name: Name of the master node
166 40a97d80 Michael Hanselmann

167 40a97d80 Michael Hanselmann
  """
168 43575108 Michael Hanselmann
  # Generate cluster secrets
169 3db3eb2a Michael Hanselmann
  GenerateClusterCrypto(True, False, False, False)
170 4a34c5cf Guido Trotter
171 f154a7a3 Michael Hanselmann
  result = utils.RunCmd([constants.DAEMON_UTIL, "start", constants.NODED])
172 a0c9f010 Michael Hanselmann
  if result.failed:
173 a0c9f010 Michael Hanselmann
    raise errors.OpExecError("Could not start the node daemon, command %s"
174 a0c9f010 Michael Hanselmann
                             " had exitcode %s and error %s" %
175 a0c9f010 Michael Hanselmann
                             (result.cmd, result.exit_code, result.output))
176 a0c9f010 Michael Hanselmann
177 5627f375 Michael Hanselmann
  _WaitForNodeDaemon(master_name)
178 5627f375 Michael Hanselmann
179 5627f375 Michael Hanselmann
180 5627f375 Michael Hanselmann
def _WaitForNodeDaemon(node_name):
181 5627f375 Michael Hanselmann
  """Wait for node daemon to become responsive.
182 5627f375 Michael Hanselmann

183 5627f375 Michael Hanselmann
  """
184 d3833ebd Michael Hanselmann
  def _CheckNodeDaemon():
185 5627f375 Michael Hanselmann
    result = rpc.RpcRunner.call_version([node_name])[node_name]
186 d3833ebd Michael Hanselmann
    if result.fail_msg:
187 d3833ebd Michael Hanselmann
      raise utils.RetryAgain()
188 8f215968 Michael Hanselmann
189 d3833ebd Michael Hanselmann
  try:
190 3b6b6129 Michael Hanselmann
    utils.Retry(_CheckNodeDaemon, 1.0, _DAEMON_READY_TIMEOUT)
191 d3833ebd Michael Hanselmann
  except utils.RetryTimeout:
192 5627f375 Michael Hanselmann
    raise errors.OpExecError("Node daemon on %s didn't answer queries within"
193 3b6b6129 Michael Hanselmann
                             " %s seconds" % (node_name, _DAEMON_READY_TIMEOUT))
194 3b6b6129 Michael Hanselmann
195 3b6b6129 Michael Hanselmann
196 3b6b6129 Michael Hanselmann
def _WaitForMasterDaemon():
197 3b6b6129 Michael Hanselmann
  """Wait for master daemon to become responsive.
198 3b6b6129 Michael Hanselmann

199 3b6b6129 Michael Hanselmann
  """
200 3b6b6129 Michael Hanselmann
  def _CheckMasterDaemon():
201 3b6b6129 Michael Hanselmann
    try:
202 3b6b6129 Michael Hanselmann
      cl = luxi.Client()
203 3b6b6129 Michael Hanselmann
      (cluster_name, ) = cl.QueryConfigValues(["cluster_name"])
204 3b6b6129 Michael Hanselmann
    except Exception:
205 3b6b6129 Michael Hanselmann
      raise utils.RetryAgain()
206 3b6b6129 Michael Hanselmann
207 3b6b6129 Michael Hanselmann
    logging.debug("Received cluster name %s from master", cluster_name)
208 3b6b6129 Michael Hanselmann
209 3b6b6129 Michael Hanselmann
  try:
210 3b6b6129 Michael Hanselmann
    utils.Retry(_CheckMasterDaemon, 1.0, _DAEMON_READY_TIMEOUT)
211 3b6b6129 Michael Hanselmann
  except utils.RetryTimeout:
212 3b6b6129 Michael Hanselmann
    raise errors.OpExecError("Master daemon didn't answer queries within"
213 3b6b6129 Michael Hanselmann
                             " %s seconds" % _DAEMON_READY_TIMEOUT)
214 5627f375 Michael Hanselmann
215 a0c9f010 Michael Hanselmann
216 0e3baaf3 Iustin Pop
def _InitFileStorage(file_storage_dir):
217 0e3baaf3 Iustin Pop
  """Initialize if needed the file storage.
218 0e3baaf3 Iustin Pop

219 0e3baaf3 Iustin Pop
  @param file_storage_dir: the user-supplied value
220 0e3baaf3 Iustin Pop
  @return: either empty string (if file storage was disabled at build
221 0e3baaf3 Iustin Pop
      time) or the normalized path to the storage directory
222 0e3baaf3 Iustin Pop

223 0e3baaf3 Iustin Pop
  """
224 0e3baaf3 Iustin Pop
  if not constants.ENABLE_FILE_STORAGE:
225 0e3baaf3 Iustin Pop
    return ""
226 0e3baaf3 Iustin Pop
227 0e3baaf3 Iustin Pop
  file_storage_dir = os.path.normpath(file_storage_dir)
228 0e3baaf3 Iustin Pop
229 0e3baaf3 Iustin Pop
  if not os.path.isabs(file_storage_dir):
230 0e3baaf3 Iustin Pop
    raise errors.OpPrereqError("The file storage directory you passed is"
231 0e3baaf3 Iustin Pop
                               " not an absolute path.", errors.ECODE_INVAL)
232 0e3baaf3 Iustin Pop
233 0e3baaf3 Iustin Pop
  if not os.path.exists(file_storage_dir):
234 0e3baaf3 Iustin Pop
    try:
235 0e3baaf3 Iustin Pop
      os.makedirs(file_storage_dir, 0750)
236 0e3baaf3 Iustin Pop
    except OSError, err:
237 0e3baaf3 Iustin Pop
      raise errors.OpPrereqError("Cannot create file storage directory"
238 0e3baaf3 Iustin Pop
                                 " '%s': %s" % (file_storage_dir, err),
239 0e3baaf3 Iustin Pop
                                 errors.ECODE_ENVIRON)
240 0e3baaf3 Iustin Pop
241 0e3baaf3 Iustin Pop
  if not os.path.isdir(file_storage_dir):
242 0e3baaf3 Iustin Pop
    raise errors.OpPrereqError("The file storage directory '%s' is not"
243 0e3baaf3 Iustin Pop
                               " a directory." % file_storage_dir,
244 0e3baaf3 Iustin Pop
                               errors.ECODE_ENVIRON)
245 0e3baaf3 Iustin Pop
  return file_storage_dir
246 0e3baaf3 Iustin Pop
247 0e3baaf3 Iustin Pop
248 7925d409 Apollon Oikonomopoulos
def _InitSharedFileStorage(shared_file_storage_dir):
249 7925d409 Apollon Oikonomopoulos
  """Initialize if needed the shared file storage.
250 7925d409 Apollon Oikonomopoulos

251 7925d409 Apollon Oikonomopoulos
  @param shared_file_storage_dir: the user-supplied value
252 7925d409 Apollon Oikonomopoulos
  @return: either empty string (if file storage was disabled at build
253 7925d409 Apollon Oikonomopoulos
      time) or the normalized path to the storage directory
254 7925d409 Apollon Oikonomopoulos

255 7925d409 Apollon Oikonomopoulos
  """
256 7925d409 Apollon Oikonomopoulos
  if not constants.ENABLE_SHARED_FILE_STORAGE:
257 7925d409 Apollon Oikonomopoulos
    return ""
258 7925d409 Apollon Oikonomopoulos
259 7925d409 Apollon Oikonomopoulos
  shared_file_storage_dir = os.path.normpath(shared_file_storage_dir)
260 7925d409 Apollon Oikonomopoulos
261 7925d409 Apollon Oikonomopoulos
  if not os.path.isabs(shared_file_storage_dir):
262 7925d409 Apollon Oikonomopoulos
    raise errors.OpPrereqError("The shared file storage directory you"
263 7925d409 Apollon Oikonomopoulos
                               " passed is not an absolute path.",
264 7925d409 Apollon Oikonomopoulos
                               errors.ECODE_INVAL)
265 7925d409 Apollon Oikonomopoulos
266 7925d409 Apollon Oikonomopoulos
  if not os.path.exists(shared_file_storage_dir):
267 7925d409 Apollon Oikonomopoulos
    try:
268 7925d409 Apollon Oikonomopoulos
      os.makedirs(shared_file_storage_dir, 0750)
269 7925d409 Apollon Oikonomopoulos
    except OSError, err:
270 7925d409 Apollon Oikonomopoulos
      raise errors.OpPrereqError("Cannot create file storage directory"
271 7925d409 Apollon Oikonomopoulos
                                 " '%s': %s" % (shared_file_storage_dir, err),
272 7925d409 Apollon Oikonomopoulos
                                 errors.ECODE_ENVIRON)
273 7925d409 Apollon Oikonomopoulos
274 7925d409 Apollon Oikonomopoulos
  if not os.path.isdir(shared_file_storage_dir):
275 7925d409 Apollon Oikonomopoulos
    raise errors.OpPrereqError("The file storage directory '%s' is not"
276 7925d409 Apollon Oikonomopoulos
                               " a directory." % shared_file_storage_dir,
277 7925d409 Apollon Oikonomopoulos
                               errors.ECODE_ENVIRON)
278 7925d409 Apollon Oikonomopoulos
  return shared_file_storage_dir
279 7925d409 Apollon Oikonomopoulos
280 7925d409 Apollon Oikonomopoulos
281 952d7515 Michael Hanselmann
def InitCluster(cluster_name, mac_prefix, # pylint: disable-msg=R0913
282 7925d409 Apollon Oikonomopoulos
                master_netdev, file_storage_dir, shared_file_storage_dir,
283 7925d409 Apollon Oikonomopoulos
                candidate_pool_size, secondary_ip=None, vg_name=None,
284 7925d409 Apollon Oikonomopoulos
                beparams=None, nicparams=None, ndparams=None, hvparams=None,
285 6204ee71 René Nussbaumer
                enabled_hypervisors=None, modify_etc_hosts=True,
286 6204ee71 René Nussbaumer
                modify_ssh_setup=True, maintain_node_health=False,
287 6204ee71 René Nussbaumer
                drbd_helper=None, uid_pool=None, default_iallocator=None,
288 3d914585 René Nussbaumer
                primary_ip_version=None, prealloc_wipe_disks=False):
289 a0c9f010 Michael Hanselmann
  """Initialise the cluster.
290 a0c9f010 Michael Hanselmann

291 ce735215 Guido Trotter
  @type candidate_pool_size: int
292 ce735215 Guido Trotter
  @param candidate_pool_size: master candidate pool size
293 ce735215 Guido Trotter

294 a0c9f010 Michael Hanselmann
  """
295 ce735215 Guido Trotter
  # TODO: complete the docstring
296 a0c9f010 Michael Hanselmann
  if config.ConfigWriter.IsCluster():
297 debac808 Iustin Pop
    raise errors.OpPrereqError("Cluster is already initialised",
298 debac808 Iustin Pop
                               errors.ECODE_STATE)
299 a0c9f010 Michael Hanselmann
300 b119bccb Guido Trotter
  if not enabled_hypervisors:
301 b119bccb Guido Trotter
    raise errors.OpPrereqError("Enabled hypervisors list must contain at"
302 debac808 Iustin Pop
                               " least one member", errors.ECODE_INVAL)
303 b119bccb Guido Trotter
  invalid_hvs = set(enabled_hypervisors) - constants.HYPER_TYPES
304 b119bccb Guido Trotter
  if invalid_hvs:
305 b119bccb Guido Trotter
    raise errors.OpPrereqError("Enabled hypervisors contains invalid"
306 debac808 Iustin Pop
                               " entries: %s" % invalid_hvs,
307 debac808 Iustin Pop
                               errors.ECODE_INVAL)
308 b119bccb Guido Trotter
309 a0c9f010 Michael Hanselmann
310 2f20d07b Manuel Franceschini
  ipcls = None
311 2f20d07b Manuel Franceschini
  if primary_ip_version == constants.IP4_VERSION:
312 2f20d07b Manuel Franceschini
    ipcls = netutils.IP4Address
313 2f20d07b Manuel Franceschini
  elif primary_ip_version == constants.IP6_VERSION:
314 2f20d07b Manuel Franceschini
    ipcls = netutils.IP6Address
315 2f20d07b Manuel Franceschini
  else:
316 2f20d07b Manuel Franceschini
    raise errors.OpPrereqError("Invalid primary ip version: %d." %
317 2f20d07b Manuel Franceschini
                               primary_ip_version)
318 2f20d07b Manuel Franceschini
319 2f20d07b Manuel Franceschini
  hostname = netutils.GetHostname(family=ipcls.family)
320 2f20d07b Manuel Franceschini
  if not ipcls.IsValid(hostname.ip):
321 2f20d07b Manuel Franceschini
    raise errors.OpPrereqError("This host's IP (%s) is not a valid IPv%d"
322 2f20d07b Manuel Franceschini
                               " address." % (hostname.ip, primary_ip_version))
323 2f20d07b Manuel Franceschini
324 2f20d07b Manuel Franceschini
  if ipcls.IsLoopback(hostname.ip):
325 8b312c1d Manuel Franceschini
    raise errors.OpPrereqError("This host's IP (%s) resolves to a loopback"
326 8b312c1d Manuel Franceschini
                               " address. Please fix DNS or %s." %
327 debac808 Iustin Pop
                               (hostname.ip, constants.ETC_HOSTS),
328 debac808 Iustin Pop
                               errors.ECODE_ENVIRON)
329 a0c9f010 Michael Hanselmann
330 2f20d07b Manuel Franceschini
  if not ipcls.Own(hostname.ip):
331 a0c9f010 Michael Hanselmann
    raise errors.OpPrereqError("Inconsistency: this host's name resolves"
332 a0c9f010 Michael Hanselmann
                               " to %s,\nbut this ip address does not"
333 7c4c22f5 Manuel Franceschini
                               " belong to this host" %
334 debac808 Iustin Pop
                               hostname.ip, errors.ECODE_ENVIRON)
335 a0c9f010 Michael Hanselmann
336 2f20d07b Manuel Franceschini
  clustername = netutils.GetHostname(name=cluster_name, family=ipcls.family)
337 a0c9f010 Michael Hanselmann
338 2f20d07b Manuel Franceschini
  if netutils.TcpPing(clustername.ip, constants.DEFAULT_NODED_PORT, timeout=5):
339 7c4c22f5 Manuel Franceschini
    raise errors.OpPrereqError("Cluster IP already active",
340 debac808 Iustin Pop
                               errors.ECODE_NOTUNIQUE)
341 a0c9f010 Michael Hanselmann
342 2f20d07b Manuel Franceschini
  if not secondary_ip:
343 2f20d07b Manuel Franceschini
    if primary_ip_version == constants.IP6_VERSION:
344 2f20d07b Manuel Franceschini
      raise errors.OpPrereqError("When using a IPv6 primary address, a valid"
345 7c4c22f5 Manuel Franceschini
                                 " IPv4 address must be given as secondary",
346 7c4c22f5 Manuel Franceschini
                                 errors.ECODE_INVAL)
347 b9eeeb02 Michael Hanselmann
    secondary_ip = hostname.ip
348 a0c9f010 Michael Hanselmann
349 2f20d07b Manuel Franceschini
  if not netutils.IP4Address.IsValid(secondary_ip):
350 2f20d07b Manuel Franceschini
    raise errors.OpPrereqError("Secondary IP address (%s) has to be a valid"
351 2f20d07b Manuel Franceschini
                               " IPv4 address." % secondary_ip,
352 2f20d07b Manuel Franceschini
                               errors.ECODE_INVAL)
353 2f20d07b Manuel Franceschini
354 2f20d07b Manuel Franceschini
  if not netutils.IP4Address.Own(secondary_ip):
355 2f20d07b Manuel Franceschini
    raise errors.OpPrereqError("You gave %s as secondary IP,"
356 2f20d07b Manuel Franceschini
                               " but it does not belong to this host." %
357 2f20d07b Manuel Franceschini
                               secondary_ip, errors.ECODE_ENVIRON)
358 2f20d07b Manuel Franceschini
359 a0c9f010 Michael Hanselmann
  if vg_name is not None:
360 a0c9f010 Michael Hanselmann
    # Check if volume group is valid
361 a0c9f010 Michael Hanselmann
    vgstatus = utils.CheckVolumeGroupSize(utils.ListVolumeGroups(), vg_name,
362 a0c9f010 Michael Hanselmann
                                          constants.MIN_VG_SIZE)
363 a0c9f010 Michael Hanselmann
    if vgstatus:
364 a0c9f010 Michael Hanselmann
      raise errors.OpPrereqError("Error: %s\nspecify --no-lvm-storage if"
365 debac808 Iustin Pop
                                 " you are not using lvm" % vgstatus,
366 debac808 Iustin Pop
                                 errors.ECODE_INVAL)
367 a0c9f010 Michael Hanselmann
368 a721e23a Luca Bigliardi
  if drbd_helper is not None:
369 a721e23a Luca Bigliardi
    try:
370 a721e23a Luca Bigliardi
      curr_helper = bdev.BaseDRBD.GetUsermodeHelper()
371 a721e23a Luca Bigliardi
    except errors.BlockDeviceError, err:
372 a721e23a Luca Bigliardi
      raise errors.OpPrereqError("Error while checking drbd helper"
373 a721e23a Luca Bigliardi
                                 " (specify --no-drbd-storage if you are not"
374 a721e23a Luca Bigliardi
                                 " using drbd): %s" % str(err),
375 a721e23a Luca Bigliardi
                                 errors.ECODE_ENVIRON)
376 a721e23a Luca Bigliardi
    if drbd_helper != curr_helper:
377 a721e23a Luca Bigliardi
      raise errors.OpPrereqError("Error: requiring %s as drbd helper but %s"
378 a721e23a Luca Bigliardi
                                 " is the current helper" % (drbd_helper,
379 a721e23a Luca Bigliardi
                                                             curr_helper),
380 a721e23a Luca Bigliardi
                                 errors.ECODE_INVAL)
381 a721e23a Luca Bigliardi
382 0e3baaf3 Iustin Pop
  file_storage_dir = _InitFileStorage(file_storage_dir)
383 7925d409 Apollon Oikonomopoulos
  shared_file_storage_dir = _InitSharedFileStorage(shared_file_storage_dir)
384 a0c9f010 Michael Hanselmann
385 a0c9f010 Michael Hanselmann
  if not re.match("^[0-9a-z]{2}:[0-9a-z]{2}:[0-9a-z]{2}$", mac_prefix):
386 debac808 Iustin Pop
    raise errors.OpPrereqError("Invalid mac prefix given '%s'" % mac_prefix,
387 debac808 Iustin Pop
                               errors.ECODE_INVAL)
388 a0c9f010 Michael Hanselmann
389 a0c9f010 Michael Hanselmann
  result = utils.RunCmd(["ip", "link", "show", "dev", master_netdev])
390 a0c9f010 Michael Hanselmann
  if result.failed:
391 a0c9f010 Michael Hanselmann
    raise errors.OpPrereqError("Invalid master netdev given (%s): '%s'" %
392 a0c9f010 Michael Hanselmann
                               (master_netdev,
393 debac808 Iustin Pop
                                result.output.strip()), errors.ECODE_INVAL)
394 a0c9f010 Michael Hanselmann
395 9dae41ad Guido Trotter
  dirs = [(constants.RUN_GANETI_DIR, constants.RUN_DIRS_MODE)]
396 9dae41ad Guido Trotter
  utils.EnsureDirs(dirs)
397 9dae41ad Guido Trotter
398 a5728081 Guido Trotter
  utils.ForceDictType(beparams, constants.BES_PARAMETER_TYPES)
399 b6a30b0d Guido Trotter
  utils.ForceDictType(nicparams, constants.NICS_PARAMETER_TYPES)
400 b6a30b0d Guido Trotter
  objects.NIC.CheckParameterSyntax(nicparams)
401 b6a30b0d Guido Trotter
402 6204ee71 René Nussbaumer
  if ndparams is not None:
403 6204ee71 René Nussbaumer
    utils.ForceDictType(ndparams, constants.NDS_PARAMETER_TYPES)
404 6204ee71 René Nussbaumer
  else:
405 6204ee71 René Nussbaumer
    ndparams = dict(constants.NDC_DEFAULTS)
406 6204ee71 René Nussbaumer
407 a5728081 Guido Trotter
  # hvparams is a mapping of hypervisor->hvparams dict
408 a5728081 Guido Trotter
  for hv_name, hv_params in hvparams.iteritems():
409 a5728081 Guido Trotter
    utils.ForceDictType(hv_params, constants.HVS_PARAMETER_TYPES)
410 a5728081 Guido Trotter
    hv_class = hypervisor.GetHypervisor(hv_name)
411 a5728081 Guido Trotter
    hv_class.CheckParameterSyntax(hv_params)
412 d4b72030 Guido Trotter
413 a0c9f010 Michael Hanselmann
  # set up ssh config and /etc/hosts
414 13998ef2 Michael Hanselmann
  sshline = utils.ReadFile(constants.SSH_HOST_RSA_PUB)
415 a0c9f010 Michael Hanselmann
  sshkey = sshline.split(" ")[1]
416 a0c9f010 Michael Hanselmann
417 b86a6bcd Guido Trotter
  if modify_etc_hosts:
418 ea8ac9c9 René Nussbaumer
    utils.AddHostToEtcHosts(hostname.name, hostname.ip)
419 b86a6bcd Guido Trotter
420 b989b9d9 Ken Wehr
  if modify_ssh_setup:
421 b989b9d9 Ken Wehr
    _InitSSHSetup()
422 a0c9f010 Michael Hanselmann
423 bf4af505 Apollon Oikonomopoulos
  if default_iallocator is not None:
424 bf4af505 Apollon Oikonomopoulos
    alloc_script = utils.FindFile(default_iallocator,
425 bf4af505 Apollon Oikonomopoulos
                                  constants.IALLOCATOR_SEARCH_PATH,
426 bf4af505 Apollon Oikonomopoulos
                                  os.path.isfile)
427 bf4af505 Apollon Oikonomopoulos
    if alloc_script is None:
428 bf4af505 Apollon Oikonomopoulos
      raise errors.OpPrereqError("Invalid default iallocator script '%s'"
429 bf4af505 Apollon Oikonomopoulos
                                 " specified" % default_iallocator,
430 bf4af505 Apollon Oikonomopoulos
                                 errors.ECODE_INVAL)
431 d1e9c98d Iustin Pop
  elif constants.HTOOLS:
432 d1e9c98d Iustin Pop
    # htools was enabled at build-time, we default to it
433 d1e9c98d Iustin Pop
    if utils.FindFile(constants.IALLOC_HAIL,
434 d1e9c98d Iustin Pop
                      constants.IALLOCATOR_SEARCH_PATH,
435 d1e9c98d Iustin Pop
                      os.path.isfile):
436 d1e9c98d Iustin Pop
      default_iallocator = constants.IALLOC_HAIL
437 bf4af505 Apollon Oikonomopoulos
438 430b923c Iustin Pop
  now = time.time()
439 430b923c Iustin Pop
440 a0c9f010 Michael Hanselmann
  # init of cluster config file
441 b9eeeb02 Michael Hanselmann
  cluster_config = objects.Cluster(
442 b9eeeb02 Michael Hanselmann
    serial_no=1,
443 b9eeeb02 Michael Hanselmann
    rsahostkeypub=sshkey,
444 b9eeeb02 Michael Hanselmann
    highest_used_port=(constants.FIRST_DRBD_PORT - 1),
445 b9eeeb02 Michael Hanselmann
    mac_prefix=mac_prefix,
446 b9eeeb02 Michael Hanselmann
    volume_group_name=vg_name,
447 b9eeeb02 Michael Hanselmann
    tcpudp_port_pool=set(),
448 f6bd6e98 Michael Hanselmann
    master_node=hostname.name,
449 f6bd6e98 Michael Hanselmann
    master_ip=clustername.ip,
450 f6bd6e98 Michael Hanselmann
    master_netdev=master_netdev,
451 f6bd6e98 Michael Hanselmann
    cluster_name=clustername.name,
452 f6bd6e98 Michael Hanselmann
    file_storage_dir=file_storage_dir,
453 4b97f902 Apollon Oikonomopoulos
    shared_file_storage_dir=shared_file_storage_dir,
454 ea3a925f Alexander Schreiber
    enabled_hypervisors=enabled_hypervisors,
455 4ef7f423 Guido Trotter
    beparams={constants.PP_DEFAULT: beparams},
456 b6a30b0d Guido Trotter
    nicparams={constants.PP_DEFAULT: nicparams},
457 6204ee71 René Nussbaumer
    ndparams=ndparams,
458 ea3a925f Alexander Schreiber
    hvparams=hvparams,
459 ce735215 Guido Trotter
    candidate_pool_size=candidate_pool_size,
460 022c3a0b Guido Trotter
    modify_etc_hosts=modify_etc_hosts,
461 b989b9d9 Ken Wehr
    modify_ssh_setup=modify_ssh_setup,
462 39b0f0c2 Balazs Lecz
    uid_pool=uid_pool,
463 430b923c Iustin Pop
    ctime=now,
464 430b923c Iustin Pop
    mtime=now,
465 3953242f Iustin Pop
    maintain_node_health=maintain_node_health,
466 a721e23a Luca Bigliardi
    drbd_usermode_helper=drbd_helper,
467 bf4af505 Apollon Oikonomopoulos
    default_iallocator=default_iallocator,
468 2f20d07b Manuel Franceschini
    primary_ip_family=ipcls.family,
469 3d914585 René Nussbaumer
    prealloc_wipe_disks=prealloc_wipe_disks,
470 b9eeeb02 Michael Hanselmann
    )
471 b9eeeb02 Michael Hanselmann
  master_node_config = objects.Node(name=hostname.name,
472 b9eeeb02 Michael Hanselmann
                                    primary_ip=hostname.ip,
473 b9222f32 Guido Trotter
                                    secondary_ip=secondary_ip,
474 c044f32c Guido Trotter
                                    serial_no=1,
475 c044f32c Guido Trotter
                                    master_candidate=True,
476 af64c0ea Iustin Pop
                                    offline=False, drained=False,
477 435e4bd6 Michael Hanselmann
                                    ctime=now, mtime=now,
478 c044f32c Guido Trotter
                                    )
479 9e1333b9 Guido Trotter
  InitConfig(constants.CONFIG_VERSION, cluster_config, master_node_config)
480 d367b66c Manuel Franceschini
  cfg = config.ConfigWriter(offline=True)
481 9e1333b9 Guido Trotter
  ssh.WriteKnownHostsFile(cfg, constants.SSH_KNOWN_HOSTS_FILE)
482 a4eae71f Michael Hanselmann
  cfg.Update(cfg.GetClusterInfo(), logging.error)
483 d367b66c Manuel Franceschini
  backend.WriteSsconfFiles(cfg.GetSsconfValues())
484 d367b66c Manuel Franceschini
485 d367b66c Manuel Franceschini
  # set up the inter-node password and certificate
486 d367b66c Manuel Franceschini
  _InitGanetiServerSetup(hostname.name)
487 827f753e Guido Trotter
488 952d7515 Michael Hanselmann
  logging.debug("Starting daemons")
489 952d7515 Michael Hanselmann
  result = utils.RunCmd([constants.DAEMON_UTIL, "start-all"])
490 952d7515 Michael Hanselmann
  if result.failed:
491 952d7515 Michael Hanselmann
    raise errors.OpExecError("Could not start daemons, command %s"
492 952d7515 Michael Hanselmann
                             " had exitcode %s and error %s" %
493 952d7515 Michael Hanselmann
                             (result.cmd, result.exit_code, result.output))
494 b3f1cf6f Iustin Pop
495 3b6b6129 Michael Hanselmann
  _WaitForMasterDaemon()
496 b3f1cf6f Iustin Pop
497 b1b6ea87 Iustin Pop
498 02f99608 Oleksiy Mishchenko
def InitConfig(version, cluster_config, master_node_config,
499 02f99608 Oleksiy Mishchenko
               cfg_file=constants.CLUSTER_CONF_FILE):
500 7b3a8fb5 Iustin Pop
  """Create the initial cluster configuration.
501 7b3a8fb5 Iustin Pop

502 7b3a8fb5 Iustin Pop
  It will contain the current node, which will also be the master
503 7b3a8fb5 Iustin Pop
  node, and no instances.
504 7b3a8fb5 Iustin Pop

505 7b3a8fb5 Iustin Pop
  @type version: int
506 c41eea6e Iustin Pop
  @param version: configuration version
507 c41eea6e Iustin Pop
  @type cluster_config: L{objects.Cluster}
508 c41eea6e Iustin Pop
  @param cluster_config: cluster configuration
509 c41eea6e Iustin Pop
  @type master_node_config: L{objects.Node}
510 c41eea6e Iustin Pop
  @param master_node_config: master node configuration
511 c41eea6e Iustin Pop
  @type cfg_file: string
512 c41eea6e Iustin Pop
  @param cfg_file: configuration file path
513 c41eea6e Iustin Pop

514 7b3a8fb5 Iustin Pop
  """
515 88b92fe3 Guido Trotter
  uuid_generator = config.TemporaryReservationManager()
516 88b92fe3 Guido Trotter
  cluster_config.uuid = uuid_generator.Generate([], utils.NewUUID,
517 88b92fe3 Guido Trotter
                                                _INITCONF_ECID)
518 88b92fe3 Guido Trotter
  master_node_config.uuid = uuid_generator.Generate([], utils.NewUUID,
519 88b92fe3 Guido Trotter
                                                    _INITCONF_ECID)
520 7b3a8fb5 Iustin Pop
  nodes = {
521 7b3a8fb5 Iustin Pop
    master_node_config.name: master_node_config,
522 7b3a8fb5 Iustin Pop
    }
523 88b92fe3 Guido Trotter
  default_nodegroup = objects.NodeGroup(
524 88b92fe3 Guido Trotter
    uuid=uuid_generator.Generate([], utils.NewUUID, _INITCONF_ECID),
525 75cf411a Adeodato Simo
    name=constants.INITIAL_NODE_GROUP_NAME,
526 88b92fe3 Guido Trotter
    members=[master_node_config.name],
527 88b92fe3 Guido Trotter
    )
528 88b92fe3 Guido Trotter
  nodegroups = {
529 88b92fe3 Guido Trotter
    default_nodegroup.uuid: default_nodegroup,
530 88b92fe3 Guido Trotter
    }
531 d693c864 Iustin Pop
  now = time.time()
532 7b3a8fb5 Iustin Pop
  config_data = objects.ConfigData(version=version,
533 7b3a8fb5 Iustin Pop
                                   cluster=cluster_config,
534 88b92fe3 Guido Trotter
                                   nodegroups=nodegroups,
535 7b3a8fb5 Iustin Pop
                                   nodes=nodes,
536 7b3a8fb5 Iustin Pop
                                   instances={},
537 d693c864 Iustin Pop
                                   serial_no=1,
538 d693c864 Iustin Pop
                                   ctime=now, mtime=now)
539 a33848a5 Guido Trotter
  utils.WriteFile(cfg_file,
540 a33848a5 Guido Trotter
                  data=serializer.Dump(config_data.ToDict()),
541 a33848a5 Guido Trotter
                  mode=0600)
542 02f99608 Oleksiy Mishchenko
543 02f99608 Oleksiy Mishchenko
544 140aa4a8 Iustin Pop
def FinalizeClusterDestroy(master):
545 140aa4a8 Iustin Pop
  """Execute the last steps of cluster destroy
546 140aa4a8 Iustin Pop

547 140aa4a8 Iustin Pop
  This function shuts down all the daemons, completing the destroy
548 140aa4a8 Iustin Pop
  begun in cmdlib.LUDestroyOpcode.
549 140aa4a8 Iustin Pop

550 140aa4a8 Iustin Pop
  """
551 b989b9d9 Ken Wehr
  cfg = config.ConfigWriter()
552 b989b9d9 Ken Wehr
  modify_ssh_setup = cfg.GetClusterInfo().modify_ssh_setup
553 781de953 Iustin Pop
  result = rpc.RpcRunner.call_node_stop_master(master, True)
554 3cebe102 Michael Hanselmann
  msg = result.fail_msg
555 6c00d19a Iustin Pop
  if msg:
556 099c52ad Iustin Pop
    logging.warning("Could not disable the master role: %s", msg)
557 b989b9d9 Ken Wehr
  result = rpc.RpcRunner.call_node_leave_cluster(master, modify_ssh_setup)
558 3cebe102 Michael Hanselmann
  msg = result.fail_msg
559 0623d351 Iustin Pop
  if msg:
560 0623d351 Iustin Pop
    logging.warning("Could not shutdown the node daemon and cleanup"
561 0623d351 Iustin Pop
                    " the node: %s", msg)
562 140aa4a8 Iustin Pop
563 140aa4a8 Iustin Pop
564 87622829 Iustin Pop
def SetupNodeDaemon(cluster_name, node, ssh_key_check):
565 827f753e Guido Trotter
  """Add a node to the cluster.
566 827f753e Guido Trotter

567 b1b6ea87 Iustin Pop
  This function must be called before the actual opcode, and will ssh
568 b1b6ea87 Iustin Pop
  to the remote node, copy the needed files, and start ganeti-noded,
569 b1b6ea87 Iustin Pop
  allowing the master to do the rest via normal rpc calls.
570 827f753e Guido Trotter

571 87622829 Iustin Pop
  @param cluster_name: the cluster name
572 87622829 Iustin Pop
  @param node: the name of the new node
573 87622829 Iustin Pop
  @param ssh_key_check: whether to do a strict key check
574 827f753e Guido Trotter

575 827f753e Guido Trotter
  """
576 b43dcc5a Manuel Franceschini
  family = ssconf.SimpleStore().GetPrimaryIPFamily()
577 b43dcc5a Manuel Franceschini
  sshrunner = ssh.SshRunner(cluster_name,
578 72729d6e Michael Hanselmann
                            ipv6=(family == netutils.IP6Address.family))
579 827f753e Guido Trotter
580 b43dcc5a Manuel Franceschini
  bind_address = constants.IP4_ADDRESS_ANY
581 b43dcc5a Manuel Franceschini
  if family == netutils.IP6Address.family:
582 b43dcc5a Manuel Franceschini
    bind_address = constants.IP6_ADDRESS_ANY
583 b43dcc5a Manuel Franceschini
584 827f753e Guido Trotter
  # set up inter-node password and certificate and restarts the node daemon
585 827f753e Guido Trotter
  # and then connect with ssh to set password and start ganeti-noded
586 827f753e Guido Trotter
  # note that all the below variables are sanitized at this point,
587 827f753e Guido Trotter
  # either by being constants or by the checks above
588 9294514d René Nussbaumer
  sshrunner.CopyFileToNode(node, constants.NODED_CERT_FILE)
589 9294514d René Nussbaumer
  sshrunner.CopyFileToNode(node, constants.RAPI_CERT_FILE)
590 9294514d René Nussbaumer
  sshrunner.CopyFileToNode(node, constants.CONFD_HMAC_KEY)
591 72729d6e Michael Hanselmann
  mycommand = ("%s stop-all; %s start %s -b %s" %
592 72729d6e Michael Hanselmann
               (constants.DAEMON_UTIL, constants.DAEMON_UTIL, constants.NODED,
593 72729d6e Michael Hanselmann
                utils.ShellQuote(bind_address)))
594 827f753e Guido Trotter
595 c4b6c29c Michael Hanselmann
  result = sshrunner.Run(node, 'root', mycommand, batch=False,
596 c4b6c29c Michael Hanselmann
                         ask_key=ssh_key_check,
597 9294514d René Nussbaumer
                         use_cluster_key=True,
598 c4b6c29c Michael Hanselmann
                         strict_host_check=ssh_key_check)
599 827f753e Guido Trotter
  if result.failed:
600 827f753e Guido Trotter
    raise errors.OpExecError("Remote command on node %s, error: %s,"
601 827f753e Guido Trotter
                             " output: %s" %
602 827f753e Guido Trotter
                             (node, result.fail_reason, result.output))
603 827f753e Guido Trotter
604 5627f375 Michael Hanselmann
  _WaitForNodeDaemon(node)
605 5627f375 Michael Hanselmann
606 b1b6ea87 Iustin Pop
607 8e2524c3 Guido Trotter
def MasterFailover(no_voting=False):
608 b1b6ea87 Iustin Pop
  """Failover the master node.
609 b1b6ea87 Iustin Pop

610 b1b6ea87 Iustin Pop
  This checks that we are not already the master, and will cause the
611 b1b6ea87 Iustin Pop
  current master to cease being master, and the non-master to become
612 b1b6ea87 Iustin Pop
  new master.
613 b1b6ea87 Iustin Pop

614 8e2524c3 Guido Trotter
  @type no_voting: boolean
615 8e2524c3 Guido Trotter
  @param no_voting: force the operation without remote nodes agreement
616 8e2524c3 Guido Trotter
                      (dangerous)
617 8e2524c3 Guido Trotter

618 b1b6ea87 Iustin Pop
  """
619 8135a2db Iustin Pop
  sstore = ssconf.SimpleStore()
620 b1b6ea87 Iustin Pop
621 8135a2db Iustin Pop
  old_master, new_master = ssconf.GetMasterAndMyself(sstore)
622 8135a2db Iustin Pop
  node_list = sstore.GetNodeList()
623 8135a2db Iustin Pop
  mc_list = sstore.GetMasterCandidates()
624 b1b6ea87 Iustin Pop
625 b1b6ea87 Iustin Pop
  if old_master == new_master:
626 b1b6ea87 Iustin Pop
    raise errors.OpPrereqError("This commands must be run on the node"
627 b1b6ea87 Iustin Pop
                               " where you want the new master to be."
628 b1b6ea87 Iustin Pop
                               " %s is already the master" %
629 debac808 Iustin Pop
                               old_master, errors.ECODE_INVAL)
630 d5927e48 Iustin Pop
631 8135a2db Iustin Pop
  if new_master not in mc_list:
632 8135a2db Iustin Pop
    mc_no_master = [name for name in mc_list if name != old_master]
633 8135a2db Iustin Pop
    raise errors.OpPrereqError("This node is not among the nodes marked"
634 8135a2db Iustin Pop
                               " as master candidates. Only these nodes"
635 8135a2db Iustin Pop
                               " can become masters. Current list of"
636 8135a2db Iustin Pop
                               " master candidates is:\n"
637 debac808 Iustin Pop
                               "%s" % ('\n'.join(mc_no_master)),
638 debac808 Iustin Pop
                               errors.ECODE_STATE)
639 8135a2db Iustin Pop
640 8e2524c3 Guido Trotter
  if not no_voting:
641 8e2524c3 Guido Trotter
    vote_list = GatherMasterVotes(node_list)
642 8e2524c3 Guido Trotter
643 8e2524c3 Guido Trotter
    if vote_list:
644 8e2524c3 Guido Trotter
      voted_master = vote_list[0][0]
645 8e2524c3 Guido Trotter
      if voted_master is None:
646 8e2524c3 Guido Trotter
        raise errors.OpPrereqError("Cluster is inconsistent, most nodes did"
647 debac808 Iustin Pop
                                   " not respond.", errors.ECODE_ENVIRON)
648 8e2524c3 Guido Trotter
      elif voted_master != old_master:
649 8e2524c3 Guido Trotter
        raise errors.OpPrereqError("I have a wrong configuration, I believe"
650 8e2524c3 Guido Trotter
                                   " the master is %s but the other nodes"
651 8e2524c3 Guido Trotter
                                   " voted %s. Please resync the configuration"
652 8e2524c3 Guido Trotter
                                   " of this node." %
653 debac808 Iustin Pop
                                   (old_master, voted_master),
654 debac808 Iustin Pop
                                   errors.ECODE_STATE)
655 b1b6ea87 Iustin Pop
  # end checks
656 b1b6ea87 Iustin Pop
657 b1b6ea87 Iustin Pop
  rcode = 0
658 b1b6ea87 Iustin Pop
659 d5927e48 Iustin Pop
  logging.info("Setting master to %s, old master: %s", new_master, old_master)
660 b1b6ea87 Iustin Pop
661 21004460 Iustin Pop
  try:
662 21004460 Iustin Pop
    # instantiate a real config writer, as we now know we have the
663 21004460 Iustin Pop
    # configuration data
664 eb180fe2 Iustin Pop
    cfg = config.ConfigWriter(accept_foreign=True)
665 21004460 Iustin Pop
666 21004460 Iustin Pop
    cluster_info = cfg.GetClusterInfo()
667 21004460 Iustin Pop
    cluster_info.master_node = new_master
668 21004460 Iustin Pop
    # this will also regenerate the ssconf files, since we updated the
669 21004460 Iustin Pop
    # cluster info
670 21004460 Iustin Pop
    cfg.Update(cluster_info, logging.error)
671 21004460 Iustin Pop
  except errors.ConfigurationError, err:
672 21004460 Iustin Pop
    logging.error("Error while trying to set the new master: %s",
673 21004460 Iustin Pop
                  str(err))
674 21004460 Iustin Pop
    return 1
675 21004460 Iustin Pop
676 21004460 Iustin Pop
  # if cfg.Update worked, then it means the old master daemon won't be
677 21004460 Iustin Pop
  # able now to write its own config file (we rely on locking in both
678 21004460 Iustin Pop
  # backend.UploadFile() and ConfigWriter._Write(); hence the next
679 21004460 Iustin Pop
  # step is to kill the old master
680 21004460 Iustin Pop
681 21004460 Iustin Pop
  logging.info("Stopping the master daemon on node %s", old_master)
682 21004460 Iustin Pop
683 781de953 Iustin Pop
  result = rpc.RpcRunner.call_node_stop_master(old_master, True)
684 3cebe102 Michael Hanselmann
  msg = result.fail_msg
685 6c00d19a Iustin Pop
  if msg:
686 d5927e48 Iustin Pop
    logging.error("Could not disable the master role on the old master"
687 6c00d19a Iustin Pop
                 " %s, please disable manually: %s", old_master, msg)
688 b1b6ea87 Iustin Pop
689 21004460 Iustin Pop
  logging.info("Checking master IP non-reachability...")
690 21004460 Iustin Pop
691 425f0f54 Iustin Pop
  master_ip = sstore.GetMasterIP()
692 425f0f54 Iustin Pop
  total_timeout = 30
693 d23ef431 Michael Hanselmann
  # Here we have a phase where no master should be running
694 425f0f54 Iustin Pop
  def _check_ip():
695 a744b676 Manuel Franceschini
    if netutils.TcpPing(master_ip, constants.DEFAULT_NODED_PORT):
696 425f0f54 Iustin Pop
      raise utils.RetryAgain()
697 425f0f54 Iustin Pop
698 425f0f54 Iustin Pop
  try:
699 425f0f54 Iustin Pop
    utils.Retry(_check_ip, (1, 1.5, 5), total_timeout)
700 425f0f54 Iustin Pop
  except utils.RetryTimeout:
701 425f0f54 Iustin Pop
    logging.warning("The master IP is still reachable after %s seconds,"
702 425f0f54 Iustin Pop
                    " continuing but activating the master on the current"
703 425f0f54 Iustin Pop
                    " node will probably fail", total_timeout)
704 b1b6ea87 Iustin Pop
705 ff699aa9 Michael Hanselmann
  if jstore.CheckDrainFlag():
706 ff699aa9 Michael Hanselmann
    logging.info("Undraining job queue")
707 ff699aa9 Michael Hanselmann
    jstore.SetDrainFlag(False)
708 ff699aa9 Michael Hanselmann
709 21004460 Iustin Pop
  logging.info("Starting the master daemons on the new master")
710 d5927e48 Iustin Pop
711 3583908a Guido Trotter
  result = rpc.RpcRunner.call_node_start_master(new_master, True, no_voting)
712 3cebe102 Michael Hanselmann
  msg = result.fail_msg
713 b726aff0 Iustin Pop
  if msg:
714 d5927e48 Iustin Pop
    logging.error("Could not start the master role on the new master"
715 b726aff0 Iustin Pop
                  " %s, please check: %s", new_master, msg)
716 b1b6ea87 Iustin Pop
    rcode = 1
717 b1b6ea87 Iustin Pop
718 21004460 Iustin Pop
  logging.info("Master failed over from %s to %s", old_master, new_master)
719 b1b6ea87 Iustin Pop
  return rcode
720 d7cdb55d Iustin Pop
721 d7cdb55d Iustin Pop
722 8eb148ae Iustin Pop
def GetMaster():
723 8eb148ae Iustin Pop
  """Returns the current master node.
724 8eb148ae Iustin Pop

725 8eb148ae Iustin Pop
  This is a separate function in bootstrap since it's needed by
726 8eb148ae Iustin Pop
  gnt-cluster, and instead of importing directly ssconf, it's better
727 8eb148ae Iustin Pop
  to abstract it in bootstrap, where we do use ssconf in other
728 8eb148ae Iustin Pop
  functions too.
729 8eb148ae Iustin Pop

730 8eb148ae Iustin Pop
  """
731 8eb148ae Iustin Pop
  sstore = ssconf.SimpleStore()
732 8eb148ae Iustin Pop
733 8eb148ae Iustin Pop
  old_master, _ = ssconf.GetMasterAndMyself(sstore)
734 8eb148ae Iustin Pop
735 8eb148ae Iustin Pop
  return old_master
736 8eb148ae Iustin Pop
737 8eb148ae Iustin Pop
738 d7cdb55d Iustin Pop
def GatherMasterVotes(node_list):
739 d7cdb55d Iustin Pop
  """Check the agreement on who is the master.
740 d7cdb55d Iustin Pop

741 d7cdb55d Iustin Pop
  This function will return a list of (node, number of votes), ordered
742 d7cdb55d Iustin Pop
  by the number of votes. Errors will be denoted by the key 'None'.
743 d7cdb55d Iustin Pop

744 d7cdb55d Iustin Pop
  Note that the sum of votes is the number of nodes this machine
745 d7cdb55d Iustin Pop
  knows, whereas the number of entries in the list could be different
746 d7cdb55d Iustin Pop
  (if some nodes vote for another master).
747 d7cdb55d Iustin Pop

748 d7cdb55d Iustin Pop
  We remove ourselves from the list since we know that (bugs aside)
749 d7cdb55d Iustin Pop
  since we use the same source for configuration information for both
750 d7cdb55d Iustin Pop
  backend and boostrap, we'll always vote for ourselves.
751 d7cdb55d Iustin Pop

752 d7cdb55d Iustin Pop
  @type node_list: list
753 d7cdb55d Iustin Pop
  @param node_list: the list of nodes to query for master info; the current
754 5bbd3f7f Michael Hanselmann
      node will be removed if it is in the list
755 d7cdb55d Iustin Pop
  @rtype: list
756 d7cdb55d Iustin Pop
  @return: list of (node, votes)
757 d7cdb55d Iustin Pop

758 d7cdb55d Iustin Pop
  """
759 b705c7a6 Manuel Franceschini
  myself = netutils.Hostname.GetSysName()
760 d7cdb55d Iustin Pop
  try:
761 d7cdb55d Iustin Pop
    node_list.remove(myself)
762 d7cdb55d Iustin Pop
  except ValueError:
763 d7cdb55d Iustin Pop
    pass
764 d7cdb55d Iustin Pop
  if not node_list:
765 d7cdb55d Iustin Pop
    # no nodes left (eventually after removing myself)
766 d7cdb55d Iustin Pop
    return []
767 d7cdb55d Iustin Pop
  results = rpc.RpcRunner.call_master_info(node_list)
768 d7cdb55d Iustin Pop
  if not isinstance(results, dict):
769 d7cdb55d Iustin Pop
    # this should not happen (unless internal error in rpc)
770 d7cdb55d Iustin Pop
    logging.critical("Can't complete rpc call, aborting master startup")
771 d7cdb55d Iustin Pop
    return [(None, len(node_list))]
772 d7cdb55d Iustin Pop
  votes = {}
773 d7cdb55d Iustin Pop
  for node in results:
774 781de953 Iustin Pop
    nres = results[node]
775 2a52a064 Iustin Pop
    data = nres.payload
776 3cebe102 Michael Hanselmann
    msg = nres.fail_msg
777 2a52a064 Iustin Pop
    fail = False
778 2a52a064 Iustin Pop
    if msg:
779 2a52a064 Iustin Pop
      logging.warning("Error contacting node %s: %s", node, msg)
780 2a52a064 Iustin Pop
      fail = True
781 d8e0caa6 Manuel Franceschini
    # for now we accept both length 3 and 4 (data[3] is primary ip version)
782 2a52a064 Iustin Pop
    elif not isinstance(data, (tuple, list)) or len(data) < 3:
783 2a52a064 Iustin Pop
      logging.warning("Invalid data received from node %s: %s", node, data)
784 2a52a064 Iustin Pop
      fail = True
785 2a52a064 Iustin Pop
    if fail:
786 d7cdb55d Iustin Pop
      if None not in votes:
787 d7cdb55d Iustin Pop
        votes[None] = 0
788 d7cdb55d Iustin Pop
      votes[None] += 1
789 d7cdb55d Iustin Pop
      continue
790 781de953 Iustin Pop
    master_node = data[2]
791 d7cdb55d Iustin Pop
    if master_node not in votes:
792 d7cdb55d Iustin Pop
      votes[master_node] = 0
793 d7cdb55d Iustin Pop
    votes[master_node] += 1
794 d7cdb55d Iustin Pop
795 d7cdb55d Iustin Pop
  vote_list = [v for v in votes.items()]
796 d7cdb55d Iustin Pop
  # sort first on number of votes then on name, since we want None
797 d7cdb55d Iustin Pop
  # sorted later if we have the half of the nodes not responding, and
798 d7cdb55d Iustin Pop
  # half voting all for the same master
799 d7cdb55d Iustin Pop
  vote_list.sort(key=lambda x: (x[1], x[0]), reverse=True)
800 d7cdb55d Iustin Pop
801 d7cdb55d Iustin Pop
  return vote_list