Statistics
| Branch: | Tag: | Revision:

root / tools / burnin @ 7eda951b

History | View | Annotate | Download (38.8 kB)

1 a8083063 Iustin Pop
#!/usr/bin/python
2 a8083063 Iustin Pop
#
3 a8083063 Iustin Pop
4 4ff922a2 Iustin Pop
# Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011 Google Inc.
5 01b69ec5 Michael Hanselmann
#
6 01b69ec5 Michael Hanselmann
# This program is free software; you can redistribute it and/or modify
7 01b69ec5 Michael Hanselmann
# it under the terms of the GNU General Public License as published by
8 01b69ec5 Michael Hanselmann
# the Free Software Foundation; either version 2 of the License, or
9 01b69ec5 Michael Hanselmann
# (at your option) any later version.
10 01b69ec5 Michael Hanselmann
#
11 01b69ec5 Michael Hanselmann
# This program is distributed in the hope that it will be useful, but
12 01b69ec5 Michael Hanselmann
# WITHOUT ANY WARRANTY; without even the implied warranty of
13 01b69ec5 Michael Hanselmann
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 01b69ec5 Michael Hanselmann
# General Public License for more details.
15 01b69ec5 Michael Hanselmann
#
16 01b69ec5 Michael Hanselmann
# You should have received a copy of the GNU General Public License
17 01b69ec5 Michael Hanselmann
# along with this program; if not, write to the Free Software
18 01b69ec5 Michael Hanselmann
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
19 01b69ec5 Michael Hanselmann
# 02110-1301, USA.
20 01b69ec5 Michael Hanselmann
21 01b69ec5 Michael Hanselmann
22 01b69ec5 Michael Hanselmann
"""Burnin program
23 01b69ec5 Michael Hanselmann
24 01b69ec5 Michael Hanselmann
"""
25 175f44c2 Iustin Pop
26 a8083063 Iustin Pop
import sys
27 a8083063 Iustin Pop
import optparse
28 e2212007 Iustin Pop
import time
29 5178f1bc Iustin Pop
import socket
30 5dc626fd Iustin Pop
import urllib
31 175f44c2 Iustin Pop
from itertools import izip, islice, cycle
32 21546b1c Iustin Pop
from cStringIO import StringIO
33 a8083063 Iustin Pop
34 a8083063 Iustin Pop
from ganeti import opcodes
35 a8083063 Iustin Pop
from ganeti import constants
36 a8083063 Iustin Pop
from ganeti import cli
37 9f13fc7a Iustin Pop
from ganeti import errors
38 9f13fc7a Iustin Pop
from ganeti import utils
39 aac0352d Iustin Pop
from ganeti import hypervisor
40 cea881e5 Michael Hanselmann
from ganeti import compat
41 a619a1dd Guido Trotter
42 a619a1dd Guido Trotter
from ganeti.confd import client as confd_client
43 a8083063 Iustin Pop
44 01b69ec5 Michael Hanselmann
45 9f13fc7a Iustin Pop
USAGE = ("\tburnin -o OS_NAME [options...] instance_name ...")
46 a8083063 Iustin Pop
47 73ff3118 Iustin Pop
MAX_RETRIES = 3
48 1b334175 Iustin Pop
LOG_HEADERS = {
49 1b334175 Iustin Pop
  0: "- ",
50 1b334175 Iustin Pop
  1: "* ",
51 1b334175 Iustin Pop
  2: ""
52 1b334175 Iustin Pop
  }
53 01b69ec5 Michael Hanselmann
54 e687ec01 Michael Hanselmann
55 5178f1bc Iustin Pop
class InstanceDown(Exception):
56 5178f1bc Iustin Pop
  """The checked instance was not up"""
57 5178f1bc Iustin Pop
58 5178f1bc Iustin Pop
59 73ff3118 Iustin Pop
class BurninFailure(Exception):
60 73ff3118 Iustin Pop
  """Failure detected during burning"""
61 73ff3118 Iustin Pop
62 73ff3118 Iustin Pop
63 a8083063 Iustin Pop
def Usage():
64 a8083063 Iustin Pop
  """Shows program usage information and exits the program."""
65 a8083063 Iustin Pop
66 a8083063 Iustin Pop
  print >> sys.stderr, "Usage:"
67 a8083063 Iustin Pop
  print >> sys.stderr, USAGE
68 a8083063 Iustin Pop
  sys.exit(2)
69 a8083063 Iustin Pop
70 01b69ec5 Michael Hanselmann
71 1b334175 Iustin Pop
def Log(msg, *args, **kwargs):
72 3ecf6786 Iustin Pop
  """Simple function that prints out its argument.
73 3ecf6786 Iustin Pop
74 3ecf6786 Iustin Pop
  """
75 1b334175 Iustin Pop
  if args:
76 1b334175 Iustin Pop
    msg = msg % args
77 e687ec01 Michael Hanselmann
  indent = kwargs.get("indent", 0)
78 e687ec01 Michael Hanselmann
  sys.stdout.write("%*s%s%s\n" % (2 * indent, "",
79 1b334175 Iustin Pop
                                  LOG_HEADERS.get(indent, "  "), msg))
80 84cc52ab Michael Hanselmann
  sys.stdout.flush()
81 a8083063 Iustin Pop
82 1b334175 Iustin Pop
83 836d59d7 Iustin Pop
def Err(msg, exit_code=1):
84 836d59d7 Iustin Pop
  """Simple error logging that prints to stderr.
85 836d59d7 Iustin Pop
86 836d59d7 Iustin Pop
  """
87 836d59d7 Iustin Pop
  sys.stderr.write(msg + "\n")
88 836d59d7 Iustin Pop
  sys.stderr.flush()
89 836d59d7 Iustin Pop
  sys.exit(exit_code)
90 01b69ec5 Michael Hanselmann
91 5dc626fd Iustin Pop
92 5dc626fd Iustin Pop
class SimpleOpener(urllib.FancyURLopener):
93 5dc626fd Iustin Pop
  """A simple url opener"""
94 b459a848 Andrea Spadaccini
  # pylint: disable=W0221
95 5dc626fd Iustin Pop
96 7260cfbe Iustin Pop
  def prompt_user_passwd(self, host, realm, clear_cache=0):
97 5dc626fd Iustin Pop
    """No-interaction version of prompt_user_passwd."""
98 2d54e29c Iustin Pop
    # we follow parent class' API
99 b459a848 Andrea Spadaccini
    # pylint: disable=W0613
100 5dc626fd Iustin Pop
    return None, None
101 5dc626fd Iustin Pop
102 5dc626fd Iustin Pop
  def http_error_default(self, url, fp, errcode, errmsg, headers):
103 5dc626fd Iustin Pop
    """Custom error handling"""
104 5dc626fd Iustin Pop
    # make sure sockets are not left in CLOSE_WAIT, this is similar
105 5dc626fd Iustin Pop
    # but with a different exception to the BasicURLOpener class
106 5dc626fd Iustin Pop
    _ = fp.read() # throw away data
107 5dc626fd Iustin Pop
    fp.close()
108 5dc626fd Iustin Pop
    raise InstanceDown("HTTP error returned: code %s, msg %s" %
109 5dc626fd Iustin Pop
                       (errcode, errmsg))
110 5dc626fd Iustin Pop
111 5dc626fd Iustin Pop
112 10eb54fb Michael Hanselmann
OPTIONS = [
113 10eb54fb Michael Hanselmann
  cli.cli_option("-o", "--os", dest="os", default=None,
114 10eb54fb Michael Hanselmann
                 help="OS to use during burnin",
115 a52ba89d Michael Hanselmann
                 metavar="<OS>",
116 a52ba89d Michael Hanselmann
                 completion_suggest=cli.OPT_COMPL_ONE_OS),
117 aac0352d Iustin Pop
  cli.HYPERVISOR_OPT,
118 062a7100 Iustin Pop
  cli.OSPARAMS_OPT,
119 10eb54fb Michael Hanselmann
  cli.cli_option("--disk-size", dest="disk_size",
120 10eb54fb Michael Hanselmann
                 help="Disk size (determines disk count)",
121 10eb54fb Michael Hanselmann
                 default="128m", type="string", metavar="<size,size,...>",
122 10eb54fb Michael Hanselmann
                 completion_suggest=("128M 512M 1G 4G 1G,256M"
123 10eb54fb Michael Hanselmann
                                     " 4G,1G,1G 10G").split()),
124 10eb54fb Michael Hanselmann
  cli.cli_option("--disk-growth", dest="disk_growth", help="Disk growth",
125 10eb54fb Michael Hanselmann
                 default="128m", type="string", metavar="<size,size,...>"),
126 10eb54fb Michael Hanselmann
  cli.cli_option("--mem-size", dest="mem_size", help="Memory size",
127 10eb54fb Michael Hanselmann
                 default=128, type="unit", metavar="<size>",
128 10eb54fb Michael Hanselmann
                 completion_suggest=("128M 256M 512M 1G 4G 8G"
129 10eb54fb Michael Hanselmann
                                     " 12G 16G").split()),
130 d0ffa390 Pedro Macedo
  cli.cli_option("--vcpu-count", dest="vcpu_count", help="VCPU count",
131 d0ffa390 Pedro Macedo
                 default=3, type="unit", metavar="<count>",
132 d0ffa390 Pedro Macedo
                 completion_suggest=("1 2 3 4").split()),
133 265e6244 Iustin Pop
  cli.DEBUG_OPT,
134 9cdb9578 Iustin Pop
  cli.VERBOSE_OPT,
135 544ca43b Iustin Pop
  cli.NOIPCHECK_OPT,
136 544ca43b Iustin Pop
  cli.NONAMECHECK_OPT,
137 7ea7bcf6 Iustin Pop
  cli.EARLY_RELEASE_OPT,
138 10eb54fb Michael Hanselmann
  cli.cli_option("--no-replace1", dest="do_replace1",
139 10eb54fb Michael Hanselmann
                 help="Skip disk replacement with the same secondary",
140 10eb54fb Michael Hanselmann
                 action="store_false", default=True),
141 10eb54fb Michael Hanselmann
  cli.cli_option("--no-replace2", dest="do_replace2",
142 10eb54fb Michael Hanselmann
                 help="Skip disk replacement with a different secondary",
143 10eb54fb Michael Hanselmann
                 action="store_false", default=True),
144 10eb54fb Michael Hanselmann
  cli.cli_option("--no-failover", dest="do_failover",
145 10eb54fb Michael Hanselmann
                 help="Skip instance failovers", action="store_false",
146 10eb54fb Michael Hanselmann
                 default=True),
147 10eb54fb Michael Hanselmann
  cli.cli_option("--no-migrate", dest="do_migrate",
148 10eb54fb Michael Hanselmann
                 help="Skip instance live migration",
149 10eb54fb Michael Hanselmann
                 action="store_false", default=True),
150 5b9107ff Iustin Pop
  cli.cli_option("--no-move", dest="do_move",
151 5b9107ff Iustin Pop
                 help="Skip instance moves", action="store_false",
152 5b9107ff Iustin Pop
                 default=True),
153 10eb54fb Michael Hanselmann
  cli.cli_option("--no-importexport", dest="do_importexport",
154 10eb54fb Michael Hanselmann
                 help="Skip instance export/import", action="store_false",
155 10eb54fb Michael Hanselmann
                 default=True),
156 10eb54fb Michael Hanselmann
  cli.cli_option("--no-startstop", dest="do_startstop",
157 10eb54fb Michael Hanselmann
                 help="Skip instance stop/start", action="store_false",
158 10eb54fb Michael Hanselmann
                 default=True),
159 10eb54fb Michael Hanselmann
  cli.cli_option("--no-reinstall", dest="do_reinstall",
160 10eb54fb Michael Hanselmann
                 help="Skip instance reinstall", action="store_false",
161 10eb54fb Michael Hanselmann
                 default=True),
162 10eb54fb Michael Hanselmann
  cli.cli_option("--no-reboot", dest="do_reboot",
163 10eb54fb Michael Hanselmann
                 help="Skip instance reboot", action="store_false",
164 10eb54fb Michael Hanselmann
                 default=True),
165 1d103c02 Iustin Pop
  cli.cli_option("--reboot-types", dest="reboot_types",
166 1d103c02 Iustin Pop
                 help="Specify the reboot types", default=None),
167 10eb54fb Michael Hanselmann
  cli.cli_option("--no-activate-disks", dest="do_activate_disks",
168 10eb54fb Michael Hanselmann
                 help="Skip disk activation/deactivation",
169 10eb54fb Michael Hanselmann
                 action="store_false", default=True),
170 10eb54fb Michael Hanselmann
  cli.cli_option("--no-add-disks", dest="do_addremove_disks",
171 10eb54fb Michael Hanselmann
                 help="Skip disk addition/removal",
172 10eb54fb Michael Hanselmann
                 action="store_false", default=True),
173 10eb54fb Michael Hanselmann
  cli.cli_option("--no-add-nics", dest="do_addremove_nics",
174 10eb54fb Michael Hanselmann
                 help="Skip NIC addition/removal",
175 10eb54fb Michael Hanselmann
                 action="store_false", default=True),
176 10eb54fb Michael Hanselmann
  cli.cli_option("--no-nics", dest="nics",
177 10eb54fb Michael Hanselmann
                 help="No network interfaces", action="store_const",
178 10eb54fb Michael Hanselmann
                 const=[], default=[{}]),
179 a619a1dd Guido Trotter
  cli.cli_option("--no-confd", dest="do_confd_tests",
180 a619a1dd Guido Trotter
                 help="Skip confd queries",
181 a619a1dd Guido Trotter
                 action="store_false", default=True),
182 10eb54fb Michael Hanselmann
  cli.cli_option("--rename", dest="rename", default=None,
183 10eb54fb Michael Hanselmann
                 help=("Give one unused instance name which is taken"
184 10eb54fb Michael Hanselmann
                       " to start the renaming sequence"),
185 10eb54fb Michael Hanselmann
                 metavar="<instance_name>"),
186 10eb54fb Michael Hanselmann
  cli.cli_option("-t", "--disk-template", dest="disk_template",
187 10eb54fb Michael Hanselmann
                 choices=list(constants.DISK_TEMPLATES),
188 10eb54fb Michael Hanselmann
                 default=constants.DT_DRBD8,
189 9869360c Apollon Oikonomopoulos
                 help="Disk template (diskless, file, plain, sharedfile"
190 9869360c Apollon Oikonomopoulos
                 " or drbd) [drbd]"),
191 10eb54fb Michael Hanselmann
  cli.cli_option("-n", "--nodes", dest="nodes", default="",
192 10eb54fb Michael Hanselmann
                 help=("Comma separated list of nodes to perform"
193 a52ba89d Michael Hanselmann
                       " the burnin on (defaults to all nodes)"),
194 a52ba89d Michael Hanselmann
                 completion_suggest=cli.OPT_COMPL_MANY_NODES),
195 10eb54fb Michael Hanselmann
  cli.cli_option("-I", "--iallocator", dest="iallocator",
196 10eb54fb Michael Hanselmann
                 default=None, type="string",
197 10eb54fb Michael Hanselmann
                 help=("Perform the allocation using an iallocator"
198 10eb54fb Michael Hanselmann
                       " instead of fixed node spread (node restrictions no"
199 10eb54fb Michael Hanselmann
                       " longer apply, therefore -n/--nodes must not be"
200 a52ba89d Michael Hanselmann
                       " used"),
201 a52ba89d Michael Hanselmann
                 completion_suggest=cli.OPT_COMPL_ONE_IALLOCATOR),
202 10eb54fb Michael Hanselmann
  cli.cli_option("-p", "--parallel", default=False, action="store_true",
203 10eb54fb Michael Hanselmann
                 dest="parallel",
204 10eb54fb Michael Hanselmann
                 help=("Enable parallelization of some operations in"
205 10eb54fb Michael Hanselmann
                       " order to speed burnin or to test granular locking")),
206 10eb54fb Michael Hanselmann
  cli.cli_option("--net-timeout", default=15, type="int",
207 10eb54fb Michael Hanselmann
                 dest="net_timeout",
208 10eb54fb Michael Hanselmann
                 help=("The instance check network timeout in seconds"
209 10eb54fb Michael Hanselmann
                       " (defaults to 15 seconds)"),
210 10eb54fb Michael Hanselmann
                 completion_suggest="15 60 300 900".split()),
211 10eb54fb Michael Hanselmann
  cli.cli_option("-C", "--http-check", default=False, action="store_true",
212 10eb54fb Michael Hanselmann
                 dest="http_check",
213 10eb54fb Michael Hanselmann
                 help=("Enable checking of instance status via http,"
214 10eb54fb Michael Hanselmann
                       " looking for /hostname.txt that should contain the"
215 10eb54fb Michael Hanselmann
                       " name of the instance")),
216 10eb54fb Michael Hanselmann
  cli.cli_option("-K", "--keep-instances", default=False,
217 10eb54fb Michael Hanselmann
                 action="store_true",
218 10eb54fb Michael Hanselmann
                 dest="keep_instances",
219 10eb54fb Michael Hanselmann
                 help=("Leave instances on the cluster after burnin,"
220 10eb54fb Michael Hanselmann
                       " for investigation in case of errors or simply"
221 10eb54fb Michael Hanselmann
                       " to use them")),
222 10eb54fb Michael Hanselmann
  ]
223 10eb54fb Michael Hanselmann
224 10eb54fb Michael Hanselmann
# Mainly used for bash completion
225 10eb54fb Michael Hanselmann
ARGUMENTS = [cli.ArgInstance(min=1)]
226 10eb54fb Michael Hanselmann
227 10eb54fb Michael Hanselmann
228 e7beaa02 Iustin Pop
def _DoCheckInstances(fn):
229 e7beaa02 Iustin Pop
  """Decorator for checking instances.
230 e7beaa02 Iustin Pop
231 e7beaa02 Iustin Pop
  """
232 e7beaa02 Iustin Pop
  def wrapper(self, *args, **kwargs):
233 e7beaa02 Iustin Pop
    val = fn(self, *args, **kwargs)
234 e7beaa02 Iustin Pop
    for instance in self.instances:
235 b459a848 Andrea Spadaccini
      self._CheckInstanceAlive(instance) # pylint: disable=W0212
236 e7beaa02 Iustin Pop
    return val
237 e7beaa02 Iustin Pop
238 e7beaa02 Iustin Pop
  return wrapper
239 e7beaa02 Iustin Pop
240 e7beaa02 Iustin Pop
241 e7beaa02 Iustin Pop
def _DoBatch(retry):
242 e7beaa02 Iustin Pop
  """Decorator for possible batch operations.
243 e7beaa02 Iustin Pop
244 e7beaa02 Iustin Pop
  Must come after the _DoCheckInstances decorator (if any).
245 e7beaa02 Iustin Pop
246 e7beaa02 Iustin Pop
  @param retry: whether this is a retryable batch, will be
247 e7beaa02 Iustin Pop
      passed to StartBatch
248 e7beaa02 Iustin Pop
249 e7beaa02 Iustin Pop
  """
250 e7beaa02 Iustin Pop
  def wrap(fn):
251 e7beaa02 Iustin Pop
    def batched(self, *args, **kwargs):
252 e7beaa02 Iustin Pop
      self.StartBatch(retry)
253 e7beaa02 Iustin Pop
      val = fn(self, *args, **kwargs)
254 e7beaa02 Iustin Pop
      self.CommitQueue()
255 e7beaa02 Iustin Pop
      return val
256 e7beaa02 Iustin Pop
    return batched
257 e7beaa02 Iustin Pop
258 e7beaa02 Iustin Pop
  return wrap
259 e7beaa02 Iustin Pop
260 e7beaa02 Iustin Pop
261 175f44c2 Iustin Pop
class Burner(object):
262 175f44c2 Iustin Pop
  """Burner class."""
263 175f44c2 Iustin Pop
264 175f44c2 Iustin Pop
  def __init__(self):
265 175f44c2 Iustin Pop
    """Constructor."""
266 5dc626fd Iustin Pop
    self.url_opener = SimpleOpener()
267 21546b1c Iustin Pop
    self._feed_buf = StringIO()
268 175f44c2 Iustin Pop
    self.nodes = []
269 175f44c2 Iustin Pop
    self.instances = []
270 175f44c2 Iustin Pop
    self.to_rem = []
271 c723c163 Iustin Pop
    self.queued_ops = []
272 175f44c2 Iustin Pop
    self.opts = None
273 73ff3118 Iustin Pop
    self.queue_retry = False
274 73ff3118 Iustin Pop
    self.disk_count = self.disk_growth = self.disk_size = None
275 73ff3118 Iustin Pop
    self.hvp = self.bep = None
276 175f44c2 Iustin Pop
    self.ParseOptions()
277 320eda24 Iustin Pop
    self.cl = cli.GetClient()
278 175f44c2 Iustin Pop
    self.GetState()
279 175f44c2 Iustin Pop
280 21546b1c Iustin Pop
  def ClearFeedbackBuf(self):
281 21546b1c Iustin Pop
    """Clear the feedback buffer."""
282 21546b1c Iustin Pop
    self._feed_buf.truncate(0)
283 21546b1c Iustin Pop
284 21546b1c Iustin Pop
  def GetFeedbackBuf(self):
285 21546b1c Iustin Pop
    """Return the contents of the buffer."""
286 21546b1c Iustin Pop
    return self._feed_buf.getvalue()
287 21546b1c Iustin Pop
288 21546b1c Iustin Pop
  def Feedback(self, msg):
289 21546b1c Iustin Pop
    """Acumulate feedback in our buffer."""
290 88d31e5c Iustin Pop
    formatted_msg = "%s %s" % (time.ctime(utils.MergeTime(msg[0])), msg[2])
291 88d31e5c Iustin Pop
    self._feed_buf.write(formatted_msg + "\n")
292 d7b47a77 Iustin Pop
    if self.opts.verbose:
293 88d31e5c Iustin Pop
      Log(formatted_msg, indent=3)
294 21546b1c Iustin Pop
295 73ff3118 Iustin Pop
  def MaybeRetry(self, retry_count, msg, fn, *args):
296 73ff3118 Iustin Pop
    """Possibly retry a given function execution.
297 73ff3118 Iustin Pop
298 73ff3118 Iustin Pop
    @type retry_count: int
299 73ff3118 Iustin Pop
    @param retry_count: retry counter:
300 73ff3118 Iustin Pop
        - 0: non-retryable action
301 73ff3118 Iustin Pop
        - 1: last retry for a retryable action
302 73ff3118 Iustin Pop
        - MAX_RETRIES: original try for a retryable action
303 73ff3118 Iustin Pop
    @type msg: str
304 73ff3118 Iustin Pop
    @param msg: the kind of the operation
305 73ff3118 Iustin Pop
    @type fn: callable
306 73ff3118 Iustin Pop
    @param fn: the function to be called
307 73ff3118 Iustin Pop
308 73ff3118 Iustin Pop
    """
309 73ff3118 Iustin Pop
    try:
310 73ff3118 Iustin Pop
      val = fn(*args)
311 73ff3118 Iustin Pop
      if retry_count > 0 and retry_count < MAX_RETRIES:
312 1b334175 Iustin Pop
        Log("Idempotent %s succeeded after %d retries",
313 1b334175 Iustin Pop
            msg, MAX_RETRIES - retry_count)
314 73ff3118 Iustin Pop
      return val
315 b459a848 Andrea Spadaccini
    except Exception, err: # pylint: disable=W0703
316 73ff3118 Iustin Pop
      if retry_count == 0:
317 1b334175 Iustin Pop
        Log("Non-idempotent %s failed, aborting", msg)
318 73ff3118 Iustin Pop
        raise
319 73ff3118 Iustin Pop
      elif retry_count == 1:
320 1b334175 Iustin Pop
        Log("Idempotent %s repeated failure, aborting", msg)
321 73ff3118 Iustin Pop
        raise
322 73ff3118 Iustin Pop
      else:
323 1b334175 Iustin Pop
        Log("Idempotent %s failed, retry #%d/%d: %s",
324 1b334175 Iustin Pop
            msg, MAX_RETRIES - retry_count + 1, MAX_RETRIES, err)
325 73ff3118 Iustin Pop
        self.MaybeRetry(retry_count - 1, msg, fn, *args)
326 73ff3118 Iustin Pop
327 73ff3118 Iustin Pop
  def _ExecOp(self, *ops):
328 c723c163 Iustin Pop
    """Execute one or more opcodes and manage the exec buffer.
329 c723c163 Iustin Pop
330 454723b5 Iustin Pop
    @return: if only opcode has been passed, we return its result;
331 c723c163 Iustin Pop
        otherwise we return the list of results
332 c723c163 Iustin Pop
333 c723c163 Iustin Pop
    """
334 c723c163 Iustin Pop
    job_id = cli.SendJob(ops, cl=self.cl)
335 c723c163 Iustin Pop
    results = cli.PollJob(job_id, cl=self.cl, feedback_fn=self.Feedback)
336 c723c163 Iustin Pop
    if len(ops) == 1:
337 c723c163 Iustin Pop
      return results[0]
338 c723c163 Iustin Pop
    else:
339 c723c163 Iustin Pop
      return results
340 c723c163 Iustin Pop
341 73ff3118 Iustin Pop
  def ExecOp(self, retry, *ops):
342 73ff3118 Iustin Pop
    """Execute one or more opcodes and manage the exec buffer.
343 73ff3118 Iustin Pop
344 454723b5 Iustin Pop
    @return: if only opcode has been passed, we return its result;
345 73ff3118 Iustin Pop
        otherwise we return the list of results
346 73ff3118 Iustin Pop
347 73ff3118 Iustin Pop
    """
348 73ff3118 Iustin Pop
    if retry:
349 73ff3118 Iustin Pop
      rval = MAX_RETRIES
350 73ff3118 Iustin Pop
    else:
351 73ff3118 Iustin Pop
      rval = 0
352 ddfde77e Michael Hanselmann
    cli.SetGenericOpcodeOpts(ops, self.opts)
353 73ff3118 Iustin Pop
    return self.MaybeRetry(rval, "opcode", self._ExecOp, *ops)
354 73ff3118 Iustin Pop
355 2c035435 Guido Trotter
  def ExecOrQueue(self, name, ops, post_process=None):
356 21546b1c Iustin Pop
    """Execute an opcode and manage the exec buffer."""
357 c723c163 Iustin Pop
    if self.opts.parallel:
358 ddfde77e Michael Hanselmann
      cli.SetGenericOpcodeOpts(ops, self.opts)
359 2c035435 Guido Trotter
      self.queued_ops.append((ops, name, post_process))
360 c723c163 Iustin Pop
    else:
361 b459a848 Andrea Spadaccini
      val = self.ExecOp(self.queue_retry, *ops) # pylint: disable=W0142
362 2c035435 Guido Trotter
      if post_process is not None:
363 2c035435 Guido Trotter
        post_process()
364 2c035435 Guido Trotter
      return val
365 73ff3118 Iustin Pop
366 73ff3118 Iustin Pop
  def StartBatch(self, retry):
367 73ff3118 Iustin Pop
    """Start a new batch of jobs.
368 73ff3118 Iustin Pop
369 73ff3118 Iustin Pop
    @param retry: whether this is a retryable batch
370 73ff3118 Iustin Pop
371 73ff3118 Iustin Pop
    """
372 73ff3118 Iustin Pop
    self.queued_ops = []
373 73ff3118 Iustin Pop
    self.queue_retry = retry
374 c723c163 Iustin Pop
375 c723c163 Iustin Pop
  def CommitQueue(self):
376 c723c163 Iustin Pop
    """Execute all submitted opcodes in case of parallel burnin"""
377 78bb78b1 Iustin Pop
    if not self.opts.parallel or not self.queued_ops:
378 c723c163 Iustin Pop
      return
379 c723c163 Iustin Pop
380 73ff3118 Iustin Pop
    if self.queue_retry:
381 73ff3118 Iustin Pop
      rval = MAX_RETRIES
382 73ff3118 Iustin Pop
    else:
383 73ff3118 Iustin Pop
      rval = 0
384 73ff3118 Iustin Pop
385 c723c163 Iustin Pop
    try:
386 73ff3118 Iustin Pop
      results = self.MaybeRetry(rval, "jobset", self.ExecJobSet,
387 73ff3118 Iustin Pop
                                self.queued_ops)
388 c723c163 Iustin Pop
    finally:
389 c723c163 Iustin Pop
      self.queued_ops = []
390 c723c163 Iustin Pop
    return results
391 ec5c88dc Iustin Pop
392 ec5c88dc Iustin Pop
  def ExecJobSet(self, jobs):
393 ec5c88dc Iustin Pop
    """Execute a set of jobs and return once all are done.
394 ec5c88dc Iustin Pop
395 ec5c88dc Iustin Pop
    The method will return the list of results, if all jobs are
396 c723c163 Iustin Pop
    successful. Otherwise, OpExecError will be raised from within
397 ec5c88dc Iustin Pop
    cli.py.
398 ec5c88dc Iustin Pop
399 ec5c88dc Iustin Pop
    """
400 ec5c88dc Iustin Pop
    self.ClearFeedbackBuf()
401 14947dbc Iustin Pop
    jex = cli.JobExecutor(cl=self.cl, feedback_fn=self.Feedback)
402 2c035435 Guido Trotter
    for ops, name, _ in jobs:
403 b459a848 Andrea Spadaccini
      jex.QueueJob(name, *ops) # pylint: disable=W0142
404 14947dbc Iustin Pop
    try:
405 14947dbc Iustin Pop
      results = jex.GetResults()
406 b459a848 Andrea Spadaccini
    except Exception, err: # pylint: disable=W0703
407 14947dbc Iustin Pop
      Log("Jobs failed: %s", err)
408 73ff3118 Iustin Pop
      raise BurninFailure()
409 14947dbc Iustin Pop
410 2c035435 Guido Trotter
    fail = False
411 2c035435 Guido Trotter
    val = []
412 2c035435 Guido Trotter
    for (_, name, post_process), (success, result) in zip(jobs, results):
413 2c035435 Guido Trotter
      if success:
414 2c035435 Guido Trotter
        if post_process:
415 2c035435 Guido Trotter
          try:
416 2c035435 Guido Trotter
            post_process()
417 b459a848 Andrea Spadaccini
          except Exception, err: # pylint: disable=W0703
418 2c035435 Guido Trotter
            Log("Post process call for job %s failed: %s", name, err)
419 2c035435 Guido Trotter
            fail = True
420 2c035435 Guido Trotter
        val.append(result)
421 2c035435 Guido Trotter
      else:
422 2c035435 Guido Trotter
        fail = True
423 2c035435 Guido Trotter
424 2c035435 Guido Trotter
    if fail:
425 14947dbc Iustin Pop
      raise BurninFailure()
426 14947dbc Iustin Pop
427 2c035435 Guido Trotter
    return val
428 21546b1c Iustin Pop
429 175f44c2 Iustin Pop
  def ParseOptions(self):
430 175f44c2 Iustin Pop
    """Parses the command line options.
431 175f44c2 Iustin Pop
432 175f44c2 Iustin Pop
    In case of command line errors, it will show the usage and exit the
433 175f44c2 Iustin Pop
    program.
434 175f44c2 Iustin Pop
435 175f44c2 Iustin Pop
    """
436 175f44c2 Iustin Pop
    parser = optparse.OptionParser(usage="\n%s" % USAGE,
437 10eb54fb Michael Hanselmann
                                   version=("%%prog (ganeti) %s" %
438 10eb54fb Michael Hanselmann
                                            constants.RELEASE_VERSION),
439 10eb54fb Michael Hanselmann
                                   option_list=OPTIONS)
440 175f44c2 Iustin Pop
441 175f44c2 Iustin Pop
    options, args = parser.parse_args()
442 175f44c2 Iustin Pop
    if len(args) < 1 or options.os is None:
443 175f44c2 Iustin Pop
      Usage()
444 175f44c2 Iustin Pop
445 f9193417 Iustin Pop
    supported_disk_templates = (constants.DT_DISKLESS,
446 2f505cb5 Manuel Franceschini
                                constants.DT_FILE,
447 9869360c Apollon Oikonomopoulos
                                constants.DT_SHARED_FILE,
448 f9193417 Iustin Pop
                                constants.DT_PLAIN,
449 12c3449a Michael Hanselmann
                                constants.DT_DRBD8)
450 12c3449a Michael Hanselmann
    if options.disk_template not in supported_disk_templates:
451 836d59d7 Iustin Pop
      Err("Unknown disk template '%s'" % options.disk_template)
452 175f44c2 Iustin Pop
453 b518a14a Iustin Pop
    if options.disk_template == constants.DT_DISKLESS:
454 b518a14a Iustin Pop
      disk_size = disk_growth = []
455 5178f1bc Iustin Pop
      options.do_addremove_disks = False
456 b518a14a Iustin Pop
    else:
457 b518a14a Iustin Pop
      disk_size = [utils.ParseUnit(v) for v in options.disk_size.split(",")]
458 b518a14a Iustin Pop
      disk_growth = [utils.ParseUnit(v)
459 b518a14a Iustin Pop
                     for v in options.disk_growth.split(",")]
460 b518a14a Iustin Pop
      if len(disk_growth) != len(disk_size):
461 836d59d7 Iustin Pop
        Err("Wrong disk sizes/growth combination")
462 08db7c5c Iustin Pop
    if ((disk_size and options.disk_template == constants.DT_DISKLESS) or
463 08db7c5c Iustin Pop
        (not disk_size and options.disk_template != constants.DT_DISKLESS)):
464 836d59d7 Iustin Pop
      Err("Wrong disk count/disk template combination")
465 08db7c5c Iustin Pop
466 08db7c5c Iustin Pop
    self.disk_size = disk_size
467 08db7c5c Iustin Pop
    self.disk_growth = disk_growth
468 08db7c5c Iustin Pop
    self.disk_count = len(disk_size)
469 08db7c5c Iustin Pop
470 b91bde14 Iustin Pop
    if options.nodes and options.iallocator:
471 836d59d7 Iustin Pop
      Err("Give either the nodes option or the iallocator option, not both")
472 b91bde14 Iustin Pop
473 544ca43b Iustin Pop
    if options.http_check and not options.name_check:
474 544ca43b Iustin Pop
      Err("Can't enable HTTP checks without name checks")
475 544ca43b Iustin Pop
476 175f44c2 Iustin Pop
    self.opts = options
477 175f44c2 Iustin Pop
    self.instances = args
478 338e51e8 Iustin Pop
    self.bep = {
479 338e51e8 Iustin Pop
      constants.BE_MEMORY: options.mem_size,
480 d0ffa390 Pedro Macedo
      constants.BE_VCPUS: options.vcpu_count,
481 338e51e8 Iustin Pop
      }
482 aac0352d Iustin Pop
483 aac0352d Iustin Pop
    self.hypervisor = None
484 338e51e8 Iustin Pop
    self.hvp = {}
485 aac0352d Iustin Pop
    if options.hypervisor:
486 aac0352d Iustin Pop
      self.hypervisor, self.hvp = options.hypervisor
487 175f44c2 Iustin Pop
488 1d103c02 Iustin Pop
    if options.reboot_types is None:
489 1d103c02 Iustin Pop
      options.reboot_types = constants.REBOOT_TYPES
490 1d103c02 Iustin Pop
    else:
491 1d103c02 Iustin Pop
      options.reboot_types = options.reboot_types.split(",")
492 1d103c02 Iustin Pop
      rt_diff = set(options.reboot_types).difference(constants.REBOOT_TYPES)
493 1d103c02 Iustin Pop
      if rt_diff:
494 1d103c02 Iustin Pop
        Err("Invalid reboot types specified: %s" % utils.CommaJoin(rt_diff))
495 1d103c02 Iustin Pop
496 5178f1bc Iustin Pop
    socket.setdefaulttimeout(options.net_timeout)
497 5178f1bc Iustin Pop
498 175f44c2 Iustin Pop
  def GetState(self):
499 be0636e3 Guido Trotter
    """Read the cluster state from the master daemon."""
500 175f44c2 Iustin Pop
    if self.opts.nodes:
501 175f44c2 Iustin Pop
      names = self.opts.nodes.split(",")
502 175f44c2 Iustin Pop
    else:
503 175f44c2 Iustin Pop
      names = []
504 175f44c2 Iustin Pop
    try:
505 2237687b Iustin Pop
      op = opcodes.OpNodeQuery(output_fields=["name", "offline", "drained"],
506 2237687b Iustin Pop
                               names=names, use_locking=True)
507 73ff3118 Iustin Pop
      result = self.ExecOp(True, op)
508 175f44c2 Iustin Pop
    except errors.GenericError, err:
509 175f44c2 Iustin Pop
      err_code, msg = cli.FormatError(err)
510 836d59d7 Iustin Pop
      Err(msg, exit_code=err_code)
511 e8d47209 Iustin Pop
    self.nodes = [data[0] for data in result if not (data[1] or data[2])]
512 175f44c2 Iustin Pop
513 da2d02e7 Iustin Pop
    op_diagnose = opcodes.OpOsDiagnose(output_fields=["name",
514 eddc9815 Guido Trotter
                                                      "variants",
515 eddc9815 Guido Trotter
                                                      "hidden"],
516 d22dfef7 Iustin Pop
                                       names=[])
517 e3ac208c Guido Trotter
    result = self.ExecOp(True, op_diagnose)
518 175f44c2 Iustin Pop
519 175f44c2 Iustin Pop
    if not result:
520 836d59d7 Iustin Pop
      Err("Can't get the OS list")
521 175f44c2 Iustin Pop
522 e3ac208c Guido Trotter
    found = False
523 eddc9815 Guido Trotter
    for (name, variants, _) in result:
524 d22dfef7 Iustin Pop
      if self.opts.os in cli.CalculateOSNames(name, variants):
525 e3ac208c Guido Trotter
        found = True
526 e3ac208c Guido Trotter
        break
527 175f44c2 Iustin Pop
528 e3ac208c Guido Trotter
    if not found:
529 836d59d7 Iustin Pop
      Err("OS '%s' not found" % self.opts.os)
530 175f44c2 Iustin Pop
531 be0636e3 Guido Trotter
    cluster_info = self.cl.QueryClusterInfo()
532 be0636e3 Guido Trotter
    self.cluster_info = cluster_info
533 be0636e3 Guido Trotter
    if not self.cluster_info:
534 be0636e3 Guido Trotter
      Err("Can't get cluster info")
535 be0636e3 Guido Trotter
536 be0636e3 Guido Trotter
    default_nic_params = self.cluster_info["nicparams"][constants.PP_DEFAULT]
537 be0636e3 Guido Trotter
    self.cluster_default_nicparams = default_nic_params
538 aac0352d Iustin Pop
    if self.hypervisor is None:
539 aac0352d Iustin Pop
      self.hypervisor = self.cluster_info["default_hypervisor"]
540 aac0352d Iustin Pop
    self.hv_class = hypervisor.GetHypervisorClass(self.hypervisor)
541 be0636e3 Guido Trotter
542 d9b7a0b4 Iustin Pop
  @_DoCheckInstances
543 c70481ab Iustin Pop
  @_DoBatch(False)
544 c723c163 Iustin Pop
  def BurnCreateInstances(self):
545 175f44c2 Iustin Pop
    """Create the given instances.
546 175f44c2 Iustin Pop
547 175f44c2 Iustin Pop
    """
548 175f44c2 Iustin Pop
    self.to_rem = []
549 175f44c2 Iustin Pop
    mytor = izip(cycle(self.nodes),
550 175f44c2 Iustin Pop
                 islice(cycle(self.nodes), 1, None),
551 175f44c2 Iustin Pop
                 self.instances)
552 338e51e8 Iustin Pop
553 836d59d7 Iustin Pop
    Log("Creating instances")
554 175f44c2 Iustin Pop
    for pnode, snode, instance in mytor:
555 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
556 b91bde14 Iustin Pop
      if self.opts.iallocator:
557 b91bde14 Iustin Pop
        pnode = snode = None
558 836d59d7 Iustin Pop
        msg = "with iallocator %s" % self.opts.iallocator
559 3429a076 Apollon Oikonomopoulos
      elif self.opts.disk_template not in constants.DTS_INT_MIRROR:
560 6d54548e Guido Trotter
        snode = None
561 836d59d7 Iustin Pop
        msg = "on %s" % pnode
562 6d54548e Guido Trotter
      else:
563 836d59d7 Iustin Pop
        msg = "on %s, %s" % (pnode, snode)
564 836d59d7 Iustin Pop
565 836d59d7 Iustin Pop
      Log(msg, indent=2)
566 6d54548e Guido Trotter
567 e1530b10 Iustin Pop
      op = opcodes.OpInstanceCreate(instance_name=instance,
568 e687ec01 Michael Hanselmann
                                    disks=[{"size": size}
569 e687ec01 Michael Hanselmann
                                           for size in self.disk_size],
570 175f44c2 Iustin Pop
                                    disk_template=self.opts.disk_template,
571 b518a14a Iustin Pop
                                    nics=self.opts.nics,
572 a8083063 Iustin Pop
                                    mode=constants.INSTANCE_CREATE,
573 175f44c2 Iustin Pop
                                    os_type=self.opts.os,
574 175f44c2 Iustin Pop
                                    pnode=pnode,
575 175f44c2 Iustin Pop
                                    snode=snode,
576 a8083063 Iustin Pop
                                    start=True,
577 544ca43b Iustin Pop
                                    ip_check=self.opts.ip_check,
578 544ca43b Iustin Pop
                                    name_check=self.opts.name_check,
579 4501af56 Iustin Pop
                                    wait_for_sync=True,
580 2f505cb5 Manuel Franceschini
                                    file_driver="loop",
581 b91bde14 Iustin Pop
                                    file_storage_dir=None,
582 6e2dc934 Alexander Schreiber
                                    iallocator=self.opts.iallocator,
583 338e51e8 Iustin Pop
                                    beparams=self.bep,
584 338e51e8 Iustin Pop
                                    hvparams=self.hvp,
585 aac0352d Iustin Pop
                                    hypervisor=self.hypervisor,
586 062a7100 Iustin Pop
                                    osparams=self.opts.osparams,
587 338e51e8 Iustin Pop
                                    )
588 1e82a86b Guido Trotter
      remove_instance = lambda name: lambda: self.to_rem.append(name)
589 1e82a86b Guido Trotter
      self.ExecOrQueue(instance, [op], post_process=remove_instance(instance))
590 c723c163 Iustin Pop
591 c70481ab Iustin Pop
  @_DoBatch(False)
592 c723c163 Iustin Pop
  def BurnGrowDisks(self):
593 659712c8 Iustin Pop
    """Grow both the os and the swap disks by the requested amount, if any."""
594 836d59d7 Iustin Pop
    Log("Growing disks")
595 659712c8 Iustin Pop
    for instance in self.instances:
596 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
597 08db7c5c Iustin Pop
      for idx, growth in enumerate(self.disk_growth):
598 659712c8 Iustin Pop
        if growth > 0:
599 60472d29 Iustin Pop
          op = opcodes.OpInstanceGrowDisk(instance_name=instance, disk=idx,
600 60472d29 Iustin Pop
                                          amount=growth, wait_for_sync=True)
601 1b334175 Iustin Pop
          Log("increase disk/%s by %s MB", idx, growth, indent=2)
602 b05f29a6 Guido Trotter
          self.ExecOrQueue(instance, [op])
603 659712c8 Iustin Pop
604 c70481ab Iustin Pop
  @_DoBatch(True)
605 c723c163 Iustin Pop
  def BurnReplaceDisks1D8(self):
606 175f44c2 Iustin Pop
    """Replace disks on primary and secondary for drbd8."""
607 836d59d7 Iustin Pop
    Log("Replacing disks on the same nodes")
608 668f755d Iustin Pop
    early_release = self.opts.early_release
609 175f44c2 Iustin Pop
    for instance in self.instances:
610 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
611 c723c163 Iustin Pop
      ops = []
612 175f44c2 Iustin Pop
      for mode in constants.REPLACE_DISK_SEC, constants.REPLACE_DISK_PRI:
613 668f755d Iustin Pop
        op = opcodes.OpInstanceReplaceDisks(instance_name=instance,
614 668f755d Iustin Pop
                                            mode=mode,
615 668f755d Iustin Pop
                                            disks=list(range(self.disk_count)),
616 668f755d Iustin Pop
                                            early_release=early_release)
617 1b334175 Iustin Pop
        Log("run %s", mode, indent=2)
618 c723c163 Iustin Pop
        ops.append(op)
619 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, ops)
620 175f44c2 Iustin Pop
621 c70481ab Iustin Pop
  @_DoBatch(True)
622 c723c163 Iustin Pop
  def BurnReplaceDisks2(self):
623 175f44c2 Iustin Pop
    """Replace secondary node."""
624 836d59d7 Iustin Pop
    Log("Changing the secondary node")
625 cfacfd6e Iustin Pop
    mode = constants.REPLACE_DISK_CHG
626 175f44c2 Iustin Pop
627 175f44c2 Iustin Pop
    mytor = izip(islice(cycle(self.nodes), 2, None),
628 175f44c2 Iustin Pop
                 self.instances)
629 175f44c2 Iustin Pop
    for tnode, instance in mytor:
630 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
631 b6e82a65 Iustin Pop
      if self.opts.iallocator:
632 b6e82a65 Iustin Pop
        tnode = None
633 836d59d7 Iustin Pop
        msg = "with iallocator %s" % self.opts.iallocator
634 836d59d7 Iustin Pop
      else:
635 836d59d7 Iustin Pop
        msg = tnode
636 668f755d Iustin Pop
      op = opcodes.OpInstanceReplaceDisks(instance_name=instance,
637 668f755d Iustin Pop
                                          mode=mode,
638 668f755d Iustin Pop
                                          remote_node=tnode,
639 668f755d Iustin Pop
                                          iallocator=self.opts.iallocator,
640 668f755d Iustin Pop
                                          disks=[],
641 668f755d Iustin Pop
                                          early_release=self.opts.early_release)
642 1b334175 Iustin Pop
      Log("run %s %s", mode, msg, indent=2)
643 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, [op])
644 175f44c2 Iustin Pop
645 d9b7a0b4 Iustin Pop
  @_DoCheckInstances
646 c70481ab Iustin Pop
  @_DoBatch(False)
647 c723c163 Iustin Pop
  def BurnFailover(self):
648 175f44c2 Iustin Pop
    """Failover the instances."""
649 836d59d7 Iustin Pop
    Log("Failing over instances")
650 175f44c2 Iustin Pop
    for instance in self.instances:
651 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
652 019dbee1 Iustin Pop
      op = opcodes.OpInstanceFailover(instance_name=instance,
653 175f44c2 Iustin Pop
                                      ignore_consistency=False)
654 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, [op])
655 175f44c2 Iustin Pop
656 5b9107ff Iustin Pop
  @_DoCheckInstances
657 5b9107ff Iustin Pop
  @_DoBatch(False)
658 5b9107ff Iustin Pop
  def BurnMove(self):
659 5b9107ff Iustin Pop
    """Move the instances."""
660 5b9107ff Iustin Pop
    Log("Moving instances")
661 5b9107ff Iustin Pop
    mytor = izip(islice(cycle(self.nodes), 1, None),
662 5b9107ff Iustin Pop
                 self.instances)
663 5b9107ff Iustin Pop
    for tnode, instance in mytor:
664 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
665 0091b480 Iustin Pop
      op = opcodes.OpInstanceMove(instance_name=instance,
666 5b9107ff Iustin Pop
                                  target_node=tnode)
667 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, [op])
668 175f44c2 Iustin Pop
669 c70481ab Iustin Pop
  @_DoBatch(False)
670 c723c163 Iustin Pop
  def BurnMigrate(self):
671 99bdd139 Iustin Pop
    """Migrate the instances."""
672 801cda94 Iustin Pop
    Log("Migrating instances")
673 99bdd139 Iustin Pop
    for instance in self.instances:
674 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
675 75c866c2 Iustin Pop
      op1 = opcodes.OpInstanceMigrate(instance_name=instance, mode=None,
676 c723c163 Iustin Pop
                                      cleanup=False)
677 99bdd139 Iustin Pop
678 75c866c2 Iustin Pop
      op2 = opcodes.OpInstanceMigrate(instance_name=instance, mode=None,
679 c723c163 Iustin Pop
                                      cleanup=True)
680 c723c163 Iustin Pop
      Log("migration and migration cleanup", indent=2)
681 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, [op1, op2])
682 99bdd139 Iustin Pop
683 d9b7a0b4 Iustin Pop
  @_DoCheckInstances
684 c70481ab Iustin Pop
  @_DoBatch(False)
685 c723c163 Iustin Pop
  def BurnImportExport(self):
686 bd5e77f9 Guido Trotter
    """Export the instance, delete it, and import it back.
687 bd5e77f9 Guido Trotter
688 bd5e77f9 Guido Trotter
    """
689 836d59d7 Iustin Pop
    Log("Exporting and re-importing instances")
690 bd5e77f9 Guido Trotter
    mytor = izip(cycle(self.nodes),
691 bd5e77f9 Guido Trotter
                 islice(cycle(self.nodes), 1, None),
692 bd5e77f9 Guido Trotter
                 islice(cycle(self.nodes), 2, None),
693 bd5e77f9 Guido Trotter
                 self.instances)
694 bd5e77f9 Guido Trotter
695 bd5e77f9 Guido Trotter
    for pnode, snode, enode, instance in mytor:
696 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
697 c723c163 Iustin Pop
      # read the full name of the instance
698 f2af0bec Iustin Pop
      nam_op = opcodes.OpInstanceQuery(output_fields=["name"],
699 f2af0bec Iustin Pop
                                       names=[instance], use_locking=True)
700 73ff3118 Iustin Pop
      full_name = self.ExecOp(False, nam_op)[0][0]
701 c723c163 Iustin Pop
702 f9af35c8 Guido Trotter
      if self.opts.iallocator:
703 f9af35c8 Guido Trotter
        pnode = snode = None
704 836d59d7 Iustin Pop
        import_log_msg = ("import from %s"
705 836d59d7 Iustin Pop
                          " with iallocator %s" %
706 836d59d7 Iustin Pop
                          (enode, self.opts.iallocator))
707 3429a076 Apollon Oikonomopoulos
      elif self.opts.disk_template not in constants.DTS_INT_MIRROR:
708 f9af35c8 Guido Trotter
        snode = None
709 836d59d7 Iustin Pop
        import_log_msg = ("import from %s to %s" %
710 836d59d7 Iustin Pop
                          (enode, pnode))
711 f9af35c8 Guido Trotter
      else:
712 836d59d7 Iustin Pop
        import_log_msg = ("import from %s to %s, %s" %
713 836d59d7 Iustin Pop
                          (enode, pnode, snode))
714 f9af35c8 Guido Trotter
715 4ff922a2 Iustin Pop
      exp_op = opcodes.OpBackupExport(instance_name=instance,
716 4ff922a2 Iustin Pop
                                      target_node=enode,
717 4ff922a2 Iustin Pop
                                      mode=constants.EXPORT_MODE_LOCAL,
718 4ff922a2 Iustin Pop
                                      shutdown=True)
719 3cd2d4b1 Iustin Pop
      rem_op = opcodes.OpInstanceRemove(instance_name=instance,
720 5c54b832 Iustin Pop
                                        ignore_failures=True)
721 c4feafe8 Iustin Pop
      imp_dir = utils.PathJoin(constants.EXPORT_DIR, full_name)
722 e1530b10 Iustin Pop
      imp_op = opcodes.OpInstanceCreate(instance_name=instance,
723 e687ec01 Michael Hanselmann
                                        disks=[{"size": size}
724 e687ec01 Michael Hanselmann
                                               for size in self.disk_size],
725 bd5e77f9 Guido Trotter
                                        disk_template=self.opts.disk_template,
726 b518a14a Iustin Pop
                                        nics=self.opts.nics,
727 bd5e77f9 Guido Trotter
                                        mode=constants.INSTANCE_IMPORT,
728 bd5e77f9 Guido Trotter
                                        src_node=enode,
729 bd5e77f9 Guido Trotter
                                        src_path=imp_dir,
730 bd5e77f9 Guido Trotter
                                        pnode=pnode,
731 bd5e77f9 Guido Trotter
                                        snode=snode,
732 bd5e77f9 Guido Trotter
                                        start=True,
733 544ca43b Iustin Pop
                                        ip_check=self.opts.ip_check,
734 544ca43b Iustin Pop
                                        name_check=self.opts.name_check,
735 bd5e77f9 Guido Trotter
                                        wait_for_sync=True,
736 96bb2f71 Manuel Franceschini
                                        file_storage_dir=None,
737 0ca35d45 Guido Trotter
                                        file_driver="loop",
738 6e2dc934 Alexander Schreiber
                                        iallocator=self.opts.iallocator,
739 338e51e8 Iustin Pop
                                        beparams=self.bep,
740 338e51e8 Iustin Pop
                                        hvparams=self.hvp,
741 062a7100 Iustin Pop
                                        osparams=self.opts.osparams,
742 338e51e8 Iustin Pop
                                        )
743 6e2dc934 Alexander Schreiber
744 ca5890ad Iustin Pop
      erem_op = opcodes.OpBackupRemove(instance_name=instance)
745 bd5e77f9 Guido Trotter
746 1b334175 Iustin Pop
      Log("export to node %s", enode, indent=2)
747 836d59d7 Iustin Pop
      Log("remove instance", indent=2)
748 836d59d7 Iustin Pop
      Log(import_log_msg, indent=2)
749 836d59d7 Iustin Pop
      Log("remove export", indent=2)
750 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, [exp_op, rem_op, imp_op, erem_op])
751 bd5e77f9 Guido Trotter
752 7e950d31 Iustin Pop
  @staticmethod
753 7e950d31 Iustin Pop
  def StopInstanceOp(instance):
754 054a8696 Manuel Franceschini
    """Stop given instance."""
755 ee3e37a7 Iustin Pop
    return opcodes.OpInstanceShutdown(instance_name=instance)
756 054a8696 Manuel Franceschini
757 7e950d31 Iustin Pop
  @staticmethod
758 7e950d31 Iustin Pop
  def StartInstanceOp(instance):
759 054a8696 Manuel Franceschini
    """Start given instance."""
760 c873d91c Iustin Pop
    return opcodes.OpInstanceStartup(instance_name=instance, force=False)
761 054a8696 Manuel Franceschini
762 7e950d31 Iustin Pop
  @staticmethod
763 7e950d31 Iustin Pop
  def RenameInstanceOp(instance, instance_new):
764 054a8696 Manuel Franceschini
    """Rename instance."""
765 5659e2e2 Iustin Pop
    return opcodes.OpInstanceRename(instance_name=instance,
766 c723c163 Iustin Pop
                                    new_name=instance_new)
767 054a8696 Manuel Franceschini
768 d9b7a0b4 Iustin Pop
  @_DoCheckInstances
769 c70481ab Iustin Pop
  @_DoBatch(True)
770 c723c163 Iustin Pop
  def BurnStopStart(self):
771 175f44c2 Iustin Pop
    """Stop/start the instances."""
772 836d59d7 Iustin Pop
    Log("Stopping and starting instances")
773 175f44c2 Iustin Pop
    for instance in self.instances:
774 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
775 c723c163 Iustin Pop
      op1 = self.StopInstanceOp(instance)
776 c723c163 Iustin Pop
      op2 = self.StartInstanceOp(instance)
777 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, [op1, op2])
778 c723c163 Iustin Pop
779 c70481ab Iustin Pop
  @_DoBatch(False)
780 c723c163 Iustin Pop
  def BurnRemove(self):
781 175f44c2 Iustin Pop
    """Remove the instances."""
782 836d59d7 Iustin Pop
    Log("Removing instances")
783 175f44c2 Iustin Pop
    for instance in self.to_rem:
784 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
785 3cd2d4b1 Iustin Pop
      op = opcodes.OpInstanceRemove(instance_name=instance,
786 5c54b832 Iustin Pop
                                    ignore_failures=True)
787 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, [op])
788 c723c163 Iustin Pop
789 c723c163 Iustin Pop
  def BurnRename(self):
790 c723c163 Iustin Pop
    """Rename the instances.
791 175f44c2 Iustin Pop
792 c723c163 Iustin Pop
    Note that this function will not execute in parallel, since we
793 c723c163 Iustin Pop
    only have one target for rename.
794 c723c163 Iustin Pop
795 c723c163 Iustin Pop
    """
796 836d59d7 Iustin Pop
    Log("Renaming instances")
797 054a8696 Manuel Franceschini
    rename = self.opts.rename
798 054a8696 Manuel Franceschini
    for instance in self.instances:
799 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
800 2e39ab98 Iustin Pop
      op_stop1 = self.StopInstanceOp(instance)
801 2e39ab98 Iustin Pop
      op_stop2 = self.StopInstanceOp(rename)
802 c723c163 Iustin Pop
      op_rename1 = self.RenameInstanceOp(instance, rename)
803 c723c163 Iustin Pop
      op_rename2 = self.RenameInstanceOp(rename, instance)
804 c723c163 Iustin Pop
      op_start1 = self.StartInstanceOp(rename)
805 c723c163 Iustin Pop
      op_start2 = self.StartInstanceOp(instance)
806 73ff3118 Iustin Pop
      self.ExecOp(False, op_stop1, op_rename1, op_start1)
807 5178f1bc Iustin Pop
      self._CheckInstanceAlive(rename)
808 73ff3118 Iustin Pop
      self.ExecOp(False, op_stop2, op_rename2, op_start2)
809 5178f1bc Iustin Pop
      self._CheckInstanceAlive(instance)
810 5178f1bc Iustin Pop
811 d9b7a0b4 Iustin Pop
  @_DoCheckInstances
812 c70481ab Iustin Pop
  @_DoBatch(True)
813 c723c163 Iustin Pop
  def BurnReinstall(self):
814 00f91f29 Iustin Pop
    """Reinstall the instances."""
815 836d59d7 Iustin Pop
    Log("Reinstalling instances")
816 00f91f29 Iustin Pop
    for instance in self.instances:
817 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
818 c723c163 Iustin Pop
      op1 = self.StopInstanceOp(instance)
819 5073fd8f Iustin Pop
      op2 = opcodes.OpInstanceReinstall(instance_name=instance)
820 836d59d7 Iustin Pop
      Log("reinstall without passing the OS", indent=2)
821 5073fd8f Iustin Pop
      op3 = opcodes.OpInstanceReinstall(instance_name=instance,
822 c723c163 Iustin Pop
                                        os_type=self.opts.os)
823 836d59d7 Iustin Pop
      Log("reinstall specifying the OS", indent=2)
824 c723c163 Iustin Pop
      op4 = self.StartInstanceOp(instance)
825 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, [op1, op2, op3, op4])
826 c723c163 Iustin Pop
827 d9b7a0b4 Iustin Pop
  @_DoCheckInstances
828 c70481ab Iustin Pop
  @_DoBatch(True)
829 c723c163 Iustin Pop
  def BurnReboot(self):
830 836d59d7 Iustin Pop
    """Reboot the instances."""
831 836d59d7 Iustin Pop
    Log("Rebooting instances")
832 00f91f29 Iustin Pop
    for instance in self.instances:
833 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
834 c723c163 Iustin Pop
      ops = []
835 1d103c02 Iustin Pop
      for reboot_type in self.opts.reboot_types:
836 90ab1a95 Iustin Pop
        op = opcodes.OpInstanceReboot(instance_name=instance,
837 00f91f29 Iustin Pop
                                      reboot_type=reboot_type,
838 00f91f29 Iustin Pop
                                      ignore_secondaries=False)
839 1b334175 Iustin Pop
        Log("reboot with type '%s'", reboot_type, indent=2)
840 c723c163 Iustin Pop
        ops.append(op)
841 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, ops)
842 c723c163 Iustin Pop
843 d9b7a0b4 Iustin Pop
  @_DoCheckInstances
844 c70481ab Iustin Pop
  @_DoBatch(True)
845 c723c163 Iustin Pop
  def BurnActivateDisks(self):
846 90e722d1 Iustin Pop
    """Activate and deactivate disks of the instances."""
847 836d59d7 Iustin Pop
    Log("Activating/deactivating disks")
848 90e722d1 Iustin Pop
    for instance in self.instances:
849 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
850 c723c163 Iustin Pop
      op_start = self.StartInstanceOp(instance)
851 83f5d475 Iustin Pop
      op_act = opcodes.OpInstanceActivateDisks(instance_name=instance)
852 e176281f Iustin Pop
      op_deact = opcodes.OpInstanceDeactivateDisks(instance_name=instance)
853 c723c163 Iustin Pop
      op_stop = self.StopInstanceOp(instance)
854 836d59d7 Iustin Pop
      Log("activate disks when online", indent=2)
855 836d59d7 Iustin Pop
      Log("activate disks when offline", indent=2)
856 836d59d7 Iustin Pop
      Log("deactivate disks (when offline)", indent=2)
857 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, [op_act, op_stop, op_act, op_deact, op_start])
858 90e722d1 Iustin Pop
859 d9b7a0b4 Iustin Pop
  @_DoCheckInstances
860 c70481ab Iustin Pop
  @_DoBatch(False)
861 c723c163 Iustin Pop
  def BurnAddRemoveDisks(self):
862 5c22d16e Iustin Pop
    """Add and remove an extra disk for the instances."""
863 836d59d7 Iustin Pop
    Log("Adding and removing disks")
864 5c22d16e Iustin Pop
    for instance in self.instances:
865 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
866 9a3cc7ae Iustin Pop
      op_add = opcodes.OpInstanceSetParams(\
867 5c22d16e Iustin Pop
        instance_name=instance,
868 5c22d16e Iustin Pop
        disks=[(constants.DDM_ADD, {"size": self.disk_size[0]})])
869 9a3cc7ae Iustin Pop
      op_rem = opcodes.OpInstanceSetParams(\
870 5c22d16e Iustin Pop
        instance_name=instance, disks=[(constants.DDM_REMOVE, {})])
871 c723c163 Iustin Pop
      op_stop = self.StopInstanceOp(instance)
872 c723c163 Iustin Pop
      op_start = self.StartInstanceOp(instance)
873 836d59d7 Iustin Pop
      Log("adding a disk", indent=2)
874 836d59d7 Iustin Pop
      Log("removing last disk", indent=2)
875 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, [op_add, op_stop, op_rem, op_start])
876 5c22d16e Iustin Pop
877 c70481ab Iustin Pop
  @_DoBatch(False)
878 c723c163 Iustin Pop
  def BurnAddRemoveNICs(self):
879 5c22d16e Iustin Pop
    """Add and remove an extra NIC for the instances."""
880 836d59d7 Iustin Pop
    Log("Adding and removing NICs")
881 5c22d16e Iustin Pop
    for instance in self.instances:
882 1b334175 Iustin Pop
      Log("instance %s", instance, indent=1)
883 9a3cc7ae Iustin Pop
      op_add = opcodes.OpInstanceSetParams(\
884 5c22d16e Iustin Pop
        instance_name=instance, nics=[(constants.DDM_ADD, {})])
885 9a3cc7ae Iustin Pop
      op_rem = opcodes.OpInstanceSetParams(\
886 5c22d16e Iustin Pop
        instance_name=instance, nics=[(constants.DDM_REMOVE, {})])
887 836d59d7 Iustin Pop
      Log("adding a NIC", indent=2)
888 836d59d7 Iustin Pop
      Log("removing last NIC", indent=2)
889 b05f29a6 Guido Trotter
      self.ExecOrQueue(instance, [op_add, op_rem])
890 5c22d16e Iustin Pop
891 a619a1dd Guido Trotter
  def ConfdCallback(self, reply):
892 a619a1dd Guido Trotter
    """Callback for confd queries"""
893 a619a1dd Guido Trotter
    if reply.type == confd_client.UPCALL_REPLY:
894 a619a1dd Guido Trotter
      if reply.server_reply.status != constants.CONFD_REPL_STATUS_OK:
895 a619a1dd Guido Trotter
        Err("Query %s gave non-ok status %s: %s" % (reply.orig_request,
896 a619a1dd Guido Trotter
                                                    reply.server_reply.status,
897 a619a1dd Guido Trotter
                                                    reply.server_reply))
898 a619a1dd Guido Trotter
      if reply.orig_request.type == constants.CONFD_REQ_PING:
899 a619a1dd Guido Trotter
        Log("Ping: OK", indent=1)
900 a619a1dd Guido Trotter
      elif reply.orig_request.type == constants.CONFD_REQ_CLUSTER_MASTER:
901 a619a1dd Guido Trotter
        if reply.server_reply.answer == self.cluster_info["master"]:
902 a619a1dd Guido Trotter
          Log("Master: OK", indent=1)
903 a619a1dd Guido Trotter
        else:
904 a619a1dd Guido Trotter
          Err("Master: wrong: %s" % reply.server_reply.answer)
905 a619a1dd Guido Trotter
      elif reply.orig_request.type == constants.CONFD_REQ_NODE_ROLE_BYNAME:
906 a619a1dd Guido Trotter
        if reply.server_reply.answer == constants.CONFD_NODE_ROLE_MASTER:
907 a619a1dd Guido Trotter
          Log("Node role for master: OK", indent=1)
908 a619a1dd Guido Trotter
        else:
909 a619a1dd Guido Trotter
          Err("Node role for master: wrong: %s" % reply.server_reply.answer)
910 a619a1dd Guido Trotter
911 a619a1dd Guido Trotter
  def DoConfdRequestReply(self, req):
912 a619a1dd Guido Trotter
    self.confd_counting_callback.RegisterQuery(req.rsalt)
913 a619a1dd Guido Trotter
    self.confd_client.SendRequest(req, async=False)
914 a619a1dd Guido Trotter
    while not self.confd_counting_callback.AllAnswered():
915 a619a1dd Guido Trotter
      if not self.confd_client.ReceiveReply():
916 a619a1dd Guido Trotter
        Err("Did not receive all expected confd replies")
917 a619a1dd Guido Trotter
        break
918 a619a1dd Guido Trotter
919 a619a1dd Guido Trotter
  def BurnConfd(self):
920 a619a1dd Guido Trotter
    """Run confd queries for our instances.
921 a619a1dd Guido Trotter
922 a619a1dd Guido Trotter
    The following confd queries are tested:
923 454723b5 Iustin Pop
      - CONFD_REQ_PING: simple ping
924 454723b5 Iustin Pop
      - CONFD_REQ_CLUSTER_MASTER: cluster master
925 454723b5 Iustin Pop
      - CONFD_REQ_NODE_ROLE_BYNAME: node role, for the master
926 a619a1dd Guido Trotter
927 a619a1dd Guido Trotter
    """
928 a619a1dd Guido Trotter
    Log("Checking confd results")
929 a619a1dd Guido Trotter
930 a619a1dd Guido Trotter
    filter_callback = confd_client.ConfdFilterCallback(self.ConfdCallback)
931 a619a1dd Guido Trotter
    counting_callback = confd_client.ConfdCountingCallback(filter_callback)
932 a619a1dd Guido Trotter
    self.confd_counting_callback = counting_callback
933 a619a1dd Guido Trotter
934 5b349fd1 Iustin Pop
    self.confd_client = confd_client.GetConfdClient(counting_callback)
935 a619a1dd Guido Trotter
936 a619a1dd Guido Trotter
    req = confd_client.ConfdClientRequest(type=constants.CONFD_REQ_PING)
937 a619a1dd Guido Trotter
    self.DoConfdRequestReply(req)
938 a619a1dd Guido Trotter
939 a619a1dd Guido Trotter
    req = confd_client.ConfdClientRequest(
940 a619a1dd Guido Trotter
      type=constants.CONFD_REQ_CLUSTER_MASTER)
941 a619a1dd Guido Trotter
    self.DoConfdRequestReply(req)
942 a619a1dd Guido Trotter
943 a619a1dd Guido Trotter
    req = confd_client.ConfdClientRequest(
944 a619a1dd Guido Trotter
        type=constants.CONFD_REQ_NODE_ROLE_BYNAME,
945 a619a1dd Guido Trotter
        query=self.cluster_info["master"])
946 a619a1dd Guido Trotter
    self.DoConfdRequestReply(req)
947 a619a1dd Guido Trotter
948 5178f1bc Iustin Pop
  def _CheckInstanceAlive(self, instance):
949 5178f1bc Iustin Pop
    """Check if an instance is alive by doing http checks.
950 5178f1bc Iustin Pop
951 5178f1bc Iustin Pop
    This will try to retrieve the url on the instance /hostname.txt
952 5178f1bc Iustin Pop
    and check that it contains the hostname of the instance. In case
953 5178f1bc Iustin Pop
    we get ECONNREFUSED, we retry up to the net timeout seconds, for
954 5178f1bc Iustin Pop
    any other error we abort.
955 5178f1bc Iustin Pop
956 5178f1bc Iustin Pop
    """
957 5178f1bc Iustin Pop
    if not self.opts.http_check:
958 5178f1bc Iustin Pop
      return
959 5dc626fd Iustin Pop
    end_time = time.time() + self.opts.net_timeout
960 5dc626fd Iustin Pop
    url = None
961 5dc626fd Iustin Pop
    while time.time() < end_time and url is None:
962 5dc626fd Iustin Pop
      try:
963 5dc626fd Iustin Pop
        url = self.url_opener.open("http://%s/hostname.txt" % instance)
964 c723c163 Iustin Pop
      except IOError:
965 5dc626fd Iustin Pop
        # here we can have connection refused, no route to host, etc.
966 5dc626fd Iustin Pop
        time.sleep(1)
967 5dc626fd Iustin Pop
    if url is None:
968 5dc626fd Iustin Pop
      raise InstanceDown(instance, "Cannot contact instance")
969 5178f1bc Iustin Pop
    hostname = url.read().strip()
970 5dc626fd Iustin Pop
    url.close()
971 5178f1bc Iustin Pop
    if hostname != instance:
972 5178f1bc Iustin Pop
      raise InstanceDown(instance, ("Hostname mismatch, expected %s, got %s" %
973 5178f1bc Iustin Pop
                                    (instance, hostname)))
974 5178f1bc Iustin Pop
975 175f44c2 Iustin Pop
  def BurninCluster(self):
976 175f44c2 Iustin Pop
    """Test a cluster intensively.
977 175f44c2 Iustin Pop
978 175f44c2 Iustin Pop
    This will create instances and then start/stop/failover them.
979 175f44c2 Iustin Pop
    It is safe for existing instances but could impact performance.
980 175f44c2 Iustin Pop
981 175f44c2 Iustin Pop
    """
982 175f44c2 Iustin Pop
983 175f44c2 Iustin Pop
    opts = self.opts
984 175f44c2 Iustin Pop
985 836d59d7 Iustin Pop
    Log("Testing global parameters")
986 175f44c2 Iustin Pop
987 bd249e2f Iustin Pop
    if (len(self.nodes) == 1 and
988 2f505cb5 Manuel Franceschini
        opts.disk_template not in (constants.DT_DISKLESS, constants.DT_PLAIN,
989 9869360c Apollon Oikonomopoulos
                                   constants.DT_FILE,
990 9869360c Apollon Oikonomopoulos
                                   constants.DT_SHARED_FILE)):
991 836d59d7 Iustin Pop
      Err("When one node is available/selected the disk template must"
992 2f505cb5 Manuel Franceschini
          " be 'diskless', 'file' or 'plain'")
993 175f44c2 Iustin Pop
994 21546b1c Iustin Pop
    has_err = True
995 175f44c2 Iustin Pop
    try:
996 c723c163 Iustin Pop
      self.BurnCreateInstances()
997 3429a076 Apollon Oikonomopoulos
      if opts.do_replace1 and opts.disk_template in constants.DTS_INT_MIRROR:
998 c723c163 Iustin Pop
        self.BurnReplaceDisks1D8()
999 175f44c2 Iustin Pop
      if (opts.do_replace2 and len(self.nodes) > 2 and
1000 e687ec01 Michael Hanselmann
          opts.disk_template in constants.DTS_INT_MIRROR):
1001 c723c163 Iustin Pop
        self.BurnReplaceDisks2()
1002 175f44c2 Iustin Pop
1003 728489a3 Guido Trotter
      if (opts.disk_template in constants.DTS_GROWABLE and
1004 403f5172 Guido Trotter
          compat.any(n > 0 for n in self.disk_growth)):
1005 c723c163 Iustin Pop
        self.BurnGrowDisks()
1006 659712c8 Iustin Pop
1007 9869360c Apollon Oikonomopoulos
      if opts.do_failover and opts.disk_template in constants.DTS_MIRRORED:
1008 c723c163 Iustin Pop
        self.BurnFailover()
1009 175f44c2 Iustin Pop
1010 aac0352d Iustin Pop
      if opts.do_migrate:
1011 9869360c Apollon Oikonomopoulos
        if opts.disk_template not in constants.DTS_MIRRORED:
1012 9869360c Apollon Oikonomopoulos
          Log("Skipping migration (disk template %s does not support it)",
1013 9869360c Apollon Oikonomopoulos
              opts.disk_template)
1014 aac0352d Iustin Pop
        elif not self.hv_class.CAN_MIGRATE:
1015 aac0352d Iustin Pop
          Log("Skipping migration (hypervisor %s does not support it)",
1016 aac0352d Iustin Pop
              self.hypervisor)
1017 aac0352d Iustin Pop
        else:
1018 aac0352d Iustin Pop
          self.BurnMigrate()
1019 99bdd139 Iustin Pop
1020 9e32b93b Guido Trotter
      if (opts.do_move and len(self.nodes) > 1 and
1021 9e32b93b Guido Trotter
          opts.disk_template in [constants.DT_PLAIN, constants.DT_FILE]):
1022 5b9107ff Iustin Pop
        self.BurnMove()
1023 5b9107ff Iustin Pop
1024 a5e7be6b Iustin Pop
      if (opts.do_importexport and
1025 a5e7be6b Iustin Pop
          opts.disk_template not in (constants.DT_DISKLESS,
1026 9869360c Apollon Oikonomopoulos
                                     constants.DT_SHARED_FILE,
1027 a5e7be6b Iustin Pop
                                     constants.DT_FILE)):
1028 c723c163 Iustin Pop
        self.BurnImportExport()
1029 bd5e77f9 Guido Trotter
1030 00f91f29 Iustin Pop
      if opts.do_reinstall:
1031 c723c163 Iustin Pop
        self.BurnReinstall()
1032 00f91f29 Iustin Pop
1033 00f91f29 Iustin Pop
      if opts.do_reboot:
1034 c723c163 Iustin Pop
        self.BurnReboot()
1035 00f91f29 Iustin Pop
1036 5c22d16e Iustin Pop
      if opts.do_addremove_disks:
1037 c723c163 Iustin Pop
        self.BurnAddRemoveDisks()
1038 5c22d16e Iustin Pop
1039 be0636e3 Guido Trotter
      default_nic_mode = self.cluster_default_nicparams[constants.NIC_MODE]
1040 be0636e3 Guido Trotter
      # Don't add/remove nics in routed mode, as we would need an ip to add
1041 be0636e3 Guido Trotter
      # them with
1042 5c22d16e Iustin Pop
      if opts.do_addremove_nics:
1043 be0636e3 Guido Trotter
        if default_nic_mode == constants.NIC_MODE_BRIDGED:
1044 be0636e3 Guido Trotter
          self.BurnAddRemoveNICs()
1045 be0636e3 Guido Trotter
        else:
1046 be0636e3 Guido Trotter
          Log("Skipping nic add/remove as the cluster is not in bridged mode")
1047 5c22d16e Iustin Pop
1048 90e722d1 Iustin Pop
      if opts.do_activate_disks:
1049 c723c163 Iustin Pop
        self.BurnActivateDisks()
1050 90e722d1 Iustin Pop
1051 054a8696 Manuel Franceschini
      if opts.rename:
1052 c723c163 Iustin Pop
        self.BurnRename()
1053 054a8696 Manuel Franceschini
1054 a619a1dd Guido Trotter
      if opts.do_confd_tests:
1055 a619a1dd Guido Trotter
        self.BurnConfd()
1056 a619a1dd Guido Trotter
1057 eb61f8d3 Iustin Pop
      if opts.do_startstop:
1058 c723c163 Iustin Pop
        self.BurnStopStart()
1059 eb61f8d3 Iustin Pop
1060 21546b1c Iustin Pop
      has_err = False
1061 175f44c2 Iustin Pop
    finally:
1062 21546b1c Iustin Pop
      if has_err:
1063 21546b1c Iustin Pop
        Log("Error detected: opcode buffer follows:\n\n")
1064 21546b1c Iustin Pop
        Log(self.GetFeedbackBuf())
1065 21546b1c Iustin Pop
        Log("\n\n")
1066 320eda24 Iustin Pop
      if not self.opts.keep_instances:
1067 8629a543 Iustin Pop
        try:
1068 8629a543 Iustin Pop
          self.BurnRemove()
1069 b459a848 Andrea Spadaccini
        except Exception, err:  # pylint: disable=W0703
1070 8629a543 Iustin Pop
          if has_err: # already detected errors, so errors in removal
1071 8629a543 Iustin Pop
                      # are quite expected
1072 1b334175 Iustin Pop
            Log("Note: error detected during instance remove: %s", err)
1073 8629a543 Iustin Pop
          else: # non-expected error
1074 8629a543 Iustin Pop
            raise
1075 175f44c2 Iustin Pop
1076 3f17ef02 Michael Hanselmann
    return constants.EXIT_SUCCESS
1077 a8083063 Iustin Pop
1078 01b69ec5 Michael Hanselmann
1079 a8083063 Iustin Pop
def main():
1080 3f17ef02 Michael Hanselmann
  """Main function.
1081 3ecf6786 Iustin Pop
1082 3f17ef02 Michael Hanselmann
  """
1083 cfcc79c6 Michael Hanselmann
  utils.SetupLogging(constants.LOG_BURNIN, sys.argv[0],
1084 cfcc79c6 Michael Hanselmann
                     debug=False, stderr_logging=True)
1085 3f17ef02 Michael Hanselmann
1086 3f17ef02 Michael Hanselmann
  return Burner().BurninCluster()
1087 a8083063 Iustin Pop
1088 01b69ec5 Michael Hanselmann
1089 a8083063 Iustin Pop
if __name__ == "__main__":
1090 3ecf6786 Iustin Pop
  main()