Statistics
| Branch: | Tag: | Revision:

root / lib / client / gnt_cluster.py @ bf93ae69

History | View | Annotate | Download (42.6 kB)

1
#
2
#
3

    
4
# Copyright (C) 2006, 2007, 2010, 2011 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
"""Cluster related commands"""
22

    
23
# pylint: disable-msg=W0401,W0613,W0614,C0103
24
# W0401: Wildcard import ganeti.cli
25
# W0613: Unused argument, since all functions follow the same API
26
# W0614: Unused import %s from wildcard import (since we need cli)
27
# C0103: Invalid name gnt-cluster
28

    
29
import os.path
30
import time
31
import OpenSSL
32
import itertools
33

    
34
from ganeti.cli import *
35
from ganeti import opcodes
36
from ganeti import constants
37
from ganeti import errors
38
from ganeti import utils
39
from ganeti import bootstrap
40
from ganeti import ssh
41
from ganeti import objects
42
from ganeti import uidpool
43
from ganeti import compat
44
from ganeti import netutils
45

    
46

    
47
ON_OPT = cli_option("--on", default=False,
48
                    action="store_true", dest="on",
49
                    help="Recover from an EPO")
50

    
51
GROUPS_OPT = cli_option("--groups", default=False,
52
                    action="store_true", dest="groups",
53
                    help="Arguments are node groups instead of nodes")
54

    
55
_EPO_PING_INTERVAL = 30 # 30 seconds between pings
56
_EPO_PING_TIMEOUT = 1 # 1 second
57
_EPO_REACHABLE_TIMEOUT = 15 * 60 # 15 minutes
58

    
59

    
60
@UsesRPC
61
def InitCluster(opts, args):
62
  """Initialize the cluster.
63

64
  @param opts: the command line options selected by the user
65
  @type args: list
66
  @param args: should contain only one element, the desired
67
      cluster name
68
  @rtype: int
69
  @return: the desired exit code
70

71
  """
72
  if not opts.lvm_storage and opts.vg_name:
73
    ToStderr("Options --no-lvm-storage and --vg-name conflict.")
74
    return 1
75

    
76
  vg_name = opts.vg_name
77
  if opts.lvm_storage and not opts.vg_name:
78
    vg_name = constants.DEFAULT_VG
79

    
80
  if not opts.drbd_storage and opts.drbd_helper:
81
    ToStderr("Options --no-drbd-storage and --drbd-usermode-helper conflict.")
82
    return 1
83

    
84
  drbd_helper = opts.drbd_helper
85
  if opts.drbd_storage and not opts.drbd_helper:
86
    drbd_helper = constants.DEFAULT_DRBD_HELPER
87

    
88
  master_netdev = opts.master_netdev
89
  if master_netdev is None:
90
    master_netdev = constants.DEFAULT_BRIDGE
91

    
92
  hvlist = opts.enabled_hypervisors
93
  if hvlist is None:
94
    hvlist = constants.DEFAULT_ENABLED_HYPERVISOR
95
  hvlist = hvlist.split(",")
96

    
97
  hvparams = dict(opts.hvparams)
98
  beparams = opts.beparams
99
  nicparams = opts.nicparams
100

    
101
  # prepare beparams dict
102
  beparams = objects.FillDict(constants.BEC_DEFAULTS, beparams)
103
  utils.ForceDictType(beparams, constants.BES_PARAMETER_TYPES)
104

    
105
  # prepare nicparams dict
106
  nicparams = objects.FillDict(constants.NICC_DEFAULTS, nicparams)
107
  utils.ForceDictType(nicparams, constants.NICS_PARAMETER_TYPES)
108

    
109
  # prepare ndparams dict
110
  if opts.ndparams is None:
111
    ndparams = dict(constants.NDC_DEFAULTS)
112
  else:
113
    ndparams = objects.FillDict(constants.NDC_DEFAULTS, opts.ndparams)
114
    utils.ForceDictType(ndparams, constants.NDS_PARAMETER_TYPES)
115

    
116
  # prepare hvparams dict
117
  for hv in constants.HYPER_TYPES:
118
    if hv not in hvparams:
119
      hvparams[hv] = {}
120
    hvparams[hv] = objects.FillDict(constants.HVC_DEFAULTS[hv], hvparams[hv])
121
    utils.ForceDictType(hvparams[hv], constants.HVS_PARAMETER_TYPES)
122

    
123
  if opts.candidate_pool_size is None:
124
    opts.candidate_pool_size = constants.MASTER_POOL_SIZE_DEFAULT
125

    
126
  if opts.mac_prefix is None:
127
    opts.mac_prefix = constants.DEFAULT_MAC_PREFIX
128

    
129
  uid_pool = opts.uid_pool
130
  if uid_pool is not None:
131
    uid_pool = uidpool.ParseUidPool(uid_pool)
132

    
133
  if opts.prealloc_wipe_disks is None:
134
    opts.prealloc_wipe_disks = False
135

    
136
  try:
137
    primary_ip_version = int(opts.primary_ip_version)
138
  except (ValueError, TypeError), err:
139
    ToStderr("Invalid primary ip version value: %s" % str(err))
140
    return 1
141

    
142
  bootstrap.InitCluster(cluster_name=args[0],
143
                        secondary_ip=opts.secondary_ip,
144
                        vg_name=vg_name,
145
                        mac_prefix=opts.mac_prefix,
146
                        master_netdev=master_netdev,
147
                        file_storage_dir=opts.file_storage_dir,
148
                        shared_file_storage_dir=opts.shared_file_storage_dir,
149
                        enabled_hypervisors=hvlist,
150
                        hvparams=hvparams,
151
                        beparams=beparams,
152
                        nicparams=nicparams,
153
                        ndparams=ndparams,
154
                        candidate_pool_size=opts.candidate_pool_size,
155
                        modify_etc_hosts=opts.modify_etc_hosts,
156
                        modify_ssh_setup=opts.modify_ssh_setup,
157
                        maintain_node_health=opts.maintain_node_health,
158
                        drbd_helper=drbd_helper,
159
                        uid_pool=uid_pool,
160
                        default_iallocator=opts.default_iallocator,
161
                        primary_ip_version=primary_ip_version,
162
                        prealloc_wipe_disks=opts.prealloc_wipe_disks,
163
                        )
164
  op = opcodes.OpClusterPostInit()
165
  SubmitOpCode(op, opts=opts)
166
  return 0
167

    
168

    
169
@UsesRPC
170
def DestroyCluster(opts, args):
171
  """Destroy the cluster.
172

173
  @param opts: the command line options selected by the user
174
  @type args: list
175
  @param args: should be an empty list
176
  @rtype: int
177
  @return: the desired exit code
178

179
  """
180
  if not opts.yes_do_it:
181
    ToStderr("Destroying a cluster is irreversible. If you really want"
182
             " destroy this cluster, supply the --yes-do-it option.")
183
    return 1
184

    
185
  op = opcodes.OpClusterDestroy()
186
  master = SubmitOpCode(op, opts=opts)
187
  # if we reached this, the opcode didn't fail; we can proceed to
188
  # shutdown all the daemons
189
  bootstrap.FinalizeClusterDestroy(master)
190
  return 0
191

    
192

    
193
def RenameCluster(opts, args):
194
  """Rename the cluster.
195

196
  @param opts: the command line options selected by the user
197
  @type args: list
198
  @param args: should contain only one element, the new cluster name
199
  @rtype: int
200
  @return: the desired exit code
201

202
  """
203
  cl = GetClient()
204

    
205
  (cluster_name, ) = cl.QueryConfigValues(["cluster_name"])
206

    
207
  new_name = args[0]
208
  if not opts.force:
209
    usertext = ("This will rename the cluster from '%s' to '%s'. If you are"
210
                " connected over the network to the cluster name, the"
211
                " operation is very dangerous as the IP address will be"
212
                " removed from the node and the change may not go through."
213
                " Continue?") % (cluster_name, new_name)
214
    if not AskUser(usertext):
215
      return 1
216

    
217
  op = opcodes.OpClusterRename(name=new_name)
218
  result = SubmitOpCode(op, opts=opts, cl=cl)
219

    
220
  if result:
221
    ToStdout("Cluster renamed from '%s' to '%s'", cluster_name, result)
222

    
223
  return 0
224

    
225

    
226
def RedistributeConfig(opts, args):
227
  """Forces push of the cluster configuration.
228

229
  @param opts: the command line options selected by the user
230
  @type args: list
231
  @param args: empty list
232
  @rtype: int
233
  @return: the desired exit code
234

235
  """
236
  op = opcodes.OpClusterRedistConf()
237
  SubmitOrSend(op, opts)
238
  return 0
239

    
240

    
241
def ShowClusterVersion(opts, args):
242
  """Write version of ganeti software to the standard output.
243

244
  @param opts: the command line options selected by the user
245
  @type args: list
246
  @param args: should be an empty list
247
  @rtype: int
248
  @return: the desired exit code
249

250
  """
251
  cl = GetClient()
252
  result = cl.QueryClusterInfo()
253
  ToStdout("Software version: %s", result["software_version"])
254
  ToStdout("Internode protocol: %s", result["protocol_version"])
255
  ToStdout("Configuration format: %s", result["config_version"])
256
  ToStdout("OS api version: %s", result["os_api_version"])
257
  ToStdout("Export interface: %s", result["export_version"])
258
  return 0
259

    
260

    
261
def ShowClusterMaster(opts, args):
262
  """Write name of master node to the standard output.
263

264
  @param opts: the command line options selected by the user
265
  @type args: list
266
  @param args: should be an empty list
267
  @rtype: int
268
  @return: the desired exit code
269

270
  """
271
  master = bootstrap.GetMaster()
272
  ToStdout(master)
273
  return 0
274

    
275

    
276
def _PrintGroupedParams(paramsdict, level=1, roman=False):
277
  """Print Grouped parameters (be, nic, disk) by group.
278

279
  @type paramsdict: dict of dicts
280
  @param paramsdict: {group: {param: value, ...}, ...}
281
  @type level: int
282
  @param level: Level of indention
283

284
  """
285
  indent = "  " * level
286
  for item, val in sorted(paramsdict.items()):
287
    if isinstance(val, dict):
288
      ToStdout("%s- %s:", indent, item)
289
      _PrintGroupedParams(val, level=level + 1, roman=roman)
290
    elif roman and isinstance(val, int):
291
      ToStdout("%s  %s: %s", indent, item, compat.TryToRoman(val))
292
    else:
293
      ToStdout("%s  %s: %s", indent, item, val)
294

    
295

    
296
def ShowClusterConfig(opts, args):
297
  """Shows cluster information.
298

299
  @param opts: the command line options selected by the user
300
  @type args: list
301
  @param args: should be an empty list
302
  @rtype: int
303
  @return: the desired exit code
304

305
  """
306
  cl = GetClient()
307
  result = cl.QueryClusterInfo()
308

    
309
  ToStdout("Cluster name: %s", result["name"])
310
  ToStdout("Cluster UUID: %s", result["uuid"])
311

    
312
  ToStdout("Creation time: %s", utils.FormatTime(result["ctime"]))
313
  ToStdout("Modification time: %s", utils.FormatTime(result["mtime"]))
314

    
315
  ToStdout("Master node: %s", result["master"])
316

    
317
  ToStdout("Architecture (this node): %s (%s)",
318
           result["architecture"][0], result["architecture"][1])
319

    
320
  if result["tags"]:
321
    tags = utils.CommaJoin(utils.NiceSort(result["tags"]))
322
  else:
323
    tags = "(none)"
324

    
325
  ToStdout("Tags: %s", tags)
326

    
327
  ToStdout("Default hypervisor: %s", result["default_hypervisor"])
328
  ToStdout("Enabled hypervisors: %s",
329
           utils.CommaJoin(result["enabled_hypervisors"]))
330

    
331
  ToStdout("Hypervisor parameters:")
332
  _PrintGroupedParams(result["hvparams"])
333

    
334
  ToStdout("OS-specific hypervisor parameters:")
335
  _PrintGroupedParams(result["os_hvp"])
336

    
337
  ToStdout("OS parameters:")
338
  _PrintGroupedParams(result["osparams"])
339

    
340
  ToStdout("Hidden OSes: %s", utils.CommaJoin(result["hidden_os"]))
341
  ToStdout("Blacklisted OSes: %s", utils.CommaJoin(result["blacklisted_os"]))
342

    
343
  ToStdout("Cluster parameters:")
344
  ToStdout("  - candidate pool size: %s",
345
            compat.TryToRoman(result["candidate_pool_size"],
346
                              convert=opts.roman_integers))
347
  ToStdout("  - master netdev: %s", result["master_netdev"])
348
  ToStdout("  - lvm volume group: %s", result["volume_group_name"])
349
  if result["reserved_lvs"]:
350
    reserved_lvs = utils.CommaJoin(result["reserved_lvs"])
351
  else:
352
    reserved_lvs = "(none)"
353
  ToStdout("  - lvm reserved volumes: %s", reserved_lvs)
354
  ToStdout("  - drbd usermode helper: %s", result["drbd_usermode_helper"])
355
  ToStdout("  - file storage path: %s", result["file_storage_dir"])
356
  ToStdout("  - shared file storage path: %s",
357
           result["shared_file_storage_dir"])
358
  ToStdout("  - maintenance of node health: %s",
359
           result["maintain_node_health"])
360
  ToStdout("  - uid pool: %s",
361
            uidpool.FormatUidPool(result["uid_pool"],
362
                                  roman=opts.roman_integers))
363
  ToStdout("  - default instance allocator: %s", result["default_iallocator"])
364
  ToStdout("  - primary ip version: %d", result["primary_ip_version"])
365
  ToStdout("  - preallocation wipe disks: %s", result["prealloc_wipe_disks"])
366

    
367
  ToStdout("Default node parameters:")
368
  _PrintGroupedParams(result["ndparams"], roman=opts.roman_integers)
369

    
370
  ToStdout("Default instance parameters:")
371
  _PrintGroupedParams(result["beparams"], roman=opts.roman_integers)
372

    
373
  ToStdout("Default nic parameters:")
374
  _PrintGroupedParams(result["nicparams"], roman=opts.roman_integers)
375

    
376
  return 0
377

    
378

    
379
def ClusterCopyFile(opts, args):
380
  """Copy a file from master to some nodes.
381

382
  @param opts: the command line options selected by the user
383
  @type args: list
384
  @param args: should contain only one element, the path of
385
      the file to be copied
386
  @rtype: int
387
  @return: the desired exit code
388

389
  """
390
  filename = args[0]
391
  if not os.path.exists(filename):
392
    raise errors.OpPrereqError("No such filename '%s'" % filename,
393
                               errors.ECODE_INVAL)
394

    
395
  cl = GetClient()
396

    
397
  cluster_name = cl.QueryConfigValues(["cluster_name"])[0]
398

    
399
  results = GetOnlineNodes(nodes=opts.nodes, cl=cl, filter_master=True,
400
                           secondary_ips=opts.use_replication_network)
401

    
402
  srun = ssh.SshRunner(cluster_name=cluster_name)
403
  for node in results:
404
    if not srun.CopyFileToNode(node, filename):
405
      ToStderr("Copy of file %s to node %s failed", filename, node)
406

    
407
  return 0
408

    
409

    
410
def RunClusterCommand(opts, args):
411
  """Run a command on some nodes.
412

413
  @param opts: the command line options selected by the user
414
  @type args: list
415
  @param args: should contain the command to be run and its arguments
416
  @rtype: int
417
  @return: the desired exit code
418

419
  """
420
  cl = GetClient()
421

    
422
  command = " ".join(args)
423

    
424
  nodes = GetOnlineNodes(nodes=opts.nodes, cl=cl)
425

    
426
  cluster_name, master_node = cl.QueryConfigValues(["cluster_name",
427
                                                    "master_node"])
428

    
429
  srun = ssh.SshRunner(cluster_name=cluster_name)
430

    
431
  # Make sure master node is at list end
432
  if master_node in nodes:
433
    nodes.remove(master_node)
434
    nodes.append(master_node)
435

    
436
  for name in nodes:
437
    result = srun.Run(name, "root", command)
438
    ToStdout("------------------------------------------------")
439
    ToStdout("node: %s", name)
440
    ToStdout("%s", result.output)
441
    ToStdout("return code = %s", result.exit_code)
442

    
443
  return 0
444

    
445

    
446
def VerifyCluster(opts, args):
447
  """Verify integrity of cluster, performing various test on nodes.
448

449
  @param opts: the command line options selected by the user
450
  @type args: list
451
  @param args: should be an empty list
452
  @rtype: int
453
  @return: the desired exit code
454

455
  """
456
  simulate = opts.simulate_errors
457
  skip_checks = []
458

    
459
  # Verify cluster config.
460
  op = opcodes.OpClusterVerifyConfig(verbose=opts.verbose,
461
                                     error_codes=opts.error_codes,
462
                                     debug_simulate_errors=simulate)
463

    
464
  success, all_groups = SubmitOpCode(op, opts=opts)
465

    
466
  if opts.skip_nplusone_mem:
467
    skip_checks.append(constants.VERIFY_NPLUSONE_MEM)
468

    
469
  jex = JobExecutor(opts=opts, verbose=False)
470

    
471
  for group in all_groups:
472
    op = opcodes.OpClusterVerifyGroup(group_name=group,
473
                                      skip_checks=skip_checks,
474
                                      verbose=opts.verbose,
475
                                      error_codes=opts.error_codes,
476
                                      debug_simulate_errors=simulate)
477
    jex.QueueJob('group ' + group, op)
478

    
479
  results = jex.GetResults()
480
  success &= compat.all(r[1][0] for r in results)
481

    
482
  return (not success and 1 or 0)
483

    
484

    
485
def VerifyDisks(opts, args):
486
  """Verify integrity of cluster disks.
487

488
  @param opts: the command line options selected by the user
489
  @type args: list
490
  @param args: should be an empty list
491
  @rtype: int
492
  @return: the desired exit code
493

494
  """
495
  cl = GetClient()
496

    
497
  op = opcodes.OpClusterVerifyDisks()
498
  result = SubmitOpCode(op, opts=opts, cl=cl)
499
  if not isinstance(result, (list, tuple)) or len(result) != 3:
500
    raise errors.ProgrammerError("Unknown result type for OpClusterVerifyDisks")
501

    
502
  bad_nodes, instances, missing = result
503

    
504
  retcode = constants.EXIT_SUCCESS
505

    
506
  if bad_nodes:
507
    for node, text in bad_nodes.items():
508
      ToStdout("Error gathering data on node %s: %s",
509
               node, utils.SafeEncode(text[-400:]))
510
      retcode |= 1
511
      ToStdout("You need to fix these nodes first before fixing instances")
512

    
513
  if instances:
514
    for iname in instances:
515
      if iname in missing:
516
        continue
517
      op = opcodes.OpInstanceActivateDisks(instance_name=iname)
518
      try:
519
        ToStdout("Activating disks for instance '%s'", iname)
520
        SubmitOpCode(op, opts=opts, cl=cl)
521
      except errors.GenericError, err:
522
        nret, msg = FormatError(err)
523
        retcode |= nret
524
        ToStderr("Error activating disks for instance %s: %s", iname, msg)
525

    
526
  if missing:
527
    for iname, ival in missing.iteritems():
528
      all_missing = compat.all(x[0] in bad_nodes for x in ival)
529
      if all_missing:
530
        ToStdout("Instance %s cannot be verified as it lives on"
531
                 " broken nodes", iname)
532
      else:
533
        ToStdout("Instance %s has missing logical volumes:", iname)
534
        ival.sort()
535
        for node, vol in ival:
536
          if node in bad_nodes:
537
            ToStdout("\tbroken node %s /dev/%s", node, vol)
538
          else:
539
            ToStdout("\t%s /dev/%s", node, vol)
540

    
541
    ToStdout("You need to run replace or recreate disks for all the above"
542
             " instances, if this message persist after fixing nodes.")
543
    retcode |= 1
544

    
545
  return retcode
546

    
547

    
548
def RepairDiskSizes(opts, args):
549
  """Verify sizes of cluster disks.
550

551
  @param opts: the command line options selected by the user
552
  @type args: list
553
  @param args: optional list of instances to restrict check to
554
  @rtype: int
555
  @return: the desired exit code
556

557
  """
558
  op = opcodes.OpClusterRepairDiskSizes(instances=args)
559
  SubmitOpCode(op, opts=opts)
560

    
561

    
562
@UsesRPC
563
def MasterFailover(opts, args):
564
  """Failover the master node.
565

566
  This command, when run on a non-master node, will cause the current
567
  master to cease being master, and the non-master to become new
568
  master.
569

570
  @param opts: the command line options selected by the user
571
  @type args: list
572
  @param args: should be an empty list
573
  @rtype: int
574
  @return: the desired exit code
575

576
  """
577
  if opts.no_voting:
578
    usertext = ("This will perform the failover even if most other nodes"
579
                " are down, or if this node is outdated. This is dangerous"
580
                " as it can lead to a non-consistent cluster. Check the"
581
                " gnt-cluster(8) man page before proceeding. Continue?")
582
    if not AskUser(usertext):
583
      return 1
584

    
585
  return bootstrap.MasterFailover(no_voting=opts.no_voting)
586

    
587

    
588
def MasterPing(opts, args):
589
  """Checks if the master is alive.
590

591
  @param opts: the command line options selected by the user
592
  @type args: list
593
  @param args: should be an empty list
594
  @rtype: int
595
  @return: the desired exit code
596

597
  """
598
  try:
599
    cl = GetClient()
600
    cl.QueryClusterInfo()
601
    return 0
602
  except Exception: # pylint: disable-msg=W0703
603
    return 1
604

    
605

    
606
def SearchTags(opts, args):
607
  """Searches the tags on all the cluster.
608

609
  @param opts: the command line options selected by the user
610
  @type args: list
611
  @param args: should contain only one element, the tag pattern
612
  @rtype: int
613
  @return: the desired exit code
614

615
  """
616
  op = opcodes.OpTagsSearch(pattern=args[0])
617
  result = SubmitOpCode(op, opts=opts)
618
  if not result:
619
    return 1
620
  result = list(result)
621
  result.sort()
622
  for path, tag in result:
623
    ToStdout("%s %s", path, tag)
624

    
625

    
626
def _RenewCrypto(new_cluster_cert, new_rapi_cert, rapi_cert_filename,
627
                 new_confd_hmac_key, new_cds, cds_filename,
628
                 force):
629
  """Renews cluster certificates, keys and secrets.
630

631
  @type new_cluster_cert: bool
632
  @param new_cluster_cert: Whether to generate a new cluster certificate
633
  @type new_rapi_cert: bool
634
  @param new_rapi_cert: Whether to generate a new RAPI certificate
635
  @type rapi_cert_filename: string
636
  @param rapi_cert_filename: Path to file containing new RAPI certificate
637
  @type new_confd_hmac_key: bool
638
  @param new_confd_hmac_key: Whether to generate a new HMAC key
639
  @type new_cds: bool
640
  @param new_cds: Whether to generate a new cluster domain secret
641
  @type cds_filename: string
642
  @param cds_filename: Path to file containing new cluster domain secret
643
  @type force: bool
644
  @param force: Whether to ask user for confirmation
645

646
  """
647
  if new_rapi_cert and rapi_cert_filename:
648
    ToStderr("Only one of the --new-rapi-certficate and --rapi-certificate"
649
             " options can be specified at the same time.")
650
    return 1
651

    
652
  if new_cds and cds_filename:
653
    ToStderr("Only one of the --new-cluster-domain-secret and"
654
             " --cluster-domain-secret options can be specified at"
655
             " the same time.")
656
    return 1
657

    
658
  if rapi_cert_filename:
659
    # Read and verify new certificate
660
    try:
661
      rapi_cert_pem = utils.ReadFile(rapi_cert_filename)
662

    
663
      OpenSSL.crypto.load_certificate(OpenSSL.crypto.FILETYPE_PEM,
664
                                      rapi_cert_pem)
665
    except Exception, err: # pylint: disable-msg=W0703
666
      ToStderr("Can't load new RAPI certificate from %s: %s" %
667
               (rapi_cert_filename, str(err)))
668
      return 1
669

    
670
    try:
671
      OpenSSL.crypto.load_privatekey(OpenSSL.crypto.FILETYPE_PEM, rapi_cert_pem)
672
    except Exception, err: # pylint: disable-msg=W0703
673
      ToStderr("Can't load new RAPI private key from %s: %s" %
674
               (rapi_cert_filename, str(err)))
675
      return 1
676

    
677
  else:
678
    rapi_cert_pem = None
679

    
680
  if cds_filename:
681
    try:
682
      cds = utils.ReadFile(cds_filename)
683
    except Exception, err: # pylint: disable-msg=W0703
684
      ToStderr("Can't load new cluster domain secret from %s: %s" %
685
               (cds_filename, str(err)))
686
      return 1
687
  else:
688
    cds = None
689

    
690
  if not force:
691
    usertext = ("This requires all daemons on all nodes to be restarted and"
692
                " may take some time. Continue?")
693
    if not AskUser(usertext):
694
      return 1
695

    
696
  def _RenewCryptoInner(ctx):
697
    ctx.feedback_fn("Updating certificates and keys")
698
    bootstrap.GenerateClusterCrypto(new_cluster_cert, new_rapi_cert,
699
                                    new_confd_hmac_key,
700
                                    new_cds,
701
                                    rapi_cert_pem=rapi_cert_pem,
702
                                    cds=cds)
703

    
704
    files_to_copy = []
705

    
706
    if new_cluster_cert:
707
      files_to_copy.append(constants.NODED_CERT_FILE)
708

    
709
    if new_rapi_cert or rapi_cert_pem:
710
      files_to_copy.append(constants.RAPI_CERT_FILE)
711

    
712
    if new_confd_hmac_key:
713
      files_to_copy.append(constants.CONFD_HMAC_KEY)
714

    
715
    if new_cds or cds:
716
      files_to_copy.append(constants.CLUSTER_DOMAIN_SECRET_FILE)
717

    
718
    if files_to_copy:
719
      for node_name in ctx.nonmaster_nodes:
720
        ctx.feedback_fn("Copying %s to %s" %
721
                        (", ".join(files_to_copy), node_name))
722
        for file_name in files_to_copy:
723
          ctx.ssh.CopyFileToNode(node_name, file_name)
724

    
725
  RunWhileClusterStopped(ToStdout, _RenewCryptoInner)
726

    
727
  ToStdout("All requested certificates and keys have been replaced."
728
           " Running \"gnt-cluster verify\" now is recommended.")
729

    
730
  return 0
731

    
732

    
733
def RenewCrypto(opts, args):
734
  """Renews cluster certificates, keys and secrets.
735

736
  """
737
  return _RenewCrypto(opts.new_cluster_cert,
738
                      opts.new_rapi_cert,
739
                      opts.rapi_cert,
740
                      opts.new_confd_hmac_key,
741
                      opts.new_cluster_domain_secret,
742
                      opts.cluster_domain_secret,
743
                      opts.force)
744

    
745

    
746
def SetClusterParams(opts, args):
747
  """Modify the cluster.
748

749
  @param opts: the command line options selected by the user
750
  @type args: list
751
  @param args: should be an empty list
752
  @rtype: int
753
  @return: the desired exit code
754

755
  """
756
  if not (not opts.lvm_storage or opts.vg_name or
757
          not opts.drbd_storage or opts.drbd_helper or
758
          opts.enabled_hypervisors or opts.hvparams or
759
          opts.beparams or opts.nicparams or opts.ndparams or
760
          opts.candidate_pool_size is not None or
761
          opts.uid_pool is not None or
762
          opts.maintain_node_health is not None or
763
          opts.add_uids is not None or
764
          opts.remove_uids is not None or
765
          opts.default_iallocator is not None or
766
          opts.reserved_lvs is not None or
767
          opts.master_netdev is not None or
768
          opts.prealloc_wipe_disks is not None):
769
    ToStderr("Please give at least one of the parameters.")
770
    return 1
771

    
772
  vg_name = opts.vg_name
773
  if not opts.lvm_storage and opts.vg_name:
774
    ToStderr("Options --no-lvm-storage and --vg-name conflict.")
775
    return 1
776

    
777
  if not opts.lvm_storage:
778
    vg_name = ""
779

    
780
  drbd_helper = opts.drbd_helper
781
  if not opts.drbd_storage and opts.drbd_helper:
782
    ToStderr("Options --no-drbd-storage and --drbd-usermode-helper conflict.")
783
    return 1
784

    
785
  if not opts.drbd_storage:
786
    drbd_helper = ""
787

    
788
  hvlist = opts.enabled_hypervisors
789
  if hvlist is not None:
790
    hvlist = hvlist.split(",")
791

    
792
  # a list of (name, dict) we can pass directly to dict() (or [])
793
  hvparams = dict(opts.hvparams)
794
  for hv_params in hvparams.values():
795
    utils.ForceDictType(hv_params, constants.HVS_PARAMETER_TYPES)
796

    
797
  beparams = opts.beparams
798
  utils.ForceDictType(beparams, constants.BES_PARAMETER_TYPES)
799

    
800
  nicparams = opts.nicparams
801
  utils.ForceDictType(nicparams, constants.NICS_PARAMETER_TYPES)
802

    
803
  ndparams = opts.ndparams
804
  if ndparams is not None:
805
    utils.ForceDictType(ndparams, constants.NDS_PARAMETER_TYPES)
806

    
807
  mnh = opts.maintain_node_health
808

    
809
  uid_pool = opts.uid_pool
810
  if uid_pool is not None:
811
    uid_pool = uidpool.ParseUidPool(uid_pool)
812

    
813
  add_uids = opts.add_uids
814
  if add_uids is not None:
815
    add_uids = uidpool.ParseUidPool(add_uids)
816

    
817
  remove_uids = opts.remove_uids
818
  if remove_uids is not None:
819
    remove_uids = uidpool.ParseUidPool(remove_uids)
820

    
821
  if opts.reserved_lvs is not None:
822
    if opts.reserved_lvs == "":
823
      opts.reserved_lvs = []
824
    else:
825
      opts.reserved_lvs = utils.UnescapeAndSplit(opts.reserved_lvs, sep=",")
826

    
827
  op = opcodes.OpClusterSetParams(vg_name=vg_name,
828
                                  drbd_helper=drbd_helper,
829
                                  enabled_hypervisors=hvlist,
830
                                  hvparams=hvparams,
831
                                  os_hvp=None,
832
                                  beparams=beparams,
833
                                  nicparams=nicparams,
834
                                  ndparams=ndparams,
835
                                  candidate_pool_size=opts.candidate_pool_size,
836
                                  maintain_node_health=mnh,
837
                                  uid_pool=uid_pool,
838
                                  add_uids=add_uids,
839
                                  remove_uids=remove_uids,
840
                                  default_iallocator=opts.default_iallocator,
841
                                  prealloc_wipe_disks=opts.prealloc_wipe_disks,
842
                                  master_netdev=opts.master_netdev,
843
                                  reserved_lvs=opts.reserved_lvs)
844
  SubmitOpCode(op, opts=opts)
845
  return 0
846

    
847

    
848
def QueueOps(opts, args):
849
  """Queue operations.
850

851
  @param opts: the command line options selected by the user
852
  @type args: list
853
  @param args: should contain only one element, the subcommand
854
  @rtype: int
855
  @return: the desired exit code
856

857
  """
858
  command = args[0]
859
  client = GetClient()
860
  if command in ("drain", "undrain"):
861
    drain_flag = command == "drain"
862
    client.SetQueueDrainFlag(drain_flag)
863
  elif command == "info":
864
    result = client.QueryConfigValues(["drain_flag"])
865
    if result[0]:
866
      val = "set"
867
    else:
868
      val = "unset"
869
    ToStdout("The drain flag is %s" % val)
870
  else:
871
    raise errors.OpPrereqError("Command '%s' is not valid." % command,
872
                               errors.ECODE_INVAL)
873

    
874
  return 0
875

    
876

    
877
def _ShowWatcherPause(until):
878
  if until is None or until < time.time():
879
    ToStdout("The watcher is not paused.")
880
  else:
881
    ToStdout("The watcher is paused until %s.", time.ctime(until))
882

    
883

    
884
def WatcherOps(opts, args):
885
  """Watcher operations.
886

887
  @param opts: the command line options selected by the user
888
  @type args: list
889
  @param args: should contain only one element, the subcommand
890
  @rtype: int
891
  @return: the desired exit code
892

893
  """
894
  command = args[0]
895
  client = GetClient()
896

    
897
  if command == "continue":
898
    client.SetWatcherPause(None)
899
    ToStdout("The watcher is no longer paused.")
900

    
901
  elif command == "pause":
902
    if len(args) < 2:
903
      raise errors.OpPrereqError("Missing pause duration", errors.ECODE_INVAL)
904

    
905
    result = client.SetWatcherPause(time.time() + ParseTimespec(args[1]))
906
    _ShowWatcherPause(result)
907

    
908
  elif command == "info":
909
    result = client.QueryConfigValues(["watcher_pause"])
910
    _ShowWatcherPause(result[0])
911

    
912
  else:
913
    raise errors.OpPrereqError("Command '%s' is not valid." % command,
914
                               errors.ECODE_INVAL)
915

    
916
  return 0
917

    
918

    
919
def _OobPower(opts, node_list, power):
920
  """Puts the node in the list to desired power state.
921

922
  @param opts: The command line options selected by the user
923
  @param node_list: The list of nodes to operate on
924
  @param power: True if they should be powered on, False otherwise
925
  @return: The success of the operation (none failed)
926

927
  """
928
  if power:
929
    command = constants.OOB_POWER_ON
930
  else:
931
    command = constants.OOB_POWER_OFF
932

    
933
  op = opcodes.OpOobCommand(node_names=node_list,
934
                            command=command,
935
                            ignore_status=True,
936
                            timeout=opts.oob_timeout,
937
                            power_delay=opts.power_delay)
938
  result = SubmitOpCode(op, opts=opts)
939
  errs = 0
940
  for node_result in result:
941
    (node_tuple, data_tuple) = node_result
942
    (_, node_name) = node_tuple
943
    (data_status, _) = data_tuple
944
    if data_status != constants.RS_NORMAL:
945
      assert data_status != constants.RS_UNAVAIL
946
      errs += 1
947
      ToStderr("There was a problem changing power for %s, please investigate",
948
               node_name)
949

    
950
  if errs > 0:
951
    return False
952

    
953
  return True
954

    
955

    
956
def _InstanceStart(opts, inst_list, start):
957
  """Puts the instances in the list to desired state.
958

959
  @param opts: The command line options selected by the user
960
  @param inst_list: The list of instances to operate on
961
  @param start: True if they should be started, False for shutdown
962
  @return: The success of the operation (none failed)
963

964
  """
965
  if start:
966
    opcls = opcodes.OpInstanceStartup
967
    text_submit, text_success, text_failed = ("startup", "started", "starting")
968
  else:
969
    opcls = compat.partial(opcodes.OpInstanceShutdown,
970
                           timeout=opts.shutdown_timeout)
971
    text_submit, text_success, text_failed = ("shutdown", "stopped", "stopping")
972

    
973
  jex = JobExecutor(opts=opts)
974

    
975
  for inst in inst_list:
976
    ToStdout("Submit %s of instance %s", text_submit, inst)
977
    op = opcls(instance_name=inst)
978
    jex.QueueJob(inst, op)
979

    
980
  results = jex.GetResults()
981
  bad_cnt = len([1 for (success, _) in results if not success])
982

    
983
  if bad_cnt == 0:
984
    ToStdout("All instances have been %s successfully", text_success)
985
  else:
986
    ToStderr("There were errors while %s instances:\n"
987
             "%d error(s) out of %d instance(s)", text_failed, bad_cnt,
988
             len(results))
989
    return False
990

    
991
  return True
992

    
993

    
994
class _RunWhenNodesReachableHelper:
995
  """Helper class to make shared internal state sharing easier.
996

997
  @ivar success: Indicates if all action_cb calls were successful
998

999
  """
1000
  def __init__(self, node_list, action_cb, node2ip, port, feedback_fn,
1001
               _ping_fn=netutils.TcpPing, _sleep_fn=time.sleep):
1002
    """Init the object.
1003

1004
    @param node_list: The list of nodes to be reachable
1005
    @param action_cb: Callback called when a new host is reachable
1006
    @type node2ip: dict
1007
    @param node2ip: Node to ip mapping
1008
    @param port: The port to use for the TCP ping
1009
    @param feedback_fn: The function used for feedback
1010
    @param _ping_fn: Function to check reachabilty (for unittest use only)
1011
    @param _sleep_fn: Function to sleep (for unittest use only)
1012

1013
    """
1014
    self.down = set(node_list)
1015
    self.up = set()
1016
    self.node2ip = node2ip
1017
    self.success = True
1018
    self.action_cb = action_cb
1019
    self.port = port
1020
    self.feedback_fn = feedback_fn
1021
    self._ping_fn = _ping_fn
1022
    self._sleep_fn = _sleep_fn
1023

    
1024
  def __call__(self):
1025
    """When called we run action_cb.
1026

1027
    @raises utils.RetryAgain: When there are still down nodes
1028

1029
    """
1030
    if not self.action_cb(self.up):
1031
      self.success = False
1032

    
1033
    if self.down:
1034
      raise utils.RetryAgain()
1035
    else:
1036
      return self.success
1037

    
1038
  def Wait(self, secs):
1039
    """Checks if a host is up or waits remaining seconds.
1040

1041
    @param secs: The secs remaining
1042

1043
    """
1044
    start = time.time()
1045
    for node in self.down:
1046
      if self._ping_fn(self.node2ip[node], self.port, timeout=_EPO_PING_TIMEOUT,
1047
                       live_port_needed=True):
1048
        self.feedback_fn("Node %s became available" % node)
1049
        self.up.add(node)
1050
        self.down -= self.up
1051
        # If we have a node available there is the possibility to run the
1052
        # action callback successfully, therefore we don't wait and return
1053
        return
1054

    
1055
    self._sleep_fn(max(0.0, start + secs - time.time()))
1056

    
1057

    
1058
def _RunWhenNodesReachable(node_list, action_cb, interval):
1059
  """Run action_cb when nodes become reachable.
1060

1061
  @param node_list: The list of nodes to be reachable
1062
  @param action_cb: Callback called when a new host is reachable
1063
  @param interval: The earliest time to retry
1064

1065
  """
1066
  client = GetClient()
1067
  cluster_info = client.QueryClusterInfo()
1068
  if cluster_info["primary_ip_version"] == constants.IP4_VERSION:
1069
    family = netutils.IPAddress.family
1070
  else:
1071
    family = netutils.IP6Address.family
1072

    
1073
  node2ip = dict((node, netutils.GetHostname(node, family=family).ip)
1074
                 for node in node_list)
1075

    
1076
  port = netutils.GetDaemonPort(constants.NODED)
1077
  helper = _RunWhenNodesReachableHelper(node_list, action_cb, node2ip, port,
1078
                                        ToStdout)
1079

    
1080
  try:
1081
    return utils.Retry(helper, interval, _EPO_REACHABLE_TIMEOUT,
1082
                       wait_fn=helper.Wait)
1083
  except utils.RetryTimeout:
1084
    ToStderr("Time exceeded while waiting for nodes to become reachable"
1085
             " again:\n  - %s", "  - ".join(helper.down))
1086
    return False
1087

    
1088

    
1089
def _MaybeInstanceStartup(opts, inst_map, nodes_online,
1090
                          _instance_start_fn=_InstanceStart):
1091
  """Start the instances conditional based on node_states.
1092

1093
  @param opts: The command line options selected by the user
1094
  @param inst_map: A dict of inst -> nodes mapping
1095
  @param nodes_online: A list of nodes online
1096
  @param _instance_start_fn: Callback to start instances (unittest use only)
1097
  @return: Success of the operation on all instances
1098

1099
  """
1100
  start_inst_list = []
1101
  for (inst, nodes) in inst_map.items():
1102
    if not (nodes - nodes_online):
1103
      # All nodes the instance lives on are back online
1104
      start_inst_list.append(inst)
1105

    
1106
  for inst in start_inst_list:
1107
    del inst_map[inst]
1108

    
1109
  if start_inst_list:
1110
    return _instance_start_fn(opts, start_inst_list, True)
1111

    
1112
  return True
1113

    
1114

    
1115
def _EpoOn(opts, full_node_list, node_list, inst_map):
1116
  """Does the actual power on.
1117

1118
  @param opts: The command line options selected by the user
1119
  @param full_node_list: All nodes to operate on (includes nodes not supporting
1120
                         OOB)
1121
  @param node_list: The list of nodes to operate on (all need to support OOB)
1122
  @param inst_map: A dict of inst -> nodes mapping
1123
  @return: The desired exit status
1124

1125
  """
1126
  if node_list and not _OobPower(opts, node_list, False):
1127
    ToStderr("Not all nodes seem to get back up, investigate and start"
1128
             " manually if needed")
1129

    
1130
  # Wait for the nodes to be back up
1131
  action_cb = compat.partial(_MaybeInstanceStartup, opts, dict(inst_map))
1132

    
1133
  ToStdout("Waiting until all nodes are available again")
1134
  if not _RunWhenNodesReachable(full_node_list, action_cb, _EPO_PING_INTERVAL):
1135
    ToStderr("Please investigate and start stopped instances manually")
1136
    return constants.EXIT_FAILURE
1137

    
1138
  return constants.EXIT_SUCCESS
1139

    
1140

    
1141
def _EpoOff(opts, node_list, inst_map):
1142
  """Does the actual power off.
1143

1144
  @param opts: The command line options selected by the user
1145
  @param node_list: The list of nodes to operate on (all need to support OOB)
1146
  @param inst_map: A dict of inst -> nodes mapping
1147
  @return: The desired exit status
1148

1149
  """
1150
  if not _InstanceStart(opts, inst_map.keys(), False):
1151
    ToStderr("Please investigate and stop instances manually before continuing")
1152
    return constants.EXIT_FAILURE
1153

    
1154
  if not node_list:
1155
    return constants.EXIT_SUCCESS
1156

    
1157
  if _OobPower(opts, node_list, False):
1158
    return constants.EXIT_SUCCESS
1159
  else:
1160
    return constants.EXIT_FAILURE
1161

    
1162

    
1163
def Epo(opts, args):
1164
  """EPO operations.
1165

1166
  @param opts: the command line options selected by the user
1167
  @type args: list
1168
  @param args: should contain only one element, the subcommand
1169
  @rtype: int
1170
  @return: the desired exit code
1171

1172
  """
1173
  if opts.groups and opts.show_all:
1174
    ToStderr("Only one of --groups or --all are allowed")
1175
    return constants.EXIT_FAILURE
1176
  elif args and opts.show_all:
1177
    ToStderr("Arguments in combination with --all are not allowed")
1178
    return constants.EXIT_FAILURE
1179

    
1180
  client = GetClient()
1181

    
1182
  if opts.groups:
1183
    node_query_list = itertools.chain(*client.QueryGroups(names=args,
1184
                                                          fields=["node_list"],
1185
                                                          use_locking=False))
1186
  else:
1187
    node_query_list = args
1188

    
1189
  result = client.QueryNodes(names=node_query_list,
1190
                             fields=["name", "master", "pinst_list",
1191
                                     "sinst_list", "powered", "offline"],
1192
                             use_locking=False)
1193
  node_list = []
1194
  inst_map = {}
1195
  for (idx, (node, master, pinsts, sinsts, powered,
1196
             offline)) in enumerate(result):
1197
    # Normalize the node_query_list as well
1198
    if not opts.show_all:
1199
      node_query_list[idx] = node
1200
    if not offline:
1201
      for inst in (pinsts + sinsts):
1202
        if inst in inst_map:
1203
          if not master:
1204
            inst_map[inst].add(node)
1205
        elif master:
1206
          inst_map[inst] = set()
1207
        else:
1208
          inst_map[inst] = set([node])
1209

    
1210
    if master and opts.on:
1211
      # We ignore the master for turning on the machines, in fact we are
1212
      # already operating on the master at this point :)
1213
      continue
1214
    elif master and not opts.show_all:
1215
      ToStderr("%s is the master node, please do a master-failover to another"
1216
               " node not affected by the EPO or use --all if you intend to"
1217
               " shutdown the whole cluster", node)
1218
      return constants.EXIT_FAILURE
1219
    elif powered is None:
1220
      ToStdout("Node %s does not support out-of-band handling, it can not be"
1221
               " handled in a fully automated manner", node)
1222
    elif powered == opts.on:
1223
      ToStdout("Node %s is already in desired power state, skipping", node)
1224
    elif not offline or (offline and powered):
1225
      node_list.append(node)
1226

    
1227
  if not opts.force and not ConfirmOperation(node_query_list, "nodes", "epo"):
1228
    return constants.EXIT_FAILURE
1229

    
1230
  if opts.on:
1231
    return _EpoOn(opts, node_query_list, node_list, inst_map)
1232
  else:
1233
    return _EpoOff(opts, node_list, inst_map)
1234

    
1235

    
1236
commands = {
1237
  'init': (
1238
    InitCluster, [ArgHost(min=1, max=1)],
1239
    [BACKEND_OPT, CP_SIZE_OPT, ENABLED_HV_OPT, GLOBAL_FILEDIR_OPT,
1240
     HVLIST_OPT, MAC_PREFIX_OPT, MASTER_NETDEV_OPT, NIC_PARAMS_OPT,
1241
     NOLVM_STORAGE_OPT, NOMODIFY_ETCHOSTS_OPT, NOMODIFY_SSH_SETUP_OPT,
1242
     SECONDARY_IP_OPT, VG_NAME_OPT, MAINTAIN_NODE_HEALTH_OPT,
1243
     UIDPOOL_OPT, DRBD_HELPER_OPT, NODRBD_STORAGE_OPT,
1244
     DEFAULT_IALLOCATOR_OPT, PRIMARY_IP_VERSION_OPT, PREALLOC_WIPE_DISKS_OPT,
1245
     NODE_PARAMS_OPT, GLOBAL_SHARED_FILEDIR_OPT],
1246
    "[opts...] <cluster_name>", "Initialises a new cluster configuration"),
1247
  'destroy': (
1248
    DestroyCluster, ARGS_NONE, [YES_DOIT_OPT],
1249
    "", "Destroy cluster"),
1250
  'rename': (
1251
    RenameCluster, [ArgHost(min=1, max=1)],
1252
    [FORCE_OPT, DRY_RUN_OPT],
1253
    "<new_name>",
1254
    "Renames the cluster"),
1255
  'redist-conf': (
1256
    RedistributeConfig, ARGS_NONE, [SUBMIT_OPT, DRY_RUN_OPT, PRIORITY_OPT],
1257
    "", "Forces a push of the configuration file and ssconf files"
1258
    " to the nodes in the cluster"),
1259
  'verify': (
1260
    VerifyCluster, ARGS_NONE,
1261
    [VERBOSE_OPT, DEBUG_SIMERR_OPT, ERROR_CODES_OPT, NONPLUS1_OPT,
1262
     DRY_RUN_OPT, PRIORITY_OPT],
1263
    "", "Does a check on the cluster configuration"),
1264
  'verify-disks': (
1265
    VerifyDisks, ARGS_NONE, [PRIORITY_OPT],
1266
    "", "Does a check on the cluster disk status"),
1267
  'repair-disk-sizes': (
1268
    RepairDiskSizes, ARGS_MANY_INSTANCES, [DRY_RUN_OPT, PRIORITY_OPT],
1269
    "", "Updates mismatches in recorded disk sizes"),
1270
  'master-failover': (
1271
    MasterFailover, ARGS_NONE, [NOVOTING_OPT],
1272
    "", "Makes the current node the master"),
1273
  'master-ping': (
1274
    MasterPing, ARGS_NONE, [],
1275
    "", "Checks if the master is alive"),
1276
  'version': (
1277
    ShowClusterVersion, ARGS_NONE, [],
1278
    "", "Shows the cluster version"),
1279
  'getmaster': (
1280
    ShowClusterMaster, ARGS_NONE, [],
1281
    "", "Shows the cluster master"),
1282
  'copyfile': (
1283
    ClusterCopyFile, [ArgFile(min=1, max=1)],
1284
    [NODE_LIST_OPT, USE_REPL_NET_OPT],
1285
    "[-n node...] <filename>", "Copies a file to all (or only some) nodes"),
1286
  'command': (
1287
    RunClusterCommand, [ArgCommand(min=1)],
1288
    [NODE_LIST_OPT],
1289
    "[-n node...] <command>", "Runs a command on all (or only some) nodes"),
1290
  'info': (
1291
    ShowClusterConfig, ARGS_NONE, [ROMAN_OPT],
1292
    "[--roman]", "Show cluster configuration"),
1293
  'list-tags': (
1294
    ListTags, ARGS_NONE, [], "", "List the tags of the cluster"),
1295
  'add-tags': (
1296
    AddTags, [ArgUnknown()], [TAG_SRC_OPT, PRIORITY_OPT],
1297
    "tag...", "Add tags to the cluster"),
1298
  'remove-tags': (
1299
    RemoveTags, [ArgUnknown()], [TAG_SRC_OPT, PRIORITY_OPT],
1300
    "tag...", "Remove tags from the cluster"),
1301
  'search-tags': (
1302
    SearchTags, [ArgUnknown(min=1, max=1)], [PRIORITY_OPT], "",
1303
    "Searches the tags on all objects on"
1304
    " the cluster for a given pattern (regex)"),
1305
  'queue': (
1306
    QueueOps,
1307
    [ArgChoice(min=1, max=1, choices=["drain", "undrain", "info"])],
1308
    [], "drain|undrain|info", "Change queue properties"),
1309
  'watcher': (
1310
    WatcherOps,
1311
    [ArgChoice(min=1, max=1, choices=["pause", "continue", "info"]),
1312
     ArgSuggest(min=0, max=1, choices=["30m", "1h", "4h"])],
1313
    [],
1314
    "{pause <timespec>|continue|info}", "Change watcher properties"),
1315
  'modify': (
1316
    SetClusterParams, ARGS_NONE,
1317
    [BACKEND_OPT, CP_SIZE_OPT, ENABLED_HV_OPT, HVLIST_OPT, MASTER_NETDEV_OPT,
1318
     NIC_PARAMS_OPT, NOLVM_STORAGE_OPT, VG_NAME_OPT, MAINTAIN_NODE_HEALTH_OPT,
1319
     UIDPOOL_OPT, ADD_UIDS_OPT, REMOVE_UIDS_OPT, DRBD_HELPER_OPT,
1320
     NODRBD_STORAGE_OPT, DEFAULT_IALLOCATOR_OPT, RESERVED_LVS_OPT,
1321
     DRY_RUN_OPT, PRIORITY_OPT, PREALLOC_WIPE_DISKS_OPT, NODE_PARAMS_OPT],
1322
    "[opts...]",
1323
    "Alters the parameters of the cluster"),
1324
  "renew-crypto": (
1325
    RenewCrypto, ARGS_NONE,
1326
    [NEW_CLUSTER_CERT_OPT, NEW_RAPI_CERT_OPT, RAPI_CERT_OPT,
1327
     NEW_CONFD_HMAC_KEY_OPT, FORCE_OPT,
1328
     NEW_CLUSTER_DOMAIN_SECRET_OPT, CLUSTER_DOMAIN_SECRET_OPT],
1329
    "[opts...]",
1330
    "Renews cluster certificates, keys and secrets"),
1331
  "epo": (
1332
    Epo, [ArgUnknown()],
1333
    [FORCE_OPT, ON_OPT, GROUPS_OPT, ALL_OPT, OOB_TIMEOUT_OPT,
1334
     SHUTDOWN_TIMEOUT_OPT, POWER_DELAY_OPT],
1335
    "[opts...] [args]",
1336
    "Performs an emergency power-off on given args"),
1337
  }
1338

    
1339

    
1340
#: dictionary with aliases for commands
1341
aliases = {
1342
  'masterfailover': 'master-failover',
1343
}
1344

    
1345

    
1346
def Main():
1347
  return GenericMain(commands, override={"tag_type": constants.TAG_CLUSTER},
1348
                     aliases=aliases)