Statistics
| Branch: | Tag: | Revision:

root / lib / cmdlib / instance.py @ 602db636

History | View | Annotate | Download (145.7 kB)

1 22b7f6f8 Thomas Thrainer
#
2 22b7f6f8 Thomas Thrainer
#
3 22b7f6f8 Thomas Thrainer
4 22b7f6f8 Thomas Thrainer
# Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013 Google Inc.
5 22b7f6f8 Thomas Thrainer
#
6 22b7f6f8 Thomas Thrainer
# This program is free software; you can redistribute it and/or modify
7 22b7f6f8 Thomas Thrainer
# it under the terms of the GNU General Public License as published by
8 22b7f6f8 Thomas Thrainer
# the Free Software Foundation; either version 2 of the License, or
9 22b7f6f8 Thomas Thrainer
# (at your option) any later version.
10 22b7f6f8 Thomas Thrainer
#
11 22b7f6f8 Thomas Thrainer
# This program is distributed in the hope that it will be useful, but
12 22b7f6f8 Thomas Thrainer
# WITHOUT ANY WARRANTY; without even the implied warranty of
13 22b7f6f8 Thomas Thrainer
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 22b7f6f8 Thomas Thrainer
# General Public License for more details.
15 22b7f6f8 Thomas Thrainer
#
16 22b7f6f8 Thomas Thrainer
# You should have received a copy of the GNU General Public License
17 22b7f6f8 Thomas Thrainer
# along with this program; if not, write to the Free Software
18 22b7f6f8 Thomas Thrainer
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
19 22b7f6f8 Thomas Thrainer
# 02110-1301, USA.
20 22b7f6f8 Thomas Thrainer
21 22b7f6f8 Thomas Thrainer
22 22b7f6f8 Thomas Thrainer
"""Logical units dealing with instances."""
23 22b7f6f8 Thomas Thrainer
24 22b7f6f8 Thomas Thrainer
import OpenSSL
25 22b7f6f8 Thomas Thrainer
import copy
26 22b7f6f8 Thomas Thrainer
import logging
27 22b7f6f8 Thomas Thrainer
import os
28 22b7f6f8 Thomas Thrainer
29 22b7f6f8 Thomas Thrainer
from ganeti import compat
30 22b7f6f8 Thomas Thrainer
from ganeti import constants
31 22b7f6f8 Thomas Thrainer
from ganeti import errors
32 22b7f6f8 Thomas Thrainer
from ganeti import ht
33 22b7f6f8 Thomas Thrainer
from ganeti import hypervisor
34 22b7f6f8 Thomas Thrainer
from ganeti import locking
35 22b7f6f8 Thomas Thrainer
from ganeti.masterd import iallocator
36 22b7f6f8 Thomas Thrainer
from ganeti import masterd
37 22b7f6f8 Thomas Thrainer
from ganeti import netutils
38 22b7f6f8 Thomas Thrainer
from ganeti import objects
39 22b7f6f8 Thomas Thrainer
from ganeti import pathutils
40 22b7f6f8 Thomas Thrainer
from ganeti import rpc
41 22b7f6f8 Thomas Thrainer
from ganeti import utils
42 22b7f6f8 Thomas Thrainer
43 8aa8f6b1 Thomas Thrainer
from ganeti.cmdlib.base import NoHooksLU, LogicalUnit, ResultWithJobs
44 22b7f6f8 Thomas Thrainer
45 13f6af81 Thomas Thrainer
from ganeti.cmdlib.common import INSTANCE_DOWN, \
46 5eacbcae Thomas Thrainer
  INSTANCE_NOT_RUNNING, CAN_CHANGE_INSTANCE_OFFLINE, CheckNodeOnline, \
47 5eacbcae Thomas Thrainer
  ShareAll, GetDefaultIAllocator, CheckInstanceNodeGroups, \
48 5eacbcae Thomas Thrainer
  LoadNodeEvacResult, CheckIAllocatorOrNode, CheckParamsNotGlobal, \
49 5eacbcae Thomas Thrainer
  IsExclusiveStorageEnabledNode, CheckHVParams, CheckOSParams, \
50 da4a52a3 Thomas Thrainer
  AnnotateDiskParams, GetUpdatedParams, ExpandInstanceUuidAndName, \
51 1f7c8208 Helga Velroyen
  ComputeIPolicySpecViolation, CheckInstanceState, ExpandNodeUuidAndName, \
52 294254b1 Raffa Santi
  CheckDiskTemplateEnabled, IsValidDiskAccessModeCombination
53 5eacbcae Thomas Thrainer
from ganeti.cmdlib.instance_storage import CreateDisks, \
54 a365b47f Bernardo Dal Seno
  CheckNodesFreeDiskPerVG, WipeDisks, WipeOrCleanupDisks, WaitForSync, \
55 1c3231aa Thomas Thrainer
  IsExclusiveStorageEnabledNodeUuid, CreateSingleBlockDev, ComputeDisks, \
56 5eacbcae Thomas Thrainer
  CheckRADOSFreeSpace, ComputeDiskSizePerVG, GenerateDiskTemplate, \
57 3f3ea14c Bernardo Dal Seno
  StartInstanceDisks, ShutdownInstanceDisks, AssembleInstanceDisks, \
58 3f3ea14c Bernardo Dal Seno
  CheckSpindlesExclusiveStorage
59 5eacbcae Thomas Thrainer
from ganeti.cmdlib.instance_utils import BuildInstanceHookEnvByObject, \
60 5eacbcae Thomas Thrainer
  GetClusterDomainSecret, BuildInstanceHookEnv, NICListToTuple, \
61 5eacbcae Thomas Thrainer
  NICToTuple, CheckNodeNotDrained, RemoveInstance, CopyLockList, \
62 5eacbcae Thomas Thrainer
  ReleaseLocks, CheckNodeVmCapable, CheckTargetNodeIPolicy, \
63 5eacbcae Thomas Thrainer
  GetInstanceInfoText, RemoveDisks, CheckNodeFreeMemory, \
64 5eacbcae Thomas Thrainer
  CheckInstanceBridgesExist, CheckNicsBridgesExist, CheckNodeHasOS
65 22b7f6f8 Thomas Thrainer
66 22b7f6f8 Thomas Thrainer
import ganeti.masterd.instance
67 22b7f6f8 Thomas Thrainer
68 22b7f6f8 Thomas Thrainer
69 5eacbcae Thomas Thrainer
#: Type description for changes as returned by L{_ApplyContainerMods}'s
70 22b7f6f8 Thomas Thrainer
#: callbacks
71 22b7f6f8 Thomas Thrainer
_TApplyContModsCbChanges = \
72 22b7f6f8 Thomas Thrainer
  ht.TMaybeListOf(ht.TAnd(ht.TIsLength(2), ht.TItems([
73 22b7f6f8 Thomas Thrainer
    ht.TNonEmptyString,
74 22b7f6f8 Thomas Thrainer
    ht.TAny,
75 22b7f6f8 Thomas Thrainer
    ])))
76 22b7f6f8 Thomas Thrainer
77 22b7f6f8 Thomas Thrainer
78 22b7f6f8 Thomas Thrainer
def _CheckHostnameSane(lu, name):
79 22b7f6f8 Thomas Thrainer
  """Ensures that a given hostname resolves to a 'sane' name.
80 22b7f6f8 Thomas Thrainer

81 22b7f6f8 Thomas Thrainer
  The given name is required to be a prefix of the resolved hostname,
82 22b7f6f8 Thomas Thrainer
  to prevent accidental mismatches.
83 22b7f6f8 Thomas Thrainer

84 22b7f6f8 Thomas Thrainer
  @param lu: the logical unit on behalf of which we're checking
85 22b7f6f8 Thomas Thrainer
  @param name: the name we should resolve and check
86 22b7f6f8 Thomas Thrainer
  @return: the resolved hostname object
87 22b7f6f8 Thomas Thrainer

88 22b7f6f8 Thomas Thrainer
  """
89 22b7f6f8 Thomas Thrainer
  hostname = netutils.GetHostname(name=name)
90 22b7f6f8 Thomas Thrainer
  if hostname.name != name:
91 22b7f6f8 Thomas Thrainer
    lu.LogInfo("Resolved given name '%s' to '%s'", name, hostname.name)
92 22b7f6f8 Thomas Thrainer
  if not utils.MatchNameComponent(name, [hostname.name]):
93 22b7f6f8 Thomas Thrainer
    raise errors.OpPrereqError(("Resolved hostname '%s' does not look the"
94 22b7f6f8 Thomas Thrainer
                                " same as given hostname '%s'") %
95 22b7f6f8 Thomas Thrainer
                               (hostname.name, name), errors.ECODE_INVAL)
96 22b7f6f8 Thomas Thrainer
  return hostname
97 22b7f6f8 Thomas Thrainer
98 22b7f6f8 Thomas Thrainer
99 22b7f6f8 Thomas Thrainer
def _CheckOpportunisticLocking(op):
100 22b7f6f8 Thomas Thrainer
  """Generate error if opportunistic locking is not possible.
101 22b7f6f8 Thomas Thrainer

102 22b7f6f8 Thomas Thrainer
  """
103 22b7f6f8 Thomas Thrainer
  if op.opportunistic_locking and not op.iallocator:
104 22b7f6f8 Thomas Thrainer
    raise errors.OpPrereqError("Opportunistic locking is only available in"
105 22b7f6f8 Thomas Thrainer
                               " combination with an instance allocator",
106 22b7f6f8 Thomas Thrainer
                               errors.ECODE_INVAL)
107 22b7f6f8 Thomas Thrainer
108 22b7f6f8 Thomas Thrainer
109 1c3231aa Thomas Thrainer
def _CreateInstanceAllocRequest(op, disks, nics, beparams, node_name_whitelist):
110 22b7f6f8 Thomas Thrainer
  """Wrapper around IAReqInstanceAlloc.
111 22b7f6f8 Thomas Thrainer

112 22b7f6f8 Thomas Thrainer
  @param op: The instance opcode
113 22b7f6f8 Thomas Thrainer
  @param disks: The computed disks
114 22b7f6f8 Thomas Thrainer
  @param nics: The computed nics
115 22b7f6f8 Thomas Thrainer
  @param beparams: The full filled beparams
116 1c3231aa Thomas Thrainer
  @param node_name_whitelist: List of nodes which should appear as online to the
117 22b7f6f8 Thomas Thrainer
    allocator (unless the node is already marked offline)
118 22b7f6f8 Thomas Thrainer

119 22b7f6f8 Thomas Thrainer
  @returns: A filled L{iallocator.IAReqInstanceAlloc}
120 22b7f6f8 Thomas Thrainer

121 22b7f6f8 Thomas Thrainer
  """
122 22b7f6f8 Thomas Thrainer
  spindle_use = beparams[constants.BE_SPINDLE_USE]
123 22b7f6f8 Thomas Thrainer
  return iallocator.IAReqInstanceAlloc(name=op.instance_name,
124 22b7f6f8 Thomas Thrainer
                                       disk_template=op.disk_template,
125 22b7f6f8 Thomas Thrainer
                                       tags=op.tags,
126 22b7f6f8 Thomas Thrainer
                                       os=op.os_type,
127 22b7f6f8 Thomas Thrainer
                                       vcpus=beparams[constants.BE_VCPUS],
128 22b7f6f8 Thomas Thrainer
                                       memory=beparams[constants.BE_MAXMEM],
129 22b7f6f8 Thomas Thrainer
                                       spindle_use=spindle_use,
130 22b7f6f8 Thomas Thrainer
                                       disks=disks,
131 22b7f6f8 Thomas Thrainer
                                       nics=[n.ToDict() for n in nics],
132 22b7f6f8 Thomas Thrainer
                                       hypervisor=op.hypervisor,
133 1c3231aa Thomas Thrainer
                                       node_whitelist=node_name_whitelist)
134 22b7f6f8 Thomas Thrainer
135 22b7f6f8 Thomas Thrainer
136 22b7f6f8 Thomas Thrainer
def _ComputeFullBeParams(op, cluster):
137 22b7f6f8 Thomas Thrainer
  """Computes the full beparams.
138 22b7f6f8 Thomas Thrainer

139 22b7f6f8 Thomas Thrainer
  @param op: The instance opcode
140 22b7f6f8 Thomas Thrainer
  @param cluster: The cluster config object
141 22b7f6f8 Thomas Thrainer

142 22b7f6f8 Thomas Thrainer
  @return: The fully filled beparams
143 22b7f6f8 Thomas Thrainer

144 22b7f6f8 Thomas Thrainer
  """
145 22b7f6f8 Thomas Thrainer
  default_beparams = cluster.beparams[constants.PP_DEFAULT]
146 22b7f6f8 Thomas Thrainer
  for param, value in op.beparams.iteritems():
147 22b7f6f8 Thomas Thrainer
    if value == constants.VALUE_AUTO:
148 22b7f6f8 Thomas Thrainer
      op.beparams[param] = default_beparams[param]
149 22b7f6f8 Thomas Thrainer
  objects.UpgradeBeParams(op.beparams)
150 22b7f6f8 Thomas Thrainer
  utils.ForceDictType(op.beparams, constants.BES_PARAMETER_TYPES)
151 22b7f6f8 Thomas Thrainer
  return cluster.SimpleFillBE(op.beparams)
152 22b7f6f8 Thomas Thrainer
153 22b7f6f8 Thomas Thrainer
154 22b7f6f8 Thomas Thrainer
def _ComputeNics(op, cluster, default_ip, cfg, ec_id):
155 22b7f6f8 Thomas Thrainer
  """Computes the nics.
156 22b7f6f8 Thomas Thrainer

157 22b7f6f8 Thomas Thrainer
  @param op: The instance opcode
158 22b7f6f8 Thomas Thrainer
  @param cluster: Cluster configuration object
159 22b7f6f8 Thomas Thrainer
  @param default_ip: The default ip to assign
160 22b7f6f8 Thomas Thrainer
  @param cfg: An instance of the configuration object
161 22b7f6f8 Thomas Thrainer
  @param ec_id: Execution context ID
162 22b7f6f8 Thomas Thrainer

163 22b7f6f8 Thomas Thrainer
  @returns: The build up nics
164 22b7f6f8 Thomas Thrainer

165 22b7f6f8 Thomas Thrainer
  """
166 22b7f6f8 Thomas Thrainer
  nics = []
167 22b7f6f8 Thomas Thrainer
  for nic in op.nics:
168 22b7f6f8 Thomas Thrainer
    nic_mode_req = nic.get(constants.INIC_MODE, None)
169 22b7f6f8 Thomas Thrainer
    nic_mode = nic_mode_req
170 22b7f6f8 Thomas Thrainer
    if nic_mode is None or nic_mode == constants.VALUE_AUTO:
171 22b7f6f8 Thomas Thrainer
      nic_mode = cluster.nicparams[constants.PP_DEFAULT][constants.NIC_MODE]
172 22b7f6f8 Thomas Thrainer
173 22b7f6f8 Thomas Thrainer
    net = nic.get(constants.INIC_NETWORK, None)
174 22b7f6f8 Thomas Thrainer
    link = nic.get(constants.NIC_LINK, None)
175 22b7f6f8 Thomas Thrainer
    ip = nic.get(constants.INIC_IP, None)
176 9f7d5fe4 Sebastian Gebhard
    vlan = nic.get(constants.INIC_VLAN, None)
177 22b7f6f8 Thomas Thrainer
178 22b7f6f8 Thomas Thrainer
    if net is None or net.lower() == constants.VALUE_NONE:
179 22b7f6f8 Thomas Thrainer
      net = None
180 22b7f6f8 Thomas Thrainer
    else:
181 22b7f6f8 Thomas Thrainer
      if nic_mode_req is not None or link is not None:
182 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("If network is given, no mode or link"
183 22b7f6f8 Thomas Thrainer
                                   " is allowed to be passed",
184 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
185 22b7f6f8 Thomas Thrainer
186 22b7f6f8 Thomas Thrainer
    # ip validity checks
187 22b7f6f8 Thomas Thrainer
    if ip is None or ip.lower() == constants.VALUE_NONE:
188 22b7f6f8 Thomas Thrainer
      nic_ip = None
189 22b7f6f8 Thomas Thrainer
    elif ip.lower() == constants.VALUE_AUTO:
190 22b7f6f8 Thomas Thrainer
      if not op.name_check:
191 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("IP address set to auto but name checks"
192 22b7f6f8 Thomas Thrainer
                                   " have been skipped",
193 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
194 22b7f6f8 Thomas Thrainer
      nic_ip = default_ip
195 22b7f6f8 Thomas Thrainer
    else:
196 22b7f6f8 Thomas Thrainer
      # We defer pool operations until later, so that the iallocator has
197 22b7f6f8 Thomas Thrainer
      # filled in the instance's node(s) dimara
198 22b7f6f8 Thomas Thrainer
      if ip.lower() == constants.NIC_IP_POOL:
199 22b7f6f8 Thomas Thrainer
        if net is None:
200 22b7f6f8 Thomas Thrainer
          raise errors.OpPrereqError("if ip=pool, parameter network"
201 22b7f6f8 Thomas Thrainer
                                     " must be passed too",
202 22b7f6f8 Thomas Thrainer
                                     errors.ECODE_INVAL)
203 22b7f6f8 Thomas Thrainer
204 22b7f6f8 Thomas Thrainer
      elif not netutils.IPAddress.IsValid(ip):
205 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Invalid IP address '%s'" % ip,
206 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
207 22b7f6f8 Thomas Thrainer
208 22b7f6f8 Thomas Thrainer
      nic_ip = ip
209 22b7f6f8 Thomas Thrainer
210 22b7f6f8 Thomas Thrainer
    # TODO: check the ip address for uniqueness
211 22b7f6f8 Thomas Thrainer
    if nic_mode == constants.NIC_MODE_ROUTED and not nic_ip:
212 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Routed nic mode requires an ip address",
213 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_INVAL)
214 22b7f6f8 Thomas Thrainer
215 22b7f6f8 Thomas Thrainer
    # MAC address verification
216 22b7f6f8 Thomas Thrainer
    mac = nic.get(constants.INIC_MAC, constants.VALUE_AUTO)
217 22b7f6f8 Thomas Thrainer
    if mac not in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
218 22b7f6f8 Thomas Thrainer
      mac = utils.NormalizeAndValidateMac(mac)
219 22b7f6f8 Thomas Thrainer
220 22b7f6f8 Thomas Thrainer
      try:
221 22b7f6f8 Thomas Thrainer
        # TODO: We need to factor this out
222 22b7f6f8 Thomas Thrainer
        cfg.ReserveMAC(mac, ec_id)
223 22b7f6f8 Thomas Thrainer
      except errors.ReservationError:
224 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("MAC address %s already in use"
225 22b7f6f8 Thomas Thrainer
                                   " in cluster" % mac,
226 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_NOTUNIQUE)
227 22b7f6f8 Thomas Thrainer
228 22b7f6f8 Thomas Thrainer
    #  Build nic parameters
229 22b7f6f8 Thomas Thrainer
    nicparams = {}
230 22b7f6f8 Thomas Thrainer
    if nic_mode_req:
231 22b7f6f8 Thomas Thrainer
      nicparams[constants.NIC_MODE] = nic_mode
232 22b7f6f8 Thomas Thrainer
    if link:
233 22b7f6f8 Thomas Thrainer
      nicparams[constants.NIC_LINK] = link
234 9f7d5fe4 Sebastian Gebhard
    if vlan:
235 9f7d5fe4 Sebastian Gebhard
      nicparams[constants.NIC_VLAN] = vlan
236 22b7f6f8 Thomas Thrainer
237 22b7f6f8 Thomas Thrainer
    check_params = cluster.SimpleFillNIC(nicparams)
238 22b7f6f8 Thomas Thrainer
    objects.NIC.CheckParameterSyntax(check_params)
239 22b7f6f8 Thomas Thrainer
    net_uuid = cfg.LookupNetwork(net)
240 22b7f6f8 Thomas Thrainer
    name = nic.get(constants.INIC_NAME, None)
241 22b7f6f8 Thomas Thrainer
    if name is not None and name.lower() == constants.VALUE_NONE:
242 22b7f6f8 Thomas Thrainer
      name = None
243 22b7f6f8 Thomas Thrainer
    nic_obj = objects.NIC(mac=mac, ip=nic_ip, name=name,
244 22b7f6f8 Thomas Thrainer
                          network=net_uuid, nicparams=nicparams)
245 22b7f6f8 Thomas Thrainer
    nic_obj.uuid = cfg.GenerateUniqueID(ec_id)
246 22b7f6f8 Thomas Thrainer
    nics.append(nic_obj)
247 22b7f6f8 Thomas Thrainer
248 22b7f6f8 Thomas Thrainer
  return nics
249 22b7f6f8 Thomas Thrainer
250 22b7f6f8 Thomas Thrainer
251 1c3231aa Thomas Thrainer
def _CheckForConflictingIp(lu, ip, node_uuid):
252 22b7f6f8 Thomas Thrainer
  """In case of conflicting IP address raise error.
253 22b7f6f8 Thomas Thrainer

254 22b7f6f8 Thomas Thrainer
  @type ip: string
255 22b7f6f8 Thomas Thrainer
  @param ip: IP address
256 1c3231aa Thomas Thrainer
  @type node_uuid: string
257 1c3231aa Thomas Thrainer
  @param node_uuid: node UUID
258 22b7f6f8 Thomas Thrainer

259 22b7f6f8 Thomas Thrainer
  """
260 1c3231aa Thomas Thrainer
  (conf_net, _) = lu.cfg.CheckIPInNodeGroup(ip, node_uuid)
261 22b7f6f8 Thomas Thrainer
  if conf_net is not None:
262 22b7f6f8 Thomas Thrainer
    raise errors.OpPrereqError(("The requested IP address (%s) belongs to"
263 22b7f6f8 Thomas Thrainer
                                " network %s, but the target NIC does not." %
264 22b7f6f8 Thomas Thrainer
                                (ip, conf_net)),
265 22b7f6f8 Thomas Thrainer
                               errors.ECODE_STATE)
266 22b7f6f8 Thomas Thrainer
267 22b7f6f8 Thomas Thrainer
  return (None, None)
268 22b7f6f8 Thomas Thrainer
269 22b7f6f8 Thomas Thrainer
270 22b7f6f8 Thomas Thrainer
def _ComputeIPolicyInstanceSpecViolation(
271 22b7f6f8 Thomas Thrainer
  ipolicy, instance_spec, disk_template,
272 5eacbcae Thomas Thrainer
  _compute_fn=ComputeIPolicySpecViolation):
273 22b7f6f8 Thomas Thrainer
  """Compute if instance specs meets the specs of ipolicy.
274 22b7f6f8 Thomas Thrainer

275 22b7f6f8 Thomas Thrainer
  @type ipolicy: dict
276 22b7f6f8 Thomas Thrainer
  @param ipolicy: The ipolicy to verify against
277 22b7f6f8 Thomas Thrainer
  @param instance_spec: dict
278 22b7f6f8 Thomas Thrainer
  @param instance_spec: The instance spec to verify
279 22b7f6f8 Thomas Thrainer
  @type disk_template: string
280 22b7f6f8 Thomas Thrainer
  @param disk_template: the disk template of the instance
281 22b7f6f8 Thomas Thrainer
  @param _compute_fn: The function to verify ipolicy (unittest only)
282 5eacbcae Thomas Thrainer
  @see: L{ComputeIPolicySpecViolation}
283 22b7f6f8 Thomas Thrainer

284 22b7f6f8 Thomas Thrainer
  """
285 22b7f6f8 Thomas Thrainer
  mem_size = instance_spec.get(constants.ISPEC_MEM_SIZE, None)
286 22b7f6f8 Thomas Thrainer
  cpu_count = instance_spec.get(constants.ISPEC_CPU_COUNT, None)
287 22b7f6f8 Thomas Thrainer
  disk_count = instance_spec.get(constants.ISPEC_DISK_COUNT, 0)
288 22b7f6f8 Thomas Thrainer
  disk_sizes = instance_spec.get(constants.ISPEC_DISK_SIZE, [])
289 22b7f6f8 Thomas Thrainer
  nic_count = instance_spec.get(constants.ISPEC_NIC_COUNT, 0)
290 22b7f6f8 Thomas Thrainer
  spindle_use = instance_spec.get(constants.ISPEC_SPINDLE_USE, None)
291 22b7f6f8 Thomas Thrainer
292 22b7f6f8 Thomas Thrainer
  return _compute_fn(ipolicy, mem_size, cpu_count, disk_count, nic_count,
293 22b7f6f8 Thomas Thrainer
                     disk_sizes, spindle_use, disk_template)
294 22b7f6f8 Thomas Thrainer
295 22b7f6f8 Thomas Thrainer
296 22b7f6f8 Thomas Thrainer
def _CheckOSVariant(os_obj, name):
297 22b7f6f8 Thomas Thrainer
  """Check whether an OS name conforms to the os variants specification.
298 22b7f6f8 Thomas Thrainer

299 22b7f6f8 Thomas Thrainer
  @type os_obj: L{objects.OS}
300 22b7f6f8 Thomas Thrainer
  @param os_obj: OS object to check
301 22b7f6f8 Thomas Thrainer
  @type name: string
302 22b7f6f8 Thomas Thrainer
  @param name: OS name passed by the user, to check for validity
303 22b7f6f8 Thomas Thrainer

304 22b7f6f8 Thomas Thrainer
  """
305 22b7f6f8 Thomas Thrainer
  variant = objects.OS.GetVariant(name)
306 22b7f6f8 Thomas Thrainer
  if not os_obj.supported_variants:
307 22b7f6f8 Thomas Thrainer
    if variant:
308 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("OS '%s' doesn't support variants ('%s'"
309 22b7f6f8 Thomas Thrainer
                                 " passed)" % (os_obj.name, variant),
310 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_INVAL)
311 22b7f6f8 Thomas Thrainer
    return
312 22b7f6f8 Thomas Thrainer
  if not variant:
313 22b7f6f8 Thomas Thrainer
    raise errors.OpPrereqError("OS name must include a variant",
314 22b7f6f8 Thomas Thrainer
                               errors.ECODE_INVAL)
315 22b7f6f8 Thomas Thrainer
316 22b7f6f8 Thomas Thrainer
  if variant not in os_obj.supported_variants:
317 22b7f6f8 Thomas Thrainer
    raise errors.OpPrereqError("Unsupported OS variant", errors.ECODE_INVAL)
318 22b7f6f8 Thomas Thrainer
319 22b7f6f8 Thomas Thrainer
320 22b7f6f8 Thomas Thrainer
class LUInstanceCreate(LogicalUnit):
321 22b7f6f8 Thomas Thrainer
  """Create an instance.
322 22b7f6f8 Thomas Thrainer

323 22b7f6f8 Thomas Thrainer
  """
324 22b7f6f8 Thomas Thrainer
  HPATH = "instance-add"
325 22b7f6f8 Thomas Thrainer
  HTYPE = constants.HTYPE_INSTANCE
326 22b7f6f8 Thomas Thrainer
  REQ_BGL = False
327 22b7f6f8 Thomas Thrainer
328 dab6ea3d Helga Velroyen
  def _CheckDiskTemplateValid(self):
329 dab6ea3d Helga Velroyen
    """Checks validity of disk template.
330 22b7f6f8 Thomas Thrainer

331 22b7f6f8 Thomas Thrainer
    """
332 22b7f6f8 Thomas Thrainer
    cluster = self.cfg.GetClusterInfo()
333 72080db1 Helga Velroyen
    if self.op.disk_template is None:
334 72080db1 Helga Velroyen
      # FIXME: It would be better to take the default disk template from the
335 72080db1 Helga Velroyen
      # ipolicy, but for the ipolicy we need the primary node, which we get from
336 72080db1 Helga Velroyen
      # the iallocator, which wants the disk template as input. To solve this
337 72080db1 Helga Velroyen
      # chicken-and-egg problem, it should be possible to specify just a node
338 72080db1 Helga Velroyen
      # group from the iallocator and take the ipolicy from that.
339 72080db1 Helga Velroyen
      self.op.disk_template = cluster.enabled_disk_templates[0]
340 1f7c8208 Helga Velroyen
    CheckDiskTemplateEnabled(cluster, self.op.disk_template)
341 22b7f6f8 Thomas Thrainer
342 dab6ea3d Helga Velroyen
  def _CheckDiskArguments(self):
343 dab6ea3d Helga Velroyen
    """Checks validity of disk-related arguments.
344 dab6ea3d Helga Velroyen

345 dab6ea3d Helga Velroyen
    """
346 dab6ea3d Helga Velroyen
    # check that disk's names are unique and valid
347 dab6ea3d Helga Velroyen
    utils.ValidateDeviceNames("disk", self.op.disks)
348 dab6ea3d Helga Velroyen
349 dab6ea3d Helga Velroyen
    self._CheckDiskTemplateValid()
350 dab6ea3d Helga Velroyen
351 22b7f6f8 Thomas Thrainer
    # check disks. parameter names and consistent adopt/no-adopt strategy
352 22b7f6f8 Thomas Thrainer
    has_adopt = has_no_adopt = False
353 22b7f6f8 Thomas Thrainer
    for disk in self.op.disks:
354 22b7f6f8 Thomas Thrainer
      if self.op.disk_template != constants.DT_EXT:
355 22b7f6f8 Thomas Thrainer
        utils.ForceDictType(disk, constants.IDISK_PARAMS_TYPES)
356 22b7f6f8 Thomas Thrainer
      if constants.IDISK_ADOPT in disk:
357 22b7f6f8 Thomas Thrainer
        has_adopt = True
358 22b7f6f8 Thomas Thrainer
      else:
359 22b7f6f8 Thomas Thrainer
        has_no_adopt = True
360 22b7f6f8 Thomas Thrainer
    if has_adopt and has_no_adopt:
361 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Either all disks are adopted or none is",
362 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_INVAL)
363 22b7f6f8 Thomas Thrainer
    if has_adopt:
364 22b7f6f8 Thomas Thrainer
      if self.op.disk_template not in constants.DTS_MAY_ADOPT:
365 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Disk adoption is not supported for the"
366 22b7f6f8 Thomas Thrainer
                                   " '%s' disk template" %
367 22b7f6f8 Thomas Thrainer
                                   self.op.disk_template,
368 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
369 22b7f6f8 Thomas Thrainer
      if self.op.iallocator is not None:
370 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Disk adoption not allowed with an"
371 22b7f6f8 Thomas Thrainer
                                   " iallocator script", errors.ECODE_INVAL)
372 22b7f6f8 Thomas Thrainer
      if self.op.mode == constants.INSTANCE_IMPORT:
373 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Disk adoption not allowed for"
374 22b7f6f8 Thomas Thrainer
                                   " instance import", errors.ECODE_INVAL)
375 22b7f6f8 Thomas Thrainer
    else:
376 22b7f6f8 Thomas Thrainer
      if self.op.disk_template in constants.DTS_MUST_ADOPT:
377 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Disk template %s requires disk adoption,"
378 22b7f6f8 Thomas Thrainer
                                   " but no 'adopt' parameter given" %
379 22b7f6f8 Thomas Thrainer
                                   self.op.disk_template,
380 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
381 22b7f6f8 Thomas Thrainer
382 22b7f6f8 Thomas Thrainer
    self.adopt_disks = has_adopt
383 22b7f6f8 Thomas Thrainer
384 9f7d5fe4 Sebastian Gebhard
  def _CheckVLANArguments(self):
385 9f7d5fe4 Sebastian Gebhard
    """ Check validity of VLANs if given
386 9f7d5fe4 Sebastian Gebhard

387 9f7d5fe4 Sebastian Gebhard
    """
388 9f7d5fe4 Sebastian Gebhard
    for nic in self.op.nics:
389 5ce6fa9a Sebastian Gebhard
      vlan = nic.get(constants.INIC_VLAN, None)
390 5ce6fa9a Sebastian Gebhard
      if vlan:
391 9f7d5fe4 Sebastian Gebhard
        if vlan[0] == ".":
392 9f7d5fe4 Sebastian Gebhard
          # vlan starting with dot means single untagged vlan,
393 9f7d5fe4 Sebastian Gebhard
          # might be followed by trunk (:)
394 9f7d5fe4 Sebastian Gebhard
          if not vlan[1:].isdigit():
395 9f7d5fe4 Sebastian Gebhard
            vlanlist = vlan[1:].split(':')
396 9f7d5fe4 Sebastian Gebhard
            for vl in vlanlist:
397 9f7d5fe4 Sebastian Gebhard
              if not vl.isdigit():
398 9f7d5fe4 Sebastian Gebhard
                raise errors.OpPrereqError("Specified VLAN parameter is "
399 9f7d5fe4 Sebastian Gebhard
                                           "invalid : %s" % vlan,
400 9f7d5fe4 Sebastian Gebhard
                                             errors.ECODE_INVAL)
401 9f7d5fe4 Sebastian Gebhard
        elif vlan[0] == ":":
402 9f7d5fe4 Sebastian Gebhard
          # Trunk - tagged only
403 9f7d5fe4 Sebastian Gebhard
          vlanlist = vlan[1:].split(':')
404 9f7d5fe4 Sebastian Gebhard
          for vl in vlanlist:
405 9f7d5fe4 Sebastian Gebhard
            if not vl.isdigit():
406 9f7d5fe4 Sebastian Gebhard
              raise errors.OpPrereqError("Specified VLAN parameter is invalid"
407 9f7d5fe4 Sebastian Gebhard
                                           " : %s" % vlan, errors.ECODE_INVAL)
408 9f7d5fe4 Sebastian Gebhard
        elif vlan.isdigit():
409 9f7d5fe4 Sebastian Gebhard
          # This is the simplest case. No dots, only single digit
410 9f7d5fe4 Sebastian Gebhard
          # -> Create untagged access port, dot needs to be added
411 9f7d5fe4 Sebastian Gebhard
          nic[constants.INIC_VLAN] = "." + vlan
412 9f7d5fe4 Sebastian Gebhard
        else:
413 9f7d5fe4 Sebastian Gebhard
          raise errors.OpPrereqError("Specified VLAN parameter is invalid"
414 9f7d5fe4 Sebastian Gebhard
                                       " : %s" % vlan, errors.ECODE_INVAL)
415 9f7d5fe4 Sebastian Gebhard
416 dab6ea3d Helga Velroyen
  def CheckArguments(self):
417 dab6ea3d Helga Velroyen
    """Check arguments.
418 dab6ea3d Helga Velroyen

419 dab6ea3d Helga Velroyen
    """
420 dab6ea3d Helga Velroyen
    # do not require name_check to ease forward/backward compatibility
421 dab6ea3d Helga Velroyen
    # for tools
422 dab6ea3d Helga Velroyen
    if self.op.no_install and self.op.start:
423 dab6ea3d Helga Velroyen
      self.LogInfo("No-installation mode selected, disabling startup")
424 dab6ea3d Helga Velroyen
      self.op.start = False
425 dab6ea3d Helga Velroyen
    # validate/normalize the instance name
426 dab6ea3d Helga Velroyen
    self.op.instance_name = \
427 dab6ea3d Helga Velroyen
      netutils.Hostname.GetNormalizedName(self.op.instance_name)
428 dab6ea3d Helga Velroyen
429 dab6ea3d Helga Velroyen
    if self.op.ip_check and not self.op.name_check:
430 dab6ea3d Helga Velroyen
      # TODO: make the ip check more flexible and not depend on the name check
431 dab6ea3d Helga Velroyen
      raise errors.OpPrereqError("Cannot do IP address check without a name"
432 dab6ea3d Helga Velroyen
                                 " check", errors.ECODE_INVAL)
433 dab6ea3d Helga Velroyen
434 dab6ea3d Helga Velroyen
    # check nics' parameter names
435 dab6ea3d Helga Velroyen
    for nic in self.op.nics:
436 dab6ea3d Helga Velroyen
      utils.ForceDictType(nic, constants.INIC_PARAMS_TYPES)
437 dab6ea3d Helga Velroyen
    # check that NIC's parameters names are unique and valid
438 dab6ea3d Helga Velroyen
    utils.ValidateDeviceNames("NIC", self.op.nics)
439 dab6ea3d Helga Velroyen
440 9f7d5fe4 Sebastian Gebhard
    self._CheckVLANArguments()
441 9f7d5fe4 Sebastian Gebhard
442 dab6ea3d Helga Velroyen
    self._CheckDiskArguments()
443 66222813 Thomas Thrainer
    assert self.op.disk_template is not None
444 dab6ea3d Helga Velroyen
445 22b7f6f8 Thomas Thrainer
    # instance name verification
446 22b7f6f8 Thomas Thrainer
    if self.op.name_check:
447 da4a52a3 Thomas Thrainer
      self.hostname = _CheckHostnameSane(self, self.op.instance_name)
448 da4a52a3 Thomas Thrainer
      self.op.instance_name = self.hostname.name
449 22b7f6f8 Thomas Thrainer
      # used in CheckPrereq for ip ping check
450 da4a52a3 Thomas Thrainer
      self.check_ip = self.hostname.ip
451 22b7f6f8 Thomas Thrainer
    else:
452 22b7f6f8 Thomas Thrainer
      self.check_ip = None
453 22b7f6f8 Thomas Thrainer
454 22b7f6f8 Thomas Thrainer
    # file storage checks
455 22b7f6f8 Thomas Thrainer
    if (self.op.file_driver and
456 22b7f6f8 Thomas Thrainer
        not self.op.file_driver in constants.FILE_DRIVER):
457 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Invalid file driver name '%s'" %
458 22b7f6f8 Thomas Thrainer
                                 self.op.file_driver, errors.ECODE_INVAL)
459 22b7f6f8 Thomas Thrainer
460 ff34fb97 Michele Tartara
    # set default file_driver if unset and required
461 ff34fb97 Michele Tartara
    if (not self.op.file_driver and
462 a09639d1 Santi Raffa
        self.op.disk_template in constants.DTS_FILEBASED):
463 ff34fb97 Michele Tartara
      self.op.file_driver = constants.FD_LOOP
464 ff34fb97 Michele Tartara
465 22b7f6f8 Thomas Thrainer
    ### Node/iallocator related checks
466 5eacbcae Thomas Thrainer
    CheckIAllocatorOrNode(self, "iallocator", "pnode")
467 22b7f6f8 Thomas Thrainer
468 22b7f6f8 Thomas Thrainer
    if self.op.pnode is not None:
469 22b7f6f8 Thomas Thrainer
      if self.op.disk_template in constants.DTS_INT_MIRROR:
470 22b7f6f8 Thomas Thrainer
        if self.op.snode is None:
471 22b7f6f8 Thomas Thrainer
          raise errors.OpPrereqError("The networked disk templates need"
472 22b7f6f8 Thomas Thrainer
                                     " a mirror node", errors.ECODE_INVAL)
473 22b7f6f8 Thomas Thrainer
      elif self.op.snode:
474 22b7f6f8 Thomas Thrainer
        self.LogWarning("Secondary node will be ignored on non-mirrored disk"
475 22b7f6f8 Thomas Thrainer
                        " template")
476 22b7f6f8 Thomas Thrainer
        self.op.snode = None
477 22b7f6f8 Thomas Thrainer
478 22b7f6f8 Thomas Thrainer
    _CheckOpportunisticLocking(self.op)
479 22b7f6f8 Thomas Thrainer
480 22b7f6f8 Thomas Thrainer
    if self.op.mode == constants.INSTANCE_IMPORT:
481 22b7f6f8 Thomas Thrainer
      # On import force_variant must be True, because if we forced it at
482 22b7f6f8 Thomas Thrainer
      # initial install, our only chance when importing it back is that it
483 22b7f6f8 Thomas Thrainer
      # works again!
484 22b7f6f8 Thomas Thrainer
      self.op.force_variant = True
485 22b7f6f8 Thomas Thrainer
486 22b7f6f8 Thomas Thrainer
      if self.op.no_install:
487 22b7f6f8 Thomas Thrainer
        self.LogInfo("No-installation mode has no effect during import")
488 22b7f6f8 Thomas Thrainer
489 22b7f6f8 Thomas Thrainer
    elif self.op.mode == constants.INSTANCE_CREATE:
490 22b7f6f8 Thomas Thrainer
      if self.op.os_type is None:
491 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("No guest OS specified",
492 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
493 22b7f6f8 Thomas Thrainer
      if self.op.os_type in self.cfg.GetClusterInfo().blacklisted_os:
494 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Guest OS '%s' is not allowed for"
495 22b7f6f8 Thomas Thrainer
                                   " installation" % self.op.os_type,
496 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_STATE)
497 22b7f6f8 Thomas Thrainer
    elif self.op.mode == constants.INSTANCE_REMOTE_IMPORT:
498 66222813 Thomas Thrainer
      self._cds = GetClusterDomainSecret()
499 66222813 Thomas Thrainer
500 22b7f6f8 Thomas Thrainer
      # Check handshake to ensure both clusters have the same domain secret
501 22b7f6f8 Thomas Thrainer
      src_handshake = self.op.source_handshake
502 22b7f6f8 Thomas Thrainer
      if not src_handshake:
503 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Missing source handshake",
504 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
505 22b7f6f8 Thomas Thrainer
506 22b7f6f8 Thomas Thrainer
      errmsg = masterd.instance.CheckRemoteExportHandshake(self._cds,
507 22b7f6f8 Thomas Thrainer
                                                           src_handshake)
508 22b7f6f8 Thomas Thrainer
      if errmsg:
509 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Invalid handshake: %s" % errmsg,
510 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
511 22b7f6f8 Thomas Thrainer
512 22b7f6f8 Thomas Thrainer
      # Load and check source CA
513 22b7f6f8 Thomas Thrainer
      self.source_x509_ca_pem = self.op.source_x509_ca
514 22b7f6f8 Thomas Thrainer
      if not self.source_x509_ca_pem:
515 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Missing source X509 CA",
516 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
517 22b7f6f8 Thomas Thrainer
518 22b7f6f8 Thomas Thrainer
      try:
519 22b7f6f8 Thomas Thrainer
        (cert, _) = utils.LoadSignedX509Certificate(self.source_x509_ca_pem,
520 22b7f6f8 Thomas Thrainer
                                                    self._cds)
521 22b7f6f8 Thomas Thrainer
      except OpenSSL.crypto.Error, err:
522 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Unable to load source X509 CA (%s)" %
523 22b7f6f8 Thomas Thrainer
                                   (err, ), errors.ECODE_INVAL)
524 22b7f6f8 Thomas Thrainer
525 22b7f6f8 Thomas Thrainer
      (errcode, msg) = utils.VerifyX509Certificate(cert, None, None)
526 22b7f6f8 Thomas Thrainer
      if errcode is not None:
527 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Invalid source X509 CA (%s)" % (msg, ),
528 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
529 22b7f6f8 Thomas Thrainer
530 22b7f6f8 Thomas Thrainer
      self.source_x509_ca = cert
531 22b7f6f8 Thomas Thrainer
532 22b7f6f8 Thomas Thrainer
      src_instance_name = self.op.source_instance_name
533 22b7f6f8 Thomas Thrainer
      if not src_instance_name:
534 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Missing source instance name",
535 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
536 22b7f6f8 Thomas Thrainer
537 22b7f6f8 Thomas Thrainer
      self.source_instance_name = \
538 22b7f6f8 Thomas Thrainer
        netutils.GetHostname(name=src_instance_name).name
539 22b7f6f8 Thomas Thrainer
540 22b7f6f8 Thomas Thrainer
    else:
541 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Invalid instance creation mode %r" %
542 22b7f6f8 Thomas Thrainer
                                 self.op.mode, errors.ECODE_INVAL)
543 22b7f6f8 Thomas Thrainer
544 22b7f6f8 Thomas Thrainer
  def ExpandNames(self):
545 22b7f6f8 Thomas Thrainer
    """ExpandNames for CreateInstance.
546 22b7f6f8 Thomas Thrainer

547 22b7f6f8 Thomas Thrainer
    Figure out the right locks for instance creation.
548 22b7f6f8 Thomas Thrainer

549 22b7f6f8 Thomas Thrainer
    """
550 22b7f6f8 Thomas Thrainer
    self.needed_locks = {}
551 22b7f6f8 Thomas Thrainer
552 22b7f6f8 Thomas Thrainer
    # this is just a preventive check, but someone might still add this
553 22b7f6f8 Thomas Thrainer
    # instance in the meantime, and creation will fail at lock-add time
554 da4a52a3 Thomas Thrainer
    if self.op.instance_name in\
555 da4a52a3 Thomas Thrainer
      [inst.name for inst in self.cfg.GetAllInstancesInfo().values()]:
556 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Instance '%s' is already in the cluster" %
557 d0d7d7cf Thomas Thrainer
                                 self.op.instance_name, errors.ECODE_EXISTS)
558 22b7f6f8 Thomas Thrainer
559 d0d7d7cf Thomas Thrainer
    self.add_locks[locking.LEVEL_INSTANCE] = self.op.instance_name
560 22b7f6f8 Thomas Thrainer
561 22b7f6f8 Thomas Thrainer
    if self.op.iallocator:
562 22b7f6f8 Thomas Thrainer
      # TODO: Find a solution to not lock all nodes in the cluster, e.g. by
563 22b7f6f8 Thomas Thrainer
      # specifying a group on instance creation and then selecting nodes from
564 22b7f6f8 Thomas Thrainer
      # that group
565 22b7f6f8 Thomas Thrainer
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
566 22b7f6f8 Thomas Thrainer
      self.needed_locks[locking.LEVEL_NODE_ALLOC] = locking.ALL_SET
567 22b7f6f8 Thomas Thrainer
568 22b7f6f8 Thomas Thrainer
      if self.op.opportunistic_locking:
569 22b7f6f8 Thomas Thrainer
        self.opportunistic_locks[locking.LEVEL_NODE] = True
570 22b7f6f8 Thomas Thrainer
        self.opportunistic_locks[locking.LEVEL_NODE_RES] = True
571 22b7f6f8 Thomas Thrainer
    else:
572 1c3231aa Thomas Thrainer
      (self.op.pnode_uuid, self.op.pnode) = \
573 1c3231aa Thomas Thrainer
        ExpandNodeUuidAndName(self.cfg, self.op.pnode_uuid, self.op.pnode)
574 1c3231aa Thomas Thrainer
      nodelist = [self.op.pnode_uuid]
575 22b7f6f8 Thomas Thrainer
      if self.op.snode is not None:
576 1c3231aa Thomas Thrainer
        (self.op.snode_uuid, self.op.snode) = \
577 1c3231aa Thomas Thrainer
          ExpandNodeUuidAndName(self.cfg, self.op.snode_uuid, self.op.snode)
578 1c3231aa Thomas Thrainer
        nodelist.append(self.op.snode_uuid)
579 22b7f6f8 Thomas Thrainer
      self.needed_locks[locking.LEVEL_NODE] = nodelist
580 22b7f6f8 Thomas Thrainer
581 22b7f6f8 Thomas Thrainer
    # in case of import lock the source node too
582 22b7f6f8 Thomas Thrainer
    if self.op.mode == constants.INSTANCE_IMPORT:
583 22b7f6f8 Thomas Thrainer
      src_node = self.op.src_node
584 22b7f6f8 Thomas Thrainer
      src_path = self.op.src_path
585 22b7f6f8 Thomas Thrainer
586 22b7f6f8 Thomas Thrainer
      if src_path is None:
587 22b7f6f8 Thomas Thrainer
        self.op.src_path = src_path = self.op.instance_name
588 22b7f6f8 Thomas Thrainer
589 22b7f6f8 Thomas Thrainer
      if src_node is None:
590 22b7f6f8 Thomas Thrainer
        self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
591 22b7f6f8 Thomas Thrainer
        self.needed_locks[locking.LEVEL_NODE_ALLOC] = locking.ALL_SET
592 22b7f6f8 Thomas Thrainer
        self.op.src_node = None
593 22b7f6f8 Thomas Thrainer
        if os.path.isabs(src_path):
594 22b7f6f8 Thomas Thrainer
          raise errors.OpPrereqError("Importing an instance from a path"
595 22b7f6f8 Thomas Thrainer
                                     " requires a source node option",
596 22b7f6f8 Thomas Thrainer
                                     errors.ECODE_INVAL)
597 22b7f6f8 Thomas Thrainer
      else:
598 1c3231aa Thomas Thrainer
        (self.op.src_node_uuid, self.op.src_node) = (_, src_node) = \
599 1c3231aa Thomas Thrainer
          ExpandNodeUuidAndName(self.cfg, self.op.src_node_uuid, src_node)
600 22b7f6f8 Thomas Thrainer
        if self.needed_locks[locking.LEVEL_NODE] is not locking.ALL_SET:
601 1c3231aa Thomas Thrainer
          self.needed_locks[locking.LEVEL_NODE].append(self.op.src_node_uuid)
602 22b7f6f8 Thomas Thrainer
        if not os.path.isabs(src_path):
603 a61a0813 Thomas Thrainer
          self.op.src_path = \
604 22b7f6f8 Thomas Thrainer
            utils.PathJoin(pathutils.EXPORT_DIR, src_path)
605 22b7f6f8 Thomas Thrainer
606 22b7f6f8 Thomas Thrainer
    self.needed_locks[locking.LEVEL_NODE_RES] = \
607 5eacbcae Thomas Thrainer
      CopyLockList(self.needed_locks[locking.LEVEL_NODE])
608 22b7f6f8 Thomas Thrainer
609 22b7f6f8 Thomas Thrainer
  def _RunAllocator(self):
610 22b7f6f8 Thomas Thrainer
    """Run the allocator based on input opcode.
611 22b7f6f8 Thomas Thrainer

612 22b7f6f8 Thomas Thrainer
    """
613 22b7f6f8 Thomas Thrainer
    if self.op.opportunistic_locking:
614 22b7f6f8 Thomas Thrainer
      # Only consider nodes for which a lock is held
615 4665b94f Thomas Thrainer
      node_name_whitelist = self.cfg.GetNodeNames(
616 4665b94f Thomas Thrainer
        self.owned_locks(locking.LEVEL_NODE))
617 22b7f6f8 Thomas Thrainer
    else:
618 4665b94f Thomas Thrainer
      node_name_whitelist = None
619 22b7f6f8 Thomas Thrainer
620 22b7f6f8 Thomas Thrainer
    req = _CreateInstanceAllocRequest(self.op, self.disks,
621 22b7f6f8 Thomas Thrainer
                                      self.nics, self.be_full,
622 4665b94f Thomas Thrainer
                                      node_name_whitelist)
623 22b7f6f8 Thomas Thrainer
    ial = iallocator.IAllocator(self.cfg, self.rpc, req)
624 22b7f6f8 Thomas Thrainer
625 22b7f6f8 Thomas Thrainer
    ial.Run(self.op.iallocator)
626 22b7f6f8 Thomas Thrainer
627 22b7f6f8 Thomas Thrainer
    if not ial.success:
628 22b7f6f8 Thomas Thrainer
      # When opportunistic locks are used only a temporary failure is generated
629 22b7f6f8 Thomas Thrainer
      if self.op.opportunistic_locking:
630 22b7f6f8 Thomas Thrainer
        ecode = errors.ECODE_TEMP_NORES
631 22b7f6f8 Thomas Thrainer
      else:
632 22b7f6f8 Thomas Thrainer
        ecode = errors.ECODE_NORES
633 22b7f6f8 Thomas Thrainer
634 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Can't compute nodes using"
635 22b7f6f8 Thomas Thrainer
                                 " iallocator '%s': %s" %
636 22b7f6f8 Thomas Thrainer
                                 (self.op.iallocator, ial.info),
637 22b7f6f8 Thomas Thrainer
                                 ecode)
638 22b7f6f8 Thomas Thrainer
639 1c3231aa Thomas Thrainer
    (self.op.pnode_uuid, self.op.pnode) = \
640 1c3231aa Thomas Thrainer
      ExpandNodeUuidAndName(self.cfg, None, ial.result[0])
641 22b7f6f8 Thomas Thrainer
    self.LogInfo("Selected nodes for instance %s via iallocator %s: %s",
642 22b7f6f8 Thomas Thrainer
                 self.op.instance_name, self.op.iallocator,
643 22b7f6f8 Thomas Thrainer
                 utils.CommaJoin(ial.result))
644 22b7f6f8 Thomas Thrainer
645 22b7f6f8 Thomas Thrainer
    assert req.RequiredNodes() in (1, 2), "Wrong node count from iallocator"
646 22b7f6f8 Thomas Thrainer
647 22b7f6f8 Thomas Thrainer
    if req.RequiredNodes() == 2:
648 1c3231aa Thomas Thrainer
      (self.op.snode_uuid, self.op.snode) = \
649 1c3231aa Thomas Thrainer
        ExpandNodeUuidAndName(self.cfg, None, ial.result[1])
650 22b7f6f8 Thomas Thrainer
651 22b7f6f8 Thomas Thrainer
  def BuildHooksEnv(self):
652 22b7f6f8 Thomas Thrainer
    """Build hooks env.
653 22b7f6f8 Thomas Thrainer

654 22b7f6f8 Thomas Thrainer
    This runs on master, primary and secondary nodes of the instance.
655 22b7f6f8 Thomas Thrainer

656 22b7f6f8 Thomas Thrainer
    """
657 22b7f6f8 Thomas Thrainer
    env = {
658 22b7f6f8 Thomas Thrainer
      "ADD_MODE": self.op.mode,
659 22b7f6f8 Thomas Thrainer
      }
660 22b7f6f8 Thomas Thrainer
    if self.op.mode == constants.INSTANCE_IMPORT:
661 22b7f6f8 Thomas Thrainer
      env["SRC_NODE"] = self.op.src_node
662 22b7f6f8 Thomas Thrainer
      env["SRC_PATH"] = self.op.src_path
663 22b7f6f8 Thomas Thrainer
      env["SRC_IMAGES"] = self.src_images
664 22b7f6f8 Thomas Thrainer
665 5eacbcae Thomas Thrainer
    env.update(BuildInstanceHookEnv(
666 22b7f6f8 Thomas Thrainer
      name=self.op.instance_name,
667 1c3231aa Thomas Thrainer
      primary_node_name=self.op.pnode,
668 1c3231aa Thomas Thrainer
      secondary_node_names=self.cfg.GetNodeNames(self.secondaries),
669 22b7f6f8 Thomas Thrainer
      status=self.op.start,
670 22b7f6f8 Thomas Thrainer
      os_type=self.op.os_type,
671 22b7f6f8 Thomas Thrainer
      minmem=self.be_full[constants.BE_MINMEM],
672 22b7f6f8 Thomas Thrainer
      maxmem=self.be_full[constants.BE_MAXMEM],
673 22b7f6f8 Thomas Thrainer
      vcpus=self.be_full[constants.BE_VCPUS],
674 5eacbcae Thomas Thrainer
      nics=NICListToTuple(self, self.nics),
675 22b7f6f8 Thomas Thrainer
      disk_template=self.op.disk_template,
676 8a348b15 Christos Stavrakakis
      disks=[(d[constants.IDISK_NAME], d.get("uuid", ""),
677 8a348b15 Christos Stavrakakis
              d[constants.IDISK_SIZE], d[constants.IDISK_MODE])
678 8a348b15 Christos Stavrakakis
             for d in self.disks],
679 22b7f6f8 Thomas Thrainer
      bep=self.be_full,
680 22b7f6f8 Thomas Thrainer
      hvp=self.hv_full,
681 22b7f6f8 Thomas Thrainer
      hypervisor_name=self.op.hypervisor,
682 22b7f6f8 Thomas Thrainer
      tags=self.op.tags,
683 22b7f6f8 Thomas Thrainer
      ))
684 22b7f6f8 Thomas Thrainer
685 22b7f6f8 Thomas Thrainer
    return env
686 22b7f6f8 Thomas Thrainer
687 22b7f6f8 Thomas Thrainer
  def BuildHooksNodes(self):
688 22b7f6f8 Thomas Thrainer
    """Build hooks nodes.
689 22b7f6f8 Thomas Thrainer

690 22b7f6f8 Thomas Thrainer
    """
691 1c3231aa Thomas Thrainer
    nl = [self.cfg.GetMasterNode(), self.op.pnode_uuid] + self.secondaries
692 22b7f6f8 Thomas Thrainer
    return nl, nl
693 22b7f6f8 Thomas Thrainer
694 22b7f6f8 Thomas Thrainer
  def _ReadExportInfo(self):
695 22b7f6f8 Thomas Thrainer
    """Reads the export information from disk.
696 22b7f6f8 Thomas Thrainer

697 22b7f6f8 Thomas Thrainer
    It will override the opcode source node and path with the actual
698 22b7f6f8 Thomas Thrainer
    information, if these two were not specified before.
699 22b7f6f8 Thomas Thrainer

700 22b7f6f8 Thomas Thrainer
    @return: the export information
701 22b7f6f8 Thomas Thrainer

702 22b7f6f8 Thomas Thrainer
    """
703 22b7f6f8 Thomas Thrainer
    assert self.op.mode == constants.INSTANCE_IMPORT
704 22b7f6f8 Thomas Thrainer
705 d0d7d7cf Thomas Thrainer
    if self.op.src_node_uuid is None:
706 22b7f6f8 Thomas Thrainer
      locked_nodes = self.owned_locks(locking.LEVEL_NODE)
707 22b7f6f8 Thomas Thrainer
      exp_list = self.rpc.call_export_list(locked_nodes)
708 22b7f6f8 Thomas Thrainer
      found = False
709 a61a0813 Thomas Thrainer
      for node_uuid in exp_list:
710 a61a0813 Thomas Thrainer
        if exp_list[node_uuid].fail_msg:
711 22b7f6f8 Thomas Thrainer
          continue
712 a61a0813 Thomas Thrainer
        if self.op.src_path in exp_list[node_uuid].payload:
713 22b7f6f8 Thomas Thrainer
          found = True
714 a61a0813 Thomas Thrainer
          self.op.src_node = self.cfg.GetNodeInfo(node_uuid).name
715 a61a0813 Thomas Thrainer
          self.op.src_node_uuid = node_uuid
716 d0d7d7cf Thomas Thrainer
          self.op.src_path = utils.PathJoin(pathutils.EXPORT_DIR,
717 d0d7d7cf Thomas Thrainer
                                            self.op.src_path)
718 22b7f6f8 Thomas Thrainer
          break
719 22b7f6f8 Thomas Thrainer
      if not found:
720 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("No export found for relative path %s" %
721 d0d7d7cf Thomas Thrainer
                                   self.op.src_path, errors.ECODE_INVAL)
722 22b7f6f8 Thomas Thrainer
723 d0d7d7cf Thomas Thrainer
    CheckNodeOnline(self, self.op.src_node_uuid)
724 d0d7d7cf Thomas Thrainer
    result = self.rpc.call_export_info(self.op.src_node_uuid, self.op.src_path)
725 d0d7d7cf Thomas Thrainer
    result.Raise("No export or invalid export found in dir %s" %
726 d0d7d7cf Thomas Thrainer
                 self.op.src_path)
727 22b7f6f8 Thomas Thrainer
728 22b7f6f8 Thomas Thrainer
    export_info = objects.SerializableConfigParser.Loads(str(result.payload))
729 22b7f6f8 Thomas Thrainer
    if not export_info.has_section(constants.INISECT_EXP):
730 22b7f6f8 Thomas Thrainer
      raise errors.ProgrammerError("Corrupted export config",
731 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_ENVIRON)
732 22b7f6f8 Thomas Thrainer
733 22b7f6f8 Thomas Thrainer
    ei_version = export_info.get(constants.INISECT_EXP, "version")
734 1c3231aa Thomas Thrainer
    if int(ei_version) != constants.EXPORT_VERSION:
735 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Wrong export version %s (wanted %d)" %
736 22b7f6f8 Thomas Thrainer
                                 (ei_version, constants.EXPORT_VERSION),
737 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_ENVIRON)
738 22b7f6f8 Thomas Thrainer
    return export_info
739 22b7f6f8 Thomas Thrainer
740 22b7f6f8 Thomas Thrainer
  def _ReadExportParams(self, einfo):
741 22b7f6f8 Thomas Thrainer
    """Use export parameters as defaults.
742 22b7f6f8 Thomas Thrainer

743 22b7f6f8 Thomas Thrainer
    In case the opcode doesn't specify (as in override) some instance
744 22b7f6f8 Thomas Thrainer
    parameters, then try to use them from the export information, if
745 22b7f6f8 Thomas Thrainer
    that declares them.
746 22b7f6f8 Thomas Thrainer

747 22b7f6f8 Thomas Thrainer
    """
748 22b7f6f8 Thomas Thrainer
    self.op.os_type = einfo.get(constants.INISECT_EXP, "os")
749 22b7f6f8 Thomas Thrainer
750 22b7f6f8 Thomas Thrainer
    if not self.op.disks:
751 22b7f6f8 Thomas Thrainer
      disks = []
752 22b7f6f8 Thomas Thrainer
      # TODO: import the disk iv_name too
753 22b7f6f8 Thomas Thrainer
      for idx in range(constants.MAX_DISKS):
754 22b7f6f8 Thomas Thrainer
        if einfo.has_option(constants.INISECT_INS, "disk%d_size" % idx):
755 22b7f6f8 Thomas Thrainer
          disk_sz = einfo.getint(constants.INISECT_INS, "disk%d_size" % idx)
756 22b7f6f8 Thomas Thrainer
          disks.append({constants.IDISK_SIZE: disk_sz})
757 22b7f6f8 Thomas Thrainer
      self.op.disks = disks
758 22b7f6f8 Thomas Thrainer
      if not disks and self.op.disk_template != constants.DT_DISKLESS:
759 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("No disk info specified and the export"
760 22b7f6f8 Thomas Thrainer
                                   " is missing the disk information",
761 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
762 22b7f6f8 Thomas Thrainer
763 22b7f6f8 Thomas Thrainer
    if not self.op.nics:
764 22b7f6f8 Thomas Thrainer
      nics = []
765 22b7f6f8 Thomas Thrainer
      for idx in range(constants.MAX_NICS):
766 22b7f6f8 Thomas Thrainer
        if einfo.has_option(constants.INISECT_INS, "nic%d_mac" % idx):
767 22b7f6f8 Thomas Thrainer
          ndict = {}
768 22b7f6f8 Thomas Thrainer
          for name in list(constants.NICS_PARAMETERS) + ["ip", "mac"]:
769 a61a0813 Thomas Thrainer
            nic_param_name = "nic%d_%s" % (idx, name)
770 a61a0813 Thomas Thrainer
            if einfo.has_option(constants.INISECT_INS, nic_param_name):
771 a61a0813 Thomas Thrainer
              v = einfo.get(constants.INISECT_INS, nic_param_name)
772 a61a0813 Thomas Thrainer
              ndict[name] = v
773 22b7f6f8 Thomas Thrainer
          nics.append(ndict)
774 22b7f6f8 Thomas Thrainer
        else:
775 22b7f6f8 Thomas Thrainer
          break
776 22b7f6f8 Thomas Thrainer
      self.op.nics = nics
777 22b7f6f8 Thomas Thrainer
778 22b7f6f8 Thomas Thrainer
    if not self.op.tags and einfo.has_option(constants.INISECT_INS, "tags"):
779 22b7f6f8 Thomas Thrainer
      self.op.tags = einfo.get(constants.INISECT_INS, "tags").split()
780 22b7f6f8 Thomas Thrainer
781 22b7f6f8 Thomas Thrainer
    if (self.op.hypervisor is None and
782 22b7f6f8 Thomas Thrainer
        einfo.has_option(constants.INISECT_INS, "hypervisor")):
783 22b7f6f8 Thomas Thrainer
      self.op.hypervisor = einfo.get(constants.INISECT_INS, "hypervisor")
784 22b7f6f8 Thomas Thrainer
785 22b7f6f8 Thomas Thrainer
    if einfo.has_section(constants.INISECT_HYP):
786 22b7f6f8 Thomas Thrainer
      # use the export parameters but do not override the ones
787 22b7f6f8 Thomas Thrainer
      # specified by the user
788 22b7f6f8 Thomas Thrainer
      for name, value in einfo.items(constants.INISECT_HYP):
789 22b7f6f8 Thomas Thrainer
        if name not in self.op.hvparams:
790 22b7f6f8 Thomas Thrainer
          self.op.hvparams[name] = value
791 22b7f6f8 Thomas Thrainer
792 22b7f6f8 Thomas Thrainer
    if einfo.has_section(constants.INISECT_BEP):
793 22b7f6f8 Thomas Thrainer
      # use the parameters, without overriding
794 22b7f6f8 Thomas Thrainer
      for name, value in einfo.items(constants.INISECT_BEP):
795 22b7f6f8 Thomas Thrainer
        if name not in self.op.beparams:
796 22b7f6f8 Thomas Thrainer
          self.op.beparams[name] = value
797 22b7f6f8 Thomas Thrainer
        # Compatibility for the old "memory" be param
798 22b7f6f8 Thomas Thrainer
        if name == constants.BE_MEMORY:
799 22b7f6f8 Thomas Thrainer
          if constants.BE_MAXMEM not in self.op.beparams:
800 22b7f6f8 Thomas Thrainer
            self.op.beparams[constants.BE_MAXMEM] = value
801 22b7f6f8 Thomas Thrainer
          if constants.BE_MINMEM not in self.op.beparams:
802 22b7f6f8 Thomas Thrainer
            self.op.beparams[constants.BE_MINMEM] = value
803 22b7f6f8 Thomas Thrainer
    else:
804 22b7f6f8 Thomas Thrainer
      # try to read the parameters old style, from the main section
805 22b7f6f8 Thomas Thrainer
      for name in constants.BES_PARAMETERS:
806 22b7f6f8 Thomas Thrainer
        if (name not in self.op.beparams and
807 22b7f6f8 Thomas Thrainer
            einfo.has_option(constants.INISECT_INS, name)):
808 22b7f6f8 Thomas Thrainer
          self.op.beparams[name] = einfo.get(constants.INISECT_INS, name)
809 22b7f6f8 Thomas Thrainer
810 22b7f6f8 Thomas Thrainer
    if einfo.has_section(constants.INISECT_OSP):
811 22b7f6f8 Thomas Thrainer
      # use the parameters, without overriding
812 22b7f6f8 Thomas Thrainer
      for name, value in einfo.items(constants.INISECT_OSP):
813 22b7f6f8 Thomas Thrainer
        if name not in self.op.osparams:
814 22b7f6f8 Thomas Thrainer
          self.op.osparams[name] = value
815 22b7f6f8 Thomas Thrainer
816 22b7f6f8 Thomas Thrainer
  def _RevertToDefaults(self, cluster):
817 22b7f6f8 Thomas Thrainer
    """Revert the instance parameters to the default values.
818 22b7f6f8 Thomas Thrainer

819 22b7f6f8 Thomas Thrainer
    """
820 22b7f6f8 Thomas Thrainer
    # hvparams
821 22b7f6f8 Thomas Thrainer
    hv_defs = cluster.SimpleFillHV(self.op.hypervisor, self.op.os_type, {})
822 22b7f6f8 Thomas Thrainer
    for name in self.op.hvparams.keys():
823 22b7f6f8 Thomas Thrainer
      if name in hv_defs and hv_defs[name] == self.op.hvparams[name]:
824 22b7f6f8 Thomas Thrainer
        del self.op.hvparams[name]
825 22b7f6f8 Thomas Thrainer
    # beparams
826 22b7f6f8 Thomas Thrainer
    be_defs = cluster.SimpleFillBE({})
827 22b7f6f8 Thomas Thrainer
    for name in self.op.beparams.keys():
828 22b7f6f8 Thomas Thrainer
      if name in be_defs and be_defs[name] == self.op.beparams[name]:
829 22b7f6f8 Thomas Thrainer
        del self.op.beparams[name]
830 22b7f6f8 Thomas Thrainer
    # nic params
831 22b7f6f8 Thomas Thrainer
    nic_defs = cluster.SimpleFillNIC({})
832 22b7f6f8 Thomas Thrainer
    for nic in self.op.nics:
833 22b7f6f8 Thomas Thrainer
      for name in constants.NICS_PARAMETERS:
834 22b7f6f8 Thomas Thrainer
        if name in nic and name in nic_defs and nic[name] == nic_defs[name]:
835 22b7f6f8 Thomas Thrainer
          del nic[name]
836 22b7f6f8 Thomas Thrainer
    # osparams
837 22b7f6f8 Thomas Thrainer
    os_defs = cluster.SimpleFillOS(self.op.os_type, {})
838 22b7f6f8 Thomas Thrainer
    for name in self.op.osparams.keys():
839 22b7f6f8 Thomas Thrainer
      if name in os_defs and os_defs[name] == self.op.osparams[name]:
840 22b7f6f8 Thomas Thrainer
        del self.op.osparams[name]
841 22b7f6f8 Thomas Thrainer
842 22b7f6f8 Thomas Thrainer
  def _CalculateFileStorageDir(self):
843 22b7f6f8 Thomas Thrainer
    """Calculate final instance file storage dir.
844 22b7f6f8 Thomas Thrainer

845 22b7f6f8 Thomas Thrainer
    """
846 22b7f6f8 Thomas Thrainer
    # file storage dir calculation/check
847 22b7f6f8 Thomas Thrainer
    self.instance_file_storage_dir = None
848 22b7f6f8 Thomas Thrainer
    if self.op.disk_template in constants.DTS_FILEBASED:
849 22b7f6f8 Thomas Thrainer
      # build the full file storage dir path
850 22b7f6f8 Thomas Thrainer
      joinargs = []
851 22b7f6f8 Thomas Thrainer
852 22b7f6f8 Thomas Thrainer
      if self.op.disk_template == constants.DT_SHARED_FILE:
853 22b7f6f8 Thomas Thrainer
        get_fsd_fn = self.cfg.GetSharedFileStorageDir
854 22b7f6f8 Thomas Thrainer
      else:
855 22b7f6f8 Thomas Thrainer
        get_fsd_fn = self.cfg.GetFileStorageDir
856 22b7f6f8 Thomas Thrainer
857 22b7f6f8 Thomas Thrainer
      cfg_storagedir = get_fsd_fn()
858 22b7f6f8 Thomas Thrainer
      if not cfg_storagedir:
859 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Cluster file storage dir not defined",
860 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_STATE)
861 22b7f6f8 Thomas Thrainer
      joinargs.append(cfg_storagedir)
862 22b7f6f8 Thomas Thrainer
863 22b7f6f8 Thomas Thrainer
      if self.op.file_storage_dir is not None:
864 22b7f6f8 Thomas Thrainer
        joinargs.append(self.op.file_storage_dir)
865 22b7f6f8 Thomas Thrainer
866 22b7f6f8 Thomas Thrainer
      joinargs.append(self.op.instance_name)
867 22b7f6f8 Thomas Thrainer
868 22b7f6f8 Thomas Thrainer
      # pylint: disable=W0142
869 22b7f6f8 Thomas Thrainer
      self.instance_file_storage_dir = utils.PathJoin(*joinargs)
870 22b7f6f8 Thomas Thrainer
871 22b7f6f8 Thomas Thrainer
  def CheckPrereq(self): # pylint: disable=R0914
872 22b7f6f8 Thomas Thrainer
    """Check prerequisites.
873 22b7f6f8 Thomas Thrainer

874 22b7f6f8 Thomas Thrainer
    """
875 22b7f6f8 Thomas Thrainer
    self._CalculateFileStorageDir()
876 22b7f6f8 Thomas Thrainer
877 22b7f6f8 Thomas Thrainer
    if self.op.mode == constants.INSTANCE_IMPORT:
878 22b7f6f8 Thomas Thrainer
      export_info = self._ReadExportInfo()
879 22b7f6f8 Thomas Thrainer
      self._ReadExportParams(export_info)
880 22b7f6f8 Thomas Thrainer
      self._old_instance_name = export_info.get(constants.INISECT_INS, "name")
881 22b7f6f8 Thomas Thrainer
    else:
882 22b7f6f8 Thomas Thrainer
      self._old_instance_name = None
883 22b7f6f8 Thomas Thrainer
884 22b7f6f8 Thomas Thrainer
    if (not self.cfg.GetVGName() and
885 22b7f6f8 Thomas Thrainer
        self.op.disk_template not in constants.DTS_NOT_LVM):
886 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Cluster does not support lvm-based"
887 22b7f6f8 Thomas Thrainer
                                 " instances", errors.ECODE_STATE)
888 22b7f6f8 Thomas Thrainer
889 22b7f6f8 Thomas Thrainer
    if (self.op.hypervisor is None or
890 22b7f6f8 Thomas Thrainer
        self.op.hypervisor == constants.VALUE_AUTO):
891 22b7f6f8 Thomas Thrainer
      self.op.hypervisor = self.cfg.GetHypervisorType()
892 22b7f6f8 Thomas Thrainer
893 22b7f6f8 Thomas Thrainer
    cluster = self.cfg.GetClusterInfo()
894 22b7f6f8 Thomas Thrainer
    enabled_hvs = cluster.enabled_hypervisors
895 22b7f6f8 Thomas Thrainer
    if self.op.hypervisor not in enabled_hvs:
896 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Selected hypervisor (%s) not enabled in the"
897 22b7f6f8 Thomas Thrainer
                                 " cluster (%s)" %
898 22b7f6f8 Thomas Thrainer
                                 (self.op.hypervisor, ",".join(enabled_hvs)),
899 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_STATE)
900 22b7f6f8 Thomas Thrainer
901 22b7f6f8 Thomas Thrainer
    # Check tag validity
902 22b7f6f8 Thomas Thrainer
    for tag in self.op.tags:
903 22b7f6f8 Thomas Thrainer
      objects.TaggableObject.ValidateTag(tag)
904 22b7f6f8 Thomas Thrainer
905 22b7f6f8 Thomas Thrainer
    # check hypervisor parameter syntax (locally)
906 22b7f6f8 Thomas Thrainer
    utils.ForceDictType(self.op.hvparams, constants.HVS_PARAMETER_TYPES)
907 22b7f6f8 Thomas Thrainer
    filled_hvp = cluster.SimpleFillHV(self.op.hypervisor, self.op.os_type,
908 22b7f6f8 Thomas Thrainer
                                      self.op.hvparams)
909 22b7f6f8 Thomas Thrainer
    hv_type = hypervisor.GetHypervisorClass(self.op.hypervisor)
910 22b7f6f8 Thomas Thrainer
    hv_type.CheckParameterSyntax(filled_hvp)
911 22b7f6f8 Thomas Thrainer
    self.hv_full = filled_hvp
912 22b7f6f8 Thomas Thrainer
    # check that we don't specify global parameters on an instance
913 5eacbcae Thomas Thrainer
    CheckParamsNotGlobal(self.op.hvparams, constants.HVC_GLOBALS, "hypervisor",
914 5eacbcae Thomas Thrainer
                         "instance", "cluster")
915 22b7f6f8 Thomas Thrainer
916 22b7f6f8 Thomas Thrainer
    # fill and remember the beparams dict
917 22b7f6f8 Thomas Thrainer
    self.be_full = _ComputeFullBeParams(self.op, cluster)
918 22b7f6f8 Thomas Thrainer
919 22b7f6f8 Thomas Thrainer
    # build os parameters
920 22b7f6f8 Thomas Thrainer
    self.os_full = cluster.SimpleFillOS(self.op.os_type, self.op.osparams)
921 22b7f6f8 Thomas Thrainer
922 22b7f6f8 Thomas Thrainer
    # now that hvp/bep are in final format, let's reset to defaults,
923 22b7f6f8 Thomas Thrainer
    # if told to do so
924 22b7f6f8 Thomas Thrainer
    if self.op.identify_defaults:
925 22b7f6f8 Thomas Thrainer
      self._RevertToDefaults(cluster)
926 22b7f6f8 Thomas Thrainer
927 22b7f6f8 Thomas Thrainer
    # NIC buildup
928 22b7f6f8 Thomas Thrainer
    self.nics = _ComputeNics(self.op, cluster, self.check_ip, self.cfg,
929 22b7f6f8 Thomas Thrainer
                             self.proc.GetECId())
930 22b7f6f8 Thomas Thrainer
931 22b7f6f8 Thomas Thrainer
    # disk checks/pre-build
932 22b7f6f8 Thomas Thrainer
    default_vg = self.cfg.GetVGName()
933 5eacbcae Thomas Thrainer
    self.disks = ComputeDisks(self.op, default_vg)
934 22b7f6f8 Thomas Thrainer
935 22b7f6f8 Thomas Thrainer
    if self.op.mode == constants.INSTANCE_IMPORT:
936 22b7f6f8 Thomas Thrainer
      disk_images = []
937 22b7f6f8 Thomas Thrainer
      for idx in range(len(self.disks)):
938 22b7f6f8 Thomas Thrainer
        option = "disk%d_dump" % idx
939 22b7f6f8 Thomas Thrainer
        if export_info.has_option(constants.INISECT_INS, option):
940 22b7f6f8 Thomas Thrainer
          # FIXME: are the old os-es, disk sizes, etc. useful?
941 22b7f6f8 Thomas Thrainer
          export_name = export_info.get(constants.INISECT_INS, option)
942 22b7f6f8 Thomas Thrainer
          image = utils.PathJoin(self.op.src_path, export_name)
943 22b7f6f8 Thomas Thrainer
          disk_images.append(image)
944 22b7f6f8 Thomas Thrainer
        else:
945 22b7f6f8 Thomas Thrainer
          disk_images.append(False)
946 22b7f6f8 Thomas Thrainer
947 22b7f6f8 Thomas Thrainer
      self.src_images = disk_images
948 22b7f6f8 Thomas Thrainer
949 22b7f6f8 Thomas Thrainer
      if self.op.instance_name == self._old_instance_name:
950 22b7f6f8 Thomas Thrainer
        for idx, nic in enumerate(self.nics):
951 22b7f6f8 Thomas Thrainer
          if nic.mac == constants.VALUE_AUTO:
952 22b7f6f8 Thomas Thrainer
            nic_mac_ini = "nic%d_mac" % idx
953 22b7f6f8 Thomas Thrainer
            nic.mac = export_info.get(constants.INISECT_INS, nic_mac_ini)
954 22b7f6f8 Thomas Thrainer
955 22b7f6f8 Thomas Thrainer
    # ENDIF: self.op.mode == constants.INSTANCE_IMPORT
956 22b7f6f8 Thomas Thrainer
957 22b7f6f8 Thomas Thrainer
    # ip ping checks (we use the same ip that was resolved in ExpandNames)
958 22b7f6f8 Thomas Thrainer
    if self.op.ip_check:
959 22b7f6f8 Thomas Thrainer
      if netutils.TcpPing(self.check_ip, constants.DEFAULT_NODED_PORT):
960 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("IP %s of instance %s already in use" %
961 22b7f6f8 Thomas Thrainer
                                   (self.check_ip, self.op.instance_name),
962 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_NOTUNIQUE)
963 22b7f6f8 Thomas Thrainer
964 22b7f6f8 Thomas Thrainer
    #### mac address generation
965 22b7f6f8 Thomas Thrainer
    # By generating here the mac address both the allocator and the hooks get
966 22b7f6f8 Thomas Thrainer
    # the real final mac address rather than the 'auto' or 'generate' value.
967 22b7f6f8 Thomas Thrainer
    # There is a race condition between the generation and the instance object
968 22b7f6f8 Thomas Thrainer
    # creation, which means that we know the mac is valid now, but we're not
969 22b7f6f8 Thomas Thrainer
    # sure it will be when we actually add the instance. If things go bad
970 22b7f6f8 Thomas Thrainer
    # adding the instance will abort because of a duplicate mac, and the
971 22b7f6f8 Thomas Thrainer
    # creation job will fail.
972 22b7f6f8 Thomas Thrainer
    for nic in self.nics:
973 22b7f6f8 Thomas Thrainer
      if nic.mac in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
974 22b7f6f8 Thomas Thrainer
        nic.mac = self.cfg.GenerateMAC(nic.network, self.proc.GetECId())
975 22b7f6f8 Thomas Thrainer
976 22b7f6f8 Thomas Thrainer
    #### allocator run
977 22b7f6f8 Thomas Thrainer
978 22b7f6f8 Thomas Thrainer
    if self.op.iallocator is not None:
979 22b7f6f8 Thomas Thrainer
      self._RunAllocator()
980 22b7f6f8 Thomas Thrainer
981 22b7f6f8 Thomas Thrainer
    # Release all unneeded node locks
982 1c3231aa Thomas Thrainer
    keep_locks = filter(None, [self.op.pnode_uuid, self.op.snode_uuid,
983 1c3231aa Thomas Thrainer
                               self.op.src_node_uuid])
984 5eacbcae Thomas Thrainer
    ReleaseLocks(self, locking.LEVEL_NODE, keep=keep_locks)
985 5eacbcae Thomas Thrainer
    ReleaseLocks(self, locking.LEVEL_NODE_RES, keep=keep_locks)
986 5eacbcae Thomas Thrainer
    ReleaseLocks(self, locking.LEVEL_NODE_ALLOC)
987 22b7f6f8 Thomas Thrainer
988 22b7f6f8 Thomas Thrainer
    assert (self.owned_locks(locking.LEVEL_NODE) ==
989 22b7f6f8 Thomas Thrainer
            self.owned_locks(locking.LEVEL_NODE_RES)), \
990 22b7f6f8 Thomas Thrainer
      "Node locks differ from node resource locks"
991 22b7f6f8 Thomas Thrainer
992 22b7f6f8 Thomas Thrainer
    #### node related checks
993 22b7f6f8 Thomas Thrainer
994 22b7f6f8 Thomas Thrainer
    # check primary node
995 1c3231aa Thomas Thrainer
    self.pnode = pnode = self.cfg.GetNodeInfo(self.op.pnode_uuid)
996 22b7f6f8 Thomas Thrainer
    assert self.pnode is not None, \
997 1c3231aa Thomas Thrainer
      "Cannot retrieve locked node %s" % self.op.pnode_uuid
998 22b7f6f8 Thomas Thrainer
    if pnode.offline:
999 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Cannot use offline primary node '%s'" %
1000 22b7f6f8 Thomas Thrainer
                                 pnode.name, errors.ECODE_STATE)
1001 22b7f6f8 Thomas Thrainer
    if pnode.drained:
1002 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Cannot use drained primary node '%s'" %
1003 22b7f6f8 Thomas Thrainer
                                 pnode.name, errors.ECODE_STATE)
1004 22b7f6f8 Thomas Thrainer
    if not pnode.vm_capable:
1005 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Cannot use non-vm_capable primary node"
1006 22b7f6f8 Thomas Thrainer
                                 " '%s'" % pnode.name, errors.ECODE_STATE)
1007 22b7f6f8 Thomas Thrainer
1008 22b7f6f8 Thomas Thrainer
    self.secondaries = []
1009 22b7f6f8 Thomas Thrainer
1010 22b7f6f8 Thomas Thrainer
    # Fill in any IPs from IP pools. This must happen here, because we need to
1011 22b7f6f8 Thomas Thrainer
    # know the nic's primary node, as specified by the iallocator
1012 22b7f6f8 Thomas Thrainer
    for idx, nic in enumerate(self.nics):
1013 22b7f6f8 Thomas Thrainer
      net_uuid = nic.network
1014 22b7f6f8 Thomas Thrainer
      if net_uuid is not None:
1015 22b7f6f8 Thomas Thrainer
        nobj = self.cfg.GetNetwork(net_uuid)
1016 1c3231aa Thomas Thrainer
        netparams = self.cfg.GetGroupNetParams(net_uuid, self.pnode.uuid)
1017 22b7f6f8 Thomas Thrainer
        if netparams is None:
1018 22b7f6f8 Thomas Thrainer
          raise errors.OpPrereqError("No netparams found for network"
1019 66222813 Thomas Thrainer
                                     " %s. Probably not connected to"
1020 22b7f6f8 Thomas Thrainer
                                     " node's %s nodegroup" %
1021 22b7f6f8 Thomas Thrainer
                                     (nobj.name, self.pnode.name),
1022 22b7f6f8 Thomas Thrainer
                                     errors.ECODE_INVAL)
1023 22b7f6f8 Thomas Thrainer
        self.LogInfo("NIC/%d inherits netparams %s" %
1024 22b7f6f8 Thomas Thrainer
                     (idx, netparams.values()))
1025 22b7f6f8 Thomas Thrainer
        nic.nicparams = dict(netparams)
1026 22b7f6f8 Thomas Thrainer
        if nic.ip is not None:
1027 22b7f6f8 Thomas Thrainer
          if nic.ip.lower() == constants.NIC_IP_POOL:
1028 22b7f6f8 Thomas Thrainer
            try:
1029 22b7f6f8 Thomas Thrainer
              nic.ip = self.cfg.GenerateIp(net_uuid, self.proc.GetECId())
1030 22b7f6f8 Thomas Thrainer
            except errors.ReservationError:
1031 22b7f6f8 Thomas Thrainer
              raise errors.OpPrereqError("Unable to get a free IP for NIC %d"
1032 22b7f6f8 Thomas Thrainer
                                         " from the address pool" % idx,
1033 22b7f6f8 Thomas Thrainer
                                         errors.ECODE_STATE)
1034 22b7f6f8 Thomas Thrainer
            self.LogInfo("Chose IP %s from network %s", nic.ip, nobj.name)
1035 22b7f6f8 Thomas Thrainer
          else:
1036 22b7f6f8 Thomas Thrainer
            try:
1037 22b7f6f8 Thomas Thrainer
              self.cfg.ReserveIp(net_uuid, nic.ip, self.proc.GetECId())
1038 22b7f6f8 Thomas Thrainer
            except errors.ReservationError:
1039 22b7f6f8 Thomas Thrainer
              raise errors.OpPrereqError("IP address %s already in use"
1040 22b7f6f8 Thomas Thrainer
                                         " or does not belong to network %s" %
1041 22b7f6f8 Thomas Thrainer
                                         (nic.ip, nobj.name),
1042 22b7f6f8 Thomas Thrainer
                                         errors.ECODE_NOTUNIQUE)
1043 22b7f6f8 Thomas Thrainer
1044 22b7f6f8 Thomas Thrainer
      # net is None, ip None or given
1045 22b7f6f8 Thomas Thrainer
      elif self.op.conflicts_check:
1046 1c3231aa Thomas Thrainer
        _CheckForConflictingIp(self, nic.ip, self.pnode.uuid)
1047 22b7f6f8 Thomas Thrainer
1048 22b7f6f8 Thomas Thrainer
    # mirror node verification
1049 22b7f6f8 Thomas Thrainer
    if self.op.disk_template in constants.DTS_INT_MIRROR:
1050 1c3231aa Thomas Thrainer
      if self.op.snode_uuid == pnode.uuid:
1051 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("The secondary node cannot be the"
1052 22b7f6f8 Thomas Thrainer
                                   " primary node", errors.ECODE_INVAL)
1053 1c3231aa Thomas Thrainer
      CheckNodeOnline(self, self.op.snode_uuid)
1054 1c3231aa Thomas Thrainer
      CheckNodeNotDrained(self, self.op.snode_uuid)
1055 1c3231aa Thomas Thrainer
      CheckNodeVmCapable(self, self.op.snode_uuid)
1056 1c3231aa Thomas Thrainer
      self.secondaries.append(self.op.snode_uuid)
1057 22b7f6f8 Thomas Thrainer
1058 1c3231aa Thomas Thrainer
      snode = self.cfg.GetNodeInfo(self.op.snode_uuid)
1059 22b7f6f8 Thomas Thrainer
      if pnode.group != snode.group:
1060 22b7f6f8 Thomas Thrainer
        self.LogWarning("The primary and secondary nodes are in two"
1061 22b7f6f8 Thomas Thrainer
                        " different node groups; the disk parameters"
1062 22b7f6f8 Thomas Thrainer
                        " from the first disk's node group will be"
1063 22b7f6f8 Thomas Thrainer
                        " used")
1064 22b7f6f8 Thomas Thrainer
1065 1bb99a33 Bernardo Dal Seno
    nodes = [pnode]
1066 1bb99a33 Bernardo Dal Seno
    if self.op.disk_template in constants.DTS_INT_MIRROR:
1067 1bb99a33 Bernardo Dal Seno
      nodes.append(snode)
1068 1bb99a33 Bernardo Dal Seno
    has_es = lambda n: IsExclusiveStorageEnabledNode(self.cfg, n)
1069 3f3ea14c Bernardo Dal Seno
    excl_stor = compat.any(map(has_es, nodes))
1070 3f3ea14c Bernardo Dal Seno
    if excl_stor and not self.op.disk_template in constants.DTS_EXCL_STORAGE:
1071 3f3ea14c Bernardo Dal Seno
      raise errors.OpPrereqError("Disk template %s not supported with"
1072 3f3ea14c Bernardo Dal Seno
                                 " exclusive storage" % self.op.disk_template,
1073 3f3ea14c Bernardo Dal Seno
                                 errors.ECODE_STATE)
1074 3f3ea14c Bernardo Dal Seno
    for disk in self.disks:
1075 7c848a6a Bernardo Dal Seno
      CheckSpindlesExclusiveStorage(disk, excl_stor, True)
1076 22b7f6f8 Thomas Thrainer
1077 1c3231aa Thomas Thrainer
    node_uuids = [pnode.uuid] + self.secondaries
1078 22b7f6f8 Thomas Thrainer
1079 22b7f6f8 Thomas Thrainer
    if not self.adopt_disks:
1080 22b7f6f8 Thomas Thrainer
      if self.op.disk_template == constants.DT_RBD:
1081 22b7f6f8 Thomas Thrainer
        # _CheckRADOSFreeSpace() is just a placeholder.
1082 22b7f6f8 Thomas Thrainer
        # Any function that checks prerequisites can be placed here.
1083 22b7f6f8 Thomas Thrainer
        # Check if there is enough space on the RADOS cluster.
1084 5eacbcae Thomas Thrainer
        CheckRADOSFreeSpace()
1085 22b7f6f8 Thomas Thrainer
      elif self.op.disk_template == constants.DT_EXT:
1086 22b7f6f8 Thomas Thrainer
        # FIXME: Function that checks prereqs if needed
1087 22b7f6f8 Thomas Thrainer
        pass
1088 d48c944b Helga Velroyen
      elif self.op.disk_template in constants.DTS_LVM:
1089 22b7f6f8 Thomas Thrainer
        # Check lv size requirements, if not adopting
1090 5eacbcae Thomas Thrainer
        req_sizes = ComputeDiskSizePerVG(self.op.disk_template, self.disks)
1091 1c3231aa Thomas Thrainer
        CheckNodesFreeDiskPerVG(self, node_uuids, req_sizes)
1092 4b92e992 Helga Velroyen
      else:
1093 4b92e992 Helga Velroyen
        # FIXME: add checks for other, non-adopting, non-lvm disk templates
1094 4b92e992 Helga Velroyen
        pass
1095 22b7f6f8 Thomas Thrainer
1096 22b7f6f8 Thomas Thrainer
    elif self.op.disk_template == constants.DT_PLAIN: # Check the adoption data
1097 22b7f6f8 Thomas Thrainer
      all_lvs = set(["%s/%s" % (disk[constants.IDISK_VG],
1098 22b7f6f8 Thomas Thrainer
                                disk[constants.IDISK_ADOPT])
1099 22b7f6f8 Thomas Thrainer
                     for disk in self.disks])
1100 22b7f6f8 Thomas Thrainer
      if len(all_lvs) != len(self.disks):
1101 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Duplicate volume names given for adoption",
1102 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
1103 22b7f6f8 Thomas Thrainer
      for lv_name in all_lvs:
1104 22b7f6f8 Thomas Thrainer
        try:
1105 22b7f6f8 Thomas Thrainer
          # FIXME: lv_name here is "vg/lv" need to ensure that other calls
1106 22b7f6f8 Thomas Thrainer
          # to ReserveLV uses the same syntax
1107 22b7f6f8 Thomas Thrainer
          self.cfg.ReserveLV(lv_name, self.proc.GetECId())
1108 22b7f6f8 Thomas Thrainer
        except errors.ReservationError:
1109 22b7f6f8 Thomas Thrainer
          raise errors.OpPrereqError("LV named %s used by another instance" %
1110 22b7f6f8 Thomas Thrainer
                                     lv_name, errors.ECODE_NOTUNIQUE)
1111 22b7f6f8 Thomas Thrainer
1112 1c3231aa Thomas Thrainer
      vg_names = self.rpc.call_vg_list([pnode.uuid])[pnode.uuid]
1113 22b7f6f8 Thomas Thrainer
      vg_names.Raise("Cannot get VG information from node %s" % pnode.name)
1114 22b7f6f8 Thomas Thrainer
1115 1c3231aa Thomas Thrainer
      node_lvs = self.rpc.call_lv_list([pnode.uuid],
1116 1c3231aa Thomas Thrainer
                                       vg_names.payload.keys())[pnode.uuid]
1117 22b7f6f8 Thomas Thrainer
      node_lvs.Raise("Cannot get LV information from node %s" % pnode.name)
1118 22b7f6f8 Thomas Thrainer
      node_lvs = node_lvs.payload
1119 22b7f6f8 Thomas Thrainer
1120 22b7f6f8 Thomas Thrainer
      delta = all_lvs.difference(node_lvs.keys())
1121 22b7f6f8 Thomas Thrainer
      if delta:
1122 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Missing logical volume(s): %s" %
1123 22b7f6f8 Thomas Thrainer
                                   utils.CommaJoin(delta),
1124 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
1125 22b7f6f8 Thomas Thrainer
      online_lvs = [lv for lv in all_lvs if node_lvs[lv][2]]
1126 22b7f6f8 Thomas Thrainer
      if online_lvs:
1127 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Online logical volumes found, cannot"
1128 22b7f6f8 Thomas Thrainer
                                   " adopt: %s" % utils.CommaJoin(online_lvs),
1129 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_STATE)
1130 22b7f6f8 Thomas Thrainer
      # update the size of disk based on what is found
1131 22b7f6f8 Thomas Thrainer
      for dsk in self.disks:
1132 22b7f6f8 Thomas Thrainer
        dsk[constants.IDISK_SIZE] = \
1133 22b7f6f8 Thomas Thrainer
          int(float(node_lvs["%s/%s" % (dsk[constants.IDISK_VG],
1134 22b7f6f8 Thomas Thrainer
                                        dsk[constants.IDISK_ADOPT])][0]))
1135 22b7f6f8 Thomas Thrainer
1136 22b7f6f8 Thomas Thrainer
    elif self.op.disk_template == constants.DT_BLOCK:
1137 22b7f6f8 Thomas Thrainer
      # Normalize and de-duplicate device paths
1138 22b7f6f8 Thomas Thrainer
      all_disks = set([os.path.abspath(disk[constants.IDISK_ADOPT])
1139 22b7f6f8 Thomas Thrainer
                       for disk in self.disks])
1140 22b7f6f8 Thomas Thrainer
      if len(all_disks) != len(self.disks):
1141 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Duplicate disk names given for adoption",
1142 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
1143 22b7f6f8 Thomas Thrainer
      baddisks = [d for d in all_disks
1144 22b7f6f8 Thomas Thrainer
                  if not d.startswith(constants.ADOPTABLE_BLOCKDEV_ROOT)]
1145 22b7f6f8 Thomas Thrainer
      if baddisks:
1146 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Device node(s) %s lie outside %s and"
1147 22b7f6f8 Thomas Thrainer
                                   " cannot be adopted" %
1148 22b7f6f8 Thomas Thrainer
                                   (utils.CommaJoin(baddisks),
1149 22b7f6f8 Thomas Thrainer
                                    constants.ADOPTABLE_BLOCKDEV_ROOT),
1150 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
1151 22b7f6f8 Thomas Thrainer
1152 1c3231aa Thomas Thrainer
      node_disks = self.rpc.call_bdev_sizes([pnode.uuid],
1153 1c3231aa Thomas Thrainer
                                            list(all_disks))[pnode.uuid]
1154 22b7f6f8 Thomas Thrainer
      node_disks.Raise("Cannot get block device information from node %s" %
1155 22b7f6f8 Thomas Thrainer
                       pnode.name)
1156 22b7f6f8 Thomas Thrainer
      node_disks = node_disks.payload
1157 22b7f6f8 Thomas Thrainer
      delta = all_disks.difference(node_disks.keys())
1158 22b7f6f8 Thomas Thrainer
      if delta:
1159 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Missing block device(s): %s" %
1160 22b7f6f8 Thomas Thrainer
                                   utils.CommaJoin(delta),
1161 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
1162 22b7f6f8 Thomas Thrainer
      for dsk in self.disks:
1163 22b7f6f8 Thomas Thrainer
        dsk[constants.IDISK_SIZE] = \
1164 22b7f6f8 Thomas Thrainer
          int(float(node_disks[dsk[constants.IDISK_ADOPT]]))
1165 22b7f6f8 Thomas Thrainer
1166 294254b1 Raffa Santi
    # Check disk access param to be compatible with specified hypervisor
1167 294254b1 Raffa Santi
    node_info = self.cfg.GetNodeInfo(self.op.pnode_uuid)
1168 294254b1 Raffa Santi
    node_group = self.cfg.GetNodeGroup(node_info.group)
1169 294254b1 Raffa Santi
    disk_params = self.cfg.GetGroupDiskParams(node_group)
1170 294254b1 Raffa Santi
    access_type = disk_params[self.op.disk_template].get(
1171 294254b1 Raffa Santi
      constants.RBD_ACCESS, constants.DISK_KERNELSPACE
1172 294254b1 Raffa Santi
    )
1173 294254b1 Raffa Santi
1174 294254b1 Raffa Santi
    if not IsValidDiskAccessModeCombination(self.op.hypervisor,
1175 294254b1 Raffa Santi
                                            self.op.disk_template,
1176 294254b1 Raffa Santi
                                            access_type):
1177 294254b1 Raffa Santi
      raise errors.OpPrereqError("Selected hypervisor (%s) cannot be"
1178 294254b1 Raffa Santi
                                 " used with %s disk access param" %
1179 294254b1 Raffa Santi
                                 (self.op.hypervisor, access_type),
1180 294254b1 Raffa Santi
                                  errors.ECODE_STATE)
1181 294254b1 Raffa Santi
1182 22b7f6f8 Thomas Thrainer
    # Verify instance specs
1183 22b7f6f8 Thomas Thrainer
    spindle_use = self.be_full.get(constants.BE_SPINDLE_USE, None)
1184 22b7f6f8 Thomas Thrainer
    ispec = {
1185 22b7f6f8 Thomas Thrainer
      constants.ISPEC_MEM_SIZE: self.be_full.get(constants.BE_MAXMEM, None),
1186 22b7f6f8 Thomas Thrainer
      constants.ISPEC_CPU_COUNT: self.be_full.get(constants.BE_VCPUS, None),
1187 22b7f6f8 Thomas Thrainer
      constants.ISPEC_DISK_COUNT: len(self.disks),
1188 22b7f6f8 Thomas Thrainer
      constants.ISPEC_DISK_SIZE: [disk[constants.IDISK_SIZE]
1189 22b7f6f8 Thomas Thrainer
                                  for disk in self.disks],
1190 22b7f6f8 Thomas Thrainer
      constants.ISPEC_NIC_COUNT: len(self.nics),
1191 22b7f6f8 Thomas Thrainer
      constants.ISPEC_SPINDLE_USE: spindle_use,
1192 22b7f6f8 Thomas Thrainer
      }
1193 22b7f6f8 Thomas Thrainer
1194 22b7f6f8 Thomas Thrainer
    group_info = self.cfg.GetNodeGroup(pnode.group)
1195 22b7f6f8 Thomas Thrainer
    ipolicy = ganeti.masterd.instance.CalculateGroupIPolicy(cluster, group_info)
1196 22b7f6f8 Thomas Thrainer
    res = _ComputeIPolicyInstanceSpecViolation(ipolicy, ispec,
1197 22b7f6f8 Thomas Thrainer
                                               self.op.disk_template)
1198 22b7f6f8 Thomas Thrainer
    if not self.op.ignore_ipolicy and res:
1199 22b7f6f8 Thomas Thrainer
      msg = ("Instance allocation to group %s (%s) violates policy: %s" %
1200 22b7f6f8 Thomas Thrainer
             (pnode.group, group_info.name, utils.CommaJoin(res)))
1201 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError(msg, errors.ECODE_INVAL)
1202 22b7f6f8 Thomas Thrainer
1203 1c3231aa Thomas Thrainer
    CheckHVParams(self, node_uuids, self.op.hypervisor, self.op.hvparams)
1204 22b7f6f8 Thomas Thrainer
1205 1c3231aa Thomas Thrainer
    CheckNodeHasOS(self, pnode.uuid, self.op.os_type, self.op.force_variant)
1206 22b7f6f8 Thomas Thrainer
    # check OS parameters (remotely)
1207 1c3231aa Thomas Thrainer
    CheckOSParams(self, True, node_uuids, self.op.os_type, self.os_full)
1208 22b7f6f8 Thomas Thrainer
1209 1c3231aa Thomas Thrainer
    CheckNicsBridgesExist(self, self.nics, self.pnode.uuid)
1210 22b7f6f8 Thomas Thrainer
1211 22b7f6f8 Thomas Thrainer
    #TODO: _CheckExtParams (remotely)
1212 22b7f6f8 Thomas Thrainer
    # Check parameters for extstorage
1213 22b7f6f8 Thomas Thrainer
1214 22b7f6f8 Thomas Thrainer
    # memory check on primary node
1215 22b7f6f8 Thomas Thrainer
    #TODO(dynmem): use MINMEM for checking
1216 22b7f6f8 Thomas Thrainer
    if self.op.start:
1217 a295eb80 Helga Velroyen
      hvfull = objects.FillDict(cluster.hvparams.get(self.op.hypervisor, {}),
1218 a295eb80 Helga Velroyen
                                self.op.hvparams)
1219 1c3231aa Thomas Thrainer
      CheckNodeFreeMemory(self, self.pnode.uuid,
1220 5eacbcae Thomas Thrainer
                          "creating instance %s" % self.op.instance_name,
1221 5eacbcae Thomas Thrainer
                          self.be_full[constants.BE_MAXMEM],
1222 a295eb80 Helga Velroyen
                          self.op.hypervisor, hvfull)
1223 22b7f6f8 Thomas Thrainer
1224 1c3231aa Thomas Thrainer
    self.dry_run_result = list(node_uuids)
1225 22b7f6f8 Thomas Thrainer
1226 22b7f6f8 Thomas Thrainer
  def Exec(self, feedback_fn):
1227 22b7f6f8 Thomas Thrainer
    """Create and add the instance to the cluster.
1228 22b7f6f8 Thomas Thrainer

1229 22b7f6f8 Thomas Thrainer
    """
1230 22b7f6f8 Thomas Thrainer
    assert not (self.owned_locks(locking.LEVEL_NODE_RES) -
1231 22b7f6f8 Thomas Thrainer
                self.owned_locks(locking.LEVEL_NODE)), \
1232 22b7f6f8 Thomas Thrainer
      "Node locks differ from node resource locks"
1233 22b7f6f8 Thomas Thrainer
    assert not self.glm.is_owned(locking.LEVEL_NODE_ALLOC)
1234 22b7f6f8 Thomas Thrainer
1235 22b7f6f8 Thomas Thrainer
    ht_kind = self.op.hypervisor
1236 22b7f6f8 Thomas Thrainer
    if ht_kind in constants.HTS_REQ_PORT:
1237 22b7f6f8 Thomas Thrainer
      network_port = self.cfg.AllocatePort()
1238 22b7f6f8 Thomas Thrainer
    else:
1239 22b7f6f8 Thomas Thrainer
      network_port = None
1240 22b7f6f8 Thomas Thrainer
1241 da4a52a3 Thomas Thrainer
    instance_uuid = self.cfg.GenerateUniqueID(self.proc.GetECId())
1242 da4a52a3 Thomas Thrainer
1243 22b7f6f8 Thomas Thrainer
    # This is ugly but we got a chicken-egg problem here
1244 22b7f6f8 Thomas Thrainer
    # We can only take the group disk parameters, as the instance
1245 22b7f6f8 Thomas Thrainer
    # has no disks yet (we are generating them right here).
1246 1c3231aa Thomas Thrainer
    nodegroup = self.cfg.GetNodeGroup(self.pnode.group)
1247 5eacbcae Thomas Thrainer
    disks = GenerateDiskTemplate(self,
1248 5eacbcae Thomas Thrainer
                                 self.op.disk_template,
1249 da4a52a3 Thomas Thrainer
                                 instance_uuid, self.pnode.uuid,
1250 5eacbcae Thomas Thrainer
                                 self.secondaries,
1251 5eacbcae Thomas Thrainer
                                 self.disks,
1252 5eacbcae Thomas Thrainer
                                 self.instance_file_storage_dir,
1253 5eacbcae Thomas Thrainer
                                 self.op.file_driver,
1254 5eacbcae Thomas Thrainer
                                 0,
1255 5eacbcae Thomas Thrainer
                                 feedback_fn,
1256 5eacbcae Thomas Thrainer
                                 self.cfg.GetGroupDiskParams(nodegroup))
1257 22b7f6f8 Thomas Thrainer
1258 da4a52a3 Thomas Thrainer
    iobj = objects.Instance(name=self.op.instance_name,
1259 da4a52a3 Thomas Thrainer
                            uuid=instance_uuid,
1260 da4a52a3 Thomas Thrainer
                            os=self.op.os_type,
1261 1c3231aa Thomas Thrainer
                            primary_node=self.pnode.uuid,
1262 22b7f6f8 Thomas Thrainer
                            nics=self.nics, disks=disks,
1263 22b7f6f8 Thomas Thrainer
                            disk_template=self.op.disk_template,
1264 1d4a4b26 Thomas Thrainer
                            disks_active=False,
1265 22b7f6f8 Thomas Thrainer
                            admin_state=constants.ADMINST_DOWN,
1266 22b7f6f8 Thomas Thrainer
                            network_port=network_port,
1267 22b7f6f8 Thomas Thrainer
                            beparams=self.op.beparams,
1268 22b7f6f8 Thomas Thrainer
                            hvparams=self.op.hvparams,
1269 22b7f6f8 Thomas Thrainer
                            hypervisor=self.op.hypervisor,
1270 22b7f6f8 Thomas Thrainer
                            osparams=self.op.osparams,
1271 22b7f6f8 Thomas Thrainer
                            )
1272 22b7f6f8 Thomas Thrainer
1273 22b7f6f8 Thomas Thrainer
    if self.op.tags:
1274 22b7f6f8 Thomas Thrainer
      for tag in self.op.tags:
1275 22b7f6f8 Thomas Thrainer
        iobj.AddTag(tag)
1276 22b7f6f8 Thomas Thrainer
1277 22b7f6f8 Thomas Thrainer
    if self.adopt_disks:
1278 22b7f6f8 Thomas Thrainer
      if self.op.disk_template == constants.DT_PLAIN:
1279 22b7f6f8 Thomas Thrainer
        # rename LVs to the newly-generated names; we need to construct
1280 22b7f6f8 Thomas Thrainer
        # 'fake' LV disks with the old data, plus the new unique_id
1281 22b7f6f8 Thomas Thrainer
        tmp_disks = [objects.Disk.FromDict(v.ToDict()) for v in disks]
1282 22b7f6f8 Thomas Thrainer
        rename_to = []
1283 22b7f6f8 Thomas Thrainer
        for t_dsk, a_dsk in zip(tmp_disks, self.disks):
1284 22b7f6f8 Thomas Thrainer
          rename_to.append(t_dsk.logical_id)
1285 22b7f6f8 Thomas Thrainer
          t_dsk.logical_id = (t_dsk.logical_id[0], a_dsk[constants.IDISK_ADOPT])
1286 1c3231aa Thomas Thrainer
        result = self.rpc.call_blockdev_rename(self.pnode.uuid,
1287 22b7f6f8 Thomas Thrainer
                                               zip(tmp_disks, rename_to))
1288 22b7f6f8 Thomas Thrainer
        result.Raise("Failed to rename adoped LVs")
1289 22b7f6f8 Thomas Thrainer
    else:
1290 22b7f6f8 Thomas Thrainer
      feedback_fn("* creating instance disks...")
1291 22b7f6f8 Thomas Thrainer
      try:
1292 5eacbcae Thomas Thrainer
        CreateDisks(self, iobj)
1293 22b7f6f8 Thomas Thrainer
      except errors.OpExecError:
1294 22b7f6f8 Thomas Thrainer
        self.LogWarning("Device creation failed")
1295 d0d7d7cf Thomas Thrainer
        self.cfg.ReleaseDRBDMinors(self.op.instance_name)
1296 22b7f6f8 Thomas Thrainer
        raise
1297 22b7f6f8 Thomas Thrainer
1298 d0d7d7cf Thomas Thrainer
    feedback_fn("adding instance %s to cluster config" % self.op.instance_name)
1299 22b7f6f8 Thomas Thrainer
1300 22b7f6f8 Thomas Thrainer
    self.cfg.AddInstance(iobj, self.proc.GetECId())
1301 22b7f6f8 Thomas Thrainer
1302 22b7f6f8 Thomas Thrainer
    # Declare that we don't want to remove the instance lock anymore, as we've
1303 22b7f6f8 Thomas Thrainer
    # added the instance to the config
1304 22b7f6f8 Thomas Thrainer
    del self.remove_locks[locking.LEVEL_INSTANCE]
1305 22b7f6f8 Thomas Thrainer
1306 22b7f6f8 Thomas Thrainer
    if self.op.mode == constants.INSTANCE_IMPORT:
1307 22b7f6f8 Thomas Thrainer
      # Release unused nodes
1308 1c3231aa Thomas Thrainer
      ReleaseLocks(self, locking.LEVEL_NODE, keep=[self.op.src_node_uuid])
1309 22b7f6f8 Thomas Thrainer
    else:
1310 22b7f6f8 Thomas Thrainer
      # Release all nodes
1311 5eacbcae Thomas Thrainer
      ReleaseLocks(self, locking.LEVEL_NODE)
1312 22b7f6f8 Thomas Thrainer
1313 22b7f6f8 Thomas Thrainer
    disk_abort = False
1314 22b7f6f8 Thomas Thrainer
    if not self.adopt_disks and self.cfg.GetClusterInfo().prealloc_wipe_disks:
1315 22b7f6f8 Thomas Thrainer
      feedback_fn("* wiping instance disks...")
1316 22b7f6f8 Thomas Thrainer
      try:
1317 5eacbcae Thomas Thrainer
        WipeDisks(self, iobj)
1318 22b7f6f8 Thomas Thrainer
      except errors.OpExecError, err:
1319 22b7f6f8 Thomas Thrainer
        logging.exception("Wiping disks failed")
1320 22b7f6f8 Thomas Thrainer
        self.LogWarning("Wiping instance disks failed (%s)", err)
1321 22b7f6f8 Thomas Thrainer
        disk_abort = True
1322 22b7f6f8 Thomas Thrainer
1323 22b7f6f8 Thomas Thrainer
    if disk_abort:
1324 22b7f6f8 Thomas Thrainer
      # Something is already wrong with the disks, don't do anything else
1325 22b7f6f8 Thomas Thrainer
      pass
1326 22b7f6f8 Thomas Thrainer
    elif self.op.wait_for_sync:
1327 5eacbcae Thomas Thrainer
      disk_abort = not WaitForSync(self, iobj)
1328 22b7f6f8 Thomas Thrainer
    elif iobj.disk_template in constants.DTS_INT_MIRROR:
1329 22b7f6f8 Thomas Thrainer
      # make sure the disks are not degraded (still sync-ing is ok)
1330 22b7f6f8 Thomas Thrainer
      feedback_fn("* checking mirrors status")
1331 5eacbcae Thomas Thrainer
      disk_abort = not WaitForSync(self, iobj, oneshot=True)
1332 22b7f6f8 Thomas Thrainer
    else:
1333 22b7f6f8 Thomas Thrainer
      disk_abort = False
1334 22b7f6f8 Thomas Thrainer
1335 22b7f6f8 Thomas Thrainer
    if disk_abort:
1336 5eacbcae Thomas Thrainer
      RemoveDisks(self, iobj)
1337 da4a52a3 Thomas Thrainer
      self.cfg.RemoveInstance(iobj.uuid)
1338 22b7f6f8 Thomas Thrainer
      # Make sure the instance lock gets removed
1339 22b7f6f8 Thomas Thrainer
      self.remove_locks[locking.LEVEL_INSTANCE] = iobj.name
1340 22b7f6f8 Thomas Thrainer
      raise errors.OpExecError("There are some degraded disks for"
1341 22b7f6f8 Thomas Thrainer
                               " this instance")
1342 22b7f6f8 Thomas Thrainer
1343 1d4a4b26 Thomas Thrainer
    # instance disks are now active
1344 1d4a4b26 Thomas Thrainer
    iobj.disks_active = True
1345 1d4a4b26 Thomas Thrainer
1346 22b7f6f8 Thomas Thrainer
    # Release all node resource locks
1347 5eacbcae Thomas Thrainer
    ReleaseLocks(self, locking.LEVEL_NODE_RES)
1348 22b7f6f8 Thomas Thrainer
1349 22b7f6f8 Thomas Thrainer
    if iobj.disk_template != constants.DT_DISKLESS and not self.adopt_disks:
1350 22b7f6f8 Thomas Thrainer
      if self.op.mode == constants.INSTANCE_CREATE:
1351 22b7f6f8 Thomas Thrainer
        if not self.op.no_install:
1352 22b7f6f8 Thomas Thrainer
          pause_sync = (iobj.disk_template in constants.DTS_INT_MIRROR and
1353 22b7f6f8 Thomas Thrainer
                        not self.op.wait_for_sync)
1354 22b7f6f8 Thomas Thrainer
          if pause_sync:
1355 22b7f6f8 Thomas Thrainer
            feedback_fn("* pausing disk sync to install instance OS")
1356 1c3231aa Thomas Thrainer
            result = self.rpc.call_blockdev_pause_resume_sync(self.pnode.uuid,
1357 22b7f6f8 Thomas Thrainer
                                                              (iobj.disks,
1358 22b7f6f8 Thomas Thrainer
                                                               iobj), True)
1359 22b7f6f8 Thomas Thrainer
            for idx, success in enumerate(result.payload):
1360 22b7f6f8 Thomas Thrainer
              if not success:
1361 22b7f6f8 Thomas Thrainer
                logging.warn("pause-sync of instance %s for disk %d failed",
1362 d0d7d7cf Thomas Thrainer
                             self.op.instance_name, idx)
1363 22b7f6f8 Thomas Thrainer
1364 22b7f6f8 Thomas Thrainer
          feedback_fn("* running the instance OS create scripts...")
1365 22b7f6f8 Thomas Thrainer
          # FIXME: pass debug option from opcode to backend
1366 22b7f6f8 Thomas Thrainer
          os_add_result = \
1367 1c3231aa Thomas Thrainer
            self.rpc.call_instance_os_add(self.pnode.uuid, (iobj, None), False,
1368 22b7f6f8 Thomas Thrainer
                                          self.op.debug_level)
1369 22b7f6f8 Thomas Thrainer
          if pause_sync:
1370 22b7f6f8 Thomas Thrainer
            feedback_fn("* resuming disk sync")
1371 1c3231aa Thomas Thrainer
            result = self.rpc.call_blockdev_pause_resume_sync(self.pnode.uuid,
1372 22b7f6f8 Thomas Thrainer
                                                              (iobj.disks,
1373 22b7f6f8 Thomas Thrainer
                                                               iobj), False)
1374 22b7f6f8 Thomas Thrainer
            for idx, success in enumerate(result.payload):
1375 22b7f6f8 Thomas Thrainer
              if not success:
1376 22b7f6f8 Thomas Thrainer
                logging.warn("resume-sync of instance %s for disk %d failed",
1377 d0d7d7cf Thomas Thrainer
                             self.op.instance_name, idx)
1378 22b7f6f8 Thomas Thrainer
1379 22b7f6f8 Thomas Thrainer
          os_add_result.Raise("Could not add os for instance %s"
1380 d0d7d7cf Thomas Thrainer
                              " on node %s" % (self.op.instance_name,
1381 d0d7d7cf Thomas Thrainer
                                               self.pnode.name))
1382 22b7f6f8 Thomas Thrainer
1383 22b7f6f8 Thomas Thrainer
      else:
1384 22b7f6f8 Thomas Thrainer
        if self.op.mode == constants.INSTANCE_IMPORT:
1385 22b7f6f8 Thomas Thrainer
          feedback_fn("* running the instance OS import scripts...")
1386 22b7f6f8 Thomas Thrainer
1387 22b7f6f8 Thomas Thrainer
          transfers = []
1388 22b7f6f8 Thomas Thrainer
1389 22b7f6f8 Thomas Thrainer
          for idx, image in enumerate(self.src_images):
1390 22b7f6f8 Thomas Thrainer
            if not image:
1391 22b7f6f8 Thomas Thrainer
              continue
1392 22b7f6f8 Thomas Thrainer
1393 22b7f6f8 Thomas Thrainer
            # FIXME: pass debug option from opcode to backend
1394 22b7f6f8 Thomas Thrainer
            dt = masterd.instance.DiskTransfer("disk/%s" % idx,
1395 22b7f6f8 Thomas Thrainer
                                               constants.IEIO_FILE, (image, ),
1396 22b7f6f8 Thomas Thrainer
                                               constants.IEIO_SCRIPT,
1397 0c3d9c7c Thomas Thrainer
                                               ((iobj.disks[idx], iobj), idx),
1398 22b7f6f8 Thomas Thrainer
                                               None)
1399 22b7f6f8 Thomas Thrainer
            transfers.append(dt)
1400 22b7f6f8 Thomas Thrainer
1401 22b7f6f8 Thomas Thrainer
          import_result = \
1402 22b7f6f8 Thomas Thrainer
            masterd.instance.TransferInstanceData(self, feedback_fn,
1403 1c3231aa Thomas Thrainer
                                                  self.op.src_node_uuid,
1404 1c3231aa Thomas Thrainer
                                                  self.pnode.uuid,
1405 22b7f6f8 Thomas Thrainer
                                                  self.pnode.secondary_ip,
1406 51d7ac96 Thomas Thrainer
                                                  self.op.compress,
1407 22b7f6f8 Thomas Thrainer
                                                  iobj, transfers)
1408 22b7f6f8 Thomas Thrainer
          if not compat.all(import_result):
1409 22b7f6f8 Thomas Thrainer
            self.LogWarning("Some disks for instance %s on node %s were not"
1410 d0d7d7cf Thomas Thrainer
                            " imported successfully" % (self.op.instance_name,
1411 d0d7d7cf Thomas Thrainer
                                                        self.pnode.name))
1412 22b7f6f8 Thomas Thrainer
1413 22b7f6f8 Thomas Thrainer
          rename_from = self._old_instance_name
1414 22b7f6f8 Thomas Thrainer
1415 22b7f6f8 Thomas Thrainer
        elif self.op.mode == constants.INSTANCE_REMOTE_IMPORT:
1416 22b7f6f8 Thomas Thrainer
          feedback_fn("* preparing remote import...")
1417 22b7f6f8 Thomas Thrainer
          # The source cluster will stop the instance before attempting to make
1418 22b7f6f8 Thomas Thrainer
          # a connection. In some cases stopping an instance can take a long
1419 22b7f6f8 Thomas Thrainer
          # time, hence the shutdown timeout is added to the connection
1420 22b7f6f8 Thomas Thrainer
          # timeout.
1421 22b7f6f8 Thomas Thrainer
          connect_timeout = (constants.RIE_CONNECT_TIMEOUT +
1422 22b7f6f8 Thomas Thrainer
                             self.op.source_shutdown_timeout)
1423 22b7f6f8 Thomas Thrainer
          timeouts = masterd.instance.ImportExportTimeouts(connect_timeout)
1424 22b7f6f8 Thomas Thrainer
1425 1c3231aa Thomas Thrainer
          assert iobj.primary_node == self.pnode.uuid
1426 22b7f6f8 Thomas Thrainer
          disk_results = \
1427 22b7f6f8 Thomas Thrainer
            masterd.instance.RemoteImport(self, feedback_fn, iobj, self.pnode,
1428 22b7f6f8 Thomas Thrainer
                                          self.source_x509_ca,
1429 88acff3f Thomas Thrainer
                                          self._cds, self.op.compress, timeouts)
1430 22b7f6f8 Thomas Thrainer
          if not compat.all(disk_results):
1431 22b7f6f8 Thomas Thrainer
            # TODO: Should the instance still be started, even if some disks
1432 22b7f6f8 Thomas Thrainer
            # failed to import (valid for local imports, too)?
1433 22b7f6f8 Thomas Thrainer
            self.LogWarning("Some disks for instance %s on node %s were not"
1434 d0d7d7cf Thomas Thrainer
                            " imported successfully" % (self.op.instance_name,
1435 d0d7d7cf Thomas Thrainer
                                                        self.pnode.name))
1436 22b7f6f8 Thomas Thrainer
1437 22b7f6f8 Thomas Thrainer
          rename_from = self.source_instance_name
1438 22b7f6f8 Thomas Thrainer
1439 22b7f6f8 Thomas Thrainer
        else:
1440 22b7f6f8 Thomas Thrainer
          # also checked in the prereq part
1441 22b7f6f8 Thomas Thrainer
          raise errors.ProgrammerError("Unknown OS initialization mode '%s'"
1442 22b7f6f8 Thomas Thrainer
                                       % self.op.mode)
1443 22b7f6f8 Thomas Thrainer
1444 22b7f6f8 Thomas Thrainer
        # Run rename script on newly imported instance
1445 d0d7d7cf Thomas Thrainer
        assert iobj.name == self.op.instance_name
1446 d0d7d7cf Thomas Thrainer
        feedback_fn("Running rename script for %s" % self.op.instance_name)
1447 1c3231aa Thomas Thrainer
        result = self.rpc.call_instance_run_rename(self.pnode.uuid, iobj,
1448 22b7f6f8 Thomas Thrainer
                                                   rename_from,
1449 22b7f6f8 Thomas Thrainer
                                                   self.op.debug_level)
1450 c7dd65be Klaus Aehlig
        result.Warn("Failed to run rename script for %s on node %s" %
1451 d0d7d7cf Thomas Thrainer
                    (self.op.instance_name, self.pnode.name), self.LogWarning)
1452 22b7f6f8 Thomas Thrainer
1453 22b7f6f8 Thomas Thrainer
    assert not self.owned_locks(locking.LEVEL_NODE_RES)
1454 22b7f6f8 Thomas Thrainer
1455 22b7f6f8 Thomas Thrainer
    if self.op.start:
1456 22b7f6f8 Thomas Thrainer
      iobj.admin_state = constants.ADMINST_UP
1457 22b7f6f8 Thomas Thrainer
      self.cfg.Update(iobj, feedback_fn)
1458 d0d7d7cf Thomas Thrainer
      logging.info("Starting instance %s on node %s", self.op.instance_name,
1459 d0d7d7cf Thomas Thrainer
                   self.pnode.name)
1460 22b7f6f8 Thomas Thrainer
      feedback_fn("* starting instance...")
1461 1c3231aa Thomas Thrainer
      result = self.rpc.call_instance_start(self.pnode.uuid, (iobj, None, None),
1462 22b7f6f8 Thomas Thrainer
                                            False, self.op.reason)
1463 22b7f6f8 Thomas Thrainer
      result.Raise("Could not start instance")
1464 22b7f6f8 Thomas Thrainer
1465 22b7f6f8 Thomas Thrainer
    return list(iobj.all_nodes)
1466 22b7f6f8 Thomas Thrainer
1467 22b7f6f8 Thomas Thrainer
1468 22b7f6f8 Thomas Thrainer
class LUInstanceRename(LogicalUnit):
1469 22b7f6f8 Thomas Thrainer
  """Rename an instance.
1470 22b7f6f8 Thomas Thrainer

1471 22b7f6f8 Thomas Thrainer
  """
1472 22b7f6f8 Thomas Thrainer
  HPATH = "instance-rename"
1473 22b7f6f8 Thomas Thrainer
  HTYPE = constants.HTYPE_INSTANCE
1474 22b7f6f8 Thomas Thrainer
1475 22b7f6f8 Thomas Thrainer
  def CheckArguments(self):
1476 22b7f6f8 Thomas Thrainer
    """Check arguments.
1477 22b7f6f8 Thomas Thrainer

1478 22b7f6f8 Thomas Thrainer
    """
1479 22b7f6f8 Thomas Thrainer
    if self.op.ip_check and not self.op.name_check:
1480 22b7f6f8 Thomas Thrainer
      # TODO: make the ip check more flexible and not depend on the name check
1481 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("IP address check requires a name check",
1482 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_INVAL)
1483 22b7f6f8 Thomas Thrainer
1484 22b7f6f8 Thomas Thrainer
  def BuildHooksEnv(self):
1485 22b7f6f8 Thomas Thrainer
    """Build hooks env.
1486 22b7f6f8 Thomas Thrainer

1487 22b7f6f8 Thomas Thrainer
    This runs on master, primary and secondary nodes of the instance.
1488 22b7f6f8 Thomas Thrainer

1489 22b7f6f8 Thomas Thrainer
    """
1490 5eacbcae Thomas Thrainer
    env = BuildInstanceHookEnvByObject(self, self.instance)
1491 22b7f6f8 Thomas Thrainer
    env["INSTANCE_NEW_NAME"] = self.op.new_name
1492 22b7f6f8 Thomas Thrainer
    return env
1493 22b7f6f8 Thomas Thrainer
1494 22b7f6f8 Thomas Thrainer
  def BuildHooksNodes(self):
1495 22b7f6f8 Thomas Thrainer
    """Build hooks nodes.
1496 22b7f6f8 Thomas Thrainer

1497 22b7f6f8 Thomas Thrainer
    """
1498 22b7f6f8 Thomas Thrainer
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
1499 22b7f6f8 Thomas Thrainer
    return (nl, nl)
1500 22b7f6f8 Thomas Thrainer
1501 22b7f6f8 Thomas Thrainer
  def CheckPrereq(self):
1502 22b7f6f8 Thomas Thrainer
    """Check prerequisites.
1503 22b7f6f8 Thomas Thrainer

1504 22b7f6f8 Thomas Thrainer
    This checks that the instance is in the cluster and is not running.
1505 22b7f6f8 Thomas Thrainer

1506 22b7f6f8 Thomas Thrainer
    """
1507 da4a52a3 Thomas Thrainer
    (self.op.instance_uuid, self.op.instance_name) = \
1508 da4a52a3 Thomas Thrainer
      ExpandInstanceUuidAndName(self.cfg, self.op.instance_uuid,
1509 da4a52a3 Thomas Thrainer
                                self.op.instance_name)
1510 da4a52a3 Thomas Thrainer
    instance = self.cfg.GetInstanceInfo(self.op.instance_uuid)
1511 22b7f6f8 Thomas Thrainer
    assert instance is not None
1512 1f7c8208 Helga Velroyen
1513 1f7c8208 Helga Velroyen
    # It should actually not happen that an instance is running with a disabled
1514 1f7c8208 Helga Velroyen
    # disk template, but in case it does, the renaming of file-based instances
1515 1f7c8208 Helga Velroyen
    # will fail horribly. Thus, we test it before.
1516 1f7c8208 Helga Velroyen
    if (instance.disk_template in constants.DTS_FILEBASED and
1517 1f7c8208 Helga Velroyen
        self.op.new_name != instance.name):
1518 1f7c8208 Helga Velroyen
      CheckDiskTemplateEnabled(self.cfg.GetClusterInfo(),
1519 1f7c8208 Helga Velroyen
                               instance.disk_template)
1520 1f7c8208 Helga Velroyen
1521 5eacbcae Thomas Thrainer
    CheckNodeOnline(self, instance.primary_node)
1522 5eacbcae Thomas Thrainer
    CheckInstanceState(self, instance, INSTANCE_NOT_RUNNING,
1523 5eacbcae Thomas Thrainer
                       msg="cannot rename")
1524 22b7f6f8 Thomas Thrainer
    self.instance = instance
1525 22b7f6f8 Thomas Thrainer
1526 22b7f6f8 Thomas Thrainer
    new_name = self.op.new_name
1527 22b7f6f8 Thomas Thrainer
    if self.op.name_check:
1528 22b7f6f8 Thomas Thrainer
      hostname = _CheckHostnameSane(self, new_name)
1529 22b7f6f8 Thomas Thrainer
      new_name = self.op.new_name = hostname.name
1530 22b7f6f8 Thomas Thrainer
      if (self.op.ip_check and
1531 22b7f6f8 Thomas Thrainer
          netutils.TcpPing(hostname.ip, constants.DEFAULT_NODED_PORT)):
1532 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("IP %s of instance %s already in use" %
1533 22b7f6f8 Thomas Thrainer
                                   (hostname.ip, new_name),
1534 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_NOTUNIQUE)
1535 22b7f6f8 Thomas Thrainer
1536 da4a52a3 Thomas Thrainer
    instance_names = [inst.name for
1537 da4a52a3 Thomas Thrainer
                      inst in self.cfg.GetAllInstancesInfo().values()]
1538 da4a52a3 Thomas Thrainer
    if new_name in instance_names and new_name != instance.name:
1539 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Instance '%s' is already in the cluster" %
1540 22b7f6f8 Thomas Thrainer
                                 new_name, errors.ECODE_EXISTS)
1541 22b7f6f8 Thomas Thrainer
1542 22b7f6f8 Thomas Thrainer
  def Exec(self, feedback_fn):
1543 22b7f6f8 Thomas Thrainer
    """Rename the instance.
1544 22b7f6f8 Thomas Thrainer

1545 22b7f6f8 Thomas Thrainer
    """
1546 d0d7d7cf Thomas Thrainer
    old_name = self.instance.name
1547 22b7f6f8 Thomas Thrainer
1548 22b7f6f8 Thomas Thrainer
    rename_file_storage = False
1549 d0d7d7cf Thomas Thrainer
    if (self.instance.disk_template in constants.DTS_FILEBASED and
1550 d0d7d7cf Thomas Thrainer
        self.op.new_name != self.instance.name):
1551 d0d7d7cf Thomas Thrainer
      old_file_storage_dir = os.path.dirname(
1552 d0d7d7cf Thomas Thrainer
                               self.instance.disks[0].logical_id[1])
1553 22b7f6f8 Thomas Thrainer
      rename_file_storage = True
1554 22b7f6f8 Thomas Thrainer
1555 da4a52a3 Thomas Thrainer
    self.cfg.RenameInstance(self.instance.uuid, self.op.new_name)
1556 22b7f6f8 Thomas Thrainer
    # Change the instance lock. This is definitely safe while we hold the BGL.
1557 22b7f6f8 Thomas Thrainer
    # Otherwise the new lock would have to be added in acquired mode.
1558 22b7f6f8 Thomas Thrainer
    assert self.REQ_BGL
1559 22b7f6f8 Thomas Thrainer
    assert locking.BGL in self.owned_locks(locking.LEVEL_CLUSTER)
1560 22b7f6f8 Thomas Thrainer
    self.glm.remove(locking.LEVEL_INSTANCE, old_name)
1561 22b7f6f8 Thomas Thrainer
    self.glm.add(locking.LEVEL_INSTANCE, self.op.new_name)
1562 22b7f6f8 Thomas Thrainer
1563 22b7f6f8 Thomas Thrainer
    # re-read the instance from the configuration after rename
1564 da4a52a3 Thomas Thrainer
    renamed_inst = self.cfg.GetInstanceInfo(self.instance.uuid)
1565 22b7f6f8 Thomas Thrainer
1566 22b7f6f8 Thomas Thrainer
    if rename_file_storage:
1567 d0d7d7cf Thomas Thrainer
      new_file_storage_dir = os.path.dirname(
1568 d0d7d7cf Thomas Thrainer
                               renamed_inst.disks[0].logical_id[1])
1569 d0d7d7cf Thomas Thrainer
      result = self.rpc.call_file_storage_dir_rename(renamed_inst.primary_node,
1570 22b7f6f8 Thomas Thrainer
                                                     old_file_storage_dir,
1571 22b7f6f8 Thomas Thrainer
                                                     new_file_storage_dir)
1572 22b7f6f8 Thomas Thrainer
      result.Raise("Could not rename on node %s directory '%s' to '%s'"
1573 22b7f6f8 Thomas Thrainer
                   " (but the instance has been renamed in Ganeti)" %
1574 d0d7d7cf Thomas Thrainer
                   (self.cfg.GetNodeName(renamed_inst.primary_node),
1575 1c3231aa Thomas Thrainer
                    old_file_storage_dir, new_file_storage_dir))
1576 22b7f6f8 Thomas Thrainer
1577 d0d7d7cf Thomas Thrainer
    StartInstanceDisks(self, renamed_inst, None)
1578 22b7f6f8 Thomas Thrainer
    # update info on disks
1579 d0d7d7cf Thomas Thrainer
    info = GetInstanceInfoText(renamed_inst)
1580 d0d7d7cf Thomas Thrainer
    for (idx, disk) in enumerate(renamed_inst.disks):
1581 d0d7d7cf Thomas Thrainer
      for node_uuid in renamed_inst.all_nodes:
1582 0c3d9c7c Thomas Thrainer
        result = self.rpc.call_blockdev_setinfo(node_uuid,
1583 0c3d9c7c Thomas Thrainer
                                                (disk, renamed_inst), info)
1584 1c3231aa Thomas Thrainer
        result.Warn("Error setting info on node %s for disk %s" %
1585 1c3231aa Thomas Thrainer
                    (self.cfg.GetNodeName(node_uuid), idx), self.LogWarning)
1586 22b7f6f8 Thomas Thrainer
    try:
1587 d0d7d7cf Thomas Thrainer
      result = self.rpc.call_instance_run_rename(renamed_inst.primary_node,
1588 d0d7d7cf Thomas Thrainer
                                                 renamed_inst, old_name,
1589 d0d7d7cf Thomas Thrainer
                                                 self.op.debug_level)
1590 c7dd65be Klaus Aehlig
      result.Warn("Could not run OS rename script for instance %s on node %s"
1591 c7dd65be Klaus Aehlig
                  " (but the instance has been renamed in Ganeti)" %
1592 d0d7d7cf Thomas Thrainer
                  (renamed_inst.name,
1593 d0d7d7cf Thomas Thrainer
                   self.cfg.GetNodeName(renamed_inst.primary_node)),
1594 1c3231aa Thomas Thrainer
                  self.LogWarning)
1595 22b7f6f8 Thomas Thrainer
    finally:
1596 d0d7d7cf Thomas Thrainer
      ShutdownInstanceDisks(self, renamed_inst)
1597 22b7f6f8 Thomas Thrainer
1598 d0d7d7cf Thomas Thrainer
    return renamed_inst.name
1599 22b7f6f8 Thomas Thrainer
1600 22b7f6f8 Thomas Thrainer
1601 22b7f6f8 Thomas Thrainer
class LUInstanceRemove(LogicalUnit):
1602 22b7f6f8 Thomas Thrainer
  """Remove an instance.
1603 22b7f6f8 Thomas Thrainer

1604 22b7f6f8 Thomas Thrainer
  """
1605 22b7f6f8 Thomas Thrainer
  HPATH = "instance-remove"
1606 22b7f6f8 Thomas Thrainer
  HTYPE = constants.HTYPE_INSTANCE
1607 22b7f6f8 Thomas Thrainer
  REQ_BGL = False
1608 22b7f6f8 Thomas Thrainer
1609 22b7f6f8 Thomas Thrainer
  def ExpandNames(self):
1610 22b7f6f8 Thomas Thrainer
    self._ExpandAndLockInstance()
1611 22b7f6f8 Thomas Thrainer
    self.needed_locks[locking.LEVEL_NODE] = []
1612 22b7f6f8 Thomas Thrainer
    self.needed_locks[locking.LEVEL_NODE_RES] = []
1613 22b7f6f8 Thomas Thrainer
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
1614 22b7f6f8 Thomas Thrainer
1615 22b7f6f8 Thomas Thrainer
  def DeclareLocks(self, level):
1616 22b7f6f8 Thomas Thrainer
    if level == locking.LEVEL_NODE:
1617 22b7f6f8 Thomas Thrainer
      self._LockInstancesNodes()
1618 22b7f6f8 Thomas Thrainer
    elif level == locking.LEVEL_NODE_RES:
1619 22b7f6f8 Thomas Thrainer
      # Copy node locks
1620 22b7f6f8 Thomas Thrainer
      self.needed_locks[locking.LEVEL_NODE_RES] = \
1621 5eacbcae Thomas Thrainer
        CopyLockList(self.needed_locks[locking.LEVEL_NODE])
1622 22b7f6f8 Thomas Thrainer
1623 22b7f6f8 Thomas Thrainer
  def BuildHooksEnv(self):
1624 22b7f6f8 Thomas Thrainer
    """Build hooks env.
1625 22b7f6f8 Thomas Thrainer

1626 22b7f6f8 Thomas Thrainer
    This runs on master, primary and secondary nodes of the instance.
1627 22b7f6f8 Thomas Thrainer

1628 22b7f6f8 Thomas Thrainer
    """
1629 5eacbcae Thomas Thrainer
    env = BuildInstanceHookEnvByObject(self, self.instance)
1630 22b7f6f8 Thomas Thrainer
    env["SHUTDOWN_TIMEOUT"] = self.op.shutdown_timeout
1631 22b7f6f8 Thomas Thrainer
    return env
1632 22b7f6f8 Thomas Thrainer
1633 22b7f6f8 Thomas Thrainer
  def BuildHooksNodes(self):
1634 22b7f6f8 Thomas Thrainer
    """Build hooks nodes.
1635 22b7f6f8 Thomas Thrainer

1636 22b7f6f8 Thomas Thrainer
    """
1637 22b7f6f8 Thomas Thrainer
    nl = [self.cfg.GetMasterNode()]
1638 22b7f6f8 Thomas Thrainer
    nl_post = list(self.instance.all_nodes) + nl
1639 22b7f6f8 Thomas Thrainer
    return (nl, nl_post)
1640 22b7f6f8 Thomas Thrainer
1641 22b7f6f8 Thomas Thrainer
  def CheckPrereq(self):
1642 22b7f6f8 Thomas Thrainer
    """Check prerequisites.
1643 22b7f6f8 Thomas Thrainer

1644 22b7f6f8 Thomas Thrainer
    This checks that the instance is in the cluster.
1645 22b7f6f8 Thomas Thrainer

1646 22b7f6f8 Thomas Thrainer
    """
1647 da4a52a3 Thomas Thrainer
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_uuid)
1648 22b7f6f8 Thomas Thrainer
    assert self.instance is not None, \
1649 22b7f6f8 Thomas Thrainer
      "Cannot retrieve locked instance %s" % self.op.instance_name
1650 22b7f6f8 Thomas Thrainer
1651 22b7f6f8 Thomas Thrainer
  def Exec(self, feedback_fn):
1652 22b7f6f8 Thomas Thrainer
    """Remove the instance.
1653 22b7f6f8 Thomas Thrainer

1654 22b7f6f8 Thomas Thrainer
    """
1655 d0d7d7cf Thomas Thrainer
    logging.info("Shutting down instance %s on node %s", self.instance.name,
1656 d0d7d7cf Thomas Thrainer
                 self.cfg.GetNodeName(self.instance.primary_node))
1657 22b7f6f8 Thomas Thrainer
1658 d0d7d7cf Thomas Thrainer
    result = self.rpc.call_instance_shutdown(self.instance.primary_node,
1659 d0d7d7cf Thomas Thrainer
                                             self.instance,
1660 22b7f6f8 Thomas Thrainer
                                             self.op.shutdown_timeout,
1661 22b7f6f8 Thomas Thrainer
                                             self.op.reason)
1662 c7dd65be Klaus Aehlig
    if self.op.ignore_failures:
1663 c7dd65be Klaus Aehlig
      result.Warn("Warning: can't shutdown instance", feedback_fn)
1664 c7dd65be Klaus Aehlig
    else:
1665 c7dd65be Klaus Aehlig
      result.Raise("Could not shutdown instance %s on node %s" %
1666 d0d7d7cf Thomas Thrainer
                   (self.instance.name,
1667 d0d7d7cf Thomas Thrainer
                    self.cfg.GetNodeName(self.instance.primary_node)))
1668 22b7f6f8 Thomas Thrainer
1669 22b7f6f8 Thomas Thrainer
    assert (self.owned_locks(locking.LEVEL_NODE) ==
1670 22b7f6f8 Thomas Thrainer
            self.owned_locks(locking.LEVEL_NODE_RES))
1671 d0d7d7cf Thomas Thrainer
    assert not (set(self.instance.all_nodes) -
1672 22b7f6f8 Thomas Thrainer
                self.owned_locks(locking.LEVEL_NODE)), \
1673 22b7f6f8 Thomas Thrainer
      "Not owning correct locks"
1674 22b7f6f8 Thomas Thrainer
1675 d0d7d7cf Thomas Thrainer
    RemoveInstance(self, feedback_fn, self.instance, self.op.ignore_failures)
1676 22b7f6f8 Thomas Thrainer
1677 22b7f6f8 Thomas Thrainer
1678 22b7f6f8 Thomas Thrainer
class LUInstanceMove(LogicalUnit):
1679 22b7f6f8 Thomas Thrainer
  """Move an instance by data-copying.
1680 22b7f6f8 Thomas Thrainer

1681 22b7f6f8 Thomas Thrainer
  """
1682 22b7f6f8 Thomas Thrainer
  HPATH = "instance-move"
1683 22b7f6f8 Thomas Thrainer
  HTYPE = constants.HTYPE_INSTANCE
1684 22b7f6f8 Thomas Thrainer
  REQ_BGL = False
1685 22b7f6f8 Thomas Thrainer
1686 22b7f6f8 Thomas Thrainer
  def ExpandNames(self):
1687 22b7f6f8 Thomas Thrainer
    self._ExpandAndLockInstance()
1688 1c3231aa Thomas Thrainer
    (self.op.target_node_uuid, self.op.target_node) = \
1689 1c3231aa Thomas Thrainer
      ExpandNodeUuidAndName(self.cfg, self.op.target_node_uuid,
1690 1c3231aa Thomas Thrainer
                            self.op.target_node)
1691 b9aae98b Thomas Thrainer
    self.needed_locks[locking.LEVEL_NODE] = [self.op.target_node_uuid]
1692 22b7f6f8 Thomas Thrainer
    self.needed_locks[locking.LEVEL_NODE_RES] = []
1693 22b7f6f8 Thomas Thrainer
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_APPEND
1694 22b7f6f8 Thomas Thrainer
1695 22b7f6f8 Thomas Thrainer
  def DeclareLocks(self, level):
1696 22b7f6f8 Thomas Thrainer
    if level == locking.LEVEL_NODE:
1697 22b7f6f8 Thomas Thrainer
      self._LockInstancesNodes(primary_only=True)
1698 22b7f6f8 Thomas Thrainer
    elif level == locking.LEVEL_NODE_RES:
1699 22b7f6f8 Thomas Thrainer
      # Copy node locks
1700 22b7f6f8 Thomas Thrainer
      self.needed_locks[locking.LEVEL_NODE_RES] = \
1701 5eacbcae Thomas Thrainer
        CopyLockList(self.needed_locks[locking.LEVEL_NODE])
1702 22b7f6f8 Thomas Thrainer
1703 22b7f6f8 Thomas Thrainer
  def BuildHooksEnv(self):
1704 22b7f6f8 Thomas Thrainer
    """Build hooks env.
1705 22b7f6f8 Thomas Thrainer

1706 83266db6 Thomas Thrainer
    This runs on master, primary and target nodes of the instance.
1707 22b7f6f8 Thomas Thrainer

1708 22b7f6f8 Thomas Thrainer
    """
1709 22b7f6f8 Thomas Thrainer
    env = {
1710 22b7f6f8 Thomas Thrainer
      "TARGET_NODE": self.op.target_node,
1711 22b7f6f8 Thomas Thrainer
      "SHUTDOWN_TIMEOUT": self.op.shutdown_timeout,
1712 22b7f6f8 Thomas Thrainer
      }
1713 5eacbcae Thomas Thrainer
    env.update(BuildInstanceHookEnvByObject(self, self.instance))
1714 22b7f6f8 Thomas Thrainer
    return env
1715 22b7f6f8 Thomas Thrainer
1716 22b7f6f8 Thomas Thrainer
  def BuildHooksNodes(self):
1717 22b7f6f8 Thomas Thrainer
    """Build hooks nodes.
1718 22b7f6f8 Thomas Thrainer

1719 22b7f6f8 Thomas Thrainer
    """
1720 22b7f6f8 Thomas Thrainer
    nl = [
1721 22b7f6f8 Thomas Thrainer
      self.cfg.GetMasterNode(),
1722 22b7f6f8 Thomas Thrainer
      self.instance.primary_node,
1723 1c3231aa Thomas Thrainer
      self.op.target_node_uuid,
1724 22b7f6f8 Thomas Thrainer
      ]
1725 22b7f6f8 Thomas Thrainer
    return (nl, nl)
1726 22b7f6f8 Thomas Thrainer
1727 22b7f6f8 Thomas Thrainer
  def CheckPrereq(self):
1728 22b7f6f8 Thomas Thrainer
    """Check prerequisites.
1729 22b7f6f8 Thomas Thrainer

1730 22b7f6f8 Thomas Thrainer
    This checks that the instance is in the cluster.
1731 22b7f6f8 Thomas Thrainer

1732 22b7f6f8 Thomas Thrainer
    """
1733 da4a52a3 Thomas Thrainer
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_uuid)
1734 22b7f6f8 Thomas Thrainer
    assert self.instance is not None, \
1735 22b7f6f8 Thomas Thrainer
      "Cannot retrieve locked instance %s" % self.op.instance_name
1736 22b7f6f8 Thomas Thrainer
1737 d0d7d7cf Thomas Thrainer
    if self.instance.disk_template not in constants.DTS_COPYABLE:
1738 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Disk template %s not suitable for copying" %
1739 d0d7d7cf Thomas Thrainer
                                 self.instance.disk_template,
1740 d0d7d7cf Thomas Thrainer
                                 errors.ECODE_STATE)
1741 22b7f6f8 Thomas Thrainer
1742 1c3231aa Thomas Thrainer
    target_node = self.cfg.GetNodeInfo(self.op.target_node_uuid)
1743 1c3231aa Thomas Thrainer
    assert target_node is not None, \
1744 22b7f6f8 Thomas Thrainer
      "Cannot retrieve locked node %s" % self.op.target_node
1745 22b7f6f8 Thomas Thrainer
1746 1c3231aa Thomas Thrainer
    self.target_node_uuid = target_node.uuid
1747 d0d7d7cf Thomas Thrainer
    if target_node.uuid == self.instance.primary_node:
1748 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Instance %s is already on the node %s" %
1749 d0d7d7cf Thomas Thrainer
                                 (self.instance.name, target_node.name),
1750 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_STATE)
1751 22b7f6f8 Thomas Thrainer
1752 d29f52a6 Thomas Thrainer
    cluster = self.cfg.GetClusterInfo()
1753 d29f52a6 Thomas Thrainer
    bep = cluster.FillBE(self.instance)
1754 22b7f6f8 Thomas Thrainer
1755 d0d7d7cf Thomas Thrainer
    for idx, dsk in enumerate(self.instance.disks):
1756 cd3b4ff4 Helga Velroyen
      if dsk.dev_type not in (constants.DT_PLAIN, constants.DT_FILE,
1757 cd3b4ff4 Helga Velroyen
                              constants.DT_SHARED_FILE):
1758 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Instance disk %d has a complex layout,"
1759 22b7f6f8 Thomas Thrainer
                                   " cannot copy" % idx, errors.ECODE_STATE)
1760 22b7f6f8 Thomas Thrainer
1761 1c3231aa Thomas Thrainer
    CheckNodeOnline(self, target_node.uuid)
1762 1c3231aa Thomas Thrainer
    CheckNodeNotDrained(self, target_node.uuid)
1763 1c3231aa Thomas Thrainer
    CheckNodeVmCapable(self, target_node.uuid)
1764 1c3231aa Thomas Thrainer
    group_info = self.cfg.GetNodeGroup(target_node.group)
1765 22b7f6f8 Thomas Thrainer
    ipolicy = ganeti.masterd.instance.CalculateGroupIPolicy(cluster, group_info)
1766 d0d7d7cf Thomas Thrainer
    CheckTargetNodeIPolicy(self, ipolicy, self.instance, target_node, self.cfg,
1767 5eacbcae Thomas Thrainer
                           ignore=self.op.ignore_ipolicy)
1768 22b7f6f8 Thomas Thrainer
1769 d0d7d7cf Thomas Thrainer
    if self.instance.admin_state == constants.ADMINST_UP:
1770 83266db6 Thomas Thrainer
      # check memory requirements on the target node
1771 a295eb80 Helga Velroyen
      CheckNodeFreeMemory(
1772 1c3231aa Thomas Thrainer
          self, target_node.uuid, "failing over instance %s" %
1773 d0d7d7cf Thomas Thrainer
          self.instance.name, bep[constants.BE_MAXMEM],
1774 d0d7d7cf Thomas Thrainer
          self.instance.hypervisor,
1775 d29f52a6 Thomas Thrainer
          cluster.hvparams[self.instance.hypervisor])
1776 22b7f6f8 Thomas Thrainer
    else:
1777 22b7f6f8 Thomas Thrainer
      self.LogInfo("Not checking memory on the secondary node as"
1778 22b7f6f8 Thomas Thrainer
                   " instance will not be started")
1779 22b7f6f8 Thomas Thrainer
1780 22b7f6f8 Thomas Thrainer
    # check bridge existance
1781 d0d7d7cf Thomas Thrainer
    CheckInstanceBridgesExist(self, self.instance, node_uuid=target_node.uuid)
1782 22b7f6f8 Thomas Thrainer
1783 22b7f6f8 Thomas Thrainer
  def Exec(self, feedback_fn):
1784 22b7f6f8 Thomas Thrainer
    """Move an instance.
1785 22b7f6f8 Thomas Thrainer

1786 22b7f6f8 Thomas Thrainer
    The move is done by shutting it down on its present node, copying
1787 22b7f6f8 Thomas Thrainer
    the data over (slow) and starting it on the new node.
1788 22b7f6f8 Thomas Thrainer

1789 22b7f6f8 Thomas Thrainer
    """
1790 d0d7d7cf Thomas Thrainer
    source_node = self.cfg.GetNodeInfo(self.instance.primary_node)
1791 1c3231aa Thomas Thrainer
    target_node = self.cfg.GetNodeInfo(self.target_node_uuid)
1792 22b7f6f8 Thomas Thrainer
1793 22b7f6f8 Thomas Thrainer
    self.LogInfo("Shutting down instance %s on source node %s",
1794 d0d7d7cf Thomas Thrainer
                 self.instance.name, source_node.name)
1795 22b7f6f8 Thomas Thrainer
1796 22b7f6f8 Thomas Thrainer
    assert (self.owned_locks(locking.LEVEL_NODE) ==
1797 22b7f6f8 Thomas Thrainer
            self.owned_locks(locking.LEVEL_NODE_RES))
1798 22b7f6f8 Thomas Thrainer
1799 d0d7d7cf Thomas Thrainer
    result = self.rpc.call_instance_shutdown(source_node.uuid, self.instance,
1800 22b7f6f8 Thomas Thrainer
                                             self.op.shutdown_timeout,
1801 22b7f6f8 Thomas Thrainer
                                             self.op.reason)
1802 c7dd65be Klaus Aehlig
    if self.op.ignore_consistency:
1803 c7dd65be Klaus Aehlig
      result.Warn("Could not shutdown instance %s on node %s. Proceeding"
1804 c7dd65be Klaus Aehlig
                  " anyway. Please make sure node %s is down. Error details" %
1805 d0d7d7cf Thomas Thrainer
                  (self.instance.name, source_node.name, source_node.name),
1806 1c3231aa Thomas Thrainer
                  self.LogWarning)
1807 c7dd65be Klaus Aehlig
    else:
1808 c7dd65be Klaus Aehlig
      result.Raise("Could not shutdown instance %s on node %s" %
1809 d0d7d7cf Thomas Thrainer
                   (self.instance.name, source_node.name))
1810 22b7f6f8 Thomas Thrainer
1811 22b7f6f8 Thomas Thrainer
    # create the target disks
1812 22b7f6f8 Thomas Thrainer
    try:
1813 d0d7d7cf Thomas Thrainer
      CreateDisks(self, self.instance, target_node_uuid=target_node.uuid)
1814 22b7f6f8 Thomas Thrainer
    except errors.OpExecError:
1815 22b7f6f8 Thomas Thrainer
      self.LogWarning("Device creation failed")
1816 da4a52a3 Thomas Thrainer
      self.cfg.ReleaseDRBDMinors(self.instance.uuid)
1817 22b7f6f8 Thomas Thrainer
      raise
1818 22b7f6f8 Thomas Thrainer
1819 22b7f6f8 Thomas Thrainer
    errs = []
1820 063613aa Thomas Thrainer
    transfers = []
1821 063613aa Thomas Thrainer
    # activate, get path, create transfer jobs
1822 d0d7d7cf Thomas Thrainer
    for idx, disk in enumerate(self.instance.disks):
1823 063613aa Thomas Thrainer
      # FIXME: pass debug option from opcode to backend
1824 063613aa Thomas Thrainer
      dt = masterd.instance.DiskTransfer("disk/%s" % idx,
1825 063613aa Thomas Thrainer
                                         constants.IEIO_RAW_DISK,
1826 063613aa Thomas Thrainer
                                         (disk, self.instance),
1827 063613aa Thomas Thrainer
                                         constants.IEIO_RAW_DISK,
1828 063613aa Thomas Thrainer
                                         (disk, self.instance),
1829 063613aa Thomas Thrainer
                                         None)
1830 063613aa Thomas Thrainer
      transfers.append(dt)
1831 063613aa Thomas Thrainer
1832 063613aa Thomas Thrainer
    import_result = \
1833 063613aa Thomas Thrainer
      masterd.instance.TransferInstanceData(self, feedback_fn,
1834 063613aa Thomas Thrainer
                                            source_node.uuid,
1835 063613aa Thomas Thrainer
                                            target_node.uuid,
1836 063613aa Thomas Thrainer
                                            target_node.secondary_ip,
1837 f198cf91 Thomas Thrainer
                                            self.op.compress,
1838 063613aa Thomas Thrainer
                                            self.instance, transfers)
1839 063613aa Thomas Thrainer
    if not compat.all(import_result):
1840 063613aa Thomas Thrainer
      errs.append("Failed to transfer instance data")
1841 22b7f6f8 Thomas Thrainer
1842 22b7f6f8 Thomas Thrainer
    if errs:
1843 22b7f6f8 Thomas Thrainer
      self.LogWarning("Some disks failed to copy, aborting")
1844 22b7f6f8 Thomas Thrainer
      try:
1845 d0d7d7cf Thomas Thrainer
        RemoveDisks(self, self.instance, target_node_uuid=target_node.uuid)
1846 22b7f6f8 Thomas Thrainer
      finally:
1847 da4a52a3 Thomas Thrainer
        self.cfg.ReleaseDRBDMinors(self.instance.uuid)
1848 22b7f6f8 Thomas Thrainer
        raise errors.OpExecError("Errors during disk copy: %s" %
1849 22b7f6f8 Thomas Thrainer
                                 (",".join(errs),))
1850 22b7f6f8 Thomas Thrainer
1851 d0d7d7cf Thomas Thrainer
    self.instance.primary_node = target_node.uuid
1852 d0d7d7cf Thomas Thrainer
    self.cfg.Update(self.instance, feedback_fn)
1853 22b7f6f8 Thomas Thrainer
1854 22b7f6f8 Thomas Thrainer
    self.LogInfo("Removing the disks on the original node")
1855 d0d7d7cf Thomas Thrainer
    RemoveDisks(self, self.instance, target_node_uuid=source_node.uuid)
1856 22b7f6f8 Thomas Thrainer
1857 22b7f6f8 Thomas Thrainer
    # Only start the instance if it's marked as up
1858 d0d7d7cf Thomas Thrainer
    if self.instance.admin_state == constants.ADMINST_UP:
1859 22b7f6f8 Thomas Thrainer
      self.LogInfo("Starting instance %s on node %s",
1860 d0d7d7cf Thomas Thrainer
                   self.instance.name, target_node.name)
1861 22b7f6f8 Thomas Thrainer
1862 d0d7d7cf Thomas Thrainer
      disks_ok, _ = AssembleInstanceDisks(self, self.instance,
1863 5eacbcae Thomas Thrainer
                                          ignore_secondaries=True)
1864 22b7f6f8 Thomas Thrainer
      if not disks_ok:
1865 d0d7d7cf Thomas Thrainer
        ShutdownInstanceDisks(self, self.instance)
1866 22b7f6f8 Thomas Thrainer
        raise errors.OpExecError("Can't activate the instance's disks")
1867 22b7f6f8 Thomas Thrainer
1868 1c3231aa Thomas Thrainer
      result = self.rpc.call_instance_start(target_node.uuid,
1869 d0d7d7cf Thomas Thrainer
                                            (self.instance, None, None), False,
1870 22b7f6f8 Thomas Thrainer
                                            self.op.reason)
1871 22b7f6f8 Thomas Thrainer
      msg = result.fail_msg
1872 22b7f6f8 Thomas Thrainer
      if msg:
1873 d0d7d7cf Thomas Thrainer
        ShutdownInstanceDisks(self, self.instance)
1874 22b7f6f8 Thomas Thrainer
        raise errors.OpExecError("Could not start instance %s on node %s: %s" %
1875 d0d7d7cf Thomas Thrainer
                                 (self.instance.name, target_node.name, msg))
1876 22b7f6f8 Thomas Thrainer
1877 22b7f6f8 Thomas Thrainer
1878 22b7f6f8 Thomas Thrainer
class LUInstanceMultiAlloc(NoHooksLU):
1879 22b7f6f8 Thomas Thrainer
  """Allocates multiple instances at the same time.
1880 22b7f6f8 Thomas Thrainer

1881 22b7f6f8 Thomas Thrainer
  """
1882 22b7f6f8 Thomas Thrainer
  REQ_BGL = False
1883 22b7f6f8 Thomas Thrainer
1884 22b7f6f8 Thomas Thrainer
  def CheckArguments(self):
1885 22b7f6f8 Thomas Thrainer
    """Check arguments.
1886 22b7f6f8 Thomas Thrainer

1887 22b7f6f8 Thomas Thrainer
    """
1888 22b7f6f8 Thomas Thrainer
    nodes = []
1889 22b7f6f8 Thomas Thrainer
    for inst in self.op.instances:
1890 22b7f6f8 Thomas Thrainer
      if inst.iallocator is not None:
1891 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("iallocator are not allowed to be set on"
1892 22b7f6f8 Thomas Thrainer
                                   " instance objects", errors.ECODE_INVAL)
1893 22b7f6f8 Thomas Thrainer
      nodes.append(bool(inst.pnode))
1894 22b7f6f8 Thomas Thrainer
      if inst.disk_template in constants.DTS_INT_MIRROR:
1895 22b7f6f8 Thomas Thrainer
        nodes.append(bool(inst.snode))
1896 22b7f6f8 Thomas Thrainer
1897 22b7f6f8 Thomas Thrainer
    has_nodes = compat.any(nodes)
1898 22b7f6f8 Thomas Thrainer
    if compat.all(nodes) ^ has_nodes:
1899 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("There are instance objects providing"
1900 22b7f6f8 Thomas Thrainer
                                 " pnode/snode while others do not",
1901 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_INVAL)
1902 22b7f6f8 Thomas Thrainer
1903 0c072225 Thomas Thrainer
    if not has_nodes and self.op.iallocator is None:
1904 22b7f6f8 Thomas Thrainer
      default_iallocator = self.cfg.GetDefaultIAllocator()
1905 0c072225 Thomas Thrainer
      if default_iallocator:
1906 22b7f6f8 Thomas Thrainer
        self.op.iallocator = default_iallocator
1907 22b7f6f8 Thomas Thrainer
      else:
1908 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("No iallocator or nodes on the instances"
1909 22b7f6f8 Thomas Thrainer
                                   " given and no cluster-wide default"
1910 22b7f6f8 Thomas Thrainer
                                   " iallocator found; please specify either"
1911 22b7f6f8 Thomas Thrainer
                                   " an iallocator or nodes on the instances"
1912 22b7f6f8 Thomas Thrainer
                                   " or set a cluster-wide default iallocator",
1913 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
1914 22b7f6f8 Thomas Thrainer
1915 22b7f6f8 Thomas Thrainer
    _CheckOpportunisticLocking(self.op)
1916 22b7f6f8 Thomas Thrainer
1917 22b7f6f8 Thomas Thrainer
    dups = utils.FindDuplicates([op.instance_name for op in self.op.instances])
1918 22b7f6f8 Thomas Thrainer
    if dups:
1919 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("There are duplicate instance names: %s" %
1920 22b7f6f8 Thomas Thrainer
                                 utils.CommaJoin(dups), errors.ECODE_INVAL)
1921 22b7f6f8 Thomas Thrainer
1922 22b7f6f8 Thomas Thrainer
  def ExpandNames(self):
1923 22b7f6f8 Thomas Thrainer
    """Calculate the locks.
1924 22b7f6f8 Thomas Thrainer

1925 22b7f6f8 Thomas Thrainer
    """
1926 5eacbcae Thomas Thrainer
    self.share_locks = ShareAll()
1927 22b7f6f8 Thomas Thrainer
    self.needed_locks = {
1928 22b7f6f8 Thomas Thrainer
      # iallocator will select nodes and even if no iallocator is used,
1929 22b7f6f8 Thomas Thrainer
      # collisions with LUInstanceCreate should be avoided
1930 22b7f6f8 Thomas Thrainer
      locking.LEVEL_NODE_ALLOC: locking.ALL_SET,
1931 22b7f6f8 Thomas Thrainer
      }
1932 22b7f6f8 Thomas Thrainer
1933 22b7f6f8 Thomas Thrainer
    if self.op.iallocator:
1934 22b7f6f8 Thomas Thrainer
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
1935 22b7f6f8 Thomas Thrainer
      self.needed_locks[locking.LEVEL_NODE_RES] = locking.ALL_SET
1936 22b7f6f8 Thomas Thrainer
1937 22b7f6f8 Thomas Thrainer
      if self.op.opportunistic_locking:
1938 22b7f6f8 Thomas Thrainer
        self.opportunistic_locks[locking.LEVEL_NODE] = True
1939 22b7f6f8 Thomas Thrainer
        self.opportunistic_locks[locking.LEVEL_NODE_RES] = True
1940 22b7f6f8 Thomas Thrainer
    else:
1941 22b7f6f8 Thomas Thrainer
      nodeslist = []
1942 22b7f6f8 Thomas Thrainer
      for inst in self.op.instances:
1943 1c3231aa Thomas Thrainer
        (inst.pnode_uuid, inst.pnode) = \
1944 1c3231aa Thomas Thrainer
          ExpandNodeUuidAndName(self.cfg, inst.pnode_uuid, inst.pnode)
1945 6869f673 Thomas Thrainer
        nodeslist.append(inst.pnode_uuid)
1946 22b7f6f8 Thomas Thrainer
        if inst.snode is not None:
1947 1c3231aa Thomas Thrainer
          (inst.snode_uuid, inst.snode) = \
1948 1c3231aa Thomas Thrainer
            ExpandNodeUuidAndName(self.cfg, inst.snode_uuid, inst.snode)
1949 6869f673 Thomas Thrainer
          nodeslist.append(inst.snode_uuid)
1950 22b7f6f8 Thomas Thrainer
1951 22b7f6f8 Thomas Thrainer
      self.needed_locks[locking.LEVEL_NODE] = nodeslist
1952 22b7f6f8 Thomas Thrainer
      # Lock resources of instance's primary and secondary nodes (copy to
1953 22b7f6f8 Thomas Thrainer
      # prevent accidential modification)
1954 22b7f6f8 Thomas Thrainer
      self.needed_locks[locking.LEVEL_NODE_RES] = list(nodeslist)
1955 22b7f6f8 Thomas Thrainer
1956 22b7f6f8 Thomas Thrainer
  def CheckPrereq(self):
1957 22b7f6f8 Thomas Thrainer
    """Check prerequisite.
1958 22b7f6f8 Thomas Thrainer

1959 22b7f6f8 Thomas Thrainer
    """
1960 0c072225 Thomas Thrainer
    if self.op.iallocator:
1961 0c072225 Thomas Thrainer
      cluster = self.cfg.GetClusterInfo()
1962 0c072225 Thomas Thrainer
      default_vg = self.cfg.GetVGName()
1963 0c072225 Thomas Thrainer
      ec_id = self.proc.GetECId()
1964 22b7f6f8 Thomas Thrainer
1965 0c072225 Thomas Thrainer
      if self.op.opportunistic_locking:
1966 0c072225 Thomas Thrainer
        # Only consider nodes for which a lock is held
1967 804d72eb Thomas Thrainer
        node_whitelist = self.cfg.GetNodeNames(
1968 804d72eb Thomas Thrainer
                           list(self.owned_locks(locking.LEVEL_NODE)))
1969 0c072225 Thomas Thrainer
      else:
1970 0c072225 Thomas Thrainer
        node_whitelist = None
1971 22b7f6f8 Thomas Thrainer
1972 0c072225 Thomas Thrainer
      insts = [_CreateInstanceAllocRequest(op, ComputeDisks(op, default_vg),
1973 0c072225 Thomas Thrainer
                                           _ComputeNics(op, cluster, None,
1974 0c072225 Thomas Thrainer
                                                        self.cfg, ec_id),
1975 0c072225 Thomas Thrainer
                                           _ComputeFullBeParams(op, cluster),
1976 0c072225 Thomas Thrainer
                                           node_whitelist)
1977 0c072225 Thomas Thrainer
               for op in self.op.instances]
1978 22b7f6f8 Thomas Thrainer
1979 0c072225 Thomas Thrainer
      req = iallocator.IAReqMultiInstanceAlloc(instances=insts)
1980 0c072225 Thomas Thrainer
      ial = iallocator.IAllocator(self.cfg, self.rpc, req)
1981 22b7f6f8 Thomas Thrainer
1982 0c072225 Thomas Thrainer
      ial.Run(self.op.iallocator)
1983 22b7f6f8 Thomas Thrainer
1984 0c072225 Thomas Thrainer
      if not ial.success:
1985 0c072225 Thomas Thrainer
        raise errors.OpPrereqError("Can't compute nodes using"
1986 0c072225 Thomas Thrainer
                                   " iallocator '%s': %s" %
1987 0c072225 Thomas Thrainer
                                   (self.op.iallocator, ial.info),
1988 0c072225 Thomas Thrainer
                                   errors.ECODE_NORES)
1989 22b7f6f8 Thomas Thrainer
1990 0c072225 Thomas Thrainer
      self.ia_result = ial.result
1991 22b7f6f8 Thomas Thrainer
1992 22b7f6f8 Thomas Thrainer
    if self.op.dry_run:
1993 22b7f6f8 Thomas Thrainer
      self.dry_run_result = objects.FillDict(self._ConstructPartialResult(), {
1994 22b7f6f8 Thomas Thrainer
        constants.JOB_IDS_KEY: [],
1995 22b7f6f8 Thomas Thrainer
        })
1996 22b7f6f8 Thomas Thrainer
1997 22b7f6f8 Thomas Thrainer
  def _ConstructPartialResult(self):
1998 22b7f6f8 Thomas Thrainer
    """Contructs the partial result.
1999 22b7f6f8 Thomas Thrainer

2000 22b7f6f8 Thomas Thrainer
    """
2001 0c072225 Thomas Thrainer
    if self.op.iallocator:
2002 0c072225 Thomas Thrainer
      (allocatable, failed_insts) = self.ia_result
2003 0c072225 Thomas Thrainer
      allocatable_insts = map(compat.fst, allocatable)
2004 0c072225 Thomas Thrainer
    else:
2005 0c072225 Thomas Thrainer
      allocatable_insts = [op.instance_name for op in self.op.instances]
2006 0c072225 Thomas Thrainer
      failed_insts = []
2007 0c072225 Thomas Thrainer
2008 22b7f6f8 Thomas Thrainer
    return {
2009 1ca326c8 Thomas Thrainer
      constants.ALLOCATABLE_KEY: allocatable_insts,
2010 1ca326c8 Thomas Thrainer
      constants.FAILED_KEY: failed_insts,
2011 22b7f6f8 Thomas Thrainer
      }
2012 22b7f6f8 Thomas Thrainer
2013 22b7f6f8 Thomas Thrainer
  def Exec(self, feedback_fn):
2014 22b7f6f8 Thomas Thrainer
    """Executes the opcode.
2015 22b7f6f8 Thomas Thrainer

2016 22b7f6f8 Thomas Thrainer
    """
2017 22b7f6f8 Thomas Thrainer
    jobs = []
2018 0c072225 Thomas Thrainer
    if self.op.iallocator:
2019 0c072225 Thomas Thrainer
      op2inst = dict((op.instance_name, op) for op in self.op.instances)
2020 0c072225 Thomas Thrainer
      (allocatable, failed) = self.ia_result
2021 22b7f6f8 Thomas Thrainer
2022 804d72eb Thomas Thrainer
      for (name, node_names) in allocatable:
2023 0c072225 Thomas Thrainer
        op = op2inst.pop(name)
2024 22b7f6f8 Thomas Thrainer
2025 804d72eb Thomas Thrainer
        (op.pnode_uuid, op.pnode) = \
2026 804d72eb Thomas Thrainer
          ExpandNodeUuidAndName(self.cfg, None, node_names[0])
2027 804d72eb Thomas Thrainer
        if len(node_names) > 1:
2028 804d72eb Thomas Thrainer
          (op.snode_uuid, op.snode) = \
2029 804d72eb Thomas Thrainer
            ExpandNodeUuidAndName(self.cfg, None, node_names[1])
2030 22b7f6f8 Thomas Thrainer
2031 804d72eb Thomas Thrainer
          jobs.append([op])
2032 22b7f6f8 Thomas Thrainer
2033 804d72eb Thomas Thrainer
        missing = set(op2inst.keys()) - set(failed)
2034 804d72eb Thomas Thrainer
        assert not missing, \
2035 804d72eb Thomas Thrainer
          "Iallocator did return incomplete result: %s" % \
2036 804d72eb Thomas Thrainer
          utils.CommaJoin(missing)
2037 0c072225 Thomas Thrainer
    else:
2038 0c072225 Thomas Thrainer
      jobs.extend([op] for op in self.op.instances)
2039 22b7f6f8 Thomas Thrainer
2040 22b7f6f8 Thomas Thrainer
    return ResultWithJobs(jobs, **self._ConstructPartialResult())
2041 22b7f6f8 Thomas Thrainer
2042 22b7f6f8 Thomas Thrainer
2043 22b7f6f8 Thomas Thrainer
class _InstNicModPrivate:
2044 22b7f6f8 Thomas Thrainer
  """Data structure for network interface modifications.
2045 22b7f6f8 Thomas Thrainer

2046 22b7f6f8 Thomas Thrainer
  Used by L{LUInstanceSetParams}.
2047 22b7f6f8 Thomas Thrainer

2048 22b7f6f8 Thomas Thrainer
  """
2049 22b7f6f8 Thomas Thrainer
  def __init__(self):
2050 22b7f6f8 Thomas Thrainer
    self.params = None
2051 22b7f6f8 Thomas Thrainer
    self.filled = None
2052 22b7f6f8 Thomas Thrainer
2053 22b7f6f8 Thomas Thrainer
2054 5eacbcae Thomas Thrainer
def _PrepareContainerMods(mods, private_fn):
2055 22b7f6f8 Thomas Thrainer
  """Prepares a list of container modifications by adding a private data field.
2056 22b7f6f8 Thomas Thrainer

2057 22b7f6f8 Thomas Thrainer
  @type mods: list of tuples; (operation, index, parameters)
2058 22b7f6f8 Thomas Thrainer
  @param mods: List of modifications
2059 22b7f6f8 Thomas Thrainer
  @type private_fn: callable or None
2060 22b7f6f8 Thomas Thrainer
  @param private_fn: Callable for constructing a private data field for a
2061 22b7f6f8 Thomas Thrainer
    modification
2062 22b7f6f8 Thomas Thrainer
  @rtype: list
2063 22b7f6f8 Thomas Thrainer

2064 22b7f6f8 Thomas Thrainer
  """
2065 22b7f6f8 Thomas Thrainer
  if private_fn is None:
2066 22b7f6f8 Thomas Thrainer
    fn = lambda: None
2067 22b7f6f8 Thomas Thrainer
  else:
2068 22b7f6f8 Thomas Thrainer
    fn = private_fn
2069 22b7f6f8 Thomas Thrainer
2070 22b7f6f8 Thomas Thrainer
  return [(op, idx, params, fn()) for (op, idx, params) in mods]
2071 22b7f6f8 Thomas Thrainer
2072 22b7f6f8 Thomas Thrainer
2073 1c3231aa Thomas Thrainer
def _CheckNodesPhysicalCPUs(lu, node_uuids, requested, hypervisor_specs):
2074 22b7f6f8 Thomas Thrainer
  """Checks if nodes have enough physical CPUs
2075 22b7f6f8 Thomas Thrainer

2076 22b7f6f8 Thomas Thrainer
  This function checks if all given nodes have the needed number of
2077 22b7f6f8 Thomas Thrainer
  physical CPUs. In case any node has less CPUs or we cannot get the
2078 22b7f6f8 Thomas Thrainer
  information from the node, this function raises an OpPrereqError
2079 22b7f6f8 Thomas Thrainer
  exception.
2080 22b7f6f8 Thomas Thrainer

2081 22b7f6f8 Thomas Thrainer
  @type lu: C{LogicalUnit}
2082 22b7f6f8 Thomas Thrainer
  @param lu: a logical unit from which we get configuration data
2083 1c3231aa Thomas Thrainer
  @type node_uuids: C{list}
2084 1c3231aa Thomas Thrainer
  @param node_uuids: the list of node UUIDs to check
2085 22b7f6f8 Thomas Thrainer
  @type requested: C{int}
2086 22b7f6f8 Thomas Thrainer
  @param requested: the minimum acceptable number of physical CPUs
2087 a295eb80 Helga Velroyen
  @type hypervisor_specs: list of pairs (string, dict of strings)
2088 a295eb80 Helga Velroyen
  @param hypervisor_specs: list of hypervisor specifications in
2089 a295eb80 Helga Velroyen
      pairs (hypervisor_name, hvparams)
2090 22b7f6f8 Thomas Thrainer
  @raise errors.OpPrereqError: if the node doesn't have enough CPUs,
2091 22b7f6f8 Thomas Thrainer
      or we cannot check the node
2092 22b7f6f8 Thomas Thrainer

2093 22b7f6f8 Thomas Thrainer
  """
2094 da803ff1 Helga Velroyen
  nodeinfo = lu.rpc.call_node_info(node_uuids, None, hypervisor_specs)
2095 1c3231aa Thomas Thrainer
  for node_uuid in node_uuids:
2096 1c3231aa Thomas Thrainer
    info = nodeinfo[node_uuid]
2097 1c3231aa Thomas Thrainer
    node_name = lu.cfg.GetNodeName(node_uuid)
2098 1c3231aa Thomas Thrainer
    info.Raise("Cannot get current information from node %s" % node_name,
2099 22b7f6f8 Thomas Thrainer
               prereq=True, ecode=errors.ECODE_ENVIRON)
2100 22b7f6f8 Thomas Thrainer
    (_, _, (hv_info, )) = info.payload
2101 22b7f6f8 Thomas Thrainer
    num_cpus = hv_info.get("cpu_total", None)
2102 22b7f6f8 Thomas Thrainer
    if not isinstance(num_cpus, int):
2103 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Can't compute the number of physical CPUs"
2104 22b7f6f8 Thomas Thrainer
                                 " on node %s, result was '%s'" %
2105 1c3231aa Thomas Thrainer
                                 (node_name, num_cpus), errors.ECODE_ENVIRON)
2106 22b7f6f8 Thomas Thrainer
    if requested > num_cpus:
2107 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Node %s has %s physical CPUs, but %s are "
2108 1c3231aa Thomas Thrainer
                                 "required" % (node_name, num_cpus, requested),
2109 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_NORES)
2110 22b7f6f8 Thomas Thrainer
2111 22b7f6f8 Thomas Thrainer
2112 22b7f6f8 Thomas Thrainer
def GetItemFromContainer(identifier, kind, container):
2113 22b7f6f8 Thomas Thrainer
  """Return the item refered by the identifier.
2114 22b7f6f8 Thomas Thrainer

2115 22b7f6f8 Thomas Thrainer
  @type identifier: string
2116 22b7f6f8 Thomas Thrainer
  @param identifier: Item index or name or UUID
2117 22b7f6f8 Thomas Thrainer
  @type kind: string
2118 22b7f6f8 Thomas Thrainer
  @param kind: One-word item description
2119 22b7f6f8 Thomas Thrainer
  @type container: list
2120 22b7f6f8 Thomas Thrainer
  @param container: Container to get the item from
2121 22b7f6f8 Thomas Thrainer

2122 22b7f6f8 Thomas Thrainer
  """
2123 22b7f6f8 Thomas Thrainer
  # Index
2124 22b7f6f8 Thomas Thrainer
  try:
2125 22b7f6f8 Thomas Thrainer
    idx = int(identifier)
2126 22b7f6f8 Thomas Thrainer
    if idx == -1:
2127 22b7f6f8 Thomas Thrainer
      # Append
2128 22b7f6f8 Thomas Thrainer
      absidx = len(container) - 1
2129 22b7f6f8 Thomas Thrainer
    elif idx < 0:
2130 22b7f6f8 Thomas Thrainer
      raise IndexError("Not accepting negative indices other than -1")
2131 22b7f6f8 Thomas Thrainer
    elif idx > len(container):
2132 22b7f6f8 Thomas Thrainer
      raise IndexError("Got %s index %s, but there are only %s" %
2133 22b7f6f8 Thomas Thrainer
                       (kind, idx, len(container)))
2134 22b7f6f8 Thomas Thrainer
    else:
2135 22b7f6f8 Thomas Thrainer
      absidx = idx
2136 22b7f6f8 Thomas Thrainer
    return (absidx, container[idx])
2137 22b7f6f8 Thomas Thrainer
  except ValueError:
2138 22b7f6f8 Thomas Thrainer
    pass
2139 22b7f6f8 Thomas Thrainer
2140 22b7f6f8 Thomas Thrainer
  for idx, item in enumerate(container):
2141 22b7f6f8 Thomas Thrainer
    if item.uuid == identifier or item.name == identifier:
2142 22b7f6f8 Thomas Thrainer
      return (idx, item)
2143 22b7f6f8 Thomas Thrainer
2144 22b7f6f8 Thomas Thrainer
  raise errors.OpPrereqError("Cannot find %s with identifier %s" %
2145 22b7f6f8 Thomas Thrainer
                             (kind, identifier), errors.ECODE_NOENT)
2146 22b7f6f8 Thomas Thrainer
2147 22b7f6f8 Thomas Thrainer
2148 5eacbcae Thomas Thrainer
def _ApplyContainerMods(kind, container, chgdesc, mods,
2149 922a9e65 Thomas Thrainer
                        create_fn, modify_fn, remove_fn,
2150 922a9e65 Thomas Thrainer
                        post_add_fn=None):
2151 22b7f6f8 Thomas Thrainer
  """Applies descriptions in C{mods} to C{container}.
2152 22b7f6f8 Thomas Thrainer

2153 22b7f6f8 Thomas Thrainer
  @type kind: string
2154 22b7f6f8 Thomas Thrainer
  @param kind: One-word item description
2155 22b7f6f8 Thomas Thrainer
  @type container: list
2156 22b7f6f8 Thomas Thrainer
  @param container: Container to modify
2157 22b7f6f8 Thomas Thrainer
  @type chgdesc: None or list
2158 22b7f6f8 Thomas Thrainer
  @param chgdesc: List of applied changes
2159 22b7f6f8 Thomas Thrainer
  @type mods: list
2160 5eacbcae Thomas Thrainer
  @param mods: Modifications as returned by L{_PrepareContainerMods}
2161 22b7f6f8 Thomas Thrainer
  @type create_fn: callable
2162 22b7f6f8 Thomas Thrainer
  @param create_fn: Callback for creating a new item (L{constants.DDM_ADD});
2163 22b7f6f8 Thomas Thrainer
    receives absolute item index, parameters and private data object as added
2164 5eacbcae Thomas Thrainer
    by L{_PrepareContainerMods}, returns tuple containing new item and changes
2165 22b7f6f8 Thomas Thrainer
    as list
2166 22b7f6f8 Thomas Thrainer
  @type modify_fn: callable
2167 22b7f6f8 Thomas Thrainer
  @param modify_fn: Callback for modifying an existing item
2168 22b7f6f8 Thomas Thrainer
    (L{constants.DDM_MODIFY}); receives absolute item index, item, parameters
2169 5eacbcae Thomas Thrainer
    and private data object as added by L{_PrepareContainerMods}, returns
2170 22b7f6f8 Thomas Thrainer
    changes as list
2171 22b7f6f8 Thomas Thrainer
  @type remove_fn: callable
2172 22b7f6f8 Thomas Thrainer
  @param remove_fn: Callback on removing item; receives absolute item index,
2173 5eacbcae Thomas Thrainer
    item and private data object as added by L{_PrepareContainerMods}
2174 922a9e65 Thomas Thrainer
  @type post_add_fn: callable
2175 922a9e65 Thomas Thrainer
  @param post_add_fn: Callable for post-processing a newly created item after
2176 922a9e65 Thomas Thrainer
    it has been put into the container. It receives the index of the new item
2177 922a9e65 Thomas Thrainer
    and the new item as parameters.
2178 22b7f6f8 Thomas Thrainer

2179 22b7f6f8 Thomas Thrainer
  """
2180 22b7f6f8 Thomas Thrainer
  for (op, identifier, params, private) in mods:
2181 22b7f6f8 Thomas Thrainer
    changes = None
2182 22b7f6f8 Thomas Thrainer
2183 22b7f6f8 Thomas Thrainer
    if op == constants.DDM_ADD:
2184 22b7f6f8 Thomas Thrainer
      # Calculate where item will be added
2185 22b7f6f8 Thomas Thrainer
      # When adding an item, identifier can only be an index
2186 22b7f6f8 Thomas Thrainer
      try:
2187 22b7f6f8 Thomas Thrainer
        idx = int(identifier)
2188 22b7f6f8 Thomas Thrainer
      except ValueError:
2189 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Only possitive integer or -1 is accepted as"
2190 22b7f6f8 Thomas Thrainer
                                   " identifier for %s" % constants.DDM_ADD,
2191 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
2192 22b7f6f8 Thomas Thrainer
      if idx == -1:
2193 22b7f6f8 Thomas Thrainer
        addidx = len(container)
2194 22b7f6f8 Thomas Thrainer
      else:
2195 22b7f6f8 Thomas Thrainer
        if idx < 0:
2196 22b7f6f8 Thomas Thrainer
          raise IndexError("Not accepting negative indices other than -1")
2197 22b7f6f8 Thomas Thrainer
        elif idx > len(container):
2198 22b7f6f8 Thomas Thrainer
          raise IndexError("Got %s index %s, but there are only %s" %
2199 22b7f6f8 Thomas Thrainer
                           (kind, idx, len(container)))
2200 22b7f6f8 Thomas Thrainer
        addidx = idx
2201 22b7f6f8 Thomas Thrainer
2202 22b7f6f8 Thomas Thrainer
      if create_fn is None:
2203 22b7f6f8 Thomas Thrainer
        item = params
2204 22b7f6f8 Thomas Thrainer
      else:
2205 22b7f6f8 Thomas Thrainer
        (item, changes) = create_fn(addidx, params, private)
2206 22b7f6f8 Thomas Thrainer
2207 22b7f6f8 Thomas Thrainer
      if idx == -1:
2208 22b7f6f8 Thomas Thrainer
        container.append(item)
2209 22b7f6f8 Thomas Thrainer
      else:
2210 22b7f6f8 Thomas Thrainer
        assert idx >= 0
2211 22b7f6f8 Thomas Thrainer
        assert idx <= len(container)
2212 22b7f6f8 Thomas Thrainer
        # list.insert does so before the specified index
2213 22b7f6f8 Thomas Thrainer
        container.insert(idx, item)
2214 922a9e65 Thomas Thrainer
2215 922a9e65 Thomas Thrainer
      if post_add_fn is not None:
2216 922a9e65 Thomas Thrainer
        post_add_fn(addidx, item)
2217 922a9e65 Thomas Thrainer
2218 22b7f6f8 Thomas Thrainer
    else:
2219 22b7f6f8 Thomas Thrainer
      # Retrieve existing item
2220 22b7f6f8 Thomas Thrainer
      (absidx, item) = GetItemFromContainer(identifier, kind, container)
2221 22b7f6f8 Thomas Thrainer
2222 22b7f6f8 Thomas Thrainer
      if op == constants.DDM_REMOVE:
2223 22b7f6f8 Thomas Thrainer
        assert not params
2224 22b7f6f8 Thomas Thrainer
2225 22b7f6f8 Thomas Thrainer
        changes = [("%s/%s" % (kind, absidx), "remove")]
2226 22b7f6f8 Thomas Thrainer
2227 e15a00dc Dimitris Aragiorgis
        if remove_fn is not None:
2228 e15a00dc Dimitris Aragiorgis
          msg = remove_fn(absidx, item, private)
2229 e15a00dc Dimitris Aragiorgis
          if msg:
2230 e15a00dc Dimitris Aragiorgis
            changes.append(("%s/%s" % (kind, absidx), msg))
2231 e15a00dc Dimitris Aragiorgis
2232 22b7f6f8 Thomas Thrainer
        assert container[absidx] == item
2233 22b7f6f8 Thomas Thrainer
        del container[absidx]
2234 22b7f6f8 Thomas Thrainer
      elif op == constants.DDM_MODIFY:
2235 22b7f6f8 Thomas Thrainer
        if modify_fn is not None:
2236 22b7f6f8 Thomas Thrainer
          changes = modify_fn(absidx, item, params, private)
2237 22b7f6f8 Thomas Thrainer
      else:
2238 22b7f6f8 Thomas Thrainer
        raise errors.ProgrammerError("Unhandled operation '%s'" % op)
2239 22b7f6f8 Thomas Thrainer
2240 22b7f6f8 Thomas Thrainer
    assert _TApplyContModsCbChanges(changes)
2241 22b7f6f8 Thomas Thrainer
2242 22b7f6f8 Thomas Thrainer
    if not (chgdesc is None or changes is None):
2243 22b7f6f8 Thomas Thrainer
      chgdesc.extend(changes)
2244 22b7f6f8 Thomas Thrainer
2245 22b7f6f8 Thomas Thrainer
2246 22b7f6f8 Thomas Thrainer
def _UpdateIvNames(base_index, disks):
2247 22b7f6f8 Thomas Thrainer
  """Updates the C{iv_name} attribute of disks.
2248 22b7f6f8 Thomas Thrainer

2249 22b7f6f8 Thomas Thrainer
  @type disks: list of L{objects.Disk}
2250 22b7f6f8 Thomas Thrainer

2251 22b7f6f8 Thomas Thrainer
  """
2252 22b7f6f8 Thomas Thrainer
  for (idx, disk) in enumerate(disks):
2253 22b7f6f8 Thomas Thrainer
    disk.iv_name = "disk/%s" % (base_index + idx, )
2254 22b7f6f8 Thomas Thrainer
2255 22b7f6f8 Thomas Thrainer
2256 22b7f6f8 Thomas Thrainer
class LUInstanceSetParams(LogicalUnit):
2257 22b7f6f8 Thomas Thrainer
  """Modifies an instances's parameters.
2258 22b7f6f8 Thomas Thrainer

2259 22b7f6f8 Thomas Thrainer
  """
2260 22b7f6f8 Thomas Thrainer
  HPATH = "instance-modify"
2261 22b7f6f8 Thomas Thrainer
  HTYPE = constants.HTYPE_INSTANCE
2262 22b7f6f8 Thomas Thrainer
  REQ_BGL = False
2263 22b7f6f8 Thomas Thrainer
2264 22b7f6f8 Thomas Thrainer
  @staticmethod
2265 22b7f6f8 Thomas Thrainer
  def _UpgradeDiskNicMods(kind, mods, verify_fn):
2266 22b7f6f8 Thomas Thrainer
    assert ht.TList(mods)
2267 22b7f6f8 Thomas Thrainer
    assert not mods or len(mods[0]) in (2, 3)
2268 22b7f6f8 Thomas Thrainer
2269 22b7f6f8 Thomas Thrainer
    if mods and len(mods[0]) == 2:
2270 22b7f6f8 Thomas Thrainer
      result = []
2271 22b7f6f8 Thomas Thrainer
2272 22b7f6f8 Thomas Thrainer
      addremove = 0
2273 22b7f6f8 Thomas Thrainer
      for op, params in mods:
2274 22b7f6f8 Thomas Thrainer
        if op in (constants.DDM_ADD, constants.DDM_REMOVE):
2275 22b7f6f8 Thomas Thrainer
          result.append((op, -1, params))
2276 22b7f6f8 Thomas Thrainer
          addremove += 1
2277 22b7f6f8 Thomas Thrainer
2278 22b7f6f8 Thomas Thrainer
          if addremove > 1:
2279 22b7f6f8 Thomas Thrainer
            raise errors.OpPrereqError("Only one %s add or remove operation is"
2280 22b7f6f8 Thomas Thrainer
                                       " supported at a time" % kind,
2281 22b7f6f8 Thomas Thrainer
                                       errors.ECODE_INVAL)
2282 22b7f6f8 Thomas Thrainer
        else:
2283 22b7f6f8 Thomas Thrainer
          result.append((constants.DDM_MODIFY, op, params))
2284 22b7f6f8 Thomas Thrainer
2285 22b7f6f8 Thomas Thrainer
      assert verify_fn(result)
2286 22b7f6f8 Thomas Thrainer
    else:
2287 22b7f6f8 Thomas Thrainer
      result = mods
2288 22b7f6f8 Thomas Thrainer
2289 22b7f6f8 Thomas Thrainer
    return result
2290 22b7f6f8 Thomas Thrainer
2291 22b7f6f8 Thomas Thrainer
  @staticmethod
2292 22b7f6f8 Thomas Thrainer
  def _CheckMods(kind, mods, key_types, item_fn):
2293 22b7f6f8 Thomas Thrainer
    """Ensures requested disk/NIC modifications are valid.
2294 22b7f6f8 Thomas Thrainer

2295 22b7f6f8 Thomas Thrainer
    """
2296 22b7f6f8 Thomas Thrainer
    for (op, _, params) in mods:
2297 22b7f6f8 Thomas Thrainer
      assert ht.TDict(params)
2298 22b7f6f8 Thomas Thrainer
2299 22b7f6f8 Thomas Thrainer
      # If 'key_types' is an empty dict, we assume we have an
2300 22b7f6f8 Thomas Thrainer
      # 'ext' template and thus do not ForceDictType
2301 22b7f6f8 Thomas Thrainer
      if key_types:
2302 22b7f6f8 Thomas Thrainer
        utils.ForceDictType(params, key_types)
2303 22b7f6f8 Thomas Thrainer
2304 22b7f6f8 Thomas Thrainer
      if op == constants.DDM_REMOVE:
2305 22b7f6f8 Thomas Thrainer
        if params:
2306 22b7f6f8 Thomas Thrainer
          raise errors.OpPrereqError("No settings should be passed when"
2307 22b7f6f8 Thomas Thrainer
                                     " removing a %s" % kind,
2308 22b7f6f8 Thomas Thrainer
                                     errors.ECODE_INVAL)
2309 22b7f6f8 Thomas Thrainer
      elif op in (constants.DDM_ADD, constants.DDM_MODIFY):
2310 22b7f6f8 Thomas Thrainer
        item_fn(op, params)
2311 22b7f6f8 Thomas Thrainer
      else:
2312 22b7f6f8 Thomas Thrainer
        raise errors.ProgrammerError("Unhandled operation '%s'" % op)
2313 22b7f6f8 Thomas Thrainer
2314 22b7f6f8 Thomas Thrainer
  @staticmethod
2315 3f3ea14c Bernardo Dal Seno
  def _VerifyDiskModification(op, params, excl_stor):
2316 22b7f6f8 Thomas Thrainer
    """Verifies a disk modification.
2317 22b7f6f8 Thomas Thrainer

2318 22b7f6f8 Thomas Thrainer
    """
2319 22b7f6f8 Thomas Thrainer
    if op == constants.DDM_ADD:
2320 22b7f6f8 Thomas Thrainer
      mode = params.setdefault(constants.IDISK_MODE, constants.DISK_RDWR)
2321 22b7f6f8 Thomas Thrainer
      if mode not in constants.DISK_ACCESS_SET:
2322 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Invalid disk access mode '%s'" % mode,
2323 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
2324 22b7f6f8 Thomas Thrainer
2325 22b7f6f8 Thomas Thrainer
      size = params.get(constants.IDISK_SIZE, None)
2326 22b7f6f8 Thomas Thrainer
      if size is None:
2327 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Required disk parameter '%s' missing" %
2328 22b7f6f8 Thomas Thrainer
                                   constants.IDISK_SIZE, errors.ECODE_INVAL)
2329 34956ece Thomas Thrainer
      size = int(size)
2330 22b7f6f8 Thomas Thrainer
2331 22b7f6f8 Thomas Thrainer
      params[constants.IDISK_SIZE] = size
2332 22b7f6f8 Thomas Thrainer
      name = params.get(constants.IDISK_NAME, None)
2333 22b7f6f8 Thomas Thrainer
      if name is not None and name.lower() == constants.VALUE_NONE:
2334 22b7f6f8 Thomas Thrainer
        params[constants.IDISK_NAME] = None
2335 22b7f6f8 Thomas Thrainer
2336 7c848a6a Bernardo Dal Seno
      CheckSpindlesExclusiveStorage(params, excl_stor, True)
2337 3f3ea14c Bernardo Dal Seno
2338 22b7f6f8 Thomas Thrainer
    elif op == constants.DDM_MODIFY:
2339 22b7f6f8 Thomas Thrainer
      if constants.IDISK_SIZE in params:
2340 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Disk size change not possible, use"
2341 22b7f6f8 Thomas Thrainer
                                   " grow-disk", errors.ECODE_INVAL)
2342 22b7f6f8 Thomas Thrainer
      if len(params) > 2:
2343 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Disk modification doesn't support"
2344 22b7f6f8 Thomas Thrainer
                                   " additional arbitrary parameters",
2345 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
2346 22b7f6f8 Thomas Thrainer
      name = params.get(constants.IDISK_NAME, None)
2347 22b7f6f8 Thomas Thrainer
      if name is not None and name.lower() == constants.VALUE_NONE:
2348 22b7f6f8 Thomas Thrainer
        params[constants.IDISK_NAME] = None
2349 22b7f6f8 Thomas Thrainer
2350 22b7f6f8 Thomas Thrainer
  @staticmethod
2351 22b7f6f8 Thomas Thrainer
  def _VerifyNicModification(op, params):
2352 22b7f6f8 Thomas Thrainer
    """Verifies a network interface modification.
2353 22b7f6f8 Thomas Thrainer

2354 22b7f6f8 Thomas Thrainer
    """
2355 22b7f6f8 Thomas Thrainer
    if op in (constants.DDM_ADD, constants.DDM_MODIFY):
2356 22b7f6f8 Thomas Thrainer
      ip = params.get(constants.INIC_IP, None)
2357 22b7f6f8 Thomas Thrainer
      name = params.get(constants.INIC_NAME, None)
2358 22b7f6f8 Thomas Thrainer
      req_net = params.get(constants.INIC_NETWORK, None)
2359 22b7f6f8 Thomas Thrainer
      link = params.get(constants.NIC_LINK, None)
2360 22b7f6f8 Thomas Thrainer
      mode = params.get(constants.NIC_MODE, None)
2361 22b7f6f8 Thomas Thrainer
      if name is not None and name.lower() == constants.VALUE_NONE:
2362 22b7f6f8 Thomas Thrainer
        params[constants.INIC_NAME] = None
2363 22b7f6f8 Thomas Thrainer
      if req_net is not None:
2364 22b7f6f8 Thomas Thrainer
        if req_net.lower() == constants.VALUE_NONE:
2365 22b7f6f8 Thomas Thrainer
          params[constants.INIC_NETWORK] = None
2366 22b7f6f8 Thomas Thrainer
          req_net = None
2367 22b7f6f8 Thomas Thrainer
        elif link is not None or mode is not None:
2368 22b7f6f8 Thomas Thrainer
          raise errors.OpPrereqError("If network is given"
2369 22b7f6f8 Thomas Thrainer
                                     " mode or link should not",
2370 22b7f6f8 Thomas Thrainer
                                     errors.ECODE_INVAL)
2371 22b7f6f8 Thomas Thrainer
2372 22b7f6f8 Thomas Thrainer
      if op == constants.DDM_ADD:
2373 22b7f6f8 Thomas Thrainer
        macaddr = params.get(constants.INIC_MAC, None)
2374 22b7f6f8 Thomas Thrainer
        if macaddr is None:
2375 22b7f6f8 Thomas Thrainer
          params[constants.INIC_MAC] = constants.VALUE_AUTO
2376 22b7f6f8 Thomas Thrainer
2377 22b7f6f8 Thomas Thrainer
      if ip is not None:
2378 22b7f6f8 Thomas Thrainer
        if ip.lower() == constants.VALUE_NONE:
2379 22b7f6f8 Thomas Thrainer
          params[constants.INIC_IP] = None
2380 22b7f6f8 Thomas Thrainer
        else:
2381 22b7f6f8 Thomas Thrainer
          if ip.lower() == constants.NIC_IP_POOL:
2382 22b7f6f8 Thomas Thrainer
            if op == constants.DDM_ADD and req_net is None:
2383 22b7f6f8 Thomas Thrainer
              raise errors.OpPrereqError("If ip=pool, parameter network"
2384 22b7f6f8 Thomas Thrainer
                                         " cannot be none",
2385 22b7f6f8 Thomas Thrainer
                                         errors.ECODE_INVAL)
2386 22b7f6f8 Thomas Thrainer
          else:
2387 22b7f6f8 Thomas Thrainer
            if not netutils.IPAddress.IsValid(ip):
2388 22b7f6f8 Thomas Thrainer
              raise errors.OpPrereqError("Invalid IP address '%s'" % ip,
2389 22b7f6f8 Thomas Thrainer
                                         errors.ECODE_INVAL)
2390 22b7f6f8 Thomas Thrainer
2391 22b7f6f8 Thomas Thrainer
      if constants.INIC_MAC in params:
2392 22b7f6f8 Thomas Thrainer
        macaddr = params[constants.INIC_MAC]
2393 22b7f6f8 Thomas Thrainer
        if macaddr not in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
2394 22b7f6f8 Thomas Thrainer
          macaddr = utils.NormalizeAndValidateMac(macaddr)
2395 22b7f6f8 Thomas Thrainer
2396 22b7f6f8 Thomas Thrainer
        if op == constants.DDM_MODIFY and macaddr == constants.VALUE_AUTO:
2397 22b7f6f8 Thomas Thrainer
          raise errors.OpPrereqError("'auto' is not a valid MAC address when"
2398 22b7f6f8 Thomas Thrainer
                                     " modifying an existing NIC",
2399 22b7f6f8 Thomas Thrainer
                                     errors.ECODE_INVAL)
2400 22b7f6f8 Thomas Thrainer
2401 22b7f6f8 Thomas Thrainer
  def CheckArguments(self):
2402 22b7f6f8 Thomas Thrainer
    if not (self.op.nics or self.op.disks or self.op.disk_template or
2403 22b7f6f8 Thomas Thrainer
            self.op.hvparams or self.op.beparams or self.op.os_name or
2404 5eae613c Thomas Thrainer
            self.op.osparams or self.op.offline is not None or
2405 5eae613c Thomas Thrainer
            self.op.runtime_mem or self.op.pnode):
2406 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("No changes submitted", errors.ECODE_INVAL)
2407 22b7f6f8 Thomas Thrainer
2408 22b7f6f8 Thomas Thrainer
    if self.op.hvparams:
2409 5eacbcae Thomas Thrainer
      CheckParamsNotGlobal(self.op.hvparams, constants.HVC_GLOBALS,
2410 5eacbcae Thomas Thrainer
                           "hypervisor", "instance", "cluster")
2411 22b7f6f8 Thomas Thrainer
2412 22b7f6f8 Thomas Thrainer
    self.op.disks = self._UpgradeDiskNicMods(
2413 72cd5493 Jose A. Lopes
      "disk", self.op.disks, ht.TSetParamsMods(ht.TIDiskParams))
2414 22b7f6f8 Thomas Thrainer
    self.op.nics = self._UpgradeDiskNicMods(
2415 72cd5493 Jose A. Lopes
      "NIC", self.op.nics, ht.TSetParamsMods(ht.TINicParams))
2416 22b7f6f8 Thomas Thrainer
2417 22b7f6f8 Thomas Thrainer
    if self.op.disks and self.op.disk_template is not None:
2418 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Disk template conversion and other disk"
2419 22b7f6f8 Thomas Thrainer
                                 " changes not supported at the same time",
2420 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_INVAL)
2421 22b7f6f8 Thomas Thrainer
2422 22b7f6f8 Thomas Thrainer
    if (self.op.disk_template and
2423 22b7f6f8 Thomas Thrainer
        self.op.disk_template in constants.DTS_INT_MIRROR and
2424 22b7f6f8 Thomas Thrainer
        self.op.remote_node is None):
2425 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Changing the disk template to a mirrored"
2426 22b7f6f8 Thomas Thrainer
                                 " one requires specifying a secondary node",
2427 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_INVAL)
2428 22b7f6f8 Thomas Thrainer
2429 22b7f6f8 Thomas Thrainer
    # Check NIC modifications
2430 22b7f6f8 Thomas Thrainer
    self._CheckMods("NIC", self.op.nics, constants.INIC_PARAMS_TYPES,
2431 22b7f6f8 Thomas Thrainer
                    self._VerifyNicModification)
2432 22b7f6f8 Thomas Thrainer
2433 22b7f6f8 Thomas Thrainer
    if self.op.pnode:
2434 1c3231aa Thomas Thrainer
      (self.op.pnode_uuid, self.op.pnode) = \
2435 1c3231aa Thomas Thrainer
        ExpandNodeUuidAndName(self.cfg, self.op.pnode_uuid, self.op.pnode)
2436 22b7f6f8 Thomas Thrainer
2437 22b7f6f8 Thomas Thrainer
  def ExpandNames(self):
2438 22b7f6f8 Thomas Thrainer
    self._ExpandAndLockInstance()
2439 22b7f6f8 Thomas Thrainer
    self.needed_locks[locking.LEVEL_NODEGROUP] = []
2440 22b7f6f8 Thomas Thrainer
    # Can't even acquire node locks in shared mode as upcoming changes in
2441 22b7f6f8 Thomas Thrainer
    # Ganeti 2.6 will start to modify the node object on disk conversion
2442 22b7f6f8 Thomas Thrainer
    self.needed_locks[locking.LEVEL_NODE] = []
2443 22b7f6f8 Thomas Thrainer
    self.needed_locks[locking.LEVEL_NODE_RES] = []
2444 22b7f6f8 Thomas Thrainer
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
2445 22b7f6f8 Thomas Thrainer
    # Look node group to look up the ipolicy
2446 22b7f6f8 Thomas Thrainer
    self.share_locks[locking.LEVEL_NODEGROUP] = 1
2447 22b7f6f8 Thomas Thrainer
2448 22b7f6f8 Thomas Thrainer
  def DeclareLocks(self, level):
2449 22b7f6f8 Thomas Thrainer
    if level == locking.LEVEL_NODEGROUP:
2450 22b7f6f8 Thomas Thrainer
      assert not self.needed_locks[locking.LEVEL_NODEGROUP]
2451 22b7f6f8 Thomas Thrainer
      # Acquire locks for the instance's nodegroups optimistically. Needs
2452 22b7f6f8 Thomas Thrainer
      # to be verified in CheckPrereq
2453 22b7f6f8 Thomas Thrainer
      self.needed_locks[locking.LEVEL_NODEGROUP] = \
2454 da4a52a3 Thomas Thrainer
        self.cfg.GetInstanceNodeGroups(self.op.instance_uuid)
2455 22b7f6f8 Thomas Thrainer
    elif level == locking.LEVEL_NODE:
2456 22b7f6f8 Thomas Thrainer
      self._LockInstancesNodes()
2457 22b7f6f8 Thomas Thrainer
      if self.op.disk_template and self.op.remote_node:
2458 1c3231aa Thomas Thrainer
        (self.op.remote_node_uuid, self.op.remote_node) = \
2459 1c3231aa Thomas Thrainer
          ExpandNodeUuidAndName(self.cfg, self.op.remote_node_uuid,
2460 1c3231aa Thomas Thrainer
                                self.op.remote_node)
2461 1c3231aa Thomas Thrainer
        self.needed_locks[locking.LEVEL_NODE].append(self.op.remote_node_uuid)
2462 22b7f6f8 Thomas Thrainer
    elif level == locking.LEVEL_NODE_RES and self.op.disk_template:
2463 22b7f6f8 Thomas Thrainer
      # Copy node locks
2464 22b7f6f8 Thomas Thrainer
      self.needed_locks[locking.LEVEL_NODE_RES] = \
2465 5eacbcae Thomas Thrainer
        CopyLockList(self.needed_locks[locking.LEVEL_NODE])
2466 22b7f6f8 Thomas Thrainer
2467 22b7f6f8 Thomas Thrainer
  def BuildHooksEnv(self):
2468 22b7f6f8 Thomas Thrainer
    """Build hooks env.
2469 22b7f6f8 Thomas Thrainer

2470 22b7f6f8 Thomas Thrainer
    This runs on the master, primary and secondaries.
2471 22b7f6f8 Thomas Thrainer

2472 22b7f6f8 Thomas Thrainer
    """
2473 22b7f6f8 Thomas Thrainer
    args = {}
2474 22b7f6f8 Thomas Thrainer
    if constants.BE_MINMEM in self.be_new:
2475 22b7f6f8 Thomas Thrainer
      args["minmem"] = self.be_new[constants.BE_MINMEM]
2476 22b7f6f8 Thomas Thrainer
    if constants.BE_MAXMEM in self.be_new:
2477 22b7f6f8 Thomas Thrainer
      args["maxmem"] = self.be_new[constants.BE_MAXMEM]
2478 22b7f6f8 Thomas Thrainer
    if constants.BE_VCPUS in self.be_new:
2479 22b7f6f8 Thomas Thrainer
      args["vcpus"] = self.be_new[constants.BE_VCPUS]
2480 22b7f6f8 Thomas Thrainer
    # TODO: export disk changes. Note: _BuildInstanceHookEnv* don't export disk
2481 22b7f6f8 Thomas Thrainer
    # information at all.
2482 22b7f6f8 Thomas Thrainer
2483 22b7f6f8 Thomas Thrainer
    if self._new_nics is not None:
2484 22b7f6f8 Thomas Thrainer
      nics = []
2485 22b7f6f8 Thomas Thrainer
2486 22b7f6f8 Thomas Thrainer
      for nic in self._new_nics:
2487 22b7f6f8 Thomas Thrainer
        n = copy.deepcopy(nic)
2488 22b7f6f8 Thomas Thrainer
        nicparams = self.cluster.SimpleFillNIC(n.nicparams)
2489 22b7f6f8 Thomas Thrainer
        n.nicparams = nicparams
2490 5eacbcae Thomas Thrainer
        nics.append(NICToTuple(self, n))
2491 22b7f6f8 Thomas Thrainer
2492 22b7f6f8 Thomas Thrainer
      args["nics"] = nics
2493 22b7f6f8 Thomas Thrainer
2494 5eacbcae Thomas Thrainer
    env = BuildInstanceHookEnvByObject(self, self.instance, override=args)
2495 22b7f6f8 Thomas Thrainer
    if self.op.disk_template:
2496 22b7f6f8 Thomas Thrainer
      env["NEW_DISK_TEMPLATE"] = self.op.disk_template
2497 22b7f6f8 Thomas Thrainer
    if self.op.runtime_mem:
2498 22b7f6f8 Thomas Thrainer
      env["RUNTIME_MEMORY"] = self.op.runtime_mem
2499 22b7f6f8 Thomas Thrainer
2500 22b7f6f8 Thomas Thrainer
    return env
2501 22b7f6f8 Thomas Thrainer
2502 22b7f6f8 Thomas Thrainer
  def BuildHooksNodes(self):
2503 22b7f6f8 Thomas Thrainer
    """Build hooks nodes.
2504 22b7f6f8 Thomas Thrainer

2505 22b7f6f8 Thomas Thrainer
    """
2506 22b7f6f8 Thomas Thrainer
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
2507 22b7f6f8 Thomas Thrainer
    return (nl, nl)
2508 22b7f6f8 Thomas Thrainer
2509 22b7f6f8 Thomas Thrainer
  def _PrepareNicModification(self, params, private, old_ip, old_net_uuid,
2510 1c3231aa Thomas Thrainer
                              old_params, cluster, pnode_uuid):
2511 22b7f6f8 Thomas Thrainer
2512 22b7f6f8 Thomas Thrainer
    update_params_dict = dict([(key, params[key])
2513 22b7f6f8 Thomas Thrainer
                               for key in constants.NICS_PARAMETERS
2514 22b7f6f8 Thomas Thrainer
                               if key in params])
2515 22b7f6f8 Thomas Thrainer
2516 22b7f6f8 Thomas Thrainer
    req_link = update_params_dict.get(constants.NIC_LINK, None)
2517 22b7f6f8 Thomas Thrainer
    req_mode = update_params_dict.get(constants.NIC_MODE, None)
2518 22b7f6f8 Thomas Thrainer
2519 22b7f6f8 Thomas Thrainer
    new_net_uuid = None
2520 22b7f6f8 Thomas Thrainer
    new_net_uuid_or_name = params.get(constants.INIC_NETWORK, old_net_uuid)
2521 22b7f6f8 Thomas Thrainer
    if new_net_uuid_or_name:
2522 22b7f6f8 Thomas Thrainer
      new_net_uuid = self.cfg.LookupNetwork(new_net_uuid_or_name)
2523 22b7f6f8 Thomas Thrainer
      new_net_obj = self.cfg.GetNetwork(new_net_uuid)
2524 22b7f6f8 Thomas Thrainer
2525 22b7f6f8 Thomas Thrainer
    if old_net_uuid:
2526 22b7f6f8 Thomas Thrainer
      old_net_obj = self.cfg.GetNetwork(old_net_uuid)
2527 22b7f6f8 Thomas Thrainer
2528 22b7f6f8 Thomas Thrainer
    if new_net_uuid:
2529 1c3231aa Thomas Thrainer
      netparams = self.cfg.GetGroupNetParams(new_net_uuid, pnode_uuid)
2530 22b7f6f8 Thomas Thrainer
      if not netparams:
2531 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("No netparams found for the network"
2532 22b7f6f8 Thomas Thrainer
                                   " %s, probably not connected" %
2533 22b7f6f8 Thomas Thrainer
                                   new_net_obj.name, errors.ECODE_INVAL)
2534 22b7f6f8 Thomas Thrainer
      new_params = dict(netparams)
2535 22b7f6f8 Thomas Thrainer
    else:
2536 5eacbcae Thomas Thrainer
      new_params = GetUpdatedParams(old_params, update_params_dict)
2537 22b7f6f8 Thomas Thrainer
2538 22b7f6f8 Thomas Thrainer
    utils.ForceDictType(new_params, constants.NICS_PARAMETER_TYPES)
2539 22b7f6f8 Thomas Thrainer
2540 22b7f6f8 Thomas Thrainer
    new_filled_params = cluster.SimpleFillNIC(new_params)
2541 22b7f6f8 Thomas Thrainer
    objects.NIC.CheckParameterSyntax(new_filled_params)
2542 22b7f6f8 Thomas Thrainer
2543 22b7f6f8 Thomas Thrainer
    new_mode = new_filled_params[constants.NIC_MODE]
2544 22b7f6f8 Thomas Thrainer
    if new_mode == constants.NIC_MODE_BRIDGED:
2545 22b7f6f8 Thomas Thrainer
      bridge = new_filled_params[constants.NIC_LINK]
2546 1c3231aa Thomas Thrainer
      msg = self.rpc.call_bridges_exist(pnode_uuid, [bridge]).fail_msg
2547 22b7f6f8 Thomas Thrainer
      if msg:
2548 1c3231aa Thomas Thrainer
        msg = "Error checking bridges on node '%s': %s" % \
2549 1c3231aa Thomas Thrainer
                (self.cfg.GetNodeName(pnode_uuid), msg)
2550 22b7f6f8 Thomas Thrainer
        if self.op.force:
2551 22b7f6f8 Thomas Thrainer
          self.warn.append(msg)
2552 22b7f6f8 Thomas Thrainer
        else:
2553 22b7f6f8 Thomas Thrainer
          raise errors.OpPrereqError(msg, errors.ECODE_ENVIRON)
2554 22b7f6f8 Thomas Thrainer
2555 22b7f6f8 Thomas Thrainer
    elif new_mode == constants.NIC_MODE_ROUTED:
2556 22b7f6f8 Thomas Thrainer
      ip = params.get(constants.INIC_IP, old_ip)
2557 22b7f6f8 Thomas Thrainer
      if ip is None:
2558 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Cannot set the NIC IP address to None"
2559 22b7f6f8 Thomas Thrainer
                                   " on a routed NIC", errors.ECODE_INVAL)
2560 22b7f6f8 Thomas Thrainer
2561 22b7f6f8 Thomas Thrainer
    elif new_mode == constants.NIC_MODE_OVS:
2562 22b7f6f8 Thomas Thrainer
      # TODO: check OVS link
2563 22b7f6f8 Thomas Thrainer
      self.LogInfo("OVS links are currently not checked for correctness")
2564 22b7f6f8 Thomas Thrainer
2565 22b7f6f8 Thomas Thrainer
    if constants.INIC_MAC in params:
2566 22b7f6f8 Thomas Thrainer
      mac = params[constants.INIC_MAC]
2567 22b7f6f8 Thomas Thrainer
      if mac is None:
2568 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Cannot unset the NIC MAC address",
2569 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
2570 22b7f6f8 Thomas Thrainer
      elif mac in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
2571 22b7f6f8 Thomas Thrainer
        # otherwise generate the MAC address
2572 22b7f6f8 Thomas Thrainer
        params[constants.INIC_MAC] = \
2573 22b7f6f8 Thomas Thrainer
          self.cfg.GenerateMAC(new_net_uuid, self.proc.GetECId())
2574 22b7f6f8 Thomas Thrainer
      else:
2575 22b7f6f8 Thomas Thrainer
        # or validate/reserve the current one
2576 22b7f6f8 Thomas Thrainer
        try:
2577 22b7f6f8 Thomas Thrainer
          self.cfg.ReserveMAC(mac, self.proc.GetECId())
2578 22b7f6f8 Thomas Thrainer
        except errors.ReservationError:
2579 22b7f6f8 Thomas Thrainer
          raise errors.OpPrereqError("MAC address '%s' already in use"
2580 22b7f6f8 Thomas Thrainer
                                     " in cluster" % mac,
2581 22b7f6f8 Thomas Thrainer
                                     errors.ECODE_NOTUNIQUE)
2582 22b7f6f8 Thomas Thrainer
    elif new_net_uuid != old_net_uuid:
2583 22b7f6f8 Thomas Thrainer
2584 22b7f6f8 Thomas Thrainer
      def get_net_prefix(net_uuid):
2585 22b7f6f8 Thomas Thrainer
        mac_prefix = None
2586 22b7f6f8 Thomas Thrainer
        if net_uuid:
2587 22b7f6f8 Thomas Thrainer
          nobj = self.cfg.GetNetwork(net_uuid)
2588 22b7f6f8 Thomas Thrainer
          mac_prefix = nobj.mac_prefix
2589 22b7f6f8 Thomas Thrainer
2590 22b7f6f8 Thomas Thrainer
        return mac_prefix
2591 22b7f6f8 Thomas Thrainer
2592 22b7f6f8 Thomas Thrainer
      new_prefix = get_net_prefix(new_net_uuid)
2593 22b7f6f8 Thomas Thrainer
      old_prefix = get_net_prefix(old_net_uuid)
2594 22b7f6f8 Thomas Thrainer
      if old_prefix != new_prefix:
2595 22b7f6f8 Thomas Thrainer
        params[constants.INIC_MAC] = \
2596 22b7f6f8 Thomas Thrainer
          self.cfg.GenerateMAC(new_net_uuid, self.proc.GetECId())
2597 22b7f6f8 Thomas Thrainer
2598 22b7f6f8 Thomas Thrainer
    # if there is a change in (ip, network) tuple
2599 22b7f6f8 Thomas Thrainer
    new_ip = params.get(constants.INIC_IP, old_ip)
2600 22b7f6f8 Thomas Thrainer
    if (new_ip, new_net_uuid) != (old_ip, old_net_uuid):
2601 22b7f6f8 Thomas Thrainer
      if new_ip:
2602 22b7f6f8 Thomas Thrainer
        # if IP is pool then require a network and generate one IP
2603 22b7f6f8 Thomas Thrainer
        if new_ip.lower() == constants.NIC_IP_POOL:
2604 22b7f6f8 Thomas Thrainer
          if new_net_uuid:
2605 22b7f6f8 Thomas Thrainer
            try:
2606 22b7f6f8 Thomas Thrainer
              new_ip = self.cfg.GenerateIp(new_net_uuid, self.proc.GetECId())
2607 22b7f6f8 Thomas Thrainer
            except errors.ReservationError:
2608 22b7f6f8 Thomas Thrainer
              raise errors.OpPrereqError("Unable to get a free IP"
2609 22b7f6f8 Thomas Thrainer
                                         " from the address pool",
2610 22b7f6f8 Thomas Thrainer
                                         errors.ECODE_STATE)
2611 22b7f6f8 Thomas Thrainer
            self.LogInfo("Chose IP %s from network %s",
2612 22b7f6f8 Thomas Thrainer
                         new_ip,
2613 22b7f6f8 Thomas Thrainer
                         new_net_obj.name)
2614 22b7f6f8 Thomas Thrainer
            params[constants.INIC_IP] = new_ip
2615 22b7f6f8 Thomas Thrainer
          else:
2616 22b7f6f8 Thomas Thrainer
            raise errors.OpPrereqError("ip=pool, but no network found",
2617 22b7f6f8 Thomas Thrainer
                                       errors.ECODE_INVAL)
2618 22b7f6f8 Thomas Thrainer
        # Reserve new IP if in the new network if any
2619 22b7f6f8 Thomas Thrainer
        elif new_net_uuid:
2620 22b7f6f8 Thomas Thrainer
          try:
2621 22b7f6f8 Thomas Thrainer
            self.cfg.ReserveIp(new_net_uuid, new_ip, self.proc.GetECId())
2622 22b7f6f8 Thomas Thrainer
            self.LogInfo("Reserving IP %s in network %s",
2623 22b7f6f8 Thomas Thrainer
                         new_ip, new_net_obj.name)
2624 22b7f6f8 Thomas Thrainer
          except errors.ReservationError:
2625 22b7f6f8 Thomas Thrainer
            raise errors.OpPrereqError("IP %s not available in network %s" %
2626 22b7f6f8 Thomas Thrainer
                                       (new_ip, new_net_obj.name),
2627 22b7f6f8 Thomas Thrainer
                                       errors.ECODE_NOTUNIQUE)
2628 22b7f6f8 Thomas Thrainer
        # new network is None so check if new IP is a conflicting IP
2629 22b7f6f8 Thomas Thrainer
        elif self.op.conflicts_check:
2630 1c3231aa Thomas Thrainer
          _CheckForConflictingIp(self, new_ip, pnode_uuid)
2631 22b7f6f8 Thomas Thrainer
2632 22b7f6f8 Thomas Thrainer
      # release old IP if old network is not None
2633 22b7f6f8 Thomas Thrainer
      if old_ip and old_net_uuid:
2634 22b7f6f8 Thomas Thrainer
        try:
2635 22b7f6f8 Thomas Thrainer
          self.cfg.ReleaseIp(old_net_uuid, old_ip, self.proc.GetECId())
2636 22b7f6f8 Thomas Thrainer
        except errors.AddressPoolError:
2637 22b7f6f8 Thomas Thrainer
          logging.warning("Release IP %s not contained in network %s",
2638 22b7f6f8 Thomas Thrainer
                          old_ip, old_net_obj.name)
2639 22b7f6f8 Thomas Thrainer
2640 22b7f6f8 Thomas Thrainer
    # there are no changes in (ip, network) tuple and old network is not None
2641 22b7f6f8 Thomas Thrainer
    elif (old_net_uuid is not None and
2642 22b7f6f8 Thomas Thrainer
          (req_link is not None or req_mode is not None)):
2643 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Not allowed to change link or mode of"
2644 22b7f6f8 Thomas Thrainer
                                 " a NIC that is connected to a network",
2645 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_INVAL)
2646 22b7f6f8 Thomas Thrainer
2647 22b7f6f8 Thomas Thrainer
    private.params = new_params
2648 22b7f6f8 Thomas Thrainer
    private.filled = new_filled_params
2649 22b7f6f8 Thomas Thrainer
2650 22b7f6f8 Thomas Thrainer
  def _PreCheckDiskTemplate(self, pnode_info):
2651 22b7f6f8 Thomas Thrainer
    """CheckPrereq checks related to a new disk template."""
2652 22b7f6f8 Thomas Thrainer
    # Arguments are passed to avoid configuration lookups
2653 d0d7d7cf Thomas Thrainer
    pnode_uuid = self.instance.primary_node
2654 d0d7d7cf Thomas Thrainer
    if self.instance.disk_template == self.op.disk_template:
2655 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Instance already has disk template %s" %
2656 d0d7d7cf Thomas Thrainer
                                 self.instance.disk_template,
2657 d0d7d7cf Thomas Thrainer
                                 errors.ECODE_INVAL)
2658 22b7f6f8 Thomas Thrainer
2659 7bb0c47f Thomas Thrainer
    if not self.cluster.IsDiskTemplateEnabled(self.op.disk_template):
2660 9d276e93 Helga Velroyen
      raise errors.OpPrereqError("Disk template '%s' is not enabled for this"
2661 7bb0c47f Thomas Thrainer
                                 " cluster." % self.op.disk_template)
2662 9d276e93 Helga Velroyen
2663 d0d7d7cf Thomas Thrainer
    if (self.instance.disk_template,
2664 22b7f6f8 Thomas Thrainer
        self.op.disk_template) not in self._DISK_CONVERSIONS:
2665 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Unsupported disk template conversion from"
2666 d0d7d7cf Thomas Thrainer
                                 " %s to %s" % (self.instance.disk_template,
2667 22b7f6f8 Thomas Thrainer
                                                self.op.disk_template),
2668 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_INVAL)
2669 d0d7d7cf Thomas Thrainer
    CheckInstanceState(self, self.instance, INSTANCE_DOWN,
2670 5eacbcae Thomas Thrainer
                       msg="cannot change disk template")
2671 22b7f6f8 Thomas Thrainer
    if self.op.disk_template in constants.DTS_INT_MIRROR:
2672 1c3231aa Thomas Thrainer
      if self.op.remote_node_uuid == pnode_uuid:
2673 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Given new secondary node %s is the same"
2674 22b7f6f8 Thomas Thrainer
                                   " as the primary node of the instance" %
2675 22b7f6f8 Thomas Thrainer
                                   self.op.remote_node, errors.ECODE_STATE)
2676 1c3231aa Thomas Thrainer
      CheckNodeOnline(self, self.op.remote_node_uuid)
2677 1c3231aa Thomas Thrainer
      CheckNodeNotDrained(self, self.op.remote_node_uuid)
2678 22b7f6f8 Thomas Thrainer
      # FIXME: here we assume that the old instance type is DT_PLAIN
2679 d0d7d7cf Thomas Thrainer
      assert self.instance.disk_template == constants.DT_PLAIN
2680 22b7f6f8 Thomas Thrainer
      disks = [{constants.IDISK_SIZE: d.size,
2681 22b7f6f8 Thomas Thrainer
                constants.IDISK_VG: d.logical_id[0]}
2682 d0d7d7cf Thomas Thrainer
               for d in self.instance.disks]
2683 5eacbcae Thomas Thrainer
      required = ComputeDiskSizePerVG(self.op.disk_template, disks)
2684 1c3231aa Thomas Thrainer
      CheckNodesFreeDiskPerVG(self, [self.op.remote_node_uuid], required)
2685 22b7f6f8 Thomas Thrainer
2686 1c3231aa Thomas Thrainer
      snode_info = self.cfg.GetNodeInfo(self.op.remote_node_uuid)
2687 22b7f6f8 Thomas Thrainer
      snode_group = self.cfg.GetNodeGroup(snode_info.group)
2688 d0d7d7cf Thomas Thrainer
      ipolicy = ganeti.masterd.instance.CalculateGroupIPolicy(self.cluster,
2689 22b7f6f8 Thomas Thrainer
                                                              snode_group)
2690 d0d7d7cf Thomas Thrainer
      CheckTargetNodeIPolicy(self, ipolicy, self.instance, snode_info, self.cfg,
2691 5eacbcae Thomas Thrainer
                             ignore=self.op.ignore_ipolicy)
2692 22b7f6f8 Thomas Thrainer
      if pnode_info.group != snode_info.group:
2693 22b7f6f8 Thomas Thrainer
        self.LogWarning("The primary and secondary nodes are in two"
2694 22b7f6f8 Thomas Thrainer
                        " different node groups; the disk parameters"
2695 22b7f6f8 Thomas Thrainer
                        " from the first disk's node group will be"
2696 22b7f6f8 Thomas Thrainer
                        " used")
2697 22b7f6f8 Thomas Thrainer
2698 22b7f6f8 Thomas Thrainer
    if not self.op.disk_template in constants.DTS_EXCL_STORAGE:
2699 22b7f6f8 Thomas Thrainer
      # Make sure none of the nodes require exclusive storage
2700 22b7f6f8 Thomas Thrainer
      nodes = [pnode_info]
2701 22b7f6f8 Thomas Thrainer
      if self.op.disk_template in constants.DTS_INT_MIRROR:
2702 22b7f6f8 Thomas Thrainer
        assert snode_info
2703 22b7f6f8 Thomas Thrainer
        nodes.append(snode_info)
2704 5eacbcae Thomas Thrainer
      has_es = lambda n: IsExclusiveStorageEnabledNode(self.cfg, n)
2705 22b7f6f8 Thomas Thrainer
      if compat.any(map(has_es, nodes)):
2706 22b7f6f8 Thomas Thrainer
        errmsg = ("Cannot convert disk template from %s to %s when exclusive"
2707 d0d7d7cf Thomas Thrainer
                  " storage is enabled" % (self.instance.disk_template,
2708 22b7f6f8 Thomas Thrainer
                                           self.op.disk_template))
2709 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError(errmsg, errors.ECODE_STATE)
2710 22b7f6f8 Thomas Thrainer
2711 1bb99a33 Bernardo Dal Seno
  def _PreCheckDisks(self, ispec):
2712 1bb99a33 Bernardo Dal Seno
    """CheckPrereq checks related to disk changes.
2713 22b7f6f8 Thomas Thrainer

2714 1bb99a33 Bernardo Dal Seno
    @type ispec: dict
2715 1bb99a33 Bernardo Dal Seno
    @param ispec: instance specs to be updated with the new disks
2716 22b7f6f8 Thomas Thrainer

2717 22b7f6f8 Thomas Thrainer
    """
2718 d0d7d7cf Thomas Thrainer
    self.diskparams = self.cfg.GetInstanceDiskParams(self.instance)
2719 22b7f6f8 Thomas Thrainer
2720 3f3ea14c Bernardo Dal Seno
    excl_stor = compat.any(
2721 d0d7d7cf Thomas Thrainer
      rpc.GetExclusiveStorageForNodes(self.cfg,
2722 d0d7d7cf Thomas Thrainer
                                      self.instance.all_nodes).values()
2723 3f3ea14c Bernardo Dal Seno
      )
2724 3f3ea14c Bernardo Dal Seno
2725 22b7f6f8 Thomas Thrainer
    # Check disk modifications. This is done here and not in CheckArguments
2726 22b7f6f8 Thomas Thrainer
    # (as with NICs), because we need to know the instance's disk template
2727 3f3ea14c Bernardo Dal Seno
    ver_fn = lambda op, par: self._VerifyDiskModification(op, par, excl_stor)
2728 d0d7d7cf Thomas Thrainer
    if self.instance.disk_template == constants.DT_EXT:
2729 3f3ea14c Bernardo Dal Seno
      self._CheckMods("disk", self.op.disks, {}, ver_fn)
2730 22b7f6f8 Thomas Thrainer
    else:
2731 22b7f6f8 Thomas Thrainer
      self._CheckMods("disk", self.op.disks, constants.IDISK_PARAMS_TYPES,
2732 3f3ea14c Bernardo Dal Seno
                      ver_fn)
2733 22b7f6f8 Thomas Thrainer
2734 5eacbcae Thomas Thrainer
    self.diskmod = _PrepareContainerMods(self.op.disks, None)
2735 22b7f6f8 Thomas Thrainer
2736 22b7f6f8 Thomas Thrainer
    # Check the validity of the `provider' parameter
2737 d0d7d7cf Thomas Thrainer
    if self.instance.disk_template in constants.DT_EXT:
2738 22b7f6f8 Thomas Thrainer
      for mod in self.diskmod:
2739 22b7f6f8 Thomas Thrainer
        ext_provider = mod[2].get(constants.IDISK_PROVIDER, None)
2740 22b7f6f8 Thomas Thrainer
        if mod[0] == constants.DDM_ADD:
2741 22b7f6f8 Thomas Thrainer
          if ext_provider is None:
2742 22b7f6f8 Thomas Thrainer
            raise errors.OpPrereqError("Instance template is '%s' and parameter"
2743 22b7f6f8 Thomas Thrainer
                                       " '%s' missing, during disk add" %
2744 22b7f6f8 Thomas Thrainer
                                       (constants.DT_EXT,
2745 22b7f6f8 Thomas Thrainer
                                        constants.IDISK_PROVIDER),
2746 22b7f6f8 Thomas Thrainer
                                       errors.ECODE_NOENT)
2747 22b7f6f8 Thomas Thrainer
        elif mod[0] == constants.DDM_MODIFY:
2748 22b7f6f8 Thomas Thrainer
          if ext_provider:
2749 22b7f6f8 Thomas Thrainer
            raise errors.OpPrereqError("Parameter '%s' is invalid during disk"
2750 22b7f6f8 Thomas Thrainer
                                       " modification" %
2751 22b7f6f8 Thomas Thrainer
                                       constants.IDISK_PROVIDER,
2752 22b7f6f8 Thomas Thrainer
                                       errors.ECODE_INVAL)
2753 22b7f6f8 Thomas Thrainer
    else:
2754 22b7f6f8 Thomas Thrainer
      for mod in self.diskmod:
2755 22b7f6f8 Thomas Thrainer
        ext_provider = mod[2].get(constants.IDISK_PROVIDER, None)
2756 22b7f6f8 Thomas Thrainer
        if ext_provider is not None:
2757 22b7f6f8 Thomas Thrainer
          raise errors.OpPrereqError("Parameter '%s' is only valid for"
2758 22b7f6f8 Thomas Thrainer
                                     " instances of type '%s'" %
2759 22b7f6f8 Thomas Thrainer
                                     (constants.IDISK_PROVIDER,
2760 22b7f6f8 Thomas Thrainer
                                      constants.DT_EXT),
2761 22b7f6f8 Thomas Thrainer
                                     errors.ECODE_INVAL)
2762 22b7f6f8 Thomas Thrainer
2763 3c260845 Thomas Thrainer
    if not self.op.wait_for_sync and self.instance.disks_active:
2764 3c260845 Thomas Thrainer
      for mod in self.diskmod:
2765 3c260845 Thomas Thrainer
        if mod[0] == constants.DDM_ADD:
2766 3c260845 Thomas Thrainer
          raise errors.OpPrereqError("Can't add a disk to an instance with"
2767 3c260845 Thomas Thrainer
                                     " activated disks and"
2768 3c260845 Thomas Thrainer
                                     " --no-wait-for-sync given.",
2769 3c260845 Thomas Thrainer
                                     errors.ECODE_INVAL)
2770 3c260845 Thomas Thrainer
2771 d0d7d7cf Thomas Thrainer
    if self.op.disks and self.instance.disk_template == constants.DT_DISKLESS:
2772 1bb99a33 Bernardo Dal Seno
      raise errors.OpPrereqError("Disk operations not supported for"
2773 1bb99a33 Bernardo Dal Seno
                                 " diskless instances", errors.ECODE_INVAL)
2774 1bb99a33 Bernardo Dal Seno
2775 1bb99a33 Bernardo Dal Seno
    def _PrepareDiskMod(_, disk, params, __):
2776 1bb99a33 Bernardo Dal Seno
      disk.name = params.get(constants.IDISK_NAME, None)
2777 1bb99a33 Bernardo Dal Seno
2778 1bb99a33 Bernardo Dal Seno
    # Verify disk changes (operating on a copy)
2779 d0d7d7cf Thomas Thrainer
    disks = copy.deepcopy(self.instance.disks)
2780 1bb99a33 Bernardo Dal Seno
    _ApplyContainerMods("disk", disks, None, self.diskmod, None,
2781 1bb99a33 Bernardo Dal Seno
                        _PrepareDiskMod, None)
2782 1bb99a33 Bernardo Dal Seno
    utils.ValidateDeviceNames("disk", disks)
2783 1bb99a33 Bernardo Dal Seno
    if len(disks) > constants.MAX_DISKS:
2784 1bb99a33 Bernardo Dal Seno
      raise errors.OpPrereqError("Instance has too many disks (%d), cannot add"
2785 1bb99a33 Bernardo Dal Seno
                                 " more" % constants.MAX_DISKS,
2786 1bb99a33 Bernardo Dal Seno
                                 errors.ECODE_STATE)
2787 d0d7d7cf Thomas Thrainer
    disk_sizes = [disk.size for disk in self.instance.disks]
2788 1bb99a33 Bernardo Dal Seno
    disk_sizes.extend(params["size"] for (op, idx, params, private) in
2789 1bb99a33 Bernardo Dal Seno
                      self.diskmod if op == constants.DDM_ADD)
2790 1bb99a33 Bernardo Dal Seno
    ispec[constants.ISPEC_DISK_COUNT] = len(disk_sizes)
2791 1bb99a33 Bernardo Dal Seno
    ispec[constants.ISPEC_DISK_SIZE] = disk_sizes
2792 1bb99a33 Bernardo Dal Seno
2793 1bb99a33 Bernardo Dal Seno
    if self.op.offline is not None and self.op.offline:
2794 d0d7d7cf Thomas Thrainer
      CheckInstanceState(self, self.instance, CAN_CHANGE_INSTANCE_OFFLINE,
2795 1bb99a33 Bernardo Dal Seno
                         msg="can't change to offline")
2796 1bb99a33 Bernardo Dal Seno
2797 1bb99a33 Bernardo Dal Seno
  def CheckPrereq(self):
2798 1bb99a33 Bernardo Dal Seno
    """Check prerequisites.
2799 1bb99a33 Bernardo Dal Seno

2800 1bb99a33 Bernardo Dal Seno
    This only checks the instance list against the existing names.
2801 1bb99a33 Bernardo Dal Seno

2802 1bb99a33 Bernardo Dal Seno
    """
2803 1bb99a33 Bernardo Dal Seno
    assert self.op.instance_name in self.owned_locks(locking.LEVEL_INSTANCE)
2804 da4a52a3 Thomas Thrainer
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_uuid)
2805 d0d7d7cf Thomas Thrainer
    self.cluster = self.cfg.GetClusterInfo()
2806 9c8f7bf4 Helga Velroyen
    cluster_hvparams = self.cluster.hvparams[self.instance.hypervisor]
2807 1bb99a33 Bernardo Dal Seno
2808 1bb99a33 Bernardo Dal Seno
    assert self.instance is not None, \
2809 1bb99a33 Bernardo Dal Seno
      "Cannot retrieve locked instance %s" % self.op.instance_name
2810 1bb99a33 Bernardo Dal Seno
2811 d0d7d7cf Thomas Thrainer
    pnode_uuid = self.instance.primary_node
2812 1bb99a33 Bernardo Dal Seno
2813 1bb99a33 Bernardo Dal Seno
    self.warn = []
2814 1bb99a33 Bernardo Dal Seno
2815 1c3231aa Thomas Thrainer
    if (self.op.pnode_uuid is not None and self.op.pnode_uuid != pnode_uuid and
2816 1bb99a33 Bernardo Dal Seno
        not self.op.force):
2817 1bb99a33 Bernardo Dal Seno
      # verify that the instance is not up
2818 0bbec3af Helga Velroyen
      instance_info = self.rpc.call_instance_info(
2819 d0d7d7cf Thomas Thrainer
          pnode_uuid, self.instance.name, self.instance.hypervisor,
2820 9c8f7bf4 Helga Velroyen
          cluster_hvparams)
2821 1bb99a33 Bernardo Dal Seno
      if instance_info.fail_msg:
2822 1bb99a33 Bernardo Dal Seno
        self.warn.append("Can't get instance runtime information: %s" %
2823 1bb99a33 Bernardo Dal Seno
                         instance_info.fail_msg)
2824 1bb99a33 Bernardo Dal Seno
      elif instance_info.payload:
2825 1c3231aa Thomas Thrainer
        raise errors.OpPrereqError("Instance is still running on %s" %
2826 1c3231aa Thomas Thrainer
                                   self.cfg.GetNodeName(pnode_uuid),
2827 1bb99a33 Bernardo Dal Seno
                                   errors.ECODE_STATE)
2828 1bb99a33 Bernardo Dal Seno
2829 1c3231aa Thomas Thrainer
    assert pnode_uuid in self.owned_locks(locking.LEVEL_NODE)
2830 d0d7d7cf Thomas Thrainer
    node_uuids = list(self.instance.all_nodes)
2831 1c3231aa Thomas Thrainer
    pnode_info = self.cfg.GetNodeInfo(pnode_uuid)
2832 1bb99a33 Bernardo Dal Seno
2833 1bb99a33 Bernardo Dal Seno
    #_CheckInstanceNodeGroups(self.cfg, self.op.instance_name, owned_groups)
2834 1bb99a33 Bernardo Dal Seno
    assert pnode_info.group in self.owned_locks(locking.LEVEL_NODEGROUP)
2835 1bb99a33 Bernardo Dal Seno
    group_info = self.cfg.GetNodeGroup(pnode_info.group)
2836 1bb99a33 Bernardo Dal Seno
2837 1bb99a33 Bernardo Dal Seno
    # dictionary with instance information after the modification
2838 1bb99a33 Bernardo Dal Seno
    ispec = {}
2839 1bb99a33 Bernardo Dal Seno
2840 24711492 Dimitris Aragiorgis
    if self.op.hotplug:
2841 24711492 Dimitris Aragiorgis
      result = self.rpc.call_hotplug_supported(self.instance.primary_node,
2842 24711492 Dimitris Aragiorgis
                                               self.instance)
2843 24711492 Dimitris Aragiorgis
      result.Raise("Hotplug is not supported.")
2844 24711492 Dimitris Aragiorgis
2845 1bb99a33 Bernardo Dal Seno
    # Prepare NIC modifications
2846 1bb99a33 Bernardo Dal Seno
    self.nicmod = _PrepareContainerMods(self.op.nics, _InstNicModPrivate)
2847 1bb99a33 Bernardo Dal Seno
2848 22b7f6f8 Thomas Thrainer
    # OS change
2849 22b7f6f8 Thomas Thrainer
    if self.op.os_name and not self.op.force:
2850 d0d7d7cf Thomas Thrainer
      CheckNodeHasOS(self, self.instance.primary_node, self.op.os_name,
2851 5eacbcae Thomas Thrainer
                     self.op.force_variant)
2852 22b7f6f8 Thomas Thrainer
      instance_os = self.op.os_name
2853 22b7f6f8 Thomas Thrainer
    else:
2854 d0d7d7cf Thomas Thrainer
      instance_os = self.instance.os
2855 22b7f6f8 Thomas Thrainer
2856 22b7f6f8 Thomas Thrainer
    assert not (self.op.disk_template and self.op.disks), \
2857 22b7f6f8 Thomas Thrainer
      "Can't modify disk template and apply disk changes at the same time"
2858 22b7f6f8 Thomas Thrainer
2859 22b7f6f8 Thomas Thrainer
    if self.op.disk_template:
2860 22b7f6f8 Thomas Thrainer
      self._PreCheckDiskTemplate(pnode_info)
2861 22b7f6f8 Thomas Thrainer
2862 1bb99a33 Bernardo Dal Seno
    self._PreCheckDisks(ispec)
2863 1bb99a33 Bernardo Dal Seno
2864 22b7f6f8 Thomas Thrainer
    # hvparams processing
2865 22b7f6f8 Thomas Thrainer
    if self.op.hvparams:
2866 d0d7d7cf Thomas Thrainer
      hv_type = self.instance.hypervisor
2867 d0d7d7cf Thomas Thrainer
      i_hvdict = GetUpdatedParams(self.instance.hvparams, self.op.hvparams)
2868 22b7f6f8 Thomas Thrainer
      utils.ForceDictType(i_hvdict, constants.HVS_PARAMETER_TYPES)
2869 d0d7d7cf Thomas Thrainer
      hv_new = self.cluster.SimpleFillHV(hv_type, self.instance.os, i_hvdict)
2870 22b7f6f8 Thomas Thrainer
2871 22b7f6f8 Thomas Thrainer
      # local check
2872 22b7f6f8 Thomas Thrainer
      hypervisor.GetHypervisorClass(hv_type).CheckParameterSyntax(hv_new)
2873 d0d7d7cf Thomas Thrainer
      CheckHVParams(self, node_uuids, self.instance.hypervisor, hv_new)
2874 22b7f6f8 Thomas Thrainer
      self.hv_proposed = self.hv_new = hv_new # the new actual values
2875 22b7f6f8 Thomas Thrainer
      self.hv_inst = i_hvdict # the new dict (without defaults)
2876 22b7f6f8 Thomas Thrainer
    else:
2877 d0d7d7cf Thomas Thrainer
      self.hv_proposed = self.cluster.SimpleFillHV(self.instance.hypervisor,
2878 d0d7d7cf Thomas Thrainer
                                                   self.instance.os,
2879 d0d7d7cf Thomas Thrainer
                                                   self.instance.hvparams)
2880 22b7f6f8 Thomas Thrainer
      self.hv_new = self.hv_inst = {}
2881 22b7f6f8 Thomas Thrainer
2882 22b7f6f8 Thomas Thrainer
    # beparams processing
2883 22b7f6f8 Thomas Thrainer
    if self.op.beparams:
2884 d0d7d7cf Thomas Thrainer
      i_bedict = GetUpdatedParams(self.instance.beparams, self.op.beparams,
2885 5eacbcae Thomas Thrainer
                                  use_none=True)
2886 22b7f6f8 Thomas Thrainer
      objects.UpgradeBeParams(i_bedict)
2887 22b7f6f8 Thomas Thrainer
      utils.ForceDictType(i_bedict, constants.BES_PARAMETER_TYPES)
2888 d0d7d7cf Thomas Thrainer
      be_new = self.cluster.SimpleFillBE(i_bedict)
2889 22b7f6f8 Thomas Thrainer
      self.be_proposed = self.be_new = be_new # the new actual values
2890 22b7f6f8 Thomas Thrainer
      self.be_inst = i_bedict # the new dict (without defaults)
2891 22b7f6f8 Thomas Thrainer
    else:
2892 22b7f6f8 Thomas Thrainer
      self.be_new = self.be_inst = {}
2893 d0d7d7cf Thomas Thrainer
      self.be_proposed = self.cluster.SimpleFillBE(self.instance.beparams)
2894 d0d7d7cf Thomas Thrainer
    be_old = self.cluster.FillBE(self.instance)
2895 22b7f6f8 Thomas Thrainer
2896 22b7f6f8 Thomas Thrainer
    # CPU param validation -- checking every time a parameter is
2897 22b7f6f8 Thomas Thrainer
    # changed to cover all cases where either CPU mask or vcpus have
2898 22b7f6f8 Thomas Thrainer
    # changed
2899 22b7f6f8 Thomas Thrainer
    if (constants.BE_VCPUS in self.be_proposed and
2900 22b7f6f8 Thomas Thrainer
        constants.HV_CPU_MASK in self.hv_proposed):
2901 22b7f6f8 Thomas Thrainer
      cpu_list = \
2902 22b7f6f8 Thomas Thrainer
        utils.ParseMultiCpuMask(self.hv_proposed[constants.HV_CPU_MASK])
2903 22b7f6f8 Thomas Thrainer
      # Verify mask is consistent with number of vCPUs. Can skip this
2904 22b7f6f8 Thomas Thrainer
      # test if only 1 entry in the CPU mask, which means same mask
2905 22b7f6f8 Thomas Thrainer
      # is applied to all vCPUs.
2906 22b7f6f8 Thomas Thrainer
      if (len(cpu_list) > 1 and
2907 22b7f6f8 Thomas Thrainer
          len(cpu_list) != self.be_proposed[constants.BE_VCPUS]):
2908 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Number of vCPUs [%d] does not match the"
2909 22b7f6f8 Thomas Thrainer
                                   " CPU mask [%s]" %
2910 22b7f6f8 Thomas Thrainer
                                   (self.be_proposed[constants.BE_VCPUS],
2911 22b7f6f8 Thomas Thrainer
                                    self.hv_proposed[constants.HV_CPU_MASK]),
2912 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
2913 22b7f6f8 Thomas Thrainer
2914 22b7f6f8 Thomas Thrainer
      # Only perform this test if a new CPU mask is given
2915 22b7f6f8 Thomas Thrainer
      if constants.HV_CPU_MASK in self.hv_new:
2916 22b7f6f8 Thomas Thrainer
        # Calculate the largest CPU number requested
2917 22b7f6f8 Thomas Thrainer
        max_requested_cpu = max(map(max, cpu_list))
2918 22b7f6f8 Thomas Thrainer
        # Check that all of the instance's nodes have enough physical CPUs to
2919 22b7f6f8 Thomas Thrainer
        # satisfy the requested CPU mask
2920 d0d7d7cf Thomas Thrainer
        hvspecs = [(self.instance.hypervisor,
2921 d0d7d7cf Thomas Thrainer
                    self.cfg.GetClusterInfo()
2922 d0d7d7cf Thomas Thrainer
                      .hvparams[self.instance.hypervisor])]
2923 d0d7d7cf Thomas Thrainer
        _CheckNodesPhysicalCPUs(self, self.instance.all_nodes,
2924 a295eb80 Helga Velroyen
                                max_requested_cpu + 1,
2925 a295eb80 Helga Velroyen
                                hvspecs)
2926 22b7f6f8 Thomas Thrainer
2927 22b7f6f8 Thomas Thrainer
    # osparams processing
2928 22b7f6f8 Thomas Thrainer
    if self.op.osparams:
2929 d0d7d7cf Thomas Thrainer
      i_osdict = GetUpdatedParams(self.instance.osparams, self.op.osparams)
2930 1c3231aa Thomas Thrainer
      CheckOSParams(self, True, node_uuids, instance_os, i_osdict)
2931 22b7f6f8 Thomas Thrainer
      self.os_inst = i_osdict # the new dict (without defaults)
2932 22b7f6f8 Thomas Thrainer
    else:
2933 22b7f6f8 Thomas Thrainer
      self.os_inst = {}
2934 22b7f6f8 Thomas Thrainer
2935 22b7f6f8 Thomas Thrainer
    #TODO(dynmem): do the appropriate check involving MINMEM
2936 22b7f6f8 Thomas Thrainer
    if (constants.BE_MAXMEM in self.op.beparams and not self.op.force and
2937 22b7f6f8 Thomas Thrainer
        be_new[constants.BE_MAXMEM] > be_old[constants.BE_MAXMEM]):
2938 1c3231aa Thomas Thrainer
      mem_check_list = [pnode_uuid]
2939 22b7f6f8 Thomas Thrainer
      if be_new[constants.BE_AUTO_BALANCE]:
2940 22b7f6f8 Thomas Thrainer
        # either we changed auto_balance to yes or it was from before
2941 d0d7d7cf Thomas Thrainer
        mem_check_list.extend(self.instance.secondary_nodes)
2942 a295eb80 Helga Velroyen
      instance_info = self.rpc.call_instance_info(
2943 d0d7d7cf Thomas Thrainer
          pnode_uuid, self.instance.name, self.instance.hypervisor,
2944 9c8f7bf4 Helga Velroyen
          cluster_hvparams)
2945 d0d7d7cf Thomas Thrainer
      hvspecs = [(self.instance.hypervisor,
2946 9c8f7bf4 Helga Velroyen
                  cluster_hvparams)]
2947 22b7f6f8 Thomas Thrainer
      nodeinfo = self.rpc.call_node_info(mem_check_list, None,
2948 da803ff1 Helga Velroyen
                                         hvspecs)
2949 1c3231aa Thomas Thrainer
      pninfo = nodeinfo[pnode_uuid]
2950 22b7f6f8 Thomas Thrainer
      msg = pninfo.fail_msg
2951 22b7f6f8 Thomas Thrainer
      if msg:
2952 22b7f6f8 Thomas Thrainer
        # Assume the primary node is unreachable and go ahead
2953 22b7f6f8 Thomas Thrainer
        self.warn.append("Can't get info from primary node %s: %s" %
2954 1c3231aa Thomas Thrainer
                         (self.cfg.GetNodeName(pnode_uuid), msg))
2955 22b7f6f8 Thomas Thrainer
      else:
2956 22b7f6f8 Thomas Thrainer
        (_, _, (pnhvinfo, )) = pninfo.payload
2957 22b7f6f8 Thomas Thrainer
        if not isinstance(pnhvinfo.get("memory_free", None), int):
2958 22b7f6f8 Thomas Thrainer
          self.warn.append("Node data from primary node %s doesn't contain"
2959 1c3231aa Thomas Thrainer
                           " free memory information" %
2960 1c3231aa Thomas Thrainer
                           self.cfg.GetNodeName(pnode_uuid))
2961 22b7f6f8 Thomas Thrainer
        elif instance_info.fail_msg:
2962 22b7f6f8 Thomas Thrainer
          self.warn.append("Can't get instance runtime information: %s" %
2963 22b7f6f8 Thomas Thrainer
                           instance_info.fail_msg)
2964 22b7f6f8 Thomas Thrainer
        else:
2965 22b7f6f8 Thomas Thrainer
          if instance_info.payload:
2966 22b7f6f8 Thomas Thrainer
            current_mem = int(instance_info.payload["memory"])
2967 22b7f6f8 Thomas Thrainer
          else:
2968 22b7f6f8 Thomas Thrainer
            # Assume instance not running
2969 22b7f6f8 Thomas Thrainer
            # (there is a slight race condition here, but it's not very
2970 22b7f6f8 Thomas Thrainer
            # probable, and we have no other way to check)
2971 22b7f6f8 Thomas Thrainer
            # TODO: Describe race condition
2972 22b7f6f8 Thomas Thrainer
            current_mem = 0
2973 22b7f6f8 Thomas Thrainer
          #TODO(dynmem): do the appropriate check involving MINMEM
2974 22b7f6f8 Thomas Thrainer
          miss_mem = (be_new[constants.BE_MAXMEM] - current_mem -
2975 22b7f6f8 Thomas Thrainer
                      pnhvinfo["memory_free"])
2976 22b7f6f8 Thomas Thrainer
          if miss_mem > 0:
2977 22b7f6f8 Thomas Thrainer
            raise errors.OpPrereqError("This change will prevent the instance"
2978 22b7f6f8 Thomas Thrainer
                                       " from starting, due to %d MB of memory"
2979 22b7f6f8 Thomas Thrainer
                                       " missing on its primary node" %
2980 22b7f6f8 Thomas Thrainer
                                       miss_mem, errors.ECODE_NORES)
2981 22b7f6f8 Thomas Thrainer
2982 22b7f6f8 Thomas Thrainer
      if be_new[constants.BE_AUTO_BALANCE]:
2983 1c3231aa Thomas Thrainer
        for node_uuid, nres in nodeinfo.items():
2984 d0d7d7cf Thomas Thrainer
          if node_uuid not in self.instance.secondary_nodes:
2985 22b7f6f8 Thomas Thrainer
            continue
2986 1c3231aa Thomas Thrainer
          nres.Raise("Can't get info from secondary node %s" %
2987 1c3231aa Thomas Thrainer
                     self.cfg.GetNodeName(node_uuid), prereq=True,
2988 1c3231aa Thomas Thrainer
                     ecode=errors.ECODE_STATE)
2989 22b7f6f8 Thomas Thrainer
          (_, _, (nhvinfo, )) = nres.payload
2990 22b7f6f8 Thomas Thrainer
          if not isinstance(nhvinfo.get("memory_free", None), int):
2991 22b7f6f8 Thomas Thrainer
            raise errors.OpPrereqError("Secondary node %s didn't return free"
2992 1c3231aa Thomas Thrainer
                                       " memory information" %
2993 1c3231aa Thomas Thrainer
                                       self.cfg.GetNodeName(node_uuid),
2994 22b7f6f8 Thomas Thrainer
                                       errors.ECODE_STATE)
2995 22b7f6f8 Thomas Thrainer
          #TODO(dynmem): do the appropriate check involving MINMEM
2996 22b7f6f8 Thomas Thrainer
          elif be_new[constants.BE_MAXMEM] > nhvinfo["memory_free"]:
2997 22b7f6f8 Thomas Thrainer
            raise errors.OpPrereqError("This change will prevent the instance"
2998 22b7f6f8 Thomas Thrainer
                                       " from failover to its secondary node"
2999 1c3231aa Thomas Thrainer
                                       " %s, due to not enough memory" %
3000 1c3231aa Thomas Thrainer
                                       self.cfg.GetNodeName(node_uuid),
3001 22b7f6f8 Thomas Thrainer
                                       errors.ECODE_STATE)
3002 22b7f6f8 Thomas Thrainer
3003 22b7f6f8 Thomas Thrainer
    if self.op.runtime_mem:
3004 0bbec3af Helga Velroyen
      remote_info = self.rpc.call_instance_info(
3005 d0d7d7cf Thomas Thrainer
         self.instance.primary_node, self.instance.name,
3006 b666a94c Helga Velroyen
         self.instance.hypervisor,
3007 9c8f7bf4 Helga Velroyen
         cluster_hvparams)
3008 1c3231aa Thomas Thrainer
      remote_info.Raise("Error checking node %s" %
3009 d0d7d7cf Thomas Thrainer
                        self.cfg.GetNodeName(self.instance.primary_node))
3010 22b7f6f8 Thomas Thrainer
      if not remote_info.payload: # not running already
3011 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Instance %s is not running" %
3012 d0d7d7cf Thomas Thrainer
                                   self.instance.name, errors.ECODE_STATE)
3013 22b7f6f8 Thomas Thrainer
3014 22b7f6f8 Thomas Thrainer
      current_memory = remote_info.payload["memory"]
3015 22b7f6f8 Thomas Thrainer
      if (not self.op.force and
3016 22b7f6f8 Thomas Thrainer
           (self.op.runtime_mem > self.be_proposed[constants.BE_MAXMEM] or
3017 22b7f6f8 Thomas Thrainer
            self.op.runtime_mem < self.be_proposed[constants.BE_MINMEM])):
3018 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError("Instance %s must have memory between %d"
3019 22b7f6f8 Thomas Thrainer
                                   " and %d MB of memory unless --force is"
3020 22b7f6f8 Thomas Thrainer
                                   " given" %
3021 d0d7d7cf Thomas Thrainer
                                   (self.instance.name,
3022 22b7f6f8 Thomas Thrainer
                                    self.be_proposed[constants.BE_MINMEM],
3023 22b7f6f8 Thomas Thrainer
                                    self.be_proposed[constants.BE_MAXMEM]),
3024 22b7f6f8 Thomas Thrainer
                                   errors.ECODE_INVAL)
3025 22b7f6f8 Thomas Thrainer
3026 22b7f6f8 Thomas Thrainer
      delta = self.op.runtime_mem - current_memory
3027 22b7f6f8 Thomas Thrainer
      if delta > 0:
3028 a295eb80 Helga Velroyen
        CheckNodeFreeMemory(
3029 d0d7d7cf Thomas Thrainer
            self, self.instance.primary_node,
3030 d0d7d7cf Thomas Thrainer
            "ballooning memory for instance %s" % self.instance.name, delta,
3031 d0d7d7cf Thomas Thrainer
            self.instance.hypervisor,
3032 d0d7d7cf Thomas Thrainer
            self.cfg.GetClusterInfo().hvparams[self.instance.hypervisor])
3033 d0d7d7cf Thomas Thrainer
3034 d0d7d7cf Thomas Thrainer
    # make self.cluster visible in the functions below
3035 d0d7d7cf Thomas Thrainer
    cluster = self.cluster
3036 22b7f6f8 Thomas Thrainer
3037 22b7f6f8 Thomas Thrainer
    def _PrepareNicCreate(_, params, private):
3038 22b7f6f8 Thomas Thrainer
      self._PrepareNicModification(params, private, None, None,
3039 1c3231aa Thomas Thrainer
                                   {}, cluster, pnode_uuid)
3040 22b7f6f8 Thomas Thrainer
      return (None, None)
3041 22b7f6f8 Thomas Thrainer
3042 22b7f6f8 Thomas Thrainer
    def _PrepareNicMod(_, nic, params, private):
3043 22b7f6f8 Thomas Thrainer
      self._PrepareNicModification(params, private, nic.ip, nic.network,
3044 1c3231aa Thomas Thrainer
                                   nic.nicparams, cluster, pnode_uuid)
3045 22b7f6f8 Thomas Thrainer
      return None
3046 22b7f6f8 Thomas Thrainer
3047 22b7f6f8 Thomas Thrainer
    def _PrepareNicRemove(_, params, __):
3048 22b7f6f8 Thomas Thrainer
      ip = params.ip
3049 22b7f6f8 Thomas Thrainer
      net = params.network
3050 22b7f6f8 Thomas Thrainer
      if net is not None and ip is not None:
3051 22b7f6f8 Thomas Thrainer
        self.cfg.ReleaseIp(net, ip, self.proc.GetECId())
3052 22b7f6f8 Thomas Thrainer
3053 22b7f6f8 Thomas Thrainer
    # Verify NIC changes (operating on copy)
3054 d0d7d7cf Thomas Thrainer
    nics = self.instance.nics[:]
3055 5eacbcae Thomas Thrainer
    _ApplyContainerMods("NIC", nics, None, self.nicmod,
3056 5eacbcae Thomas Thrainer
                        _PrepareNicCreate, _PrepareNicMod, _PrepareNicRemove)
3057 22b7f6f8 Thomas Thrainer
    if len(nics) > constants.MAX_NICS:
3058 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Instance has too many network interfaces"
3059 22b7f6f8 Thomas Thrainer
                                 " (%d), cannot add more" % constants.MAX_NICS,
3060 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_STATE)
3061 22b7f6f8 Thomas Thrainer
3062 22b7f6f8 Thomas Thrainer
    # Pre-compute NIC changes (necessary to use result in hooks)
3063 22b7f6f8 Thomas Thrainer
    self._nic_chgdesc = []
3064 22b7f6f8 Thomas Thrainer
    if self.nicmod:
3065 22b7f6f8 Thomas Thrainer
      # Operate on copies as this is still in prereq
3066 d0d7d7cf Thomas Thrainer
      nics = [nic.Copy() for nic in self.instance.nics]
3067 5eacbcae Thomas Thrainer
      _ApplyContainerMods("NIC", nics, self._nic_chgdesc, self.nicmod,
3068 ba924970 Dimitris Aragiorgis
                          self._CreateNewNic, self._ApplyNicMods,
3069 ba924970 Dimitris Aragiorgis
                          self._RemoveNic)
3070 22b7f6f8 Thomas Thrainer
      # Verify that NIC names are unique and valid
3071 22b7f6f8 Thomas Thrainer
      utils.ValidateDeviceNames("NIC", nics)
3072 22b7f6f8 Thomas Thrainer
      self._new_nics = nics
3073 22b7f6f8 Thomas Thrainer
      ispec[constants.ISPEC_NIC_COUNT] = len(self._new_nics)
3074 22b7f6f8 Thomas Thrainer
    else:
3075 22b7f6f8 Thomas Thrainer
      self._new_nics = None
3076 d0d7d7cf Thomas Thrainer
      ispec[constants.ISPEC_NIC_COUNT] = len(self.instance.nics)
3077 22b7f6f8 Thomas Thrainer
3078 22b7f6f8 Thomas Thrainer
    if not self.op.ignore_ipolicy:
3079 d0d7d7cf Thomas Thrainer
      ipolicy = ganeti.masterd.instance.CalculateGroupIPolicy(self.cluster,
3080 22b7f6f8 Thomas Thrainer
                                                              group_info)
3081 22b7f6f8 Thomas Thrainer
3082 22b7f6f8 Thomas Thrainer
      # Fill ispec with backend parameters
3083 22b7f6f8 Thomas Thrainer
      ispec[constants.ISPEC_SPINDLE_USE] = \
3084 22b7f6f8 Thomas Thrainer
        self.be_new.get(constants.BE_SPINDLE_USE, None)
3085 22b7f6f8 Thomas Thrainer
      ispec[constants.ISPEC_CPU_COUNT] = self.be_new.get(constants.BE_VCPUS,
3086 22b7f6f8 Thomas Thrainer
                                                         None)
3087 22b7f6f8 Thomas Thrainer
3088 22b7f6f8 Thomas Thrainer
      # Copy ispec to verify parameters with min/max values separately
3089 22b7f6f8 Thomas Thrainer
      if self.op.disk_template:
3090 22b7f6f8 Thomas Thrainer
        new_disk_template = self.op.disk_template
3091 22b7f6f8 Thomas Thrainer
      else:
3092 d0d7d7cf Thomas Thrainer
        new_disk_template = self.instance.disk_template
3093 22b7f6f8 Thomas Thrainer
      ispec_max = ispec.copy()
3094 22b7f6f8 Thomas Thrainer
      ispec_max[constants.ISPEC_MEM_SIZE] = \
3095 22b7f6f8 Thomas Thrainer
        self.be_new.get(constants.BE_MAXMEM, None)
3096 22b7f6f8 Thomas Thrainer
      res_max = _ComputeIPolicyInstanceSpecViolation(ipolicy, ispec_max,
3097 22b7f6f8 Thomas Thrainer
                                                     new_disk_template)
3098 22b7f6f8 Thomas Thrainer
      ispec_min = ispec.copy()
3099 22b7f6f8 Thomas Thrainer
      ispec_min[constants.ISPEC_MEM_SIZE] = \
3100 22b7f6f8 Thomas Thrainer
        self.be_new.get(constants.BE_MINMEM, None)
3101 22b7f6f8 Thomas Thrainer
      res_min = _ComputeIPolicyInstanceSpecViolation(ipolicy, ispec_min,
3102 22b7f6f8 Thomas Thrainer
                                                     new_disk_template)
3103 22b7f6f8 Thomas Thrainer
3104 22b7f6f8 Thomas Thrainer
      if (res_max or res_min):
3105 22b7f6f8 Thomas Thrainer
        # FIXME: Improve error message by including information about whether
3106 22b7f6f8 Thomas Thrainer
        # the upper or lower limit of the parameter fails the ipolicy.
3107 22b7f6f8 Thomas Thrainer
        msg = ("Instance allocation to group %s (%s) violates policy: %s" %
3108 22b7f6f8 Thomas Thrainer
               (group_info, group_info.name,
3109 22b7f6f8 Thomas Thrainer
                utils.CommaJoin(set(res_max + res_min))))
3110 22b7f6f8 Thomas Thrainer
        raise errors.OpPrereqError(msg, errors.ECODE_INVAL)
3111 22b7f6f8 Thomas Thrainer
3112 22b7f6f8 Thomas Thrainer
  def _ConvertPlainToDrbd(self, feedback_fn):
3113 22b7f6f8 Thomas Thrainer
    """Converts an instance from plain to drbd.
3114 22b7f6f8 Thomas Thrainer

3115 22b7f6f8 Thomas Thrainer
    """
3116 22b7f6f8 Thomas Thrainer
    feedback_fn("Converting template to drbd")
3117 d0d7d7cf Thomas Thrainer
    pnode_uuid = self.instance.primary_node
3118 1c3231aa Thomas Thrainer
    snode_uuid = self.op.remote_node_uuid
3119 22b7f6f8 Thomas Thrainer
3120 d0d7d7cf Thomas Thrainer
    assert self.instance.disk_template == constants.DT_PLAIN
3121 22b7f6f8 Thomas Thrainer
3122 22b7f6f8 Thomas Thrainer
    # create a fake disk info for _GenerateDiskTemplate
3123 22b7f6f8 Thomas Thrainer
    disk_info = [{constants.IDISK_SIZE: d.size, constants.IDISK_MODE: d.mode,
3124 22b7f6f8 Thomas Thrainer
                  constants.IDISK_VG: d.logical_id[0],
3125 22b7f6f8 Thomas Thrainer
                  constants.IDISK_NAME: d.name}
3126 d0d7d7cf Thomas Thrainer
                 for d in self.instance.disks]
3127 5eacbcae Thomas Thrainer
    new_disks = GenerateDiskTemplate(self, self.op.disk_template,
3128 da4a52a3 Thomas Thrainer
                                     self.instance.uuid, pnode_uuid,
3129 d0d7d7cf Thomas Thrainer
                                     [snode_uuid], disk_info, None, None, 0,
3130 d0d7d7cf Thomas Thrainer
                                     feedback_fn, self.diskparams)
3131 0c3d9c7c Thomas Thrainer
    anno_disks = rpc.AnnotateDiskParams(new_disks, self.diskparams)
3132 1c3231aa Thomas Thrainer
    p_excl_stor = IsExclusiveStorageEnabledNodeUuid(self.cfg, pnode_uuid)
3133 1c3231aa Thomas Thrainer
    s_excl_stor = IsExclusiveStorageEnabledNodeUuid(self.cfg, snode_uuid)
3134 d0d7d7cf Thomas Thrainer
    info = GetInstanceInfoText(self.instance)
3135 22b7f6f8 Thomas Thrainer
    feedback_fn("Creating additional volumes...")
3136 22b7f6f8 Thomas Thrainer
    # first, create the missing data and meta devices
3137 22b7f6f8 Thomas Thrainer
    for disk in anno_disks:
3138 22b7f6f8 Thomas Thrainer
      # unfortunately this is... not too nice
3139 d0d7d7cf Thomas Thrainer
      CreateSingleBlockDev(self, pnode_uuid, self.instance, disk.children[1],
3140 5eacbcae Thomas Thrainer
                           info, True, p_excl_stor)
3141 22b7f6f8 Thomas Thrainer
      for child in disk.children:
3142 d0d7d7cf Thomas Thrainer
        CreateSingleBlockDev(self, snode_uuid, self.instance, child, info, True,
3143 5eacbcae Thomas Thrainer
                             s_excl_stor)
3144 22b7f6f8 Thomas Thrainer
    # at this stage, all new LVs have been created, we can rename the
3145 22b7f6f8 Thomas Thrainer
    # old ones
3146 22b7f6f8 Thomas Thrainer
    feedback_fn("Renaming original volumes...")
3147 22b7f6f8 Thomas Thrainer
    rename_list = [(o, n.children[0].logical_id)
3148 d0d7d7cf Thomas Thrainer
                   for (o, n) in zip(self.instance.disks, new_disks)]
3149 1c3231aa Thomas Thrainer
    result = self.rpc.call_blockdev_rename(pnode_uuid, rename_list)
3150 22b7f6f8 Thomas Thrainer
    result.Raise("Failed to rename original LVs")
3151 22b7f6f8 Thomas Thrainer
3152 22b7f6f8 Thomas Thrainer
    feedback_fn("Initializing DRBD devices...")
3153 22b7f6f8 Thomas Thrainer
    # all child devices are in place, we can now create the DRBD devices
3154 22b7f6f8 Thomas Thrainer
    try:
3155 22b7f6f8 Thomas Thrainer
      for disk in anno_disks:
3156 1c3231aa Thomas Thrainer
        for (node_uuid, excl_stor) in [(pnode_uuid, p_excl_stor),
3157 1c3231aa Thomas Thrainer
                                       (snode_uuid, s_excl_stor)]:
3158 1c3231aa Thomas Thrainer
          f_create = node_uuid == pnode_uuid
3159 d0d7d7cf Thomas Thrainer
          CreateSingleBlockDev(self, node_uuid, self.instance, disk, info,
3160 d0d7d7cf Thomas Thrainer
                               f_create, excl_stor)
3161 22b7f6f8 Thomas Thrainer
    except errors.GenericError, e:
3162 22b7f6f8 Thomas Thrainer
      feedback_fn("Initializing of DRBD devices failed;"
3163 22b7f6f8 Thomas Thrainer
                  " renaming back original volumes...")
3164 22b7f6f8 Thomas Thrainer
      rename_back_list = [(n.children[0], o.logical_id)
3165 d0d7d7cf Thomas Thrainer
                          for (n, o) in zip(new_disks, self.instance.disks)]
3166 1c3231aa Thomas Thrainer
      result = self.rpc.call_blockdev_rename(pnode_uuid, rename_back_list)
3167 22b7f6f8 Thomas Thrainer
      result.Raise("Failed to rename LVs back after error %s" % str(e))
3168 22b7f6f8 Thomas Thrainer
      raise
3169 22b7f6f8 Thomas Thrainer
3170 22b7f6f8 Thomas Thrainer
    # at this point, the instance has been modified
3171 d0d7d7cf Thomas Thrainer
    self.instance.disk_template = constants.DT_DRBD8
3172 d0d7d7cf Thomas Thrainer
    self.instance.disks = new_disks
3173 d0d7d7cf Thomas Thrainer
    self.cfg.Update(self.instance, feedback_fn)
3174 22b7f6f8 Thomas Thrainer
3175 22b7f6f8 Thomas Thrainer
    # Release node locks while waiting for sync
3176 5eacbcae Thomas Thrainer
    ReleaseLocks(self, locking.LEVEL_NODE)
3177 22b7f6f8 Thomas Thrainer
3178 22b7f6f8 Thomas Thrainer
    # disks are created, waiting for sync
3179 d0d7d7cf Thomas Thrainer
    disk_abort = not WaitForSync(self, self.instance,
3180 5eacbcae Thomas Thrainer
                                 oneshot=not self.op.wait_for_sync)
3181 22b7f6f8 Thomas Thrainer
    if disk_abort:
3182 22b7f6f8 Thomas Thrainer
      raise errors.OpExecError("There are some degraded disks for"
3183 22b7f6f8 Thomas Thrainer
                               " this instance, please cleanup manually")
3184 22b7f6f8 Thomas Thrainer
3185 22b7f6f8 Thomas Thrainer
    # Node resource locks will be released by caller
3186 22b7f6f8 Thomas Thrainer
3187 22b7f6f8 Thomas Thrainer
  def _ConvertDrbdToPlain(self, feedback_fn):
3188 22b7f6f8 Thomas Thrainer
    """Converts an instance from drbd to plain.
3189 22b7f6f8 Thomas Thrainer

3190 22b7f6f8 Thomas Thrainer
    """
3191 d0d7d7cf Thomas Thrainer
    assert len(self.instance.secondary_nodes) == 1
3192 d0d7d7cf Thomas Thrainer
    assert self.instance.disk_template == constants.DT_DRBD8
3193 22b7f6f8 Thomas Thrainer
3194 d0d7d7cf Thomas Thrainer
    pnode_uuid = self.instance.primary_node
3195 d0d7d7cf Thomas Thrainer
    snode_uuid = self.instance.secondary_nodes[0]
3196 22b7f6f8 Thomas Thrainer
    feedback_fn("Converting template to plain")
3197 22b7f6f8 Thomas Thrainer
3198 d0d7d7cf Thomas Thrainer
    old_disks = AnnotateDiskParams(self.instance, self.instance.disks, self.cfg)
3199 d0d7d7cf Thomas Thrainer
    new_disks = [d.children[0] for d in self.instance.disks]
3200 22b7f6f8 Thomas Thrainer
3201 22b7f6f8 Thomas Thrainer
    # copy over size, mode and name
3202 22b7f6f8 Thomas Thrainer
    for parent, child in zip(old_disks, new_disks):
3203 22b7f6f8 Thomas Thrainer
      child.size = parent.size
3204 22b7f6f8 Thomas Thrainer
      child.mode = parent.mode
3205 22b7f6f8 Thomas Thrainer
      child.name = parent.name
3206 22b7f6f8 Thomas Thrainer
3207 22b7f6f8 Thomas Thrainer
    # this is a DRBD disk, return its port to the pool
3208 22b7f6f8 Thomas Thrainer
    # NOTE: this must be done right before the call to cfg.Update!
3209 22b7f6f8 Thomas Thrainer
    for disk in old_disks:
3210 22b7f6f8 Thomas Thrainer
      tcp_port = disk.logical_id[2]
3211 22b7f6f8 Thomas Thrainer
      self.cfg.AddTcpUdpPort(tcp_port)
3212 22b7f6f8 Thomas Thrainer
3213 22b7f6f8 Thomas Thrainer
    # update instance structure
3214 d0d7d7cf Thomas Thrainer
    self.instance.disks = new_disks
3215 d0d7d7cf Thomas Thrainer
    self.instance.disk_template = constants.DT_PLAIN
3216 d0d7d7cf Thomas Thrainer
    _UpdateIvNames(0, self.instance.disks)
3217 d0d7d7cf Thomas Thrainer
    self.cfg.Update(self.instance, feedback_fn)
3218 22b7f6f8 Thomas Thrainer
3219 22b7f6f8 Thomas Thrainer
    # Release locks in case removing disks takes a while
3220 5eacbcae Thomas Thrainer
    ReleaseLocks(self, locking.LEVEL_NODE)
3221 22b7f6f8 Thomas Thrainer
3222 22b7f6f8 Thomas Thrainer
    feedback_fn("Removing volumes on the secondary node...")
3223 22b7f6f8 Thomas Thrainer
    for disk in old_disks:
3224 0c3d9c7c Thomas Thrainer
      result = self.rpc.call_blockdev_remove(snode_uuid, (disk, self.instance))
3225 aefc2f89 Thomas Thrainer
      result.Warn("Could not remove block device %s on node %s,"
3226 aefc2f89 Thomas Thrainer
                  " continuing anyway" %
3227 aefc2f89 Thomas Thrainer
                  (disk.iv_name, self.cfg.GetNodeName(snode_uuid)),
3228 aefc2f89 Thomas Thrainer
                  self.LogWarning)
3229 22b7f6f8 Thomas Thrainer
3230 22b7f6f8 Thomas Thrainer
    feedback_fn("Removing unneeded volumes on the primary node...")
3231 22b7f6f8 Thomas Thrainer
    for idx, disk in enumerate(old_disks):
3232 22b7f6f8 Thomas Thrainer
      meta = disk.children[1]
3233 0c3d9c7c Thomas Thrainer
      result = self.rpc.call_blockdev_remove(pnode_uuid, (meta, self.instance))
3234 aefc2f89 Thomas Thrainer
      result.Warn("Could not remove metadata for disk %d on node %s,"
3235 aefc2f89 Thomas Thrainer
                  " continuing anyway" %
3236 aefc2f89 Thomas Thrainer
                  (idx, self.cfg.GetNodeName(pnode_uuid)),
3237 aefc2f89 Thomas Thrainer
                  self.LogWarning)
3238 22b7f6f8 Thomas Thrainer
3239 ba924970 Dimitris Aragiorgis
  def _HotplugDevice(self, action, dev_type, device, extra, seq):
3240 ba924970 Dimitris Aragiorgis
    self.LogInfo("Trying to hotplug device...")
3241 e15a00dc Dimitris Aragiorgis
    msg = "hotplug:"
3242 ba924970 Dimitris Aragiorgis
    result = self.rpc.call_hotplug_device(self.instance.primary_node,
3243 ba924970 Dimitris Aragiorgis
                                          self.instance, action, dev_type,
3244 51951d38 Dimitris Aragiorgis
                                          (device, self.instance),
3245 51951d38 Dimitris Aragiorgis
                                          extra, seq)
3246 ba924970 Dimitris Aragiorgis
    if result.fail_msg:
3247 ba924970 Dimitris Aragiorgis
      self.LogWarning("Could not hotplug device: %s" % result.fail_msg)
3248 ba924970 Dimitris Aragiorgis
      self.LogInfo("Continuing execution..")
3249 e15a00dc Dimitris Aragiorgis
      msg += "failed"
3250 ba924970 Dimitris Aragiorgis
    else:
3251 ba924970 Dimitris Aragiorgis
      self.LogInfo("Hotplug done.")
3252 e15a00dc Dimitris Aragiorgis
      msg += "done"
3253 e15a00dc Dimitris Aragiorgis
    return msg
3254 ba924970 Dimitris Aragiorgis
3255 22b7f6f8 Thomas Thrainer
  def _CreateNewDisk(self, idx, params, _):
3256 22b7f6f8 Thomas Thrainer
    """Creates a new disk.
3257 22b7f6f8 Thomas Thrainer

3258 22b7f6f8 Thomas Thrainer
    """
3259 22b7f6f8 Thomas Thrainer
    # add a new disk
3260 d0d7d7cf Thomas Thrainer
    if self.instance.disk_template in constants.DTS_FILEBASED:
3261 d0d7d7cf Thomas Thrainer
      (file_driver, file_path) = self.instance.disks[0].logical_id
3262 22b7f6f8 Thomas Thrainer
      file_path = os.path.dirname(file_path)
3263 22b7f6f8 Thomas Thrainer
    else:
3264 22b7f6f8 Thomas Thrainer
      file_driver = file_path = None
3265 22b7f6f8 Thomas Thrainer
3266 22b7f6f8 Thomas Thrainer
    disk = \
3267 d0d7d7cf Thomas Thrainer
      GenerateDiskTemplate(self, self.instance.disk_template,
3268 da4a52a3 Thomas Thrainer
                           self.instance.uuid, self.instance.primary_node,
3269 d0d7d7cf Thomas Thrainer
                           self.instance.secondary_nodes, [params], file_path,
3270 d0d7d7cf Thomas Thrainer
                           file_driver, idx, self.Log, self.diskparams)[0]
3271 22b7f6f8 Thomas Thrainer
3272 d0d7d7cf Thomas Thrainer
    new_disks = CreateDisks(self, self.instance, disks=[disk])
3273 22b7f6f8 Thomas Thrainer
3274 22b7f6f8 Thomas Thrainer
    if self.cluster.prealloc_wipe_disks:
3275 22b7f6f8 Thomas Thrainer
      # Wipe new disk
3276 d0d7d7cf Thomas Thrainer
      WipeOrCleanupDisks(self, self.instance,
3277 a365b47f Bernardo Dal Seno
                         disks=[(idx, disk, 0)],
3278 a365b47f Bernardo Dal Seno
                         cleanup=new_disks)
3279 22b7f6f8 Thomas Thrainer
3280 e15a00dc Dimitris Aragiorgis
    changes = [
3281 e15a00dc Dimitris Aragiorgis
      ("disk/%d" % idx,
3282 b15d5bd3 Petr Pudlak
       "add:size=%s,mode=%s" % (disk.size, disk.mode)),
3283 e15a00dc Dimitris Aragiorgis
      ]
3284 ba924970 Dimitris Aragiorgis
    if self.op.hotplug:
3285 ba924970 Dimitris Aragiorgis
      result = self.rpc.call_blockdev_assemble(self.instance.primary_node,
3286 ba924970 Dimitris Aragiorgis
                                               (disk, self.instance),
3287 ba924970 Dimitris Aragiorgis
                                               self.instance.name, True, idx)
3288 ba924970 Dimitris Aragiorgis
      if result.fail_msg:
3289 e15a00dc Dimitris Aragiorgis
        changes.append(("disk/%d" % idx, "assemble:failed"))
3290 ba924970 Dimitris Aragiorgis
        self.LogWarning("Can't assemble newly created disk %d: %s",
3291 ba924970 Dimitris Aragiorgis
                        idx, result.fail_msg)
3292 ba924970 Dimitris Aragiorgis
      else:
3293 ba924970 Dimitris Aragiorgis
        _, link_name = result.payload
3294 e15a00dc Dimitris Aragiorgis
        msg = self._HotplugDevice(constants.HOTPLUG_ACTION_ADD,
3295 e15a00dc Dimitris Aragiorgis
                                  constants.HOTPLUG_TARGET_DISK,
3296 e15a00dc Dimitris Aragiorgis
                                  disk, link_name, idx)
3297 e15a00dc Dimitris Aragiorgis
        changes.append(("disk/%d" % idx, msg))
3298 ba924970 Dimitris Aragiorgis
3299 e15a00dc Dimitris Aragiorgis
    return (disk, changes)
3300 22b7f6f8 Thomas Thrainer
3301 922a9e65 Thomas Thrainer
  def _PostAddDisk(self, _, disk):
3302 922a9e65 Thomas Thrainer
    if not WaitForSync(self, self.instance, disks=[disk],
3303 922a9e65 Thomas Thrainer
                       oneshot=not self.op.wait_for_sync):
3304 922a9e65 Thomas Thrainer
      raise errors.OpExecError("Failed to sync disks of %s" %
3305 922a9e65 Thomas Thrainer
                               self.instance.name)
3306 922a9e65 Thomas Thrainer
3307 3c260845 Thomas Thrainer
    # the disk is active at this point, so deactivate it if the instance disks
3308 3c260845 Thomas Thrainer
    # are supposed to be inactive
3309 3c260845 Thomas Thrainer
    if not self.instance.disks_active:
3310 3c260845 Thomas Thrainer
      ShutdownInstanceDisks(self, self.instance, disks=[disk])
3311 3c260845 Thomas Thrainer
3312 22b7f6f8 Thomas Thrainer
  @staticmethod
3313 22b7f6f8 Thomas Thrainer
  def _ModifyDisk(idx, disk, params, _):
3314 22b7f6f8 Thomas Thrainer
    """Modifies a disk.
3315 22b7f6f8 Thomas Thrainer

3316 22b7f6f8 Thomas Thrainer
    """
3317 22b7f6f8 Thomas Thrainer
    changes = []
3318 22b7f6f8 Thomas Thrainer
    mode = params.get(constants.IDISK_MODE, None)
3319 22b7f6f8 Thomas Thrainer
    if mode:
3320 22b7f6f8 Thomas Thrainer
      disk.mode = mode
3321 22b7f6f8 Thomas Thrainer
      changes.append(("disk.mode/%d" % idx, disk.mode))
3322 22b7f6f8 Thomas Thrainer
3323 22b7f6f8 Thomas Thrainer
    name = params.get(constants.IDISK_NAME, None)
3324 22b7f6f8 Thomas Thrainer
    disk.name = name
3325 22b7f6f8 Thomas Thrainer
    changes.append(("disk.name/%d" % idx, disk.name))
3326 22b7f6f8 Thomas Thrainer
3327 22b7f6f8 Thomas Thrainer
    return changes
3328 22b7f6f8 Thomas Thrainer
3329 22b7f6f8 Thomas Thrainer
  def _RemoveDisk(self, idx, root, _):
3330 22b7f6f8 Thomas Thrainer
    """Removes a disk.
3331 22b7f6f8 Thomas Thrainer

3332 22b7f6f8 Thomas Thrainer
    """
3333 e15a00dc Dimitris Aragiorgis
    hotmsg = ""
3334 ba924970 Dimitris Aragiorgis
    if self.op.hotplug:
3335 e15a00dc Dimitris Aragiorgis
      hotmsg = self._HotplugDevice(constants.HOTPLUG_ACTION_REMOVE,
3336 e15a00dc Dimitris Aragiorgis
                                   constants.HOTPLUG_TARGET_DISK,
3337 e15a00dc Dimitris Aragiorgis
                                   root, None, idx)
3338 ba924970 Dimitris Aragiorgis
      ShutdownInstanceDisks(self, self.instance, [root])
3339 ba924970 Dimitris Aragiorgis
3340 5eacbcae Thomas Thrainer
    (anno_disk,) = AnnotateDiskParams(self.instance, [root], self.cfg)
3341 1c3231aa Thomas Thrainer
    for node_uuid, disk in anno_disk.ComputeNodeTree(
3342 1c3231aa Thomas Thrainer
                             self.instance.primary_node):
3343 0c3d9c7c Thomas Thrainer
      msg = self.rpc.call_blockdev_remove(node_uuid, (disk, self.instance)) \
3344 0c3d9c7c Thomas Thrainer
              .fail_msg
3345 22b7f6f8 Thomas Thrainer
      if msg:
3346 22b7f6f8 Thomas Thrainer
        self.LogWarning("Could not remove disk/%d on node '%s': %s,"
3347 1c3231aa Thomas Thrainer
                        " continuing anyway", idx,
3348 1c3231aa Thomas Thrainer
                        self.cfg.GetNodeName(node_uuid), msg)
3349 22b7f6f8 Thomas Thrainer
3350 22b7f6f8 Thomas Thrainer
    # if this is a DRBD disk, return its port to the pool
3351 66a37e7a Helga Velroyen
    if root.dev_type in constants.DTS_DRBD:
3352 22b7f6f8 Thomas Thrainer
      self.cfg.AddTcpUdpPort(root.logical_id[2])
3353 22b7f6f8 Thomas Thrainer
3354 e15a00dc Dimitris Aragiorgis
    return hotmsg
3355 e15a00dc Dimitris Aragiorgis
3356 22b7f6f8 Thomas Thrainer
  def _CreateNewNic(self, idx, params, private):
3357 22b7f6f8 Thomas Thrainer
    """Creates data structure for a new network interface.
3358 22b7f6f8 Thomas Thrainer

3359 22b7f6f8 Thomas Thrainer
    """
3360 22b7f6f8 Thomas Thrainer
    mac = params[constants.INIC_MAC]
3361 22b7f6f8 Thomas Thrainer
    ip = params.get(constants.INIC_IP, None)
3362 22b7f6f8 Thomas Thrainer
    net = params.get(constants.INIC_NETWORK, None)
3363 22b7f6f8 Thomas Thrainer
    name = params.get(constants.INIC_NAME, None)
3364 22b7f6f8 Thomas Thrainer
    net_uuid = self.cfg.LookupNetwork(net)
3365 22b7f6f8 Thomas Thrainer
    #TODO: not private.filled?? can a nic have no nicparams??
3366 22b7f6f8 Thomas Thrainer
    nicparams = private.filled
3367 22b7f6f8 Thomas Thrainer
    nobj = objects.NIC(mac=mac, ip=ip, network=net_uuid, name=name,
3368 22b7f6f8 Thomas Thrainer
                       nicparams=nicparams)
3369 22b7f6f8 Thomas Thrainer
    nobj.uuid = self.cfg.GenerateUniqueID(self.proc.GetECId())
3370 22b7f6f8 Thomas Thrainer
3371 e15a00dc Dimitris Aragiorgis
    changes = [
3372 22b7f6f8 Thomas Thrainer
      ("nic.%d" % idx,
3373 22b7f6f8 Thomas Thrainer
       "add:mac=%s,ip=%s,mode=%s,link=%s,network=%s" %
3374 22b7f6f8 Thomas Thrainer
       (mac, ip, private.filled[constants.NIC_MODE],
3375 ba924970 Dimitris Aragiorgis
       private.filled[constants.NIC_LINK], net)),
3376 ba924970 Dimitris Aragiorgis
      ]
3377 ba924970 Dimitris Aragiorgis
3378 e15a00dc Dimitris Aragiorgis
    if self.op.hotplug:
3379 e15a00dc Dimitris Aragiorgis
      msg = self._HotplugDevice(constants.HOTPLUG_ACTION_ADD,
3380 e15a00dc Dimitris Aragiorgis
                                constants.HOTPLUG_TARGET_NIC,
3381 e15a00dc Dimitris Aragiorgis
                                nobj, None, idx)
3382 e15a00dc Dimitris Aragiorgis
      changes.append(("nic.%d" % idx, msg))
3383 e15a00dc Dimitris Aragiorgis
3384 e15a00dc Dimitris Aragiorgis
    return (nobj, changes)
3385 22b7f6f8 Thomas Thrainer
3386 22b7f6f8 Thomas Thrainer
  def _ApplyNicMods(self, idx, nic, params, private):
3387 22b7f6f8 Thomas Thrainer
    """Modifies a network interface.
3388 22b7f6f8 Thomas Thrainer

3389 22b7f6f8 Thomas Thrainer
    """
3390 22b7f6f8 Thomas Thrainer
    changes = []
3391 22b7f6f8 Thomas Thrainer
3392 22b7f6f8 Thomas Thrainer
    for key in [constants.INIC_MAC, constants.INIC_IP, constants.INIC_NAME]:
3393 22b7f6f8 Thomas Thrainer
      if key in params:
3394 22b7f6f8 Thomas Thrainer
        changes.append(("nic.%s/%d" % (key, idx), params[key]))
3395 22b7f6f8 Thomas Thrainer
        setattr(nic, key, params[key])
3396 22b7f6f8 Thomas Thrainer
3397 22b7f6f8 Thomas Thrainer
    new_net = params.get(constants.INIC_NETWORK, nic.network)
3398 22b7f6f8 Thomas Thrainer
    new_net_uuid = self.cfg.LookupNetwork(new_net)
3399 22b7f6f8 Thomas Thrainer
    if new_net_uuid != nic.network:
3400 22b7f6f8 Thomas Thrainer
      changes.append(("nic.network/%d" % idx, new_net))
3401 22b7f6f8 Thomas Thrainer
      nic.network = new_net_uuid
3402 22b7f6f8 Thomas Thrainer
3403 22b7f6f8 Thomas Thrainer
    if private.filled:
3404 22b7f6f8 Thomas Thrainer
      nic.nicparams = private.filled
3405 22b7f6f8 Thomas Thrainer
3406 22b7f6f8 Thomas Thrainer
      for (key, val) in nic.nicparams.items():
3407 22b7f6f8 Thomas Thrainer
        changes.append(("nic.%s/%d" % (key, idx), val))
3408 22b7f6f8 Thomas Thrainer
3409 ba924970 Dimitris Aragiorgis
    if self.op.hotplug:
3410 e15a00dc Dimitris Aragiorgis
      msg = self._HotplugDevice(constants.HOTPLUG_ACTION_MODIFY,
3411 e15a00dc Dimitris Aragiorgis
                                constants.HOTPLUG_TARGET_NIC,
3412 e15a00dc Dimitris Aragiorgis
                                nic, None, idx)
3413 e15a00dc Dimitris Aragiorgis
      changes.append(("nic/%d" % idx, msg))
3414 ba924970 Dimitris Aragiorgis
3415 22b7f6f8 Thomas Thrainer
    return changes
3416 22b7f6f8 Thomas Thrainer
3417 ba924970 Dimitris Aragiorgis
  def _RemoveNic(self, idx, nic, _):
3418 ba924970 Dimitris Aragiorgis
    if self.op.hotplug:
3419 e15a00dc Dimitris Aragiorgis
      return self._HotplugDevice(constants.HOTPLUG_ACTION_REMOVE,
3420 e15a00dc Dimitris Aragiorgis
                                 constants.HOTPLUG_TARGET_NIC,
3421 e15a00dc Dimitris Aragiorgis
                                 nic, None, idx)
3422 ba924970 Dimitris Aragiorgis
3423 22b7f6f8 Thomas Thrainer
  def Exec(self, feedback_fn):
3424 22b7f6f8 Thomas Thrainer
    """Modifies an instance.
3425 22b7f6f8 Thomas Thrainer

3426 22b7f6f8 Thomas Thrainer
    All parameters take effect only at the next restart of the instance.
3427 22b7f6f8 Thomas Thrainer

3428 22b7f6f8 Thomas Thrainer
    """
3429 22b7f6f8 Thomas Thrainer
    # Process here the warnings from CheckPrereq, as we don't have a
3430 22b7f6f8 Thomas Thrainer
    # feedback_fn there.
3431 22b7f6f8 Thomas Thrainer
    # TODO: Replace with self.LogWarning
3432 22b7f6f8 Thomas Thrainer
    for warn in self.warn:
3433 22b7f6f8 Thomas Thrainer
      feedback_fn("WARNING: %s" % warn)
3434 22b7f6f8 Thomas Thrainer
3435 22b7f6f8 Thomas Thrainer
    assert ((self.op.disk_template is None) ^
3436 22b7f6f8 Thomas Thrainer
            bool(self.owned_locks(locking.LEVEL_NODE_RES))), \
3437 22b7f6f8 Thomas Thrainer
      "Not owning any node resource locks"
3438 22b7f6f8 Thomas Thrainer
3439 22b7f6f8 Thomas Thrainer
    result = []
3440 22b7f6f8 Thomas Thrainer
3441 22b7f6f8 Thomas Thrainer
    # New primary node
3442 1c3231aa Thomas Thrainer
    if self.op.pnode_uuid:
3443 d0d7d7cf Thomas Thrainer
      self.instance.primary_node = self.op.pnode_uuid
3444 22b7f6f8 Thomas Thrainer
3445 22b7f6f8 Thomas Thrainer
    # runtime memory
3446 22b7f6f8 Thomas Thrainer
    if self.op.runtime_mem:
3447 d0d7d7cf Thomas Thrainer
      rpcres = self.rpc.call_instance_balloon_memory(self.instance.primary_node,
3448 d0d7d7cf Thomas Thrainer
                                                     self.instance,
3449 22b7f6f8 Thomas Thrainer
                                                     self.op.runtime_mem)
3450 22b7f6f8 Thomas Thrainer
      rpcres.Raise("Cannot modify instance runtime memory")
3451 22b7f6f8 Thomas Thrainer
      result.append(("runtime_memory", self.op.runtime_mem))
3452 22b7f6f8 Thomas Thrainer
3453 22b7f6f8 Thomas Thrainer
    # Apply disk changes
3454 d0d7d7cf Thomas Thrainer
    _ApplyContainerMods("disk", self.instance.disks, result, self.diskmod,
3455 5eacbcae Thomas Thrainer
                        self._CreateNewDisk, self._ModifyDisk,
3456 922a9e65 Thomas Thrainer
                        self._RemoveDisk, post_add_fn=self._PostAddDisk)
3457 d0d7d7cf Thomas Thrainer
    _UpdateIvNames(0, self.instance.disks)
3458 22b7f6f8 Thomas Thrainer
3459 22b7f6f8 Thomas Thrainer
    if self.op.disk_template:
3460 22b7f6f8 Thomas Thrainer
      if __debug__:
3461 d0d7d7cf Thomas Thrainer
        check_nodes = set(self.instance.all_nodes)
3462 1c3231aa Thomas Thrainer
        if self.op.remote_node_uuid:
3463 1c3231aa Thomas Thrainer
          check_nodes.add(self.op.remote_node_uuid)
3464 22b7f6f8 Thomas Thrainer
        for level in [locking.LEVEL_NODE, locking.LEVEL_NODE_RES]:
3465 22b7f6f8 Thomas Thrainer
          owned = self.owned_locks(level)
3466 22b7f6f8 Thomas Thrainer
          assert not (check_nodes - owned), \
3467 22b7f6f8 Thomas Thrainer
            ("Not owning the correct locks, owning %r, expected at least %r" %
3468 22b7f6f8 Thomas Thrainer
             (owned, check_nodes))
3469 22b7f6f8 Thomas Thrainer
3470 d0d7d7cf Thomas Thrainer
      r_shut = ShutdownInstanceDisks(self, self.instance)
3471 22b7f6f8 Thomas Thrainer
      if not r_shut:
3472 22b7f6f8 Thomas Thrainer
        raise errors.OpExecError("Cannot shutdown instance disks, unable to"
3473 22b7f6f8 Thomas Thrainer
                                 " proceed with disk template conversion")
3474 d0d7d7cf Thomas Thrainer
      mode = (self.instance.disk_template, self.op.disk_template)
3475 22b7f6f8 Thomas Thrainer
      try:
3476 22b7f6f8 Thomas Thrainer
        self._DISK_CONVERSIONS[mode](self, feedback_fn)
3477 22b7f6f8 Thomas Thrainer
      except:
3478 da4a52a3 Thomas Thrainer
        self.cfg.ReleaseDRBDMinors(self.instance.uuid)
3479 22b7f6f8 Thomas Thrainer
        raise
3480 22b7f6f8 Thomas Thrainer
      result.append(("disk_template", self.op.disk_template))
3481 22b7f6f8 Thomas Thrainer
3482 d0d7d7cf Thomas Thrainer
      assert self.instance.disk_template == self.op.disk_template, \
3483 22b7f6f8 Thomas Thrainer
        ("Expected disk template '%s', found '%s'" %
3484 d0d7d7cf Thomas Thrainer
         (self.op.disk_template, self.instance.disk_template))
3485 22b7f6f8 Thomas Thrainer
3486 22b7f6f8 Thomas Thrainer
    # Release node and resource locks if there are any (they might already have
3487 22b7f6f8 Thomas Thrainer
    # been released during disk conversion)
3488 5eacbcae Thomas Thrainer
    ReleaseLocks(self, locking.LEVEL_NODE)
3489 5eacbcae Thomas Thrainer
    ReleaseLocks(self, locking.LEVEL_NODE_RES)
3490 22b7f6f8 Thomas Thrainer
3491 22b7f6f8 Thomas Thrainer
    # Apply NIC changes
3492 22b7f6f8 Thomas Thrainer
    if self._new_nics is not None:
3493 d0d7d7cf Thomas Thrainer
      self.instance.nics = self._new_nics
3494 22b7f6f8 Thomas Thrainer
      result.extend(self._nic_chgdesc)
3495 22b7f6f8 Thomas Thrainer
3496 22b7f6f8 Thomas Thrainer
    # hvparams changes
3497 22b7f6f8 Thomas Thrainer
    if self.op.hvparams:
3498 d0d7d7cf Thomas Thrainer
      self.instance.hvparams = self.hv_inst
3499 22b7f6f8 Thomas Thrainer
      for key, val in self.op.hvparams.iteritems():
3500 22b7f6f8 Thomas Thrainer
        result.append(("hv/%s" % key, val))
3501 22b7f6f8 Thomas Thrainer
3502 22b7f6f8 Thomas Thrainer
    # beparams changes
3503 22b7f6f8 Thomas Thrainer
    if self.op.beparams:
3504 d0d7d7cf Thomas Thrainer
      self.instance.beparams = self.be_inst
3505 22b7f6f8 Thomas Thrainer
      for key, val in self.op.beparams.iteritems():
3506 22b7f6f8 Thomas Thrainer
        result.append(("be/%s" % key, val))
3507 22b7f6f8 Thomas Thrainer
3508 22b7f6f8 Thomas Thrainer
    # OS change
3509 22b7f6f8 Thomas Thrainer
    if self.op.os_name:
3510 d0d7d7cf Thomas Thrainer
      self.instance.os = self.op.os_name
3511 22b7f6f8 Thomas Thrainer
3512 22b7f6f8 Thomas Thrainer
    # osparams changes
3513 22b7f6f8 Thomas Thrainer
    if self.op.osparams:
3514 d0d7d7cf Thomas Thrainer
      self.instance.osparams = self.os_inst
3515 22b7f6f8 Thomas Thrainer
      for key, val in self.op.osparams.iteritems():
3516 22b7f6f8 Thomas Thrainer
        result.append(("os/%s" % key, val))
3517 22b7f6f8 Thomas Thrainer
3518 22b7f6f8 Thomas Thrainer
    if self.op.offline is None:
3519 22b7f6f8 Thomas Thrainer
      # Ignore
3520 22b7f6f8 Thomas Thrainer
      pass
3521 22b7f6f8 Thomas Thrainer
    elif self.op.offline:
3522 22b7f6f8 Thomas Thrainer
      # Mark instance as offline
3523 da4a52a3 Thomas Thrainer
      self.cfg.MarkInstanceOffline(self.instance.uuid)
3524 22b7f6f8 Thomas Thrainer
      result.append(("admin_state", constants.ADMINST_OFFLINE))
3525 22b7f6f8 Thomas Thrainer
    else:
3526 22b7f6f8 Thomas Thrainer
      # Mark instance as online, but stopped
3527 da4a52a3 Thomas Thrainer
      self.cfg.MarkInstanceDown(self.instance.uuid)
3528 22b7f6f8 Thomas Thrainer
      result.append(("admin_state", constants.ADMINST_DOWN))
3529 22b7f6f8 Thomas Thrainer
3530 d0d7d7cf Thomas Thrainer
    self.cfg.Update(self.instance, feedback_fn, self.proc.GetECId())
3531 22b7f6f8 Thomas Thrainer
3532 22b7f6f8 Thomas Thrainer
    assert not (self.owned_locks(locking.LEVEL_NODE_RES) or
3533 22b7f6f8 Thomas Thrainer
                self.owned_locks(locking.LEVEL_NODE)), \
3534 22b7f6f8 Thomas Thrainer
      "All node locks should have been released by now"
3535 22b7f6f8 Thomas Thrainer
3536 22b7f6f8 Thomas Thrainer
    return result
3537 22b7f6f8 Thomas Thrainer
3538 22b7f6f8 Thomas Thrainer
  _DISK_CONVERSIONS = {
3539 22b7f6f8 Thomas Thrainer
    (constants.DT_PLAIN, constants.DT_DRBD8): _ConvertPlainToDrbd,
3540 22b7f6f8 Thomas Thrainer
    (constants.DT_DRBD8, constants.DT_PLAIN): _ConvertDrbdToPlain,
3541 22b7f6f8 Thomas Thrainer
    }
3542 22b7f6f8 Thomas Thrainer
3543 22b7f6f8 Thomas Thrainer
3544 22b7f6f8 Thomas Thrainer
class LUInstanceChangeGroup(LogicalUnit):
3545 22b7f6f8 Thomas Thrainer
  HPATH = "instance-change-group"
3546 22b7f6f8 Thomas Thrainer
  HTYPE = constants.HTYPE_INSTANCE
3547 22b7f6f8 Thomas Thrainer
  REQ_BGL = False
3548 22b7f6f8 Thomas Thrainer
3549 22b7f6f8 Thomas Thrainer
  def ExpandNames(self):
3550 5eacbcae Thomas Thrainer
    self.share_locks = ShareAll()
3551 22b7f6f8 Thomas Thrainer
3552 22b7f6f8 Thomas Thrainer
    self.needed_locks = {
3553 22b7f6f8 Thomas Thrainer
      locking.LEVEL_NODEGROUP: [],
3554 22b7f6f8 Thomas Thrainer
      locking.LEVEL_NODE: [],
3555 22b7f6f8 Thomas Thrainer
      locking.LEVEL_NODE_ALLOC: locking.ALL_SET,
3556 22b7f6f8 Thomas Thrainer
      }
3557 22b7f6f8 Thomas Thrainer
3558 22b7f6f8 Thomas Thrainer
    self._ExpandAndLockInstance()
3559 22b7f6f8 Thomas Thrainer
3560 22b7f6f8 Thomas Thrainer
    if self.op.target_groups:
3561 22b7f6f8 Thomas Thrainer
      self.req_target_uuids = map(self.cfg.LookupNodeGroup,
3562 22b7f6f8 Thomas Thrainer
                                  self.op.target_groups)
3563 22b7f6f8 Thomas Thrainer
    else:
3564 22b7f6f8 Thomas Thrainer
      self.req_target_uuids = None
3565 22b7f6f8 Thomas Thrainer
3566 5eacbcae Thomas Thrainer
    self.op.iallocator = GetDefaultIAllocator(self.cfg, self.op.iallocator)
3567 22b7f6f8 Thomas Thrainer
3568 22b7f6f8 Thomas Thrainer
  def DeclareLocks(self, level):
3569 22b7f6f8 Thomas Thrainer
    if level == locking.LEVEL_NODEGROUP:
3570 22b7f6f8 Thomas Thrainer
      assert not self.needed_locks[locking.LEVEL_NODEGROUP]
3571 22b7f6f8 Thomas Thrainer
3572 22b7f6f8 Thomas Thrainer
      if self.req_target_uuids:
3573 22b7f6f8 Thomas Thrainer
        lock_groups = set(self.req_target_uuids)
3574 22b7f6f8 Thomas Thrainer
3575 22b7f6f8 Thomas Thrainer
        # Lock all groups used by instance optimistically; this requires going
3576 22b7f6f8 Thomas Thrainer
        # via the node before it's locked, requiring verification later on
3577 da4a52a3 Thomas Thrainer
        instance_groups = self.cfg.GetInstanceNodeGroups(self.op.instance_uuid)
3578 22b7f6f8 Thomas Thrainer
        lock_groups.update(instance_groups)
3579 22b7f6f8 Thomas Thrainer
      else:
3580 22b7f6f8 Thomas Thrainer
        # No target groups, need to lock all of them
3581 22b7f6f8 Thomas Thrainer
        lock_groups = locking.ALL_SET
3582 22b7f6f8 Thomas Thrainer
3583 22b7f6f8 Thomas Thrainer
      self.needed_locks[locking.LEVEL_NODEGROUP] = lock_groups
3584 22b7f6f8 Thomas Thrainer
3585 22b7f6f8 Thomas Thrainer
    elif level == locking.LEVEL_NODE:
3586 22b7f6f8 Thomas Thrainer
      if self.req_target_uuids:
3587 22b7f6f8 Thomas Thrainer
        # Lock all nodes used by instances
3588 22b7f6f8 Thomas Thrainer
        self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_APPEND
3589 22b7f6f8 Thomas Thrainer
        self._LockInstancesNodes()
3590 22b7f6f8 Thomas Thrainer
3591 22b7f6f8 Thomas Thrainer
        # Lock all nodes in all potential target groups
3592 22b7f6f8 Thomas Thrainer
        lock_groups = (frozenset(self.owned_locks(locking.LEVEL_NODEGROUP)) -
3593 da4a52a3 Thomas Thrainer
                       self.cfg.GetInstanceNodeGroups(self.op.instance_uuid))
3594 1c3231aa Thomas Thrainer
        member_nodes = [node_uuid
3595 22b7f6f8 Thomas Thrainer
                        for group in lock_groups
3596 1c3231aa Thomas Thrainer
                        for node_uuid in self.cfg.GetNodeGroup(group).members]
3597 22b7f6f8 Thomas Thrainer
        self.needed_locks[locking.LEVEL_NODE].extend(member_nodes)
3598 22b7f6f8 Thomas Thrainer
      else:
3599 22b7f6f8 Thomas Thrainer
        # Lock all nodes as all groups are potential targets
3600 22b7f6f8 Thomas Thrainer
        self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
3601 22b7f6f8 Thomas Thrainer
3602 22b7f6f8 Thomas Thrainer
  def CheckPrereq(self):
3603 da4a52a3 Thomas Thrainer
    owned_instance_names = frozenset(self.owned_locks(locking.LEVEL_INSTANCE))
3604 22b7f6f8 Thomas Thrainer
    owned_groups = frozenset(self.owned_locks(locking.LEVEL_NODEGROUP))
3605 22b7f6f8 Thomas Thrainer
    owned_nodes = frozenset(self.owned_locks(locking.LEVEL_NODE))
3606 22b7f6f8 Thomas Thrainer
3607 22b7f6f8 Thomas Thrainer
    assert (self.req_target_uuids is None or
3608 22b7f6f8 Thomas Thrainer
            owned_groups.issuperset(self.req_target_uuids))
3609 da4a52a3 Thomas Thrainer
    assert owned_instance_names == set([self.op.instance_name])
3610 22b7f6f8 Thomas Thrainer
3611 22b7f6f8 Thomas Thrainer
    # Get instance information
3612 da4a52a3 Thomas Thrainer
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_uuid)
3613 22b7f6f8 Thomas Thrainer
3614 22b7f6f8 Thomas Thrainer
    # Check if node groups for locked instance are still correct
3615 22b7f6f8 Thomas Thrainer
    assert owned_nodes.issuperset(self.instance.all_nodes), \
3616 22b7f6f8 Thomas Thrainer
      ("Instance %s's nodes changed while we kept the lock" %
3617 22b7f6f8 Thomas Thrainer
       self.op.instance_name)
3618 22b7f6f8 Thomas Thrainer
3619 da4a52a3 Thomas Thrainer
    inst_groups = CheckInstanceNodeGroups(self.cfg, self.op.instance_uuid,
3620 5eacbcae Thomas Thrainer
                                          owned_groups)
3621 22b7f6f8 Thomas Thrainer
3622 22b7f6f8 Thomas Thrainer
    if self.req_target_uuids:
3623 22b7f6f8 Thomas Thrainer
      # User requested specific target groups
3624 22b7f6f8 Thomas Thrainer
      self.target_uuids = frozenset(self.req_target_uuids)
3625 22b7f6f8 Thomas Thrainer
    else:
3626 22b7f6f8 Thomas Thrainer
      # All groups except those used by the instance are potential targets
3627 22b7f6f8 Thomas Thrainer
      self.target_uuids = owned_groups - inst_groups
3628 22b7f6f8 Thomas Thrainer
3629 22b7f6f8 Thomas Thrainer
    conflicting_groups = self.target_uuids & inst_groups
3630 22b7f6f8 Thomas Thrainer
    if conflicting_groups:
3631 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Can't use group(s) '%s' as targets, they are"
3632 22b7f6f8 Thomas Thrainer
                                 " used by the instance '%s'" %
3633 22b7f6f8 Thomas Thrainer
                                 (utils.CommaJoin(conflicting_groups),
3634 22b7f6f8 Thomas Thrainer
                                  self.op.instance_name),
3635 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_INVAL)
3636 22b7f6f8 Thomas Thrainer
3637 22b7f6f8 Thomas Thrainer
    if not self.target_uuids:
3638 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("There are no possible target groups",
3639 22b7f6f8 Thomas Thrainer
                                 errors.ECODE_INVAL)
3640 22b7f6f8 Thomas Thrainer
3641 22b7f6f8 Thomas Thrainer
  def BuildHooksEnv(self):
3642 22b7f6f8 Thomas Thrainer
    """Build hooks env.
3643 22b7f6f8 Thomas Thrainer

3644 22b7f6f8 Thomas Thrainer
    """
3645 22b7f6f8 Thomas Thrainer
    assert self.target_uuids
3646 22b7f6f8 Thomas Thrainer
3647 22b7f6f8 Thomas Thrainer
    env = {
3648 22b7f6f8 Thomas Thrainer
      "TARGET_GROUPS": " ".join(self.target_uuids),
3649 22b7f6f8 Thomas Thrainer
      }
3650 22b7f6f8 Thomas Thrainer
3651 5eacbcae Thomas Thrainer
    env.update(BuildInstanceHookEnvByObject(self, self.instance))
3652 22b7f6f8 Thomas Thrainer
3653 22b7f6f8 Thomas Thrainer
    return env
3654 22b7f6f8 Thomas Thrainer
3655 22b7f6f8 Thomas Thrainer
  def BuildHooksNodes(self):
3656 22b7f6f8 Thomas Thrainer
    """Build hooks nodes.
3657 22b7f6f8 Thomas Thrainer

3658 22b7f6f8 Thomas Thrainer
    """
3659 22b7f6f8 Thomas Thrainer
    mn = self.cfg.GetMasterNode()
3660 22b7f6f8 Thomas Thrainer
    return ([mn], [mn])
3661 22b7f6f8 Thomas Thrainer
3662 22b7f6f8 Thomas Thrainer
  def Exec(self, feedback_fn):
3663 22b7f6f8 Thomas Thrainer
    instances = list(self.owned_locks(locking.LEVEL_INSTANCE))
3664 22b7f6f8 Thomas Thrainer
3665 22b7f6f8 Thomas Thrainer
    assert instances == [self.op.instance_name], "Instance not locked"
3666 22b7f6f8 Thomas Thrainer
3667 22b7f6f8 Thomas Thrainer
    req = iallocator.IAReqGroupChange(instances=instances,
3668 22b7f6f8 Thomas Thrainer
                                      target_groups=list(self.target_uuids))
3669 22b7f6f8 Thomas Thrainer
    ial = iallocator.IAllocator(self.cfg, self.rpc, req)
3670 22b7f6f8 Thomas Thrainer
3671 22b7f6f8 Thomas Thrainer
    ial.Run(self.op.iallocator)
3672 22b7f6f8 Thomas Thrainer
3673 22b7f6f8 Thomas Thrainer
    if not ial.success:
3674 22b7f6f8 Thomas Thrainer
      raise errors.OpPrereqError("Can't compute solution for changing group of"
3675 22b7f6f8 Thomas Thrainer
                                 " instance '%s' using iallocator '%s': %s" %
3676 22b7f6f8 Thomas Thrainer
                                 (self.op.instance_name, self.op.iallocator,
3677 22b7f6f8 Thomas Thrainer
                                  ial.info), errors.ECODE_NORES)
3678 22b7f6f8 Thomas Thrainer
3679 5eacbcae Thomas Thrainer
    jobs = LoadNodeEvacResult(self, ial.result, self.op.early_release, False)
3680 22b7f6f8 Thomas Thrainer
3681 22b7f6f8 Thomas Thrainer
    self.LogInfo("Iallocator returned %s job(s) for changing group of"
3682 22b7f6f8 Thomas Thrainer
                 " instance '%s'", len(jobs), self.op.instance_name)
3683 22b7f6f8 Thomas Thrainer
3684 22b7f6f8 Thomas Thrainer
    return ResultWithJobs(jobs)