Statistics
| Branch: | Tag: | Revision:

root / lib / bootstrap.py @ 33c730a2

History | View | Annotate | Download (35.3 kB)

1
#
2
#
3

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

    
21

    
22
"""Functions to bootstrap a new cluster.
23

24
"""
25

    
26
import os
27
import os.path
28
import re
29
import logging
30
import time
31
import tempfile
32

    
33
from ganeti import rpc
34
from ganeti import ssh
35
from ganeti import utils
36
from ganeti import errors
37
from ganeti import config
38
from ganeti import constants
39
from ganeti import objects
40
from ganeti import ssconf
41
from ganeti import serializer
42
from ganeti import hypervisor
43
from ganeti import bdev
44
from ganeti import netutils
45
from ganeti import luxi
46
from ganeti import jstore
47
from ganeti import pathutils
48

    
49

    
50
# ec_id for InitConfig's temporary reservation manager
51
_INITCONF_ECID = "initconfig-ecid"
52

    
53
#: After how many seconds daemon must be responsive
54
_DAEMON_READY_TIMEOUT = 10.0
55

    
56

    
57
def _InitSSHSetup():
58
  """Setup the SSH configuration for the cluster.
59

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

63
  """
64
  priv_key, pub_key, auth_keys = ssh.GetUserFiles(constants.SSH_LOGIN_USER)
65

    
66
  for name in priv_key, pub_key:
67
    if os.path.exists(name):
68
      utils.CreateBackup(name)
69
    utils.RemoveFile(name)
70

    
71
  result = utils.RunCmd(["ssh-keygen", "-t", "dsa",
72
                         "-f", priv_key,
73
                         "-q", "-N", ""])
74
  if result.failed:
75
    raise errors.OpExecError("Could not generate ssh keypair, error %s" %
76
                             result.output)
77

    
78
  utils.AddAuthorizedKey(auth_keys, utils.ReadFile(pub_key))
79

    
80

    
81
def GenerateHmacKey(file_name):
82
  """Writes a new HMAC key.
83

84
  @type file_name: str
85
  @param file_name: Path to output file
86

87
  """
88
  utils.WriteFile(file_name, data="%s\n" % utils.GenerateSecret(), mode=0400,
89
                  backup=True)
90

    
91

    
92
def GenerateClusterCrypto(new_cluster_cert, new_rapi_cert, new_spice_cert,
93
                          new_confd_hmac_key, new_cds,
94
                          rapi_cert_pem=None, spice_cert_pem=None,
95
                          spice_cacert_pem=None, cds=None,
96
                          nodecert_file=pathutils.NODED_CERT_FILE,
97
                          rapicert_file=pathutils.RAPI_CERT_FILE,
98
                          spicecert_file=pathutils.SPICE_CERT_FILE,
99
                          spicecacert_file=pathutils.SPICE_CACERT_FILE,
100
                          hmackey_file=pathutils.CONFD_HMAC_KEY,
101
                          cds_file=pathutils.CLUSTER_DOMAIN_SECRET_FILE):
102
  """Updates the cluster certificates, keys and secrets.
103

104
  @type new_cluster_cert: bool
105
  @param new_cluster_cert: Whether to generate a new cluster certificate
106
  @type new_rapi_cert: bool
107
  @param new_rapi_cert: Whether to generate a new RAPI certificate
108
  @type new_spice_cert: bool
109
  @param new_spice_cert: Whether to generate a new SPICE certificate
110
  @type new_confd_hmac_key: bool
111
  @param new_confd_hmac_key: Whether to generate a new HMAC key
112
  @type new_cds: bool
113
  @param new_cds: Whether to generate a new cluster domain secret
114
  @type rapi_cert_pem: string
115
  @param rapi_cert_pem: New RAPI certificate in PEM format
116
  @type spice_cert_pem: string
117
  @param spice_cert_pem: New SPICE certificate in PEM format
118
  @type spice_cacert_pem: string
119
  @param spice_cacert_pem: Certificate of the CA that signed the SPICE
120
                           certificate, in PEM format
121
  @type cds: string
122
  @param cds: New cluster domain secret
123
  @type nodecert_file: string
124
  @param nodecert_file: optional override of the node cert file path
125
  @type rapicert_file: string
126
  @param rapicert_file: optional override of the rapi cert file path
127
  @type spicecert_file: string
128
  @param spicecert_file: optional override of the spice cert file path
129
  @type spicecacert_file: string
130
  @param spicecacert_file: optional override of the spice CA cert file path
131
  @type hmackey_file: string
132
  @param hmackey_file: optional override of the hmac key file path
133

134
  """
135
  # noded SSL certificate
136
  cluster_cert_exists = os.path.exists(nodecert_file)
137
  if new_cluster_cert or not cluster_cert_exists:
138
    if cluster_cert_exists:
139
      utils.CreateBackup(nodecert_file)
140

    
141
    logging.debug("Generating new cluster certificate at %s", nodecert_file)
142
    utils.GenerateSelfSignedSslCert(nodecert_file)
143

    
144
  # confd HMAC key
145
  if new_confd_hmac_key or not os.path.exists(hmackey_file):
146
    logging.debug("Writing new confd HMAC key to %s", hmackey_file)
147
    GenerateHmacKey(hmackey_file)
148

    
149
  # RAPI
150
  rapi_cert_exists = os.path.exists(rapicert_file)
151

    
152
  if rapi_cert_pem:
153
    # Assume rapi_pem contains a valid PEM-formatted certificate and key
154
    logging.debug("Writing RAPI certificate at %s", rapicert_file)
155
    utils.WriteFile(rapicert_file, data=rapi_cert_pem, backup=True)
156

    
157
  elif new_rapi_cert or not rapi_cert_exists:
158
    if rapi_cert_exists:
159
      utils.CreateBackup(rapicert_file)
160

    
161
    logging.debug("Generating new RAPI certificate at %s", rapicert_file)
162
    utils.GenerateSelfSignedSslCert(rapicert_file)
163

    
164
  # SPICE
165
  spice_cert_exists = os.path.exists(spicecert_file)
166
  spice_cacert_exists = os.path.exists(spicecacert_file)
167
  if spice_cert_pem:
168
    # spice_cert_pem implies also spice_cacert_pem
169
    logging.debug("Writing SPICE certificate at %s", spicecert_file)
170
    utils.WriteFile(spicecert_file, data=spice_cert_pem, backup=True)
171
    logging.debug("Writing SPICE CA certificate at %s", spicecacert_file)
172
    utils.WriteFile(spicecacert_file, data=spice_cacert_pem, backup=True)
173
  elif new_spice_cert or not spice_cert_exists:
174
    if spice_cert_exists:
175
      utils.CreateBackup(spicecert_file)
176
    if spice_cacert_exists:
177
      utils.CreateBackup(spicecacert_file)
178

    
179
    logging.debug("Generating new self-signed SPICE certificate at %s",
180
                  spicecert_file)
181
    (_, cert_pem) = utils.GenerateSelfSignedSslCert(spicecert_file)
182

    
183
    # Self-signed certificate -> the public certificate is also the CA public
184
    # certificate
185
    logging.debug("Writing the public certificate to %s",
186
                  spicecert_file)
187
    utils.io.WriteFile(spicecacert_file, mode=0400, data=cert_pem)
188

    
189
  # Cluster domain secret
190
  if cds:
191
    logging.debug("Writing cluster domain secret to %s", cds_file)
192
    utils.WriteFile(cds_file, data=cds, backup=True)
193

    
194
  elif new_cds or not os.path.exists(cds_file):
195
    logging.debug("Generating new cluster domain secret at %s", cds_file)
196
    GenerateHmacKey(cds_file)
197

    
198

    
199
def _InitGanetiServerSetup(master_name):
200
  """Setup the necessary configuration for the initial node daemon.
201

202
  This creates the nodepass file containing the shared password for
203
  the cluster, generates the SSL certificate and starts the node daemon.
204

205
  @type master_name: str
206
  @param master_name: Name of the master node
207

208
  """
209
  # Generate cluster secrets
210
  GenerateClusterCrypto(True, False, False, False, False)
211

    
212
  result = utils.RunCmd([pathutils.DAEMON_UTIL, "start", constants.NODED])
213
  if result.failed:
214
    raise errors.OpExecError("Could not start the node daemon, command %s"
215
                             " had exitcode %s and error %s" %
216
                             (result.cmd, result.exit_code, result.output))
217

    
218
  _WaitForNodeDaemon(master_name)
219

    
220

    
221
def _WaitForNodeDaemon(node_name):
222
  """Wait for node daemon to become responsive.
223

224
  """
225
  def _CheckNodeDaemon():
226
    # Pylint bug <http://www.logilab.org/ticket/35642>
227
    # pylint: disable=E1101
228
    result = rpc.BootstrapRunner().call_version([node_name])[node_name]
229
    if result.fail_msg:
230
      raise utils.RetryAgain()
231

    
232
  try:
233
    utils.Retry(_CheckNodeDaemon, 1.0, _DAEMON_READY_TIMEOUT)
234
  except utils.RetryTimeout:
235
    raise errors.OpExecError("Node daemon on %s didn't answer queries within"
236
                             " %s seconds" % (node_name, _DAEMON_READY_TIMEOUT))
237

    
238

    
239
def _WaitForMasterDaemon():
240
  """Wait for master daemon to become responsive.
241

242
  """
243
  def _CheckMasterDaemon():
244
    try:
245
      cl = luxi.Client()
246
      (cluster_name, ) = cl.QueryConfigValues(["cluster_name"])
247
    except Exception:
248
      raise utils.RetryAgain()
249

    
250
    logging.debug("Received cluster name %s from master", cluster_name)
251

    
252
  try:
253
    utils.Retry(_CheckMasterDaemon, 1.0, _DAEMON_READY_TIMEOUT)
254
  except utils.RetryTimeout:
255
    raise errors.OpExecError("Master daemon didn't answer queries within"
256
                             " %s seconds" % _DAEMON_READY_TIMEOUT)
257

    
258

    
259
def _WaitForSshDaemon(hostname, port, family):
260
  """Wait for SSH daemon to become responsive.
261

262
  """
263
  hostip = netutils.GetHostname(name=hostname, family=family).ip
264

    
265
  def _CheckSshDaemon():
266
    if netutils.TcpPing(hostip, port, timeout=1.0, live_port_needed=True):
267
      logging.debug("SSH daemon on %s:%s (IP address %s) has become"
268
                    " responsive", hostname, port, hostip)
269
    else:
270
      raise utils.RetryAgain()
271

    
272
  try:
273
    utils.Retry(_CheckSshDaemon, 1.0, _DAEMON_READY_TIMEOUT)
274
  except utils.RetryTimeout:
275
    raise errors.OpExecError("SSH daemon on %s:%s (IP address %s) didn't"
276
                             " become responsive within %s seconds" %
277
                             (hostname, port, hostip, _DAEMON_READY_TIMEOUT))
278

    
279

    
280
def RunNodeSetupCmd(cluster_name, node, basecmd, debug, verbose,
281
                    use_cluster_key, ask_key, strict_host_check, data):
282
  """Runs a command to configure something on a remote machine.
283

284
  @type cluster_name: string
285
  @param cluster_name: Cluster name
286
  @type node: string
287
  @param node: Node name
288
  @type basecmd: string
289
  @param basecmd: Base command (path on the remote machine)
290
  @type debug: bool
291
  @param debug: Enable debug output
292
  @type verbose: bool
293
  @param verbose: Enable verbose output
294
  @type use_cluster_key: bool
295
  @param use_cluster_key: See L{ssh.SshRunner.BuildCmd}
296
  @type ask_key: bool
297
  @param ask_key: See L{ssh.SshRunner.BuildCmd}
298
  @type strict_host_check: bool
299
  @param strict_host_check: See L{ssh.SshRunner.BuildCmd}
300
  @param data: JSON-serializable input data for script (passed to stdin)
301

302
  """
303
  cmd = [basecmd]
304

    
305
  # Pass --debug/--verbose to the external script if set on our invocation
306
  if debug:
307
    cmd.append("--debug")
308

    
309
  if verbose:
310
    cmd.append("--verbose")
311

    
312
  family = ssconf.SimpleStore().GetPrimaryIPFamily()
313
  srun = ssh.SshRunner(cluster_name,
314
                       ipv6=(family == netutils.IP6Address.family))
315
  scmd = srun.BuildCmd(node, constants.SSH_LOGIN_USER,
316
                       utils.ShellQuoteArgs(cmd),
317
                       batch=False, ask_key=ask_key, quiet=False,
318
                       strict_host_check=strict_host_check,
319
                       use_cluster_key=use_cluster_key)
320

    
321
  tempfh = tempfile.TemporaryFile()
322
  try:
323
    tempfh.write(serializer.DumpJson(data))
324
    tempfh.seek(0)
325

    
326
    result = utils.RunCmd(scmd, interactive=True, input_fd=tempfh)
327
  finally:
328
    tempfh.close()
329

    
330
  if result.failed:
331
    raise errors.OpExecError("Command '%s' failed: %s" %
332
                             (result.cmd, result.fail_reason))
333

    
334
  _WaitForSshDaemon(node, netutils.GetDaemonPort(constants.SSH), family)
335

    
336

    
337
def _InitFileStorage(file_storage_dir):
338
  """Initialize if needed the file storage.
339

340
  @param file_storage_dir: the user-supplied value
341
  @return: either empty string (if file storage was disabled at build
342
      time) or the normalized path to the storage directory
343

344
  """
345
  file_storage_dir = os.path.normpath(file_storage_dir)
346

    
347
  if not os.path.isabs(file_storage_dir):
348
    raise errors.OpPrereqError("File storage directory '%s' is not an absolute"
349
                               " path" % file_storage_dir, errors.ECODE_INVAL)
350

    
351
  if not os.path.exists(file_storage_dir):
352
    try:
353
      os.makedirs(file_storage_dir, 0750)
354
    except OSError, err:
355
      raise errors.OpPrereqError("Cannot create file storage directory"
356
                                 " '%s': %s" % (file_storage_dir, err),
357
                                 errors.ECODE_ENVIRON)
358

    
359
  if not os.path.isdir(file_storage_dir):
360
    raise errors.OpPrereqError("The file storage directory '%s' is not"
361
                               " a directory." % file_storage_dir,
362
                               errors.ECODE_ENVIRON)
363
  return file_storage_dir
364

    
365

    
366
def InitCluster(cluster_name, mac_prefix, # pylint: disable=R0913, R0914
367
                master_netmask, master_netdev, file_storage_dir,
368
                shared_file_storage_dir, candidate_pool_size, secondary_ip=None,
369
                vg_name=None, beparams=None, nicparams=None, ndparams=None,
370
                hvparams=None, diskparams=None, enabled_hypervisors=None,
371
                modify_etc_hosts=True, modify_ssh_setup=True,
372
                maintain_node_health=False, drbd_helper=None, uid_pool=None,
373
                default_iallocator=None, primary_ip_version=None, ipolicy=None,
374
                prealloc_wipe_disks=False, use_external_mip_script=False,
375
                hv_state=None, disk_state=None):
376
  """Initialise the cluster.
377

378
  @type candidate_pool_size: int
379
  @param candidate_pool_size: master candidate pool size
380

381
  """
382
  # TODO: complete the docstring
383
  if config.ConfigWriter.IsCluster():
384
    raise errors.OpPrereqError("Cluster is already initialised",
385
                               errors.ECODE_STATE)
386

    
387
  if not enabled_hypervisors:
388
    raise errors.OpPrereqError("Enabled hypervisors list must contain at"
389
                               " least one member", errors.ECODE_INVAL)
390
  invalid_hvs = set(enabled_hypervisors) - constants.HYPER_TYPES
391
  if invalid_hvs:
392
    raise errors.OpPrereqError("Enabled hypervisors contains invalid"
393
                               " entries: %s" % invalid_hvs,
394
                               errors.ECODE_INVAL)
395

    
396
  try:
397
    ipcls = netutils.IPAddress.GetClassFromIpVersion(primary_ip_version)
398
  except errors.ProgrammerError:
399
    raise errors.OpPrereqError("Invalid primary ip version: %d." %
400
                               primary_ip_version, errors.ECODE_INVAL)
401

    
402
  hostname = netutils.GetHostname(family=ipcls.family)
403
  if not ipcls.IsValid(hostname.ip):
404
    raise errors.OpPrereqError("This host's IP (%s) is not a valid IPv%d"
405
                               " address." % (hostname.ip, primary_ip_version),
406
                               errors.ECODE_INVAL)
407

    
408
  if ipcls.IsLoopback(hostname.ip):
409
    raise errors.OpPrereqError("This host's IP (%s) resolves to a loopback"
410
                               " address. Please fix DNS or %s." %
411
                               (hostname.ip, pathutils.ETC_HOSTS),
412
                               errors.ECODE_ENVIRON)
413

    
414
  if not ipcls.Own(hostname.ip):
415
    raise errors.OpPrereqError("Inconsistency: this host's name resolves"
416
                               " to %s,\nbut this ip address does not"
417
                               " belong to this host" %
418
                               hostname.ip, errors.ECODE_ENVIRON)
419

    
420
  clustername = netutils.GetHostname(name=cluster_name, family=ipcls.family)
421

    
422
  if netutils.TcpPing(clustername.ip, constants.DEFAULT_NODED_PORT, timeout=5):
423
    raise errors.OpPrereqError("Cluster IP already active",
424
                               errors.ECODE_NOTUNIQUE)
425

    
426
  if not secondary_ip:
427
    if primary_ip_version == constants.IP6_VERSION:
428
      raise errors.OpPrereqError("When using a IPv6 primary address, a valid"
429
                                 " IPv4 address must be given as secondary",
430
                                 errors.ECODE_INVAL)
431
    secondary_ip = hostname.ip
432

    
433
  if not netutils.IP4Address.IsValid(secondary_ip):
434
    raise errors.OpPrereqError("Secondary IP address (%s) has to be a valid"
435
                               " IPv4 address." % secondary_ip,
436
                               errors.ECODE_INVAL)
437

    
438
  if not netutils.IP4Address.Own(secondary_ip):
439
    raise errors.OpPrereqError("You gave %s as secondary IP,"
440
                               " but it does not belong to this host." %
441
                               secondary_ip, errors.ECODE_ENVIRON)
442

    
443
  if master_netmask is not None:
444
    if not ipcls.ValidateNetmask(master_netmask):
445
      raise errors.OpPrereqError("CIDR netmask (%s) not valid for IPv%s " %
446
                                  (master_netmask, primary_ip_version),
447
                                 errors.ECODE_INVAL)
448
  else:
449
    master_netmask = ipcls.iplen
450

    
451
  if vg_name is not None:
452
    # Check if volume group is valid
453
    vgstatus = utils.CheckVolumeGroupSize(utils.ListVolumeGroups(), vg_name,
454
                                          constants.MIN_VG_SIZE)
455
    if vgstatus:
456
      raise errors.OpPrereqError("Error: %s\nspecify --no-lvm-storage if"
457
                                 " you are not using lvm" % vgstatus,
458
                                 errors.ECODE_INVAL)
459

    
460
  if drbd_helper is not None:
461
    try:
462
      curr_helper = bdev.BaseDRBD.GetUsermodeHelper()
463
    except errors.BlockDeviceError, err:
464
      raise errors.OpPrereqError("Error while checking drbd helper"
465
                                 " (specify --no-drbd-storage if you are not"
466
                                 " using drbd): %s" % str(err),
467
                                 errors.ECODE_ENVIRON)
468
    if drbd_helper != curr_helper:
469
      raise errors.OpPrereqError("Error: requiring %s as drbd helper but %s"
470
                                 " is the current helper" % (drbd_helper,
471
                                                             curr_helper),
472
                                 errors.ECODE_INVAL)
473

    
474
  logging.debug("Stopping daemons (if any are running)")
475
  result = utils.RunCmd([pathutils.DAEMON_UTIL, "stop-all"])
476
  if result.failed:
477
    raise errors.OpExecError("Could not stop daemons, command %s"
478
                             " had exitcode %s and error '%s'" %
479
                             (result.cmd, result.exit_code, result.output))
480

    
481
  if constants.ENABLE_FILE_STORAGE:
482
    file_storage_dir = _InitFileStorage(file_storage_dir)
483
  else:
484
    file_storage_dir = ""
485

    
486
  if constants.ENABLE_SHARED_FILE_STORAGE:
487
    shared_file_storage_dir = _InitFileStorage(shared_file_storage_dir)
488
  else:
489
    shared_file_storage_dir = ""
490

    
491
  if not re.match("^[0-9a-z]{2}:[0-9a-z]{2}:[0-9a-z]{2}$", mac_prefix):
492
    raise errors.OpPrereqError("Invalid mac prefix given '%s'" % mac_prefix,
493
                               errors.ECODE_INVAL)
494

    
495
  result = utils.RunCmd(["ip", "link", "show", "dev", master_netdev])
496
  if result.failed:
497
    raise errors.OpPrereqError("Invalid master netdev given (%s): '%s'" %
498
                               (master_netdev,
499
                                result.output.strip()), errors.ECODE_INVAL)
500

    
501
  dirs = [(pathutils.RUN_DIR, constants.RUN_DIRS_MODE)]
502
  utils.EnsureDirs(dirs)
503

    
504
  objects.UpgradeBeParams(beparams)
505
  utils.ForceDictType(beparams, constants.BES_PARAMETER_TYPES)
506
  utils.ForceDictType(nicparams, constants.NICS_PARAMETER_TYPES)
507

    
508
  objects.NIC.CheckParameterSyntax(nicparams)
509

    
510
  full_ipolicy = objects.FillIPolicy(constants.IPOLICY_DEFAULTS, ipolicy)
511

    
512
  if ndparams is not None:
513
    utils.ForceDictType(ndparams, constants.NDS_PARAMETER_TYPES)
514
  else:
515
    ndparams = dict(constants.NDC_DEFAULTS)
516

    
517
  # This is ugly, as we modify the dict itself
518
  # FIXME: Make utils.ForceDictType pure functional or write a wrapper
519
  # around it
520
  if hv_state:
521
    for hvname, hvs_data in hv_state.items():
522
      utils.ForceDictType(hvs_data, constants.HVSTS_PARAMETER_TYPES)
523
      hv_state[hvname] = objects.Cluster.SimpleFillHvState(hvs_data)
524
  else:
525
    hv_state = dict((hvname, constants.HVST_DEFAULTS)
526
                    for hvname in enabled_hypervisors)
527

    
528
  # FIXME: disk_state has no default values yet
529
  if disk_state:
530
    for storage, ds_data in disk_state.items():
531
      if storage not in constants.DS_VALID_TYPES:
532
        raise errors.OpPrereqError("Invalid storage type in disk state: %s" %
533
                                   storage, errors.ECODE_INVAL)
534
      for ds_name, state in ds_data.items():
535
        utils.ForceDictType(state, constants.DSS_PARAMETER_TYPES)
536
        ds_data[ds_name] = objects.Cluster.SimpleFillDiskState(state)
537

    
538
  # hvparams is a mapping of hypervisor->hvparams dict
539
  for hv_name, hv_params in hvparams.iteritems():
540
    utils.ForceDictType(hv_params, constants.HVS_PARAMETER_TYPES)
541
    hv_class = hypervisor.GetHypervisor(hv_name)
542
    hv_class.CheckParameterSyntax(hv_params)
543

    
544
  # diskparams is a mapping of disk-template->diskparams dict
545
  for template, dt_params in diskparams.items():
546
    param_keys = set(dt_params.keys())
547
    default_param_keys = set(constants.DISK_DT_DEFAULTS[template].keys())
548
    if not (param_keys <= default_param_keys):
549
      unknown_params = param_keys - default_param_keys
550
      raise errors.OpPrereqError("Invalid parameters for disk template %s:"
551
                                 " %s" % (template,
552
                                          utils.CommaJoin(unknown_params)),
553
                                 errors.ECODE_INVAL)
554
    utils.ForceDictType(dt_params, constants.DISK_DT_TYPES)
555
  try:
556
    utils.VerifyDictOptions(diskparams, constants.DISK_DT_DEFAULTS)
557
  except errors.OpPrereqError, err:
558
    raise errors.OpPrereqError("While verify diskparam options: %s" % err,
559
                               errors.ECODE_INVAL)
560

    
561
  # set up ssh config and /etc/hosts
562
  sshline = utils.ReadFile(pathutils.SSH_HOST_RSA_PUB)
563
  sshkey = sshline.split(" ")[1]
564

    
565
  if modify_etc_hosts:
566
    utils.AddHostToEtcHosts(hostname.name, hostname.ip)
567

    
568
  if modify_ssh_setup:
569
    _InitSSHSetup()
570

    
571
  if default_iallocator is not None:
572
    alloc_script = utils.FindFile(default_iallocator,
573
                                  constants.IALLOCATOR_SEARCH_PATH,
574
                                  os.path.isfile)
575
    if alloc_script is None:
576
      raise errors.OpPrereqError("Invalid default iallocator script '%s'"
577
                                 " specified" % default_iallocator,
578
                                 errors.ECODE_INVAL)
579
  elif constants.HTOOLS:
580
    # htools was enabled at build-time, we default to it
581
    if utils.FindFile(constants.IALLOC_HAIL,
582
                      constants.IALLOCATOR_SEARCH_PATH,
583
                      os.path.isfile):
584
      default_iallocator = constants.IALLOC_HAIL
585

    
586
  now = time.time()
587

    
588
  # init of cluster config file
589
  cluster_config = objects.Cluster(
590
    serial_no=1,
591
    rsahostkeypub=sshkey,
592
    highest_used_port=(constants.FIRST_DRBD_PORT - 1),
593
    mac_prefix=mac_prefix,
594
    volume_group_name=vg_name,
595
    tcpudp_port_pool=set(),
596
    master_node=hostname.name,
597
    master_ip=clustername.ip,
598
    master_netmask=master_netmask,
599
    master_netdev=master_netdev,
600
    cluster_name=clustername.name,
601
    file_storage_dir=file_storage_dir,
602
    shared_file_storage_dir=shared_file_storage_dir,
603
    enabled_hypervisors=enabled_hypervisors,
604
    beparams={constants.PP_DEFAULT: beparams},
605
    nicparams={constants.PP_DEFAULT: nicparams},
606
    ndparams=ndparams,
607
    hvparams=hvparams,
608
    diskparams=diskparams,
609
    candidate_pool_size=candidate_pool_size,
610
    modify_etc_hosts=modify_etc_hosts,
611
    modify_ssh_setup=modify_ssh_setup,
612
    uid_pool=uid_pool,
613
    ctime=now,
614
    mtime=now,
615
    maintain_node_health=maintain_node_health,
616
    drbd_usermode_helper=drbd_helper,
617
    default_iallocator=default_iallocator,
618
    primary_ip_family=ipcls.family,
619
    prealloc_wipe_disks=prealloc_wipe_disks,
620
    use_external_mip_script=use_external_mip_script,
621
    ipolicy=full_ipolicy,
622
    hv_state_static=hv_state,
623
    disk_state_static=disk_state,
624
    )
625
  master_node_config = objects.Node(name=hostname.name,
626
                                    primary_ip=hostname.ip,
627
                                    secondary_ip=secondary_ip,
628
                                    serial_no=1,
629
                                    master_candidate=True,
630
                                    offline=False, drained=False,
631
                                    ctime=now, mtime=now,
632
                                    )
633
  InitConfig(constants.CONFIG_VERSION, cluster_config, master_node_config)
634
  cfg = config.ConfigWriter(offline=True)
635
  ssh.WriteKnownHostsFile(cfg, pathutils.SSH_KNOWN_HOSTS_FILE)
636
  cfg.Update(cfg.GetClusterInfo(), logging.error)
637
  ssconf.WriteSsconfFiles(cfg.GetSsconfValues())
638

    
639
  # set up the inter-node password and certificate
640
  _InitGanetiServerSetup(hostname.name)
641

    
642
  logging.debug("Starting daemons")
643
  result = utils.RunCmd([pathutils.DAEMON_UTIL, "start-all"])
644
  if result.failed:
645
    raise errors.OpExecError("Could not start daemons, command %s"
646
                             " had exitcode %s and error %s" %
647
                             (result.cmd, result.exit_code, result.output))
648

    
649
  _WaitForMasterDaemon()
650

    
651

    
652
def InitConfig(version, cluster_config, master_node_config,
653
               cfg_file=pathutils.CLUSTER_CONF_FILE):
654
  """Create the initial cluster configuration.
655

656
  It will contain the current node, which will also be the master
657
  node, and no instances.
658

659
  @type version: int
660
  @param version: configuration version
661
  @type cluster_config: L{objects.Cluster}
662
  @param cluster_config: cluster configuration
663
  @type master_node_config: L{objects.Node}
664
  @param master_node_config: master node configuration
665
  @type cfg_file: string
666
  @param cfg_file: configuration file path
667

668
  """
669
  uuid_generator = config.TemporaryReservationManager()
670
  cluster_config.uuid = uuid_generator.Generate([], utils.NewUUID,
671
                                                _INITCONF_ECID)
672
  master_node_config.uuid = uuid_generator.Generate([], utils.NewUUID,
673
                                                    _INITCONF_ECID)
674
  nodes = {
675
    master_node_config.name: master_node_config,
676
    }
677
  default_nodegroup = objects.NodeGroup(
678
    uuid=uuid_generator.Generate([], utils.NewUUID, _INITCONF_ECID),
679
    name=constants.INITIAL_NODE_GROUP_NAME,
680
    members=[master_node_config.name],
681
    diskparams={},
682
    )
683
  nodegroups = {
684
    default_nodegroup.uuid: default_nodegroup,
685
    }
686
  now = time.time()
687
  config_data = objects.ConfigData(version=version,
688
                                   cluster=cluster_config,
689
                                   nodegroups=nodegroups,
690
                                   nodes=nodes,
691
                                   instances={},
692
                                   networks={},
693
                                   serial_no=1,
694
                                   ctime=now, mtime=now)
695
  utils.WriteFile(cfg_file,
696
                  data=serializer.Dump(config_data.ToDict()),
697
                  mode=0600)
698

    
699

    
700
def FinalizeClusterDestroy(master):
701
  """Execute the last steps of cluster destroy
702

703
  This function shuts down all the daemons, completing the destroy
704
  begun in cmdlib.LUDestroyOpcode.
705

706
  """
707
  cfg = config.ConfigWriter()
708
  modify_ssh_setup = cfg.GetClusterInfo().modify_ssh_setup
709
  runner = rpc.BootstrapRunner()
710

    
711
  master_params = cfg.GetMasterNetworkParameters()
712
  master_params.name = master
713
  ems = cfg.GetUseExternalMipScript()
714
  result = runner.call_node_deactivate_master_ip(master_params.name,
715
                                                 master_params, ems)
716

    
717
  msg = result.fail_msg
718
  if msg:
719
    logging.warning("Could not disable the master IP: %s", msg)
720

    
721
  result = runner.call_node_stop_master(master)
722
  msg = result.fail_msg
723
  if msg:
724
    logging.warning("Could not disable the master role: %s", msg)
725

    
726
  result = runner.call_node_leave_cluster(master, modify_ssh_setup)
727
  msg = result.fail_msg
728
  if msg:
729
    logging.warning("Could not shutdown the node daemon and cleanup"
730
                    " the node: %s", msg)
731

    
732

    
733
def SetupNodeDaemon(opts, cluster_name, node):
734
  """Add a node to the cluster.
735

736
  This function must be called before the actual opcode, and will ssh
737
  to the remote node, copy the needed files, and start ganeti-noded,
738
  allowing the master to do the rest via normal rpc calls.
739

740
  @param cluster_name: the cluster name
741
  @param node: the name of the new node
742

743
  """
744
  data = {
745
    constants.NDS_CLUSTER_NAME: cluster_name,
746
    constants.NDS_NODE_DAEMON_CERTIFICATE:
747
      utils.ReadFile(pathutils.NODED_CERT_FILE),
748
    constants.NDS_SSCONF: ssconf.SimpleStore().ReadAll(),
749
    constants.NDS_START_NODE_DAEMON: True,
750
    }
751

    
752
  RunNodeSetupCmd(cluster_name, node, pathutils.NODE_DAEMON_SETUP,
753
                  opts.debug, opts.verbose,
754
                  True, opts.ssh_key_check, opts.ssh_key_check, data)
755

    
756
  _WaitForNodeDaemon(node)
757

    
758

    
759
def MasterFailover(no_voting=False):
760
  """Failover the master node.
761

762
  This checks that we are not already the master, and will cause the
763
  current master to cease being master, and the non-master to become
764
  new master.
765

766
  @type no_voting: boolean
767
  @param no_voting: force the operation without remote nodes agreement
768
                      (dangerous)
769

770
  """
771
  sstore = ssconf.SimpleStore()
772

    
773
  old_master, new_master = ssconf.GetMasterAndMyself(sstore)
774
  node_list = sstore.GetNodeList()
775
  mc_list = sstore.GetMasterCandidates()
776

    
777
  if old_master == new_master:
778
    raise errors.OpPrereqError("This commands must be run on the node"
779
                               " where you want the new master to be."
780
                               " %s is already the master" %
781
                               old_master, errors.ECODE_INVAL)
782

    
783
  if new_master not in mc_list:
784
    mc_no_master = [name for name in mc_list if name != old_master]
785
    raise errors.OpPrereqError("This node is not among the nodes marked"
786
                               " as master candidates. Only these nodes"
787
                               " can become masters. Current list of"
788
                               " master candidates is:\n"
789
                               "%s" % ("\n".join(mc_no_master)),
790
                               errors.ECODE_STATE)
791

    
792
  if not no_voting:
793
    vote_list = GatherMasterVotes(node_list)
794

    
795
    if vote_list:
796
      voted_master = vote_list[0][0]
797
      if voted_master is None:
798
        raise errors.OpPrereqError("Cluster is inconsistent, most nodes did"
799
                                   " not respond.", errors.ECODE_ENVIRON)
800
      elif voted_master != old_master:
801
        raise errors.OpPrereqError("I have a wrong configuration, I believe"
802
                                   " the master is %s but the other nodes"
803
                                   " voted %s. Please resync the configuration"
804
                                   " of this node." %
805
                                   (old_master, voted_master),
806
                                   errors.ECODE_STATE)
807
  # end checks
808

    
809
  rcode = 0
810

    
811
  logging.info("Setting master to %s, old master: %s", new_master, old_master)
812

    
813
  try:
814
    # instantiate a real config writer, as we now know we have the
815
    # configuration data
816
    cfg = config.ConfigWriter(accept_foreign=True)
817

    
818
    cluster_info = cfg.GetClusterInfo()
819
    cluster_info.master_node = new_master
820
    # this will also regenerate the ssconf files, since we updated the
821
    # cluster info
822
    cfg.Update(cluster_info, logging.error)
823
  except errors.ConfigurationError, err:
824
    logging.error("Error while trying to set the new master: %s",
825
                  str(err))
826
    return 1
827

    
828
  # if cfg.Update worked, then it means the old master daemon won't be
829
  # able now to write its own config file (we rely on locking in both
830
  # backend.UploadFile() and ConfigWriter._Write(); hence the next
831
  # step is to kill the old master
832

    
833
  logging.info("Stopping the master daemon on node %s", old_master)
834

    
835
  runner = rpc.BootstrapRunner()
836
  master_params = cfg.GetMasterNetworkParameters()
837
  master_params.name = old_master
838
  ems = cfg.GetUseExternalMipScript()
839
  result = runner.call_node_deactivate_master_ip(master_params.name,
840
                                                 master_params, ems)
841

    
842
  msg = result.fail_msg
843
  if msg:
844
    logging.warning("Could not disable the master IP: %s", msg)
845

    
846
  result = runner.call_node_stop_master(old_master)
847
  msg = result.fail_msg
848
  if msg:
849
    logging.error("Could not disable the master role on the old master"
850
                  " %s, please disable manually: %s", old_master, msg)
851

    
852
  logging.info("Checking master IP non-reachability...")
853

    
854
  master_ip = sstore.GetMasterIP()
855
  total_timeout = 30
856

    
857
  # Here we have a phase where no master should be running
858
  def _check_ip():
859
    if netutils.TcpPing(master_ip, constants.DEFAULT_NODED_PORT):
860
      raise utils.RetryAgain()
861

    
862
  try:
863
    utils.Retry(_check_ip, (1, 1.5, 5), total_timeout)
864
  except utils.RetryTimeout:
865
    logging.warning("The master IP is still reachable after %s seconds,"
866
                    " continuing but activating the master on the current"
867
                    " node will probably fail", total_timeout)
868

    
869
  if jstore.CheckDrainFlag():
870
    logging.info("Undraining job queue")
871
    jstore.SetDrainFlag(False)
872

    
873
  logging.info("Starting the master daemons on the new master")
874

    
875
  result = rpc.BootstrapRunner().call_node_start_master_daemons(new_master,
876
                                                                no_voting)
877
  msg = result.fail_msg
878
  if msg:
879
    logging.error("Could not start the master role on the new master"
880
                  " %s, please check: %s", new_master, msg)
881
    rcode = 1
882

    
883
  logging.info("Master failed over from %s to %s", old_master, new_master)
884
  return rcode
885

    
886

    
887
def GetMaster():
888
  """Returns the current master node.
889

890
  This is a separate function in bootstrap since it's needed by
891
  gnt-cluster, and instead of importing directly ssconf, it's better
892
  to abstract it in bootstrap, where we do use ssconf in other
893
  functions too.
894

895
  """
896
  sstore = ssconf.SimpleStore()
897

    
898
  old_master, _ = ssconf.GetMasterAndMyself(sstore)
899

    
900
  return old_master
901

    
902

    
903
def GatherMasterVotes(node_list):
904
  """Check the agreement on who is the master.
905

906
  This function will return a list of (node, number of votes), ordered
907
  by the number of votes. Errors will be denoted by the key 'None'.
908

909
  Note that the sum of votes is the number of nodes this machine
910
  knows, whereas the number of entries in the list could be different
911
  (if some nodes vote for another master).
912

913
  We remove ourselves from the list since we know that (bugs aside)
914
  since we use the same source for configuration information for both
915
  backend and boostrap, we'll always vote for ourselves.
916

917
  @type node_list: list
918
  @param node_list: the list of nodes to query for master info; the current
919
      node will be removed if it is in the list
920
  @rtype: list
921
  @return: list of (node, votes)
922

923
  """
924
  myself = netutils.Hostname.GetSysName()
925
  try:
926
    node_list.remove(myself)
927
  except ValueError:
928
    pass
929
  if not node_list:
930
    # no nodes left (eventually after removing myself)
931
    return []
932
  results = rpc.BootstrapRunner().call_master_info(node_list)
933
  if not isinstance(results, dict):
934
    # this should not happen (unless internal error in rpc)
935
    logging.critical("Can't complete rpc call, aborting master startup")
936
    return [(None, len(node_list))]
937
  votes = {}
938
  for node in results:
939
    nres = results[node]
940
    data = nres.payload
941
    msg = nres.fail_msg
942
    fail = False
943
    if msg:
944
      logging.warning("Error contacting node %s: %s", node, msg)
945
      fail = True
946
    # for now we accept both length 3, 4 and 5 (data[3] is primary ip version
947
    # and data[4] is the master netmask)
948
    elif not isinstance(data, (tuple, list)) or len(data) < 3:
949
      logging.warning("Invalid data received from node %s: %s", node, data)
950
      fail = True
951
    if fail:
952
      if None not in votes:
953
        votes[None] = 0
954
      votes[None] += 1
955
      continue
956
    master_node = data[2]
957
    if master_node not in votes:
958
      votes[master_node] = 0
959
    votes[master_node] += 1
960

    
961
  vote_list = [v for v in votes.items()]
962
  # sort first on number of votes then on name, since we want None
963
  # sorted later if we have the half of the nodes not responding, and
964
  # half voting all for the same master
965
  vote_list.sort(key=lambda x: (x[1], x[0]), reverse=True)
966

    
967
  return vote_list